Refactor: Re-order helper functions based on like-kind
This commit is contained in:
junderw
parent
2fd4b9dc54
commit
479c56bbb4
2 changed files with 681 additions and 698 deletions
606
src/psbt.js
606
src/psbt.js
|
|
@ -352,31 +352,120 @@ class Psbt extends bip174_1.Psbt {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
exports.Psbt = Psbt;
|
exports.Psbt = Psbt;
|
||||||
function addNonWitnessTxCache(cache, input, inputIndex) {
|
function canFinalize(input, script, scriptType) {
|
||||||
cache.__NON_WITNESS_UTXO_BUF_CACHE[inputIndex] = input.nonWitnessUtxo;
|
switch (scriptType) {
|
||||||
const tx = transaction_1.Transaction.fromBuffer(input.nonWitnessUtxo);
|
case 'pubkey':
|
||||||
cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex] = tx;
|
case 'pubkeyhash':
|
||||||
const self = cache;
|
case 'witnesspubkeyhash':
|
||||||
const selfIndex = inputIndex;
|
return hasSigs(1, input.partialSig);
|
||||||
delete input.nonWitnessUtxo;
|
case 'multisig':
|
||||||
Object.defineProperty(input, 'nonWitnessUtxo', {
|
const p2ms = payments.p2ms({ output: script });
|
||||||
enumerable: true,
|
return hasSigs(p2ms.m, input.partialSig);
|
||||||
get() {
|
default:
|
||||||
const buf = self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex];
|
return false;
|
||||||
const txCache = self.__NON_WITNESS_UTXO_TX_CACHE[selfIndex];
|
}
|
||||||
if (buf !== undefined) {
|
}
|
||||||
return buf;
|
function hasSigs(neededSigs, partialSig) {
|
||||||
} else {
|
if (!partialSig) return false;
|
||||||
const newBuf = txCache.toBuffer();
|
if (partialSig.length > neededSigs) throw new Error('Too many signatures');
|
||||||
self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex] = newBuf;
|
return partialSig.length === neededSigs;
|
||||||
return newBuf;
|
}
|
||||||
|
function isFinalized(input) {
|
||||||
|
return !!input.finalScriptSig || !!input.finalScriptWitness;
|
||||||
|
}
|
||||||
|
function isPaymentFactory(payment) {
|
||||||
|
return script => {
|
||||||
|
try {
|
||||||
|
payment({ output: script });
|
||||||
|
return true;
|
||||||
|
} catch (err) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
const isP2MS = isPaymentFactory(payments.p2ms);
|
||||||
|
const isP2PK = isPaymentFactory(payments.p2pk);
|
||||||
|
const isP2PKH = isPaymentFactory(payments.p2pkh);
|
||||||
|
const isP2WPKH = isPaymentFactory(payments.p2wpkh);
|
||||||
|
function check32Bit(num) {
|
||||||
|
if (
|
||||||
|
typeof num !== 'number' ||
|
||||||
|
num !== Math.floor(num) ||
|
||||||
|
num > 0xffffffff ||
|
||||||
|
num < 0
|
||||||
|
) {
|
||||||
|
throw new Error('Invalid 32 bit integer');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function checkFees(psbt, cache, opts) {
|
||||||
|
const feeRate = cache.__FEE_RATE || psbt.getFeeRate();
|
||||||
|
const vsize = cache.__EXTRACTED_TX.virtualSize();
|
||||||
|
const satoshis = feeRate * vsize;
|
||||||
|
if (feeRate >= opts.maximumFeeRate) {
|
||||||
|
throw new Error(
|
||||||
|
`Warning: You are paying around ${(satoshis / 1e8).toFixed(8)} in ` +
|
||||||
|
`fees, which is ${feeRate} satoshi per byte for a transaction ` +
|
||||||
|
`with a VSize of ${vsize} bytes (segwit counted as 0.25 byte per ` +
|
||||||
|
`byte). Use setMaximumFeeRate method to raise your threshold, or ` +
|
||||||
|
`pass true to the first arg of extractTransaction.`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function checkInputsForPartialSig(inputs, action) {
|
||||||
|
inputs.forEach(input => {
|
||||||
|
let throws = false;
|
||||||
|
if ((input.partialSig || []).length === 0) return;
|
||||||
|
input.partialSig.forEach(pSig => {
|
||||||
|
const { hashType } = bscript.signature.decode(pSig.signature);
|
||||||
|
const whitelist = [];
|
||||||
|
const isAnyoneCanPay =
|
||||||
|
hashType & transaction_1.Transaction.SIGHASH_ANYONECANPAY;
|
||||||
|
if (isAnyoneCanPay) whitelist.push('addInput');
|
||||||
|
const hashMod = hashType & 0x1f;
|
||||||
|
switch (hashMod) {
|
||||||
|
case transaction_1.Transaction.SIGHASH_ALL:
|
||||||
|
break;
|
||||||
|
case transaction_1.Transaction.SIGHASH_SINGLE:
|
||||||
|
case transaction_1.Transaction.SIGHASH_NONE:
|
||||||
|
whitelist.push('addOutput');
|
||||||
|
whitelist.push('setSequence');
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
},
|
if (whitelist.indexOf(action) === -1) {
|
||||||
set(data) {
|
throws = true;
|
||||||
self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex] = data;
|
}
|
||||||
},
|
});
|
||||||
|
if (throws) {
|
||||||
|
throw new Error('Can not modify transaction, signatures exist.');
|
||||||
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
function checkScriptForPubkey(pubkey, script, action) {
|
||||||
|
const pubkeyHash = crypto_1.hash160(pubkey);
|
||||||
|
const decompiled = bscript.decompile(script);
|
||||||
|
if (decompiled === null) throw new Error('Unknown script error');
|
||||||
|
const hasKey = decompiled.some(element => {
|
||||||
|
if (typeof element === 'number') return false;
|
||||||
|
return element.equals(pubkey) || element.equals(pubkeyHash);
|
||||||
|
});
|
||||||
|
if (!hasKey) {
|
||||||
|
throw new Error(
|
||||||
|
`Can not ${action} for this input with the key ${pubkey.toString('hex')}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function checkTxEmpty(tx) {
|
||||||
|
const isEmpty = tx.ins.every(
|
||||||
|
input =>
|
||||||
|
input.script &&
|
||||||
|
input.script.length === 0 &&
|
||||||
|
input.witness &&
|
||||||
|
input.witness.length === 0,
|
||||||
|
);
|
||||||
|
if (!isEmpty) {
|
||||||
|
throw new Error('Format Error: Transaction ScriptSigs are not empty');
|
||||||
|
}
|
||||||
|
}
|
||||||
function checkTxForDupeIns(tx, cache) {
|
function checkTxForDupeIns(tx, cache) {
|
||||||
tx.ins.forEach(input => {
|
tx.ins.forEach(input => {
|
||||||
checkTxInputCache(cache, input);
|
checkTxInputCache(cache, input);
|
||||||
|
|
@ -390,18 +479,23 @@ function checkTxInputCache(cache, input) {
|
||||||
if (cache.__TX_IN_CACHE[key]) throw new Error('Duplicate input detected.');
|
if (cache.__TX_IN_CACHE[key]) throw new Error('Duplicate input detected.');
|
||||||
cache.__TX_IN_CACHE[key] = 1;
|
cache.__TX_IN_CACHE[key] = 1;
|
||||||
}
|
}
|
||||||
function isFinalized(input) {
|
function scriptCheckerFactory(payment, paymentScriptName) {
|
||||||
return !!input.finalScriptSig || !!input.finalScriptWitness;
|
return (inputIndex, scriptPubKey, redeemScript) => {
|
||||||
}
|
const redeemScriptOutput = payment({
|
||||||
function getHashAndSighashType(inputs, inputIndex, pubkey, cache) {
|
redeem: { output: redeemScript },
|
||||||
const input = utils_1.checkForInput(inputs, inputIndex);
|
}).output;
|
||||||
const { hash, sighashType, script } = getHashForSig(inputIndex, input, cache);
|
if (!scriptPubKey.equals(redeemScriptOutput)) {
|
||||||
checkScriptForPubkey(pubkey, script, 'sign');
|
throw new Error(
|
||||||
return {
|
`${paymentScriptName} for input #${inputIndex} doesn't match the scriptPubKey in the prevout`,
|
||||||
hash,
|
);
|
||||||
sighashType,
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
const checkRedeemScript = scriptCheckerFactory(payments.p2sh, 'Redeem script');
|
||||||
|
const checkWitnessScript = scriptCheckerFactory(
|
||||||
|
payments.p2wsh,
|
||||||
|
'Witness script',
|
||||||
|
);
|
||||||
function getFinalScripts(
|
function getFinalScripts(
|
||||||
script,
|
script,
|
||||||
scriptType,
|
scriptType,
|
||||||
|
|
@ -437,81 +531,14 @@ function getFinalScripts(
|
||||||
finalScriptWitness,
|
finalScriptWitness,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
function getSortedSigs(script, partialSig) {
|
function getHashAndSighashType(inputs, inputIndex, pubkey, cache) {
|
||||||
const p2ms = payments.p2ms({ output: script });
|
const input = utils_1.checkForInput(inputs, inputIndex);
|
||||||
// for each pubkey in order of p2ms script
|
const { hash, sighashType, script } = getHashForSig(inputIndex, input, cache);
|
||||||
return p2ms.pubkeys
|
checkScriptForPubkey(pubkey, script, 'sign');
|
||||||
.map(pk => {
|
return {
|
||||||
// filter partialSig array by pubkey being equal
|
hash,
|
||||||
return (
|
sighashType,
|
||||||
partialSig.filter(ps => {
|
};
|
||||||
return ps.pubkey.equals(pk);
|
|
||||||
})[0] || {}
|
|
||||||
).signature;
|
|
||||||
// Any pubkey without a match will return undefined
|
|
||||||
// this last filter removes all the undefined items in the array.
|
|
||||||
})
|
|
||||||
.filter(v => !!v);
|
|
||||||
}
|
|
||||||
function getPayment(script, scriptType, partialSig) {
|
|
||||||
let payment;
|
|
||||||
switch (scriptType) {
|
|
||||||
case 'multisig':
|
|
||||||
const sigs = getSortedSigs(script, partialSig);
|
|
||||||
payment = payments.p2ms({
|
|
||||||
output: script,
|
|
||||||
signatures: sigs,
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
case 'pubkey':
|
|
||||||
payment = payments.p2pk({
|
|
||||||
output: script,
|
|
||||||
signature: partialSig[0].signature,
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
case 'pubkeyhash':
|
|
||||||
payment = payments.p2pkh({
|
|
||||||
output: script,
|
|
||||||
pubkey: partialSig[0].pubkey,
|
|
||||||
signature: partialSig[0].signature,
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
case 'witnesspubkeyhash':
|
|
||||||
payment = payments.p2wpkh({
|
|
||||||
output: script,
|
|
||||||
pubkey: partialSig[0].pubkey,
|
|
||||||
signature: partialSig[0].signature,
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
return payment;
|
|
||||||
}
|
|
||||||
function canFinalize(input, script, scriptType) {
|
|
||||||
switch (scriptType) {
|
|
||||||
case 'pubkey':
|
|
||||||
case 'pubkeyhash':
|
|
||||||
case 'witnesspubkeyhash':
|
|
||||||
return hasSigs(1, input.partialSig);
|
|
||||||
case 'multisig':
|
|
||||||
const p2ms = payments.p2ms({ output: script });
|
|
||||||
return hasSigs(p2ms.m, input.partialSig);
|
|
||||||
default:
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
function checkScriptForPubkey(pubkey, script, action) {
|
|
||||||
const pubkeyHash = crypto_1.hash160(pubkey);
|
|
||||||
const decompiled = bscript.decompile(script);
|
|
||||||
if (decompiled === null) throw new Error('Unknown script error');
|
|
||||||
const hasKey = decompiled.some(element => {
|
|
||||||
if (typeof element === 'number') return false;
|
|
||||||
return element.equals(pubkey) || element.equals(pubkeyHash);
|
|
||||||
});
|
|
||||||
if (!hasKey) {
|
|
||||||
throw new Error(
|
|
||||||
`Can not ${action} for this input with the key ${pubkey.toString('hex')}`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
function getHashForSig(inputIndex, input, cache) {
|
function getHashForSig(inputIndex, input, cache) {
|
||||||
const unsignedTx = cache.__TX;
|
const unsignedTx = cache.__TX;
|
||||||
|
|
@ -597,182 +624,6 @@ function getHashForSig(inputIndex, input, cache) {
|
||||||
hash,
|
hash,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
function scriptCheckerFactory(payment, paymentScriptName) {
|
|
||||||
return (inputIndex, scriptPubKey, redeemScript) => {
|
|
||||||
const redeemScriptOutput = payment({
|
|
||||||
redeem: { output: redeemScript },
|
|
||||||
}).output;
|
|
||||||
if (!scriptPubKey.equals(redeemScriptOutput)) {
|
|
||||||
throw new Error(
|
|
||||||
`${paymentScriptName} for input #${inputIndex} doesn't match the scriptPubKey in the prevout`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
const checkRedeemScript = scriptCheckerFactory(payments.p2sh, 'Redeem script');
|
|
||||||
const checkWitnessScript = scriptCheckerFactory(
|
|
||||||
payments.p2wsh,
|
|
||||||
'Witness script',
|
|
||||||
);
|
|
||||||
function isPaymentFactory(payment) {
|
|
||||||
return script => {
|
|
||||||
try {
|
|
||||||
payment({ output: script });
|
|
||||||
return true;
|
|
||||||
} catch (err) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
const isP2WPKH = isPaymentFactory(payments.p2wpkh);
|
|
||||||
const isP2PKH = isPaymentFactory(payments.p2pkh);
|
|
||||||
const isP2MS = isPaymentFactory(payments.p2ms);
|
|
||||||
const isP2PK = isPaymentFactory(payments.p2pk);
|
|
||||||
function classifyScript(script) {
|
|
||||||
if (isP2WPKH(script)) return 'witnesspubkeyhash';
|
|
||||||
if (isP2PKH(script)) return 'pubkeyhash';
|
|
||||||
if (isP2MS(script)) return 'multisig';
|
|
||||||
if (isP2PK(script)) return 'pubkey';
|
|
||||||
return 'nonstandard';
|
|
||||||
}
|
|
||||||
function getScriptFromInput(inputIndex, input, cache) {
|
|
||||||
const unsignedTx = cache.__TX;
|
|
||||||
const res = {
|
|
||||||
script: null,
|
|
||||||
isSegwit: false,
|
|
||||||
isP2SH: false,
|
|
||||||
isP2WSH: false,
|
|
||||||
};
|
|
||||||
if (input.nonWitnessUtxo) {
|
|
||||||
if (input.redeemScript) {
|
|
||||||
res.isP2SH = true;
|
|
||||||
res.script = input.redeemScript;
|
|
||||||
} else {
|
|
||||||
const nonWitnessUtxoTx = nonWitnessUtxoTxFromCache(
|
|
||||||
cache,
|
|
||||||
input,
|
|
||||||
inputIndex,
|
|
||||||
);
|
|
||||||
const prevoutIndex = unsignedTx.ins[inputIndex].index;
|
|
||||||
res.script = nonWitnessUtxoTx.outs[prevoutIndex].script;
|
|
||||||
}
|
|
||||||
} else if (input.witnessUtxo) {
|
|
||||||
res.isSegwit = true;
|
|
||||||
res.isP2SH = !!input.redeemScript;
|
|
||||||
res.isP2WSH = !!input.witnessScript;
|
|
||||||
if (input.witnessScript) {
|
|
||||||
res.script = input.witnessScript;
|
|
||||||
} else if (input.redeemScript) {
|
|
||||||
res.script = payments.p2wpkh({
|
|
||||||
hash: input.redeemScript.slice(2),
|
|
||||||
}).output;
|
|
||||||
} else {
|
|
||||||
res.script = payments.p2wpkh({
|
|
||||||
hash: input.witnessUtxo.script.slice(2),
|
|
||||||
}).output;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return res;
|
|
||||||
}
|
|
||||||
function hasSigs(neededSigs, partialSig) {
|
|
||||||
if (!partialSig) return false;
|
|
||||||
if (partialSig.length > neededSigs) throw new Error('Too many signatures');
|
|
||||||
return partialSig.length === neededSigs;
|
|
||||||
}
|
|
||||||
function witnessStackToScriptWitness(witness) {
|
|
||||||
let buffer = Buffer.allocUnsafe(0);
|
|
||||||
function writeSlice(slice) {
|
|
||||||
buffer = Buffer.concat([buffer, Buffer.from(slice)]);
|
|
||||||
}
|
|
||||||
function writeVarInt(i) {
|
|
||||||
const currentLen = buffer.length;
|
|
||||||
const varintLen = varuint.encodingLength(i);
|
|
||||||
buffer = Buffer.concat([buffer, Buffer.allocUnsafe(varintLen)]);
|
|
||||||
varuint.encode(i, buffer, currentLen);
|
|
||||||
}
|
|
||||||
function writeVarSlice(slice) {
|
|
||||||
writeVarInt(slice.length);
|
|
||||||
writeSlice(slice);
|
|
||||||
}
|
|
||||||
function writeVector(vector) {
|
|
||||||
writeVarInt(vector.length);
|
|
||||||
vector.forEach(writeVarSlice);
|
|
||||||
}
|
|
||||||
writeVector(witness);
|
|
||||||
return buffer;
|
|
||||||
}
|
|
||||||
function scriptWitnessToWitnessStack(buffer) {
|
|
||||||
let offset = 0;
|
|
||||||
function readSlice(n) {
|
|
||||||
offset += n;
|
|
||||||
return buffer.slice(offset - n, offset);
|
|
||||||
}
|
|
||||||
function readVarInt() {
|
|
||||||
const vi = varuint.decode(buffer, offset);
|
|
||||||
offset += varuint.decode.bytes;
|
|
||||||
return vi;
|
|
||||||
}
|
|
||||||
function readVarSlice() {
|
|
||||||
return readSlice(readVarInt());
|
|
||||||
}
|
|
||||||
function readVector() {
|
|
||||||
const count = readVarInt();
|
|
||||||
const vector = [];
|
|
||||||
for (let i = 0; i < count; i++) vector.push(readVarSlice());
|
|
||||||
return vector;
|
|
||||||
}
|
|
||||||
return readVector();
|
|
||||||
}
|
|
||||||
function range(n) {
|
|
||||||
return [...Array(n).keys()];
|
|
||||||
}
|
|
||||||
function checkTxEmpty(tx) {
|
|
||||||
const isEmpty = tx.ins.every(
|
|
||||||
input =>
|
|
||||||
input.script &&
|
|
||||||
input.script.length === 0 &&
|
|
||||||
input.witness &&
|
|
||||||
input.witness.length === 0,
|
|
||||||
);
|
|
||||||
if (!isEmpty) {
|
|
||||||
throw new Error('Format Error: Transaction ScriptSigs are not empty');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
function checkInputsForPartialSig(inputs, action) {
|
|
||||||
inputs.forEach(input => {
|
|
||||||
let throws = false;
|
|
||||||
if ((input.partialSig || []).length === 0) return;
|
|
||||||
input.partialSig.forEach(pSig => {
|
|
||||||
const { hashType } = bscript.signature.decode(pSig.signature);
|
|
||||||
const whitelist = [];
|
|
||||||
const isAnyoneCanPay =
|
|
||||||
hashType & transaction_1.Transaction.SIGHASH_ANYONECANPAY;
|
|
||||||
if (isAnyoneCanPay) whitelist.push('addInput');
|
|
||||||
const hashMod = hashType & 0x1f;
|
|
||||||
switch (hashMod) {
|
|
||||||
case transaction_1.Transaction.SIGHASH_ALL:
|
|
||||||
break;
|
|
||||||
case transaction_1.Transaction.SIGHASH_SINGLE:
|
|
||||||
case transaction_1.Transaction.SIGHASH_NONE:
|
|
||||||
whitelist.push('addOutput');
|
|
||||||
whitelist.push('setSequence');
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
if (whitelist.indexOf(action) === -1) {
|
|
||||||
throws = true;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
if (throws) {
|
|
||||||
throw new Error('Can not modify transaction, signatures exist.');
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
function nonWitnessUtxoTxFromCache(cache, input, inputIndex) {
|
|
||||||
if (!cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex]) {
|
|
||||||
addNonWitnessTxCache(cache, input, inputIndex);
|
|
||||||
}
|
|
||||||
return cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex];
|
|
||||||
}
|
|
||||||
function getInputAdder(cache) {
|
function getInputAdder(cache) {
|
||||||
const selfCache = cache;
|
const selfCache = cache;
|
||||||
return (_inputData, txBuf) => {
|
return (_inputData, txBuf) => {
|
||||||
|
|
@ -822,19 +673,162 @@ function getOutputAdder(cache) {
|
||||||
return selfCache.__TX.toBuffer();
|
return selfCache.__TX.toBuffer();
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
function checkFees(psbt, cache, opts) {
|
function getPayment(script, scriptType, partialSig) {
|
||||||
const feeRate = cache.__FEE_RATE || psbt.getFeeRate();
|
let payment;
|
||||||
const vsize = cache.__EXTRACTED_TX.virtualSize();
|
switch (scriptType) {
|
||||||
const satoshis = feeRate * vsize;
|
case 'multisig':
|
||||||
if (feeRate >= opts.maximumFeeRate) {
|
const sigs = getSortedSigs(script, partialSig);
|
||||||
throw new Error(
|
payment = payments.p2ms({
|
||||||
`Warning: You are paying around ${(satoshis / 1e8).toFixed(8)} in ` +
|
output: script,
|
||||||
`fees, which is ${feeRate} satoshi per byte for a transaction ` +
|
signatures: sigs,
|
||||||
`with a VSize of ${vsize} bytes (segwit counted as 0.25 byte per ` +
|
});
|
||||||
`byte). Use setMaximumFeeRate method to raise your threshold, or ` +
|
break;
|
||||||
`pass true to the first arg of extractTransaction.`,
|
case 'pubkey':
|
||||||
);
|
payment = payments.p2pk({
|
||||||
|
output: script,
|
||||||
|
signature: partialSig[0].signature,
|
||||||
|
});
|
||||||
|
break;
|
||||||
|
case 'pubkeyhash':
|
||||||
|
payment = payments.p2pkh({
|
||||||
|
output: script,
|
||||||
|
pubkey: partialSig[0].pubkey,
|
||||||
|
signature: partialSig[0].signature,
|
||||||
|
});
|
||||||
|
break;
|
||||||
|
case 'witnesspubkeyhash':
|
||||||
|
payment = payments.p2wpkh({
|
||||||
|
output: script,
|
||||||
|
pubkey: partialSig[0].pubkey,
|
||||||
|
signature: partialSig[0].signature,
|
||||||
|
});
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
|
return payment;
|
||||||
|
}
|
||||||
|
function getScriptFromInput(inputIndex, input, cache) {
|
||||||
|
const unsignedTx = cache.__TX;
|
||||||
|
const res = {
|
||||||
|
script: null,
|
||||||
|
isSegwit: false,
|
||||||
|
isP2SH: false,
|
||||||
|
isP2WSH: false,
|
||||||
|
};
|
||||||
|
if (input.nonWitnessUtxo) {
|
||||||
|
if (input.redeemScript) {
|
||||||
|
res.isP2SH = true;
|
||||||
|
res.script = input.redeemScript;
|
||||||
|
} else {
|
||||||
|
const nonWitnessUtxoTx = nonWitnessUtxoTxFromCache(
|
||||||
|
cache,
|
||||||
|
input,
|
||||||
|
inputIndex,
|
||||||
|
);
|
||||||
|
const prevoutIndex = unsignedTx.ins[inputIndex].index;
|
||||||
|
res.script = nonWitnessUtxoTx.outs[prevoutIndex].script;
|
||||||
|
}
|
||||||
|
} else if (input.witnessUtxo) {
|
||||||
|
res.isSegwit = true;
|
||||||
|
res.isP2SH = !!input.redeemScript;
|
||||||
|
res.isP2WSH = !!input.witnessScript;
|
||||||
|
if (input.witnessScript) {
|
||||||
|
res.script = input.witnessScript;
|
||||||
|
} else if (input.redeemScript) {
|
||||||
|
res.script = payments.p2wpkh({
|
||||||
|
hash: input.redeemScript.slice(2),
|
||||||
|
}).output;
|
||||||
|
} else {
|
||||||
|
res.script = payments.p2wpkh({
|
||||||
|
hash: input.witnessUtxo.script.slice(2),
|
||||||
|
}).output;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
function getSortedSigs(script, partialSig) {
|
||||||
|
const p2ms = payments.p2ms({ output: script });
|
||||||
|
// for each pubkey in order of p2ms script
|
||||||
|
return p2ms.pubkeys
|
||||||
|
.map(pk => {
|
||||||
|
// filter partialSig array by pubkey being equal
|
||||||
|
return (
|
||||||
|
partialSig.filter(ps => {
|
||||||
|
return ps.pubkey.equals(pk);
|
||||||
|
})[0] || {}
|
||||||
|
).signature;
|
||||||
|
// Any pubkey without a match will return undefined
|
||||||
|
// this last filter removes all the undefined items in the array.
|
||||||
|
})
|
||||||
|
.filter(v => !!v);
|
||||||
|
}
|
||||||
|
function scriptWitnessToWitnessStack(buffer) {
|
||||||
|
let offset = 0;
|
||||||
|
function readSlice(n) {
|
||||||
|
offset += n;
|
||||||
|
return buffer.slice(offset - n, offset);
|
||||||
|
}
|
||||||
|
function readVarInt() {
|
||||||
|
const vi = varuint.decode(buffer, offset);
|
||||||
|
offset += varuint.decode.bytes;
|
||||||
|
return vi;
|
||||||
|
}
|
||||||
|
function readVarSlice() {
|
||||||
|
return readSlice(readVarInt());
|
||||||
|
}
|
||||||
|
function readVector() {
|
||||||
|
const count = readVarInt();
|
||||||
|
const vector = [];
|
||||||
|
for (let i = 0; i < count; i++) vector.push(readVarSlice());
|
||||||
|
return vector;
|
||||||
|
}
|
||||||
|
return readVector();
|
||||||
|
}
|
||||||
|
function witnessStackToScriptWitness(witness) {
|
||||||
|
let buffer = Buffer.allocUnsafe(0);
|
||||||
|
function writeSlice(slice) {
|
||||||
|
buffer = Buffer.concat([buffer, Buffer.from(slice)]);
|
||||||
|
}
|
||||||
|
function writeVarInt(i) {
|
||||||
|
const currentLen = buffer.length;
|
||||||
|
const varintLen = varuint.encodingLength(i);
|
||||||
|
buffer = Buffer.concat([buffer, Buffer.allocUnsafe(varintLen)]);
|
||||||
|
varuint.encode(i, buffer, currentLen);
|
||||||
|
}
|
||||||
|
function writeVarSlice(slice) {
|
||||||
|
writeVarInt(slice.length);
|
||||||
|
writeSlice(slice);
|
||||||
|
}
|
||||||
|
function writeVector(vector) {
|
||||||
|
writeVarInt(vector.length);
|
||||||
|
vector.forEach(writeVarSlice);
|
||||||
|
}
|
||||||
|
writeVector(witness);
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
function addNonWitnessTxCache(cache, input, inputIndex) {
|
||||||
|
cache.__NON_WITNESS_UTXO_BUF_CACHE[inputIndex] = input.nonWitnessUtxo;
|
||||||
|
const tx = transaction_1.Transaction.fromBuffer(input.nonWitnessUtxo);
|
||||||
|
cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex] = tx;
|
||||||
|
const self = cache;
|
||||||
|
const selfIndex = inputIndex;
|
||||||
|
delete input.nonWitnessUtxo;
|
||||||
|
Object.defineProperty(input, 'nonWitnessUtxo', {
|
||||||
|
enumerable: true,
|
||||||
|
get() {
|
||||||
|
const buf = self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex];
|
||||||
|
const txCache = self.__NON_WITNESS_UTXO_TX_CACHE[selfIndex];
|
||||||
|
if (buf !== undefined) {
|
||||||
|
return buf;
|
||||||
|
} else {
|
||||||
|
const newBuf = txCache.toBuffer();
|
||||||
|
self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex] = newBuf;
|
||||||
|
return newBuf;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
set(data) {
|
||||||
|
self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex] = data;
|
||||||
|
},
|
||||||
|
});
|
||||||
}
|
}
|
||||||
function inputFinalizeGetAmts(inputs, tx, cache, mustFinalize, getAmounts) {
|
function inputFinalizeGetAmts(inputs, tx, cache, mustFinalize, getAmounts) {
|
||||||
let inputAmount = 0;
|
let inputAmount = 0;
|
||||||
|
|
@ -857,13 +851,19 @@ function inputFinalizeGetAmts(inputs, tx, cache, mustFinalize, getAmounts) {
|
||||||
});
|
});
|
||||||
return inputAmount;
|
return inputAmount;
|
||||||
}
|
}
|
||||||
function check32Bit(num) {
|
function nonWitnessUtxoTxFromCache(cache, input, inputIndex) {
|
||||||
if (
|
if (!cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex]) {
|
||||||
typeof num !== 'number' ||
|
addNonWitnessTxCache(cache, input, inputIndex);
|
||||||
num !== Math.floor(num) ||
|
|
||||||
num > 0xffffffff ||
|
|
||||||
num < 0
|
|
||||||
) {
|
|
||||||
throw new Error('Invalid 32 bit integer');
|
|
||||||
}
|
}
|
||||||
|
return cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex];
|
||||||
|
}
|
||||||
|
function classifyScript(script) {
|
||||||
|
if (isP2WPKH(script)) return 'witnesspubkeyhash';
|
||||||
|
if (isP2PKH(script)) return 'pubkeyhash';
|
||||||
|
if (isP2MS(script)) return 'multisig';
|
||||||
|
if (isP2PK(script)) return 'pubkey';
|
||||||
|
return 'nonstandard';
|
||||||
|
}
|
||||||
|
function range(n) {
|
||||||
|
return [...Array(n).keys()];
|
||||||
}
|
}
|
||||||
|
|
|
||||||
773
ts_src/psbt.ts
773
ts_src/psbt.ts
|
|
@ -425,16 +425,6 @@ export class Psbt extends PsbtBase {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
//
|
|
||||||
//
|
|
||||||
//
|
|
||||||
//
|
|
||||||
// Helper functions
|
|
||||||
//
|
|
||||||
//
|
|
||||||
//
|
|
||||||
//
|
|
||||||
|
|
||||||
interface PsbtCache {
|
interface PsbtCache {
|
||||||
__NON_WITNESS_UTXO_TX_CACHE: Transaction[];
|
__NON_WITNESS_UTXO_TX_CACHE: Transaction[];
|
||||||
__NON_WITNESS_UTXO_BUF_CACHE: Buffer[];
|
__NON_WITNESS_UTXO_BUF_CACHE: Buffer[];
|
||||||
|
|
@ -455,38 +445,139 @@ interface PsbtOpts {
|
||||||
maximumFeeRate: number;
|
maximumFeeRate: number;
|
||||||
}
|
}
|
||||||
|
|
||||||
function addNonWitnessTxCache(
|
function canFinalize(
|
||||||
cache: PsbtCache,
|
|
||||||
input: PsbtInput,
|
input: PsbtInput,
|
||||||
inputIndex: number,
|
script: Buffer,
|
||||||
): void {
|
scriptType: string,
|
||||||
cache.__NON_WITNESS_UTXO_BUF_CACHE[inputIndex] = input.nonWitnessUtxo!;
|
): boolean {
|
||||||
|
switch (scriptType) {
|
||||||
|
case 'pubkey':
|
||||||
|
case 'pubkeyhash':
|
||||||
|
case 'witnesspubkeyhash':
|
||||||
|
return hasSigs(1, input.partialSig);
|
||||||
|
case 'multisig':
|
||||||
|
const p2ms = payments.p2ms({ output: script });
|
||||||
|
return hasSigs(p2ms.m!, input.partialSig);
|
||||||
|
default:
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
const tx = Transaction.fromBuffer(input.nonWitnessUtxo!);
|
function hasSigs(neededSigs: number, partialSig?: any[]): boolean {
|
||||||
cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex] = tx;
|
if (!partialSig) return false;
|
||||||
|
if (partialSig.length > neededSigs) throw new Error('Too many signatures');
|
||||||
|
return partialSig.length === neededSigs;
|
||||||
|
}
|
||||||
|
|
||||||
const self = cache;
|
function isFinalized(input: PsbtInput): boolean {
|
||||||
const selfIndex = inputIndex;
|
return !!input.finalScriptSig || !!input.finalScriptWitness;
|
||||||
delete input.nonWitnessUtxo;
|
}
|
||||||
Object.defineProperty(input, 'nonWitnessUtxo', {
|
|
||||||
enumerable: true,
|
function isPaymentFactory(payment: any): (script: Buffer) => boolean {
|
||||||
get(): Buffer {
|
return (script: Buffer): boolean => {
|
||||||
const buf = self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex];
|
try {
|
||||||
const txCache = self.__NON_WITNESS_UTXO_TX_CACHE[selfIndex];
|
payment({ output: script });
|
||||||
if (buf !== undefined) {
|
return true;
|
||||||
return buf;
|
} catch (err) {
|
||||||
} else {
|
return false;
|
||||||
const newBuf = txCache.toBuffer();
|
}
|
||||||
self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex] = newBuf;
|
};
|
||||||
return newBuf;
|
}
|
||||||
|
const isP2MS = isPaymentFactory(payments.p2ms);
|
||||||
|
const isP2PK = isPaymentFactory(payments.p2pk);
|
||||||
|
const isP2PKH = isPaymentFactory(payments.p2pkh);
|
||||||
|
const isP2WPKH = isPaymentFactory(payments.p2wpkh);
|
||||||
|
|
||||||
|
function check32Bit(num: number): void {
|
||||||
|
if (
|
||||||
|
typeof num !== 'number' ||
|
||||||
|
num !== Math.floor(num) ||
|
||||||
|
num > 0xffffffff ||
|
||||||
|
num < 0
|
||||||
|
) {
|
||||||
|
throw new Error('Invalid 32 bit integer');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function checkFees(psbt: Psbt, cache: PsbtCache, opts: PsbtOpts): void {
|
||||||
|
const feeRate = cache.__FEE_RATE || psbt.getFeeRate();
|
||||||
|
const vsize = cache.__EXTRACTED_TX!.virtualSize();
|
||||||
|
const satoshis = feeRate * vsize;
|
||||||
|
if (feeRate >= opts.maximumFeeRate) {
|
||||||
|
throw new Error(
|
||||||
|
`Warning: You are paying around ${(satoshis / 1e8).toFixed(8)} in ` +
|
||||||
|
`fees, which is ${feeRate} satoshi per byte for a transaction ` +
|
||||||
|
`with a VSize of ${vsize} bytes (segwit counted as 0.25 byte per ` +
|
||||||
|
`byte). Use setMaximumFeeRate method to raise your threshold, or ` +
|
||||||
|
`pass true to the first arg of extractTransaction.`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function checkInputsForPartialSig(inputs: PsbtInput[], action: string): void {
|
||||||
|
inputs.forEach(input => {
|
||||||
|
let throws = false;
|
||||||
|
if ((input.partialSig || []).length === 0) return;
|
||||||
|
input.partialSig!.forEach(pSig => {
|
||||||
|
const { hashType } = bscript.signature.decode(pSig.signature);
|
||||||
|
const whitelist: string[] = [];
|
||||||
|
const isAnyoneCanPay = hashType & Transaction.SIGHASH_ANYONECANPAY;
|
||||||
|
if (isAnyoneCanPay) whitelist.push('addInput');
|
||||||
|
const hashMod = hashType & 0x1f;
|
||||||
|
switch (hashMod) {
|
||||||
|
case Transaction.SIGHASH_ALL:
|
||||||
|
break;
|
||||||
|
case Transaction.SIGHASH_SINGLE:
|
||||||
|
case Transaction.SIGHASH_NONE:
|
||||||
|
whitelist.push('addOutput');
|
||||||
|
whitelist.push('setSequence');
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
},
|
if (whitelist.indexOf(action) === -1) {
|
||||||
set(data: Buffer): void {
|
throws = true;
|
||||||
self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex] = data;
|
}
|
||||||
},
|
});
|
||||||
|
if (throws) {
|
||||||
|
throw new Error('Can not modify transaction, signatures exist.');
|
||||||
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function checkScriptForPubkey(
|
||||||
|
pubkey: Buffer,
|
||||||
|
script: Buffer,
|
||||||
|
action: string,
|
||||||
|
): void {
|
||||||
|
const pubkeyHash = hash160(pubkey);
|
||||||
|
|
||||||
|
const decompiled = bscript.decompile(script);
|
||||||
|
if (decompiled === null) throw new Error('Unknown script error');
|
||||||
|
|
||||||
|
const hasKey = decompiled.some(element => {
|
||||||
|
if (typeof element === 'number') return false;
|
||||||
|
return element.equals(pubkey) || element.equals(pubkeyHash);
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!hasKey) {
|
||||||
|
throw new Error(
|
||||||
|
`Can not ${action} for this input with the key ${pubkey.toString('hex')}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function checkTxEmpty(tx: Transaction): void {
|
||||||
|
const isEmpty = tx.ins.every(
|
||||||
|
input =>
|
||||||
|
input.script &&
|
||||||
|
input.script.length === 0 &&
|
||||||
|
input.witness &&
|
||||||
|
input.witness.length === 0,
|
||||||
|
);
|
||||||
|
if (!isEmpty) {
|
||||||
|
throw new Error('Format Error: Transaction ScriptSigs are not empty');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
function checkTxForDupeIns(tx: Transaction, cache: PsbtCache): void {
|
function checkTxForDupeIns(tx: Transaction, cache: PsbtCache): void {
|
||||||
tx.ins.forEach(input => {
|
tx.ins.forEach(input => {
|
||||||
checkTxInputCache(cache, input);
|
checkTxInputCache(cache, input);
|
||||||
|
|
@ -503,27 +594,31 @@ function checkTxInputCache(
|
||||||
cache.__TX_IN_CACHE[key] = 1;
|
cache.__TX_IN_CACHE[key] = 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
function isFinalized(input: PsbtInput): boolean {
|
function scriptCheckerFactory(
|
||||||
return !!input.finalScriptSig || !!input.finalScriptWitness;
|
payment: any,
|
||||||
}
|
paymentScriptName: string,
|
||||||
|
): (idx: number, spk: Buffer, rs: Buffer) => void {
|
||||||
|
return (
|
||||||
|
inputIndex: number,
|
||||||
|
scriptPubKey: Buffer,
|
||||||
|
redeemScript: Buffer,
|
||||||
|
): void => {
|
||||||
|
const redeemScriptOutput = payment({
|
||||||
|
redeem: { output: redeemScript },
|
||||||
|
}).output as Buffer;
|
||||||
|
|
||||||
function getHashAndSighashType(
|
if (!scriptPubKey.equals(redeemScriptOutput)) {
|
||||||
inputs: PsbtInput[],
|
throw new Error(
|
||||||
inputIndex: number,
|
`${paymentScriptName} for input #${inputIndex} doesn't match the scriptPubKey in the prevout`,
|
||||||
pubkey: Buffer,
|
);
|
||||||
cache: PsbtCache,
|
}
|
||||||
): {
|
|
||||||
hash: Buffer;
|
|
||||||
sighashType: number;
|
|
||||||
} {
|
|
||||||
const input = checkForInput(inputs, inputIndex);
|
|
||||||
const { hash, sighashType, script } = getHashForSig(inputIndex, input, cache);
|
|
||||||
checkScriptForPubkey(pubkey, script, 'sign');
|
|
||||||
return {
|
|
||||||
hash,
|
|
||||||
sighashType,
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
const checkRedeemScript = scriptCheckerFactory(payments.p2sh, 'Redeem script');
|
||||||
|
const checkWitnessScript = scriptCheckerFactory(
|
||||||
|
payments.p2wsh,
|
||||||
|
'Witness script',
|
||||||
|
);
|
||||||
|
|
||||||
function getFinalScripts(
|
function getFinalScripts(
|
||||||
script: Buffer,
|
script: Buffer,
|
||||||
|
|
@ -566,112 +661,33 @@ function getFinalScripts(
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
function getSortedSigs(script: Buffer, partialSig: PartialSig[]): Buffer[] {
|
function getHashAndSighashType(
|
||||||
const p2ms = payments.p2ms({ output: script });
|
inputs: PsbtInput[],
|
||||||
// for each pubkey in order of p2ms script
|
inputIndex: number,
|
||||||
return p2ms
|
|
||||||
.pubkeys!.map(pk => {
|
|
||||||
// filter partialSig array by pubkey being equal
|
|
||||||
return (
|
|
||||||
partialSig.filter(ps => {
|
|
||||||
return ps.pubkey.equals(pk);
|
|
||||||
})[0] || {}
|
|
||||||
).signature;
|
|
||||||
// Any pubkey without a match will return undefined
|
|
||||||
// this last filter removes all the undefined items in the array.
|
|
||||||
})
|
|
||||||
.filter(v => !!v);
|
|
||||||
}
|
|
||||||
|
|
||||||
function getPayment(
|
|
||||||
script: Buffer,
|
|
||||||
scriptType: string,
|
|
||||||
partialSig: PartialSig[],
|
|
||||||
): payments.Payment {
|
|
||||||
let payment: payments.Payment;
|
|
||||||
switch (scriptType) {
|
|
||||||
case 'multisig':
|
|
||||||
const sigs = getSortedSigs(script, partialSig);
|
|
||||||
payment = payments.p2ms({
|
|
||||||
output: script,
|
|
||||||
signatures: sigs,
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
case 'pubkey':
|
|
||||||
payment = payments.p2pk({
|
|
||||||
output: script,
|
|
||||||
signature: partialSig[0].signature,
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
case 'pubkeyhash':
|
|
||||||
payment = payments.p2pkh({
|
|
||||||
output: script,
|
|
||||||
pubkey: partialSig[0].pubkey,
|
|
||||||
signature: partialSig[0].signature,
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
case 'witnesspubkeyhash':
|
|
||||||
payment = payments.p2wpkh({
|
|
||||||
output: script,
|
|
||||||
pubkey: partialSig[0].pubkey,
|
|
||||||
signature: partialSig[0].signature,
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
return payment!;
|
|
||||||
}
|
|
||||||
|
|
||||||
function canFinalize(
|
|
||||||
input: PsbtInput,
|
|
||||||
script: Buffer,
|
|
||||||
scriptType: string,
|
|
||||||
): boolean {
|
|
||||||
switch (scriptType) {
|
|
||||||
case 'pubkey':
|
|
||||||
case 'pubkeyhash':
|
|
||||||
case 'witnesspubkeyhash':
|
|
||||||
return hasSigs(1, input.partialSig);
|
|
||||||
case 'multisig':
|
|
||||||
const p2ms = payments.p2ms({ output: script });
|
|
||||||
return hasSigs(p2ms.m!, input.partialSig);
|
|
||||||
default:
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function checkScriptForPubkey(
|
|
||||||
pubkey: Buffer,
|
pubkey: Buffer,
|
||||||
script: Buffer,
|
cache: PsbtCache,
|
||||||
action: string,
|
): {
|
||||||
): void {
|
|
||||||
const pubkeyHash = hash160(pubkey);
|
|
||||||
|
|
||||||
const decompiled = bscript.decompile(script);
|
|
||||||
if (decompiled === null) throw new Error('Unknown script error');
|
|
||||||
|
|
||||||
const hasKey = decompiled.some(element => {
|
|
||||||
if (typeof element === 'number') return false;
|
|
||||||
return element.equals(pubkey) || element.equals(pubkeyHash);
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!hasKey) {
|
|
||||||
throw new Error(
|
|
||||||
`Can not ${action} for this input with the key ${pubkey.toString('hex')}`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
interface HashForSigData {
|
|
||||||
script: Buffer;
|
|
||||||
hash: Buffer;
|
hash: Buffer;
|
||||||
sighashType: number;
|
sighashType: number;
|
||||||
|
} {
|
||||||
|
const input = checkForInput(inputs, inputIndex);
|
||||||
|
const { hash, sighashType, script } = getHashForSig(inputIndex, input, cache);
|
||||||
|
checkScriptForPubkey(pubkey, script, 'sign');
|
||||||
|
return {
|
||||||
|
hash,
|
||||||
|
sighashType,
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
function getHashForSig(
|
function getHashForSig(
|
||||||
inputIndex: number,
|
inputIndex: number,
|
||||||
input: PsbtInput,
|
input: PsbtInput,
|
||||||
cache: PsbtCache,
|
cache: PsbtCache,
|
||||||
): HashForSigData {
|
): {
|
||||||
|
script: Buffer;
|
||||||
|
hash: Buffer;
|
||||||
|
sighashType: number;
|
||||||
|
} {
|
||||||
const unsignedTx = cache.__TX;
|
const unsignedTx = cache.__TX;
|
||||||
const sighashType = input.sighashType || Transaction.SIGHASH_ALL;
|
const sighashType = input.sighashType || Transaction.SIGHASH_ALL;
|
||||||
let hash: Buffer;
|
let hash: Buffer;
|
||||||
|
|
@ -760,231 +776,6 @@ function getHashForSig(
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
type ScriptCheckerFunction = (idx: number, spk: Buffer, rs: Buffer) => void;
|
|
||||||
|
|
||||||
function scriptCheckerFactory(
|
|
||||||
payment: any,
|
|
||||||
paymentScriptName: string,
|
|
||||||
): ScriptCheckerFunction {
|
|
||||||
return (
|
|
||||||
inputIndex: number,
|
|
||||||
scriptPubKey: Buffer,
|
|
||||||
redeemScript: Buffer,
|
|
||||||
): void => {
|
|
||||||
const redeemScriptOutput = payment({
|
|
||||||
redeem: { output: redeemScript },
|
|
||||||
}).output as Buffer;
|
|
||||||
|
|
||||||
if (!scriptPubKey.equals(redeemScriptOutput)) {
|
|
||||||
throw new Error(
|
|
||||||
`${paymentScriptName} for input #${inputIndex} doesn't match the scriptPubKey in the prevout`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
const checkRedeemScript = scriptCheckerFactory(payments.p2sh, 'Redeem script');
|
|
||||||
const checkWitnessScript = scriptCheckerFactory(
|
|
||||||
payments.p2wsh,
|
|
||||||
'Witness script',
|
|
||||||
);
|
|
||||||
|
|
||||||
type isPaymentFunction = (script: Buffer) => boolean;
|
|
||||||
|
|
||||||
function isPaymentFactory(payment: any): isPaymentFunction {
|
|
||||||
return (script: Buffer): boolean => {
|
|
||||||
try {
|
|
||||||
payment({ output: script });
|
|
||||||
return true;
|
|
||||||
} catch (err) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
const isP2WPKH = isPaymentFactory(payments.p2wpkh);
|
|
||||||
const isP2PKH = isPaymentFactory(payments.p2pkh);
|
|
||||||
const isP2MS = isPaymentFactory(payments.p2ms);
|
|
||||||
const isP2PK = isPaymentFactory(payments.p2pk);
|
|
||||||
|
|
||||||
function classifyScript(script: Buffer): string {
|
|
||||||
if (isP2WPKH(script)) return 'witnesspubkeyhash';
|
|
||||||
if (isP2PKH(script)) return 'pubkeyhash';
|
|
||||||
if (isP2MS(script)) return 'multisig';
|
|
||||||
if (isP2PK(script)) return 'pubkey';
|
|
||||||
return 'nonstandard';
|
|
||||||
}
|
|
||||||
|
|
||||||
interface GetScriptReturn {
|
|
||||||
script: Buffer | null;
|
|
||||||
isSegwit: boolean;
|
|
||||||
isP2SH: boolean;
|
|
||||||
isP2WSH: boolean;
|
|
||||||
}
|
|
||||||
function getScriptFromInput(
|
|
||||||
inputIndex: number,
|
|
||||||
input: PsbtInput,
|
|
||||||
cache: PsbtCache,
|
|
||||||
): GetScriptReturn {
|
|
||||||
const unsignedTx = cache.__TX;
|
|
||||||
const res: GetScriptReturn = {
|
|
||||||
script: null,
|
|
||||||
isSegwit: false,
|
|
||||||
isP2SH: false,
|
|
||||||
isP2WSH: false,
|
|
||||||
};
|
|
||||||
if (input.nonWitnessUtxo) {
|
|
||||||
if (input.redeemScript) {
|
|
||||||
res.isP2SH = true;
|
|
||||||
res.script = input.redeemScript;
|
|
||||||
} else {
|
|
||||||
const nonWitnessUtxoTx = nonWitnessUtxoTxFromCache(
|
|
||||||
cache,
|
|
||||||
input,
|
|
||||||
inputIndex,
|
|
||||||
);
|
|
||||||
const prevoutIndex = unsignedTx.ins[inputIndex].index;
|
|
||||||
res.script = nonWitnessUtxoTx.outs[prevoutIndex].script;
|
|
||||||
}
|
|
||||||
} else if (input.witnessUtxo) {
|
|
||||||
res.isSegwit = true;
|
|
||||||
res.isP2SH = !!input.redeemScript;
|
|
||||||
res.isP2WSH = !!input.witnessScript;
|
|
||||||
if (input.witnessScript) {
|
|
||||||
res.script = input.witnessScript;
|
|
||||||
} else if (input.redeemScript) {
|
|
||||||
res.script = payments.p2wpkh({
|
|
||||||
hash: input.redeemScript.slice(2),
|
|
||||||
}).output!;
|
|
||||||
} else {
|
|
||||||
res.script = payments.p2wpkh({
|
|
||||||
hash: input.witnessUtxo.script.slice(2),
|
|
||||||
}).output!;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return res;
|
|
||||||
}
|
|
||||||
|
|
||||||
function hasSigs(neededSigs: number, partialSig?: any[]): boolean {
|
|
||||||
if (!partialSig) return false;
|
|
||||||
if (partialSig.length > neededSigs) throw new Error('Too many signatures');
|
|
||||||
return partialSig.length === neededSigs;
|
|
||||||
}
|
|
||||||
|
|
||||||
function witnessStackToScriptWitness(witness: Buffer[]): Buffer {
|
|
||||||
let buffer = Buffer.allocUnsafe(0);
|
|
||||||
|
|
||||||
function writeSlice(slice: Buffer): void {
|
|
||||||
buffer = Buffer.concat([buffer, Buffer.from(slice)]);
|
|
||||||
}
|
|
||||||
|
|
||||||
function writeVarInt(i: number): void {
|
|
||||||
const currentLen = buffer.length;
|
|
||||||
const varintLen = varuint.encodingLength(i);
|
|
||||||
|
|
||||||
buffer = Buffer.concat([buffer, Buffer.allocUnsafe(varintLen)]);
|
|
||||||
varuint.encode(i, buffer, currentLen);
|
|
||||||
}
|
|
||||||
|
|
||||||
function writeVarSlice(slice: Buffer): void {
|
|
||||||
writeVarInt(slice.length);
|
|
||||||
writeSlice(slice);
|
|
||||||
}
|
|
||||||
|
|
||||||
function writeVector(vector: Buffer[]): void {
|
|
||||||
writeVarInt(vector.length);
|
|
||||||
vector.forEach(writeVarSlice);
|
|
||||||
}
|
|
||||||
|
|
||||||
writeVector(witness);
|
|
||||||
|
|
||||||
return buffer;
|
|
||||||
}
|
|
||||||
|
|
||||||
function scriptWitnessToWitnessStack(buffer: Buffer): Buffer[] {
|
|
||||||
let offset = 0;
|
|
||||||
|
|
||||||
function readSlice(n: number): Buffer {
|
|
||||||
offset += n;
|
|
||||||
return buffer.slice(offset - n, offset);
|
|
||||||
}
|
|
||||||
|
|
||||||
function readVarInt(): number {
|
|
||||||
const vi = varuint.decode(buffer, offset);
|
|
||||||
offset += varuint.decode.bytes;
|
|
||||||
return vi;
|
|
||||||
}
|
|
||||||
|
|
||||||
function readVarSlice(): Buffer {
|
|
||||||
return readSlice(readVarInt());
|
|
||||||
}
|
|
||||||
|
|
||||||
function readVector(): Buffer[] {
|
|
||||||
const count = readVarInt();
|
|
||||||
const vector: Buffer[] = [];
|
|
||||||
for (let i = 0; i < count; i++) vector.push(readVarSlice());
|
|
||||||
return vector;
|
|
||||||
}
|
|
||||||
|
|
||||||
return readVector();
|
|
||||||
}
|
|
||||||
|
|
||||||
function range(n: number): number[] {
|
|
||||||
return [...Array(n).keys()];
|
|
||||||
}
|
|
||||||
|
|
||||||
function checkTxEmpty(tx: Transaction): void {
|
|
||||||
const isEmpty = tx.ins.every(
|
|
||||||
input =>
|
|
||||||
input.script &&
|
|
||||||
input.script.length === 0 &&
|
|
||||||
input.witness &&
|
|
||||||
input.witness.length === 0,
|
|
||||||
);
|
|
||||||
if (!isEmpty) {
|
|
||||||
throw new Error('Format Error: Transaction ScriptSigs are not empty');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function checkInputsForPartialSig(inputs: PsbtInput[], action: string): void {
|
|
||||||
inputs.forEach(input => {
|
|
||||||
let throws = false;
|
|
||||||
if ((input.partialSig || []).length === 0) return;
|
|
||||||
input.partialSig!.forEach(pSig => {
|
|
||||||
const { hashType } = bscript.signature.decode(pSig.signature);
|
|
||||||
const whitelist: string[] = [];
|
|
||||||
const isAnyoneCanPay = hashType & Transaction.SIGHASH_ANYONECANPAY;
|
|
||||||
if (isAnyoneCanPay) whitelist.push('addInput');
|
|
||||||
const hashMod = hashType & 0x1f;
|
|
||||||
switch (hashMod) {
|
|
||||||
case Transaction.SIGHASH_ALL:
|
|
||||||
break;
|
|
||||||
case Transaction.SIGHASH_SINGLE:
|
|
||||||
case Transaction.SIGHASH_NONE:
|
|
||||||
whitelist.push('addOutput');
|
|
||||||
whitelist.push('setSequence');
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
if (whitelist.indexOf(action) === -1) {
|
|
||||||
throws = true;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
if (throws) {
|
|
||||||
throw new Error('Can not modify transaction, signatures exist.');
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
function nonWitnessUtxoTxFromCache(
|
|
||||||
cache: PsbtCache,
|
|
||||||
input: PsbtInput,
|
|
||||||
inputIndex: number,
|
|
||||||
): Transaction {
|
|
||||||
if (!cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex]) {
|
|
||||||
addNonWitnessTxCache(cache, input, inputIndex);
|
|
||||||
}
|
|
||||||
return cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex];
|
|
||||||
}
|
|
||||||
|
|
||||||
function getInputAdder(
|
function getInputAdder(
|
||||||
cache: PsbtCache,
|
cache: PsbtCache,
|
||||||
): (_inputData: TransactionInput, txBuf: Buffer) => Buffer {
|
): (_inputData: TransactionInput, txBuf: Buffer) => Buffer {
|
||||||
|
|
@ -1040,19 +831,199 @@ function getOutputAdder(
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
function checkFees(psbt: Psbt, cache: PsbtCache, opts: PsbtOpts): void {
|
function getPayment(
|
||||||
const feeRate = cache.__FEE_RATE || psbt.getFeeRate();
|
script: Buffer,
|
||||||
const vsize = cache.__EXTRACTED_TX!.virtualSize();
|
scriptType: string,
|
||||||
const satoshis = feeRate * vsize;
|
partialSig: PartialSig[],
|
||||||
if (feeRate >= opts.maximumFeeRate) {
|
): payments.Payment {
|
||||||
throw new Error(
|
let payment: payments.Payment;
|
||||||
`Warning: You are paying around ${(satoshis / 1e8).toFixed(8)} in ` +
|
switch (scriptType) {
|
||||||
`fees, which is ${feeRate} satoshi per byte for a transaction ` +
|
case 'multisig':
|
||||||
`with a VSize of ${vsize} bytes (segwit counted as 0.25 byte per ` +
|
const sigs = getSortedSigs(script, partialSig);
|
||||||
`byte). Use setMaximumFeeRate method to raise your threshold, or ` +
|
payment = payments.p2ms({
|
||||||
`pass true to the first arg of extractTransaction.`,
|
output: script,
|
||||||
);
|
signatures: sigs,
|
||||||
|
});
|
||||||
|
break;
|
||||||
|
case 'pubkey':
|
||||||
|
payment = payments.p2pk({
|
||||||
|
output: script,
|
||||||
|
signature: partialSig[0].signature,
|
||||||
|
});
|
||||||
|
break;
|
||||||
|
case 'pubkeyhash':
|
||||||
|
payment = payments.p2pkh({
|
||||||
|
output: script,
|
||||||
|
pubkey: partialSig[0].pubkey,
|
||||||
|
signature: partialSig[0].signature,
|
||||||
|
});
|
||||||
|
break;
|
||||||
|
case 'witnesspubkeyhash':
|
||||||
|
payment = payments.p2wpkh({
|
||||||
|
output: script,
|
||||||
|
pubkey: partialSig[0].pubkey,
|
||||||
|
signature: partialSig[0].signature,
|
||||||
|
});
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
|
return payment!;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface GetScriptReturn {
|
||||||
|
script: Buffer | null;
|
||||||
|
isSegwit: boolean;
|
||||||
|
isP2SH: boolean;
|
||||||
|
isP2WSH: boolean;
|
||||||
|
}
|
||||||
|
function getScriptFromInput(
|
||||||
|
inputIndex: number,
|
||||||
|
input: PsbtInput,
|
||||||
|
cache: PsbtCache,
|
||||||
|
): GetScriptReturn {
|
||||||
|
const unsignedTx = cache.__TX;
|
||||||
|
const res: GetScriptReturn = {
|
||||||
|
script: null,
|
||||||
|
isSegwit: false,
|
||||||
|
isP2SH: false,
|
||||||
|
isP2WSH: false,
|
||||||
|
};
|
||||||
|
if (input.nonWitnessUtxo) {
|
||||||
|
if (input.redeemScript) {
|
||||||
|
res.isP2SH = true;
|
||||||
|
res.script = input.redeemScript;
|
||||||
|
} else {
|
||||||
|
const nonWitnessUtxoTx = nonWitnessUtxoTxFromCache(
|
||||||
|
cache,
|
||||||
|
input,
|
||||||
|
inputIndex,
|
||||||
|
);
|
||||||
|
const prevoutIndex = unsignedTx.ins[inputIndex].index;
|
||||||
|
res.script = nonWitnessUtxoTx.outs[prevoutIndex].script;
|
||||||
|
}
|
||||||
|
} else if (input.witnessUtxo) {
|
||||||
|
res.isSegwit = true;
|
||||||
|
res.isP2SH = !!input.redeemScript;
|
||||||
|
res.isP2WSH = !!input.witnessScript;
|
||||||
|
if (input.witnessScript) {
|
||||||
|
res.script = input.witnessScript;
|
||||||
|
} else if (input.redeemScript) {
|
||||||
|
res.script = payments.p2wpkh({
|
||||||
|
hash: input.redeemScript.slice(2),
|
||||||
|
}).output!;
|
||||||
|
} else {
|
||||||
|
res.script = payments.p2wpkh({
|
||||||
|
hash: input.witnessUtxo.script.slice(2),
|
||||||
|
}).output!;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
function getSortedSigs(script: Buffer, partialSig: PartialSig[]): Buffer[] {
|
||||||
|
const p2ms = payments.p2ms({ output: script });
|
||||||
|
// for each pubkey in order of p2ms script
|
||||||
|
return p2ms
|
||||||
|
.pubkeys!.map(pk => {
|
||||||
|
// filter partialSig array by pubkey being equal
|
||||||
|
return (
|
||||||
|
partialSig.filter(ps => {
|
||||||
|
return ps.pubkey.equals(pk);
|
||||||
|
})[0] || {}
|
||||||
|
).signature;
|
||||||
|
// Any pubkey without a match will return undefined
|
||||||
|
// this last filter removes all the undefined items in the array.
|
||||||
|
})
|
||||||
|
.filter(v => !!v);
|
||||||
|
}
|
||||||
|
|
||||||
|
function scriptWitnessToWitnessStack(buffer: Buffer): Buffer[] {
|
||||||
|
let offset = 0;
|
||||||
|
|
||||||
|
function readSlice(n: number): Buffer {
|
||||||
|
offset += n;
|
||||||
|
return buffer.slice(offset - n, offset);
|
||||||
|
}
|
||||||
|
|
||||||
|
function readVarInt(): number {
|
||||||
|
const vi = varuint.decode(buffer, offset);
|
||||||
|
offset += varuint.decode.bytes;
|
||||||
|
return vi;
|
||||||
|
}
|
||||||
|
|
||||||
|
function readVarSlice(): Buffer {
|
||||||
|
return readSlice(readVarInt());
|
||||||
|
}
|
||||||
|
|
||||||
|
function readVector(): Buffer[] {
|
||||||
|
const count = readVarInt();
|
||||||
|
const vector: Buffer[] = [];
|
||||||
|
for (let i = 0; i < count; i++) vector.push(readVarSlice());
|
||||||
|
return vector;
|
||||||
|
}
|
||||||
|
|
||||||
|
return readVector();
|
||||||
|
}
|
||||||
|
|
||||||
|
function witnessStackToScriptWitness(witness: Buffer[]): Buffer {
|
||||||
|
let buffer = Buffer.allocUnsafe(0);
|
||||||
|
|
||||||
|
function writeSlice(slice: Buffer): void {
|
||||||
|
buffer = Buffer.concat([buffer, Buffer.from(slice)]);
|
||||||
|
}
|
||||||
|
|
||||||
|
function writeVarInt(i: number): void {
|
||||||
|
const currentLen = buffer.length;
|
||||||
|
const varintLen = varuint.encodingLength(i);
|
||||||
|
|
||||||
|
buffer = Buffer.concat([buffer, Buffer.allocUnsafe(varintLen)]);
|
||||||
|
varuint.encode(i, buffer, currentLen);
|
||||||
|
}
|
||||||
|
|
||||||
|
function writeVarSlice(slice: Buffer): void {
|
||||||
|
writeVarInt(slice.length);
|
||||||
|
writeSlice(slice);
|
||||||
|
}
|
||||||
|
|
||||||
|
function writeVector(vector: Buffer[]): void {
|
||||||
|
writeVarInt(vector.length);
|
||||||
|
vector.forEach(writeVarSlice);
|
||||||
|
}
|
||||||
|
|
||||||
|
writeVector(witness);
|
||||||
|
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
function addNonWitnessTxCache(
|
||||||
|
cache: PsbtCache,
|
||||||
|
input: PsbtInput,
|
||||||
|
inputIndex: number,
|
||||||
|
): void {
|
||||||
|
cache.__NON_WITNESS_UTXO_BUF_CACHE[inputIndex] = input.nonWitnessUtxo!;
|
||||||
|
|
||||||
|
const tx = Transaction.fromBuffer(input.nonWitnessUtxo!);
|
||||||
|
cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex] = tx;
|
||||||
|
|
||||||
|
const self = cache;
|
||||||
|
const selfIndex = inputIndex;
|
||||||
|
delete input.nonWitnessUtxo;
|
||||||
|
Object.defineProperty(input, 'nonWitnessUtxo', {
|
||||||
|
enumerable: true,
|
||||||
|
get(): Buffer {
|
||||||
|
const buf = self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex];
|
||||||
|
const txCache = self.__NON_WITNESS_UTXO_TX_CACHE[selfIndex];
|
||||||
|
if (buf !== undefined) {
|
||||||
|
return buf;
|
||||||
|
} else {
|
||||||
|
const newBuf = txCache.toBuffer();
|
||||||
|
self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex] = newBuf;
|
||||||
|
return newBuf;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
set(data: Buffer): void {
|
||||||
|
self.__NON_WITNESS_UTXO_BUF_CACHE[selfIndex] = data;
|
||||||
|
},
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
function inputFinalizeGetAmts(
|
function inputFinalizeGetAmts(
|
||||||
|
|
@ -1083,13 +1054,25 @@ function inputFinalizeGetAmts(
|
||||||
return inputAmount;
|
return inputAmount;
|
||||||
}
|
}
|
||||||
|
|
||||||
function check32Bit(num: number): void {
|
function nonWitnessUtxoTxFromCache(
|
||||||
if (
|
cache: PsbtCache,
|
||||||
typeof num !== 'number' ||
|
input: PsbtInput,
|
||||||
num !== Math.floor(num) ||
|
inputIndex: number,
|
||||||
num > 0xffffffff ||
|
): Transaction {
|
||||||
num < 0
|
if (!cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex]) {
|
||||||
) {
|
addNonWitnessTxCache(cache, input, inputIndex);
|
||||||
throw new Error('Invalid 32 bit integer');
|
|
||||||
}
|
}
|
||||||
|
return cache.__NON_WITNESS_UTXO_TX_CACHE[inputIndex];
|
||||||
|
}
|
||||||
|
|
||||||
|
function classifyScript(script: Buffer): string {
|
||||||
|
if (isP2WPKH(script)) return 'witnesspubkeyhash';
|
||||||
|
if (isP2PKH(script)) return 'pubkeyhash';
|
||||||
|
if (isP2MS(script)) return 'multisig';
|
||||||
|
if (isP2PK(script)) return 'pubkey';
|
||||||
|
return 'nonstandard';
|
||||||
|
}
|
||||||
|
|
||||||
|
function range(n: number): number[] {
|
||||||
|
return [...Array(n).keys()];
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue