mirror of
https://git.wownero.com/wownero/wownero.git
synced 2024-08-15 01:03:23 +00:00
611 lines
24 KiB
C++
611 lines
24 KiB
C++
// Copyright (c) 2012-2013 The Cryptonote developers
|
|
// Distributed under the MIT/X11 software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
|
|
#include <boost/archive/binary_oarchive.hpp>
|
|
#include <boost/archive/binary_iarchive.hpp>
|
|
|
|
#include <boost/utility/value_init.hpp>
|
|
#include "include_base_utils.h"
|
|
using namespace epee;
|
|
|
|
#include "wallet2.h"
|
|
#include "cryptonote_core/cryptonote_format_utils.h"
|
|
#include "rpc/core_rpc_server_commands_defs.h"
|
|
#include "misc_language.h"
|
|
#include "cryptonote_core/cryptonote_basic_impl.h"
|
|
#include "common/boost_serialization_helper.h"
|
|
#include "profile_tools.h"
|
|
#include "crypto/crypto.h"
|
|
#include "serialization/binary_utils.h"
|
|
|
|
using namespace cryptonote;
|
|
|
|
namespace tools
|
|
{
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::init(const std::string& daemon_address, uint64_t upper_transaction_size_limit)
|
|
{
|
|
m_upper_transaction_size_limit = upper_transaction_size_limit;
|
|
m_daemon_address = daemon_address;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::process_new_transaction(const cryptonote::transaction& tx, uint64_t height)
|
|
{
|
|
process_unconfirmed(tx);
|
|
std::vector<size_t> outs;
|
|
uint64_t tx_money_got_in_outs = 0;
|
|
crypto::public_key tx_pub_key = null_pkey;
|
|
bool r = parse_and_validate_tx_extra(tx, tx_pub_key);
|
|
|
|
// Temporarily disabled due to messed up tx_extra from someone
|
|
// screwing around with MMing. 2014-04-30
|
|
// THROW_WALLET_EXCEPTION_IF(!r, error::tx_extra_parse_error, tx);
|
|
|
|
// We don't know how to handle this weird tx, so return
|
|
if (!r) return;
|
|
|
|
r = lookup_acc_outs(m_account.get_keys(), tx, tx_pub_key, outs, tx_money_got_in_outs);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::acc_outs_lookup_error, tx, tx_pub_key, m_account.get_keys());
|
|
|
|
if(!outs.empty() && tx_money_got_in_outs)
|
|
{
|
|
//good news - got money! take care about it
|
|
//usually we have only one transfer for user in transaction
|
|
cryptonote::COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::request req = AUTO_VAL_INIT(req);
|
|
cryptonote::COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::response res = AUTO_VAL_INIT(res);
|
|
req.txid = get_transaction_hash(tx);
|
|
bool r = net_utils::invoke_http_bin_remote_command2(m_daemon_address + "/get_o_indexes.bin", req, res, m_http_client, WALLET_RCP_CONNECTION_TIMEOUT);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::no_connection_to_daemon, "get_o_indexes.bin");
|
|
THROW_WALLET_EXCEPTION_IF(res.status == CORE_RPC_STATUS_BUSY, error::daemon_busy, "get_o_indexes.bin");
|
|
THROW_WALLET_EXCEPTION_IF(res.status != CORE_RPC_STATUS_OK, error::get_out_indices_error, res.status);
|
|
THROW_WALLET_EXCEPTION_IF(res.o_indexes.size() != tx.vout.size(), error::wallet_internal_error,
|
|
"transactions outputs size=" + std::to_string(tx.vout.size()) +
|
|
" not match with COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES response size=" + std::to_string(res.o_indexes.size()));
|
|
|
|
BOOST_FOREACH(size_t o, outs)
|
|
{
|
|
THROW_WALLET_EXCEPTION_IF(tx.vout.size() <= o, error::wallet_internal_error, "wrong out in transaction: internal index=" +
|
|
std::to_string(o) + ", total_outs=" + std::to_string(tx.vout.size()));
|
|
|
|
m_transfers.push_back(boost::value_initialized<transfer_details>());
|
|
transfer_details& td = m_transfers.back();
|
|
td.m_block_height = height;
|
|
td.m_internal_output_index = o;
|
|
td.m_global_output_index = res.o_indexes[o];
|
|
td.m_tx = tx;
|
|
td.m_spent = false;
|
|
cryptonote::keypair in_ephemeral;
|
|
cryptonote::generate_key_image_helper(m_account.get_keys(), tx_pub_key, o, in_ephemeral, td.m_key_image);
|
|
THROW_WALLET_EXCEPTION_IF(in_ephemeral.pub != boost::get<cryptonote::txout_to_key>(tx.vout[o].target).key,
|
|
error::wallet_internal_error, "key_image generated ephemeral public key not matched with output_key");
|
|
|
|
m_key_images[td.m_key_image] = m_transfers.size()-1;
|
|
LOG_PRINT_L0("Received money: " << print_money(td.amount()) << ", with tx: " << get_transaction_hash(tx));
|
|
if (0 != m_callback)
|
|
m_callback->on_money_received(height, td.m_tx, td.m_internal_output_index);
|
|
}
|
|
}
|
|
// check all outputs for spending (compare key images)
|
|
BOOST_FOREACH(auto& in, tx.vin)
|
|
{
|
|
if(in.type() != typeid(cryptonote::txin_to_key))
|
|
continue;
|
|
auto it = m_key_images.find(boost::get<cryptonote::txin_to_key>(in).k_image);
|
|
if(it != m_key_images.end())
|
|
{
|
|
LOG_PRINT_L0("Spent money: " << print_money(boost::get<cryptonote::txin_to_key>(in).amount) << ", with tx: " << get_transaction_hash(tx));
|
|
transfer_details& td = m_transfers[it->second];
|
|
td.m_spent = true;
|
|
if (0 != m_callback)
|
|
m_callback->on_money_spent(height, td.m_tx, td.m_internal_output_index, tx);
|
|
}
|
|
}
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::process_unconfirmed(const cryptonote::transaction& tx)
|
|
{
|
|
auto unconf_it = m_unconfirmed_txs.find(get_transaction_hash(tx));
|
|
if(unconf_it != m_unconfirmed_txs.end())
|
|
m_unconfirmed_txs.erase(unconf_it);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::process_new_blockchain_entry(const cryptonote::block& b, cryptonote::block_complete_entry& bche, crypto::hash& bl_id, uint64_t height)
|
|
{
|
|
//handle transactions from new block
|
|
THROW_WALLET_EXCEPTION_IF(height != m_blockchain.size(), error::wallet_internal_error,
|
|
"current_index=" + std::to_string(height) + ", m_blockchain.size()=" + std::to_string(m_blockchain.size()));
|
|
|
|
//optimization: seeking only for blocks that are not older then the wallet creation time plus 1 day. 1 day is for possible user incorrect time setup
|
|
if(b.timestamp + 60*60*24 > m_account.get_createtime())
|
|
{
|
|
TIME_MEASURE_START(miner_tx_handle_time);
|
|
process_new_transaction(b.miner_tx, height);
|
|
TIME_MEASURE_FINISH(miner_tx_handle_time);
|
|
|
|
TIME_MEASURE_START(txs_handle_time);
|
|
BOOST_FOREACH(auto& txblob, bche.txs)
|
|
{
|
|
cryptonote::transaction tx;
|
|
bool r = parse_and_validate_tx_from_blob(txblob, tx);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::tx_parse_error, txblob);
|
|
process_new_transaction(tx, height);
|
|
}
|
|
TIME_MEASURE_FINISH(txs_handle_time);
|
|
LOG_PRINT_L2("Processed block: " << bl_id << ", height " << height << ", " << miner_tx_handle_time + txs_handle_time << "(" << miner_tx_handle_time << "/" << txs_handle_time <<")ms");
|
|
}else
|
|
{
|
|
LOG_PRINT_L2( "Skipped block by timestamp, height: " << height << ", block time " << b.timestamp << ", account time " << m_account.get_createtime());
|
|
}
|
|
m_blockchain.push_back(bl_id);
|
|
++m_local_bc_height;
|
|
|
|
if (0 != m_callback)
|
|
m_callback->on_new_block(height, b);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::get_short_chain_history(std::list<crypto::hash>& ids)
|
|
{
|
|
size_t i = 0;
|
|
size_t current_multiplier = 1;
|
|
size_t sz = m_blockchain.size();
|
|
if(!sz)
|
|
return;
|
|
size_t current_back_offset = 1;
|
|
bool genesis_included = false;
|
|
while(current_back_offset < sz)
|
|
{
|
|
ids.push_back(m_blockchain[sz-current_back_offset]);
|
|
if(sz-current_back_offset == 0)
|
|
genesis_included = true;
|
|
if(i < 10)
|
|
{
|
|
++current_back_offset;
|
|
}else
|
|
{
|
|
current_back_offset += current_multiplier *= 2;
|
|
}
|
|
++i;
|
|
}
|
|
if(!genesis_included)
|
|
ids.push_back(m_blockchain[0]);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::pull_blocks(size_t& blocks_added)
|
|
{
|
|
blocks_added = 0;
|
|
cryptonote::COMMAND_RPC_GET_BLOCKS_FAST::request req = AUTO_VAL_INIT(req);
|
|
cryptonote::COMMAND_RPC_GET_BLOCKS_FAST::response res = AUTO_VAL_INIT(res);
|
|
get_short_chain_history(req.block_ids);
|
|
bool r = net_utils::invoke_http_bin_remote_command2(m_daemon_address + "/getblocks.bin", req, res, m_http_client, WALLET_RCP_CONNECTION_TIMEOUT);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::no_connection_to_daemon, "getblocks.bin");
|
|
THROW_WALLET_EXCEPTION_IF(res.status == CORE_RPC_STATUS_BUSY, error::daemon_busy, "getblocks.bin");
|
|
THROW_WALLET_EXCEPTION_IF(res.status != CORE_RPC_STATUS_OK, error::get_blocks_error, res.status);
|
|
THROW_WALLET_EXCEPTION_IF(m_blockchain.size() <= res.start_height, error::wallet_internal_error,
|
|
"wrong daemon response: m_start_height=" + std::to_string(res.start_height) +
|
|
" not less than local blockchain size=" + std::to_string(m_blockchain.size()));
|
|
|
|
size_t current_index = res.start_height;
|
|
BOOST_FOREACH(auto& bl_entry, res.blocks)
|
|
{
|
|
cryptonote::block bl;
|
|
r = cryptonote::parse_and_validate_block_from_blob(bl_entry.block, bl);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::block_parse_error, bl_entry.block);
|
|
|
|
crypto::hash bl_id = get_block_hash(bl);
|
|
if(current_index >= m_blockchain.size())
|
|
{
|
|
process_new_blockchain_entry(bl, bl_entry, bl_id, current_index);
|
|
++blocks_added;
|
|
}
|
|
else if(bl_id != m_blockchain[current_index])
|
|
{
|
|
//split detected here !!!
|
|
THROW_WALLET_EXCEPTION_IF(current_index == res.start_height, error::wallet_internal_error,
|
|
"wrong daemon response: split starts from the first block in response " + string_tools::pod_to_hex(bl_id) +
|
|
" (height " + std::to_string(res.start_height) + "), local block id at this height: " +
|
|
string_tools::pod_to_hex(m_blockchain[current_index]));
|
|
|
|
detach_blockchain(current_index);
|
|
process_new_blockchain_entry(bl, bl_entry, bl_id, current_index);
|
|
}
|
|
else
|
|
{
|
|
LOG_PRINT_L2("Block is already in blockchain: " << string_tools::pod_to_hex(bl_id));
|
|
}
|
|
|
|
++current_index;
|
|
}
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::refresh()
|
|
{
|
|
size_t blocks_fetched = 0;
|
|
refresh(blocks_fetched);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::refresh(size_t & blocks_fetched)
|
|
{
|
|
bool received_money = false;
|
|
refresh(blocks_fetched, received_money);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::refresh(size_t & blocks_fetched, bool& received_money)
|
|
{
|
|
received_money = false;
|
|
blocks_fetched = 0;
|
|
size_t added_blocks = 0;
|
|
size_t try_count = 0;
|
|
crypto::hash last_tx_hash_id = m_transfers.size() ? get_transaction_hash(m_transfers.back().m_tx) : null_hash;
|
|
|
|
while(m_run.load(std::memory_order_relaxed))
|
|
{
|
|
try
|
|
{
|
|
pull_blocks(added_blocks);
|
|
blocks_fetched += added_blocks;
|
|
if(!added_blocks)
|
|
break;
|
|
}
|
|
catch (const std::exception&)
|
|
{
|
|
blocks_fetched += added_blocks;
|
|
if(try_count < 3)
|
|
{
|
|
LOG_PRINT_L1("Another try pull_blocks (try_count=" << try_count << ")...");
|
|
++try_count;
|
|
}
|
|
else
|
|
{
|
|
LOG_ERROR("pull_blocks failed, try_count=" << try_count);
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
if(last_tx_hash_id != (m_transfers.size() ? get_transaction_hash(m_transfers.back().m_tx) : null_hash))
|
|
received_money = true;
|
|
|
|
LOG_PRINT_L1("Refresh done, blocks received: " << blocks_fetched << ", balance: " << print_money(balance()) << ", unlocked: " << print_money(unlocked_balance()));
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
bool wallet2::refresh(size_t & blocks_fetched, bool& received_money, bool& ok)
|
|
{
|
|
try
|
|
{
|
|
refresh(blocks_fetched, received_money);
|
|
ok = true;
|
|
}
|
|
catch (...)
|
|
{
|
|
ok = false;
|
|
}
|
|
return ok;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::detach_blockchain(uint64_t height)
|
|
{
|
|
LOG_PRINT_L0("Detaching blockchain on height " << height);
|
|
size_t transfers_detached = 0;
|
|
|
|
auto it = std::find_if(m_transfers.begin(), m_transfers.end(), [&](const transfer_details& td){return td.m_block_height >= height;});
|
|
size_t i_start = it - m_transfers.begin();
|
|
|
|
for(size_t i = i_start; i!= m_transfers.size();i++)
|
|
{
|
|
auto it_ki = m_key_images.find(m_transfers[i].m_key_image);
|
|
THROW_WALLET_EXCEPTION_IF(it_ki == m_key_images.end(), error::wallet_internal_error, "key image not found");
|
|
m_key_images.erase(it_ki);
|
|
++transfers_detached;
|
|
}
|
|
m_transfers.erase(it, m_transfers.end());
|
|
|
|
size_t blocks_detached = m_blockchain.end() - (m_blockchain.begin()+height);
|
|
m_blockchain.erase(m_blockchain.begin()+height, m_blockchain.end());
|
|
m_local_bc_height -= blocks_detached;
|
|
|
|
LOG_PRINT_L0("Detached blockchain on height " << height << ", transfers detached " << transfers_detached << ", blocks detached " << blocks_detached);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
bool wallet2::deinit()
|
|
{
|
|
return true;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
bool wallet2::clear()
|
|
{
|
|
m_blockchain.clear();
|
|
m_transfers.clear();
|
|
cryptonote::block b;
|
|
cryptonote::generate_genesis_block(b);
|
|
m_blockchain.push_back(get_block_hash(b));
|
|
m_local_bc_height = 1;
|
|
return true;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
bool wallet2::store_keys(const std::string& keys_file_name, const std::string& password)
|
|
{
|
|
std::string account_data;
|
|
bool r = epee::serialization::store_t_to_binary(m_account, account_data);
|
|
CHECK_AND_ASSERT_MES(r, false, "failed to serialize wallet keys");
|
|
wallet2::keys_file_data keys_file_data = boost::value_initialized<wallet2::keys_file_data>();
|
|
|
|
crypto::chacha8_key key;
|
|
crypto::generate_chacha8_key(password, key);
|
|
std::string cipher;
|
|
cipher.resize(account_data.size());
|
|
keys_file_data.iv = crypto::rand<crypto::chacha8_iv>();
|
|
crypto::chacha8(account_data.data(), account_data.size(), key, keys_file_data.iv, &cipher[0]);
|
|
keys_file_data.account_data = cipher;
|
|
|
|
std::string buf;
|
|
r = ::serialization::dump_binary(keys_file_data, buf);
|
|
r = r && epee::file_io_utils::save_string_to_file(keys_file_name, buf); //and never touch wallet_keys_file again, only read
|
|
CHECK_AND_ASSERT_MES(r, false, "failed to generate wallet keys file " << keys_file_name);
|
|
|
|
return true;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
namespace
|
|
{
|
|
bool verify_keys(const crypto::secret_key& sec, const crypto::public_key& expected_pub)
|
|
{
|
|
crypto::public_key pub;
|
|
bool r = crypto::secret_key_to_public_key(sec, pub);
|
|
return r && expected_pub == pub;
|
|
}
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::load_keys(const std::string& keys_file_name, const std::string& password)
|
|
{
|
|
wallet2::keys_file_data keys_file_data;
|
|
std::string buf;
|
|
bool r = epee::file_io_utils::load_file_to_string(keys_file_name, buf);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::file_read_error, keys_file_name);
|
|
r = ::serialization::parse_binary(buf, keys_file_data);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::wallet_internal_error, "internal error: failed to deserialize \"" + keys_file_name + '\"');
|
|
|
|
crypto::chacha8_key key;
|
|
crypto::generate_chacha8_key(password, key);
|
|
std::string account_data;
|
|
account_data.resize(keys_file_data.account_data.size());
|
|
crypto::chacha8(keys_file_data.account_data.data(), keys_file_data.account_data.size(), key, keys_file_data.iv, &account_data[0]);
|
|
|
|
const cryptonote::account_keys& keys = m_account.get_keys();
|
|
r = epee::serialization::load_t_from_binary(m_account, account_data);
|
|
r = r && verify_keys(keys.m_view_secret_key, keys.m_account_address.m_view_public_key);
|
|
r = r && verify_keys(keys.m_spend_secret_key, keys.m_account_address.m_spend_public_key);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::invalid_password);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::generate(const std::string& wallet_, const std::string& password)
|
|
{
|
|
clear();
|
|
prepare_file_names(wallet_);
|
|
|
|
boost::system::error_code ignored_ec;
|
|
THROW_WALLET_EXCEPTION_IF(boost::filesystem::exists(m_wallet_file, ignored_ec), error::file_exists, m_wallet_file);
|
|
THROW_WALLET_EXCEPTION_IF(boost::filesystem::exists(m_keys_file, ignored_ec), error::file_exists, m_keys_file);
|
|
|
|
m_account.generate();
|
|
m_account_public_address = m_account.get_keys().m_account_address;
|
|
|
|
bool r = store_keys(m_keys_file, password);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::file_save_error, m_keys_file);
|
|
|
|
r = file_io_utils::save_string_to_file(m_wallet_file + ".address.txt", m_account.get_public_address_str());
|
|
if(!r) LOG_PRINT_RED_L0("String with address text not saved");
|
|
|
|
store();
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
bool wallet2::prepare_file_names(const std::string& file_path)
|
|
{
|
|
m_keys_file = file_path;
|
|
m_wallet_file = file_path;
|
|
boost::system::error_code e;
|
|
if(string_tools::get_extension(m_keys_file) == "keys")
|
|
{//provided keys file name
|
|
m_wallet_file = string_tools::cut_off_extension(m_wallet_file);
|
|
}else
|
|
{//provided wallet file name
|
|
m_keys_file += ".keys";
|
|
}
|
|
return true;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
bool wallet2::check_connection()
|
|
{
|
|
if(m_http_client.is_connected())
|
|
return true;
|
|
|
|
net_utils::http::url_content u;
|
|
net_utils::parse_url(m_daemon_address, u);
|
|
if(!u.port)
|
|
u.port = RPC_DEFAULT_PORT;
|
|
return m_http_client.connect(u.host, std::to_string(u.port), WALLET_RCP_CONNECTION_TIMEOUT);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::load(const std::string& wallet_, const std::string& password)
|
|
{
|
|
clear();
|
|
prepare_file_names(wallet_);
|
|
|
|
boost::system::error_code e;
|
|
bool exists = boost::filesystem::exists(m_keys_file, e);
|
|
THROW_WALLET_EXCEPTION_IF(e || !exists, error::file_not_found, m_keys_file);
|
|
|
|
load_keys(m_keys_file, password);
|
|
LOG_PRINT_L0("Loaded wallet keys file, with public address: " << m_account.get_public_address_str());
|
|
|
|
//keys loaded ok!
|
|
//try to load wallet file. but even if we failed, it is not big problem
|
|
if(!boost::filesystem::exists(m_wallet_file, e) || e)
|
|
{
|
|
LOG_PRINT_L0("file not found: " << m_wallet_file << ", starting with empty blockchain");
|
|
m_account_public_address = m_account.get_keys().m_account_address;
|
|
return;
|
|
}
|
|
bool r = tools::unserialize_obj_from_file(*this, m_wallet_file);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::file_read_error, m_wallet_file);
|
|
THROW_WALLET_EXCEPTION_IF(
|
|
m_account_public_address.m_spend_public_key != m_account.get_keys().m_account_address.m_spend_public_key ||
|
|
m_account_public_address.m_view_public_key != m_account.get_keys().m_account_address.m_view_public_key,
|
|
error::wallet_files_doesnt_correspond, m_keys_file, m_wallet_file);
|
|
|
|
if(m_blockchain.empty())
|
|
{
|
|
cryptonote::block b;
|
|
cryptonote::generate_genesis_block(b);
|
|
m_blockchain.push_back(get_block_hash(b));
|
|
}
|
|
m_local_bc_height = m_blockchain.size();
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::store()
|
|
{
|
|
bool r = tools::serialize_obj_to_file(*this, m_wallet_file);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::file_save_error, m_wallet_file);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
uint64_t wallet2::unlocked_balance()
|
|
{
|
|
uint64_t amount = 0;
|
|
BOOST_FOREACH(transfer_details& td, m_transfers)
|
|
if(!td.m_spent && is_transfer_unlocked(td))
|
|
amount += td.amount();
|
|
|
|
return amount;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
uint64_t wallet2::balance()
|
|
{
|
|
uint64_t amount = 0;
|
|
BOOST_FOREACH(auto& td, m_transfers)
|
|
if(!td.m_spent)
|
|
amount += td.amount();
|
|
|
|
|
|
BOOST_FOREACH(auto& utx, m_unconfirmed_txs)
|
|
amount+= utx.second.m_change;
|
|
|
|
return amount;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::get_transfers(wallet2::transfer_container& incoming_transfers) const
|
|
{
|
|
incoming_transfers = m_transfers;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
bool wallet2::is_transfer_unlocked(const transfer_details& td) const
|
|
{
|
|
if(!is_tx_spendtime_unlocked(td.m_tx.unlock_time))
|
|
return false;
|
|
|
|
if(td.m_block_height + DEFAULT_TX_SPENDABLE_AGE > m_blockchain.size())
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
bool wallet2::is_tx_spendtime_unlocked(uint64_t unlock_time) const
|
|
{
|
|
if(unlock_time < CRYPTONOTE_MAX_BLOCK_NUMBER)
|
|
{
|
|
//interpret as block index
|
|
if(m_blockchain.size()-1 + CRYPTONOTE_LOCKED_TX_ALLOWED_DELTA_BLOCKS >= unlock_time)
|
|
return true;
|
|
else
|
|
return false;
|
|
}else
|
|
{
|
|
//interpret as time
|
|
uint64_t current_time = static_cast<uint64_t>(time(NULL));
|
|
if(current_time + CRYPTONOTE_LOCKED_TX_ALLOWED_DELTA_SECONDS >= unlock_time)
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
return false;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
namespace
|
|
{
|
|
template<typename T>
|
|
T pop_random_value(std::vector<T>& vec)
|
|
{
|
|
CHECK_AND_ASSERT_MES(!vec.empty(), T(), "Vector must be non-empty");
|
|
|
|
size_t idx = crypto::rand<size_t>() % vec.size();
|
|
T res = vec[idx];
|
|
if (idx + 1 != vec.size())
|
|
{
|
|
vec[idx] = vec.back();
|
|
}
|
|
vec.resize(vec.size() - 1);
|
|
|
|
return res;
|
|
}
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
uint64_t wallet2::select_transfers(uint64_t needed_money, bool add_dust, uint64_t dust, std::list<transfer_container::iterator>& selected_transfers)
|
|
{
|
|
std::vector<size_t> unused_transfers_indices;
|
|
std::vector<size_t> unused_dust_indices;
|
|
for (size_t i = 0; i < m_transfers.size(); ++i)
|
|
{
|
|
const transfer_details& td = m_transfers[i];
|
|
if (!td.m_spent && is_transfer_unlocked(td))
|
|
{
|
|
if (dust < td.amount())
|
|
unused_transfers_indices.push_back(i);
|
|
else
|
|
unused_dust_indices.push_back(i);
|
|
}
|
|
}
|
|
|
|
bool select_one_dust = add_dust && !unused_dust_indices.empty();
|
|
uint64_t found_money = 0;
|
|
while (found_money < needed_money && (!unused_transfers_indices.empty() || !unused_dust_indices.empty()))
|
|
{
|
|
size_t idx;
|
|
if (select_one_dust)
|
|
{
|
|
idx = pop_random_value(unused_dust_indices);
|
|
select_one_dust = false;
|
|
}
|
|
else
|
|
{
|
|
idx = !unused_transfers_indices.empty() ? pop_random_value(unused_transfers_indices) : pop_random_value(unused_dust_indices);
|
|
}
|
|
|
|
transfer_container::iterator it = m_transfers.begin() + idx;
|
|
selected_transfers.push_back(it);
|
|
found_money += it->amount();
|
|
}
|
|
|
|
return found_money;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::add_unconfirmed_tx(const cryptonote::transaction& tx, uint64_t change_amount)
|
|
{
|
|
unconfirmed_transfer_details& utd = m_unconfirmed_txs[cryptonote::get_transaction_hash(tx)];
|
|
utd.m_change = change_amount;
|
|
utd.m_sent_time = time(NULL);
|
|
utd.m_tx = tx;
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count,
|
|
uint64_t unlock_time, uint64_t fee, cryptonote::transaction& tx)
|
|
{
|
|
transfer(dsts, fake_outputs_count, unlock_time, fee, detail::digit_split_strategy, tx_dust_policy(fee), tx);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
void wallet2::transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count,
|
|
uint64_t unlock_time, uint64_t fee)
|
|
{
|
|
cryptonote::transaction tx;
|
|
transfer(dsts, fake_outputs_count, unlock_time, fee, tx);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
}
|