2010-08-29 18:58:15 +02:00
|
|
|
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
2018-01-02 18:12:05 +01:00
|
|
|
// Copyright (c) 2009-2017 The Bitcoin Core developers
|
2014-12-13 05:09:33 +01:00
|
|
|
// Distributed under the MIT software license, see the accompanying
|
2012-05-18 16:02:28 +02:00
|
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
2010-08-29 18:58:15 +02:00
|
|
|
|
2017-11-10 01:57:53 +01:00
|
|
|
#include <wallet/db.h>
|
2013-04-13 07:13:08 +02:00
|
|
|
|
2017-11-10 01:57:53 +01:00
|
|
|
#include <addrman.h>
|
|
|
|
#include <hash.h>
|
|
|
|
#include <protocol.h>
|
|
|
|
#include <util.h>
|
|
|
|
#include <utilstrencodings.h>
|
2017-10-08 22:48:07 +02:00
|
|
|
#include <wallet/walletutil.h>
|
2013-04-13 07:13:08 +02:00
|
|
|
|
|
|
|
#include <stdint.h>
|
2010-08-29 18:58:15 +02:00
|
|
|
|
2012-04-15 22:10:54 +02:00
|
|
|
#ifndef WIN32
|
2013-04-13 07:13:08 +02:00
|
|
|
#include <sys/stat.h>
|
2012-04-15 22:10:54 +02:00
|
|
|
#endif
|
|
|
|
|
2017-12-14 23:06:22 +01:00
|
|
|
#include <boost/interprocess/sync/file_lock.hpp>
|
Split up util.cpp/h
Split up util.cpp/h into:
- string utilities (hex, base32, base64): no internal dependencies, no dependency on boost (apart from foreach)
- money utilities (parsesmoney, formatmoney)
- time utilities (gettime*, sleep, format date):
- and the rest (logging, argument parsing, config file parsing)
The latter is basically the environment and OS handling,
and is stripped of all utility functions, so we may want to
rename it to something else than util.cpp/h for clarity (Matt suggested
osinterface).
Breaks dependency of sha256.cpp on all the things pulled in by util.
2014-08-21 16:11:09 +02:00
|
|
|
#include <boost/thread.hpp>
|
2014-09-14 12:43:56 +02:00
|
|
|
|
2017-10-10 21:27:26 +02:00
|
|
|
namespace {
|
|
|
|
//! Make sure database has a unique fileid within the environment. If it
|
|
|
|
//! doesn't, throw an error. BDB caches do not work properly when more than one
|
|
|
|
//! open database has the same fileid (values written to one database may show
|
|
|
|
//! up in reads to other databases).
|
|
|
|
//!
|
|
|
|
//! BerkeleyDB generates unique fileids by default
|
|
|
|
//! (https://docs.oracle.com/cd/E17275_01/html/programmer_reference/program_copy.html),
|
|
|
|
//! so bitcoin should never create different databases with the same fileid, but
|
|
|
|
//! this error can be triggered if users manually copy database files.
|
|
|
|
void CheckUniqueFileid(const CDBEnv& env, const std::string& filename, Db& db)
|
|
|
|
{
|
|
|
|
if (env.IsMock()) return;
|
|
|
|
|
|
|
|
u_int8_t fileid[DB_FILE_ID_LEN];
|
|
|
|
int ret = db.get_mpf()->get_fileid(fileid);
|
|
|
|
if (ret != 0) {
|
|
|
|
throw std::runtime_error(strprintf("CDB: Can't open database %s (get_fileid failed with %d)", filename, ret));
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const auto& item : env.mapDb) {
|
|
|
|
u_int8_t item_fileid[DB_FILE_ID_LEN];
|
|
|
|
if (item.second && item.second->get_mpf()->get_fileid(item_fileid) == 0 &&
|
|
|
|
memcmp(fileid, item_fileid, sizeof(fileid)) == 0) {
|
|
|
|
const char* item_filename = nullptr;
|
|
|
|
item.second->get_dbname(&item_filename, nullptr);
|
|
|
|
throw std::runtime_error(strprintf("CDB: Can't open database %s (duplicates fileid %s from %s)", filename,
|
|
|
|
HexStr(std::begin(item_fileid), std::end(item_fileid)),
|
|
|
|
item_filename ? item_filename : "(unknown database)"));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2017-12-14 23:06:22 +01:00
|
|
|
|
|
|
|
bool LockEnvDirectory(const fs::path& env_path)
|
|
|
|
{
|
|
|
|
// Make sure only a single Bitcoin process is using the wallet directory.
|
2017-12-26 07:18:39 +01:00
|
|
|
fs::path lock_file_path = env_path / ".walletlock";
|
2017-12-14 23:06:22 +01:00
|
|
|
FILE* file = fsbridge::fopen(lock_file_path, "a"); // empty lock file; created if it doesn't exist.
|
|
|
|
if (file) fclose(file);
|
|
|
|
|
|
|
|
try {
|
|
|
|
static boost::interprocess::file_lock lock(lock_file_path.string().c_str());
|
|
|
|
if (!lock.try_lock()) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
} catch (const boost::interprocess::interprocess_exception& e) {
|
|
|
|
return error("Error obtaining lock on wallet directory %s: %s.", env_path.string(), e.what());
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
2017-10-10 21:27:26 +02:00
|
|
|
} // namespace
|
|
|
|
|
2010-08-29 18:58:15 +02:00
|
|
|
//
|
|
|
|
// CDB
|
|
|
|
//
|
|
|
|
|
2012-05-14 03:37:39 +02:00
|
|
|
CDBEnv bitdb;
|
2010-08-29 18:58:15 +02:00
|
|
|
|
2012-05-14 03:37:39 +02:00
|
|
|
void CDBEnv::EnvShutdown()
|
2011-11-11 03:12:46 +01:00
|
|
|
{
|
|
|
|
if (!fDbEnvInit)
|
|
|
|
return;
|
|
|
|
|
|
|
|
fDbEnvInit = false;
|
2015-03-03 16:49:12 +01:00
|
|
|
int ret = dbenv->close(0);
|
2012-10-08 21:18:04 +02:00
|
|
|
if (ret != 0)
|
2015-01-08 11:44:25 +01:00
|
|
|
LogPrintf("CDBEnv::EnvShutdown: Error %d shutting down database environment: %s\n", ret, DbEnv::strerror(ret));
|
2012-05-22 21:51:13 +02:00
|
|
|
if (!fMockDb)
|
2016-08-30 08:00:55 +02:00
|
|
|
DbEnv((u_int32_t)0).remove(strPath.c_str(), 0);
|
2011-11-11 03:12:46 +01:00
|
|
|
}
|
|
|
|
|
2015-03-03 16:49:12 +01:00
|
|
|
void CDBEnv::Reset()
|
2010-08-29 18:58:15 +02:00
|
|
|
{
|
2017-08-09 16:24:12 +02:00
|
|
|
dbenv.reset(new DbEnv(DB_CXX_NO_EXCEPTIONS));
|
2012-11-18 11:58:32 +01:00
|
|
|
fDbEnvInit = false;
|
|
|
|
fMockDb = false;
|
2010-08-29 18:58:15 +02:00
|
|
|
}
|
|
|
|
|
2017-08-09 16:24:12 +02:00
|
|
|
CDBEnv::CDBEnv()
|
2015-03-03 16:49:12 +01:00
|
|
|
{
|
|
|
|
Reset();
|
|
|
|
}
|
|
|
|
|
2012-05-14 03:37:39 +02:00
|
|
|
CDBEnv::~CDBEnv()
|
|
|
|
{
|
|
|
|
EnvShutdown();
|
|
|
|
}
|
|
|
|
|
|
|
|
void CDBEnv::Close()
|
|
|
|
{
|
|
|
|
EnvShutdown();
|
|
|
|
}
|
2010-08-29 18:58:15 +02:00
|
|
|
|
2017-12-14 23:06:22 +01:00
|
|
|
bool CDBEnv::Open(const fs::path& pathIn, bool retry)
|
2012-05-14 03:37:39 +02:00
|
|
|
{
|
|
|
|
if (fDbEnvInit)
|
|
|
|
return true;
|
|
|
|
|
2013-03-09 18:02:57 +01:00
|
|
|
boost::this_thread::interruption_point();
|
2012-05-14 03:37:39 +02:00
|
|
|
|
2017-12-24 00:45:33 +01:00
|
|
|
strPath = pathIn.string();
|
2017-12-14 23:06:22 +01:00
|
|
|
if (!LockEnvDirectory(pathIn)) {
|
2017-12-26 07:18:39 +01:00
|
|
|
LogPrintf("Cannot obtain a lock on wallet directory %s. Another instance of bitcoin may be using it.\n", strPath);
|
2017-12-14 23:06:22 +01:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2017-03-01 17:05:50 +01:00
|
|
|
fs::path pathLogDir = pathIn / "database";
|
2017-02-22 10:10:00 +01:00
|
|
|
TryCreateDirectories(pathLogDir);
|
2017-03-01 17:05:50 +01:00
|
|
|
fs::path pathErrorFile = pathIn / "db.log";
|
2015-01-08 11:44:25 +01:00
|
|
|
LogPrintf("CDBEnv::Open: LogDir=%s ErrorFile=%s\n", pathLogDir.string(), pathErrorFile.string());
|
2012-05-14 03:37:39 +02:00
|
|
|
|
2012-05-22 23:45:00 +02:00
|
|
|
unsigned int nEnvFlags = 0;
|
2017-08-01 21:17:40 +02:00
|
|
|
if (gArgs.GetBoolArg("-privdb", DEFAULT_WALLET_PRIVDB))
|
2012-05-22 23:45:00 +02:00
|
|
|
nEnvFlags |= DB_PRIVATE;
|
|
|
|
|
2015-03-03 16:49:12 +01:00
|
|
|
dbenv->set_lg_dir(pathLogDir.string().c_str());
|
|
|
|
dbenv->set_cachesize(0, 0x100000, 1); // 1 MiB should be enough for just the wallet
|
|
|
|
dbenv->set_lg_bsize(0x10000);
|
|
|
|
dbenv->set_lg_max(1048576);
|
|
|
|
dbenv->set_lk_max_locks(40000);
|
|
|
|
dbenv->set_lk_max_objects(40000);
|
2017-03-01 17:28:39 +01:00
|
|
|
dbenv->set_errfile(fsbridge::fopen(pathErrorFile, "a")); /// debug
|
2015-03-03 16:49:12 +01:00
|
|
|
dbenv->set_flags(DB_AUTO_COMMIT, 1);
|
|
|
|
dbenv->set_flags(DB_TXN_WRITE_NOSYNC, 1);
|
|
|
|
dbenv->log_set_config(DB_LOG_AUTO_REMOVE, 1);
|
2015-06-15 07:46:51 +02:00
|
|
|
int ret = dbenv->open(strPath.c_str(),
|
2014-09-19 19:21:46 +02:00
|
|
|
DB_CREATE |
|
|
|
|
DB_INIT_LOCK |
|
|
|
|
DB_INIT_LOG |
|
|
|
|
DB_INIT_MPOOL |
|
|
|
|
DB_INIT_TXN |
|
|
|
|
DB_THREAD |
|
|
|
|
DB_RECOVER |
|
|
|
|
nEnvFlags,
|
|
|
|
S_IRUSR | S_IWUSR);
|
2017-08-09 10:24:15 +02:00
|
|
|
if (ret != 0) {
|
|
|
|
dbenv->close(0);
|
2017-12-14 23:06:22 +01:00
|
|
|
LogPrintf("CDBEnv::Open: Error %d opening database environment: %s\n", ret, DbEnv::strerror(ret));
|
|
|
|
if (retry) {
|
|
|
|
// try moving the database env out of the way
|
|
|
|
fs::path pathDatabaseBak = pathIn / strprintf("database.%d.bak", GetTime());
|
|
|
|
try {
|
|
|
|
fs::rename(pathLogDir, pathDatabaseBak);
|
|
|
|
LogPrintf("Moved old %s to %s. Retrying.\n", pathLogDir.string(), pathDatabaseBak.string());
|
|
|
|
} catch (const fs::filesystem_error&) {
|
|
|
|
// failure is ok (well, not really, but it's not worse than what we started with)
|
|
|
|
}
|
|
|
|
// try opening it again one more time
|
|
|
|
if (!Open(pathIn, false)) {
|
|
|
|
// if it still fails, it probably means we can't even create the database env
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
return false;
|
|
|
|
}
|
2017-08-09 10:24:15 +02:00
|
|
|
}
|
2012-05-14 03:37:39 +02:00
|
|
|
|
|
|
|
fDbEnvInit = true;
|
2012-05-22 21:51:13 +02:00
|
|
|
fMockDb = false;
|
2012-05-14 03:37:39 +02:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2012-05-22 21:51:13 +02:00
|
|
|
void CDBEnv::MakeMock()
|
|
|
|
{
|
|
|
|
if (fDbEnvInit)
|
2017-01-27 02:33:45 +01:00
|
|
|
throw std::runtime_error("CDBEnv::MakeMock: Already initialized");
|
2012-05-22 21:51:13 +02:00
|
|
|
|
2013-03-09 18:02:57 +01:00
|
|
|
boost::this_thread::interruption_point();
|
2012-05-22 21:51:13 +02:00
|
|
|
|
2016-12-25 21:19:40 +01:00
|
|
|
LogPrint(BCLog::DB, "CDBEnv::MakeMock\n");
|
2012-05-22 21:51:13 +02:00
|
|
|
|
2015-03-03 16:49:12 +01:00
|
|
|
dbenv->set_cachesize(1, 0, 1);
|
|
|
|
dbenv->set_lg_bsize(10485760 * 4);
|
|
|
|
dbenv->set_lg_max(10485760);
|
|
|
|
dbenv->set_lk_max_locks(10000);
|
|
|
|
dbenv->set_lk_max_objects(10000);
|
|
|
|
dbenv->set_flags(DB_AUTO_COMMIT, 1);
|
|
|
|
dbenv->log_set_config(DB_LOG_IN_MEMORY, 1);
|
2017-08-07 07:36:37 +02:00
|
|
|
int ret = dbenv->open(nullptr,
|
2014-09-19 19:21:46 +02:00
|
|
|
DB_CREATE |
|
|
|
|
DB_INIT_LOCK |
|
|
|
|
DB_INIT_LOG |
|
|
|
|
DB_INIT_MPOOL |
|
|
|
|
DB_INIT_TXN |
|
|
|
|
DB_THREAD |
|
|
|
|
DB_PRIVATE,
|
|
|
|
S_IRUSR | S_IWUSR);
|
2012-05-22 21:51:13 +02:00
|
|
|
if (ret > 0)
|
2017-01-27 02:33:45 +01:00
|
|
|
throw std::runtime_error(strprintf("CDBEnv::MakeMock: Error %d opening database environment.", ret));
|
2012-05-22 21:51:13 +02:00
|
|
|
|
|
|
|
fDbEnvInit = true;
|
|
|
|
fMockDb = true;
|
|
|
|
}
|
|
|
|
|
2017-06-06 00:01:48 +02:00
|
|
|
CDBEnv::VerifyResult CDBEnv::Verify(const std::string& strFile, recoverFunc_type recoverFunc, std::string& out_backup_filename)
|
2012-09-18 20:30:47 +02:00
|
|
|
{
|
|
|
|
LOCK(cs_db);
|
|
|
|
assert(mapFileUseCount.count(strFile) == 0);
|
|
|
|
|
2017-08-09 16:24:12 +02:00
|
|
|
Db db(dbenv.get(), 0);
|
2017-08-07 07:36:37 +02:00
|
|
|
int result = db.verify(strFile.c_str(), nullptr, nullptr, 0);
|
2012-09-18 20:30:47 +02:00
|
|
|
if (result == 0)
|
|
|
|
return VERIFY_OK;
|
2017-08-07 07:36:37 +02:00
|
|
|
else if (recoverFunc == nullptr)
|
2012-09-18 20:30:47 +02:00
|
|
|
return RECOVER_FAIL;
|
|
|
|
|
|
|
|
// Try to recover:
|
2017-06-06 00:01:48 +02:00
|
|
|
bool fRecovered = (*recoverFunc)(strFile, out_backup_filename);
|
2012-09-18 20:30:47 +02:00
|
|
|
return (fRecovered ? RECOVER_OK : RECOVER_FAIL);
|
|
|
|
}
|
|
|
|
|
2017-06-06 00:01:48 +02:00
|
|
|
bool CDB::Recover(const std::string& filename, void *callbackDataIn, bool (*recoverKVcallback)(void* callbackData, CDataStream ssKey, CDataStream ssValue), std::string& newFilename)
|
2016-08-24 09:57:23 +02:00
|
|
|
{
|
|
|
|
// Recovery procedure:
|
2017-06-06 00:02:32 +02:00
|
|
|
// move wallet file to walletfilename.timestamp.bak
|
2016-08-24 09:57:23 +02:00
|
|
|
// Call Salvage with fAggressive=true to
|
|
|
|
// get as much data as possible.
|
|
|
|
// Rewrite salvaged data to fresh wallet file
|
|
|
|
// Set -rescan so any missing transactions will be
|
|
|
|
// found.
|
|
|
|
int64_t now = GetTime();
|
2017-06-06 00:02:32 +02:00
|
|
|
newFilename = strprintf("%s.%d.bak", filename, now);
|
2016-08-24 09:57:23 +02:00
|
|
|
|
2017-08-07 07:36:37 +02:00
|
|
|
int result = bitdb.dbenv->dbrename(nullptr, filename.c_str(), nullptr,
|
2016-08-24 09:57:23 +02:00
|
|
|
newFilename.c_str(), DB_AUTO_COMMIT);
|
|
|
|
if (result == 0)
|
|
|
|
LogPrintf("Renamed %s to %s\n", filename, newFilename);
|
|
|
|
else
|
|
|
|
{
|
|
|
|
LogPrintf("Failed to rename %s to %s\n", filename, newFilename);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::vector<CDBEnv::KeyValPair> salvagedData;
|
|
|
|
bool fSuccess = bitdb.Salvage(newFilename, true, salvagedData);
|
|
|
|
if (salvagedData.empty())
|
|
|
|
{
|
|
|
|
LogPrintf("Salvage(aggressive) found no records in %s.\n", newFilename);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
LogPrintf("Salvage(aggressive) found %u records\n", salvagedData.size());
|
|
|
|
|
2017-08-15 07:46:56 +02:00
|
|
|
std::unique_ptr<Db> pdbCopy = MakeUnique<Db>(bitdb.dbenv.get(), 0);
|
2017-08-07 07:36:37 +02:00
|
|
|
int ret = pdbCopy->open(nullptr, // Txn pointer
|
2016-08-24 09:57:23 +02:00
|
|
|
filename.c_str(), // Filename
|
|
|
|
"main", // Logical db name
|
|
|
|
DB_BTREE, // Database type
|
|
|
|
DB_CREATE, // Flags
|
|
|
|
0);
|
2017-08-09 10:24:15 +02:00
|
|
|
if (ret > 0) {
|
2016-08-24 09:57:23 +02:00
|
|
|
LogPrintf("Cannot create database file %s\n", filename);
|
2017-08-09 10:24:15 +02:00
|
|
|
pdbCopy->close(0);
|
2016-08-24 09:57:23 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
DbTxn* ptxn = bitdb.TxnBegin();
|
2017-06-02 03:18:57 +02:00
|
|
|
for (CDBEnv::KeyValPair& row : salvagedData)
|
2016-08-24 09:57:23 +02:00
|
|
|
{
|
|
|
|
if (recoverKVcallback)
|
|
|
|
{
|
|
|
|
CDataStream ssKey(row.first, SER_DISK, CLIENT_VERSION);
|
|
|
|
CDataStream ssValue(row.second, SER_DISK, CLIENT_VERSION);
|
|
|
|
if (!(*recoverKVcallback)(callbackDataIn, ssKey, ssValue))
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
Dbt datKey(&row.first[0], row.first.size());
|
|
|
|
Dbt datValue(&row.second[0], row.second.size());
|
|
|
|
int ret2 = pdbCopy->put(ptxn, &datKey, &datValue, DB_NOOVERWRITE);
|
|
|
|
if (ret2 > 0)
|
|
|
|
fSuccess = false;
|
|
|
|
}
|
|
|
|
ptxn->commit(0);
|
|
|
|
pdbCopy->close(0);
|
|
|
|
|
|
|
|
return fSuccess;
|
|
|
|
}
|
|
|
|
|
2017-10-08 22:48:07 +02:00
|
|
|
bool CDB::VerifyEnvironment(const std::string& walletFile, const fs::path& walletDir, std::string& errorStr)
|
2016-08-24 09:57:23 +02:00
|
|
|
{
|
|
|
|
LogPrintf("Using BerkeleyDB version %s\n", DbEnv::version(0, 0, 0));
|
|
|
|
LogPrintf("Using wallet %s\n", walletFile);
|
|
|
|
|
|
|
|
// Wallet file must be a plain filename without a directory
|
2017-03-01 17:05:50 +01:00
|
|
|
if (walletFile != fs::basename(walletFile) + fs::extension(walletFile))
|
2016-08-24 09:57:23 +02:00
|
|
|
{
|
2017-10-08 22:48:07 +02:00
|
|
|
errorStr = strprintf(_("Wallet %s resides outside wallet directory %s"), walletFile, walletDir.string());
|
2016-08-24 09:57:23 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2017-12-14 23:06:22 +01:00
|
|
|
if (!bitdb.Open(walletDir, true)) {
|
2017-12-24 00:45:33 +01:00
|
|
|
errorStr = strprintf(_("Error initializing wallet database environment %s!"), walletDir);
|
2017-12-14 23:06:22 +01:00
|
|
|
return false;
|
2016-08-24 09:57:23 +02:00
|
|
|
}
|
2017-12-14 23:06:22 +01:00
|
|
|
|
2016-08-24 09:57:23 +02:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2017-10-08 22:48:07 +02:00
|
|
|
bool CDB::VerifyDatabaseFile(const std::string& walletFile, const fs::path& walletDir, std::string& warningStr, std::string& errorStr, CDBEnv::recoverFunc_type recoverFunc)
|
2016-08-24 09:57:23 +02:00
|
|
|
{
|
2017-10-08 22:48:07 +02:00
|
|
|
if (fs::exists(walletDir / walletFile))
|
2016-08-24 09:57:23 +02:00
|
|
|
{
|
2017-06-06 00:01:48 +02:00
|
|
|
std::string backup_filename;
|
|
|
|
CDBEnv::VerifyResult r = bitdb.Verify(walletFile, recoverFunc, backup_filename);
|
2016-08-24 09:57:23 +02:00
|
|
|
if (r == CDBEnv::RECOVER_OK)
|
|
|
|
{
|
|
|
|
warningStr = strprintf(_("Warning: Wallet file corrupt, data salvaged!"
|
|
|
|
" Original %s saved as %s in %s; if"
|
|
|
|
" your balance or transactions are incorrect you should"
|
|
|
|
" restore from a backup."),
|
2017-10-08 22:48:07 +02:00
|
|
|
walletFile, backup_filename, walletDir);
|
2016-08-24 09:57:23 +02:00
|
|
|
}
|
|
|
|
if (r == CDBEnv::RECOVER_FAIL)
|
|
|
|
{
|
|
|
|
errorStr = strprintf(_("%s corrupt, salvage failed"), walletFile);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// also return true if files does not exists
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2016-02-15 16:09:13 +01:00
|
|
|
/* End of headers, beginning of key/value data */
|
|
|
|
static const char *HEADER_END = "HEADER=END";
|
|
|
|
/* End of key/value data */
|
|
|
|
static const char *DATA_END = "DATA=END";
|
|
|
|
|
2015-03-21 18:40:51 +01:00
|
|
|
bool CDBEnv::Salvage(const std::string& strFile, bool fAggressive, std::vector<CDBEnv::KeyValPair>& vResult)
|
2012-09-18 20:30:47 +02:00
|
|
|
{
|
|
|
|
LOCK(cs_db);
|
|
|
|
assert(mapFileUseCount.count(strFile) == 0);
|
|
|
|
|
|
|
|
u_int32_t flags = DB_SALVAGE;
|
2014-09-19 19:21:46 +02:00
|
|
|
if (fAggressive)
|
|
|
|
flags |= DB_AGGRESSIVE;
|
2012-09-18 20:30:47 +02:00
|
|
|
|
2017-01-27 02:33:45 +01:00
|
|
|
std::stringstream strDump;
|
2012-09-18 20:30:47 +02:00
|
|
|
|
2017-08-09 16:24:12 +02:00
|
|
|
Db db(dbenv.get(), 0);
|
2017-08-07 07:36:37 +02:00
|
|
|
int result = db.verify(strFile.c_str(), nullptr, &strDump, flags);
|
2014-09-19 19:21:46 +02:00
|
|
|
if (result == DB_VERIFY_BAD) {
|
2015-01-08 11:44:25 +01:00
|
|
|
LogPrintf("CDBEnv::Salvage: Database salvage found errors, all data may not be recoverable.\n");
|
2014-09-19 19:21:46 +02:00
|
|
|
if (!fAggressive) {
|
2015-01-08 11:44:25 +01:00
|
|
|
LogPrintf("CDBEnv::Salvage: Rerun with aggressive mode to ignore errors and continue.\n");
|
2013-03-24 20:59:03 +01:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
2014-09-19 19:21:46 +02:00
|
|
|
if (result != 0 && result != DB_VERIFY_BAD) {
|
2015-01-08 11:44:25 +01:00
|
|
|
LogPrintf("CDBEnv::Salvage: Database salvage failed with result %d.\n", result);
|
2012-09-18 20:30:47 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Format of bdb dump is ascii lines:
|
|
|
|
// header lines...
|
|
|
|
// HEADER=END
|
2016-02-15 15:50:28 +01:00
|
|
|
// hexadecimal key
|
|
|
|
// hexadecimal value
|
|
|
|
// ... repeated
|
2012-09-18 20:30:47 +02:00
|
|
|
// DATA=END
|
|
|
|
|
2017-01-27 02:33:45 +01:00
|
|
|
std::string strLine;
|
2016-02-15 16:09:13 +01:00
|
|
|
while (!strDump.eof() && strLine != HEADER_END)
|
2012-09-18 20:30:47 +02:00
|
|
|
getline(strDump, strLine); // Skip past header
|
|
|
|
|
|
|
|
std::string keyHex, valueHex;
|
2016-02-15 16:09:13 +01:00
|
|
|
while (!strDump.eof() && keyHex != DATA_END) {
|
2012-09-18 20:30:47 +02:00
|
|
|
getline(strDump, keyHex);
|
2016-02-15 16:09:13 +01:00
|
|
|
if (keyHex != DATA_END) {
|
|
|
|
if (strDump.eof())
|
|
|
|
break;
|
2012-09-18 20:30:47 +02:00
|
|
|
getline(strDump, valueHex);
|
2016-02-15 16:09:13 +01:00
|
|
|
if (valueHex == DATA_END) {
|
|
|
|
LogPrintf("CDBEnv::Salvage: WARNING: Number of keys in data does not match number of values.\n");
|
|
|
|
break;
|
|
|
|
}
|
2014-09-19 19:21:46 +02:00
|
|
|
vResult.push_back(make_pair(ParseHex(keyHex), ParseHex(valueHex)));
|
2012-09-18 20:30:47 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-02-15 16:09:13 +01:00
|
|
|
if (keyHex != DATA_END) {
|
|
|
|
LogPrintf("CDBEnv::Salvage: WARNING: Unexpected end of file while reading salvage output.\n");
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2012-09-18 20:30:47 +02:00
|
|
|
return (result == 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2014-09-16 15:16:29 +02:00
|
|
|
void CDBEnv::CheckpointLSN(const std::string& strFile)
|
2012-05-14 03:37:39 +02:00
|
|
|
{
|
2015-03-03 16:49:12 +01:00
|
|
|
dbenv->txn_checkpoint(0, 0, 0);
|
2012-05-22 21:51:13 +02:00
|
|
|
if (fMockDb)
|
|
|
|
return;
|
2015-03-03 16:49:12 +01:00
|
|
|
dbenv->lsn_reset(strFile.c_str(), 0);
|
2012-05-14 03:37:39 +02:00
|
|
|
}
|
2010-08-29 18:58:15 +02:00
|
|
|
|
|
|
|
|
2017-08-07 07:36:37 +02:00
|
|
|
CDB::CDB(CWalletDBWrapper& dbw, const char* pszMode, bool fFlushOnCloseIn) : pdb(nullptr), activeTxn(nullptr)
|
2010-08-29 18:58:15 +02:00
|
|
|
{
|
2014-08-28 15:28:57 +02:00
|
|
|
fReadOnly = (!strchr(pszMode, '+') && !strchr(pszMode, 'w'));
|
2014-08-31 05:55:27 +02:00
|
|
|
fFlushOnClose = fFlushOnCloseIn;
|
2017-03-08 14:34:47 +01:00
|
|
|
env = dbw.env;
|
2017-03-08 13:08:26 +01:00
|
|
|
if (dbw.IsDummy()) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
const std::string &strFilename = dbw.strFile;
|
2010-08-29 18:58:15 +02:00
|
|
|
|
2017-08-07 07:36:37 +02:00
|
|
|
bool fCreate = strchr(pszMode, 'c') != nullptr;
|
2010-08-29 18:58:15 +02:00
|
|
|
unsigned int nFlags = DB_THREAD;
|
|
|
|
if (fCreate)
|
|
|
|
nFlags |= DB_CREATE;
|
|
|
|
|
|
|
|
{
|
2017-03-08 14:34:47 +01:00
|
|
|
LOCK(env->cs_db);
|
2017-10-08 22:48:07 +02:00
|
|
|
if (!env->Open(GetWalletDir()))
|
2017-01-27 02:33:45 +01:00
|
|
|
throw std::runtime_error("CDB: Failed to open database environment.");
|
2010-08-29 18:58:15 +02:00
|
|
|
|
2017-10-13 00:14:46 +02:00
|
|
|
pdb = env->mapDb[strFilename];
|
2017-08-07 07:36:37 +02:00
|
|
|
if (pdb == nullptr) {
|
2017-06-04 22:45:22 +02:00
|
|
|
int ret;
|
2017-10-18 23:21:06 +02:00
|
|
|
std::unique_ptr<Db> pdb_temp = MakeUnique<Db>(env->dbenv.get(), 0);
|
2010-08-29 18:58:15 +02:00
|
|
|
|
2017-03-08 14:34:47 +01:00
|
|
|
bool fMockDb = env->IsMock();
|
2014-09-19 19:21:46 +02:00
|
|
|
if (fMockDb) {
|
2017-10-13 00:14:46 +02:00
|
|
|
DbMpoolFile* mpf = pdb_temp->get_mpf();
|
2012-05-22 21:51:13 +02:00
|
|
|
ret = mpf->set_flags(DB_MPOOL_NOFILE, 1);
|
2017-10-13 00:14:46 +02:00
|
|
|
if (ret != 0) {
|
|
|
|
throw std::runtime_error(strprintf("CDB: Failed to configure for no temp file backing for database %s", strFilename));
|
|
|
|
}
|
2012-05-22 21:51:13 +02:00
|
|
|
}
|
|
|
|
|
2017-10-13 00:14:46 +02:00
|
|
|
ret = pdb_temp->open(nullptr, // Txn pointer
|
|
|
|
fMockDb ? nullptr : strFilename.c_str(), // Filename
|
|
|
|
fMockDb ? strFilename.c_str() : "main", // Logical db name
|
|
|
|
DB_BTREE, // Database type
|
|
|
|
nFlags, // Flags
|
2010-08-29 18:58:15 +02:00
|
|
|
0);
|
|
|
|
|
2014-09-19 19:21:46 +02:00
|
|
|
if (ret != 0) {
|
2017-01-27 02:33:45 +01:00
|
|
|
throw std::runtime_error(strprintf("CDB: Error %d, can't open database %s", ret, strFilename));
|
2010-08-29 18:58:15 +02:00
|
|
|
}
|
2017-10-10 21:27:26 +02:00
|
|
|
CheckUniqueFileid(*env, strFilename, *pdb_temp);
|
2010-08-29 18:58:15 +02:00
|
|
|
|
2017-10-13 00:14:46 +02:00
|
|
|
pdb = pdb_temp.release();
|
|
|
|
env->mapDb[strFilename] = pdb;
|
|
|
|
|
2017-01-27 02:33:45 +01:00
|
|
|
if (fCreate && !Exists(std::string("version"))) {
|
2010-08-29 18:58:15 +02:00
|
|
|
bool fTmp = fReadOnly;
|
|
|
|
fReadOnly = false;
|
2011-12-16 22:26:14 +01:00
|
|
|
WriteVersion(CLIENT_VERSION);
|
2010-08-29 18:58:15 +02:00
|
|
|
fReadOnly = fTmp;
|
|
|
|
}
|
|
|
|
}
|
2017-10-13 00:14:46 +02:00
|
|
|
++env->mapFileUseCount[strFilename];
|
|
|
|
strFile = strFilename;
|
2010-08-29 18:58:15 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-07-06 16:33:34 +02:00
|
|
|
void CDB::Flush()
|
2010-08-29 18:58:15 +02:00
|
|
|
{
|
2012-05-14 18:39:29 +02:00
|
|
|
if (activeTxn)
|
2012-07-06 16:33:34 +02:00
|
|
|
return;
|
2010-08-29 18:58:15 +02:00
|
|
|
|
|
|
|
// Flush database activity from memory pool to disk log
|
|
|
|
unsigned int nMinutes = 0;
|
2010-12-05 10:29:30 +01:00
|
|
|
if (fReadOnly)
|
|
|
|
nMinutes = 1;
|
2012-03-28 22:09:18 +02:00
|
|
|
|
2017-08-01 21:17:40 +02:00
|
|
|
env->dbenv->txn_checkpoint(nMinutes ? gArgs.GetArg("-dblogsize", DEFAULT_WALLET_DBLOGSIZE) * 1024 : 0, nMinutes, 0);
|
2012-07-06 16:33:34 +02:00
|
|
|
}
|
|
|
|
|
2017-03-09 21:56:58 +01:00
|
|
|
void CWalletDBWrapper::IncrementUpdateCounter()
|
|
|
|
{
|
|
|
|
++nUpdateCounter;
|
|
|
|
}
|
|
|
|
|
2012-07-06 16:33:34 +02:00
|
|
|
void CDB::Close()
|
|
|
|
{
|
|
|
|
if (!pdb)
|
|
|
|
return;
|
|
|
|
if (activeTxn)
|
|
|
|
activeTxn->abort();
|
2017-08-07 07:36:37 +02:00
|
|
|
activeTxn = nullptr;
|
|
|
|
pdb = nullptr;
|
2012-07-06 16:33:34 +02:00
|
|
|
|
2014-08-31 05:55:27 +02:00
|
|
|
if (fFlushOnClose)
|
|
|
|
Flush();
|
2010-08-29 18:58:15 +02:00
|
|
|
|
2012-04-06 18:39:12 +02:00
|
|
|
{
|
2017-03-08 14:34:47 +01:00
|
|
|
LOCK(env->cs_db);
|
|
|
|
--env->mapFileUseCount[strFile];
|
2012-04-06 18:39:12 +02:00
|
|
|
}
|
2010-08-29 18:58:15 +02:00
|
|
|
}
|
|
|
|
|
2017-01-27 02:33:45 +01:00
|
|
|
void CDBEnv::CloseDb(const std::string& strFile)
|
2010-08-29 18:58:15 +02:00
|
|
|
{
|
|
|
|
{
|
2012-04-06 18:39:12 +02:00
|
|
|
LOCK(cs_db);
|
2017-08-07 07:36:37 +02:00
|
|
|
if (mapDb[strFile] != nullptr) {
|
2010-08-29 18:58:15 +02:00
|
|
|
// Close the database handle
|
|
|
|
Db* pdb = mapDb[strFile];
|
|
|
|
pdb->close(0);
|
|
|
|
delete pdb;
|
2017-08-07 07:36:37 +02:00
|
|
|
mapDb[strFile] = nullptr;
|
2010-08-29 18:58:15 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-03-08 11:48:58 +01:00
|
|
|
bool CDB::Rewrite(CWalletDBWrapper& dbw, const char* pszSkip)
|
2011-11-10 21:29:23 +01:00
|
|
|
{
|
2017-03-08 13:08:26 +01:00
|
|
|
if (dbw.IsDummy()) {
|
2017-03-08 11:48:58 +01:00
|
|
|
return true;
|
|
|
|
}
|
2017-03-08 14:34:47 +01:00
|
|
|
CDBEnv *env = dbw.env;
|
2017-03-08 11:48:58 +01:00
|
|
|
const std::string& strFile = dbw.strFile;
|
2014-09-19 19:21:46 +02:00
|
|
|
while (true) {
|
2011-11-10 21:29:23 +01:00
|
|
|
{
|
2017-03-08 14:34:47 +01:00
|
|
|
LOCK(env->cs_db);
|
|
|
|
if (!env->mapFileUseCount.count(strFile) || env->mapFileUseCount[strFile] == 0) {
|
2011-11-10 21:29:23 +01:00
|
|
|
// Flush log data to the dat file
|
2017-03-08 14:34:47 +01:00
|
|
|
env->CloseDb(strFile);
|
|
|
|
env->CheckpointLSN(strFile);
|
|
|
|
env->mapFileUseCount.erase(strFile);
|
2011-11-10 21:29:23 +01:00
|
|
|
|
|
|
|
bool fSuccess = true;
|
2015-01-08 11:44:25 +01:00
|
|
|
LogPrintf("CDB::Rewrite: Rewriting %s...\n", strFile);
|
2017-01-27 02:33:45 +01:00
|
|
|
std::string strFileRes = strFile + ".rewrite";
|
2011-11-20 17:12:00 +01:00
|
|
|
{ // surround usage of db with extra {}
|
2017-03-08 11:48:58 +01:00
|
|
|
CDB db(dbw, "r");
|
2017-08-15 07:46:56 +02:00
|
|
|
std::unique_ptr<Db> pdbCopy = MakeUnique<Db>(env->dbenv.get(), 0);
|
2012-09-18 21:07:58 +02:00
|
|
|
|
2017-08-07 07:36:37 +02:00
|
|
|
int ret = pdbCopy->open(nullptr, // Txn pointer
|
2014-09-19 19:21:46 +02:00
|
|
|
strFileRes.c_str(), // Filename
|
|
|
|
"main", // Logical db name
|
|
|
|
DB_BTREE, // Database type
|
|
|
|
DB_CREATE, // Flags
|
2011-11-20 17:12:00 +01:00
|
|
|
0);
|
2014-09-19 19:21:46 +02:00
|
|
|
if (ret > 0) {
|
2015-01-08 11:44:25 +01:00
|
|
|
LogPrintf("CDB::Rewrite: Can't create database file %s\n", strFileRes);
|
2011-11-20 17:12:00 +01:00
|
|
|
fSuccess = false;
|
|
|
|
}
|
2012-09-18 21:07:58 +02:00
|
|
|
|
2011-11-20 17:12:00 +01:00
|
|
|
Dbc* pcursor = db.GetCursor();
|
|
|
|
if (pcursor)
|
2014-09-19 19:21:46 +02:00
|
|
|
while (fSuccess) {
|
2012-04-16 14:56:45 +02:00
|
|
|
CDataStream ssKey(SER_DISK, CLIENT_VERSION);
|
|
|
|
CDataStream ssValue(SER_DISK, CLIENT_VERSION);
|
2016-06-11 16:35:19 +02:00
|
|
|
int ret1 = db.ReadAtCursor(pcursor, ssKey, ssValue);
|
|
|
|
if (ret1 == DB_NOTFOUND) {
|
2011-11-20 17:12:00 +01:00
|
|
|
pcursor->close();
|
|
|
|
break;
|
2016-06-11 16:35:19 +02:00
|
|
|
} else if (ret1 != 0) {
|
2011-11-20 17:12:00 +01:00
|
|
|
pcursor->close();
|
|
|
|
fSuccess = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (pszSkip &&
|
2016-12-15 17:34:59 +01:00
|
|
|
strncmp(ssKey.data(), pszSkip, std::min(ssKey.size(), strlen(pszSkip))) == 0)
|
2011-11-20 17:12:00 +01:00
|
|
|
continue;
|
2016-12-15 17:34:59 +01:00
|
|
|
if (strncmp(ssKey.data(), "\x07version", 8) == 0) {
|
2011-11-20 17:12:00 +01:00
|
|
|
// Update version:
|
|
|
|
ssValue.clear();
|
2011-12-16 22:26:14 +01:00
|
|
|
ssValue << CLIENT_VERSION;
|
2011-11-20 17:12:00 +01:00
|
|
|
}
|
2016-12-15 17:34:59 +01:00
|
|
|
Dbt datKey(ssKey.data(), ssKey.size());
|
|
|
|
Dbt datValue(ssValue.data(), ssValue.size());
|
2017-08-07 07:36:37 +02:00
|
|
|
int ret2 = pdbCopy->put(nullptr, &datKey, &datValue, DB_NOOVERWRITE);
|
2011-11-20 17:12:00 +01:00
|
|
|
if (ret2 > 0)
|
|
|
|
fSuccess = false;
|
2011-11-11 03:12:46 +01:00
|
|
|
}
|
2014-09-19 19:21:46 +02:00
|
|
|
if (fSuccess) {
|
2011-11-20 17:12:00 +01:00
|
|
|
db.Close();
|
2017-03-08 14:34:47 +01:00
|
|
|
env->CloseDb(strFile);
|
2011-11-20 17:12:00 +01:00
|
|
|
if (pdbCopy->close(0))
|
2011-11-10 21:29:23 +01:00
|
|
|
fSuccess = false;
|
2017-08-09 10:24:15 +02:00
|
|
|
} else {
|
|
|
|
pdbCopy->close(0);
|
2011-11-10 21:29:23 +01:00
|
|
|
}
|
|
|
|
}
|
2014-09-19 19:21:46 +02:00
|
|
|
if (fSuccess) {
|
2017-08-09 16:24:12 +02:00
|
|
|
Db dbA(env->dbenv.get(), 0);
|
2017-08-07 07:36:37 +02:00
|
|
|
if (dbA.remove(strFile.c_str(), nullptr, 0))
|
2011-11-10 21:29:23 +01:00
|
|
|
fSuccess = false;
|
2017-08-09 16:24:12 +02:00
|
|
|
Db dbB(env->dbenv.get(), 0);
|
2017-08-07 07:36:37 +02:00
|
|
|
if (dbB.rename(strFileRes.c_str(), nullptr, strFile.c_str(), 0))
|
2011-11-10 21:29:23 +01:00
|
|
|
fSuccess = false;
|
|
|
|
}
|
|
|
|
if (!fSuccess)
|
2015-01-08 11:44:25 +01:00
|
|
|
LogPrintf("CDB::Rewrite: Failed to rewrite database file %s\n", strFileRes);
|
2011-11-10 21:29:23 +01:00
|
|
|
return fSuccess;
|
|
|
|
}
|
|
|
|
}
|
2013-03-07 20:25:21 +01:00
|
|
|
MilliSleep(100);
|
2011-11-10 21:29:23 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-05-14 03:37:39 +02:00
|
|
|
void CDBEnv::Flush(bool fShutdown)
|
2010-08-29 18:58:15 +02:00
|
|
|
{
|
2013-04-13 07:13:08 +02:00
|
|
|
int64_t nStart = GetTimeMillis();
|
2014-01-22 14:41:24 +01:00
|
|
|
// Flush log data to the actual data file on all files that are not in use
|
2016-12-25 21:19:40 +01:00
|
|
|
LogPrint(BCLog::DB, "CDBEnv::Flush: Flush(%s)%s\n", fShutdown ? "true" : "false", fDbEnvInit ? "" : " database not started");
|
2010-08-29 18:58:15 +02:00
|
|
|
if (!fDbEnvInit)
|
|
|
|
return;
|
|
|
|
{
|
2012-04-06 18:39:12 +02:00
|
|
|
LOCK(cs_db);
|
2017-01-27 02:33:45 +01:00
|
|
|
std::map<std::string, int>::iterator mi = mapFileUseCount.begin();
|
2014-09-19 19:21:46 +02:00
|
|
|
while (mi != mapFileUseCount.end()) {
|
2017-01-27 02:33:45 +01:00
|
|
|
std::string strFile = (*mi).first;
|
2010-08-29 18:58:15 +02:00
|
|
|
int nRefCount = (*mi).second;
|
2016-12-25 21:19:40 +01:00
|
|
|
LogPrint(BCLog::DB, "CDBEnv::Flush: Flushing %s (refcount = %d)...\n", strFile, nRefCount);
|
2014-09-19 19:21:46 +02:00
|
|
|
if (nRefCount == 0) {
|
2010-08-29 18:58:15 +02:00
|
|
|
// Move log data to the dat file
|
|
|
|
CloseDb(strFile);
|
2016-12-25 21:19:40 +01:00
|
|
|
LogPrint(BCLog::DB, "CDBEnv::Flush: %s checkpoint\n", strFile);
|
2015-03-03 16:49:12 +01:00
|
|
|
dbenv->txn_checkpoint(0, 0, 0);
|
2016-12-25 21:19:40 +01:00
|
|
|
LogPrint(BCLog::DB, "CDBEnv::Flush: %s detach\n", strFile);
|
2012-11-04 12:48:45 +01:00
|
|
|
if (!fMockDb)
|
2015-03-03 16:49:12 +01:00
|
|
|
dbenv->lsn_reset(strFile.c_str(), 0);
|
2016-12-25 21:19:40 +01:00
|
|
|
LogPrint(BCLog::DB, "CDBEnv::Flush: %s closed\n", strFile);
|
2010-08-29 18:58:15 +02:00
|
|
|
mapFileUseCount.erase(mi++);
|
2014-09-19 19:21:46 +02:00
|
|
|
} else
|
2010-08-29 18:58:15 +02:00
|
|
|
mi++;
|
|
|
|
}
|
2016-12-25 21:19:40 +01:00
|
|
|
LogPrint(BCLog::DB, "CDBEnv::Flush: Flush(%s)%s took %15dms\n", fShutdown ? "true" : "false", fDbEnvInit ? "" : " database not started", GetTimeMillis() - nStart);
|
2014-09-19 19:21:46 +02:00
|
|
|
if (fShutdown) {
|
2010-08-29 18:58:15 +02:00
|
|
|
char** listp;
|
2014-09-19 19:21:46 +02:00
|
|
|
if (mapFileUseCount.empty()) {
|
2015-03-03 16:49:12 +01:00
|
|
|
dbenv->log_archive(&listp, DB_ARCH_REMOVE);
|
2012-05-14 03:37:39 +02:00
|
|
|
Close();
|
2013-04-24 00:43:14 +02:00
|
|
|
if (!fMockDb)
|
2017-03-01 17:05:50 +01:00
|
|
|
fs::remove_all(fs::path(strPath) / "database");
|
2011-11-11 03:12:46 +01:00
|
|
|
}
|
2010-08-29 18:58:15 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-08-24 09:57:23 +02:00
|
|
|
|
2017-03-08 11:48:58 +01:00
|
|
|
bool CDB::PeriodicFlush(CWalletDBWrapper& dbw)
|
2016-08-24 09:57:23 +02:00
|
|
|
{
|
2017-03-08 13:08:26 +01:00
|
|
|
if (dbw.IsDummy()) {
|
|
|
|
return true;
|
|
|
|
}
|
2016-08-24 09:57:23 +02:00
|
|
|
bool ret = false;
|
2017-03-08 14:34:47 +01:00
|
|
|
CDBEnv *env = dbw.env;
|
2017-03-08 11:48:58 +01:00
|
|
|
const std::string& strFile = dbw.strFile;
|
2016-08-24 09:57:23 +02:00
|
|
|
TRY_LOCK(bitdb.cs_db,lockDb);
|
|
|
|
if (lockDb)
|
|
|
|
{
|
|
|
|
// Don't do this if any databases are in use
|
|
|
|
int nRefCount = 0;
|
2017-03-08 14:34:47 +01:00
|
|
|
std::map<std::string, int>::iterator mit = env->mapFileUseCount.begin();
|
|
|
|
while (mit != env->mapFileUseCount.end())
|
2016-08-24 09:57:23 +02:00
|
|
|
{
|
2017-03-18 07:19:16 +01:00
|
|
|
nRefCount += (*mit).second;
|
|
|
|
mit++;
|
2016-08-24 09:57:23 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (nRefCount == 0)
|
|
|
|
{
|
|
|
|
boost::this_thread::interruption_point();
|
2017-03-08 14:34:47 +01:00
|
|
|
std::map<std::string, int>::iterator mi = env->mapFileUseCount.find(strFile);
|
|
|
|
if (mi != env->mapFileUseCount.end())
|
2016-08-24 09:57:23 +02:00
|
|
|
{
|
2016-12-25 21:19:40 +01:00
|
|
|
LogPrint(BCLog::DB, "Flushing %s\n", strFile);
|
2016-08-24 09:57:23 +02:00
|
|
|
int64_t nStart = GetTimeMillis();
|
|
|
|
|
|
|
|
// Flush wallet file so it's self contained
|
2017-03-08 14:34:47 +01:00
|
|
|
env->CloseDb(strFile);
|
|
|
|
env->CheckpointLSN(strFile);
|
2016-08-24 09:57:23 +02:00
|
|
|
|
2017-03-08 14:34:47 +01:00
|
|
|
env->mapFileUseCount.erase(mi++);
|
2016-12-25 21:19:40 +01:00
|
|
|
LogPrint(BCLog::DB, "Flushed %s %dms\n", strFile, GetTimeMillis() - nStart);
|
2016-08-24 09:57:23 +02:00
|
|
|
ret = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
}
|
2017-03-08 11:48:58 +01:00
|
|
|
|
|
|
|
bool CWalletDBWrapper::Rewrite(const char* pszSkip)
|
|
|
|
{
|
|
|
|
return CDB::Rewrite(*this, pszSkip);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool CWalletDBWrapper::Backup(const std::string& strDest)
|
|
|
|
{
|
2017-03-08 13:08:26 +01:00
|
|
|
if (IsDummy()) {
|
2017-03-08 11:48:58 +01:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
while (true)
|
|
|
|
{
|
|
|
|
{
|
2017-03-08 14:34:47 +01:00
|
|
|
LOCK(env->cs_db);
|
|
|
|
if (!env->mapFileUseCount.count(strFile) || env->mapFileUseCount[strFile] == 0)
|
2017-03-08 11:48:58 +01:00
|
|
|
{
|
|
|
|
// Flush log data to the dat file
|
2017-03-08 14:34:47 +01:00
|
|
|
env->CloseDb(strFile);
|
|
|
|
env->CheckpointLSN(strFile);
|
|
|
|
env->mapFileUseCount.erase(strFile);
|
2017-03-08 11:48:58 +01:00
|
|
|
|
|
|
|
// Copy wallet file
|
2017-10-08 22:48:07 +02:00
|
|
|
fs::path pathSrc = GetWalletDir() / strFile;
|
2017-03-08 11:48:58 +01:00
|
|
|
fs::path pathDest(strDest);
|
|
|
|
if (fs::is_directory(pathDest))
|
|
|
|
pathDest /= strFile;
|
|
|
|
|
|
|
|
try {
|
2017-09-21 00:10:46 +02:00
|
|
|
if (fs::equivalent(pathSrc, pathDest)) {
|
|
|
|
LogPrintf("cannot backup to wallet source file %s\n", pathDest.string());
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2017-03-08 11:48:58 +01:00
|
|
|
fs::copy_file(pathSrc, pathDest, fs::copy_option::overwrite_if_exists);
|
|
|
|
LogPrintf("copied %s to %s\n", strFile, pathDest.string());
|
|
|
|
return true;
|
|
|
|
} catch (const fs::filesystem_error& e) {
|
|
|
|
LogPrintf("error copying %s to %s - %s\n", strFile, pathDest.string(), e.what());
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
MilliSleep(100);
|
|
|
|
}
|
|
|
|
}
|
2017-03-08 14:34:47 +01:00
|
|
|
|
|
|
|
void CWalletDBWrapper::Flush(bool shutdown)
|
|
|
|
{
|
|
|
|
if (!IsDummy()) {
|
|
|
|
env->Flush(shutdown);
|
|
|
|
}
|
|
|
|
}
|