moneroexamples
6 years ago
parent
8f59acd34c
commit
9e10445241
@ -1,755 +0,0 @@
|
||||
//
|
||||
// Created by mwo on 27/04/16.
|
||||
//
|
||||
#ifndef XMRLMDBCPP_MYLMDB_H
|
||||
#define XMRLMDBCPP_MYLMDB_H
|
||||
|
||||
#include "../ext/lmdb++.h"
|
||||
|
||||
#include <iostream>
|
||||
#include <memory>
|
||||
|
||||
namespace xmreg
|
||||
{
|
||||
|
||||
using epee::string_tools::pod_to_hex;
|
||||
using epee::string_tools::hex_to_pod;
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* Stores info about outputs useful
|
||||
* for checking which ouputs belong to a
|
||||
* given address and viewkey
|
||||
*/
|
||||
struct output_info
|
||||
{
|
||||
crypto::public_key out_pub_key;
|
||||
crypto::hash tx_hash;
|
||||
crypto::public_key tx_pub_key;
|
||||
uint64_t amount;
|
||||
uint64_t index_in_tx;
|
||||
};
|
||||
|
||||
std::ostream& operator<<(std::ostream& os, const output_info& out_info)
|
||||
{
|
||||
os << ", out_pub_key: " << out_info.out_pub_key
|
||||
<< ", tx_hash: " << out_info.tx_hash
|
||||
<< ", tx_pub_key: " << out_info.tx_pub_key
|
||||
<< ", amount: " << XMR_AMOUNT(out_info.amount)
|
||||
<< ", index_in_tx: " << out_info.index_in_tx;
|
||||
|
||||
return os;
|
||||
}
|
||||
|
||||
class MyLMDB
|
||||
{
|
||||
|
||||
|
||||
static const uint64_t DEFAULT_MAPSIZE = 30UL * 1024UL * 1024UL * 1024UL; /* 30 GiB */
|
||||
static const uint64_t DEFAULT_NO_DBs = 10;
|
||||
|
||||
|
||||
string m_db_path;
|
||||
|
||||
uint64_t m_mapsize;
|
||||
uint64_t m_no_dbs;
|
||||
|
||||
lmdb::env m_env;
|
||||
|
||||
|
||||
public:
|
||||
MyLMDB(string _path,
|
||||
uint64_t _mapsize = DEFAULT_MAPSIZE,
|
||||
uint64_t _no_dbs = DEFAULT_NO_DBs)
|
||||
: m_db_path {_path},
|
||||
m_mapsize {_mapsize},
|
||||
m_no_dbs {_no_dbs},
|
||||
m_env {nullptr}
|
||||
{
|
||||
create_and_open_env();
|
||||
}
|
||||
|
||||
bool
|
||||
create_and_open_env()
|
||||
{
|
||||
try
|
||||
{ m_env = lmdb::env::create();
|
||||
m_env.set_mapsize(m_mapsize);
|
||||
m_env.set_max_dbs(m_no_dbs);
|
||||
m_env.open(m_db_path.c_str(), MDB_CREATE, 0664);
|
||||
}
|
||||
catch (lmdb::error& e )
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
bool
|
||||
write_key_images(const transaction& tx)
|
||||
{
|
||||
crypto::hash tx_hash = get_transaction_hash(tx);
|
||||
|
||||
string tx_hash_str = pod_to_hex(tx_hash);
|
||||
|
||||
vector<cryptonote::txin_to_key> key_images
|
||||
= xmreg::get_key_images(tx);
|
||||
|
||||
lmdb::txn wtxn {nullptr};
|
||||
lmdb::dbi wdbi {0};
|
||||
|
||||
unsigned int flags = MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED;
|
||||
|
||||
try
|
||||
{
|
||||
wtxn = lmdb::txn::begin(m_env);
|
||||
wdbi = lmdb::dbi::open(wtxn, "key_images", flags);
|
||||
}
|
||||
catch (lmdb::error& e )
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
for (const cryptonote::txin_to_key& key_image: key_images)
|
||||
{
|
||||
string key_img_str = pod_to_hex(key_image.k_image);
|
||||
|
||||
lmdb::val key_img_val {key_img_str};
|
||||
lmdb::val tx_hash_val {tx_hash_str};
|
||||
|
||||
wdbi.put(wtxn, key_img_val, tx_hash_val);
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
wtxn.commit();
|
||||
}
|
||||
catch (lmdb::error& e )
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool
|
||||
write_output_public_keys(const transaction& tx, const block& blk)
|
||||
{
|
||||
crypto::hash tx_hash = get_transaction_hash(tx);
|
||||
|
||||
crypto::public_key tx_pub_key = get_tx_pub_key_from_extra(tx);
|
||||
|
||||
string tx_hash_str = pod_to_hex(tx_hash);
|
||||
|
||||
vector<tuple<txout_to_key, uint64_t, uint64_t>> outputs =
|
||||
xmreg::get_ouputs_tuple(tx);
|
||||
|
||||
lmdb::txn wtxn {nullptr};
|
||||
lmdb::dbi wdbi1 {0};
|
||||
lmdb::dbi wdbi2 {0};
|
||||
lmdb::dbi wdbi3 {0};
|
||||
|
||||
unsigned int flags = MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED;
|
||||
|
||||
try
|
||||
{
|
||||
wtxn = lmdb::txn::begin(m_env);
|
||||
wdbi1 = lmdb::dbi::open(wtxn, "output_public_keys", flags);
|
||||
wdbi2 = lmdb::dbi::open(wtxn, "output_amounts", flags);
|
||||
wdbi3 = lmdb::dbi::open(wtxn, "output_info",
|
||||
flags | MDB_INTEGERKEY | MDB_INTEGERDUP);
|
||||
}
|
||||
catch (lmdb::error& e )
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
for (auto& output: outputs)
|
||||
{
|
||||
|
||||
public_key out_pub_key = std::get<0>(output).key;
|
||||
|
||||
string public_key_str = pod_to_hex(out_pub_key);
|
||||
|
||||
lmdb::val public_key_val {public_key_str};
|
||||
lmdb::val tx_hash_val {tx_hash_str};
|
||||
|
||||
uint64_t amount = std::get<1>(output);
|
||||
|
||||
lmdb::val amount_val {static_cast<void*>(&amount), sizeof(amount)};
|
||||
|
||||
uint64_t index_in_tx = std::get<2>(output);
|
||||
|
||||
output_info out_info {out_pub_key, tx_hash,
|
||||
tx_pub_key, amount,
|
||||
index_in_tx};
|
||||
|
||||
uint64_t out_timestamp = blk.timestamp;
|
||||
|
||||
lmdb::val out_timestamp_val {static_cast<void*>(&out_timestamp),
|
||||
sizeof(out_timestamp)};
|
||||
lmdb::val out_info_val {static_cast<void*>(&out_info),
|
||||
sizeof(out_info)};
|
||||
|
||||
wdbi1.put(wtxn, public_key_val, tx_hash_val);
|
||||
wdbi2.put(wtxn, public_key_val, amount_val);
|
||||
wdbi3.put(wtxn, out_timestamp_val, out_info_val);
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
wtxn.commit();
|
||||
}
|
||||
catch (lmdb::error& e )
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool
|
||||
write_tx_public_key(const transaction& tx)
|
||||
{
|
||||
crypto::hash tx_hash = get_transaction_hash(tx);
|
||||
|
||||
string tx_hash_str = pod_to_hex(tx_hash);
|
||||
|
||||
unsigned int flags = MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED;
|
||||
|
||||
public_key pk = get_tx_pub_key_from_extra(tx);
|
||||
|
||||
string pk_str = pod_to_hex(pk);
|
||||
|
||||
try
|
||||
{
|
||||
lmdb::txn wtxn = lmdb::txn::begin(m_env);
|
||||
lmdb::dbi wdbi = lmdb::dbi::open(wtxn, "tx_public_keys", flags);
|
||||
|
||||
//cout << "Saving public_key: " << pk_str << endl;
|
||||
|
||||
lmdb::val public_key_val {pk_str};
|
||||
lmdb::val tx_hash_val {tx_hash_str};
|
||||
|
||||
wdbi.put(wtxn, public_key_val, tx_hash_val);
|
||||
|
||||
wtxn.commit();
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool
|
||||
write_payment_id(const transaction& tx)
|
||||
{
|
||||
crypto::hash tx_hash = get_transaction_hash(tx);
|
||||
|
||||
string tx_hash_str = pod_to_hex(tx_hash);
|
||||
|
||||
crypto::hash payment_id;
|
||||
crypto::hash8 payment_id8;
|
||||
|
||||
get_payment_id(tx, payment_id, payment_id8);
|
||||
|
||||
if (payment_id == null_hash)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
unsigned int flags = MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED;
|
||||
|
||||
string payment_id_str = pod_to_hex(payment_id);
|
||||
|
||||
try
|
||||
{
|
||||
lmdb::txn wtxn = lmdb::txn::begin(m_env);
|
||||
lmdb::dbi wdbi = lmdb::dbi::open(wtxn, "payments_id", flags);
|
||||
|
||||
//cout << "Saving payiment_id: " << payment_id_str << endl;
|
||||
|
||||
lmdb::val payment_id_val {payment_id_str};
|
||||
lmdb::val tx_hash_val {tx_hash_str};
|
||||
|
||||
wdbi.put(wtxn, payment_id_val, tx_hash_val);
|
||||
|
||||
wtxn.commit();
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool
|
||||
write_encrypted_payment_id(const transaction& tx)
|
||||
{
|
||||
crypto::hash tx_hash = get_transaction_hash(tx);
|
||||
|
||||
string tx_hash_str = pod_to_hex(tx_hash);
|
||||
|
||||
crypto::hash payment_id;
|
||||
crypto::hash8 payment_id8;
|
||||
|
||||
get_payment_id(tx, payment_id, payment_id8);
|
||||
|
||||
if (payment_id8 == null_hash8)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
unsigned int flags = MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED;
|
||||
|
||||
string payment_id_str = pod_to_hex(payment_id8);
|
||||
|
||||
try
|
||||
{
|
||||
lmdb::txn wtxn = lmdb::txn::begin(m_env);
|
||||
lmdb::dbi wdbi = lmdb::dbi::open(wtxn, "encrypted_payments_id", flags);
|
||||
|
||||
//cout << "Saving encrypted payiment_id: " << payment_id_str << endl;
|
||||
//string wait_for_enter;
|
||||
//cin >> wait_for_enter;
|
||||
|
||||
lmdb::val payment_id_val {payment_id_str};
|
||||
lmdb::val tx_hash_val {tx_hash_str};
|
||||
|
||||
wdbi.put(wtxn, payment_id_val, tx_hash_val);
|
||||
|
||||
wtxn.commit();
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// // this seems to be not needed as outputs are written based on timestamps
|
||||
//
|
||||
// bool
|
||||
// write_block_timestamp(uint64_t& blk_timestamp, uint64_t& blk_height)
|
||||
// {
|
||||
//
|
||||
// unsigned int flags = MDB_CREATE | MDB_INTEGERKEY;
|
||||
//
|
||||
// try
|
||||
// {
|
||||
// lmdb::txn wtxn = lmdb::txn::begin(m_env);
|
||||
// lmdb::dbi wdbi = lmdb::dbi::open(wtxn, "block_timestamps", flags);
|
||||
//
|
||||
// lmdb::val blk_timestamp_val {static_cast<void*>(&blk_timestamp),
|
||||
// sizeof(blk_timestamp)};
|
||||
// lmdb::val blk_height_val {static_cast<void*>(&blk_height),
|
||||
// sizeof(blk_height)};
|
||||
//
|
||||
// wdbi.put(wtxn, blk_timestamp_val, blk_height_val);
|
||||
//
|
||||
// wtxn.commit();
|
||||
// }
|
||||
// catch (lmdb::error& e)
|
||||
// {
|
||||
// cerr << e.what() << endl;
|
||||
// return false;
|
||||
// }
|
||||
//
|
||||
// return true;
|
||||
// }
|
||||
|
||||
bool
|
||||
search(const string& key,
|
||||
vector<string>& found_tx_hashes,
|
||||
const string& db_name = "key_images")
|
||||
{
|
||||
unsigned int flags = MDB_DUPSORT | MDB_DUPFIXED;
|
||||
|
||||
try
|
||||
{
|
||||
|
||||
lmdb::txn rtxn = lmdb::txn::begin(m_env, nullptr, MDB_RDONLY);
|
||||
lmdb::dbi rdbi = lmdb::dbi::open(rtxn, db_name.c_str(), flags);
|
||||
lmdb::cursor cr = lmdb::cursor::open(rtxn, rdbi);
|
||||
|
||||
lmdb::val key_to_find{key};
|
||||
lmdb::val tx_hash_val;
|
||||
|
||||
// set cursor the the first item
|
||||
if (cr.get(key_to_find, tx_hash_val, MDB_SET))
|
||||
{
|
||||
//cout << key_val_to_str(key_to_find, tx_hash_val) << endl;
|
||||
found_tx_hashes.push_back(string(tx_hash_val.data(), tx_hash_val.size()));
|
||||
|
||||
// process other values for the same key
|
||||
while (cr.get(key_to_find, tx_hash_val, MDB_NEXT_DUP))
|
||||
{
|
||||
//cout << key_val_to_str(key_to_find, tx_hash_val) << endl;
|
||||
found_tx_hashes.push_back(string(tx_hash_val.data(), tx_hash_val.size()));
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
cr.close();
|
||||
rtxn.abort();
|
||||
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool
|
||||
get_output_amount(const string& key,
|
||||
uint64_t& amount,
|
||||
const string& db_name = "output_amounts")
|
||||
{
|
||||
|
||||
unsigned int flags = 0;
|
||||
|
||||
try
|
||||
{
|
||||
|
||||
lmdb::txn rtxn = lmdb::txn::begin(m_env, nullptr, MDB_RDONLY);
|
||||
lmdb::dbi rdbi = lmdb::dbi::open(rtxn, db_name.c_str(), flags);
|
||||
|
||||
lmdb::val key_to_find{key};
|
||||
lmdb::val amount_val;
|
||||
|
||||
if(!rdbi.get(rtxn, key_to_find, amount_val))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
amount = *(amount_val.data<uint64_t>());
|
||||
|
||||
rtxn.abort();
|
||||
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
bool
|
||||
get_output_info(uint64_t key_timestamp,
|
||||
vector<output_info>& out_infos,
|
||||
const string& db_name = "output_info")
|
||||
{
|
||||
|
||||
unsigned int flags = 0;
|
||||
|
||||
try
|
||||
{
|
||||
|
||||
lmdb::txn rtxn = lmdb::txn::begin(m_env, nullptr, MDB_RDONLY);
|
||||
lmdb::dbi rdbi = lmdb::dbi::open(rtxn, db_name.c_str(), flags);
|
||||
|
||||
lmdb::val key_to_find{static_cast<void*>(&key_timestamp),
|
||||
sizeof(key_timestamp)};
|
||||
lmdb::val info_val;
|
||||
|
||||
lmdb::cursor cr = lmdb::cursor::open(rtxn, rdbi);
|
||||
|
||||
// set cursor the the first item
|
||||
if (cr.get(key_to_find, info_val, MDB_SET_RANGE))
|
||||
{
|
||||
out_infos.push_back(*(info_val.data<output_info>()));
|
||||
|
||||
// process other values for the same key
|
||||
while (cr.get(key_to_find, info_val, MDB_NEXT_DUP))
|
||||
{
|
||||
//cout << key_val_to_str(key_to_find, tx_hash_val) << endl;
|
||||
out_infos.push_back(*(info_val.data<output_info>()));
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
rtxn.abort();
|
||||
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool
|
||||
get_output_info_range(uint64_t key_timestamp_start,
|
||||
uint64_t key_timestamp_end,
|
||||
vector<pair<uint64_t, output_info>>& out_infos,
|
||||
const string& db_name = "output_info")
|
||||
{
|
||||
|
||||
unsigned int flags = 0;
|
||||
|
||||
try
|
||||
{
|
||||
lmdb::txn rtxn = lmdb::txn::begin(m_env, nullptr, MDB_RDONLY);
|
||||
lmdb::dbi rdbi = lmdb::dbi::open(rtxn, db_name.c_str(), flags);
|
||||
|
||||
lmdb::val key_to_find{static_cast<void*>(&key_timestamp_start),
|
||||
sizeof(key_timestamp_start)};
|
||||
lmdb::val info_val;
|
||||
|
||||
lmdb::cursor cr = lmdb::cursor::open(rtxn, rdbi);
|
||||
|
||||
uint64_t current_timestamp = key_timestamp_start;
|
||||
|
||||
// set cursor the the first item
|
||||
if (cr.get(key_to_find, info_val, MDB_SET_RANGE))
|
||||
{
|
||||
|
||||
current_timestamp = *key_to_find.data<uint64_t>();
|
||||
|
||||
if (current_timestamp > key_timestamp_end)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
out_infos.push_back(make_pair(
|
||||
current_timestamp,
|
||||
*(info_val.data<output_info>())));
|
||||
|
||||
// process other values for the same key
|
||||
while (cr.get(key_to_find, info_val, MDB_NEXT))
|
||||
{
|
||||
current_timestamp = *key_to_find.data<uint64_t>();
|
||||
|
||||
if (current_timestamp > key_timestamp_end)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
out_infos.push_back(make_pair(
|
||||
current_timestamp,
|
||||
*(info_val.data<output_info>())));
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
rtxn.abort();
|
||||
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns sorted and unique tx hashes withing a
|
||||
* given timestamp range
|
||||
*
|
||||
* @param key_timestamp_start
|
||||
* @param key_timestamp_end
|
||||
* @param out_txs
|
||||
* @return bool
|
||||
*/
|
||||
bool
|
||||
get_txs_from_timestamp_range(uint64_t key_timestamp_start,
|
||||
uint64_t key_timestamp_end,
|
||||
vector<crypto::hash>& out_txs)
|
||||
{
|
||||
using output_pair = pair<uint64_t, output_info>;
|
||||
|
||||
auto sort_by_timestamp = [](const output_pair& l,
|
||||
const output_pair& r)
|
||||
{
|
||||
return l.first < r.first;
|
||||
};
|
||||
|
||||
vector<output_pair> out_infos;
|
||||
|
||||
if (get_output_info_range(key_timestamp_start,
|
||||
key_timestamp_end,
|
||||
out_infos))
|
||||
{
|
||||
|
||||
set<output_pair, decltype(sort_by_timestamp)> unique_txs(sort_by_timestamp);
|
||||
|
||||
for (auto oi: out_infos)
|
||||
unique_txs.insert(oi);
|
||||
|
||||
for (auto ut: unique_txs)
|
||||
out_txs.push_back(ut.second.tx_hash);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
|
||||
void
|
||||
for_all_outputs(
|
||||
std::function<bool(public_key& out_pubkey,
|
||||
output_info& out_info)> f)
|
||||
{
|
||||
unsigned int flags = MDB_DUPSORT | MDB_DUPFIXED;
|
||||
|
||||
try
|
||||
{
|
||||
|
||||
lmdb::txn rtxn = lmdb::txn::begin(m_env, nullptr, MDB_RDONLY);
|
||||
lmdb::dbi rdbi = lmdb::dbi::open(rtxn, "output_info", flags);
|
||||
lmdb::cursor cr = lmdb::cursor::open(rtxn, rdbi);
|
||||
|
||||
lmdb::val key_to_find;
|
||||
lmdb::val amount_val;
|
||||
|
||||
|
||||
// process all values for the same key
|
||||
while (cr.get(key_to_find, amount_val, MDB_NEXT))
|
||||
{
|
||||
public_key pub_key;
|
||||
|
||||
hex_to_pod(string(key_to_find.data(), key_to_find.size()),
|
||||
pub_key);
|
||||
|
||||
output_info out_info = *(amount_val.data<output_info>());
|
||||
|
||||
if (f(pub_key, out_info) == false)
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
cr.close();
|
||||
rtxn.abort();
|
||||
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
print_all(const string& db_name)
|
||||
{
|
||||
unsigned int flags = MDB_DUPSORT | MDB_DUPFIXED;
|
||||
|
||||
try
|
||||
{
|
||||
|
||||
lmdb::txn rtxn = lmdb::txn::begin(m_env, nullptr, MDB_RDONLY);
|
||||
lmdb::dbi rdbi = lmdb::dbi::open(rtxn, db_name.c_str(), flags);
|
||||
lmdb::cursor cr = lmdb::cursor::open(rtxn, rdbi);
|
||||
|
||||
lmdb::val key_to_find;
|
||||
lmdb::val tx_hash_val;
|
||||
|
||||
|
||||
// process other values for the same key
|
||||
while (cr.get(key_to_find, tx_hash_val, MDB_NEXT))
|
||||
{
|
||||
cout << key_val_to_str(key_to_find, tx_hash_val) << endl;
|
||||
}
|
||||
|
||||
cr.close();
|
||||
rtxn.abort();
|
||||
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
}
|
||||
}
|
||||
|
||||
static uint64_t
|
||||
get_blockchain_height(string blk_path = "/home/mwo/.blockchain/lmdb")
|
||||
{
|
||||
uint64_t height {0};
|
||||
|
||||
try
|
||||
{
|
||||
auto env = lmdb::env::create();
|
||||
env.set_mapsize(DEFAULT_MAPSIZE * 3);
|
||||
env.set_max_dbs(20);
|
||||
env.open(blk_path.c_str(), MDB_CREATE, 0664);
|
||||
|
||||
//auto rtxn = lmdb::txn::begin(env, nullptr, MDB_RDONLY);
|
||||
auto rtxn = lmdb::txn::begin(env, nullptr);
|
||||
auto rdbi = lmdb::dbi::open(rtxn, "blocks");
|
||||
|
||||
MDB_stat stats = rdbi.stat(rtxn);
|
||||
|
||||
height = static_cast<uint64_t>(stats.ms_entries);
|
||||
|
||||
rtxn.abort();
|
||||
}
|
||||
catch (lmdb::error& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return height;
|
||||
}
|
||||
catch (exception& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
return height;
|
||||
}
|
||||
|
||||
//cout << height << endl;
|
||||
|
||||
return height;
|
||||
|
||||
}
|
||||
|
||||
string
|
||||
key_val_to_str(const lmdb::val& key, const lmdb::val& val)
|
||||
{
|
||||
return "key: " + string(key.data(), key.size())
|
||||
+ ", val: " + string(val.data(), val.size());
|
||||
}
|
||||
|
||||
|
||||
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif //XMRLMDBCPP_MYLMDB_H
|
||||
@ -0,0 +1,83 @@
|
||||
//
|
||||
// Created by mwo on 15/06/18.
|
||||
//
|
||||
|
||||
#include "../src/MicroCore.h"
|
||||
#include "../src/CurrentBlockchainStatus.h"
|
||||
#include "../src/ThreadRAII.h"
|
||||
|
||||
#include "gmock/gmock.h"
|
||||
#include "gtest/gtest.h"
|
||||
|
||||
|
||||
|
||||
namespace
|
||||
{
|
||||
|
||||
|
||||
using json = nlohmann::json;
|
||||
using namespace std;
|
||||
using namespace cryptonote;
|
||||
using namespace epee::string_tools;
|
||||
using namespace std::chrono_literals;
|
||||
|
||||
using ::testing::AllOf;
|
||||
using ::testing::Ge;
|
||||
using ::testing::Le;
|
||||
using ::testing::HasSubstr;
|
||||
using ::testing::Not;
|
||||
using ::testing::internal::FilePath;
|
||||
|
||||
|
||||
class BCSTATUS_TEST : public ::testing::Test
|
||||
{
|
||||
public:
|
||||
|
||||
static void
|
||||
SetUpTestCase()
|
||||
{
|
||||
string config_path {"../config/config.json"};
|
||||
config_json = xmreg::BlockchainSetup::read_config(config_path);
|
||||
}
|
||||
|
||||
protected:
|
||||
|
||||
virtual void
|
||||
SetUp()
|
||||
{
|
||||
bc_setup = xmreg::BlockchainSetup{net_type, do_not_relay, config_json};
|
||||
}
|
||||
|
||||
network_type net_type {network_type::STAGENET};
|
||||
bool do_not_relay {false};
|
||||
xmreg::BlockchainSetup bc_setup;
|
||||
|
||||
static json config_json;
|
||||
};
|
||||
|
||||
json BCSTATUS_TEST::config_json;
|
||||
|
||||
TEST_F(BCSTATUS_TEST, DefaultConstruction)
|
||||
{
|
||||
xmreg::CurrentBlockchainStatus bcs {bc_setup, nullptr};
|
||||
EXPECT_TRUE(true);
|
||||
}
|
||||
|
||||
class MockMicroCore : public xmreg::MicroCore
|
||||
{
|
||||
bool
|
||||
init(const string& _blockchain_path, network_type nt)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
};
|
||||
|
||||
TEST_F(BCSTATUS_TEST, InitMoneroBlockchain)
|
||||
{
|
||||
std::unique_ptr<MockMicroCore> mcore = std::make_unique<MockMicroCore>();
|
||||
xmreg::CurrentBlockchainStatus bcs {bc_setup, std::move(mcore)};
|
||||
EXPECT_TRUE(bcs.init_monero_blockchain());
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
Loading…
Reference in new issue