@onekeyfe/hd-core 1.1.18 → 1.1.19-alpha.1

package/src/api/FirmwareUpdate.ts

@@ -1,24 +1,25 @@
 import {
+  EFirmwareType,
   ERRORS,
-  HardwareErrorCode,
   HardwareError,
+  HardwareErrorCode,
   createDeferred,
-  Deferred,
 } from '@onekeyfe/hd-shared';
+
 import { UI_REQUEST } from '../constants/ui-request';
 import { BaseMethod } from './BaseMethod';
 import { validateParams } from './helpers/paramsValidator';
 import { getBinary } from './firmware/getBinary';
 import { uploadFirmware } from './firmware/uploadFirmware';
 import { createUiMessage } from '../events';
-import { type KnownDevice, DeviceModelToTypes } from '../types';
-
+import { DeviceModelToTypes, type KnownDevice } from '../types';
 import { isEnteredManuallyBoot } from './firmware/bootloaderHelper';
-
 import { LoggerNames, getDeviceType, getDeviceUUID, getLogger, wait } from '../utils';
 import { DataManager } from '../data-manager';
 import { DevicePool } from '../device/DevicePool';
 
+import type { Deferred } from '@onekeyfe/hd-shared';
+
 type Params = {
   binary?: ArrayBuffer;
   version?: number[];
@@ -189,6 +190,7 @@ export default class FirmwareUpdate extends BaseMethod<Params> {
           features: device.features,
           version: params.version,
           updateType: params.updateType,
+          firmwareType: EFirmwareType.Universal,
         });
         binary = firmware.binary;
         this.postTipMessage('DownloadFirmwareSuccess');

package/src/api/FirmwareUpdateV2.ts

@@ -1,12 +1,14 @@
 import {
-  createDeferred,
-  Deferred,
+  type Deferred,
   EDeviceType,
+  type EFirmwareType,
   ERRORS,
   HardwareError,
   HardwareErrorCode,
+  createDeferred,
 } from '@onekeyfe/hd-shared';
 import semver from 'semver';
+
 import { UI_REQUEST } from '../constants/ui-request';
 import { BaseMethod } from './BaseMethod';
 import { validateParams } from './helpers/paramsValidator';
@@ -14,20 +16,21 @@ import { DevicePool } from '../device/DevicePool';
 import { getBinary, getInfo, getSysResourceBinary } from './firmware/getBinary';
 import { updateResources, uploadFirmware } from './firmware/uploadFirmware';
 import {
+  LoggerNames,
+  getDeviceFirmwareVersion,
   getDeviceType,
   getDeviceUUID,
-  wait,
+  getFirmwareType,
   getLogger,
-  LoggerNames,
-  getDeviceFirmwareVersion,
+  wait,
 } from '../utils';
-import { createUiMessage, FirmwareUpdateTipMessage } from '../events/ui-request';
+import { FirmwareUpdateTipMessage, createUiMessage } from '../events/ui-request';
 import { DeviceModelToTypes } from '../types';
 import { DataManager } from '../data-manager';
+import { DEVICE } from '../events';
 
-import type { KnownDevice, Features } from '../types';
+import type { Features, KnownDevice } from '../types';
 import type { Device } from '../device/Device';
