@onekeyfe/hd-core 1.1.34-alpha.2 → 1.1.34-alpha.3

package/src/api/helpers/protocolV2FileWrite.ts

@@ -0,0 +1,193 @@
+import {
+  PROTOCOL_V2_BLE_FILE_CHUNK_SIZE,
+  PROTOCOL_V2_WEBUSB_FILE_CHUNK_SIZE,
+} from '@onekeyfe/hd-transport';
+import { ERRORS, HardwareErrorCode } from '@onekeyfe/hd-shared';
+
+import { DataManager } from '../../data-manager';
+
+import type { DeviceCommands } from '../../device/DeviceCommands';
+
+export type ProtocolV2FileWriteData = ArrayBuffer | Uint8Array | Blob | string;
+
+export type ProtocolV2FileWriteProgress = {
+  progress: number;
+  transferredBytes: number;
+  totalBytes: number;
+  rateBytesPerSecond?: number;
+  elapsedMs: number;
+};
+
+export type ProtocolV2FileWriteOptions = {
+  commands: Pick<DeviceCommands, 'typedCall'>;
+  path: string;
+  data: ProtocolV2FileWriteData;
+  offset?: number;
+  totalSize?: number;
+  chunkSize?: number;
+  chunkLen?: number;
+  overwrite?: boolean;
+  append?: boolean;
+  uiPercentage?: number;
+  timeoutMs?: number;
+  maxChunkRetries?: number;
+  throwIfAborted?: () => void;
+  onProgress?: (progress: ProtocolV2FileWriteProgress) => void;
+};
+
+const MIN_FILE_CHUNK_SIZE = 64;
+
+function isRetryableFileWriteTimeout(error: unknown) {
+  if (!error || typeof error !== 'object') return false;
+  const candidate = error as { errorCode?: number; code?: number; message?: string };
+  const code = candidate.errorCode ?? candidate.code;
+  return (
+    code === HardwareErrorCode.BleTimeoutError ||
+    /Lowlevel response timeout/i.test(candidate.message ?? '')
+  );
+}
+
+function getProtocolV2FileChunkLimit() {
+  const env = DataManager.getSettings('env');
+  return env && DataManager.isBleConnect(env)
+    ? PROTOCOL_V2_BLE_FILE_CHUNK_SIZE
+    : PROTOCOL_V2_WEBUSB_FILE_CHUNK_SIZE;
+}
+
+async function dataToUint8Array(data: ProtocolV2FileWriteData): Promise<Uint8Array> {
+  if (typeof data === 'string') return new TextEncoder().encode(data);
+  if (data instanceof ArrayBuffer) return new Uint8Array(data);
+  if (ArrayBuffer.isView(data)) {
+    return new Uint8Array(data.buffer, data.byteOffset, data.byteLength);
+  }
+  if (typeof Blob !== 'undefined' && data instanceof Blob) {
+    return new Uint8Array(await data.arrayBuffer());
+  }
+  throw ERRORS.TypedError(
+    HardwareErrorCode.CallMethodInvalidParameter,
+    'Unsupported FilesystemFileWrite data'
+  );
+}
+
+function normalizeChunkSize(value: unknown, maxChunkSize: number): number {
+  const numeric = Number(value);
+  if (!Number.isFinite(numeric) || numeric <= 0) return maxChunkSize;
+  return Math.min(Math.max(Math.floor(numeric), MIN_FILE_CHUNK_SIZE), maxChunkSize);
+}
+
+function getDeviceTransferProgress(before: number, after: number, total: number) {
+  if (!Number.isFinite(total) || total <= 0) return 100;
+  if (before <= 0 && after < total) return 0;
+  if (after >= total) return 100;
+  return Math.min(Math.max(Math.ceil((after / total) * 100), 1), 99);
+}
+
+function getConfirmedProgress(processed: number, total: number, written: number, length: number) {
+  if (Number.isFinite(processed) && Number.isFinite(total) && total > 0) {
+    if (processed >= total) return 100;
+    return Math.min(Math.max(Math.floor((processed / total) * 100), 0), 99);
+  }
+  if (length > 0) return written >= length ? 100 : Math.floor((written / length) * 100);
+  return 100;
+}
+
+export async function writeProtocolV2File(options: ProtocolV2FileWriteOptions) {
+  options.throwIfAborted?.();
+  const data = await dataToUint8Array(options.data);
+  const dataLength = data.byteLength;
+  const startOffset =
+    Number.isFinite(options.offset) && Number(options.offset) > 0 ? Number(options.offset) : 0;
+  const totalSize =
+    Number.isFinite(options.totalSize) && Number(options.totalSize) > 0
+      ? Number(options.totalSize)
+      : startOffset + dataLength;
+
+  if (totalSize < startOffset + dataLength) {
+    throw ERRORS.TypedError(
+      HardwareErrorCode.RuntimeError,
+      `FilesystemFileWrite totalSize ${totalSize} is smaller than offset + data length ${
+        startOffset + dataLength
+      }`
+    );
+  }
+
+  const chunkSize = normalizeChunkSize(
+    options.chunkSize ?? options.chunkLen,
+    getProtocolV2FileChunkLimit()
+  );
+  let written = 0;
+  let chunks = 0;
+  let lastMessage: Record<string, unknown> | undefined;
+  const startTime = Date.now();
+
+  while (written < dataLength) {
+    options.throwIfAborted?.();
+    const chunk = data.slice(written, Math.min(written + chunkSize, dataLength));
+    const offset = startOffset + written;
+    const progress =
+      options.uiPercentage ??
+      getDeviceTransferProgress(offset, offset + chunk.byteLength, totalSize);
+    const request = {
+      file: { path: options.path, offset, total_size: totalSize, data: chunk },
+      overwrite: chunks === 0 ? options.overwrite ?? false : false,
+      append: options.append ?? false,
+      ui_percentage: progress,
+    };
+    const maxChunkRetries = Math.max(Math.floor(options.maxChunkRetries ?? 0), 0);
+    let retryCount = 0;
+    let response;
+    let isWritePending = true;
+    while (isWritePending) {
+      try {
+        response = await options.commands.typedCall(
+          'FilesystemFileWrite',
+          'FilesystemFile',
+          request,
+          { timeoutMs: options.timeoutMs }
+        );
+        isWritePending = false;
+      } catch (error) {
+        if (retryCount >= maxChunkRetries || !isRetryableFileWriteTimeout(error)) throw error;
+        retryCount += 1;
+        options.throwIfAborted?.();
+      }
+    }
+    options.throwIfAborted?.();
+    lastMessage = response.message;
+    const rawProcessedByte = response.message?.processed_byte;
+    const processedByte = Number(rawProcessedByte);
+    if (
+      rawProcessedByte !== undefined &&
+      (!Number.isFinite(processedByte) ||
+        processedByte <= offset ||
+        processedByte > offset + chunk.byteLength)
+    ) {
+      throw ERRORS.TypedError(
+        HardwareErrorCode.RuntimeError,
+        `FilesystemFileWrite invalid processed_byte ${processedByte}`
+      );
+    }
+    written =
+      rawProcessedByte === undefined ? written + chunk.byteLength : processedByte - startOffset;
+    chunks += 1;
+    const elapsedMs = Date.now() - startTime;
+    const transferredBytes = Math.min(written, dataLength);
+    options.onProgress?.({
+      progress: getConfirmedProgress(startOffset + written, totalSize, written, dataLength),
+      transferredBytes,
+      totalBytes: dataLength,
+      rateBytesPerSecond:
+        elapsedMs > 0 ? Math.round((transferredBytes / elapsedMs) * 1000) : undefined,
+      elapsedMs,
+    });
+  }
+
+  return {
+    ...lastMessage,
+    path: options.path,
+    offset: startOffset,
+    total_size: totalSize,
+    processed_byte: startOffset + written,
+    chunks,
+  };
+}

