Skip to content

Commit d240dea

Browse files
committed
descriptors: Move FlatSigningProvider pubkey filling to GetPubKey
Instead of MakeScripts inconsistently filling the output FlatSigningProvider with the pubkeys involved, just do it in GetPubKey.
1 parent bda628a commit d240dea

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/script/descriptor.cpp

+3-4
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,7 @@ class OriginPubkeyProvider final : public PubkeyProvider
242242
{
243243
std::optional<CPubKey> pub = m_provider->GetPubKey(pos, arg, out, read_cache, write_cache);
244244
if (!pub) return std::nullopt;
245+
Assert(out.pubkeys.contains(pub->GetID()));
245246
auto& [pubkey, suborigin] = out.origins[pub->GetID()];
246247
Assert(pubkey == *pub); // All subproviders must be inserting a valid origin already
247248
std::copy(std::begin(m_origin.fingerprint), std::end(m_origin.fingerprint), suborigin.fingerprint);
@@ -305,6 +306,7 @@ class ConstPubkeyProvider final : public PubkeyProvider
305306
CKeyID keyid = m_pubkey.GetID();
306307
std::copy(keyid.begin(), keyid.begin() + sizeof(info.fingerprint), info.fingerprint);
307308
out.origins.emplace(keyid, std::make_pair(m_pubkey, info));
309+
out.pubkeys.emplace(keyid, m_pubkey);
308310
return m_pubkey;
309311
}
310312
bool IsRange() const override { return false; }
@@ -449,6 +451,7 @@ class BIP32PubkeyProvider final : public PubkeyProvider
449451
if (!der) return std::nullopt;
450452

451453
out.origins.emplace(final_extkey.pubkey.GetID(), std::make_pair(final_extkey.pubkey, info));
454+
out.pubkeys.emplace(final_extkey.pubkey.GetID(), final_extkey.pubkey);
452455

453456
if (write_cache) {
454457
// Only cache parent if there is any unhardened derivation
@@ -887,7 +890,6 @@ class PKHDescriptor final : public DescriptorImpl
887890
std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript>, FlatSigningProvider& out) const override
888891
{
889892
CKeyID id = keys[0].GetID();
890-
out.pubkeys.emplace(id, keys[0]);
891893
return Vector(GetScriptForDestination(PKHash(id)));
892894
}
893895
public:
@@ -922,7 +924,6 @@ class WPKHDescriptor final : public DescriptorImpl
922924
std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript>, FlatSigningProvider& out) const override
923925
{
924926
CKeyID id = keys[0].GetID();
925-
out.pubkeys.emplace(id, keys[0]);
926927
return Vector(GetScriptForDestination(WitnessV0KeyHash(id)));
927928
}
928929
public:
@@ -958,7 +959,6 @@ class ComboDescriptor final : public DescriptorImpl
958959
{
959960
std::vector<CScript> ret;
960961
CKeyID id = keys[0].GetID();
961-
out.pubkeys.emplace(id, keys[0]);
962962
ret.emplace_back(GetScriptForRawPubKey(keys[0])); // P2PK
963963
ret.emplace_back(GetScriptForDestination(PKHash(id))); // P2PKH
964964
if (keys[0].IsCompressed()) {
@@ -1185,7 +1185,6 @@ class TRDescriptor final : public DescriptorImpl
11851185
builder.Finalize(xpk);
11861186
WitnessV1Taproot output = builder.GetOutput();
11871187
out.tr_trees[output] = builder;
1188-
out.pubkeys.emplace(keys[0].GetID(), keys[0]);
11891188
return Vector(GetScriptForDestination(output));
11901189
}
11911190
bool ToStringSubScriptHelper(const SigningProvider* arg, std::string& ret, const StringType type, const DescriptorCache* cache = nullptr) const override

0 commit comments

Comments
 (0)