Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-06-06 22:26:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NOD-1532] Check UTXOCommitment for all blocks

Browse Source
This commit is contained in:
Mike Zak 2020-11-16 12:42:26 +02:00 committed by Svarog
parent fca8ed57bd
commit 151910c27a
1 changed files with 44 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
domain/consensus/processes/consensusstatemanager/calculate_past_utxo_test.go
View File

@ -3,6 +3,8 @@ package consensusstatemanager_test
import ( import (
"testing" "testing"
"github.com/kaspanet/kaspad/domain/consensus/model/testapi"
"github.com/kaspanet/kaspad/domain/consensus/utils/transactionhelper" "github.com/kaspanet/kaspad/domain/consensus/utils/transactionhelper"
"github.com/kaspanet/kaspad/domain/consensus" "github.com/kaspanet/kaspad/domain/consensus"
@ -34,6 +36,7 @@ func TestUTXOCommitment(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("Error creating block A: %+v", err) t.Fatalf("Error creating block A: %+v", err)
} }
checkBlockUTXOCommitment(t, consensus, blockAHash, "A")
// Block B: // Block B:
blockBHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockAHash}, nil, nil) blockBHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockAHash}, nil, nil)
if err != nil { if err != nil {
@ -43,11 +46,13 @@ func TestUTXOCommitment(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("Error getting block B: %+v", err) t.Fatalf("Error getting block B: %+v", err)
} }
checkBlockUTXOCommitment(t, consensus, blockBHash, "B")
// Block C: // Block C:
blockCHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockBHash}, nil, nil) blockCHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockBHash}, nil, nil)
if err != nil { if err != nil {
t.Fatalf("Error creating block C: %+v", err) t.Fatalf("Error creating block C: %+v", err)
} }
checkBlockUTXOCommitment(t, consensus, blockCHash, "C")
// Block D: // Block D:
blockDTransaction, err := testutils.CreateTransaction( blockDTransaction, err := testutils.CreateTransaction(
blockB.Transactions[transactionhelper.CoinbaseTransactionIndex]) blockB.Transactions[transactionhelper.CoinbaseTransactionIndex])
@ -59,48 +64,53 @@ func TestUTXOCommitment(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("Error creating block D: %+v", err) t.Fatalf("Error creating block D: %+v", err)
} }
checkBlockUTXOCommitment(t, consensus, blockDHash, "D")
// Block E: // Block E:
blockEHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockCHash, blockDHash}, nil, nil) blockEHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockCHash, blockDHash}, nil, nil)
if err != nil { if err != nil {
t.Fatalf("Error creating block E: %+v", err) t.Fatalf("Error creating block E: %+v", err)
} }
blockE, err := consensus.GetBlock(blockEHash) checkBlockUTXOCommitment(t, consensus, blockEHash, "E")
if err != nil {
t.Fatalf("Error getting block E: %+v", err)
}
// Get the past UTXO set of block E
csm := consensus.ConsensusStateManager()
utxoSetIterator, err := csm.RestorePastUTXOSetIterator(blockEHash)
if err != nil {
t.Fatalf("Error restoring past UTXO of block E: %+v", err)
}
// Build a Multiset for block E
ms := multiset.New()
for utxoSetIterator.Next() {
outpoint, entry, err := utxoSetIterator.Get()
if err != nil {
t.Fatalf("Error getting from UTXOSet iterator: %+v", err)
}
err = consensus.ConsensusStateManager().AddUTXOToMultiset(ms, entry, outpoint)
if err != nil {
t.Fatalf("Error adding utxo to multiset: %+v", err)
}
}
// Turn the multiset into a UTXO commitment
utxoCommitment := ms.Hash()
// Make sure that the two commitments are equal
if *utxoCommitment != blockE.Header.UTXOCommitment {
t.Fatalf("TestUTXOCommitment: calculated UTXO commitment and "+
"actual UTXO commitment don't match. Want: %s, got: %s",
utxoCommitment, blockE.Header.UTXOCommitment)
}
}) })
} }
func checkBlockUTXOCommitment(t *testing.T, consensus testapi.TestConsensus, blockHash *externalapi.DomainHash, blockName string) {
block, err := consensus.GetBlock(blockHash)
if err != nil {
t.Fatalf("Error getting block %s: %+v", blockName, err)
}
// Get the past UTXO set of block
csm := consensus.ConsensusStateManager()
utxoSetIterator, err := csm.RestorePastUTXOSetIterator(blockHash)
if err != nil {
t.Fatalf("Error restoring past UTXO of block %s: %+v", blockName, err)
}
// Build a Multiset
ms := multiset.New()
for utxoSetIterator.Next() {
outpoint, entry, err := utxoSetIterator.Get()
if err != nil {
t.Fatalf("Error getting from UTXOSet iterator: %+v", err)
}
err = consensus.ConsensusStateManager().AddUTXOToMultiset(ms, entry, outpoint)
if err != nil {
t.Fatalf("Error adding utxo to multiset: %+v", err)
}
}
// Turn the multiset into a UTXO commitment
utxoCommitment := ms.Hash()
// Make sure that the two commitments are equal
if *utxoCommitment != block.Header.UTXOCommitment {
t.Fatalf("TestUTXOCommitment: calculated UTXO commitment for block %s and "+
"actual UTXO commitment don't match. Want: %s, got: %s", blockName,
utxoCommitment, block.Header.UTXOCommitment)
}
}
func TestPastUTXOMultiset(t *testing.T) { func TestPastUTXOMultiset(t *testing.T) {
testutils.ForAllNets(t, true, func(t *testing.T, params *dagconfig.Params) { testutils.ForAllNets(t, true, func(t *testing.T, params *dagconfig.Params) {
factory := consensus.NewFactory() factory := consensus.NewFactory()