tkms 0.12.5 → 0.12.8

package/kms_lib.d.ts

@@ -1,60 +1,13 @@
 /* tslint:disable */
 /* eslint-disable */
-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 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_public_sig_key(v: Uint8Array): PublicSigKey;
+export function ml_kem_pke_pk_len(): number;
 export function u8vec_to_private_sig_key(v: Uint8Array): PrivateSigKey;
-/**
- * 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`!
- */
-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_keygen(): PrivateEncKeyMlKem512;
-export function u8vec_to_ml_kem_pke_pk(v: Uint8Array): PublicEncKeyMlKem512;
 /**
  * Process the user_decryption response from JavaScript objects.
  * The returned result is a byte array representing a plaintext of any length,
@@ -117,6 +70,52 @@ export function u8vec_to_ml_kem_pke_pk(v: Uint8Array): PublicEncKeyMlKem512;
  * 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 private_sig_key_to_u8vec(sk: PrivateSigKey): 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.
+ *
+ * * `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 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[];
+/**
+ * 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 ml_kem_pke_pk_to_u8vec(pk: PublicEncKeyMlKem512): Uint8Array;
+export function get_client_secret_key(client: Client): PrivateSigKey | undefined;
+export function public_sig_key_to_u8vec(pk: PublicSigKey): Uint8Array;
+export function ml_kem_pke_keygen(): PrivateEncKeyMlKem512;
+export function get_client_address(client: Client): string;
+export function ml_kem_pke_sk_to_u8vec(sk: PrivateEncKeyMlKem512): Uint8Array;
+export function get_server_addrs(client: Client): ServerIdAddr[];
+export function u8vec_to_ml_kem_pke_pk(v: Uint8Array): PublicEncKeyMlKem512;
+export function u8vec_to_ml_kem_pke_sk(v: Uint8Array): PrivateEncKeyMlKem512;
+export function ml_kem_pke_sk_len(): number;
+/**
+ * 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;
 /**
  * Instantiate a new client.
  *
@@ -130,6 +129,7 @@ export function process_user_decryption_resp_from_js(client: Client, request: an
  * The "default" parameter choice is selected if no matching string is found.
  */
 export function new_client(server_addrs: ServerIdAddr[], client_address_hex: string, fhe_parameter: string): Client;
