@onekeyfe/hd-core 1.2.0-alpha.9 → 1.2.0-alpha.90

package/src/api/UploadPortfolio.ts

@@ -0,0 +1,58 @@
+import { createDeviceNotSupportMethodError } from '@onekeyfe/hd-shared';
+
+import { supportsProtocolV2Message } from '../protocols/protocol-v2/features';
+import FileWrite from './FileWrite';
+
+export type UploadPortfolioParams = {
+  packageBytes: ArrayBuffer | Uint8Array | Blob;
+  timeoutMs?: number | string;
+};
+
+const PORTFOLIO_PENDING_PATH = 'vol1:/portfolio/portfolio.okpkg.pending';
+const PORTFOLIO_CHUNK_SIZE = 2048;
+const FILESYSTEM_FILE_WRITE_MESSAGE_TYPE = 60805;
+const PORTFOLIO_UPDATE_MESSAGE_TYPE = 61400;
+
+export default class UploadPortfolio extends FileWrite {
+  init() {
+    const { packageBytes, timeoutMs } = this.payload as UploadPortfolioParams;
+    this.payload = {
+      ...this.payload,
+      path: PORTFOLIO_PENDING_PATH,
+      offset: 0,
+      data: packageBytes,
+      chunkSize: PORTFOLIO_CHUNK_SIZE,
+      overwrite: true,
+      append: false,
+      emitProgress: false,
+      timeoutMs,
+    };
+    super.init();
+    this.unlockPolicy = 'none';
+    // Portfolio is a background write/apply flow and never synthesizes UI events.
+    this.protocolV2UiMode = 'none';
+  }
+
+  async run() {
+    const protocolInfo = await this.device.ensureProtocolV2RuntimeContext();
+    const hasFileWrite = supportsProtocolV2Message(
+      protocolInfo,
+      FILESYSTEM_FILE_WRITE_MESSAGE_TYPE
+    );
+    const hasPortfolioUpdate = supportsProtocolV2Message(
+      protocolInfo,
+      PORTFOLIO_UPDATE_MESSAGE_TYPE
+    );
+    if (!hasFileWrite || !hasPortfolioUpdate) {
+      throw createDeviceNotSupportMethodError(this.name, this.device.getCurrentFirmwareType());
+    }
+
+    const stagedFile = await super.run();
+    this.throwIfAborted();
+    await this.device.commands.typedCall('PortfolioUpdate', 'Success', {});
+    return {
+      ...stagedFile,
+      portfolioUpdated: true,
+    };
+  }
+}

package/src/api/alephium/AlephiumGetAddress.ts

