tkms 0.13.0-rc.1 → 0.13.0-rc.2

package/kms_lib.d.ts

@@ -1,37 +1,41 @@
 /* tslint:disable */
 /* eslint-disable */
+export function get_client_address(client: Client): string;
+export function ml_kem_pke_sk_len(): number;
+export function ml_kem_pke_sk_to_u8vec(sk: PrivateEncKeyMlKem512): Uint8Array;
 /**
- * This function is *not* used by relayer-sdk because the encryption
- * happens on the KMS side. It's just here for completeness and tests.
- */
-export function ml_kem_pke_encrypt(msg: Uint8Array, their_pk: PublicEncKeyMlKem512): Uint8Array;
-/**
- * Instantiate a new client.
+ * Process the user_decryption response from Rust objects.
+ * Consider using [process_user_decryption_resp_from_js]
+ * when using the JS API.
+ * The result is a byte array representing a plaintext of any length.
  *
- * * `server_addrs` - a list of KMS server ID with EIP-55 addresses,
- * the elements in the list can be created using [new_server_id_addr].
+ * * `client` - client that wants to perform user_decryption.
  *
- * * `client_address_hex` - the client (wallet) address in hex,
- * must be prefixed with "0x".
+ * * `request` - the initial user_decryption request.
+ * Must be given if `verify` is true.
  *
- * * `fhe_parameter` - the parameter choice, which can be either `"test"` or `"default"`.
- * The "default" parameter choice is selected if no matching string is found.
+ * * `eip712_domain` - the EIP-712 domain.
+ * Must be given if `verify` is true.
+ *
+ * * `agg_resp` - the vector of user_decryption responses.
+ *
+ * * `enc_pk` - The ephemeral public key.
+ *
+ * * `enc_sk` - The ephemeral secret key.
+ *
+ * * `verify` - Whether to perform signature verification for the response.
+ * It is insecure if `verify = false`!
  */
-export function new_client(server_addrs: ServerIdAddr[], client_address_hex: string, fhe_parameter: string): Client;
-export function ml_kem_pke_keygen(): PrivateEncKeyMlKem512;
+export function process_user_decryption_resp(client: Client, request: ParsedUserDecryptionRequest | null | undefined, eip712_domain: Eip712DomainMsg | null | undefined, agg_resp: UserDecryptionResponse[], enc_pk: PublicEncKeyMlKem512, enc_sk: PrivateEncKeyMlKem512, verify: boolean): TypedPlaintext[];
 export function ml_kem_pke_get_pk(sk: PrivateEncKeyMlKem512): PublicEncKeyMlKem512;
-export function u8vec_to_private_sig_key(v: Uint8Array): PrivateSigKey;
-export function public_sig_key_to_u8vec(pk: PublicSigKey): Uint8Array;
+export function u8vec_to_ml_kem_pke_pk(v: Uint8Array): PublicEncKeyMlKem512;
 /**
- * This function is *not* used by relayer-sdk because the decryption
- * is handled by [process_user_decryption_resp].
- * It's just here for completeness and tests.
+ * Create a new [ServerIdAddr] structure that holds an ID and an address
+ * which must be a valid EIP-55 address, notably prefixed with "0x".
  */
-export function ml_kem_pke_decrypt(ct: Uint8Array, my_sk: PrivateEncKeyMlKem512): Uint8Array;
-export function u8vec_to_public_sig_key(v: Uint8Array): PublicSigKey;
-export function u8vec_to_ml_kem_pke_pk(v: Uint8Array): PublicEncKeyMlKem512;
-export function ml_kem_pke_pk_len(): number;
-export function get_server_addrs(client: Client): ServerIdAddr[];
+export function new_server_id_addr(id: number, addr: string): ServerIdAddr;
+export function u8vec_to_ml_kem_pke_sk(v: Uint8Array): PrivateEncKeyMlKem512;
+export function ml_kem_pke_pk_to_u8vec(pk: PublicEncKeyMlKem512): Uint8Array;
 /**
  * Process the user_decryption response from JavaScript objects.
  * The returned result is a byte array representing a plaintext of any length,
@@ -94,42 +98,38 @@ export function get_server_addrs(client: Client): ServerIdAddr[];
  * It is insecure if `verify = false`!
  */
 export function process_user_decryption_resp_from_js(client: Client, request: any, eip712_domain: any, agg_resp: any, enc_pk: PublicEncKeyMlKem512, enc_sk: PrivateEncKeyMlKem512, verify: boolean): TypedPlaintext[];
