Cleaning up message.js
This commit is contained in:
parent
65fb593be2
commit
096f8c982e
1 changed files with 41 additions and 44 deletions
|
@ -1,68 +1,65 @@
|
||||||
/// Implements Bitcoin's feature for signing arbitrary messages.
|
/// Implements Bitcoin's feature for signing arbitrary messages.
|
||||||
|
|
||||||
var SHA256 = require('crypto-js/sha256');
|
var SHA256 = require('crypto-js/sha256')
|
||||||
var ecdsa = require('./ecdsa');
|
var ecdsa = require('./ecdsa')
|
||||||
var convert = require('./convert');
|
var convert = require('./convert')
|
||||||
|
|
||||||
var Message = {};
|
var Message = {}
|
||||||
|
|
||||||
Message.magicPrefix = "Bitcoin Signed Message:\n";
|
Message.magicPrefix = "Bitcoin Signed Message:\n"
|
||||||
|
|
||||||
Message.makeMagicMessage = function (message) {
|
Message.makeMagicMessage = function (message) {
|
||||||
var magicBytes = convert.stringToBytes(Message.magicPrefix);
|
var magicBytes = convert.stringToBytes(Message.magicPrefix)
|
||||||
var messageBytes = convert.stringToBytes(message);
|
var messageBytes = convert.stringToBytes(message)
|
||||||
|
|
||||||
var buffer = [];
|
return [].concat(
|
||||||
buffer = buffer.concat(convert.numToVarInt(magicBytes.length));
|
convert.numToVarInt(magicBytes.length),
|
||||||
buffer = buffer.concat(magicBytes);
|
magicBytes,
|
||||||
buffer = buffer.concat(convert.numToVarInt(messageBytes.length));
|
convert.numToVarInt(messageBytes.length),
|
||||||
buffer = buffer.concat(messageBytes);
|
messageBytes
|
||||||
|
)
|
||||||
|
|
||||||
return buffer;
|
}
|
||||||
};
|
|
||||||
|
|
||||||
Message.getHash = function (message) {
|
Message.getHash = function (message) {
|
||||||
var buffer = Message.makeMagicMessage(message);
|
var buffer = Message.makeMagicMessage(message)
|
||||||
return convert.wordArrayToBytes(SHA256(SHA256(convert.bytesToWordArray(buffer))));
|
return convert.wordArrayToBytes(SHA256(SHA256(convert.bytesToWordArray(buffer))))
|
||||||
};
|
}
|
||||||
|
|
||||||
Message.signMessage = function (key, message) {
|
Message.signMessage = function (key, message) {
|
||||||
var hash = Message.getHash(message);
|
var hash = Message.getHash(message)
|
||||||
|
var sig = key.sign(hash)
|
||||||
|
var obj = ecdsa.parseSig(sig)
|
||||||
|
var i = ecdsa.calcPubkeyRecoveryParam(key, obj.r, obj.s, hash)
|
||||||
|
|
||||||
var sig = key.sign(hash);
|
i += 27
|
||||||
|
if (key.compressed) {
|
||||||
|
i += 4
|
||||||
|
}
|
||||||
|
|
||||||
var obj = ecdsa.parseSig(sig);
|
var rBa = obj.r.toByteArrayUnsigned()
|
||||||
|
var sBa = obj.s.toByteArrayUnsigned()
|
||||||
var i = ecdsa.calcPubkeyRecoveryParam(key, obj.r, obj.s, hash);
|
|
||||||
|
|
||||||
i += 27;
|
|
||||||
if (key.compressed) i += 4;
|
|
||||||
|
|
||||||
var rBa = obj.r.toByteArrayUnsigned();
|
|
||||||
var sBa = obj.s.toByteArrayUnsigned();
|
|
||||||
|
|
||||||
// Pad to 32 bytes per value
|
// Pad to 32 bytes per value
|
||||||
while (rBa.length < 32) rBa.unshift(0);
|
while (rBa.length < 32) rBa.unshift(0)
|
||||||
while (sBa.length < 32) sBa.unshift(0);
|
while (sBa.length < 32) sBa.unshift(0)
|
||||||
|
|
||||||
sig = [i].concat(rBa).concat(sBa);
|
sig = [i].concat(rBa, sBa)
|
||||||
|
|
||||||
return convert.bytesToHex(sig);
|
return convert.bytesToHex(sig)
|
||||||
};
|
}
|
||||||
|
|
||||||
Message.verifyMessage = function (address, sig, message) {
|
Message.verifyMessage = function (address, sig, message) {
|
||||||
sig = convert.hexToBytes(sig);
|
sig = ecdsa.parseSigCompact(convert.hexToBytes(sig))
|
||||||
sig = ecdsa.parseSigCompact(sig);
|
|
||||||
|
|
||||||
var hash = Message.getHash(message);
|
var hash = Message.getHash(message)
|
||||||
|
|
||||||
var isCompressed = !!(sig.i & 4);
|
var isCompressed = !!(sig.i & 4)
|
||||||
var pubKey = ecdsa.recoverPubKey(sig.r, sig.s, hash, sig.i);
|
var pubKey = ecdsa.recoverPubKey(sig.r, sig.s, hash, sig.i)
|
||||||
pubKey.compressed = isCompressed;
|
pubKey.compressed = isCompressed
|
||||||
|
|
||||||
var expectedAddress = pubKey.getAddress().toString();
|
// Compare address to expected address
|
||||||
|
return address === pubKey.getAddress().toString()
|
||||||
|
}
|
||||||
|
|
||||||
return (address === expectedAddress);
|
module.exports = Message
|
||||||
};
|
|
||||||
|
|
||||||
module.exports = Message;
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue