feat(coinjoin): coinjoin backend identities

(cherry picked from commit 6238877)
trezor · Nov 8, 2022 · 826c3fd · 826c3fd
1 parent ff6d3f3
commit 826c3fd
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 9 deletions.
diff --git a/packages/coinjoin/src/backend/CoinjoinBackendClient.ts b/packages/coinjoin/src/backend/CoinjoinBackendClient.ts
@@ -15,17 +15,28 @@ type CoinjoinBackendClientSettings = CoinjoinBackendSettings & {
 
 export class CoinjoinBackendClient extends EventEmitter {
     protected readonly wabisabiUrl;
-    protected readonly blockbookUrl;
+    protected readonly blockbookUrls;
+
+    private readonly identityWabisabi = 'Satoshi';
+    private readonly identitiesBlockbook = [
+        'Blockbook_1',
+        'Blockbook_2',
+        'Blockbook_3',
+        'Blockbook_4',
+    ];
 
     constructor(settings: CoinjoinBackendClientSettings) {
         super();
         this.wabisabiUrl = `${settings.wabisabiBackendUrl}api/v4/btc`;
-        this.blockbookUrl =
-            settings.blockbookUrls[Math.floor(Math.random() * settings.blockbookUrls.length)];
+        this.blockbookUrls = settings.blockbookUrls;
+    }
+
+    getIdentityForBlock(height: number | undefined) {
+        return this.identitiesBlockbook[(height ?? 0) & 0x3]; // Works only when identities.length === 4
     }
 
     fetchBlock(height: number, options?: RequestOptions): Promise<BlockbookBlock> {
-        return this.blockbook(options)
+        return this.blockbook({ identity: this.getIdentityForBlock(height), ...options })
             .get(`block/${height}`)
             .then(this.handleBlockbookResponse.bind(this));
     }
@@ -113,11 +124,12 @@ export class CoinjoinBackendClient extends EventEmitter {
     }
 
     protected wabisabi(options?: RequestOptions) {
-        return this.request(this.wabisabiUrl, options);
+        return this.request(this.wabisabiUrl, { identity: this.identityWabisabi, ...options });
     }
 
     protected blockbook(options?: RequestOptions) {
-        return this.request(this.blockbookUrl, {
+        const url = this.blockbookUrls[Math.floor(Math.random() * this.blockbookUrls.length)];
+        return this.request(url, {
             ...options,
             userAgent: '', // blockbook api requires user-agent to be sent, see ./utils/http.ts
         });

diff --git a/packages/coinjoin/src/backend/CoinjoinMempoolController.ts b/packages/coinjoin/src/backend/CoinjoinMempoolController.ts
@@ -22,7 +22,11 @@ export class CoinjoinMempoolController implements MempoolController {
         const txids = await this.client.fetchMempoolTxids();
         const entries = await Promise.allSettled(
             txids.map(async txid => {
-                const tx = this.mempool[txid] ?? (await this.client.fetchTransaction(txid));
+                const tx =
+                    this.mempool[txid] ??
+                    (await this.client.fetchTransaction(txid, {
+                        identity: this.client.getIdentityForBlock(undefined),
+                    }));
                 return [txid, tx] as const;
             }),
         )

diff --git a/packages/coinjoin/src/backend/backendUtils.ts b/packages/coinjoin/src/backend/backendUtils.ts
@@ -34,7 +34,9 @@ export const deriveAddresses = (
 export const fixTx = (transactions: Transaction[], client: CoinjoinBackendClient) =>
     Promise.all(
         transactions.map(async tx => {
-            const fetched = await client.fetchTransaction(tx.txid);
+            const fetched = await client.fetchTransaction(tx.txid, {
+                identity: client.getIdentityForBlock(tx.blockHeight),
+            });
             // tx.vsize missing in transactions from /block endpoint
             tx.feeRate = fetched.vsize
                 ? new BigNumber(tx.fee).div(fetched.vsize).decimalPlaces(2).toString()

diff --git a/packages/coinjoin/src/types/backend.ts b/packages/coinjoin/src/types/backend.ts
@@ -121,7 +121,10 @@ export interface FilterController {
 
 export type FilterClient = Pick<CoinjoinBackendClient, 'fetchFilters'>;
 
-export type MempoolClient = Pick<CoinjoinBackendClient, 'fetchMempoolTxids' | 'fetchTransaction'>;
+export type MempoolClient = Pick<
+    CoinjoinBackendClient,
+    'fetchMempoolTxids' | 'fetchTransaction' | 'getIdentityForBlock'
+>;
 
 export type AccountInfo = AccountInfoBase & {
     utxo: Utxo[];

diff --git a/packages/coinjoin/tests/mocks/MockMempoolClient.ts b/packages/coinjoin/tests/mocks/MockMempoolClient.ts
@@ -11,6 +11,10 @@ export class MockMempoolClient implements MempoolClient {
         this.fetched = [];
     }
 
+    getIdentityForBlock() {
+        return 'default';
+    }
+
     fetchMempoolTxids() {
         return Promise.resolve(this.transactions.map(tx => tx.txid));
     }