[Wallet] CStakerStatus: save a pointer to the tip instead of the hash

random-zebra · Fuzzbawls · commit f9f437a02720 · 2020-01-11T15:43:40.000-08:00
Github-Pull: #1245 Rebased-From: 968d861
diff --git a/src/wallet/wallet.cpp b/src/wallet/wallet.cpp
@@ -2701,7 +2701,7 @@ bool CWallet::CreateCoinStake(
     int nAttempts = 0;
 
     // update staker status (hash)
-    pStakerStatus->SetLastHash(pindexPrev->GetBlockHash());
+    pStakerStatus->SetLastTip(pindexPrev);
 
     for (std::unique_ptr<CStakeInput>& stakeInput : listInputs) {
         //new block came in, move on
diff --git a/src/wallet/wallet.h b/src/wallet/wallet.h
@@ -147,19 +147,23 @@ class CKeyPool
     }
 };
 
-/** Record metadata about last kernel stake operation (time and prev hash)**/
+/** Record info about last kernel stake operation (time and chainTip)**/
 class CStakerStatus {
 private:
-    uint256 hashLastStakeAttempt;
+    const CBlockIndex* tipLastStakeAttempt;
     int64_t timeLastStakeAttempt;
 public:
-    uint256 GetLastHash() const { return hashLastStakeAttempt; }
+    const CBlockIndex* GetLastTip() const { return tipLastStakeAttempt; }
+    uint256 GetLastHash() const
+    {
+        return (tipLastStakeAttempt == nullptr ? 0 : tipLastStakeAttempt->GetBlockHash());
+    }
     int64_t GetLastTime() const { return timeLastStakeAttempt; }
-    void SetLastHash(const uint256& lastHash) { hashLastStakeAttempt = lastHash; }
+    void SetLastTip(const CBlockIndex* lastTip) { tipLastStakeAttempt = lastTip; }
     void SetLastTime(const uint64_t lastTime) { timeLastStakeAttempt = lastTime; }
-    void Update(const uint256& lastHash, const uint64_t lastTime)
+    void Update(CBlockIndex* lastTip, const uint64_t lastTime)
     {
-        SetLastHash(lastHash);
+        SetLastTip(lastTip);
         SetLastTime(lastTime);
     }
     bool IsActive() { return (timeLastStakeAttempt + 30) >= GetTime(); }
