@onekeyfe/hd-core 1.2.2-alpha.0 → 1.2.2-alpha.10

package/__tests__/pro2HostAssetPackage.test.ts

@@ -1,3 +1,4 @@
+/* eslint-disable no-bitwise -- LZ4 decoding and deterministic test data generation require bitwise operations. */
 import { sha3_512 } from '@noble/hashes/sha3';
 
 import {
@@ -5,6 +6,82 @@ import {
   supportsPro2HostAssetPackage,
 } from '../src/utils/pro2HostAssetPackage';
 
+const decodeRawLz4Block = (compressed: Uint8Array, expectedLength: number) => {
+  const output = new Uint8Array(expectedLength);
+  let inputOffset = 0;
+  let outputOffset = 0;
+
+  const readLength = (initialLength: number) => {
+    let length = initialLength;
+    if (length === 15) {
+      let extension = 255;
+      while (extension === 255) {
+        extension = compressed[inputOffset];
+        inputOffset += 1;
+        length += extension;
+      }
+    }
+    return length;
+  };
+
+  while (inputOffset < compressed.byteLength) {
+    const token = compressed[inputOffset];
+    inputOffset += 1;
+    const literalLength = readLength(token >>> 4);
+    output.set(compressed.subarray(inputOffset, inputOffset + literalLength), outputOffset);
+    inputOffset += literalLength;
+    outputOffset += literalLength;
+    if (inputOffset >= compressed.byteLength) break;
+
+    const matchOffset = compressed[inputOffset] | (compressed[inputOffset + 1] << 8);
+    inputOffset += 2;
+    const matchLength = readLength(token & 0x0f) + 4;
+    for (let index = 0; index < matchLength; index += 1) {
+      output[outputOffset] = output[outputOffset - matchOffset];
+      outputOffset += 1;
+    }
+  }
+
+  expect(outputOffset).toBe(expectedLength);
+  return output;
+};
+
+const decodeFirstPackageEntry = (packageData: Uint8Array, rawLength: number) => {
+  const containerHeaderSize = 0x5f90;
+  const archive = packageData.subarray(containerHeaderSize);
+  const archiveView = new DataView(archive.buffer, archive.byteOffset, archive.byteLength);
+  const compressedOffset = archiveView.getUint32(42 + 0x100, true);
+  const compressed = archive.subarray(compressedOffset);
+  const compressedView = new DataView(
+    compressed.buffer,
+    compressed.byteOffset,
+    compressed.byteLength
+  );
+  const blockCount = compressedView.getUint16(0, true);
+  const blockSize = 1 << compressedView.getUint16(2, true);
+  let blockOffset = 8 + blockCount * 4;
+  const decodedBlocks: Uint8Array[] = [];
+
+  for (let index = 0; index < blockCount; index += 1) {
+    const compressedLength = compressedView.getUint32(8 + index * 4, true);
+    const expectedLength = Math.min(blockSize, rawLength - index * blockSize);
+    decodedBlocks.push(
+      decodeRawLz4Block(
+        compressed.subarray(blockOffset, blockOffset + compressedLength),
+        expectedLength
+      )
+    );
+    blockOffset += compressedLength;
+  }
+
+  const decoded = new Uint8Array(rawLength);
+  decodedBlocks.reduce((offset, block) => {
+    decoded.set(block, offset);
+    return offset + block.byteLength;
+  }, 0);
+  return decoded;
+};
+
 describe('Pro2 host asset package', () => {
   test('builds the unsigned RESOURCE container and LZ4-blocked archive expected by firmware', () => {
     const raw = new TextEncoder().encode('123456789');
@@ -40,12 +117,42 @@ describe('Pro2 host asset package', () => {
 
     const compressedOffset = archiveView.getUint32(42 + 0x100, true);
     expect(archiveView.getUint16(compressedOffset, true)).toBe(1);
-    expect(archiveView.getUint16(compressedOffset + 2, true)).toBe(12);
+    expect(archiveView.getUint16(compressedOffset + 2, true)).toBe(14);
     expect(archiveView.getUint32(compressedOffset + 4, true)).toBe(0);
     expect(archiveView.getUint32(compressedOffset + 8, true)).toBe(10);
     expect(payload.subarray(compressedOffset + 12)).toEqual(new Uint8Array([0x90, ...raw]));
   });
 
+  test('round-trips multi-block data byte-for-byte with best-match compression', () => {
+    const raw = Uint8Array.from({ length: 16_384 * 3 + 137 }, (_, index) => {
+      const column = index % 604;
+      const row = Math.floor(index / 604);
+      return (column * 31 + row * 17) & 0xff;
+    });
+
+    const packageData = buildPro2HostAssetPackage([{ name: 'wallpaper.bin', data: raw }]);
+
+    expect(decodeFirstPackageEntry(packageData, raw.byteLength)).toEqual(raw);
+  });
+
+  test('falls back to 8 KiB blocks when a compressed 16 KiB block exceeds firmware capacity', () => {
+    let state = 0x12345678;
+    const raw = Uint8Array.from({ length: 16_384 }, () => {
+      state ^= state << 13;
+      state ^= state >>> 17;
+      state ^= state << 5;
+      return state & 0xff;
+    });
+
+    const packageData = buildPro2HostAssetPackage([{ name: 'wallpaper.bin', data: raw }]);
+    const archive = packageData.subarray(0x5f90);
+    const archiveView = new DataView(archive.buffer, archive.byteOffset, archive.byteLength);
+    const compressedOffset = archiveView.getUint32(42 + 0x100, true);
+
+    expect(archiveView.getUint16(compressedOffset + 2, true)).toBe(13);
+    expect(decodeFirstPackageEntry(packageData, raw.byteLength)).toEqual(raw);
+  });
+
   test.each([
     ['1.0.0', false],
     ['1.0.1-beta.1', false],

package/__tests__/protocol-v2.test.ts

@@ -7,6 +7,7 @@ import { encode as encodeJpeg } from 'jpeg-js';
 import {
   DeviceRebootType,
   DeviceSessionPinType,
+  DeviceSessionSeedDomain,
   DeviceSettingsPage,
   DeviceType,
 } from '@onekeyfe/hd-transport';
@@ -255,6 +256,9 @@ describe('DeviceUploadWallpaper', () => {
   });
 
   test('uploads and applies the fixed wallpaper package on firmware 1.0.1', async () => {
+    const getSettingsSpy = jest
+      .spyOn(DataManager, 'getSettings')
+      .mockReturnValue('react-native' as any);
     const typedCall = jest.fn().mockImplementation((request, _response, params) => {
       if (request === 'FilesystemDirMake') return { message: {} };
       if (request === 'FilesystemFileWrite') {
@@ -271,17 +275,24 @@ describe('DeviceUploadWallpaper', () => {
       payload: {
         method: 'deviceUploadWallpaper',
         jpegBase64: createJpegBase64(604, 1024),
+        fileName: 'ignored-on-package-firmware.bin',
       },
     });
     const device = stubWallpaperDevice({
       commands: { typedCall },
+      state: { versions: { firmware: '1.0.1' } },
       getCurrentFirmwareVersionString: jest.fn(() => '1.0.1'),
     });
     (method as any).device = device;
     method.postMessage = jest.fn();
 
-    method.init();
-    const result = await method.run();
+    let result;
+    try {
+      method.init();
+      result = await method.run();
+    } finally {
+      getSettingsSpy.mockRestore();
+    }
 
     const fileWrites = typedCall.mock.calls.filter(call => call[0] === 'FilesystemFileWrite');
     expect(new Set(fileWrites.map(call => call[2].file.path))).toEqual(
@@ -290,6 +301,7 @@ describe('DeviceUploadWallpaper', () => {
     expect(fileWrites[0][2].file.data.subarray(0, 4)).toEqual(
       new Uint8Array([0x4f, 0x4b, 0x50, 0x50])
     );
+    expect(fileWrites[0][2].file.data).toHaveLength(1960);
     expect(typedCall).toHaveBeenLastCalledWith('DeviceSettingsSet', 'Success', {
       settings: { wallpaper_path: 'vol1:/wallpapers/wallpaper.okpkg' },
     });
@@ -300,6 +312,40 @@ describe('DeviceUploadWallpaper', () => {
     });
   });
 
+  test('keeps the legacy wallpaper path on a 1.0.1 prerelease', async () => {
+    const typedCall = jest.fn().mockImplementation((request, _response, params) => {
+      if (request === 'FilesystemDirMake') return { message: {} };
+      if (request === 'FilesystemFileWrite') {
+        const file = params.file as { data: Uint8Array; offset: number };
+        return { message: { processed_byte: file.offset + file.data.byteLength } };
+      }
+      if (request === 'DeviceSettingsSet') return { message: {} };
+      throw new Error(`Unexpected request: ${request}`);
+    });
+    const method = new DeviceUploadWallpaper({
+      id: 1,
+      payload: {
+        method: 'deviceUploadWallpaper',
+        jpegBase64: createJpegBase64(604, 1024),
+        fileName: 'prerelease-wallpaper.bin',
+      },
+    });
+    const device = stubWallpaperDevice({
+      commands: { typedCall },
+      state: { versions: { firmware: '1.0.1-beta.1' } },
+    });
+    (method as any).device = device;
+
+    method.init();
+    const result = await method.run();
+
+    expect(result.path).toBe('vol1:/wallpapers/prerelease-wallpaper.bin');
+    const fileWrites = typedCall.mock.calls.filter(call => call[0] === 'FilesystemFileWrite');
+    expect(new Set(fileWrites.map(call => call[2].file.path))).toEqual(
+      new Set(['vol1:/wallpapers/prerelease-wallpaper.bin'])
+    );
+  });
+
   test('文件上传失败时不修改 wallpaper_path', async () => {
     const typedCall = jest.fn().mockImplementation(request => {
       if (request === 'FilesystemDirMake') return { message: {} };
@@ -1079,10 +1125,9 @@ describe('Protocol V2 feature adapter', () => {
     expect(typedCall).toHaveBeenCalledWith('DeviceSessionAskPassphrase', 'Success', {
       passphrase: '',
       on_device: false,
+      seed_domains: [DeviceSessionSeedDomain.SeedDomain_Standard],
     });
-    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
-      seed_domains: [],
-    });
+    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {});
     device.passphraseState = 'state-1';
     expect(device.getInternalState()).toBe('session-1');
   });
@@ -1135,13 +1180,10 @@ describe('Protocol V2 feature adapter', () => {
     expect(
       typedCall.mock.calls.filter(call => call[0] === 'DeviceSessionAskPassphrase')
     ).toHaveLength(1);
-    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
-      seed_domains: [],
-    });
+    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {});
     expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
       session_id: 'standard-session',
       btc_test_address: 'standard-state',
-      seed_domains: [],
     });
     expect(deviceWalletSessionStore.getStandard(deviceId)).toEqual({
       passphraseState: 'standard-state',
@@ -1506,10 +1548,9 @@ describe('Protocol V2 feature adapter', () => {
     expect(typedCall).toHaveBeenCalledWith('DeviceSessionAskPassphrase', 'Success', {
       passphrase: 'host hidden wallet',
       on_device: false,
+      seed_domains: [DeviceSessionSeedDomain.SeedDomain_Standard],
     });
-    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
-      seed_domains: [],
-    });
+    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {});
   });
 
   test('deviceStatusGet returns raw DeviceStatus and updates dynamic features', async () => {
@@ -1618,7 +1659,6 @@ describe('Protocol V2 feature adapter', () => {
     expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
       session_id: 'session-a',
       btc_test_address: 'state-a',
-      seed_domains: [],
     });
     expect(device.getInternalState()).toBe('session-b');
   });
@@ -1692,7 +1732,6 @@ describe('Protocol V2 feature adapter', () => {
         {
           session_id: 'session-a',
           btc_test_address: 'state-a',
-          seed_domains: [],
         },
       ],
     ]);
@@ -1739,10 +1778,14 @@ describe('Protocol V2 feature adapter', () => {
       [
         'DeviceSessionAskPassphrase',
         'Success',
-        { passphrase: 'host hidden wallet', on_device: false },
+        {
+          passphrase: 'host hidden wallet',
+          on_device: false,
+          seed_domains: [DeviceSessionSeedDomain.SeedDomain_Standard],
+        },
       ],
       ['DeviceStatusGet', 'DeviceStatus', {}],
-      ['DeviceSessionGet', 'DeviceSession', { seed_domains: [] }],
+      ['DeviceSessionGet', 'DeviceSession', {}],
     ]);
   });
 
@@ -1904,7 +1947,6 @@ describe('Protocol V2 feature adapter', () => {
       expect(promptPassphrase).not.toHaveBeenCalled();
       expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
         btc_test_address: 'expected-state',
-        seed_domains: [],
       });
       expect(typedCall).toHaveBeenCalledWith('LockDevice', 'Success', {});
       expect(typedCall.mock.calls.filter(call => call[0] === 'DeviceSessionGet')).toHaveLength(1);
@@ -2202,7 +2244,7 @@ describe('Protocol V2 feature adapter', () => {
       firmwareVersion: '4.15.0',
       passphraseProtection: true,
       sessionId: 'feature-session',
-      unlockedAttachPin: true,
+      unlockedAttachPin: false,
     };
     const typedCall = jest
       .fn()
@@ -2295,7 +2337,6 @@ describe('Protocol V2 feature adapter', () => {
     ).resolves.toMatchObject({ passphraseState: 'expected-state' });
     expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
       btc_test_address: 'expected-state',
-      seed_domains: [],
     });
     expect(promptPassphrase).not.toHaveBeenCalled();
   });
@@ -2369,7 +2410,6 @@ describe('Protocol V2 feature adapter', () => {
         {
           session_id: 'session-pro2-app',
           btc_test_address: 'state-pro2-app',
-          seed_domains: [],
         },
       ],
     ]);
@@ -2559,10 +2599,9 @@ describe('Protocol V2 feature adapter', () => {
     expect(typedCall).toHaveBeenNthCalledWith(2, 'DeviceSessionAskPassphrase', 'Success', {
       passphrase: 'host hidden wallet',
       on_device: false,
+      seed_domains: [DeviceSessionSeedDomain.SeedDomain_Standard],
     });
-    expect(typedCall).toHaveBeenLastCalledWith('DeviceSessionGet', 'DeviceSession', {
-      seed_domains: [],
-    });
+    expect(typedCall).toHaveBeenLastCalledWith('DeviceSessionGet', 'DeviceSession', {});
   });
 
   test('does not mark Pro2 passphrase enabled from a main PIN session alone', async () => {
@@ -2642,16 +2681,14 @@ describe('Protocol V2 feature adapter', () => {
     expect(device.getInternalState()).toBeUndefined();
     expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
       btc_test_address: 'expected-state',
-      seed_domains: [],
     });
     expect(typedCall).toHaveBeenCalledWith('DeviceSessionAskPassphrase', 'Success', {
       passphrase: 'host hidden wallet',
       on_device: false,
+      seed_domains: [DeviceSessionSeedDomain.SeedDomain_Standard],
     });
     expect(typedCall).toHaveBeenCalledWith('DeviceStatusGet', 'DeviceStatus', {});
-    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
-      seed_domains: [],
-    });
+    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {});
   });
 
   test('fails closed instead of switching to Main PIN during a standard-wallet safety check', async () => {
@@ -2749,20 +2786,19 @@ describe('Protocol V2 feature adapter', () => {
     await expect(
       device.checkPassphraseStateSafety('stale-hidden-state', true, false)
     ).resolves.toBe(true);
-    expect(typedCall).toHaveBeenCalledTimes(5);
+    expect(typedCall).toHaveBeenCalledTimes(6);
     expect(typedCall).toHaveBeenCalledWith('ProtocolInfoRequest', 'ProtocolInfo', {
       eventless_wallet_session: true,
     });
-    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {
-      seed_domains: [],
-    });
+    expect(typedCall).toHaveBeenCalledWith('DeviceSessionGet', 'DeviceSession', {});
     expect(typedCall).toHaveBeenCalledWith('DeviceSessionAskPassphrase', 'Success', {
       passphrase: '',
       on_device: false,
+      seed_domains: [DeviceSessionSeedDomain.SeedDomain_Standard],
     });
     expect(typedCall).toHaveBeenCalledWith('DeviceStatusGet', 'DeviceStatus', {});
     expect(typedCall.mock.calls.filter(([request]) => request === 'DeviceStatusGet')).toHaveLength(
-      2
+      3
     );
     expect(typedCall).not.toHaveBeenCalledWith('DeviceSessionAskPin', 'Success', expect.anything());
   });
@@ -5627,7 +5663,7 @@ describe('Protocol V2 firmware update targets', () => {
     expect(method.postTipMessage).not.toHaveBeenCalled();
     expect(method.postProgressMessage).not.toHaveBeenCalled();
     await cancelableAction?.();
-    expect(cancelDevice).toHaveBeenCalledTimes(1);
+    expect(cancelDevice).not.toHaveBeenCalled();
   });
 
   test('does not send the install request when Protocol V2 staging fails', async () => {
@@ -5732,6 +5768,7 @@ describe('Protocol V2 firmware update targets', () => {
     );
     expect(typedCall).not.toHaveBeenCalledWith('DeviceFirmwareUpdateRequest', 'Success', {});
     expect((method as any).protocolV2InstallNeedsReconnect).toBe(true);
+    expect((method as any).protocolV2InstallDisconnectObserved).toBe(true);
   });
 
   test.each([
@@ -7147,7 +7184,14 @@ describe('Protocol V2 firmware update targets', () => {
       true
     );
     expect((method as any).exitProtocolV2BootloaderToNormal).not.toHaveBeenCalled();
-    expect(method.postProgressMessage).toHaveBeenCalledWith(100, 'transferData');
+    expect(method.postProgressMessage).toHaveBeenCalledWith(
+      100,
+      'transferData',
+      expect.objectContaining({
+        transferredBytes: 5,
+        totalBytes: 5,
+      })
+    );
     expect((method as any).completeProtocolV2FinalVerification).toHaveBeenCalledTimes(1);
   });
 
@@ -7221,7 +7265,14 @@ describe('Protocol V2 firmware update targets', () => {
       expect.objectContaining({ processedSize: 2, totalSize: 3 })
     );
     expect(method.postProgressMessage).toHaveBeenCalledTimes(1);
-    expect(method.postProgressMessage).toHaveBeenCalledWith(100, 'transferData');
+    expect(method.postProgressMessage).toHaveBeenCalledWith(
+      100,
+      'transferData',
+      expect.objectContaining({
+        transferredBytes: 3,
+        totalBytes: 3,
+      })
+    );
     expect((method as any).protocolV2StartFirmwareUpdate).toHaveBeenCalledTimes(1);
     expect((method as any).protocolV2StartFirmwareUpdate).toHaveBeenCalledWith({
       targets: [{ target_id: 4, path: 'vol0:/application_p1.bin' }],
@@ -8541,26 +8592,43 @@ describe('Protocol V2 firmware update targets', () => {
     (method as any).verifyProtocolV2StagedFile = jest.fn().mockResolvedValue(undefined);
     (method as any).protocolV2StartFirmwareUpdate = jest.fn();
     (method as any).waitForProtocolV2FirmwareUpdateComplete = jest.fn();
+    const dateNowSpy = jest
+      .spyOn(Date, 'now')
+      .mockReturnValueOnce(1_000)
+      .mockReturnValueOnce(5_000);
 
-    await (method as any).executeProtocolV2SourceUpdate({
-      installSources: [],
-      resourceSources: [
-        {
-          name: 'images.okpkg',
-          source: {
-            size: 3,
-            readAt: jest.fn(),
-            close: jest.fn(),
+    try {
+      await (method as any).executeProtocolV2SourceUpdate({
+        installSources: [],
+        resourceSources: [
+          {
+            name: 'images.okpkg',
+            source: {
+              size: 3,
+              readAt: jest.fn(),
+              close: jest.fn(),
+            },
+            devicePath: 'vol0:/bundles/images/images.okpkg',
           },
-          devicePath: 'vol0:/bundles/images/images.okpkg',
-        },
-      ],
-    });
+        ],
+      });
+    } finally {
+      dateNowSpy.mockRestore();
+    }
 
     expect((method as any).protocolV2SourceUpdateProcess).toHaveBeenCalledTimes(1);
     expect((method as any).protocolV2SourceUpdateProcess).toHaveBeenCalledWith(
-      expect.objectContaining({ filePath: 'vol0:/bundles/images/images.okpkg' })
+      expect.objectContaining({
+        filePath: 'vol0:/bundles/images/images.okpkg',
+        transferStartedAt: 1_000,
+      })
     );
+    expect(method.postProgressMessage).toHaveBeenLastCalledWith(100, 'transferData', {
+      transferredBytes: 3,
+      totalBytes: 3,
+      rateBytesPerSecond: 1,
+      elapsedMs: 4_000,
+    });
     expect((method as any).verifyProtocolV2StagedFile).toHaveBeenCalledWith(
       'vol0:/bundles/images/images.okpkg',
       3
@@ -8790,6 +8858,84 @@ describe('Protocol V2 firmware update targets', () => {
     expect(recoverProtocolV2FileTransfer).not.toHaveBeenCalled();
   });
 
+  test('does not recover or wrap a cancelled V4 file transfer', async () => {
+    const method = new FirmwareUpdateV4({
+      id: 1,
+      payload: {
+        method: 'firmwareUpdateV4',
+      },
+    });
+    const abortController = new AbortController();
+    method.abortSignal = abortController.signal;
+    (method as any).fileWriteChunk = jest.fn().mockImplementation(() => {
+      abortController.abort();
+      return Promise.reject(new Error('transport disposed'));
+    });
+    const recoverProtocolV2FileTransfer = jest.fn();
+    (method as any).recoverProtocolV2FileTransfer = recoverProtocolV2FileTransfer;
+    const source = await openFirmwareByteSource({
+      binary: new Uint8Array([1]).buffer,
+    });
+
+    try {
+      await expect(
+        (method as any).protocolV2SourceUpdateProcess({
+          source,
+          filePath: 'vol1:firmware.bin',
+          processedSize: 0,
+          totalSize: 1,
+        })
+      ).rejects.toMatchObject({ errorCode: HardwareErrorCode.CallQueueActionCancelled });
+    } finally {
+      await source?.close();
+    }
+    expect(recoverProtocolV2FileTransfer).not.toHaveBeenCalled();
+  });
+
+  test('keeps public transfer bytes monotonic when a V4 file retry restarts at zero', async () => {
+    const method = new FirmwareUpdateV4({
+      id: 1,
+      payload: {
+        method: 'firmwareUpdateV4',
+      },
+    });
+    let writeCount = 0;
+    (method as any).fileWriteChunk = jest.fn(
+      (_path: string, _size: number, offset: number, data: Uint8Array) => {
+        writeCount += 1;
+        if (writeCount === 3) {
+          return Promise.reject(new Error('transport timeout'));
+        }
+        return Promise.resolve({
+          message: { processed_byte: offset + data.byteLength },
+        });
+      }
+    );
+    (method as any).getProtocolV2FirmwareChunkSize = jest.fn().mockReturnValue(1000);
+    (method as any).recoverProtocolV2FileTransfer = jest.fn().mockResolvedValue(undefined);
+    method.postProgressMessage = jest.fn();
+    const source = await openFirmwareByteSource({
+      binary: new Uint8Array(3000).buffer,
+    });
+
+    try {
+      await (method as any).protocolV2SourceUpdateProcess({
+        source,
+        filePath: 'vol1:firmware.bin',
+        processedSize: 0,
+        totalSize: 3000,
+      });
+    } finally {
+      await source?.close();
+    }
+
+    const transferredBytes = (method.postProgressMessage as jest.Mock).mock.calls.map(
+      ([, , metrics]) => metrics.transferredBytes
+    );
+    expect(transferredBytes).toEqual([1000, 2000, 3000]);
+    expect((method as any).recoverProtocolV2FileTransfer).toHaveBeenCalledTimes(1);
+  });
+
   test('throttles repeated transfer progress while preserving file completion', async () => {
     const method = new FirmwareUpdateV4({
       id: 1,
@@ -9148,6 +9294,42 @@ describe('Protocol V2 firmware update targets', () => {
     expect(typedCall).toHaveBeenCalledTimes(6);
   });
 
+  test('preserves the underlying transport error code after firmware transfer retries', async () => {
+    const method = new FirmwareUpdateV4({
+      id: 1,
+      payload: {
+        method: 'firmwareUpdateV4',
+      },
+    });
+    const typedCall = jest
+      .fn()
+      .mockRejectedValue(ERRORS.TypedError(HardwareErrorCode.BleTimeoutError, 'response timeout'));
+
+    (method as any).device = stubDevice({
+      getCommands: () => ({ typedCall }),
+    });
+    method.postProgressMessage = jest.fn();
+    method.postTipMessage = jest.fn();
+    (method as any).recoverProtocolV2FileTransfer = jest.fn().mockResolvedValue(undefined);
+
+    const source = await openFirmwareByteSource({
+      binary: new Uint8Array([1, 2, 3]).buffer,
+    });
+    await expect(
+      (method as any).protocolV2SourceUpdateProcess({
+        source,
+        filePath: 'vol0:/firmware.bin',
+        processedSize: 0,
+        totalSize: 3,
+      })
+    ).rejects.toMatchObject({
+      errorCode: HardwareErrorCode.EmmcFileWriteFirmwareError,
+      params: { causeCode: HardwareErrorCode.BleTimeoutError },
+    });
+    await source?.close();
+    expect(typedCall).toHaveBeenCalledTimes(3);
+  });
+
   // TODO(#850/#855): PR #855 added resume-on-retry and per-chunk retry on the
   // writeProtocolV2File path. PR #850 replaced that path with FirmwareByteSource
   // streaming (protocolV2SourceUpdateProcess), which restarts a failed transfer from

package/__tests__/protocolV2FileWrite.test.ts

@@ -10,6 +10,46 @@ jest.mock('../src/data/config', () => ({
 }));
 
 describe('writeProtocolV2File', () => {
+  test('allows a verified caller-specific BLE chunk limit', async () => {
+    const getSettingsSpy = jest
+      .spyOn(DataManager, 'getSettings')
+      .mockReturnValue('react-native' as any);
+    const isBleConnectSpy = jest.spyOn(DataManager, 'isBleConnect').mockReturnValue(true);
+    const data = new Uint8Array(1961);
+    const typedCall = jest.fn().mockResolvedValue({ message: {} });
+
+    try {
+      await writeProtocolV2File({
+        commands: { typedCall } as any,
+        path: 'vol1:/wallpapers/wallpaper.okpkg',
+        data,
+        bleChunkSizeLimit: 1960,
+      });
+    } finally {
+      getSettingsSpy.mockRestore();
+      isBleConnectSpy.mockRestore();
+    }
+
+    expect(typedCall).toHaveBeenCalledTimes(2);
+    expect(typedCall.mock.calls[0][2].file.data).toEqual(data.slice(0, 1960));
+    expect(typedCall.mock.calls[1][2].file.data).toEqual(data.slice(1960));
+  });
+
+  test('does not apply the BLE-only limit to WebUSB', async () => {
+    const data = new Uint8Array(1961);
+    const typedCall = jest.fn().mockResolvedValue({ message: {} });
+
+    await writeProtocolV2File({
+      commands: { typedCall } as any,
+      path: 'vol1:/wallpapers/wallpaper.okpkg',
+      data,
+      bleChunkSizeLimit: 1960,
+    });
+
+    expect(typedCall).toHaveBeenCalledTimes(1);
+    expect(typedCall.mock.calls[0][2].file.data).toEqual(data);
+  });
+
   test('按分片写入并只在首片设置 overwrite', async () => {
     const data = new Uint8Array(4097);
     const typedCall = jest.fn().mockResolvedValue({ message: {} });

package/dist/api/FirmwareUpdateV4.d.ts

@@ -18,10 +18,12 @@ export default class FirmwareUpdateV4 extends FirmwareUpdateBaseMethod<FirmwareU
     private protocolV2LatestFinalDeviceInfo?;
     private protocolV2InstallBaselineVersions;
     private protocolV2InstallNeedsReconnect;
+    private protocolV2InstallDisconnectObserved;
     private protocolV2InstallTerminalSuccessObserved;
     private protocolV2LastRuntimeProbeFeatures?;
     private protocolV2LastTransferProgress?;
     private protocolV2LastTransferProgressAt;
+    private protocolV2LastTransferredBytes;
     init(): void;
     private getProtocolV2FirmwareChunkSize;
     run(): Promise<{