mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-28 17:56:50 +00:00

* Use selectedParent instead of selectedTip for non-selectedTip blocks in restoreSingleBlockStatus * Cache the selectedParent for re-use in a resolveSingleBlockStatus chain * Implement and use reverseUTXOSet * Reverse blocks in correct order * Support resolveBlockStatus without separate stagingAreas for usage of testConsensus * Handle the case where the tip of the resolved block is not the next selectedTip * Unify isResolveTip * Some minor fixes and cleanup * Add full finality window re-org test to stability-slow * rename: useSeparateStagingAreasPerBlock -> useSeparateStagingAreaPerBlock * Better logs in resolveSingleBlockStatus * A few retouches to reverseUTXODiffs * TEMPORARY COMMIT: EXTRAT ALL DIFFFROMS TO SEPARATE METHODS * TEMPORARY COMMIT: REMOVE DIFFICULTY CHECKS IN DEVNET * Don't pre-allocate in utxo-algebra, since the numbers are not known ahead-of-time * Add some logs to reverseUTXODiffs * Revert "TEMPORARY COMMIT: REMOVE DIFFICULTY CHECKS IN DEVNET" This reverts commit c0af9dc6ade78a914c970e11bc63c34605565f57. * Revert "TEMPORARY COMMIT: EXTRAT ALL DIFFFROMS TO SEPARATE METHODS" This reverts commit 4fcca1b48c3a1183598833a355b9bfaf169edba1. * Remove redundant paranthesis * Revise some logs messages * Rename:oneBlockBeforeCurrentUTXOSet -> lastResolvedBlockUTXOSet * Don't break if the block was resolved as invalid * rename unverifiedBlocks to recentlyVerifiedBlcks in reverseUTXODiffs * Add errors.New to the panic, for a stack trace * Reverse the UTXODiffs after the main block has been commited * Use the correct value for previousUTXODiff * Add test for ReverseUTXODiff * Fix some names and comments * Update TestReverseUTXODiffs to use consensus.Config * Fix comments mentioning 'oneBlockBeforeTip'
165 lines
4.1 KiB
Go
165 lines
4.1 KiB
Go
package main
|
|
|
|
import (
|
|
"compress/gzip"
|
|
"fmt"
|
|
"math/rand"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/kaspanet/kaspad/domain/consensus/model"
|
|
|
|
"github.com/kaspanet/kaspad/domain/consensus"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/testapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/consensushashing"
|
|
"github.com/kaspanet/kaspad/domain/dagconfig"
|
|
)
|
|
|
|
func testReorg(cfg *configFlags) {
|
|
consensusConfig := consensus.Config{Params: dagconfig.DevnetParams}
|
|
consensusConfig.SkipProofOfWork = true
|
|
consensusConfig.DisableDifficultyAdjustment = true
|
|
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(&consensusConfig, "ReorgHonest")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
f, err := os.Open(cfg.DAGFile)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
gzipReader, err := gzip.NewReader(f)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer gzipReader.Close()
|
|
|
|
_, err = tc.MineJSON(gzipReader, testapi.MineJSONBlockTypeUTXOValidBlock)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
tcAttacker, teardownAttacker, err := factory.NewTestConsensus(&consensusConfig, "ReorgAttacker")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer teardownAttacker(false)
|
|
|
|
virtualSelectedParent, err := tc.GetVirtualSelectedParent()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
stagingArea := model.NewStagingArea()
|
|
virtualSelectedParentGHOSTDAGData, err := tc.GHOSTDAGDataStore().Get(tc.DatabaseContext(), stagingArea, virtualSelectedParent)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Infof("Selected tip blue score %d", virtualSelectedParentGHOSTDAGData.BlueScore())
|
|
|
|
sideChain := make([]*externalapi.DomainBlock, 0)
|
|
|
|
for i := uint64(0); ; i++ {
|
|
tips, err := tcAttacker.Tips()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
block, _, err := tcAttacker.BuildBlockWithParents(tips, nil, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// We change the nonce of the first block so its hash won't be similar to any of the
|
|
// honest DAG blocks. As a result the rest of the side chain should have unique hashes
|
|
// as well.
|
|
if i == 0 {
|
|
mutableHeader := block.Header.ToMutable()
|
|
mutableHeader.SetNonce(uint64(rand.NewSource(84147).Int63()))
|
|
block.Header = mutableHeader.ToImmutable()
|
|
}
|
|
|
|
_, err = tcAttacker.ValidateAndInsertBlock(block)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
sideChain = append(sideChain, block)
|
|
|
|
if i%100 == 0 {
|
|
log.Infof("Attacker side chain mined %d blocks", i)
|
|
}
|
|
|
|
blockHash := consensushashing.BlockHash(block)
|
|
ghostdagData, err := tcAttacker.GHOSTDAGDataStore().Get(tcAttacker.DatabaseContext(), stagingArea, blockHash)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if virtualSelectedParentGHOSTDAGData.BlueWork().Cmp(ghostdagData.BlueWork()) == -1 {
|
|
break
|
|
}
|
|
}
|
|
|
|
sideChainTipHash := consensushashing.BlockHash(sideChain[len(sideChain)-1])
|
|
sideChainTipGHOSTDAGData, err := tcAttacker.GHOSTDAGDataStore().Get(tcAttacker.DatabaseContext(), stagingArea, sideChainTipHash)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Infof("Side chain tip (%s) blue score %d", sideChainTipHash, sideChainTipGHOSTDAGData.BlueScore())
|
|
|
|
doneChan := make(chan struct{})
|
|
spawn("add-sidechain-to-honest", func() {
|
|
for i, block := range sideChain {
|
|
if i%100 == 0 {
|
|
log.Infof("Validated %d blocks from the attacker chain", i)
|
|
}
|
|
_, err := tc.ValidateAndInsertBlock(block)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
doneChan <- struct{}{}
|
|
})
|
|
|
|
const timeout = 12 * time.Hour
|
|
select {
|
|
case <-doneChan:
|
|
case <-time.After(timeout):
|
|
fail("Adding the side chain took more than %s", timeout)
|
|
}
|
|
|
|
sideChainTipGHOSTDAGData, err = tc.GHOSTDAGDataStore().Get(tc.DatabaseContext(), stagingArea, sideChainTipHash)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Infof("Side chain tip (%s) blue score %d", sideChainTipHash, sideChainTipGHOSTDAGData.BlueScore())
|
|
|
|
newVirtualSelectedParent, err := tc.GetVirtualSelectedParent()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if !newVirtualSelectedParent.Equal(sideChainTipHash) {
|
|
fail("No reorg happened")
|
|
}
|
|
}
|
|
|
|
func fail(format string, args ...interface{}) {
|
|
msg := fmt.Sprintf(format, args...)
|
|
fmt.Fprintln(os.Stderr, msg)
|
|
log.Criticalf(msg)
|
|
backendLog.Close()
|
|
os.Exit(1)
|
|
}
|