Check signatures for finalized inputs too

Author: junderw

src/psbt.js

@@ -626,8 +626,27 @@ function checkFees(psbt, cache, opts) {
 function checkInputsForPartialSig(inputs, action) {
   inputs.forEach(input => {
     let throws = false;
-    if ((input.partialSig || []).length === 0) return;
-    input.partialSig.forEach(pSig => {
+    let pSigs = [];
+    if ((input.partialSig || []).length === 0) {
+      if (!input.finalScriptSig && !input.finalScriptWitness) return;
+      const scriptItems = !input.finalScriptSig
+        ? []
+        : bscript.decompile(input.finalScriptSig) || [];
+      const witnessItems = !input.finalScriptWitness
+        ? []
+        : bscript.decompile(input.finalScriptWitness) || [];
+      pSigs = scriptItems
+        .concat(witnessItems)
+        .filter(item => {
+          return (
+            Buffer.isBuffer(item) && bscript.isCanonicalScriptSignature(item)
+          );
+        })
+        .map(sig => ({ signature: sig }));
+    } else {
+      pSigs = input.partialSig;
+    }
+    pSigs.forEach(pSig => {
       const { hashType } = bscript.signature.decode(pSig.signature);
       const whitelist = [];
       const isAnyoneCanPay =

test/psbt.js

@@ -625,6 +625,9 @@ describe(`Psbt`, () => {
     }, new RegExp('Can not modify transaction, signatures exist.'))
     psbt.validateSignaturesOfInput(0)
     psbt.finalizeAllInputs()
+    assert.throws(() => {
+      psbt.setVersion(3)
+    }, new RegExp('Can not modify transaction, signatures exist.'))
     assert.strictEqual(
       psbt.extractTransaction().toHex(),
       '02000000013ebc8203037dda39d482bf41ff3be955996c50d9d4f7cfc3d2097a694a7' +

ts_src/psbt.ts

@@ -795,8 +795,27 @@ function checkFees(psbt: Psbt, cache: PsbtCache, opts: PsbtOpts): void {
 function checkInputsForPartialSig(inputs: PsbtInput[], action: string): void {
   inputs.forEach(input => {
     let throws = false;
-    if ((input.partialSig || []).length === 0) return;
-    input.partialSig!.forEach(pSig => {
+    let pSigs: PartialSig[] = [];
+    if ((input.partialSig || []).length === 0) {
+      if (!input.finalScriptSig && !input.finalScriptWitness) return;
+      const scriptItems = !input.finalScriptSig
+        ? []
+        : bscript.decompile(input.finalScriptSig) || [];
+      const witnessItems = !input.finalScriptWitness
+        ? []
+        : bscript.decompile(input.finalScriptWitness) || [];
+      pSigs = scriptItems
+        .concat(witnessItems)
+        .filter(item => {
+          return (
+            Buffer.isBuffer(item) && bscript.isCanonicalScriptSignature(item)
+          );
+        })
+        .map(sig => ({ signature: sig })) as PartialSig[];
+    } else {
+      pSigs = input.partialSig!;
+    }
+    pSigs.forEach(pSig => {
       const { hashType } = bscript.signature.decode(pSig.signature);
       const whitelist: string[] = [];
       const isAnyoneCanPay = hashType & Transaction.SIGHASH_ANYONECANPAY;