mirror of
https://github.com/AntelopeIO/spring.git
synced 2026-07-23 15:03:39 +00:00
Kevin Heifner
269 lines
11 KiB
C++
269 lines
11 KiB
C++
#include <eosio/chain/genesis_state.hpp>
|
|
#include <eosio/wallet_plugin/wallet.hpp>
|
|
#include <eosio/wallet_plugin/wallet_manager.hpp>
|
|
|
|
#include <boost/test/unit_test.hpp>
|
|
#include <eosio/chain/authority.hpp>
|
|
#include <eosio/chain/exceptions.hpp>
|
|
|
|
namespace eosio {
|
|
|
|
BOOST_AUTO_TEST_SUITE(wallet_tests)
|
|
|
|
/// Test creating the wallet
|
|
BOOST_AUTO_TEST_CASE(wallet_test)
|
|
{ try {
|
|
using namespace eosio::wallet;
|
|
|
|
wallet_data d;
|
|
soft_wallet wallet(d);
|
|
BOOST_CHECK(wallet.is_locked());
|
|
|
|
wallet.set_password("pass");
|
|
BOOST_CHECK(wallet.is_locked());
|
|
|
|
wallet.unlock("pass");
|
|
BOOST_CHECK(!wallet.is_locked());
|
|
|
|
wallet.set_wallet_filename("test");
|
|
BOOST_CHECK_EQUAL("test", wallet.get_wallet_filename());
|
|
|
|
BOOST_CHECK_EQUAL(0u, wallet.list_keys().size());
|
|
|
|
auto priv = fc::crypto::private_key::generate();
|
|
auto pub = priv.get_public_key();
|
|
auto wif = priv.to_string({});
|
|
wallet.import_key(wif);
|
|
BOOST_CHECK_EQUAL(1u, wallet.list_keys().size());
|
|
|
|
auto privCopy = wallet.get_private_key(pub);
|
|
BOOST_CHECK_EQUAL(wif, privCopy.to_string({}));
|
|
|
|
wallet.lock();
|
|
BOOST_CHECK(wallet.is_locked());
|
|
wallet.unlock("pass");
|
|
BOOST_CHECK_EQUAL(1u, wallet.list_keys().size());
|
|
wallet.save_wallet_file("wallet_test.json");
|
|
BOOST_CHECK(std::filesystem::exists("wallet_test.json"));
|
|
|
|
wallet_data d2;
|
|
soft_wallet wallet2(d2);
|
|
|
|
BOOST_CHECK(wallet2.is_locked());
|
|
wallet2.load_wallet_file("wallet_test.json");
|
|
BOOST_CHECK(wallet2.is_locked());
|
|
|
|
wallet2.unlock("pass");
|
|
BOOST_CHECK_EQUAL(1u, wallet2.list_keys().size());
|
|
|
|
auto privCopy2 = wallet2.get_private_key(pub);
|
|
BOOST_CHECK_EQUAL(wif, privCopy2.to_string({}));
|
|
|
|
std::filesystem::remove("wallet_test.json");
|
|
} FC_LOG_AND_RETHROW() }
|
|
|
|
/// Test wallet manager
|
|
BOOST_AUTO_TEST_CASE(wallet_manager_test)
|
|
{ try {
|
|
using namespace eosio::wallet;
|
|
|
|
if (std::filesystem::exists("test.wallet")) std::filesystem::remove("test.wallet");
|
|
if (std::filesystem::exists("test2.wallet")) std::filesystem::remove("test2.wallet");
|
|
if (std::filesystem::exists("testgen.wallet")) std::filesystem::remove("testgen.wallet");
|
|
|
|
constexpr auto key1 = "5JktVNHnRX48BUdtewU7N1CyL4Z886c42x7wYW7XhNWkDQRhdcS";
|
|
constexpr auto key2 = "5Ju5RTcVDo35ndtzHioPMgebvBM6LkJ6tvuU6LTNQv8yaz3ggZr";
|
|
constexpr auto key3 = "5KQwrPbwdL6PhXujxW37FSSQZ1JiwsST4cqQzDeyXtP79zkvFD3";
|
|
|
|
wallet_manager wm;
|
|
BOOST_CHECK_EQUAL(0u, wm.list_wallets().size());
|
|
BOOST_CHECK_THROW(wm.get_public_keys(), wallet_not_available_exception);
|
|
BOOST_CHECK_NO_THROW(wm.lock_all());
|
|
|
|
BOOST_CHECK_THROW(wm.lock("test"), fc::exception);
|
|
BOOST_CHECK_THROW(wm.unlock("test", "pw"), fc::exception);
|
|
BOOST_CHECK_THROW(wm.import_key("test", "pw"), fc::exception);
|
|
|
|
auto pw = wm.create("test");
|
|
BOOST_CHECK(!pw.empty());
|
|
BOOST_CHECK_EQUAL(0u, pw.find("PW")); // starts with PW
|
|
BOOST_CHECK_EQUAL(1u, wm.list_wallets().size());
|
|
// wallet has no keys when it is created
|
|
BOOST_CHECK_EQUAL(0u, wm.get_public_keys().size());
|
|
BOOST_CHECK_EQUAL(0u, wm.list_keys("test", pw).size());
|
|
BOOST_CHECK(wm.list_wallets().at(0).find("*") != std::string::npos);
|
|
wm.lock("test");
|
|
BOOST_CHECK(wm.list_wallets().at(0).find("*") == std::string::npos);
|
|
wm.unlock("test", pw);
|
|
BOOST_CHECK_THROW(wm.unlock("test", pw), chain::wallet_unlocked_exception);
|
|
BOOST_CHECK(wm.list_wallets().at(0).find("*") != std::string::npos);
|
|
wm.import_key("test", key1);
|
|
BOOST_CHECK_EQUAL(1u, wm.get_public_keys().size());
|
|
auto keys = wm.list_keys("test", pw);
|
|
|
|
auto pub_pri_pair = [](const char *key) -> auto {
|
|
private_key_type prikey = private_key_type(std::string(key));
|
|
return std::pair<const public_key_type, private_key_type>(prikey.get_public_key(), prikey);
|
|
};
|
|
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key1)) != keys.cend());
|
|
|
|
wm.import_key("test", key2);
|
|
keys = wm.list_keys("test", pw);
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key1)) != keys.cend());
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key2)) != keys.cend());
|
|
// key3 was not automatically imported
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key3)) == keys.cend());
|
|
|
|
wm.remove_key("test", pw, pub_pri_pair(key2).first.to_string({}));
|
|
BOOST_CHECK_EQUAL(1u, wm.get_public_keys().size());
|
|
keys = wm.list_keys("test", pw);
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key2)) == keys.cend());
|
|
wm.import_key("test", key2);
|
|
BOOST_CHECK_EQUAL(2u, wm.get_public_keys().size());
|
|
keys = wm.list_keys("test", pw);
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key2)) != keys.cend());
|
|
BOOST_CHECK_THROW(wm.remove_key("test", pw, pub_pri_pair(key3).first.to_string({})), fc::exception);
|
|
BOOST_CHECK_EQUAL(2u, wm.get_public_keys().size());
|
|
BOOST_CHECK_THROW(wm.remove_key("test", "PWnogood", pub_pri_pair(key2).first.to_string({})), wallet_invalid_password_exception);
|
|
BOOST_CHECK_EQUAL(2u, wm.get_public_keys().size());
|
|
|
|
wm.lock("test");
|
|
BOOST_CHECK_THROW(wm.list_keys("test", pw), wallet_locked_exception);
|
|
BOOST_CHECK_THROW(wm.get_public_keys(), wallet_locked_exception);
|
|
wm.unlock("test", pw);
|
|
BOOST_CHECK_EQUAL(2u, wm.get_public_keys().size());
|
|
BOOST_CHECK_EQUAL(2u, wm.list_keys("test", pw).size());
|
|
wm.lock_all();
|
|
BOOST_CHECK_THROW(wm.get_public_keys(), wallet_locked_exception);
|
|
BOOST_CHECK(wm.list_wallets().at(0).find("*") == std::string::npos);
|
|
|
|
auto pw2 = wm.create("test2");
|
|
BOOST_CHECK_EQUAL(2u, wm.list_wallets().size());
|
|
// wallet has no keys when it is created
|
|
BOOST_CHECK_EQUAL(0u, wm.get_public_keys().size());
|
|
wm.import_key("test2", key3);
|
|
BOOST_CHECK_EQUAL(1u, wm.get_public_keys().size());
|
|
BOOST_CHECK_THROW(wm.import_key("test2", key3), fc::exception);
|
|
keys = wm.list_keys("test2", pw2);
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key1)) == keys.cend());
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key2)) == keys.cend());
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key3)) != keys.cend());
|
|
wm.unlock("test", pw);
|
|
keys = wm.list_keys("test", pw);
|
|
auto keys2 = wm.list_keys("test2", pw2);
|
|
keys.insert(keys2.begin(), keys2.end());
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key1)) != keys.cend());
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key2)) != keys.cend());
|
|
BOOST_CHECK(std::find(keys.cbegin(), keys.cend(), pub_pri_pair(key3)) != keys.cend());
|
|
BOOST_CHECK_EQUAL(3u, keys.size());
|
|
|
|
BOOST_CHECK_THROW(wm.list_keys("test2", "PWnogood"), wallet_invalid_password_exception);
|
|
|
|
private_key_type pkey1{std::string(key1)};
|
|
private_key_type pkey2{std::string(key2)};
|
|
|
|
chain::signed_transaction trx;
|
|
auto chain_id = genesis_state().compute_chain_id();
|
|
flat_set<public_key_type> pubkeys;
|
|
pubkeys.emplace(pkey1.get_public_key());
|
|
pubkeys.emplace(pkey2.get_public_key());
|
|
trx = wm.sign_transaction(trx, pubkeys, chain_id );
|
|
flat_set<public_key_type> pks;
|
|
trx.get_signature_keys(chain_id, fc::time_point::maximum(), pks);
|
|
BOOST_CHECK_EQUAL(2u, pks.size());
|
|
BOOST_CHECK(find(pks.cbegin(), pks.cend(), pkey1.get_public_key()) != pks.cend());
|
|
BOOST_CHECK(find(pks.cbegin(), pks.cend(), pkey2.get_public_key()) != pks.cend());
|
|
|
|
BOOST_CHECK_EQUAL(3u, wm.get_public_keys().size());
|
|
wm.set_timeout(chrono::seconds(0));
|
|
BOOST_CHECK_THROW(wm.get_public_keys(), wallet_locked_exception);
|
|
BOOST_CHECK_THROW(wm.list_keys("test", pw), wallet_locked_exception);
|
|
|
|
wm.set_timeout(chrono::seconds(15));
|
|
|
|
wm.create("testgen");
|
|
BOOST_CHECK_THROW(wm.create_key("testgen", "xxx"), chain::wallet_exception);
|
|
wm.lock("testgen");
|
|
std::filesystem::remove("testgen.wallet");
|
|
|
|
const string test_key_create_types[] = {"K1", "R1", "k1", ""};
|
|
for(const string& key_type_to_create : test_key_create_types) {
|
|
string pw = wm.create("testgen");
|
|
|
|
//check that the public key returned looks legit through a string conversion
|
|
// (would throw otherwise)
|
|
public_key_type create_key_pub(wm.create_key("testgen", key_type_to_create));
|
|
|
|
//now pluck out the private key from the wallet and see if the public key of said
|
|
// private key matches what was returned earlier from the create_key() call
|
|
private_key_type create_key_priv(wm.list_keys("testgen", pw).cbegin()->second);
|
|
BOOST_CHECK_EQUAL(create_key_pub.to_string({}), create_key_priv.get_public_key().to_string({}));
|
|
|
|
wm.lock("testgen");
|
|
BOOST_CHECK(std::filesystem::exists("testgen.wallet"));
|
|
std::filesystem::remove("testgen.wallet");
|
|
}
|
|
|
|
BOOST_CHECK(std::filesystem::exists("test.wallet"));
|
|
BOOST_CHECK(std::filesystem::exists("test2.wallet"));
|
|
std::filesystem::remove("test.wallet");
|
|
std::filesystem::remove("test2.wallet");
|
|
|
|
} FC_LOG_AND_RETHROW() }
|
|
|
|
/// Test wallet manager
|
|
BOOST_AUTO_TEST_CASE(wallet_manager_create_test) {
|
|
try {
|
|
using namespace eosio::wallet;
|
|
|
|
if (std::filesystem::exists("test.wallet")) std::filesystem::remove("test.wallet");
|
|
|
|
wallet_manager wm;
|
|
wm.create("test");
|
|
constexpr auto key1 = "5JktVNHnRX48BUdtewU7N1CyL4Z886c42x7wYW7XhNWkDQRhdcS";
|
|
wm.import_key("test", key1);
|
|
BOOST_CHECK_THROW(wm.create("test"), wallet_exist_exception);
|
|
|
|
BOOST_CHECK_THROW(wm.create("./test"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create("../../test"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create("/tmp/test"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create("/tmp/"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create("/"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create(",/"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create(","), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create("<<"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create("<"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create(",<"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create(",<<"), wallet_exception);
|
|
BOOST_CHECK_THROW(wm.create(""), wallet_exception);
|
|
|
|
std::filesystem::remove("test.wallet");
|
|
|
|
wm.create(".test");
|
|
BOOST_CHECK(std::filesystem::exists(".test.wallet"));
|
|
std::filesystem::remove(".test.wallet");
|
|
wm.create("..test");
|
|
BOOST_CHECK(std::filesystem::exists("..test.wallet"));
|
|
std::filesystem::remove("..test.wallet");
|
|
wm.create("...test");
|
|
BOOST_CHECK(std::filesystem::exists("...test.wallet"));
|
|
std::filesystem::remove("...test.wallet");
|
|
wm.create(".");
|
|
BOOST_CHECK(std::filesystem::exists("..wallet"));
|
|
std::filesystem::remove("..wallet");
|
|
wm.create("__test_test");
|
|
BOOST_CHECK(std::filesystem::exists("__test_test.wallet"));
|
|
std::filesystem::remove("__test_test.wallet");
|
|
wm.create("t-t");
|
|
BOOST_CHECK(std::filesystem::exists("t-t.wallet"));
|
|
std::filesystem::remove("t-t.wallet");
|
|
|
|
} FC_LOG_AND_RETHROW()
|
|
}
|
|
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|
|
|
|
} // namespace eos
|