@onekeyfe/hd-core 1.2.0-alpha.66 → 1.2.0-alpha.67

package/src/device/DeviceConnector.ts

@@ -14,7 +14,7 @@ import type { OneKeyDeviceInfo as DeviceDescriptor, Transport } from '@onekeyfe/
 const Log = getLogger(LoggerNames.DeviceConnector);
 
 export default class DeviceConnector {
-  transport?: Transport;
+  transport: Transport;
 
   listenTimestamp = 0;
 
@@ -30,21 +30,9 @@ export default class DeviceConnector {
     DevicePool.setConnector(this);
   }
 
-  private getActiveTransport(): Transport {
-    const transport = this.transport ?? TransportManager.getTransport();
-    if (!transport) {
-      throw ERRORS.TypedError(
-        HardwareErrorCode.TransportNotConfigured,
-        'Device connector was created before transport initialization'
-      );
-    }
-    this.transport = transport;
-    return transport;
-  }
-
   async enumerate() {
     try {
-      const descriptors = await this.getActiveTransport().enumerate();
+      const descriptors = await this.transport.enumerate();
       this.upcoming = descriptors;
       this._reportDevicesChange();
       return { descriptors } as DeviceDescriptorDiff;
@@ -62,10 +50,11 @@ export default class DeviceConnector {
     let descriptors: DeviceDescriptor[];
 
     try {
-      const transport = this.getActiveTransport();
       Log.debug('Start listening', current);
       this.listenTimestamp = new Date().getTime();
-      descriptors = waitForEvent ? await transport.listen(current) : await transport.enumerate();
+      descriptors = waitForEvent
+        ? await this.transport.listen(current)
+        : await this.transport.enumerate();
       if (!this.listening) return; // do not continue if stop() was called
 
       this.upcoming = descriptors;
@@ -99,17 +88,16 @@ export default class DeviceConnector {
     Log.debug('acquire', path, session, expectedProtocol, protocolHint);
     const env = DataManager.getSettings('env');
     try {
-      const transport = this.getActiveTransport();
       let res;
       if (DataManager.isBleConnect(env)) {
-        res = await transport.acquire({
+        res = await this.transport.acquire({
           uuid: path,
           forceCleanRunPromise,
           expectedProtocol,
           protocolHint,
         });
       } else {
-        res = await transport.acquire({
+        res = await this.transport.acquire({
           path,
           previous: session ?? null,
           expectedProtocol,
@@ -117,13 +105,7 @@ export default class DeviceConnector {
         });
       }
       if (expectedProtocol) {
-        // The acquire response is the stable snapshot from this active probe. A delayed
-        // disconnect from an older generation may clear caches, so do not rely on cache alone.
-        const acquiredProtocol =
-          typeof res === 'object' && res !== null
-            ? (res as DeviceDescriptor).protocolType
-            : undefined;
-        const detectedProtocol = acquiredProtocol ?? transport.getProtocolType(path);
+        const detectedProtocol = this.transport.getProtocolType(path);
         if (detectedProtocol !== expectedProtocol) {
           throw ERRORS.TypedError(
             HardwareErrorCode.RuntimeError,
@@ -140,7 +122,7 @@ export default class DeviceConnector {
 
   async release(session: string, onclose: boolean) {
     try {
-      const res = await this.getActiveTransport().release(session, onclose);
+      const res = await this.transport.release(session, onclose);
       return res;
     } catch (error) {
       safeThrowError(error);
@@ -149,9 +131,8 @@ export default class DeviceConnector {
 
   async disconnect(session: string | undefined | null) {
     try {
-      const transport = this.getActiveTransport();
-      if (transport.disconnect && !!session) {
-        await transport.disconnect(session);
+      if (this.transport.disconnect && !!session) {
+        await this.transport.disconnect(session);
       }
     } catch (error) {
       safeThrowError(error);
@@ -159,11 +140,10 @@ export default class DeviceConnector {
   }
 
   promptDeviceAccess(): Promise<USBDevice | BluetoothDevice | null> {
-    const transport = this.getActiveTransport();
-    if (!transport.promptDeviceAccess) {
+    if (!this.transport.promptDeviceAccess) {
       return Promise.resolve(null);
     }
-    return transport.promptDeviceAccess();
+    return this.transport.promptDeviceAccess();
   }
 
   _reportDevicesChange() {

package/src/device/DevicePool.ts

@@ -103,7 +103,7 @@ export class DevicePool extends EventEmitter {
       const device = this.devicesCache[connectId];
       if (device) {
         const exist = descriptorList.find(d => d.path === device.originalDescriptor.path);
-        if (exist && !initOptions?.forceProtocolDetection) {
+        if (exist) {
           // Log.debug('find existed Device: ', connectId);
           device.updateDescriptor(exist, true);
           await this._refreshRuntimeState(device, initOptions);
@@ -151,23 +151,12 @@ export class DevicePool extends EventEmitter {
     }
   }
 
-  static async _createDevice(
-    descriptor: DeviceDescriptor,
-    initOptions?: InitOptions
-  ): Promise<Device> {
-    const cachedDevice = this.getDeviceByPath(descriptor.path);
-    const forceProtocolDetection = initOptions?.forceProtocolDetection === true;
-    const isNewDevice = !cachedDevice;
-    const device = cachedDevice ?? Device.fromDescriptor(descriptor);
-    if (isNewDevice || forceProtocolDetection) {
+  static async _createDevice(descriptor: DeviceDescriptor, initOptions?: InitOptions) {
+    let device = this.getDeviceByPath(descriptor.path);
+    if (!device) {
+      device = Device.fromDescriptor(descriptor);
       device.deviceConnector = this.connector;
-      if (forceProtocolDetection && !isNewDevice) {
-        await device.acquire(initOptions?.connectProtocol, { forceProtocolDetection: true });
-      } else {
-        await device.connect(initOptions?.connectProtocol, {
-          forceProtocolDetection: initOptions?.forceProtocolDetection,
-        });
-      }
+      await device.connect(initOptions?.connectProtocol);
       try {
         await device.initialize(initOptions);
         if (initOptions?.refreshRuntimeState && device.isProtocolV2()) {
@@ -192,10 +181,7 @@ export class DevicePool extends EventEmitter {
           refreshError = error;
         }
       },
-      {
-        connectProtocol: initOptions.connectProtocol,
-        forceProtocolDetection: initOptions.forceProtocolDetection,
-      }
+      { connectProtocol: initOptions.connectProtocol }
     );
     if (refreshError instanceof Error) throw refreshError;
     if (refreshError) throw new Error(String(refreshError));

package/src/deviceProfile/buildDeviceFeatures.ts

@@ -1,4 +1,4 @@
-import { EDeviceType, EFirmwareType } from '@onekeyfe/hd-shared';
+import { EFirmwareType } from '@onekeyfe/hd-shared';
 
 import {
   resolveDeviceBleFirmwareVersion,
@@ -386,7 +386,7 @@ export const buildProtocolV2FeaturesPayload = ({
     device_id: deviceId ?? undefined,
     session_id: cached?.sessionId ?? undefined,
     ble_name: bleName ?? undefined,
-    onekey_device_type: deviceType === EDeviceType.Unknown ? undefined : deviceType.toUpperCase(),
+    onekey_device_type: 'PRO2',
     passphrase_protection: passphraseProtection ?? undefined,
     bootloader_mode: bootloaderMode,
     passphraseState: cached?.passphraseState,

package/src/deviceProfile/protocolV2DeviceIdentity.ts

@@ -26,9 +26,6 @@ export const resolveProtocolV2DeviceIdentity = (
     case DeviceType.PRO2:
     case 'PRO2':
       return { deviceType: EDeviceType.Pro2, model: 'pro2' };
-    case DeviceType.NEO:
-    case 'NEO':
-      return { deviceType: EDeviceType.Neo, model: 'neo' };
     default:
       return { deviceType: EDeviceType.Unknown, model: null };
   }

package/src/events/device.ts

@@ -1,5 +1,4 @@
 import type { MessageFactoryFn } from './utils';
-import type { HardwareUiInteractionMeta } from './ui-request';
 import type { PROTO } from '../constants';
 import type {
   KnownDevice as Device,
@@ -27,8 +26,6 @@ export const DEVICE = {
   // onekey-transport events in protobuf format
   BUTTON: 'button',
   PIN: 'pin',
-  PIN_ON_DEVICE: 'pin_on_device',
-  PIN_ON_DEVICE_COMPLETE: 'pin_on_device_complete',
   PASSPHRASE: 'passphrase',
   PASSPHRASE_ON_DEVICE: 'passphrase_on_device',
   ATTACH_PIN_ON_DEVICE: 'attach_pin_on_device',
@@ -61,7 +58,6 @@ export type PassphraseRequestPayload = {
   source?: 'wallet-session-coordinator';
   reason?: 'open-wallet' | 'session-recovery';
   expectedPassphraseState?: string;
-  interaction?: HardwareUiInteractionMeta;
 };
 
 export interface DeviceButtonRequest {

package/src/events/logBlockEvent.ts

@@ -10,7 +10,6 @@ export const LogBlockEvent: Set<string> = new Set([
 
 const LogLabelMethod: Set<string> = new Set([
   'openWalletSession',
-  'deviceUploadNft',
   'deviceUploadWallpaper',
   'uploadPortfolio',
   'fileWrite',

package/src/events/ui-promise.ts

@@ -1,7 +1,7 @@
 import type { Deferred } from '@onekeyfe/hd-shared';
 import type { DEVICE } from './device';
 import type { Device } from '../device/Device';
-import type { UiResponseCorrelation, UiResponseEvent } from './ui-response';
+import type { UiResponseEvent } from './ui-response';
 
 export type UiPromiseResponse =
   | UiResponseEvent
@@ -11,6 +11,4 @@ export type UiPromise<T extends UiPromiseResponse['type']> = Deferred<
   Extract<UiPromiseResponse, { type: T }>,
   T,
   Device
-> & {
-  responseCorrelation?: UiResponseCorrelation;
-};
+>;

package/src/events/ui-request.ts

@@ -1,7 +1,6 @@
 import type { PROTO } from '../constants';
 import type { Device } from '../types';
 import type { DeviceButtonRequest } from './device';
-import type { UiResponseCorrelation } from './ui-response';
 import type { MessageFactoryFn } from './utils';
 
 export const UI_EVENT = 'UI_EVENT';
@@ -57,17 +56,6 @@ export type ProtocolV2UiEventSource =
 
 export type ProtocolV2UiCompletion = 'page-accepted' | 'operation-completed';
 
-export type HardwareUiInteractionMeta = {
-  interactionId: string;
-  phaseId: string;
-  sequence: number;
-  phase: 'pin' | 'passphrase' | 'passphrase-on-device' | 'button' | 'processing';
-  transition: 'start' | 'complete' | 'finish';
-  outcome?: 'submitted' | 'succeeded' | 'failed' | 'cancelled' | 'disconnected';
-  protocol: 'V2';
-  device?: Device;
-};
-
 export type ProtocolV2UiEventMetadata = {
   source?: ProtocolV2UiEventSource;
   reason?: string;
@@ -76,29 +64,12 @@ export type ProtocolV2UiEventMetadata = {
   method?: string;
   page?: string | number;
   operation?: string;
-  interaction?: HardwareUiInteractionMeta;
 };
 
-export type UiRequestWindowClose =
-  | {
-      type: typeof UI_REQUEST.CLOSE_UI_WINDOW;
-      payload: HardwareUiInteractionMeta;
-    }
-  | {
-      type: typeof UI_REQUEST.CLOSE_UI_PIN_WINDOW;
-      payload: HardwareUiInteractionMeta;
-    }
-  | {
-      type: typeof UI_REQUEST.CLOSE_UI_WINDOW;
-      payload?: undefined;
-    }
-  | {
-      type: typeof UI_REQUEST.CLOSE_UI_PIN_WINDOW;
-      payload?: undefined;
-    };
-
 export interface UiRequestWithoutPayload {
   type:
+    | typeof UI_REQUEST.CLOSE_UI_WINDOW
+    | typeof UI_REQUEST.CLOSE_UI_PIN_WINDOW
     | typeof UI_REQUEST.BLUETOOTH_PERMISSION
     | typeof UI_REQUEST.BLUETOOTH_UNSUPPORTED
     | typeof UI_REQUEST.BLUETOOTH_POWERED_OFF
@@ -122,7 +93,6 @@ export type UiRequestDeviceAction = {
   payload: ProtocolV2UiEventMetadata & {
     device: Device;
     type?: PROTO.PinMatrixRequestType | 'ButtonRequest_PinEntry' | 'ButtonRequest_AttachPin';
-    responseCorrelation?: UiResponseCorrelation;
   };
 };
 
@@ -141,8 +111,6 @@ export interface UiRequestPassphrase {
     source?: 'wallet-session-coordinator';
     reason?: 'open-wallet' | 'session-recovery';
     expectedPassphraseState?: string;
-    interaction?: HardwareUiInteractionMeta;
-    responseCorrelation?: UiResponseCorrelation;
   };
 }
 
@@ -153,7 +121,6 @@ export interface UiRequestPassphraseOnDevice {
     passphraseState?: string;
     source?: 'wallet-session-coordinator';
     reason?: 'open-wallet' | 'session-recovery';
-    interaction?: HardwareUiInteractionMeta;
   };
 }
 
@@ -224,7 +191,6 @@ export interface PreviousAddressResult {
 
 export type UiEvent =
   | UiRequestWithoutPayload
-  | UiRequestWindowClose
   | UiRequestDeviceAction
   | UiRequestButton
   | UiRequestPassphraseOnDevice

package/src/events/ui-response.ts

@@ -11,22 +11,12 @@ export const UI_RESPONSE = {
     'ui-receive_select-device-for-switch-firmware-web-device',
 } as const;
 
-export interface UiResponseCorrelation {
-  interactionId: string;
-  deviceId: string;
-}
-
-export interface UiResponseCorrelationFields {
-  interactionId?: string;
-  deviceId?: string;
-}
-
-export interface UiResponsePin extends UiResponseCorrelationFields {
+export interface UiResponsePin {
   type: typeof UI_RESPONSE.RECEIVE_PIN;
   payload: string;
 }
 
-export interface UiResponsePassphrase extends UiResponseCorrelationFields {
+export interface UiResponsePassphrase {
   type: typeof UI_RESPONSE.RECEIVE_PASSPHRASE;
   payload: {
     value: string;

package/src/index.ts

@@ -19,6 +19,11 @@ export * from './types';
 export { whitelist, whitelistExtension } from './data/config';
 export { executeCallback, cleanupCallback };
 export { preloadSessionCache } from './device/Device';
+export {
+  getFirmwareUpdateHostBindingGeneration,
+  registerFirmwareUpdateHostBinding,
+  unregisterFirmwareUpdateHostBinding,
+} from './api/firmware/FirmwareHostBinding';
 
 const HardwareSdk = ({
   init,

package/src/inject.ts

@@ -1,7 +1,17 @@
-import type { HardwareConnectProtocol } from '@onekeyfe/hd-shared';
-import type { Unsuccessful } from './types';
+import { getFirmwareUpdateCapabilities } from './api/firmware/FirmwareUpdateCapabilities';
+import {
+  prepareFirmwareUpdatePlan,
+  validateFirmwareUpdatePreparedPlan,
+} from './api/firmware/FirmwareUpdatePreparedPlan';
+import {
+  getFirmwareUpdateHostBindingGeneration,
+  registerFirmwareUpdateHostBinding,
+  unregisterFirmwareUpdateHostBinding,
+} from './api/firmware/FirmwareHostBinding';
+
 import type { EventEmitter } from 'events';
 import type { CallMethod } from './events';
+import type { Unsuccessful } from './types';
 import type { CoreApi } from './types/api';
 import type { AllNetworkAddress } from './types/api/allNetworkGetAddress';
 
@@ -40,50 +50,6 @@ export interface InjectApi {
   switchTransport: CoreApi['switchTransport'];
 }
 
-const normalizeConnectId = (connectId: string) => connectId.trim().toLowerCase();
-
-/**
- * Keeps verified protocols isolated per transport endpoint and injects them at
- * the single public call boundary, including APIs that do not expose params.
- */
-export const createProtocolAwareCall = (rawCall: CoreApi['call']) => {
-  const protocolByConnectId = new Map<string, HardwareConnectProtocol>();
-
-  const setDeviceConnectProtocol: CoreApi['setDeviceConnectProtocol'] = (
-    connectId,
-    connectProtocol
-  ) => {
-    const normalizedConnectId = normalizeConnectId(connectId);
-    if (!normalizedConnectId) return;
-    if (connectProtocol) {
-      protocolByConnectId.set(normalizedConnectId, connectProtocol);
-    } else {
-      protocolByConnectId.delete(normalizedConnectId);
-    }
-  };
-
-  const call: CoreApi['call'] = params => {
-    if (!params || typeof params !== 'object') {
-      return rawCall(params);
-    }
-
-    const connectId = typeof params.connectId === 'string' ? params.connectId : undefined;
-    const boundProtocol = connectId
-      ? protocolByConnectId.get(normalizeConnectId(connectId))
-      : undefined;
-    if (
-      boundProtocol &&
-      params.connectProtocol === undefined &&
-      params.forceProtocolDetection !== true
-    ) {
-      return rawCall({ ...params, connectProtocol: boundProtocol });
-    }
-    return rawCall(params);
-  };
-
-  return { call, setDeviceConnectProtocol };
-};
-
 export const inject = ({
   call,
   cancel,
@@ -94,7 +60,6 @@ export const inject = ({
   switchTransport,
   uiResponse,
 }: InjectApi): CoreApi => {
-  const protocolAwareCall = createProtocolAwareCall(call);
   const api: CoreApi = {
     on: <T extends string, P extends (...args: any[]) => any>(type: T, fn: P) => {
       eventEmitter.on(type, fn);
@@ -112,11 +77,12 @@ export const inject = ({
 
     init,
 
-    call: protocolAwareCall.call,
+    call,
 
-    setDeviceConnectProtocol: protocolAwareCall.setDeviceConnectProtocol,
-
-    dispose,
+    dispose: async () => {
+      unregisterFirmwareUpdateHostBinding();
+      await dispose();
+    },
 
     uiResponse,
 
@@ -126,7 +92,7 @@ export const inject = ({
 
     switchTransport,
 
-    ...createCoreApi(protocolAwareCall.call),
+    ...createCoreApi(call),
   };
   return api;
 };
@@ -141,25 +107,24 @@ export const createCoreApi = (
   | 'removeAllListeners'
   | 'init'
   | 'call'
-  | 'setDeviceConnectProtocol'
   | 'dispose'
   | 'uiResponse'
   | 'cancel'
   | 'updateSettings'
   | 'switchTransport'
 > => ({
+  getFirmwareUpdateCapabilities,
+  registerFirmwareUpdateHostBinding,
+  unregisterFirmwareUpdateHostBinding,
+  getFirmwareUpdateHostBindingGeneration,
+  prepareFirmwareUpdatePlan,
+  validateFirmwareUpdatePreparedPlan,
   getLogs: () => call({ method: 'getLogs' }),
   clearSessionCache: params => call({ ...params, method: 'clearSessionCache' }),
   /**
    * 搜索设备
    */
   searchDevices: params => call({ ...params, method: 'searchDevices' }),
-  detectDeviceConnectProtocol: connectId =>
-    call({
-      connectId,
-      method: 'detectDeviceConnectProtocol',
-      forceProtocolDetection: true,
-    }),
 
   /**
    * 获取设备信息
@@ -217,7 +182,6 @@ export const createCoreApi = (
     call({ ...params, connectId, method: 'deviceGetOnboardingStatus' }),
   deviceUploadWallpaper: (connectId, params) =>
     call({ ...params, connectId, method: 'deviceUploadWallpaper' }),
-  deviceUploadNft: (connectId, params) => call({ ...params, connectId, method: 'deviceUploadNft' }),
   uploadPortfolio: (connectId, params) => call({ ...params, connectId, method: 'uploadPortfolio' }),
   deviceRecovery: (connectId, params) => call({ ...params, connectId, method: 'deviceRecovery' }),
   deviceReset: (connectId, params) => call({ ...params, connectId, method: 'deviceReset' }),

package/src/lowLevelInject.ts

@@ -1,4 +1,5 @@
-import { createCoreApi, createProtocolAwareCall } from './inject';
+import { createCoreApi } from './inject';
+import { unregisterFirmwareUpdateHostBinding } from './api/firmware/FirmwareHostBinding';
 
 import type { EventEmitter } from 'events';
 import type { CallMethod, CoreMessage } from './events';
@@ -33,7 +34,6 @@ export const lowLevelInject = ({
   switchTransport,
   addHardwareGlobalEventListener,
 }: LowLevelInjectApi): LowLevelCoreApi => {
-  const protocolAwareCall = createProtocolAwareCall(call);
   const api: LowLevelCoreApi = {
     addHardwareGlobalEventListener,
     removeAllListeners: type => {
@@ -42,11 +42,12 @@ export const lowLevelInject = ({
 
     init,
 
-    call: protocolAwareCall.call,
+    call,
 
-    setDeviceConnectProtocol: protocolAwareCall.setDeviceConnectProtocol,
-
-    dispose,
+    dispose: async () => {
+      unregisterFirmwareUpdateHostBinding();
+      await dispose();
+    },
 
     uiResponse,
 
@@ -58,7 +59,7 @@ export const lowLevelInject = ({
 
     emit: () => {},
 
-    ...createCoreApi(protocolAwareCall.call),
+    ...createCoreApi(call),
   };
   return api;
 };

package/src/protocols/protocol-v2/features.ts

@@ -107,6 +107,10 @@ export const PROTOCOL_V2_FEATURES_DEVICE_INFO_REQUEST = {
     hw: true,
     fw: true,
     coprocessor: true,
+    se1: true,
+    se2: true,
+    se3: true,
+    se4: true,
   },
   types: {
     version: true,
@@ -153,7 +157,7 @@ export const PROTOCOL_V2_FULL_DEVICE_INFO_REQUEST = {
 };
 
 export const PROTOCOL_V2_DEVICE_INFO_REQUEST = PROTOCOL_V2_FULL_DEVICE_INFO_REQUEST;
-export const PROTOCOL_V2_DEVICE_INFO_TIMEOUT_MS = 30 * 1000;
+export const PROTOCOL_V2_DEVICE_INFO_TIMEOUT_MS = 10 * 1000;
 
 export async function requestProtocolV2ProtocolInfo({
   commands,
@@ -164,15 +168,8 @@ export async function requestProtocolV2ProtocolInfo({
 }): Promise<ProtocolInfo> {
   const response =
     timeoutMs === undefined
-      ? await commands.typedCall('ProtocolInfoRequest', 'ProtocolInfo', {
-          eventless_wallet_session: true,
-        })
-      : await commands.typedCall(
-          'ProtocolInfoRequest',
-          'ProtocolInfo',
-          { eventless_wallet_session: true },
-          { timeoutMs }
-        );
+      ? await commands.typedCall('ProtocolInfoRequest', 'ProtocolInfo', {})
+      : await commands.typedCall('ProtocolInfoRequest', 'ProtocolInfo', {}, { timeoutMs });
   return response.message;
 }
 

package/src/protocols/protocol-v2/firmware.ts

@@ -15,7 +15,5 @@ export const ProtocolV2FirmwareTargetType = {
   FW_MGMT_TARGET_SE04: 10,
 } as const;
 
-// 这里有意同时导出同名值与类型，以保持现有公共 API 不变。
-// eslint-disable-next-line @typescript-eslint/no-redeclare
 export type ProtocolV2FirmwareTargetType =
   (typeof ProtocolV2FirmwareTargetType)[keyof typeof ProtocolV2FirmwareTargetType];

package/src/protocols/protocol-v2/lockedError.ts

@@ -0,0 +1,10 @@
+import { HardwareErrorCode } from '@onekeyfe/hd-shared';
+
+export function isDeviceLockedError(error: unknown): boolean {
+  return (
+    typeof error === 'object' &&
+    error !== null &&
+    'errorCode' in error &&
+    error.errorCode === HardwareErrorCode.DeviceLocked
+  );
+}