feat: optimize version limit

Author: ByteZhang1024

packages/core/src/api/BaseMethod.ts

@@ -1,3 +1,5 @@
+import semver from 'semver';
+import { ERRORS, HardwareError, HardwareErrorCode } from '@onekeyfe/hd-shared';
 import { supportInputPinOnSoftware, supportModifyHomescreen } from '../utils/deviceFeaturesUtils';
 import { createDeviceMessage } from '../events/device';
 import { UI_REQUEST } from '../constants/ui-request';
@@ -6,7 +8,8 @@ import DeviceConnector from '../device/DeviceConnector';
 import { DeviceFirmwareRange, KnownDevice } from '../types';
 import { CoreMessage, createFirmwareMessage, createUiMessage, DEVICE, FIRMWARE } from '../events';
 import { getBleFirmwareReleaseInfo, getFirmwareReleaseInfo } from './firmware/releaseHelper';
-import { getLogger, LoggerNames } from '../utils';
+import { getDeviceFirmwareVersion, getLogger, getMethodVersionRange, LoggerNames } from '../utils';
+import { DataManager } from '../data-manager';
 
 const Log = getLogger(LoggerNames.Method);
 
@@ -81,6 +84,12 @@ export abstract class BaseMethod<Params = undefined> {
    */
   skipForceUpdateCheck = false;
 
+  /**
+   * 是否需要预先检查版本限制
+   * @default false
+   */
+  preCheckVersionLimit = false;
+
   // @ts-expect-error: strictPropertyInitialization
   postMessage: (message: CoreMessage) => void;
 
@@ -127,6 +136,46 @@ export abstract class BaseMethod<Params = undefined> {
     );
   }
 
