Add CAffectedKeysVisitor

Tranz5 · May 21, 2014 · 09cea91 · 09cea91
1 parent 58519a5
commit 09cea91
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/src/script.cpp b/src/script.cpp
@@ -1539,6 +1539,43 @@ bool ExtractDestination(const CScript& scriptPubKey, CTxDestination& addressRet)
     return false;
 }
 
+class CAffectedKeysVisitor : public boost::static_visitor<void> {
+  private:
+      const CKeyStore &keystore;
+      std::vector<CKeyID> &vKeys;
+
+  public:
+      CAffectedKeysVisitor(const CKeyStore &keystoreIn, std::vector<CKeyID> &vKeysIn) : keystore(keystoreIn), vKeys(vKeysIn) {}
+
+      void Process(const CScript &script) {
+          txnouttype type;
+          std::vector<CTxDestination> vDest;
+          int nRequired;
+          if (ExtractDestinations(script, type, vDest, nRequired)) {
+              BOOST_FOREACH(const CTxDestination &dest, vDest)
+                  boost::apply_visitor(*this, dest);
+          }
+      }
+
+      void operator()(const CKeyID &keyId) {
+          if (keystore.HaveKey(keyId))
+              vKeys.push_back(keyId);
+      }
+
+      void operator()(const CScriptID &scriptId) {
+          CScript script;
+          if (keystore.GetCScript(scriptId, script))
+              Process(script);
+      }
+
+      void operator()(const CNoDestination &none) {}
+  };
+
+
+  void ExtractAffectedKeys(const CKeyStore &keystore, const CScript& scriptPubKey, std::vector<CKeyID> &vKeys) {
+      CAffectedKeysVisitor(keystore, vKeys).Process(scriptPubKey);
+  }
+
 bool ExtractDestinations(const CScript& scriptPubKey, txnouttype& typeRet, vector<CTxDestination>& addressRet, int& nRequiredRet)
 {
     addressRet.clear();

diff --git a/src/script.h b/src/script.h
@@ -591,6 +591,7 @@ int ScriptSigArgsExpected(txnouttype t, const std::vector<std::vector<unsigned c
 bool IsStandard(const CScript& scriptPubKey);
 bool IsMine(const CKeyStore& keystore, const CScript& scriptPubKey);
 bool IsMine(const CKeyStore& keystore, const CTxDestination &dest);
+void ExtractAffectedKeys(const CKeyStore &keystore, const CScript& scriptPubKey, std::vector<CKeyID> &vKeys);
 bool ExtractDestination(const CScript& scriptPubKey, CTxDestination& addressRet);
 bool ExtractDestinations(const CScript& scriptPubKey, txnouttype& typeRet, std::vector<CTxDestination>& addressRet, int& nRequiredRet);
 bool SignSignature(const CKeyStore& keystore, const CScript& fromPubKey, CTransaction& txTo, unsigned int nIn, int nHashType=SIGHASH_ALL);