+export function ml_kem_pke_pk_len(): number;
+export function get_client_secret_key(client: Client): PrivateSigKey | undefined;
+export function get_server_addrs(client: Client): ServerIdAddr[];
+export function u8vec_to_public_sig_key(v: Uint8Array): PublicSigKey;
+export function u8vec_to_private_sig_key(v: Uint8Array): PrivateSigKey;
+export function ml_kem_pke_keygen(): PrivateEncKeyMlKem512;
 /**
- * Create a new [ServerIdAddr] structure that holds an ID and an address
- * which must be a valid EIP-55 address, notably prefixed with "0x".
+ * This function is *not* used by relayer-sdk because the encryption
+ * happens on the KMS side. It's just here for completeness and tests.
  */
-export function new_server_id_addr(id: number, addr: string): ServerIdAddr;
-export function ml_kem_pke_sk_len(): number;
-export function private_sig_key_to_u8vec(sk: PrivateSigKey): Uint8Array;
-export function u8vec_to_ml_kem_pke_sk(v: Uint8Array): PrivateEncKeyMlKem512;
-export function get_client_address(client: Client): string;
+export function ml_kem_pke_encrypt(msg: Uint8Array, their_pk: PublicEncKeyMlKem512): Uint8Array;
 /**
- * Process the user_decryption response from Rust objects.
- * Consider using [process_user_decryption_resp_from_js]
- * when using the JS API.
- * The result is a byte array representing a plaintext of any length.
- *
- * * `client` - client that wants to perform user_decryption.
- *
- * * `request` - the initial user_decryption request.
- * Must be given if `verify` is true.
- *
- * * `eip712_domain` - the EIP-712 domain.
- * Must be given if `verify` is true.
- *
- * * `agg_resp` - the vector of user_decryption responses.
+ * Instantiate a new client.
  *
- * * `enc_pk` - The ephemeral public key.
+ * * `server_addrs` - a list of KMS server ID with EIP-55 addresses,
+ * the elements in the list can be created using [new_server_id_addr].
  *
- * * `enc_sk` - The ephemeral secret key.
+ * * `client_address_hex` - the client (wallet) address in hex,
+ * must be prefixed with "0x".
  *
- * * `verify` - Whether to perform signature verification for the response.
- * It is insecure if `verify = false`!
+ * * `fhe_parameter` - the parameter choice, which can be either `"test"` or `"default"`.
+ * The "default" parameter choice is selected if no matching string is found.
  */
