Browse Source

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

0.8 v0.5.0rc6
Gavin Andresen 13 years ago
parent
commit
1c15f88653
  1. 13
      src/db.cpp
  2. 3
      src/db.h
  3. 4
      src/init.cpp
  4. 5
      src/wallet.cpp

13
src/db.cpp

@ -28,7 +28,12 @@ DbEnv dbenv(0); @@ -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: @@ -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) @@ -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) @@ -322,7 +327,7 @@ void DBFlush(bool fShutdown, bool fRemoveLogFiles)
if (mapFileUseCount.empty())
{
dbenv.log_archive(&listp, DB_ARCH_REMOVE);
EnvShutdown(fRemoveLogFiles);
EnvShutdown();
}
}
}

3
src/db.h

@ -29,7 +29,8 @@ extern unsigned int nWalletDBUpdated; @@ -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);

4
src/init.cpp

@ -55,9 +55,9 @@ void Shutdown(void* parg) @@ -55,9 +55,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;

5
src/wallet.cpp

@ -193,8 +193,8 @@ bool CWallet::EncryptWallet(const string& strWalletPassphrase) @@ -193,8 +193,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;
@ -1154,6 +1154,7 @@ int CWallet::LoadWallet(bool& fFirstRunRet) @@ -1154,6 +1154,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

Loading…
Cancel
Save