@xiaoyuyu6420/dsh-backup 0.14.1 → 0.14.3

package/lib/encrypt.js

@@ -23,6 +23,10 @@ import {
   randomBytes,
   scrypt as scryptCb,
 } from 'node:crypto';
+import { createReadStream, createWriteStream } from 'node:fs';
+import { rm } from 'node:fs/promises';
+import { Transform } from 'node:stream';
+import { pipeline } from 'node:stream/promises';
 import { promisify } from 'node:util';
 
 const scrypt = promisify(scryptCb);
@@ -91,15 +95,10 @@ export async function encryptBuffer(plaintext, password) {
 }
 
 /**
- * 解密一段密文。口令错误 / 密文被篡改时 AES-GCM 认证失败，抛出口令错误人话报错
- * （调用方可直接展示）；未知版本 / 未知算法 / 边车字段缺失一律拒绝，不猜着解。
- *
- * @param {Buffer|Uint8Array} ciphertext 密文（归档文件内容）
- * @param {object} sidecar `.encrypted.json` 解析出的对象
- * @param {string} password 口令（内存态）
- * @returns {Promise<Buffer>} 明文（tar.gz 字节）
+ * 校验边车并派生出解开这段密文所需的 { key, nonce, authTag }。
+ * 版本/算法/字段缺失一律拒绝（宁可不恢复，也不拿不认识的参数猜着解）。
  */
-export async function decryptBuffer(ciphertext, sidecar, password) {
+async function unlock(sidecar, password) {
   if (typeof password !== 'string' || !password) {
     throw new Error('这份备份已加密，恢复需要口令（口令只经内存传入，绝不写盘）');
   }
@@ -120,16 +119,116 @@ export async function decryptBuffer(ciphertext, sidecar, password) {
     throw new Error('加密边车字段缺失（salt/nonce/authTag），无法解密');
   }
   const key = await deriveKey(password, { ...(sidecar.kdfParams || KDF_PARAMS), salt });
+  return { key, nonce, authTag };
+}
+
+/** GCM 认证失败的统一人话报错（不泄露具体原因）。 */
+function authFailed(err) {
+  if (/unable to authenticate|authentication/i.test(String(err && err.message ? err.message : err))) {
+    return new Error('口令错误或归档已损坏（AES-GCM 认证失败）——请核对口令后重试');
+  }
+  return err;
+}
+
+/**
+ * 解密一段密文（内存版，供小体积数据/测试用）。口令错误 / 密文被篡改时 AES-GCM
+ * 认证失败，抛出口令错误人话报错（调用方可直接展示）。
+ *
+ * @param {Buffer|Uint8Array} ciphertext 密文（归档文件内容）
+ * @param {object} sidecar `.encrypted.json` 解析出的对象
+ * @param {string} password 口令（内存态）
+ * @returns {Promise<Buffer>} 明文（tar.gz 字节）
+ */
+export async function decryptBuffer(ciphertext, sidecar, password) {
+  const { key, nonce, authTag } = await unlock(sidecar, password);
   const decipher = createDecipheriv(CIPHER, key, nonce);
   decipher.setAuthTag(authTag);
   const body = Buffer.isBuffer(ciphertext) ? ciphertext : Buffer.from(ciphertext);
   try {
     return Buffer.concat([decipher.update(body), decipher.final()]);
   } catch (err) {
-    // GCM final() 认证失败：口令错或密文被改。两条都指向同一句人话，不泄露具体原因。
-    if (/unable to authenticate|authentication/i.test(String(err && err.message ? err.message : err))) {
-      throw new Error('口令错误或归档已损坏（AES-GCM 认证失败）——请核对口令后重试');
-    }
+    throw authFailed(err);
+  }
+}
+
+/**
+ * 加密文件：**分块流式**，不整包入内存（归档可以远超 Buffer 上限）。
+ * 调用方负责把输出写到临时路径后自己 `rename` 覆盖（本函数只保证「要么完整
+ * 密文、要么失败且不留半截文件」）。
+ *
+ * @param {string} inPath 明文文件
+ * @param {string} outPath 密文输出文件
+ * @param {string} password 口令
+ * @returns {Promise<{ sidecar: object, ciphertextSize: number }>}
+ */
+export async function encryptFile(inPath, outPath, password) {
+  if (typeof password !== 'string' || !password) {
+    throw new Error('加密需要非空口令（口令只经内存传入，绝不写盘）');
+  }
+  const salt = randomBytes(SALT_BYTES);
+  const nonce = randomBytes(NONCE_BYTES);
+  const key = await deriveKey(password, { ...KDF_PARAMS, salt });
+  const cipher = createCipheriv(CIPHER, key, nonce);
+  let ciphertextSize = 0;
+  const counter = new Transform({
+    transform(chunk, _enc, done) {
+      ciphertextSize += chunk.length;
+      done(null, chunk);
+    },
+  });
+  try {
+    await pipeline(
+      createReadStream(inPath),
+      cipher,
+      counter,
+      createWriteStream(outPath, { mode: 0o600 }),
+    );
+  } catch (err) {
+    await rm(outPath, { force: true }).catch(() => {});
     throw err;
   }
+  const authTag = cipher.getAuthTag();
+  return {
+    ciphertextSize,
+    sidecar: {
+      version: ENCRYPT_VERSION,
+      cipher: CIPHER,
+      kdf: KDF_PARAMS.algorithm,
+      kdfParams: {
+        N: KDF_PARAMS.N,
+        r: KDF_PARAMS.r,
+        p: KDF_PARAMS.p,
+        keyLen: KDF_PARAMS.keyLen,
+      },
+      salt: salt.toString('base64'),
+      nonce: nonce.toString('base64'),
+      authTag: authTag.toString('base64'),
+      ciphertextSize,
+    },
+  };
+}
+
+/**
+ * 解密文件：**分块流式**。认证通过才产出完整明文；口令错/密文被改时删除半截
+ * 输出并抛人话报错（绝不留下「解了一半」的明文）。
+ *
+ * @param {string} inPath 密文文件
+ * @param {string} outPath 明文输出文件（可能不完整；失败时已被删除）
+ * @param {object} sidecar `.encrypted.json` 内容
+ * @param {string} password 口令
+ */
+export async function decryptFile(inPath, outPath, sidecar, password) {
+  const { key, nonce, authTag } = await unlock(sidecar, password);
+  const decipher = createDecipheriv(CIPHER, key, nonce);
+  decipher.setAuthTag(authTag);
+  try {
+    await pipeline(
+      createReadStream(inPath),
+      decipher,
+      createWriteStream(outPath, { mode: 0o600 }),
+    );
+  } catch (err) {
+    await rm(outPath, { force: true }).catch(() => {});
+    throw authFailed(err);
+  }
 }