123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313 |
- // ECOin - Copyright (c) - 2014/2022 - GPLv3 - epsylon@riseup.net (https://03c8.net)
- #include <iostream>
- #include <fstream>
- #include "init.h" // for pwalletMain
- #include "ecoinrpc.h"
- #include "ui_interface.h"
- #include "base58.h"
- #include <boost/date_time/posix_time/posix_time.hpp>
- #include <boost/lexical_cast.hpp>
- #include <boost/variant/get.hpp>
- #include <boost/algorithm/string.hpp>
- #define printf OutputDebugStringF
- using namespace json_spirit;
- using namespace std;
- void EnsureWalletIsUnlocked();
- namespace bt = boost::posix_time;
- const std::locale formats[] = {
- std::locale(std::locale::classic(),new bt::time_input_facet("%Y-%m-%dT%H:%M:%SZ")),
- std::locale(std::locale::classic(),new bt::time_input_facet("%Y-%m-%d %H:%M:%S")),
- std::locale(std::locale::classic(),new bt::time_input_facet("%Y/%m/%d %H:%M:%S")),
- std::locale(std::locale::classic(),new bt::time_input_facet("%d.%m.%Y %H:%M:%S")),
- std::locale(std::locale::classic(),new bt::time_input_facet("%Y-%m-%d"))
- };
- const size_t formats_n = sizeof(formats)/sizeof(formats[0]);
- std::time_t pt_to_time_t(const bt::ptime& pt)
- {
- bt::ptime timet_start(boost::gregorian::date(1970,1,1));
- bt::time_duration diff = pt - timet_start;
- return diff.ticks()/bt::time_duration::rep_type::ticks_per_second;
- }
- int64 DecodeDumpTime(const std::string& s)
- {
- bt::ptime pt;
- for(size_t i=0; i<formats_n; ++i)
- {
- std::istringstream is(s);
- is.imbue(formats[i]);
- is >> pt;
- if(pt != bt::ptime()) break;
- }
- return pt_to_time_t(pt);
- }
- std::string static EncodeDumpTime(int64 nTime) {
- return DateTimeStrFormat("%Y-%m-%dT%H:%M:%SZ", nTime);
- }
- std::string static EncodeDumpString(const std::string &str) {
- std::stringstream ret;
- BOOST_FOREACH(unsigned char c, str) {
- if (c <= 32 || c >= 128 || c == '%') {
- ret << '%' << HexStr(&c, &c + 1);
- } else {
- ret << c;
- }
- }
- return ret.str();
- }
- std::string DecodeDumpString(const std::string &str) {
- std::stringstream ret;
- for (unsigned int pos = 0; pos < str.length(); pos++) {
- unsigned char c = str[pos];
- if (c == '%' && pos+2 < str.length()) {
- c = (((str[pos+1]>>6)*9+((str[pos+1]-'0')&15)) << 4) |
- ((str[pos+2]>>6)*9+((str[pos+2]-'0')&15));
- pos += 2;
- }
- ret << c;
- }
- return ret.str();
- }
- class CTxDump
- {
- public:
- CBlockIndex *pindex;
- int64 nValue;
- bool fSpent;
- CWalletTx* ptx;
- int nOut;
- CTxDump(CWalletTx* ptx = NULL, int nOut = -1)
- {
- pindex = NULL;
- nValue = 0;
- fSpent = false;
- this->ptx = ptx;
- this->nOut = nOut;
- }
- };
- Value importprivkey(const Array& params, bool fHelp)
- {
- if (fHelp || params.size() < 1 || params.size() > 2)
- throw runtime_error(
- "importprivkey <ecoinprivkey> [label]\n"
- "Adds a private key (as returned by dumpprivkey) to your wallet.");
- string strSecret = params[0].get_str();
- string strLabel = "";
- if (params.size() > 1)
- strLabel = params[1].get_str();
- CEcoinSecret vchSecret;
- bool fGood = vchSecret.SetString(strSecret);
- if (!fGood) throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Invalid private key");
- if (fWalletUnlockStakingOnly) // ecoin: no importprivkey in mint-only mode
- throw JSONRPCError(RPC_WALLET_UNLOCK_NEEDED, "Wallet is unlocked for minting only.");
- CKey key;
- bool fCompressed;
- CSecret secret = vchSecret.GetSecret(fCompressed);
- key.SetSecret(secret, fCompressed);
- CKeyID vchAddress = key.GetPubKey().GetID();
- {
- LOCK2(cs_main, pwalletMain->cs_wallet);
- pwalletMain->MarkDirty();
- pwalletMain->SetAddressBookName(vchAddress, strLabel);
- if (!pwalletMain->AddKey(key))
- throw JSONRPCError(RPC_WALLET_ERROR, "Error adding key to wallet");
- pwalletMain->ScanForWalletTransactions(pindexGenesisBlock, true);
- pwalletMain->ReacceptWalletTransactions();
- }
- return Value::null;
- }
- Value importwallet(const Array& params, bool fHelp)
- {
- if (fHelp || params.size() != 1)
- throw runtime_error(
- "importwallet <filename>\n"
- "Imports keys from a wallet dump file (see dumpwallet).");
- EnsureWalletIsUnlocked();
- ifstream file;
- file.open(params[0].get_str().c_str());
- if (!file.is_open())
- throw JSONRPCError(RPC_INVALID_PARAMETER, "Cannot open wallet dump file");
- int64 nTimeBegin = pindexBest->nTime;
- bool fGood = true;
- while (file.good()) {
- std::string line;
- std::getline(file, line);
- if (line.empty() || line[0] == '#')
- continue;
- std::vector<std::string> vstr;
- boost::split(vstr, line, boost::is_any_of(" "));
- if (vstr.size() < 2)
- continue;
- CEcoinSecret vchSecret;
- if (!vchSecret.SetString(vstr[0]))
- continue;
- bool fCompressed;
- CKey key;
- CSecret secret = vchSecret.GetSecret(fCompressed);
- key.SetSecret(secret, fCompressed);
- CKeyID keyid = key.GetPubKey().GetID();
- if (pwalletMain->HaveKey(keyid)) {
- printf("Skipping import of %s (key already present)\n", CEcoinAddress(keyid).ToString().c_str());
- continue;
- }
- int64 nTime = DecodeDumpTime(vstr[1]);
- std::string strLabel;
- bool fLabel = true;
- for (unsigned int nStr = 2; nStr < vstr.size(); nStr++) {
- if (boost::algorithm::starts_with(vstr[nStr], "#"))
- break;
- if (vstr[nStr] == "change=1")
- fLabel = false;
- if (vstr[nStr] == "reserve=1")
- fLabel = false;
- if (boost::algorithm::starts_with(vstr[nStr], "label=")) {
- strLabel = DecodeDumpString(vstr[nStr].substr(6));
- fLabel = true;
- }
- }
- printf("Importing %s...\n", CEcoinAddress(keyid).ToString().c_str());
- if (!pwalletMain->AddKey(key)) {
- fGood = false;
- continue;
- }
- pwalletMain->mapKeyMetadata[keyid].nCreateTime = nTime;
- if (fLabel)
- pwalletMain->SetAddressBookName(keyid, strLabel);
- nTimeBegin = std::min(nTimeBegin, nTime);
- }
- file.close();
- CBlockIndex *pindex = pindexBest;
- while (pindex && pindex->pprev && pindex->nTime > nTimeBegin - 7200)
- pindex = pindex->pprev;
- printf("Rescanning last %i blocks\n", pindexBest->nHeight - pindex->nHeight + 1);
- pwalletMain->ScanForWalletTransactions(pindex);
- pwalletMain->ReacceptWalletTransactions();
- pwalletMain->MarkDirty();
- if (!fGood)
- throw JSONRPCError(RPC_WALLET_ERROR, "Error adding some keys to wallet");
- return Value::null;
- }
- Value dumpprivkey(const Array& params, bool fHelp)
- {
- if (fHelp || params.size() != 1)
- throw runtime_error(
- "dumpprivkey <ecoinaddress>\n"
- "Reveals the private key corresponding to <ecoinaddress>.");
- EnsureWalletIsUnlocked();
- string strAddress = params[0].get_str();
- CEcoinAddress address;
- if (!address.SetString(strAddress))
- throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Invalid Ecoin address");
- if (fWalletUnlockStakingOnly) // ecoin: no dumpprivkey in mint-only mode
- throw JSONRPCError(RPC_WALLET_UNLOCK_NEEDED, "Wallet is unlocked for minting only.");
- CKeyID keyID;
- if (!address.GetKeyID(keyID))
- throw JSONRPCError(RPC_TYPE_ERROR, "Address does not refer to a key");
- CSecret vchSecret;
- bool fCompressed;
- if (!pwalletMain->GetSecret(keyID, vchSecret, fCompressed))
- throw JSONRPCError(RPC_WALLET_ERROR, "Private key for address " + strAddress + " is not known");
- return CEcoinSecret(vchSecret, fCompressed).ToString();
- }
- Value dumpwallet(const Array& params, bool fHelp)
- {
- if (fHelp || params.size() != 1)
- throw runtime_error(
- "dumpwallet <filename>\n"
- "Dumps all wallet keys in a human-readable format.");
- EnsureWalletIsUnlocked();
- ofstream file;
- file.open(params[0].get_str().c_str());
- if (!file.is_open())
- throw JSONRPCError(RPC_INVALID_PARAMETER, "Cannot open wallet dump file");
- std::map<CKeyID, int64> mapKeyBirth;
- std::set<CKeyID> setKeyPool;
- pwalletMain->GetKeyBirthTimes(mapKeyBirth);
- pwalletMain->GetAllReserveKeys(setKeyPool);
- // sort time/key pairs
- std::vector<std::pair<int64, CKeyID> > vKeyBirth;
- for (std::map<CKeyID, int64>::const_iterator it = mapKeyBirth.begin(); it != mapKeyBirth.end(); it++) {
- vKeyBirth.push_back(std::make_pair(it->second, it->first));
- }
- mapKeyBirth.clear();
- std::sort(vKeyBirth.begin(), vKeyBirth.end());
- // produce output
- file << strprintf("# Wallet dump created by Ecoin %s (%s)\n", CLIENT_BUILD.c_str(), CLIENT_DATE.c_str());
- file << strprintf("# * Created on %s\n", EncodeDumpTime(GetTime()).c_str());
- file << strprintf("# * Best block at time of backup was %i (%s),\n", nBestHeight, hashBestChain.ToString().c_str());
- file << strprintf("# mined on %s\n", EncodeDumpTime(pindexBest->nTime).c_str());
- file << "\n";
- for (std::vector<std::pair<int64, CKeyID> >::const_iterator it = vKeyBirth.begin(); it != vKeyBirth.end(); it++) {
- const CKeyID &keyid = it->second;
- std::string strTime = EncodeDumpTime(it->first);
- std::string strAddr = CEcoinAddress(keyid).ToString();
- bool IsCompressed;
- CKey key;
- if (pwalletMain->GetKey(keyid, key)) {
- if (pwalletMain->mapAddressBook.count(keyid)) {
- CSecret secret = key.GetSecret(IsCompressed);
- file << strprintf("%s %s label=%s # addr=%s\n", CEcoinSecret(secret, IsCompressed).ToString().c_str(), strTime.c_str(), EncodeDumpString(pwalletMain->mapAddressBook[keyid]).c_str(), strAddr.c_str());
- } else if (setKeyPool.count(keyid)) {
- CSecret secret = key.GetSecret(IsCompressed);
- file << strprintf("%s %s reserve=1 # addr=%s\n", CEcoinSecret(secret, IsCompressed).ToString().c_str(), strTime.c_str(), strAddr.c_str());
- } else {
- CSecret secret = key.GetSecret(IsCompressed);
- file << strprintf("%s %s change=1 # addr=%s\n", CEcoinSecret(secret, IsCompressed).ToString().c_str(), strTime.c_str(), strAddr.c_str());
- }
- }
- }
- file << "\n";
- file << "# End of dump\n";
- file.close();
- return Value::null;
- }
|