node-tkms 0.12.4 → 0.12.5

package/kms_lib.d.ts

@@ -1,49 +1,60 @@
 /* tslint:disable */
 /* eslint-disable */
-export function ml_kem_pke_pk_len(): number;
-export function ml_kem_pke_sk_len(): number;
-export function public_sig_key_to_u8vec(pk: PublicSigKey): Uint8Array;
+export function get_server_addrs(client: Client): ServerIdAddr[];
+export function ml_kem_pke_sk_to_u8vec(sk: PrivateEncKeyMlKem512): Uint8Array;
+export function get_client_address(client: Client): string;
 export function u8vec_to_public_sig_key(v: Uint8Array): PublicSigKey;
 export function private_sig_key_to_u8vec(sk: PrivateSigKey): Uint8Array;
-export function u8vec_to_private_sig_key(v: Uint8Array): PrivateSigKey;
+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 ml_kem_pke_sk_len(): number;
+export function ml_kem_pke_pk_len(): number;
+/**
+ * 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;
 /**
  * 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 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_get_pk(sk: PrivateEncKeyMlKem512): PublicEncKeyMlKem512;
+export function ml_kem_pke_pk_to_u8vec(pk: PublicEncKeyMlKem512): Uint8Array;
+export function u8vec_to_private_sig_key(v: Uint8Array): PrivateSigKey;
 /**
- * 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 get_server_addrs(client: Client): ServerIdAddr[];
+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 get_client_address(client: Client): string;
 export function ml_kem_pke_keygen(): PrivateEncKeyMlKem512;
-export function ml_kem_pke_get_pk(sk: PrivateEncKeyMlKem512): PublicEncKeyMlKem512;
-export function ml_kem_pke_pk_to_u8vec(pk: PublicEncKeyMlKem512): Uint8Array;
-export function ml_kem_pke_sk_to_u8vec(sk: PrivateEncKeyMlKem512): Uint8Array;
 export function u8vec_to_ml_kem_pke_pk(v: Uint8Array): PublicEncKeyMlKem512;
-export function u8vec_to_ml_kem_pke_sk(v: Uint8Array): PrivateEncKeyMlKem512;
-/**
- * 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;
-/**
- * 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;
 /**
  * Process the user_decryption response from JavaScript objects.
  * The returned result is a byte array representing a plaintext of any length,
@@ -107,29 +118,18 @@ export function ml_kem_pke_decrypt(ct: Uint8Array, my_sk: PrivateEncKeyMlKem512)
  */
 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[];
 /**
- * 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 new_client(server_addrs: ServerIdAddr[], client_address_hex: string, fhe_parameter: string): Client;
 export class CiphertextHandle {
   private constructor();
   free(): void;

package/kms_lib.js

@@ -170,21 +170,6 @@ function debugString(val) {
     // TODO we could test for more things here, like `Set`s and `Map`s.
     return className;
 }
-/**
- * @returns {number}
- */
-module.exports.ml_kem_pke_pk_len = function() {
-    const ret = wasm.ml_kem_pke_pk_len();
-    return ret >>> 0;
-};
-
-/**
- * @returns {number}
- */
-module.exports.ml_kem_pke_sk_len = function() {
-    const ret = wasm.ml_kem_pke_sk_len();
-    return ret >>> 0;
-};
 
 function _assertClass(instance, klass) {
     if (!(instance instanceof klass)) {
@@ -192,34 +177,77 @@ function _assertClass(instance, klass) {
     }
 }
 
+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;
+}
+/**
+ * @param {Client} client
+ * @returns {ServerIdAddr[]}
+ */
+module.exports.get_server_addrs = function(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;
+};
+
+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);
 }
 /**
- * @param {PublicSigKey} pk
+ * @param {PrivateEncKeyMlKem512} sk
  * @returns {Uint8Array}
  */
