src/rpcdump.cpp

   1 // Copyright (c) 2009-2012 Bitcoin Developers
   2 // Distributed under the MIT/X11 software license, see the accompanying
   3 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   4
   5 #include "init.h" // for pwalletMain
   6 #include "bitcoinrpc.h"
   7 #include "ui_interface.h"
   8 #include "base58.h"
   9
  10 #define printf OutputDebugStringF
  11
  12 using namespace json_spirit;
  13 using namespace std;
  14
  15 void EnsureWalletIsUnlocked();
  16
  17 class CTxDump
  18 {
  19 public:
  20     CBlockIndex *pindex;
  21     int64_t nValue;
  22     bool fSpent;
  23     CWalletTx* ptx;
  24     int nOut;
  25     CTxDump(CWalletTx* ptx = NULL, int nOut = -1)
  26     {
  27         pindex = NULL;
  28         nValue = 0;
  29         fSpent = false;
  30         this->ptx = ptx;
  31         this->nOut = nOut;
  32     }
  33 };
  34
  35 Value importprivkey(const Array& params, bool fHelp)
  36 {
  37     if (fHelp || params.size() < 1 || params.size() > 2)
  38         throw runtime_error(
  39             "importprivkey <novacoinprivkey> [label]\n"
  40             "Adds a private key (as returned by dumpprivkey) to your wallet.");
  41
  42     string strSecret = params[0].get_str();
  43     string strLabel = "";
  44     if (params.size() > 1)
  45         strLabel = params[1].get_str();
  46     CBitcoinSecret vchSecret;
  47     bool fGood = vchSecret.SetString(strSecret);
  48
  49     if (!fGood) throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Invalid private key");
  50     if (fWalletUnlockMintOnly) // ppcoin: no importprivkey in mint-only mode
  51         throw JSONRPCError(RPC_WALLET_UNLOCK_NEEDED, "Wallet is unlocked for minting only.");
  52
  53     CKey key;
  54     bool fCompressed;
  55     CSecret secret = vchSecret.GetSecret(fCompressed);
  56     key.SetSecret(secret, fCompressed);
  57     CKeyID vchAddress = key.GetPubKey().GetID();
  58     {
  59         LOCK2(cs_main, pwalletMain->cs_wallet);
  60
  61         pwalletMain->MarkDirty();
  62         pwalletMain->SetAddressBookName(vchAddress, strLabel);
  63
  64         if (!pwalletMain->AddKey(key))
  65             throw JSONRPCError(RPC_WALLET_ERROR, "Error adding key to wallet");
  66
  67         pwalletMain->ScanForWalletTransactions(pindexGenesisBlock, true);
  68         pwalletMain->ReacceptWalletTransactions();
  69     }
  70
  71     return Value::null;
  72 }
  73
  74 Value importaddress(const Array& params, bool fHelp)
  75 {
  76     if (fHelp || params.size() < 1 || params.size() > 3)
  77         throw runtime_error(
  78             "importaddress <address> [label] [rescan=true]\n"
  79             "Adds an address or script (in hex) that can be watched as if it were in your wallet but cannot be used to spend.");
  80
  81     CScript script;
  82     CBitcoinAddress address(params[0].get_str());
  83     if (address.IsValid()) {
  84         script.SetDestination(address.Get());
  85     } else if (IsHex(params[0].get_str())) {
  86         std::vector<unsigned char> data(ParseHex(params[0].get_str()));
  87         script = CScript(data.begin(), data.end());
  88     } else {
  89         throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Invalid Novacoin address or script");
  90     }
  91
  92     string strLabel = "";
  93     if (params.size() > 1)
  94         strLabel = params[1].get_str();
  95
  96     // Whether to perform rescan after import
  97     bool fRescan = true;
  98     if (params.size() > 2)
  99         fRescan = params[2].get_bool();
 100
 101     {
 102         LOCK2(cs_main, pwalletMain->cs_wallet);
 103         if (::IsMine(*pwalletMain, script) == MINE_SPENDABLE)
 104             throw JSONRPCError(RPC_WALLET_ERROR, "The wallet already contains the private key for this address or script");
 105
 106         // Don't throw error in case an address is already there
 107         if (pwalletMain->HaveWatchOnly(script))
 108             return Value::null;
 109
 110         pwalletMain->MarkDirty();
 111
 112         if (address.IsValid())
 113             pwalletMain->SetAddressBookName(address.Get(), strLabel);
 114
 115         if (!pwalletMain->AddWatchOnly(script))
 116             throw JSONRPCError(RPC_WALLET_ERROR, "Error adding address to wallet");
 117
 118         if (fRescan)
 119         {
 120             pwalletMain->ScanForWalletTransactions(pindexGenesisBlock, true);
 121             pwalletMain->ReacceptWalletTransactions();
 122         }
 123     }
 124
 125     return Value::null;
 126 }
 127
 128 Value removeaddress(const Array& params, bool fHelp)
 129 {
 130     if (fHelp || params.size() != 1)
 131         throw runtime_error(
 132             "removeaddress 'address'\n"
 133             "\nRemoves watch-only address or script (in hex) added by importaddress.\n"
 134             "\nArguments:\n"
 135             "1. 'address' (string, required) The address\n"
 136             "\nExamples:\n"
 137             "\nremoveaddress 4EqHMPgEAf56CQmU6ZWS8Ug4d7N3gsQVQA\n"
 138             "\nRemove watch-only address 4EqHMPgEAf56CQmU6ZWS8Ug4d7N3gsQVQA\n");
 139
 140     CScript script;
 141
 142     CBitcoinAddress address(params[0].get_str());
 143     if (address.IsValid()) {
 144         script.SetDestination(address.Get());
 145     } else if (IsHex(params[0].get_str())) {
 146         std::vector<unsigned char> data(ParseHex(params[0].get_str()));
 147         script = CScript(data.begin(), data.end());
 148     } else {
 149         throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Invalid Bitcoin address or script");
 150     }
 151
 152     if (::IsMine(*pwalletMain, script) == MINE_SPENDABLE)
 153         throw JSONRPCError(RPC_WALLET_ERROR, "The wallet contains the private key for this address or script - can't remove it");
 154
 155     if (!pwalletMain->HaveWatchOnly(script))
 156         throw JSONRPCError(RPC_WALLET_ERROR, "The wallet does not contain this address or script");
 157
 158     LOCK2(cs_main, pwalletMain->cs_wallet);
 159
 160     pwalletMain->MarkDirty();
 161
 162     if (!pwalletMain->RemoveWatchOnly(script))
 163         throw JSONRPCError(RPC_WALLET_ERROR, "Error removing address from wallet");
 164
 165     return Value::null;
 166 }
 167
 168 Value importwallet(const Array& params, bool fHelp)
 169 {
 170     if (fHelp || params.size() != 1)
 171         throw runtime_error(
 172             "importwallet <filename>\n"
 173             "Imports keys from a wallet dump file (see dumpwallet)."
 174             + HelpRequiringPassphrase());
 175
 176     EnsureWalletIsUnlocked();
 177
 178     if(!ImportWallet(pwalletMain, params[0].get_str().c_str()))
 179        throw JSONRPCError(RPC_WALLET_ERROR, "Error adding some keys to wallet");
 180
 181     return Value::null;
 182 }
 183
 184 Value dumpprivkey(const Array& params, bool fHelp)
 185 {
 186     if (fHelp || params.size() != 1)
 187         throw runtime_error(
 188             "dumpprivkey <novacoinaddress>\n"
 189             "Reveals the private key corresponding to <novacoinaddress>.");
 190
 191     EnsureWalletIsUnlocked();
 192
 193     string strAddress = params[0].get_str();
 194     CBitcoinAddress address;
 195     if (!address.SetString(strAddress))
 196         throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Invalid NovaCoin address");
 197     if (fWalletUnlockMintOnly) // ppcoin: no dumpprivkey in mint-only mode
 198         throw JSONRPCError(RPC_WALLET_UNLOCK_NEEDED, "Wallet is unlocked for minting only.");
 199     CKeyID keyID;
 200     if (!address.GetKeyID(keyID))
 201         throw JSONRPCError(RPC_TYPE_ERROR, "Address does not refer to a key");
 202     CSecret vchSecret;
 203     bool fCompressed;
 204     if (!pwalletMain->GetSecret(keyID, vchSecret, fCompressed))
 205         throw JSONRPCError(RPC_WALLET_ERROR, "Private key for address " + strAddress + " is not known");
 206     return CBitcoinSecret(vchSecret, fCompressed).ToString();
 207 }
 208
 209 Value dumpwallet(const Array& params, bool fHelp)
 210 {
 211     if (fHelp || params.size() != 1)
 212         throw runtime_error(
 213             "dumpwallet <filename>\n"
 214             "Dumps all wallet keys in a human-readable format."
 215             + HelpRequiringPassphrase());
 216
 217     EnsureWalletIsUnlocked();
 218
 219     if(!DumpWallet(pwalletMain, params[0].get_str().c_str() ))
 220       throw JSONRPCError(RPC_WALLET_ERROR, "Error dumping wallet keys to file");
 221
 222     return Value::null;
 223 }