110 lines
3.6 KiB
TypeScript
110 lines
3.6 KiB
TypeScript
'use strict';
|
|
|
|
import { ec as EC } from 'elliptic';
|
|
|
|
import { getAddress } from './address';
|
|
|
|
import { arrayify, hexlify, hexZeroPad, splitSignature } from './bytes';
|
|
import { hashMessage } from './hash';
|
|
import { keccak256 } from './keccak256';
|
|
import { defineReadOnly } from './properties';
|
|
|
|
import * as errors from '../errors';
|
|
|
|
///////////////////////////////
|
|
// Imported Types
|
|
|
|
import { Arrayish, Signature } from './bytes';
|
|
|
|
///////////////////////////////
|
|
|
|
let _curve: EC = null
|
|
function getCurve() {
|
|
if (!_curve) {
|
|
_curve = new EC('secp256k1');
|
|
}
|
|
return _curve;
|
|
}
|
|
|
|
|
|
export class KeyPair {
|
|
|
|
readonly privateKey: string;
|
|
|
|
readonly publicKey: string;
|
|
readonly compressedPublicKey: string;
|
|
|
|
readonly publicKeyBytes: Uint8Array;
|
|
|
|
constructor(privateKey: Arrayish | string) {
|
|
let keyPair = getCurve().keyFromPrivate(arrayify(privateKey));
|
|
|
|
defineReadOnly(this, 'privateKey', hexlify(keyPair.priv.toArray('be', 32)));
|
|
defineReadOnly(this, 'publicKey', '0x' + keyPair.getPublic(false, 'hex'));
|
|
defineReadOnly(this, 'compressedPublicKey', '0x' + keyPair.getPublic(true, 'hex'));
|
|
defineReadOnly(this, 'publicKeyBytes', keyPair.getPublic().encode(null, true));
|
|
}
|
|
|
|
sign(digest: Arrayish | string): Signature {
|
|
let keyPair = getCurve().keyFromPrivate(arrayify(this.privateKey));
|
|
let signature = keyPair.sign(arrayify(digest), {canonical: true});
|
|
return {
|
|
recoveryParam: signature.recoveryParam,
|
|
r: hexZeroPad('0x' + signature.r.toString(16), 32),
|
|
s: hexZeroPad('0x' + signature.s.toString(16), 32),
|
|
v: 27 + signature.recoveryParam
|
|
}
|
|
|
|
}
|
|
|
|
computeSharedSecret(otherKey: Arrayish | string): string {
|
|
let keyPair = getCurve().keyFromPrivate(arrayify(this.privateKey));
|
|
let otherKeyPair = getCurve().keyFromPublic(arrayify(computePublicKey(otherKey)));
|
|
return hexZeroPad('0x' + keyPair.derive(otherKeyPair.getPublic()).toString(16), 32);
|
|
}
|
|
}
|
|
|
|
export function computePublicKey(key: Arrayish | string, compressed?: boolean): string {
|
|
|
|
let bytes = arrayify(key);
|
|
|
|
if (bytes.length === 32) {
|
|
let keyPair: KeyPair = new KeyPair(bytes);
|
|
if (compressed) {
|
|
return keyPair.compressedPublicKey;
|
|
}
|
|
return keyPair.publicKey;
|
|
|
|
} else if (bytes.length === 33) {
|
|
if (compressed) { return hexlify(bytes); }
|
|
return '0x' + getCurve().keyFromPublic(bytes).getPublic(false, 'hex');
|
|
|
|
} else if (bytes.length === 65) {
|
|
if (!compressed) { return hexlify(bytes); }
|
|
return '0x' + getCurve().keyFromPublic(bytes).getPublic(true, 'hex');
|
|
}
|
|
|
|
errors.throwError('invalid public or private key', errors.INVALID_ARGUMENT, { arg: 'key', value: '[REDACTED]' });
|
|
return null;
|
|
}
|
|
|
|
export function computeAddress(key: Arrayish | string): string {
|
|
// Strip off the leading "0x04"
|
|
let publicKey = '0x' + computePublicKey(key).slice(4);
|
|
return getAddress('0x' + keccak256(publicKey).substring(26));
|
|
}
|
|
|
|
export function recoverPublicKey(digest: Arrayish | string, signature: Signature | string): string {
|
|
let sig = splitSignature(signature);
|
|
let rs = { r: arrayify(sig.r), s: arrayify(sig.s) };
|
|
return '0x' + getCurve().recoverPubKey(arrayify(digest), rs, sig.recoveryParam).encode('hex', false);
|
|
}
|
|
|
|
export function recoverAddress(digest: Arrayish | string, signature: Signature | string): string {
|
|
return computeAddress(recoverPublicKey(arrayify(digest), signature));
|
|
}
|
|
|
|
export function verifyMessage(message: Arrayish | string, signature: Signature | string): string {
|
|
return recoverAddress(hashMessage(message), signature);
|
|
}
|