space-data-module-sdk 0.4.1 → 0.5.4

package/src/utils/wasmCrypto.js

@@ -116,3 +116,46 @@ export async function aesGcmDecrypt(key, ciphertext, tag, iv, aad = null) {
     aad ? toUint8Array(aad) : undefined,
   );
 }
+
+function normalizeCtrIv(nonceStart) {
+  const nonce = toUint8Array(nonceStart);
+  if (nonce.length !== 12) {
+    throw new Error("AES-256-CTR expects a 12-byte NONCE_START value.");
+  }
+  const iv = new Uint8Array(16);
+  iv.set(nonce, 0);
+  return iv;
+}
+
+async function getAesCtrApi() {
+  const wallet = await getWasmWallet();
+  const aesCtr = wallet?.aesCtr;
+  if (
+    !aesCtr ||
+    typeof aesCtr.encrypt !== "function" ||
+    typeof aesCtr.decrypt !== "function"
+  ) {
+    throw new Error(
+      "hd-wallet-wasm aesCtr API is unavailable; cannot process ENC payloads.",
+    );
+  }
+  return aesCtr;
+}
+
+export async function aesCtrEncrypt(key, plaintext, nonceStart) {
+  const aesCtr = await getAesCtrApi();
+  return aesCtr.encrypt(
+    toUint8Array(key),
+    toUint8Array(plaintext),
+    normalizeCtrIv(nonceStart),
+  );
+}
+
+export async function aesCtrDecrypt(key, ciphertext, nonceStart) {
+  const aesCtr = await getAesCtrApi();
+  return aesCtr.decrypt(
+    toUint8Array(key),
+    toUint8Array(ciphertext),
+    normalizeCtrIv(nonceStart),
+  );
+}