hacktober fest #214
2 changed files with 60 additions and 12 deletions
|
@ -1595,10 +1595,12 @@ static bool ApplyTxInUndo(const CTxInUndo& undo, CCoinsViewCache& view, CNCCTrie
|
|||
std::string name(vvchParams[0].begin(), vvchParams[0].end());
|
||||
LogPrintf("%s: Restoring %s to the NCC trie due to a block being disconnected\n", __func__, name.c_str());
|
||||
int nValidHeight = undo.nNCCValidHeight;
|
||||
assert(nValidHeight > 0 && nValidHeight >= coins->nHeight);
|
||||
if (nValidHeight > 0 && nValidHeight >= coins->nHeight)
|
||||
{
|
||||
if (!trieCache.undoSpendClaim(name, out.hash, out.n, undo.txout.nValue, coins->nHeight, nValidHeight))
|
||||
LogPrintf("%s: Something went wrong inserting the name\n", __func__);
|
||||
}
|
||||
}
|
||||
|
||||
coins->vout[out.n] = undo.txout;
|
||||
|
||||
|
@ -1874,14 +1876,15 @@ bool ConnectBlock(const CBlock& block, CValidationState& state, CBlockIndex* pin
|
|||
std::string name(vvchParams[0].begin(), vvchParams[0].end());
|
||||
int nValidAtHeight;
|
||||
LogPrintf("%s: Removing %s from the ncc trie. Tx: %s, nOut: %d\n", __func__, name, txin.prevout.hash.GetHex(), txin.prevout.n);
|
||||
if (!trieCache.spendClaim(name, txin.prevout.hash, txin.prevout.n, coins->nHeight, nValidAtHeight))
|
||||
LogPrintf("%s: Something went wrong removing the name\n", __func__);
|
||||
if (trieCache.spendClaim(name, txin.prevout.hash, txin.prevout.n, coins->nHeight, nValidAtHeight))
|
||||
{
|
||||
mNCCUndoHeights[i] = nValidAtHeight;
|
||||
std::pair<uint256, unsigned int> val(txin.prevout.hash, txin.prevout.n);
|
||||
std::pair<std::string, std::pair<uint256, unsigned int> > entry(name, val);
|
||||
spentClaims.insert(entry);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (unsigned int i = 0; i < tx.vout.size(); ++i)
|
||||
{
|
||||
|
|
|
@ -139,7 +139,7 @@ const unsigned int expire_nonces[] = {
|
|||
19076, 179631, 36500, 152550, 56501, 81202, 77561, 134713, 81130, 22321,
|
||||
112081, 32992, 144573, 21369, 2471, 18371, 63050, 44211, 6147, 206052,
|
||||
34252, 534, 20176, 58035, 24268, 19608, 37770, 57588, 120961, 58415,
|
||||
4780, 4614, 229320, 42279, 41295, 23501, 78183};
|
||||
4780, 4614, 229320, 42279, 41295, 23501, 78183, 171983, 24229};
|
||||
|
||||
BOOST_FIXTURE_TEST_SUITE(ncctrie_tests, TestingSetup)
|
||||
|
||||
|
@ -199,9 +199,7 @@ bool CreateBlock(CBlockTemplate* pblocktemplate, int nonce)
|
|||
pblock->vtx[0] = CTransaction(txCoinbase);
|
||||
pblock->hashMerkleRoot = pblock->BuildMerkleTree();
|
||||
//if (nonce != -1)
|
||||
//{
|
||||
pblock->nNonce = nonce;
|
||||
//}
|
||||
/*else
|
||||
{
|
||||
for (int i = 0; ; ++i)
|
||||
|
@ -1071,6 +1069,8 @@ BOOST_AUTO_TEST_CASE(ncctrie_claim_expiration)
|
|||
BOOST_CHECK(pnccTrie->queueEmpty());
|
||||
BOOST_CHECK(!pnccTrie->expirationQueueEmpty());
|
||||
|
||||
mempool.clear();
|
||||
|
||||
// advance until the expiration event occurs. verify the event occurs on time.
|
||||
|
||||
BOOST_CHECK(pblocktemplate = CreateNewBlock(scriptPubKey));
|
||||
|
@ -1095,6 +1095,31 @@ BOOST_AUTO_TEST_CASE(ncctrie_claim_expiration)
|
|||
BOOST_CHECK(pnccTrie->queueEmpty());
|
||||
BOOST_CHECK(pnccTrie->expirationQueueEmpty());
|
||||
|
||||
// spend the expired claim
|
||||
|
||||
AddToMempool(tx2);
|
||||
|
||||
BOOST_CHECK(pblocktemplate = CreateNewBlock(scriptPubKey));
|
||||
BOOST_CHECK(pblocktemplate->block.vtx.size() == 2);
|
||||
pblocktemplate->block.hashPrevBlock = chainActive.Tip()->GetBlockHash();
|
||||
BOOST_CHECK(CreateBlock(pblocktemplate, expire_nonces[block_counter++]));
|
||||
blocks_to_invalidate.push_back(pblocktemplate->block.hashPrevBlock);
|
||||
delete pblocktemplate;
|
||||
|
||||
BOOST_CHECK(pnccTrie->empty());
|
||||
BOOST_CHECK(pnccTrie->queueEmpty());
|
||||
BOOST_CHECK(pnccTrie->expirationQueueEmpty());
|
||||
|
||||
// undo the spend. verify everything remains empty.
|
||||
|
||||
BOOST_CHECK(RemoveBlock(blocks_to_invalidate.back()));
|
||||
blocks_to_invalidate.pop_back();
|
||||
BOOST_CHECK(pnccTrie->empty());
|
||||
BOOST_CHECK(pnccTrie->queueEmpty());
|
||||
BOOST_CHECK(pnccTrie->expirationQueueEmpty());
|
||||
|
||||
mempool.clear();
|
||||
|
||||
// roll back to before the expiration event. verify the claim is reinserted. verify the expiration event is scheduled again.
|
||||
|
||||
BOOST_CHECK(RemoveBlock(blocks_to_invalidate.back()));
|
||||
|
@ -1103,6 +1128,26 @@ BOOST_AUTO_TEST_CASE(ncctrie_claim_expiration)
|
|||
BOOST_CHECK(pnccTrie->queueEmpty());
|
||||
BOOST_CHECK(!pnccTrie->expirationQueueEmpty());
|
||||
|
||||
// verify the expiration event happens at the right time again
|
||||
|
||||
BOOST_CHECK(pblocktemplate = CreateNewBlock(scriptPubKey));
|
||||
pblocktemplate->block.hashPrevBlock = chainActive.Tip()->GetBlockHash();
|
||||
BOOST_CHECK(CreateBlock(pblocktemplate, expire_nonces[block_counter++]));
|
||||
blocks_to_invalidate.push_back(pblocktemplate->block.hashPrevBlock);
|
||||
delete pblocktemplate;
|
||||
|
||||
BOOST_CHECK(pnccTrie->empty());
|
||||
BOOST_CHECK(pnccTrie->queueEmpty());
|
||||
BOOST_CHECK(pnccTrie->expirationQueueEmpty());
|
||||
|
||||
// roll back to before the expiration event. verify it gets reinserted and expiration gets scheduled.
|
||||
|
||||
BOOST_CHECK(RemoveBlock(blocks_to_invalidate.back()));
|
||||
blocks_to_invalidate.pop_back();
|
||||
BOOST_CHECK(!pnccTrie->empty());
|
||||
BOOST_CHECK(pnccTrie->queueEmpty());
|
||||
BOOST_CHECK(!pnccTrie->expirationQueueEmpty());
|
||||
|
||||
// roll back to before the claim is valid. verify the claim is removed but the expiration event is not.
|
||||
|
||||
BOOST_CHECK(RemoveBlock(blocks_to_invalidate.back()));
|
||||
|
|
Loading…
Reference in a new issue