fixup! feat(transport): implement THP

trezor · Jul 30, 2024 · 6157652 · 6157652
1 parent 57e072f
commit 6157652
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/packages/transport/src/transports/abstractApi.ts b/packages/transport/src/transports/abstractApi.ts
@@ -350,8 +350,26 @@ export abstract class AbstractApiTransport extends AbstractTransport {
                 }
                 const { path } = getPathBySessionResponse.payload;
 
+                const apiWrite = (chunk: Buffer, attemptSignal?: AbortSignal) =>
+                    this.api.write(path, chunk, attemptSignal || signal);
+                const apiRead = (attemptSignal?: AbortSignal) =>
+                    this.api.read(path, attemptSignal || signal);
+
                 try {
                     const protocol = customProtocol || v1Protocol;
+
+                    if (protocol.name === 'v2') {
+                        const message = await receiveThpMessage({
+                            messages: this.messages,
+                            protocolState,
+                            apiWrite,
+                            apiRead,
+                            signal,
+                        });
+
+                        return this.success(message);
+                    }
+
                     const message = await receiveAndParse(
                         this.messages,
                         () =>