|
1 | 1 | import * as assert from 'assert'; |
2 | 2 | import 'should'; |
3 | | -import { getDerivationPath } from '@bitgo/sdk-lib-mpc'; |
| 3 | +import { deriveUnhardenedMps, getDerivationPath } from '@bitgo/sdk-lib-mpc'; |
4 | 4 |
|
5 | 5 | function getAddressVerificationModule() { |
6 | 6 | return require('../../../../../src/bitgo/utils/tss/addressVerification'); |
7 | 7 | } |
8 | 8 |
|
9 | 9 | const getExtractCommonKeychain = () => getAddressVerificationModule().extractCommonKeychain; |
| 10 | +const getVerifyEddsaTssWalletAddress = () => getAddressVerificationModule().verifyEddsaTssWalletAddress; |
| 11 | + |
| 12 | +// RFC 8032 test vector: known valid Ed25519 public key + arbitrary chaincode = 128 hex chars. |
| 13 | +const TEST_PK = 'd75a980182b10ab7d54bfed3c964073a0ee172f3daa62325af021a68f707511a'; |
| 14 | +const TEST_CHAINCODE = 'deadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeef'; |
| 15 | +const TEST_KEYCHAIN = TEST_PK + TEST_CHAINCODE; |
10 | 16 |
|
11 | 17 | describe('TSS Address Verification - Derivation Path with Prefix', function () { |
12 | 18 | const commonKeychain = |
@@ -61,3 +67,81 @@ describe('TSS Address Verification - Derivation Path with Prefix', function () { |
61 | 67 | }); |
62 | 68 | }); |
63 | 69 | }); |
| 70 | + |
| 71 | +describe('verifyEddsaTssWalletAddress', function () { |
| 72 | + const keychains = [ |
| 73 | + { commonKeychain: TEST_KEYCHAIN }, |
| 74 | + { commonKeychain: TEST_KEYCHAIN }, |
| 75 | + { commonKeychain: TEST_KEYCHAIN }, |
| 76 | + ]; |
| 77 | + const isValidAddress = (addr: string) => addr.length === 64; |
| 78 | + const getAddressFromPublicKey = (pk: string) => pk; |
| 79 | + |
| 80 | + describe('MPCv2 wallets (Silence Labs / MPS formula)', function () { |
| 81 | + it('verifies a correct address derived with deriveUnhardenedMps at index 0', async function () { |
| 82 | + const verifyEddsaTssWalletAddress = getVerifyEddsaTssWalletAddress(); |
| 83 | + const expectedAddress = deriveUnhardenedMps(TEST_KEYCHAIN, 'm/0').slice(0, 64); |
| 84 | + |
| 85 | + const result = await verifyEddsaTssWalletAddress( |
| 86 | + { address: expectedAddress, keychains, index: 0, multisigTypeVersion: 'MPCv2' }, |
| 87 | + isValidAddress, |
| 88 | + getAddressFromPublicKey |
| 89 | + ); |
| 90 | + result.should.be.true(); |
| 91 | + }); |
| 92 | + |
| 93 | + it('verifies a correct address derived with deriveUnhardenedMps at index 1', async function () { |
| 94 | + const verifyEddsaTssWalletAddress = getVerifyEddsaTssWalletAddress(); |
| 95 | + const expectedAddress = deriveUnhardenedMps(TEST_KEYCHAIN, 'm/1').slice(0, 64); |
| 96 | + |
| 97 | + const result = await verifyEddsaTssWalletAddress( |
| 98 | + { address: expectedAddress, keychains, index: 1, multisigTypeVersion: 'MPCv2' }, |
| 99 | + isValidAddress, |
| 100 | + getAddressFromPublicKey |
| 101 | + ); |
| 102 | + result.should.be.true(); |
| 103 | + }); |
| 104 | + |
| 105 | + it('rejects an address derived at a different index', async function () { |
| 106 | + const verifyEddsaTssWalletAddress = getVerifyEddsaTssWalletAddress(); |
| 107 | + const addressFromIndex0 = deriveUnhardenedMps(TEST_KEYCHAIN, 'm/0').slice(0, 64); |
| 108 | + |
| 109 | + const result = await verifyEddsaTssWalletAddress( |
| 110 | + { address: addressFromIndex0, keychains, index: 1, multisigTypeVersion: 'MPCv2' }, |
| 111 | + isValidAddress, |
| 112 | + getAddressFromPublicKey |
| 113 | + ); |
| 114 | + result.should.be.false(); |
| 115 | + }); |
| 116 | + |
| 117 | + it('rejects a random address that was not derived from the keychain', async function () { |
| 118 | + const verifyEddsaTssWalletAddress = getVerifyEddsaTssWalletAddress(); |
| 119 | + const randomAddress = 'ab'.repeat(32); // 64 hex chars, wrong address |
| 120 | + |
| 121 | + const result = await verifyEddsaTssWalletAddress( |
| 122 | + { address: randomAddress, keychains, index: 0, multisigTypeVersion: 'MPCv2' }, |
| 123 | + isValidAddress, |
| 124 | + getAddressFromPublicKey |
| 125 | + ); |
| 126 | + result.should.be.false(); |
| 127 | + }); |
| 128 | + }); |
| 129 | + |
| 130 | + describe('non-MPCv2 wallets (MPCv1 formula)', function () { |
| 131 | + it('rejects an MPCv2-derived address when multisigTypeVersion is not set', async function () { |
| 132 | + const verifyEddsaTssWalletAddress = getVerifyEddsaTssWalletAddress(); |
| 133 | + // MPCv2 (Silence Labs) and MPCv1 formulas produce different addresses for the same keychain. |
| 134 | + // Without multisigTypeVersion: 'MPCv2', the MPCv1 formula is used, so the MPCv2-derived |
| 135 | + // address should not match. |
| 136 | + const mpcv2Address = deriveUnhardenedMps(TEST_KEYCHAIN, 'm/0').slice(0, 64); |
| 137 | + |
| 138 | + const result = await verifyEddsaTssWalletAddress( |
| 139 | + { address: mpcv2Address, keychains, index: 0 }, |
| 140 | + isValidAddress, |
| 141 | + getAddressFromPublicKey |
| 142 | + ); |
| 143 | + |
| 144 | + result.should.be.false(); |
| 145 | + }); |
| 146 | + }); |
| 147 | +}); |
0 commit comments