131 lines
3.4 KiB
TypeScript
131 lines
3.4 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
|
|
}
|