beacon/light: add CommitteeChain unit tests

This commit is contained in:
Zsolt Felfoldi 2023-04-09 05:40:21 +02:00
parent a051e30ef7
commit fcb1d9cb15
4 changed files with 459 additions and 13 deletions

View file

@ -165,11 +165,30 @@ func NewCommitteeChain(db ethdb.KeyValueStore, forks types.Forks, signerThreshol
s.committees.periodRange.First >= s.fixedRoots.periodRange.AfterLast { s.committees.periodRange.First >= s.fixedRoots.periodRange.AfterLast {
log.Crit("Inconsistent database error: first committee is not in the fixed roots range") log.Crit("Inconsistent database error: first committee is not in the fixed roots range")
} }
if s.committees.periodRange.AfterLast > s.fixedRoots.periodRange.AfterLast && s.committees.periodRange.AfterLast >= s.updates.periodRange.AfterLast { if s.committees.periodRange.AfterLast > s.fixedRoots.periodRange.AfterLast && s.committees.periodRange.AfterLast > s.updates.periodRange.AfterLast+1 {
log.Crit("Inconsistent database error: last committee is neither in the fixed roots range nor proven by updates") log.Crit("Inconsistent database error: last committee is neither in the fixed roots range nor proven by updates")
} }
log.Trace("Sync committee chain loaded", "first period", s.committees.periodRange.First, "last period", s.committees.periodRange.AfterLast-1) log.Trace("Sync committee chain loaded", "first period", s.committees.periodRange.First, "last period", s.committees.periodRange.AfterLast-1)
} }
// roll back invalid updates (might be necessary if forks have been changed since last time)
var batch ethdb.Batch
for !s.updates.periodRange.IsEmpty() {
if update := s.updates.get(s.updates.periodRange.AfterLast - 1); update == nil || s.verifyUpdate(update) {
if update == nil {
log.Crit("Sync committee update missing", "period", s.updates.periodRange.AfterLast-1)
}
break
}
if batch == nil {
batch = s.db.NewBatch()
}
s.rollback(batch, s.updates.periodRange.AfterLast)
}
if batch != nil {
if err := batch.Write(); err != nil {
log.Error("Error writing batch into chain database", "error", err)
}
}
return s return s
} }
@ -307,6 +326,9 @@ func (s *CommitteeChain) InsertUpdate(update *types.LightClientUpdate, nextCommi
if !s.updates.periodRange.CanExpand(period) || !s.committees.periodRange.Includes(period) { if !s.updates.periodRange.CanExpand(period) || !s.committees.periodRange.Includes(period) {
return ErrInvalidPeriod return ErrInvalidPeriod
} }
if s.minimumUpdateScore.BetterThan(update.Score()) {
return ErrInvalidUpdate
}
oldRoot := s.getCommitteeRoot(period + 1) oldRoot := s.getCommitteeRoot(period + 1)
reorg := oldRoot != (common.Hash{}) && oldRoot != update.NextSyncCommitteeRoot reorg := oldRoot != (common.Hash{}) && oldRoot != update.NextSyncCommitteeRoot
if oldUpdate := s.updates.get(period); oldUpdate != nil && !update.Score().BetterThan(oldUpdate.Score()) { if oldUpdate := s.updates.get(period); oldUpdate != nil && !update.Score().BetterThan(oldUpdate.Score()) {

View file

@ -0,0 +1,424 @@
// Copyright 2022 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
package light
import (
"math/rand"
"testing"
"time"
"github.com/ethereum/go-ethereum/beacon/light/types"
"github.com/ethereum/go-ethereum/beacon/merkle"
"github.com/ethereum/go-ethereum/beacon/params"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/common/mclock"
"github.com/ethereum/go-ethereum/ethdb/memorydb"
"github.com/minio/sha256-simd"
)
var (
testGenesis = newTestGenesis()
testGenesis2 = newTestGenesis()
tfBase = newTestForks(testGenesis, types.Forks{
types.Fork{Epoch: 0, Version: []byte{0}},
})
tfAlternative = newTestForks(testGenesis, types.Forks{
types.Fork{Epoch: 0, Version: []byte{0}},
types.Fork{Epoch: 0x700, Version: []byte{1}},
})
tfAnotherGenesis = newTestForks(testGenesis2, types.Forks{
types.Fork{Epoch: 0, Version: []byte{0}},
})
tcBase = newTestCommitteeChain(nil, testGenesis, tfBase, true, 0, 10, 400, 400)
tcBaseWithInvalidUpdates = newTestCommitteeChain(tcBase, testGenesis, tfBase, false, 5, 10, 400, 200) // signer count too low
tcBaseWithBetterUpdates = newTestCommitteeChain(tcBase, testGenesis, tfBase, false, 5, 10, 400, 440)
tcReorgWithWorseUpdates = newTestCommitteeChain(tcBase, testGenesis, tfBase, true, 5, 10, 20, 400)
tcReorgWithWorseUpdates2 = newTestCommitteeChain(tcBase, testGenesis, tfBase, true, 5, 10, 400, 380)
tcReorgWithBetterUpdates = newTestCommitteeChain(tcBase, testGenesis, tfBase, true, 5, 10, 400, 420)
tcReorgWithFinalizedUpdates = newTestCommitteeChain(tcBase, testGenesis, tfBase, true, 5, 10, finalizedTestUpdate, 400)
tcFork = newTestCommitteeChain(tcBase, testGenesis, tfAlternative, true, 7, 10, 400, 400)
tcAnotherGenesis = newTestCommitteeChain(nil, testGenesis2, tfAnotherGenesis, true, 0, 10, 400, 400)
)
func TestCommitteeChainFixedRoots(t *testing.T) {
for _, reload := range []bool{false, true} {
c := newCommitteeChainTest(t, testGenesis, tfBase, 300, true)
c.setClockPeriod(7)
c.addFixedRoot(tcBase, 4, nil)
c.addFixedRoot(tcBase, 5, nil)
c.addFixedRoot(tcBase, 6, nil)
c.addFixedRoot(tcBase, 8, ErrInvalidPeriod) // range has to be continuoous
c.addFixedRoot(tcBase, 3, nil)
c.addFixedRoot(tcBase, 2, nil)
if reload {
c.reloadChain()
}
c.addCommittee(tcBase, 4, nil)
c.addCommittee(tcBase, 6, ErrInvalidPeriod) // range has to be continuoous
c.addCommittee(tcBase, 5, nil)
c.addCommittee(tcBase, 6, nil)
c.addCommittee(tcAnotherGenesis, 3, ErrWrongCommitteeRoot)
c.addCommittee(tcBase, 3, nil)
if reload {
c.reloadChain()
}
c.verifyRange(tcBase, 3, 6)
}
}
func TestCommitteeChainCheckpointSync(t *testing.T) {
for _, enforceTime := range []bool{false, true} {
for _, reload := range []bool{false, true} {
c := newCommitteeChainTest(t, testGenesis, tfBase, 300, enforceTime)
if enforceTime {
c.setClockPeriod(6)
}
c.insertUpdate(tcBase, 3, true, ErrInvalidPeriod)
c.addFixedRoot(tcBase, 3, nil)
c.addFixedRoot(tcBase, 4, nil)
c.insertUpdate(tcBase, 4, true, ErrInvalidPeriod) // still no committee
c.addCommittee(tcBase, 3, nil)
c.addCommittee(tcBase, 4, nil)
if reload {
c.reloadChain()
}
c.verifyRange(tcBase, 3, 4)
c.insertUpdate(tcBase, 3, false, nil) // update can be added without committee here
c.insertUpdate(tcBase, 4, false, ErrNeedCommittee) // but not here as committee 5 is not there yet
c.insertUpdate(tcBase, 4, true, nil)
c.verifyRange(tcBase, 3, 5)
c.insertUpdate(tcBaseWithInvalidUpdates, 5, true, ErrInvalidUpdate) // signer count too low
c.insertUpdate(tcBase, 5, true, nil)
if reload {
c.reloadChain()
}
if enforceTime {
c.insertUpdate(tcBase, 6, true, ErrInvalidUpdate) // future update rejected
c.setClockPeriod(7)
}
c.insertUpdate(tcBase, 6, true, nil) // when the time comes it's accepted
if reload {
c.reloadChain()
}
if enforceTime {
c.verifyRange(tcBase, 3, 6) // committee 7 is there but still in the future
c.setClockPeriod(8)
}
c.verifyRange(tcBase, 3, 7) // now period 7 can also be verified
// try reverse syncing an update
c.insertUpdate(tcBase, 2, false, ErrInvalidPeriod) // fixed committee is needed first
c.addFixedRoot(tcBase, 2, nil)
c.addCommittee(tcBase, 2, nil)
c.insertUpdate(tcBase, 2, false, nil)
c.verifyRange(tcBase, 2, 7)
}
}
}
func TestCommitteeChainReorg(t *testing.T) {
for _, reload := range []bool{false, true} {
for _, addBetterUpdates := range []bool{false, true} {
c := newCommitteeChainTest(t, testGenesis, tfBase, 300, true)
c.setClockPeriod(11)
c.addFixedRoot(tcBase, 3, nil)
c.addFixedRoot(tcBase, 4, nil)
c.addCommittee(tcBase, 3, nil)
for period := uint64(3); period < 10; period++ {
c.insertUpdate(tcBase, period, true, nil)
}
if reload {
c.reloadChain()
}
c.verifyRange(tcBase, 3, 10)
c.insertUpdate(tcReorgWithWorseUpdates, 5, true, ErrCannotReorg)
c.insertUpdate(tcReorgWithWorseUpdates2, 5, true, ErrCannotReorg)
if addBetterUpdates {
// add better updates for the base chain and expect first reorg to fail
// (only add updates as committees should be the same)
for period := uint64(5); period < 10; period++ {
c.insertUpdate(tcBaseWithBetterUpdates, period, false, nil)
}
if reload {
c.reloadChain()
}
c.verifyRange(tcBase, 3, 10) // still on the same chain
c.insertUpdate(tcReorgWithBetterUpdates, 5, true, ErrCannotReorg)
} else {
// reorg with better updates
c.insertUpdate(tcReorgWithBetterUpdates, 5, false, ErrNeedCommittee)
c.verifyRange(tcBase, 3, 10) // no success yet, still on the base chain
c.verifyRange(tcReorgWithBetterUpdates, 3, 5)
c.insertUpdate(tcReorgWithBetterUpdates, 5, true, nil)
// successful reorg, base chain should only match before the reorg period
if reload {
c.reloadChain()
}
c.verifyRange(tcBase, 3, 5)
c.verifyRange(tcReorgWithBetterUpdates, 3, 6)
for period := uint64(6); period < 10; period++ {
c.insertUpdate(tcReorgWithBetterUpdates, period, true, nil)
}
c.verifyRange(tcReorgWithBetterUpdates, 3, 10)
}
// reorg with finalized updates; should succeed even if base chain updates
// have been improved beacuse a finalized update beats everything else
c.insertUpdate(tcReorgWithFinalizedUpdates, 5, false, ErrNeedCommittee)
c.insertUpdate(tcReorgWithFinalizedUpdates, 5, true, nil)
if reload {
c.reloadChain()
}
c.verifyRange(tcReorgWithFinalizedUpdates, 3, 6)
for period := uint64(6); period < 10; period++ {
c.insertUpdate(tcReorgWithFinalizedUpdates, period, true, nil)
}
c.verifyRange(tcReorgWithFinalizedUpdates, 3, 10)
}
}
}
func TestCommitteeChainFork(t *testing.T) {
c := newCommitteeChainTest(t, testGenesis, tfAlternative, 300, true)
c.setClockPeriod(11)
// trying to sync a chain on an alternative fork with the base chain data
c.addFixedRoot(tcBase, 0, nil)
c.addFixedRoot(tcBase, 1, nil)
c.addCommittee(tcBase, 0, nil)
// shared section should sync without errors
for period := uint64(0); period < 7; period++ {
c.insertUpdate(tcBase, period, true, nil)
}
c.insertUpdate(tcBase, 7, true, ErrInvalidUpdate) // wrong fork
// committee root #7 is still the same but signatures are already signed with
// a different fork id so period 7 should only verify on the alternative fork
c.verifyRange(tcBase, 0, 6)
c.verifyRange(tcFork, 0, 7)
for period := uint64(7); period < 10; period++ {
c.insertUpdate(tcFork, period, true, nil)
}
c.verifyRange(tcFork, 0, 10)
// reload the chain while switching to the base fork
c.forks = tfBase
c.reloadChain()
// updates 7..9 should be rolled back now
c.verifyRange(tcFork, 0, 6) // again, period 7 only verifies on the right fork
c.verifyRange(tcBase, 0, 7)
c.insertUpdate(tcFork, 7, true, ErrInvalidUpdate) // wrong fork
for period := uint64(7); period < 10; period++ {
c.insertUpdate(tcBase, period, true, nil)
}
c.verifyRange(tcBase, 0, 10)
}
type committeeChainTest struct {
t *testing.T
db *memorydb.Database
clock *mclock.Simulated
genesis GenesisData
forks types.Forks
signerThreshold int
enforceTime bool
chain *CommitteeChain
}
func newCommitteeChainTest(t *testing.T, genesis GenesisData, forks types.Forks, signerThreshold int, enforceTime bool) *committeeChainTest {
c := &committeeChainTest{
t: t,
db: memorydb.New(),
clock: &mclock.Simulated{},
genesis: genesis,
forks: forks,
signerThreshold: signerThreshold,
enforceTime: enforceTime,
}
c.chain = NewCommitteeChain(c.db, forks, signerThreshold, enforceTime, dummyVerifier{}, c.clock, func() int64 { return int64(c.clock.Now()) })
c.chain.SetGenesisData(genesis)
return c
}
func (c *committeeChainTest) reloadChain() {
c.chain = NewCommitteeChain(c.db, c.forks, c.signerThreshold, c.enforceTime, dummyVerifier{}, c.clock, func() int64 { return int64(c.clock.Now()) })
c.chain.SetGenesisData(c.genesis)
}
func (c *committeeChainTest) setClockPeriod(period float64) {
target := mclock.AbsTime(period * float64(time.Second*12*params.SyncPeriodLength))
wait := time.Duration(target - c.clock.Now())
if wait < 0 {
c.t.Fatalf("Invalid setClockPeriod")
}
c.clock.Run(wait)
}
func (c *committeeChainTest) addFixedRoot(tc *testCommitteeChain, period uint64, expErr error) {
if err := c.chain.AddFixedRoot(period, tc.periods[period].committeeRoot); err != expErr {
c.t.Errorf("Incorrect error output from AddFixedRoot at period %d (expected %v, got %v)", period, expErr, err)
}
}
func (c *committeeChainTest) addCommittee(tc *testCommitteeChain, period uint64, expErr error) {
if err := c.chain.AddCommittee(period, serializeDummySyncCommittee(tc.periods[period].committee)); err != expErr {
c.t.Errorf("Incorrect error output from AddCommittee at period %d (expected %v, got %v)", period, expErr, err)
}
}
func (c *committeeChainTest) insertUpdate(tc *testCommitteeChain, period uint64, addCommittee bool, expErr error) {
var committee *types.SerializedCommittee
if addCommittee {
committee = serializeDummySyncCommittee(tc.periods[period+1].committee)
}
if err := c.chain.InsertUpdate(&tc.periods[period].update, committee); err != expErr {
c.t.Errorf("Incorrect error output from InsertUpdate at period %d (expected %v, got %v)", period, expErr, err)
}
}
func (c *committeeChainTest) verifySignedHead(tc *testCommitteeChain, period float64, expOk bool) {
signedHead := tc.makeTestSignedHead(types.Header{Slot: uint64(period * float64(params.SyncPeriodLength))}, 400)
if ok, _ := c.chain.VerifySignedHead(signedHead); ok != expOk {
c.t.Errorf("Incorrect output from VerifySignedHead at period %f (expected %v, got %v)", period, expOk, ok)
}
}
func (c *committeeChainTest) verifyRange(tc *testCommitteeChain, begin, end uint64) {
if begin > 0 {
c.verifySignedHead(tc, float64(begin)-0.5, false)
}
for period := begin; period <= end; period++ {
c.verifySignedHead(tc, float64(period)+0.5, true)
}
c.verifySignedHead(tc, float64(end)+1.5, false)
}
func newTestGenesis() GenesisData {
var genesisData GenesisData
rand.Read(genesisData.GenesisValidatorsRoot[:])
return genesisData
}
func newTestForks(genesisData GenesisData, forks types.Forks) types.Forks {
forks.ComputeDomains(genesisData.GenesisValidatorsRoot)
return forks
}
func newTestCommitteeChain(parent *testCommitteeChain, genesisData GenesisData, forks types.Forks, newCommittees bool, begin, end int, subPeriodIndex uint64, signerCount int) *testCommitteeChain {
tc := &testCommitteeChain{
genesisData: genesisData,
forks: forks,
}
if parent != nil {
tc.periods = make([]testPeriod, len(parent.periods))
copy(tc.periods, parent.periods)
}
if newCommittees {
if begin == 0 {
tc.fillCommittees(begin, end+1)
} else {
tc.fillCommittees(begin+1, end+1)
}
}
tc.fillUpdates(begin, end, subPeriodIndex, signerCount)
return tc
}
func makeTestHeaderWithSingleProof(slot, index uint64, value merkle.Value) (types.Header, merkle.Values) {
var branch merkle.Values
hasher := sha256.New()
for index > 1 {
var proofHash merkle.Value
rand.Read(proofHash[:])
hasher.Reset()
if index&1 == 0 {
hasher.Write(value[:])
hasher.Write(proofHash[:])
} else {
hasher.Write(proofHash[:])
hasher.Write(value[:])
}
hasher.Sum(value[:0])
index /= 2
branch = append(branch, proofHash)
}
return types.Header{Slot: slot, StateRoot: common.Hash(value)}, branch
}
func makeBitmask(signerCount int) (bitmask [params.SyncCommitteeBitmaskSize]byte) {
for i := 0; i < params.SyncCommitteeSize; i++ {
if rand.Intn(params.SyncCommitteeSize-i) < signerCount {
bitmask[i/8] += byte(1) << (i & 7)
signerCount--
}
}
return
}
type testPeriod struct {
committee dummySyncCommittee
committeeRoot common.Hash
update types.LightClientUpdate
}
type testCommitteeChain struct {
periods []testPeriod
forks types.Forks
genesisData GenesisData
}
func (tc *testCommitteeChain) makeTestSignedHead(header types.Header, signerCount int) types.SignedHead {
bitmask := makeBitmask(signerCount)
return types.SignedHead{
Header: header,
SyncAggregate: types.SyncAggregate{
BitMask: bitmask,
Signature: makeDummySignature(tc.periods[types.PeriodOfSlot(header.Slot+1)].committee, tc.forks.SigningRoot(header), bitmask),
},
SignatureSlot: header.Slot + 1,
}
}
const finalizedTestUpdate = params.SyncPeriodLength - 1 // if subPeriodIndex == finalizedTestUpdate then a finalized update is generated
func (tc *testCommitteeChain) makeTestUpdate(period, subPeriodIndex uint64, signerCount int) types.LightClientUpdate {
var update types.LightClientUpdate
update.NextSyncCommitteeRoot = tc.periods[period+1].committeeRoot
if subPeriodIndex == finalizedTestUpdate {
update.FinalizedHeader, update.NextSyncCommitteeBranch = makeTestHeaderWithSingleProof(types.PeriodStart(period)+100, params.BsiNextSyncCommittee, merkle.Value(update.NextSyncCommitteeRoot))
update.Header, update.FinalityBranch = makeTestHeaderWithSingleProof(types.PeriodStart(period)+200, params.BsiFinalBlock, merkle.Value(update.FinalizedHeader.Hash()))
} else {
update.Header, update.NextSyncCommitteeBranch = makeTestHeaderWithSingleProof(types.PeriodStart(period)+subPeriodIndex, params.BsiNextSyncCommittee, merkle.Value(update.NextSyncCommitteeRoot))
}
signedHead := tc.makeTestSignedHead(update.Header, signerCount)
update.SyncAggregate = signedHead.SyncAggregate
update.SignatureSlot = update.Header.Slot
return update
}
func (tc *testCommitteeChain) fillCommittees(begin, end int) {
if len(tc.periods) <= end {
tc.periods = append(tc.periods, make([]testPeriod, end+1-len(tc.periods))...)
}
for i := begin; i <= end; i++ {
tc.periods[i].committee = randomDummySyncCommittee()
tc.periods[i].committeeRoot = serializeDummySyncCommittee(tc.periods[i].committee).Root()
}
}
func (tc *testCommitteeChain) fillUpdates(begin, end int, subPeriodIndex uint64, signerCount int) {
for i := begin; i <= end; i++ {
tc.periods[i].update = tc.makeTestUpdate(uint64(i), subPeriodIndex, signerCount)
}
}

View file

@ -28,7 +28,7 @@ import (
func TestLightChainSetHead(t *testing.T) { func TestLightChainSetHead(t *testing.T) {
for _, reload := range []bool{false, true} { for _, reload := range []bool{false, true} {
c := newChainTest(t) c := newLightChainTest(t)
a1, a2 := c.makeChain(types.Header{}, 100, false, false) a1, a2 := c.makeChain(types.Header{}, 100, false, false)
b1, b2 := c.makeChain(a2, 150, true, false) b1, b2 := c.makeChain(a2, 150, true, false)
c1, c2 := c.makeChain(b2, 200, true, true) c1, c2 := c.makeChain(b2, 200, true, true)
@ -88,7 +88,7 @@ func TestLightChainSetHead(t *testing.T) {
func TestLightChainExtendHeaderTail(t *testing.T) { func TestLightChainExtendHeaderTail(t *testing.T) {
for _, reload := range []bool{false, true} { for _, reload := range []bool{false, true} {
for _, reverse := range []bool{false, true} { for _, reverse := range []bool{false, true} {
c := newChainTest(t) c := newLightChainTest(t)
a1, a2 := c.makeChain(types.Header{}, 50, false, false) a1, a2 := c.makeChain(types.Header{}, 50, false, false)
b1, b2 := c.makeChain(a2, 100, true, false) b1, b2 := c.makeChain(a2, 100, true, false)
c.chain.SetHead(b2) c.chain.SetHead(b2)
@ -118,7 +118,7 @@ func TestLightChainExtendHeaderTail(t *testing.T) {
func TestLightChainExtendStateRange(t *testing.T) { func TestLightChainExtendStateRange(t *testing.T) {
for _, reload := range []bool{false /*, true*/} { for _, reload := range []bool{false /*, true*/} {
for _, reverse := range []bool{false, true} { for _, reverse := range []bool{false, true} {
c := newChainTest(t) c := newLightChainTest(t)
a1, a2 := c.makeChain(types.Header{}, 50, true, false) a1, a2 := c.makeChain(types.Header{}, 50, true, false)
b1, b2 := c.makeChain(a2, 100, true, true) b1, b2 := c.makeChain(a2, 100, true, true)
_, c2 := c.makeChain(b2, 150, true, false) _, c2 := c.makeChain(b2, 150, true, false)
@ -145,7 +145,7 @@ func TestLightChainExtendStateRange(t *testing.T) {
} }
} }
type chainTest struct { type lightChainTest struct {
t *testing.T t *testing.T
db *memorydb.Database db *memorydb.Database
proofFormat merkle.ProofFormat proofFormat merkle.ProofFormat
@ -159,8 +159,8 @@ type testProof struct {
proof merkle.MultiProof proof merkle.MultiProof
} }
func newChainTest(t *testing.T) *chainTest { func newLightChainTest(t *testing.T) *lightChainTest {
c := &chainTest{ c := &lightChainTest{
t: t, t: t,
db: memorydb.New(), db: memorydb.New(),
proofFormat: merkle.NewIndexMapFormat().AddLeaf(42, nil).AddLeaf(67, nil), proofFormat: merkle.NewIndexMapFormat().AddLeaf(42, nil).AddLeaf(67, nil),
@ -169,7 +169,7 @@ func newChainTest(t *testing.T) *chainTest {
return c return c
} }
func (c *chainTest) checkRange(chainInit bool, chainTail, chainHead types.Header, stateInit bool, stateTail, stateHead types.Header) { func (c *lightChainTest) checkRange(chainInit bool, chainTail, chainHead types.Header, stateInit bool, stateTail, stateHead types.Header) {
ch, ct, ci := c.chain.HeaderRange() ch, ct, ci := c.chain.HeaderRange()
if ci != chainInit || (ci && (ct != chainTail || ch != chainHead)) { if ci != chainInit || (ci && (ct != chainTail || ch != chainHead)) {
c.t.Errorf("Incorrect header chain range (expected: %v %d %d, got: %v %d %d)", chainInit, chainTail.Slot, chainHead.Slot, ci, ct.Slot, ch.Slot) c.t.Errorf("Incorrect header chain range (expected: %v %d %d, got: %v %d %d)", chainInit, chainTail.Slot, chainHead.Slot, ci, ct.Slot, ch.Slot)
@ -188,13 +188,13 @@ func (c *chainTest) checkRange(chainInit bool, chainTail, chainHead types.Header
} }
} }
func (c *chainTest) checkCanonical(header types.Header, expected bool) { func (c *lightChainTest) checkCanonical(header types.Header, expected bool) {
if canonical := c.chain.IsCanonical(header); canonical != expected { if canonical := c.chain.IsCanonical(header); canonical != expected {
c.t.Errorf("Canonical status of header at slot %d is incorrect (expected: %v, got: %v)", header.Slot, expected, canonical) c.t.Errorf("Canonical status of header at slot %d is incorrect (expected: %v, got: %v)", header.Slot, expected, canonical)
} }
} }
func (c *chainTest) checkTail(header, expTail types.Header) { func (c *lightChainTest) checkTail(header, expTail types.Header) {
for { for {
if parent, err := c.chain.GetParent(header); err == nil { if parent, err := c.chain.GetParent(header); err == nil {
header = parent header = parent
@ -207,11 +207,11 @@ func (c *chainTest) checkTail(header, expTail types.Header) {
} }
} }
func (c *chainTest) reloadChain() { func (c *lightChainTest) reloadChain() {
c.chain = NewLightChain(c.db, c.proofFormat) c.chain = NewLightChain(c.db, c.proofFormat)
} }
func (c *chainTest) makeChain(from types.Header, targetHeadSlot uint64, addHeaders, addStateProofs bool) (tail, head types.Header) { func (c *lightChainTest) makeChain(from types.Header, targetHeadSlot uint64, addHeaders, addStateProofs bool) (tail, head types.Header) {
head = from head = from
valueCount := merkle.ValueCount(c.proofFormat) valueCount := merkle.ValueCount(c.proofFormat)
for head.Slot < targetHeadSlot { for head.Slot < targetHeadSlot {

View file

@ -136,7 +136,7 @@ func (update *LightClientUpdate) Validate() error {
// Note that in addition to this, a sufficient signer participation is also needed // Note that in addition to this, a sufficient signer participation is also needed
// in order to fulfill the quasi-finality condition (see UpdateScore.isFinalized). // in order to fulfill the quasi-finality condition (see UpdateScore.isFinalized).
func (l *LightClientUpdate) hasFinalizedHeader() bool { func (l *LightClientUpdate) hasFinalizedHeader() bool {
return l.FinalizedHeader.BodyRoot != (common.Hash{}) && l.FinalizedHeader.SyncPeriod() == l.Header.SyncPeriod() return l.FinalizedHeader.StateRoot != (common.Hash{}) && l.FinalizedHeader.SyncPeriod() == l.Header.SyncPeriod()
} }
// Score returns the UpdateScore describing the proof strength of the update // Score returns the UpdateScore describing the proof strength of the update