-export function process_user_decryption_resp(client: Client, request: ParsedUserDecryptionRequest | null | undefined, eip712_domain: Eip712DomainMsg | null | undefined, agg_resp: UserDecryptionResponse[], enc_pk: PublicEncKeyMlKem512, enc_sk: PrivateEncKeyMlKem512, verify: boolean): TypedPlaintext[];
-export function get_client_secret_key(client: Client): PrivateSigKey | undefined;
-export function ml_kem_pke_pk_to_u8vec(pk: PublicEncKeyMlKem512): Uint8Array;
-export function ml_kem_pke_sk_to_u8vec(sk: PrivateEncKeyMlKem512): Uint8Array;
+export function new_client(server_addrs: ServerIdAddr[], client_address_hex: string, fhe_parameter: string): Client;
+export function public_sig_key_to_u8vec(pk: PublicSigKey): Uint8Array;
+/**
+ * This function is *not* used by relayer-sdk because the decryption
+ * is handled by [process_user_decryption_resp].
+ * It's just here for completeness and tests.
+ */
+export function ml_kem_pke_decrypt(ct: Uint8Array, my_sk: PrivateEncKeyMlKem512): Uint8Array;
+export function private_sig_key_to_u8vec(sk: PrivateSigKey): Uint8Array;
 export class CiphertextHandle {
   private constructor();
   free(): void;
@@ -391,8 +391,6 @@ export type InitInput = RequestInfo | URL | Response | BufferSource | WebAssembl
 export interface InitOutput {
   readonly memory: WebAssembly.Memory;
   readonly __wbg_client_free: (a: number, b: number) => void;
-  readonly __wbg_ciphertexthandle_free: (a: number, b: number) => void;
-  readonly __wbg_parseduserdecryptionrequest_free: (a: number, b: number) => void;
   readonly __wbg_privateenckeymlkem512_free: (a: number, b: number) => void;
   readonly __wbg_publicenckeymlkem512_free: (a: number, b: number) => void;
   readonly __wbg_serveridaddr_free: (a: number, b: number) => void;
@@ -417,6 +415,8 @@ export interface InitOutput {
   readonly u8vec_to_ml_kem_pke_sk: (a: number, b: number) => [number, number, number];
   readonly u8vec_to_private_sig_key: (a: number, b: number) => [number, number, number];
   readonly u8vec_to_public_sig_key: (a: number, b: number) => [number, number, number];
+  readonly __wbg_ciphertexthandle_free: (a: number, b: number) => void;
+  readonly __wbg_parseduserdecryptionrequest_free: (a: number, b: number) => void;
   readonly __wbg_privatesigkey_free: (a: number, b: number) => void;
   readonly __wbg_publicsigkey_free: (a: number, b: number) => void;
   readonly __wbg_eip712domainmsg_free: (a: number, b: number) => void;
@@ -466,10 +466,10 @@ export interface InitOutput {
   readonly __wbg_userdecryptionrequest_free: (a: number, b: number) => void;
   readonly __wbg_userdecryptionresponse_free: (a: number, b: number) => void;
   readonly __wbg_userdecryptionresponsepayload_free: (a: number, b: number) => void;
-  readonly __wbg_get_requestid_request_id: (a: number) => [number, number];
-  readonly __wbg_get_userdecryptionrequest_client_address: (a: number) => [number, number];
   readonly __wbg_set_typedsigncryptedciphertext_fhe_type: (a: number, b: number) => void;
   readonly __wbg_set_typedsigncryptedciphertext_packing_factor: (a: number, b: number) => void;
+  readonly __wbg_get_requestid_request_id: (a: number) => [number, number];
+  readonly __wbg_get_userdecryptionrequest_client_address: (a: number) => [number, number];
   readonly __wbg_set_requestid_request_id: (a: number, b: number, c: number) => void;
   readonly __wbg_set_typedciphertext_ciphertext: (a: number, b: number, c: number) => void;
   readonly __wbg_set_typedciphertext_external_handle: (a: number, b: number, c: number) => void;

package/kms_lib.js

@@ -167,38 +167,60 @@ function debugString(val) {
     return className;
 }
 
-function passArray8ToWasm0(arg, malloc) {
-    const ptr = malloc(arg.length * 1, 1) >>> 0;
-    getUint8ArrayMemory0().set(arg, ptr / 1);
-    WASM_VECTOR_LEN = arg.length;
-    return ptr;
-}
-
 function _assertClass(instance, klass) {
     if (!(instance instanceof klass)) {
         throw new Error(`expected instance of ${klass.name}`);
     }
 }
+/**
+ * @param {Client} client
+ * @returns {string}
+ */
+export function get_client_address(client) {
+    let deferred1_0;
+    let deferred1_1;
+    try {
+        _assertClass(client, Client);
+        const ret = wasm.get_client_address(client.__wbg_ptr);
+        deferred1_0 = ret[0];
+        deferred1_1 = ret[1];
+        return getStringFromWasm0(ret[0], ret[1]);
+    } finally {
+        wasm.__wbindgen_free(deferred1_0, deferred1_1, 1);
+    }
+}
+
+/**
+ * @returns {number}
+ */
+export function ml_kem_pke_sk_len() {
+    const ret = wasm.ml_kem_pke_sk_len();
+    return ret >>> 0;
+}
+
+function takeFromExternrefTable0(idx) {
+    const value = wasm.__wbindgen_export_4.get(idx);
+    wasm.__externref_table_dealloc(idx);
+    return value;
+}
 
 function getArrayU8FromWasm0(ptr, len) {
     ptr = ptr >>> 0;
     return getUint8ArrayMemory0().subarray(ptr / 1, ptr / 1 + len);
 }
 /**
- * This function is *not* used by relayer-sdk because the encryption
- * happens on the KMS side. It's just here for completeness and tests.
- * @param {Uint8Array} msg
- * @param {PublicEncKeyMlKem512} their_pk
+ * @param {PrivateEncKeyMlKem512} sk
  * @returns {Uint8Array}
  */
-export function ml_kem_pke_encrypt(msg, their_pk) {
-    const ptr0 = passArray8ToWasm0(msg, wasm.__wbindgen_malloc);
-    const len0 = WASM_VECTOR_LEN;
-    _assertClass(their_pk, PublicEncKeyMlKem512);
-    const ret = wasm.ml_kem_pke_encrypt(ptr0, len0, their_pk.__wbg_ptr);
-    var v2 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
+export function ml_kem_pke_sk_to_u8vec(sk) {
+    _assertClass(sk, PrivateEncKeyMlKem512);
+    const ret = wasm.ml_kem_pke_sk_to_u8vec(sk.__wbg_ptr);
+    if (ret[3]) {
+        throw takeFromExternrefTable0(ret[2]);
+    }
+    var v1 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
     wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
-    return v2;
+    return v1;
 }
 
 function passArrayJsValueToWasm0(array, malloc) {
@@ -211,47 +233,70 @@ function passArrayJsValueToWasm0(array, malloc) {
     return ptr;
 }
 
-function takeFromExternrefTable0(idx) {
-    const value = wasm.__wbindgen_export_4.get(idx);
-    wasm.__externref_table_dealloc(idx);
-    return value;
+function getArrayJsValueFromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    const mem = getDataViewMemory0();
+    const result = [];
+    for (let i = ptr; i < ptr + 4 * len; i += 4) {
+        result.push(wasm.__wbindgen_export_4.get(mem.getUint32(i, true)));
+    }
+    wasm.__externref_drop_slice(ptr, len);
+    return result;
 }
 /**
- * Instantiate a new client.
+ * Process the user_decryption response from Rust objects.
+ * Consider using [process_user_decryption_resp_from_js]
+ * when using the JS API.
+ * The result is a byte array representing a plaintext of any length.
  *
- * * `server_addrs` - a list of KMS server ID with EIP-55 addresses,
- * the elements in the list can be created using [new_server_id_addr].
+ * * `client` - client that wants to perform user_decryption.
  *
- * * `client_address_hex` - the client (wallet) address in hex,
- * must be prefixed with "0x".
+ * * `request` - the initial user_decryption request.
+ * Must be given if `verify` is true.
  *
- * * `fhe_parameter` - the parameter choice, which can be either `"test"` or `"default"`.
- * The "default" parameter choice is selected if no matching string is found.
- * @param {ServerIdAddr[]} server_addrs
- * @param {string} client_address_hex
- * @param {string} fhe_parameter
- * @returns {Client}
+ * * `eip712_domain` - the EIP-712 domain.
+ * Must be given if `verify` is true.
+ *
+ * * `agg_resp` - the vector of user_decryption responses.
+ *
+ * * `enc_pk` - The ephemeral public key.
+ *
+ * * `enc_sk` - The ephemeral secret key.
+ *
+ * * `verify` - Whether to perform signature verification for the response.
+ * It is insecure if `verify = false`!
+ * @param {Client} client
+ * @param {ParsedUserDecryptionRequest | null | undefined} request
+ * @param {Eip712DomainMsg | null | undefined} eip712_domain
+ * @param {UserDecryptionResponse[]} agg_resp
+ * @param {PublicEncKeyMlKem512} enc_pk
+ * @param {PrivateEncKeyMlKem512} enc_sk
+ * @param {boolean} verify
+ * @returns {TypedPlaintext[]}
  */
-export function new_client(server_addrs, client_address_hex, fhe_parameter) {
-    const ptr0 = passArrayJsValueToWasm0(server_addrs, wasm.__wbindgen_malloc);
-    const len0 = WASM_VECTOR_LEN;
-    const ptr1 = passStringToWasm0(client_address_hex, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
-    const len1 = WASM_VECTOR_LEN;
-    const ptr2 = passStringToWasm0(fhe_parameter, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+export function process_user_decryption_resp(client, request, eip712_domain, agg_resp, enc_pk, enc_sk, verify) {
+    _assertClass(client, Client);
+    let ptr0 = 0;
+    if (!isLikeNone(request)) {
+        _assertClass(request, ParsedUserDecryptionRequest);
+        ptr0 = request.__destroy_into_raw();
+    }
+    let ptr1 = 0;
+    if (!isLikeNone(eip712_domain)) {
+        _assertClass(eip712_domain, Eip712DomainMsg);
+        ptr1 = eip712_domain.__destroy_into_raw();
+    }
+    const ptr2 = passArrayJsValueToWasm0(agg_resp, wasm.__wbindgen_malloc);
     const len2 = WASM_VECTOR_LEN;
-    const ret = wasm.new_client(ptr0, len0, ptr1, len1, ptr2, len2);
-    if (ret[2]) {
-        throw takeFromExternrefTable0(ret[1]);
+    _assertClass(enc_pk, PublicEncKeyMlKem512);
+    _assertClass(enc_sk, PrivateEncKeyMlKem512);
+    const ret = wasm.process_user_decryption_resp(client.__wbg_ptr, ptr0, ptr1, ptr2, len2, enc_pk.__wbg_ptr, enc_sk.__wbg_ptr, verify);
+    if (ret[3]) {
+        throw takeFromExternrefTable0(ret[2]);
     }
-    return Client.__wrap(ret[0]);
-}
-
-/**
- * @returns {PrivateEncKeyMlKem512}
- */
-export function ml_kem_pke_keygen() {
-    const ret = wasm.ml_kem_pke_keygen();
-    return PrivateEncKeyMlKem512.__wrap(ret);
+    var v4 = getArrayJsValueFromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 4, 4);
+    return v4;
 }
 
 /**
@@ -264,105 +309,69 @@ export function ml_kem_pke_get_pk(sk) {
     return PublicEncKeyMlKem512.__wrap(ret);
 }
 
+function passArray8ToWasm0(arg, malloc) {
+    const ptr = malloc(arg.length * 1, 1) >>> 0;
+    getUint8ArrayMemory0().set(arg, ptr / 1);
+    WASM_VECTOR_LEN = arg.length;
+    return ptr;
+}
 /**
  * @param {Uint8Array} v
- * @returns {PrivateSigKey}
+ * @returns {PublicEncKeyMlKem512}
  */
-export function u8vec_to_private_sig_key(v) {
+export function u8vec_to_ml_kem_pke_pk(v) {
     const ptr0 = passArray8ToWasm0(v, wasm.__wbindgen_malloc);
     const len0 = WASM_VECTOR_LEN;
-    const ret = wasm.u8vec_to_private_sig_key(ptr0, len0);
+    const ret = wasm.u8vec_to_ml_kem_pke_pk(ptr0, len0);
     if (ret[2]) {
         throw takeFromExternrefTable0(ret[1]);
     }
-    return PrivateSigKey.__wrap(ret[0]);
-}
-
-/**
- * @param {PublicSigKey} pk
- * @returns {Uint8Array}
- */
-export function public_sig_key_to_u8vec(pk) {
-    _assertClass(pk, PublicSigKey);
-    const ret = wasm.public_sig_key_to_u8vec(pk.__wbg_ptr);
-    var v1 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
-    wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
-    return v1;
-}
-
-/**
- * This function is *not* used by relayer-sdk because the decryption
- * is handled by [process_user_decryption_resp].
- * It's just here for completeness and tests.
- * @param {Uint8Array} ct
- * @param {PrivateEncKeyMlKem512} my_sk
- * @returns {Uint8Array}
- */
-export function ml_kem_pke_decrypt(ct, my_sk) {
-    const ptr0 = passArray8ToWasm0(ct, wasm.__wbindgen_malloc);
-    const len0 = WASM_VECTOR_LEN;
-    _assertClass(my_sk, PrivateEncKeyMlKem512);
-    const ret = wasm.ml_kem_pke_decrypt(ptr0, len0, my_sk.__wbg_ptr);
-    var v2 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
-    wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
-    return v2;
+    return PublicEncKeyMlKem512.__wrap(ret[0]);
 }
 
 /**
- * @param {Uint8Array} v
- * @returns {PublicSigKey}
+ * Create a new [ServerIdAddr] structure that holds an ID and an address
+ * which must be a valid EIP-55 address, notably prefixed with "0x".
+ * @param {number} id
+ * @param {string} addr
+ * @returns {ServerIdAddr}
  */
-export function u8vec_to_public_sig_key(v) {
-    const ptr0 = passArray8ToWasm0(v, wasm.__wbindgen_malloc);
+export function new_server_id_addr(id, addr) {
+    const ptr0 = passStringToWasm0(addr, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
     const len0 = WASM_VECTOR_LEN;
-    const ret = wasm.u8vec_to_public_sig_key(ptr0, len0);
+    const ret = wasm.new_server_id_addr(id, ptr0, len0);
     if (ret[2]) {
         throw takeFromExternrefTable0(ret[1]);
     }
-    return PublicSigKey.__wrap(ret[0]);
+    return ServerIdAddr.__wrap(ret[0]);
 }
 
 /**
  * @param {Uint8Array} v
- * @returns {PublicEncKeyMlKem512}
+ * @returns {PrivateEncKeyMlKem512}
  */
-export function u8vec_to_ml_kem_pke_pk(v) {
+export function u8vec_to_ml_kem_pke_sk(v) {
     const ptr0 = passArray8ToWasm0(v, wasm.__wbindgen_malloc);
     const len0 = WASM_VECTOR_LEN;
-    const ret = wasm.u8vec_to_ml_kem_pke_pk(ptr0, len0);
+    const ret = wasm.u8vec_to_ml_kem_pke_sk(ptr0, len0);
     if (ret[2]) {
         throw takeFromExternrefTable0(ret[1]);
     }
-    return PublicEncKeyMlKem512.__wrap(ret[0]);
+    return PrivateEncKeyMlKem512.__wrap(ret[0]);
 }
 
 /**
- * @returns {number}
+ * @param {PublicEncKeyMlKem512} pk
+ * @returns {Uint8Array}
  */
-export function ml_kem_pke_pk_len() {
-    const ret = wasm.ml_kem_pke_pk_len();
-    return ret >>> 0;
-}
-
-function getArrayJsValueFromWasm0(ptr, len) {
-    ptr = ptr >>> 0;
-    const mem = getDataViewMemory0();
-    const result = [];
-    for (let i = ptr; i < ptr + 4 * len; i += 4) {
-        result.push(wasm.__wbindgen_export_4.get(mem.getUint32(i, true)));
+export function ml_kem_pke_pk_to_u8vec(pk) {
+    _assertClass(pk, PublicEncKeyMlKem512);
+    const ret = wasm.ml_kem_pke_pk_to_u8vec(pk.__wbg_ptr);
+    if (ret[3]) {
+        throw takeFromExternrefTable0(ret[2]);
     }
-    wasm.__externref_drop_slice(ptr, len);
-    return result;
-}
-/**
- * @param {Client} client
- * @returns {ServerIdAddr[]}
- */
-export function get_server_addrs(client) {
-    _assertClass(client, Client);
-    const ret = wasm.get_server_addrs(client.__wbg_ptr);
-    var v1 = getArrayJsValueFromWasm0(ret[0], ret[1]).slice();
-    wasm.__wbindgen_free(ret[0], ret[1] * 4, 4);
+    var v1 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
     return v1;
 }
 
@@ -449,165 +458,155 @@ export function process_user_decryption_resp_from_js(client, request, eip712_dom
 }
 
 /**
- * Create a new [ServerIdAddr] structure that holds an ID and an address
- * which must be a valid EIP-55 address, notably prefixed with "0x".
- * @param {number} id
- * @param {string} addr
- * @returns {ServerIdAddr}
+ * @returns {number}
  */
-export function new_server_id_addr(id, addr) {
-    const ptr0 = passStringToWasm0(addr, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
-    const len0 = WASM_VECTOR_LEN;
-    const ret = wasm.new_server_id_addr(id, ptr0, len0);
-    if (ret[2]) {
-        throw takeFromExternrefTable0(ret[1]);
-    }
-    return ServerIdAddr.__wrap(ret[0]);
+export function ml_kem_pke_pk_len() {
+    const ret = wasm.ml_kem_pke_pk_len();
+    return ret >>> 0;
 }
 
 /**
- * @returns {number}
+ * @param {Client} client
+ * @returns {PrivateSigKey | undefined}
  */
-export function ml_kem_pke_sk_len() {
-    const ret = wasm.ml_kem_pke_sk_len();
-    return ret >>> 0;
+export function get_client_secret_key(client) {
+    _assertClass(client, Client);
+    const ret = wasm.get_client_secret_key(client.__wbg_ptr);
+    return ret === 0 ? undefined : PrivateSigKey.__wrap(ret);
 }
 
 /**
- * @param {PrivateSigKey} sk
- * @returns {Uint8Array}
+ * @param {Client} client
+ * @returns {ServerIdAddr[]}
  */
-export function private_sig_key_to_u8vec(sk) {
-    _assertClass(sk, PrivateSigKey);
-    const ret = wasm.private_sig_key_to_u8vec(sk.__wbg_ptr);
-    if (ret[3]) {
-        throw takeFromExternrefTable0(ret[2]);
-    }
-    var v1 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
-    wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
+export function get_server_addrs(client) {
+    _assertClass(client, Client);
+    const ret = wasm.get_server_addrs(client.__wbg_ptr);
+    var v1 = getArrayJsValueFromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 4, 4);
     return v1;
 }
 
 /**
  * @param {Uint8Array} v
- * @returns {PrivateEncKeyMlKem512}
+ * @returns {PublicSigKey}
  */
-export function u8vec_to_ml_kem_pke_sk(v) {
+export function u8vec_to_public_sig_key(v) {
     const ptr0 = passArray8ToWasm0(v, wasm.__wbindgen_malloc);
     const len0 = WASM_VECTOR_LEN;
-    const ret = wasm.u8vec_to_ml_kem_pke_sk(ptr0, len0);
+    const ret = wasm.u8vec_to_public_sig_key(ptr0, len0);
     if (ret[2]) {
         throw takeFromExternrefTable0(ret[1]);
     }
-    return PrivateEncKeyMlKem512.__wrap(ret[0]);
+    return PublicSigKey.__wrap(ret[0]);
 }
 
 /**
- * @param {Client} client
- * @returns {string}
+ * @param {Uint8Array} v
+ * @returns {PrivateSigKey}
  */
-export function get_client_address(client) {
-    let deferred1_0;
-    let deferred1_1;
-    try {
-        _assertClass(client, Client);
-        const ret = wasm.get_client_address(client.__wbg_ptr);
-        deferred1_0 = ret[0];
-        deferred1_1 = ret[1];
-        return getStringFromWasm0(ret[0], ret[1]);
-    } finally {
-        wasm.__wbindgen_free(deferred1_0, deferred1_1, 1);
+export function u8vec_to_private_sig_key(v) {
+    const ptr0 = passArray8ToWasm0(v, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.u8vec_to_private_sig_key(ptr0, len0);
+    if (ret[2]) {
+        throw takeFromExternrefTable0(ret[1]);
     }
+    return PrivateSigKey.__wrap(ret[0]);
 }
 
 /**
- * Process the user_decryption response from Rust objects.
- * Consider using [process_user_decryption_resp_from_js]
- * when using the JS API.
- * The result is a byte array representing a plaintext of any length.
- *
- * * `client` - client that wants to perform user_decryption.
- *
- * * `request` - the initial user_decryption request.
- * Must be given if `verify` is true.
- *
- * * `eip712_domain` - the EIP-712 domain.
- * Must be given if `verify` is true.
- *
- * * `agg_resp` - the vector of user_decryption responses.
+ * @returns {PrivateEncKeyMlKem512}
+ */
+export function ml_kem_pke_keygen() {
+    const ret = wasm.ml_kem_pke_keygen();
+    return PrivateEncKeyMlKem512.__wrap(ret);
+}
+
+/**
+ * This function is *not* used by relayer-sdk because the encryption
+ * happens on the KMS side. It's just here for completeness and tests.
+ * @param {Uint8Array} msg
+ * @param {PublicEncKeyMlKem512} their_pk
+ * @returns {Uint8Array}
+ */
+export function ml_kem_pke_encrypt(msg, their_pk) {
+    const ptr0 = passArray8ToWasm0(msg, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    _assertClass(their_pk, PublicEncKeyMlKem512);
+    const ret = wasm.ml_kem_pke_encrypt(ptr0, len0, their_pk.__wbg_ptr);
+    var v2 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
+    return v2;
+}
+
+/**
+ * Instantiate a new client.
  *
- * * `enc_pk` - The ephemeral public key.
+ * * `server_addrs` - a list of KMS server ID with EIP-55 addresses,
+ * the elements in the list can be created using [new_server_id_addr].
  *
- * * `enc_sk` - The ephemeral secret key.
+ * * `client_address_hex` - the client (wallet) address in hex,
+ * must be prefixed with "0x".
  *
- * * `verify` - Whether to perform signature verification for the response.
- * It is insecure if `verify = false`!
- * @param {Client} client
- * @param {ParsedUserDecryptionRequest | null | undefined} request
- * @param {Eip712DomainMsg | null | undefined} eip712_domain
- * @param {UserDecryptionResponse[]} agg_resp
- * @param {PublicEncKeyMlKem512} enc_pk
- * @param {PrivateEncKeyMlKem512} enc_sk
- * @param {boolean} verify
- * @returns {TypedPlaintext[]}
+ * * `fhe_parameter` - the parameter choice, which can be either `"test"` or `"default"`.
+ * The "default" parameter choice is selected if no matching string is found.
+ * @param {ServerIdAddr[]} server_addrs
+ * @param {string} client_address_hex
+ * @param {string} fhe_parameter
+ * @returns {Client}
  */
-export function process_user_decryption_resp(client, request, eip712_domain, agg_resp, enc_pk, enc_sk, verify) {
-    _assertClass(client, Client);
-    let ptr0 = 0;
-    if (!isLikeNone(request)) {
-        _assertClass(request, ParsedUserDecryptionRequest);
-        ptr0 = request.__destroy_into_raw();
-    }
-    let ptr1 = 0;
-    if (!isLikeNone(eip712_domain)) {
-        _assertClass(eip712_domain, Eip712DomainMsg);
-        ptr1 = eip712_domain.__destroy_into_raw();
-    }
-    const ptr2 = passArrayJsValueToWasm0(agg_resp, wasm.__wbindgen_malloc);
+export function new_client(server_addrs, client_address_hex, fhe_parameter) {
+    const ptr0 = passArrayJsValueToWasm0(server_addrs, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ptr1 = passStringToWasm0(client_address_hex, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+    const len1 = WASM_VECTOR_LEN;
+    const ptr2 = passStringToWasm0(fhe_parameter, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
     const len2 = WASM_VECTOR_LEN;
-    _assertClass(enc_pk, PublicEncKeyMlKem512);
-    _assertClass(enc_sk, PrivateEncKeyMlKem512);
-    const ret = wasm.process_user_decryption_resp(client.__wbg_ptr, ptr0, ptr1, ptr2, len2, enc_pk.__wbg_ptr, enc_sk.__wbg_ptr, verify);
-    if (ret[3]) {
-        throw takeFromExternrefTable0(ret[2]);
+    const ret = wasm.new_client(ptr0, len0, ptr1, len1, ptr2, len2);
+    if (ret[2]) {
+        throw takeFromExternrefTable0(ret[1]);
     }
-    var v4 = getArrayJsValueFromWasm0(ret[0], ret[1]).slice();
-    wasm.__wbindgen_free(ret[0], ret[1] * 4, 4);
-    return v4;
+    return Client.__wrap(ret[0]);
 }
 
 /**
- * @param {Client} client
- * @returns {PrivateSigKey | undefined}
+ * @param {PublicSigKey} pk
+ * @returns {Uint8Array}
  */
-export function get_client_secret_key(client) {
-    _assertClass(client, Client);
-    const ret = wasm.get_client_secret_key(client.__wbg_ptr);
-    return ret === 0 ? undefined : PrivateSigKey.__wrap(ret);
+export function public_sig_key_to_u8vec(pk) {
+    _assertClass(pk, PublicSigKey);
+    const ret = wasm.public_sig_key_to_u8vec(pk.__wbg_ptr);
+    var v1 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
+    return v1;
 }
 
 /**
- * @param {PublicEncKeyMlKem512} pk
+ * This function is *not* used by relayer-sdk because the decryption
+ * is handled by [process_user_decryption_resp].
+ * It's just here for completeness and tests.
+ * @param {Uint8Array} ct
+ * @param {PrivateEncKeyMlKem512} my_sk
  * @returns {Uint8Array}
  */
-export function ml_kem_pke_pk_to_u8vec(pk) {
-    _assertClass(pk, PublicEncKeyMlKem512);
-    const ret = wasm.ml_kem_pke_pk_to_u8vec(pk.__wbg_ptr);
-    if (ret[3]) {
-        throw takeFromExternrefTable0(ret[2]);
-    }
-    var v1 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
+export function ml_kem_pke_decrypt(ct, my_sk) {
+    const ptr0 = passArray8ToWasm0(ct, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    _assertClass(my_sk, PrivateEncKeyMlKem512);
+    const ret = wasm.ml_kem_pke_decrypt(ptr0, len0, my_sk.__wbg_ptr);
+    var v2 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
     wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
-    return v1;
+    return v2;
 }
 
 /**
- * @param {PrivateEncKeyMlKem512} sk
+ * @param {PrivateSigKey} sk
  * @returns {Uint8Array}
  */
-export function ml_kem_pke_sk_to_u8vec(sk) {
-    _assertClass(sk, PrivateEncKeyMlKem512);
-    const ret = wasm.ml_kem_pke_sk_to_u8vec(sk.__wbg_ptr);
+export function private_sig_key_to_u8vec(sk) {
+    _assertClass(sk, PrivateSigKey);
+    const ret = wasm.private_sig_key_to_u8vec(sk.__wbg_ptr);
     if (ret[3]) {
         throw takeFromExternrefTable0(ret[2]);
     }

package/package.json

@@ -5,7 +5,7 @@
     "Zama"
   ],
   "description": "Key Management System for the Zama Protocol.",
-  "version": "v0.13.0-rc.1",
+  "version": "v0.13.0-rc.2",
   "license": "BSD-3-Clause-Clear",
   "repository": {
     "type": "git",