112 lines
3.4 KiB
JavaScript
112 lines
3.4 KiB
JavaScript
|
'use strict';
|
||
|
|
||
|
var bitcore = require('bitcore');
|
||
|
var PrivateKey = bitcore.PrivateKey;
|
||
|
var PublicKey = bitcore.PublicKey;
|
||
|
var Address = bitcore.Address;
|
||
|
var BufferWriter = bitcore.encoding.BufferWriter;
|
||
|
var ECDSA = bitcore.crypto.ECDSA;
|
||
|
var Signature = bitcore.crypto.Signature;
|
||
|
var sha256sha256 = bitcore.crypto.Hash.sha256sha256;
|
||
|
|
||
|
/**
|
||
|
* Will construct a new message to sign and verify.
|
||
|
*
|
||
|
* @param {String} message
|
||
|
* @returns {Message}
|
||
|
*/
|
||
|
var Message = function Message(message) {
|
||
|
if (!(this instanceof Message)) {
|
||
|
return new Message(message);
|
||
|
}
|
||
|
if (typeof message !== 'string') {
|
||
|
throw new TypeError('First argument should be a string');
|
||
|
}
|
||
|
this.message = message;
|
||
|
|
||
|
return this;
|
||
|
};
|
||
|
|
||
|
Message.MAGIC_BYTES = new Buffer('Bitcoin Signed Message:\n');
|
||
|
|
||
|
Message.prototype.magicHash = function magicHash() {
|
||
|
var prefix1 = BufferWriter.varintBufNum(Message.MAGIC_BYTES.length);
|
||
|
var messageBuffer = new Buffer(this.message);
|
||
|
var prefix2 = BufferWriter.varintBufNum(messageBuffer.length);
|
||
|
var buf = Buffer.concat([prefix1, Message.MAGIC_BYTES, prefix2, messageBuffer]);
|
||
|
var hash = sha256sha256(buf);
|
||
|
return hash;
|
||
|
};
|
||
|
|
||
|
Message.prototype._sign = function _sign(privateKey) {
|
||
|
if (!(privateKey instanceof PrivateKey)) {
|
||
|
throw new TypeError('First argument should be an instance of PrivateKey');
|
||
|
}
|
||
|
var hash = this.magicHash();
|
||
|
var ecdsa = new ECDSA();
|
||
|
ecdsa.hashbuf = hash;
|
||
|
ecdsa.privkey = privateKey;
|
||
|
ecdsa.pubkey = privateKey.toPublicKey();
|
||
|
ecdsa.signRandomK();
|
||
|
ecdsa.calci();
|
||
|
return ecdsa.sig;
|
||
|
};
|
||
|
|
||
|
/**
|
||
|
* Will sign a message with a given bitcoin private key.
|
||
|
*
|
||
|
* @param {PrivateKey} privateKey - An instance of PrivateKey
|
||
|
* @returns {String} A base64 encoded compact signature
|
||
|
*/
|
||
|
Message.prototype.sign = function sign(privateKey) {
|
||
|
var signature = this._sign(privateKey);
|
||
|
return signature.toCompact().toString('base64');
|
||
|
};
|
||
|
|
||
|
Message.prototype._verify = function _verify(publicKey, signature) {
|
||
|
if (!(publicKey instanceof PublicKey)) {
|
||
|
throw new TypeError('First argument should be an instance of PublicKey');
|
||
|
}
|
||
|
if (!(signature instanceof Signature)) {
|
||
|
throw new TypeError('Second argument should be an instance of Signature');
|
||
|
}
|
||
|
var hash = this.magicHash();
|
||
|
var verified = ECDSA.verify(hash, signature, publicKey);
|
||
|
if (!verified) {
|
||
|
this.error = 'The signature was invalid';
|
||
|
}
|
||
|
return verified;
|
||
|
};
|
||
|
|
||
|
/**
|
||
|
* Will return a boolean of the signature is valid for a given bitcoin address.
|
||
|
* If it isn't the specific reason is accessible via the "error" member.
|
||
|
*
|
||
|
* @param {String} bitcoinAddress - A bitcoin address
|
||
|
* @param {String} signatureString - A base64 encoded compact signature
|
||
|
* @returns {Boolean}
|
||
|
*/
|
||
|
Message.prototype.verify = function verify(bitcoinAddress, signatureString) {
|
||
|
var signature = Signature.fromCompact(new Buffer(signatureString, 'base64'));
|
||
|
|
||
|
// recover the public key
|
||
|
var ecdsa = new ECDSA();
|
||
|
ecdsa.hashbuf = this.magicHash();
|
||
|
ecdsa.sig = signature;
|
||
|
var publicKey = ecdsa.toPublicKey();
|
||
|
|
||
|
var expectedAddress = Address.fromString(bitcoinAddress);
|
||
|
var signatureAddress = Address.fromPublicKey(publicKey, expectedAddress.network);
|
||
|
|
||
|
// check that the recovered address and specified address match
|
||
|
if (expectedAddress.toString() !== signatureAddress.toString()) {
|
||
|
this.error = 'The signature did not match the message digest';
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
return this._verify(publicKey, signature);
|
||
|
};
|
||
|
|
||
|
module.exports = Message;
|
||
|
|