@onekeyfe/hd-transport 1.2.0-alpha.8 → 1.2.0-alpha.80

package/src/types/transport.ts

@@ -1,5 +1,15 @@
 import type EventEmitter from 'events';
 
+export const TRANSPORT_EVENT = {
+  DEVICE_DISCONNECT: 'transport-device-disconnect',
+} as const;
+
+export type TransportDeviceDisconnectEvent = {
+  id: string;
+  connectId: string;
+  name: string | null;
+};
+
 export type ProtocolType = 'V1' | 'V2';
 
 export type OneKeyDeviceCommType =
@@ -43,15 +53,25 @@ export type AcquireInput = {
   uuid?: string;
   forceCleanRunPromise?: boolean;
   expectedProtocol?: ProtocolType;
+  protocolHint?: ProtocolType;
 };
 
 export type MessageFromOneKey = { type: string; message: Record<string, any> };
 
+export type TransportWriteMetrics = {
+  elapsedMs: number;
+  frameBytes: number;
+};
+
 export type TransportCallOptions = {
   timeoutMs?: number;
   expectedTypes?: string[];
   intermediateTypes?: string[];
   onIntermediateResponse?: (response: MessageFromOneKey) => void;
+  /** Called after the complete request frame has been submitted to the transport. */
+  onWriteCompleted?: (metrics: TransportWriteMetrics) => void;
+  /** Prefer acknowledged BLE characteristic writes for this call when supported. */
+  writeWithResponse?: boolean;
 };
 
 type ITransportInitFn = (
@@ -64,7 +84,12 @@ export type Transport = {
   enumerate(): Promise<Array<OneKeyDeviceInfo>>;
   listen(old?: Array<OneKeyDeviceInfo>): Promise<Array<OneKeyDeviceInfo>>;
   acquire(input: AcquireInput): Promise<string>;
-  release(session: string, onclose: boolean): Promise<void>;
+  /**
+   * `keepSession` tells a transport that the caller intends to keep using this
+   * device across calls (firmware update, batched signing). Transports that
+   * hold the link open may use it to pick a longer idle window; the rest ignore it.
+   */
+  release(session: string, onclose: boolean, keepSession?: boolean): Promise<void>;
   configure(signedData: JSON | string): Promise<void>;
   configureProtocolV2?: (signedData: JSON | string) => Promise<void> | void;
   call(
@@ -92,7 +117,7 @@ export type Transport = {
 
   // resolves when the transport can be used; rejects when it cannot
   init: ITransportInitFn;
-  stop(): void;
+  stop(): void | Promise<void>;
 
   configured: boolean;
   version: string;
@@ -105,8 +130,8 @@ export type Transport = {
 export type LowLevelDevice = OneKeyDeviceInfoBase & { id: string; name: string };
 export type LowlevelTransportSharedPlugin = {
   enumerate: () => Promise<LowLevelDevice[]>;
-  send: (uuid: string, data: string) => Promise<void>;
-  receive: () => Promise<string>;
+  send: (uuid: string, data: string, options?: { withoutResponse?: boolean }) => Promise<void>;
+  receive: (uuid?: string) => Promise<string>;
   connect: (uuid: string) => Promise<void>;
   disconnect: (uuid: string) => Promise<void>;
 

package/src/utils/transportLog.ts

@@ -0,0 +1,102 @@
+import type { ProtocolType } from '../types';
+
+const REDACTED = '[REDACTED]';
+
+const SENSITIVE_LOG_KEYS = new Set([
+  'accesstoken',
+  'address',
+  'apikey',
+  'connectid',
+  'devicestate',
+  'deviceid',
+  'entropy',
+  'expectedpassphrasestate',
+  'mnemonic',
+  'passphrase',
+  'passphrasestate',
+  'password',
+  'pin',
+  'privatekey',
+  'publickey',
+  'seed',
+  'serialnumber',
+  'session',
+  'sessionid',
+  'signature',
+  'token',
+  'uuid',
+  'walletsessionid',
+  'xprv',
+  'xpub',
+]);
+
+const normalizeLogKey = (key: string) => key.replace(/[_-]/g, '').toLowerCase();
+
+const isSensitiveLogKey = (key: string) => {
+  const normalizedKey = normalizeLogKey(key);
+  return (
+    SENSITIVE_LOG_KEYS.has(normalizedKey) ||
+    normalizedKey.endsWith('address') ||
+    normalizedKey.endsWith('privatekey') ||
+    normalizedKey.endsWith('publickey') ||
+    normalizedKey.endsWith('signature')
+  );
+};
+
+const isSigningMessage = (messageName?: string) =>
+  Boolean(messageName && /(?:sign|txack|transaction)/i.test(messageName));
+
+const redactLogValue = (value: unknown, seen: WeakSet<object>): unknown => {
+  if (ArrayBuffer.isView(value)) {
+    return `[BINARY:${value.byteLength}]`;
+  }
+  if (value instanceof ArrayBuffer) {
+    return `[BINARY:${value.byteLength}]`;
+  }
+  if (Array.isArray(value)) {
+    return value.map(item => redactLogValue(item, seen));
+  }
+  if (!value || typeof value !== 'object') {
+    return value;
+  }
+  if (seen.has(value)) {
+    return '[CIRCULAR]';
+  }
+
+  seen.add(value);
+  const redacted = Object.fromEntries(
+    Object.entries(value as Record<string, unknown>).map(([key, item]) => [
+      key,
+      isSensitiveLogKey(key) && item !== null && item !== undefined
+        ? REDACTED
+        : redactLogValue(item, seen),
+    ])
+  );
+  seen.delete(value);
+  return redacted;
+};
+
+export function getSafeTransportLogPayload(value: unknown, messageName?: string): unknown {
+  if (isSigningMessage(messageName)) {
+    return REDACTED;
+  }
+  return redactLogValue(value, new WeakSet());
+}
+
+export function createTransportCallLog(
+  name: string,
+  protocol: ProtocolType,
+  data: Record<string, unknown>
+) {
+  return {
+    name,
+    protocol,
+    request: getSafeTransportLogPayload(data, name),
+  };
+}
+
+const HIGH_VOLUME_CALLS = new Set(['FileWrite', 'FilesystemFileWrite', 'EmmcFileWrite']);
+
+export function shouldSuppressHighVolumeCallLog(name: string) {
+  return HIGH_VOLUME_CALLS.has(name);
+}

package/dist/protocols/v2/debug.d.ts

@@ -1,13 +0,0 @@
-export type ProtocolV2DebugLogger = {
-    debug?: (...args: unknown[]) => void;
-};
-export type ProtocolV2FrameDebugOptions = {
-    logger?: ProtocolV2DebugLogger;
-    logPrefix?: string;
-    context?: string;
-    messageName?: string;
-    messageTypeId?: number;
-    pbPayloadLength?: number;
-};
-export declare function logProtocolV2Debug(options: ProtocolV2FrameDebugOptions | undefined, stage: string, details: Record<string, unknown>): void;
-//# sourceMappingURL=debug.d.ts.map

package/dist/protocols/v2/debug.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"debug.d.ts","sourceRoot":"","sources":["../../../src/protocols/v2/debug.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,qBAAqB,GAAG;IAClC,KAAK,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CACtC,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,MAAM,CAAC,EAAE,qBAAqB,CAAC;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B,CAAC;AAEF,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,2BAA2B,GAAG,SAAS,EAChD,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,QASjC"}

package/src/protocols/v2/debug.ts

@@ -1,26 +0,0 @@
-export type ProtocolV2DebugLogger = {
-  debug?: (...args: unknown[]) => void;
-};
-
-export type ProtocolV2FrameDebugOptions = {
-  logger?: ProtocolV2DebugLogger;
-  logPrefix?: string;
-  context?: string;
-  messageName?: string;
-  messageTypeId?: number;
-  pbPayloadLength?: number;
-};
-
-export function logProtocolV2Debug(
-  options: ProtocolV2FrameDebugOptions | undefined,
-  stage: string,
-  details: Record<string, unknown>
-) {
-  options?.logger?.debug?.(`[${options.logPrefix ?? 'ProtocolV2'}] ${stage}`, {
-    ...(options.context ? { context: options.context } : {}),
-    ...(options.messageName ? { messageName: options.messageName } : {}),
-    ...(options.messageTypeId !== undefined ? { messageTypeId: options.messageTypeId } : {}),
-    ...(options.pbPayloadLength !== undefined ? { pbPayloadLength: options.pbPayloadLength } : {}),
-    ...details,
-  });
-}