mirror of
https://github.com/status-im/nimbus-eth2.git
synced 2025-01-09 13:56:23 +00:00
f70ff38b53
Some upstream repos still need fixes, but this gets us close enough that style hints can be enabled by default. In general, "canonical" spellings are preferred even if they violate nep-1 - this applies in particular to spec-related stuff like `genesis_validators_root` which appears throughout the codebase.
246 lines
8.7 KiB
Nim
246 lines
8.7 KiB
Nim
# beacon_chain
|
|
# Copyright (c) 2022 Status Research & Development GmbH
|
|
# Licensed and distributed under either of
|
|
# * MIT license (license terms in the root directory or at https://opensource.org/licenses/MIT).
|
|
# * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
{.used.}
|
|
|
|
import
|
|
# Status libraries
|
|
chronos, eth/keys,
|
|
# Beacon chain internals
|
|
../beacon_chain/consensus_object_pools/
|
|
[block_clearance, block_quarantine, blockchain_dag],
|
|
../beacon_chain/gossip_processing/light_client_processor,
|
|
../beacon_chain/spec/[beacon_time, light_client_sync, state_transition],
|
|
# Test utilities
|
|
./testutil, ./testdbutil
|
|
|
|
suite "Light client processor" & preset():
|
|
let
|
|
cfg = block:
|
|
var res = defaultRuntimeConfig
|
|
res.ALTAIR_FORK_EPOCH = GENESIS_EPOCH + 1
|
|
res
|
|
|
|
const numValidators = SLOTS_PER_EPOCH
|
|
let
|
|
validatorMonitor = newClone(ValidatorMonitor.init())
|
|
dag = ChainDAGRef.init(
|
|
cfg, makeTestDB(numValidators), validatorMonitor, {},
|
|
serveLightClientData = true,
|
|
importLightClientData = ImportLightClientData.OnlyNew)
|
|
quarantine = newClone(Quarantine.init())
|
|
taskpool = Taskpool.new()
|
|
var verifier = BatchVerifier(rng: keys.newRng(), taskpool: taskpool)
|
|
|
|
var cache: StateCache
|
|
proc addBlocks(blocks: uint64, syncCommitteeRatio: float) =
|
|
for blck in makeTestBlocks(dag.headState, cache, blocks.int,
|
|
attested = true, syncCommitteeRatio, cfg):
|
|
let added =
|
|
case blck.kind
|
|
of BeaconBlockFork.Phase0:
|
|
const nilCallback = OnPhase0BlockAdded(nil)
|
|
dag.addHeadBlock(verifier, blck.phase0Data, nilCallback)
|
|
of BeaconBlockFork.Altair:
|
|
const nilCallback = OnAltairBlockAdded(nil)
|
|
dag.addHeadBlock(verifier, blck.altairData, nilCallback)
|
|
of BeaconBlockFork.Bellatrix:
|
|
const nilCallback = OnBellatrixBlockAdded(nil)
|
|
dag.addHeadBlock(verifier, blck.bellatrixData, nilCallback)
|
|
doAssert added.isOk()
|
|
dag.updateHead(added[], quarantine[])
|
|
|
|
addBlocks(SLOTS_PER_EPOCH, 0.75)
|
|
let
|
|
genesis_validators_root = dag.genesis_validators_root
|
|
trustedBlockRoot = dag.head.root
|
|
|
|
const
|
|
lowPeriod = 0.SyncCommitteePeriod
|
|
lastPeriodWithSupermajority = 3.SyncCommitteePeriod
|
|
highPeriod = 5.SyncCommitteePeriod
|
|
for period in lowPeriod .. highPeriod:
|
|
const numFilledEpochsPerPeriod = 3
|
|
let slot = ((period + 1).start_epoch - numFilledEpochsPerPeriod).start_slot
|
|
var info: ForkedEpochInfo
|
|
doAssert process_slots(cfg, dag.headState, slot,
|
|
cache, info, flags = {}).isOk()
|
|
let syncCommitteeRatio =
|
|
if period > lastPeriodWithSupermajority:
|
|
0.25
|
|
else:
|
|
0.75
|
|
addBlocks(numFilledEpochsPerPeriod * SLOTS_PER_EPOCH, syncCommitteeRatio)
|
|
|
|
setup:
|
|
var time = chronos.seconds(0)
|
|
proc getBeaconTime(): BeaconTime =
|
|
BeaconTime(ns_since_genesis: time.nanoseconds)
|
|
func setTimeToSlot(slot: Slot) =
|
|
time = chronos.seconds((slot * SECONDS_PER_SLOT).int64)
|
|
|
|
var numDidInitializeStoreCalls = 0
|
|
proc didInitializeStore() = inc numDidInitializeStoreCalls
|
|
|
|
let store = (ref Option[LightClientStore])()
|
|
var
|
|
processor = LightClientProcessor.new(
|
|
false, "", "", cfg, genesis_validators_root, trustedBlockRoot,
|
|
store, getBeaconTime, didInitializeStore)
|
|
res: Result[void, BlockError]
|
|
|
|
test "Standard sync" & preset():
|
|
let bootstrap = dag.getLightClientBootstrap(trustedBlockRoot)
|
|
check bootstrap.isOk
|
|
setTimeToSlot(bootstrap.get.header.slot)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), bootstrap.get)
|
|
check:
|
|
res.isOk
|
|
numDidInitializeStoreCalls == 1
|
|
|
|
for period in lowPeriod .. lastPeriodWithSupermajority:
|
|
let update = dag.getBestLightClientUpdateForPeriod(period)
|
|
check update.isSome
|
|
setTimeToSlot(update.get.attested_header.slot + 1)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), update.get)
|
|
check:
|
|
res.isOk
|
|
store[].isSome
|
|
store[].get.finalized_header == update.get.finalized_header
|
|
store[].get.optimistic_header == update.get.attested_header
|
|
|
|
test "Forced update" & preset():
|
|
let bootstrap = dag.getLightClientBootstrap(trustedBlockRoot)
|
|
check bootstrap.isOk
|
|
setTimeToSlot(bootstrap.get.header.slot)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), bootstrap.get)
|
|
check:
|
|
res.isOk
|
|
numDidInitializeStoreCalls == 1
|
|
|
|
for period in lowPeriod .. lastPeriodWithSupermajority:
|
|
let update = dag.getBestLightClientUpdateForPeriod(period)
|
|
check update.isSome
|
|
setTimeToSlot(update.get.attested_header.slot + 1)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), update.get)
|
|
check:
|
|
res.isOk
|
|
store[].isSome
|
|
store[].get.finalized_header == update.get.finalized_header
|
|
store[].get.optimistic_header == update.get.attested_header
|
|
|
|
for period in lastPeriodWithSupermajority + 1 .. highPeriod:
|
|
let update = dag.getBestLightClientUpdateForPeriod(period)
|
|
check update.isSome
|
|
setTimeToSlot(update.get.attested_header.slot + 1)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), update.get)
|
|
check:
|
|
res.isOk
|
|
store[].isSome
|
|
store[].get.best_valid_update.isSome
|
|
store[].get.best_valid_update.get == update.get
|
|
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), update.get)
|
|
check:
|
|
res.isErr
|
|
res.error == BlockError.Duplicate
|
|
store[].isSome
|
|
store[].get.best_valid_update.isSome
|
|
store[].get.best_valid_update.get == update.get
|
|
time += chronos.minutes(15)
|
|
|
|
for _ in 0 ..< 150:
|
|
time += chronos.seconds(5)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), update.get)
|
|
check:
|
|
res.isErr
|
|
res.error == BlockError.Duplicate
|
|
store[].isSome
|
|
store[].get.best_valid_update.isSome
|
|
store[].get.best_valid_update.get == update.get
|
|
|
|
time += chronos.minutes(15)
|
|
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), update.get)
|
|
check:
|
|
res.isErr
|
|
res.error == BlockError.Duplicate
|
|
store[].isSome
|
|
store[].get.best_valid_update.isNone
|
|
store[].get.finalized_header == update.get.finalized_header
|
|
|
|
let optimisticUpdate = dag.getOptimisticLightClientUpdate()
|
|
check optimisticUpdate.isSome
|
|
setTimeToSlot(optimisticUpdate.get.attested_header.slot + 1)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), optimisticUpdate.get)
|
|
if res.isOk:
|
|
check:
|
|
store[].isSome
|
|
store[].get.optimistic_header == optimisticUpdate.get.attested_header
|
|
else:
|
|
check res.error == BlockError.Duplicate
|
|
check numDidInitializeStoreCalls == 1
|
|
|
|
test "Invalid bootstrap" & preset():
|
|
var bootstrap = dag.getLightClientBootstrap(trustedBlockRoot)
|
|
check bootstrap.isOk
|
|
bootstrap.get.header.slot.inc()
|
|
setTimeToSlot(bootstrap.get.header.slot)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), bootstrap.get)
|
|
check:
|
|
res.isErr
|
|
res.error == BlockError.Invalid
|
|
numDidInitializeStoreCalls == 0
|
|
|
|
test "Duplicate bootstrap" & preset():
|
|
let bootstrap = dag.getLightClientBootstrap(trustedBlockRoot)
|
|
check bootstrap.isOk
|
|
setTimeToSlot(bootstrap.get.header.slot)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), bootstrap.get)
|
|
check:
|
|
res.isOk
|
|
numDidInitializeStoreCalls == 1
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), bootstrap.get)
|
|
check:
|
|
res.isErr
|
|
res.error == BlockError.Duplicate
|
|
numDidInitializeStoreCalls == 1
|
|
|
|
test "Missing bootstrap (update)" & preset():
|
|
let update = dag.getBestLightClientUpdateForPeriod(lowPeriod)
|
|
check update.isSome
|
|
setTimeToSlot(update.get.attested_header.slot + 1)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), update.get)
|
|
check:
|
|
res.isErr
|
|
res.error == BlockError.MissingParent
|
|
numDidInitializeStoreCalls == 0
|
|
|
|
test "Missing bootstrap (optimistic update)" & preset():
|
|
let optimisticUpdate = dag.getOptimisticLightClientUpdate()
|
|
check optimisticUpdate.isSome
|
|
setTimeToSlot(optimisticUpdate.get.attested_header.slot + 1)
|
|
res = processor[].storeObject(
|
|
MsgSource.gossip, getBeaconTime(), optimisticUpdate.get)
|
|
check:
|
|
res.isErr
|
|
res.error == BlockError.MissingParent
|
|
numDidInitializeStoreCalls == 0
|