Pieter Wuille
12 years ago
11 changed files with 78 additions and 351 deletions
@ -1,171 +0,0 @@ |
|||||||
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
|
||||||
// Copyright (c) 2009-2012 The Bitcoin developers
|
|
||||||
// Distributed under the MIT/X11 software license, see the accompanying
|
|
||||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
||||||
|
|
||||||
#include "txdb-bdb.h" |
|
||||||
|
|
||||||
bool CCoinsDB::HaveCoins(uint256 hash) { |
|
||||||
assert(!fClient); |
|
||||||
return Exists(make_pair('c', hash)); |
|
||||||
} |
|
||||||
|
|
||||||
bool CCoinsDB::ReadCoins(uint256 hash, CCoins &coins) { |
|
||||||
assert(!fClient); |
|
||||||
return Read(make_pair('c', hash), coins); |
|
||||||
} |
|
||||||
|
|
||||||
bool CCoinsDB::WriteCoins(uint256 hash, const CCoins &coins) { |
|
||||||
assert(!fClient); |
|
||||||
if (coins.IsPruned()) |
|
||||||
return Erase(make_pair('c', hash)); |
|
||||||
else |
|
||||||
return Write(make_pair('c', hash), coins); |
|
||||||
} |
|
||||||
|
|
||||||
bool CCoinsDB::ReadHashBestChain(uint256& hashBestChain) |
|
||||||
{ |
|
||||||
return Read('B', hashBestChain); |
|
||||||
} |
|
||||||
|
|
||||||
bool CCoinsDB::WriteHashBestChain(uint256 hashBestChain) |
|
||||||
{ |
|
||||||
return Write('B', hashBestChain); |
|
||||||
} |
|
||||||
|
|
||||||
bool CBlockTreeDB::WriteBlockIndex(const CDiskBlockIndex& blockindex) |
|
||||||
{ |
|
||||||
return Write(make_pair('b', blockindex.GetBlockHash()), blockindex); |
|
||||||
} |
|
||||||
|
|
||||||
bool CBlockTreeDB::ReadBestInvalidWork(CBigNum& bnBestInvalidWork) |
|
||||||
{ |
|
||||||
return Read('I', bnBestInvalidWork); |
|
||||||
} |
|
||||||
|
|
||||||
bool CBlockTreeDB::WriteBestInvalidWork(CBigNum bnBestInvalidWork) |
|
||||||
{ |
|
||||||
return Write('I', bnBestInvalidWork); |
|
||||||
} |
|
||||||
|
|
||||||
bool CBlockTreeDB::WriteBlockFileInfo(int nFile, const CBlockFileInfo &info) { |
|
||||||
return Write(make_pair('f', nFile), info); |
|
||||||
} |
|
||||||
|
|
||||||
bool CBlockTreeDB::ReadBlockFileInfo(int nFile, CBlockFileInfo &info) { |
|
||||||
return Read(make_pair('f', nFile), info); |
|
||||||
} |
|
||||||
|
|
||||||
bool CBlockTreeDB::WriteLastBlockFile(int nFile) { |
|
||||||
return Write('l', nFile); |
|
||||||
} |
|
||||||
|
|
||||||
bool CBlockTreeDB::ReadLastBlockFile(int &nFile) { |
|
||||||
return Read('l', nFile); |
|
||||||
} |
|
||||||
|
|
||||||
CCoinsViewDB::CCoinsViewDB() : db() {} |
|
||||||
bool CCoinsViewDB::GetCoins(uint256 txid, CCoins &coins) { return db.ReadCoins(txid, coins); } |
|
||||||
bool CCoinsViewDB::SetCoins(uint256 txid, const CCoins &coins) { return db.WriteCoins(txid, coins); } |
|
||||||
bool CCoinsViewDB::HaveCoins(uint256 txid) { return db.HaveCoins(txid); } |
|
||||||
CBlockIndex *CCoinsViewDB::GetBestBlock() { |
|
||||||
uint256 hashBestChain; |
|
||||||
if (!db.ReadHashBestChain(hashBestChain)) |
|
||||||
return NULL; |
|
||||||
std::map<uint256, CBlockIndex*>::iterator it = mapBlockIndex.find(hashBestChain); |
|
||||||
if (it == mapBlockIndex.end()) |
|
||||||
return NULL; |
|
||||||
return it->second; |
|
||||||
} |
|
||||||
bool CCoinsViewDB::SetBestBlock(CBlockIndex *pindex) { return db.WriteHashBestChain(pindex->GetBlockHash()); } |
|
||||||
bool CCoinsViewDB::BatchWrite(const std::map<uint256, CCoins> &mapCoins, CBlockIndex *pindex) { |
|
||||||
printf("Committing %u changed transactions to coin database...\n", (unsigned int)mapCoins.size()); |
|
||||||
|
|
||||||
if (!db.TxnBegin()) |
|
||||||
return false; |
|
||||||
bool fOk = true; |
|
||||||
for (std::map<uint256, CCoins>::const_iterator it = mapCoins.begin(); it != mapCoins.end(); it++) { |
|
||||||
fOk = db.WriteCoins(it->first, it->second); |
|
||||||
if (!fOk) |
|
||||||
break; |
|
||||||
} |
|
||||||
if (fOk) |
|
||||||
fOk = db.WriteHashBestChain(pindex->GetBlockHash()); |
|
||||||
|
|
||||||
if (!fOk) |
|
||||||
db.TxnAbort(); |
|
||||||
else |
|
||||||
fOk = db.TxnCommit(); |
|
||||||
|
|
||||||
return fOk; |
|
||||||
} |
|
||||||
|
|
||||||
|
|
||||||
bool CBlockTreeDB::LoadBlockIndexGuts() |
|
||||||
{ |
|
||||||
// Get database cursor
|
|
||||||
Dbc* pcursor = GetCursor(); |
|
||||||
if (!pcursor) |
|
||||||
return false; |
|
||||||
|
|
||||||
// Load mapBlockIndex
|
|
||||||
unsigned int fFlags = DB_SET_RANGE; |
|
||||||
loop |
|
||||||
{ |
|
||||||
// Read next record
|
|
||||||
CDataStream ssKey(SER_DISK, CLIENT_VERSION); |
|
||||||
if (fFlags == DB_SET_RANGE) |
|
||||||
ssKey << make_pair('b', uint256(0)); |
|
||||||
CDataStream ssValue(SER_DISK, CLIENT_VERSION); |
|
||||||
int ret = ReadAtCursor(pcursor, ssKey, ssValue, fFlags); |
|
||||||
fFlags = DB_NEXT; |
|
||||||
if (ret == DB_NOTFOUND) |
|
||||||
break; |
|
||||||
else if (ret != 0) |
|
||||||
return false; |
|
||||||
|
|
||||||
// Unserialize
|
|
||||||
|
|
||||||
try { |
|
||||||
char chType; |
|
||||||
ssKey >> chType; |
|
||||||
if (chType == 'b' && !fRequestShutdown) |
|
||||||
{ |
|
||||||
CDiskBlockIndex diskindex; |
|
||||||
ssValue >> diskindex; |
|
||||||
|
|
||||||
// Construct block index object
|
|
||||||
CBlockIndex* pindexNew = InsertBlockIndex(diskindex.GetBlockHash()); |
|
||||||
pindexNew->pprev = InsertBlockIndex(diskindex.hashPrev); |
|
||||||
pindexNew->nHeight = diskindex.nHeight; |
|
||||||
pindexNew->nFile = diskindex.nFile; |
|
||||||
pindexNew->nDataPos = diskindex.nDataPos; |
|
||||||
pindexNew->nUndoPos = diskindex.nUndoPos; |
|
||||||
pindexNew->nVersion = diskindex.nVersion; |
|
||||||
pindexNew->hashMerkleRoot = diskindex.hashMerkleRoot; |
|
||||||
pindexNew->nTime = diskindex.nTime; |
|
||||||
pindexNew->nBits = diskindex.nBits; |
|
||||||
pindexNew->nNonce = diskindex.nNonce; |
|
||||||
pindexNew->nStatus = diskindex.nStatus; |
|
||||||
pindexNew->nTx = diskindex.nTx; |
|
||||||
|
|
||||||
// Watch for genesis block
|
|
||||||
if (pindexGenesisBlock == NULL && diskindex.GetBlockHash() == hashGenesisBlock) |
|
||||||
pindexGenesisBlock = pindexNew; |
|
||||||
|
|
||||||
if (!pindexNew->CheckIndex()) |
|
||||||
return error("LoadBlockIndex() : CheckIndex failed: %s", pindexNew->ToString().c_str()); |
|
||||||
} |
|
||||||
else |
|
||||||
{ |
|
||||||
break; // if shutdown requested or finished loading block index
|
|
||||||
} |
|
||||||
} // try
|
|
||||||
catch (std::exception &e) { |
|
||||||
return error("%s() : deserialize error", __PRETTY_FUNCTION__); |
|
||||||
} |
|
||||||
} |
|
||||||
pcursor->close(); |
|
||||||
|
|
||||||
return true; |
|
||||||
} |
|
||||||
@ -1,61 +0,0 @@ |
|||||||
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
|
||||||
// Copyright (c) 2009-2012 The Bitcoin developers
|
|
||||||
// Distributed under the MIT/X11 software license, see the accompanying
|
|
||||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
||||||
#ifndef BITCOIN_TXDB_BDB_H |
|
||||||
#define BITCOIN_TXDB_BDB_H |
|
||||||
|
|
||||||
#include "db.h" |
|
||||||
|
|
||||||
/** Access to the transaction database (coins.dat) */ |
|
||||||
class CCoinsDB : public CDB |
|
||||||
{ |
|
||||||
public: |
|
||||||
CCoinsDB() : CDB("coins.dat", "cr+") { } |
|
||||||
private: |
|
||||||
CCoinsDB(const CCoinsDB&); |
|
||||||
void operator=(const CCoinsDB&); |
|
||||||
public: |
|
||||||
bool ReadCoins(uint256 hash, CCoins &coins); |
|
||||||
bool WriteCoins(uint256 hash, const CCoins& coins); |
|
||||||
bool HaveCoins(uint256 hash); |
|
||||||
bool ReadHashBestChain(uint256& hashBestChain); |
|
||||||
bool WriteHashBestChain(uint256 hashBestChain); |
|
||||||
}; |
|
||||||
|
|
||||||
/** CCoinsView backed by a CCoinsDB */ |
|
||||||
class CCoinsViewDB : public CCoinsView |
|
||||||
{ |
|
||||||
protected: |
|
||||||
CCoinsDB db; |
|
||||||
public: |
|
||||||
CCoinsViewDB(); |
|
||||||
|
|
||||||
bool GetCoins(uint256 txid, CCoins &coins); |
|
||||||
bool SetCoins(uint256 txid, const CCoins &coins); |
|
||||||
bool HaveCoins(uint256 txid); |
|
||||||
CBlockIndex *GetBestBlock(); |
|
||||||
bool SetBestBlock(CBlockIndex *pindex); |
|
||||||
bool BatchWrite(const std::map<uint256, CCoins> &mapCoins, CBlockIndex *pindex); |
|
||||||
}; |
|
||||||
|
|
||||||
/** Access to the block database (blktree.dat) */ |
|
||||||
class CBlockTreeDB : public CDB |
|
||||||
{ |
|
||||||
public: |
|
||||||
CBlockTreeDB() : CDB("blktree.dat", "cr+") { } |
|
||||||
private: |
|
||||||
CBlockTreeDB(const CBlockTreeDB&); |
|
||||||
void operator=(const CBlockTreeDB&); |
|
||||||
public: |
|
||||||
bool WriteBlockIndex(const CDiskBlockIndex& blockindex); |
|
||||||
bool ReadBestInvalidWork(CBigNum& bnBestInvalidWork); |
|
||||||
bool WriteBestInvalidWork(CBigNum bnBestInvalidWork); |
|
||||||
bool ReadBlockFileInfo(int nFile, CBlockFileInfo &fileinfo); |
|
||||||
bool WriteBlockFileInfo(int nFile, const CBlockFileInfo &fileinfo); |
|
||||||
bool ReadLastBlockFile(int &nFile); |
|
||||||
bool WriteLastBlockFile(int nFile); |
|
||||||
bool LoadBlockIndexGuts(); |
|
||||||
}; |
|
||||||
|
|
||||||
#endif // BITCOIN_TXDB_BDB_H
|
|
||||||
@ -1,47 +0,0 @@ |
|||||||
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
|
||||||
// Copyright (c) 2009-2012 The Bitcoin developers
|
|
||||||
// Distributed under the MIT/X11 software license, see the accompanying
|
|
||||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
||||||
#ifndef BITCOIN_TXDB_LEVELDB_H |
|
||||||
#define BITCOIN_TXDB_LEVELDB_H |
|
||||||
|
|
||||||
#include "main.h" |
|
||||||
#include "leveldb.h" |
|
||||||
|
|
||||||
/** CCoinsView backed by the LevelDB coin database (coins/) */ |
|
||||||
class CCoinsViewDB : public CCoinsView |
|
||||||
{ |
|
||||||
protected: |
|
||||||
CLevelDB db; |
|
||||||
public: |
|
||||||
CCoinsViewDB(bool fMemory = false); |
|
||||||
|
|
||||||
bool GetCoins(uint256 txid, CCoins &coins); |
|
||||||
bool SetCoins(uint256 txid, const CCoins &coins); |
|
||||||
bool HaveCoins(uint256 txid); |
|
||||||
CBlockIndex *GetBestBlock(); |
|
||||||
bool SetBestBlock(CBlockIndex *pindex); |
|
||||||
bool BatchWrite(const std::map<uint256, CCoins> &mapCoins, CBlockIndex *pindex); |
|
||||||
bool GetStats(CCoinsStats &stats); |
|
||||||
}; |
|
||||||
|
|
||||||
/** Access to the block database (blktree/) */ |
|
||||||
class CBlockTreeDB : public CLevelDB |
|
||||||
{ |
|
||||||
public: |
|
||||||
CBlockTreeDB(bool fMemory = false); |
|
||||||
private: |
|
||||||
CBlockTreeDB(const CBlockTreeDB&); |
|
||||||
void operator=(const CBlockTreeDB&); |
|
||||||
public: |
|
||||||
bool WriteBlockIndex(const CDiskBlockIndex& blockindex); |
|
||||||
bool ReadBestInvalidWork(CBigNum& bnBestInvalidWork); |
|
||||||
bool WriteBestInvalidWork(const CBigNum& bnBestInvalidWork); |
|
||||||
bool ReadBlockFileInfo(int nFile, CBlockFileInfo &fileinfo); |
|
||||||
bool WriteBlockFileInfo(int nFile, const CBlockFileInfo &fileinfo); |
|
||||||
bool ReadLastBlockFile(int &nFile); |
|
||||||
bool WriteLastBlockFile(int nFile); |
|
||||||
bool LoadBlockIndexGuts(); |
|
||||||
}; |
|
||||||
|
|
||||||
#endif // BITCOIN_TXDB_LEVELDB_H
|
|
||||||
Loading…
Reference in new issue