+export function ml_kem_pke_get_pk(sk: PrivateEncKeyMlKem512): PublicEncKeyMlKem512;
 export class CiphertextHandle {
   private constructor();
   free(): void;
@@ -425,13 +425,11 @@ export interface InitOutput {
   readonly __wbg_get_eip712domainmsg_salt: (a: number) => [number, number];
   readonly __wbg_get_eip712domainmsg_verifying_contract: (a: number) => [number, number];
   readonly __wbg_get_eip712domainmsg_version: (a: number) => [number, number];
-  readonly __wbg_get_requestid_request_id: (a: number) => [number, number];
   readonly __wbg_get_typedciphertext_ciphertext: (a: number) => [number, number];
   readonly __wbg_get_typedciphertext_ciphertext_format: (a: number) => number;
   readonly __wbg_get_typedciphertext_external_handle: (a: number) => [number, number];
   readonly __wbg_get_typedciphertext_fhe_type: (a: number) => number;
   readonly __wbg_get_typedplaintext_fhe_type: (a: number) => number;
-  readonly __wbg_get_userdecryptionrequest_client_address: (a: number) => [number, number];
   readonly __wbg_get_userdecryptionrequest_context_id: (a: number) => number;
   readonly __wbg_get_userdecryptionrequest_domain: (a: number) => number;
   readonly __wbg_get_userdecryptionrequest_epoch_id: (a: number) => number;
@@ -468,6 +466,8 @@ 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_set_requestid_request_id: (a: number, b: number, c: number) => void;
@@ -501,8 +501,8 @@ export interface InitOutput {
   readonly __externref_table_alloc: () => number;
   readonly __wbindgen_export_4: WebAssembly.Table;
   readonly __wbindgen_free: (a: number, b: number, c: number) => void;
-  readonly __externref_drop_slice: (a: number, b: number) => void;
   readonly __externref_table_dealloc: (a: number) => void;
+  readonly __externref_drop_slice: (a: number, b: number) => void;
   readonly __wbindgen_start: () => void;
 }
 

package/kms_lib.js

@@ -167,82 +167,44 @@ 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}`);
     }
 }
 
-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[]}
- */
-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;
-}
-
-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 {PrivateEncKeyMlKem512} sk
+ * 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_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();
+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 v1;
-}
-
-/**
- * @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);
-    }
+    return v2;
 }
 
-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
@@ -258,59 +220,6 @@ export function u8vec_to_public_sig_key(v) {
     return PublicSigKey.__wrap(ret[0]);
 }
 
-/**
- * @param {PrivateSigKey} sk
- * @returns {Uint8Array}
- */
-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);
-    return v1;
-}
-
-/**
- * @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;
-}
-
-/**
- * @returns {number}
- */
-export function ml_kem_pke_sk_len() {
-    const ret = wasm.ml_kem_pke_sk_len();
-    return ret >>> 0;
-}
-
 /**
  * @returns {number}
  */
@@ -319,71 +228,119 @@ export function ml_kem_pke_pk_len() {
     return ret >>> 0;
 }
 
-/**
- * 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;
-}
-
-/**
- * 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 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]);
-}
-
 /**
  * @param {Uint8Array} v
- * @returns {PrivateEncKeyMlKem512}
+ * @returns {PrivateSigKey}
  */
-export function u8vec_to_ml_kem_pke_sk(v) {
+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_ml_kem_pke_sk(ptr0, len0);
+    const ret = wasm.u8vec_to_private_sig_key(ptr0, len0);
     if (ret[2]) {
         throw takeFromExternrefTable0(ret[1]);
     }
-    return PrivateEncKeyMlKem512.__wrap(ret[0]);
+    return PrivateSigKey.__wrap(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 {PrivateEncKeyMlKem512} sk
- * @returns {PublicEncKeyMlKem512}
+ * Process the user_decryption response from JavaScript objects.
+ * The returned result is a byte array representing a plaintext of any length,
+ * postprocessing is returned to turn it into an integer.
+ *
+ * * `client` - client that wants to perform user_decryption.
+ *
+ * * `request` - the initial user_decryption request JS object.
+ * It can be set to null if `verify` is false.
+ * Otherwise the caller needs to give the following JS object.
+ * Note that `client_address` and `eip712_verifying_contract` follow EIP-55.
+ * The signature field is not needed.
+ * ```
+ * {
+ *   signature: undefined,
+ *   client_address: '0x17853A630aAe15AED549B2B874de08B73C0F59c5',
+ *   enc_key: '2000000000000000df2fcacb774f03187f3802a27259f45c06d33cefa68d9c53426b15ad531aa822',
+ *   ciphertext_handles: [ '0748b542afe2353c86cb707e3d21044b0be1fd18efc7cbaa6a415af055bfb358' ]
+ *   eip712_verifying_contract: '0x66f9664f97F2b50F62D13eA064982f936dE76657'
+ * }
+ * ```
+ *
+ * * `eip712_domain` - the EIP-712 domain JS object.
+ * It can be set to null if `verify` is false.
+ * Otherwise the caller needs to give the following JS object.
+ * Note that `salt` is optional and `verifying_contract` follows EIP-55,
+ * additionally, `chain_id` is an array of u8.
+ * ```
+ * {
+ *   name: 'Authorization token',
+ *   version: '1',
+ *   chain_id: [
+ *     70, 31, 0, 0, 0, 0, 0, 0, 0,
+ *      0,  0, 0, 0, 0, 0, 0, 0, 0,
+ *      0,  0, 0, 0, 0, 0, 0, 0, 0,
+ *      0,  0, 0, 0, 0
+ *   ],
+ *   verifying_contract: '0x66f9664f97F2b50F62D13eA064982f936dE76657',
+ *   salt: []
+ * }
+ * ```
+ *
+ * * `agg_resp` - the response JS object from the gateway.
+ * It has two fields like so, both are hex encoded byte arrays.
+ * ```
+ * [
+ *   {
+ *     signature: '69e7e040cab157aa819015b321c012dccb1545ffefd325b359b492653f0347517e28e66c572cdc299e259024329859ff9fcb0096e1ce072af0b6e1ca1fe25ec6',
+ *     payload: '0100000029...',
+ *     extra_data: '01234...',
+ *   }
+ * ]
+ * ```
+ *
+ * * `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 {any} request
+ * @param {any} eip712_domain
+ * @param {any} agg_resp
+ * @param {PublicEncKeyMlKem512} enc_pk
+ * @param {PrivateEncKeyMlKem512} enc_sk
+ * @param {boolean} verify
+ * @returns {TypedPlaintext[]}
  */
-export function ml_kem_pke_get_pk(sk) {
-    _assertClass(sk, PrivateEncKeyMlKem512);
-    const ret = wasm.ml_kem_pke_get_pk(sk.__wbg_ptr);
-    return PublicEncKeyMlKem512.__wrap(ret);
+export function process_user_decryption_resp_from_js(client, request, eip712_domain, agg_resp, enc_pk, enc_sk, verify) {
+    _assertClass(client, Client);
+    _assertClass(enc_pk, PublicEncKeyMlKem512);
+    _assertClass(enc_sk, PrivateEncKeyMlKem512);
+    const ret = wasm.process_user_decryption_resp_from_js(client.__wbg_ptr, request, eip712_domain, agg_resp, enc_pk.__wbg_ptr, enc_sk.__wbg_ptr, verify);
+    if (ret[3]) {
+        throw takeFromExternrefTable0(ret[2]);
+    }
+    var v1 = getArrayJsValueFromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 4, 4);
+    return v1;
 }
 
 /**
- * @param {PublicEncKeyMlKem512} pk
+ * @param {PrivateSigKey} 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);
+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]);
     }
@@ -392,20 +349,6 @@ export function ml_kem_pke_pk_to_u8vec(pk) {
     return v1;
 }
 
-/**
- * @param {Uint8Array} v
- * @returns {PrivateSigKey}
- */
-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]);
-}
-
 function passArrayJsValueToWasm0(array, malloc) {
     const ptr = malloc(array.length * 4, 4) >>> 0;
     for (let i = 0; i < array.length; i++) {
@@ -471,6 +414,38 @@ export function process_user_decryption_resp(client, request, eip712_domain, agg
     return v4;
 }
 
+/**
+ * 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 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]);
+}
+
+/**
+ * @param {PublicEncKeyMlKem512} pk
+ * @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();
+    wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
+    return v1;
+}
+
 /**
  * @param {Client} client
  * @returns {PrivateSigKey | undefined}
@@ -481,6 +456,18 @@ export function get_client_secret_key(client) {
     return ret === 0 ? undefined : PrivateSigKey.__wrap(ret);
 }
 
+/**
+ * @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;
+}
+
 /**
  * @returns {PrivateEncKeyMlKem512}
  */
@@ -489,6 +476,51 @@ export function ml_kem_pke_keygen() {
     return PrivateEncKeyMlKem512.__wrap(ret);
 }
 
+/**
+ * @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);
+    }
+}
+
+/**
+ * @param {PrivateEncKeyMlKem512} 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);
+    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 {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);
+    return v1;
+}
+
 /**
  * @param {Uint8Array} v
  * @returns {PublicEncKeyMlKem512}
@@ -504,85 +536,43 @@ export function u8vec_to_ml_kem_pke_pk(v) {
 }
 
 /**
- * Process the user_decryption response from JavaScript objects.
- * The returned result is a byte array representing a plaintext of any length,
- * postprocessing is returned to turn it into an integer.
- *
- * * `client` - client that wants to perform user_decryption.
- *
- * * `request` - the initial user_decryption request JS object.
- * It can be set to null if `verify` is false.
- * Otherwise the caller needs to give the following JS object.
- * Note that `client_address` and `eip712_verifying_contract` follow EIP-55.
- * The signature field is not needed.
- * ```
- * {
- *   signature: undefined,
- *   client_address: '0x17853A630aAe15AED549B2B874de08B73C0F59c5',
- *   enc_key: '2000000000000000df2fcacb774f03187f3802a27259f45c06d33cefa68d9c53426b15ad531aa822',
- *   ciphertext_handles: [ '0748b542afe2353c86cb707e3d21044b0be1fd18efc7cbaa6a415af055bfb358' ]
- *   eip712_verifying_contract: '0x66f9664f97F2b50F62D13eA064982f936dE76657'
- * }
- * ```
- *
- * * `eip712_domain` - the EIP-712 domain JS object.
- * It can be set to null if `verify` is false.
- * Otherwise the caller needs to give the following JS object.
- * Note that `salt` is optional and `verifying_contract` follows EIP-55,
- * additionally, `chain_id` is an array of u8.
- * ```
- * {
- *   name: 'Authorization token',
- *   version: '1',
- *   chain_id: [
- *     70, 31, 0, 0, 0, 0, 0, 0, 0,
- *      0,  0, 0, 0, 0, 0, 0, 0, 0,
- *      0,  0, 0, 0, 0, 0, 0, 0, 0,
- *      0,  0, 0, 0, 0
- *   ],
- *   verifying_contract: '0x66f9664f97F2b50F62D13eA064982f936dE76657',
- *   salt: []
- * }
- * ```
- *
- * * `agg_resp` - the response JS object from the gateway.
- * It has two fields like so, both are hex encoded byte arrays.
- * ```
- * [
- *   {
- *     signature: '69e7e040cab157aa819015b321c012dccb1545ffefd325b359b492653f0347517e28e66c572cdc299e259024329859ff9fcb0096e1ce072af0b6e1ca1fe25ec6',
- *     payload: '0100000029...',
- *     extra_data: '01234...',
- *   }
- * ]
- * ```
- *
- * * `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 {any} request
- * @param {any} eip712_domain
- * @param {any} agg_resp
- * @param {PublicEncKeyMlKem512} enc_pk
- * @param {PrivateEncKeyMlKem512} enc_sk
- * @param {boolean} verify
- * @returns {TypedPlaintext[]}
+ * @param {Uint8Array} v
+ * @returns {PrivateEncKeyMlKem512}
  */
-export function process_user_decryption_resp_from_js(client, request, eip712_domain, agg_resp, enc_pk, enc_sk, verify) {
-    _assertClass(client, Client);
-    _assertClass(enc_pk, PublicEncKeyMlKem512);
-    _assertClass(enc_sk, PrivateEncKeyMlKem512);
-    const ret = wasm.process_user_decryption_resp_from_js(client.__wbg_ptr, request, eip712_domain, agg_resp, enc_pk.__wbg_ptr, enc_sk.__wbg_ptr, verify);
-    if (ret[3]) {
-        throw takeFromExternrefTable0(ret[2]);
+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_sk(ptr0, len0);
+    if (ret[2]) {
+        throw takeFromExternrefTable0(ret[1]);
     }
-    var v1 = getArrayJsValueFromWasm0(ret[0], ret[1]).slice();
-    wasm.__wbindgen_free(ret[0], ret[1] * 4, 4);
-    return v1;
+    return PrivateEncKeyMlKem512.__wrap(ret[0]);
+}
+
+/**
+ * @returns {number}
+ */
+export function ml_kem_pke_sk_len() {
+    const ret = wasm.ml_kem_pke_sk_len();
+    return ret >>> 0;
+}
+
+/**
+ * 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;
 }
 
 /**
@@ -615,6 +605,16 @@ export function new_client(server_addrs, client_address_hex, fhe_parameter) {
     return Client.__wrap(ret[0]);
 }
 
+/**
+ * @param {PrivateEncKeyMlKem512} sk
+ * @returns {PublicEncKeyMlKem512}
+ */
+export function ml_kem_pke_get_pk(sk) {
+    _assertClass(sk, PrivateEncKeyMlKem512);
+    const ret = wasm.ml_kem_pke_get_pk(sk.__wbg_ptr);
+    return PublicEncKeyMlKem512.__wrap(ret);
+}
+
 const CiphertextHandleFinalization = (typeof FinalizationRegistry === 'undefined')
     ? { register: () => {}, unregister: () => {} }
     : new FinalizationRegistry(ptr => wasm.__wbg_ciphertexthandle_free(ptr >>> 0, 1));

package/package.json

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