2016-04-28 03:37:48 +00:00
|
|
|
//
|
|
|
|
// 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;
|
2016-05-13 01:38:23 +00:00
|
|
|
using epee::string_tools::hex_to_pod;
|
2016-04-28 03:37:48 +00:00
|
|
|
|
|
|
|
using namespace std;
|
|
|
|
|
|
|
|
|
2016-05-11 06:32:02 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Stores info about outputs useful
|
|
|
|
* for checking which ouputs belong to a
|
|
|
|
* given address and viewkey
|
|
|
|
*/
|
|
|
|
struct output_info
|
|
|
|
{
|
2016-05-21 00:17:31 +00:00
|
|
|
crypto::public_key out_pub_key;
|
2016-05-11 06:32:02 +00:00
|
|
|
crypto::hash tx_hash;
|
|
|
|
crypto::public_key tx_pub_key;
|
|
|
|
uint64_t amount;
|
|
|
|
uint64_t index_in_tx;
|
|
|
|
};
|
|
|
|
|
2016-05-21 00:17:31 +00:00
|
|
|
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;
|
2016-05-11 06:32:02 +00:00
|
|
|
|
2016-05-21 00:17:31 +00:00
|
|
|
return os;
|
|
|
|
}
|
2016-05-11 06:32:02 +00:00
|
|
|
|
2016-04-28 03:37:48 +00:00
|
|
|
class MyLMDB
|
|
|
|
{
|
|
|
|
|
|
|
|
|
2016-05-12 05:57:14 +00:00
|
|
|
static const uint64_t DEFAULT_MAPSIZE = 30UL * 1024UL * 1024UL * 1024UL; /* 30 GiB */
|
2016-04-30 09:23:37 +00:00
|
|
|
static const uint64_t DEFAULT_NO_DBs = 10;
|
2016-04-28 03:37:48 +00:00
|
|
|
|
|
|
|
|
|
|
|
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
|
2016-05-21 00:17:31 +00:00
|
|
|
write_output_public_keys(const transaction& tx, const block& blk)
|
2016-04-28 03:37:48 +00:00
|
|
|
{
|
|
|
|
crypto::hash tx_hash = get_transaction_hash(tx);
|
|
|
|
|
2016-05-11 06:32:02 +00:00
|
|
|
crypto::public_key tx_pub_key = get_tx_pub_key_from_extra(tx);
|
|
|
|
|
2016-04-28 03:37:48 +00:00
|
|
|
string tx_hash_str = pod_to_hex(tx_hash);
|
|
|
|
|
2016-05-11 06:32:02 +00:00
|
|
|
vector<tuple<txout_to_key, uint64_t, uint64_t>> outputs =
|
|
|
|
xmreg::get_ouputs_tuple(tx);
|
2016-04-28 03:37:48 +00:00
|
|
|
|
2016-04-30 09:23:37 +00:00
|
|
|
lmdb::txn wtxn {nullptr};
|
|
|
|
lmdb::dbi wdbi1 {0};
|
|
|
|
lmdb::dbi wdbi2 {0};
|
2016-05-11 06:32:02 +00:00
|
|
|
lmdb::dbi wdbi3 {0};
|
2016-04-28 03:37:48 +00:00
|
|
|
|
|
|
|
unsigned int flags = MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED;
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
2016-04-30 09:23:37 +00:00
|
|
|
wtxn = lmdb::txn::begin(m_env);
|
|
|
|
wdbi1 = lmdb::dbi::open(wtxn, "output_public_keys", flags);
|
|
|
|
wdbi2 = lmdb::dbi::open(wtxn, "output_amounts", flags);
|
2016-05-21 00:17:31 +00:00
|
|
|
wdbi3 = lmdb::dbi::open(wtxn, "output_info",
|
|
|
|
flags | MDB_INTEGERKEY | MDB_INTEGERDUP);
|
2016-04-28 03:37:48 +00:00
|
|
|
}
|
|
|
|
catch (lmdb::error& e )
|
|
|
|
{
|
|
|
|
cerr << e.what() << endl;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2016-05-11 06:32:02 +00:00
|
|
|
for (auto& output: outputs)
|
2016-04-28 03:37:48 +00:00
|
|
|
{
|
2016-05-21 00:17:31 +00:00
|
|
|
|
|
|
|
public_key out_pub_key = std::get<0>(output).key;
|
|
|
|
|
|
|
|
string public_key_str = pod_to_hex(out_pub_key);
|
2016-04-28 03:37:48 +00:00
|
|
|
|
|
|
|
lmdb::val public_key_val {public_key_str};
|
|
|
|
lmdb::val tx_hash_val {tx_hash_str};
|
|
|
|
|
2016-05-11 06:32:02 +00:00
|
|
|
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);
|
|
|
|
|
2016-11-25 00:43:59 +00:00
|
|
|
output_info out_info {out_pub_key, tx_hash,
|
|
|
|
tx_pub_key, amount,
|
|
|
|
index_in_tx};
|
2016-05-11 06:32:02 +00:00
|
|
|
|
2016-05-21 00:17:31 +00:00
|
|
|
uint64_t out_timestamp = blk.timestamp;
|
2016-05-11 06:32:02 +00:00
|
|
|
|
2016-05-21 00:17:31 +00:00
|
|
|
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)};
|
2016-04-30 09:23:37 +00:00
|
|
|
|
|
|
|
wdbi1.put(wtxn, public_key_val, tx_hash_val);
|
|
|
|
wdbi2.put(wtxn, public_key_val, amount_val);
|
2016-05-21 00:17:31 +00:00
|
|
|
wdbi3.put(wtxn, out_timestamp_val, out_info_val);
|
2016-04-28 03:37:48 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
wtxn.commit();
|
|
|
|
}
|
|
|
|
catch (lmdb::error& e )
|
|
|
|
{
|
|
|
|
cerr << e.what() << endl;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2016-04-28 22:49:10 +00:00
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2016-04-28 03:37:48 +00:00
|
|
|
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;
|
|
|
|
|
2016-04-28 22:49:10 +00:00
|
|
|
lmdb::val payment_id_val {payment_id_str};
|
|
|
|
lmdb::val tx_hash_val {tx_hash_str};
|
2016-04-28 03:37:48 +00:00
|
|
|
|
|
|
|
wdbi.put(wtxn, payment_id_val, tx_hash_val);
|
|
|
|
|
|
|
|
wtxn.commit();
|
|
|
|
}
|
|
|
|
catch (lmdb::error& e)
|
|
|
|
{
|
|
|
|
cerr << e.what() << endl;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2016-04-30 09:23:37 +00:00
|
|
|
bool
|
|
|
|
write_encrypted_payment_id(const transaction& tx)
|
2016-04-28 03:37:48 +00:00
|
|
|
{
|
2016-04-30 09:23:37 +00:00
|
|
|
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);
|
|
|
|
|
2016-05-11 06:32:02 +00:00
|
|
|
//cout << "Saving encrypted payiment_id: " << payment_id_str << endl;
|
|
|
|
//string wait_for_enter;
|
|
|
|
//cin >> wait_for_enter;
|
|
|
|
|
2016-04-30 09:23:37 +00:00
|
|
|
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;
|
|
|
|
}
|
2016-04-28 03:37:48 +00:00
|
|
|
|
2016-04-30 09:23:37 +00:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2016-11-25 00:43:59 +00:00
|
|
|
// // 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;
|
|
|
|
// }
|
|
|
|
|
2016-04-30 09:23:37 +00:00
|
|
|
bool
|
|
|
|
search(const string& key,
|
|
|
|
vector<string>& found_tx_hashes,
|
|
|
|
const string& db_name = "key_images")
|
|
|
|
{
|
2016-04-28 03:37:48 +00:00
|
|
|
unsigned int flags = MDB_DUPSORT | MDB_DUPFIXED;
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
2016-05-10 10:21:20 +00:00
|
|
|
|
2016-04-28 03:37:48 +00:00
|
|
|
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))
|
|
|
|
{
|
2016-04-28 04:18:49 +00:00
|
|
|
//cout << key_val_to_str(key_to_find, tx_hash_val) << endl;
|
2016-04-30 09:23:37 +00:00
|
|
|
found_tx_hashes.push_back(string(tx_hash_val.data(), tx_hash_val.size()));
|
2016-04-28 03:37:48 +00:00
|
|
|
|
|
|
|
// process other values for the same key
|
2016-04-28 22:49:10 +00:00
|
|
|
while (cr.get(key_to_find, tx_hash_val, MDB_NEXT_DUP))
|
|
|
|
{
|
2016-04-28 04:18:49 +00:00
|
|
|
//cout << key_val_to_str(key_to_find, tx_hash_val) << endl;
|
2016-04-30 09:23:37 +00:00
|
|
|
found_tx_hashes.push_back(string(tx_hash_val.data(), tx_hash_val.size()));
|
2016-04-28 03:37:48 +00:00
|
|
|
}
|
|
|
|
}
|
2016-04-30 09:23:37 +00:00
|
|
|
else
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
2016-04-28 03:37:48 +00:00
|
|
|
|
|
|
|
cr.close();
|
|
|
|
rtxn.abort();
|
|
|
|
|
|
|
|
}
|
|
|
|
catch (lmdb::error& e)
|
|
|
|
{
|
|
|
|
cerr << e.what() << endl;
|
2016-04-30 09:23:37 +00:00
|
|
|
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;
|
2016-04-28 03:37:48 +00:00
|
|
|
}
|
|
|
|
|
2016-04-30 09:23:37 +00:00
|
|
|
return true;
|
2016-04-28 03:37:48 +00:00
|
|
|
}
|
|
|
|
|
2016-05-10 10:21:20 +00:00
|
|
|
|
2016-05-11 06:32:02 +00:00
|
|
|
bool
|
2016-05-21 00:17:31 +00:00
|
|
|
get_output_info(uint64_t key_timestamp,
|
|
|
|
vector<output_info>& out_infos,
|
2016-05-11 06:32:02 +00:00
|
|
|
const string& db_name = "output_info")
|
2016-05-10 10:21:20 +00:00
|
|
|
{
|
2016-05-11 06:32:02 +00:00
|
|
|
|
|
|
|
unsigned int flags = 0;
|
2016-05-10 10:21:20 +00:00
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
|
|
|
|
lmdb::txn rtxn = lmdb::txn::begin(m_env, nullptr, MDB_RDONLY);
|
2016-05-11 06:32:02 +00:00
|
|
|
lmdb::dbi rdbi = lmdb::dbi::open(rtxn, db_name.c_str(), flags);
|
2016-05-10 10:21:20 +00:00
|
|
|
|
2016-05-21 00:17:31 +00:00
|
|
|
lmdb::val key_to_find{static_cast<void*>(&key_timestamp),
|
|
|
|
sizeof(key_timestamp)};
|
2016-05-11 06:32:02 +00:00
|
|
|
lmdb::val info_val;
|
2016-05-10 10:21:20 +00:00
|
|
|
|
2016-05-21 00:17:31 +00:00
|
|
|
|
|
|
|
|
|
|
|
lmdb::cursor cr = lmdb::cursor::open(rtxn, rdbi);
|
|
|
|
|
|
|
|
|
|
|
|
// set cursor the the first item
|
2016-11-23 06:24:05 +00:00
|
|
|
if (cr.get(key_to_find, info_val, MDB_SET_RANGE))
|
2016-05-21 00:17:31 +00:00
|
|
|
{
|
|
|
|
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
|
2016-05-10 10:21:20 +00:00
|
|
|
{
|
2016-05-11 06:32:02 +00:00
|
|
|
return false;
|
2016-05-10 10:21:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
rtxn.abort();
|
|
|
|
|
|
|
|
}
|
|
|
|
catch (lmdb::error& e)
|
|
|
|
{
|
|
|
|
cerr << e.what() << endl;
|
2016-05-11 06:32:02 +00:00
|
|
|
return false;
|
2016-05-10 10:21:20 +00:00
|
|
|
}
|
2016-05-11 06:32:02 +00:00
|
|
|
|
|
|
|
return true;
|
2016-05-10 10:21:20 +00:00
|
|
|
}
|
|
|
|
|
2016-11-25 00:43:59 +00:00
|
|
|
bool
|
|
|
|
get_output_info_range(uint64_t key_timestamp_start,
|
|
|
|
uint64_t key_timestamp_end,
|
|
|
|
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_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))
|
|
|
|
{
|
|
|
|
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))
|
|
|
|
{
|
|
|
|
current_timestamp = *key_to_find.data<uint64_t>();
|
|
|
|
//cout << current_timestamp << endl;
|
|
|
|
out_infos.push_back(*(info_val.data<output_info>()));
|
|
|
|
if (current_timestamp > key_timestamp_end)
|
|
|
|
{
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
rtxn.abort();
|
|
|
|
|
|
|
|
}
|
|
|
|
catch (lmdb::error& e)
|
|
|
|
{
|
|
|
|
cerr << e.what() << endl;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2016-05-13 01:38:23 +00:00
|
|
|
|
2016-05-11 06:33:53 +00:00
|
|
|
void
|
2016-05-13 01:38:23 +00:00
|
|
|
for_all_outputs(
|
|
|
|
std::function<bool(public_key& out_pubkey,
|
|
|
|
output_info& out_info)> f)
|
2016-05-11 06:33:53 +00:00
|
|
|
{
|
|
|
|
unsigned int flags = MDB_DUPSORT | MDB_DUPFIXED;
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
|
|
|
|
lmdb::txn rtxn = lmdb::txn::begin(m_env, nullptr, MDB_RDONLY);
|
2016-05-13 01:38:23 +00:00
|
|
|
lmdb::dbi rdbi = lmdb::dbi::open(rtxn, "output_info", flags);
|
2016-05-11 06:33:53 +00:00
|
|
|
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;
|
|
|
|
|
2016-05-13 01:38:23 +00:00
|
|
|
hex_to_pod(string(key_to_find.data(), key_to_find.size()),
|
2016-05-21 00:17:31 +00:00
|
|
|
pub_key);
|
2016-05-11 06:33:53 +00:00
|
|
|
|
2016-05-13 01:38:23 +00:00
|
|
|
output_info out_info = *(amount_val.data<output_info>());
|
2016-05-11 06:33:53 +00:00
|
|
|
|
2016-05-13 01:38:23 +00:00
|
|
|
if (f(pub_key, out_info) == false)
|
2016-05-11 06:33:53 +00:00
|
|
|
{
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
cr.close();
|
|
|
|
rtxn.abort();
|
|
|
|
|
|
|
|
}
|
|
|
|
catch (lmdb::error& e)
|
|
|
|
{
|
|
|
|
cerr << e.what() << endl;
|
|
|
|
}
|
2016-05-12 05:57:14 +00:00
|
|
|
}
|
2016-05-11 06:33:53 +00:00
|
|
|
|
2016-05-13 01:38:23 +00:00
|
|
|
|
2016-04-28 03:37:48 +00:00
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2016-11-25 03:09:25 +00:00
|
|
|
//cout << height << endl;
|
2016-04-28 03:37:48 +00:00
|
|
|
|
|
|
|
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
|