Preliminary undo file creation
Create files (one per block) with undo information for the transactions in it.
This commit is contained in:
parent
630fd8dcb6
commit
8adf48dc9b
2 changed files with 32 additions and 0 deletions
17
src/main.cpp
17
src/main.cpp
|
@ -1432,6 +1432,8 @@ bool CBlock::ConnectBlock(CTxDB& txdb, CBlockIndex* pindex, bool fJustCheck)
|
|||
else
|
||||
nTxPos = ::GetSerializeSize(CBlock(), SER_DISK, CLIENT_VERSION) - 1 + GetSizeOfCompactSize(vtx.size());
|
||||
|
||||
CBlockUndo blockundo;
|
||||
|
||||
map<uint256, CTxIndex> mapQueuedChanges;
|
||||
int64 nFees = 0;
|
||||
unsigned int nSigOps = 0;
|
||||
|
@ -1461,6 +1463,8 @@ bool CBlock::ConnectBlock(CTxDB& txdb, CBlockIndex* pindex, bool fJustCheck)
|
|||
MapPrevTx mapInputs;
|
||||
if (!tx.IsCoinBase())
|
||||
{
|
||||
CTxUndo undo;
|
||||
|
||||
bool fInvalid;
|
||||
if (!tx.FetchInputs(txdb, mapQueuedChanges, true, false, mapInputs, fInvalid))
|
||||
return false;
|
||||
|
@ -1477,8 +1481,14 @@ bool CBlock::ConnectBlock(CTxDB& txdb, CBlockIndex* pindex, bool fJustCheck)
|
|||
|
||||
nFees += tx.GetValueIn(mapInputs)-tx.GetValueOut();
|
||||
|
||||
BOOST_FOREACH(const CTxIn &in, tx.vin) {
|
||||
undo.vprevout.push_back(CTxInUndo(mapInputs[in.prevout.hash].second.vout[in.prevout.n], pindex->nHeight));
|
||||
}
|
||||
|
||||
if (!tx.ConnectInputs(mapInputs, mapQueuedChanges, posThisTx, pindex, true, false, fStrictPayToScriptHash))
|
||||
return false;
|
||||
|
||||
blockundo.vtxundo.push_back(undo);
|
||||
}
|
||||
|
||||
mapQueuedChanges[hashTx] = CTxIndex(posThisTx, tx.vout.size());
|
||||
|
@ -1507,6 +1517,13 @@ bool CBlock::ConnectBlock(CTxDB& txdb, CBlockIndex* pindex, bool fJustCheck)
|
|||
return error("ConnectBlock() : WriteBlockIndex failed");
|
||||
}
|
||||
|
||||
// Write undo information to disk
|
||||
if (pindex->nHeight > Checkpoints::GetTotalBlocksEstimate())
|
||||
{
|
||||
CAutoFile fileUndo(fopen(pindex->GetBlockPos().GetUndoFile(GetDataDir()).string().c_str(), "wb"), SER_DISK, CLIENT_VERSION);
|
||||
fileUndo << blockundo;
|
||||
}
|
||||
|
||||
// Watch for transactions paying to me
|
||||
BOOST_FOREACH(CTransaction& tx, vtx)
|
||||
SyncWithWallets(tx, this, true);
|
||||
|
|
15
src/main.h
15
src/main.h
|
@ -155,6 +155,10 @@ public:
|
|||
return GetDirectory(base) / strprintf("%08u%s.blk", nHeight, GetAlternative().c_str());
|
||||
}
|
||||
|
||||
boost::filesystem::path GetUndoFile(const boost::filesystem::path &base) const {
|
||||
return GetDirectory(base) / strprintf("%08u%s.und", nHeight, GetAlternative().c_str());
|
||||
}
|
||||
|
||||
// TODO: make thread-safe (lockfile, atomic file creation, ...?)
|
||||
void MakeUnique(const boost::filesystem::path &base) {
|
||||
while (boost::filesystem::exists(GetFileName(base)))
|
||||
|
@ -810,6 +814,17 @@ public:
|
|||
)
|
||||
};
|
||||
|
||||
/** Undo information for a CBlock */
|
||||
class CBlockUndo
|
||||
{
|
||||
public:
|
||||
std::vector<CTxUndo> vtxundo;
|
||||
|
||||
IMPLEMENT_SERIALIZE(
|
||||
READWRITE(vtxundo);
|
||||
)
|
||||
};
|
||||
|
||||
/** pruned version of CTransaction: only retains metadata and unspent transaction outputs
|
||||
*
|
||||
* Serialized format:
|
||||
|
|
Loading…
Reference in a new issue