danicoin/src/wallet/wallet2_api.cpp

594 lines
17 KiB
C++
Raw Normal View History

2016-02-15 18:04:00 +00:00
// Copyright (c) 2014-2016, The Monero Project
//
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without modification, are
// permitted provided that the following conditions are met:
//
// 1. Redistributions of source code must retain the above copyright notice, this list of
// conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright notice, this list
// of conditions and the following disclaimer in the documentation and/or other
// materials provided with the distribution.
//
// 3. Neither the name of the copyright holder nor the names of its contributors may be
// used to endorse or promote products derived from this software without specific
// prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
// THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
// THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
#include "wallet2_api.h"
#include "wallet2.h"
#include "mnemonics/electrum-words.h"
2016-03-31 12:39:28 +00:00
#include "cryptonote_core/cryptonote_format_utils.h"
2016-04-03 11:34:38 +00:00
#include "cryptonote_core/cryptonote_basic_impl.h"
2016-04-04 11:56:17 +00:00
#include "cryptonote_core/cryptonote_format_utils.h"
2016-04-03 11:34:38 +00:00
2016-02-15 18:04:00 +00:00
#include <memory>
2016-04-03 11:34:38 +00:00
#include <vector>
2016-04-04 11:56:17 +00:00
#include <sstream>
#include <boost/format.hpp>
#define tr(x) (x)
2016-02-15 18:04:00 +00:00
namespace epee {
unsigned int g_test_dbg_lock_sleep = 0;
}
2016-02-20 16:04:56 +00:00
2016-02-15 18:04:00 +00:00
namespace Bitmonero {
struct WalletManagerImpl;
namespace {
static WalletManagerImpl * g_walletManager = nullptr;
2016-04-03 11:34:38 +00:00
// copy-pasted from
static const size_t DEFAULT_MIX = 4;
2016-02-20 16:04:56 +00:00
2016-02-15 18:04:00 +00:00
}
2016-04-03 11:34:38 +00:00
2016-04-04 11:56:17 +00:00
2016-04-03 11:34:38 +00:00
using namespace std;
2016-04-04 11:56:17 +00:00
using namespace cryptonote;
2016-04-03 11:34:38 +00:00
2016-02-21 18:18:16 +00:00
Wallet::~Wallet() {}
2016-02-15 18:04:00 +00:00
2016-04-04 11:56:17 +00:00
///////////////////////// Transaction implementation ///////////////////////////
class TransactionImpl : public Transaction
{
public:
TransactionImpl(Wallet * wallet);
~TransactionImpl();
int status() const;
std::string errorString() const;
bool commit();
private:
std::vector<tools::wallet2::pending_tx> & transactions();
private:
friend class WalletImpl;
Wallet * m_wallet;
int m_status;
std::string m_errorString;
std::vector<tools::wallet2::pending_tx> m_pending_tx;
};
TransactionImpl::TransactionImpl(Wallet *wallet)
: m_wallet(wallet)
{
}
TransactionImpl::~TransactionImpl()
{
}
int TransactionImpl::status() const
{
return m_status;
}
string TransactionImpl::errorString() const
{
return m_errorString;
}
bool TransactionImpl::commit()
{
// while (!m_pending_tx.empty()) {
// }
return false;
}
2016-02-15 18:04:00 +00:00
///////////////////////// Wallet implementation ////////////////////////////////
class WalletImpl : public Wallet
{
public:
2016-03-25 14:06:30 +00:00
WalletImpl(bool testnet = false);
2016-02-15 18:04:00 +00:00
~WalletImpl();
bool create(const std::string &path, const std::string &password,
const std::string &language);
bool open(const std::string &path, const std::string &password);
bool recover(const std::string &path, const std::string &seed);
bool close();
2016-02-15 18:04:00 +00:00
std::string seed() const;
2016-02-21 18:18:16 +00:00
std::string getSeedLanguage() const;
void setSeedLanguage(const std::string &arg);
void setListener(Listener *) {}
int status() const;
std::string errorString() const;
bool setPassword(const std::string &password);
2016-03-12 14:52:58 +00:00
std::string address() const;
bool store(const std::string &path);
2016-03-25 14:06:30 +00:00
bool init(const std::string &daemon_address, uint64_t upper_transaction_size_limit);
2016-03-31 12:39:28 +00:00
bool connectToDaemon();
2016-03-25 14:06:30 +00:00
uint64_t balance() const;
uint64_t unlockedBalance() const;
2016-03-31 12:39:28 +00:00
std::string displayAmount(uint64_t amount) const;
2016-03-31 13:38:57 +00:00
bool refresh();
2016-04-03 11:34:38 +00:00
bool transfer(const std::string &dst_addr, uint64_t amount);
2016-03-31 12:39:28 +00:00
private:
void clearStatus();
2016-02-15 18:04:00 +00:00
private:
2016-02-21 18:18:16 +00:00
//std::unique_ptr<tools::wallet2> m_wallet;
tools::wallet2 * m_wallet;
int m_status;
std::string m_errorString;
std::string m_password;
2016-02-15 18:04:00 +00:00
};
2016-03-25 14:06:30 +00:00
WalletImpl::WalletImpl(bool testnet)
:m_wallet(nullptr), m_status(Wallet::Status_Ok)
2016-02-15 18:04:00 +00:00
{
2016-03-25 14:06:30 +00:00
m_wallet = new tools::wallet2(testnet);
2016-02-15 18:04:00 +00:00
}
WalletImpl::~WalletImpl()
{
2016-02-21 18:18:16 +00:00
delete m_wallet;
2016-02-15 18:04:00 +00:00
}
bool WalletImpl::create(const std::string &path, const std::string &password, const std::string &language)
{
clearStatus();
2016-02-15 18:04:00 +00:00
bool keys_file_exists;
bool wallet_file_exists;
tools::wallet2::wallet_exists(path, keys_file_exists, wallet_file_exists);
2016-02-21 18:18:16 +00:00
// TODO: figure out how to setup logger;
2016-02-15 18:04:00 +00:00
LOG_PRINT_L3("wallet_path: " << path << "");
LOG_PRINT_L3("keys_file_exists: " << std::boolalpha << keys_file_exists << std::noboolalpha
<< " wallet_file_exists: " << std::boolalpha << wallet_file_exists << std::noboolalpha);
// add logic to error out if new wallet requested but named wallet file exists
if (keys_file_exists || wallet_file_exists) {
m_errorString = "attempting to generate or restore wallet, but specified file(s) exist. Exiting to not risk overwriting.";
LOG_ERROR(m_errorString);
m_status = Status_Error;
2016-02-15 18:04:00 +00:00
return false;
}
// TODO: validate language
m_wallet->set_seed_language(language);
2016-02-15 18:04:00 +00:00
crypto::secret_key recovery_val, secret_key;
try {
recovery_val = m_wallet->generate(path, password, secret_key, false, false);
m_password = password;
m_status = Status_Ok;
} catch (const std::exception &e) {
LOG_ERROR("Error creating wallet: " << e.what());
m_status = Status_Error;
m_errorString = e.what();
2016-02-15 18:04:00 +00:00
return false;
}
2016-02-15 18:04:00 +00:00
return true;
}
bool WalletImpl::open(const std::string &path, const std::string &password)
{
clearStatus();
try {
// TODO: handle "deprecated"
m_wallet->load(path, password);
m_password = password;
} catch (const std::exception &e) {
LOG_ERROR("Error opening wallet: " << e.what());
m_status = Status_Error;
m_errorString = e.what();
}
return m_status == Status_Ok;
}
bool WalletImpl::recover(const std::string &path, const std::string &seed)
{
clearStatus();
m_errorString.clear();
if (seed.empty()) {
m_errorString = "Electrum seed is empty";
LOG_ERROR(m_errorString);
m_status = Status_Error;
return false;
}
crypto::secret_key recovery_key;
std::string old_language;
if (!crypto::ElectrumWords::words_to_bytes(seed, recovery_key, old_language)) {
m_errorString = "Electrum-style word list failed verification";
m_status = Status_Error;
return false;
}
try {
m_wallet->set_seed_language(old_language);
m_wallet->generate(path, "", recovery_key, true, false);
// TODO: wallet->init(daemon_address);
} catch (const std::exception &e) {
m_status = Status_Error;
m_errorString = e.what();
}
return m_status == Status_Ok;
}
bool WalletImpl::close()
{
clearStatus();
bool result = false;
try {
m_wallet->store();
m_wallet->stop();
result = true;
} catch (const std::exception &e) {
m_status = Status_Error;
m_errorString = e.what();
LOG_ERROR("Error closing wallet: " << e.what());
}
return result;
}
2016-02-20 16:04:56 +00:00
std::string WalletImpl::seed() const
{
2016-02-21 18:18:16 +00:00
std::string seed;
if (m_wallet)
m_wallet->get_seed(seed);
return seed;
}
std::string WalletImpl::getSeedLanguage() const
{
return m_wallet->get_seed_language();
2016-02-20 16:04:56 +00:00
}
2016-02-21 18:18:16 +00:00
void WalletImpl::setSeedLanguage(const std::string &arg)
{
m_wallet->set_seed_language(arg);
}
2016-02-15 18:04:00 +00:00
int WalletImpl::status() const
{
return m_status;
}
std::string WalletImpl::errorString() const
{
return m_errorString;
}
bool WalletImpl::setPassword(const std::string &password)
{
clearStatus();
try {
m_wallet->rewrite(m_wallet->get_wallet_file(), password);
m_password = password;
} catch (const std::exception &e) {
m_status = Status_Error;
m_errorString = e.what();
}
return m_status == Status_Ok;
}
2016-03-12 14:52:58 +00:00
std::string WalletImpl::address() const
{
return m_wallet->get_account().get_public_address_str(m_wallet->testnet());
}
bool WalletImpl::store(const std::string &path)
{
clearStatus();
try {
if (path.empty()) {
m_wallet->store();
} else {
m_wallet->store_to(path, m_password);
}
} catch (const std::exception &e) {
LOG_ERROR("Error storing wallet: " << e.what());
m_status = Status_Error;
m_errorString = e.what();
}
return m_status == Status_Ok;
}
2016-03-25 14:06:30 +00:00
bool WalletImpl::init(const std::string &daemon_address, uint64_t upper_transaction_size_limit)
{
clearStatus();
try {
m_wallet->init(daemon_address, upper_transaction_size_limit);
} catch (const std::exception &e) {
LOG_ERROR("Error initializing wallet: " << e.what());
m_status = Status_Error;
m_errorString = e.what();
}
return m_status == Status_Ok;
}
uint64_t WalletImpl::balance() const
{
return m_wallet->balance();
}
uint64_t WalletImpl::unlockedBalance() const
{
return m_wallet->unlocked_balance();
}
2016-03-31 12:39:28 +00:00
std::string WalletImpl::displayAmount(uint64_t amount) const
{
return cryptonote::print_money(amount);
}
2016-03-31 13:38:57 +00:00
bool WalletImpl::refresh()
{
clearStatus();
try {
m_wallet->refresh();
} catch (const std::exception &e) {
m_status = Status_Error;
m_errorString = e.what();
}
return m_status == Status_Ok;
}
2016-04-03 11:34:38 +00:00
bool WalletImpl::transfer(const std::string &dst_addr, uint64_t amount)
{
clearStatus();
vector<cryptonote::tx_destination_entry> dsts;
cryptonote::tx_destination_entry de;
bool has_payment_id;
bool payment_id_seen = false;
crypto::hash8 new_payment_id;
2016-04-04 11:56:17 +00:00
size_t fake_outs_count = m_wallet->default_mixin();
if (fake_outs_count == 0)
fake_outs_count = DEFAULT_MIX;
2016-04-03 11:34:38 +00:00
if(!cryptonote::get_account_integrated_address_from_str(de.addr, has_payment_id, new_payment_id, m_wallet->testnet(), dst_addr)) {
// TODO: copy-paste 'if treating as an address fails, try as url' from simplewallet.cpp:1982
m_status = Status_Error;
m_errorString = "Invalid destination address";
return false;
}
de.amount = amount;
if (de.amount <= 0) {
m_status = Status_Error;
m_errorString = "Invalid amount";
return false;
}
dsts.push_back(de);
2016-04-04 11:56:17 +00:00
std::vector<tools::wallet2::pending_tx> ptx_vector;
std::vector<uint8_t> extra;
try {
ptx_vector = m_wallet->create_transactions(dsts, fake_outs_count, 0 /* unlock_time */, 0 /* unused fee arg*/, extra);
// TODO: move it to transaction class
while (!ptx_vector.empty()) {
auto & ptx = ptx_vector.back();
m_wallet->commit_tx(ptx);
// success_msg_writer(true) << tr("Money successfully sent, transaction ") << get_transaction_hash(ptx.tx);
// if no exception, remove element from vector
ptx_vector.pop_back();
} // TODO: extract method;
} catch (const tools::error::daemon_busy&) {
// TODO: make it translatable with "tr"?
m_errorString = tr("daemon is busy. Please try again later.");
m_status = Status_Error;
} catch (const tools::error::no_connection_to_daemon&) {
m_errorString = tr("no connection to daemon. Please make sure daemon is running.");
m_status = Status_Error;
} catch (const tools::error::wallet_rpc_error& e) {
m_errorString = tr("RPC error: ") + e.to_string();
m_status = Status_Error;
} catch (const tools::error::get_random_outs_error&) {
m_errorString = tr("failed to get random outputs to mix");
m_status = Status_Error;
2016-04-03 11:34:38 +00:00
2016-04-04 11:56:17 +00:00
} catch (const tools::error::not_enough_money& e) {
m_status = Status_Error;
std::ostringstream writer(m_errorString);
writer << boost::format(tr("not enough money to transfer, available only %s, transaction amount %s = %s + %s (fee)")) %
print_money(e.available()) %
print_money(e.tx_amount() + e.fee()) %
print_money(e.tx_amount()) %
print_money(e.fee());
} catch (const tools::error::not_enough_outs_to_mix& e) {
std::ostringstream writer(m_errorString);
writer << tr("not enough outputs for specified mixin_count") << " = " << e.mixin_count() << ":";
for (const cryptonote::COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::outs_for_amount& outs_for_amount : e.scanty_outs()) {
writer << "\n" << tr("output amount") << " = " << print_money(outs_for_amount.amount) << ", " << tr("found outputs to mix") << " = " << outs_for_amount.outs.size();
}
m_status = Status_Error;
} catch (const tools::error::tx_not_constructed&) {
m_errorString = tr("transaction was not constructed");
m_status = Status_Error;
} catch (const tools::error::tx_rejected& e) {
std::ostringstream writer(m_errorString);
writer << (boost::format(tr("transaction %s was rejected by daemon with status: ")) % get_transaction_hash(e.tx())) << e.status();
m_status = Status_Error;
} catch (const tools::error::tx_sum_overflow& e) {
m_errorString = e.what();
m_status = Status_Error;
} catch (const tools::error::zero_destination&) {
m_errorString = tr("one of destinations is zero");
m_status = Status_Error;
} catch (const tools::error::tx_too_big& e) {
m_errorString = tr("failed to find a suitable way to split transactions");
m_status = Status_Error;
} catch (const tools::error::transfer_error& e) {
m_errorString = string(tr("unknown transfer error: ")) + e.what();
m_status = Status_Error;
} catch (const tools::error::wallet_internal_error& e) {
m_errorString = string(tr("internal error: ")) + e.what();
m_status = Status_Error;
} catch (const std::exception& e) {
m_errorString = string(tr("unexpected error: ")) + e.what();
m_status = Status_Error;
} catch (...) {
m_errorString = tr("unknown error");
m_status = Status_Error;
}
2016-04-03 11:34:38 +00:00
return m_status == Status_Ok;
}
2016-03-25 14:06:30 +00:00
bool WalletImpl::connectToDaemon()
{
bool result = m_wallet->check_connection();
m_status = result ? Status_Ok : Status_Error;
if (!result) {
m_errorString = "Error connecting to daemon at " + m_wallet->get_daemon_address();
}
return result;
}
void WalletImpl::clearStatus()
{
m_status = Status_Ok;
m_errorString.clear();
}
2016-02-15 18:04:00 +00:00
///////////////////////// WalletManager implementation /////////////////////////
class WalletManagerImpl : public WalletManager
{
public:
Wallet * createWallet(const std::string &path, const std::string &password,
2016-03-25 14:06:30 +00:00
const std::string &language, bool testnet);
Wallet * openWallet(const std::string &path, const std::string &password, bool testnet);
virtual Wallet * recoveryWallet(const std::string &path, const std::string &memo, bool testnet);
virtual bool closeWallet(Wallet *wallet);
2016-02-20 16:04:56 +00:00
bool walletExists(const std::string &path);
std::string errorString() const;
void setDaemonHost(const std::string &hostname);
2016-02-15 18:04:00 +00:00
private:
WalletManagerImpl() {}
friend struct WalletManagerFactory;
std::string m_errorString;
2016-02-15 18:04:00 +00:00
};
2016-02-20 16:04:56 +00:00
Wallet *WalletManagerImpl::createWallet(const std::string &path, const std::string &password,
2016-03-25 14:06:30 +00:00
const std::string &language, bool testnet)
2016-02-15 18:04:00 +00:00
{
2016-03-25 14:06:30 +00:00
WalletImpl * wallet = new WalletImpl(testnet);
wallet->create(path, password, language);
2016-02-15 18:04:00 +00:00
return wallet;
}
2016-03-25 14:06:30 +00:00
Wallet *WalletManagerImpl::openWallet(const std::string &path, const std::string &password, bool testnet)
{
2016-03-25 14:06:30 +00:00
WalletImpl * wallet = new WalletImpl(testnet);
wallet->open(path, password);
return wallet;
}
Wallet *WalletManagerImpl::recoveryWallet(const std::string &path, const std::string &memo, bool testnet)
2016-02-20 16:04:56 +00:00
{
WalletImpl * wallet = new WalletImpl(testnet);
wallet->recover(path, memo);
return wallet;
2016-02-20 16:04:56 +00:00
}
bool WalletManagerImpl::closeWallet(Wallet *wallet)
{
WalletImpl * wallet_ = dynamic_cast<WalletImpl*>(wallet);
bool result = wallet_->close();
if (!result) {
m_errorString = wallet_->errorString();
} else {
delete wallet_;
}
return result;
}
2016-02-20 16:04:56 +00:00
bool WalletManagerImpl::walletExists(const std::string &path)
{
return false;
}
std::string WalletManagerImpl::errorString() const
2016-02-20 16:04:56 +00:00
{
return m_errorString;
2016-02-20 16:04:56 +00:00
}
void WalletManagerImpl::setDaemonHost(const std::string &hostname)
{
}
2016-02-20 16:04:56 +00:00
2016-02-15 18:04:00 +00:00
///////////////////// WalletManagerFactory implementation //////////////////////
WalletManager *WalletManagerFactory::getWalletManager()
{
2016-02-15 18:04:00 +00:00
if (!g_walletManager) {
epee::log_space::log_singletone::add_logger(LOGGER_CONSOLE, NULL, NULL, LOG_LEVEL_0);
2016-02-15 18:04:00 +00:00
g_walletManager = new WalletManagerImpl();
}
return g_walletManager;
}
2016-03-31 12:39:28 +00:00
2016-04-04 11:56:17 +00:00
2016-02-15 18:04:00 +00:00
}