[Wallet] IsSaplingSpent back port.

PIVX-Project · Oct 14, 2020 · 6156b15 · 6156b15
1 parent 46f4f92
commit 6156b15
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/sapling/saplingscriptpubkeyman.cpp b/src/sapling/saplingscriptpubkeyman.cpp
@@ -17,6 +17,21 @@ void SaplingScriptPubKeyMan::AddToSaplingSpends(const uint256& nullifier, const
     wallet->SyncMetaDataN(range);
 }
 
+bool SaplingScriptPubKeyMan::IsSaplingSpent(const uint256& nullifier) const {
+    LOCK(cs_main);
+    std::pair<TxNullifiers::const_iterator, TxNullifiers::const_iterator> range;
+    range = mapTxSaplingNullifiers.equal_range(nullifier);
+
+    for (TxNullifiers::const_iterator it = range.first; it != range.second; ++it) {
+        const uint256& wtxid = it->second;
+        std::map<uint256, CWalletTx>::const_iterator mit = wallet->mapWallet.find(wtxid);
+        if (mit != wallet->mapWallet.end() && mit->second.GetDepthInMainChain() >= 0) {
+            return true; // Spent
+        }
+    }
+    return false;
+}
+
 void SaplingScriptPubKeyMan::UpdateSaplingNullifierNoteMapForBlock(const CBlock *pblock) {
     LOCK(wallet->cs_wallet);
 

diff --git a/src/sapling/saplingscriptpubkeyman.h b/src/sapling/saplingscriptpubkeyman.h
@@ -103,6 +103,7 @@ class SaplingScriptPubKeyMan {
      * Keep track of the used nullifier.
      */
     void AddToSaplingSpends(const uint256& nullifier, const uint256& wtxid);
+    bool IsSaplingSpent(const uint256& nullifier) const;
 
     /**
      * pindex is the new tip being connected.