package/src/api/index.ts

@@ -3,9 +3,12 @@ export { default as preInitialize } from './device/PreInitialize';
 
 export { default as searchDevices } from './SearchDevices';
 export { default as getFeatures } from './GetFeatures';
+export { default as getDeviceState } from './GetDeviceState';
 export { default as getOnekeyFeatures } from './GetOnekeyFeatures';
 export { default as getPassphraseState } from './GetPassphraseState';
+export { default as openWalletSession } from './OpenWalletSession';
 export { default as getLogs } from './GetLogs';
+export { default as clearSessionCache } from './ClearSessionCache';
 export { default as checkFirmwareRelease } from './CheckFirmwareRelease';
 export { default as checkBLEFirmwareRelease } from './CheckBLEFirmwareRelease';
 export { default as checkBridgeStatus } from './CheckBridgeStatus';
@@ -38,8 +41,37 @@ export { default as getNextU2FCounter } from './u2f/GetNextU2FCounter';
 export { default as firmwareUpdate } from './FirmwareUpdate';
 export { default as firmwareUpdateV2 } from './FirmwareUpdateV2';
 export { default as firmwareUpdateV3 } from './FirmwareUpdateV3';
+export { default as firmwareUpdateV4 } from './FirmwareUpdateV4';
 export { default as promptWebDeviceAccess } from './PromptWebDeviceAccess';
 
