Move importwallet and dumpwallet implementations to walletdb.cpp;
[novacoin.git] / 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 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 importwallet(const Array& params, bool fHelp)
75 {
76     if (fHelp || params.size() != 1)
77         throw runtime_error(
78             "importwallet <filename>\n"
79             "Imports keys from a wallet dump file (see dumpwallet)."
80             + HelpRequiringPassphrase());
81
82     EnsureWalletIsUnlocked();
83
84     if(!ImportWallet(pwalletMain, params[0].get_str().c_str()))
85        throw JSONRPCError(RPC_WALLET_ERROR, "Error adding some keys to wallet");
86
87     return Value::null;
88 }
89
90 Value dumpprivkey(const Array& params, bool fHelp)
91 {
92     if (fHelp || params.size() != 1)
93         throw runtime_error(
94             "dumpprivkey <novacoinaddress>\n"
95             "Reveals the private key corresponding to <novacoinaddress>.");
96
97     EnsureWalletIsUnlocked();
98
99     string strAddress = params[0].get_str();
100     CBitcoinAddress address;
101     if (!address.SetString(strAddress))
102         throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Invalid NovaCoin address");
103     if (fWalletUnlockMintOnly) // ppcoin: no dumpprivkey in mint-only mode
104         throw JSONRPCError(RPC_WALLET_UNLOCK_NEEDED, "Wallet is unlocked for minting only.");
105     CKeyID keyID;
106     if (!address.GetKeyID(keyID))
107         throw JSONRPCError(RPC_TYPE_ERROR, "Address does not refer to a key");
108     CSecret vchSecret;
109     bool fCompressed;
110     if (!pwalletMain->GetSecret(keyID, vchSecret, fCompressed))
111         throw JSONRPCError(RPC_WALLET_ERROR, "Private key for address " + strAddress + " is not known");
112     return CBitcoinSecret(vchSecret, fCompressed).ToString();
113 }
114
115 Value dumpwallet(const Array& params, bool fHelp)
116 {
117     if (fHelp || params.size() != 1)
118         throw runtime_error(
119             "dumpwallet <filename>\n"
120             "Dumps all wallet keys in a human-readable format."
121             + HelpRequiringPassphrase());
122
123     EnsureWalletIsUnlocked();
124
125     if(!DumpWallet(pwalletMain, params[0].get_str().c_str() ))
126       throw JSONRPCError(RPC_WALLET_ERROR, "Error dumping wallet keys to file");
127
128     return Value::null;
129 }