mirror of
https://git.wownero.com/wownero/wownero.git
synced 2024-08-15 01:03:23 +00:00
Refactored password prompting for wallets
This commit is contained in:
parent
5fcf7c1c18
commit
2bddb8ebee
7 changed files with 225 additions and 375 deletions
|
@ -335,7 +335,6 @@ bool simple_wallet::seed(const std::vector<std::string> &args/* = std::vector<st
|
||||||
|
|
||||||
bool simple_wallet::seed_set_language(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
bool simple_wallet::seed_set_language(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
||||||
{
|
{
|
||||||
bool success = false;
|
|
||||||
if (m_wallet->watch_only())
|
if (m_wallet->watch_only())
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("wallet is watch-only and has no seed");
|
fail_msg_writer() << tr("wallet is watch-only and has no seed");
|
||||||
|
@ -347,87 +346,49 @@ bool simple_wallet::seed_set_language(const std::vector<std::string> &args/* = s
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
tools::password_container pwd_container(m_wallet_file.empty());
|
const auto pwd_container = get_and_verify_password();
|
||||||
success = pwd_container.read_password();
|
if (pwd_container)
|
||||||
if (!success)
|
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* verify password before using so user doesn't accidentally set a new password for rewritten wallet */
|
|
||||||
success = m_wallet->verify_password(pwd_container.password());
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("invalid password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string mnemonic_language = get_mnemonic_language();
|
std::string mnemonic_language = get_mnemonic_language();
|
||||||
if (mnemonic_language.empty())
|
if (mnemonic_language.empty())
|
||||||
return true;
|
return true;
|
||||||
m_wallet->set_seed_language(mnemonic_language);
|
|
||||||
m_wallet->rewrite(m_wallet_file, pwd_container.password());
|
m_wallet->set_seed_language(std::move(mnemonic_language));
|
||||||
|
m_wallet->rewrite(m_wallet_file, pwd_container->password());
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool simple_wallet::set_always_confirm_transfers(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
bool simple_wallet::set_always_confirm_transfers(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
||||||
{
|
{
|
||||||
bool success = false;
|
const auto pwd_container = get_and_verify_password();
|
||||||
tools::password_container pwd_container(m_wallet_file.empty());
|
if (pwd_container)
|
||||||
success = pwd_container.read_password();
|
|
||||||
if (!success)
|
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* verify password before using so user doesn't accidentally set a new password for rewritten wallet */
|
|
||||||
success = m_wallet->verify_password(pwd_container.password());
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("invalid password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
m_wallet->always_confirm_transfers(is_it_true(args[1]));
|
m_wallet->always_confirm_transfers(is_it_true(args[1]));
|
||||||
m_wallet->rewrite(m_wallet_file, pwd_container.password());
|
m_wallet->rewrite(m_wallet_file, pwd_container->password());
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool simple_wallet::set_store_tx_info(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
bool simple_wallet::set_store_tx_info(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
||||||
{
|
{
|
||||||
bool success = false;
|
|
||||||
if (m_wallet->watch_only())
|
if (m_wallet->watch_only())
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("wallet is watch-only and cannot transfer");
|
fail_msg_writer() << tr("wallet is watch-only and cannot transfer");
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
tools::password_container pwd_container(m_wallet_file.empty());
|
const auto pwd_container = get_and_verify_password();
|
||||||
success = pwd_container.read_password();
|
if (pwd_container)
|
||||||
if (!success)
|
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* verify password before using so user doesn't accidentally set a new password for rewritten wallet */
|
|
||||||
success = m_wallet->verify_password(pwd_container.password());
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("invalid password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
m_wallet->store_tx_info(is_it_true(args[1]));
|
m_wallet->store_tx_info(is_it_true(args[1]));
|
||||||
m_wallet->rewrite(m_wallet_file, pwd_container.password());
|
m_wallet->rewrite(m_wallet_file, pwd_container->password());
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool simple_wallet::set_default_mixin(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
bool simple_wallet::set_default_mixin(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
||||||
{
|
{
|
||||||
bool success = false;
|
|
||||||
if (m_wallet->watch_only())
|
if (m_wallet->watch_only())
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("wallet is watch-only and cannot transfer");
|
fail_msg_writer() << tr("wallet is watch-only and cannot transfer");
|
||||||
|
@ -449,25 +410,12 @@ bool simple_wallet::set_default_mixin(const std::vector<std::string> &args/* = s
|
||||||
if (mixin == 0)
|
if (mixin == 0)
|
||||||
mixin = DEFAULT_MIX;
|
mixin = DEFAULT_MIX;
|
||||||
|
|
||||||
tools::password_container pwd_container(m_wallet_file.empty());
|
const auto pwd_container = get_and_verify_password();
|
||||||
|
if (pwd_container)
|
||||||
success = pwd_container.read_password();
|
|
||||||
if (!success)
|
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* verify password before using so user doesn't accidentally set a new password for rewritten wallet */
|
|
||||||
success = m_wallet->verify_password(pwd_container.password());
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("invalid password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
m_wallet->default_mixin(mixin);
|
m_wallet->default_mixin(mixin);
|
||||||
m_wallet->rewrite(m_wallet_file, pwd_container.password());
|
m_wallet->rewrite(m_wallet_file, pwd_container->password());
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
catch(const boost::bad_lexical_cast &)
|
catch(const boost::bad_lexical_cast &)
|
||||||
|
@ -484,7 +432,6 @@ bool simple_wallet::set_default_mixin(const std::vector<std::string> &args/* = s
|
||||||
|
|
||||||
bool simple_wallet::set_default_priority(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
bool simple_wallet::set_default_priority(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
||||||
{
|
{
|
||||||
bool success = false;
|
|
||||||
int priority = 0;
|
int priority = 0;
|
||||||
if (m_wallet->watch_only())
|
if (m_wallet->watch_only())
|
||||||
{
|
{
|
||||||
|
@ -512,24 +459,12 @@ bool simple_wallet::set_default_priority(const std::vector<std::string> &args/*
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
tools::password_container pwd_container(m_wallet_file.empty());
|
const auto pwd_container = get_and_verify_password();
|
||||||
success = pwd_container.read_password();
|
if (pwd_container)
|
||||||
if (!success)
|
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* verify password before using so user doesn't accidentally set a new password for rewritten wallet */
|
|
||||||
success = m_wallet->verify_password(pwd_container.password());
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("invalid password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
m_wallet->set_default_priority(priority);
|
m_wallet->set_default_priority(priority);
|
||||||
m_wallet->rewrite(m_wallet_file, pwd_container.password());
|
m_wallet->rewrite(m_wallet_file, pwd_container->password());
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
catch(const boost::bad_lexical_cast &)
|
catch(const boost::bad_lexical_cast &)
|
||||||
|
@ -546,93 +481,52 @@ bool simple_wallet::set_default_priority(const std::vector<std::string> &args/*
|
||||||
|
|
||||||
bool simple_wallet::set_auto_refresh(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
bool simple_wallet::set_auto_refresh(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
||||||
{
|
{
|
||||||
|
const auto pwd_container = get_and_verify_password();
|
||||||
tools::password_container pwd_container(m_wallet_file.empty());
|
if (pwd_container)
|
||||||
|
|
||||||
bool success = pwd_container.read_password();
|
|
||||||
if (!success)
|
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
const bool auto_refresh = is_it_true(args[1]);
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* verify password before using so user doesn't accidentally set a new password for rewritten wallet */
|
|
||||||
success = m_wallet->verify_password(pwd_container.password());
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("invalid password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool auto_refresh = is_it_true(args[1]);
|
|
||||||
m_wallet->auto_refresh(auto_refresh);
|
m_wallet->auto_refresh(auto_refresh);
|
||||||
m_idle_mutex.lock();
|
m_idle_mutex.lock();
|
||||||
m_auto_refresh_enabled.store(auto_refresh, std::memory_order_relaxed);
|
m_auto_refresh_enabled.store(auto_refresh, std::memory_order_relaxed);
|
||||||
m_idle_cond.notify_one();
|
m_idle_cond.notify_one();
|
||||||
m_idle_mutex.unlock();
|
m_idle_mutex.unlock();
|
||||||
|
|
||||||
m_wallet->rewrite(m_wallet_file, pwd_container.password());
|
m_wallet->rewrite(m_wallet_file, pwd_container->password());
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool simple_wallet::set_refresh_type(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
bool simple_wallet::set_refresh_type(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
||||||
{
|
{
|
||||||
bool success = false;
|
|
||||||
|
|
||||||
tools::wallet2::RefreshType refresh_type;
|
tools::wallet2::RefreshType refresh_type;
|
||||||
if (!parse_refresh_type(args[1], refresh_type))
|
if (!parse_refresh_type(args[1], refresh_type))
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
tools::password_container pwd_container(m_wallet_file.empty());
|
const auto pwd_container = get_and_verify_password();
|
||||||
success = pwd_container.read_password();
|
if (pwd_container)
|
||||||
if (!success)
|
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* verify password before using so user doesn't accidentally set a new password for rewritten wallet */
|
|
||||||
success = m_wallet->verify_password(pwd_container.password());
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("invalid password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
m_wallet->set_refresh_type(refresh_type);
|
m_wallet->set_refresh_type(refresh_type);
|
||||||
|
m_wallet->rewrite(m_wallet_file, pwd_container->password());
|
||||||
m_wallet->rewrite(m_wallet_file, pwd_container.password());
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool simple_wallet::set_confirm_missing_payment_id(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
bool simple_wallet::set_confirm_missing_payment_id(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
||||||
{
|
{
|
||||||
bool success = false;
|
|
||||||
if (m_wallet->watch_only())
|
if (m_wallet->watch_only())
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("wallet is watch-only and cannot transfer");
|
fail_msg_writer() << tr("wallet is watch-only and cannot transfer");
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
tools::password_container pwd_container(m_wallet_file.empty());
|
|
||||||
success = pwd_container.read_password();
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* verify password before using so user doesn't accidentally set a new password for rewritten wallet */
|
const auto pwd_container = get_and_verify_password();
|
||||||
success = m_wallet->verify_password(pwd_container.password());
|
if (pwd_container)
|
||||||
if (!success)
|
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("invalid password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
m_wallet->confirm_missing_payment_id(is_it_true(args[1]));
|
m_wallet->confirm_missing_payment_id(is_it_true(args[1]));
|
||||||
m_wallet->rewrite(m_wallet_file, pwd_container.password());
|
m_wallet->rewrite(m_wallet_file, pwd_container->password());
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1298,7 +1192,20 @@ std::string simple_wallet::get_mnemonic_language()
|
||||||
}
|
}
|
||||||
return language_list[language_number];
|
return language_list[language_number];
|
||||||
}
|
}
|
||||||
|
//----------------------------------------------------------------------------------------------------
|
||||||
|
boost::optional<tools::password_container> simple_wallet::get_and_verify_password() const
|
||||||
|
{
|
||||||
|
auto pwd_container = tools::wallet2::password_prompt(m_wallet_file.empty());
|
||||||
|
if (!pwd_container)
|
||||||
|
return boost::none;
|
||||||
|
|
||||||
|
if (!m_wallet->verify_password(pwd_container->password()))
|
||||||
|
{
|
||||||
|
fail_msg_writer() << tr("invalid password");
|
||||||
|
return boost::none;
|
||||||
|
}
|
||||||
|
return pwd_container;
|
||||||
|
}
|
||||||
//----------------------------------------------------------------------------------------------------
|
//----------------------------------------------------------------------------------------------------
|
||||||
bool simple_wallet::new_wallet(const boost::program_options::variables_map& vm,
|
bool simple_wallet::new_wallet(const boost::program_options::variables_map& vm,
|
||||||
const crypto::secret_key& recovery_key, bool recover, bool two_random, const std::string &old_language)
|
const crypto::secret_key& recovery_key, bool recover, bool two_random, const std::string &old_language)
|
||||||
|
@ -1532,29 +1439,15 @@ bool simple_wallet::save(const std::vector<std::string> &args)
|
||||||
//----------------------------------------------------------------------------------------------------
|
//----------------------------------------------------------------------------------------------------
|
||||||
bool simple_wallet::save_watch_only(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
bool simple_wallet::save_watch_only(const std::vector<std::string> &args/* = std::vector<std::string>()*/)
|
||||||
{
|
{
|
||||||
bool success = false;
|
const auto pwd_container = tools::password_container::prompt(true, tr("Password for new watch-only wallet"));
|
||||||
tools::password_container pwd_container(m_wallet_file.empty());
|
|
||||||
|
|
||||||
success = pwd_container.read_password(tr("Password for the new watch-only wallet"));
|
if (!pwd_container)
|
||||||
if (!success)
|
|
||||||
{
|
{
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
fail_msg_writer() << tr("failed to read wallet password");
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
std::string password = pwd_container.password();
|
|
||||||
success = pwd_container.read_password(tr("Enter new password again"));
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("failed to read wallet password");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
if (password != pwd_container.password())
|
|
||||||
{
|
|
||||||
fail_msg_writer() << tr("passwords do not match");
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
m_wallet->write_watch_only_wallet(m_wallet_file, pwd_container.password());
|
m_wallet->write_watch_only_wallet(m_wallet_file, pwd_container->password());
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -81,6 +81,9 @@ namespace cryptonote
|
||||||
|
|
||||||
void wallet_idle_thread();
|
void wallet_idle_thread();
|
||||||
|
|
||||||
|
//! \return Prompts user for password and verifies against local file. Logs on error and returns `none`
|
||||||
|
boost::optional<tools::password_container> get_and_verify_password() const;
|
||||||
|
|
||||||
bool new_wallet(const boost::program_options::variables_map& vm, const crypto::secret_key& recovery_key,
|
bool new_wallet(const boost::program_options::variables_map& vm, const crypto::secret_key& recovery_key,
|
||||||
bool recover, bool two_random, const std::string &old_language);
|
bool recover, bool two_random, const std::string &old_language);
|
||||||
bool new_wallet(const boost::program_options::variables_map& vm, const cryptonote::account_public_address& address,
|
bool new_wallet(const boost::program_options::variables_map& vm, const cryptonote::account_public_address& address,
|
||||||
|
|
|
@ -42,153 +42,17 @@
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
namespace tools
|
|
||||||
{
|
|
||||||
namespace
|
namespace
|
||||||
{
|
{
|
||||||
bool is_cin_tty();
|
|
||||||
}
|
|
||||||
// deleted via private member
|
|
||||||
password_container::password_container()
|
|
||||||
: m_empty(true),m_verify(true)
|
|
||||||
{
|
|
||||||
|
|
||||||
}
|
|
||||||
password_container::password_container(bool verify)
|
|
||||||
: m_empty(true),m_verify(verify)
|
|
||||||
{
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
password_container::password_container(std::string&& password)
|
|
||||||
: m_empty(false)
|
|
||||||
, m_password(std::move(password))
|
|
||||||
, m_verify(false)
|
|
||||||
{
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
password_container::password_container(password_container&& rhs)
|
|
||||||
: m_empty(std::move(rhs.m_empty))
|
|
||||||
, m_password(std::move(rhs.m_password))
|
|
||||||
, m_verify(std::move(rhs.m_verify))
|
|
||||||
{
|
|
||||||
}
|
|
||||||
password_container::~password_container()
|
|
||||||
{
|
|
||||||
clear();
|
|
||||||
}
|
|
||||||
|
|
||||||
void password_container::clear()
|
|
||||||
{
|
|
||||||
if (0 < m_password.capacity())
|
|
||||||
{
|
|
||||||
m_password.replace(0, m_password.capacity(), m_password.capacity(), '\0');
|
|
||||||
m_password.resize(0);
|
|
||||||
}
|
|
||||||
m_empty = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool password_container::read_password(const char *message)
|
|
||||||
{
|
|
||||||
clear();
|
|
||||||
|
|
||||||
bool r;
|
|
||||||
if (is_cin_tty())
|
|
||||||
{
|
|
||||||
r = read_from_tty_double_check(message);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
r = read_from_file();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (r)
|
|
||||||
{
|
|
||||||
m_empty = false;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
clear();
|
|
||||||
}
|
|
||||||
|
|
||||||
return r;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool password_container::read_from_file()
|
|
||||||
{
|
|
||||||
m_password.reserve(max_password_size);
|
|
||||||
for (size_t i = 0; i < max_password_size; ++i)
|
|
||||||
{
|
|
||||||
char ch = static_cast<char>(std::cin.get());
|
|
||||||
if (std::cin.eof() || ch == '\n' || ch == '\r')
|
|
||||||
{
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
else if (std::cin.fail())
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
m_password.push_back(ch);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool password_container::read_from_tty_double_check(const char *message) {
|
|
||||||
std::string pass1;
|
|
||||||
std::string pass2;
|
|
||||||
bool match=false;
|
|
||||||
bool doNotVerifyEntry=false;
|
|
||||||
if (m_verify){message = "Enter a password for your new wallet";}
|
|
||||||
do{
|
|
||||||
if (message)
|
|
||||||
std::cout << message <<": ";
|
|
||||||
if (!password_container::read_from_tty(pass1))
|
|
||||||
return false;
|
|
||||||
if (m_verify==true){//double check password;
|
|
||||||
std::cout << "Confirm Password: ";
|
|
||||||
if (!password_container::read_from_tty(pass2))
|
|
||||||
return false;
|
|
||||||
if(pass1!=pass2){ //new password entered did not match
|
|
||||||
|
|
||||||
std::cout << "Passwords do not match! Please try again." << std::endl;
|
|
||||||
pass1="";
|
|
||||||
pass2="";
|
|
||||||
match=false;
|
|
||||||
}
|
|
||||||
else{//new password matches
|
|
||||||
match=true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
doNotVerifyEntry=true; //do not verify
|
|
||||||
//No need to verify password entered at this point in the code
|
|
||||||
|
|
||||||
}while(match==false && doNotVerifyEntry==false);
|
|
||||||
|
|
||||||
m_password=pass1;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
#if defined(_WIN32)
|
#if defined(_WIN32)
|
||||||
|
bool is_cin_tty() noexcept
|
||||||
namespace
|
|
||||||
{
|
|
||||||
bool is_cin_tty()
|
|
||||||
{
|
{
|
||||||
return 0 != _isatty(_fileno(stdin));
|
return 0 != _isatty(_fileno(stdin));
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
bool password_container::read_from_tty(std::string & pass)
|
bool read_from_tty(std::string& pass)
|
||||||
{
|
{
|
||||||
const char BACKSPACE = 8;
|
static constexpr const char BACKSPACE = 8;
|
||||||
|
|
||||||
HANDLE h_cin = ::GetStdHandle(STD_INPUT_HANDLE);
|
HANDLE h_cin = ::GetStdHandle(STD_INPUT_HANDLE);
|
||||||
|
|
||||||
|
@ -198,8 +62,8 @@ bool password_container::read_from_tty_double_check(const char *message) {
|
||||||
::SetConsoleMode(h_cin, mode_new);
|
::SetConsoleMode(h_cin, mode_new);
|
||||||
|
|
||||||
bool r = true;
|
bool r = true;
|
||||||
pass.reserve(max_password_size);
|
pass.reserve(tools::password_container::max_password_size);
|
||||||
while (pass.size() < max_password_size)
|
while (pass.size() < tools::password_container::max_password_size)
|
||||||
{
|
{
|
||||||
DWORD read;
|
DWORD read;
|
||||||
char ch;
|
char ch;
|
||||||
|
@ -235,16 +99,14 @@ bool password_container::read_from_tty_double_check(const char *message) {
|
||||||
return r;
|
return r;
|
||||||
}
|
}
|
||||||
|
|
||||||
#else
|
#else // end WIN32
|
||||||
|
|
||||||
namespace
|
bool is_cin_tty() noexcept
|
||||||
{
|
|
||||||
bool is_cin_tty()
|
|
||||||
{
|
{
|
||||||
return 0 != isatty(fileno(stdin));
|
return 0 != isatty(fileno(stdin));
|
||||||
}
|
}
|
||||||
|
|
||||||
int getch()
|
int getch() noexcept
|
||||||
{
|
{
|
||||||
struct termios tty_old;
|
struct termios tty_old;
|
||||||
tcgetattr(STDIN_FILENO, &tty_old);
|
tcgetattr(STDIN_FILENO, &tty_old);
|
||||||
|
@ -260,13 +122,13 @@ bool password_container::read_from_tty_double_check(const char *message) {
|
||||||
|
|
||||||
return ch;
|
return ch;
|
||||||
}
|
}
|
||||||
}
|
|
||||||
bool password_container::read_from_tty(std::string &aPass)
|
|
||||||
{
|
|
||||||
const char BACKSPACE = 127;
|
|
||||||
|
|
||||||
aPass.reserve(max_password_size);
|
bool read_from_tty(std::string& aPass)
|
||||||
while (aPass.size() < max_password_size)
|
{
|
||||||
|
static constexpr const char BACKSPACE = 127;
|
||||||
|
|
||||||
|
aPass.reserve(tools::password_container::max_password_size);
|
||||||
|
while (aPass.size() < tools::password_container::max_password_size)
|
||||||
{
|
{
|
||||||
int ch = getch();
|
int ch = getch();
|
||||||
if (EOF == ch)
|
if (EOF == ch)
|
||||||
|
@ -297,5 +159,90 @@ bool password_container::read_from_tty_double_check(const char *message) {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif
|
#endif // end !WIN32
|
||||||
|
|
||||||
|
void clear(std::string& pass) noexcept
|
||||||
|
{
|
||||||
|
//! TODO Call a memory wipe function that hopefully is not optimized out
|
||||||
|
pass.replace(0, pass.capacity(), pass.capacity(), '\0');
|
||||||
|
pass.clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool read_from_tty(const bool verify, const char *message, std::string& pass1, std::string& pass2)
|
||||||
|
{
|
||||||
|
while (true)
|
||||||
|
{
|
||||||
|
if (message)
|
||||||
|
std::cout << message <<": ";
|
||||||
|
if (!read_from_tty(pass1))
|
||||||
|
return false;
|
||||||
|
if (verify)
|
||||||
|
{
|
||||||
|
std::cout << "Confirm Password: ";
|
||||||
|
if (!read_from_tty(pass2))
|
||||||
|
return false;
|
||||||
|
if(pass1!=pass2)
|
||||||
|
{
|
||||||
|
std::cout << "Passwords do not match! Please try again." << std::endl;
|
||||||
|
clear(pass1);
|
||||||
|
clear(pass2);
|
||||||
|
}
|
||||||
|
else //new password matches
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
return true;
|
||||||
|
//No need to verify password entered at this point in the code
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool read_from_file(std::string& pass)
|
||||||
|
{
|
||||||
|
pass.reserve(tools::password_container::max_password_size);
|
||||||
|
for (size_t i = 0; i < tools::password_container::max_password_size; ++i)
|
||||||
|
{
|
||||||
|
char ch = static_cast<char>(std::cin.get());
|
||||||
|
if (std::cin.eof() || ch == '\n' || ch == '\r')
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
else if (std::cin.fail())
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
pass.push_back(ch);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // anonymous namespace
|
||||||
|
|
||||||
|
namespace tools
|
||||||
|
{
|
||||||
|
// deleted via private member
|
||||||
|
password_container::password_container() noexcept : m_password() {}
|
||||||
|
password_container::password_container(std::string&& password) noexcept
|
||||||
|
: m_password(std::move(password))
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
password_container::~password_container() noexcept
|
||||||
|
{
|
||||||
|
clear(m_password);
|
||||||
|
}
|
||||||
|
|
||||||
|
boost::optional<password_container> password_container::prompt(const bool verify, const char *message)
|
||||||
|
{
|
||||||
|
password_container pass1{};
|
||||||
|
password_container pass2{};
|
||||||
|
if (is_cin_tty() ? read_from_tty(verify, message, pass1.m_password, pass2.m_password) : read_from_file(pass1.m_password))
|
||||||
|
return {std::move(pass1)};
|
||||||
|
|
||||||
|
return boost::none;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -31,34 +31,37 @@
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <boost/program_options/variables_map.hpp>
|
#include <boost/optional/optional.hpp>
|
||||||
|
|
||||||
namespace tools
|
namespace tools
|
||||||
{
|
{
|
||||||
class password_container
|
class password_container
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
static const size_t max_password_size = 1024;
|
static constexpr const size_t max_password_size = 1024;
|
||||||
password_container(bool verify);
|
|
||||||
password_container(password_container&& rhs);
|
|
||||||
password_container(std::string&& password);
|
|
||||||
~password_container();
|
|
||||||
|
|
||||||
void clear();
|
//! Empty password
|
||||||
bool empty() const { return m_empty; }
|
password_container() noexcept;
|
||||||
const std::string& password() const { return m_password; }
|
|
||||||
void password(std::string&& val) { m_password = std::move(val); m_empty = false; }
|
//! `password` is used as password
|
||||||
bool read_password(const char *message = "Password");
|
password_container(std::string&& password) noexcept;
|
||||||
|
|
||||||
|
//! \return A password from stdin TTY prompt or `std::cin` pipe.
|
||||||
|
static boost::optional<password_container> prompt(bool verify, const char *mesage = "Password");
|
||||||
|
|
||||||
|
password_container(const password_container&) = delete;
|
||||||
|
password_container(password_container&& rhs) = default;
|
||||||
|
|
||||||
|
//! Wipes internal password
|
||||||
|
~password_container() noexcept;
|
||||||
|
|
||||||
|
password_container& operator=(const password_container&) = delete;
|
||||||
|
password_container& operator=(password_container&&) = default;
|
||||||
|
|
||||||
|
const std::string& password() const noexcept { return m_password; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
//delete constructor with no parameters
|
//! TODO Custom allocator that locks to RAM?
|
||||||
password_container();
|
|
||||||
bool read_from_file();
|
|
||||||
bool read_from_tty(std::string & pass);
|
|
||||||
bool read_from_tty_double_check(const char *message);
|
|
||||||
|
|
||||||
bool m_empty;
|
|
||||||
std::string m_password;
|
std::string m_password;
|
||||||
bool m_verify;
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
|
@ -174,9 +174,7 @@ boost::optional<tools::password_container> get_password(const boost::program_opt
|
||||||
|
|
||||||
if (command_line::has_arg(vm, opts.password))
|
if (command_line::has_arg(vm, opts.password))
|
||||||
{
|
{
|
||||||
tools::password_container pwd(false);
|
return tools::password_container{command_line::get_arg(vm, opts.password)};
|
||||||
pwd.password(command_line::get_arg(vm, opts.password));
|
|
||||||
return {std::move(pwd)};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (command_line::has_arg(vm, opts.password_file))
|
if (command_line::has_arg(vm, opts.password_file))
|
||||||
|
@ -192,19 +190,10 @@ boost::optional<tools::password_container> get_password(const boost::program_opt
|
||||||
|
|
||||||
// Remove line breaks the user might have inserted
|
// Remove line breaks the user might have inserted
|
||||||
boost::trim_right_if(password, boost::is_any_of("\r\n"));
|
boost::trim_right_if(password, boost::is_any_of("\r\n"));
|
||||||
return {tools::password_container(std::move(password))};
|
return {tools::password_container{std::move(password)}};
|
||||||
}
|
}
|
||||||
|
|
||||||
//vm is already part of the password container class. just need to check vm for an already existing wallet
|
return tools::wallet2::password_prompt(verify);
|
||||||
//here need to pass in variable map. This will indicate if the wallet already exists to the read password function
|
|
||||||
tools::password_container pwd(verify);
|
|
||||||
if (pwd.read_password())
|
|
||||||
{
|
|
||||||
return {std::move(pwd)};
|
|
||||||
}
|
|
||||||
|
|
||||||
tools::fail_msg_writer() << tools::wallet2::tr("failed to read wallet password");
|
|
||||||
return boost::none;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
std::unique_ptr<tools::wallet2> generate_from_json(const std::string& json_file, bool testnet, bool restricted)
|
std::unique_ptr<tools::wallet2> generate_from_json(const std::string& json_file, bool testnet, bool restricted)
|
||||||
|
@ -431,6 +420,18 @@ void wallet2::init_options(boost::program_options::options_description& desc_par
|
||||||
command_line::add_arg(desc_params, opts.restricted);
|
command_line::add_arg(desc_params, opts.restricted);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
boost::optional<password_container> wallet2::password_prompt(const bool is_new_wallet)
|
||||||
|
{
|
||||||
|
auto pwd_container = tools::password_container::prompt(
|
||||||
|
is_new_wallet, (is_new_wallet ? tr("Enter a password for your new wallet") : tr("Wallet password"))
|
||||||
|
);
|
||||||
|
if (!pwd_container)
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << tr("failed to read wallet password");
|
||||||
|
}
|
||||||
|
return pwd_container;
|
||||||
|
}
|
||||||
|
|
||||||
std::unique_ptr<wallet2> wallet2::make_from_json(const boost::program_options::variables_map& vm, const std::string& json_file)
|
std::unique_ptr<wallet2> wallet2::make_from_json(const boost::program_options::variables_map& vm, const std::string& json_file)
|
||||||
{
|
{
|
||||||
const options opts{};
|
const options opts{};
|
||||||
|
@ -444,7 +445,7 @@ std::pair<std::unique_ptr<wallet2>, password_container> wallet2::make_from_file(
|
||||||
auto pwd = get_password(vm, opts, false);
|
auto pwd = get_password(vm, opts, false);
|
||||||
if (!pwd)
|
if (!pwd)
|
||||||
{
|
{
|
||||||
return {nullptr, password_container(false)};
|
return {nullptr, password_container{}};
|
||||||
}
|
}
|
||||||
auto wallet = make_basic(vm, opts);
|
auto wallet = make_basic(vm, opts);
|
||||||
if (wallet)
|
if (wallet)
|
||||||
|
@ -460,7 +461,7 @@ std::pair<std::unique_ptr<wallet2>, password_container> wallet2::make_new(const
|
||||||
auto pwd = get_password(vm, opts, true);
|
auto pwd = get_password(vm, opts, true);
|
||||||
if (!pwd)
|
if (!pwd)
|
||||||
{
|
{
|
||||||
return {nullptr, password_container(false)};
|
return {nullptr, password_container{}};
|
||||||
}
|
}
|
||||||
return {make_basic(vm, opts), std::move(*pwd)};
|
return {make_basic(vm, opts), std::move(*pwd)};
|
||||||
}
|
}
|
||||||
|
|
|
@ -104,6 +104,9 @@ namespace tools
|
||||||
static bool has_testnet_option(const boost::program_options::variables_map& vm);
|
static bool has_testnet_option(const boost::program_options::variables_map& vm);
|
||||||
static void init_options(boost::program_options::options_description& desc_params);
|
static void init_options(boost::program_options::options_description& desc_params);
|
||||||
|
|
||||||
|
//! \return Password retrieved from prompt. Logs error on failure.
|
||||||
|
static boost::optional<password_container> password_prompt(const bool is_new_wallet);
|
||||||
|
|
||||||
//! Uses stdin and stdout. Returns a wallet2 if no errors.
|
//! Uses stdin and stdout. Returns a wallet2 if no errors.
|
||||||
static std::unique_ptr<wallet2> make_from_json(const boost::program_options::variables_map& vm, const std::string& json_file);
|
static std::unique_ptr<wallet2> make_from_json(const boost::program_options::variables_map& vm, const std::string& json_file);
|
||||||
|
|
||||||
|
|
|
@ -159,9 +159,9 @@ namespace tools
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
tools::password_container pwd(true);
|
login.password = tools::password_container::prompt(true, "RPC password").value_or(
|
||||||
pwd.read_password("RPC password");
|
tools::password_container{}
|
||||||
login.password = pwd.password();
|
).password();
|
||||||
}
|
}
|
||||||
|
|
||||||
if (login.username.empty() || login.password.empty())
|
if (login.username.empty() || login.password.empty())
|
||||||
|
|
Loading…
Reference in a new issue