456 lines
19 KiB
C++
456 lines
19 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.
|
|
|
|
#pragma once
|
|
|
|
#include <memory>
|
|
#include <boost/serialization/list.hpp>
|
|
#include <boost/serialization/vector.hpp>
|
|
#include <atomic>
|
|
|
|
#include "include_base_utils.h"
|
|
#include "cryptonote_core/account.h"
|
|
#include "cryptonote_core/account_boost_serialization.h"
|
|
#include "cryptonote_core/cryptonote_basic_impl.h"
|
|
#include "net/http_client.h"
|
|
#include "storages/http_abstract_invoke.h"
|
|
#include "rpc/core_rpc_server_commands_defs.h"
|
|
#include "cryptonote_core/cryptonote_format_utils.h"
|
|
#include "common/unordered_containers_boost_serialization.h"
|
|
#include "crypto/chacha8.h"
|
|
#include "crypto/hash.h"
|
|
|
|
#include "wallet_errors.h"
|
|
|
|
#define DEFAULT_TX_SPENDABLE_AGE 10
|
|
#define WALLET_RCP_CONNECTION_TIMEOUT 200000
|
|
|
|
namespace tools
|
|
{
|
|
class i_wallet2_callback
|
|
{
|
|
public:
|
|
virtual void on_new_block(uint64_t height, const cryptonote::block& block) {}
|
|
virtual void on_money_received(uint64_t height, const cryptonote::transaction& tx, size_t out_index) {}
|
|
virtual void on_money_spent(uint64_t height, const cryptonote::transaction& in_tx, size_t out_index, const cryptonote::transaction& spend_tx) {}
|
|
virtual void on_skip_transaction(uint64_t height, const cryptonote::transaction& tx) {}
|
|
};
|
|
|
|
struct tx_dust_policy
|
|
{
|
|
uint64_t dust_threshold;
|
|
bool add_to_fee;
|
|
cryptonote::account_public_address addr_for_dust;
|
|
|
|
tx_dust_policy(uint64_t a_dust_threshold = 0, bool an_add_to_fee = true, cryptonote::account_public_address an_addr_for_dust = cryptonote::account_public_address())
|
|
: dust_threshold(a_dust_threshold)
|
|
, add_to_fee(an_add_to_fee)
|
|
, addr_for_dust(an_addr_for_dust)
|
|
{
|
|
}
|
|
};
|
|
|
|
class wallet2
|
|
{
|
|
wallet2(const wallet2&) : m_run(true), m_callback(0) {};
|
|
public:
|
|
wallet2() : m_run(true), m_callback(0) {};
|
|
struct transfer_details
|
|
{
|
|
uint64_t m_block_height;
|
|
cryptonote::transaction m_tx;
|
|
size_t m_internal_output_index;
|
|
uint64_t m_global_output_index;
|
|
bool m_spent;
|
|
crypto::key_image m_key_image; //TODO: key_image stored twice :(
|
|
|
|
uint64_t amount() const { return m_tx.vout[m_internal_output_index].amount; }
|
|
};
|
|
|
|
struct payment_details
|
|
{
|
|
crypto::hash m_tx_hash;
|
|
uint64_t m_amount;
|
|
uint64_t m_block_height;
|
|
uint64_t m_unlock_time;
|
|
};
|
|
|
|
struct unconfirmed_transfer_details
|
|
{
|
|
cryptonote::transaction m_tx;
|
|
uint64_t m_change;
|
|
time_t m_sent_time;
|
|
};
|
|
|
|
typedef std::vector<transfer_details> transfer_container;
|
|
typedef std::unordered_multimap<crypto::hash, payment_details> payment_container;
|
|
|
|
struct pending_tx
|
|
{
|
|
cryptonote::transaction tx;
|
|
uint64_t dust, fee;
|
|
cryptonote::tx_destination_entry change_dts;
|
|
std::list<transfer_container::iterator> selected_transfers;
|
|
std::string key_images;
|
|
};
|
|
|
|
struct keys_file_data
|
|
{
|
|
crypto::chacha8_iv iv;
|
|
std::string account_data;
|
|
|
|
BEGIN_SERIALIZE_OBJECT()
|
|
FIELD(iv)
|
|
FIELD(account_data)
|
|
END_SERIALIZE()
|
|
};
|
|
|
|
crypto::secret_key generate(const std::string& wallet, const std::string& password, const crypto::secret_key& recovery_param = crypto::secret_key(), bool recover = false, bool two_random = false);
|
|
void load(const std::string& wallet, const std::string& password);
|
|
void store();
|
|
cryptonote::account_base& get_account(){return m_account;}
|
|
|
|
// upper_transaction_size_limit as defined below is set to
|
|
// approximately 125% of the fixed minimum allowable penalty
|
|
// free block size. TODO: fix this so that it actually takes
|
|
// into account the current median block size rather than
|
|
// the minimum block size.
|
|
void init(const std::string& daemon_address = "http://localhost:8080", uint64_t upper_transaction_size_limit = ((CRYPTONOTE_BLOCK_GRANTED_FULL_REWARD_ZONE * 125) / 100) - CRYPTONOTE_COINBASE_BLOB_RESERVED_SIZE);
|
|
bool deinit();
|
|
|
|
void stop() { m_run.store(false, std::memory_order_relaxed); }
|
|
|
|
i_wallet2_callback* callback() const { return m_callback; }
|
|
void callback(i_wallet2_callback* callback) { m_callback = callback; }
|
|
|
|
void refresh();
|
|
void refresh(size_t & blocks_fetched);
|
|
void refresh(size_t & blocks_fetched, bool& received_money);
|
|
bool refresh(size_t & blocks_fetched, bool& received_money, bool& ok);
|
|
|
|
uint64_t balance();
|
|
uint64_t unlocked_balance();
|
|
template<typename T>
|
|
void transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee, const std::vector<uint8_t>& extra, T destination_split_strategy, const tx_dust_policy& dust_policy);
|
|
template<typename T>
|
|
void transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee, const std::vector<uint8_t>& extra, T destination_split_strategy, const tx_dust_policy& dust_policy, cryptonote::transaction& tx, pending_tx& ptx);
|
|
void transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee, const std::vector<uint8_t>& extra);
|
|
void transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee, const std::vector<uint8_t>& extra, cryptonote::transaction& tx, pending_tx& ptx);
|
|
void commit_tx(pending_tx& ptx);
|
|
bool check_connection();
|
|
void get_transfers(wallet2::transfer_container& incoming_transfers) const;
|
|
void get_payments(const crypto::hash& payment_id, std::list<wallet2::payment_details>& payments) const;
|
|
uint64_t get_blockchain_current_height() const { return m_local_bc_height; }
|
|
template <class t_archive>
|
|
inline void serialize(t_archive &a, const unsigned int ver)
|
|
{
|
|
if(ver < 5)
|
|
return;
|
|
a & m_blockchain;
|
|
a & m_transfers;
|
|
a & m_account_public_address;
|
|
a & m_key_images;
|
|
if(ver < 6)
|
|
return;
|
|
a & m_unconfirmed_txs;
|
|
if(ver < 7)
|
|
return;
|
|
a & m_payments;
|
|
}
|
|
|
|
static void wallet_exists(const std::string& file_path, bool& keys_file_exists, bool& wallet_file_exists);
|
|
|
|
static bool parse_payment_id(const std::string& payment_id_str, crypto::hash& payment_id);
|
|
|
|
private:
|
|
bool store_keys(const std::string& keys_file_name, const std::string& password);
|
|
void load_keys(const std::string& keys_file_name, const std::string& password);
|
|
void process_new_transaction(const cryptonote::transaction& tx, uint64_t height);
|
|
void process_new_blockchain_entry(const cryptonote::block& b, cryptonote::block_complete_entry& bche, crypto::hash& bl_id, uint64_t height);
|
|
void detach_blockchain(uint64_t height);
|
|
void get_short_chain_history(std::list<crypto::hash>& ids);
|
|
bool is_tx_spendtime_unlocked(uint64_t unlock_time) const;
|
|
bool is_transfer_unlocked(const transfer_details& td) const;
|
|
bool clear();
|
|
void pull_blocks(size_t& blocks_added);
|
|
uint64_t select_transfers(uint64_t needed_money, bool add_dust, uint64_t dust, std::list<transfer_container::iterator>& selected_transfers);
|
|
bool prepare_file_names(const std::string& file_path);
|
|
void process_unconfirmed(const cryptonote::transaction& tx);
|
|
void add_unconfirmed_tx(const cryptonote::transaction& tx, uint64_t change_amount);
|
|
|
|
cryptonote::account_base m_account;
|
|
std::string m_daemon_address;
|
|
std::string m_wallet_file;
|
|
std::string m_keys_file;
|
|
epee::net_utils::http::http_simple_client m_http_client;
|
|
std::vector<crypto::hash> m_blockchain;
|
|
std::atomic<uint64_t> m_local_bc_height; //temporary workaround
|
|
std::unordered_map<crypto::hash, unconfirmed_transfer_details> m_unconfirmed_txs;
|
|
|
|
transfer_container m_transfers;
|
|
payment_container m_payments;
|
|
std::unordered_map<crypto::key_image, size_t> m_key_images;
|
|
cryptonote::account_public_address m_account_public_address;
|
|
uint64_t m_upper_transaction_size_limit; //TODO: auto-calc this value or request from daemon, now use some fixed value
|
|
|
|
std::atomic<bool> m_run;
|
|
|
|
i_wallet2_callback* m_callback;
|
|
};
|
|
}
|
|
BOOST_CLASS_VERSION(tools::wallet2, 7)
|
|
|
|
namespace boost
|
|
{
|
|
namespace serialization
|
|
{
|
|
template <class Archive>
|
|
inline void serialize(Archive &a, tools::wallet2::transfer_details &x, const boost::serialization::version_type ver)
|
|
{
|
|
a & x.m_block_height;
|
|
a & x.m_global_output_index;
|
|
a & x.m_internal_output_index;
|
|
a & x.m_tx;
|
|
a & x.m_spent;
|
|
a & x.m_key_image;
|
|
}
|
|
|
|
template <class Archive>
|
|
inline void serialize(Archive &a, tools::wallet2::unconfirmed_transfer_details &x, const boost::serialization::version_type ver)
|
|
{
|
|
a & x.m_change;
|
|
a & x.m_sent_time;
|
|
a & x.m_tx;
|
|
}
|
|
|
|
template <class Archive>
|
|
inline void serialize(Archive& a, tools::wallet2::payment_details& x, const boost::serialization::version_type ver)
|
|
{
|
|
a & x.m_tx_hash;
|
|
a & x.m_amount;
|
|
a & x.m_block_height;
|
|
a & x.m_unlock_time;
|
|
}
|
|
}
|
|
}
|
|
|
|
namespace tools
|
|
{
|
|
|
|
namespace detail
|
|
{
|
|
//----------------------------------------------------------------------------------------------------
|
|
inline void digit_split_strategy(const std::vector<cryptonote::tx_destination_entry>& dsts,
|
|
const cryptonote::tx_destination_entry& change_dst, uint64_t dust_threshold,
|
|
std::vector<cryptonote::tx_destination_entry>& splitted_dsts, uint64_t& dust)
|
|
{
|
|
splitted_dsts.clear();
|
|
dust = 0;
|
|
|
|
BOOST_FOREACH(auto& de, dsts)
|
|
{
|
|
cryptonote::decompose_amount_into_digits(de.amount, dust_threshold,
|
|
[&](uint64_t chunk) { splitted_dsts.push_back(cryptonote::tx_destination_entry(chunk, de.addr)); },
|
|
[&](uint64_t a_dust) { splitted_dsts.push_back(cryptonote::tx_destination_entry(a_dust, de.addr)); } );
|
|
}
|
|
|
|
cryptonote::decompose_amount_into_digits(change_dst.amount, dust_threshold,
|
|
[&](uint64_t chunk) { splitted_dsts.push_back(cryptonote::tx_destination_entry(chunk, change_dst.addr)); },
|
|
[&](uint64_t a_dust) { dust = a_dust; } );
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
inline void null_split_strategy(const std::vector<cryptonote::tx_destination_entry>& dsts,
|
|
const cryptonote::tx_destination_entry& change_dst, uint64_t dust_threshold,
|
|
std::vector<cryptonote::tx_destination_entry>& splitted_dsts, uint64_t& dust)
|
|
{
|
|
splitted_dsts = dsts;
|
|
|
|
dust = 0;
|
|
uint64_t change = change_dst.amount;
|
|
if (0 < dust_threshold)
|
|
{
|
|
for (uint64_t order = 10; order <= 10 * dust_threshold; order *= 10)
|
|
{
|
|
uint64_t dust_candidate = change_dst.amount % order;
|
|
uint64_t change_candidate = (change_dst.amount / order) * order;
|
|
if (dust_candidate <= dust_threshold)
|
|
{
|
|
dust = dust_candidate;
|
|
change = change_candidate;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (0 != change)
|
|
{
|
|
splitted_dsts.push_back(cryptonote::tx_destination_entry(change, change_dst.addr));
|
|
}
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
inline void print_source_entry(const cryptonote::tx_source_entry& src)
|
|
{
|
|
std::string indexes;
|
|
std::for_each(src.outputs.begin(), src.outputs.end(), [&](const cryptonote::tx_source_entry::output_entry& s_e) { indexes += boost::to_string(s_e.first) + " "; });
|
|
LOG_PRINT_L0("amount=" << cryptonote::print_money(src.amount) << ", real_output=" <<src.real_output << ", real_output_in_tx_index=" << src.real_output_in_tx_index << ", indexes: " << indexes);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
}
|
|
//----------------------------------------------------------------------------------------------------
|
|
template<typename T>
|
|
void wallet2::transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count,
|
|
uint64_t unlock_time, uint64_t fee, const std::vector<uint8_t>& extra, T destination_split_strategy, const tx_dust_policy& dust_policy)
|
|
{
|
|
pending_tx ptx;
|
|
cryptonote::transaction tx;
|
|
transfer(dsts, fake_outputs_count, unlock_time, fee, extra, destination_split_strategy, dust_policy, tx, ptx);
|
|
}
|
|
|
|
template<typename T>
|
|
void wallet2::transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count,
|
|
uint64_t unlock_time, uint64_t fee, const std::vector<uint8_t>& extra, T destination_split_strategy, const tx_dust_policy& dust_policy, cryptonote::transaction& tx, pending_tx &ptx)
|
|
{
|
|
using namespace cryptonote;
|
|
// throw if attempting a transaction with no destinations
|
|
THROW_WALLET_EXCEPTION_IF(dsts.empty(), error::zero_destination);
|
|
|
|
uint64_t needed_money = fee;
|
|
|
|
// calculate total amount being sent to all destinations
|
|
// throw if total amount overflows uint64_t
|
|
BOOST_FOREACH(auto& dt, dsts)
|
|
{
|
|
THROW_WALLET_EXCEPTION_IF(0 == dt.amount, error::zero_destination);
|
|
needed_money += dt.amount;
|
|
THROW_WALLET_EXCEPTION_IF(needed_money < dt.amount, error::tx_sum_overflow, dsts, fee);
|
|
}
|
|
|
|
// randomly select inputs for transaction
|
|
// throw if requested send amount is greater than amount available to send
|
|
std::list<transfer_container::iterator> selected_transfers;
|
|
uint64_t found_money = select_transfers(needed_money, 0 == fake_outputs_count, dust_policy.dust_threshold, selected_transfers);
|
|
THROW_WALLET_EXCEPTION_IF(found_money < needed_money, error::not_enough_money, found_money, needed_money - fee, fee);
|
|
|
|
typedef COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::out_entry out_entry;
|
|
typedef cryptonote::tx_source_entry::output_entry tx_output_entry;
|
|
|
|
COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::response daemon_resp = AUTO_VAL_INIT(daemon_resp);
|
|
if(fake_outputs_count)
|
|
{
|
|
COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::request req = AUTO_VAL_INIT(req);
|
|
req.outs_count = fake_outputs_count + 1;// add one to make possible (if need) to skip real output key
|
|
BOOST_FOREACH(transfer_container::iterator it, selected_transfers)
|
|
{
|
|
THROW_WALLET_EXCEPTION_IF(it->m_tx.vout.size() <= it->m_internal_output_index, error::wallet_internal_error,
|
|
"m_internal_output_index = " + std::to_string(it->m_internal_output_index) +
|
|
" is greater or equal to outputs count = " + std::to_string(it->m_tx.vout.size()));
|
|
req.amounts.push_back(it->amount());
|
|
}
|
|
|
|
bool r = epee::net_utils::invoke_http_bin_remote_command2(m_daemon_address + "/getrandom_outs.bin", req, daemon_resp, m_http_client, 200000);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::no_connection_to_daemon, "getrandom_outs.bin");
|
|
THROW_WALLET_EXCEPTION_IF(daemon_resp.status == CORE_RPC_STATUS_BUSY, error::daemon_busy, "getrandom_outs.bin");
|
|
THROW_WALLET_EXCEPTION_IF(daemon_resp.status != CORE_RPC_STATUS_OK, error::get_random_outs_error, daemon_resp.status);
|
|
THROW_WALLET_EXCEPTION_IF(daemon_resp.outs.size() != selected_transfers.size(), error::wallet_internal_error,
|
|
"daemon returned wrong response for getrandom_outs.bin, wrong amounts count = " +
|
|
std::to_string(daemon_resp.outs.size()) + ", expected " + std::to_string(selected_transfers.size()));
|
|
|
|
std::vector<COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::outs_for_amount> scanty_outs;
|
|
BOOST_FOREACH(COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::outs_for_amount& amount_outs, daemon_resp.outs)
|
|
{
|
|
if (amount_outs.outs.size() < fake_outputs_count)
|
|
{
|
|
scanty_outs.push_back(amount_outs);
|
|
}
|
|
}
|
|
THROW_WALLET_EXCEPTION_IF(!scanty_outs.empty(), error::not_enough_outs_to_mix, scanty_outs, fake_outputs_count);
|
|
}
|
|
|
|
//prepare inputs
|
|
size_t i = 0;
|
|
std::vector<cryptonote::tx_source_entry> sources;
|
|
BOOST_FOREACH(transfer_container::iterator it, selected_transfers)
|
|
{
|
|
sources.resize(sources.size()+1);
|
|
cryptonote::tx_source_entry& src = sources.back();
|
|
transfer_details& td = *it;
|
|
src.amount = td.amount();
|
|
//paste mixin transaction
|
|
if(daemon_resp.outs.size())
|
|
{
|
|
daemon_resp.outs[i].outs.sort([](const out_entry& a, const out_entry& b){return a.global_amount_index < b.global_amount_index;});
|
|
BOOST_FOREACH(out_entry& daemon_oe, daemon_resp.outs[i].outs)
|
|
{
|
|
if(td.m_global_output_index == daemon_oe.global_amount_index)
|
|
continue;
|
|
tx_output_entry oe;
|
|
oe.first = daemon_oe.global_amount_index;
|
|
oe.second = daemon_oe.out_key;
|
|
src.outputs.push_back(oe);
|
|
if(src.outputs.size() >= fake_outputs_count)
|
|
break;
|
|
}
|
|
}
|
|
|
|
//paste real transaction to the random index
|
|
auto it_to_insert = std::find_if(src.outputs.begin(), src.outputs.end(), [&](const tx_output_entry& a)
|
|
{
|
|
return a.first >= td.m_global_output_index;
|
|
});
|
|
//size_t real_index = src.outputs.size() ? (rand() % src.outputs.size() ):0;
|
|
tx_output_entry real_oe;
|
|
real_oe.first = td.m_global_output_index;
|
|
real_oe.second = boost::get<txout_to_key>(td.m_tx.vout[td.m_internal_output_index].target).key;
|
|
auto interted_it = src.outputs.insert(it_to_insert, real_oe);
|
|
src.real_out_tx_key = get_tx_pub_key_from_extra(td.m_tx);
|
|
src.real_output = interted_it - src.outputs.begin();
|
|
src.real_output_in_tx_index = td.m_internal_output_index;
|
|
detail::print_source_entry(src);
|
|
++i;
|
|
}
|
|
|
|
cryptonote::tx_destination_entry change_dts = AUTO_VAL_INIT(change_dts);
|
|
if (needed_money < found_money)
|
|
{
|
|
change_dts.addr = m_account.get_keys().m_account_address;
|
|
change_dts.amount = found_money - needed_money;
|
|
}
|
|
|
|
uint64_t dust = 0;
|
|
std::vector<cryptonote::tx_destination_entry> splitted_dsts;
|
|
destination_split_strategy(dsts, change_dts, dust_policy.dust_threshold, splitted_dsts, dust);
|
|
THROW_WALLET_EXCEPTION_IF(dust_policy.dust_threshold < dust, error::wallet_internal_error, "invalid dust value: dust = " +
|
|
std::to_string(dust) + ", dust_threshold = " + std::to_string(dust_policy.dust_threshold));
|
|
if (0 != dust && !dust_policy.add_to_fee)
|
|
{
|
|
splitted_dsts.push_back(cryptonote::tx_destination_entry(dust, dust_policy.addr_for_dust));
|
|
}
|
|
|
|
bool r = cryptonote::construct_tx(m_account.get_keys(), sources, splitted_dsts, extra, tx, unlock_time);
|
|
THROW_WALLET_EXCEPTION_IF(!r, error::tx_not_constructed, sources, splitted_dsts, unlock_time);
|
|
THROW_WALLET_EXCEPTION_IF(m_upper_transaction_size_limit <= get_object_blobsize(tx), error::tx_too_big, tx, m_upper_transaction_size_limit);
|
|
|
|
std::string key_images;
|
|
bool all_are_txin_to_key = std::all_of(tx.vin.begin(), tx.vin.end(), [&](const txin_v& s_e) -> bool
|
|
{
|
|
CHECKED_GET_SPECIFIC_VARIANT(s_e, const txin_to_key, in, false);
|
|
key_images += boost::to_string(in.k_image) + " ";
|
|
return true;
|
|
});
|
|
THROW_WALLET_EXCEPTION_IF(!all_are_txin_to_key, error::unexpected_txin_type, tx);
|
|
|
|
ptx.key_images = key_images;
|
|
ptx.fee = fee;
|
|
ptx.dust = dust;
|
|
ptx.tx = tx;
|
|
ptx.change_dts = change_dts;
|
|
ptx.selected_transfers = selected_transfers;
|
|
|
|
}
|
|
|
|
|
|
}
|