sm-crypto-wasm 0.1.2 → 0.2.1

package/dist/sm2/index.d.ts

@@ -70,14 +70,54 @@ export interface Sm2DecryptOptions {
  * @returns PKCS#8 私钥 PEM 与 SPKI 公钥 PEM。
  */
 export declare function generateKeyPair(): Promise<Sm2KeyPair>;
+/**
+ * 池中的一个预计算签名点。
+ *
+ * `point`（`k·G` 的坐标）是公开信息；**`k` 被封在模块内部**，外部读不到，被
+ * `sign()` 取走后立刻从索引中摘除并抹掉字节 —— 所以"复用同一个 `k`"在 API 层面
+ * 就不可能发生，调用方不需要（也无法）自己去保证这一点。
+ */
+export declare class SignaturePoint {
+    /** `k·G` 的仿射坐标 `x ‖ y`（64 字节 hex）。公开信息，可用于审计。 */
+    readonly point: string;
+    /** 只允许本模块的 `createSignaturePointPool` 构造。 */
+    private constructor();
+    /** @internal 仅供 `createSignaturePointPool` 使用。 */
+    static _create(nonce: Uint8Array, point: string): SignaturePoint;
+}
+/** `sign` 的选项。 */
+export interface SignOptions {
+    /**
+     * 预计算的签名点池（见 `createSignaturePointPool`）。
+     *
+     * 命中时省掉本次签名的固定基点标量乘（实测占单次签名的 56%）；池被取空后
+     * 自动回退到现算。**该数组会被原地消费**（每签一次 `pop()` 一个点），
+     * 调用方无需自己维护游标。
+     */
+    pointPool?: SignaturePoint[];
+}
+/**
+ * 预计算 `n` 个签名点，供 `sign()` 的 `pointPool` 选项消费。
+ *
+ * 每个点 = 一次 `k·G`（本机约 130 µs）。批量签名时把这些固定基点标量乘提前
+ * 摊销，签名的关键路径只剩 `e = SM3(Z ‖ M)` 与两条模运算 —— 实测快 5.9×。
+ *
+ * `k` 取自宿主 CSPRNG，并由本模块独占保管：调用方拿不到它，也就无从误用成复用。
+ * 唯一的约定是"同一个点只喂一次"，而真喂两次会直接抛错。
+ *
+ * @param n 生成个数。
+ * @returns `n` 个签名点。
+ */
+export declare function createSignaturePointPool(n: number): Promise<SignaturePoint[]>;
 /**
  * SM2 签名（默认 SM3withSM2、默认用户标识）。
  *
  * @param privateKey 32 字节私钥。
  * @param message 待签名消息。
+ * @param opts `pointPool`：预计算的签名点池（见 `createSignaturePointPool`）。
  * @returns 64 字节签名（r || s）的 hex。
  */
-export declare function sign(privateKey: string | Uint8Array, message: string | Uint8Array): string;
+export declare function sign(privateKey: string | Uint8Array, message: string | Uint8Array, opts?: SignOptions): string;
 /**
  * SM2 验签。
  *

package/dist/sm2/index.js

@@ -215,34 +215,122 @@ function randomK() {
     }
     return out;
 }
+/**
+ * 池中的一个预计算签名点。
+ *
+ * `point`（`k·G` 的坐标）是公开信息；**`k` 被封在模块内部**，外部读不到，被
+ * `sign()` 取走后立刻从索引中摘除并抹掉字节 —— 所以"复用同一个 `k`"在 API 层面
+ * 就不可能发生，调用方不需要（也无法）自己去保证这一点。
+ */
+export class SignaturePoint {
+    /** `k·G` 的仿射坐标 `x ‖ y`（64 字节 hex）。公开信息，可用于审计。 */
+    point;
+    /** 只允许本模块的 `createSignaturePointPool` 构造。 */
+    constructor(point) {
+        this.point = point;
+    }
+    /** @internal 仅供 `createSignaturePointPool` 使用。 */
+    static _create(nonce, point) {
+        const p = new SignaturePoint(point);
+        POINT_NONCE.set(p, nonce);
+        return p;
+    }
+}
+/**
+ * 点位 → 一次性 `k`。取走即 `delete`：同一个点位对象无论被谁持有引用，都只能
+ * 取到一次 `k`。用 `WeakMap` 而非对象属性，是为了连"枚举键名"都做不到。
+ */
+const POINT_NONCE = new WeakMap();
+/**
+ * 取走签名点的一次性 `k`。
+ *
+ * 索引里没有（已被取过，或不是本库产出的伪造对象）就抛错 —— 于是"同一个点签两次"
+ * 会当场失败，而不是悄悄复用 `k` 把私钥送出去。
+ */
+function takeNonce(p) {
+    const nonce = POINT_NONCE.get(p);
+    if (nonce === undefined) {
+        throw new Error('sm2: 该签名点不可用（已被使用过，或不是 createSignaturePointPool 产出的）；签名点是一次性的。');
+    }
+    POINT_NONCE.delete(p);
+    return nonce;
+}
+/**
+ * 预计算 `n` 个签名点，供 `sign()` 的 `pointPool` 选项消费。
+ *
+ * 每个点 = 一次 `k·G`（本机约 130 µs）。批量签名时把这些固定基点标量乘提前
+ * 摊销，签名的关键路径只剩 `e = SM3(Z ‖ M)` 与两条模运算 —— 实测快 5.9×。
+ *
+ * `k` 取自宿主 CSPRNG，并由本模块独占保管：调用方拿不到它，也就无从误用成复用。
+ * 唯一的约定是"同一个点只喂一次"，而真喂两次会直接抛错。
+ *
+ * @param n 生成个数。
+ * @returns `n` 个签名点。
+ */
+export async function createSignaturePointPool(n) {
+    if (!Number.isInteger(n) || n < 0) {
+        throw new RangeError(`sm2: 签名点池大小必须是非负整数，收到 ${String(n)}`);
+    }
+    await ensureLoaded();
+    const { exports, memory } = getModule();
+    const pool = [];
+    for (let i = 0; i < n; i++) {
+        const nonce = randomK();
+        const kPtr = exports.alloc(32);
+        const ptPtr = exports.alloc(64);
+        try {
+            new Uint8Array(memory.buffer, kPtr, 32).set(nonce);
+            exports.scalarMulGOut(kPtr, ptPtr);
+            pool.push(SignaturePoint._create(nonce, fromBytes(new Uint8Array(memory.buffer, ptPtr, 64).slice(), 'hex')));
+        }
+        finally {
+            exports.freeAll();
+        }
+    }
+    return pool;
+}
 /**
  * SM2 签名（默认 SM3withSM2、默认用户标识）。
  *
  * @param privateKey 32 字节私钥。
  * @param message 待签名消息。
+ * @param opts `pointPool`：预计算的签名点池（见 `createSignaturePointPool`）。
  * @returns 64 字节签名（r || s）的 hex。
  */
-export function sign(privateKey, message) {
+export function sign(privateKey, message, opts = {}) {
     const d = toPrivateKeyBytes(privateKey);
     const m = toBytes(message, 'message');
     const pub = derivePublicKeyBytes(d);
-    const { exports } = getModule();
+    const { exports, memory } = getModule();
     const dp = exports.alloc(32);
     const pxp = exports.alloc(32);
     const pyp = exports.alloc(32);
     const mp = exports.alloc(m.length);
     const kp = exports.alloc(32);
+    const xp = exports.alloc(32);
     const op = exports.alloc(64);
+    const pool = opts.pointPool;
     try {
         putBytes(d, dp);
         putBytes(pub.subarray(0, 32), pxp);
         putBytes(pub.subarray(32, 64), pyp);
         putBytes(m, mp);
-        // 最多重试几次 k（极小概率 r/s 为 0 或 r+k==n）。
+        // 最多重试几次 k（极小概率 r/s 为 0 或 r+k==n）。池命中时每轮换一个点。
         for (let attempt = 0; attempt < 16; attempt++) {
-            const k = randomK();
-            putBytes(k, kp);
-            const rc = exports.signOut(dp, pxp, pyp, mp, m.length, kp, op);
+            const pooled = pool?.pop();
+            let rc;
+            if (pooled !== undefined) {
+                // 一次性取走 k；此处之后该点不可再用（再喂会抛错）。
+                const nonce = takeNonce(pooled);
+                putBytes(nonce, kp);
+                nonce.fill(0); // JS 侧那份随即抹掉
+                new Uint8Array(memory.buffer, xp, 32).set(toBytes(pooled.point, 'point').subarray(0, 32));
+                rc = exports.signWithPointOut(dp, pxp, pyp, mp, m.length, kp, xp, op);
+            }
+            else {
+                putBytes(randomK(), kp);
+                rc = exports.signOut(dp, pxp, pyp, mp, m.length, kp, op);
+            }
             if (rc === 0) {
                 return fromBytes(readBytes(op, 64), 'hex');
             }
@@ -250,6 +338,8 @@ export function sign(privateKey, message) {
         throw new Error('sm2.sign: failed to produce a valid signature after retries');
     }
     finally {
+        // 抹掉 WASM 内存里那份 k（缓冲区在 freeAll 前仍可写）
+        new Uint8Array(memory.buffer, kp, 32).fill(0);
         exports.freeAll();
     }
 }

package/dist/sm3/index.d.ts

@@ -7,6 +7,9 @@
  * 3. 拷入输入、调 `instance.exports.sm3(inputPtr, inputLen, outPtr)`。
  * 4. 从 outPtr 拷出 32 字节。
  *
+ * HMAC 的 ipad/opad 拼接与两次摘要调用都在 WASM 内完成（`exports.sm3Hmac`），
+ * 这一层只做参数归一化与缓冲区搬运。
+ *
  * spec/plan 对应：plan Task 7；spec §6 API、§9 安全姿态。
  */
 /**
@@ -22,7 +25,7 @@ export declare function ensureLoaded(): Promise<void>;
  */
 export declare function hash(input: string | Uint8Array): string;
 /**
- * HMAC-SM3（同步）。
+ * HMAC-SM3（同步，RFC 2104）。
  *
  * @param key 任意长度密钥。
  * @param msg 任意长度消息。

package/dist/sm3/index.js

@@ -7,6 +7,9 @@
  * 3. 拷入输入、调 `instance.exports.sm3(inputPtr, inputLen, outPtr)`。
  * 4. 从 outPtr 拷出 32 字节。
  *
+ * HMAC 的 ipad/opad 拼接与两次摘要调用都在 WASM 内完成（`exports.sm3Hmac`），
+ * 这一层只做参数归一化与缓冲区搬运。
+ *
  * spec/plan 对应：plan Task 7；spec §6 API、§9 安全姿态。
  */
 import { fromBytes, toBytes } from '../runtime/convert.js';
@@ -42,6 +45,10 @@ function digestInput(input, label) {
         return new Uint8Array(0);
     return toBytes(input, label);
 }
+/** `Uint8Array` 直接使用：紧接着就会 `set` 进 WASM 私有缓冲，再复制一份是纯浪费。 */
+function inputBytes(input, label) {
+    return input instanceof Uint8Array ? input : digestInput(input, label);
+}
 /**
  * SM3 摘要（同步）。
  *
@@ -49,7 +56,7 @@ function digestInput(input, label) {
  * @returns 32 字节摘要的 hex 字符串（64 字符）。
  */
 export function hash(input) {
-    const bytes = digestInput(input, 'msg');
+    const bytes = inputBytes(input, 'msg');
     const { exports, memory } = getModule();
     const inputLen = bytes.length;
     // 从 WASM 堆分配缓冲区；**不可**用固定地址，否则覆盖 AS runtime 数据。
@@ -66,30 +73,30 @@ export function hash(input) {
     }
 }
 /**
- * HMAC-SM3（同步）。
+ * HMAC-SM3（同步，RFC 2104）。
  *
  * @param key 任意长度密钥。
  * @param msg 任意长度消息。
  * @returns 32 字节摘要的 hex 字符串。
  */
 export function hmac(key, msg) {
-    const k = digestInput(key, 'key');
-    const m = digestInput(msg, 'msg');
-    const blockKey = k.length > 64 ? toBytes(hash(k), 'key64') : k;
-    const padded = new Uint8Array(64);
-    padded.set(blockKey);
-    const innerKey = new Uint8Array(64);
-    for (let i = 0; i < 64; i++)
-        innerKey[i] = (padded[i] ?? 0) ^ 0x36;
-    const outerKey = new Uint8Array(64);
-    for (let i = 0; i < 64; i++)
-        outerKey[i] = (padded[i] ?? 0) ^ 0x5c;
-    const innerMsg = new Uint8Array(64 + m.length);
-    innerMsg.set(innerKey, 0);
-    innerMsg.set(m, 64);
-    const inner = toBytes(hash(innerMsg), 'inner');
-    const outerMsg = new Uint8Array(64 + 32);
-    outerMsg.set(outerKey, 0);
-    outerMsg.set(inner, 64);
-    return hash(outerMsg);
+    const k = inputBytes(key, 'key');
+    const m = inputBytes(msg, 'msg');
+    const { exports, memory } = getModule();
+    // 空 key / 空 msg 也要占 1 字节：`alloc(0)` 返回的地址不保证可写。
+    const keyPtr = exports.alloc(k.length === 0 ? 1 : k.length);
+    const msgPtr = exports.alloc(m.length === 0 ? 1 : m.length);
+    const outPtr = exports.alloc(32);
+    try {
+        if (k.length > 0)
+            new Uint8Array(memory.buffer, keyPtr, k.length).set(k);
+        if (m.length > 0)
+            new Uint8Array(memory.buffer, msgPtr, m.length).set(m);
+        exports.sm3Hmac(keyPtr, k.length, msgPtr, m.length, outPtr);
+        const out = new Uint8Array(memory.buffer, outPtr, 32).slice();
+        return fromBytes(out, 'hex');
+    }
+    finally {
+        exports.freeAll();
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sm-crypto-wasm",
-  "version": "0.1.2",
+  "version": "0.2.1",
   "description": "国密 SM2/SM3/SM4/SM9/ZUC 的 WASM 实现，Node 18+ 与现代浏览器均可使用，按算法分包懒加载。",
   "description:en": "SM2/SM3/SM4/SM9/ZUC (China crypto) implemented in AssemblyScript -> WASM. Node 18+ & modern browsers, per-algorithm lazy loading.",
   "license": "Apache-2.0",
@@ -92,6 +92,7 @@
     "format": "biome format --write src tests bench",
     "typecheck": "tsc --noEmit",
     "bench": "node --import tsx bench/run.ts",
+    "bench:cubelrti": "node --import tsx bench/compare-cubelrti.bench.ts",
     "docs:dev": "pnpm build && vitepress dev website",
     "docs:build": "pnpm build && vitepress build website",
     "docs:preview": "vitepress preview website",
@@ -106,6 +107,7 @@
     "happy-dom": "21.0.1",
     "mitata": "^1.0.34",
     "release-it": "^21.1.1",
+    "sm-crypto-v2": "1.15.1",
     "tsx": "^4.19.2",
     "typescript": "^7.0.2",
     "vitepress": "^1.6.4",