]>
Commit | Line | Data |
---|---|---|
e89b9f6a PW |
1 | // Copyright (c) 2009-2011 Satoshi Nakamoto & Bitcoin developers |
2 | // Distributed under the MIT/X11 software license, see the accompanying | |
3 | // file license.txt or http://www.opensource.org/licenses/mit-license.php. | |
4 | ||
5 | #include "headers.h" | |
6 | #include "db.h" | |
7 | #include "cryptopp/sha.h" | |
8 | ||
9 | using namespace std; | |
10 | ||
11 | ||
12 | ||
13 | ////////////////////////////////////////////////////////////////////////////// | |
14 | // | |
15 | // mapWallet | |
16 | // | |
17 | ||
64c7ee7e PW |
18 | bool CWallet::AddKey(const CKey& key) |
19 | { | |
20 | this->CKeyStore::AddKey(key); | |
21 | if (!fFileBacked) | |
22 | return true; | |
23 | return CWalletDB(strWalletFile).WriteKey(key.GetPubKey(), key.GetPrivKey()); | |
24 | } | |
25 | ||
26 | void CWallet::WalletUpdateSpent(const CTransaction &tx) | |
e89b9f6a PW |
27 | { |
28 | // Anytime a signature is successfully verified, it's proof the outpoint is spent. | |
29 | // Update the wallet spent flag if it doesn't know due to wallet.dat being | |
30 | // restored from backup or the user making copies of wallet.dat. | |
31 | CRITICAL_BLOCK(cs_mapWallet) | |
32 | { | |
64c7ee7e | 33 | BOOST_FOREACH(const CTxIn& txin, tx.vin) |
e89b9f6a | 34 | { |
64c7ee7e PW |
35 | map<uint256, CWalletTx>::iterator mi = mapWallet.find(txin.prevout.hash); |
36 | if (mi != mapWallet.end()) | |
e89b9f6a | 37 | { |
64c7ee7e PW |
38 | CWalletTx& wtx = (*mi).second; |
39 | if (!wtx.IsSpent(txin.prevout.n) && IsMine(wtx.vout[txin.prevout.n])) | |
40 | { | |
41 | printf("WalletUpdateSpent found spent coin %sbc %s\n", FormatMoney(wtx.GetCredit()).c_str(), wtx.GetHash().ToString().c_str()); | |
42 | wtx.MarkSpent(txin.prevout.n); | |
43 | wtx.WriteToDisk(); | |
44 | vWalletUpdated.push_back(txin.prevout.hash); | |
45 | } | |
e89b9f6a PW |
46 | } |
47 | } | |
48 | } | |
49 | } | |
50 | ||
64c7ee7e | 51 | bool CWallet::AddToWallet(const CWalletTx& wtxIn) |
e89b9f6a PW |
52 | { |
53 | uint256 hash = wtxIn.GetHash(); | |
54 | CRITICAL_BLOCK(cs_mapWallet) | |
55 | { | |
56 | // Inserts only if not already there, returns tx inserted or tx found | |
57 | pair<map<uint256, CWalletTx>::iterator, bool> ret = mapWallet.insert(make_pair(hash, wtxIn)); | |
58 | CWalletTx& wtx = (*ret.first).second; | |
64c7ee7e | 59 | wtx.pwallet = this; |
e89b9f6a PW |
60 | bool fInsertedNew = ret.second; |
61 | if (fInsertedNew) | |
62 | wtx.nTimeReceived = GetAdjustedTime(); | |
63 | ||
64 | bool fUpdated = false; | |
65 | if (!fInsertedNew) | |
66 | { | |
67 | // Merge | |
68 | if (wtxIn.hashBlock != 0 && wtxIn.hashBlock != wtx.hashBlock) | |
69 | { | |
70 | wtx.hashBlock = wtxIn.hashBlock; | |
71 | fUpdated = true; | |
72 | } | |
73 | if (wtxIn.nIndex != -1 && (wtxIn.vMerkleBranch != wtx.vMerkleBranch || wtxIn.nIndex != wtx.nIndex)) | |
74 | { | |
75 | wtx.vMerkleBranch = wtxIn.vMerkleBranch; | |
76 | wtx.nIndex = wtxIn.nIndex; | |
77 | fUpdated = true; | |
78 | } | |
79 | if (wtxIn.fFromMe && wtxIn.fFromMe != wtx.fFromMe) | |
80 | { | |
81 | wtx.fFromMe = wtxIn.fFromMe; | |
82 | fUpdated = true; | |
83 | } | |
84 | fUpdated |= wtx.UpdateSpent(wtxIn.vfSpent); | |
85 | } | |
86 | ||
87 | //// debug print | |
88 | printf("AddToWallet %s %s%s\n", wtxIn.GetHash().ToString().substr(0,10).c_str(), (fInsertedNew ? "new" : ""), (fUpdated ? "update" : "")); | |
89 | ||
90 | // Write to disk | |
91 | if (fInsertedNew || fUpdated) | |
92 | if (!wtx.WriteToDisk()) | |
93 | return false; | |
94 | ||
95 | // If default receiving address gets used, replace it with a new one | |
96 | CScript scriptDefaultKey; | |
97 | scriptDefaultKey.SetBitcoinAddress(vchDefaultKey); | |
98 | BOOST_FOREACH(const CTxOut& txout, wtx.vout) | |
99 | { | |
100 | if (txout.scriptPubKey == scriptDefaultKey) | |
e9f61c87 | 101 | SetDefaultKey(GetKeyFromKeyPool()); |
e89b9f6a PW |
102 | } |
103 | ||
104 | // Notify UI | |
105 | vWalletUpdated.push_back(hash); | |
64c7ee7e PW |
106 | |
107 | // since AddToWallet is called directly for self-originating transactions, check for consumption of own coins | |
108 | WalletUpdateSpent(wtx); | |
e89b9f6a PW |
109 | } |
110 | ||
111 | // Refresh UI | |
112 | MainFrameRepaint(); | |
113 | return true; | |
114 | } | |
115 | ||
64c7ee7e | 116 | bool CWallet::AddToWalletIfInvolvingMe(const CTransaction& tx, const CBlock* pblock, bool fUpdate) |
e89b9f6a PW |
117 | { |
118 | uint256 hash = tx.GetHash(); | |
119 | bool fExisted = mapWallet.count(hash); | |
120 | if (fExisted && !fUpdate) return false; | |
64c7ee7e | 121 | if (fExisted || IsMine(tx) || IsFromMe(tx)) |
e89b9f6a | 122 | { |
64c7ee7e | 123 | CWalletTx wtx(this,tx); |
e89b9f6a PW |
124 | // Get merkle branch if transaction was found in a block |
125 | if (pblock) | |
126 | wtx.SetMerkleBranch(pblock); | |
127 | return AddToWallet(wtx); | |
128 | } | |
64c7ee7e PW |
129 | else |
130 | WalletUpdateSpent(tx); | |
e89b9f6a PW |
131 | return false; |
132 | } | |
133 | ||
64c7ee7e | 134 | bool CWallet::EraseFromWallet(uint256 hash) |
e89b9f6a | 135 | { |
64c7ee7e PW |
136 | if (!fFileBacked) |
137 | return false; | |
e89b9f6a PW |
138 | CRITICAL_BLOCK(cs_mapWallet) |
139 | { | |
140 | if (mapWallet.erase(hash)) | |
64c7ee7e | 141 | CWalletDB(strWalletFile).EraseTx(hash); |
e89b9f6a PW |
142 | } |
143 | return true; | |
144 | } | |
145 | ||
146 | ||
64c7ee7e | 147 | bool CWallet::IsMine(const CTxIn &txin) const |
e89b9f6a PW |
148 | { |
149 | CRITICAL_BLOCK(cs_mapWallet) | |
150 | { | |
64c7ee7e | 151 | map<uint256, CWalletTx>::const_iterator mi = mapWallet.find(txin.prevout.hash); |
e89b9f6a PW |
152 | if (mi != mapWallet.end()) |
153 | { | |
154 | const CWalletTx& prev = (*mi).second; | |
64c7ee7e PW |
155 | if (txin.prevout.n < prev.vout.size()) |
156 | if (IsMine(prev.vout[txin.prevout.n])) | |
e89b9f6a PW |
157 | return true; |
158 | } | |
159 | } | |
160 | return false; | |
161 | } | |
162 | ||
64c7ee7e | 163 | int64 CWallet::GetDebit(const CTxIn &txin) const |
e89b9f6a PW |
164 | { |
165 | CRITICAL_BLOCK(cs_mapWallet) | |
166 | { | |
64c7ee7e | 167 | map<uint256, CWalletTx>::const_iterator mi = mapWallet.find(txin.prevout.hash); |
e89b9f6a PW |
168 | if (mi != mapWallet.end()) |
169 | { | |
170 | const CWalletTx& prev = (*mi).second; | |
64c7ee7e PW |
171 | if (txin.prevout.n < prev.vout.size()) |
172 | if (IsMine(prev.vout[txin.prevout.n])) | |
173 | return prev.vout[txin.prevout.n].nValue; | |
e89b9f6a PW |
174 | } |
175 | } | |
176 | return 0; | |
177 | } | |
178 | ||
179 | int64 CWalletTx::GetTxTime() const | |
180 | { | |
181 | if (!fTimeReceivedIsTxTime && hashBlock != 0) | |
182 | { | |
183 | // If we did not receive the transaction directly, we rely on the block's | |
184 | // time to figure out when it happened. We use the median over a range | |
185 | // of blocks to try to filter out inaccurate block times. | |
186 | map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashBlock); | |
187 | if (mi != mapBlockIndex.end()) | |
188 | { | |
189 | CBlockIndex* pindex = (*mi).second; | |
190 | if (pindex) | |
191 | return pindex->GetMedianTime(); | |
192 | } | |
193 | } | |
194 | return nTimeReceived; | |
195 | } | |
196 | ||
197 | int CWalletTx::GetRequestCount() const | |
198 | { | |
199 | // Returns -1 if it wasn't being tracked | |
200 | int nRequests = -1; | |
64c7ee7e | 201 | CRITICAL_BLOCK(pwallet->cs_mapRequestCount) |
e89b9f6a PW |
202 | { |
203 | if (IsCoinBase()) | |
204 | { | |
205 | // Generated block | |
206 | if (hashBlock != 0) | |
207 | { | |
64c7ee7e PW |
208 | map<uint256, int>::const_iterator mi = pwallet->mapRequestCount.find(hashBlock); |
209 | if (mi != pwallet->mapRequestCount.end()) | |
e89b9f6a PW |
210 | nRequests = (*mi).second; |
211 | } | |
212 | } | |
213 | else | |
214 | { | |
215 | // Did anyone request this transaction? | |
64c7ee7e PW |
216 | map<uint256, int>::const_iterator mi = pwallet->mapRequestCount.find(GetHash()); |
217 | if (mi != pwallet->mapRequestCount.end()) | |
e89b9f6a PW |
218 | { |
219 | nRequests = (*mi).second; | |
220 | ||
221 | // How about the block it's in? | |
222 | if (nRequests == 0 && hashBlock != 0) | |
223 | { | |
64c7ee7e PW |
224 | map<uint256, int>::const_iterator mi = pwallet->mapRequestCount.find(hashBlock); |
225 | if (mi != pwallet->mapRequestCount.end()) | |
e89b9f6a PW |
226 | nRequests = (*mi).second; |
227 | else | |
228 | nRequests = 1; // If it's in someone else's block it must have got out | |
229 | } | |
230 | } | |
231 | } | |
232 | } | |
233 | return nRequests; | |
234 | } | |
235 | ||
236 | void CWalletTx::GetAmounts(int64& nGeneratedImmature, int64& nGeneratedMature, list<pair<string, int64> >& listReceived, | |
237 | list<pair<string, int64> >& listSent, int64& nFee, string& strSentAccount) const | |
238 | { | |
239 | nGeneratedImmature = nGeneratedMature = nFee = 0; | |
240 | listReceived.clear(); | |
241 | listSent.clear(); | |
242 | strSentAccount = strFromAccount; | |
243 | ||
244 | if (IsCoinBase()) | |
245 | { | |
246 | if (GetBlocksToMaturity() > 0) | |
64c7ee7e | 247 | nGeneratedImmature = pwallet->GetCredit(*this); |
e89b9f6a PW |
248 | else |
249 | nGeneratedMature = GetCredit(); | |
250 | return; | |
251 | } | |
252 | ||
253 | // Compute fee: | |
254 | int64 nDebit = GetDebit(); | |
255 | if (nDebit > 0) // debit>0 means we signed/sent this transaction | |
256 | { | |
257 | int64 nValueOut = GetValueOut(); | |
258 | nFee = nDebit - nValueOut; | |
259 | } | |
260 | ||
261 | // Sent/received. Standard client will never generate a send-to-multiple-recipients, | |
262 | // but non-standard clients might (so return a list of address/amount pairs) | |
263 | BOOST_FOREACH(const CTxOut& txout, vout) | |
264 | { | |
265 | string address; | |
266 | uint160 hash160; | |
267 | vector<unsigned char> vchPubKey; | |
268 | if (ExtractHash160(txout.scriptPubKey, hash160)) | |
269 | address = Hash160ToAddress(hash160); | |
aa0c5e87 | 270 | else if (ExtractPubKey(txout.scriptPubKey, NULL, vchPubKey)) |
e89b9f6a PW |
271 | address = PubKeyToAddress(vchPubKey); |
272 | else | |
273 | { | |
274 | printf("CWalletTx::GetAmounts: Unknown transaction type found, txid %s\n", | |
275 | this->GetHash().ToString().c_str()); | |
276 | address = " unknown "; | |
277 | } | |
278 | ||
279 | // Don't report 'change' txouts | |
64c7ee7e | 280 | if (nDebit > 0 && pwallet->IsChange(txout)) |
e89b9f6a PW |
281 | continue; |
282 | ||
283 | if (nDebit > 0) | |
284 | listSent.push_back(make_pair(address, txout.nValue)); | |
285 | ||
64c7ee7e | 286 | if (pwallet->IsMine(txout)) |
e89b9f6a PW |
287 | listReceived.push_back(make_pair(address, txout.nValue)); |
288 | } | |
289 | ||
290 | } | |
291 | ||
292 | void CWalletTx::GetAccountAmounts(const string& strAccount, int64& nGenerated, int64& nReceived, | |
293 | int64& nSent, int64& nFee) const | |
294 | { | |
295 | nGenerated = nReceived = nSent = nFee = 0; | |
296 | ||
297 | int64 allGeneratedImmature, allGeneratedMature, allFee; | |
298 | allGeneratedImmature = allGeneratedMature = allFee = 0; | |
299 | string strSentAccount; | |
300 | list<pair<string, int64> > listReceived; | |
301 | list<pair<string, int64> > listSent; | |
302 | GetAmounts(allGeneratedImmature, allGeneratedMature, listReceived, listSent, allFee, strSentAccount); | |
303 | ||
304 | if (strAccount == "") | |
305 | nGenerated = allGeneratedMature; | |
306 | if (strAccount == strSentAccount) | |
307 | { | |
308 | BOOST_FOREACH(const PAIRTYPE(string,int64)& s, listSent) | |
309 | nSent += s.second; | |
310 | nFee = allFee; | |
311 | } | |
64c7ee7e | 312 | CRITICAL_BLOCK(pwallet->cs_mapAddressBook) |
e89b9f6a PW |
313 | { |
314 | BOOST_FOREACH(const PAIRTYPE(string,int64)& r, listReceived) | |
315 | { | |
64c7ee7e | 316 | if (pwallet->mapAddressBook.count(r.first)) |
e89b9f6a | 317 | { |
64c7ee7e PW |
318 | map<string, string>::const_iterator mi = pwallet->mapAddressBook.find(r.first); |
319 | if (mi != pwallet->mapAddressBook.end() && (*mi).second == strAccount) | |
e89b9f6a | 320 | nReceived += r.second; |
e89b9f6a PW |
321 | } |
322 | else if (strAccount.empty()) | |
323 | { | |
324 | nReceived += r.second; | |
325 | } | |
326 | } | |
327 | } | |
328 | } | |
329 | ||
330 | void CWalletTx::AddSupportingTransactions(CTxDB& txdb) | |
331 | { | |
332 | vtxPrev.clear(); | |
333 | ||
334 | const int COPY_DEPTH = 3; | |
335 | if (SetMerkleBranch() < COPY_DEPTH) | |
336 | { | |
337 | vector<uint256> vWorkQueue; | |
338 | BOOST_FOREACH(const CTxIn& txin, vin) | |
339 | vWorkQueue.push_back(txin.prevout.hash); | |
340 | ||
341 | // This critsect is OK because txdb is already open | |
64c7ee7e | 342 | CRITICAL_BLOCK(pwallet->cs_mapWallet) |
e89b9f6a PW |
343 | { |
344 | map<uint256, const CMerkleTx*> mapWalletPrev; | |
345 | set<uint256> setAlreadyDone; | |
346 | for (int i = 0; i < vWorkQueue.size(); i++) | |
347 | { | |
348 | uint256 hash = vWorkQueue[i]; | |
349 | if (setAlreadyDone.count(hash)) | |
350 | continue; | |
351 | setAlreadyDone.insert(hash); | |
352 | ||
353 | CMerkleTx tx; | |
64c7ee7e PW |
354 | map<uint256, CWalletTx>::const_iterator mi = pwallet->mapWallet.find(hash); |
355 | if (mi != pwallet->mapWallet.end()) | |
e89b9f6a | 356 | { |
64c7ee7e PW |
357 | tx = (*mi).second; |
358 | BOOST_FOREACH(const CMerkleTx& txWalletPrev, (*mi).second.vtxPrev) | |
e89b9f6a PW |
359 | mapWalletPrev[txWalletPrev.GetHash()] = &txWalletPrev; |
360 | } | |
361 | else if (mapWalletPrev.count(hash)) | |
362 | { | |
363 | tx = *mapWalletPrev[hash]; | |
364 | } | |
365 | else if (!fClient && txdb.ReadDiskTx(hash, tx)) | |
366 | { | |
367 | ; | |
368 | } | |
369 | else | |
370 | { | |
371 | printf("ERROR: AddSupportingTransactions() : unsupported transaction\n"); | |
372 | continue; | |
373 | } | |
374 | ||
375 | int nDepth = tx.SetMerkleBranch(); | |
376 | vtxPrev.push_back(tx); | |
377 | ||
378 | if (nDepth < COPY_DEPTH) | |
379 | BOOST_FOREACH(const CTxIn& txin, tx.vin) | |
380 | vWorkQueue.push_back(txin.prevout.hash); | |
381 | } | |
382 | } | |
383 | } | |
384 | ||
385 | reverse(vtxPrev.begin(), vtxPrev.end()); | |
386 | } | |
387 | ||
388 | bool CWalletTx::WriteToDisk() | |
389 | { | |
64c7ee7e | 390 | return CWalletDB(pwallet->strWalletFile).WriteTx(GetHash(), *this); |
e89b9f6a PW |
391 | } |
392 | ||
64c7ee7e | 393 | int CWallet::ScanForWalletTransactions(CBlockIndex* pindexStart, bool fUpdate) |
e89b9f6a PW |
394 | { |
395 | int ret = 0; | |
396 | ||
397 | CBlockIndex* pindex = pindexStart; | |
398 | CRITICAL_BLOCK(cs_mapWallet) | |
399 | { | |
400 | while (pindex) | |
401 | { | |
402 | CBlock block; | |
403 | block.ReadFromDisk(pindex, true); | |
404 | BOOST_FOREACH(CTransaction& tx, block.vtx) | |
405 | { | |
406 | if (AddToWalletIfInvolvingMe(tx, &block, fUpdate)) | |
407 | ret++; | |
408 | } | |
409 | pindex = pindex->pnext; | |
410 | } | |
411 | } | |
412 | return ret; | |
413 | } | |
414 | ||
64c7ee7e | 415 | void CWallet::ReacceptWalletTransactions() |
e89b9f6a PW |
416 | { |
417 | CTxDB txdb("r"); | |
418 | bool fRepeat = true; | |
419 | while (fRepeat) CRITICAL_BLOCK(cs_mapWallet) | |
420 | { | |
421 | fRepeat = false; | |
422 | vector<CDiskTxPos> vMissingTx; | |
423 | BOOST_FOREACH(PAIRTYPE(const uint256, CWalletTx)& item, mapWallet) | |
424 | { | |
425 | CWalletTx& wtx = item.second; | |
426 | if (wtx.IsCoinBase() && wtx.IsSpent(0)) | |
427 | continue; | |
428 | ||
429 | CTxIndex txindex; | |
430 | bool fUpdated = false; | |
431 | if (txdb.ReadTxIndex(wtx.GetHash(), txindex)) | |
432 | { | |
433 | // Update fSpent if a tx got spent somewhere else by a copy of wallet.dat | |
434 | if (txindex.vSpent.size() != wtx.vout.size()) | |
435 | { | |
436 | printf("ERROR: ReacceptWalletTransactions() : txindex.vSpent.size() %d != wtx.vout.size() %d\n", txindex.vSpent.size(), wtx.vout.size()); | |
437 | continue; | |
438 | } | |
439 | for (int i = 0; i < txindex.vSpent.size(); i++) | |
440 | { | |
441 | if (wtx.IsSpent(i)) | |
442 | continue; | |
64c7ee7e | 443 | if (!txindex.vSpent[i].IsNull() && IsMine(wtx.vout[i])) |
e89b9f6a PW |
444 | { |
445 | wtx.MarkSpent(i); | |
446 | fUpdated = true; | |
447 | vMissingTx.push_back(txindex.vSpent[i]); | |
448 | } | |
449 | } | |
450 | if (fUpdated) | |
451 | { | |
452 | printf("ReacceptWalletTransactions found spent coin %sbc %s\n", FormatMoney(wtx.GetCredit()).c_str(), wtx.GetHash().ToString().c_str()); | |
453 | wtx.MarkDirty(); | |
454 | wtx.WriteToDisk(); | |
455 | } | |
456 | } | |
457 | else | |
458 | { | |
459 | // Reaccept any txes of ours that aren't already in a block | |
460 | if (!wtx.IsCoinBase()) | |
461 | wtx.AcceptWalletTransaction(txdb, false); | |
462 | } | |
463 | } | |
464 | if (!vMissingTx.empty()) | |
465 | { | |
466 | // TODO: optimize this to scan just part of the block chain? | |
467 | if (ScanForWalletTransactions(pindexGenesisBlock)) | |
468 | fRepeat = true; // Found missing transactions: re-do Reaccept. | |
469 | } | |
470 | } | |
471 | } | |
472 | ||
473 | void CWalletTx::RelayWalletTransaction(CTxDB& txdb) | |
474 | { | |
475 | BOOST_FOREACH(const CMerkleTx& tx, vtxPrev) | |
476 | { | |
477 | if (!tx.IsCoinBase()) | |
478 | { | |
479 | uint256 hash = tx.GetHash(); | |
480 | if (!txdb.ContainsTx(hash)) | |
481 | RelayMessage(CInv(MSG_TX, hash), (CTransaction)tx); | |
482 | } | |
483 | } | |
484 | if (!IsCoinBase()) | |
485 | { | |
486 | uint256 hash = GetHash(); | |
487 | if (!txdb.ContainsTx(hash)) | |
488 | { | |
489 | printf("Relaying wtx %s\n", hash.ToString().substr(0,10).c_str()); | |
490 | RelayMessage(CInv(MSG_TX, hash), (CTransaction)*this); | |
491 | } | |
492 | } | |
493 | } | |
494 | ||
495 | void CWalletTx::RelayWalletTransaction() | |
496 | { | |
497 | CTxDB txdb("r"); | |
498 | RelayWalletTransaction(txdb); | |
499 | } | |
500 | ||
64c7ee7e | 501 | void CWallet::ResendWalletTransactions() |
e89b9f6a PW |
502 | { |
503 | // Do this infrequently and randomly to avoid giving away | |
504 | // that these are our transactions. | |
505 | static int64 nNextTime; | |
506 | if (GetTime() < nNextTime) | |
507 | return; | |
508 | bool fFirst = (nNextTime == 0); | |
509 | nNextTime = GetTime() + GetRand(30 * 60); | |
510 | if (fFirst) | |
511 | return; | |
512 | ||
513 | // Only do it if there's been a new block since last time | |
514 | static int64 nLastTime; | |
515 | if (nTimeBestReceived < nLastTime) | |
516 | return; | |
517 | nLastTime = GetTime(); | |
518 | ||
519 | // Rebroadcast any of our txes that aren't in a block yet | |
520 | printf("ResendWalletTransactions()\n"); | |
521 | CTxDB txdb("r"); | |
522 | CRITICAL_BLOCK(cs_mapWallet) | |
523 | { | |
524 | // Sort them in chronological order | |
525 | multimap<unsigned int, CWalletTx*> mapSorted; | |
526 | BOOST_FOREACH(PAIRTYPE(const uint256, CWalletTx)& item, mapWallet) | |
527 | { | |
528 | CWalletTx& wtx = item.second; | |
529 | // Don't rebroadcast until it's had plenty of time that | |
530 | // it should have gotten in already by now. | |
531 | if (nTimeBestReceived - (int64)wtx.nTimeReceived > 5 * 60) | |
532 | mapSorted.insert(make_pair(wtx.nTimeReceived, &wtx)); | |
533 | } | |
534 | BOOST_FOREACH(PAIRTYPE(const unsigned int, CWalletTx*)& item, mapSorted) | |
535 | { | |
536 | CWalletTx& wtx = *item.second; | |
537 | wtx.RelayWalletTransaction(txdb); | |
538 | } | |
539 | } | |
540 | } | |
541 | ||
542 | ||
543 | ||
544 | ||
545 | ||
546 | ||
547 | ////////////////////////////////////////////////////////////////////////////// | |
548 | // | |
549 | // Actions | |
550 | // | |
551 | ||
552 | ||
64c7ee7e | 553 | int64 CWallet::GetBalance() const |
e89b9f6a PW |
554 | { |
555 | int64 nStart = GetTimeMillis(); | |
556 | ||
557 | int64 nTotal = 0; | |
558 | CRITICAL_BLOCK(cs_mapWallet) | |
559 | { | |
64c7ee7e | 560 | for (map<uint256, CWalletTx>::const_iterator it = mapWallet.begin(); it != mapWallet.end(); ++it) |
e89b9f6a | 561 | { |
64c7ee7e | 562 | const CWalletTx* pcoin = &(*it).second; |
e89b9f6a PW |
563 | if (!pcoin->IsFinal() || !pcoin->IsConfirmed()) |
564 | continue; | |
565 | nTotal += pcoin->GetAvailableCredit(); | |
566 | } | |
567 | } | |
568 | ||
569 | //printf("GetBalance() %"PRI64d"ms\n", GetTimeMillis() - nStart); | |
570 | return nTotal; | |
571 | } | |
572 | ||
573 | ||
64c7ee7e | 574 | bool CWallet::SelectCoinsMinConf(int64 nTargetValue, int nConfMine, int nConfTheirs, set<pair<const CWalletTx*,unsigned int> >& setCoinsRet, int64& nValueRet) const |
e89b9f6a PW |
575 | { |
576 | setCoinsRet.clear(); | |
577 | nValueRet = 0; | |
578 | ||
579 | // List of values less than target | |
64c7ee7e | 580 | pair<int64, pair<const CWalletTx*,unsigned int> > coinLowestLarger; |
e89b9f6a PW |
581 | coinLowestLarger.first = INT64_MAX; |
582 | coinLowestLarger.second.first = NULL; | |
64c7ee7e | 583 | vector<pair<int64, pair<const CWalletTx*,unsigned int> > > vValue; |
e89b9f6a PW |
584 | int64 nTotalLower = 0; |
585 | ||
586 | CRITICAL_BLOCK(cs_mapWallet) | |
587 | { | |
64c7ee7e | 588 | vector<const CWalletTx*> vCoins; |
e89b9f6a | 589 | vCoins.reserve(mapWallet.size()); |
64c7ee7e | 590 | for (map<uint256, CWalletTx>::const_iterator it = mapWallet.begin(); it != mapWallet.end(); ++it) |
e89b9f6a PW |
591 | vCoins.push_back(&(*it).second); |
592 | random_shuffle(vCoins.begin(), vCoins.end(), GetRandInt); | |
593 | ||
64c7ee7e | 594 | BOOST_FOREACH(const CWalletTx* pcoin, vCoins) |
e89b9f6a PW |
595 | { |
596 | if (!pcoin->IsFinal() || !pcoin->IsConfirmed()) | |
597 | continue; | |
598 | ||
599 | if (pcoin->IsCoinBase() && pcoin->GetBlocksToMaturity() > 0) | |
600 | continue; | |
601 | ||
602 | int nDepth = pcoin->GetDepthInMainChain(); | |
603 | if (nDepth < (pcoin->IsFromMe() ? nConfMine : nConfTheirs)) | |
604 | continue; | |
605 | ||
606 | for (int i = 0; i < pcoin->vout.size(); i++) | |
607 | { | |
64c7ee7e | 608 | if (pcoin->IsSpent(i) || !IsMine(pcoin->vout[i])) |
e89b9f6a PW |
609 | continue; |
610 | ||
611 | int64 n = pcoin->vout[i].nValue; | |
612 | ||
613 | if (n <= 0) | |
614 | continue; | |
615 | ||
64c7ee7e | 616 | pair<int64,pair<const CWalletTx*,unsigned int> > coin = make_pair(n,make_pair(pcoin,i)); |
e89b9f6a PW |
617 | |
618 | if (n == nTargetValue) | |
619 | { | |
620 | setCoinsRet.insert(coin.second); | |
621 | nValueRet += coin.first; | |
622 | return true; | |
623 | } | |
624 | else if (n < nTargetValue + CENT) | |
625 | { | |
626 | vValue.push_back(coin); | |
627 | nTotalLower += n; | |
628 | } | |
629 | else if (n < coinLowestLarger.first) | |
630 | { | |
631 | coinLowestLarger = coin; | |
632 | } | |
633 | } | |
634 | } | |
635 | } | |
636 | ||
637 | if (nTotalLower == nTargetValue || nTotalLower == nTargetValue + CENT) | |
638 | { | |
639 | for (int i = 0; i < vValue.size(); ++i) | |
640 | { | |
641 | setCoinsRet.insert(vValue[i].second); | |
642 | nValueRet += vValue[i].first; | |
643 | } | |
644 | return true; | |
645 | } | |
646 | ||
647 | if (nTotalLower < nTargetValue + (coinLowestLarger.second.first ? CENT : 0)) | |
648 | { | |
649 | if (coinLowestLarger.second.first == NULL) | |
650 | return false; | |
651 | setCoinsRet.insert(coinLowestLarger.second); | |
652 | nValueRet += coinLowestLarger.first; | |
653 | return true; | |
654 | } | |
655 | ||
656 | if (nTotalLower >= nTargetValue + CENT) | |
657 | nTargetValue += CENT; | |
658 | ||
659 | // Solve subset sum by stochastic approximation | |
660 | sort(vValue.rbegin(), vValue.rend()); | |
661 | vector<char> vfIncluded; | |
662 | vector<char> vfBest(vValue.size(), true); | |
663 | int64 nBest = nTotalLower; | |
664 | ||
665 | for (int nRep = 0; nRep < 1000 && nBest != nTargetValue; nRep++) | |
666 | { | |
667 | vfIncluded.assign(vValue.size(), false); | |
668 | int64 nTotal = 0; | |
669 | bool fReachedTarget = false; | |
670 | for (int nPass = 0; nPass < 2 && !fReachedTarget; nPass++) | |
671 | { | |
672 | for (int i = 0; i < vValue.size(); i++) | |
673 | { | |
674 | if (nPass == 0 ? rand() % 2 : !vfIncluded[i]) | |
675 | { | |
676 | nTotal += vValue[i].first; | |
677 | vfIncluded[i] = true; | |
678 | if (nTotal >= nTargetValue) | |
679 | { | |
680 | fReachedTarget = true; | |
681 | if (nTotal < nBest) | |
682 | { | |
683 | nBest = nTotal; | |
684 | vfBest = vfIncluded; | |
685 | } | |
686 | nTotal -= vValue[i].first; | |
687 | vfIncluded[i] = false; | |
688 | } | |
689 | } | |
690 | } | |
691 | } | |
692 | } | |
693 | ||
694 | // If the next larger is still closer, return it | |
695 | if (coinLowestLarger.second.first && coinLowestLarger.first - nTargetValue <= nBest - nTargetValue) | |
696 | { | |
697 | setCoinsRet.insert(coinLowestLarger.second); | |
698 | nValueRet += coinLowestLarger.first; | |
699 | } | |
700 | else { | |
701 | for (int i = 0; i < vValue.size(); i++) | |
702 | if (vfBest[i]) | |
703 | { | |
704 | setCoinsRet.insert(vValue[i].second); | |
705 | nValueRet += vValue[i].first; | |
706 | } | |
707 | ||
708 | //// debug print | |
709 | printf("SelectCoins() best subset: "); | |
710 | for (int i = 0; i < vValue.size(); i++) | |
711 | if (vfBest[i]) | |
712 | printf("%s ", FormatMoney(vValue[i].first).c_str()); | |
713 | printf("total %s\n", FormatMoney(nBest).c_str()); | |
714 | } | |
715 | ||
716 | return true; | |
717 | } | |
718 | ||
64c7ee7e | 719 | bool CWallet::SelectCoins(int64 nTargetValue, set<pair<const CWalletTx*,unsigned int> >& setCoinsRet, int64& nValueRet) const |
e89b9f6a PW |
720 | { |
721 | return (SelectCoinsMinConf(nTargetValue, 1, 6, setCoinsRet, nValueRet) || | |
722 | SelectCoinsMinConf(nTargetValue, 1, 1, setCoinsRet, nValueRet) || | |
723 | SelectCoinsMinConf(nTargetValue, 0, 1, setCoinsRet, nValueRet)); | |
724 | } | |
725 | ||
726 | ||
727 | ||
728 | ||
64c7ee7e | 729 | bool CWallet::CreateTransaction(const vector<pair<CScript, int64> >& vecSend, CWalletTx& wtxNew, CReserveKey& reservekey, int64& nFeeRet) |
e89b9f6a PW |
730 | { |
731 | int64 nValue = 0; | |
732 | BOOST_FOREACH (const PAIRTYPE(CScript, int64)& s, vecSend) | |
733 | { | |
734 | if (nValue < 0) | |
735 | return false; | |
736 | nValue += s.second; | |
737 | } | |
738 | if (vecSend.empty() || nValue < 0) | |
739 | return false; | |
740 | ||
64c7ee7e PW |
741 | wtxNew.pwallet = this; |
742 | ||
e89b9f6a PW |
743 | CRITICAL_BLOCK(cs_main) |
744 | { | |
745 | // txdb must be opened before the mapWallet lock | |
746 | CTxDB txdb("r"); | |
747 | CRITICAL_BLOCK(cs_mapWallet) | |
748 | { | |
749 | nFeeRet = nTransactionFee; | |
750 | loop | |
751 | { | |
752 | wtxNew.vin.clear(); | |
753 | wtxNew.vout.clear(); | |
754 | wtxNew.fFromMe = true; | |
755 | ||
756 | int64 nTotalValue = nValue + nFeeRet; | |
757 | double dPriority = 0; | |
758 | // vouts to the payees | |
759 | BOOST_FOREACH (const PAIRTYPE(CScript, int64)& s, vecSend) | |
760 | wtxNew.vout.push_back(CTxOut(s.second, s.first)); | |
761 | ||
762 | // Choose coins to use | |
64c7ee7e | 763 | set<pair<const CWalletTx*,unsigned int> > setCoins; |
e89b9f6a PW |
764 | int64 nValueIn = 0; |
765 | if (!SelectCoins(nTotalValue, setCoins, nValueIn)) | |
766 | return false; | |
64c7ee7e | 767 | BOOST_FOREACH(PAIRTYPE(const CWalletTx*, unsigned int) pcoin, setCoins) |
e89b9f6a PW |
768 | { |
769 | int64 nCredit = pcoin.first->vout[pcoin.second].nValue; | |
770 | dPriority += (double)nCredit * pcoin.first->GetDepthInMainChain(); | |
771 | } | |
772 | ||
773 | // Fill a vout back to self with any change | |
774 | int64 nChange = nValueIn - nTotalValue; | |
775 | if (nChange >= CENT) | |
776 | { | |
777 | // Note: We use a new key here to keep it from being obvious which side is the change. | |
778 | // The drawback is that by not reusing a previous key, the change may be lost if a | |
779 | // backup is restored, if the backup doesn't have the new private key for the change. | |
780 | // If we reused the old key, it would be possible to add code to look for and | |
781 | // rediscover unknown transactions that were written with keys of ours to recover | |
782 | // post-backup change. | |
783 | ||
784 | // Reserve a new key pair from key pool | |
785 | vector<unsigned char> vchPubKey = reservekey.GetReservedKey(); | |
786 | assert(mapKeys.count(vchPubKey)); | |
787 | ||
788 | // Fill a vout to ourself, using same address type as the payment | |
789 | CScript scriptChange; | |
790 | if (vecSend[0].first.GetBitcoinAddressHash160() != 0) | |
791 | scriptChange.SetBitcoinAddress(vchPubKey); | |
792 | else | |
793 | scriptChange << vchPubKey << OP_CHECKSIG; | |
794 | ||
795 | // Insert change txn at random position: | |
796 | vector<CTxOut>::iterator position = wtxNew.vout.begin()+GetRandInt(wtxNew.vout.size()); | |
797 | wtxNew.vout.insert(position, CTxOut(nChange, scriptChange)); | |
798 | } | |
799 | else | |
800 | reservekey.ReturnKey(); | |
801 | ||
802 | // Fill vin | |
64c7ee7e | 803 | BOOST_FOREACH(const PAIRTYPE(const CWalletTx*,unsigned int)& coin, setCoins) |
e89b9f6a PW |
804 | wtxNew.vin.push_back(CTxIn(coin.first->GetHash(),coin.second)); |
805 | ||
806 | // Sign | |
807 | int nIn = 0; | |
64c7ee7e PW |
808 | BOOST_FOREACH(const PAIRTYPE(const CWalletTx*,unsigned int)& coin, setCoins) |
809 | if (!SignSignature(*this, *coin.first, wtxNew, nIn++)) | |
e89b9f6a PW |
810 | return false; |
811 | ||
812 | // Limit size | |
813 | unsigned int nBytes = ::GetSerializeSize(*(CTransaction*)&wtxNew, SER_NETWORK); | |
814 | if (nBytes >= MAX_BLOCK_SIZE_GEN/5) | |
815 | return false; | |
816 | dPriority /= nBytes; | |
817 | ||
818 | // Check that enough fee is included | |
819 | int64 nPayFee = nTransactionFee * (1 + (int64)nBytes / 1000); | |
820 | bool fAllowFree = CTransaction::AllowFree(dPriority); | |
821 | int64 nMinFee = wtxNew.GetMinFee(1, fAllowFree); | |
822 | if (nFeeRet < max(nPayFee, nMinFee)) | |
823 | { | |
824 | nFeeRet = max(nPayFee, nMinFee); | |
825 | continue; | |
826 | } | |
827 | ||
828 | // Fill vtxPrev by copying from previous transactions vtxPrev | |
829 | wtxNew.AddSupportingTransactions(txdb); | |
830 | wtxNew.fTimeReceivedIsTxTime = true; | |
831 | ||
832 | break; | |
833 | } | |
834 | } | |
835 | } | |
836 | return true; | |
837 | } | |
838 | ||
64c7ee7e | 839 | bool CWallet::CreateTransaction(CScript scriptPubKey, int64 nValue, CWalletTx& wtxNew, CReserveKey& reservekey, int64& nFeeRet) |
e89b9f6a PW |
840 | { |
841 | vector< pair<CScript, int64> > vecSend; | |
842 | vecSend.push_back(make_pair(scriptPubKey, nValue)); | |
843 | return CreateTransaction(vecSend, wtxNew, reservekey, nFeeRet); | |
844 | } | |
845 | ||
846 | // Call after CreateTransaction unless you want to abort | |
64c7ee7e | 847 | bool CWallet::CommitTransaction(CWalletTx& wtxNew, CReserveKey& reservekey) |
e89b9f6a PW |
848 | { |
849 | CRITICAL_BLOCK(cs_main) | |
850 | { | |
851 | printf("CommitTransaction:\n%s", wtxNew.ToString().c_str()); | |
852 | CRITICAL_BLOCK(cs_mapWallet) | |
853 | { | |
854 | // This is only to keep the database open to defeat the auto-flush for the | |
855 | // duration of this scope. This is the only place where this optimization | |
856 | // maybe makes sense; please don't do it anywhere else. | |
64c7ee7e | 857 | CWalletDB* pwalletdb = fFileBacked ? new CWalletDB(strWalletFile,"r") : NULL; |
e89b9f6a PW |
858 | |
859 | // Take key pair from key pool so it won't be used again | |
860 | reservekey.KeepKey(); | |
861 | ||
862 | // Add tx to wallet, because if it has change it's also ours, | |
863 | // otherwise just for transaction history. | |
864 | AddToWallet(wtxNew); | |
865 | ||
866 | // Mark old coins as spent | |
867 | set<CWalletTx*> setCoins; | |
868 | BOOST_FOREACH(const CTxIn& txin, wtxNew.vin) | |
869 | { | |
64c7ee7e PW |
870 | CWalletTx &coin = mapWallet[txin.prevout.hash]; |
871 | coin.pwallet = this; | |
872 | coin.MarkSpent(txin.prevout.n); | |
873 | coin.WriteToDisk(); | |
874 | vWalletUpdated.push_back(coin.GetHash()); | |
e89b9f6a | 875 | } |
64c7ee7e PW |
876 | |
877 | if (fFileBacked) | |
878 | delete pwalletdb; | |
e89b9f6a PW |
879 | } |
880 | ||
881 | // Track how many getdata requests our transaction gets | |
882 | CRITICAL_BLOCK(cs_mapRequestCount) | |
883 | mapRequestCount[wtxNew.GetHash()] = 0; | |
884 | ||
885 | // Broadcast | |
886 | if (!wtxNew.AcceptToMemoryPool()) | |
887 | { | |
888 | // This must not fail. The transaction has already been signed and recorded. | |
889 | printf("CommitTransaction() : Error: Transaction not valid"); | |
890 | return false; | |
891 | } | |
892 | wtxNew.RelayWalletTransaction(); | |
893 | } | |
894 | MainFrameRepaint(); | |
895 | return true; | |
896 | } | |
897 | ||
898 | ||
899 | ||
900 | ||
901 | // requires cs_main lock | |
64c7ee7e | 902 | string CWallet::SendMoney(CScript scriptPubKey, int64 nValue, CWalletTx& wtxNew, bool fAskFee) |
e89b9f6a | 903 | { |
64c7ee7e | 904 | CReserveKey reservekey(this); |
e89b9f6a PW |
905 | int64 nFeeRequired; |
906 | if (!CreateTransaction(scriptPubKey, nValue, wtxNew, reservekey, nFeeRequired)) | |
907 | { | |
908 | string strError; | |
909 | if (nValue + nFeeRequired > GetBalance()) | |
910 | strError = strprintf(_("Error: This transaction requires a transaction fee of at least %s because of its amount, complexity, or use of recently received funds "), FormatMoney(nFeeRequired).c_str()); | |
911 | else | |
912 | strError = _("Error: Transaction creation failed "); | |
913 | printf("SendMoney() : %s", strError.c_str()); | |
914 | return strError; | |
915 | } | |
916 | ||
917 | if (fAskFee && !ThreadSafeAskFee(nFeeRequired, _("Sending..."), NULL)) | |
918 | return "ABORTED"; | |
919 | ||
920 | if (!CommitTransaction(wtxNew, reservekey)) | |
921 | return _("Error: The transaction was rejected. This might happen if some of the coins in your wallet were already spent, such as if you used a copy of wallet.dat and coins were spent in the copy but not marked as spent here."); | |
922 | ||
923 | MainFrameRepaint(); | |
924 | return ""; | |
925 | } | |
926 | ||
927 | ||
928 | ||
929 | // requires cs_main lock | |
64c7ee7e | 930 | string CWallet::SendMoneyToBitcoinAddress(string strAddress, int64 nValue, CWalletTx& wtxNew, bool fAskFee) |
e89b9f6a PW |
931 | { |
932 | // Check amount | |
933 | if (nValue <= 0) | |
934 | return _("Invalid amount"); | |
935 | if (nValue + nTransactionFee > GetBalance()) | |
936 | return _("Insufficient funds"); | |
937 | ||
938 | // Parse bitcoin address | |
939 | CScript scriptPubKey; | |
940 | if (!scriptPubKey.SetBitcoinAddress(strAddress)) | |
941 | return _("Invalid bitcoin address"); | |
942 | ||
943 | return SendMoney(scriptPubKey, nValue, wtxNew, fAskFee); | |
944 | } | |
945 | ||
946 | ||
947 | ||
948 | ||
64c7ee7e | 949 | bool CWallet::LoadWallet(bool& fFirstRunRet) |
e89b9f6a | 950 | { |
64c7ee7e PW |
951 | if (!fFileBacked) |
952 | return false; | |
e89b9f6a | 953 | fFirstRunRet = false; |
64c7ee7e | 954 | if (!CWalletDB(strWalletFile,"cr+").LoadWallet(this)) |
e89b9f6a PW |
955 | return false; |
956 | fFirstRunRet = vchDefaultKey.empty(); | |
957 | ||
d3800d59 | 958 | if (!mapKeys.count(vchDefaultKey)) |
e89b9f6a | 959 | { |
d3800d59 | 960 | // Create new default key |
e89b9f6a PW |
961 | RandAddSeedPerfmon(); |
962 | ||
e9f61c87 | 963 | SetDefaultKey(GetKeyFromKeyPool()); |
64c7ee7e PW |
964 | if (!SetAddressBookName(PubKeyToAddress(vchDefaultKey), "")) |
965 | return false; | |
e89b9f6a PW |
966 | } |
967 | ||
64c7ee7e | 968 | CreateThread(ThreadFlushWalletDB, &strWalletFile); |
e89b9f6a PW |
969 | return true; |
970 | } | |
971 | ||
4d410cfc SG |
972 | |
973 | bool CWallet::SetAddressBookName(const string& strAddress, const string& strName) | |
974 | { | |
975 | mapAddressBook[strAddress] = strName; | |
976 | if (!fFileBacked) | |
977 | return false; | |
978 | return CWalletDB(strWalletFile).WriteName(strAddress, strName); | |
979 | } | |
980 | ||
981 | bool CWallet::DelAddressBookName(const string& strAddress) | |
982 | { | |
983 | mapAddressBook.erase(strAddress); | |
984 | if (!fFileBacked) | |
985 | return false; | |
986 | return CWalletDB(strWalletFile).EraseName(strAddress); | |
987 | } | |
988 | ||
989 | ||
64c7ee7e PW |
990 | void CWallet::PrintWallet(const CBlock& block) |
991 | { | |
992 | CRITICAL_BLOCK(cs_mapWallet) | |
993 | { | |
994 | if (mapWallet.count(block.vtx[0].GetHash())) | |
995 | { | |
996 | CWalletTx& wtx = mapWallet[block.vtx[0].GetHash()]; | |
997 | printf(" mine: %d %d %d", wtx.GetDepthInMainChain(), wtx.GetBlocksToMaturity(), wtx.GetCredit()); | |
998 | } | |
999 | } | |
1000 | printf("\n"); | |
1001 | } | |
1002 | ||
1003 | bool CWallet::GetTransaction(const uint256 &hashTx, CWalletTx& wtx) | |
1004 | { | |
1005 | CRITICAL_BLOCK(cs_mapWallet) | |
1006 | { | |
1007 | map<uint256, CWalletTx>::iterator mi = mapWallet.find(hashTx); | |
1008 | if (mi != mapWallet.end()) | |
1009 | { | |
1010 | wtx = (*mi).second; | |
1011 | return true; | |
1012 | } | |
64c7ee7e | 1013 | } |
98705aa5 | 1014 | return false; |
64c7ee7e PW |
1015 | } |
1016 | ||
e9f61c87 PW |
1017 | bool CWallet::SetDefaultKey(const std::vector<unsigned char> &vchPubKey) |
1018 | { | |
1019 | if (fFileBacked) | |
1020 | { | |
1021 | if (!CWalletDB(strWalletFile).WriteDefaultKey(vchPubKey)) | |
1022 | return false; | |
1023 | } | |
1024 | vchDefaultKey = vchPubKey; | |
1025 | return true; | |
1026 | } | |
1027 | ||
64c7ee7e PW |
1028 | bool GetWalletFile(CWallet* pwallet, string &strWalletFileOut) |
1029 | { | |
1030 | if (!pwallet->fFileBacked) | |
1031 | return false; | |
1032 | strWalletFileOut = pwallet->strWalletFile; | |
1033 | return true; | |
1034 | } | |
1035 | ||
1036 | void CWallet::ReserveKeyFromKeyPool(int64& nIndex, CKeyPool& keypool) | |
e89b9f6a PW |
1037 | { |
1038 | nIndex = -1; | |
1039 | keypool.vchPubKey.clear(); | |
1040 | CRITICAL_BLOCK(cs_main) | |
1041 | CRITICAL_BLOCK(cs_mapWallet) | |
1042 | CRITICAL_BLOCK(cs_setKeyPool) | |
1043 | { | |
64c7ee7e PW |
1044 | CWalletDB walletdb(strWalletFile); |
1045 | ||
e89b9f6a PW |
1046 | // Top up key pool |
1047 | int64 nTargetSize = max(GetArg("-keypool", 100), (int64)0); | |
1048 | while (setKeyPool.size() < nTargetSize+1) | |
1049 | { | |
1050 | int64 nEnd = 1; | |
1051 | if (!setKeyPool.empty()) | |
1052 | nEnd = *(--setKeyPool.end()) + 1; | |
64c7ee7e | 1053 | if (!walletdb.WritePool(nEnd, CKeyPool(GenerateNewKey()))) |
e89b9f6a PW |
1054 | throw runtime_error("ReserveKeyFromKeyPool() : writing generated key failed"); |
1055 | setKeyPool.insert(nEnd); | |
1056 | printf("keypool added key %"PRI64d", size=%d\n", nEnd, setKeyPool.size()); | |
1057 | } | |
1058 | ||
1059 | // Get the oldest key | |
1060 | assert(!setKeyPool.empty()); | |
1061 | nIndex = *(setKeyPool.begin()); | |
1062 | setKeyPool.erase(setKeyPool.begin()); | |
64c7ee7e | 1063 | if (!walletdb.ReadPool(nIndex, keypool)) |
e89b9f6a PW |
1064 | throw runtime_error("ReserveKeyFromKeyPool() : read failed"); |
1065 | if (!mapKeys.count(keypool.vchPubKey)) | |
1066 | throw runtime_error("ReserveKeyFromKeyPool() : unknown key in key pool"); | |
1067 | assert(!keypool.vchPubKey.empty()); | |
1068 | printf("keypool reserve %"PRI64d"\n", nIndex); | |
1069 | } | |
1070 | } | |
1071 | ||
64c7ee7e | 1072 | void CWallet::KeepKey(int64 nIndex) |
e89b9f6a PW |
1073 | { |
1074 | // Remove from key pool | |
64c7ee7e | 1075 | if (fFileBacked) |
e89b9f6a | 1076 | { |
64c7ee7e PW |
1077 | CWalletDB walletdb(strWalletFile); |
1078 | CRITICAL_BLOCK(cs_main) | |
1079 | { | |
1080 | walletdb.ErasePool(nIndex); | |
1081 | } | |
e89b9f6a PW |
1082 | } |
1083 | printf("keypool keep %"PRI64d"\n", nIndex); | |
1084 | } | |
1085 | ||
64c7ee7e | 1086 | void CWallet::ReturnKey(int64 nIndex) |
e89b9f6a PW |
1087 | { |
1088 | // Return to key pool | |
1089 | CRITICAL_BLOCK(cs_setKeyPool) | |
1090 | setKeyPool.insert(nIndex); | |
1091 | printf("keypool return %"PRI64d"\n", nIndex); | |
1092 | } | |
1093 | ||
64c7ee7e | 1094 | vector<unsigned char> CWallet::GetKeyFromKeyPool() |
e89b9f6a | 1095 | { |
e89b9f6a PW |
1096 | int64 nIndex = 0; |
1097 | CKeyPool keypool; | |
64c7ee7e PW |
1098 | ReserveKeyFromKeyPool(nIndex, keypool); |
1099 | KeepKey(nIndex); | |
e89b9f6a PW |
1100 | return keypool.vchPubKey; |
1101 | } | |
1102 | ||
64c7ee7e | 1103 | int64 CWallet::GetOldestKeyPoolTime() |
e89b9f6a | 1104 | { |
e89b9f6a PW |
1105 | int64 nIndex = 0; |
1106 | CKeyPool keypool; | |
64c7ee7e PW |
1107 | ReserveKeyFromKeyPool(nIndex, keypool); |
1108 | ReturnKey(nIndex); | |
e89b9f6a PW |
1109 | return keypool.nTime; |
1110 | } | |
1111 | ||
64c7ee7e | 1112 | vector<unsigned char> CReserveKey::GetReservedKey() |
e89b9f6a PW |
1113 | { |
1114 | if (nIndex == -1) | |
1115 | { | |
1116 | CKeyPool keypool; | |
64c7ee7e | 1117 | pwallet->ReserveKeyFromKeyPool(nIndex, keypool); |
e89b9f6a PW |
1118 | vchPubKey = keypool.vchPubKey; |
1119 | } | |
1120 | assert(!vchPubKey.empty()); | |
1121 | return vchPubKey; | |
1122 | } | |
1123 | ||
1124 | void CReserveKey::KeepKey() | |
1125 | { | |
1126 | if (nIndex != -1) | |
64c7ee7e | 1127 | pwallet->KeepKey(nIndex); |
e89b9f6a PW |
1128 | nIndex = -1; |
1129 | vchPubKey.clear(); | |
1130 | } | |
1131 | ||
1132 | void CReserveKey::ReturnKey() | |
1133 | { | |
1134 | if (nIndex != -1) | |
64c7ee7e | 1135 | pwallet->ReturnKey(nIndex); |
e89b9f6a PW |
1136 | nIndex = -1; |
1137 | vchPubKey.clear(); | |
1138 | } | |
e9f61c87 | 1139 |