-module.exports.public_sig_key_to_u8vec = function(pk) {
-    _assertClass(pk, PublicSigKey);
-    const ret = wasm.public_sig_key_to_u8vec(pk.__wbg_ptr);
+module.exports.ml_kem_pke_sk_to_u8vec = function(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 v1;
 };
 
+/**
+ * @param {Client} client
+ * @returns {string}
+ */
+module.exports.get_client_address = function(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);
+    }
+};
+
 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 takeFromExternrefTable0(idx) {
-    const value = wasm.__wbindgen_export_4.get(idx);
-    wasm.__externref_table_dealloc(idx);
-    return value;
-}
 /**
  * @param {Uint8Array} v
  * @returns {PublicSigKey}
@@ -250,131 +278,97 @@ module.exports.private_sig_key_to_u8vec = function(sk) {
 };
 
 /**
- * @param {Uint8Array} v
- * @returns {PrivateSigKey}
+ * @param {PublicSigKey} pk
+ * @returns {Uint8Array}
  */
-module.exports.u8vec_to_private_sig_key = function(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]);
+module.exports.public_sig_key_to_u8vec = function(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;
 };
 
 /**
- * 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}
+ * 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}
  */
-module.exports.new_server_id_addr = function(id, addr) {
-    const ptr0 = passStringToWasm0(addr, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+module.exports.ml_kem_pke_decrypt = function(ct, my_sk) {
+    const ptr0 = passArray8ToWasm0(ct, wasm.__wbindgen_malloc);
     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]);
+    _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;
 };
 
-function passArrayJsValueToWasm0(array, malloc) {
-    const ptr = malloc(array.length * 4, 4) >>> 0;
-    for (let i = 0; i < array.length; i++) {
-        const add = addToExternrefTable0(array[i]);
-        getDataViewMemory0().setUint32(ptr + 4 * i, add, true);
-    }
-    WASM_VECTOR_LEN = array.length;
-    return ptr;
-}
 /**
- * Instantiate a new client.
- *
- * * `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_address_hex` - the client (wallet) address in hex,
- * must be prefixed with "0x".
- *
- * * `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}
+ * @returns {number}
  */
-module.exports.new_client = function(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;
-    const ret = wasm.new_client(ptr0, len0, ptr1, len1, ptr2, len2);
-    if (ret[2]) {
-        throw takeFromExternrefTable0(ret[1]);
-    }
-    return Client.__wrap(ret[0]);
+module.exports.ml_kem_pke_sk_len = function() {
+    const ret = wasm.ml_kem_pke_sk_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)));
-    }
-    wasm.__externref_drop_slice(ptr, len);
-    return result;
-}
 /**
- * @param {Client} client
- * @returns {ServerIdAddr[]}
+ * @returns {number}
  */
-module.exports.get_server_addrs = function(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;
+module.exports.ml_kem_pke_pk_len = function() {
+    const ret = wasm.ml_kem_pke_pk_len();
+    return ret >>> 0;
 };
 
 /**
- * @param {Client} client
- * @returns {PrivateSigKey | undefined}
+ * 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}
  */
-module.exports.get_client_secret_key = function(client) {
-    _assertClass(client, Client);
-    const ret = wasm.get_client_secret_key(client.__wbg_ptr);
-    return ret === 0 ? undefined : PrivateSigKey.__wrap(ret);
+module.exports.ml_kem_pke_encrypt = function(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;
 };
 
 /**
- * @param {Client} client
- * @returns {string}
+ * 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}
  */
-module.exports.get_client_address = function(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);
+module.exports.new_server_id_addr = function(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]);
 };
 
 /**
+ * @param {Uint8Array} v
  * @returns {PrivateEncKeyMlKem512}
  */
