// Copyright (c) 2014-2016, The Monero Project // All rights reserved. // // Redistribution and use in source and binary forms, with or without modification, are // permitted provided that the following conditions are met: // // 1. Redistributions of source code must retain the above copyright notice, this list of // conditions and the following disclaimer. // // 2. Redistributions in binary form must reproduce the above copyright notice, this list // of conditions and the following disclaimer in the documentation and/or other // materials provided with the distribution. // // 3. Neither the name of the copyright holder nor the names of its contributors may be // used to endorse or promote products derived from this software without specific // prior written permission. // // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF // MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL // THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, // STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF // THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. #include "db_lmdb.h" #include #include #include #include // std::unique_ptr #include // memcpy #include #include "cryptonote_core/cryptonote_format_utils.h" #include "crypto/crypto.h" #include "profile_tools.h" #if defined(__i386) || defined(__x86_64) #define MISALIGNED_OK 1 #endif using epee::string_tools::pod_to_hex; // Increase when the DB changes in a non backward compatible way, and there // is no automatic conversion, so that a full resync is needed. #define VERSION 1 namespace { struct pre_rct_output_data_t { crypto::public_key pubkey; //!< the output's public key (for spend verification) uint64_t unlock_time; //!< the output's unlock time (or height) uint64_t height; //!< the height of the block which created the output }; template inline void throw0(const T &e) { LOG_PRINT_L0(e.what()); throw e; } template inline void throw1(const T &e) { LOG_PRINT_L1(e.what()); throw e; } #define MDB_val_set(var, val) MDB_val var = {sizeof(val), (void *)&val} template struct MDB_val_copy: public MDB_val { MDB_val_copy(const T &t) : t_copy(t) { mv_size = sizeof (T); mv_data = &t_copy; } private: T t_copy; }; template<> struct MDB_val_copy: public MDB_val { MDB_val_copy(const cryptonote::blobdata &bd) : data(new char[bd.size()]) { memcpy(data.get(), bd.data(), bd.size()); mv_size = bd.size(); mv_data = data.get(); } private: std::unique_ptr data; }; template<> struct MDB_val_copy: public MDB_val { MDB_val_copy(const char *s): size(strlen(s)+1), // include the NUL, makes it easier for compares data(new char[size]) { mv_size = size; mv_data = data.get(); memcpy(mv_data, s, size); } private: size_t size; std::unique_ptr data; }; int compare_uint64(const MDB_val *a, const MDB_val *b) { const uint64_t va = *(const uint64_t *)a->mv_data; const uint64_t vb = *(const uint64_t *)b->mv_data; return (va < vb) ? -1 : va > vb; } int compare_hash32(const MDB_val *a, const MDB_val *b) { uint32_t *va = (uint32_t*) a->mv_data; uint32_t *vb = (uint32_t*) b->mv_data; for (int n = 7; n >= 0; n--) { if (va[n] == vb[n]) continue; return va[n] < vb[n] ? -1 : 1; } return 0; } int compare_string(const MDB_val *a, const MDB_val *b) { const char *va = (const char*) a->mv_data; const char *vb = (const char*) b->mv_data; return strcmp(va, vb); } /* DB schema: * * Table Key Data * ----- --- ---- * blocks block ID block blob * block_heights block hash block height * block_info block ID {block metadata} * * txs txn ID txn blob * tx_indices txn hash {txn ID, metadata} * tx_outputs txn ID [txn amount output indices] * * output_txs output ID {txn hash, local index} * output_amounts amount [{amount output index, metadata}...] * * spent_keys input hash - * * Note: where the data items are of uniform size, DUPFIXED tables have * been used to save space. In most of these cases, a dummy "zerokval" * key is used when accessing the table; the Key listed above will be * attached as a prefix on the Data to serve as the DUPSORT key. * (DUPFIXED saves 8 bytes per record.) * * The output_amounts table doesn't use a dummy key, but uses DUPSORT. */ const char* const LMDB_BLOCKS = "blocks"; const char* const LMDB_BLOCK_HEIGHTS = "block_heights"; const char* const LMDB_BLOCK_INFO = "block_info"; const char* const LMDB_TXS = "txs"; const char* const LMDB_TX_INDICES = "tx_indices"; const char* const LMDB_TX_OUTPUTS = "tx_outputs"; const char* const LMDB_OUTPUT_TXS = "output_txs"; const char* const LMDB_OUTPUT_AMOUNTS = "output_amounts"; const char* const LMDB_SPENT_KEYS = "spent_keys"; const char* const LMDB_HF_STARTING_HEIGHTS = "hf_starting_heights"; const char* const LMDB_HF_VERSIONS = "hf_versions"; const char* const LMDB_PROPERTIES = "properties"; const char zerokey[8] = {0}; const MDB_val zerokval = { sizeof(zerokey), (void *)zerokey }; const std::string lmdb_error(const std::string& error_string, int mdb_res) { const std::string full_string = error_string + mdb_strerror(mdb_res); return full_string; } inline void lmdb_db_open(MDB_txn* txn, const char* name, int flags, MDB_dbi& dbi, const std::string& error_string) { if (auto res = mdb_dbi_open(txn, name, flags, &dbi)) throw0(cryptonote::DB_OPEN_FAILURE(lmdb_error(error_string + " : ", res).c_str())); } } // anonymous namespace #define CURSOR(name) \ if (!m_cur_ ## name) { \ int result = mdb_cursor_open(*m_write_txn, m_ ## name, &m_cur_ ## name); \ if (result) \ throw0(DB_ERROR(lmdb_error("Failed to open cursor: ", result).c_str())); \ } #define RCURSOR(name) \ if (!m_cur_ ## name) { \ int result = mdb_cursor_open(m_txn, m_ ## name, (MDB_cursor **)&m_cur_ ## name); \ if (result) \ throw0(DB_ERROR(lmdb_error("Failed to open cursor: ", result).c_str())); \ if (m_cursors != &m_wcursors) \ m_tinfo->m_ti_rflags.m_rf_ ## name = true; \ } else if (m_cursors != &m_wcursors && !m_tinfo->m_ti_rflags.m_rf_ ## name) { \ int result = mdb_cursor_renew(m_txn, m_cur_ ## name); \ if (result) \ throw0(DB_ERROR(lmdb_error("Failed to renew cursor: ", result).c_str())); \ m_tinfo->m_ti_rflags.m_rf_ ## name = true; \ } namespace cryptonote { typedef struct mdb_block_info { uint64_t bi_height; uint64_t bi_timestamp; uint64_t bi_coins; uint64_t bi_size; // a size_t really but we need 32-bit compat difficulty_type bi_diff; crypto::hash bi_hash; } mdb_block_info; typedef struct blk_height { crypto::hash bh_hash; uint64_t bh_height; } blk_height; typedef struct txindex { crypto::hash key; tx_data_t data; } txindex; typedef struct pre_rct_outkey { uint64_t amount_index; uint64_t output_id; pre_rct_output_data_t data; } pre_rct_outkey; typedef struct outkey { uint64_t amount_index; uint64_t output_id; output_data_t data; } outkey; typedef struct outtx { uint64_t output_id; crypto::hash tx_hash; uint64_t local_index; } outtx; std::atomic mdb_txn_safe::num_active_txns{0}; std::atomic_flag mdb_txn_safe::creation_gate = ATOMIC_FLAG_INIT; mdb_threadinfo::~mdb_threadinfo() { MDB_cursor **cur = &m_ti_rcursors.m_txc_blocks; unsigned i; for (i=0; im_ti_rtxn); memset(&m_tinfo->m_ti_rflags, 0, sizeof(m_tinfo->m_ti_rflags)); } else if (m_txn != nullptr) { if (m_batch_txn) // this is a batch txn and should have been handled before this point for safety { LOG_PRINT_L0("WARNING: mdb_txn_safe: m_txn is a batch txn and it's not NULL in destructor - calling mdb_txn_abort()"); } else { // Example of when this occurs: a lookup fails, so a read-only txn is // aborted through this destructor. However, successful read-only txns // ideally should have been committed when done and not end up here. // // NOTE: not sure if this is ever reached for a non-batch write // transaction, but it's probably not ideal if it did. LOG_PRINT_L3("mdb_txn_safe: m_txn not NULL in destructor - calling mdb_txn_abort()"); } mdb_txn_abort(m_txn); } num_active_txns--; } void mdb_txn_safe::commit(std::string message) { if (message.size() == 0) { message = "Failed to commit a transaction to the db"; } if (auto result = mdb_txn_commit(m_txn)) { m_txn = nullptr; throw0(DB_ERROR(lmdb_error(message + ": ", result).c_str())); } m_txn = nullptr; } void mdb_txn_safe::abort() { LOG_PRINT_L3("mdb_txn_safe: abort()"); if(m_txn != nullptr) { mdb_txn_abort(m_txn); m_txn = nullptr; } else { LOG_PRINT_L0("WARNING: mdb_txn_safe: abort() called, but m_txn is NULL"); } } uint64_t mdb_txn_safe::num_active_tx() const { return num_active_txns; } void mdb_txn_safe::prevent_new_txns() { while (creation_gate.test_and_set()); } void mdb_txn_safe::wait_no_active_txns() { while (num_active_txns > 0); } void mdb_txn_safe::allow_new_txns() { creation_gate.clear(); } void BlockchainLMDB::do_resize(uint64_t increase_size) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); CRITICAL_REGION_LOCAL(m_synchronization_lock); const uint64_t add_size = 1LL << 30; // check disk capacity try { boost::filesystem::path path(m_folder); boost::filesystem::space_info si = boost::filesystem::space(path); if(si.available < add_size) { LOG_PRINT_RED_L0("!! WARNING: Insufficient free space to extend database !!: " << si.available / 1LL << 20L); return; } } catch(...) { // print something but proceed. LOG_PRINT_YELLOW("Unable to query free disk space.", LOG_LEVEL_0); } MDB_envinfo mei; mdb_env_info(m_env, &mei); MDB_stat mst; mdb_env_stat(m_env, &mst); // add 1Gb per resize, instead of doing a percentage increase uint64_t new_mapsize = (double) mei.me_mapsize + add_size; // If given, use increase_size intead of above way of resizing. // This is currently used for increasing by an estimated size at start of new // batch txn. if (increase_size > 0) new_mapsize = mei.me_mapsize + increase_size; new_mapsize += (new_mapsize % mst.ms_psize); mdb_txn_safe::prevent_new_txns(); if (m_write_txn != nullptr) { if (m_batch_active) { throw0(DB_ERROR("lmdb resizing not yet supported when batch transactions enabled!")); } else { throw0(DB_ERROR("attempting resize with write transaction in progress, this should not happen!")); } } mdb_txn_safe::wait_no_active_txns(); int result = mdb_env_set_mapsize(m_env, new_mapsize); if (result) throw0(DB_ERROR(lmdb_error("Failed to set new mapsize: ", result).c_str())); LOG_PRINT_GREEN("LMDB Mapsize increased." << " Old: " << mei.me_mapsize / (1024 * 1024) << "MiB" << ", New: " << new_mapsize / (1024 * 1024) << "MiB", LOG_LEVEL_0); mdb_txn_safe::allow_new_txns(); } // threshold_size is used for batch transactions bool BlockchainLMDB::need_resize(uint64_t threshold_size) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); #if defined(ENABLE_AUTO_RESIZE) MDB_envinfo mei; mdb_env_info(m_env, &mei); MDB_stat mst; mdb_env_stat(m_env, &mst); // size_used doesn't include data yet to be committed, which can be // significant size during batch transactions. For that, we estimate the size // needed at the beginning of the batch transaction and pass in the // additional size needed. uint64_t size_used = mst.ms_psize * mei.me_last_pgno; LOG_PRINT_L1("DB map size: " << mei.me_mapsize); LOG_PRINT_L1("Space used: " << size_used); LOG_PRINT_L1("Space remaining: " << mei.me_mapsize - size_used); LOG_PRINT_L1("Size threshold: " << threshold_size); float resize_percent_old = RESIZE_PERCENT; LOG_PRINT_L1(boost::format("Percent used: %.04f Percent threshold: %.04f") % ((double)size_used/mei.me_mapsize) % resize_percent_old); if (threshold_size > 0) { if (mei.me_mapsize - size_used < threshold_size) { LOG_PRINT_L1("Threshold met (size-based)"); return true; } else return false; } std::mt19937 engine(std::random_device{}()); std::uniform_real_distribution fdis(0.6, 0.9); double resize_percent = fdis(engine); if ((double)size_used / mei.me_mapsize > resize_percent) { LOG_PRINT_L1("Threshold met (percent-based)"); return true; } return false; #else return false; #endif } void BlockchainLMDB::check_and_resize_for_batch(uint64_t batch_num_blocks) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); LOG_PRINT_L1("[" << __func__ << "] " << "checking DB size"); const uint64_t min_increase_size = 512 * (1 << 20); uint64_t threshold_size = 0; uint64_t increase_size = 0; if (batch_num_blocks > 0) { threshold_size = get_estimated_batch_size(batch_num_blocks); LOG_PRINT_L1("calculated batch size: " << threshold_size); // The increased DB size could be a multiple of threshold_size, a fixed // size increase (> threshold_size), or other variations. // // Currently we use the greater of threshold size and a minimum size. The // minimum size increase is used to avoid frequent resizes when the batch // size is set to a very small numbers of blocks. increase_size = (threshold_size > min_increase_size) ? threshold_size : min_increase_size; LOG_PRINT_L1("increase size: " << increase_size); } // if threshold_size is 0 (i.e. number of blocks for batch not passed in), it // will fall back to the percent-based threshold check instead of the // size-based check if (need_resize(threshold_size)) { LOG_PRINT_L0("[batch] DB resize needed"); do_resize(increase_size); } } uint64_t BlockchainLMDB::get_estimated_batch_size(uint64_t batch_num_blocks) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); uint64_t threshold_size = 0; // batch size estimate * batch safety factor = final size estimate // Takes into account "reasonable" block size increases in batch. float batch_safety_factor = 1.7f; float batch_fudge_factor = batch_safety_factor * batch_num_blocks; // estimate of stored block expanded from raw block, including denormalization and db overhead. // Note that this probably doesn't grow linearly with block size. float db_expand_factor = 4.5f; uint64_t num_prev_blocks = 500; // For resizing purposes, allow for at least 4k average block size. uint64_t min_block_size = 4 * 1024; uint64_t block_stop = 0; if (m_height > 1) block_stop = m_height - 1; uint64_t block_start = 0; if (block_stop >= num_prev_blocks) block_start = block_stop - num_prev_blocks + 1; uint32_t num_blocks_used = 0; uint64_t total_block_size = 0; LOG_PRINT_L1("[" << __func__ << "] " << "m_height: " << m_height << " block_start: " << block_start << " block_stop: " << block_stop); size_t avg_block_size = 0; if (m_height == 0) { LOG_PRINT_L1("No existing blocks to check for average block size"); } else if (m_cum_count) { avg_block_size = m_cum_size / m_cum_count; LOG_PRINT_L1("average block size across recent " << m_cum_count << " blocks: " << avg_block_size); m_cum_size = 0; m_cum_count = 0; } else { for (uint64_t block_num = block_start; block_num <= block_stop; ++block_num) { uint32_t block_size = get_block_size(block_num); total_block_size += block_size; // Track number of blocks being totalled here instead of assuming, in case // some blocks were to be skipped for being outliers. ++num_blocks_used; } avg_block_size = total_block_size / num_blocks_used; LOG_PRINT_L1("average block size across recent " << num_blocks_used << " blocks: " << avg_block_size); } if (avg_block_size < min_block_size) avg_block_size = min_block_size; LOG_PRINT_L1("estimated average block size for batch: " << avg_block_size); // bigger safety margin on smaller block sizes if (batch_fudge_factor < 5000.0) batch_fudge_factor = 5000.0; threshold_size = avg_block_size * db_expand_factor * batch_fudge_factor; return threshold_size; } void BlockchainLMDB::add_block(const block& blk, const size_t& block_size, const difficulty_type& cumulative_difficulty, const uint64_t& coins_generated, const crypto::hash& blk_hash) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); mdb_txn_cursors *m_cursors = &m_wcursors; CURSOR(block_heights) blk_height bh = {blk_hash, m_height}; MDB_val_set(val_h, bh); if (mdb_cursor_get(m_cur_block_heights, (MDB_val *)&zerokval, &val_h, MDB_GET_BOTH) == 0) throw1(BLOCK_EXISTS("Attempting to add block that's already in the db")); if (m_height > 0) { MDB_val_set(parent_key, blk.prev_id); int result = mdb_cursor_get(m_cur_block_heights, (MDB_val *)&zerokval, &parent_key, MDB_GET_BOTH); if (result) { LOG_PRINT_L3("m_height: " << m_height); LOG_PRINT_L3("parent_key: " << blk.prev_id); throw0(DB_ERROR(lmdb_error("Failed to get top block hash to check for new block's parent: ", result).c_str())); } blk_height *prev = (blk_height *)parent_key.mv_data; if (prev->bh_height != m_height - 1) throw0(BLOCK_PARENT_DNE("Top block is not new block's parent")); } int result = 0; MDB_val_set(key, m_height); CURSOR(blocks) CURSOR(block_info) MDB_val_copy blob(block_to_blob(blk)); result = mdb_cursor_put(m_cur_blocks, &key, &blob, MDB_APPEND); if (result) throw0(DB_ERROR(lmdb_error("Failed to add block blob to db transaction: ", result).c_str())); mdb_block_info bi; bi.bi_height = m_height; bi.bi_timestamp = blk.timestamp; bi.bi_coins = coins_generated; bi.bi_size = block_size; bi.bi_diff = cumulative_difficulty; bi.bi_hash = blk_hash; MDB_val_set(val, bi); result = mdb_cursor_put(m_cur_block_info, (MDB_val *)&zerokval, &val, MDB_APPENDDUP); if (result) throw0(DB_ERROR(lmdb_error("Failed to add block info to db transaction: ", result).c_str())); result = mdb_cursor_put(m_cur_block_heights, (MDB_val *)&zerokval, &val_h, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to add block height by hash to db transaction: ", result).c_str())); m_cum_size += block_size; m_cum_count++; } void BlockchainLMDB::remove_block() { int result; LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); if (m_height == 0) throw0(BLOCK_DNE ("Attempting to remove block from an empty blockchain")); mdb_txn_cursors *m_cursors = &m_wcursors; CURSOR(block_info) CURSOR(block_heights) CURSOR(blocks) MDB_val_copy k(m_height - 1); MDB_val h = k; if ((result = mdb_cursor_get(m_cur_block_info, (MDB_val *)&zerokval, &h, MDB_GET_BOTH))) throw1(BLOCK_DNE(lmdb_error("Attempting to remove block that's not in the db: ", result).c_str())); // must use h now; deleting from m_block_info will invalidate it mdb_block_info *bi = (mdb_block_info *)h.mv_data; blk_height bh = {bi->bi_hash, 0}; h.mv_data = (void *)&bh; h.mv_size = sizeof(bh); if ((result = mdb_cursor_get(m_cur_block_heights, (MDB_val *)&zerokval, &h, MDB_GET_BOTH))) throw1(DB_ERROR(lmdb_error("Failed to locate block height by hash for removal: ", result).c_str())); if ((result = mdb_cursor_del(m_cur_block_heights, 0))) throw1(DB_ERROR(lmdb_error("Failed to add removal of block height by hash to db transaction: ", result).c_str())); if ((result = mdb_cursor_get(m_cur_blocks, &k, NULL, MDB_SET))) throw1(DB_ERROR(lmdb_error("Failed to locate block for removal: ", result).c_str())); if ((result = mdb_cursor_del(m_cur_blocks, 0))) throw1(DB_ERROR(lmdb_error("Failed to add removal of block to db transaction: ", result).c_str())); if ((result = mdb_cursor_del(m_cur_block_info, 0))) throw1(DB_ERROR(lmdb_error("Failed to add removal of block info to db transaction: ", result).c_str())); } uint64_t BlockchainLMDB::add_transaction_data(const crypto::hash& blk_hash, const transaction& tx, const crypto::hash& tx_hash) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); mdb_txn_cursors *m_cursors = &m_wcursors; int result; uint64_t tx_id = m_num_txs; CURSOR(txs) CURSOR(tx_indices) MDB_val_set(val_tx_id, tx_id); MDB_val_set(val_h, tx_hash); result = mdb_cursor_get(m_cur_tx_indices, (MDB_val *)&zerokval, &val_h, MDB_GET_BOTH); if (result == 0) { txindex *tip = (txindex *)val_h.mv_data; throw1(TX_EXISTS(std::string("Attempting to add transaction that's already in the db (tx id ").append(boost::lexical_cast(tip->data.tx_id)).append(")").c_str())); } else if (result != MDB_NOTFOUND) { throw1(DB_ERROR(lmdb_error(std::string("Error checking if tx index exists for tx hash ") + epee::string_tools::pod_to_hex(tx_hash) + ": ", result).c_str())); } txindex ti; ti.key = tx_hash; ti.data.tx_id = tx_id; ti.data.unlock_time = tx.unlock_time; ti.data.block_id = m_height; // we don't need blk_hash since we know m_height val_h.mv_size = sizeof(ti); val_h.mv_data = (void *)&ti; result = mdb_cursor_put(m_cur_tx_indices, (MDB_val *)&zerokval, &val_h, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to add tx data to db transaction: ", result).c_str())); MDB_val_copy blob(tx_to_blob(tx)); result = mdb_cursor_put(m_cur_txs, &val_tx_id, &blob, MDB_APPEND); if (result) throw0(DB_ERROR(lmdb_error("Failed to add tx blob to db transaction: ", result).c_str())); m_num_txs++; return tx_id; } // TODO: compare pros and cons of looking up the tx hash's tx index once and // passing it in to functions like this void BlockchainLMDB::remove_transaction_data(const crypto::hash& tx_hash, const transaction& tx) { int result; LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); mdb_txn_cursors *m_cursors = &m_wcursors; CURSOR(tx_indices) CURSOR(txs) CURSOR(tx_outputs) MDB_val_set(val_h, tx_hash); if (mdb_cursor_get(m_cur_tx_indices, (MDB_val *)&zerokval, &val_h, MDB_GET_BOTH)) throw1(TX_DNE("Attempting to remove transaction that isn't in the db")); txindex *tip = (txindex *)val_h.mv_data; MDB_val_set(val_tx_id, tip->data.tx_id); if ((result = mdb_cursor_get(m_cur_txs, &val_tx_id, NULL, MDB_SET))) throw1(DB_ERROR(lmdb_error("Failed to locate tx for removal: ", result).c_str())); result = mdb_cursor_del(m_cur_txs, 0); if (result) throw1(DB_ERROR(lmdb_error("Failed to add removal of tx to db transaction: ", result).c_str())); remove_tx_outputs(tip->data.tx_id, tx); result = mdb_cursor_get(m_cur_tx_outputs, &val_tx_id, NULL, MDB_SET); if (result == MDB_NOTFOUND) LOG_PRINT_L1("tx has no outputs to remove: " << tx_hash); else if (result) throw1(DB_ERROR(lmdb_error("Failed to locate tx outputs for removal: ", result).c_str())); if (!result) { result = mdb_cursor_del(m_cur_tx_outputs, 0); if (result) throw1(DB_ERROR(lmdb_error("Failed to add removal of tx outputs to db transaction: ", result).c_str())); } // Don't delete the tx_indices entry until the end, after we're done with val_tx_id if (mdb_cursor_del(m_cur_tx_indices, 0)) throw1(DB_ERROR("Failed to add removal of tx index to db transaction")); m_num_txs--; } uint64_t BlockchainLMDB::add_output(const crypto::hash& tx_hash, const tx_out& tx_output, const uint64_t& local_index, const uint64_t unlock_time, const rct::key *commitment) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); mdb_txn_cursors *m_cursors = &m_wcursors; int result = 0; CURSOR(output_txs) CURSOR(output_amounts) if (tx_output.target.type() != typeid(txout_to_key)) throw0(DB_ERROR("Wrong output type: expected txout_to_key")); if (tx_output.amount == 0 && !commitment) throw0(DB_ERROR("RCT output without commitment")); outtx ot = {m_num_outputs, tx_hash, local_index}; MDB_val_set(vot, ot); result = mdb_cursor_put(m_cur_output_txs, (MDB_val *)&zerokval, &vot, MDB_APPENDDUP); if (result) throw0(DB_ERROR(lmdb_error("Failed to add output tx hash to db transaction: ", result).c_str())); outkey ok; MDB_val data; MDB_val_copy val_amount(tx_output.amount); result = mdb_cursor_get(m_cur_output_amounts, &val_amount, &data, MDB_SET); if (!result) { mdb_size_t num_elems = 0; result = mdb_cursor_count(m_cur_output_amounts, &num_elems); if (result) throw0(DB_ERROR(std::string("Failed to get number of outputs for amount: ").append(mdb_strerror(result)).c_str())); ok.amount_index = num_elems; } else if (result != MDB_NOTFOUND) throw0(DB_ERROR(lmdb_error("Failed to get output amount in db transaction: ", result).c_str())); else ok.amount_index = 0; ok.output_id = m_num_outputs; ok.data.pubkey = boost::get < txout_to_key > (tx_output.target).key; ok.data.unlock_time = unlock_time; ok.data.height = m_height; if (tx_output.amount == 0) { ok.data.commitment = *commitment; data.mv_size = sizeof(ok); } else { data.mv_size = sizeof(pre_rct_outkey); } data.mv_data = &ok; if ((result = mdb_cursor_put(m_cur_output_amounts, &val_amount, &data, MDB_APPENDDUP))) throw0(DB_ERROR(lmdb_error("Failed to add output pubkey to db transaction: ", result).c_str())); m_num_outputs++; return ok.amount_index; } void BlockchainLMDB::add_tx_amount_output_indices(const uint64_t tx_id, const std::vector& amount_output_indices) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); mdb_txn_cursors *m_cursors = &m_wcursors; CURSOR(tx_outputs) int result = 0; int num_outputs = amount_output_indices.size(); MDB_val_set(k_tx_id, tx_id); MDB_val v; v.mv_data = (void *)amount_output_indices.data(); v.mv_size = sizeof(uint64_t) * num_outputs; // LOG_PRINT_L1("tx_outputs[tx_hash] size: " << v.mv_size); result = mdb_cursor_put(m_cur_tx_outputs, &k_tx_id, &v, MDB_APPEND); if (result) throw0(DB_ERROR(std::string("Failed to add to db transaction: ").append(mdb_strerror(result)).c_str())); } void BlockchainLMDB::remove_tx_outputs(const uint64_t tx_id, const transaction& tx) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); std::vector amount_output_indices = get_tx_amount_output_indices(tx_id); if (amount_output_indices.empty()) { if (tx.vout.empty()) LOG_PRINT_L2("tx has no outputs, so no output indices"); else throw0(DB_ERROR("tx has outputs, but no output indices found")); } bool is_miner_tx = tx.vin.size() == 1 && tx.vin[0].type() == typeid(txin_gen); for (uint64_t i = tx.vout.size(); i > 0; --i) { const tx_out tx_output = tx.vout[i-1]; uint64_t amount = is_miner_tx && tx.version >= 2 ? 0 : tx_output.amount; remove_output(amount, amount_output_indices[i-1]); } } void BlockchainLMDB::remove_output(const uint64_t amount, const uint64_t& out_index) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); mdb_txn_cursors *m_cursors = &m_wcursors; CURSOR(output_amounts); CURSOR(output_txs); MDB_val_set(k, amount); MDB_val_set(v, out_index); auto result = mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_GET_BOTH); if (result == MDB_NOTFOUND) throw1(OUTPUT_DNE("Attempting to get an output index by amount and amount index, but amount not found")); else if (result) throw0(DB_ERROR(lmdb_error("DB error attempting to get an output", result).c_str())); outkey *ok = (outkey *)v.mv_data; MDB_val_set(otxk, ok->output_id); result = mdb_cursor_get(m_cur_output_txs, (MDB_val *)&zerokval, &otxk, MDB_GET_BOTH); if (result == MDB_NOTFOUND) { LOG_PRINT_L0("Unexpected: global output index not found in m_output_txs"); } else if (result) { throw1(DB_ERROR(lmdb_error("Error adding removal of output tx to db transaction", result).c_str())); } result = mdb_cursor_del(m_cur_output_txs, 0); if (result) throw0(DB_ERROR(lmdb_error(std::string("Error deleting output index ").append(boost::lexical_cast(out_index).append(": ")).c_str(), result).c_str())); // now delete the amount result = mdb_cursor_del(m_cur_output_amounts, 0); if (result) throw0(DB_ERROR(lmdb_error(std::string("Error deleting amount for output index ").append(boost::lexical_cast(out_index).append(": ")).c_str(), result).c_str())); m_num_outputs--; } void BlockchainLMDB::add_spent_key(const crypto::key_image& k_image) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); mdb_txn_cursors *m_cursors = &m_wcursors; CURSOR(spent_keys) MDB_val k = {sizeof(k_image), (void *)&k_image}; if (auto result = mdb_cursor_put(m_cur_spent_keys, (MDB_val *)&zerokval, &k, MDB_NODUPDATA)) { if (result == MDB_KEYEXIST) throw1(KEY_IMAGE_EXISTS("Attempting to add spent key image that's already in the db")); else throw1(DB_ERROR(lmdb_error("Error adding spent key image to db transaction: ", result).c_str())); } } void BlockchainLMDB::remove_spent_key(const crypto::key_image& k_image) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); mdb_txn_cursors *m_cursors = &m_wcursors; CURSOR(spent_keys) MDB_val k = {sizeof(k_image), (void *)&k_image}; auto result = mdb_cursor_get(m_cur_spent_keys, (MDB_val *)&zerokval, &k, MDB_GET_BOTH); if (result != 0 && result != MDB_NOTFOUND) throw1(DB_ERROR(lmdb_error("Error finding spent key to remove", result).c_str())); if (!result) { result = mdb_cursor_del(m_cur_spent_keys, 0); if (result) throw1(DB_ERROR(lmdb_error("Error adding removal of key image to db transaction", result).c_str())); } } blobdata BlockchainLMDB::output_to_blob(const tx_out& output) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); blobdata b; if (!t_serializable_object_to_blob(output, b)) throw1(DB_ERROR("Error serializing output to blob")); return b; } tx_out BlockchainLMDB::output_from_blob(const blobdata& blob) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); std::stringstream ss; ss << blob; binary_archive ba(ss); tx_out o; if (!(::serialization::serialize(ba, o))) throw1(DB_ERROR("Error deserializing tx output blob")); return o; } void BlockchainLMDB::check_open() const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); if (!m_open) throw0(DB_ERROR("DB operation attempted on a not-open DB instance")); } BlockchainLMDB::~BlockchainLMDB() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); // batch transaction shouldn't be active at this point. If it is, consider it aborted. if (m_batch_active) batch_abort(); if (m_open) close(); } BlockchainLMDB::BlockchainLMDB(bool batch_transactions) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); // initialize folder to something "safe" just in case // someone accidentally misuses this class... m_folder = "thishsouldnotexistbecauseitisgibberish"; m_open = false; m_batch_transactions = batch_transactions; m_write_txn = nullptr; m_write_batch_txn = nullptr; m_batch_active = false; m_height = 0; m_cum_size = 0; m_cum_count = 0; m_hardfork = nullptr; } void BlockchainLMDB::open(const std::string& filename, const int mdb_flags) { int result; LOG_PRINT_L3("BlockchainLMDB::" << __func__); if (m_open) throw0(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)) throw0(DB_OPEN_FAILURE("LMDB needs a directory path, but a file was passed")); } else { if (!boost::filesystem::create_directories(direc)) throw0(DB_OPEN_FAILURE(std::string("Failed to create directory ").append(filename).c_str())); } // check for existing LMDB files in base directory boost::filesystem::path old_files = direc.parent_path(); if (boost::filesystem::exists(old_files / "data.mdb") || boost::filesystem::exists(old_files / "lock.mdb")) { LOG_PRINT_L0("Found existing LMDB files in " << old_files.string()); LOG_PRINT_L0("Move data.mdb and/or lock.mdb to " << filename << ", or delete them, and then restart"); throw DB_ERROR("Database could not be opened"); } m_folder = filename; // set up lmdb environment if ((result = mdb_env_create(&m_env))) throw0(DB_ERROR(lmdb_error("Failed to create lmdb environment: ", result).c_str())); if ((result = mdb_env_set_maxdbs(m_env, 20))) throw0(DB_ERROR(lmdb_error("Failed to set max number of dbs: ", result).c_str())); size_t mapsize = DEFAULT_MAPSIZE; if (auto result = mdb_env_open(m_env, filename.c_str(), mdb_flags, 0644)) throw0(DB_ERROR(lmdb_error("Failed to open lmdb environment: ", result).c_str())); MDB_envinfo mei; mdb_env_info(m_env, &mei); uint64_t cur_mapsize = (double)mei.me_mapsize; if (cur_mapsize < mapsize) { if (auto result = mdb_env_set_mapsize(m_env, mapsize)) throw0(DB_ERROR(lmdb_error("Failed to set max memory map size: ", result).c_str())); mdb_env_info(m_env, &mei); cur_mapsize = (double)mei.me_mapsize; LOG_PRINT_L1("LMDB memory map size: " << cur_mapsize); } if (need_resize()) { LOG_PRINT_L0("LMDB memory map needs resized, doing that now."); do_resize(); } int txn_flags = 0; if (mdb_flags & MDB_RDONLY) txn_flags |= MDB_RDONLY; // get a read/write MDB_txn, depending on mdb_flags mdb_txn_safe txn; if (auto mdb_res = mdb_txn_begin(m_env, NULL, txn_flags, txn)) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", mdb_res).c_str())); // open necessary databases, and set properties as needed // uses macros to avoid having to change things too many places lmdb_db_open(txn, LMDB_BLOCKS, MDB_INTEGERKEY | MDB_CREATE, m_blocks, "Failed to open db handle for m_blocks"); lmdb_db_open(txn, LMDB_BLOCK_INFO, MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_block_info, "Failed to open db handle for m_block_info"); lmdb_db_open(txn, LMDB_BLOCK_HEIGHTS, MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_block_heights, "Failed to open db handle for m_block_heights"); lmdb_db_open(txn, LMDB_TXS, MDB_INTEGERKEY | MDB_CREATE, m_txs, "Failed to open db handle for m_txs"); lmdb_db_open(txn, LMDB_TX_INDICES, MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_tx_indices, "Failed to open db handle for m_tx_indices"); lmdb_db_open(txn, LMDB_TX_OUTPUTS, MDB_INTEGERKEY | MDB_CREATE, m_tx_outputs, "Failed to open db handle for m_tx_outputs"); lmdb_db_open(txn, LMDB_OUTPUT_TXS, MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_output_txs, "Failed to open db handle for m_output_txs"); lmdb_db_open(txn, LMDB_OUTPUT_AMOUNTS, MDB_INTEGERKEY | MDB_DUPSORT | MDB_DUPFIXED | MDB_CREATE, m_output_amounts, "Failed to open db handle for m_output_amounts"); lmdb_db_open(txn, LMDB_SPENT_KEYS, MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_spent_keys, "Failed to open db handle for m_spent_keys"); // this subdb is dropped on sight, so it may not be present when we open the DB. // Since we use MDB_CREATE, we'll get an exception if we open read-only and it does not exist. // So we don't open for read-only, and also not drop below. It is not used elsewhere. if (!(mdb_flags & MDB_RDONLY)) lmdb_db_open(txn, LMDB_HF_STARTING_HEIGHTS, MDB_CREATE, m_hf_starting_heights, "Failed to open db handle for m_hf_starting_heights"); lmdb_db_open(txn, LMDB_HF_VERSIONS, MDB_INTEGERKEY | MDB_CREATE, m_hf_versions, "Failed to open db handle for m_hf_versions"); lmdb_db_open(txn, LMDB_PROPERTIES, MDB_CREATE, m_properties, "Failed to open db handle for m_properties"); mdb_set_dupsort(txn, m_spent_keys, compare_hash32); mdb_set_dupsort(txn, m_block_heights, compare_hash32); mdb_set_dupsort(txn, m_tx_indices, compare_hash32); mdb_set_dupsort(txn, m_output_amounts, compare_uint64); mdb_set_dupsort(txn, m_output_txs, compare_uint64); mdb_set_dupsort(txn, m_block_info, compare_uint64); mdb_set_compare(txn, m_properties, compare_string); if (!(mdb_flags & MDB_RDONLY)) { result = mdb_drop(txn, m_hf_starting_heights, 1); if (result) throw0(DB_ERROR(lmdb_error("Failed to drop m_hf_starting_heights: ", result).c_str())); } // get and keep current height MDB_stat db_stats; if ((result = mdb_stat(txn, m_blocks, &db_stats))) throw0(DB_ERROR(lmdb_error("Failed to query m_blocks: ", result).c_str())); LOG_PRINT_L2("Setting m_height to: " << db_stats.ms_entries); m_height = db_stats.ms_entries; // get and keep current number of txs if ((result = mdb_stat(txn, m_txs, &db_stats))) throw0(DB_ERROR(lmdb_error("Failed to query m_txs: ", result).c_str())); m_num_txs = db_stats.ms_entries; // get and keep current number of outputs if ((result = mdb_stat(txn, m_output_txs, &db_stats))) throw0(DB_ERROR(lmdb_error("Failed to query m_output_txs: ", result).c_str())); m_num_outputs = db_stats.ms_entries; bool compatible = true; MDB_val_copy k("version"); MDB_val v; auto get_result = mdb_get(txn, m_properties, &k, &v); if(get_result == MDB_SUCCESS) { if (*(const uint32_t*)v.mv_data > VERSION) { LOG_PRINT_RED_L0("Existing lmdb database was made by a later version. We don't know how it will change yet."); compatible = false; } #if VERSION > 0 else if (*(const uint32_t*)v.mv_data < VERSION) { // Note that there was a schema change within version 0 as well. // See commit e5d2680094ee15889934fe28901e4e133cda56f2 2015/07/10 // We don't handle the old format previous to that commit. txn.commit(); m_open = true; migrate(*(const uint32_t *)v.mv_data); return; } #endif } else { // if not found, and the DB is non-empty, this is probably // an "old" version 0, which we don't handle. If the DB is // empty it's fine. if (VERSION > 0 && m_height > 0) compatible = false; } if (!compatible) { txn.abort(); mdb_env_close(m_env); m_open = false; LOG_PRINT_RED_L0("Existing lmdb database is incompatible with this version."); LOG_PRINT_RED_L0("Please delete the existing database and resync."); return; } if (!(mdb_flags & MDB_RDONLY)) { // only write version on an empty DB if (m_height == 0) { MDB_val_copy k("version"); MDB_val_copy v(VERSION); auto put_result = mdb_put(txn, m_properties, &k, &v, 0); if (put_result != MDB_SUCCESS) { txn.abort(); mdb_env_close(m_env); m_open = false; LOG_PRINT_RED_L0("Failed to write version to database."); return; } } } // commit the transaction txn.commit(); m_open = true; // from here, init should be finished } void BlockchainLMDB::close() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); if (m_batch_active) { LOG_PRINT_L3("close() first calling batch_abort() due to active batch transaction"); batch_abort(); } this->sync(); m_tinfo.reset(); // FIXME: not yet thread safe!!! Use with care. mdb_env_close(m_env); m_open = false; } void BlockchainLMDB::sync() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); // Does nothing unless LMDB environment was opened with MDB_NOSYNC or in part // MDB_NOMETASYNC. Force flush to be synchronous. if (auto result = mdb_env_sync(m_env, true)) { throw0(DB_ERROR(lmdb_error("Failed to sync database: ", result).c_str())); } } void BlockchainLMDB::reset() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); mdb_txn_safe txn; if (auto result = mdb_txn_begin(m_env, NULL, 0, txn)) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); if (auto result = mdb_drop(txn, m_blocks, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_blocks: ", result).c_str())); if (auto result = mdb_drop(txn, m_block_info, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_block_info: ", result).c_str())); if (auto result = mdb_drop(txn, m_block_heights, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_block_heights: ", result).c_str())); if (auto result = mdb_drop(txn, m_txs, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_txs: ", result).c_str())); if (auto result = mdb_drop(txn, m_tx_indices, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_tx_indices: ", result).c_str())); if (auto result = mdb_drop(txn, m_tx_outputs, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_tx_outputs: ", result).c_str())); if (auto result = mdb_drop(txn, m_output_txs, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_output_txs: ", result).c_str())); if (auto result = mdb_drop(txn, m_output_amounts, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_output_amounts: ", result).c_str())); if (auto result = mdb_drop(txn, m_spent_keys, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_spent_keys: ", result).c_str())); (void)mdb_drop(txn, m_hf_starting_heights, 0); // this one is dropped in new code if (auto result = mdb_drop(txn, m_hf_versions, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_hf_versions: ", result).c_str())); if (auto result = mdb_drop(txn, m_properties, 0)) throw0(DB_ERROR(lmdb_error("Failed to drop m_properties: ", result).c_str())); // init with current version MDB_val_copy k("version"); MDB_val_copy v(VERSION); if (auto result = mdb_put(txn, m_properties, &k, &v, 0)) throw0(DB_ERROR(lmdb_error("Failed to write version to database: ", result).c_str())); txn.commit(); m_height = 0; m_num_outputs = 0; m_cum_size = 0; m_cum_count = 0; } std::vector BlockchainLMDB::get_filenames() const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); std::vector 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; } std::string BlockchainLMDB::get_db_name() const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); return std::string("lmdb"); } // TODO: this? bool BlockchainLMDB::lock() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); return false; } // TODO: this? void BlockchainLMDB::unlock() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); } #define TXN_PREFIX(flags); \ mdb_txn_safe auto_txn; \ mdb_txn_safe* txn_ptr = &auto_txn; \ if (m_batch_active) \ txn_ptr = m_write_txn; \ else \ { \ if (auto mdb_res = mdb_txn_begin(m_env, NULL, flags, auto_txn)) \ throw0(DB_ERROR(lmdb_error(std::string("Failed to create a transaction for the db in ")+__FUNCTION__+": ", mdb_res).c_str())); \ } \ #define TXN_PREFIX_RDONLY() \ MDB_txn *m_txn; \ mdb_txn_cursors *m_cursors; \ bool my_rtxn = block_rtxn_start(&m_txn, &m_cursors); \ mdb_txn_safe auto_txn(my_rtxn); \ if (my_rtxn) auto_txn.m_tinfo = m_tinfo.get() #define TXN_POSTFIX_RDONLY() #define TXN_POSTFIX_SUCCESS() \ do { \ if (! m_batch_active) \ auto_txn.commit(); \ } while(0) // The below two macros are for DB access within block add/remove, whether // regular batch txn is in use or not. m_write_txn is used as a batch txn, even // if it's only within block add/remove. // // DB access functions that may be called both within block add/remove and // without should use these. If the function will be called ONLY within block // add/remove, m_write_txn alone may be used instead of these macros. #define TXN_BLOCK_PREFIX(flags); \ mdb_txn_safe auto_txn; \ mdb_txn_safe* txn_ptr = &auto_txn; \ if (m_batch_active || m_write_txn) \ txn_ptr = m_write_txn; \ else \ { \ if (auto mdb_res = mdb_txn_begin(m_env, NULL, flags, auto_txn)) \ throw0(DB_ERROR(lmdb_error(std::string("Failed to create a transaction for the db in ")+__FUNCTION__+": ", mdb_res).c_str())); \ } \ #define TXN_BLOCK_POSTFIX_SUCCESS() \ do { \ if (! m_batch_active && ! m_write_txn) \ auto_txn.commit(); \ } while(0) bool BlockchainLMDB::block_exists(const crypto::hash& h) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(block_heights); bool ret = false; MDB_val_set(key, h); auto get_result = mdb_cursor_get(m_cur_block_heights, (MDB_val *)&zerokval, &key, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) { LOG_PRINT_L3("Block with hash " << epee::string_tools::pod_to_hex(h) << " not found in db"); } else if (get_result) throw0(DB_ERROR(lmdb_error("DB error attempting to fetch block index from hash", get_result).c_str())); else ret = true; TXN_POSTFIX_RDONLY(); return ret; } block BlockchainLMDB::get_block(const crypto::hash& h) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); return get_block_from_height(get_block_height(h)); } uint64_t BlockchainLMDB::get_block_height(const crypto::hash& h) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(block_heights); MDB_val_set(key, h); auto get_result = mdb_cursor_get(m_cur_block_heights, (MDB_val *)&zerokval, &key, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) throw1(BLOCK_DNE("Attempted to retrieve non-existent block height")); else if (get_result) throw0(DB_ERROR("Error attempting to retrieve a block height from the db")); blk_height *bhp = (blk_height *)key.mv_data; uint64_t ret = bhp->bh_height; TXN_POSTFIX_RDONLY(); return ret; } block_header BlockchainLMDB::get_block_header(const crypto::hash& h) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); // block_header object is automatically cast from block object return get_block(h); } block BlockchainLMDB::get_block_from_height(const uint64_t& height) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(blocks); MDB_val_copy key(height); MDB_val result; auto get_result = mdb_cursor_get(m_cur_blocks, &key, &result, MDB_SET); if (get_result == MDB_NOTFOUND) { throw0(BLOCK_DNE(std::string("Attempt to get block from height ").append(boost::lexical_cast(height)).append(" failed -- block not in db").c_str())); } else if (get_result) throw0(DB_ERROR("Error attempting to retrieve a block from the db")); blobdata bd; bd.assign(reinterpret_cast(result.mv_data), result.mv_size); block b; if (!parse_and_validate_block_from_blob(bd, b)) throw0(DB_ERROR("Failed to parse block from blob retrieved from the db")); TXN_POSTFIX_RDONLY(); return b; } uint64_t BlockchainLMDB::get_block_timestamp(const uint64_t& height) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(block_info); MDB_val_set(result, height); auto get_result = mdb_cursor_get(m_cur_block_info, (MDB_val *)&zerokval, &result, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) { throw0(BLOCK_DNE(std::string("Attempt to get timestamp from height ").append(boost::lexical_cast(height)).append(" failed -- timestamp not in db").c_str())); } else if (get_result) throw0(DB_ERROR("Error attempting to retrieve a timestamp from the db")); mdb_block_info *bi = (mdb_block_info *)result.mv_data; uint64_t ret = bi->bi_timestamp; TXN_POSTFIX_RDONLY(); return ret; } uint64_t BlockchainLMDB::get_top_block_timestamp() const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); // if no blocks, return 0 if (m_height == 0) { return 0; } return get_block_timestamp(m_height - 1); } size_t BlockchainLMDB::get_block_size(const uint64_t& height) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(block_info); MDB_val_set(result, height); auto get_result = mdb_cursor_get(m_cur_block_info, (MDB_val *)&zerokval, &result, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) { throw0(BLOCK_DNE(std::string("Attempt to get block size from height ").append(boost::lexical_cast(height)).append(" failed -- block size not in db").c_str())); } else if (get_result) throw0(DB_ERROR("Error attempting to retrieve a block size from the db")); mdb_block_info *bi = (mdb_block_info *)result.mv_data; size_t ret = bi->bi_size; TXN_POSTFIX_RDONLY(); return ret; } difficulty_type BlockchainLMDB::get_block_cumulative_difficulty(const uint64_t& height) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__ << " height: " << height); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(block_info); MDB_val_set(result, height); auto get_result = mdb_cursor_get(m_cur_block_info, (MDB_val *)&zerokval, &result, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) { throw0(BLOCK_DNE(std::string("Attempt to get cumulative difficulty from height ").append(boost::lexical_cast(height)).append(" failed -- difficulty not in db").c_str())); } else if (get_result) throw0(DB_ERROR("Error attempting to retrieve a cumulative difficulty from the db")); mdb_block_info *bi = (mdb_block_info *)result.mv_data; difficulty_type ret = bi->bi_diff; TXN_POSTFIX_RDONLY(); return ret; } difficulty_type BlockchainLMDB::get_block_difficulty(const uint64_t& height) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); difficulty_type diff1 = 0; difficulty_type diff2 = 0; diff1 = get_block_cumulative_difficulty(height); if (height != 0) { diff2 = get_block_cumulative_difficulty(height - 1); } return diff1 - diff2; } uint64_t BlockchainLMDB::get_block_already_generated_coins(const uint64_t& height) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(block_info); MDB_val_set(result, height); auto get_result = mdb_cursor_get(m_cur_block_info, (MDB_val *)&zerokval, &result, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) { throw0(BLOCK_DNE(std::string("Attempt to get generated coins from height ").append(boost::lexical_cast(height)).append(" failed -- block size not in db").c_str())); } else if (get_result) throw0(DB_ERROR("Error attempting to retrieve a total generated coins from the db")); mdb_block_info *bi = (mdb_block_info *)result.mv_data; uint64_t ret = bi->bi_coins; TXN_POSTFIX_RDONLY(); return ret; } crypto::hash BlockchainLMDB::get_block_hash_from_height(const uint64_t& height) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(block_info); MDB_val_set(result, height); auto get_result = mdb_cursor_get(m_cur_block_info, (MDB_val *)&zerokval, &result, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) { throw0(BLOCK_DNE(std::string("Attempt to get hash from height ").append(boost::lexical_cast(height)).append(" failed -- hash not in db").c_str())); } else if (get_result) throw0(DB_ERROR(lmdb_error("Error attempting to retrieve a block hash from the db: ", get_result).c_str())); mdb_block_info *bi = (mdb_block_info *)result.mv_data; crypto::hash ret = bi->bi_hash; TXN_POSTFIX_RDONLY(); return ret; } std::vector BlockchainLMDB::get_blocks_range(const uint64_t& h1, const uint64_t& h2) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); std::vector v; for (uint64_t height = h1; height <= h2; ++height) { v.push_back(get_block_from_height(height)); } return v; } std::vector BlockchainLMDB::get_hashes_range(const uint64_t& h1, const uint64_t& h2) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); std::vector v; for (uint64_t height = h1; height <= h2; ++height) { v.push_back(get_block_hash_from_height(height)); } return v; } crypto::hash BlockchainLMDB::top_block_hash() const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); if (m_height != 0) { return get_block_hash_from_height(m_height - 1); } return null_hash; } block BlockchainLMDB::get_top_block() const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); if (m_height != 0) { return get_block_from_height(m_height - 1); } block b; return b; } uint64_t BlockchainLMDB::height() const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); return m_height; } bool BlockchainLMDB::tx_exists(const crypto::hash& h) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(tx_indices); RCURSOR(txs); MDB_val_set(key, h); bool tx_found = false; TIME_MEASURE_START(time1); auto get_result = mdb_cursor_get(m_cur_tx_indices, (MDB_val *)&zerokval, &key, MDB_GET_BOTH); if (get_result == 0) tx_found = true; else if (get_result != MDB_NOTFOUND) throw0(DB_ERROR(lmdb_error(std::string("DB error attempting to fetch transaction index from hash ") + epee::string_tools::pod_to_hex(h) + ": ", get_result).c_str())); // This isn't needed as part of the check. we're not checking consistency of db. // get_result = mdb_cursor_get(m_cur_txs, &val_tx_index, &result, MDB_SET); TIME_MEASURE_FINISH(time1); time_tx_exists += time1; TXN_POSTFIX_RDONLY(); if (! tx_found) { LOG_PRINT_L1("transaction with hash " << epee::string_tools::pod_to_hex(h) << " not found in db"); return false; } // Below not needed due to above comment. // if (get_result == MDB_NOTFOUND) // throw0(DB_ERROR(std::string("transaction with hash ").append(epee::string_tools::pod_to_hex(h)).append(" not found at index").c_str())); // else if (get_result) // throw0(DB_ERROR(lmdb_error(std::string("DB error attempting to fetch transaction ") + epee::string_tools::pod_to_hex(h) + " at index: ", get_result).c_str())); return true; } bool BlockchainLMDB::tx_exists(const crypto::hash& h, uint64_t& tx_id) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(tx_indices); MDB_val_set(v, h); TIME_MEASURE_START(time1); auto get_result = mdb_cursor_get(m_cur_tx_indices, (MDB_val *)&zerokval, &v, MDB_GET_BOTH); TIME_MEASURE_FINISH(time1); time_tx_exists += time1; if (!get_result) { txindex *tip = (txindex *)v.mv_data; tx_id = tip->data.tx_id; } TXN_POSTFIX_RDONLY(); bool ret = false; if (get_result == MDB_NOTFOUND) { LOG_PRINT_L1("transaction with hash " << epee::string_tools::pod_to_hex(h) << " not found in db"); } else if (get_result) throw0(DB_ERROR(lmdb_error("DB error attempting to fetch transaction from hash", get_result).c_str())); else ret = true; return ret; } uint64_t BlockchainLMDB::get_tx_unlock_time(const crypto::hash& h) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(tx_indices); MDB_val_set(v, h); auto get_result = mdb_cursor_get(m_cur_tx_indices, (MDB_val *)&zerokval, &v, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) throw1(TX_DNE(lmdb_error(std::string("tx data with hash ") + epee::string_tools::pod_to_hex(h) + " not found in db: ", get_result).c_str())); else if (get_result) throw0(DB_ERROR(lmdb_error("DB error attempting to fetch tx data from hash: ", get_result).c_str())); txindex *tip = (txindex *)v.mv_data; uint64_t ret = tip->data.unlock_time; TXN_POSTFIX_RDONLY(); return ret; } transaction BlockchainLMDB::get_tx(const crypto::hash& h) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(tx_indices); RCURSOR(txs); MDB_val_set(v, h); MDB_val result; auto get_result = mdb_cursor_get(m_cur_tx_indices, (MDB_val *)&zerokval, &v, MDB_GET_BOTH); if (get_result == 0) { txindex *tip = (txindex *)v.mv_data; MDB_val_set(val_tx_id, tip->data.tx_id); get_result = mdb_cursor_get(m_cur_txs, &val_tx_id, &result, MDB_SET); } if (get_result == MDB_NOTFOUND) throw1(TX_DNE(std::string("tx with hash ").append(epee::string_tools::pod_to_hex(h)).append(" not found in db").c_str())); else if (get_result) throw0(DB_ERROR(lmdb_error("DB error attempting to fetch tx from hash", get_result).c_str())); blobdata bd; bd.assign(reinterpret_cast(result.mv_data), result.mv_size); transaction tx; if (!parse_and_validate_tx_from_blob(bd, tx)) throw0(DB_ERROR("Failed to parse tx from blob retrieved from the db")); TXN_POSTFIX_RDONLY(); return tx; } uint64_t BlockchainLMDB::get_tx_count() const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); MDB_stat db_stats; if (mdb_stat(m_txn, m_tx_indices, &db_stats)) throw0(DB_ERROR("Failed to query m_tx_indices")); TXN_POSTFIX_RDONLY(); return db_stats.ms_entries; } std::vector BlockchainLMDB::get_tx_list(const std::vector& hlist) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); std::vector v; for (auto& h : hlist) { v.push_back(get_tx(h)); } return v; } uint64_t BlockchainLMDB::get_tx_block_height(const crypto::hash& h) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(tx_indices); MDB_val_set(v, h); auto get_result = mdb_cursor_get(m_cur_tx_indices, (MDB_val *)&zerokval, &v, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) { throw1(TX_DNE(std::string("tx_data_t with hash ").append(epee::string_tools::pod_to_hex(h)).append(" not found in db").c_str())); } else if (get_result) throw0(DB_ERROR(lmdb_error("DB error attempting to fetch tx height from hash", get_result).c_str())); txindex *tip = (txindex *)v.mv_data; uint64_t ret = tip->data.block_id; TXN_POSTFIX_RDONLY(); return ret; } uint64_t BlockchainLMDB::get_num_outputs(const uint64_t& amount) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(output_amounts); MDB_val_copy k(amount); MDB_val v; mdb_size_t num_elems = 0; auto result = mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_SET); if (result == MDB_SUCCESS) { mdb_cursor_count(m_cur_output_amounts, &num_elems); } else if (result != MDB_NOTFOUND) throw0(DB_ERROR("DB error attempting to get number of outputs of an amount")); TXN_POSTFIX_RDONLY(); return num_elems; } // This is a lot harder now that we've removed the output_keys index output_data_t BlockchainLMDB::get_output_key(const uint64_t &global_index) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__ << " (unused version - does nothing)"); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(output_txs); RCURSOR(tx_indices); RCURSOR(txs); output_data_t od; MDB_val_set(v, global_index); auto get_result = mdb_cursor_get(m_cur_output_txs, (MDB_val *)&zerokval, &v, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) throw1(OUTPUT_DNE("output with given index not in db")); else if (get_result) throw0(DB_ERROR("DB error attempting to fetch output tx hash")); outtx *ot = (outtx *)v.mv_data; MDB_val_set(val_h, ot->tx_hash); get_result = mdb_cursor_get(m_cur_tx_indices, (MDB_val *)&zerokval, &val_h, MDB_GET_BOTH); if (get_result) throw0(DB_ERROR(lmdb_error(std::string("DB error attempting to fetch transaction index from hash ") + epee::string_tools::pod_to_hex(ot->tx_hash) + ": ", get_result).c_str())); txindex *tip = (txindex *)val_h.mv_data; MDB_val_set(val_tx_id, tip->data.tx_id); MDB_val result; get_result = mdb_cursor_get(m_cur_txs, &val_tx_id, &result, MDB_SET); if (get_result == MDB_NOTFOUND) throw1(TX_DNE(std::string("tx with hash ").append(epee::string_tools::pod_to_hex(ot->tx_hash)).append(" not found in db").c_str())); else if (get_result) throw0(DB_ERROR(lmdb_error("DB error attempting to fetch tx from hash", get_result).c_str())); blobdata bd; bd.assign(reinterpret_cast(result.mv_data), result.mv_size); transaction tx; if (!parse_and_validate_tx_from_blob(bd, tx)) throw0(DB_ERROR("Failed to parse tx from blob retrieved from the db")); const tx_out tx_output = tx.vout[ot->local_index]; od.unlock_time = tip->data.unlock_time; od.height = tip->data.block_id; od.pubkey = boost::get(tx_output.target).key; TXN_POSTFIX_RDONLY(); return od; } output_data_t BlockchainLMDB::get_output_key(const uint64_t& amount, const uint64_t& index) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(output_amounts); MDB_val_set(k, amount); MDB_val_set(v, index); auto get_result = mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) throw1(OUTPUT_DNE("Attempting to get output pubkey by index, but key does not exist")); else if (get_result) throw0(DB_ERROR("Error attempting to retrieve an output pubkey from the db")); output_data_t ret; if (amount == 0) { const outkey *okp = (const outkey *)v.mv_data; ret = okp->data; } else { const pre_rct_outkey *okp = (const pre_rct_outkey *)v.mv_data; memcpy(&ret, &okp->data, sizeof(pre_rct_output_data_t));; ret.commitment = rct::zeroCommit(amount); } TXN_POSTFIX_RDONLY(); return ret; } tx_out_index BlockchainLMDB::get_output_tx_and_index_from_global(const uint64_t& output_id) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(output_txs); MDB_val_set(v, output_id); auto get_result = mdb_cursor_get(m_cur_output_txs, (MDB_val *)&zerokval, &v, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) throw1(OUTPUT_DNE("output with given index not in db")); else if (get_result) throw0(DB_ERROR("DB error attempting to fetch output tx hash")); outtx *ot = (outtx *)v.mv_data; tx_out_index ret = tx_out_index(ot->tx_hash, ot->local_index); TXN_POSTFIX_RDONLY(); return ret; } tx_out_index BlockchainLMDB::get_output_tx_and_index(const uint64_t& amount, const uint64_t& index) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); std::vector < uint64_t > offsets; std::vector indices; offsets.push_back(index); get_output_tx_and_index(amount, offsets, indices); if (!indices.size()) throw1(OUTPUT_DNE("Attempting to get an output index by amount and amount index, but amount not found")); return indices[0]; } std::vector BlockchainLMDB::get_tx_amount_output_indices(const uint64_t tx_id) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(tx_outputs); int result = 0; MDB_val_set(k_tx_id, tx_id); MDB_val v; std::vector amount_output_indices; result = mdb_cursor_get(m_cur_tx_outputs, &k_tx_id, &v, MDB_SET); if (result == MDB_NOTFOUND) LOG_PRINT_L0("WARNING: Unexpected: tx has no amount indices stored in " "tx_outputs, but it should have an empty entry even if it's a tx without " "outputs"); else if (result) throw0(DB_ERROR(lmdb_error("DB error attempting to get data for tx_outputs[tx_index]", result).c_str())); uint64_t* indices = (uint64_t*)v.mv_data; int num_outputs = v.mv_size / sizeof(uint64_t); for (int i = 0; i < num_outputs; ++i) { // LOG_PRINT_L0("amount output index[" << 2*i << "]" << ": " << paired_indices[2*i] << " global output index: " << paired_indices[2*i+1]); amount_output_indices.push_back(indices[i]); } indices = nullptr; TXN_POSTFIX_RDONLY(); return amount_output_indices; } bool BlockchainLMDB::has_key_image(const crypto::key_image& img) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); bool ret; TXN_PREFIX_RDONLY(); RCURSOR(spent_keys); MDB_val k = {sizeof(img), (void *)&img}; ret = (mdb_cursor_get(m_cur_spent_keys, (MDB_val *)&zerokval, &k, MDB_GET_BOTH) == 0); TXN_POSTFIX_RDONLY(); return ret; } bool BlockchainLMDB::for_all_key_images(std::function f) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(spent_keys); MDB_val k, v; bool ret = true; k = zerokval; MDB_cursor_op op = MDB_FIRST; while (1) { int ret = mdb_cursor_get(m_cur_spent_keys, &k, &v, op); op = MDB_NEXT; if (ret == MDB_NOTFOUND) break; if (ret < 0) throw0(DB_ERROR("Failed to enumerate key images")); const crypto::key_image k_image = *(const crypto::key_image*)v.mv_data; if (!f(k_image)) { ret = false; break; } } TXN_POSTFIX_RDONLY(); return ret; } bool BlockchainLMDB::for_all_blocks(std::function f) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(blocks); MDB_val k; MDB_val v; bool ret = true; MDB_cursor_op op = MDB_FIRST; while (1) { int ret = mdb_cursor_get(m_cur_blocks, &k, &v, op); op = MDB_NEXT; if (ret == MDB_NOTFOUND) break; if (ret) throw0(DB_ERROR("Failed to enumerate blocks")); uint64_t height = *(const uint64_t*)k.mv_data; blobdata bd; bd.assign(reinterpret_cast(v.mv_data), v.mv_size); block b; if (!parse_and_validate_block_from_blob(bd, b)) throw0(DB_ERROR("Failed to parse block from blob retrieved from the db")); crypto::hash hash; if (!get_block_hash(b, hash)) throw0(DB_ERROR("Failed to get block hash from blob retrieved from the db")); if (!f(height, hash, b)) { ret = false; break; } } TXN_POSTFIX_RDONLY(); return ret; } bool BlockchainLMDB::for_all_transactions(std::function f) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(txs); RCURSOR(tx_indices); MDB_val k; MDB_val v; bool ret = true; MDB_cursor_op op = MDB_FIRST; while (1) { int ret = mdb_cursor_get(m_cur_tx_indices, &k, &v, op); op = MDB_NEXT; if (ret == MDB_NOTFOUND) break; if (ret) throw0(DB_ERROR(lmdb_error("Failed to enumerate transactions: ", ret).c_str())); txindex *ti = (txindex *)v.mv_data; const crypto::hash hash = ti->key; k.mv_data = (void *)&ti->data.tx_id; k.mv_size = sizeof(ti->data.tx_id); ret = mdb_cursor_get(m_cur_txs, &k, &v, MDB_SET); if (ret == MDB_NOTFOUND) break; if (ret) throw0(DB_ERROR(lmdb_error("Failed to enumerate transactions: ", ret).c_str())); blobdata bd; bd.assign(reinterpret_cast(v.mv_data), v.mv_size); transaction tx; if (!parse_and_validate_tx_from_blob(bd, tx)) throw0(DB_ERROR("Failed to parse tx from blob retrieved from the db")); if (!f(hash, tx)) { ret = false; break; } } TXN_POSTFIX_RDONLY(); return ret; } bool BlockchainLMDB::for_all_outputs(std::function f) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(output_amounts); MDB_val k; MDB_val v; bool ret = true; MDB_cursor_op op = MDB_FIRST; while (1) { int ret = mdb_cursor_get(m_cur_output_amounts, &k, &v, op); op = MDB_NEXT; if (ret == MDB_NOTFOUND) break; if (ret) throw0(DB_ERROR("Failed to enumerate outputs")); uint64_t amount = *(const uint64_t*)k.mv_data; outkey *ok = (outkey *)v.mv_data; tx_out_index toi = get_output_tx_and_index_from_global(ok->output_id); if (!f(amount, toi.first, toi.second)) { ret = false; break; } } TXN_POSTFIX_RDONLY(); return ret; } // batch_num_blocks: (optional) Used to check if resize needed before batch transaction starts. void BlockchainLMDB::batch_start(uint64_t batch_num_blocks) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); if (! m_batch_transactions) throw0(DB_ERROR("batch transactions not enabled")); if (m_batch_active) throw0(DB_ERROR("batch transaction already in progress")); if (m_write_batch_txn != nullptr) throw0(DB_ERROR("batch transaction already in progress")); if (m_write_txn) throw0(DB_ERROR("batch transaction attempted, but m_write_txn already in use")); check_open(); m_writer = boost::this_thread::get_id(); check_and_resize_for_batch(batch_num_blocks); m_write_batch_txn = new mdb_txn_safe(); // NOTE: need to make sure it's destroyed properly when done if (auto mdb_res = mdb_txn_begin(m_env, NULL, 0, *m_write_batch_txn)) { delete m_write_batch_txn; m_write_batch_txn = nullptr; throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", mdb_res).c_str())); } // indicates this transaction is for batch transactions, but not whether it's // active m_write_batch_txn->m_batch_txn = true; m_write_txn = m_write_batch_txn; m_batch_active = true; memset(&m_wcursors, 0, sizeof(m_wcursors)); LOG_PRINT_L3("batch transaction: begin"); } void BlockchainLMDB::batch_commit() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); if (! m_batch_transactions) throw0(DB_ERROR("batch transactions not enabled")); if (! m_batch_active) throw0(DB_ERROR("batch transaction not in progress")); if (m_write_batch_txn == nullptr) throw0(DB_ERROR("batch transaction not in progress")); check_open(); LOG_PRINT_L3("batch transaction: committing..."); TIME_MEASURE_START(time1); m_write_txn->commit(); TIME_MEASURE_FINISH(time1); time_commit1 += time1; LOG_PRINT_L3("batch transaction: committed"); m_write_txn = nullptr; delete m_write_batch_txn; m_write_batch_txn = nullptr; memset(&m_wcursors, 0, sizeof(m_wcursors)); } void BlockchainLMDB::batch_stop() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); if (! m_batch_transactions) throw0(DB_ERROR("batch transactions not enabled")); if (! m_batch_active) throw0(DB_ERROR("batch transaction not in progress")); if (m_write_batch_txn == nullptr) throw0(DB_ERROR("batch transaction not in progress")); check_open(); LOG_PRINT_L3("batch transaction: committing..."); TIME_MEASURE_START(time1); m_write_txn->commit(); TIME_MEASURE_FINISH(time1); time_commit1 += time1; // for destruction of batch transaction m_write_txn = nullptr; delete m_write_batch_txn; m_write_batch_txn = nullptr; m_batch_active = false; memset(&m_wcursors, 0, sizeof(m_wcursors)); LOG_PRINT_L3("batch transaction: end"); } void BlockchainLMDB::batch_abort() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); if (! m_batch_transactions) throw0(DB_ERROR("batch transactions not enabled")); if (! m_batch_active) throw0(DB_ERROR("batch transaction not in progress")); check_open(); // for destruction of batch transaction m_write_txn = nullptr; // explicitly call in case mdb_env_close() (BlockchainLMDB::close()) called before BlockchainLMDB destructor called. m_write_batch_txn->abort(); delete m_write_batch_txn; m_write_batch_txn = nullptr; m_batch_active = false; memset(&m_wcursors, 0, sizeof(m_wcursors)); LOG_PRINT_L3("batch transaction: aborted"); } void BlockchainLMDB::set_batch_transactions(bool batch_transactions) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); if ((batch_transactions) && (m_batch_transactions)) { LOG_PRINT_L0("WARNING: batch transaction mode already enabled, but asked to enable batch mode"); } m_batch_transactions = batch_transactions; LOG_PRINT_L3("batch transactions " << (m_batch_transactions ? "enabled" : "disabled")); } // return true if we started the txn, false if already started bool BlockchainLMDB::block_rtxn_start(MDB_txn **mtxn, mdb_txn_cursors **mcur) const { bool ret = false; if (m_write_txn && m_writer == boost::this_thread::get_id()) { *mtxn = m_write_txn->m_txn; *mcur = (mdb_txn_cursors *)&m_wcursors; return ret; } if (!m_tinfo.get()) { m_tinfo.reset(new mdb_threadinfo); memset(&m_tinfo->m_ti_rcursors, 0, sizeof(m_tinfo->m_ti_rcursors)); memset(&m_tinfo->m_ti_rflags, 0, sizeof(m_tinfo->m_ti_rflags)); if (auto mdb_res = mdb_txn_begin(m_env, NULL, MDB_RDONLY, &m_tinfo->m_ti_rtxn)) throw0(DB_ERROR_TXN_START(lmdb_error("Failed to create a read transaction for the db: ", mdb_res).c_str())); ret = true; } else if (!m_tinfo->m_ti_rflags.m_rf_txn) { if (auto mdb_res = mdb_txn_renew(m_tinfo->m_ti_rtxn)) throw0(DB_ERROR_TXN_START(lmdb_error("Failed to renew a read transaction for the db: ", mdb_res).c_str())); ret = true; } if (ret) m_tinfo->m_ti_rflags.m_rf_txn = true; *mtxn = m_tinfo->m_ti_rtxn; *mcur = &m_tinfo->m_ti_rcursors; if (ret) LOG_PRINT_L3("BlockchainLMDB::" << __func__); return ret; } void BlockchainLMDB::block_rtxn_stop() const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); mdb_txn_reset(m_tinfo->m_ti_rtxn); memset(&m_tinfo->m_ti_rflags, 0, sizeof(m_tinfo->m_ti_rflags)); } void BlockchainLMDB::block_txn_start(bool readonly) { if (readonly) { bool didit = false; if (m_write_txn && m_writer == boost::this_thread::get_id()) return; if (!m_tinfo.get()) { m_tinfo.reset(new mdb_threadinfo); memset(&m_tinfo->m_ti_rcursors, 0, sizeof(m_tinfo->m_ti_rcursors)); memset(&m_tinfo->m_ti_rflags, 0, sizeof(m_tinfo->m_ti_rflags)); if (auto mdb_res = mdb_txn_begin(m_env, NULL, MDB_RDONLY, &m_tinfo->m_ti_rtxn)) throw0(DB_ERROR_TXN_START(lmdb_error("Failed to create a read transaction for the db: ", mdb_res).c_str())); didit = true; } else if (!m_tinfo->m_ti_rflags.m_rf_txn) { if (auto mdb_res = mdb_txn_renew(m_tinfo->m_ti_rtxn)) throw0(DB_ERROR_TXN_START(lmdb_error("Failed to renew a read transaction for the db: ", mdb_res).c_str())); didit = true; } if (didit) { m_tinfo->m_ti_rflags.m_rf_txn = true; LOG_PRINT_L3("BlockchainLMDB::" << __func__ << " RO"); } return; } LOG_PRINT_L3("BlockchainLMDB::" << __func__); // Distinguish the exceptions here from exceptions that would be thrown while // using the txn and committing it. // // If an exception is thrown in this setup, we don't want the caller to catch // it and proceed as if there were an existing write txn, such as trying to // call block_txn_abort(). It also indicates a serious issue which will // probably be thrown up another layer. if (! m_batch_active && m_write_txn) throw0(DB_ERROR_TXN_START((std::string("Attempted to start new write txn when write txn already exists in ")+__FUNCTION__).c_str())); if (! m_batch_active) { m_writer = boost::this_thread::get_id(); m_write_txn = new mdb_txn_safe(); if (auto mdb_res = mdb_txn_begin(m_env, NULL, 0, *m_write_txn)) { delete m_write_txn; m_write_txn = nullptr; throw0(DB_ERROR_TXN_START(lmdb_error("Failed to create a transaction for the db: ", mdb_res).c_str())); } memset(&m_wcursors, 0, sizeof(m_wcursors)); } } void BlockchainLMDB::block_txn_stop() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); if (m_write_txn && m_writer == boost::this_thread::get_id()) { if (! m_batch_active) { TIME_MEASURE_START(time1); m_write_txn->commit(); TIME_MEASURE_FINISH(time1); time_commit1 += time1; delete m_write_txn; m_write_txn = nullptr; memset(&m_wcursors, 0, sizeof(m_wcursors)); } } else if (m_tinfo->m_ti_rtxn) { mdb_txn_reset(m_tinfo->m_ti_rtxn); memset(&m_tinfo->m_ti_rflags, 0, sizeof(m_tinfo->m_ti_rflags)); } } void BlockchainLMDB::block_txn_abort() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); if (m_write_txn && m_writer == boost::this_thread::get_id()) { if (! m_batch_active) { delete m_write_txn; m_write_txn = nullptr; memset(&m_wcursors, 0, sizeof(m_wcursors)); } } else if (m_tinfo->m_ti_rtxn) { mdb_txn_reset(m_tinfo->m_ti_rtxn); memset(&m_tinfo->m_ti_rflags, 0, sizeof(m_tinfo->m_ti_rflags)); } else { // This would probably mean an earlier exception was caught, but then we // proceeded further than we should have. throw0(DB_ERROR((std::string("BlockchainLMDB::") + __func__ + std::string(": block-level DB transaction abort called when write txn doesn't exist") ).c_str())); } } 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& txs) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); if (m_height % 1000 == 0) { // for batch mode, DB resize check is done at start of batch transaction if (! m_batch_active && need_resize()) { LOG_PRINT_L0("LMDB memory map needs resized, doing that now."); do_resize(); } } uint64_t num_txs = m_num_txs; uint64_t num_outputs = m_num_outputs; try { BlockchainDB::add_block(blk, block_size, cumulative_difficulty, coins_generated, txs); } catch (DB_ERROR_TXN_START& e) { throw; } catch (...) { m_num_txs = num_txs; m_num_outputs = num_outputs; block_txn_abort(); throw; } return ++m_height; } void BlockchainLMDB::pop_block(block& blk, std::vector& txs) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); block_txn_start(false); uint64_t num_txs = m_num_txs; uint64_t num_outputs = m_num_outputs; try { BlockchainDB::pop_block(blk, txs); block_txn_stop(); } catch (...) { m_num_txs = num_txs; m_num_outputs = num_outputs; block_txn_abort(); throw; } --m_height; } void BlockchainLMDB::get_output_tx_and_index_from_global(const std::vector &global_indices, std::vector &tx_out_indices) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); tx_out_indices.clear(); TXN_PREFIX_RDONLY(); RCURSOR(output_txs); for (const uint64_t &output_id : global_indices) { MDB_val_set(v, output_id); auto get_result = mdb_cursor_get(m_cur_output_txs, (MDB_val *)&zerokval, &v, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) throw1(OUTPUT_DNE("output with given index not in db")); else if (get_result) throw0(DB_ERROR("DB error attempting to fetch output tx hash")); outtx *ot = (outtx *)v.mv_data; auto result = tx_out_index(ot->tx_hash, ot->local_index); tx_out_indices.push_back(result); } TXN_POSTFIX_RDONLY(); } void BlockchainLMDB::get_output_key(const uint64_t &amount, const std::vector &offsets, std::vector &outputs) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); TIME_MEASURE_START(db3); check_open(); outputs.clear(); TXN_PREFIX_RDONLY(); RCURSOR(output_amounts); MDB_val_set(k, amount); for (const uint64_t &index : offsets) { MDB_val_set(v, index); auto get_result = mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) throw1(OUTPUT_DNE("Attempting to get output pubkey by global index, but key does not exist")); else if (get_result) throw0(DB_ERROR(lmdb_error("Error attempting to retrieve an output pubkey from the db", get_result).c_str())); output_data_t data; if (amount == 0) { const outkey *okp = (const outkey *)v.mv_data; data = okp->data; } else { const pre_rct_outkey *okp = (const pre_rct_outkey *)v.mv_data; memcpy(&data, &okp->data, sizeof(pre_rct_output_data_t)); data.commitment = rct::zeroCommit(amount); } outputs.push_back(data); } TXN_POSTFIX_RDONLY(); TIME_MEASURE_FINISH(db3); LOG_PRINT_L3("db3: " << db3); } void BlockchainLMDB::get_output_tx_and_index(const uint64_t& amount, const std::vector &offsets, std::vector &indices) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); indices.clear(); std::vector tx_indices; TXN_PREFIX_RDONLY(); RCURSOR(output_amounts); MDB_val_set(k, amount); for (const uint64_t &index : offsets) { MDB_val_set(v, index); auto get_result = mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_GET_BOTH); if (get_result == MDB_NOTFOUND) throw1(OUTPUT_DNE("Attempting to get output by index, but key does not exist")); else if (get_result) throw0(DB_ERROR(lmdb_error("Error attempting to retrieve an output from the db", get_result).c_str())); outkey *okp = (outkey *)v.mv_data; tx_indices.push_back(okp->output_id); } TIME_MEASURE_START(db3); if(tx_indices.size() > 0) { get_output_tx_and_index_from_global(tx_indices, indices); } TIME_MEASURE_FINISH(db3); LOG_PRINT_L3("db3: " << db3); } std::map BlockchainLMDB::get_output_histogram(const std::vector &amounts, bool unlocked) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(output_amounts); std::map histogram; MDB_val k; MDB_val v; if (amounts.empty()) { MDB_cursor_op op = MDB_FIRST; while (1) { int ret = mdb_cursor_get(m_cur_output_amounts, &k, &v, op); op = MDB_NEXT_NODUP; if (ret == MDB_NOTFOUND) break; if (ret) throw0(DB_ERROR(lmdb_error("Failed to enumerate outputs: ", ret).c_str())); mdb_size_t num_elems = 0; mdb_cursor_count(m_cur_output_amounts, &num_elems); uint64_t amount = *(const uint64_t*)k.mv_data; histogram[amount] = num_elems; } } else { for (const auto &amount: amounts) { MDB_val_copy k(amount); int ret = mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_SET); if (ret == MDB_NOTFOUND) { histogram[amount] = 0; } else if (ret == MDB_SUCCESS) { mdb_size_t num_elems = 0; mdb_cursor_count(m_cur_output_amounts, &num_elems); histogram[amount] = num_elems; } else { throw0(DB_ERROR(lmdb_error("Failed to enumerate outputs: ", ret).c_str())); } } } if (unlocked) { const uint64_t blockchain_height = height(); for (auto i: histogram) { uint64_t amount = i.first; uint64_t num_elems = i.second; while (num_elems > 0) { const tx_out_index toi = get_output_tx_and_index(amount, num_elems - 1); const uint64_t height = get_tx_block_height(toi.first); if (height + CRYPTONOTE_DEFAULT_TX_SPENDABLE_AGE <= blockchain_height) break; --num_elems; } // modifying second does not invalidate the iterator i.second = num_elems; } } TXN_POSTFIX_RDONLY(); return histogram; } void BlockchainLMDB::check_hard_fork_info() { } void BlockchainLMDB::drop_hard_fork_info() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX(0); mdb_drop(*txn_ptr, m_hf_starting_heights, 1); mdb_drop(*txn_ptr, m_hf_versions, 1); TXN_POSTFIX_SUCCESS(); } void BlockchainLMDB::set_hard_fork_version(uint64_t height, uint8_t version) { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_BLOCK_PREFIX(0); MDB_val_copy val_key(height); MDB_val_copy val_value(version); int result; result = mdb_put(*txn_ptr, m_hf_versions, &val_key, &val_value, MDB_APPEND); if (result == MDB_KEYEXIST) result = mdb_put(*txn_ptr, m_hf_versions, &val_key, &val_value, 0); if (result) throw1(DB_ERROR(lmdb_error("Error adding hard fork version to db transaction: ", result).c_str())); TXN_BLOCK_POSTFIX_SUCCESS(); } uint8_t BlockchainLMDB::get_hard_fork_version(uint64_t height) const { LOG_PRINT_L3("BlockchainLMDB::" << __func__); check_open(); TXN_PREFIX_RDONLY(); RCURSOR(hf_versions); MDB_val_copy val_key(height); MDB_val val_ret; auto result = mdb_cursor_get(m_cur_hf_versions, &val_key, &val_ret, MDB_SET); if (result == MDB_NOTFOUND || result) throw0(DB_ERROR(lmdb_error("Error attempting to retrieve a hard fork version at height " + boost::lexical_cast(height) + " from the db: ", result).c_str())); uint8_t ret = *(const uint8_t*)val_ret.mv_data; TXN_POSTFIX_RDONLY(); return ret; } bool BlockchainLMDB::is_read_only() const { unsigned int flags; auto result = mdb_env_get_flags(m_env, &flags); if (result) throw0(DB_ERROR(lmdb_error("Error getting database environment info: ", result).c_str())); if (flags & MDB_RDONLY) return true; return false; } void BlockchainLMDB::fixup() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); // Always call parent as well BlockchainDB::fixup(); } #define RENAME_DB(name) \ k.mv_data = (void *)name; \ k.mv_size = sizeof(name)-1; \ result = mdb_cursor_open(txn, 1, &c_cur); \ if (result) \ throw0(DB_ERROR(lmdb_error("Failed to open a cursor for " name ": ", result).c_str())); \ result = mdb_cursor_get(c_cur, &k, NULL, MDB_SET_KEY); \ if (result) \ throw0(DB_ERROR(lmdb_error("Failed to get DB record for " name ": ", result).c_str())); \ ptr = (char *)k.mv_data; \ ptr[sizeof(name)-2] = 's' #define LOGIF(y) if (y <= epee::log_space::log_singletone::get_log_detalisation_level()) void BlockchainLMDB::migrate_0_1() { LOG_PRINT_L3("BlockchainLMDB::" << __func__); uint64_t i, z; int result; mdb_txn_safe txn(false); MDB_val k, v; char *ptr; LOG_PRINT_YELLOW("Migrating blockchain from DB version 0 to 1 - this may take a while:", LOG_LEVEL_0); LOG_PRINT_L0("updating blocks, hf_versions, outputs, txs, and spent_keys tables..."); LOG_PRINT_L0("Total number of blocks: " << m_height); LOG_PRINT_L1("block migration will update block_heights, block_info, and hf_versions..."); do { LOG_PRINT_L1("migrating block_heights:"); MDB_dbi o_heights; unsigned int flags; result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); result = mdb_dbi_flags(txn, m_block_heights, &flags); if (result) throw0(DB_ERROR(lmdb_error("Failed to retrieve block_heights flags: ", result).c_str())); /* if the flags are what we expect, this table has already been migrated */ if ((flags & (MDB_INTEGERKEY|MDB_DUPSORT|MDB_DUPFIXED)) == (MDB_INTEGERKEY|MDB_DUPSORT|MDB_DUPFIXED)) { txn.abort(); LOG_PRINT_L1(" block_heights already migrated"); break; } /* the block_heights table name is the same but the old version and new version * have incompatible DB flags. Create a new table with the right flags. We want * the name to be similar to the old name so that it will occupy the same location * in the DB. */ o_heights = m_block_heights; lmdb_db_open(txn, "block_heightr", MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_block_heights, "Failed to open db handle for block_heightr"); mdb_set_dupsort(txn, m_block_heights, compare_hash32); MDB_cursor *c_old, *c_cur; blk_height bh; MDB_val_set(nv, bh); /* old table was k(hash), v(height). * new table is DUPFIXED, k(zeroval), v{hash, height}. */ i = 0; z = m_height; while(1) { if (!(i % 2000)) { if (i) { LOGIF(1) { std::cout << i << " / " << z << " \r" << std::flush; } txn.commit(); result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); } result = mdb_cursor_open(txn, m_block_heights, &c_cur); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for block_heightr: ", result).c_str())); result = mdb_cursor_open(txn, o_heights, &c_old); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for block_heights: ", result).c_str())); if (!i) { MDB_stat ms; mdb_stat(txn, m_block_heights, &ms); i = ms.ms_entries; } } result = mdb_cursor_get(c_old, &k, &v, MDB_NEXT); if (result == MDB_NOTFOUND) { txn.commit(); break; } else if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from block_heights: ", result).c_str())); bh.bh_hash = *(crypto::hash *)k.mv_data; bh.bh_height = *(uint64_t *)v.mv_data; result = mdb_cursor_put(c_cur, (MDB_val *)&zerokval, &nv, MDB_APPENDDUP); if (result) throw0(DB_ERROR(lmdb_error("Failed to put a record into block_heightr: ", result).c_str())); /* we delete the old records immediately, so the overall DB and mapsize should not grow. * This is a little slower than just letting mdb_drop() delete it all at the end, but * it saves a significant amount of disk space. */ result = mdb_cursor_del(c_old, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete a record from block_heights: ", result).c_str())); i++; } result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); /* Delete the old table */ result = mdb_drop(txn, o_heights, 1); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete old block_heights table: ", result).c_str())); RENAME_DB("block_heightr"); /* close and reopen to get old dbi slot back */ mdb_dbi_close(m_env, m_block_heights); lmdb_db_open(txn, "block_heights", MDB_INTEGERKEY | MDB_DUPSORT | MDB_DUPFIXED, m_block_heights, "Failed to open db handle for block_heights"); mdb_set_dupsort(txn, m_block_heights, compare_hash32); txn.commit(); } while(0); /* old tables are k(height), v(value). * new table is DUPFIXED, k(zeroval), v{height, values...}. */ do { LOG_PRINT_L1("migrating block info:"); MDB_dbi coins; result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); result = mdb_dbi_open(txn, "block_coins", 0, &coins); if (result == MDB_NOTFOUND) { txn.abort(); LOG_PRINT_L1(" block_info already migrated"); break; } MDB_dbi diffs, hashes, sizes, timestamps; mdb_block_info bi; MDB_val_set(nv, bi); lmdb_db_open(txn, "block_diffs", 0, diffs, "Failed to open db handle for block_diffs"); lmdb_db_open(txn, "block_hashes", 0, hashes, "Failed to open db handle for block_hashes"); lmdb_db_open(txn, "block_sizes", 0, sizes, "Failed to open db handle for block_sizes"); lmdb_db_open(txn, "block_timestamps", 0, timestamps, "Failed to open db handle for block_timestamps"); MDB_cursor *c_cur, *c_coins, *c_diffs, *c_hashes, *c_sizes, *c_timestamps; i = 0; z = m_height; while(1) { MDB_val k, v; if (!(i % 2000)) { if (i) { LOGIF(1) { std::cout << i << " / " << z << " \r" << std::flush; } txn.commit(); result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); } result = mdb_cursor_open(txn, m_block_info, &c_cur); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for block_info: ", result).c_str())); result = mdb_cursor_open(txn, coins, &c_coins); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for block_coins: ", result).c_str())); result = mdb_cursor_open(txn, diffs, &c_diffs); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for block_diffs: ", result).c_str())); result = mdb_cursor_open(txn, hashes, &c_hashes); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for block_hashes: ", result).c_str())); result = mdb_cursor_open(txn, sizes, &c_sizes); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for block_coins: ", result).c_str())); result = mdb_cursor_open(txn, timestamps, &c_timestamps); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for block_timestamps: ", result).c_str())); if (!i) { MDB_stat ms; mdb_stat(txn, m_block_info, &ms); i = ms.ms_entries; } } result = mdb_cursor_get(c_coins, &k, &v, MDB_NEXT); if (result == MDB_NOTFOUND) { break; } else if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from block_coins: ", result).c_str())); bi.bi_height = *(uint64_t *)k.mv_data; bi.bi_coins = *(uint64_t *)v.mv_data; result = mdb_cursor_get(c_diffs, &k, &v, MDB_NEXT); if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from block_diffs: ", result).c_str())); bi.bi_diff = *(uint64_t *)v.mv_data; result = mdb_cursor_get(c_hashes, &k, &v, MDB_NEXT); if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from block_hashes: ", result).c_str())); bi.bi_hash = *(crypto::hash *)v.mv_data; result = mdb_cursor_get(c_sizes, &k, &v, MDB_NEXT); if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from block_sizes: ", result).c_str())); if (v.mv_size == sizeof(uint32_t)) bi.bi_size = *(uint32_t *)v.mv_data; else bi.bi_size = *(uint64_t *)v.mv_data; // this is a 32/64 compat bug in version 0 result = mdb_cursor_get(c_timestamps, &k, &v, MDB_NEXT); if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from block_timestamps: ", result).c_str())); bi.bi_timestamp = *(uint64_t *)v.mv_data; result = mdb_cursor_put(c_cur, (MDB_val *)&zerokval, &nv, MDB_APPENDDUP); if (result) throw0(DB_ERROR(lmdb_error("Failed to put a record into block_info: ", result).c_str())); result = mdb_cursor_del(c_coins, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete a record from block_coins: ", result).c_str())); result = mdb_cursor_del(c_diffs, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete a record from block_diffs: ", result).c_str())); result = mdb_cursor_del(c_hashes, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete a record from block_hashes: ", result).c_str())); result = mdb_cursor_del(c_sizes, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete a record from block_sizes: ", result).c_str())); result = mdb_cursor_del(c_timestamps, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete a record from block_timestamps: ", result).c_str())); i++; } mdb_cursor_close(c_timestamps); mdb_cursor_close(c_sizes); mdb_cursor_close(c_hashes); mdb_cursor_close(c_diffs); mdb_cursor_close(c_coins); result = mdb_drop(txn, timestamps, 1); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete block_timestamps from the db: ", result).c_str())); result = mdb_drop(txn, sizes, 1); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete block_sizes from the db: ", result).c_str())); result = mdb_drop(txn, hashes, 1); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete block_hashes from the db: ", result).c_str())); result = mdb_drop(txn, diffs, 1); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete block_diffs from the db: ", result).c_str())); result = mdb_drop(txn, coins, 1); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete block_coins from the db: ", result).c_str())); txn.commit(); } while(0); do { LOG_PRINT_L1("migrating hf_versions:"); MDB_dbi o_hfv; unsigned int flags; result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); result = mdb_dbi_flags(txn, m_hf_versions, &flags); if (result) throw0(DB_ERROR(lmdb_error("Failed to retrieve hf_versions flags: ", result).c_str())); /* if the flags are what we expect, this table has already been migrated */ if (flags & MDB_INTEGERKEY) { txn.abort(); LOG_PRINT_L1(" hf_versions already migrated"); break; } /* the hf_versions table name is the same but the old version and new version * have incompatible DB flags. Create a new table with the right flags. */ o_hfv = m_hf_versions; lmdb_db_open(txn, "hf_versionr", MDB_INTEGERKEY | MDB_CREATE, m_hf_versions, "Failed to open db handle for hf_versionr"); MDB_cursor *c_old, *c_cur; i = 0; z = m_height; while(1) { if (!(i % 2000)) { if (i) { LOGIF(1) { std::cout << i << " / " << z << " \r" << std::flush; } txn.commit(); result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); } result = mdb_cursor_open(txn, m_hf_versions, &c_cur); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for spent_keyr: ", result).c_str())); result = mdb_cursor_open(txn, o_hfv, &c_old); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for spent_keys: ", result).c_str())); if (!i) { MDB_stat ms; mdb_stat(txn, m_hf_versions, &ms); i = ms.ms_entries; } } result = mdb_cursor_get(c_old, &k, &v, MDB_NEXT); if (result == MDB_NOTFOUND) { txn.commit(); break; } else if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from hf_versions: ", result).c_str())); result = mdb_cursor_put(c_cur, &k, &v, MDB_APPEND); if (result) throw0(DB_ERROR(lmdb_error("Failed to put a record into hf_versionr: ", result).c_str())); result = mdb_cursor_del(c_old, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete a record from hf_versions: ", result).c_str())); i++; } result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); /* Delete the old table */ result = mdb_drop(txn, o_hfv, 1); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete old hf_versions table: ", result).c_str())); RENAME_DB("hf_versionr"); mdb_dbi_close(m_env, m_hf_versions); lmdb_db_open(txn, "hf_versions", MDB_INTEGERKEY, m_hf_versions, "Failed to open db handle for hf_versions"); txn.commit(); } while(0); do { LOG_PRINT_L1("deleting old indices:"); /* Delete all other tables, we're just going to recreate them */ MDB_dbi dbi; result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); result = mdb_dbi_open(txn, "tx_unlocks", 0, &dbi); if (result == MDB_NOTFOUND) { txn.abort(); LOG_PRINT_L1(" old indices already deleted"); break; } txn.abort(); #define DELETE_DB(x) do { \ LOG_PRINT_L1(" " x ":"); \ result = mdb_txn_begin(m_env, NULL, 0, txn); \ if (result) \ throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); \ result = mdb_dbi_open(txn, x, 0, &dbi); \ if (!result) { \ result = mdb_drop(txn, dbi, 1); \ if (result) \ throw0(DB_ERROR(lmdb_error("Failed to delete " x ": ", result).c_str())); \ txn.commit(); \ } } while(0) DELETE_DB("tx_heights"); DELETE_DB("output_txs"); DELETE_DB("output_indices"); DELETE_DB("output_keys"); DELETE_DB("spent_keys"); DELETE_DB("output_amounts"); DELETE_DB("tx_outputs"); DELETE_DB("tx_unlocks"); /* reopen new DBs with correct flags */ result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); lmdb_db_open(txn, LMDB_OUTPUT_TXS, MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_output_txs, "Failed to open db handle for m_output_txs"); mdb_set_dupsort(txn, m_output_txs, compare_uint64); lmdb_db_open(txn, LMDB_TX_OUTPUTS, MDB_INTEGERKEY | MDB_CREATE, m_tx_outputs, "Failed to open db handle for m_tx_outputs"); lmdb_db_open(txn, LMDB_SPENT_KEYS, MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_spent_keys, "Failed to open db handle for m_spent_keys"); mdb_set_dupsort(txn, m_spent_keys, compare_hash32); lmdb_db_open(txn, LMDB_OUTPUT_AMOUNTS, MDB_INTEGERKEY | MDB_DUPSORT | MDB_DUPFIXED | MDB_CREATE, m_output_amounts, "Failed to open db handle for m_output_amounts"); mdb_set_dupsort(txn, m_output_amounts, compare_uint64); txn.commit(); m_num_txs = 0; m_num_outputs = 0; } while(0); do { LOG_PRINT_L1("migrating txs and outputs:"); unsigned int flags; result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); result = mdb_dbi_flags(txn, m_txs, &flags); if (result) throw0(DB_ERROR(lmdb_error("Failed to retrieve txs flags: ", result).c_str())); /* if the flags are what we expect, this table has already been migrated */ if (flags & MDB_INTEGERKEY) { txn.abort(); LOG_PRINT_L1(" txs already migrated"); break; } MDB_dbi o_txs; blobdata bd; block b; MDB_val hk; o_txs = m_txs; mdb_set_compare(txn, o_txs, compare_hash32); lmdb_db_open(txn, "txr", MDB_INTEGERKEY | MDB_CREATE, m_txs, "Failed to open db handle for txr"); txn.commit(); MDB_cursor *c_blocks, *c_txs, *c_props, *c_cur; i = 0; z = m_height; hk.mv_size = sizeof(crypto::hash); set_batch_transactions(true); batch_start(1000); txn.m_txn = m_write_txn->m_txn; m_height = 0; while(1) { if (!(i % 1000)) { if (i) { LOGIF(1) { std::cout << i << " / " << z << " \r" << std::flush; } MDB_val_set(pk, "txblk"); MDB_val_set(pv, m_height); result = mdb_cursor_put(c_props, &pk, &pv, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to update txblk property: ", result).c_str())); txn.commit(); result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); m_write_txn->m_txn = txn.m_txn; m_write_batch_txn->m_txn = txn.m_txn; memset(&m_wcursors, 0, sizeof(m_wcursors)); } result = mdb_cursor_open(txn, m_blocks, &c_blocks); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for blocks: ", result).c_str())); result = mdb_cursor_open(txn, m_properties, &c_props); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for properties: ", result).c_str())); result = mdb_cursor_open(txn, o_txs, &c_txs); if (result) throw0(DB_ERROR(lmdb_error("Failed to open a cursor for txs: ", result).c_str())); if (!i) { MDB_stat ms; mdb_stat(txn, m_output_txs, &ms); m_num_outputs = ms.ms_entries; mdb_stat(txn, m_txs, &ms); m_num_txs = i = ms.ms_entries; if (i) { m_num_txs = i; MDB_val_set(pk, "txblk"); result = mdb_cursor_get(c_props, &pk, &k, MDB_SET); if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from properties: ", result).c_str())); m_height = *(uint64_t *)k.mv_data; } } if (i) { result = mdb_cursor_get(c_blocks, &k, &v, MDB_SET); if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from blocks: ", result).c_str())); } } result = mdb_cursor_get(c_blocks, &k, &v, MDB_NEXT); if (result == MDB_NOTFOUND) { MDB_val_set(pk, "txblk"); result = mdb_cursor_get(c_props, &pk, &v, MDB_SET); if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from props: ", result).c_str())); result = mdb_cursor_del(c_props, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete a record from props: ", result).c_str())); batch_stop(); break; } else if (result) throw0(DB_ERROR(lmdb_error("Failed to get a record from blocks: ", result).c_str())); bd.assign(reinterpret_cast(v.mv_data), v.mv_size); if (!parse_and_validate_block_from_blob(bd, b)) throw0(DB_ERROR("Failed to parse block from blob retrieved from the db")); add_transaction(null_hash, b.miner_tx); for (unsigned int j = 0; j(v.mv_data), v.mv_size); if (!parse_and_validate_tx_from_blob(bd, tx)) throw0(DB_ERROR("Failed to parse tx from blob retrieved from the db")); add_transaction(null_hash, tx, &b.tx_hashes[j]); result = mdb_cursor_del(c_txs, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to get record from txs: ", result).c_str())); } i++; m_height = i; } result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); result = mdb_drop(txn, o_txs, 1); if (result) throw0(DB_ERROR(lmdb_error("Failed to delete txs from the db: ", result).c_str())); RENAME_DB("txr"); mdb_dbi_close(m_env, m_txs); lmdb_db_open(txn, "txs", MDB_INTEGERKEY, m_txs, "Failed to open db handle for txs"); txn.commit(); } while(0); uint32_t version = 1; v.mv_data = (void *)&version; v.mv_size = sizeof(version); MDB_val_copy vk("version"); result = mdb_txn_begin(m_env, NULL, 0, txn); if (result) throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); result = mdb_put(txn, m_properties, &vk, &v, 0); if (result) throw0(DB_ERROR(lmdb_error("Failed to update version for the db: ", result).c_str())); txn.commit(); } void BlockchainLMDB::migrate(const uint32_t oldversion) { switch(oldversion) { case 0: migrate_0_1(); /* FALLTHRU */ default: ; } } } // namespace cryptonote