1783 lines
57 KiB
C++
1783 lines
57 KiB
C++
#include <nano/crypto_lib/random_pool.hpp>
|
|
#include <nano/lib/blocks.hpp>
|
|
#include <nano/lib/threading.hpp>
|
|
#include <nano/lib/utility.hpp>
|
|
#include <nano/node/election.hpp>
|
|
#include <nano/node/node.hpp>
|
|
#include <nano/node/wallet.hpp>
|
|
#include <nano/store/lmdb/iterator.hpp>
|
|
|
|
#include <boost/format.hpp>
|
|
#include <boost/polymorphic_cast.hpp>
|
|
#include <boost/property_tree/json_parser.hpp>
|
|
|
|
#include <future>
|
|
|
|
#include <argon2.h>
|
|
|
|
nano::uint256_union nano::wallet_store::check (store::transaction const & transaction_a)
|
|
{
|
|
nano::wallet_value value (entry_get_raw (transaction_a, nano::wallet_store::check_special));
|
|
return value.key;
|
|
}
|
|
|
|
nano::uint256_union nano::wallet_store::salt (store::transaction const & transaction_a)
|
|
{
|
|
nano::wallet_value value (entry_get_raw (transaction_a, nano::wallet_store::salt_special));
|
|
return value.key;
|
|
}
|
|
|
|
void nano::wallet_store::wallet_key (nano::raw_key & prv_a, store::transaction const & transaction_a)
|
|
{
|
|
nano::lock_guard<std::recursive_mutex> lock{ mutex };
|
|
nano::raw_key wallet_l;
|
|
wallet_key_mem.value (wallet_l);
|
|
nano::raw_key password_l;
|
|
password.value (password_l);
|
|
prv_a.decrypt (wallet_l, password_l, salt (transaction_a).owords[0]);
|
|
}
|
|
|
|
void nano::wallet_store::seed (nano::raw_key & prv_a, store::transaction const & transaction_a)
|
|
{
|
|
nano::wallet_value value (entry_get_raw (transaction_a, nano::wallet_store::seed_special));
|
|
nano::raw_key password_l;
|
|
wallet_key (password_l, transaction_a);
|
|
prv_a.decrypt (value.key, password_l, salt (transaction_a).owords[seed_iv_index]);
|
|
}
|
|
|
|
void nano::wallet_store::seed_set (store::transaction const & transaction_a, nano::raw_key const & prv_a)
|
|
{
|
|
nano::raw_key password_l;
|
|
wallet_key (password_l, transaction_a);
|
|
nano::raw_key ciphertext;
|
|
ciphertext.encrypt (prv_a, password_l, salt (transaction_a).owords[seed_iv_index]);
|
|
entry_put_raw (transaction_a, nano::wallet_store::seed_special, nano::wallet_value (ciphertext, 0));
|
|
deterministic_clear (transaction_a);
|
|
}
|
|
|
|
nano::public_key nano::wallet_store::deterministic_insert (store::transaction const & transaction_a)
|
|
{
|
|
auto index (deterministic_index_get (transaction_a));
|
|
auto prv = deterministic_key (transaction_a, index);
|
|
nano::public_key result (nano::pub_key (prv));
|
|
while (exists (transaction_a, result))
|
|
{
|
|
++index;
|
|
prv = deterministic_key (transaction_a, index);
|
|
result = nano::pub_key (prv);
|
|
}
|
|
uint64_t marker (1);
|
|
marker <<= 32;
|
|
marker |= index;
|
|
entry_put_raw (transaction_a, result, nano::wallet_value (marker, 0));
|
|
++index;
|
|
deterministic_index_set (transaction_a, index);
|
|
return result;
|
|
}
|
|
|
|
nano::public_key nano::wallet_store::deterministic_insert (store::transaction const & transaction_a, uint32_t const index)
|
|
{
|
|
auto prv = deterministic_key (transaction_a, index);
|
|
nano::public_key result (nano::pub_key (prv));
|
|
uint64_t marker (1);
|
|
marker <<= 32;
|
|
marker |= index;
|
|
entry_put_raw (transaction_a, result, nano::wallet_value (marker, 0));
|
|
return result;
|
|
}
|
|
|
|
nano::raw_key nano::wallet_store::deterministic_key (store::transaction const & transaction_a, uint32_t index_a)
|
|
{
|
|
debug_assert (valid_password (transaction_a));
|
|
nano::raw_key seed_l;
|
|
seed (seed_l, transaction_a);
|
|
return nano::deterministic_key (seed_l, index_a);
|
|
}
|
|
|
|
uint32_t nano::wallet_store::deterministic_index_get (store::transaction const & transaction_a)
|
|
{
|
|
nano::wallet_value value (entry_get_raw (transaction_a, nano::wallet_store::deterministic_index_special));
|
|
return static_cast<uint32_t> (value.key.number () & static_cast<uint32_t> (-1));
|
|
}
|
|
|
|
void nano::wallet_store::deterministic_index_set (store::transaction const & transaction_a, uint32_t index_a)
|
|
{
|
|
nano::raw_key index_l (index_a);
|
|
nano::wallet_value value (index_l, 0);
|
|
entry_put_raw (transaction_a, nano::wallet_store::deterministic_index_special, value);
|
|
}
|
|
|
|
void nano::wallet_store::deterministic_clear (store::transaction const & transaction_a)
|
|
{
|
|
nano::uint256_union key (0);
|
|
for (auto i (begin (transaction_a)), n (end ()); i != n;)
|
|
{
|
|
switch (key_type (nano::wallet_value (i->second)))
|
|
{
|
|
case nano::key_type::deterministic:
|
|
{
|
|
auto const & key (i->first);
|
|
erase (transaction_a, key);
|
|
i = begin (transaction_a, key);
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
++i;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
deterministic_index_set (transaction_a, 0);
|
|
}
|
|
|
|
bool nano::wallet_store::valid_password (store::transaction const & transaction_a)
|
|
{
|
|
nano::raw_key zero;
|
|
zero.clear ();
|
|
nano::raw_key wallet_key_l;
|
|
wallet_key (wallet_key_l, transaction_a);
|
|
nano::uint256_union check_l;
|
|
check_l.encrypt (zero, wallet_key_l, salt (transaction_a).owords[check_iv_index]);
|
|
bool ok = check (transaction_a) == check_l;
|
|
return ok;
|
|
}
|
|
|
|
bool nano::wallet_store::attempt_password (store::transaction const & transaction_a, std::string const & password_a)
|
|
{
|
|
bool result = false;
|
|
{
|
|
nano::lock_guard<std::recursive_mutex> lock{ mutex };
|
|
nano::raw_key password_l;
|
|
derive_key (password_l, transaction_a, password_a);
|
|
password.value_set (password_l);
|
|
result = !valid_password (transaction_a);
|
|
}
|
|
if (!result)
|
|
{
|
|
switch (version (transaction_a))
|
|
{
|
|
case version_4:
|
|
break;
|
|
default:
|
|
debug_assert (false);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool nano::wallet_store::rekey (store::transaction const & transaction_a, std::string const & password_a)
|
|
{
|
|
nano::lock_guard<std::recursive_mutex> lock{ mutex };
|
|
bool result (false);
|
|
if (valid_password (transaction_a))
|
|
{
|
|
nano::raw_key password_new;
|
|
derive_key (password_new, transaction_a, password_a);
|
|
nano::raw_key wallet_key_l;
|
|
wallet_key (wallet_key_l, transaction_a);
|
|
nano::raw_key password_l;
|
|
password.value (password_l);
|
|
password.value_set (password_new);
|
|
nano::raw_key encrypted;
|
|
encrypted.encrypt (wallet_key_l, password_new, salt (transaction_a).owords[0]);
|
|
nano::raw_key wallet_enc;
|
|
wallet_enc = encrypted;
|
|
wallet_key_mem.value_set (wallet_enc);
|
|
entry_put_raw (transaction_a, nano::wallet_store::wallet_key_special, nano::wallet_value (encrypted, 0));
|
|
}
|
|
else
|
|
{
|
|
result = true;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void nano::wallet_store::derive_key (nano::raw_key & prv_a, store::transaction const & transaction_a, std::string const & password_a)
|
|
{
|
|
auto salt_l (salt (transaction_a));
|
|
kdf.phs (prv_a, password_a, salt_l);
|
|
}
|
|
|
|
nano::fan::fan (nano::raw_key const & key, std::size_t count_a)
|
|
{
|
|
auto first (std::make_unique<nano::raw_key> (key));
|
|
for (auto i (1); i < count_a; ++i)
|
|
{
|
|
auto entry (std::make_unique<nano::raw_key> ());
|
|
nano::random_pool::generate_block (entry->bytes.data (), entry->bytes.size ());
|
|
*first ^= *entry;
|
|
values.push_back (std::move (entry));
|
|
}
|
|
values.push_back (std::move (first));
|
|
}
|
|
|
|
void nano::fan::value (nano::raw_key & prv_a)
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ mutex };
|
|
value_get (prv_a);
|
|
}
|
|
|
|
void nano::fan::value_get (nano::raw_key & prv_a)
|
|
{
|
|
debug_assert (!mutex.try_lock ());
|
|
prv_a.clear ();
|
|
for (auto & i : values)
|
|
{
|
|
prv_a ^= *i;
|
|
}
|
|
}
|
|
|
|
void nano::fan::value_set (nano::raw_key const & value_a)
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ mutex };
|
|
nano::raw_key value_l;
|
|
value_get (value_l);
|
|
*(values[0]) ^= value_l;
|
|
*(values[0]) ^= value_a;
|
|
}
|
|
|
|
// Wallet version number
|
|
nano::account const nano::wallet_store::version_special{};
|
|
// Random number used to salt private key encryption
|
|
nano::account const nano::wallet_store::salt_special (1);
|
|
// Key used to encrypt wallet keys, encrypted itself by the user password
|
|
nano::account const nano::wallet_store::wallet_key_special (2);
|
|
// Check value used to see if password is valid
|
|
nano::account const nano::wallet_store::check_special (3);
|
|
// Representative account to be used if we open a new account
|
|
nano::account const nano::wallet_store::representative_special (4);
|
|
// Wallet seed for deterministic key generation
|
|
nano::account const nano::wallet_store::seed_special (5);
|
|
// Current key index for deterministic keys
|
|
nano::account const nano::wallet_store::deterministic_index_special (6);
|
|
int const nano::wallet_store::special_count (7);
|
|
std::size_t const nano::wallet_store::check_iv_index (0);
|
|
std::size_t const nano::wallet_store::seed_iv_index (1);
|
|
|
|
nano::wallet_store::wallet_store (bool & init_a, nano::kdf & kdf_a, store::transaction & transaction_a, store::lmdb::env & env_a, nano::account representative_a, unsigned fanout_a, std::string const & wallet_a, std::string const & json_a) :
|
|
password (0, fanout_a),
|
|
wallet_key_mem (0, fanout_a),
|
|
kdf (kdf_a),
|
|
env{ env_a }
|
|
{
|
|
init_a = false;
|
|
initialize (transaction_a, init_a, wallet_a);
|
|
if (!init_a)
|
|
{
|
|
MDB_val junk;
|
|
debug_assert (mdb_get (env.tx (transaction_a), handle, nano::store::lmdb::db_val (version_special), &junk) == MDB_NOTFOUND);
|
|
boost::property_tree::ptree wallet_l;
|
|
std::stringstream istream (json_a);
|
|
try
|
|
{
|
|
boost::property_tree::read_json (istream, wallet_l);
|
|
}
|
|
catch (...)
|
|
{
|
|
init_a = true;
|
|
}
|
|
for (auto i (wallet_l.begin ()), n (wallet_l.end ()); i != n; ++i)
|
|
{
|
|
nano::account key;
|
|
init_a = key.decode_hex (i->first);
|
|
if (!init_a)
|
|
{
|
|
nano::raw_key value;
|
|
init_a = value.decode_hex (wallet_l.get<std::string> (i->first));
|
|
if (!init_a)
|
|
{
|
|
entry_put_raw (transaction_a, key, nano::wallet_value (value, 0));
|
|
}
|
|
else
|
|
{
|
|
init_a = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
init_a = true;
|
|
}
|
|
}
|
|
init_a |= mdb_get (env.tx (transaction_a), handle, nano::store::lmdb::db_val (version_special), &junk) != 0;
|
|
init_a |= mdb_get (env.tx (transaction_a), handle, nano::store::lmdb::db_val (wallet_key_special), &junk) != 0;
|
|
init_a |= mdb_get (env.tx (transaction_a), handle, nano::store::lmdb::db_val (salt_special), &junk) != 0;
|
|
init_a |= mdb_get (env.tx (transaction_a), handle, nano::store::lmdb::db_val (check_special), &junk) != 0;
|
|
init_a |= mdb_get (env.tx (transaction_a), handle, nano::store::lmdb::db_val (representative_special), &junk) != 0;
|
|
nano::raw_key key;
|
|
key.clear ();
|
|
password.value_set (key);
|
|
key = entry_get_raw (transaction_a, nano::wallet_store::wallet_key_special).key;
|
|
wallet_key_mem.value_set (key);
|
|
}
|
|
}
|
|
|
|
nano::wallet_store::wallet_store (bool & init_a, nano::kdf & kdf_a, store::transaction & transaction_a, store::lmdb::env & env_a, nano::account representative_a, unsigned fanout_a, std::string const & wallet_a) :
|
|
password (0, fanout_a),
|
|
wallet_key_mem (0, fanout_a),
|
|
kdf (kdf_a),
|
|
env{ env_a }
|
|
{
|
|
init_a = false;
|
|
initialize (transaction_a, init_a, wallet_a);
|
|
if (!init_a)
|
|
{
|
|
int version_status;
|
|
MDB_val version_value;
|
|
version_status = mdb_get (env.tx (transaction_a), handle, nano::store::lmdb::db_val (version_special), &version_value);
|
|
if (version_status == MDB_NOTFOUND)
|
|
{
|
|
version_put (transaction_a, version_current);
|
|
nano::raw_key salt_l;
|
|
random_pool::generate_block (salt_l.bytes.data (), salt_l.bytes.size ());
|
|
entry_put_raw (transaction_a, nano::wallet_store::salt_special, nano::wallet_value (salt_l, 0));
|
|
// Wallet key is a fixed random key that encrypts all entries
|
|
nano::raw_key wallet_key;
|
|
random_pool::generate_block (wallet_key.bytes.data (), sizeof (wallet_key.bytes));
|
|
nano::raw_key password_l;
|
|
password_l.clear ();
|
|
password.value_set (password_l);
|
|
nano::raw_key zero;
|
|
zero.clear ();
|
|
// Wallet key is encrypted by the user's password
|
|
nano::raw_key encrypted;
|
|
encrypted.encrypt (wallet_key, zero, salt_l.owords[0]);
|
|
entry_put_raw (transaction_a, nano::wallet_store::wallet_key_special, nano::wallet_value (encrypted, 0));
|
|
nano::raw_key wallet_key_enc;
|
|
wallet_key_enc = encrypted;
|
|
wallet_key_mem.value_set (wallet_key_enc);
|
|
nano::raw_key check;
|
|
check.encrypt (zero, wallet_key, salt_l.owords[check_iv_index]);
|
|
entry_put_raw (transaction_a, nano::wallet_store::check_special, nano::wallet_value (check, 0));
|
|
nano::raw_key rep;
|
|
rep.bytes = representative_a.bytes;
|
|
entry_put_raw (transaction_a, nano::wallet_store::representative_special, nano::wallet_value (rep, 0));
|
|
nano::raw_key seed;
|
|
random_pool::generate_block (seed.bytes.data (), seed.bytes.size ());
|
|
seed_set (transaction_a, seed);
|
|
entry_put_raw (transaction_a, nano::wallet_store::deterministic_index_special, nano::wallet_value (0, 0));
|
|
}
|
|
}
|
|
nano::raw_key key;
|
|
key = entry_get_raw (transaction_a, nano::wallet_store::wallet_key_special).key;
|
|
wallet_key_mem.value_set (key);
|
|
}
|
|
|
|
std::vector<nano::account> nano::wallet_store::accounts (store::transaction const & transaction_a)
|
|
{
|
|
std::vector<nano::account> result;
|
|
for (auto i (begin (transaction_a)), n (end ()); i != n; ++i)
|
|
{
|
|
nano::account const & account (i->first);
|
|
result.push_back (account);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void nano::wallet_store::initialize (store::transaction const & transaction_a, bool & init_a, std::string const & path_a)
|
|
{
|
|
debug_assert (strlen (path_a.c_str ()) == path_a.size ());
|
|
auto error (0);
|
|
MDB_dbi handle_l;
|
|
error |= mdb_dbi_open (env.tx (transaction_a), path_a.c_str (), MDB_CREATE, &handle_l);
|
|
handle = handle_l;
|
|
init_a = error != 0;
|
|
}
|
|
|
|
bool nano::wallet_store::is_representative (store::transaction const & transaction_a)
|
|
{
|
|
return exists (transaction_a, representative (transaction_a));
|
|
}
|
|
|
|
void nano::wallet_store::representative_set (store::transaction const & transaction_a, nano::account const & representative_a)
|
|
{
|
|
nano::raw_key rep;
|
|
rep.bytes = representative_a.bytes;
|
|
entry_put_raw (transaction_a, nano::wallet_store::representative_special, nano::wallet_value (rep, 0));
|
|
}
|
|
|
|
nano::account nano::wallet_store::representative (store::transaction const & transaction_a)
|
|
{
|
|
nano::wallet_value value (entry_get_raw (transaction_a, nano::wallet_store::representative_special));
|
|
return reinterpret_cast<nano::account const &> (value.key);
|
|
}
|
|
|
|
nano::public_key nano::wallet_store::insert_adhoc (store::transaction const & transaction_a, nano::raw_key const & prv)
|
|
{
|
|
debug_assert (valid_password (transaction_a));
|
|
nano::public_key pub (nano::pub_key (prv));
|
|
nano::raw_key password_l;
|
|
wallet_key (password_l, transaction_a);
|
|
nano::raw_key ciphertext;
|
|
ciphertext.encrypt (prv, password_l, pub.owords[0].number ());
|
|
entry_put_raw (transaction_a, pub, nano::wallet_value (ciphertext, 0));
|
|
return pub;
|
|
}
|
|
|
|
bool nano::wallet_store::insert_watch (store::transaction const & transaction_a, nano::account const & pub_a)
|
|
{
|
|
bool error (!valid_public_key (pub_a));
|
|
if (!error)
|
|
{
|
|
entry_put_raw (transaction_a, pub_a, nano::wallet_value (nano::raw_key (0), 0));
|
|
}
|
|
return error;
|
|
}
|
|
|
|
void nano::wallet_store::erase (store::transaction const & transaction_a, nano::account const & pub)
|
|
{
|
|
auto status (mdb_del (env.tx (transaction_a), handle, nano::store::lmdb::db_val (pub), nullptr));
|
|
(void)status;
|
|
debug_assert (status == 0);
|
|
}
|
|
|
|
nano::wallet_value nano::wallet_store::entry_get_raw (store::transaction const & transaction_a, nano::account const & pub_a)
|
|
{
|
|
nano::wallet_value result;
|
|
nano::store::lmdb::db_val value;
|
|
auto status (mdb_get (env.tx (transaction_a), handle, nano::store::lmdb::db_val (pub_a), value));
|
|
if (status == 0)
|
|
{
|
|
result = nano::wallet_value (value);
|
|
}
|
|
else
|
|
{
|
|
result.key.clear ();
|
|
result.work = 0;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void nano::wallet_store::entry_put_raw (store::transaction const & transaction_a, nano::account const & pub_a, nano::wallet_value const & entry_a)
|
|
{
|
|
auto status (mdb_put (env.tx (transaction_a), handle, nano::store::lmdb::db_val (pub_a), nano::store::lmdb::db_val (sizeof (entry_a), const_cast<nano::wallet_value *> (&entry_a)), 0));
|
|
(void)status;
|
|
debug_assert (status == 0);
|
|
}
|
|
|
|
nano::key_type nano::wallet_store::key_type (nano::wallet_value const & value_a)
|
|
{
|
|
auto number (value_a.key.number ());
|
|
nano::key_type result;
|
|
auto text (number.convert_to<std::string> ());
|
|
if (number > std::numeric_limits<uint64_t>::max ())
|
|
{
|
|
result = nano::key_type::adhoc;
|
|
}
|
|
else
|
|
{
|
|
if ((number >> 32).convert_to<uint32_t> () == 1)
|
|
{
|
|
result = nano::key_type::deterministic;
|
|
}
|
|
else
|
|
{
|
|
result = nano::key_type::unknown;
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool nano::wallet_store::fetch (store::transaction const & transaction_a, nano::account const & pub, nano::raw_key & prv)
|
|
{
|
|
auto result (false);
|
|
if (valid_password (transaction_a))
|
|
{
|
|
nano::wallet_value value (entry_get_raw (transaction_a, pub));
|
|
if (!value.key.is_zero ())
|
|
{
|
|
switch (key_type (value))
|
|
{
|
|
case nano::key_type::deterministic:
|
|
{
|
|
nano::raw_key seed_l;
|
|
seed (seed_l, transaction_a);
|
|
uint32_t index (static_cast<uint32_t> (value.key.number () & static_cast<uint32_t> (-1)));
|
|
prv = deterministic_key (transaction_a, index);
|
|
break;
|
|
}
|
|
case nano::key_type::adhoc:
|
|
{
|
|
// Ad-hoc keys
|
|
nano::raw_key password_l;
|
|
wallet_key (password_l, transaction_a);
|
|
prv.decrypt (value.key, password_l, pub.owords[0].number ());
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
result = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
result = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
result = true;
|
|
}
|
|
if (!result)
|
|
{
|
|
nano::public_key compare (nano::pub_key (prv));
|
|
if (!(pub == compare))
|
|
{
|
|
result = true;
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool nano::wallet_store::valid_public_key (nano::public_key const & pub)
|
|
{
|
|
return pub.number () >= special_count;
|
|
}
|
|
|
|
bool nano::wallet_store::exists (store::transaction const & transaction_a, nano::public_key const & pub)
|
|
{
|
|
return valid_public_key (pub) && find (transaction_a, pub) != end ();
|
|
}
|
|
|
|
void nano::wallet_store::serialize_json (store::transaction const & transaction_a, std::string & string_a)
|
|
{
|
|
boost::property_tree::ptree tree;
|
|
for (store::iterator<nano::uint256_union, nano::wallet_value> i (std::make_unique<nano::store::lmdb::iterator<nano::uint256_union, nano::wallet_value>> (transaction_a, env, handle)), n (nullptr); i != n; ++i)
|
|
{
|
|
tree.put (i->first.to_string (), i->second.key.to_string ());
|
|
}
|
|
std::stringstream ostream;
|
|
boost::property_tree::write_json (ostream, tree);
|
|
string_a = ostream.str ();
|
|
}
|
|
|
|
void nano::wallet_store::write_backup (store::transaction const & transaction_a, std::filesystem::path const & path_a)
|
|
{
|
|
std::ofstream backup_file;
|
|
backup_file.open (path_a.string ());
|
|
if (!backup_file.fail ())
|
|
{
|
|
// Set permissions to 600
|
|
boost::system::error_code ec;
|
|
nano::set_secure_perm_file (path_a, ec);
|
|
|
|
std::string json;
|
|
serialize_json (transaction_a, json);
|
|
backup_file << json;
|
|
}
|
|
}
|
|
|
|
bool nano::wallet_store::move (store::transaction const & transaction_a, nano::wallet_store & other_a, std::vector<nano::public_key> const & keys)
|
|
{
|
|
debug_assert (valid_password (transaction_a));
|
|
debug_assert (other_a.valid_password (transaction_a));
|
|
auto result (false);
|
|
for (auto i (keys.begin ()), n (keys.end ()); i != n; ++i)
|
|
{
|
|
nano::raw_key prv;
|
|
auto error (other_a.fetch (transaction_a, *i, prv));
|
|
result = result | error;
|
|
if (!result)
|
|
{
|
|
insert_adhoc (transaction_a, prv);
|
|
other_a.erase (transaction_a, *i);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool nano::wallet_store::import (store::transaction const & transaction_a, nano::wallet_store & other_a)
|
|
{
|
|
debug_assert (valid_password (transaction_a));
|
|
debug_assert (other_a.valid_password (transaction_a));
|
|
auto result (false);
|
|
for (auto i (other_a.begin (transaction_a)), n (end ()); i != n; ++i)
|
|
{
|
|
nano::raw_key prv;
|
|
auto error (other_a.fetch (transaction_a, i->first, prv));
|
|
result = result | error;
|
|
if (!result)
|
|
{
|
|
if (!prv.is_zero ())
|
|
{
|
|
insert_adhoc (transaction_a, prv);
|
|
}
|
|
else
|
|
{
|
|
insert_watch (transaction_a, i->first);
|
|
}
|
|
other_a.erase (transaction_a, i->first);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool nano::wallet_store::work_get (store::transaction const & transaction_a, nano::public_key const & pub_a, uint64_t & work_a)
|
|
{
|
|
auto result (false);
|
|
auto entry (entry_get_raw (transaction_a, pub_a));
|
|
if (!entry.key.is_zero ())
|
|
{
|
|
work_a = entry.work;
|
|
}
|
|
else
|
|
{
|
|
result = true;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void nano::wallet_store::work_put (store::transaction const & transaction_a, nano::public_key const & pub_a, uint64_t work_a)
|
|
{
|
|
auto entry (entry_get_raw (transaction_a, pub_a));
|
|
debug_assert (!entry.key.is_zero ());
|
|
entry.work = work_a;
|
|
entry_put_raw (transaction_a, pub_a, entry);
|
|
}
|
|
|
|
unsigned nano::wallet_store::version (store::transaction const & transaction_a)
|
|
{
|
|
nano::wallet_value value (entry_get_raw (transaction_a, nano::wallet_store::version_special));
|
|
auto entry (value.key);
|
|
auto result (static_cast<unsigned> (entry.bytes[31]));
|
|
return result;
|
|
}
|
|
|
|
void nano::wallet_store::version_put (store::transaction const & transaction_a, unsigned version_a)
|
|
{
|
|
nano::raw_key entry (version_a);
|
|
entry_put_raw (transaction_a, nano::wallet_store::version_special, nano::wallet_value (entry, 0));
|
|
}
|
|
|
|
void nano::kdf::phs (nano::raw_key & result_a, std::string const & password_a, nano::uint256_union const & salt_a)
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ mutex };
|
|
auto success (argon2_hash (1, kdf_work, 1, password_a.data (), password_a.size (), salt_a.bytes.data (), salt_a.bytes.size (), result_a.bytes.data (), result_a.bytes.size (), NULL, 0, Argon2_d, 0x10));
|
|
debug_assert (success == 0);
|
|
(void)success;
|
|
}
|
|
|
|
nano::wallet::wallet (bool & init_a, store::transaction & transaction_a, nano::wallets & wallets_a, std::string const & wallet_a) :
|
|
lock_observer ([] (bool, bool) {}),
|
|
store (init_a, wallets_a.kdf, transaction_a, wallets_a.env, wallets_a.node.config.random_representative (), wallets_a.node.config.password_fanout, wallet_a),
|
|
wallets (wallets_a)
|
|
{
|
|
}
|
|
|
|
nano::wallet::wallet (bool & init_a, store::transaction & transaction_a, nano::wallets & wallets_a, std::string const & wallet_a, std::string const & json) :
|
|
lock_observer ([] (bool, bool) {}),
|
|
store (init_a, wallets_a.kdf, transaction_a, wallets_a.env, wallets_a.node.config.random_representative (), wallets_a.node.config.password_fanout, wallet_a, json),
|
|
wallets (wallets_a)
|
|
{
|
|
}
|
|
|
|
void nano::wallet::enter_initial_password ()
|
|
{
|
|
nano::raw_key password_l;
|
|
{
|
|
nano::lock_guard<std::recursive_mutex> lock{ store.mutex };
|
|
store.password.value (password_l);
|
|
}
|
|
if (password_l.is_zero ())
|
|
{
|
|
auto transaction (wallets.tx_begin_write ());
|
|
if (store.valid_password (transaction))
|
|
{
|
|
// Newly created wallets have a zero key
|
|
store.rekey (transaction, "");
|
|
}
|
|
else
|
|
{
|
|
enter_password (transaction, "");
|
|
}
|
|
}
|
|
}
|
|
|
|
bool nano::wallet::enter_password (store::transaction const & transaction_a, std::string const & password_a)
|
|
{
|
|
auto result (store.attempt_password (transaction_a, password_a));
|
|
if (!result)
|
|
{
|
|
wallets.node.logger.info (nano::log::type::wallet, "Wallet unlocked");
|
|
|
|
auto this_l = shared_from_this ();
|
|
wallets.queue_wallet_action (nano::wallets::high_priority, this_l, [this_l] (nano::wallet & wallet) {
|
|
// Wallets must survive node lifetime
|
|
this_l->search_receivable (this_l->wallets.tx_begin_read ());
|
|
});
|
|
}
|
|
else
|
|
{
|
|
wallets.node.logger.warn (nano::log::type::wallet, "Invalid password, wallet locked");
|
|
}
|
|
lock_observer (result, password_a.empty ());
|
|
return result;
|
|
}
|
|
|
|
nano::public_key nano::wallet::deterministic_insert (store::transaction const & transaction_a, bool generate_work_a)
|
|
{
|
|
nano::public_key key{};
|
|
if (store.valid_password (transaction_a))
|
|
{
|
|
key = store.deterministic_insert (transaction_a);
|
|
if (generate_work_a)
|
|
{
|
|
work_ensure (key, key);
|
|
}
|
|
auto half_principal_weight (wallets.node.minimum_principal_weight () / 2);
|
|
if (wallets.check_rep (key, half_principal_weight))
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ representatives_mutex };
|
|
representatives.insert (key);
|
|
}
|
|
}
|
|
return key;
|
|
}
|
|
|
|
nano::public_key nano::wallet::deterministic_insert (uint32_t const index, bool generate_work_a)
|
|
{
|
|
auto transaction (wallets.tx_begin_write ());
|
|
nano::public_key key{};
|
|
if (store.valid_password (transaction))
|
|
{
|
|
key = store.deterministic_insert (transaction, index);
|
|
if (generate_work_a)
|
|
{
|
|
work_ensure (key, key);
|
|
}
|
|
}
|
|
return key;
|
|
}
|
|
|
|
nano::public_key nano::wallet::deterministic_insert (bool generate_work_a)
|
|
{
|
|
auto transaction (wallets.tx_begin_write ());
|
|
auto result (deterministic_insert (transaction, generate_work_a));
|
|
return result;
|
|
}
|
|
|
|
nano::public_key nano::wallet::insert_adhoc (nano::raw_key const & key_a, bool generate_work_a)
|
|
{
|
|
nano::public_key key{};
|
|
auto transaction (wallets.tx_begin_write ());
|
|
if (store.valid_password (transaction))
|
|
{
|
|
key = store.insert_adhoc (transaction, key_a);
|
|
auto block_transaction (wallets.node.store.tx_begin_read ());
|
|
if (generate_work_a)
|
|
{
|
|
work_ensure (key, wallets.node.ledger.latest_root (block_transaction, key));
|
|
}
|
|
auto half_principal_weight (wallets.node.minimum_principal_weight () / 2);
|
|
// Makes sure that the representatives container will
|
|
// be in sync with any added keys.
|
|
transaction.commit ();
|
|
if (wallets.check_rep (key, half_principal_weight))
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ representatives_mutex };
|
|
representatives.insert (key);
|
|
}
|
|
}
|
|
return key;
|
|
}
|
|
|
|
bool nano::wallet::insert_watch (store::transaction const & transaction_a, nano::public_key const & pub_a)
|
|
{
|
|
return store.insert_watch (transaction_a, pub_a);
|
|
}
|
|
|
|
bool nano::wallet::exists (nano::public_key const & account_a)
|
|
{
|
|
auto transaction (wallets.tx_begin_read ());
|
|
return store.exists (transaction, account_a);
|
|
}
|
|
|
|
bool nano::wallet::import (std::string const & json_a, std::string const & password_a)
|
|
{
|
|
auto error (false);
|
|
std::unique_ptr<nano::wallet_store> temp;
|
|
{
|
|
auto transaction (wallets.tx_begin_write ());
|
|
nano::uint256_union id;
|
|
random_pool::generate_block (id.bytes.data (), id.bytes.size ());
|
|
temp = std::make_unique<nano::wallet_store> (error, wallets.node.wallets.kdf, transaction, wallets.env, 0, 1, id.to_string (), json_a);
|
|
}
|
|
if (!error)
|
|
{
|
|
auto transaction (wallets.tx_begin_write ());
|
|
error = temp->attempt_password (transaction, password_a);
|
|
}
|
|
auto transaction (wallets.tx_begin_write ());
|
|
if (!error)
|
|
{
|
|
error = store.import (transaction, *temp);
|
|
}
|
|
temp->destroy (transaction);
|
|
return error;
|
|
}
|
|
|
|
void nano::wallet::serialize (std::string & json_a)
|
|
{
|
|
auto transaction (wallets.tx_begin_read ());
|
|
store.serialize_json (transaction, json_a);
|
|
}
|
|
|
|
void nano::wallet_store::destroy (store::transaction const & transaction_a)
|
|
{
|
|
auto status (mdb_drop (env.tx (transaction_a), handle, 1));
|
|
(void)status;
|
|
debug_assert (status == 0);
|
|
handle = 0;
|
|
}
|
|
|
|
std::shared_ptr<nano::block> nano::wallet::receive_action (nano::block_hash const & send_hash_a, nano::account const & representative_a, nano::uint128_union const & amount_a, nano::account const & account_a, uint64_t work_a, bool generate_work_a)
|
|
{
|
|
std::shared_ptr<nano::block> block;
|
|
nano::block_details details;
|
|
details.is_receive = true;
|
|
if (wallets.node.config.receive_minimum.number () <= amount_a.number ())
|
|
{
|
|
auto block_transaction (wallets.node.ledger.store.tx_begin_read ());
|
|
auto transaction (wallets.tx_begin_read ());
|
|
if (wallets.node.ledger.block_or_pruned_exists (block_transaction, send_hash_a))
|
|
{
|
|
auto pending_info = wallets.node.ledger.pending_info (block_transaction, nano::pending_key (account_a, send_hash_a));
|
|
if (pending_info)
|
|
{
|
|
nano::raw_key prv;
|
|
if (!store.fetch (transaction, account_a, prv))
|
|
{
|
|
if (work_a == 0)
|
|
{
|
|
store.work_get (transaction, account_a, work_a);
|
|
}
|
|
auto info = wallets.node.ledger.account_info (block_transaction, account_a);
|
|
if (info)
|
|
{
|
|
block = std::make_shared<nano::state_block> (account_a, info->head, info->representative, info->balance.number () + pending_info->amount.number (), send_hash_a, prv, account_a, work_a);
|
|
details.epoch = std::max (info->epoch (), pending_info->epoch);
|
|
}
|
|
else
|
|
{
|
|
block = std::make_shared<nano::state_block> (account_a, 0, representative_a, pending_info->amount, reinterpret_cast<nano::link const &> (send_hash_a), prv, account_a, work_a);
|
|
details.epoch = pending_info->epoch;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
wallets.node.logger.warn (nano::log::type::wallet, "Unable to receive, wallet locked");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Ledger doesn't have this marked as available to receive anymore
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Ledger doesn't have this block anymore.
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Someone sent us something below the threshold of receiving
|
|
wallets.node.logger.warn (nano::log::type::wallet, "Not receiving block {} due to minimum receive threshold", send_hash_a.to_string ());
|
|
}
|
|
if (block != nullptr)
|
|
{
|
|
if (action_complete (block, account_a, generate_work_a, details))
|
|
{
|
|
// Return null block after work generation or ledger process error
|
|
block = nullptr;
|
|
}
|
|
}
|
|
return block;
|
|
}
|
|
|
|
std::shared_ptr<nano::block> nano::wallet::change_action (nano::account const & source_a, nano::account const & representative_a, uint64_t work_a, bool generate_work_a)
|
|
{
|
|
std::shared_ptr<nano::block> block;
|
|
nano::block_details details;
|
|
{
|
|
auto transaction (wallets.tx_begin_read ());
|
|
auto block_transaction (wallets.node.store.tx_begin_read ());
|
|
if (store.valid_password (transaction))
|
|
{
|
|
auto existing (store.find (transaction, source_a));
|
|
if (existing != store.end () && !wallets.node.ledger.latest (block_transaction, source_a).is_zero ())
|
|
{
|
|
auto info = wallets.node.ledger.account_info (block_transaction, source_a);
|
|
debug_assert (info);
|
|
nano::raw_key prv;
|
|
auto error2 (store.fetch (transaction, source_a, prv));
|
|
(void)error2;
|
|
debug_assert (!error2);
|
|
if (work_a == 0)
|
|
{
|
|
store.work_get (transaction, source_a, work_a);
|
|
}
|
|
block = std::make_shared<nano::state_block> (source_a, info->head, representative_a, info->balance, 0, prv, source_a, work_a);
|
|
details.epoch = info->epoch ();
|
|
}
|
|
}
|
|
}
|
|
if (block != nullptr)
|
|
{
|
|
if (action_complete (block, source_a, generate_work_a, details))
|
|
{
|
|
// Return null block after work generation or ledger process error
|
|
block = nullptr;
|
|
}
|
|
}
|
|
return block;
|
|
}
|
|
|
|
std::shared_ptr<nano::block> nano::wallet::send_action (nano::account const & source_a, nano::account const & account_a, nano::uint128_t const & amount_a, uint64_t work_a, bool generate_work_a, boost::optional<std::string> id_a)
|
|
{
|
|
boost::optional<nano::store::lmdb::db_val> id_mdb_val;
|
|
if (id_a)
|
|
{
|
|
id_mdb_val = nano::store::lmdb::db_val (id_a->size (), const_cast<char *> (id_a->data ()));
|
|
}
|
|
|
|
auto prepare_send = [&id_mdb_val, &wallets = this->wallets, &store = this->store, &source_a, &amount_a, &work_a, &account_a] (auto const & transaction) {
|
|
auto block_transaction (wallets.node.store.tx_begin_read ());
|
|
auto error (false);
|
|
auto cached_block (false);
|
|
std::shared_ptr<nano::block> block;
|
|
nano::block_details details;
|
|
details.is_send = true;
|
|
if (id_mdb_val)
|
|
{
|
|
nano::store::lmdb::db_val result;
|
|
auto status (mdb_get (wallets.env.tx (transaction), wallets.node.wallets.send_action_ids, *id_mdb_val, result));
|
|
if (status == 0)
|
|
{
|
|
nano::block_hash hash (result);
|
|
block = wallets.node.ledger.block (block_transaction, hash);
|
|
if (block != nullptr)
|
|
{
|
|
cached_block = true;
|
|
wallets.node.network.flood_block (block, nano::transport::buffer_drop_policy::no_limiter_drop);
|
|
}
|
|
}
|
|
else if (status != MDB_NOTFOUND)
|
|
{
|
|
error = true;
|
|
}
|
|
}
|
|
if (!error && block == nullptr)
|
|
{
|
|
if (store.valid_password (transaction))
|
|
{
|
|
auto existing (store.find (transaction, source_a));
|
|
if (existing != store.end ())
|
|
{
|
|
auto balance (wallets.node.ledger.account_balance (block_transaction, source_a));
|
|
if (!balance.is_zero () && balance >= amount_a)
|
|
{
|
|
auto info = wallets.node.ledger.account_info (block_transaction, source_a);
|
|
debug_assert (info);
|
|
nano::raw_key prv;
|
|
auto error2 (store.fetch (transaction, source_a, prv));
|
|
(void)error2;
|
|
debug_assert (!error2);
|
|
if (work_a == 0)
|
|
{
|
|
store.work_get (transaction, source_a, work_a);
|
|
}
|
|
block = std::make_shared<nano::state_block> (source_a, info->head, info->representative, balance - amount_a, account_a, prv, source_a, work_a);
|
|
details.epoch = info->epoch ();
|
|
if (id_mdb_val && block != nullptr)
|
|
{
|
|
auto status (mdb_put (wallets.env.tx (transaction), wallets.node.wallets.send_action_ids, *id_mdb_val, nano::store::lmdb::db_val (block->hash ()), 0));
|
|
if (status != 0)
|
|
{
|
|
block = nullptr;
|
|
error = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return std::make_tuple (block, error, cached_block, details);
|
|
};
|
|
|
|
std::tuple<std::shared_ptr<nano::block>, bool, bool, nano::block_details> result;
|
|
{
|
|
if (id_mdb_val)
|
|
{
|
|
result = prepare_send (wallets.tx_begin_write ());
|
|
}
|
|
else
|
|
{
|
|
result = prepare_send (wallets.tx_begin_read ());
|
|
}
|
|
}
|
|
|
|
std::shared_ptr<nano::block> block;
|
|
bool error;
|
|
bool cached_block;
|
|
nano::block_details details;
|
|
std::tie (block, error, cached_block, details) = result;
|
|
|
|
if (!error && block != nullptr && !cached_block)
|
|
{
|
|
if (action_complete (block, source_a, generate_work_a, details))
|
|
{
|
|
// Return null block after work generation or ledger process error
|
|
block = nullptr;
|
|
}
|
|
}
|
|
return block;
|
|
}
|
|
|
|
bool nano::wallet::action_complete (std::shared_ptr<nano::block> const & block_a, nano::account const & account_a, bool const generate_work_a, nano::block_details const & details_a)
|
|
{
|
|
bool error{ false };
|
|
// Unschedule any work caching for this account
|
|
wallets.delayed_work->erase (account_a);
|
|
if (block_a != nullptr)
|
|
{
|
|
auto required_difficulty{ wallets.node.network_params.work.threshold (block_a->work_version (), details_a) };
|
|
if (wallets.node.network_params.work.difficulty (*block_a) < required_difficulty)
|
|
{
|
|
wallets.node.logger.info (nano::log::type::wallet, "Cached or provided work for block {} account {} is invalid, regenerating...",
|
|
block_a->hash ().to_string (),
|
|
account_a.to_account ());
|
|
|
|
debug_assert (required_difficulty <= wallets.node.max_work_generate_difficulty (block_a->work_version ()));
|
|
error = !wallets.node.work_generate_blocking (*block_a, required_difficulty).is_initialized ();
|
|
}
|
|
if (!error)
|
|
{
|
|
auto result = wallets.node.process_local (block_a);
|
|
error = !result || result.value () != nano::block_status::progress;
|
|
debug_assert (error || block_a->sideband ().details == details_a);
|
|
}
|
|
if (!error && generate_work_a)
|
|
{
|
|
// Pregenerate work for next block based on the block just created
|
|
work_ensure (account_a, block_a->hash ());
|
|
}
|
|
}
|
|
return error;
|
|
}
|
|
|
|
bool nano::wallet::change_sync (nano::account const & source_a, nano::account const & representative_a)
|
|
{
|
|
std::promise<bool> result;
|
|
std::future<bool> future = result.get_future ();
|
|
change_async (
|
|
source_a, representative_a, [&result] (std::shared_ptr<nano::block> const & block_a) {
|
|
result.set_value (block_a == nullptr);
|
|
},
|
|
true);
|
|
return future.get ();
|
|
}
|
|
|
|
void nano::wallet::change_async (nano::account const & source_a, nano::account const & representative_a, std::function<void (std::shared_ptr<nano::block> const &)> const & action_a, uint64_t work_a, bool generate_work_a)
|
|
{
|
|
auto this_l (shared_from_this ());
|
|
wallets.node.wallets.queue_wallet_action (nano::wallets::high_priority, this_l, [this_l, source_a, representative_a, action_a, work_a, generate_work_a] (nano::wallet & wallet_a) {
|
|
auto block (wallet_a.change_action (source_a, representative_a, work_a, generate_work_a));
|
|
action_a (block);
|
|
});
|
|
}
|
|
|
|
bool nano::wallet::receive_sync (std::shared_ptr<nano::block> const & block_a, nano::account const & representative_a, nano::uint128_t const & amount_a)
|
|
{
|
|
std::promise<bool> result;
|
|
std::future<bool> future = result.get_future ();
|
|
auto destination (block_a->link ().is_zero () ? block_a->destination () : block_a->link ().as_account ());
|
|
receive_async (
|
|
block_a->hash (), representative_a, amount_a, destination, [&result] (std::shared_ptr<nano::block> const & block_a) {
|
|
result.set_value (block_a == nullptr);
|
|
},
|
|
true);
|
|
return future.get ();
|
|
}
|
|
|
|
void nano::wallet::receive_async (nano::block_hash const & hash_a, nano::account const & representative_a, nano::uint128_t const & amount_a, nano::account const & account_a, std::function<void (std::shared_ptr<nano::block> const &)> const & action_a, uint64_t work_a, bool generate_work_a)
|
|
{
|
|
auto this_l (shared_from_this ());
|
|
wallets.node.wallets.queue_wallet_action (amount_a, this_l, [this_l, hash_a, representative_a, amount_a, account_a, action_a, work_a, generate_work_a] (nano::wallet & wallet_a) {
|
|
auto block (wallet_a.receive_action (hash_a, representative_a, amount_a, account_a, work_a, generate_work_a));
|
|
action_a (block);
|
|
});
|
|
}
|
|
|
|
nano::block_hash nano::wallet::send_sync (nano::account const & source_a, nano::account const & account_a, nano::uint128_t const & amount_a)
|
|
{
|
|
std::promise<nano::block_hash> result;
|
|
std::future<nano::block_hash> future = result.get_future ();
|
|
send_async (
|
|
source_a, account_a, amount_a, [&result] (std::shared_ptr<nano::block> const & block_a) {
|
|
result.set_value (block_a->hash ());
|
|
},
|
|
true);
|
|
return future.get ();
|
|
}
|
|
|
|
void nano::wallet::send_async (nano::account const & source_a, nano::account const & account_a, nano::uint128_t const & amount_a, std::function<void (std::shared_ptr<nano::block> const &)> const & action_a, uint64_t work_a, bool generate_work_a, boost::optional<std::string> id_a)
|
|
{
|
|
auto this_l (shared_from_this ());
|
|
wallets.node.wallets.queue_wallet_action (nano::wallets::high_priority, this_l, [this_l, source_a, account_a, amount_a, action_a, work_a, generate_work_a, id_a] (nano::wallet & wallet_a) {
|
|
auto block (wallet_a.send_action (source_a, account_a, amount_a, work_a, generate_work_a, id_a));
|
|
action_a (block);
|
|
});
|
|
}
|
|
|
|
// Update work for account if latest root is root_a
|
|
void nano::wallet::work_update (store::transaction const & transaction_a, nano::account const & account_a, nano::root const & root_a, uint64_t work_a)
|
|
{
|
|
debug_assert (!wallets.node.network_params.work.validate_entry (nano::work_version::work_1, root_a, work_a));
|
|
debug_assert (store.exists (transaction_a, account_a));
|
|
auto block_transaction (wallets.node.store.tx_begin_read ());
|
|
auto latest (wallets.node.ledger.latest_root (block_transaction, account_a));
|
|
if (latest == root_a)
|
|
{
|
|
store.work_put (transaction_a, account_a, work_a);
|
|
}
|
|
else
|
|
{
|
|
wallets.node.logger.warn (nano::log::type::wallet, "Cached work no longer valid, discarding");
|
|
}
|
|
}
|
|
|
|
void nano::wallet::work_ensure (nano::account const & account_a, nano::root const & root_a)
|
|
{
|
|
using namespace std::chrono_literals;
|
|
std::chrono::seconds const precache_delay = wallets.node.network_params.network.is_dev_network () ? 1s : 10s;
|
|
|
|
wallets.delayed_work->operator[] (account_a) = root_a;
|
|
|
|
wallets.node.workers.add_timed_task (std::chrono::steady_clock::now () + precache_delay, [this_l = shared_from_this (), account_a, root_a] {
|
|
auto delayed_work = this_l->wallets.delayed_work.lock ();
|
|
auto existing (delayed_work->find (account_a));
|
|
if (existing != delayed_work->end () && existing->second == root_a)
|
|
{
|
|
delayed_work->erase (existing);
|
|
this_l->wallets.queue_wallet_action (nano::wallets::generate_priority, this_l, [account_a, root_a] (nano::wallet & wallet_a) {
|
|
wallet_a.work_cache_blocking (account_a, root_a);
|
|
});
|
|
}
|
|
});
|
|
}
|
|
|
|
bool nano::wallet::search_receivable (store::transaction const & wallet_transaction_a)
|
|
{
|
|
auto result (!store.valid_password (wallet_transaction_a));
|
|
if (!result)
|
|
{
|
|
wallets.node.logger.info (nano::log::type::wallet, "Beginning receivable block search");
|
|
|
|
for (auto i (store.begin (wallet_transaction_a)), n (store.end ()); i != n; ++i)
|
|
{
|
|
auto block_transaction (wallets.node.store.tx_begin_read ());
|
|
nano::account const & account (i->first);
|
|
// Don't search pending for watch-only accounts
|
|
if (!nano::wallet_value (i->second).key.is_zero ())
|
|
{
|
|
for (auto j (wallets.node.store.pending.begin (block_transaction, nano::pending_key (account, 0))), k (wallets.node.store.pending.end ()); j != k && nano::pending_key (j->first).account == account; ++j)
|
|
{
|
|
nano::pending_key key (j->first);
|
|
auto hash (key.hash);
|
|
nano::pending_info pending (j->second);
|
|
auto amount (pending.amount.number ());
|
|
if (wallets.node.config.receive_minimum.number () <= amount)
|
|
{
|
|
wallets.node.logger.info (nano::log::type::wallet, "Found a receivable block {} for account {}", hash.to_string (), pending.source.to_account ());
|
|
|
|
if (wallets.node.ledger.block_confirmed (block_transaction, hash))
|
|
{
|
|
auto representative = store.representative (wallet_transaction_a);
|
|
// Receive confirmed block
|
|
receive_async (hash, representative, amount, account, [] (std::shared_ptr<nano::block> const &) {});
|
|
}
|
|
else if (!wallets.node.confirmation_height_processor.is_processing_block (hash))
|
|
{
|
|
auto block = wallets.node.ledger.block (block_transaction, hash);
|
|
if (block)
|
|
{
|
|
// Request confirmation for block which is not being processed yet
|
|
wallets.node.start_election (block);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
wallets.node.logger.info (nano::log::type::wallet, "Receivable block search phase complete");
|
|
}
|
|
else
|
|
{
|
|
wallets.node.logger.warn (nano::log::type::wallet, "Stopping search, wallet is locked");
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void nano::wallet::init_free_accounts (store::transaction const & transaction_a)
|
|
{
|
|
free_accounts.clear ();
|
|
for (auto i (store.begin (transaction_a)), n (store.end ()); i != n; ++i)
|
|
{
|
|
free_accounts.insert (i->first);
|
|
}
|
|
}
|
|
|
|
uint32_t nano::wallet::deterministic_check (store::transaction const & transaction_a, uint32_t index)
|
|
{
|
|
auto block_transaction (wallets.node.store.tx_begin_read ());
|
|
for (uint32_t i (index + 1), n (index + 64); i < n; ++i)
|
|
{
|
|
auto prv = store.deterministic_key (transaction_a, i);
|
|
nano::keypair pair (prv.to_string ());
|
|
// Check if account received at least 1 block
|
|
auto latest (wallets.node.ledger.latest (block_transaction, pair.pub));
|
|
if (!latest.is_zero ())
|
|
{
|
|
index = i;
|
|
// i + 64 - Check additional 64 accounts
|
|
// i/64 - Check additional accounts for large wallets. I.e. 64000/64 = 1000 accounts to check
|
|
n = i + 64 + (i / 64);
|
|
}
|
|
else
|
|
{
|
|
// Check if there are pending blocks for account
|
|
for (auto ii (wallets.node.store.pending.begin (block_transaction, nano::pending_key (pair.pub, 0))), nn (wallets.node.store.pending.end ()); ii != nn && nano::pending_key (ii->first).account == pair.pub; ++ii)
|
|
{
|
|
index = i;
|
|
n = i + 64 + (i / 64);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return index;
|
|
}
|
|
|
|
nano::public_key nano::wallet::change_seed (store::transaction const & transaction_a, nano::raw_key const & prv_a, uint32_t count)
|
|
{
|
|
store.seed_set (transaction_a, prv_a);
|
|
auto account = deterministic_insert (transaction_a);
|
|
if (count == 0)
|
|
{
|
|
count = deterministic_check (transaction_a, 0);
|
|
}
|
|
for (uint32_t i (0); i < count; ++i)
|
|
{
|
|
// Disable work generation to prevent weak CPU nodes stuck
|
|
account = deterministic_insert (transaction_a, false);
|
|
}
|
|
return account;
|
|
}
|
|
|
|
void nano::wallet::deterministic_restore (store::transaction const & transaction_a)
|
|
{
|
|
auto index (store.deterministic_index_get (transaction_a));
|
|
auto new_index (deterministic_check (transaction_a, index));
|
|
for (uint32_t i (index); i <= new_index && index != new_index; ++i)
|
|
{
|
|
// Disable work generation to prevent weak CPU nodes stuck
|
|
deterministic_insert (transaction_a, false);
|
|
}
|
|
}
|
|
|
|
bool nano::wallet::live ()
|
|
{
|
|
return store.handle != 0;
|
|
}
|
|
|
|
void nano::wallet::work_cache_blocking (nano::account const & account_a, nano::root const & root_a)
|
|
{
|
|
if (wallets.node.work_generation_enabled ())
|
|
{
|
|
auto difficulty (wallets.node.default_difficulty (nano::work_version::work_1));
|
|
auto opt_work_l (wallets.node.work_generate_blocking (nano::work_version::work_1, root_a, difficulty, account_a));
|
|
if (opt_work_l.is_initialized ())
|
|
{
|
|
auto transaction_l (wallets.tx_begin_write ());
|
|
if (live () && store.exists (transaction_l, account_a))
|
|
{
|
|
work_update (transaction_l, account_a, root_a, *opt_work_l);
|
|
}
|
|
}
|
|
else if (!wallets.node.stopped)
|
|
{
|
|
wallets.node.logger.warn (nano::log::type::wallet, "Could not precache work for root {} due to work generation failure", root_a.to_string ());
|
|
}
|
|
}
|
|
}
|
|
|
|
void nano::wallets::do_wallet_actions ()
|
|
{
|
|
nano::unique_lock<nano::mutex> action_lock{ action_mutex };
|
|
while (!stopped)
|
|
{
|
|
if (!actions.empty ())
|
|
{
|
|
auto first (actions.begin ());
|
|
auto wallet (first->second.first);
|
|
auto current (std::move (first->second.second));
|
|
actions.erase (first);
|
|
if (wallet->live ())
|
|
{
|
|
action_lock.unlock ();
|
|
observer (true);
|
|
current (*wallet);
|
|
observer (false);
|
|
action_lock.lock ();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
condition.wait (action_lock);
|
|
}
|
|
}
|
|
}
|
|
|
|
nano::wallets::wallets (bool error_a, nano::node & node_a) :
|
|
network_params{ node_a.config.network_params },
|
|
observer ([] (bool) {}),
|
|
kdf{ node_a.config.network_params.kdf_work },
|
|
node (node_a),
|
|
env (boost::polymorphic_downcast<nano::mdb_wallets_store *> (node_a.wallets_store_impl.get ())->environment),
|
|
stopped (false)
|
|
{
|
|
nano::unique_lock<nano::mutex> lock{ mutex };
|
|
if (!error_a)
|
|
{
|
|
auto transaction (tx_begin_write ());
|
|
auto status (mdb_dbi_open (env.tx (transaction), nullptr, MDB_CREATE, &handle));
|
|
status |= mdb_dbi_open (env.tx (transaction), "send_action_ids", MDB_CREATE, &send_action_ids);
|
|
release_assert (status == 0);
|
|
std::string beginning (nano::uint256_union (0).to_string ());
|
|
std::string end ((nano::uint256_union (nano::uint256_t (0) - nano::uint256_t (1))).to_string ());
|
|
store::iterator<std::array<char, 64>, nano::no_value> i (std::make_unique<nano::store::lmdb::iterator<std::array<char, 64>, nano::no_value>> (transaction, env, handle, nano::store::lmdb::db_val (beginning.size (), const_cast<char *> (beginning.c_str ()))));
|
|
store::iterator<std::array<char, 64>, nano::no_value> n (std::make_unique<nano::store::lmdb::iterator<std::array<char, 64>, nano::no_value>> (transaction, env, handle, nano::store::lmdb::db_val (end.size (), const_cast<char *> (end.c_str ()))));
|
|
for (; i != n; ++i)
|
|
{
|
|
nano::wallet_id id;
|
|
std::string text (i->first.data (), i->first.size ());
|
|
auto error (id.decode_hex (text));
|
|
release_assert (!error);
|
|
release_assert (items.find (id) == items.end ());
|
|
auto wallet (std::make_shared<nano::wallet> (error, transaction, *this, text));
|
|
if (!error)
|
|
{
|
|
items[id] = wallet;
|
|
}
|
|
else
|
|
{
|
|
// Couldn't open wallet
|
|
}
|
|
}
|
|
}
|
|
// Backup before upgrade wallets
|
|
bool backup_required (false);
|
|
if (node.config.backup_before_upgrade)
|
|
{
|
|
auto transaction (tx_begin_read ());
|
|
for (auto & item : items)
|
|
{
|
|
if (item.second->store.version (transaction) != nano::wallet_store::version_current)
|
|
{
|
|
backup_required = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (backup_required)
|
|
{
|
|
char const * store_path;
|
|
mdb_env_get_path (env, &store_path);
|
|
std::filesystem::path const path (store_path);
|
|
nano::store::lmdb::component::create_backup_file (env, path, node_a.logger);
|
|
}
|
|
for (auto & item : items)
|
|
{
|
|
item.second->enter_initial_password ();
|
|
}
|
|
if (node_a.config.enable_voting)
|
|
{
|
|
lock.unlock ();
|
|
ongoing_compute_reps ();
|
|
}
|
|
}
|
|
|
|
nano::wallets::~wallets ()
|
|
{
|
|
stop ();
|
|
}
|
|
|
|
std::shared_ptr<nano::wallet> nano::wallets::open (nano::wallet_id const & id_a)
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ mutex };
|
|
std::shared_ptr<nano::wallet> result;
|
|
auto existing (items.find (id_a));
|
|
if (existing != items.end ())
|
|
{
|
|
result = existing->second;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
std::shared_ptr<nano::wallet> nano::wallets::create (nano::wallet_id const & id_a)
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ mutex };
|
|
debug_assert (items.find (id_a) == items.end ());
|
|
std::shared_ptr<nano::wallet> result;
|
|
bool error;
|
|
{
|
|
auto transaction (tx_begin_write ());
|
|
result = std::make_shared<nano::wallet> (error, transaction, *this, id_a.to_string ());
|
|
}
|
|
if (!error)
|
|
{
|
|
items[id_a] = result;
|
|
result->enter_initial_password ();
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool nano::wallets::search_receivable (nano::wallet_id const & wallet_a)
|
|
{
|
|
auto result (false);
|
|
if (auto wallet = open (wallet_a); wallet != nullptr)
|
|
{
|
|
result = wallet->search_receivable (tx_begin_read ());
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void nano::wallets::search_receivable_all ()
|
|
{
|
|
nano::unique_lock<nano::mutex> lk{ mutex };
|
|
auto wallets_l = get_wallets ();
|
|
auto wallet_transaction (tx_begin_read ());
|
|
lk.unlock ();
|
|
for (auto const & [id, wallet] : wallets_l)
|
|
{
|
|
wallet->search_receivable (wallet_transaction);
|
|
}
|
|
}
|
|
|
|
void nano::wallets::destroy (nano::wallet_id const & id_a)
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ mutex };
|
|
auto transaction (tx_begin_write ());
|
|
// action_mutex should be after transactions to prevent deadlocks in deterministic_insert () & insert_adhoc ()
|
|
nano::lock_guard<nano::mutex> action_lock{ action_mutex };
|
|
auto existing (items.find (id_a));
|
|
debug_assert (existing != items.end ());
|
|
auto wallet (existing->second);
|
|
items.erase (existing);
|
|
wallet->store.destroy (transaction);
|
|
}
|
|
|
|
void nano::wallets::reload ()
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ mutex };
|
|
auto transaction (tx_begin_write ());
|
|
std::unordered_set<nano::uint256_union> stored_items;
|
|
std::string beginning (nano::uint256_union (0).to_string ());
|
|
std::string end ((nano::uint256_union (nano::uint256_t (0) - nano::uint256_t (1))).to_string ());
|
|
store::iterator<std::array<char, 64>, nano::no_value> i (std::make_unique<nano::store::lmdb::iterator<std::array<char, 64>, nano::no_value>> (transaction, env, handle, nano::store::lmdb::db_val (beginning.size (), const_cast<char *> (beginning.c_str ()))));
|
|
store::iterator<std::array<char, 64>, nano::no_value> n (std::make_unique<nano::store::lmdb::iterator<std::array<char, 64>, nano::no_value>> (transaction, env, handle, nano::store::lmdb::db_val (end.size (), const_cast<char *> (end.c_str ()))));
|
|
for (; i != n; ++i)
|
|
{
|
|
nano::wallet_id id;
|
|
std::string text (i->first.data (), i->first.size ());
|
|
auto error (id.decode_hex (text));
|
|
debug_assert (!error);
|
|
// New wallet
|
|
if (items.find (id) == items.end ())
|
|
{
|
|
auto wallet (std::make_shared<nano::wallet> (error, transaction, *this, text));
|
|
if (!error)
|
|
{
|
|
items[id] = wallet;
|
|
}
|
|
}
|
|
// List of wallets on disk
|
|
stored_items.insert (id);
|
|
}
|
|
// Delete non existing wallets from memory
|
|
std::vector<nano::wallet_id> deleted_items;
|
|
for (auto i : items)
|
|
{
|
|
if (stored_items.find (i.first) == stored_items.end ())
|
|
{
|
|
deleted_items.push_back (i.first);
|
|
}
|
|
}
|
|
for (auto & i : deleted_items)
|
|
{
|
|
debug_assert (items.find (i) == items.end ());
|
|
items.erase (i);
|
|
}
|
|
}
|
|
|
|
void nano::wallets::queue_wallet_action (nano::uint128_t const & amount_a, std::shared_ptr<nano::wallet> const & wallet_a, std::function<void (nano::wallet &)> action_a)
|
|
{
|
|
{
|
|
nano::lock_guard<nano::mutex> action_lock{ action_mutex };
|
|
actions.emplace (amount_a, std::make_pair (wallet_a, action_a));
|
|
}
|
|
condition.notify_all ();
|
|
}
|
|
|
|
void nano::wallets::foreach_representative (std::function<void (nano::public_key const & pub_a, nano::raw_key const & prv_a)> const & action_a)
|
|
{
|
|
if (node.config.enable_voting)
|
|
{
|
|
std::vector<std::pair<nano::public_key const, nano::raw_key const>> action_accounts_l;
|
|
{
|
|
auto transaction_l (tx_begin_read ());
|
|
nano::lock_guard<nano::mutex> lock{ mutex };
|
|
for (auto i (items.begin ()), n (items.end ()); i != n; ++i)
|
|
{
|
|
auto & wallet (*i->second);
|
|
nano::lock_guard<std::recursive_mutex> store_lock{ wallet.store.mutex };
|
|
decltype (wallet.representatives) representatives_l;
|
|
{
|
|
nano::lock_guard<nano::mutex> representatives_lock{ wallet.representatives_mutex };
|
|
representatives_l = wallet.representatives;
|
|
}
|
|
for (auto const & account : representatives_l)
|
|
{
|
|
if (wallet.store.exists (transaction_l, account))
|
|
{
|
|
if (!node.ledger.weight (account).is_zero ())
|
|
{
|
|
if (wallet.store.valid_password (transaction_l))
|
|
{
|
|
nano::raw_key prv;
|
|
auto error (wallet.store.fetch (transaction_l, account, prv));
|
|
(void)error;
|
|
debug_assert (!error);
|
|
action_accounts_l.emplace_back (account, prv);
|
|
}
|
|
else
|
|
{
|
|
// TODO: Better logging interval handling
|
|
static auto last_log = std::chrono::steady_clock::time_point ();
|
|
if (last_log < std::chrono::steady_clock::now () - std::chrono::seconds (60))
|
|
{
|
|
last_log = std::chrono::steady_clock::now ();
|
|
|
|
node.logger.warn (nano::log::type::wallet, "Representative locked inside wallet: {}", i->first.to_string ());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
for (auto const & representative : action_accounts_l)
|
|
{
|
|
action_a (representative.first, representative.second);
|
|
}
|
|
}
|
|
}
|
|
|
|
bool nano::wallets::exists (store::transaction const & transaction_a, nano::account const & account_a)
|
|
{
|
|
nano::lock_guard<nano::mutex> lock{ mutex };
|
|
auto result (false);
|
|
for (auto i (items.begin ()), n (items.end ()); !result && i != n; ++i)
|
|
{
|
|
result = i->second->store.exists (transaction_a, account_a);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void nano::wallets::stop ()
|
|
{
|
|
{
|
|
nano::lock_guard<nano::mutex> action_lock{ action_mutex };
|
|
stopped = true;
|
|
actions.clear ();
|
|
}
|
|
condition.notify_all ();
|
|
if (thread.joinable ())
|
|
{
|
|
thread.join ();
|
|
}
|
|
}
|
|
|
|
void nano::wallets::start ()
|
|
{
|
|
thread = std::thread{ [this] () {
|
|
nano::thread_role::set (nano::thread_role::name::wallet_actions);
|
|
do_wallet_actions ();
|
|
} };
|
|
}
|
|
|
|
nano::store::write_transaction nano::wallets::tx_begin_write ()
|
|
{
|
|
return env.tx_begin_write ();
|
|
}
|
|
|
|
nano::store::read_transaction nano::wallets::tx_begin_read ()
|
|
{
|
|
return env.tx_begin_read ();
|
|
}
|
|
|
|
void nano::wallets::clear_send_ids (store::transaction const & transaction_a)
|
|
{
|
|
auto status (mdb_drop (env.tx (transaction_a), send_action_ids, 0));
|
|
(void)status;
|
|
debug_assert (status == 0);
|
|
}
|
|
|
|
nano::wallet_representatives nano::wallets::reps () const
|
|
{
|
|
nano::lock_guard<nano::mutex> counts_guard{ reps_cache_mutex };
|
|
return representatives;
|
|
}
|
|
|
|
bool nano::wallets::check_rep (nano::account const & account_a, nano::uint128_t const & half_principal_weight_a, bool const acquire_lock_a)
|
|
{
|
|
auto weight = node.ledger.weight (account_a);
|
|
|
|
if (weight < node.config.vote_minimum.number ())
|
|
{
|
|
return false; // account not a representative
|
|
}
|
|
|
|
nano::unique_lock<nano::mutex> lock;
|
|
if (acquire_lock_a)
|
|
{
|
|
lock = nano::unique_lock<nano::mutex>{ reps_cache_mutex };
|
|
}
|
|
|
|
if (weight >= half_principal_weight_a)
|
|
{
|
|
representatives.half_principal = true;
|
|
}
|
|
|
|
auto insert_result = representatives.accounts.insert (account_a);
|
|
if (!insert_result.second)
|
|
{
|
|
return false; // account already exists
|
|
}
|
|
|
|
++representatives.voting;
|
|
|
|
return true;
|
|
}
|
|
|
|
void nano::wallets::compute_reps ()
|
|
{
|
|
nano::lock_guard<nano::mutex> guard{ mutex };
|
|
nano::lock_guard<nano::mutex> counts_guard{ reps_cache_mutex };
|
|
representatives.clear ();
|
|
auto half_principal_weight (node.minimum_principal_weight () / 2);
|
|
auto transaction (tx_begin_read ());
|
|
for (auto i (items.begin ()), n (items.end ()); i != n; ++i)
|
|
{
|
|
auto & wallet (*i->second);
|
|
decltype (wallet.representatives) representatives_l;
|
|
for (auto ii (wallet.store.begin (transaction)), nn (wallet.store.end ()); ii != nn; ++ii)
|
|
{
|
|
auto account (ii->first);
|
|
if (check_rep (account, half_principal_weight, false))
|
|
{
|
|
representatives_l.insert (account);
|
|
}
|
|
}
|
|
nano::lock_guard<nano::mutex> representatives_guard{ wallet.representatives_mutex };
|
|
wallet.representatives.swap (representatives_l);
|
|
}
|
|
}
|
|
|
|
void nano::wallets::ongoing_compute_reps ()
|
|
{
|
|
compute_reps ();
|
|
auto & node_l (node);
|
|
// Representation drifts quickly on the test network but very slowly on the live network
|
|
auto compute_delay = network_params.network.is_dev_network () ? std::chrono::milliseconds (10) : (network_params.network.is_test_network () ? std::chrono::milliseconds (nano::test_scan_wallet_reps_delay ()) : std::chrono::minutes (15));
|
|
node.workers.add_timed_task (std::chrono::steady_clock::now () + compute_delay, [&node_l] () {
|
|
node_l.wallets.ongoing_compute_reps ();
|
|
});
|
|
}
|
|
|
|
std::unordered_map<nano::wallet_id, std::shared_ptr<nano::wallet>> nano::wallets::get_wallets ()
|
|
{
|
|
debug_assert (!mutex.try_lock ());
|
|
return items;
|
|
}
|
|
|
|
nano::uint128_t const nano::wallets::generate_priority = std::numeric_limits<nano::uint128_t>::max ();
|
|
nano::uint128_t const nano::wallets::high_priority = std::numeric_limits<nano::uint128_t>::max () - 1;
|
|
|
|
nano::store::iterator<nano::account, nano::wallet_value> nano::wallet_store::begin (store::transaction const & transaction_a)
|
|
{
|
|
store::iterator<nano::account, nano::wallet_value> result (std::make_unique<nano::store::lmdb::iterator<nano::account, nano::wallet_value>> (transaction_a, env, handle, nano::store::lmdb::db_val (nano::account (special_count))));
|
|
return result;
|
|
}
|
|
|
|
nano::store::iterator<nano::account, nano::wallet_value> nano::wallet_store::begin (store::transaction const & transaction_a, nano::account const & key)
|
|
{
|
|
store::iterator<nano::account, nano::wallet_value> result (std::make_unique<nano::store::lmdb::iterator<nano::account, nano::wallet_value>> (transaction_a, env, handle, nano::store::lmdb::db_val (key)));
|
|
return result;
|
|
}
|
|
|
|
nano::store::iterator<nano::account, nano::wallet_value> nano::wallet_store::find (store::transaction const & transaction_a, nano::account const & key)
|
|
{
|
|
auto result (begin (transaction_a, key));
|
|
store::iterator<nano::account, nano::wallet_value> end (nullptr);
|
|
if (result != end)
|
|
{
|
|
if (result->first == key)
|
|
{
|
|
return result;
|
|
}
|
|
else
|
|
{
|
|
return end;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return end;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
nano::store::iterator<nano::account, nano::wallet_value> nano::wallet_store::end ()
|
|
{
|
|
return store::iterator<nano::account, nano::wallet_value> (nullptr);
|
|
}
|
|
nano::mdb_wallets_store::mdb_wallets_store (std::filesystem::path const & path_a, nano::lmdb_config const & lmdb_config_a) :
|
|
environment (error, path_a, nano::store::lmdb::env::options::make ().set_config (lmdb_config_a).override_config_sync (nano::lmdb_config::sync_strategy::always).override_config_map_size (1ULL * 1024 * 1024 * 1024))
|
|
{
|
|
}
|
|
|
|
bool nano::mdb_wallets_store::init_error () const
|
|
{
|
|
return error;
|
|
}
|
|
|
|
std::unique_ptr<nano::container_info_component> nano::collect_container_info (wallets & wallets, std::string const & name)
|
|
{
|
|
std::size_t items_count;
|
|
std::size_t actions_count;
|
|
{
|
|
nano::lock_guard<nano::mutex> guard{ wallets.mutex };
|
|
items_count = wallets.items.size ();
|
|
actions_count = wallets.actions.size ();
|
|
}
|
|
|
|
auto sizeof_item_element = sizeof (decltype (wallets.items)::value_type);
|
|
auto sizeof_actions_element = sizeof (decltype (wallets.actions)::value_type);
|
|
auto composite = std::make_unique<container_info_composite> (name);
|
|
composite->add_component (std::make_unique<container_info_leaf> (container_info{ "items", items_count, sizeof_item_element }));
|
|
composite->add_component (std::make_unique<container_info_leaf> (container_info{ "actions", actions_count, sizeof_actions_element }));
|
|
return composite;
|
|
}
|