From: Gavin Andresen <gavinandresen@gmail.com>
Date: Thu, 17 Nov 2011 19:21:32 +0000 (-0500)
Subject: Only remove database log files on shutdown after wallet encryption/rewrite
X-Git-Tag: v0.4.0-unstable~129^2~1^2^2~2^2^2~101
X-Git-Url: https://git.novaco.in/?p=novacoin.git;a=commitdiff_plain;h=c4a3bf9e552f25a5ad0efb8cf62d344a7dad53ed

Only remove database log files on shutdown after wallet encryption/rewrite
---

diff --git a/src/db.cpp b/src/db.cpp
index b48be7a..77000ab 100644
--- a/src/db.cpp
+++ b/src/db.cpp
@@ -28,7 +28,12 @@ DbEnv dbenv(0);
 static map<string, int> mapFileUseCount;
 static map<string, Db*> mapDb;
 
-static void EnvShutdown(bool fRemoveLogFiles)
+static bool fRemoveLogFiles = false;
+void RemoveLogFilesOnShutdown(bool fIn)
+{
+    fRemoveLogFiles = fIn;
+}
+static void EnvShutdown()
 {
     if (!fDbEnvInit)
         return;
@@ -71,7 +76,7 @@ public:
     }
     ~CDBInit()
     {
-        EnvShutdown(false);
+        EnvShutdown();
     }
 }
 instance_of_cdbinit;
@@ -289,7 +294,7 @@ bool CDB::Rewrite(const string& strFile, const char* pszSkip)
 }
 
 
-void DBFlush(bool fShutdown, bool fRemoveLogFiles)
+void DBFlush(bool fShutdown)
 {
     // Flush log data to the actual data file
     //  on all files that are not in use
@@ -322,7 +327,7 @@ void DBFlush(bool fShutdown, bool fRemoveLogFiles)
             if (mapFileUseCount.empty())
             {
                 dbenv.log_archive(&listp, DB_ARCH_REMOVE);
-                EnvShutdown(fRemoveLogFiles);
+                EnvShutdown();
             }
         }
     }
diff --git a/src/db.h b/src/db.h
index 98af4be..425dc2e 100644
--- a/src/db.h
+++ b/src/db.h
@@ -29,7 +29,8 @@ extern unsigned int nWalletDBUpdated;
 extern DbEnv dbenv;
 
 
-extern void DBFlush(bool fShutdown, bool fRemoveLogFiles);
+extern void RemoveLogFilesOnShutdown(bool fRemoveLogFiles);
+extern void DBFlush(bool fShutdown);
 void ThreadFlushWalletDB(void* parg);
 bool BackupWallet(const CWallet& wallet, const std::string& strDest);
 
diff --git a/src/init.cpp b/src/init.cpp
index 4a149b3..2d21a26 100644
--- a/src/init.cpp
+++ b/src/init.cpp
@@ -45,9 +45,9 @@ void Shutdown(void* parg)
     {
         fShutdown = true;
         nTransactionsUpdated++;
-        DBFlush(false, false);
+        DBFlush(false);
         StopNode();
-        DBFlush(true, true);
+        DBFlush(true);
         boost::filesystem::remove(GetPidFile());
         UnregisterWallet(pwalletMain);
         delete pwalletMain;
diff --git a/src/wallet.cpp b/src/wallet.cpp
index 9b9586f..737f0e6 100644
--- a/src/wallet.cpp
+++ b/src/wallet.cpp
@@ -194,8 +194,8 @@ bool CWallet::EncryptWallet(const string& strWalletPassphrase)
 
         // Need to completely rewrite the wallet file; if we don't, bdb might keep
         // bits of the unencrypted private key in slack space in the database file.
-        setKeyPool.clear();
-        CDB::Rewrite(strWalletFile);
+        if (CDB::Rewrite(strWalletFile))
+            RemoveLogFilesOnShutdown(true);
     }
 
     return true;
@@ -1134,6 +1134,7 @@ int CWallet::LoadWallet(bool& fFirstRunRet)
     {
         if (CDB::Rewrite(strWalletFile, "\x04pool"))
         {
+            RemoveLogFilesOnShutdown(true);
             setKeyPool.clear();
             // Note: can't top-up keypool here, because wallet is locked.
             // User will be prompted to unlock wallet the next operation