react-native-quick-crypto 1.0.1 → 1.0.3

package/lib/module/subtle.js

@@ -17,8 +17,9 @@ import { rsa_generateKeyPair } from './rsa';
 import { getRandomValues } from './random';
 import { createHmac } from './hmac';
 import { createSign, createVerify } from './keys/signVerify';
-import { ed_generateKeyPairWebCrypto, Ed } from './ed';
+import { ed_generateKeyPairWebCrypto, x_generateKeyPairWebCrypto, xDeriveBits, Ed } from './ed';
 import { mldsa_generateKeyPairWebCrypto } from './mldsa';
+import { hkdfDeriveBits } from './hkdf';
 // import { pbkdf2DeriveBits } from './pbkdf2';
 // import { aesCipher, aesGenerateKey, aesImportKey, getAlgorithmName } from './aes';
 // import { rsaCipher, rsaExportKey, rsaImportKey, rsaKeyGenerate } from './rsa';
@@ -267,6 +268,120 @@ async function aesGcmCipher(mode, key, data, algorithm) {
     return result.buffer;
   }
 }
+async function aesKwCipher(mode, key, data) {
+  const isWrap = mode === CipherOrWrapMode.kWebCryptoCipherEncrypt;
+
+  // AES-KW requires input to be a multiple of 8 bytes (64 bits)
+  if (data.byteLength % 8 !== 0) {
+    throw lazyDOMException(`AES-KW input length must be a multiple of 8 bytes, got ${data.byteLength}`, 'OperationError');
+  }
+
+  // AES-KW requires at least 16 bytes of input (128 bits)
+  if (isWrap && data.byteLength < 16) {
+    throw lazyDOMException(`AES-KW input must be at least 16 bytes, got ${data.byteLength}`, 'OperationError');
+  }
+
+  // Get cipher type based on key length
+  const keyLength = key.algorithm.length;
+  // Use aes*-wrap for both operations (matching Node.js)
+  const cipherType = `aes${keyLength}-wrap`;
+
+  // Export key material
+  const exportedKey = key.keyObject.export();
+  const cipherKey = bufferLikeToArrayBuffer(exportedKey);
+
+  // AES-KW uses a default IV as specified in RFC 3394
+  const defaultWrapIV = new Uint8Array([0xa6, 0xa6, 0xa6, 0xa6, 0xa6, 0xa6, 0xa6, 0xa6]);
+  const factory = NitroModules.createHybridObject('CipherFactory');
+  const cipher = factory.createCipher({
+    isCipher: isWrap,
+    cipherType,
+    cipherKey,
+    iv: defaultWrapIV.buffer // RFC 3394 default IV for AES-KW
+  });
+
+  // Process data
+  const updated = cipher.update(data);
+  const final = cipher.final();
+
+  // Concatenate results
+  const result = new Uint8Array(updated.byteLength + final.byteLength);
+  result.set(new Uint8Array(updated), 0);
+  result.set(new Uint8Array(final), updated.byteLength);
+  return result.buffer;
+}
+async function chaCha20Poly1305Cipher(mode, key, data, algorithm) {
+  const {
+    iv,
+    additionalData,
+    tagLength = 128
+  } = algorithm;
+
+  // Validate IV (must be 12 bytes for ChaCha20-Poly1305)
+  const ivBuffer = bufferLikeToArrayBuffer(iv);
+  if (!ivBuffer || ivBuffer.byteLength !== 12) {
+    throw lazyDOMException('ChaCha20-Poly1305 IV must be exactly 12 bytes', 'OperationError');
+  }
+
+  // Validate tag length (only 128-bit supported)
+  if (tagLength !== 128) {
+    throw lazyDOMException('ChaCha20-Poly1305 only supports 128-bit auth tags', 'NotSupportedError');
+  }
+  const tagByteLength = 16; // 128 bits = 16 bytes
+
+  // Create cipher using existing ChaCha20-Poly1305 implementation
+  const factory = NitroModules.createHybridObject('CipherFactory');
+  const cipher = factory.createCipher({
+    isCipher: mode === CipherOrWrapMode.kWebCryptoCipherEncrypt,
+    cipherType: 'chacha20-poly1305',
+    cipherKey: bufferLikeToArrayBuffer(key.keyObject.export()),
+    iv: ivBuffer,
+    authTagLen: tagByteLength
+  });
+  let processData;
+  let authTag;
+  if (mode === CipherOrWrapMode.kWebCryptoCipherDecrypt) {
+    // For decryption, extract auth tag from end of data
+    const dataView = new Uint8Array(data);
+    if (dataView.byteLength < tagByteLength) {
+      throw lazyDOMException('The provided data is too small.', 'OperationError');
+    }
+
+    // Split data and tag
+    const ciphertextLength = dataView.byteLength - tagByteLength;
+    processData = dataView.slice(0, ciphertextLength).buffer;
+    authTag = dataView.slice(ciphertextLength).buffer;
+
+    // Set auth tag for verification
+    cipher.setAuthTag(authTag);
+  } else {
+    processData = data;
+  }
+
+  // Set additional authenticated data if provided
+  if (additionalData) {
+    cipher.setAAD(bufferLikeToArrayBuffer(additionalData));
+  }
+
+  // Process data
+  const updated = cipher.update(processData);
+  const final = cipher.final();
+  if (mode === CipherOrWrapMode.kWebCryptoCipherEncrypt) {
+    // For encryption, append auth tag to result
+    const tag = cipher.getAuthTag();
+    const result = new Uint8Array(updated.byteLength + final.byteLength + tag.byteLength);
+    result.set(new Uint8Array(updated), 0);
+    result.set(new Uint8Array(final), updated.byteLength);
+    result.set(new Uint8Array(tag), updated.byteLength + final.byteLength);
+    return result.buffer;
+  } else {
+    // For decryption, just concatenate plaintext
+    const result = new Uint8Array(updated.byteLength + final.byteLength);
+    result.set(new Uint8Array(updated), 0);
+    result.set(new Uint8Array(final), updated.byteLength);
+    return result.buffer;
+  }
+}
 async function aesGenerateKey(algorithm, extractable, keyUsages) {
   const {
     length
@@ -554,7 +669,9 @@ function edImportKey(format, data, algorithm, extractable, keyUsages) {
   } = algorithm;
 
   // Validate usages
-  if (hasAnyNotIn(keyUsages, ['sign', 'verify'])) {
+  const isX = name === 'X25519' || name === 'X448';
+  const allowedUsages = isX ? ['deriveKey', 'deriveBits'] : ['sign', 'verify'];
+  if (hasAnyNotIn(keyUsages, allowedUsages)) {
     throw lazyDOMException(`Unsupported key usage for ${name} key`, 'SyntaxError');
   }
   let keyObject;
@@ -627,8 +744,12 @@ const exportKeySpki = async key => {
     case 'Ed25519':
     // Fall through
     case 'Ed448':
+    // Fall through
+    case 'X25519':
+    // Fall through
+    case 'X448':
       if (key.type === 'public') {
-        // Export Ed key in SPKI DER format
+        // Export Ed/X key in SPKI DER format
         return bufferLikeToArrayBuffer(key.keyObject.handle.exportKey(KFormatType.DER, KeyEncoding.SPKI));
       }
       break;
@@ -666,8 +787,12 @@ const exportKeyPkcs8 = async key => {
     case 'Ed25519':
     // Fall through
     case 'Ed448':
+    // Fall through
+    case 'X25519':
+    // Fall through
+    case 'X448':
       if (key.type === 'private') {
-        // Export Ed key in PKCS8 DER format
+        // Export Ed/X key in PKCS8 DER format
         return bufferLikeToArrayBuffer(key.keyObject.handle.exportKey(KFormatType.DER, KeyEncoding.PKCS8));
       }
       break;
@@ -693,6 +818,19 @@ const exportKeyRaw = key => {
         return ecExportKey(key, KWebCryptoKeyFormat.kWebCryptoKeyFormatRaw);
       }
       break;
+    case 'Ed25519':
+    // Fall through
+    case 'Ed448':
+    // Fall through
+    case 'X25519':
+    // Fall through
+    case 'X448':
+      if (key.type === 'public') {
+        // Export raw public key
+        const exported = key.keyObject.handle.exportKey();
+        return bufferLikeToArrayBuffer(exported);
+      }
+      break;
     case 'AES-CTR':
     // Fall through
     case 'AES-CBC':
@@ -701,6 +839,8 @@ const exportKeyRaw = key => {
     // Fall through
     case 'AES-KW':
     // Fall through
+    case 'ChaCha20-Poly1305':
+    // Fall through
     case 'HMAC':
       {
         const exported = key.keyObject.export();
@@ -740,6 +880,8 @@ const exportKeyJWK = key => {
     case 'AES-GCM':
     // Fall through
     case 'AES-KW':
+    // Fall through
+    case 'ChaCha20-Poly1305':
       if (key.algorithm.length === undefined) {
         throw lazyDOMException(`Algorithm ${key.algorithm.name} missing required length property`, 'InvalidAccessError');
       }
@@ -778,6 +920,25 @@ const importGenericSecretKey = async ({
   }
   throw new Error(`Unable to import ${name} key with format ${format}`);
 };
+const hkdfImportKey = async (format, keyData, algorithm, extractable, keyUsages) => {
+  const {
+    name
+  } = algorithm;
+  if (hasAnyNotIn(keyUsages, ['deriveKey', 'deriveBits'])) {
+    throw new Error(`Unsupported key usage for a ${name} key`);
+  }
+  switch (format) {
+    case 'raw':
+      {
+        const keyObject = createSecretKey(keyData);
+        return new CryptoKey(keyObject, {
+          name
+        }, keyUsages, extractable);
+      }
+    default:
+      throw new Error(`Unable to import ${name} key with format ${format}`);
+  }
+};
 const checkCryptoKeyPairUsages = pair => {
   if (pair.privateKey && pair.privateKey instanceof CryptoKey && pair.privateKey.keyUsages && pair.privateKey.keyUsages.length > 0) {
     return;
@@ -937,6 +1098,10 @@ const cipherOrWrap = async (mode, algorithm, key, data, op) => {
     // Fall through
     case 'AES-GCM':
       return aesCipher(mode, key, data, algorithm);
+    case 'AES-KW':
+      return aesKwCipher(mode, key, data);
+    case 'ChaCha20-Poly1305':
+      return chaCha20Poly1305Cipher(mode, key, data, algorithm);
   }
 };
 export class Subtle {
@@ -949,16 +1114,54 @@ export class Subtle {
     return asyncDigest(normalizedAlgorithm, data);
   }
   async deriveBits(algorithm, baseKey, length) {
-    if (!baseKey.keyUsages.includes('deriveBits')) {
-      throw new Error('baseKey does not have deriveBits usage');
+    // Allow either deriveBits OR deriveKey usage (WebCrypto spec allows both)
+    if (!baseKey.keyUsages.includes('deriveBits') && !baseKey.keyUsages.includes('deriveKey')) {
+      throw new Error('baseKey does not have deriveBits or deriveKey usage');
     }
     if (baseKey.algorithm.name !== algorithm.name) throw new Error('Key algorithm mismatch');
     switch (algorithm.name) {
       case 'PBKDF2':
         return pbkdf2DeriveBits(algorithm, baseKey, length);
+      case 'X25519':
+      // Fall through
+      case 'X448':
+        return xDeriveBits(algorithm, baseKey, length);
+      case 'HKDF':
+        return hkdfDeriveBits(algorithm, baseKey, length);
     }
     throw new Error(`'subtle.deriveBits()' for ${algorithm.name} is not implemented.`);
   }
+  async deriveKey(algorithm, baseKey, derivedKeyAlgorithm, extractable, keyUsages) {
+    // Validate baseKey usage
+    if (!baseKey.usages.includes('deriveKey') && !baseKey.usages.includes('deriveBits')) {
+      throw lazyDOMException('baseKey does not have deriveKey or deriveBits usage', 'InvalidAccessError');
+    }
+
+    // Calculate required key length
+    const length = getKeyLength(derivedKeyAlgorithm);
+
+    // Step 1: Derive bits
+    let derivedBits;
+    if (baseKey.algorithm.name !== algorithm.name) throw new Error('Key algorithm mismatch');
+    switch (algorithm.name) {
+      case 'PBKDF2':
+        derivedBits = await pbkdf2DeriveBits(algorithm, baseKey, length);
+        break;
+      case 'X25519':
+      // Fall through
+      case 'X448':
+        derivedBits = await xDeriveBits(algorithm, baseKey, length);
+        break;
+      case 'HKDF':
+        derivedBits = hkdfDeriveBits(algorithm, baseKey, length);
+        break;
+      default:
+        throw new Error(`'subtle.deriveKey()' for ${algorithm.name} is not implemented.`);
+    }
+
+    // Step 2: Import as key
+    return this.importKey('raw', derivedBits, derivedKeyAlgorithm, extractable, keyUsages);
+  }
   async encrypt(algorithm, key, data) {
     const normalizedAlgorithm = normalizeAlgorithm(algorithm, 'encrypt');
     return cipherOrWrap(CipherOrWrapMode.kWebCryptoCipherEncrypt, normalizedAlgorithm, key, bufferLikeToArrayBuffer(data), 'encrypt');
@@ -976,6 +1179,76 @@ export class Subtle {
         return exportKeyRaw(key);
     }
   }
+  async wrapKey(format, key, wrappingKey, wrapAlgorithm) {
+    // Validate wrappingKey usage
+    if (!wrappingKey.usages.includes('wrapKey')) {
+      throw lazyDOMException('wrappingKey does not have wrapKey usage', 'InvalidAccessError');
+    }
+
+    // Step 1: Export the key
+    const exported = await this.exportKey(format, key);
+
+    // Step 2: Convert to ArrayBuffer if JWK
+    let keyData;
+    if (format === 'jwk') {
+      const jwkString = JSON.stringify(exported);
+      const buffer = SBuffer.from(jwkString, 'utf8');
+
+      // For AES-KW, pad to multiple of 8 bytes (accounting for null terminator)
+      if (wrapAlgorithm.name === 'AES-KW') {
+        const length = buffer.length;
+        // Add 1 for null terminator, then pad to multiple of 8
+        const paddedLength = Math.ceil((length + 1) / 8) * 8;
+        const paddedBuffer = SBuffer.alloc(paddedLength);
+        buffer.copy(paddedBuffer);
+        // Null terminator for JSON string (remaining bytes are already zeros from alloc)
+        paddedBuffer.writeUInt8(0, length);
+        keyData = bufferLikeToArrayBuffer(paddedBuffer);
+      } else {
+        keyData = bufferLikeToArrayBuffer(buffer);
+      }
+    } else {
+      keyData = exported;
+    }
+
+    // Step 3: Encrypt the exported key
+    return cipherOrWrap(CipherOrWrapMode.kWebCryptoCipherEncrypt, wrapAlgorithm, wrappingKey, keyData, 'wrapKey');
+  }
+  async unwrapKey(format, wrappedKey, unwrappingKey, unwrapAlgorithm, unwrappedKeyAlgorithm, extractable, keyUsages) {
+    // Validate unwrappingKey usage
+    if (!unwrappingKey.usages.includes('unwrapKey')) {
+      throw lazyDOMException('unwrappingKey does not have unwrapKey usage', 'InvalidAccessError');
+    }
+
+    // Step 1: Decrypt the wrapped key
+    const decrypted = await cipherOrWrap(CipherOrWrapMode.kWebCryptoCipherDecrypt, unwrapAlgorithm, unwrappingKey, bufferLikeToArrayBuffer(wrappedKey), 'unwrapKey');
+
+    // Step 2: Convert to appropriate format
+    let keyData;
+    if (format === 'jwk') {
+      const buffer = SBuffer.from(decrypted);
+      // For AES-KW, the data may be padded - find the null terminator
+      let jwkString;
+      if (unwrapAlgorithm.name === 'AES-KW') {
+        // Find the null terminator (if present) to get the original string
+        const nullIndex = buffer.indexOf(0);
+        if (nullIndex !== -1) {
+          jwkString = buffer.toString('utf8', 0, nullIndex);
+        } else {
+          // No null terminator, try to parse the whole buffer
+          jwkString = buffer.toString('utf8').trim();
+        }
+      } else {
+        jwkString = buffer.toString('utf8');
+      }
+      keyData = JSON.parse(jwkString);
+    } else {
+      keyData = decrypted;
+    }
+
+    // Step 3: Import the key
+    return this.importKey(format, keyData, unwrappedKeyAlgorithm, extractable, keyUsages);
+  }
   async generateKey(algorithm, extractable, keyUsages) {
     algorithm = normalizeAlgorithm(algorithm, 'generateKey');
     let result;
@@ -1002,6 +1275,18 @@ export class Subtle {
       case 'AES-KW':
         result = await aesGenerateKey(algorithm, extractable, keyUsages);
         break;
+      case 'ChaCha20-Poly1305':
+        {
+          const length = algorithm.length ?? 256;
+          if (length !== 256) {
+            throw lazyDOMException('ChaCha20-Poly1305 only supports 256-bit keys', 'NotSupportedError');
+          }
+          result = await aesGenerateKey({
+            name: 'ChaCha20-Poly1305',
+            length: 256
+          }, extractable, keyUsages);
+          break;
+        }
       case 'HMAC':
         result = await hmacGenerateKey(algorithm, extractable, keyUsages);
         break;
@@ -1019,6 +1304,12 @@ export class Subtle {
         result = await mldsa_generateKeyPairWebCrypto(algorithm.name, extractable, keyUsages);
         checkCryptoKeyPairUsages(result);
         break;
+      case 'X25519':
+      // Fall through
+      case 'X448':
+        result = await x_generateKeyPairWebCrypto(algorithm.name.toLowerCase(), extractable, keyUsages);
+        checkCryptoKeyPairUsages(result);
+        break;
       default:
         throw new Error(`'subtle.generateKey()' is not implemented for ${algorithm.name}.
             Unrecognized algorithm name`);
@@ -1051,11 +1342,20 @@ export class Subtle {
       case 'AES-GCM':
       // Fall through
       case 'AES-KW':
+      // Fall through
+      case 'ChaCha20-Poly1305':
         result = await aesImportKey(normalizedAlgorithm, format, data, extractable, keyUsages);
         break;
       case 'PBKDF2':
         result = await importGenericSecretKey(normalizedAlgorithm, format, data, extractable, keyUsages);
         break;
+      case 'HKDF':
+        result = await hkdfImportKey(format, data, normalizedAlgorithm, extractable, keyUsages);
+        break;
+      case 'X25519':
+      // Fall through
+      case 'X448':
+      // Fall through
       case 'Ed25519':
       // Fall through
       case 'Ed448':
@@ -1077,11 +1377,104 @@ export class Subtle {
     return result;
   }
   async sign(algorithm, key, data) {
-    return signVerify(algorithm, key, data);
+    const normalizedAlgorithm = normalizeAlgorithm(algorithm, 'sign');
+    if (normalizedAlgorithm.name === 'HMAC') {
+      // Validate key usage
+      if (!key.usages.includes('sign')) {
+        throw lazyDOMException('Key does not have sign usage', 'InvalidAccessError');
+      }
+
+      // Get hash algorithm from key or algorithm params
+      // Hash can be either a string or an object with name property
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      const alg = normalizedAlgorithm;
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      const keyAlg = key.algorithm;
+      let hashAlgorithm = 'SHA-256';
+      if (typeof alg.hash === 'string') {
+        hashAlgorithm = alg.hash;
+      } else if (alg.hash?.name) {
+        hashAlgorithm = alg.hash.name;
+      } else if (typeof keyAlg.hash === 'string') {
+        hashAlgorithm = keyAlg.hash;
+      } else if (keyAlg.hash?.name) {
+        hashAlgorithm = keyAlg.hash.name;
+      }
+
+      // Create HMAC and sign
+      const keyData = key.keyObject.export();
+      const hmac = createHmac(hashAlgorithm, keyData);
+      hmac.update(bufferLikeToArrayBuffer(data));
+      return bufferLikeToArrayBuffer(hmac.digest());
+    }
+    return signVerify(normalizedAlgorithm, key, data);
   }
   async verify(algorithm, key, signature, data) {
-    return signVerify(algorithm, key, data, signature);
+    const normalizedAlgorithm = normalizeAlgorithm(algorithm, 'verify');
+    if (normalizedAlgorithm.name === 'HMAC') {
+      // Validate key usage
+      if (!key.usages.includes('verify')) {
+        throw lazyDOMException('Key does not have verify usage', 'InvalidAccessError');
+      }
+
+      // Get hash algorithm
+      // Hash can be either a string or an object with name property
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      const alg = normalizedAlgorithm;
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      const keyAlg = key.algorithm;
+      let hashAlgorithm = 'SHA-256';
+      if (typeof alg.hash === 'string') {
+        hashAlgorithm = alg.hash;
+      } else if (alg.hash?.name) {
+        hashAlgorithm = alg.hash.name;
+      } else if (typeof keyAlg.hash === 'string') {
+        hashAlgorithm = keyAlg.hash;
+      } else if (keyAlg.hash?.name) {
+        hashAlgorithm = keyAlg.hash.name;
+      }
+
+      // Create HMAC and compute expected signature
+      const keyData = key.keyObject.export();
+      const hmac = createHmac(hashAlgorithm, keyData);
+      const dataBuffer = bufferLikeToArrayBuffer(data);
+      hmac.update(dataBuffer);
+      const expectedDigest = hmac.digest();
+      const expected = new Uint8Array(bufferLikeToArrayBuffer(expectedDigest));
+
+      // Constant-time comparison
+      const signatureArray = new Uint8Array(bufferLikeToArrayBuffer(signature));
+      if (expected.length !== signatureArray.length) {
+        return false;
+      }
+
+      // Manual constant-time comparison
+      let result = 0;
+      for (let i = 0; i < expected.length; i++) {
+        result |= expected[i] ^ signatureArray[i];
+      }
+      return result === 0;
+    }
+    return signVerify(normalizedAlgorithm, key, data, signature);
   }
 }
 export const subtle = new Subtle();
+function getKeyLength(algorithm) {
+  const name = algorithm.name;
+  switch (name) {
+    case 'AES-CTR':
+    case 'AES-CBC':
+    case 'AES-GCM':
+    case 'AES-KW':
+    case 'ChaCha20-Poly1305':
+      return algorithm.length || 256;
+    case 'HMAC':
+      {
+        const hmacAlg = algorithm;
+        return hmacAlg.length || 256;
+      }
+    default:
+      throw lazyDOMException(`Cannot determine key length for ${name}`, 'NotSupportedError');
+  }
+}
 //# sourceMappingURL=subtle.js.map