sparrowwallet/drongo
1
0
Fork 0
mirror of https://github.com/sparrowwallet/drongo.git synced 2024-11-02 18:26:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

support partially finalized psbts

Browse source
This commit is contained in:
Craig Raw 2020-11-03 13:52:55 +02:00
parent 8541348b5b
commit 9c9836147a
3 changed files with 43 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
src/main/java/com/sparrowwallet/drongo/psbt/PSBT.java
View file

@ -253,7 +253,9 @@ public class PSBT {
}
}
log.debug("Calculated fee at " + getFee());
if(log.isDebugEnabled()) {
log.debug("Calculated fee at " + getFee());
}
}
private void parseGlobalEntries(List<PSBTEntry> globalEntries) throws PSBTParseException {
@ -395,12 +397,12 @@ public class PSBT {
public boolean isFinalized() {
for(PSBTInput psbtInput : getPsbtInputs()) {
if(psbtInput.getFinalScriptSig() != null || psbtInput.getFinalScriptWitness() != null) {
return true;
if(!psbtInput.isFinalized()) {
return false;
}
}
return false;
return true;
}
private List<PSBTEntry> getGlobalEntries() {
@ -499,9 +501,6 @@ public class PSBT {
public Transaction extractTransaction() {
boolean hasWitness = false;
for(PSBTInput psbtInput : getPsbtInputs()) {
if(psbtInput.getFinalScriptSig() == null) {
return null;
}
if(psbtInput.getFinalScriptWitness() != null) {
hasWitness = true;
}
@ -516,7 +515,7 @@ public class PSBT {
for(int i = 0; i < finalTransaction.getInputs().size(); i++) {
TransactionInput txInput = finalTransaction.getInputs().get(i);
PSBTInput psbtInput = getPsbtInputs().get(i);
txInput.setScriptBytes(psbtInput.getFinalScriptSig().getProgram());
txInput.setScriptBytes(psbtInput.getFinalScriptSig() == null ? new byte[0] : psbtInput.getFinalScriptSig().getProgram());
if(hasWitness) {
if(psbtInput.getFinalScriptWitness() != null) {
@ -530,6 +529,27 @@ public class PSBT {
return finalTransaction;
}
public PSBT getPublicCopy() {
try {
PSBT publicCopy = new PSBT(serialize());
publicCopy.extendedPublicKeys.clear();
for(PSBTInput psbtInput : publicCopy.getPsbtInputs()) {
psbtInput.clearPrivateFields();
}
for(PSBTOutput psbtOutput : publicCopy.getPsbtOutputs()) {
psbtOutput.getDerivedPublicKeys().clear();
psbtOutput.getProprietary().clear();
}
if(publicCopy.isFinalized()) {
publicCopy.transaction = publicCopy.extractTransaction();
}
return publicCopy;
} catch(PSBTParseException e) {
throw new IllegalStateException("Could not parse PSBT", e);
}
}
public List<PSBTInput> getPsbtInputs() {
return psbtInputs;
}

12
src/main/java/com/sparrowwallet/drongo/psbt/PSBTInput.java
View file

@ -386,7 +386,7 @@ public class PSBTInput {
return false;
}
} else {
return getFinalScriptSig() != null;
return isFinalized();
}
}
@ -525,12 +525,20 @@ public class PSBTInput {
return signingScript;
}
public boolean isFinalized() {
return getFinalScriptSig() != null || getFinalScriptWitness() != null;
}
public TransactionInput getInput() {
return transaction.getInputs().get(index);
}
public TransactionOutput getUtxo() {
int vout = (int)transaction.getInputs().get(index).getOutpoint().getIndex();
return getWitnessUtxo() != null ? getWitnessUtxo() : (getNonWitnessUtxo() != null ? getNonWitnessUtxo().getOutputs().get(vout) : null);
}
public void clearFinalised() {
public void clearPrivateFields() {
partialSignatures.clear();
sigHash = null;
redeemScript = null;

6
src/main/java/com/sparrowwallet/drongo/wallet/Wallet.java
View file

@ -713,6 +713,10 @@ public class Wallet {
TransactionInput txInput = psbt.getTransaction().getInputs().get(i);
PSBTInput psbtInput = psbt.getPsbtInputs().get(i);
if(psbtInput.isFinalized()) {
continue;
}
WalletNode signingNode = signingNodes.get(psbtInput);
//Transaction parent on PSBT utxo might be null in a witness tx, so get utxo tx hash and utxo index from PSBT tx input
@ -760,7 +764,7 @@ public class Wallet {
psbtInput.setFinalScriptSig(finalizedTxInput.getScriptSig());
psbtInput.setFinalScriptWitness(finalizedTxInput.getWitness());
psbtInput.clearFinalised();
psbtInput.clearPrivateFields();
}
}
}