+  handleUnsupportedMethodError(error: HardwareError) {
+    if (!error.message.includes('Failure_UnexpectedMessage')) {
+      return undefined;
+    }
+
+    const versionRange = getMethodVersionRange(
+      this.device.features,
+      type => this.getVersionRange()[type]
+    );
+
+    if (!versionRange || !this.device.features) {
+      return ERRORS.TypedError(HardwareErrorCode.UnsupportedMethod);
+    }
+    const newVersionStatus = DataManager.getFirmwareStatus(this.device.features);
+    const currentVersion = getDeviceFirmwareVersion(this.device.features).join('.');
+    if (semver.valid(versionRange.min) && semver.lt(currentVersion, versionRange.min)) {
+      if (newVersionStatus === 'none' || newVersionStatus === 'valid') {
+        throw ERRORS.TypedError(HardwareErrorCode.NewFirmwareUnRelease);
+      }
+
+      return ERRORS.TypedError(
+        HardwareErrorCode.CallMethodNeedUpgradeFirmware,
+        `Device firmware version is too low, please update to ${versionRange.min}`,
+        { current: currentVersion, require: versionRange.min }
+      );
+    }
+
+    if (
+      versionRange.max &&
+      semver.valid(versionRange.max) &&
+      semver.gte(currentVersion, versionRange.max)
+    ) {
+      return ERRORS.TypedError(
+        HardwareErrorCode.CallMethodDeprecated,
+        `Device firmware version is too high, this method has been deprecated in ${versionRange.max}`,
+        { current: currentVersion, deprecated: versionRange.max }
+      );
+    }
+  }
+
   checkDeviceSupportFeature() {
     if (!this.device || !this.device.features) return;
     const inputPinOnSoftware = supportInputPinOnSoftware(this.device.features);

packages/core/src/api/btc/BTCGetAddress.ts

@@ -10,6 +10,8 @@ import { getBitcoinForkVersionRange } from './helpers/versionLimit';
 export default class BTCGetAddress extends BaseMethod<GetAddress[]> {
   hasBundle = false;
 
+  preCheckVersionLimit = true;
+
   init() {
     this.checkDeviceId = true;
     this.notAllowDeviceMode = [...this.notAllowDeviceMode, UI_REQUEST.INITIALIZE];

packages/core/src/api/btc/BTCGetPublicKey.ts

@@ -11,6 +11,8 @@ import { getBitcoinForkVersionRange } from './helpers/versionLimit';
 export default class BTCGetPublicKey extends BaseMethod<GetPublicKey[]> {
   hasBundle = false;
 
+  preCheckVersionLimit = true;
+
   init() {
     this.checkDeviceId = true;
     this.notAllowDeviceMode = [...this.notAllowDeviceMode, UI_REQUEST.INITIALIZE];

packages/core/src/api/btc/BTCSignMessage.ts

@@ -8,6 +8,8 @@ import { getCoinAndScriptType } from './helpers/btcParamsUtils';
 import { getBitcoinForkVersionRange } from './helpers/versionLimit';
 
 export default class BTCSignMessage extends BaseMethod<SignMessage> {
+  preCheckVersionLimit = true;
+
   init() {
     this.checkDeviceId = true;
     this.notAllowDeviceMode = [...this.notAllowDeviceMode, UI_REQUEST.INITIALIZE];

packages/core/src/api/btc/BTCSignPsbt.ts

@@ -6,6 +6,8 @@ import { formatAnyHex } from '../helpers/hexUtils';
 import { getCoinInfo } from './helpers/btcParamsUtils';
 
 export default class BTCSignPsbt extends BaseMethod<SignPsbt> {
+  preCheckVersionLimit = true;
+
   init() {
     this.checkDeviceId = true;
     this.notAllowDeviceMode = [...this.notAllowDeviceMode, UI_REQUEST.INITIALIZE];

packages/core/src/api/btc/BTCSignTransaction.ts

@@ -25,6 +25,8 @@ type Params = {
   coinName: string;
 };
 export default class BTCSignTransaction extends BaseMethod<Params> {
+  preCheckVersionLimit = true;
+
   init() {
     this.checkDeviceId = true;
     this.notAllowDeviceMode = [...this.notAllowDeviceMode, UI_REQUEST.INITIALIZE];

packages/core/src/api/btc/BTCVerifyMessage.ts

@@ -8,6 +8,8 @@ import { getCoinInfo } from './helpers/btcParamsUtils';
 import { getBitcoinForkVersionRange } from './helpers/versionLimit';
 
 export default class BTCVerifyMessage extends BaseMethod<VerifyMessage> {
+  preCheckVersionLimit = true;
+
   init() {
     this.checkDeviceId = true;
     this.notAllowDeviceMode = [...this.notAllowDeviceMode, UI_REQUEST.INITIALIZE];

packages/core/src/api/evm/EVMSignTransaction.ts

@@ -10,6 +10,8 @@ import { signTransaction } from './latest/signTransaction';
 import { signTransaction as signTransactionLegacyV1 } from './legacyV1/signTransaction';
 
 export default class EVMSignTransaction extends BaseMethod {
+  preCheckVersionLimit = true;
+
   addressN: number[] = [];
 
   isEIP1559 = false;

packages/core/src/api/polkadot/PolkadotGetAddress.ts

@@ -9,6 +9,8 @@ import { getPolkadotVersionRangeWithBundle } from './networks';
 export default class PolkadotGetAddress extends BaseMethod<HardwarePolkadotGetAddress[]> {
   hasBundle = false;
 
+  preCheckVersionLimit = true;
+
   init() {
     this.checkDeviceId = true;
     this.notAllowDeviceMode = [...this.notAllowDeviceMode];

packages/core/src/api/polkadot/PolkadotSignTransaction.ts

@@ -9,6 +9,8 @@ import { getPolkadotVersionRange } from './networks';
 export default class PolkadotSignTransaction extends BaseMethod<HardwarePolkadotSignTx> {
   hasBundle = false;
 
+  preCheckVersionLimit = true;
+
   init() {
     this.checkDeviceId = true;
     this.notAllowDeviceMode = [...this.notAllowDeviceMode];