Receive addresses recovery fixed (#1031)

firoorg · Jun 25, 2021 · d5aa685 · d5aa685
1 parent 892c64e
commit d5aa685
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/wallet/wallet.cpp b/src/wallet/wallet.cpp
@@ -6653,6 +6653,26 @@ CWallet* CWallet::CreateWalletFromFile(const std::string walletFile)
             }
         }
     }
+
+    // recover addressbook
+    if (fFirstRun)
+    {
+        for (map<uint256, CWalletTx>::iterator it = walletInstance->mapWallet.begin(); it != walletInstance->mapWallet.end(); ++it) {
+            for (uint32_t i = 0; i < (*it).second.tx->vout.size(); i++) {
+                const auto& txout = (*it).second.tx->vout[i];
+                if(txout.scriptPubKey.IsMint() || (*it).second.changes.count(i))
+                    continue;
+                if (!walletInstance->IsMine(txout))
+                    continue;
+                CTxDestination addr;
+                if(!ExtractDestination(txout.scriptPubKey, addr))
+                    continue;
+                if (walletInstance->mapAddressBook.count(addr) == 0)
+                    walletInstance->SetAddressBook(addr, "", "receive");
+            }
+        }
+    }
+
     walletInstance->SetBroadcastTransactions(GetBoolArg("-walletbroadcast", DEFAULT_WALLETBROADCAST));
 
     {