+// Pro2 business API
+export { default as deviceReboot } from './protocol-v2/DeviceReboot';
+export { default as deviceGetOnboardingStatus } from './protocol-v2/DeviceGetOnboardingStatus';
+export { default as deviceUploadWallpaper } from './protocol-v2/DeviceUploadWallpaper';
+// Protocol V2 development APIs. These remain public while Pro2 integration is in development.
+export { default as deviceInfoGet } from './protocol-v2/DeviceInfoGet';
+export { default as deviceStatusGet } from './protocol-v2/DeviceStatusGet';
+export { default as deviceSettingsGet } from './protocol-v2/DeviceSettingsGet';
+export { default as protocolInfoRequest } from './protocol-v2/ProtocolInfoRequest';
+export { default as ping } from './protocol-v2/Ping';
+export { default as deviceSessionOpen } from './protocol-v2/DeviceSessionOpen';
+export { default as deviceFirmwareUpdate } from './protocol-v2/DeviceFirmwareUpdate';
+export { default as deviceGetFirmwareUpdateStatus } from './protocol-v2/DeviceGetFirmwareUpdateStatus';
+export { default as deviceFactoryInfoSet } from './protocol-v2/DeviceFactoryInfoSet';
+export { default as deviceFactoryInfoGet } from './protocol-v2/DeviceFactoryInfoGet';
+export { default as deviceSettingsSet } from './protocol-v2/DeviceSettingsSet';
+export { default as deviceSettingsPageShow } from './protocol-v2/DeviceSettingsPageShow';
+export { default as filesystemPermissionFix } from './protocol-v2/FilesystemPermissionFix';
+export { default as filesystemFormat } from './protocol-v2/FilesystemFormat';
+export { default as fileRead } from './FileRead';
+export { default as fileWrite } from './FileWrite';
+export { default as fileDelete } from './FileDelete';
+export { default as dirList } from './DirList';
+export { default as dirMake } from './DirMake';
+export { default as dirRemove } from './DirRemove';
+export { default as pathInfo } from './PathInfo';
+export { default as uploadPortfolio } from './UploadPortfolio';
+
 export { default as cipherKeyValue } from './CipherKeyValue';
 
 export { default as allNetworkGetAddress } from './allnetwork/AllNetworkGetAddress';

package/src/api/kaspa/KaspaSignTransaction.ts

@@ -198,15 +198,14 @@ export default class KaspaSignTransaction extends BaseMethod<KaspaSignTransactio
 
       const response = await typedCall(
         'KaspaTxInputAck',
-        // @ts-expect-error
         ['KaspaTxInputRequest', 'KaspaSignedTx'],
         {
-          address_n: input.path,
+          // Normalize params.inputs paths from either strings or parsed arrays to address_n.
+          address_n: validatePath(input.path, 3),
           raw_message: bytesToHex(rawMessage),
         }
       );
 
-      // @ts-expect-error
       return this.processTxRequest(typedCall, response, nextIndex, signature);
     }
 

