Parts of LMDB impl of BlockchainDB done and working
The rest should just be tedious copypasta and modification.
This commit is contained in:
Thomas Winget
warptangent
parent
1240cf805b
commit
db00ce0173
|
|
@ -27,15 +27,21 @@
|
|||
|
||||
#include "db_lmdb.h"
|
||||
|
||||
#include <boost/filesystem.hpp>
|
||||
#include <memory> // std::unique_ptr
|
||||
#include <cstring> // memcpy
|
||||
|
||||
#include "cryptonote_core/cryptonote_format_utils.h"
|
||||
|
||||
namespace cryptonote
|
||||
{
|
||||
|
||||
#define CN_LMDB_DBI_OPEN( txn, flags, dbi ) \
|
||||
if (mdb_dbi_open(txn, NULL, flags, &dbi)) \
|
||||
{ \
|
||||
throw DB_OPEN_FAILURE( "Failed to open db handle for ##dbi" ); \
|
||||
}
|
||||
|
||||
const char* LMDB_BLOCKS = "blocks";
|
||||
const char* LMDB_BLOCK_HASHES = "block_hashes";
|
||||
const char* LMDB_BLOCK_SIZES = "block_sizes";
|
||||
const char* LMDB_BLOCK_DIFFS = "block_diffs";
|
||||
const char* LMDB_BLOCK_COINS = "block_coins";
|
||||
const char* LMDB_TXS = "txs";
|
||||
|
||||
void BlockchainLMDB::add_block( const block& blk
|
||||
, const size_t& block_size
|
||||
|
|
@ -43,34 +49,121 @@ void BlockchainLMDB::add_block( const block& blk
|
|||
, const uint64_t& coins_generated
|
||||
)
|
||||
{
|
||||
check_open();
|
||||
|
||||
crypto::hash h = get_block_hash(blk);
|
||||
MDB_val val_h;
|
||||
val_h.mv_size = sizeof(crypto::hash);
|
||||
val_h.mv_data = &h;
|
||||
|
||||
MDB_val unused;
|
||||
if (mdb_get(*m_write_txn, m_block_hashes, &val_h, &unused) == 0)
|
||||
{
|
||||
LOG_PRINT_L1("Attempting to add block that's already in the db");
|
||||
throw BLOCK_EXISTS("Attempting to add block that's already in the db");
|
||||
}
|
||||
|
||||
if (m_height > 0)
|
||||
{
|
||||
MDB_val parent_key;
|
||||
crypto::hash parent = blk.prev_id;
|
||||
parent_key.mv_size = sizeof(crypto::hash);
|
||||
parent_key.mv_data = &parent;
|
||||
|
||||
MDB_val parent_h;
|
||||
if (mdb_get(*m_write_txn, m_block_hashes, &parent_key, &parent_h))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to get top block hash to check for new block's parent");
|
||||
throw DB_ERROR("Failed to get top block hash to check for new block's parent");
|
||||
}
|
||||
|
||||
uint64_t parent_height = *(uint64_t *)parent_h.mv_data;
|
||||
if (parent_height != m_height - 1)
|
||||
{
|
||||
LOG_PRINT_L0("Top block is not new block's parent");
|
||||
throw BLOCK_PARENT_DNE("Top block is not new block's parent");
|
||||
}
|
||||
}
|
||||
|
||||
MDB_val key;
|
||||
key.mv_size = sizeof(uint64_t);
|
||||
key.mv_data = &m_height;
|
||||
|
||||
auto bd = block_to_blob(blk);
|
||||
|
||||
// const-correctness be trolling, yo
|
||||
std::unique_ptr<char[]> bd_cpy(new char[bd.size()]);
|
||||
memcpy(bd_cpy.get(), bd.data(), bd.size());
|
||||
|
||||
MDB_val blob;
|
||||
blob.mv_size = bd.size();
|
||||
blob.mv_data = bd_cpy.get();
|
||||
if (mdb_put(*m_write_txn, m_blocks, &key, &blob, 0))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to add block blob to db transaction");
|
||||
throw DB_ERROR("Failed to add block blob to db transaction");
|
||||
}
|
||||
|
||||
size_t size_cpy = block_size;
|
||||
MDB_val sz;
|
||||
sz.mv_size = sizeof(block_size);
|
||||
sz.mv_data = &size_cpy;
|
||||
if (mdb_put(*m_write_txn, m_block_sizes, &key, &sz, 0))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to add block size to db transaction");
|
||||
throw DB_ERROR("Failed to add block size to db transaction");
|
||||
}
|
||||
|
||||
if (mdb_put(*m_write_txn, m_block_hashes, &val_h, &key, 0))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to add block size to db transaction");
|
||||
throw DB_ERROR("Failed to add block size to db transaction");
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
void BlockchainLMDB::remove_block(const crypto::hash& blk_hash)
|
||||
void BlockchainLMDB::remove_block()
|
||||
{
|
||||
check_open();
|
||||
}
|
||||
|
||||
void BlockchainLMDB::add_transaction_data(const crypto::hash& blk_hash, const transaction& tx)
|
||||
{
|
||||
check_open();
|
||||
}
|
||||
|
||||
void BlockchainLMDB::remove_transaction_data(const crypto::hash& tx_hash)
|
||||
{
|
||||
check_open();
|
||||
}
|
||||
|
||||
void BlockchainLMDB::add_output(const crypto::hash& tx_hash, const tx_out& tx_output, const uint64_t& local_index)
|
||||
{
|
||||
check_open();
|
||||
}
|
||||
|
||||
void BlockchainLMDB::remove_output(const tx_out& tx_output)
|
||||
{
|
||||
check_open();
|
||||
}
|
||||
|
||||
void BlockchainLMDB::add_spent_key(const crypto::key_image& k_image)
|
||||
{
|
||||
check_open();
|
||||
}
|
||||
|
||||
void BlockchainLMDB::remove_spent_key(const crypto::key_image& k_image)
|
||||
{
|
||||
check_open();
|
||||
}
|
||||
|
||||
void BlockchainLMDB::check_open()
|
||||
{
|
||||
if (!m_open)
|
||||
{
|
||||
LOG_PRINT_L0("DB operation attempted on a not-open DB instance");
|
||||
throw DB_ERROR("DB operation attempted on a not-open DB instance");
|
||||
}
|
||||
}
|
||||
|
||||
BlockchainLMDB::~BlockchainLMDB()
|
||||
|
|
@ -79,41 +172,113 @@ BlockchainLMDB::~BlockchainLMDB()
|
|||
|
||||
BlockchainLMDB::BlockchainLMDB()
|
||||
{
|
||||
// initialize folder to something "safe" just in case
|
||||
// someone accidentally misuses this class...
|
||||
m_folder = "thishsouldnotexistbecauseitisgibberish";
|
||||
m_open = false;
|
||||
}
|
||||
|
||||
void BlockchainLMDB::open(const std::string& filename)
|
||||
{
|
||||
|
||||
if (m_open)
|
||||
{
|
||||
LOG_PRINT_L0("Attempted to open db, but it's already open");
|
||||
throw DB_OPEN_FAILURE("Attempted to open db, but it's already open");
|
||||
}
|
||||
|
||||
boost::filesystem::path direc(filename);
|
||||
if (boost::filesystem::exists(direc))
|
||||
{
|
||||
if (!boost::filesystem::is_directory(direc))
|
||||
{
|
||||
LOG_PRINT_L0("LMDB needs a directory path, but a file was passed");
|
||||
throw DB_OPEN_FAILURE("LMDB needs a directory path, but a file was passed");
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (!boost::filesystem::create_directory(direc))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to create directory " << filename);
|
||||
throw DB_OPEN_FAILURE(std::string("Failed to create directory ").append(filename).c_str());
|
||||
}
|
||||
}
|
||||
|
||||
m_folder = filename;
|
||||
|
||||
// set up lmdb environment
|
||||
if (mdb_env_create(&m_env))
|
||||
{
|
||||
throw DB_ERROR("Failed to create lmdb environment");
|
||||
}
|
||||
if (mdb_env_open(m_env, filename.c_str(), 0, 0664))
|
||||
{
|
||||
throw DB_ERROR("Failed to open lmdb environment");
|
||||
LOG_PRINT_L0("Failed to create lmdb environment");
|
||||
throw DB_OPEN_FAILURE("Failed to create lmdb environment");
|
||||
}
|
||||
if (mdb_env_set_maxdbs(m_env, 15))
|
||||
{
|
||||
throw DB_ERROR("Failed to set max number of dbs");
|
||||
LOG_PRINT_L0("Failed to set max number of dbs");
|
||||
throw DB_OPEN_FAILURE("Failed to set max number of dbs");
|
||||
}
|
||||
MDB_txn* txn;
|
||||
if (mdb_txn_begin(m_env, NULL, 0, &txn))
|
||||
if (mdb_env_open(m_env, filename.c_str(), 0, 0664))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to open lmdb environment");
|
||||
throw DB_OPEN_FAILURE("Failed to open lmdb environment");
|
||||
}
|
||||
|
||||
// get a read/write MDB_txn
|
||||
txn_safe txn;
|
||||
if (mdb_txn_begin(m_env, NULL, 0, txn))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to create a transaction for the db");
|
||||
throw DB_ERROR("Failed to create a transaction for the db");
|
||||
}
|
||||
|
||||
CN_LMDB_DBI_OPEN(txn, MDB_CREATE | MDB_DUPSORT, m_blocks)
|
||||
CN_LMDB_DBI_OPEN(txn, MDB_CREATE | MDB_DUPSORT, m_block_txs)
|
||||
CN_LMDB_DBI_OPEN(txn, MDB_CREATE | MDB_DUPSORT, m_block_hashes)
|
||||
CN_LMDB_DBI_OPEN(txn, MDB_CREATE | MDB_DUPSORT, m_block_sizes)
|
||||
CN_LMDB_DBI_OPEN(txn, MDB_CREATE | MDB_DUPSORT, m_block_diffs)
|
||||
CN_LMDB_DBI_OPEN(txn, MDB_CREATE | MDB_DUPSORT, m_block_coins)
|
||||
CN_LMDB_DBI_OPEN(txn, MDB_CREATE | MDB_DUPSORT, m_txs)
|
||||
|
||||
if (mdb_txn_commit(txn))
|
||||
// open necessary databases, and set properties as needed
|
||||
// uses macros to avoid having to change things too many places
|
||||
if (mdb_dbi_open(txn, LMDB_BLOCKS, MDB_INTEGERKEY | MDB_CREATE, &m_blocks))
|
||||
{
|
||||
throw DB_OPEN_FAILURE("Failed to commit db open transaction");
|
||||
LOG_PRINT_L0("Failed to open db handle for m_blocks");
|
||||
throw DB_OPEN_FAILURE( "Failed to open db handle for m_blocks" );
|
||||
}
|
||||
|
||||
if (mdb_dbi_open(txn, LMDB_BLOCK_HASHES, MDB_INTEGERKEY | MDB_CREATE, &m_block_hashes))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to open db handle for m_block_hashes");
|
||||
throw DB_OPEN_FAILURE( "Failed to open db handle for m_block_hashes" );
|
||||
}
|
||||
if (mdb_dbi_open(txn, LMDB_BLOCK_SIZES, MDB_INTEGERKEY | MDB_CREATE, &m_block_sizes))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to open db handle for m_block_sizes");
|
||||
throw DB_OPEN_FAILURE( "Failed to open db handle for m_block_sizes" );
|
||||
}
|
||||
if (mdb_dbi_open(txn, LMDB_BLOCK_DIFFS, MDB_INTEGERKEY | MDB_CREATE, &m_block_diffs))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to open db handle for m_block_diffs");
|
||||
throw DB_OPEN_FAILURE( "Failed to open db handle for m_block_diffs" );
|
||||
}
|
||||
if (mdb_dbi_open(txn, LMDB_BLOCK_COINS, MDB_INTEGERKEY | MDB_CREATE, &m_block_coins))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to open db handle for m_block_coins");
|
||||
throw DB_OPEN_FAILURE( "Failed to open db handle for m_block_coins" );
|
||||
}
|
||||
if (mdb_dbi_open(txn, LMDB_TXS, MDB_CREATE, &m_txs))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to open db handle for m_txs");
|
||||
throw DB_OPEN_FAILURE( "Failed to open db handle for m_txs" );
|
||||
}
|
||||
|
||||
// get and keep current height
|
||||
MDB_stat db_stats;
|
||||
if (mdb_stat(txn, m_blocks, &db_stats))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to query m_blocks");
|
||||
throw DB_ERROR("Failed to query m_blocks");
|
||||
}
|
||||
m_height = db_stats.ms_entries;
|
||||
|
||||
// commit the transaction
|
||||
txn.commit();
|
||||
|
||||
m_open = true;
|
||||
// from here, init should be finished
|
||||
}
|
||||
|
||||
|
|
@ -136,184 +301,340 @@ void BlockchainLMDB::reset()
|
|||
|
||||
std::vector<std::string> BlockchainLMDB::get_filenames()
|
||||
{
|
||||
std::vector<std::string> retval;
|
||||
return retval;
|
||||
std::vector<std::string> filenames;
|
||||
|
||||
boost::filesystem::path datafile(m_folder);
|
||||
datafile /= "data.mdb";
|
||||
boost::filesystem::path lockfile(m_folder);
|
||||
lockfile /= "lock.mdb";
|
||||
|
||||
filenames.push_back(datafile.string());
|
||||
filenames.push_back(lockfile.string());
|
||||
|
||||
return filenames;
|
||||
}
|
||||
|
||||
|
||||
bool BlockchainLMDB::lock()
|
||||
{
|
||||
check_open();
|
||||
return false;
|
||||
}
|
||||
|
||||
void BlockchainLMDB::unlock()
|
||||
{
|
||||
check_open();
|
||||
}
|
||||
|
||||
|
||||
bool BlockchainLMDB::block_exists(const crypto::hash& h)
|
||||
{
|
||||
return false;
|
||||
check_open();
|
||||
|
||||
txn_safe txn;
|
||||
if (mdb_txn_begin(m_env, NULL, MDB_RDONLY, txn))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to create a transaction for the db");
|
||||
throw DB_ERROR("Failed to create a transaction for the db");
|
||||
}
|
||||
|
||||
crypto::hash key_cpy = h;
|
||||
MDB_val key;
|
||||
key.mv_size = sizeof(crypto::hash);
|
||||
key.mv_data = &key_cpy;
|
||||
|
||||
MDB_val result;
|
||||
auto get_result = mdb_get(txn, m_block_hashes, &key, &result);
|
||||
if (get_result == MDB_NOTFOUND)
|
||||
{
|
||||
LOG_PRINT_L1("Block with hash " << epee::string_tools::pod_to_hex(h) << "not found in db");
|
||||
return false;
|
||||
}
|
||||
else if (get_result)
|
||||
{
|
||||
LOG_PRINT_L0("DB error attempting to fetch block index from hash");
|
||||
throw DB_ERROR("DB error attempting to fetch block index from hash");
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
block BlockchainLMDB::get_block(const crypto::hash& h)
|
||||
{
|
||||
block b;
|
||||
return b;
|
||||
check_open();
|
||||
|
||||
txn_safe txn;
|
||||
if (mdb_txn_begin(m_env, NULL, MDB_RDONLY, txn))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to create a transaction for the db");
|
||||
throw DB_ERROR("Failed to create a transaction for the db");
|
||||
}
|
||||
|
||||
crypto::hash key_cpy = h;
|
||||
MDB_val key;
|
||||
key.mv_size = sizeof(crypto::hash);
|
||||
key.mv_data = &key_cpy;
|
||||
|
||||
MDB_val result;
|
||||
auto get_result = mdb_get(txn, m_block_hashes, &key, &result);
|
||||
if (get_result == MDB_NOTFOUND)
|
||||
{
|
||||
LOG_PRINT_L1("Attempted to retrieve non-existent block");
|
||||
throw BLOCK_DNE("Attempted to retrieve non-existent block");
|
||||
}
|
||||
else if (get_result)
|
||||
{
|
||||
LOG_PRINT_L0("Error attempting to retrieve a block index from the db");
|
||||
throw DB_ERROR("Error attempting to retrieve a block index from the db");
|
||||
}
|
||||
|
||||
txn.commit();
|
||||
|
||||
uint64_t index = *(uint64_t*)result.mv_data;
|
||||
|
||||
return get_block_from_height(index);
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::get_block_height(const crypto::hash& h)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
block_header BlockchainLMDB::get_block_header(const crypto::hash& h)
|
||||
{
|
||||
check_open();
|
||||
block_header bh;
|
||||
return bh;
|
||||
}
|
||||
|
||||
block BlockchainLMDB::get_block_from_height(const uint64_t& height)
|
||||
{
|
||||
check_open();
|
||||
|
||||
txn_safe txn;
|
||||
if (mdb_txn_begin(m_env, NULL, MDB_RDONLY, txn))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to create a transaction for the db");
|
||||
throw DB_ERROR("Failed to create a transaction for the db");
|
||||
}
|
||||
|
||||
uint64_t height_cpy = height;
|
||||
MDB_val key;
|
||||
key.mv_size = sizeof(uint64_t);
|
||||
key.mv_data = &height_cpy;
|
||||
MDB_val result;
|
||||
auto get_result = mdb_get(txn, m_blocks, &key, &result);
|
||||
if (get_result == MDB_NOTFOUND)
|
||||
{
|
||||
LOG_PRINT_L0("Attempted to get block from height " << height << ", but no such block exists");
|
||||
throw DB_ERROR("Attempt to get block from height failed -- block not in db");
|
||||
}
|
||||
else if (get_result)
|
||||
{
|
||||
LOG_PRINT_L0("Error attempting to retrieve a block from the db");
|
||||
throw DB_ERROR("Error attempting to retrieve a block from the db");
|
||||
}
|
||||
|
||||
txn.commit();
|
||||
|
||||
blobdata bd;
|
||||
bd.assign(reinterpret_cast<char*>(result.mv_data), result.mv_size);
|
||||
|
||||
block b;
|
||||
if (!parse_and_validate_block_from_blob(bd, b))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to parse block from blob retrieved from the db");
|
||||
throw DB_ERROR("Failed to parse block from blob retrieved from the db");
|
||||
}
|
||||
|
||||
return b;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::get_block_timestamp(const uint64_t& height)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::get_top_block_timestamp()
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
size_t BlockchainLMDB::get_block_size(const uint64_t& height)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
difficulty_type BlockchainLMDB::get_block_cumulative_difficulty(const uint64_t& height)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
difficulty_type BlockchainLMDB::get_block_difficulty(const uint64_t& height)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::get_block_already_generated_coins(const uint64_t& height)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
crypto::hash BlockchainLMDB::get_block_hash_from_height(const uint64_t& height)
|
||||
{
|
||||
check_open();
|
||||
crypto::hash h;
|
||||
return h;
|
||||
}
|
||||
|
||||
std::vector<block> BlockchainLMDB::get_blocks_range(const uint64_t& h1, const uint64_t& h2)
|
||||
{
|
||||
check_open();
|
||||
std::vector<block> v;
|
||||
return v;
|
||||
}
|
||||
|
||||
std::vector<crypto::hash> BlockchainLMDB::get_hashes_range(const uint64_t& h1, const uint64_t& h2)
|
||||
{
|
||||
check_open();
|
||||
std::vector<crypto::hash> v;
|
||||
return v;
|
||||
}
|
||||
|
||||
crypto::hash BlockchainLMDB::top_block_hash()
|
||||
{
|
||||
check_open();
|
||||
crypto::hash h;
|
||||
return h;
|
||||
}
|
||||
|
||||
block BlockchainLMDB::get_top_block()
|
||||
{
|
||||
check_open();
|
||||
block b;
|
||||
return b;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::height()
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
bool BlockchainLMDB::tx_exists(const crypto::hash& h)
|
||||
{
|
||||
check_open();
|
||||
return false;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::get_tx_unlock_time(const crypto::hash& h)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
transaction BlockchainLMDB::get_tx(const crypto::hash& h)
|
||||
{
|
||||
check_open();
|
||||
transaction t;
|
||||
return t;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::get_tx_count()
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
std::vector<transaction> BlockchainLMDB::get_tx_list(const std::vector<crypto::hash>& hlist)
|
||||
{
|
||||
check_open();
|
||||
std::vector<transaction> v;
|
||||
return v;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::get_tx_block_height(const crypto::hash& h)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::get_random_output(const uint64_t& amount)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::get_num_outputs(const uint64_t& amount)
|
||||
{
|
||||
check_open();
|
||||
return 0;
|
||||
}
|
||||
|
||||
crypto::public_key BlockchainLMDB::get_output_key(const uint64_t& amount, const uint64_t& index)
|
||||
{
|
||||
check_open();
|
||||
crypto::public_key pk;
|
||||
return pk;
|
||||
}
|
||||
|
||||
tx_out BlockchainLMDB::get_output(const crypto::hash& h, const uint64_t& index)
|
||||
{
|
||||
check_open();
|
||||
tx_out o;
|
||||
return o;
|
||||
}
|
||||
|
||||
tx_out_index BlockchainLMDB::get_output_tx_and_index(const uint64_t& amount, const uint64_t& index)
|
||||
{
|
||||
check_open();
|
||||
tx_out_index i;
|
||||
return i;
|
||||
}
|
||||
|
||||
std::vector<uint64_t> BlockchainLMDB::get_tx_output_indices(const crypto::hash& h)
|
||||
{
|
||||
check_open();
|
||||
std::vector<uint64_t> v;
|
||||
return v;
|
||||
}
|
||||
|
||||
bool BlockchainLMDB::has_key_image(const crypto::key_image& img)
|
||||
{
|
||||
check_open();
|
||||
return false;
|
||||
}
|
||||
|
||||
uint64_t BlockchainLMDB::add_block( const block& blk
|
||||
, const size_t& block_size
|
||||
, const difficulty_type& cumulative_difficulty
|
||||
, const uint64_t& coins_generated
|
||||
, const std::vector<transaction>& txs
|
||||
)
|
||||
{
|
||||
check_open();
|
||||
txn_safe txn;
|
||||
if (mdb_txn_begin(m_env, NULL, 0, txn))
|
||||
{
|
||||
LOG_PRINT_L0("Failed to create a transaction for the db");
|
||||
throw DB_ERROR("Failed to create a transaction for the db");
|
||||
}
|
||||
m_write_txn = &txn;
|
||||
|
||||
BlockchainDB::add_block(blk, block_size, cumulative_difficulty, coins_generated, txs);
|
||||
|
||||
txn.commit();
|
||||
|
||||
m_height++;
|
||||
return m_height - 1;
|
||||
}
|
||||
|
||||
} // namespace cryptonote
|
||||
|
|
|
|||
|
|
@ -32,6 +32,47 @@
|
|||
namespace cryptonote
|
||||
{
|
||||
|
||||
struct txn_safe
|
||||
{
|
||||
txn_safe() : m_txn(NULL) { }
|
||||
~txn_safe()
|
||||
{
|
||||
if(m_txn != NULL)
|
||||
{
|
||||
mdb_txn_abort(m_txn);
|
||||
}
|
||||
}
|
||||
|
||||
void commit(std::string message = "")
|
||||
{
|
||||
if (message.size() == 0)
|
||||
{
|
||||
message = "Failed to commit a transaction to the db";
|
||||
}
|
||||
|
||||
if (mdb_txn_commit(m_txn))
|
||||
{
|
||||
m_txn = NULL;
|
||||
LOG_PRINT_L0(message);
|
||||
throw DB_ERROR(message.c_str());
|
||||
}
|
||||
m_txn = NULL;
|
||||
}
|
||||
|
||||
operator MDB_txn*()
|
||||
{
|
||||
return m_txn;
|
||||
}
|
||||
|
||||
operator MDB_txn**()
|
||||
{
|
||||
return &m_txn;
|
||||
}
|
||||
|
||||
MDB_txn* m_txn;
|
||||
};
|
||||
|
||||
|
||||
class BlockchainLMDB : public BlockchainDB
|
||||
{
|
||||
public:
|
||||
|
|
@ -114,6 +155,13 @@ public:
|
|||
|
||||
virtual bool has_key_image(const crypto::key_image& img);
|
||||
|
||||
virtual uint64_t add_block( const block& blk
|
||||
, const size_t& block_size
|
||||
, const difficulty_type& cumulative_difficulty
|
||||
, const uint64_t& coins_generated
|
||||
, const std::vector<transaction>& txs
|
||||
);
|
||||
|
||||
private:
|
||||
virtual void add_block( const block& blk
|
||||
, const size_t& block_size
|
||||
|
|
@ -121,7 +169,7 @@ private:
|
|||
, const uint64_t& coins_generated
|
||||
);
|
||||
|
||||
virtual void remove_block(const crypto::hash& blk_hash);
|
||||
virtual void remove_block();
|
||||
|
||||
virtual void add_transaction_data(const crypto::hash& blk_hash, const transaction& tx);
|
||||
|
||||
|
|
@ -135,10 +183,11 @@ private:
|
|||
|
||||
virtual void remove_spent_key(const crypto::key_image& k_image);
|
||||
|
||||
void check_open();
|
||||
|
||||
MDB_env* m_env;
|
||||
|
||||
MDB_dbi m_blocks;
|
||||
MDB_dbi m_block_txs;
|
||||
MDB_dbi m_block_hashes;
|
||||
MDB_dbi m_block_sizes;
|
||||
MDB_dbi m_block_diffs;
|
||||
|
|
@ -148,6 +197,11 @@ private:
|
|||
|
||||
MDB_dbi m_spent;
|
||||
MDB_dbi m_utxo;
|
||||
|
||||
bool m_open;
|
||||
uint64_t m_height;
|
||||
std::string m_folder;
|
||||
txn_safe* m_write_txn;
|
||||
};
|
||||
|
||||
} // namespace cryptonote
|
||||
|
|
|
|||
|
|
@ -62,40 +62,22 @@ void BlockchainDB::add_transaction(const crypto::hash& blk_hash, const transacti
|
|||
}
|
||||
|
||||
uint64_t BlockchainDB::add_block( const block& blk
|
||||
, const size_t& block_size
|
||||
, const difficulty_type& cumulative_difficulty
|
||||
, const uint64_t& coins_generated
|
||||
, const std::vector<transaction>& txs
|
||||
)
|
||||
, const size_t& block_size
|
||||
, const difficulty_type& cumulative_difficulty
|
||||
, const uint64_t& coins_generated
|
||||
, const std::vector<transaction>& txs
|
||||
)
|
||||
{
|
||||
try
|
||||
{
|
||||
// call out to subclass implementation to add the block & metadata
|
||||
add_block(blk, block_size, cumulative_difficulty, coins_generated);
|
||||
// call out to subclass implementation to add the block & metadata
|
||||
add_block(blk, block_size, cumulative_difficulty, coins_generated);
|
||||
|
||||
crypto::hash blk_hash = get_block_hash(blk);
|
||||
// call out to add the transactions
|
||||
for (const transaction& tx : txs)
|
||||
{
|
||||
add_transaction(blk_hash, tx);
|
||||
}
|
||||
}
|
||||
// in case any of the add_block process goes awry, undo
|
||||
catch (const std::exception& e)
|
||||
crypto::hash blk_hash = get_block_hash(blk);
|
||||
// call out to add the transactions
|
||||
|
||||
add_transaction(blk_hash, blk.miner_tx);
|
||||
for (const transaction& tx : txs)
|
||||
{
|
||||
LOG_ERROR("Error adding block to db: " << e.what());
|
||||
try
|
||||
{
|
||||
pop_block();
|
||||
}
|
||||
// if undoing goes wrong as well, we need to throw, as blockchain
|
||||
// will be in a bad state
|
||||
catch (const std::exception& e)
|
||||
{
|
||||
LOG_ERROR("Error undoing partially added block: " << e.what());
|
||||
throw;
|
||||
}
|
||||
throw;
|
||||
add_transaction(blk_hash, tx);
|
||||
}
|
||||
|
||||
return height();
|
||||
|
|
@ -105,6 +87,9 @@ void BlockchainDB::pop_block(block& blk, std::vector<transaction>& txs)
|
|||
{
|
||||
blk = get_top_block();
|
||||
|
||||
remove_block();
|
||||
|
||||
remove_transaction(get_transaction_hash(blk.miner_tx));
|
||||
for (const auto& h : blk.tx_hashes)
|
||||
{
|
||||
txs.push_back(get_tx(h));
|
||||
|
|
|
|||
|
|
@ -332,8 +332,8 @@ private:
|
|||
, const uint64_t& coins_generated
|
||||
) = 0;
|
||||
|
||||
// tells the subclass to remove data about a block
|
||||
virtual void remove_block(const crypto::hash& blk_hash) = 0;
|
||||
// tells the subclass to remove data about the top block
|
||||
virtual void remove_block() = 0;
|
||||
|
||||
// tells the subclass to store the transaction and its metadata
|
||||
virtual void add_transaction_data(const crypto::hash& blk_hash, const transaction& tx) = 0;
|
||||
|
|
@ -403,12 +403,14 @@ public:
|
|||
|
||||
|
||||
// adds a block with the given metadata to the top of the blockchain, returns the new height
|
||||
uint64_t add_block( const block& blk
|
||||
, const size_t& block_size
|
||||
, const difficulty_type& cumulative_difficulty
|
||||
, const uint64_t& coins_generated
|
||||
, const std::vector<transaction>& txs
|
||||
);
|
||||
// NOTE: subclass implementations of this (or the functions it calls) need
|
||||
// to handle undoing any partially-added blocks in the event of a failure.
|
||||
virtual uint64_t add_block( const block& blk
|
||||
, const size_t& block_size
|
||||
, const difficulty_type& cumulative_difficulty
|
||||
, const uint64_t& coins_generated
|
||||
, const std::vector<transaction>& txs
|
||||
);
|
||||
|
||||
// return true if a block with hash <h> exists in the blockchain
|
||||
virtual bool block_exists(const crypto::hash& h) = 0;
|
||||
|
|
|
|||
Loading…
Reference in New Issue