-module.exports.ml_kem_pke_keygen = function() {
-    const ret = wasm.ml_kem_pke_keygen();
-    return PrivateEncKeyMlKem512.__wrap(ret);
+module.exports.u8vec_to_ml_kem_pke_sk = function(v) {
+    const ptr0 = passArray8ToWasm0(v, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.u8vec_to_ml_kem_pke_sk(ptr0, len0);
+    if (ret[2]) {
+        throw takeFromExternrefTable0(ret[1]);
+    }
+    return PrivateEncKeyMlKem512.__wrap(ret[0]);
 };
 
 /**
@@ -402,82 +396,115 @@ module.exports.ml_kem_pke_pk_to_u8vec = function(pk) {
     return v1;
 };
 
-/**
- * @param {PrivateEncKeyMlKem512} sk
- * @returns {Uint8Array}
- */
-module.exports.ml_kem_pke_sk_to_u8vec = function(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 v1;
-};
-
 /**
  * @param {Uint8Array} v
- * @returns {PublicEncKeyMlKem512}
+ * @returns {PrivateSigKey}
  */
-module.exports.u8vec_to_ml_kem_pke_pk = function(v) {
+module.exports.u8vec_to_private_sig_key = function(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_private_sig_key(ptr0, len0);
     if (ret[2]) {
         throw takeFromExternrefTable0(ret[1]);
     }
-    return PublicEncKeyMlKem512.__wrap(ret[0]);
+    return PrivateSigKey.__wrap(ret[0]);
 };
 
+function passArrayJsValueToWasm0(array, malloc) {
+    const ptr = malloc(array.length * 4, 4) >>> 0;
+    for (let i = 0; i < array.length; i++) {
+        const add = addToExternrefTable0(array[i]);
+        getDataViewMemory0().setUint32(ptr + 4 * i, add, true);
+    }
+    WASM_VECTOR_LEN = array.length;
+    return ptr;
+}
 /**
- * @param {Uint8Array} v
- * @returns {PrivateEncKeyMlKem512}
+ * 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.
+ *
+ * * `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[]}
  */
-module.exports.u8vec_to_ml_kem_pke_sk = function(v) {
-    const ptr0 = passArray8ToWasm0(v, wasm.__wbindgen_malloc);
-    const len0 = WASM_VECTOR_LEN;
-    const ret = wasm.u8vec_to_ml_kem_pke_sk(ptr0, len0);
-    if (ret[2]) {
-        throw takeFromExternrefTable0(ret[1]);
+module.exports.process_user_decryption_resp = function(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();
     }
-    return PrivateEncKeyMlKem512.__wrap(ret[0]);
+    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;
+    _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]);
+    }
+    var v4 = getArrayJsValueFromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 4, 4);
+    return v4;
 };
 
 /**
- * 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}
+ * @param {Client} client
+ * @returns {PrivateSigKey | undefined}
  */
-module.exports.ml_kem_pke_encrypt = function(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;
+module.exports.get_client_secret_key = function(client) {
+    _assertClass(client, Client);
+    const ret = wasm.get_client_secret_key(client.__wbg_ptr);
+    return ret === 0 ? undefined : PrivateSigKey.__wrap(ret);
 };
 
 /**
- * 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}
+ * @returns {PrivateEncKeyMlKem512}
  */
-module.exports.ml_kem_pke_decrypt = function(ct, my_sk) {
-    const ptr0 = passArray8ToWasm0(ct, wasm.__wbindgen_malloc);
+module.exports.ml_kem_pke_keygen = function() {
+    const ret = wasm.ml_kem_pke_keygen();
+    return PrivateEncKeyMlKem512.__wrap(ret);
+};
+
+/**
+ * @param {Uint8Array} v
+ * @returns {PublicEncKeyMlKem512}
+ */
+module.exports.u8vec_to_ml_kem_pke_pk = function(v) {
+    const ptr0 = passArray8ToWasm0(v, 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;
+    const ret = wasm.u8vec_to_ml_kem_pke_pk(ptr0, len0);
+    if (ret[2]) {
+        throw takeFromExternrefTable0(ret[1]);
+    }
+    return PublicEncKeyMlKem512.__wrap(ret[0]);
 };
 
 /**
@@ -563,59 +590,33 @@ module.exports.process_user_decryption_resp_from_js = function(client, request,
 };
 
 /**
- * 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`!
- * @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}
  */
-module.exports.process_user_decryption_resp = function(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);
+module.exports.new_client = function(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]);
 };
 
 const CiphertextHandleFinalization = (typeof FinalizationRegistry === 'undefined')

package/package.json

@@ -4,7 +4,7 @@
     "Zama"
   ],
   "description": "Key Management System for the Zama Protocol.",
-  "version": "0.12.4",
+  "version": "0.12.5",
   "license": "BSD-3-Clause-Clear",
   "repository": {
     "type": "git",