1 // Copyright (c) 2009-2010 Satoshi Nakamoto
2 // Distributed under the MIT/X11 software license, see the accompanying
3 // file license.txt or http://www.opensource.org/licenses/mit-license.php.
7 void ThreadFlushWalletDB(void* parg
);
10 unsigned int nWalletDBUpdated
;
11 uint64 nAccountingEntryNumber
= 0;
19 static CCriticalSection cs_db
;
20 static bool fDbEnvInit
= false;
22 static map
<string
, int> mapFileUseCount
;
23 static map
<string
, Db
*> mapDb
;
43 CDB::CDB(const char* pszFile
, const char* pszMode
) : pdb(NULL
)
49 fReadOnly
= (!strchr(pszMode
, '+') && !strchr(pszMode
, 'w'));
50 bool fCreate
= strchr(pszMode
, 'c');
51 unsigned int nFlags
= DB_THREAD
;
61 string strDataDir
= GetDataDir();
62 string strLogDir
= strDataDir
+ "/database";
63 filesystem::create_directory(strLogDir
.c_str());
64 string strErrorFile
= strDataDir
+ "/db.log";
65 printf("dbenv.open strLogDir=%s strErrorFile=%s\n", strLogDir
.c_str(), strErrorFile
.c_str());
67 dbenv
.set_lg_dir(strLogDir
.c_str());
68 dbenv
.set_lg_max(10000000);
69 dbenv
.set_lk_max_locks(10000);
70 dbenv
.set_lk_max_objects(10000);
71 dbenv
.set_errfile(fopen(strErrorFile
.c_str(), "a")); /// debug
72 dbenv
.set_flags(DB_AUTO_COMMIT
, 1);
73 ret
= dbenv
.open(strDataDir
.c_str(),
83 throw runtime_error(strprintf("CDB() : error %d opening database environment", ret
));
88 ++mapFileUseCount
[strFile
];
92 pdb
= new Db(&dbenv
, 0);
94 ret
= pdb
->open(NULL
, // Txn pointer
96 "main", // Logical db name
97 DB_BTREE
, // Database type
105 CRITICAL_BLOCK(cs_db
)
106 --mapFileUseCount
[strFile
];
108 throw runtime_error(strprintf("CDB() : can't open database file %s, error %d", pszFile
, ret
));
111 if (fCreate
&& !Exists(string("version")))
113 bool fTmp
= fReadOnly
;
115 WriteVersion(VERSION
);
119 mapDb
[strFile
] = pdb
;
129 vTxn
.front()->abort();
133 // Flush database activity from memory pool to disk log
134 unsigned int nMinutes
= 0;
137 if (strFile
== "addr.dat")
139 if (strFile
== "blkindex.dat" && IsInitialBlockDownload() && nBestHeight
% 500 != 0)
141 dbenv
.txn_checkpoint(0, nMinutes
, 0);
143 CRITICAL_BLOCK(cs_db
)
144 --mapFileUseCount
[strFile
];
147 void CloseDb(const string
& strFile
)
149 CRITICAL_BLOCK(cs_db
)
151 if (mapDb
[strFile
] != NULL
)
153 // Close the database handle
154 Db
* pdb
= mapDb
[strFile
];
157 mapDb
[strFile
] = NULL
;
162 void DBFlush(bool fShutdown
)
164 // Flush log data to the actual data file
165 // on all files that are not in use
166 printf("DBFlush(%s)%s\n", fShutdown
? "true" : "false", fDbEnvInit
? "" : " db not started");
169 CRITICAL_BLOCK(cs_db
)
171 map
<string
, int>::iterator mi
= mapFileUseCount
.begin();
172 while (mi
!= mapFileUseCount
.end())
174 string strFile
= (*mi
).first
;
175 int nRefCount
= (*mi
).second
;
176 printf("%s refcount=%d\n", strFile
.c_str(), nRefCount
);
179 // Move log data to the dat file
181 dbenv
.txn_checkpoint(0, 0, 0);
182 printf("%s flush\n", strFile
.c_str());
183 dbenv
.lsn_reset(strFile
.c_str(), 0);
184 mapFileUseCount
.erase(mi
++);
192 if (mapFileUseCount
.empty())
193 dbenv
.log_archive(&listp
, DB_ARCH_REMOVE
);
209 bool CTxDB::ReadTxIndex(uint256 hash
, CTxIndex
& txindex
)
213 return Read(make_pair(string("tx"), hash
), txindex
);
216 bool CTxDB::UpdateTxIndex(uint256 hash
, const CTxIndex
& txindex
)
219 return Write(make_pair(string("tx"), hash
), txindex
);
222 bool CTxDB::AddTxIndex(const CTransaction
& tx
, const CDiskTxPos
& pos
, int nHeight
)
227 uint256 hash
= tx
.GetHash();
228 CTxIndex
txindex(pos
, tx
.vout
.size());
229 return Write(make_pair(string("tx"), hash
), txindex
);
232 bool CTxDB::EraseTxIndex(const CTransaction
& tx
)
235 uint256 hash
= tx
.GetHash();
237 return Erase(make_pair(string("tx"), hash
));
240 bool CTxDB::ContainsTx(uint256 hash
)
243 return Exists(make_pair(string("tx"), hash
));
246 bool CTxDB::ReadOwnerTxes(uint160 hash160
, int nMinHeight
, vector
<CTransaction
>& vtx
)
252 Dbc
* pcursor
= GetCursor();
256 unsigned int fFlags
= DB_SET_RANGE
;
261 if (fFlags
== DB_SET_RANGE
)
262 ssKey
<< string("owner") << hash160
<< CDiskTxPos(0, 0, 0);
264 int ret
= ReadAtCursor(pcursor
, ssKey
, ssValue
, fFlags
);
266 if (ret
== DB_NOTFOUND
)
278 ssKey
>> strType
>> hashItem
>> pos
;
280 ssValue
>> nItemHeight
;
283 if (strType
!= "owner" || hashItem
!= hash160
)
285 if (nItemHeight
>= nMinHeight
)
287 vtx
.resize(vtx
.size()+1);
288 if (!vtx
.back().ReadFromDisk(pos
))
300 bool CTxDB::ReadDiskTx(uint256 hash
, CTransaction
& tx
, CTxIndex
& txindex
)
304 if (!ReadTxIndex(hash
, txindex
))
306 return (tx
.ReadFromDisk(txindex
.pos
));
309 bool CTxDB::ReadDiskTx(uint256 hash
, CTransaction
& tx
)
312 return ReadDiskTx(hash
, tx
, txindex
);
315 bool CTxDB::ReadDiskTx(COutPoint outpoint
, CTransaction
& tx
, CTxIndex
& txindex
)
317 return ReadDiskTx(outpoint
.hash
, tx
, txindex
);
320 bool CTxDB::ReadDiskTx(COutPoint outpoint
, CTransaction
& tx
)
323 return ReadDiskTx(outpoint
.hash
, tx
, txindex
);
326 bool CTxDB::WriteBlockIndex(const CDiskBlockIndex
& blockindex
)
328 return Write(make_pair(string("blockindex"), blockindex
.GetBlockHash()), blockindex
);
331 bool CTxDB::EraseBlockIndex(uint256 hash
)
333 return Erase(make_pair(string("blockindex"), hash
));
336 bool CTxDB::ReadHashBestChain(uint256
& hashBestChain
)
338 return Read(string("hashBestChain"), hashBestChain
);
341 bool CTxDB::WriteHashBestChain(uint256 hashBestChain
)
343 return Write(string("hashBestChain"), hashBestChain
);
346 bool CTxDB::ReadBestInvalidWork(CBigNum
& bnBestInvalidWork
)
348 return Read(string("bnBestInvalidWork"), bnBestInvalidWork
);
351 bool CTxDB::WriteBestInvalidWork(CBigNum bnBestInvalidWork
)
353 return Write(string("bnBestInvalidWork"), bnBestInvalidWork
);
356 CBlockIndex
* InsertBlockIndex(uint256 hash
)
362 map
<uint256
, CBlockIndex
*>::iterator mi
= mapBlockIndex
.find(hash
);
363 if (mi
!= mapBlockIndex
.end())
367 CBlockIndex
* pindexNew
= new CBlockIndex();
369 throw runtime_error("LoadBlockIndex() : new CBlockIndex failed");
370 mi
= mapBlockIndex
.insert(make_pair(hash
, pindexNew
)).first
;
371 pindexNew
->phashBlock
= &((*mi
).first
);
376 bool CTxDB::LoadBlockIndex()
378 // Get database cursor
379 Dbc
* pcursor
= GetCursor();
383 // Load mapBlockIndex
384 unsigned int fFlags
= DB_SET_RANGE
;
389 if (fFlags
== DB_SET_RANGE
)
390 ssKey
<< make_pair(string("blockindex"), uint256(0));
392 int ret
= ReadAtCursor(pcursor
, ssKey
, ssValue
, fFlags
);
394 if (ret
== DB_NOTFOUND
)
402 if (strType
== "blockindex")
404 CDiskBlockIndex diskindex
;
405 ssValue
>> diskindex
;
407 // Construct block index object
408 CBlockIndex
* pindexNew
= InsertBlockIndex(diskindex
.GetBlockHash());
409 pindexNew
->pprev
= InsertBlockIndex(diskindex
.hashPrev
);
410 pindexNew
->pnext
= InsertBlockIndex(diskindex
.hashNext
);
411 pindexNew
->nFile
= diskindex
.nFile
;
412 pindexNew
->nBlockPos
= diskindex
.nBlockPos
;
413 pindexNew
->nHeight
= diskindex
.nHeight
;
414 pindexNew
->nVersion
= diskindex
.nVersion
;
415 pindexNew
->hashMerkleRoot
= diskindex
.hashMerkleRoot
;
416 pindexNew
->nTime
= diskindex
.nTime
;
417 pindexNew
->nBits
= diskindex
.nBits
;
418 pindexNew
->nNonce
= diskindex
.nNonce
;
420 // Watch for genesis block
421 if (pindexGenesisBlock
== NULL
&& diskindex
.GetBlockHash() == hashGenesisBlock
)
422 pindexGenesisBlock
= pindexNew
;
424 if (!pindexNew
->CheckIndex())
425 return error("LoadBlockIndex() : CheckIndex failed at %d", pindexNew
->nHeight
);
434 // Calculate bnChainWork
435 vector
<pair
<int, CBlockIndex
*> > vSortedByHeight
;
436 vSortedByHeight
.reserve(mapBlockIndex
.size());
437 foreach(const PAIRTYPE(uint256
, CBlockIndex
*)& item
, mapBlockIndex
)
439 CBlockIndex
* pindex
= item
.second
;
440 vSortedByHeight
.push_back(make_pair(pindex
->nHeight
, pindex
));
442 sort(vSortedByHeight
.begin(), vSortedByHeight
.end());
443 foreach(const PAIRTYPE(int, CBlockIndex
*)& item
, vSortedByHeight
)
445 CBlockIndex
* pindex
= item
.second
;
446 pindex
->bnChainWork
= (pindex
->pprev
? pindex
->pprev
->bnChainWork
: 0) + pindex
->GetBlockWork();
449 // Load hashBestChain pointer to end of best chain
450 if (!ReadHashBestChain(hashBestChain
))
452 if (pindexGenesisBlock
== NULL
)
454 return error("CTxDB::LoadBlockIndex() : hashBestChain not loaded");
456 if (!mapBlockIndex
.count(hashBestChain
))
457 return error("CTxDB::LoadBlockIndex() : hashBestChain not found in the block index");
458 pindexBest
= mapBlockIndex
[hashBestChain
];
459 nBestHeight
= pindexBest
->nHeight
;
460 bnBestChainWork
= pindexBest
->bnChainWork
;
461 printf("LoadBlockIndex(): hashBestChain=%s height=%d\n", hashBestChain
.ToString().substr(0,20).c_str(), nBestHeight
);
463 // Load bnBestInvalidWork, OK if it doesn't exist
464 ReadBestInvalidWork(bnBestInvalidWork
);
466 // Verify blocks in the best chain
467 CBlockIndex
* pindexFork
= NULL
;
468 for (CBlockIndex
* pindex
= pindexBest
; pindex
&& pindex
->pprev
; pindex
= pindex
->pprev
)
470 if (pindex
->nHeight
< nBestHeight
-2500 && !mapArgs
.count("-checkblocks"))
473 if (!block
.ReadFromDisk(pindex
))
474 return error("LoadBlockIndex() : block.ReadFromDisk failed");
475 if (!block
.CheckBlock())
477 printf("LoadBlockIndex() : *** found bad block at %d, hash=%s\n", pindex
->nHeight
, pindex
->GetBlockHash().ToString().c_str());
478 pindexFork
= pindex
->pprev
;
483 // Reorg back to the fork
484 printf("LoadBlockIndex() : *** moving best chain pointer back to block %d\n", pindexFork
->nHeight
);
486 if (!block
.ReadFromDisk(pindexFork
))
487 return error("LoadBlockIndex() : block.ReadFromDisk failed");
489 block
.SetBestChain(txdb
, pindexFork
);
503 bool CAddrDB::WriteAddress(const CAddress
& addr
)
505 return Write(make_pair(string("addr"), addr
.GetKey()), addr
);
508 bool CAddrDB::EraseAddress(const CAddress
& addr
)
510 return Erase(make_pair(string("addr"), addr
.GetKey()));
513 bool CAddrDB::LoadAddresses()
515 CRITICAL_BLOCK(cs_mapAddresses
)
517 // Load user provided addresses
518 CAutoFile filein
= fopen((GetDataDir() + "/addr.txt").c_str(), "rt");
524 while (fgets(psz
, sizeof(psz
), filein
))
526 CAddress
addr(psz
, NODE_NETWORK
);
527 addr
.nTime
= 0; // so it won't relay unless successfully connected
536 Dbc
* pcursor
= GetCursor();
545 int ret
= ReadAtCursor(pcursor
, ssKey
, ssValue
);
546 if (ret
== DB_NOTFOUND
)
554 if (strType
== "addr")
558 mapAddresses
.insert(make_pair(addr
.GetKey(), addr
));
563 printf("Loaded %d addresses\n", mapAddresses
.size());
571 return CAddrDB("cr+").LoadAddresses();
581 static set
<int64
> setKeyPool
;
582 static CCriticalSection cs_setKeyPool
;
584 bool CWalletDB::ReadAccount(const string
& strAccount
, CAccount
& account
)
587 return Read(make_pair(string("acc"), strAccount
), account
);
590 bool CWalletDB::WriteAccount(const string
& strAccount
, const CAccount
& account
)
592 return Write(make_pair(string("acc"), strAccount
), account
);
595 bool CWalletDB::WriteAccountingEntry(const CAccountingEntry
& acentry
)
597 return Write(make_tuple(string("acentry"), acentry
.strAccount
, ++nAccountingEntryNumber
), acentry
);
600 int64
CWalletDB::GetAccountCreditDebit(const string
& strAccount
)
602 list
<CAccountingEntry
> entries
;
603 ListAccountCreditDebit(strAccount
, entries
);
605 int64 nCreditDebit
= 0;
606 foreach (const CAccountingEntry
& entry
, entries
)
607 nCreditDebit
+= entry
.nCreditDebit
;
612 void CWalletDB::ListAccountCreditDebit(const string
& strAccount
, list
<CAccountingEntry
>& entries
)
614 int64 nCreditDebit
= 0;
616 bool fAllAccounts
= (strAccount
== "*");
618 Dbc
* pcursor
= GetCursor();
620 throw runtime_error("CWalletDB::ListAccountCreditDebit() : cannot create DB cursor");
621 unsigned int fFlags
= DB_SET_RANGE
;
626 if (fFlags
== DB_SET_RANGE
)
627 ssKey
<< make_tuple(string("acentry"), (fAllAccounts
? string("") : strAccount
), uint64(0));
629 int ret
= ReadAtCursor(pcursor
, ssKey
, ssValue
, fFlags
);
631 if (ret
== DB_NOTFOUND
)
636 throw runtime_error("CWalletDB::ListAccountCreditDebit() : error scanning DB");
642 if (strType
!= "acentry")
644 CAccountingEntry acentry
;
645 ssKey
>> acentry
.strAccount
;
646 if (!fAllAccounts
&& acentry
.strAccount
!= strAccount
)
650 entries
.push_back(acentry
);
657 bool CWalletDB::LoadWallet()
659 vchDefaultKey
.clear();
660 int nFileVersion
= 0;
661 vector
<uint256
> vWalletUpgrade
;
665 // Tray icon sometimes disappears on 9.10 karmic koala 64-bit, leaving no way to access the program
666 fMinimizeToTray
= false;
667 fMinimizeOnClose
= false;
670 //// todo: shouldn't we catch exceptions and try to recover and continue?
671 CRITICAL_BLOCK(cs_mapKeys
)
672 CRITICAL_BLOCK(cs_mapWallet
)
675 Dbc
* pcursor
= GetCursor();
684 int ret
= ReadAtCursor(pcursor
, ssKey
, ssValue
);
685 if (ret
== DB_NOTFOUND
)
691 // Taking advantage of the fact that pair serialization
692 // is just the two items serialized one after the other
695 if (strType
== "name")
699 ssValue
>> mapAddressBook
[strAddress
];
701 else if (strType
== "tx")
705 CWalletTx
& wtx
= mapWallet
[hash
];
708 if (wtx
.GetHash() != hash
)
709 printf("Error in wallet.dat, hash mismatch\n");
711 // Undo serialize changes in 31600
712 if (31404 <= wtx
.fTimeReceivedIsTxTime
&& wtx
.fTimeReceivedIsTxTime
<= 31703)
714 if (!ssValue
.empty())
718 ssValue
>> fTmp
>> fUnused
>> wtx
.strFromAccount
;
719 printf("LoadWallet() upgrading tx ver=%d %d '%s' %s\n", wtx
.fTimeReceivedIsTxTime
, fTmp
, wtx
.strFromAccount
.c_str(), hash
.ToString().c_str());
720 wtx
.fTimeReceivedIsTxTime
= fTmp
;
724 printf("LoadWallet() repairing tx ver=%d %s\n", wtx
.fTimeReceivedIsTxTime
, hash
.ToString().c_str());
725 wtx
.fTimeReceivedIsTxTime
= 0;
727 vWalletUpgrade
.push_back(hash
);
731 //printf("LoadWallet %s\n", wtx.GetHash().ToString().c_str());
732 //printf(" %12I64d %s %s %s\n",
733 // wtx.vout[0].nValue,
734 // DateTimeStrFormat("%x %H:%M:%S", wtx.GetBlockTime()).c_str(),
735 // wtx.hashBlock.ToString().substr(0,20).c_str(),
736 // wtx.mapValue["message"].c_str());
738 else if (strType
== "acentry")
744 if (nNumber
> nAccountingEntryNumber
)
745 nAccountingEntryNumber
= nNumber
;
747 else if (strType
== "key" || strType
== "wkey")
749 vector
<unsigned char> vchPubKey
;
752 if (strType
== "key")
753 ssValue
>> wkey
.vchPrivKey
;
757 mapKeys
[vchPubKey
] = wkey
.vchPrivKey
;
758 mapPubKeys
[Hash160(vchPubKey
)] = vchPubKey
;
760 else if (strType
== "defaultkey")
762 ssValue
>> vchDefaultKey
;
764 else if (strType
== "pool")
768 setKeyPool
.insert(nIndex
);
770 else if (strType
== "version")
772 ssValue
>> nFileVersion
;
773 if (nFileVersion
== 10300)
776 else if (strType
== "setting")
782 if (strKey
== "fGenerateBitcoins") ssValue
>> fGenerateBitcoins
;
785 if (strKey
== "nTransactionFee") ssValue
>> nTransactionFee
;
786 if (strKey
== "addrIncoming") ssValue
>> addrIncoming
;
787 if (strKey
== "fLimitProcessors") ssValue
>> fLimitProcessors
;
788 if (strKey
== "nLimitProcessors") ssValue
>> nLimitProcessors
;
789 if (strKey
== "fMinimizeToTray") ssValue
>> fMinimizeToTray
;
790 if (strKey
== "fMinimizeOnClose") ssValue
>> fMinimizeOnClose
;
791 if (strKey
== "fUseProxy") ssValue
>> fUseProxy
;
792 if (strKey
== "addrProxy") ssValue
>> addrProxy
;
799 foreach(uint256 hash
, vWalletUpgrade
)
800 WriteTx(hash
, mapWallet
[hash
]);
802 printf("nFileVersion = %d\n", nFileVersion
);
803 printf("fGenerateBitcoins = %d\n", fGenerateBitcoins
);
804 printf("nTransactionFee = %"PRI64d
"\n", nTransactionFee
);
805 printf("addrIncoming = %s\n", addrIncoming
.ToString().c_str());
806 printf("fMinimizeToTray = %d\n", fMinimizeToTray
);
807 printf("fMinimizeOnClose = %d\n", fMinimizeOnClose
);
808 printf("fUseProxy = %d\n", fUseProxy
);
809 printf("addrProxy = %s\n", addrProxy
.ToString().c_str());
813 if (nFileVersion
< VERSION
)
815 // Get rid of old debug.log file in current directory
816 if (nFileVersion
<= 105 && !pszSetDataDir
[0])
819 WriteVersion(VERSION
);
826 bool LoadWallet(bool& fFirstRunRet
)
828 fFirstRunRet
= false;
829 if (!CWalletDB("cr+").LoadWallet())
831 fFirstRunRet
= vchDefaultKey
.empty();
833 if (mapKeys
.count(vchDefaultKey
))
836 keyUser
.SetPubKey(vchDefaultKey
);
837 keyUser
.SetPrivKey(mapKeys
[vchDefaultKey
]);
841 // Create new keyUser and set as default key
842 RandAddSeedPerfmon();
843 keyUser
.MakeNewKey();
844 if (!AddKey(keyUser
))
846 if (!SetAddressBookName(PubKeyToAddress(keyUser
.GetPubKey()), "Your Address"))
848 CWalletDB().WriteDefaultKey(keyUser
.GetPubKey());
851 CreateThread(ThreadFlushWalletDB
, NULL
);
855 void ThreadFlushWalletDB(void* parg
)
857 static bool fOneThread
;
861 if (mapArgs
.count("-noflushwallet"))
864 unsigned int nLastSeen
= nWalletDBUpdated
;
865 unsigned int nLastFlushed
= nWalletDBUpdated
;
866 int64 nLastWalletUpdate
= GetTime();
871 if (nLastSeen
!= nWalletDBUpdated
)
873 nLastSeen
= nWalletDBUpdated
;
874 nLastWalletUpdate
= GetTime();
877 if (nLastFlushed
!= nWalletDBUpdated
&& GetTime() - nLastWalletUpdate
>= 2)
879 TRY_CRITICAL_BLOCK(cs_db
)
881 // Don't do this if any databases are in use
883 map
<string
, int>::iterator mi
= mapFileUseCount
.begin();
884 while (mi
!= mapFileUseCount
.end())
886 nRefCount
+= (*mi
).second
;
890 if (nRefCount
== 0 && !fShutdown
)
892 string strFile
= "wallet.dat";
893 map
<string
, int>::iterator mi
= mapFileUseCount
.find(strFile
);
894 if (mi
!= mapFileUseCount
.end())
896 printf("%s ", DateTimeStrFormat("%x %H:%M:%S", GetTime()).c_str());
897 printf("Flushing wallet.dat\n");
898 nLastFlushed
= nWalletDBUpdated
;
899 int64 nStart
= GetTimeMillis();
901 // Flush wallet.dat so it's self contained
903 dbenv
.txn_checkpoint(0, 0, 0);
904 dbenv
.lsn_reset(strFile
.c_str(), 0);
906 mapFileUseCount
.erase(mi
++);
907 printf("Flushed wallet.dat %"PRI64d
"ms\n", GetTimeMillis() - nStart
);
915 void BackupWallet(const string
& strDest
)
919 CRITICAL_BLOCK(cs_db
)
921 const string strFile
= "wallet.dat";
922 if (!mapFileUseCount
.count(strFile
) || mapFileUseCount
[strFile
] == 0)
924 // Flush log data to the dat file
926 dbenv
.txn_checkpoint(0, 0, 0);
927 dbenv
.lsn_reset(strFile
.c_str(), 0);
928 mapFileUseCount
.erase(strFile
);
931 filesystem::path
pathSrc(GetDataDir() + "/" + strFile
);
932 filesystem::path
pathDest(strDest
);
933 if (filesystem::is_directory(pathDest
))
934 pathDest
= pathDest
/ strFile
;
935 #if BOOST_VERSION >= 104000
936 filesystem::copy_file(pathSrc
, pathDest
, filesystem::copy_option::overwrite_if_exists
);
938 filesystem::copy_file(pathSrc
, pathDest
);
940 printf("copied wallet.dat to %s\n", pathDest
.string().c_str());
950 void CWalletDB::ReserveKeyFromKeyPool(int64
& nIndex
, CKeyPool
& keypool
)
953 keypool
.vchPubKey
.clear();
954 CRITICAL_BLOCK(cs_main
)
955 CRITICAL_BLOCK(cs_mapWallet
)
956 CRITICAL_BLOCK(cs_setKeyPool
)
959 int64 nTargetSize
= max(GetArg("-keypool", 100), (int64
)0);
960 while (setKeyPool
.size() < nTargetSize
+1)
963 if (!setKeyPool
.empty())
964 nEnd
= *(--setKeyPool
.end()) + 1;
965 if (!Write(make_pair(string("pool"), nEnd
), CKeyPool(GenerateNewKey())))
966 throw runtime_error("ReserveKeyFromKeyPool() : writing generated key failed");
967 setKeyPool
.insert(nEnd
);
968 printf("keypool added key %"PRI64d
", size=%d\n", nEnd
, setKeyPool
.size());
971 // Get the oldest key
972 assert(!setKeyPool
.empty());
973 nIndex
= *(setKeyPool
.begin());
974 setKeyPool
.erase(setKeyPool
.begin());
975 if (!Read(make_pair(string("pool"), nIndex
), keypool
))
976 throw runtime_error("ReserveKeyFromKeyPool() : read failed");
977 if (!mapKeys
.count(keypool
.vchPubKey
))
978 throw runtime_error("ReserveKeyFromKeyPool() : unknown key in key pool");
979 assert(!keypool
.vchPubKey
.empty());
980 printf("keypool reserve %"PRI64d
"\n", nIndex
);
984 void CWalletDB::KeepKey(int64 nIndex
)
986 // Remove from key pool
987 CRITICAL_BLOCK(cs_main
)
988 CRITICAL_BLOCK(cs_mapWallet
)
990 Erase(make_pair(string("pool"), nIndex
));
992 printf("keypool keep %"PRI64d
"\n", nIndex
);
995 void CWalletDB::ReturnKey(int64 nIndex
)
997 // Return to key pool
998 CRITICAL_BLOCK(cs_setKeyPool
)
999 setKeyPool
.insert(nIndex
);
1000 printf("keypool return %"PRI64d
"\n", nIndex
);
1003 vector
<unsigned char> GetKeyFromKeyPool()
1008 walletdb
.ReserveKeyFromKeyPool(nIndex
, keypool
);
1009 walletdb
.KeepKey(nIndex
);
1010 return keypool
.vchPubKey
;
1013 int64
GetOldestKeyPoolTime()
1018 walletdb
.ReserveKeyFromKeyPool(nIndex
, keypool
);
1019 walletdb
.ReturnKey(nIndex
);
1020 return keypool
.nTime
;