@@ -45,10 +45,6 @@ export default class AlephiumGetAddress extends BaseMethod<HardwareAlephiumGetAd
 
   getVersionRange(): DeviceFirmwareRange {
     return {
-      pro2: {
-        min: '0.0.0',
-        unsupported: true,
-      },
       model_touch: {
         min: '4.10.0',
       },

package/src/api/alephium/AlephiumSignMessage.ts

@@ -33,10 +33,6 @@ export default class AlephiumSignMessage extends BaseMethod<HardwareAlephiumSign
 
   getVersionRange(): DeviceFirmwareRange {
     return {
-      pro2: {
-        min: '0.0.0',
-        unsupported: true,
-      },
       model_touch: {
         min: '4.10.0',
       },

package/src/api/alephium/AlephiumSignTransaction.ts

@@ -42,10 +42,6 @@ export default class AlephiumSignTransaction extends BaseMethod<HardwareAlephium
 
   getVersionRange(): DeviceFirmwareRange {
     return {
-      pro2: {
-        min: '0.0.0',
-        unsupported: true,
-      },
       model_touch: {
         min: '4.10.0',
       },
@@ -63,8 +59,8 @@ export default class AlephiumSignTransaction extends BaseMethod<HardwareAlephium
     data: Buffer,
     scriptOpt?: Buffer,
     dataOffset = 0
-    // 设备可能在最后一个 AlephiumTxRequest（无 data_length）里返回签名，
-    // 该消息没有 address 字段，返回类型如实声明为联合类型
+    // The final AlephiumTxRequest may return a signature without data_length or address,
+    // so the result accurately uses a union type.
   ): Promise<AlephiumSignedTx | AlephiumTxRequest> => {
     const responseType = res.type;
     if (res.type === 'AlephiumSignedTx') {

package/src/api/algo/AlgoGetAddress.ts

@@ -6,6 +6,10 @@ import type { AlgorandGetAddress as HardwareAlgoGetAddress } from '@onekeyfe/hd-
 import type { AlgoAddress, AlgoGetAddressParams } from '../../types';
 
 export default class AlgoGetAddress extends BaseMethod<HardwareAlgoGetAddress[]> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   hasBundle = false;
 
   init() {

package/src/api/algo/AlgoSignTransaction.ts

@@ -7,6 +7,10 @@ import type { SolanaSignTransactionParams } from '../../types';
 import type { AlgorandSignTx as HardwareAlgorandSignTx } from '@onekeyfe/hd-transport';
 
 export default class AlgoSignTransaction extends BaseMethod<HardwareAlgorandSignTx> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   hasBundle = false;
 
   init() {

package/src/api/allnetwork/AllNetworkGetAddress.ts

@@ -24,24 +24,28 @@ export default class AllNetworkGetAddress extends AllNetworkGetAddressBase {
     const resultMap: Record<string, AllNetworkAddress> = {};
     const { bundle } = this.payload as AllNetworkGetAddressParams;
 
-    const methodGroups = bundle
-      .map((param, index) =>
-        this.generateMethodName({
-          network: param.network,
-          payload: param,
-          originalIndex: index,
-        })
-      )
-      .reduce((acc, cur) => {
-        if (!acc[cur.methodName]) {
-          acc[cur.methodName] = [];
-        }
-        acc[cur.methodName].push(cur);
-        return acc;
-      }, {} as Record<keyof CoreApi, MethodParams[]>);
+    const methodParams = bundle.map((param, index) =>
+      this.generateMethodName({
+        network: param.network,
+        payload: param,
+        originalIndex: index,
+      })
+    );
+    const groupedMethodParams = methodParams.reduce((groups, param) => {
+      const group = groups.get(param.methodName) ?? [];
+      group.push(param);
+      groups.set(param.methodName, group);
+      return groups;
+    }, new Map<keyof CoreApi, MethodParams[]>());
+    const requiresProtocolV2WalletHandoff =
+      this.device.isProtocolV2() &&
+      (this.payload.useEmptyPassphrase === true || !!this.payload.passphraseState);
+    const methodGroups: [keyof CoreApi, MethodParams[]][] = requiresProtocolV2WalletHandoff
+      ? methodParams.map(param => [param.methodName, [param]])
+      : Array.from(groupedMethodParams.entries());
 
     let i = 0;
-    for (const [methodName, params] of Object.entries(methodGroups)) {
+    for (const [methodName, params] of methodGroups) {
       const methodParams = {
         bundle: params.map(param => ({
           ...param.params,
@@ -52,11 +56,7 @@ export default class AllNetworkGetAddress extends AllNetworkGetAddressBase {
         throw new Error(HardwareErrorCodeMessage[HardwareErrorCode.RepeatUnlocking]);
       }
       // call method
-      const response = await this.callMethod(
-        methodName as keyof CoreApi,
-        methodParams,
-        rootFingerprint
-      );
+      const response = await this.callMethod(methodName, methodParams, rootFingerprint);
 
       if (this.abortController?.signal.aborted) {
         throw new Error(HardwareErrorCodeMessage[HardwareErrorCode.RepeatUnlocking]);

package/src/api/allnetwork/AllNetworkGetAddressBase.ts

@@ -12,9 +12,9 @@ import { validateParams } from '../helpers/paramsValidator';
 import { PROTO } from '../../constants';
 import { findMethod } from '../utils';
 import { DEVICE, IFRAME, createUiMessage } from '../../events';
-import { isMethodVersionRangeUnsupported } from '../../utils';
 import { UI_REQUEST } from '../../constants/ui-request';
 import { onDeviceButtonHandler } from '../../core';
+import { runMethodWithUnlockPolicy } from '../../protocols/protocol-v2/unlockPolicyRunner';
 import {
   completeRequestContext,
   createRequestContext,
@@ -260,6 +260,10 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
     chain_name?: string;
   }[]
 > {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   abortController: AbortController | null = null;
 
   init() {
@@ -352,6 +356,7 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
     };
 
     let result: AllNetworkAddress[];
+    let requestError: Error | undefined;
     try {
       method.init();
       method.setDevice?.(this.device);
@@ -371,16 +376,38 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
       this.device.on(DEVICE.PASSPHRASE, onSignalAbort);
 
       preCheckDeviceSupport(this.device, method);
-      if (this.temporarySafetyCheckPrompted) {
-        method.temporarySafetyCheckPrompted = true;
-      } else {
-        const appliedTemporarySafetyCheck = await method.checkSafetyLevelOnTestNet();
-        if (appliedTemporarySafetyCheck) {
-          this.temporarySafetyCheckPrompted = true;
-        }
-      }
-
-      const response = await method.run();
+      const response = await runMethodWithUnlockPolicy<any[]>(method, this.device, {
+        context: this.protocolV2UnlockContext,
+        prepare: async () => {
+          if (this.temporarySafetyCheckPrompted) {
+            method.temporarySafetyCheckPrompted = true;
+          } else {
+            const appliedTemporarySafetyCheck = await method.checkSafetyLevelOnTestNet();
+            if (appliedTemporarySafetyCheck) {
+              this.temporarySafetyCheckPrompted = true;
+            }
+          }
+
+          // Protocol V2 hands a wallet session to exactly one blockchain request.
+          // The parent all-network call consumes its first handoff while fetching
+          // the root fingerprint, so each nested chain method must resume the
+          // requested standard or hidden wallet before sending its device command.
+          const useEmptyPassphrase = this.payload.useEmptyPassphrase === true;
+          const deriveCardano = method.name.startsWith('cardano');
+          const shouldResumeWalletSession = useEmptyPassphrase || !!this.payload.passphraseState;
+          if (this.device.isProtocolV2() && shouldResumeWalletSession) {
+            const passphraseStateSafety = await this.device.checkPassphraseStateSafety(
+              this.payload.passphraseState,
+              useEmptyPassphrase,
+              this.payload.skipPassphraseCheck,
+              deriveCardano
+            );
+            if (!passphraseStateSafety) {
+              throw ERRORS.TypedError(HardwareErrorCode.DeviceCheckPassphraseStateError);
+            }
+          }
+        },
+      });
 
       if (!Array.isArray(response) || response.length === 0) {
         throw new Error('No response');
@@ -394,10 +421,8 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
           rootFingerprint,
         },
       }));
-      if (method.requestContext) {
-        completeRequestContext(method.requestContext.responseID);
-      }
     } catch (e: any) {
+      requestError = e instanceof Error ? e : new Error(String(e));
       const error = handleSkippableHardwareError(e, this.device, method);
 
       if (error) {
@@ -415,13 +440,10 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
       } else {
         throw e;
       }
+    } finally {
       if (method.requestContext) {
-        completeRequestContext(
-          method.requestContext.responseID,
-          e instanceof Error ? e : new Error(String(e))
-        );
+        completeRequestContext(method.requestContext.responseID, requestError);
       }
-    } finally {
       this.device.off(DEVICE.BUTTON, buttonListener);
       this.device.off(DEVICE.PIN, onSignalAbort);
       this.device.off(DEVICE.PASSPHRASE, onSignalAbort);
@@ -440,7 +462,9 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
       show_display: false,
     });
 
-    this.postMessage(createUiMessage(UI_REQUEST.CLOSE_UI_PIN_WINDOW));
+    if (!this.device.isProtocolV2()) {
+      this.postMessage(createUiMessage(UI_REQUEST.CLOSE_UI_PIN_WINDOW));
+    }
 
     if (res.message.root_fingerprint == null) {
       throw ERRORS.TypedError(HardwareErrorCode.CallMethodInvalidParameter);
@@ -463,13 +487,10 @@ export default abstract class AllNetworkGetAddressBase extends BaseMethod<
  * @param method BaseMethod
  */
 function preCheckDeviceSupport(device: Device, method: BaseMethod) {
+  method.assertProtocolSupported(device.getProtocol(), device.getCurrentFirmwareType());
   const versionRange = device.getCurrentMethodVersionRange(type => method.getVersionRange()[type]);
   const currentVersion = device.getCurrentFirmwareVersionString() ?? '0.0.0';
 
-  if (isMethodVersionRangeUnsupported(versionRange)) {
-    throw ERRORS.createDeviceNotSupportMethodError(method.name, device.getCurrentFirmwareType());
-  }
-
   if (
     versionRange &&
     semver.valid(versionRange.min) &&
@@ -509,14 +530,6 @@ function handleSkippableHardwareError(
     );
     const currentVersion = device.getCurrentFirmwareVersionString() ?? '0.0.0';
 
-    if (isMethodVersionRangeUnsupported(versionRange)) {
-      error = ERRORS.createDeviceNotSupportMethodError(
-        method.name,
-        device.getCurrentFirmwareType()
-      );
-      return error;
-    }
-
     if (
       versionRange &&
       semver.valid(versionRange.min) &&

package/src/api/aptos/AptosGetAddress.ts

@@ -12,6 +12,10 @@ import type { AptosAddress, AptosGetAddressParams } from '../../types';
 import type { AptosGetAddress as HardwareAptosGetAddress } from '@onekeyfe/hd-transport';
 
 export default class AptosGetAddress extends BaseMethod<HardwareAptosGetAddress[]> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   hasBundle = false;
 
   shouldConfirm = false;

package/src/api/aptos/AptosGetPublicKey.ts

@@ -7,6 +7,10 @@ import { batchGetPublickeys } from '../helpers/batchGetPublickeys';
 import type { AptosGetAddressParams, AptosPublicKey } from '../../types';
 
 export default class AptosGetPublicKey extends BaseMethod<any> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   hasBundle = false;
 
   init() {

package/src/api/aptos/AptosSignInMessage.ts

@@ -7,6 +7,10 @@ import type { AptosSignSIWAMessage } from '@onekeyfe/hd-transport';
 import type { AptosSignInMessageParams, AptosSignInMessageSignature } from '../../types';
 
 export default class AptosSignInMessage extends BaseMethod<AptosSignSIWAMessage> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   init() {
     this.checkDeviceId = true;
     this.allowDeviceMode = [...this.allowDeviceMode, UI_REQUEST.NOT_INITIALIZE];

package/src/api/aptos/AptosSignMessage.ts

@@ -7,6 +7,10 @@ import type { AptosSignMessage as HardwareAptosSignMessage } from '@onekeyfe/hd-
 import type { AptosMessageSignature, AptosSignMessageParams } from '../../types';
 
 export default class AptosSignMessage extends BaseMethod<HardwareAptosSignMessage> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   init() {
     this.checkDeviceId = true;
     this.allowDeviceMode = [...this.allowDeviceMode, UI_REQUEST.NOT_INITIALIZE];

package/src/api/aptos/AptosSignTransaction.ts

@@ -7,6 +7,10 @@ import { formatAnyHex } from '../helpers/hexUtils';
 import type { AptosSignTx as HardwareAptosSignTx } from '@onekeyfe/hd-transport';
 
 export default class AptosSignTransaction extends BaseMethod<HardwareAptosSignTx> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   init() {
     this.checkDeviceId = true;
     this.allowDeviceMode = [...this.allowDeviceMode, UI_REQUEST.NOT_INITIALIZE];

package/src/api/benfen/BenfenGetAddress.ts

@@ -1,3 +1,5 @@
+import { ERRORS, HardwareErrorCode } from '@onekeyfe/hd-shared';
+
 import { BaseMethod } from '../BaseMethod';
 import { validateParams, validateResult } from '../helpers/paramsValidator';
 import { serializedPath, validatePath } from '../helpers/pathUtils';
@@ -8,6 +10,13 @@ import { batchGetPublickeys, supportBatchPublicKeyByDevice } from '../helpers/ba
 import type { BenfenAddress, BenfenGetAddressParams, DeviceFirmwareRange } from '../../types';
 import type { BenfenGetAddress as HardwareBenfenGetAddress } from '@onekeyfe/hd-transport';
 
+const requireAddress = (address: string | undefined): string => {
+  if (address == null) {
+    throw ERRORS.TypedError(HardwareErrorCode.CallMethodError, "Field 'address' is null");
+  }
+  return address;
+};
+
 export default class BenfenGetAddress extends BaseMethod<HardwareBenfenGetAddress[]> {
   hasBundle = false;
 
@@ -48,10 +57,6 @@ export default class BenfenGetAddress extends BaseMethod<HardwareBenfenGetAddres
 
   getVersionRange(): DeviceFirmwareRange {
     return {
-      pro2: {
-        min: '0.0.0',
-        unsupported: true,
-      },
       pro: {
         min: '4.12.0',
       },
@@ -78,8 +83,7 @@ export default class BenfenGetAddress extends BaseMethod<HardwareBenfenGetAddres
             'BenfenAddress',
             param
           );
-          // proto 里 BenfenAddress.address 是 optional；正常响应必然携带
-          address = addressRes.message.address ?? '';
+          address = requireAddress(addressRes.message.address);
         } else {
           address = publicKeyToAddress(publicKey);
         }
@@ -107,7 +111,7 @@ export default class BenfenGetAddress extends BaseMethod<HardwareBenfenGetAddres
           );
           const result = {
             path: serializedPath(param.address_n),
-            address: hex2BfcAddress(res.message.address ?? ''),
+            address: hex2BfcAddress(requireAddress(res.message.address)),
           };
           if (this.shouldConfirm) {
             this.postPreviousAddressMessage(result);

package/src/api/benfen/BenfenGetPublicKey.ts

@@ -40,10 +40,6 @@ export default class BenfenGetPublicKey extends BaseMethod<any> {
 
   getVersionRange(): DeviceFirmwareRange {
     return {
-      pro2: {
-        min: '0.0.0',
-        unsupported: true,
-      },
       pro: {
         min: '4.12.0',
       },

package/src/api/benfen/BenfenSignMessage.ts

@@ -29,10 +29,6 @@ export default class BenfenSignMessage extends BaseMethod<HardwareBenfenSignMess
 
   getVersionRange(): DeviceFirmwareRange {
     return {
-      pro2: {
-        min: '0.0.0',
-        unsupported: true,
-      },
       pro: {
         min: '4.12.0',
       },

package/src/api/benfen/BenfenSignTransaction.ts

@@ -36,10 +36,6 @@ export default class BenfenSignTransaction extends BaseMethod<BenfenSignTx> {
 
   getVersionRange(): DeviceFirmwareRange {
     return {
-      pro2: {
-        min: '0.0.0',
-        unsupported: true,
-      },
       pro: {
         min: '4.12.0',
       },

package/src/api/btc/BTCGetAddress.ts

@@ -3,7 +3,10 @@ import { getScriptType, serializedPath, validatePath } from '../helpers/pathUtil
 import { BaseMethod } from '../BaseMethod';
 import { validateParams, validateResult } from '../helpers/paramsValidator';
 import { getCoinInfo } from './helpers/btcParamsUtils';
-import { getBitcoinForkVersionRange } from './helpers/versionLimit';
+import {
+  getBitcoinForkSupportedProtocols,
+  getBitcoinForkVersionRange,
+} from './helpers/versionLimit';
 
 import type { BTCAddress, BTCGetAddressParams } from '../../types/api/btcGetAddress';
 import type { GetAddress } from '@onekeyfe/hd-transport';
@@ -11,6 +14,10 @@ import type { GetAddress } from '@onekeyfe/hd-transport';
 export default class BTCGetAddress extends BaseMethod<GetAddress[]> {
   hasBundle = false;
 
+  getSupportedProtocols() {
+    return getBitcoinForkSupportedProtocols(this.params?.map(param => param.coin_name) ?? []);
+  }
+
   init() {
     this.checkDeviceId = true;
     this.allowDeviceMode = [...this.allowDeviceMode, UI_REQUEST.NOT_INITIALIZE];

package/src/api/btc/BTCGetPublicKey.ts

@@ -5,7 +5,10 @@ import { getScriptType, isTaprootPath, serializedPath, validatePath } from '../h
 import { BaseMethod } from '../BaseMethod';
 import { validateParams, validateResult } from '../helpers/paramsValidator';
 import { getCoinInfo } from './helpers/btcParamsUtils';
-import { getBitcoinForkVersionRange } from './helpers/versionLimit';
+import {
+  getBitcoinForkSupportedProtocols,
+  getBitcoinForkVersionRange,
+} from './helpers/versionLimit';
 import { batchGetPublickeys } from '../helpers/batchGetPublickeys';
 import { createExtendedPublicKey, getVersionBytes } from './helpers/xpubUtils';
 
@@ -14,6 +17,10 @@ import type { BTCGetAddressParams } from '../../types/api/btcGetAddress';
 import type { GetPublicKey } from '@onekeyfe/hd-transport';
 
 export default class BTCGetPublicKey extends BaseMethod<GetPublicKey[]> {
+  getSupportedProtocols() {
+    return getBitcoinForkSupportedProtocols(this.params?.map(param => param.coin_name) ?? []);
+  }
+
   hasBundle = false;
 
   init() {
@@ -78,7 +85,7 @@ export default class BTCGetPublicKey extends BaseMethod<GetPublicKey[]> {
       }
 
       for (const param of this.params) {
-        // init() 必然设置 coin_name；生成类型里该字段为 optional，这里兜底空串
+        // init() sets coin_name; keep an empty fallback for the generated optional type.
         const versionBytes = getVersionBytes(param.coin_name ?? '', param.script_type);
         if (!versionBytes) {
           throw new Error(

package/src/api/btc/BTCSignMessage.ts

@@ -4,11 +4,18 @@ import { BaseMethod } from '../BaseMethod';
 import { validateParams } from '../helpers/paramsValidator';
 import { formatAnyHex } from '../helpers/hexUtils';
 import { getCoinAndScriptType } from './helpers/btcParamsUtils';
-import { getBitcoinForkVersionRange } from './helpers/versionLimit';
+import {
+  getBitcoinForkSupportedProtocols,
+  getBitcoinForkVersionRange,
+} from './helpers/versionLimit';
 
 import type { SignMessage } from '@onekeyfe/hd-transport';
 
 export default class BTCSignMessage extends BaseMethod<SignMessage> {
+  getSupportedProtocols() {
+    return getBitcoinForkSupportedProtocols([this.params?.coin_name]);
+  }
+
   init() {
     this.checkDeviceId = true;
     this.allowDeviceMode = [...this.allowDeviceMode, UI_REQUEST.NOT_INITIALIZE];

package/src/api/btc/BTCSignPsbt.ts

@@ -9,6 +9,10 @@ import { getCoinInfo } from './helpers/btcParamsUtils';
 import type { SignPsbt } from '@onekeyfe/hd-transport';
 
 export default class BTCSignPsbt extends BaseMethod<SignPsbt> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   init() {
     this.checkDeviceId = true;
     this.allowDeviceMode = [...this.allowDeviceMode, UI_REQUEST.NOT_INITIALIZE];

package/src/api/btc/BTCSignTransaction.ts

@@ -7,7 +7,10 @@ import { validateParams } from '../helpers/paramsValidator';
 import signtx from './helpers/signtx';
 import signtxLegacy from './helpers/signtxLegacy';
 import { getCoinInfo } from './helpers/btcParamsUtils';
-import { getBitcoinForkVersionRange } from './helpers/versionLimit';
+import {
+  getBitcoinForkSupportedProtocols,
+  getBitcoinForkVersionRange,
+} from './helpers/versionLimit';
 
 import type {
   AccountAddresses,
@@ -26,6 +29,10 @@ type Params = {
   coinName: string;
 };
 export default class BTCSignTransaction extends BaseMethod<Params> {
+  getSupportedProtocols() {
+    return getBitcoinForkSupportedProtocols([this.params?.coinName]);
+  }
+
   init() {
     this.checkDeviceId = true;
     this.allowDeviceMode = [...this.allowDeviceMode, UI_REQUEST.NOT_INITIALIZE];

package/src/api/btc/BTCVerifyMessage.ts

@@ -3,11 +3,18 @@ import { BaseMethod } from '../BaseMethod';
 import { validateParams } from '../helpers/paramsValidator';
 import { formatAnyHex } from '../helpers/hexUtils';
 import { getCoinInfo } from './helpers/btcParamsUtils';
-import { getBitcoinForkVersionRange } from './helpers/versionLimit';
+import {
+  getBitcoinForkSupportedProtocols,
+  getBitcoinForkVersionRange,
+} from './helpers/versionLimit';
 
 import type { VerifyMessage } from '@onekeyfe/hd-transport';
 
 export default class BTCVerifyMessage extends BaseMethod<VerifyMessage> {
+  getSupportedProtocols() {
+    return getBitcoinForkSupportedProtocols([this.params?.coin_name]);
+  }
+
   init() {
     this.checkDeviceId = true;
     this.allowDeviceMode = [...this.allowDeviceMode, UI_REQUEST.NOT_INITIALIZE];

package/src/api/btc/helpers/versionLimit.ts

@@ -1,4 +1,5 @@
 import type { DeviceFirmwareRange } from '../../../types';
+import type { ProtocolType } from '@onekeyfe/hd-transport';
 
 function isCoinNameInList(coinName: string, coinNames: (string | undefined)[]) {
   for (let i = 0; i < coinNames.length; i++) {
@@ -13,10 +14,6 @@ function isCoinNameInList(coinName: string, coinNames: (string | undefined)[]) {
 export function getBitcoinForkVersionRange(params: (string | undefined)[]): DeviceFirmwareRange {
   if (isCoinNameInList('Neurai', params)) {
     return {
-      pro2: {
-        min: '0.0.0',
-        unsupported: true,
-      },
       model_mini: {
         min: '3.7.0',
       },
@@ -29,3 +26,9 @@ export function getBitcoinForkVersionRange(params: (string | undefined)[]): Devi
   // No version restrictions for other coins
   return {};
 }
+
+export function getBitcoinForkSupportedProtocols(
+  params: (string | undefined)[]
+): readonly ProtocolType[] {
+  return isCoinNameInList('Neurai', params) ? ['V1'] : ['V1', 'V2'];
+}

package/src/api/cardano/CardanoGetAddress.ts

@@ -13,6 +13,10 @@ import type { CardanoAddress, CardanoGetAddressParams } from '../../types/api/ca
 import type { CardanoGetAddressMethodParams } from '../../types';
 
 export default class CardanoGetAddress extends BaseMethod<CardanoGetAddressParams[]> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   hasBundle?: boolean;
 
   isCheck?: boolean;

package/src/api/cardano/CardanoGetPublicKey.ts

@@ -7,6 +7,10 @@ import { serializedPath, validatePath } from '../helpers/pathUtils';
 import type { CardanoPublicKey, CardanoPublicKeyParams } from '../../types/api/cardanoGetPublicKey';
 
 export default class CardanoGetPublicKey extends BaseMethod<CardanoPublicKeyParams[]> {
+  getSupportedProtocols() {
+    return ['V1', 'V2'] as const;
+  }
+
   hasBundle?: boolean;
 
   init() {