package/src/api/lightning/LnurlAuth.ts

@@ -32,6 +32,10 @@ export default class LnurlAuth1 extends BaseMethod<ILnurlAuth> {
       model_touch: {
         min: '4.8.0',
       },
+      pro2: {
+        min: '0.0.0',
+        unsupported: true,
+      },
     };
   }
 

package/src/api/neo/NeoGetAddress.ts

@@ -5,6 +5,7 @@ import { UI_REQUEST } from '../../constants/ui-request';
 
 import type { NeoGetAddress as HardwareNeoGetAddress } from '@onekeyfe/hd-transport';
 import type { NeoAddress, NeoGetAddressParams } from '../../types/api/neoGetAddress';
+import type { DeviceFirmwareRange } from '../../types';
 
 export default class NeoGetAddress extends BaseMethod<HardwareNeoGetAddress[]> {
   hasBundle = false;
@@ -37,8 +38,12 @@ export default class NeoGetAddress extends BaseMethod<HardwareNeoGetAddress[]> {
     });
   }
 
-  getVersionRange() {
+  getVersionRange(): DeviceFirmwareRange {
     return {
+      pro2: {
+        min: '0.0.0',
+        unsupported: true,
+      },
       pro: {
         min: '4.12.0',
       },

package/src/api/neo/NeoSignTransaction.ts

@@ -5,6 +5,7 @@ import { validateParams } from '../helpers/paramsValidator';
 import { formatAnyHex } from '../helpers/hexUtils';
 
 import type { NeoSignTx } from '@onekeyfe/hd-transport';
+import type { DeviceFirmwareRange } from '../../types';
 
 export default class NeoSignTransaction extends BaseMethod<NeoSignTx> {
   init() {
@@ -29,8 +30,12 @@ export default class NeoSignTransaction extends BaseMethod<NeoSignTx> {
     };
   }
 
-  getVersionRange() {
+  getVersionRange(): DeviceFirmwareRange {
     return {
+      pro2: {
+        min: '0.0.0',
+        unsupported: true,
+      },
       pro: {
         min: '4.12.0',
       },

package/src/api/nervos/NervosGetAddress.ts

@@ -4,7 +4,7 @@ import { BaseMethod } from '../BaseMethod';
 import { validateParams, validateResult } from '../helpers/paramsValidator';
 
 import type { NervosGetAddress as HardwareNervosGetAddress } from '@onekeyfe/hd-transport';
-import type { NervosAddress, NervosGetAddressParams } from '../../types';
+import type { DeviceFirmwareRange, NervosAddress, NervosGetAddressParams } from '../../types';
 
 export default class NervosGetAddress extends BaseMethod<HardwareNervosGetAddress[]> {
   hasBundle = false;
@@ -40,8 +40,12 @@ export default class NervosGetAddress extends BaseMethod<HardwareNervosGetAddres
     });
   }
 
-  getVersionRange() {
+  getVersionRange(): DeviceFirmwareRange {
     return {
+      pro2: {
+        min: '0.0.0',
+        unsupported: true,
+      },
       model_mini: {
         min: '3.7.0',
       },

package/src/api/nervos/NervosSignTransaction.ts

@@ -5,8 +5,12 @@ import { BaseMethod } from '../BaseMethod';
 import { validateParams } from '../helpers/paramsValidator';
 import { formatAnyHex } from '../helpers/hexUtils';
 
-import type { NervosSignTransactionParams, NervosSignedTx } from '../../types';
-import type { NervosSignTx as HardwareNervosSignTx, TypedCall } from '@onekeyfe/hd-transport';
+import type { DeviceFirmwareRange, NervosSignTransactionParams, NervosSignedTx } from '../../types';
+import type {
+  NervosSignTx as HardwareNervosSignTx,
+  NervosTxRequest,
+  TypedCall,
+} from '@onekeyfe/hd-transport';
 import type { TypedResponseMessage } from '../../device/DeviceCommands';
 
 type NervosSignTx = Omit<HardwareNervosSignTx, 'data_initial_chunk' | 'data_length'> & {
@@ -41,8 +45,12 @@ export default class NervosSignTransaction extends BaseMethod<NervosSignTx> {
     };
   }
 
-  getVersionRange() {
+  getVersionRange(): DeviceFirmwareRange {
     return {
+      pro2: {
+        min: '0.0.0',
+        unsupported: true,
+      },
       model_mini: {
         min: '3.7.0',
       },
@@ -59,7 +67,9 @@ export default class NervosSignTransaction extends BaseMethod<NervosSignTx> {
     res: TypedResponseMessage<'NervosSignedTx'> | TypedResponseMessage<'NervosTxRequest'>,
     data: Buffer,
     offset = 0
-  ): Promise<NervosSignedTx> => {
+    // The final NervosTxRequest may return a signature without data_length or address,
+    // so the result accurately uses a union type.
+  ): Promise<NervosSignedTx | (NervosTxRequest & { path: string })> => {
     if (res.type === 'NervosSignedTx') {
       if (!res?.message?.signature) {
         throw new Error('No signature returned');

package/src/api/nexa/NexaGetAddress.ts

@@ -4,7 +4,7 @@ import { BaseMethod } from '../BaseMethod';
 import { validateParams, validateResult } from '../helpers/paramsValidator';
 
 import type { NexaGetAddress as HardwareNexaGetAddress } from '@onekeyfe/hd-transport';
-import type { NexaGetAddressParams } from '../../types';
+import type { DeviceFirmwareRange, NexaGetAddressParams } from '../../types';
 
 export default class NexaGetAddress extends BaseMethod<HardwareNexaGetAddress[]> {
   hasBundle = false;
@@ -43,8 +43,12 @@ export default class NexaGetAddress extends BaseMethod<HardwareNexaGetAddress[]>
     });
   }
 
-  getVersionRange() {
+  getVersionRange(): DeviceFirmwareRange {
     return {
+      pro2: {
+        min: '0.0.0',
+        unsupported: true,
+      },
       model_mini: {
         min: '3.2.0',
       },

package/src/api/nexa/NexaSignTransaction.ts

@@ -6,7 +6,7 @@ import { validateParams } from '../helpers/paramsValidator';
 
 import type { TypedResponseMessage } from '../../device/DeviceCommands';
 import type { TypedCall } from '@onekeyfe/hd-transport';
-import type { NexaSignTransactionParams, NexaSignature } from '../../types';
+import type { DeviceFirmwareRange, NexaSignTransactionParams, NexaSignature } from '../../types';
 
 export default class NexaSignTransaction extends BaseMethod<NexaSignTransactionParams> {
   hasBundle = false;
@@ -27,8 +27,12 @@ export default class NexaSignTransaction extends BaseMethod<NexaSignTransactionP
     this.params = payload;
   }
 
-  getVersionRange() {
+  getVersionRange(): DeviceFirmwareRange {
     return {
+      pro2: {
+        min: '0.0.0',
+        unsupported: true,
+      },
       model_mini: {
         min: '3.2.0',
       },
@@ -68,17 +72,12 @@ export default class NexaSignTransaction extends BaseMethod<NexaSignTransactionP
 
       const nextIndex = res.message.request_index;
       const input = this.params.inputs[nextIndex];
-      const response = await typedCall(
-        'NexaTxInputAck',
-        // @ts-expect-error
-        ['NexaTxInputRequest', 'NexaSignedTx'],
-        {
-          address_n: input.path,
-          raw_message: input.message,
-        }
-      );
+      const response = await typedCall('NexaTxInputAck', ['NexaTxInputRequest', 'NexaSignedTx'], {
+        // Normalize a string path to address_n, matching the first input handled by run().
+        address_n: validatePath(input.path, 3),
+        raw_message: input.message,
+      });
 
-      // @ts-expect-error
       return this.processTxRequest(typedCall, response, nextIndex, signatures);
     }
 

package/src/api/polkadot/networks.ts

@@ -23,6 +23,9 @@ const baseVersionRange = {
   model_touch: {
     min: '4.3.0',
   },
+  pro2: {
+    min: '0.0.0',
+  },
 };
 
 const specialVersionRange: Record<string, DeviceFirmwareRange> = {
@@ -33,6 +36,9 @@ const specialVersionRange: Record<string, DeviceFirmwareRange> = {
     model_touch: {
       min: '4.7.0',
     },
+    pro2: {
+      min: '0.0.0',
+    },
   },
   [Networks.Manta]: {
     model_mini: {
@@ -41,6 +47,9 @@ const specialVersionRange: Record<string, DeviceFirmwareRange> = {
     model_touch: {
       min: '4.9.0',
     },
+    pro2: {
+      min: '0.0.0',
+    },
   },
 };
 

package/src/api/protocol-v2/DeviceFactoryInfoGet.ts

@@ -0,0 +1,20 @@
+import { BaseMethod } from '../BaseMethod';
+
+export default class DeviceFactoryInfoGet extends BaseMethod {
+  init() {
+    // Protocol V2 (Pro2) only; Core rejects non-V2 devices.
+    this.requireProtocolV2 = true;
+    this.skipForceUpdateCheck = true;
+    this.useDevicePassphraseState = false;
+    this.params = undefined;
+  }
+
+  async run() {
+    const res = await this.device.commands.typedCall(
+      'DeviceFactoryInfoGet',
+      'DeviceFactoryInfo',
+      {}
+    );
+    return Promise.resolve(res.message);
+  }
+}

package/src/api/protocol-v2/DeviceFactoryInfoSet.ts

@@ -0,0 +1,32 @@
+import { BaseMethod } from '../BaseMethod';
+
+import type { DeviceFactoryInfoSetParams } from './helpers';
+
+export default class DeviceFactoryInfoSet extends BaseMethod<DeviceFactoryInfoSetParams> {
+  init() {
+    // Protocol V2 (Pro2) only; Core rejects non-V2 devices.
+    this.requireProtocolV2 = true;
+    this.skipForceUpdateCheck = true;
+    this.useDevicePassphraseState = false;
+    this.params = {
+      version: this.payload.version,
+      serial_number: this.payload.serial_number,
+      burn_in_completed: this.payload.burn_in_completed,
+      factory_test_completed: this.payload.factory_test_completed,
+      manufacture_time: this.payload.manufacture_time,
+    };
+  }
+
+  async run() {
+    const res = await this.device.commands.typedCall('DeviceFactoryInfoSet', 'Success', {
+      info: {
+        version: this.params.version,
+        serial_number: this.params.serial_number,
+        burn_in_completed: this.params.burn_in_completed,
+        factory_test_completed: this.params.factory_test_completed,
+        manufacture_time: this.params.manufacture_time,
+      },
+    });
+    return Promise.resolve(res.message);
+  }
+}

package/src/api/protocol-v2/DeviceFirmwareUpdate.ts

@@ -0,0 +1,61 @@
+import { BaseMethod } from '../BaseMethod';
+import {
+  PROTOCOL_V2_FIRMWARE_UPDATE_OPTIONS,
+  PROTOCOL_V2_FIRMWARE_UPDATE_RESPONSE_TYPES,
+  isProtocolV2DeviceDisconnectedError,
+  normalizeFirmwareTargets,
+} from './helpers';
+import { UI_REQUEST, createUiMessage } from '../../events/ui-request';
+
+import type { KnownDevice } from '../../types';
+import type { MessageFromOneKey } from '@onekeyfe/hd-transport';
+import type { DeviceFirmwareUpdateParams } from './helpers';
+
+export default class DeviceFirmwareUpdate extends BaseMethod<DeviceFirmwareUpdateParams> {
+  init() {
+    // Protocol V2 (Pro2) only; Core rejects non-V2 devices.
+    this.requireProtocolV2 = true;
+    this.skipForceUpdateCheck = true;
+    this.useDevicePassphraseState = false;
+    this.params = {
+      targets: this.payload.targets,
+      targetId: this.payload.targetId,
+      target_id: this.payload.target_id,
+      path: this.payload.path,
+    };
+  }
+
+  async run() {
+    const targets = normalizeFirmwareTargets(this.params);
+    try {
+      const res = await this.device.commands.typedCall(
+        'DeviceFirmwareUpdateRequest',
+        PROTOCOL_V2_FIRMWARE_UPDATE_RESPONSE_TYPES,
+        {
+          targets,
+        },
+        {
+          ...PROTOCOL_V2_FIRMWARE_UPDATE_OPTIONS,
+          onIntermediateResponse: (response: MessageFromOneKey) => {
+            if (response.type !== 'DeviceFirmwareUpdateStatus') return;
+            this.postMessage(
+              createUiMessage(UI_REQUEST.FIRMWARE_PROGRESS, {
+                device: this.device.toMessageObject() as KnownDevice,
+                progress: 99,
+                progressType: 'installingFirmware',
+              })
+            );
+          },
+        }
+      );
+      return res.message;
+    } catch (error) {
+      if (isProtocolV2DeviceDisconnectedError(error)) {
+        return {
+          message: 'Device firmware update started',
+        };
+      }
+      throw error;
+    }
+  }
+}

package/src/api/protocol-v2/DeviceGetFirmwareUpdateStatus.ts

@@ -0,0 +1,55 @@
+import { BaseMethod } from '../BaseMethod';
+import { invalidParameter } from '../helpers/filesystemValidation';
+
+import type { DeviceFirmwareUpdateStatusGetParams } from './helpers';
+import type { DeviceFirmwareUpdateRecordFields } from '@onekeyfe/hd-transport';
+
+const DEVICE_FIRMWARE_UPDATE_STATUS_FIELDS = ['status', 'payload_version', 'path'] as const;
+
+function normalizeStatusFields(
+  fields: DeviceFirmwareUpdateStatusGetParams['fields']
+): DeviceFirmwareUpdateRecordFields | undefined {
+  if (fields === undefined || fields === null) return undefined;
+  if (typeof fields !== 'object' || Array.isArray(fields)) {
+    throw invalidParameter('Parameter [fields] must be an object.');
+  }
+
+  const unknownField = Object.keys(fields).find(
+    field => !DEVICE_FIRMWARE_UPDATE_STATUS_FIELDS.includes(field as any)
+  );
+  if (unknownField) {
+    throw invalidParameter(`Unsupported firmware update status field: ${unknownField}`);
+  }
+
+  const normalized: DeviceFirmwareUpdateRecordFields = {};
+  for (const field of DEVICE_FIRMWARE_UPDATE_STATUS_FIELDS) {
+    const value = fields[field];
+    if (value !== undefined) {
+      if (typeof value !== 'boolean') {
+        throw invalidParameter(`Parameter [fields.${field}] must be a boolean.`);
+      }
+      normalized[field] = value;
+    }
+  }
+  return normalized;
+}
+
+export default class DeviceGetFirmwareUpdateStatus extends BaseMethod<DeviceFirmwareUpdateStatusGetParams> {
+  init() {
+    // Protocol V2 (Pro2) only; Core rejects non-V2 devices.
+    this.requireProtocolV2 = true;
+    this.skipForceUpdateCheck = true;
+    this.useDevicePassphraseState = false;
+    const fields = normalizeStatusFields(this.payload.fields);
+    this.params = fields ? { fields } : {};
+  }
+
+  async run() {
+    const res = await this.device.commands.typedCall(
+      'DeviceFirmwareUpdateStatusGet',
+      'DeviceFirmwareUpdateStatus',
+      this.params
+    );
+    return Promise.resolve(res.message);
+  }
+}

package/src/api/protocol-v2/DeviceGetOnboardingStatus.ts

@@ -0,0 +1,18 @@
+import { BaseMethod } from '../BaseMethod';
+
+export default class DeviceGetOnboardingStatus extends BaseMethod {
+  init() {
+    this.requireProtocolV2 = true;
+    this.skipForceUpdateCheck = true;
+    this.useDevicePassphraseState = false;
+  }
+
+  async run() {
+    const { message } = await this.device.commands.typedCall(
+      'DevGetOnboardingStatus',
+      'DevOnboardingStatus',
+      {}
+    );
+    return message;
+  }
+}