-import { DEVICE } from '../events';
 
 type Params = {
   binary?: ArrayBuffer;
@@ -35,6 +38,7 @@ type Params = {
   updateType: 'firmware' | 'ble';
   forcedUpdateRes?: boolean;
   isUpdateBootloader?: boolean;
+  firmwareType?: EFirmwareType;
 };
 
 const Log = getLogger(LoggerNames.Method);
@@ -55,6 +59,7 @@ export default class FirmwareUpdateV2 extends BaseMethod<Params> {
       { name: 'binary', type: 'buffer' },
       { name: 'forcedUpdateRes', type: 'boolean' },
       { name: 'platform', type: 'string', required: true },
+      { name: 'firmwareType', type: 'string' },
     ]);
 
     if (!payload.updateType) {
@@ -74,6 +79,7 @@ export default class FirmwareUpdateV2 extends BaseMethod<Params> {
       this.params = {
         ...this.params,
         version: payload.version,
+        firmwareType: payload.firmwareType,
       };
     }
 
@@ -246,14 +252,14 @@ export default class FirmwareUpdateV2 extends BaseMethod<Params> {
    * Check the version number of Touch to determine if it
    * needs to be upgraded via the desktop
    */
-  checkVersionForCopyTouchResource(features?: Features) {
+  checkVersionForCopyTouchResource(features: Features | undefined, firmwareType: EFirmwareType) {
     if (!features) return;
     const deviceType = getDeviceType(features);
     const currentVersion = getDeviceFirmwareVersion(features).join('.');
     const targetVersion = this.params.version?.join('.');
     const { updateType } = this.params;
 
-    const releaseInfo = getInfo({ features, updateType });
+    const releaseInfo = getInfo({ features, updateType, firmwareType });
     if (!releaseInfo) return;
     const { fullResourceRange } = releaseInfo;
     if (!fullResourceRange) return;
@@ -275,7 +281,10 @@ export default class FirmwareUpdateV2 extends BaseMethod<Params> {
     const { features, commands } = device;
     const deviceType = getDeviceType(features);
 
-    this.checkVersionForCopyTouchResource(features);
+    const deviceFirmwareType = getFirmwareType(device.features);
+    const firmwareType = params.firmwareType ?? deviceFirmwareType;
+
+    this.checkVersionForCopyTouchResource(features, firmwareType);
 
     if (!features?.bootloader_mode && features) {
       const uuid = getDeviceUUID(features);
@@ -287,10 +296,11 @@ export default class FirmwareUpdateV2 extends BaseMethod<Params> {
       // check & upgrade firmware resource
       if (features && this.isSupportResourceUpdate(features, params.updateType)) {
         this.postTipMessage('CheckLatestUiResource');
-        const resourceUrl = DataManager.getSysResourcesLatestRelease(
+        const resourceUrl = DataManager.getSysResourcesLatestRelease({
           features,
-          params.forcedUpdateRes
-        );
+          forcedUpdateRes: params.forcedUpdateRes,
+          firmwareType,
+        });
         if (resourceUrl) {
           this.postTipMessage('DownloadLatestUiResource');
           const resource = await getSysResourceBinary(resourceUrl);
@@ -360,11 +370,13 @@ export default class FirmwareUpdateV2 extends BaseMethod<Params> {
           );
         }
         this.postTipMessage('DownloadFirmware');
+
         const firmware = await getBinary({
           features: device.features,
           version: params.version,
           updateType: params.updateType,
           isUpdateBootloader: params.isUpdateBootloader,
+          firmwareType,
         });
         binary = firmware.binary;
         this.postTipMessage('DownloadFirmwareSuccess');

package/src/api/FirmwareUpdateV3.ts

@@ -1,23 +1,26 @@
-import { Deferred, ERRORS, HardwareErrorCode, EDeviceType, wait } from '@onekeyfe/hd-shared';
+import { EDeviceType, ERRORS, HardwareErrorCode, wait } from '@onekeyfe/hd-shared';
 import semver from 'semver';
 import JSZip from 'jszip';
-import { UI_REQUEST, FirmwareUpdateTipMessage } from '../events/ui-request';
-import { validateParams } from './helpers/paramsValidator';
 
+import { FirmwareUpdateTipMessage, UI_REQUEST } from '../events/ui-request';
+import { validateParams } from './helpers/paramsValidator';
 import {
-  getDeviceType,
-  getDeviceBootloaderVersion,
+  LoggerNames,
   getDeviceBLEFirmwareVersion,
+  getDeviceBootloaderVersion,
   getDeviceFirmwareVersion,
-  LoggerNames,
+  getDeviceType,
+  getFirmwareType,
   getLogger,
 } from '../utils';
 import { getBinary, getSysResourceBinary } from './firmware/getBinary';
 import { DataManager } from '../data-manager';
-import { FirmwareUpdateV3Params } from '../types/api/firmwareUpdate';
 import { FirmwareUpdateBaseMethod } from './firmware/FirmwareUpdateBaseMethod';
 import { DevicePool } from '../device/DevicePool';
-import { TypedResponseMessage } from '../device/DeviceCommands';
+
+import type { FirmwareUpdateV3Params } from '../types/api/firmwareUpdate';
+import type { Deferred, EFirmwareType } from '@onekeyfe/hd-shared';
+import type { TypedResponseMessage } from '../device/DeviceCommands';
 
 const Log = getLogger(LoggerNames.Method);
 
@@ -54,6 +57,7 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
       { name: 'forcedUpdateRes', type: 'boolean' },
       { name: 'bootloaderVersion', type: 'array' },
       { name: 'bootloaderBinary', type: 'buffer' },
+      { name: 'firmwareType', type: 'string' },
       { name: 'platform', type: 'string' },
     ]);
 
@@ -66,6 +70,7 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
       bootloaderBinary: payload.bootloaderBinary,
       firmwareVersion: payload.firmwareVersion,
       resourceBinary: payload.resourceBinary,
+      firmwareType: payload.firmwareType,
       platform: payload.platform,
     };
   }
@@ -83,14 +88,17 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
       throw ERRORS.TypedError(HardwareErrorCode.RuntimeError, 'Device features not available');
     }
 
+    const deviceFirmwareType = getFirmwareType(features);
+    const firmwareType = this.params.firmwareType ?? deviceFirmwareType;
+
     let resourceBinary: ArrayBuffer | null = null;
     let fwBinaryMap: { fileName: string; binary: ArrayBuffer }[] = [];
     let bootloaderBinary: ArrayBuffer | null = null;
     try {
       this.postTipMessage(FirmwareUpdateTipMessage.StartDownloadFirmware);
-      resourceBinary = await this.prepareResourceBinary();
-      fwBinaryMap = await this.prepareFirmwareAndBleBinary();
-      bootloaderBinary = await this.prepareBootloaderBinary();
+      resourceBinary = await this.prepareResourceBinary(firmwareType);
+      fwBinaryMap = await this.prepareFirmwareAndBleBinary(firmwareType);
+      bootloaderBinary = await this.prepareBootloaderBinary(firmwareType);
       this.postTipMessage(FirmwareUpdateTipMessage.FinishDownloadFirmware);
     } catch (err) {
       throw ERRORS.TypedError(HardwareErrorCode.FirmwareUpdateDownloadFailed, err.message ?? err);
@@ -130,16 +138,17 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
     }
   }
 
-  private async prepareResourceBinary() {
+  private async prepareResourceBinary(firmwareType: EFirmwareType) {
     if (this.params.resourceBinary) {
       return this.params.resourceBinary;
     }
     const { features } = this.device;
     if (!features) return null;
-    const resourceUrl = DataManager.getSysResourcesLatestRelease(
+    const resourceUrl = DataManager.getSysResourcesLatestRelease({
       features,
-      this.params.forcedUpdateRes
-    );
+      forcedUpdateRes: this.params.forcedUpdateRes,
+      firmwareType,
+    });
 
     if (resourceUrl) {
       const resource = (await getSysResourceBinary(resourceUrl)).binary;
@@ -149,7 +158,7 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
     return null;
   }
 
-  private async prepareBootloaderBinary(): Promise<ArrayBuffer | null> {
+  private async prepareBootloaderBinary(firmwareType: EFirmwareType): Promise<ArrayBuffer | null> {
     if (this.params.bootloaderBinary) {
       return this.params.bootloaderBinary;
     }
@@ -157,7 +166,7 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
     if (!features) return null;
 
     if (this.params.bootloaderVersion) {
-      const bootResourceUrl = DataManager.getBootloaderResource(features);
+      const bootResourceUrl = DataManager.getBootloaderResource(features, firmwareType);
       if (bootResourceUrl) {
         const bootBinary = (await getSysResourceBinary(bootResourceUrl)).binary;
         return bootBinary;
@@ -166,7 +175,7 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
     return null;
   }
 
-  private async prepareFirmwareAndBleBinary() {
+  private async prepareFirmwareAndBleBinary(firmwareType: EFirmwareType) {
     const fwBinaryMap: { fileName: string; binary: ArrayBuffer }[] = [];
     if (this.params.firmwareBinary) {
       fwBinaryMap.push({
@@ -182,6 +191,7 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
             version: this.params.firmwareVersion,
             updateType: 'firmware',
             isUpdateBootloader: false,
+            firmwareType,
           })
         ).binary;
         fwBinaryMap.push({
@@ -203,6 +213,7 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
           features,
           version: this.params.bleVersion,
           updateType: 'ble',
+          firmwareType,
         });
         fwBinaryMap.push({
           fileName: 'ble-firmware.bin',
@@ -288,6 +299,7 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
       if (error?.errorCode) {
         const unexpectedError = [
           HardwareErrorCode.ActionCancelled,
+          HardwareErrorCode.CallQueueActionCancelled,
           HardwareErrorCode.FirmwareVerificationFailed,
           // BLE connection errors
           HardwareErrorCode.BleDeviceNotBonded,
@@ -354,6 +366,7 @@ export default class FirmwareUpdateV3 extends FirmwareUpdateBaseMethod<FirmwareU
           firmwareVersion,
         };
       } catch (error) {
+        // Hardware install firmware progress message
         if (error.message && error.message.includes('Update mode')) {
           const updateParts = error.message.split('Update mode ');
           const progressValue = updateParts[1] ?? '0';

package/src/api/allnetwork/AllNetworkGetAddressBase.ts

@@ -9,22 +9,26 @@ import {
 import { serializedPath, toHardened } from '../helpers/pathUtils';
 import { BaseMethod } from '../BaseMethod';
 import { validateParams } from '../helpers/paramsValidator';
-import { CoreApi } from '../../types';
-import type {
-  AllNetworkAddressParams,
-  INetwork,
-  AllNetworkAddress,
-  CommonResponseParams,
-} from '../../types/api/allNetworkGetAddress';
 import { PROTO } from '../../constants';
-
 import { findMethod } from '../utils';
-import { createUiMessage, DEVICE, IFRAME } from '../../events';
+import { DEVICE, IFRAME, createUiMessage } from '../../events';
 import { getDeviceFirmwareVersion, getMethodVersionRange } from '../../utils';
-import { Device, DeviceEvents } from '../../device/Device';
 import { UI_REQUEST } from '../../constants/ui-request';
 import { onDeviceButtonHandler } from '../../core';
-import { DevicePool } from '../../device/DevicePool';
+import {
+  completeRequestContext,
+  createRequestContext,
+  updateRequestContext,
+} from '../../utils/tracing';
+
+import type { Device, DeviceEvents } from '../../device/Device';
+import type { CoreApi } from '../../types';
+import type {
+  AllNetworkAddress,
+  AllNetworkAddressParams,
+  CommonResponseParams,
+  INetwork,
+} from '../../types/api/allNetworkGetAddress';
 
 const Mainnet = 'mainnet';
 
@@ -322,6 +326,27 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
 
     method.connector = this.connector;
     method.postMessage = this.postMessage;
+    if (this.context) {
+      method.setContext?.(this.context);
+    }
+
+    method.requestContext = createRequestContext(method.responseID, methodName, {
+      sdkInstanceId: this.sdkInstanceId,
+      connectId: this.payload.connectId,
+      parentResponseID: this.responseID,
+    });
+
+    const onSignalAbort = () => {
+      this.abortController?.abort(HardwareErrorCodeMessage[HardwareErrorCode.RepeatUnlocking]);
+    };
+
+    const buttonListener = (...[device, request]: [...DeviceEvents['button']]) => {
+      if (request.code === 'ButtonRequest_PinEntry' || request.code === 'ButtonRequest_AttachPin') {
+        onSignalAbort();
+      } else {
+        onDeviceButtonHandler(device, request);
+      }
+    };
 
     let result: AllNetworkAddress[];
     try {
@@ -329,23 +354,15 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
       method.setDevice?.(this.device);
       method.context = this.context;
 
-      const onSignalAbort = () => {
-        this.abortController?.abort(HardwareErrorCodeMessage[HardwareErrorCode.RepeatUnlocking]);
-      };
-
-      const _onDeviceButtonHandler = (...[device, request]: [...DeviceEvents['button']]) => {
-        if (
-          request.code === 'ButtonRequest_PinEntry' ||
-          request.code === 'ButtonRequest_AttachPin'
-        ) {
-          onSignalAbort();
-        } else {
-          onDeviceButtonHandler(device, request);
-        }
-      };
+      if (method.requestContext && this.device) {
+        updateRequestContext(method.requestContext.responseID, {
+          deviceInstanceId: this.device.instanceId,
+          commandsInstanceId: this.device.commands?.instanceId,
+        });
+      }
 
       // pro pin event
-      this.device.on(DEVICE.BUTTON, _onDeviceButtonHandler);
+      this.device.on(DEVICE.BUTTON, buttonListener);
       // classic pin event
       this.device.on(DEVICE.PIN, onSignalAbort);
       this.device.on(DEVICE.PASSPHRASE, onSignalAbort);
@@ -366,6 +383,9 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
           rootFingerprint,
         },
       }));
+      if (method.requestContext) {
+        completeRequestContext(method.requestContext.responseID);
+      }
     } catch (e: any) {
       const error = handleSkippableHardwareError(e, this.device, method);
 
@@ -384,6 +404,16 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
       } else {
         throw e;
       }
+      if (method.requestContext) {
+        completeRequestContext(
+          method.requestContext.responseID,
+          e instanceof Error ? e : new Error(String(e))
+        );
+      }
+    } finally {
+      this.device.off(DEVICE.BUTTON, buttonListener);
+      this.device.off(DEVICE.PIN, onSignalAbort);
+      this.device.off(DEVICE.PASSPHRASE, onSignalAbort);
     }
 
     return result;

package/src/api/device/DeviceFullyUploadResource.ts

@@ -4,10 +4,11 @@ import { UI_REQUEST } from '../../constants/ui-request';
 import { BaseMethod } from '../BaseMethod';
 import { getSysResourceBinary } from '../firmware/getBinary';
 import { updateResources } from '../firmware/uploadFirmware';
-import { getDeviceType, getDeviceFirmwareVersion } from '../../utils';
+import { getDeviceType, getDeviceFirmwareVersion, getFirmwareType } from '../../utils';
 import { createUiMessage } from '../../events/ui-request';
 import type { KnownDevice, Features } from '../../types';
 import { DataManager } from '../../data-manager';
+import type { DeviceFullyUploadResourceParams } from '../../types/api/deviceFullyUploadResource';
 
 export default class DeviceFullyUploadResource extends BaseMethod {
   checkPromise: Deferred<any> | null = null;
@@ -43,13 +44,18 @@ export default class DeviceFullyUploadResource extends BaseMethod {
     const { device } = this;
     const { features } = device;
 
+    const payload = this.payload as DeviceFullyUploadResourceParams;
+
+    const deviceFirmwareType = getFirmwareType(features);
+    const firmwareType = payload.firmwareType ?? deviceFirmwareType;
+
     if (!features?.bootloader_mode && features) {
       // check & upgrade firmware resource
       if (features) {
         let { binary } = this.payload;
         if (!binary) {
           this.postTipMessage('CheckLatestUiResource');
-          const resourceUrl = DataManager.getSysFullResource(features);
+          const resourceUrl = DataManager.getSysFullResource(features, firmwareType);
           if (resourceUrl) {
             this.postTipMessage('DownloadLatestUiResource');
             const resource = await getSysResourceBinary(resourceUrl);

package/src/api/device/DeviceUpdateBootloader.ts

@@ -1,4 +1,4 @@
-import { ERRORS, HardwareErrorCode } from '@onekeyfe/hd-shared';
+import { EFirmwareType, ERRORS, HardwareErrorCode } from '@onekeyfe/hd-shared';
 import { RebootType } from '@onekeyfe/hd-transport';
 import { UI_REQUEST } from '../../constants/ui-request';
 import { FirmwareUpdateTipMessage } from '../../events/ui-request';
@@ -8,10 +8,11 @@ import { updateBootloader } from '../firmware/uploadFirmware';
 import { DeviceModelToTypes } from '../../types';
 import { DataManager } from '../../data-manager';
 import { checkBootloaderLength } from '../firmware/updateBootloader';
-import { getDeviceType } from '../../utils';
+import { getDeviceType, getFirmwareType } from '../../utils';
 
 import type { Device } from '../../device/Device';
 import type { Features } from '../../types';
+import { DeviceUpdateBootloaderParams } from '../../types/api/deviceUpdateBootloader';
 
 export default class DeviceUpdateBootloader extends FirmwareUpdateBaseMethod<any> {
   init() {
@@ -41,11 +42,21 @@ export default class DeviceUpdateBootloader extends FirmwareUpdateBaseMethod<any
     return true;
   }
 
-  async updateTouchBootloader(device: Device, features?: Features) {
+  async updateTouchBootloader({
+    device,
+    features,
+    firmwareType,
+  }: {
+    device: Device;
+    features?: Features;
+    firmwareType: EFirmwareType;
+  }) {
     let { binary } = this.payload;
     if (!binary) {
       this.postTipMessage(FirmwareUpdateTipMessage.CheckLatestUiResource);
-      const resourceUrl = features ? DataManager.getBootloaderResource(features) : null;
+      const resourceUrl = features
+        ? DataManager.getBootloaderResource(features, firmwareType)
+        : null;
       if (resourceUrl) {
         this.postTipMessage(FirmwareUpdateTipMessage.DownloadLatestBootloaderResource);
         const resource = await getSysResourceBinary(resourceUrl);
@@ -83,9 +94,14 @@ export default class DeviceUpdateBootloader extends FirmwareUpdateBaseMethod<any
     const { device } = this;
     const { features } = device;
 
+    const payload = this.payload as DeviceUpdateBootloaderParams;
+
     const deviceType = getDeviceType(features);
+    const deviceFirmwareType = getFirmwareType(features);
+    const firmwareType = payload.firmwareType ?? deviceFirmwareType;
+
     if (DeviceModelToTypes.model_touch.includes(deviceType)) {
-      return this.updateTouchBootloader(device, features);
+      return this.updateTouchBootloader({ device, features, firmwareType });
     }
 
     return Promise.resolve(true);

package/src/api/firmware/getBinary.ts

@@ -1,17 +1,21 @@
 import semver from 'semver';
 import { ERRORS, HardwareErrorCode } from '@onekeyfe/hd-shared';
-import { Features } from '../../types';
+
 import { getDeviceType, httpRequest } from '../../utils';
 import { DataManager } from '../../data-manager';
 import { findLatestRelease } from '../../utils/release';
 import { getFirmwareUpdateField } from '../../utils/deviceFeaturesUtils';
-import { IFirmwareField } from '../../data-manager/DataManager';
+
+import type { Features } from '../../types';
+import type { EFirmwareType } from '@onekeyfe/hd-shared';
+import type { IFirmwareField } from '../../data-manager/DataManager';
 
 export interface GetInfoProps {
   features: Features;
   updateType: 'firmware' | 'ble';
   isUpdateBootloader?: boolean;
   targetVersion?: string;
+  firmwareType: EFirmwareType;
 }
 
 interface GetBinaryProps extends GetInfoProps {
@@ -23,8 +27,14 @@ export const getBinary = async ({
   updateType,
   version,
   isUpdateBootloader,
+  firmwareType,
 }: GetBinaryProps) => {
-  const releaseInfo = getInfo({ features, updateType, targetVersion: version?.join('.') });
+  const releaseInfo = getInfo({
+    features,
+    updateType,
+    targetVersion: version?.join('.'),
+    firmwareType,
+  });
 
   if (!releaseInfo) {
     throw ERRORS.TypedError(HardwareErrorCode.RuntimeError, 'no firmware found for this device');
@@ -71,7 +81,7 @@ export const getSysResourceBinary = async (url: string) => {
   };
 };
 
-export const getInfo = ({ features, updateType, targetVersion }: GetInfoProps) => {
+export const getInfo = ({ features, updateType, targetVersion, firmwareType }: GetInfoProps) => {
   const deviceType = getDeviceType(features);
   if (deviceType === 'unknown') {
     return null;
@@ -82,6 +92,7 @@ export const getInfo = ({ features, updateType, targetVersion }: GetInfoProps) =
     features,
     updateType,
     targetVersion,
+    firmwareType,
   });
   const releaseInfo = deviceMap?.[deviceType]?.[firmwareUpdateField] ?? [];
   return findLatestRelease(releaseInfo);

package/src/api/firmware/releaseHelper.ts

@@ -6,10 +6,12 @@ import {
   getDeviceType,
 } from '../../utils';
 
-export const getFirmwareReleaseInfo = (features: Features) => {
-  const firmwareStatus = DataManager.getFirmwareStatus(features);
-  const changelog = DataManager.getFirmwareChangelog(features);
-  const release = DataManager.getFirmwareLatestRelease(features);
+import type { EFirmwareType } from '@onekeyfe/hd-shared';
+
+export const getFirmwareReleaseInfo = (features: Features, firmwareType: EFirmwareType) => {
+  const firmwareStatus = DataManager.getFirmwareStatus(features, firmwareType);
+  const changelog = DataManager.getFirmwareChangelog(features, firmwareType);
+  const release = DataManager.getFirmwareLatestRelease(features, firmwareType);
   const bootloaderMode = !!features.bootloader_mode;
   return {
     status: firmwareStatus,
@@ -32,11 +34,16 @@ export const getBleFirmwareReleaseInfo = (features: Features) => {
   };
 };
 
-export const getBootloaderReleaseInfo = (
-  features: Features,
-  willUpdateFirmwareVersion?: string
-) => {
-  const release = DataManager.getFirmwareLatestRelease(features);
+export const getBootloaderReleaseInfo = ({
+  features,
+  willUpdateFirmwareVersion,
+  firmwareType,
+}: {
+  features: Features;
+  willUpdateFirmwareVersion?: string;
+  firmwareType: EFirmwareType;
+}) => {
+  const release = DataManager.getFirmwareLatestRelease(features, firmwareType);
   const changelog = [release?.bootloaderChangelog].filter(
     item =>
       item != null &&
@@ -52,9 +59,13 @@ export const getBootloaderReleaseInfo = (
   const deviceType = getDeviceType(features);
   // classic mini classic1s
   if (DeviceModelToTypes.model_mini.includes(deviceType)) {
-    shouldUpdate = !!checkNeedUpdateBootForClassicAndMini(features, willUpdateFirmwareVersion);
+    shouldUpdate = !!checkNeedUpdateBootForClassicAndMini({
+      features,
+      willUpdateFirmware: willUpdateFirmwareVersion,
+      firmwareType,
+    });
   } else if (DeviceModelToTypes.model_touch.includes(deviceType)) {
-    shouldUpdate = checkNeedUpdateBootForTouch(features);
+    shouldUpdate = checkNeedUpdateBootForTouch(features, firmwareType);
   }
 
   return {