diff options
Diffstat (limited to '')
| -rw-r--r-- | src/wallet/wallet2.h | 408 |
1 files changed, 408 insertions, 0 deletions
diff --git a/src/wallet/wallet2.h b/src/wallet/wallet2.h new file mode 100644 index 000000000..27799e2da --- /dev/null +++ b/src/wallet/wallet2.h @@ -0,0 +1,408 @@ +// 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" + +#define DEFAULT_TX_SPENDABLE_AGE 10 +#define WALLET_RCP_CONNECTION_TIMEOUT 200000 + +namespace tools +{ + class wallet2 + { + wallet2(const wallet2&){}; + public: + wallet2(){}; + 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; } + }; + typedef std::vector<transfer_details> transfer_container; + + 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) + { + } + }; + + struct keys_file_data + { + crypto::chacha8_iv iv; + std::string account_data; + + BEGIN_SERIALIZE_OBJECT() + FIELD(iv) + FIELD(account_data) + END_SERIALIZE() + }; + + struct transafer_fail_details + { + enum fail_reason + { + error_ok = 0, + error_not_connected, + error_rejected_by_daemon, + error_too_big_transaction, + error_not_enough_money, + error_internal_error + }; + fail_reason reason; + uint64_t tx_blob_size; + uint64_t max_expected_tx_blob_size; + }; + + bool generate(const std::string& wallet, const std::string& password); + bool load(const std::string& wallet, const std::string& password); + bool store(); + cryptonote::account_base& get_account(){return m_account;} + + bool init(const std::string& daemon_address = "http://localhost:8080", uint64_t upper_transaction_size_limit = CRYPTONOTE_BLOCK_GRANTED_FULL_REWARD_ZONE*2 - CRYPTONOTE_COINBASE_BLOB_RESERVED_SIZE); + + bool refresh(); + bool refresh(size_t & blocks_fetched); + bool refresh(size_t & blocks_fetched, bool& received_money); + bool deinit(); + + uint64_t balance(); + uint64_t unlocked_balance(); + void show_incoming_transfers(); + template<typename T> + bool transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee, T destination_split_strategy, const tx_dust_policy& dust_policy); + template<typename T> + bool transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee, T destination_split_strategy, const tx_dust_policy& dust_policy, cryptonote::transaction &tx, transafer_fail_details& tfd); + template<typename T> + bool transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee, T destination_split_strategy, const tx_dust_policy& dust_policy, cryptonote::transaction &tx); + bool transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee); + bool transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee, cryptonote::transaction& tx); + bool transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, uint64_t unlock_time, uint64_t fee, cryptonote::transaction& tx, transafer_fail_details& tfd); + bool check_connection(); + bool get_transfers(wallet2::transfer_container& incoming_transfers); + 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; + } + + private: + bool store_keys(const std::string& keys_file_name, const std::string& password); + bool load_keys(const std::string& keys_file_name, const std::string& password); + bool process_new_transaction(cryptonote::transaction& tx, uint64_t height); + bool process_new_blockchain_entry(cryptonote::block& b, cryptonote::block_complete_entry& bche, crypto::hash& bl_id, uint64_t height); + bool detach_blockchain(uint64_t height); + bool 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(); + bool pull_blocks(size_t& blocks_added); + uint64_t select_transfers(uint64_t needed_money, uint64_t dust, std::list<transfer_container::iterator>& selected_transfers); + bool prepare_file_names(const std::string& file_path); + + 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 + + transfer_container m_transfers; + 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 + }; +} +BOOST_CLASS_VERSION(tools::wallet2, 5) + +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; + } + } +} + +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) + " "; }); + std::cout << "amount=" << cryptonote::print_money(src.amount) << ", real_output=" <<src.real_output << ", real_output_in_tx_index=" << src.real_output_in_tx_index << ", indexes: " << indexes << ENDL; + } + //---------------------------------------------------------------------------------------------------- + } + //---------------------------------------------------------------------------------------------------- + template<typename T> + bool wallet2::transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, + uint64_t unlock_time, uint64_t fee, T destination_split_strategy, const tx_dust_policy& dust_policy) + { + cryptonote::transaction tx; + return transfer(dsts, fake_outputs_count, unlock_time, fee, destination_split_strategy, dust_policy, tx); + } + + template<typename T> + bool wallet2::transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, + uint64_t unlock_time, uint64_t fee, T destination_split_strategy, const tx_dust_policy& dust_policy, cryptonote::transaction &tx) + { + transafer_fail_details stub = AUTO_VAL_INIT(stub); + return transfer(dsts, fake_outputs_count, unlock_time, fee, destination_split_strategy, dust_policy, tx, stub); + } + + template<typename T> + bool wallet2::transfer(const std::vector<cryptonote::tx_destination_entry>& dsts, size_t fake_outputs_count, + uint64_t unlock_time, uint64_t fee, T destination_split_strategy, const tx_dust_policy& dust_policy, cryptonote::transaction &tx, transafer_fail_details& tfd) + { + using namespace cryptonote; + + uint64_t needed_money = fee; + BOOST_FOREACH(auto& dt, dsts) + { + CHECK_AND_ASSERT_MES(dt.amount > 0, false, "Wrong destination amount value: " << dt.amount); + needed_money += dt.amount; + } + + std::list<transfer_container::iterator> selected_transfers; + uint64_t found_money = select_transfers(needed_money, dust_policy.dust_threshold, selected_transfers); + + if(found_money < needed_money) + { + LOG_ERROR("not enough money, available only " << print_money(found_money) << ", expected " << print_money(needed_money) ); + tfd.reason = transafer_fail_details::error_not_enough_money; + return false; + } + //typedef COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::outs_for_amount outs_for_amount; + 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) + { + CHECK_AND_ASSERT_MES(it->m_tx.vout.size() > it->m_internal_output_index, false, "internal error: m_internal_output_index = " + << it->m_internal_output_index << " more than " << it->m_tx.vout.size()); + req.amounts.push_back(it->amount()); + } + bool r = net_utils::invoke_http_bin_remote_command2(m_daemon_address + "/getrandom_outs.bin", req, daemon_resp, m_http_client, 200000); + tfd.reason = transafer_fail_details::error_not_connected; + CHECK_AND_ASSERT_MES(r, false, "failed to get getrandom_outs"); + tfd.reason = transafer_fail_details::error_internal_error; + CHECK_AND_ASSERT_MES(daemon_resp.status == CORE_RPC_STATUS_OK, false, "failed to getrandom_outs.bin"); + CHECK_AND_ASSERT_MES(daemon_resp.outs.size() == selected_transfers.size(), false, "internal error: daemon returned wrong response for getrandom_outs, wrong amounts count = " + << daemon_resp.outs.size() << ", expected " << selected_transfers.size()); + } + tfd.reason = transafer_fail_details::error_ok; + + //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); + CHECK_AND_ASSERT_MES(dust <= dust_policy.dust_threshold, false, "internal error: invalid dust value"); + if (0 != dust && !dust_policy.add_to_fee) + { + splitted_dsts.push_back(cryptonote::tx_destination_entry(dust, dust_policy.addr_for_dust)); + } + + tfd.reason = transafer_fail_details::error_internal_error; + bool r = cryptonote::construct_tx(m_account.get_keys(), sources, splitted_dsts, tx, unlock_time); + CHECK_AND_ASSERT_MES(r, false, "Transaction construction failed"); + + //check transaction size + if(get_object_blobsize(tx) >= m_upper_transaction_size_limit) + { + LOG_PRINT_RED("Transaction size is too big: " << get_object_blobsize(tx) << ", expected size < " << m_upper_transaction_size_limit, LOG_LEVEL_2); + tfd.reason = transafer_fail_details::error_too_big_transaction; + tfd.tx_blob_size = get_object_blobsize(tx); + tfd.max_expected_tx_blob_size = m_upper_transaction_size_limit; + return false; + } + + + COMMAND_RPC_SEND_RAW_TX::request req; + req.tx_as_hex = epee::string_tools::buff_to_hex_nodelimer(tx_to_blob(tx)); + COMMAND_RPC_SEND_RAW_TX::response daemon_send_resp; + tfd.reason = transafer_fail_details::error_not_connected; + r = net_utils::invoke_http_json_remote_command2(m_daemon_address + "/sendrawtransaction", req, daemon_send_resp, m_http_client, 200000); + CHECK_AND_ASSERT_MES(r, false, "failed to send transaction"); + if(daemon_send_resp.status != CORE_RPC_STATUS_OK) + { + tfd.reason = transafer_fail_details::error_rejected_by_daemon; + LOG_ERROR("daemon failed to accept generated transaction, id: " << get_transaction_hash(tx) ); + return false; + } + std::string key_images; + std::for_each(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; + }); + LOG_PRINT_L2("transaction " << get_transaction_hash(tx) << " generated ok and sent to daemon, key_images: [" << key_images << "]"); + + BOOST_FOREACH(transfer_container::iterator it, selected_transfers) + it->m_spent = true; + + LOG_PRINT_L0("Transaction successfully sent. <" << get_transaction_hash(tx) << ">" << ENDL + << "Commission: " << print_money(fee+dust) << "(dust: " << print_money(dust) << ")" << ENDL + << "Balance: " << print_money(balance()) << ENDL + << "Unlocked: " << print_money(unlocked_balance()) << ENDL + << "Please, wait for confirmation for your balance to be unlocked."); + + tfd.reason = transafer_fail_details::error_ok; + return true; + } +} |