76 lines
2.4 KiB
TypeScript
76 lines
2.4 KiB
TypeScript
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<any> {
|
|
return new Promise((resolve, reject) => {
|
|
return exec(command, (err, stdout) => {
|
|
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<string> {
|
|
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<string> {
|
|
const privKeyWalletAddress = await wallet.getAddressString();
|
|
// strip checksum
|
|
return privKeyWalletAddress.toLowerCase();
|
|
}
|
|
|
|
async function getNormalizedAddressesFromWallets(wallets: IFullWallet[]): Promise<string[]> {
|
|
return Promise.all(wallets.map(getNormalizedAddressFromWallet));
|
|
}
|
|
|
|
async function testDerivation(): Promise<true> {
|
|
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);
|
|
});
|
|
});
|