import { generate, IFullWallet } from 'ethereumjs-wallet'; import { stripHexPrefix } from '../../common/libs/values'; const { exec } = require('child_process'); // FIXME pick a less magic number const derivationRounds = 500; const dockerImage = 'dternyak/eth-priv-to-addr'; const dockerTag = 'latest'; function promiseFromChildProcess(command: string): Promise { return new Promise((resolve, reject) => { return exec(command, (err: string, stdout: string) => { err ? reject(err) : resolve(stdout); }); }); } function getCleanPrivateKey(privKeyWallet: IFullWallet): string { return stripHexPrefix(privKeyWallet.getPrivateKeyString()); } function makeCommaSeparatedPrivateKeys(privKeyWallets: IFullWallet[]): string { const privateKeys = privKeyWallets.map(getCleanPrivateKey); return privateKeys.join(','); } async function privToAddrViaDocker(privKeyWallets: IFullWallet[]): Promise { const command = `docker run -e key=${makeCommaSeparatedPrivateKeys( privKeyWallets )} ${dockerImage}:${dockerTag}`; const dockerOutput = await promiseFromChildProcess(command); const newlineStrippedDockerOutput = dockerOutput.replace(/(\r\n|\n|\r)/gm, ''); return newlineStrippedDockerOutput; } function makeWallets(): IFullWallet[] { const wallets: IFullWallet[] = []; let i = 0; while (i < derivationRounds) { const privKeyWallet = generate(); wallets.push(privKeyWallet); i += 1; } return wallets; } async function getNormalizedAddressFromWallet(wallet: IFullWallet): Promise { const privKeyWalletAddress = await wallet.getAddressString(); // strip checksum return privKeyWalletAddress.toLowerCase(); } async function getNormalizedAddressesFromWallets(wallets: IFullWallet[]): Promise { return Promise.all(wallets.map(getNormalizedAddressFromWallet)); } async function testDerivation(): Promise { const wallets = makeWallets(); const walletAddrs = await getNormalizedAddressesFromWallets(wallets); const dockerAddrsCS = await privToAddrViaDocker(wallets); const dockerAddrs = dockerAddrsCS.split(','); expect(walletAddrs).toEqual(dockerAddrs); return true; } describe('Derivation Checker', () => { beforeEach(() => { // increase timer to prevent early timeout jasmine.DEFAULT_TIMEOUT_INTERVAL = 100000; }); it(`should derive identical addresses ${derivationRounds} times`, () => { return testDerivation().then(expect); }); });