Ruben de Vries
9 years ago
3 changed files with 193 additions and 0 deletions
@ -0,0 +1,165 @@ |
|||
'use strict'; |
|||
|
|||
var _ = require('lodash'); |
|||
var inherits = require('inherits'); |
|||
var Input = require('./input'); |
|||
var Output = require('../output'); |
|||
var $ = require('../../util/preconditions'); |
|||
|
|||
var Script = require('../../script'); |
|||
var Signature = require('../../crypto/signature'); |
|||
var Sighash = require('../sighash'); |
|||
var PublicKey = require('../../publickey'); |
|||
var BufferUtil = require('../../util/buffer'); |
|||
var TransactionSignature = require('../signature'); |
|||
|
|||
/** |
|||
* @constructor |
|||
*/ |
|||
function MultiSigInput(input, pubkeys, threshold, signatures) { |
|||
Input.apply(this, arguments); |
|||
var self = this; |
|||
pubkeys = pubkeys || input.publicKeys; |
|||
threshold = threshold || input.threshold; |
|||
signatures = signatures || input.signatures; |
|||
this.publicKeys = _.sortBy(pubkeys, function(publicKey) { return publicKey.toString('hex'); }); |
|||
$.checkState(Script.buildMultisigOut(this.publicKeys, threshold).equals(this.output.script), |
|||
'Provided public keys don\'t match to the provided output script'); |
|||
this.publicKeyIndex = {}; |
|||
_.each(this.publicKeys, function(publicKey, index) { |
|||
self.publicKeyIndex[publicKey.toString()] = index; |
|||
}); |
|||
this.threshold = threshold; |
|||
// Empty array of signatures
|
|||
this.signatures = signatures ? this._deserializeSignatures(signatures) : new Array(this.publicKeys.length); |
|||
} |
|||
inherits(MultiSigInput, Input); |
|||
|
|||
MultiSigInput.prototype.toObject = function() { |
|||
var obj = Input.prototype.toObject.apply(this, arguments); |
|||
obj.threshold = this.threshold; |
|||
obj.publicKeys = _.map(this.publicKeys, function(publicKey) { return publicKey.toString(); }); |
|||
obj.signatures = this._serializeSignatures(); |
|||
return obj; |
|||
}; |
|||
|
|||
MultiSigInput.prototype._deserializeSignatures = function(signatures) { |
|||
return _.map(signatures, function(signature) { |
|||
if (!signature) { |
|||
return undefined; |
|||
} |
|||
return new TransactionSignature(signature); |
|||
}); |
|||
}; |
|||
|
|||
MultiSigInput.prototype._serializeSignatures = function() { |
|||
return _.map(this.signatures, function(signature) { |
|||
if (!signature) { |
|||
return undefined; |
|||
} |
|||
return signature.toObject(); |
|||
}); |
|||
}; |
|||
|
|||
MultiSigInput.prototype.getSignatures = function(transaction, privateKey, index, sigtype) { |
|||
$.checkState(this.output instanceof Output); |
|||
sigtype = sigtype || Signature.SIGHASH_ALL; |
|||
|
|||
var self = this; |
|||
var results = []; |
|||
_.each(this.publicKeys, function(publicKey) { |
|||
if (publicKey.toString() === privateKey.publicKey.toString()) { |
|||
results.push(new TransactionSignature({ |
|||
publicKey: privateKey.publicKey, |
|||
prevTxId: self.prevTxId, |
|||
outputIndex: self.outputIndex, |
|||
inputIndex: index, |
|||
signature: Sighash.sign(transaction, privateKey, sigtype, index, self.output.script), |
|||
sigtype: sigtype |
|||
})); |
|||
} |
|||
}); |
|||
|
|||
return results; |
|||
}; |
|||
|
|||
MultiSigInput.prototype.addSignature = function(transaction, signature) { |
|||
$.checkState(!this.isFullySigned(), 'All needed signatures have already been added'); |
|||
$.checkArgument(!_.isUndefined(this.publicKeyIndex[signature.publicKey.toString()]), |
|||
'Signature has no matching public key'); |
|||
$.checkState(this.isValidSignature(transaction, signature)); |
|||
this.signatures[this.publicKeyIndex[signature.publicKey.toString()]] = signature; |
|||
this._updateScript(); |
|||
return this; |
|||
}; |
|||
|
|||
MultiSigInput.prototype._updateScript = function() { |
|||
this.setScript(Script.buildMultisigIn( |
|||
this.publicKeys, |
|||
this.threshold, |
|||
this._createSignatures() |
|||
)); |
|||
return this; |
|||
}; |
|||
|
|||
MultiSigInput.prototype._createSignatures = function() { |
|||
return _.map( |
|||
_.filter(this.signatures, function(signature) { return !_.isUndefined(signature); }), |
|||
function(signature) { |
|||
return BufferUtil.concat([ |
|||
signature.signature.toDER(), |
|||
BufferUtil.integerAsSingleByteBuffer(signature.sigtype) |
|||
]); |
|||
} |
|||
); |
|||
}; |
|||
|
|||
MultiSigInput.prototype.clearSignatures = function() { |
|||
this.signatures = new Array(this.publicKeys.length); |
|||
this._updateScript(); |
|||
}; |
|||
|
|||
MultiSigInput.prototype.isFullySigned = function() { |
|||
return this.countSignatures() === this.threshold; |
|||
}; |
|||
|
|||
MultiSigInput.prototype.countMissingSignatures = function() { |
|||
return this.threshold - this.countSignatures(); |
|||
}; |
|||
|
|||
MultiSigInput.prototype.countSignatures = function() { |
|||
return _.reduce(this.signatures, function(sum, signature) { |
|||
return sum + (!!signature); |
|||
}, 0); |
|||
}; |
|||
|
|||
MultiSigInput.prototype.publicKeysWithoutSignature = function() { |
|||
var self = this; |
|||
return _.filter(this.publicKeys, function(publicKey) { |
|||
return !(self.signatures[self.publicKeyIndex[publicKey.toString()]]); |
|||
}); |
|||
}; |
|||
|
|||
MultiSigInput.prototype.isValidSignature = function(transaction, signature) { |
|||
// FIXME: Refactor signature so this is not necessary
|
|||
signature.signature.nhashtype = signature.sigtype; |
|||
return Sighash.verify( |
|||
transaction, |
|||
signature.signature, |
|||
signature.publicKey, |
|||
signature.inputIndex, |
|||
this.output.script |
|||
); |
|||
}; |
|||
|
|||
MultiSigInput.OPCODES_SIZE = 7; // serialized size (<=3) + 0 .. N .. M OP_CHECKMULTISIG
|
|||
MultiSigInput.SIGNATURE_SIZE = 74; // size (1) + DER (<=72) + sighash (1)
|
|||
MultiSigInput.PUBKEY_SIZE = 34; // size (1) + DER (<=33)
|
|||
|
|||
MultiSigInput.prototype._estimateSize = function() { |
|||
return MultiSigInput.OPCODES_SIZE + |
|||
this.threshold * MultiSigInput.SIGNATURE_SIZE + |
|||
this.publicKeys.length * MultiSigInput.PUBKEY_SIZE; |
|||
}; |
|||
|
|||
module.exports = MultiSigInput; |
Loading…
Reference in new issue