mirror of
https://git.wownero.com/wownero/wownero.git
synced 2024-12-28 00:38:53 +00:00
453 lines
12 KiB
C++
453 lines
12 KiB
C++
// 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"
|
|
#include "cryptonote_core/cryptonote_format_utils.h"
|
|
#include "cryptonote_core/cryptonote_basic_impl.h"
|
|
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
namespace epee {
|
|
unsigned int g_test_dbg_lock_sleep = 0;
|
|
}
|
|
|
|
namespace Bitmonero {
|
|
|
|
struct WalletManagerImpl;
|
|
|
|
namespace {
|
|
static WalletManagerImpl * g_walletManager = nullptr;
|
|
// copy-pasted from
|
|
static const size_t DEFAULT_MIX = 4;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
using namespace std;
|
|
|
|
Wallet::~Wallet() {}
|
|
|
|
///////////////////////// Wallet implementation ////////////////////////////////
|
|
class WalletImpl : public Wallet
|
|
{
|
|
public:
|
|
WalletImpl(bool testnet = false);
|
|
~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();
|
|
std::string seed() const;
|
|
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);
|
|
std::string address() const;
|
|
bool store(const std::string &path);
|
|
bool init(const std::string &daemon_address, uint64_t upper_transaction_size_limit);
|
|
bool connectToDaemon();
|
|
uint64_t balance() const;
|
|
uint64_t unlockedBalance() const;
|
|
std::string displayAmount(uint64_t amount) const;
|
|
bool refresh();
|
|
bool transfer(const std::string &dst_addr, uint64_t amount);
|
|
|
|
|
|
private:
|
|
void clearStatus();
|
|
|
|
private:
|
|
//std::unique_ptr<tools::wallet2> m_wallet;
|
|
tools::wallet2 * m_wallet;
|
|
int m_status;
|
|
std::string m_errorString;
|
|
std::string m_password;
|
|
};
|
|
|
|
WalletImpl::WalletImpl(bool testnet)
|
|
:m_wallet(nullptr), m_status(Wallet::Status_Ok)
|
|
{
|
|
m_wallet = new tools::wallet2(testnet);
|
|
}
|
|
|
|
WalletImpl::~WalletImpl()
|
|
{
|
|
delete m_wallet;
|
|
}
|
|
|
|
bool WalletImpl::create(const std::string &path, const std::string &password, const std::string &language)
|
|
{
|
|
|
|
clearStatus();
|
|
|
|
bool keys_file_exists;
|
|
bool wallet_file_exists;
|
|
tools::wallet2::wallet_exists(path, keys_file_exists, wallet_file_exists);
|
|
// TODO: figure out how to setup logger;
|
|
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;
|
|
return false;
|
|
}
|
|
// TODO: validate language
|
|
m_wallet->set_seed_language(language);
|
|
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();
|
|
return false;
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
std::string WalletImpl::seed() const
|
|
{
|
|
std::string seed;
|
|
if (m_wallet)
|
|
m_wallet->get_seed(seed);
|
|
return seed;
|
|
}
|
|
|
|
std::string WalletImpl::getSeedLanguage() const
|
|
{
|
|
return m_wallet->get_seed_language();
|
|
}
|
|
|
|
void WalletImpl::setSeedLanguage(const std::string &arg)
|
|
{
|
|
m_wallet->set_seed_language(arg);
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
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();
|
|
}
|
|
|
|
std::string WalletImpl::displayAmount(uint64_t amount) const
|
|
{
|
|
return cryptonote::print_money(amount);
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
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;
|
|
size_t fake_outs_count = DEFAULT_MIX;
|
|
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);
|
|
|
|
|
|
|
|
|
|
return m_status == Status_Ok;
|
|
}
|
|
|
|
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();
|
|
}
|
|
|
|
|
|
|
|
///////////////////////// WalletManager implementation /////////////////////////
|
|
class WalletManagerImpl : public WalletManager
|
|
{
|
|
public:
|
|
Wallet * createWallet(const std::string &path, const std::string &password,
|
|
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);
|
|
bool walletExists(const std::string &path);
|
|
std::string errorString() const;
|
|
void setDaemonHost(const std::string &hostname);
|
|
|
|
|
|
private:
|
|
WalletManagerImpl() {}
|
|
friend struct WalletManagerFactory;
|
|
|
|
std::string m_errorString;
|
|
};
|
|
|
|
Wallet *WalletManagerImpl::createWallet(const std::string &path, const std::string &password,
|
|
const std::string &language, bool testnet)
|
|
{
|
|
WalletImpl * wallet = new WalletImpl(testnet);
|
|
wallet->create(path, password, language);
|
|
return wallet;
|
|
}
|
|
|
|
Wallet *WalletManagerImpl::openWallet(const std::string &path, const std::string &password, bool testnet)
|
|
{
|
|
WalletImpl * wallet = new WalletImpl(testnet);
|
|
wallet->open(path, password);
|
|
return wallet;
|
|
}
|
|
|
|
Wallet *WalletManagerImpl::recoveryWallet(const std::string &path, const std::string &memo, bool testnet)
|
|
{
|
|
WalletImpl * wallet = new WalletImpl(testnet);
|
|
wallet->recover(path, memo);
|
|
return wallet;
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
bool WalletManagerImpl::walletExists(const std::string &path)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
std::string WalletManagerImpl::errorString() const
|
|
{
|
|
return m_errorString;
|
|
}
|
|
|
|
void WalletManagerImpl::setDaemonHost(const std::string &hostname)
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////// WalletManagerFactory implementation //////////////////////
|
|
WalletManager *WalletManagerFactory::getWalletManager()
|
|
{
|
|
|
|
if (!g_walletManager) {
|
|
epee::log_space::log_singletone::add_logger(LOGGER_CONSOLE, NULL, NULL, LOG_LEVEL_0);
|
|
g_walletManager = new WalletManagerImpl();
|
|
}
|
|
|
|
return g_walletManager;
|
|
}
|
|
|
|
|
|
}
|