@ -39,17 +39,15 @@ import (
func TestHeaderVerification ( t * testing . T ) {
// Create a simple chain to verify
var (
testdb = rawdb . NewMemoryDatabase ( )
gspec = & Genesis { Config : params . TestChainConfig }
genesis = gspec . MustCommit ( testdb )
blocks , _ = GenerateChain ( params . TestChainConfig , genesis , ethash . NewFaker ( ) , testdb , 8 , nil )
gspec = & Genesis { Config : params . TestChainConfig }
_ , blocks , _ = GenerateChainWithGenesis ( gspec , ethash . NewFaker ( ) , 8 , nil )
)
headers := make ( [ ] * types . Header , len ( blocks ) )
for i , block := range blocks {
headers [ i ] = block . Header ( )
}
// Run the header checker for blocks one-by-one, checking for both valid and invalid nonces
chain , _ := NewBlockChain ( testdb , nil , gspec , nil , ethash . NewFaker ( ) , vm . Config { } , nil , nil )
chain , _ := NewBlockChain ( rawdb . NewMemoryDatabase ( ) , nil , gspec , nil , ethash . NewFaker ( ) , vm . Config { } , nil , nil )
defer chain . Stop ( )
for i := 0 ; i < len ( blocks ) ; i ++ {
@ -89,21 +87,21 @@ func TestHeaderVerificationForMergingEthash(t *testing.T) { testHeaderVerificati
// Tests the verification for eth1/2 merging, including pre-merge and post-merge
func testHeaderVerificationForMerging ( t * testing . T , isClique bool ) {
var (
testdb = rawdb . NewMemoryDatabase ( )
gspec * Genesis
preBlocks [ ] * types . Block
postBlocks [ ] * types . Block
runEngine consensus . Engine
genspec * Genesis
engine consensus . Engine
merger = consensus . NewMerger ( rawdb . NewMemoryDatabase ( ) )
)
if isClique {
var (
key , _ = crypto . HexToECDSA ( "b71c71a67e1177ad4e901695e1b4b9ee17ae16c6668d313eac2f96dbcda3f291" )
addr = crypto . PubkeyToAddress ( key . PublicKey )
engine = clique . New ( params . AllCliqueProtocolChanges . Clique , testdb )
config = * params . AllCliqueProtocolChanges
)
genspec = & Genesis {
Config : params . AllCliqueProtocolChanges ,
engine = beacon . New ( clique . New ( params . AllCliqueProtocolChanges . Clique , rawdb . NewMemoryDatabase ( ) ) )
gspec = & Genesis {
Config : & config ,
ExtraData : make ( [ ] byte , 32 + common . AddressLength + crypto . SignatureLength ) ,
Alloc : map [ common . Address ] GenesisAccount {
addr : { Balance : big . NewInt ( 1 ) } ,
@ -111,50 +109,44 @@ func testHeaderVerificationForMerging(t *testing.T, isClique bool) {
BaseFee : big . NewInt ( params . InitialBaseFee ) ,
Difficulty : new ( big . Int ) ,
}
copy ( genspec . ExtraData [ 32 : ] , addr [ : ] )
genesis := genspec . MustCommit ( testdb )
copy ( gspec . ExtraData [ 32 : ] , addr [ : ] )
genEngine := beacon . New ( engine )
preBlocks , _ = GenerateChain ( params . AllCliqueProtocolChanges , genesis , genEngine , testdb , 8 , nil )
td := 0
for i , block := range preBlocks {
genDb , blocks , _ := GenerateChainWithGenesis ( gspec , engine , 8 , nil )
for i , block := range blocks {
header := block . Header ( )
if i > 0 {
header . ParentHash = preB locks[ i - 1 ] . Hash ( )
header . ParentHash = b locks[ i - 1 ] . Hash ( )
}
header . Extra = make ( [ ] byte , 32 + crypto . SignatureLength )
header . Difficulty = big . NewInt ( 2 )
sig , _ := crypto . Sign ( g enE ngine. SealHash ( header ) . Bytes ( ) , key )
sig , _ := crypto . Sign ( engine . SealHash ( header ) . Bytes ( ) , key )
copy ( header . Extra [ len ( header . Extra ) - crypto . SignatureLength : ] , sig )
preBlocks [ i ] = block . WithSeal ( header )
blocks [ i ] = block . WithSeal ( header )
// calculate td
td += int ( block . Difficulty ( ) . Uint64 ( ) )
}
config := * params . AllCliqueProtocolChanges
config . TerminalTotalDifficulty = big . NewInt ( int64 ( td ) )
postBlocks , _ = GenerateChain ( & config , preBlocks [ len ( preBlocks ) - 1 ] , genEngine , testdb , 8 , nil )
runEngine = beacon . New ( engine )
genspec . Config = & config
preBlocks = blocks
gspec . Config . TerminalTotalDifficulty = big . NewInt ( int64 ( td ) )
postBlocks , _ = GenerateChain ( gspec . Config , preBlocks [ len ( preBlocks ) - 1 ] , engine , genDb , 8 , nil )
} else {
genspec = & Genesis { Config : params . TestChainConfig }
genesis := genspec . MustCommit ( testdb )
genEngine : = beacon . New ( ethash . NewFaker ( ) )
config := * params . TestChainConfig
gspec = & Genesis { Config : & config }
engine = beacon . New ( ethash . NewFaker ( ) )
preBlocks , _ = GenerateChain ( params . TestChainConfig , genesis , genEngine , testdb , 8 , nil )
td := 0
genDb , blocks , _ := GenerateChainWithGenesis ( gspec , engine , 8 , nil )
for _ , block := range preBlocks {
// calculate td
td += int ( block . Difficulty ( ) . Uint64 ( ) )
}
config := * params . TestChainConfig
config . TerminalTotalDifficulty = big . NewInt ( int64 ( td ) )
postBlocks , _ = GenerateChain ( params . TestChainConfig , preBlocks [ len ( preBlocks ) - 1 ] , genEngine , testdb , 8 , nil )
runEngine = beacon . New ( ethash . NewFaker ( ) )
genspec . Config = & config
preBlocks = blocks
gspec . Config . TerminalTotalDifficulty = big . NewInt ( int64 ( td ) )
postBlocks , _ = GenerateChain ( gspec . Config , preBlocks [ len ( preBlocks ) - 1 ] , engine , genDb , 8 , nil )
}
// Assemble header batch
preHeaders := make ( [ ] * types . Header , len ( preBlocks ) )
for i , block := range preBlocks {
preHeaders [ i ] = block . Header ( )
@ -170,12 +162,12 @@ func testHeaderVerificationForMerging(t *testing.T, isClique bool) {
t . Logf ( "Log header after the merging %d: %v" , block . NumberU64 ( ) , string ( blob ) )
}
// Run the header checker for blocks one-by-one, checking for both valid and invalid nonces
chain , _ := NewBlockChain ( testdb , nil , gen spec , nil , runE ngine, vm . Config { } , nil , nil )
chain , _ := NewBlockChain ( rawdb . NewMemoryDatabase ( ) , nil , gspec , nil , e ngine, vm . Config { } , nil , nil )
defer chain . Stop ( )
// Verify the blocks before the merging
for i := 0 ; i < len ( preBlocks ) ; i ++ {
_ , results := runE ngine. VerifyHeaders ( chain , [ ] * types . Header { preHeaders [ i ] } , [ ] bool { true } )
_ , results := e ngine. VerifyHeaders ( chain , [ ] * types . Header { preHeaders [ i ] } , [ ] bool { true } )
// Wait for the verification result
select {
case result := <- results :
@ -200,7 +192,7 @@ func testHeaderVerificationForMerging(t *testing.T, isClique bool) {
// Verify the blocks after the merging
for i := 0 ; i < len ( postBlocks ) ; i ++ {
_ , results := runE ngine. VerifyHeaders ( chain , [ ] * types . Header { postHeaders [ i ] } , [ ] bool { true } )
_ , results := e ngine. VerifyHeaders ( chain , [ ] * types . Header { postHeaders [ i ] } , [ ] bool { true } )
// Wait for the verification result
select {
case result := <- results :
@ -232,7 +224,7 @@ func testHeaderVerificationForMerging(t *testing.T, isClique bool) {
headers = append ( headers , block . Header ( ) )
seals = append ( seals , true )
}
_ , results := runE ngine. VerifyHeaders ( chain , headers , seals )
_ , results := e ngine. VerifyHeaders ( chain , headers , seals )
for i := 0 ; i < len ( headers ) ; i ++ {
select {
case result := <- results :
@ -259,10 +251,8 @@ func TestHeaderConcurrentVerification32(t *testing.T) { testHeaderConcurrentVeri
func testHeaderConcurrentVerification ( t * testing . T , threads int ) {
// Create a simple chain to verify
var (
testdb = rawdb . NewMemoryDatabase ( )
gspec = & Genesis { Config : params . TestChainConfig }
genesis = gspec . MustCommit ( testdb )
blocks , _ = GenerateChain ( params . TestChainConfig , genesis , ethash . NewFaker ( ) , testdb , 8 , nil )
gspec = & Genesis { Config : params . TestChainConfig }
_ , blocks , _ = GenerateChainWithGenesis ( gspec , ethash . NewFaker ( ) , 8 , nil )
)
headers := make ( [ ] * types . Header , len ( blocks ) )
seals := make ( [ ] bool , len ( blocks ) )
@ -281,11 +271,11 @@ func testHeaderConcurrentVerification(t *testing.T, threads int) {
var results <- chan error
if valid {
chain , _ := NewBlockChain ( testdb , nil , gspec , nil , ethash . NewFaker ( ) , vm . Config { } , nil , nil )
chain , _ := NewBlockChain ( rawdb . NewMemoryDatabase ( ) , nil , gspec , nil , ethash . NewFaker ( ) , vm . Config { } , nil , nil )
_ , results = chain . engine . VerifyHeaders ( chain , headers , seals )
chain . Stop ( )
} else {
chain , _ := NewBlockChain ( testdb , nil , gspec , nil , ethash . NewFakeFailer ( uint64 ( len ( headers ) - 1 ) ) , vm . Config { } , nil , nil )
chain , _ := NewBlockChain ( rawdb . NewMemoryDatabase ( ) , nil , gspec , nil , ethash . NewFakeFailer ( uint64 ( len ( headers ) - 1 ) ) , vm . Config { } , nil , nil )
_ , results = chain . engine . VerifyHeaders ( chain , headers , seals )
chain . Stop ( )
}
@ -331,10 +321,8 @@ func TestHeaderConcurrentAbortion32(t *testing.T) { testHeaderConcurrentAbortion
func testHeaderConcurrentAbortion ( t * testing . T , threads int ) {
// Create a simple chain to verify
var (
testdb = rawdb . NewMemoryDatabase ( )
gspec = & Genesis { Config : params . TestChainConfig }
genesis = gspec . MustCommit ( testdb )
blocks , _ = GenerateChain ( params . TestChainConfig , genesis , ethash . NewFaker ( ) , testdb , 1024 , nil )
gspec = & Genesis { Config : params . TestChainConfig }
_ , blocks , _ = GenerateChainWithGenesis ( gspec , ethash . NewFaker ( ) , 1024 , nil )
)
headers := make ( [ ] * types . Header , len ( blocks ) )
seals := make ( [ ] bool , len ( blocks ) )
@ -348,7 +336,7 @@ func testHeaderConcurrentAbortion(t *testing.T, threads int) {
defer runtime . GOMAXPROCS ( old )
// Start the verifications and immediately abort
chain , _ := NewBlockChain ( testdb , nil , gspec , nil , ethash . NewFakeDelayer ( time . Millisecond ) , vm . Config { } , nil , nil )
chain , _ := NewBlockChain ( rawdb . NewMemoryDatabase ( ) , nil , gspec , nil , ethash . NewFakeDelayer ( time . Millisecond ) , vm . Config { } , nil , nil )
defer chain . Stop ( )
abort , results := chain . engine . VerifyHeaders ( chain , headers , seals )