2010-08-29 18:58:15 +02:00
// Copyright (c) 2009-2010 Satoshi Nakamoto
2012-02-07 17:28:30 +01:00
// Copyright (c) 2009-2012 The Bitcoin developers
2010-08-29 18:58:15 +02:00
// Distributed under the MIT/X11 software license, see the accompanying
// file license.txt or http://www.opensource.org/licenses/mit-license.php.
2011-05-14 22:57:34 +02:00
# include "db.h"
2012-04-15 22:10:54 +02:00
# include "util.h"
# include "main.h"
2012-02-14 13:14:43 +01:00
# include <boost/version.hpp>
2011-06-20 00:12:31 +02:00
# include <boost/filesystem.hpp>
2011-05-16 05:45:35 +02:00
# include <boost/filesystem/fstream.hpp>
2010-08-29 18:58:15 +02:00
2012-04-15 22:10:54 +02:00
# ifndef WIN32
# include "sys/stat.h"
# endif
2011-05-15 09:11:04 +02:00
using namespace std ;
using namespace boost ;
2010-08-29 18:58:15 +02:00
unsigned int nWalletDBUpdated ;
//
// CDB
//
2012-04-15 23:39:49 +02:00
CCriticalSection cs_db ;
2010-08-29 18:58:15 +02:00
static bool fDbEnvInit = false ;
DbEnv dbenv ( 0 ) ;
2012-04-15 23:39:49 +02:00
map < string , int > mapFileUseCount ;
2010-08-29 18:58:15 +02:00
static map < string , Db * > mapDb ;
2011-11-17 20:21:32 +01:00
static void EnvShutdown ( )
2011-11-11 03:12:46 +01:00
{
if ( ! fDbEnvInit )
return ;
fDbEnvInit = false ;
2011-11-15 20:30:15 +01:00
try
{
dbenv . close ( 0 ) ;
}
catch ( const DbException & e )
{
printf ( " EnvShutdown exception: %s (%d) \n " , e . what ( ) , e . get_errno ( ) ) ;
}
2012-04-09 23:50:56 +02:00
DbEnv ( 0 ) . remove ( GetDataDir ( ) . string ( ) . c_str ( ) , 0 ) ;
2011-11-11 03:12:46 +01:00
}
2010-08-29 18:58:15 +02:00
class CDBInit
{
public :
CDBInit ( )
{
}
~ CDBInit ( )
{
2011-11-17 20:21:32 +01:00
EnvShutdown ( ) ;
2010-08-29 18:58:15 +02:00
}
}
instance_of_cdbinit ;
2012-04-09 23:50:56 +02:00
CDB : : CDB ( const char * pszFile , const char * pszMode ) : pdb ( NULL )
2010-08-29 18:58:15 +02:00
{
int ret ;
if ( pszFile = = NULL )
return ;
fReadOnly = ( ! strchr ( pszMode , ' + ' ) & & ! strchr ( pszMode , ' w ' ) ) ;
bool fCreate = strchr ( pszMode , ' c ' ) ;
unsigned int nFlags = DB_THREAD ;
if ( fCreate )
nFlags | = DB_CREATE ;
{
2012-04-06 18:39:12 +02:00
LOCK ( cs_db ) ;
2010-08-29 18:58:15 +02:00
if ( ! fDbEnvInit )
{
if ( fShutdown )
return ;
2012-04-09 23:50:56 +02:00
filesystem : : path pathDataDir = GetDataDir ( ) ;
filesystem : : path pathLogDir = pathDataDir / " database " ;
2012-03-21 22:15:28 +01:00
filesystem : : create_directory ( pathLogDir ) ;
2012-04-09 23:50:56 +02:00
filesystem : : path pathErrorFile = pathDataDir / " db.log " ;
2012-03-21 22:15:28 +01:00
printf ( " dbenv.open LogDir=%s ErrorFile=%s \n " , pathLogDir . string ( ) . c_str ( ) , pathErrorFile . string ( ) . c_str ( ) ) ;
2010-08-29 18:58:15 +02:00
2012-03-21 00:03:02 +01:00
int nDbCache = GetArg ( " -dbcache " , 25 ) ;
2012-03-21 22:15:28 +01:00
dbenv . set_lg_dir ( pathLogDir . string ( ) . c_str ( ) ) ;
2012-03-21 00:03:02 +01:00
dbenv . set_cachesize ( nDbCache / 1024 , ( nDbCache % 1024 ) * 1048576 , 1 ) ;
2012-03-29 15:03:11 +02:00
dbenv . set_lg_bsize ( 1048576 ) ;
dbenv . set_lg_max ( 10485760 ) ;
2010-08-29 18:58:15 +02:00
dbenv . set_lk_max_locks ( 10000 ) ;
dbenv . set_lk_max_objects ( 10000 ) ;
2012-03-21 22:15:28 +01:00
dbenv . set_errfile ( fopen ( pathErrorFile . string ( ) . c_str ( ) , " a " ) ) ; /// debug
2010-08-29 18:58:15 +02:00
dbenv . set_flags ( DB_AUTO_COMMIT , 1 ) ;
2012-03-29 15:03:11 +02:00
dbenv . log_set_config ( DB_LOG_AUTO_REMOVE , 1 ) ;
2012-04-09 23:50:56 +02:00
ret = dbenv . open ( pathDataDir . string ( ) . c_str ( ) ,
2010-08-29 18:58:15 +02:00
DB_CREATE |
DB_INIT_LOCK |
DB_INIT_LOG |
DB_INIT_MPOOL |
DB_INIT_TXN |
DB_THREAD |
DB_RECOVER ,
S_IRUSR | S_IWUSR ) ;
if ( ret > 0 )
2010-09-06 23:03:04 +02:00
throw runtime_error ( strprintf ( " CDB() : error % d opening database environment " , ret)) ;
2010-08-29 18:58:15 +02:00
fDbEnvInit = true ;
}
strFile = pszFile ;
+ + mapFileUseCount [ strFile ] ;
pdb = mapDb [ strFile ] ;
if ( pdb = = NULL )
{
pdb = new Db ( & dbenv , 0 ) ;
ret = pdb - > open ( NULL , // Txn pointer
pszFile , // Filename
" main " , // Logical db name
DB_BTREE , // Database type
nFlags , // Flags
0 ) ;
if ( ret > 0 )
{
delete pdb ;
pdb = NULL ;
2012-04-06 18:39:12 +02:00
{
LOCK ( cs_db ) ;
2010-08-29 18:58:15 +02:00
- - mapFileUseCount [ strFile ] ;
2012-04-06 18:39:12 +02:00
}
2010-08-29 18:58:15 +02:00
strFile = " " ;
2010-09-06 23:03:04 +02:00
throw runtime_error ( strprintf ( " CDB() : can ' t open database file % s , error % d " , pszFile, ret)) ;
2010-08-29 18:58:15 +02:00
}
if ( fCreate & & ! Exists ( string ( " version " ) ) )
{
bool fTmp = fReadOnly ;
fReadOnly = false ;
2011-12-16 22:26:14 +01:00
WriteVersion ( CLIENT_VERSION ) ;
2010-08-29 18:58:15 +02:00
fReadOnly = fTmp ;
}
mapDb [ strFile ] = pdb ;
}
}
}
void CDB : : Close ( )
{
if ( ! pdb )
return ;
if ( ! vTxn . empty ( ) )
vTxn . front ( ) - > abort ( ) ;
vTxn . clear ( ) ;
pdb = NULL ;
// Flush database activity from memory pool to disk log
unsigned int nMinutes = 0 ;
2010-12-05 10:29:30 +01:00
if ( fReadOnly )
nMinutes = 1 ;
2010-08-29 18:58:15 +02:00
if ( strFile = = " addr.dat " )
nMinutes = 2 ;
2012-04-17 18:27:00 +02:00
if ( strFile = = " blkindex.dat " )
nMinutes = 2 ;
2012-03-28 22:09:18 +02:00
if ( strFile = = " blkindex.dat " & & IsInitialBlockDownload ( ) )
nMinutes = 5 ;
2012-04-05 17:54:30 +02:00
dbenv . txn_checkpoint ( nMinutes ? GetArg ( " -dblogsize " , 100 ) * 1024 : 0 , nMinutes , 0 ) ;
2010-08-29 18:58:15 +02:00
2012-04-06 18:39:12 +02:00
{
LOCK ( cs_db ) ;
2010-08-29 18:58:15 +02:00
- - mapFileUseCount [ strFile ] ;
2012-04-06 18:39:12 +02:00
}
2010-08-29 18:58:15 +02:00
}
2012-04-15 23:39:49 +02:00
void CloseDb ( const string & strFile )
2010-08-29 18:58:15 +02:00
{
{
2012-04-06 18:39:12 +02:00
LOCK ( cs_db ) ;
2010-08-29 18:58:15 +02:00
if ( mapDb [ strFile ] ! = NULL )
{
// Close the database handle
Db * pdb = mapDb [ strFile ] ;
pdb - > close ( 0 ) ;
delete pdb ;
mapDb [ strFile ] = NULL ;
}
}
}
2011-11-11 03:12:46 +01:00
bool CDB : : Rewrite ( const string & strFile , const char * pszSkip )
2011-11-10 21:29:23 +01:00
{
while ( ! fShutdown )
{
{
2012-04-06 18:39:12 +02:00
LOCK ( cs_db ) ;
2011-11-10 21:29:23 +01:00
if ( ! mapFileUseCount . count ( strFile ) | | mapFileUseCount [ strFile ] = = 0 )
{
// Flush log data to the dat file
CloseDb ( strFile ) ;
dbenv . txn_checkpoint ( 0 , 0 , 0 ) ;
dbenv . lsn_reset ( strFile . c_str ( ) , 0 ) ;
mapFileUseCount . erase ( strFile ) ;
bool fSuccess = true ;
2011-11-11 03:12:46 +01:00
printf ( " Rewriting %s... \n " , strFile . c_str ( ) ) ;
string strFileRes = strFile + " .rewrite " ;
2011-11-20 17:12:00 +01:00
{ // surround usage of db with extra {}
CDB db ( strFile . c_str ( ) , " r " ) ;
Db * pdbCopy = new Db ( & dbenv , 0 ) ;
int ret = pdbCopy - > open ( NULL , // Txn pointer
strFileRes . c_str ( ) , // Filename
" main " , // Logical db name
DB_BTREE , // Database type
DB_CREATE , // Flags
0 ) ;
if ( ret > 0 )
2011-11-10 21:29:23 +01:00
{
2011-11-20 17:12:00 +01:00
printf ( " Cannot create database file %s \n " , strFileRes . c_str ( ) ) ;
fSuccess = false ;
}
Dbc * pcursor = db . GetCursor ( ) ;
if ( pcursor )
while ( fSuccess )
2011-11-11 03:12:46 +01:00
{
2012-04-16 14:56:45 +02:00
CDataStream ssKey ( SER_DISK , CLIENT_VERSION ) ;
CDataStream ssValue ( SER_DISK , CLIENT_VERSION ) ;
2011-11-20 17:12:00 +01:00
int ret = db . ReadAtCursor ( pcursor , ssKey , ssValue , DB_NEXT ) ;
if ( ret = = DB_NOTFOUND )
{
pcursor - > close ( ) ;
break ;
}
else if ( ret ! = 0 )
{
pcursor - > close ( ) ;
fSuccess = false ;
break ;
}
if ( pszSkip & &
strncmp ( & ssKey [ 0 ] , pszSkip , std : : min ( ssKey . size ( ) , strlen ( pszSkip ) ) ) = = 0 )
continue ;
if ( strncmp ( & ssKey [ 0 ] , " \x07 version " , 8 ) = = 0 )
{
// Update version:
ssValue . clear ( ) ;
2011-12-16 22:26:14 +01:00
ssValue < < CLIENT_VERSION ;
2011-11-20 17:12:00 +01:00
}
Dbt datKey ( & ssKey [ 0 ] , ssKey . size ( ) ) ;
Dbt datValue ( & ssValue [ 0 ] , ssValue . size ( ) ) ;
int ret2 = pdbCopy - > put ( NULL , & datKey , & datValue , DB_NOOVERWRITE ) ;
if ( ret2 > 0 )
fSuccess = false ;
2011-11-11 03:12:46 +01:00
}
2011-11-20 17:12:00 +01:00
if ( fSuccess )
{
db . Close ( ) ;
CloseDb ( strFile ) ;
if ( pdbCopy - > close ( 0 ) )
2011-11-10 21:29:23 +01:00
fSuccess = false ;
2011-11-20 17:12:00 +01:00
delete pdbCopy ;
2011-11-10 21:29:23 +01:00
}
}
if ( fSuccess )
{
Db dbA ( & dbenv , 0 ) ;
if ( dbA . remove ( strFile . c_str ( ) , NULL , 0 ) )
fSuccess = false ;
Db dbB ( & dbenv , 0 ) ;
if ( dbB . rename ( strFileRes . c_str ( ) , NULL , strFile . c_str ( ) , 0 ) )
fSuccess = false ;
}
if ( ! fSuccess )
2011-11-11 03:12:46 +01:00
printf ( " Rewriting of %s FAILED! \n " , strFileRes . c_str ( ) ) ;
2011-11-10 21:29:23 +01:00
return fSuccess ;
}
}
Sleep ( 100 ) ;
}
return false ;
}
2011-11-17 20:21:32 +01:00
void DBFlush ( bool fShutdown )
2010-08-29 18:58:15 +02:00
{
// Flush log data to the actual data file
// on all files that are not in use
printf ( " DBFlush(%s)%s \n " , fShutdown ? " true " : " false " , fDbEnvInit ? " " : " db not started " ) ;
if ( ! fDbEnvInit )
return ;
{
2012-04-06 18:39:12 +02:00
LOCK ( cs_db ) ;
2010-08-29 18:58:15 +02:00
map < string , int > : : iterator mi = mapFileUseCount . begin ( ) ;
while ( mi ! = mapFileUseCount . end ( ) )
{
string strFile = ( * mi ) . first ;
int nRefCount = ( * mi ) . second ;
printf ( " %s refcount=%d \n " , strFile . c_str ( ) , nRefCount ) ;
if ( nRefCount = = 0 )
{
// Move log data to the dat file
CloseDb ( strFile ) ;
dbenv . txn_checkpoint ( 0 , 0 , 0 ) ;
printf ( " %s flush \n " , strFile . c_str ( ) ) ;
dbenv . lsn_reset ( strFile . c_str ( ) , 0 ) ;
mapFileUseCount . erase ( mi + + ) ;
}
else
mi + + ;
}
if ( fShutdown )
{
char * * listp ;
if ( mapFileUseCount . empty ( ) )
2011-11-11 03:12:46 +01:00
{
2010-08-29 18:58:15 +02:00
dbenv . log_archive ( & listp , DB_ARCH_REMOVE ) ;
2011-11-17 20:21:32 +01:00
EnvShutdown ( ) ;
2011-11-11 03:12:46 +01:00
}
2010-08-29 18:58:15 +02:00
}
}
}
//
// CTxDB
//
bool CTxDB : : ReadTxIndex ( uint256 hash , CTxIndex & txindex )
{
assert ( ! fClient ) ;
txindex . SetNull ( ) ;
return Read ( make_pair ( string ( " tx " ) , hash ) , txindex ) ;
}
bool CTxDB : : UpdateTxIndex ( uint256 hash , const CTxIndex & txindex )
{
assert ( ! fClient ) ;
return Write ( make_pair ( string ( " tx " ) , hash ) , txindex ) ;
}
bool CTxDB : : AddTxIndex ( const CTransaction & tx , const CDiskTxPos & pos , int nHeight )
{
assert ( ! fClient ) ;
// Add to tx index
uint256 hash = tx . GetHash ( ) ;
CTxIndex txindex ( pos , tx . vout . size ( ) ) ;
return Write ( make_pair ( string ( " tx " ) , hash ) , txindex ) ;
}
bool CTxDB : : EraseTxIndex ( const CTransaction & tx )
{
assert ( ! fClient ) ;
uint256 hash = tx . GetHash ( ) ;
return Erase ( make_pair ( string ( " tx " ) , hash ) ) ;
}
bool CTxDB : : ContainsTx ( uint256 hash )
{
assert ( ! fClient ) ;
return Exists ( make_pair ( string ( " tx " ) , hash ) ) ;
}
bool CTxDB : : ReadOwnerTxes ( uint160 hash160 , int nMinHeight , vector < CTransaction > & vtx )
{
assert ( ! fClient ) ;
vtx . clear ( ) ;
// Get cursor
Dbc * pcursor = GetCursor ( ) ;
if ( ! pcursor )
return false ;
unsigned int fFlags = DB_SET_RANGE ;
loop
{
// Read next record
2012-04-16 14:56:45 +02:00
CDataStream ssKey ( SER_DISK , CLIENT_VERSION ) ;
2010-08-29 18:58:15 +02:00
if ( fFlags = = DB_SET_RANGE )
ssKey < < string ( " owner " ) < < hash160 < < CDiskTxPos ( 0 , 0 , 0 ) ;
2012-04-16 14:56:45 +02:00
CDataStream ssValue ( SER_DISK , CLIENT_VERSION ) ;
2010-08-29 18:58:15 +02:00
int ret = ReadAtCursor ( pcursor , ssKey , ssValue , fFlags ) ;
fFlags = DB_NEXT ;
if ( ret = = DB_NOTFOUND )
break ;
else if ( ret ! = 0 )
{
pcursor - > close ( ) ;
return false ;
}
// Unserialize
string strType ;
uint160 hashItem ;
CDiskTxPos pos ;
ssKey > > strType > > hashItem > > pos ;
int nItemHeight ;
ssValue > > nItemHeight ;
// Read transaction
if ( strType ! = " owner " | | hashItem ! = hash160 )
break ;
if ( nItemHeight > = nMinHeight )
{
vtx . resize ( vtx . size ( ) + 1 ) ;
if ( ! vtx . back ( ) . ReadFromDisk ( pos ) )
{
pcursor - > close ( ) ;
return false ;
}
}
}
pcursor - > close ( ) ;
return true ;
}
bool CTxDB : : ReadDiskTx ( uint256 hash , CTransaction & tx , CTxIndex & txindex )
{
assert ( ! fClient ) ;
tx . SetNull ( ) ;
if ( ! ReadTxIndex ( hash , txindex ) )
return false ;
return ( tx . ReadFromDisk ( txindex . pos ) ) ;
}
bool CTxDB : : ReadDiskTx ( uint256 hash , CTransaction & tx )
{
CTxIndex txindex ;
return ReadDiskTx ( hash , tx , txindex ) ;
}
bool CTxDB : : ReadDiskTx ( COutPoint outpoint , CTransaction & tx , CTxIndex & txindex )
{
return ReadDiskTx ( outpoint . hash , tx , txindex ) ;
}
bool CTxDB : : ReadDiskTx ( COutPoint outpoint , CTransaction & tx )
{
CTxIndex txindex ;
return ReadDiskTx ( outpoint . hash , tx , txindex ) ;
}
bool CTxDB : : WriteBlockIndex ( const CDiskBlockIndex & blockindex )
{
return Write ( make_pair ( string ( " blockindex " ) , blockindex . GetBlockHash ( ) ) , blockindex ) ;
}
bool CTxDB : : EraseBlockIndex ( uint256 hash )
{
return Erase ( make_pair ( string ( " blockindex " ) , hash ) ) ;
}
bool CTxDB : : ReadHashBestChain ( uint256 & hashBestChain )
{
return Read ( string ( " hashBestChain " ) , hashBestChain ) ;
}
bool CTxDB : : WriteHashBestChain ( uint256 hashBestChain )
{
return Write ( string ( " hashBestChain " ) , hashBestChain ) ;
}
bool CTxDB : : ReadBestInvalidWork ( CBigNum & bnBestInvalidWork )
{
return Read ( string ( " bnBestInvalidWork " ) , bnBestInvalidWork ) ;
}
bool CTxDB : : WriteBestInvalidWork ( CBigNum bnBestInvalidWork )
{
return Write ( string ( " bnBestInvalidWork " ) , bnBestInvalidWork ) ;
}
CWallet class
* A new class CKeyStore manages private keys, and script.cpp depends on access to CKeyStore.
* A new class CWallet extends CKeyStore, and contains all former wallet-specific globals; CWallet depends on script.cpp, not the other way around.
* Wallet-specific functions in CTransaction/CTxIn/CTxOut (GetDebit, GetCredit, GetChange, IsMine, IsFromMe), are moved to CWallet, taking their former 'this' argument as an explicit parameter
* CWalletTx objects know which CWallet they belong to, for convenience, so they have their own direct (and caching) GetDebit/... functions.
* Some code was moved from CWalletDB to CWallet, such as handling of reserve keys.
* Main.cpp keeps a set of all 'registered' wallets, which should be informed about updates to the block chain, and does not have any notion about any 'main' wallet. Function in main.cpp that require a wallet (such as GenerateCoins), take an explicit CWallet* argument.
* The actual CWallet instance used by the application is defined in init.cpp as "CWallet* pwalletMain". rpc.cpp and ui.cpp use this variable.
* Functions in main.cpp and db.cpp that are not used by other modules are marked static.
* The code for handling the 'submitorder' message is removed, as it not really compatible with the idea that a node is independent from the wallet(s) connected to it, and obsolete anyway.
2011-06-01 18:28:20 +02:00
CBlockIndex static * InsertBlockIndex ( uint256 hash )
2010-08-29 18:58:15 +02:00
{
if ( hash = = 0 )
return NULL ;
// Return existing
map < uint256 , CBlockIndex * > : : iterator mi = mapBlockIndex . find ( hash ) ;
if ( mi ! = mapBlockIndex . end ( ) )
return ( * mi ) . second ;
// Create new
CBlockIndex * pindexNew = new CBlockIndex ( ) ;
if ( ! pindexNew )
throw runtime_error ( " LoadBlockIndex() : new CBlockIndex failed " ) ;
mi = mapBlockIndex . insert ( make_pair ( hash , pindexNew ) ) . first ;
pindexNew - > phashBlock = & ( ( * mi ) . first ) ;
return pindexNew ;
}
bool CTxDB : : LoadBlockIndex ( )
{
// Get database cursor
Dbc * pcursor = GetCursor ( ) ;
if ( ! pcursor )
return false ;
// Load mapBlockIndex
unsigned int fFlags = DB_SET_RANGE ;
loop
{
// Read next record
2012-04-16 14:56:45 +02:00
CDataStream ssKey ( SER_DISK , CLIENT_VERSION ) ;
2010-08-29 18:58:15 +02:00
if ( fFlags = = DB_SET_RANGE )
ssKey < < make_pair ( string ( " blockindex " ) , uint256 ( 0 ) ) ;
2012-04-16 14:56:45 +02:00
CDataStream ssValue ( SER_DISK , CLIENT_VERSION ) ;
2010-08-29 18:58:15 +02:00
int ret = ReadAtCursor ( pcursor , ssKey , ssValue , fFlags ) ;
fFlags = DB_NEXT ;
if ( ret = = DB_NOTFOUND )
break ;
else if ( ret ! = 0 )
return false ;
// Unserialize
string strType ;
ssKey > > strType ;
2012-04-18 13:30:24 +02:00
if ( strType = = " blockindex " & & ! fRequestShutdown )
2010-08-29 18:58:15 +02:00
{
CDiskBlockIndex diskindex ;
ssValue > > diskindex ;
// Construct block index object
CBlockIndex * pindexNew = InsertBlockIndex ( diskindex . GetBlockHash ( ) ) ;
pindexNew - > pprev = InsertBlockIndex ( diskindex . hashPrev ) ;
pindexNew - > pnext = InsertBlockIndex ( diskindex . hashNext ) ;
pindexNew - > nFile = diskindex . nFile ;
pindexNew - > nBlockPos = diskindex . nBlockPos ;
pindexNew - > nHeight = diskindex . nHeight ;
pindexNew - > nVersion = diskindex . nVersion ;
pindexNew - > hashMerkleRoot = diskindex . hashMerkleRoot ;
pindexNew - > nTime = diskindex . nTime ;
pindexNew - > nBits = diskindex . nBits ;
pindexNew - > nNonce = diskindex . nNonce ;
// Watch for genesis block
if ( pindexGenesisBlock = = NULL & & diskindex . GetBlockHash ( ) = = hashGenesisBlock )
pindexGenesisBlock = pindexNew ;
if ( ! pindexNew - > CheckIndex ( ) )
return error ( " LoadBlockIndex() : CheckIndex failed at % d " , pindexNew->nHeight) ;
}
else
{
2012-04-18 13:30:24 +02:00
break ; // if shutdown requested or finished loading block index
2010-08-29 18:58:15 +02:00
}
}
pcursor - > close ( ) ;
2012-04-18 13:30:24 +02:00
if ( fRequestShutdown )
return true ;
2010-08-29 18:58:15 +02:00
// Calculate bnChainWork
vector < pair < int , CBlockIndex * > > vSortedByHeight ;
vSortedByHeight . reserve ( mapBlockIndex . size ( ) ) ;
2011-05-15 09:11:04 +02:00
BOOST_FOREACH ( const PAIRTYPE ( uint256 , CBlockIndex * ) & item , mapBlockIndex )
2010-08-29 18:58:15 +02:00
{
CBlockIndex * pindex = item . second ;
vSortedByHeight . push_back ( make_pair ( pindex - > nHeight , pindex ) ) ;
}
sort ( vSortedByHeight . begin ( ) , vSortedByHeight . end ( ) ) ;
2011-05-15 09:11:04 +02:00
BOOST_FOREACH ( const PAIRTYPE ( int , CBlockIndex * ) & item , vSortedByHeight )
2010-08-29 18:58:15 +02:00
{
CBlockIndex * pindex = item . second ;
pindex - > bnChainWork = ( pindex - > pprev ? pindex - > pprev - > bnChainWork : 0 ) + pindex - > GetBlockWork ( ) ;
}
// Load hashBestChain pointer to end of best chain
if ( ! ReadHashBestChain ( hashBestChain ) )
{
if ( pindexGenesisBlock = = NULL )
return true ;
return error ( " CTxDB::LoadBlockIndex() : hashBestChain not loaded " ) ;
}
if ( ! mapBlockIndex . count ( hashBestChain ) )
return error ( " CTxDB::LoadBlockIndex() : hashBestChain not found in the block index " ) ;
pindexBest = mapBlockIndex [ hashBestChain ] ;
nBestHeight = pindexBest - > nHeight ;
bnBestChainWork = pindexBest - > bnChainWork ;
printf ( " LoadBlockIndex(): hashBestChain=%s height=%d \n " , hashBestChain . ToString ( ) . substr ( 0 , 20 ) . c_str ( ) , nBestHeight ) ;
// Load bnBestInvalidWork, OK if it doesn't exist
ReadBestInvalidWork ( bnBestInvalidWork ) ;
// Verify blocks in the best chain
2012-03-21 23:08:13 +01:00
int nCheckLevel = GetArg ( " -checklevel " , 1 ) ;
int nCheckDepth = GetArg ( " -checkblocks " , 2500 ) ;
if ( nCheckDepth = = 0 )
nCheckDepth = 1000000000 ; // suffices until the year 19000
if ( nCheckDepth > nBestHeight )
nCheckDepth = nBestHeight ;
printf ( " Verifying last %i blocks at level %i \n " , nCheckDepth , nCheckLevel ) ;
2010-08-29 18:58:15 +02:00
CBlockIndex * pindexFork = NULL ;
2012-03-21 23:08:13 +01:00
map < pair < unsigned int , unsigned int > , CBlockIndex * > mapBlockPos ;
2010-08-29 18:58:15 +02:00
for ( CBlockIndex * pindex = pindexBest ; pindex & & pindex - > pprev ; pindex = pindex - > pprev )
{
2012-03-21 23:08:13 +01:00
if ( pindex - > nHeight < nBestHeight - nCheckDepth )
2010-08-29 18:58:15 +02:00
break ;
CBlock block ;
if ( ! block . ReadFromDisk ( pindex ) )
return error ( " LoadBlockIndex() : block . ReadFromDisk failed " ) ;
2012-03-21 23:08:13 +01:00
// check level 1: verify block validity
if ( nCheckLevel > 0 & & ! block . CheckBlock ( ) )
2010-08-29 18:58:15 +02:00
{
printf ( " LoadBlockIndex() : *** found bad block at %d, hash=%s \n " , pindex - > nHeight , pindex - > GetBlockHash ( ) . ToString ( ) . c_str ( ) ) ;
pindexFork = pindex - > pprev ;
}
2012-03-21 23:08:13 +01:00
// check level 2: verify transaction index validity
if ( nCheckLevel > 1 )
{
pair < unsigned int , unsigned int > pos = make_pair ( pindex - > nFile , pindex - > nBlockPos ) ;
mapBlockPos [ pos ] = pindex ;
BOOST_FOREACH ( const CTransaction & tx , block . vtx )
{
uint256 hashTx = tx . GetHash ( ) ;
CTxIndex txindex ;
if ( ReadTxIndex ( hashTx , txindex ) )
{
// check level 3: checker transaction hashes
if ( nCheckLevel > 2 | | pindex - > nFile ! = txindex . pos . nFile | | pindex - > nBlockPos ! = txindex . pos . nBlockPos )
{
// either an error or a duplicate transaction
CTransaction txFound ;
if ( ! txFound . ReadFromDisk ( txindex . pos ) )
{
printf ( " LoadBlockIndex() : *** cannot read mislocated transaction %s \n " , hashTx . ToString ( ) . c_str ( ) ) ;
pindexFork = pindex - > pprev ;
}
else
if ( txFound . GetHash ( ) ! = hashTx ) // not a duplicate tx
{
printf ( " LoadBlockIndex(): *** invalid tx position for %s \n " , hashTx . ToString ( ) . c_str ( ) ) ;
pindexFork = pindex - > pprev ;
}
}
// check level 4: check whether spent txouts were spent within the main chain
2012-04-22 20:01:25 +02:00
unsigned int nOutput = 0 ;
2012-03-21 23:08:13 +01:00
if ( nCheckLevel > 3 )
2012-04-15 13:27:00 +02:00
{
2012-03-21 23:08:13 +01:00
BOOST_FOREACH ( const CDiskTxPos & txpos , txindex . vSpent )
{
if ( ! txpos . IsNull ( ) )
{
pair < unsigned int , unsigned int > posFind = make_pair ( txpos . nFile , txpos . nBlockPos ) ;
if ( ! mapBlockPos . count ( posFind ) )
{
printf ( " LoadBlockIndex(): *** found bad spend at %d, hashBlock=%s, hashTx=%s \n " , pindex - > nHeight , pindex - > GetBlockHash ( ) . ToString ( ) . c_str ( ) , hashTx . ToString ( ) . c_str ( ) ) ;
pindexFork = pindex - > pprev ;
}
// check level 6: check whether spent txouts were spent by a valid transaction that consume them
if ( nCheckLevel > 5 )
{
CTransaction txSpend ;
if ( ! txSpend . ReadFromDisk ( txpos ) )
{
printf ( " LoadBlockIndex(): *** cannot read spending transaction of %s:%i from disk \n " , hashTx . ToString ( ) . c_str ( ) , nOutput ) ;
pindexFork = pindex - > pprev ;
}
else if ( ! txSpend . CheckTransaction ( ) )
{
printf ( " LoadBlockIndex(): *** spending transaction of %s:%i is invalid \n " , hashTx . ToString ( ) . c_str ( ) , nOutput ) ;
pindexFork = pindex - > pprev ;
}
else
{
bool fFound = false ;
BOOST_FOREACH ( const CTxIn & txin , txSpend . vin )
if ( txin . prevout . hash = = hashTx & & txin . prevout . n = = nOutput )
fFound = true ;
if ( ! fFound )
{
printf ( " LoadBlockIndex(): *** spending transaction of %s:%i does not spend it \n " , hashTx . ToString ( ) . c_str ( ) , nOutput ) ;
pindexFork = pindex - > pprev ;
}
}
}
}
nOutput + + ;
}
2012-04-15 13:27:00 +02:00
}
2012-03-21 23:08:13 +01:00
}
// check level 5: check whether all prevouts are marked spent
if ( nCheckLevel > 4 )
2012-04-15 13:27:00 +02:00
{
2012-03-21 23:08:13 +01:00
BOOST_FOREACH ( const CTxIn & txin , tx . vin )
{
CTxIndex txindex ;
if ( ReadTxIndex ( txin . prevout . hash , txindex ) )
if ( txindex . vSpent . size ( ) - 1 < txin . prevout . n | | txindex . vSpent [ txin . prevout . n ] . IsNull ( ) )
{
printf ( " LoadBlockIndex(): *** found unspent prevout %s:%i in %s \n " , txin . prevout . hash . ToString ( ) . c_str ( ) , txin . prevout . n , hashTx . ToString ( ) . c_str ( ) ) ;
pindexFork = pindex - > pprev ;
}
}
2012-04-15 13:27:00 +02:00
}
2012-03-21 23:08:13 +01:00
}
}
2010-08-29 18:58:15 +02:00
}
if ( pindexFork )
{
// Reorg back to the fork
printf ( " LoadBlockIndex() : *** moving best chain pointer back to block %d \n " , pindexFork - > nHeight ) ;
CBlock block ;
if ( ! block . ReadFromDisk ( pindexFork ) )
return error ( " LoadBlockIndex() : block . ReadFromDisk failed " ) ;
CTxDB txdb ;
block . SetBestChain ( txdb , pindexFork ) ;
}
return true ;
}
//
// CAddrDB
//
2012-01-04 23:39:45 +01:00
bool CAddrDB : : WriteAddrman ( const CAddrMan & addrman )
{
return Write ( string ( " addrman " ) , addrman ) ;
}
2012-03-26 20:17:35 +02:00
bool CAddrDB : : LoadAddresses ( )
2010-08-29 18:58:15 +02:00
{
2012-01-04 23:39:45 +01:00
if ( Read ( string ( " addrman " ) , addrman ) )
2010-08-29 18:58:15 +02:00
{
2012-01-04 23:39:45 +01:00
printf ( " Loaded %i addresses \n " , addrman . size ( ) ) ;
2012-03-29 19:34:07 +02:00
return true ;
2012-01-04 23:39:45 +01:00
}
2012-03-29 19:34:07 +02:00
// Read pre-0.6 addr records
2012-01-04 23:39:45 +01:00
vector < CAddress > vAddr ;
2012-03-26 20:17:35 +02:00
vector < vector < unsigned char > > vDelete ;
2012-01-04 23:39:45 +01:00
// Get cursor
Dbc * pcursor = GetCursor ( ) ;
if ( ! pcursor )
return false ;
loop
{
// Read next record
2012-04-16 14:56:45 +02:00
CDataStream ssKey ( SER_DISK , CLIENT_VERSION ) ;
CDataStream ssValue ( SER_DISK , CLIENT_VERSION ) ;
2012-01-04 23:39:45 +01:00
int ret = ReadAtCursor ( pcursor , ssKey , ssValue ) ;
if ( ret = = DB_NOTFOUND )
break ;
else if ( ret ! = 0 )
2010-08-29 18:58:15 +02:00
return false ;
2012-01-04 23:39:45 +01:00
// Unserialize
string strType ;
ssKey > > strType ;
if ( strType = = " addr " )
2010-08-29 18:58:15 +02:00
{
2012-03-29 19:34:07 +02:00
CAddress addr ;
ssValue > > addr ;
vAddr . push_back ( addr ) ;
2010-08-29 18:58:15 +02:00
}
2012-01-04 23:39:45 +01:00
}
pcursor - > close ( ) ;
2010-08-29 18:58:15 +02:00
2012-03-29 19:34:07 +02:00
addrman . Add ( vAddr , CNetAddr ( " 0.0.0.0 " ) ) ;
printf ( " Loaded %i addresses \n " , addrman . size ( ) ) ;
2012-03-26 20:17:35 +02:00
2012-03-29 19:34:07 +02:00
// Note: old records left; we ran into hangs-on-startup
// bugs for some users who (we think) were running after
// an unclean shutdown.
2010-08-29 18:58:15 +02:00
return true ;
}
bool LoadAddresses ( )
{
2012-03-26 20:17:35 +02:00
return CAddrDB ( " cr+ " ) . LoadAddresses ( ) ;
2010-08-29 18:58:15 +02:00
}