132 lines
3.5 KiB
TypeScript
132 lines
3.5 KiB
TypeScript
import { Network } from './networks';
|
|
import * as NETWORKS from './networks';
|
|
import * as types from './types';
|
|
const ecc = require('tiny-secp256k1');
|
|
const randomBytes = require('randombytes');
|
|
const typeforce = require('typeforce');
|
|
const wif = require('wif');
|
|
|
|
const isOptions = typeforce.maybe(
|
|
typeforce.compile({
|
|
compressed: types.maybe(types.Boolean),
|
|
network: types.maybe(types.Network),
|
|
}),
|
|
);
|
|
|
|
interface ECPairOptions {
|
|
compressed?: boolean;
|
|
network?: Network;
|
|
rng?(arg0: Buffer): Buffer;
|
|
}
|
|
|
|
export interface ECPairInterface {
|
|
compressed: boolean;
|
|
network: Network;
|
|
privateKey?: Buffer;
|
|
publicKey?: Buffer;
|
|
toWIF(): string;
|
|
sign(hash: Buffer): Buffer;
|
|
verify(hash: Buffer, signature: Buffer): Buffer;
|
|
getPublicKey?(): Buffer;
|
|
}
|
|
|
|
class ECPair implements ECPairInterface {
|
|
compressed: boolean;
|
|
network: Network;
|
|
private __d?: Buffer;
|
|
private __Q?: Buffer;
|
|
|
|
constructor(d?: Buffer, Q?: Buffer, options?: ECPairOptions) {
|
|
if (options === undefined) options = {};
|
|
this.compressed =
|
|
options.compressed === undefined ? true : options.compressed;
|
|
this.network = options.network || NETWORKS.bitcoin;
|
|
|
|
this.__d = undefined;
|
|
this.__Q = undefined;
|
|
if (d !== undefined) this.__d = d;
|
|
if (Q !== undefined) this.__Q = ecc.pointCompress(Q, this.compressed);
|
|
}
|
|
|
|
get privateKey(): Buffer | undefined {
|
|
return this.__d;
|
|
}
|
|
|
|
get publicKey(): Buffer | undefined {
|
|
if (!this.__Q) this.__Q = ecc.pointFromScalar(this.__d, this.compressed);
|
|
return this.__Q;
|
|
}
|
|
|
|
toWIF(): string {
|
|
if (!this.__d) throw new Error('Missing private key');
|
|
return wif.encode(this.network.wif, this.__d, this.compressed);
|
|
}
|
|
|
|
sign(hash: Buffer): Buffer {
|
|
if (!this.__d) throw new Error('Missing private key');
|
|
return ecc.sign(hash, this.__d);
|
|
}
|
|
|
|
verify(hash: Buffer, signature: Buffer): Buffer {
|
|
return ecc.verify(hash, this.publicKey, signature);
|
|
}
|
|
}
|
|
|
|
function fromPrivateKey(buffer: Buffer, options?: ECPairOptions): ECPair {
|
|
typeforce(types.Buffer256bit, buffer);
|
|
if (!ecc.isPrivate(buffer))
|
|
throw new TypeError('Private key not in range [1, n)');
|
|
typeforce(isOptions, options);
|
|
|
|
return new ECPair(buffer, undefined, <ECPairOptions>options);
|
|
}
|
|
|
|
function fromPublicKey(buffer: Buffer, options?: ECPairOptions): ECPair {
|
|
typeforce(ecc.isPoint, buffer);
|
|
typeforce(isOptions, options);
|
|
return new ECPair(undefined, buffer, <ECPairOptions>options);
|
|
}
|
|
|
|
function fromWIF(string: string, network?: Network | Array<Network>): ECPair {
|
|
const decoded = wif.decode(string);
|
|
const version = decoded.version;
|
|
|
|
// list of networks?
|
|
if (types.Array(network)) {
|
|
network = <Network>(<Array<Network>>network)
|
|
.filter(function(x: Network) {
|
|
return version === x.wif;
|
|
})
|
|
.pop();
|
|
|
|
if (!network) throw new Error('Unknown network version');
|
|
|
|
// otherwise, assume a network object (or default to bitcoin)
|
|
} else {
|
|
network = network || NETWORKS.bitcoin;
|
|
|
|
if (version !== (<Network>network).wif)
|
|
throw new Error('Invalid network version');
|
|
}
|
|
|
|
return fromPrivateKey(decoded.privateKey, {
|
|
compressed: decoded.compressed,
|
|
network: <Network>network,
|
|
});
|
|
}
|
|
|
|
function makeRandom(options?: ECPairOptions): ECPair {
|
|
typeforce(isOptions, options);
|
|
if (options === undefined) options = {};
|
|
const rng = options.rng || randomBytes;
|
|
|
|
let d;
|
|
do {
|
|
d = rng(32);
|
|
typeforce(types.Buffer256bit, d);
|
|
} while (!ecc.isPrivate(d));
|
|
|
|
return fromPrivateKey(d, options);
|
|
}
|
|
|
|
export { makeRandom, fromPrivateKey, fromPublicKey, fromWIF };
|