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

support finding signing nodes from provided psbt input derivation paths

Browse source
This commit is contained in:
Craig Raw 2024-04-30 11:54:43 +02:00
parent 143d28166a
commit d4bdd9f385
2 changed files with 31 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/main/java/com/sparrowwallet/drongo/wallet/Keystore.java
View file

@ -242,6 +242,24 @@ public class Keystore extends Persistable {
return getKeyDerivation().extend(keyPurpose.getPathIndex()).extend(new ChildNumber(keyIndex));
}
public ECKey getPubKeyForDerivation(KeyDerivation keyDerivation) {
if(keyDerivation != null) {
List<ChildNumber> derivation = keyDerivation.getDerivation();
if(derivation.size() > this.keyDerivation.getDerivation().size()) {
List<ChildNumber> xpubDerivation = derivation.subList(0, this.keyDerivation.getDerivation().size());
if(xpubDerivation.equals(this.keyDerivation.getDerivation())) {
derivation = derivation.subList(this.keyDerivation.getDerivation().size(), derivation.size());
}
}
if(derivation.size() == 2) {
return getPubKey(new WalletNode(KeyDerivation.writePath(derivation)));
}
}
return null;
}
public boolean isValid() {
try {
checkKeystore();

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

@ -1490,8 +1490,8 @@ public class Wallet extends Persistable implements Comparable<Wallet> {
return signingNodes;
}
public List<Keystore> getSigningKeystores(PSBT psbt) {
List<Keystore> signingKeystores = new ArrayList<>();
public Collection<Keystore> getSigningKeystores(PSBT psbt) {
Set<Keystore> signingKeystores = new LinkedHashSet<>();
for(Map.Entry<ExtendedKey, KeyDerivation> entry : psbt.getExtendedPublicKeys().entrySet()) {
for(Keystore keystore : getKeystores()) {
@ -1501,6 +1501,17 @@ public class Wallet extends Persistable implements Comparable<Wallet> {
}
}
for(PSBTInput psbtInput : psbt.getPsbtInputs()) {
for(Map.Entry<ECKey, KeyDerivation> entry : psbtInput.getDerivedPublicKeys().entrySet()) {
for(Keystore keystore : getKeystores().stream().filter(k -> !signingKeystores.contains(k)).toList()) {
ECKey derivedKey = keystore.getPubKeyForDerivation(entry.getValue());
if(derivedKey != null && Arrays.equals(entry.getKey().getPubKey(), derivedKey.getPubKey())) {
signingKeystores.add(keystore);
}
}
}
}
return signingKeystores;
}