664 lines
24 KiB
Nim
664 lines
24 KiB
Nim
import
|
|
std_shims/[os_shims, objects], net, sequtils, options, tables,
|
|
chronos, chronicles, confutils, eth/[p2p, keys],
|
|
spec/[datatypes, digest, crypto, beaconstate, helpers, validator], conf, time,
|
|
state_transition, fork_choice, ssz, beacon_chain_db, validator_pool, extras,
|
|
attestation_pool, block_pool,
|
|
mainchain_monitor, gossipsub_protocol, trusted_state_snapshots,
|
|
eth/trie/db, eth/trie/backends/rocksdb_backend
|
|
|
|
type
|
|
BeaconNode* = ref object
|
|
network*: EthereumNode
|
|
db*: BeaconChainDB
|
|
config*: BeaconNodeConf
|
|
keys*: KeyPair
|
|
attachedValidators: ValidatorPool
|
|
blockPool: BlockPool
|
|
state*: StateData
|
|
attestationPool: AttestationPool
|
|
mainchainMonitor: MainchainMonitor
|
|
potentialHeads: seq[Eth2Digest]
|
|
|
|
const
|
|
version = "v0.1" # TODO: read this from the nimble file
|
|
clientId = "nimbus beacon node " & version
|
|
|
|
topicBeaconBlocks = "ethereum/2.1/beacon_chain/blocks"
|
|
topicAttestations = "ethereum/2.1/beacon_chain/attestations"
|
|
topicfetchBlocks = "ethereum/2.1/beacon_chain/fetch"
|
|
|
|
|
|
proc onBeaconBlock*(node: BeaconNode, blck: BeaconBlock) {.gcsafe.}
|
|
|
|
import sync_protocol
|
|
|
|
|
|
func shortValidatorKey(node: BeaconNode, validatorIdx: int): string =
|
|
($node.state.data.validator_registry[validatorIdx].pubkey)[0..7]
|
|
|
|
func slotStart(node: BeaconNode, slot: Slot): Timestamp =
|
|
node.state.data.slotStart(slot)
|
|
|
|
proc ensureNetworkKeys*(dataDir: string): KeyPair =
|
|
# TODO:
|
|
# 1. Check if keys already exist in the data dir
|
|
# 2. Generate new ones and save them in the directory
|
|
# if necessary
|
|
return newKeyPair()
|
|
|
|
proc init*(T: type BeaconNode, conf: BeaconNodeConf): T =
|
|
new result
|
|
result.config = conf
|
|
|
|
result.attachedValidators = ValidatorPool.init
|
|
init result.mainchainMonitor, "", Port(0) # TODO: specify geth address and port
|
|
|
|
let trieDB = trieDB newChainDb(string conf.dataDir)
|
|
result.db = BeaconChainDB.init(trieDB)
|
|
|
|
# TODO this is problably not the right place to ensure that db is sane..
|
|
# TODO does it really make sense to load from DB if a state snapshot has been
|
|
# specified on command line? potentially, this should be the other way
|
|
# around...
|
|
|
|
let headBlock = result.db.getHeadBlock()
|
|
if headBlock.isNone():
|
|
let
|
|
tailState = result.config.stateSnapshot.get()
|
|
tailBlock = get_initial_beacon_block(tailState)
|
|
blockRoot = hash_tree_root_final(tailBlock)
|
|
|
|
notice "Creating new database from snapshot",
|
|
blockRoot = shortLog(blockRoot),
|
|
stateRoot = shortLog(tailBlock.state_root),
|
|
fork = tailState.fork,
|
|
validators = tailState.validator_registry.len()
|
|
|
|
result.db.putState(tailState)
|
|
result.db.putBlock(tailBlock)
|
|
result.db.putTailBlock(blockRoot)
|
|
result.db.putHeadBlock(blockRoot)
|
|
|
|
result.blockPool = BlockPool.init(result.db)
|
|
result.attestationPool = AttestationPool.init(result.blockPool)
|
|
|
|
result.keys = ensureNetworkKeys(string conf.dataDir)
|
|
|
|
var address: Address
|
|
address.ip = parseIpAddress("127.0.0.1")
|
|
address.tcpPort = Port(conf.tcpPort)
|
|
address.udpPort = Port(conf.udpPort)
|
|
|
|
result.network =
|
|
newEthereumNode(result.keys, address, 0, nil, clientId, minPeers = 1)
|
|
let state = result.network.protocolState(BeaconSync)
|
|
state.node = result
|
|
state.db = result.db
|
|
|
|
let
|
|
head = result.blockPool.get(result.db.getHeadBlock().get())
|
|
|
|
result.state = result.blockPool.loadTailState()
|
|
result.blockPool.updateState(result.state, head.get().refs)
|
|
|
|
writeFile(string(conf.dataDir) / "beacon_node.address",
|
|
$result.network.listeningAddress)
|
|
|
|
proc connectToNetwork(node: BeaconNode) {.async.} =
|
|
var bootstrapNodes = newSeq[ENode]()
|
|
|
|
for node in node.config.bootstrapNodes:
|
|
bootstrapNodes.add initENode(node)
|
|
|
|
let bootstrapFile = string node.config.bootstrapNodesFile
|
|
if bootstrapFile.len > 0:
|
|
for ln in lines(bootstrapFile):
|
|
bootstrapNodes.add initENode(string ln)
|
|
|
|
if bootstrapNodes.len > 0:
|
|
info "Connecting to bootstrap nodes", bootstrapNodes
|
|
else:
|
|
info "Waiting for connections"
|
|
await node.network.connectToNetwork(bootstrapNodes)
|
|
|
|
proc sync*(node: BeaconNode): Future[bool] {.async.} =
|
|
if node.state.data.slotDistanceFromNow() > WEAK_SUBJECTVITY_PERIOD.int64:
|
|
# node.state.data = await obtainTrustedStateSnapshot(node.db)
|
|
return false
|
|
else:
|
|
# TODO waiting for genesis should probably be moved elsewhere.. it has
|
|
# little to do with syncing..
|
|
let t = now()
|
|
if t < node.state.data.genesis_time * 1000:
|
|
notice "Waiting for genesis",
|
|
fromNow = int(node.state.data.genesis_time * 1000 - t) div 1000
|
|
await sleepAsync int(node.state.data.genesis_time * 1000 - t)
|
|
|
|
let
|
|
targetSlot = node.state.data.getSlotFromTime()
|
|
|
|
# TODO: change this to a full sync / block download
|
|
info "Syncing state from remote peers",
|
|
finalized_epoch = humaneEpochNum(node.state.data.finalized_epoch),
|
|
target_slot_epoch = humaneEpochNum(targetSlot.slot_to_epoch)
|
|
|
|
# TODO: sync is called at the beginning of the program, but doing this kind
|
|
# of catching up here is wrong - if we fall behind on processing
|
|
# for whatever reason, we want to be safe against the damage that
|
|
# might cause regardless if we just started or have been running for
|
|
# long. A classic example where this might happen is when the
|
|
# computer goes to sleep - when waking up, we'll be in the middle of
|
|
# processing, but behind everyone else.
|
|
# TOOD we now detect during epoch scheduling if we're very far behind -
|
|
# that would potentially be a good place to run the sync (?)
|
|
# while node.beaconState.finalized_epoch < targetSlot.slot_to_epoch:
|
|
# var (peer, changeLog) = await node.network.getValidatorChangeLog(
|
|
# node.beaconState.validator_registry_delta_chain_tip)
|
|
|
|
# if peer == nil:
|
|
# error "Failed to sync with any peer"
|
|
# return false
|
|
|
|
# if applyValidatorChangeLog(changeLog, node.beaconState):
|
|
# node.db.persistState(node.beaconState)
|
|
# node.db.persistBlock(changeLog.signedBlock)
|
|
# else:
|
|
# warn "Ignoring invalid validator change log", sentFrom = peer
|
|
|
|
return true
|
|
|
|
template findIt(s: openarray, predicate: untyped): int =
|
|
var res = -1
|
|
for i, it {.inject.} in s:
|
|
if predicate:
|
|
res = i
|
|
break
|
|
res
|
|
|
|
proc addLocalValidators*(node: BeaconNode) =
|
|
for privKey in node.config.validators:
|
|
let
|
|
pubKey = privKey.pubKey()
|
|
|
|
let idx = node.state.data.validator_registry.findIt(it.pubKey == pubKey)
|
|
if idx == -1:
|
|
warn "Validator not in registry", pubKey
|
|
else:
|
|
debug "Attaching validator", validator = shortValidatorKey(node, idx),
|
|
idx, pubKey
|
|
node.attachedValidators.addLocalValidator(idx, pubKey, privKey)
|
|
|
|
info "Local validators attached ", count = node.attachedValidators.count
|
|
|
|
proc getAttachedValidator(node: BeaconNode, idx: int): AttachedValidator =
|
|
let validatorKey = node.state.data.validator_registry[idx].pubkey
|
|
return node.attachedValidators.getValidator(validatorKey)
|
|
|
|
proc updateHead(node: BeaconNode) =
|
|
# TODO placeholder logic for running the fork choice
|
|
var
|
|
head = node.state.blck
|
|
headSlot = node.state.data.slot
|
|
|
|
# LRB fork choice - latest resolved block :)
|
|
for ph in node.potentialHeads:
|
|
let blck = node.blockPool.get(ph)
|
|
if blck.isNone():
|
|
continue
|
|
if blck.get().data.slot >= headSlot:
|
|
head = blck.get().refs
|
|
headSlot = blck.get().data.slot
|
|
node.potentialHeads.setLen(0)
|
|
|
|
if head.root == node.state.blck.root:
|
|
debug "No new head found",
|
|
stateRoot = shortLog(node.state.root),
|
|
blockRoot = shortLog(node.state.blck.root),
|
|
stateSlot = humaneSlotNum(node.state.data.slot)
|
|
return
|
|
|
|
node.blockPool.updateState(node.state, head)
|
|
|
|
# TODO this should probably be in blockpool, but what if updateState is
|
|
# called with a non-head block?
|
|
node.db.putHeadBlock(node.state.blck.root)
|
|
|
|
# TODO we should save the state every now and then, but which state do we
|
|
# save? When we receive a block and process it, the state from a
|
|
# particular epoch may become finalized - but we no longer have it!
|
|
# One thing that would work would be to replay from some earlier
|
|
# state (the tail?) to the new finalized state, then save that. Another
|
|
# option would be to simply save every epoch start state, and eventually
|
|
# point it out as it becomes finalized..
|
|
|
|
info "Updated head",
|
|
stateRoot = shortLog(node.state.root),
|
|
headBlockRoot = shortLog(node.state.blck.root),
|
|
stateSlot = humaneSlotNum(node.state.data.slot)
|
|
|
|
proc makeAttestation(node: BeaconNode,
|
|
validator: AttachedValidator,
|
|
slot: Slot,
|
|
shard: uint64,
|
|
committeeLen: int,
|
|
indexInCommittee: int) {.async.} =
|
|
doAssert node != nil
|
|
doAssert validator != nil
|
|
|
|
# It's time to make an attestation. To do so, we must determine what we
|
|
# consider to be the head block - this is done by the fork choice rule.
|
|
# TODO this lazy update of the head is good because it delays head resolution
|
|
# until the very latest moment - on the other hand, if it takes long, the
|
|
# attestation might be late!
|
|
node.updateHead()
|
|
|
|
# Check pending attestations - maybe we found some blocks for them
|
|
node.attestationPool.resolve(node.state.data)
|
|
|
|
# It might be that the latest block we found is an old one - if this is the
|
|
# case, we need to fast-forward the state
|
|
# TODO maybe this is not necessary? We just use the justified epoch from the
|
|
# state - investigate if it can change (and maybe restructure the state
|
|
# update code so it becomes obvious... this would require moving away
|
|
# from the huge state object)
|
|
var state = node.state.data
|
|
skipSlots(state, node.state.blck.root, slot)
|
|
|
|
let
|
|
justifiedBlockRoot =
|
|
get_block_root(state, get_epoch_start_slot(state.justified_epoch))
|
|
|
|
attestationData = AttestationData(
|
|
slot: slot,
|
|
shard: shard,
|
|
beacon_block_root: node.state.blck.root,
|
|
epoch_boundary_root: Eth2Digest(), # TODO
|
|
shard_block_root: Eth2Digest(), # TODO
|
|
latest_crosslink: state.latest_crosslinks[shard],
|
|
justified_epoch: state.justified_epoch,
|
|
justified_block_root: justifiedBlockRoot)
|
|
|
|
validatorSignature = await validator.signAttestation(attestationData)
|
|
|
|
var participationBitfield = repeat(0'u8, ceil_div8(committeeLen))
|
|
bitSet(participationBitfield, indexInCommittee)
|
|
|
|
var attestation = Attestation(
|
|
data: attestationData,
|
|
aggregate_signature: validatorSignature,
|
|
aggregation_bitfield: participationBitfield,
|
|
# Stub in phase0
|
|
custody_bitfield: newSeq[byte](participationBitfield.len)
|
|
)
|
|
|
|
# TODO what are we waiting for here? broadcast should never block, and never
|
|
# fail...
|
|
await node.network.broadcast(topicAttestations, attestation)
|
|
|
|
info "Attestation sent",
|
|
slot = humaneSlotNum(attestationData.slot),
|
|
shard = attestationData.shard,
|
|
validator = shortValidatorKey(node, validator.idx),
|
|
signature = shortLog(validatorSignature),
|
|
beaconBlockRoot = shortLog(attestationData.beacon_block_root)
|
|
|
|
proc proposeBlock(node: BeaconNode,
|
|
validator: AttachedValidator,
|
|
slot: Slot) {.async.} =
|
|
doAssert node != nil
|
|
doAssert validator != nil
|
|
doAssert validator.idx < node.state.data.validator_registry.len
|
|
|
|
# To propose a block, we should know what the head is, because that's what
|
|
# we'll be building the next block upon..
|
|
node.updateHead()
|
|
|
|
# To create a block, we'll first apply a partial block to the state, skipping
|
|
# some validations.
|
|
# TODO technically, we could leave the state with the new block applied here,
|
|
# though it works this way as well because eventually we'll receive the
|
|
# block through broadcast.. to apply or not to apply permantently, that
|
|
# is the question...
|
|
var state = node.state.data
|
|
|
|
skipSlots(state, node.state.blck.root, slot - 1)
|
|
|
|
var blockBody = BeaconBlockBody(
|
|
attestations: node.attestationPool.getAttestationsForBlock(slot))
|
|
|
|
var newBlock = BeaconBlock(
|
|
slot: slot,
|
|
parent_root: node.state.blck.root,
|
|
randao_reveal: validator.genRandaoReveal(state, state.slot),
|
|
eth1_data: node.mainchainMonitor.getBeaconBlockRef(),
|
|
signature: ValidatorSig(), # we need the rest of the block first!
|
|
body: blockBody)
|
|
|
|
let ok =
|
|
updateState(state, node.state.blck.root, some(newBlock), {skipValidation})
|
|
doAssert ok # TODO: err, could this fail somehow?
|
|
|
|
newBlock.state_root = Eth2Digest(data: hash_tree_root(state))
|
|
|
|
var signedData = ProposalSignedData(
|
|
slot: slot,
|
|
shard: BEACON_CHAIN_SHARD_NUMBER,
|
|
blockRoot: hash_tree_root_final(newBlock))
|
|
|
|
newBlock.signature = await validator.signBlockProposal(state.fork, signedData)
|
|
|
|
# TODO what are we waiting for here? broadcast should never block, and never
|
|
# fail...
|
|
await node.network.broadcast(topicBeaconBlocks, newBlock)
|
|
|
|
info "Block proposed",
|
|
slot = humaneSlotNum(slot),
|
|
stateRoot = shortLog(newBlock.state_root),
|
|
parentRoot = shortLog(newBlock.parent_root),
|
|
validator = shortValidatorKey(node, validator.idx),
|
|
idx = validator.idx
|
|
|
|
proc scheduleBlockProposal(node: BeaconNode,
|
|
slot: Slot,
|
|
validator: AttachedValidator) =
|
|
# TODO:
|
|
# This function exists only to hide a bug with Nim's closures.
|
|
# If you inline it in `scheduleEpochActions`, you'll see the
|
|
# internal `doAssert` starting to fail.
|
|
doAssert validator != nil
|
|
|
|
let
|
|
at = node.slotStart(slot)
|
|
now = fastEpochTime()
|
|
|
|
if now > at:
|
|
warn "Falling behind on block proposals", at, now, slot
|
|
|
|
info "Scheduling block proposal",
|
|
validator = shortValidatorKey(node, validator.idx),
|
|
idx = validator.idx,
|
|
slot = humaneSlotNum(slot),
|
|
fromNow = (at - now) div 1000
|
|
|
|
addTimer(at) do (x: pointer) {.gcsafe.}:
|
|
# TODO timers are generally not accurate / guaranteed to fire at the right
|
|
# time - need to guard here against early / late firings
|
|
doAssert validator != nil
|
|
asyncCheck proposeBlock(node, validator, slot)
|
|
|
|
proc scheduleAttestation(node: BeaconNode,
|
|
validator: AttachedValidator,
|
|
slot: Slot,
|
|
shard: uint64,
|
|
committeeLen: int,
|
|
indexInCommittee: int) =
|
|
# TODO:
|
|
# This function exists only to hide a bug with Nim's closures.
|
|
# If you inline it in `scheduleEpochActions`, you'll see the
|
|
# internal `doAssert` starting to fail.
|
|
doAssert validator != nil
|
|
|
|
let
|
|
at = node.slotStart(slot)
|
|
now = fastEpochTime()
|
|
|
|
if now > at:
|
|
warn "Falling behind on attestations", at, now, slot
|
|
|
|
debug "Scheduling attestation",
|
|
validator = shortValidatorKey(node, validator.idx),
|
|
fromNow = (at - now) div 1000,
|
|
slot = humaneSlotNum(slot),
|
|
shard
|
|
|
|
addTimer(at) do (p: pointer) {.gcsafe.}:
|
|
doAssert validator != nil
|
|
asyncCheck makeAttestation(node, validator, slot,
|
|
shard, committeeLen, indexInCommittee)
|
|
|
|
proc scheduleEpochActions(node: BeaconNode, epoch: Epoch) =
|
|
## This schedules the required block proposals and
|
|
## attestations from our attached validators.
|
|
doAssert node != nil
|
|
doAssert epoch >= GENESIS_EPOCH,
|
|
"Epoch: " & $epoch & ", humane epoch: " & $humaneEpochNum(epoch)
|
|
|
|
debug "Scheduling epoch actions",
|
|
epoch = humaneEpochNum(epoch),
|
|
stateEpoch = humaneEpochNum(node.state.data.slot.slot_to_epoch())
|
|
|
|
# Sanity check - verify that the current head block is not too far behind
|
|
if node.state.data.slot.slot_to_epoch() + 1 < epoch:
|
|
# Normally, we update the head state lazily, just before making an
|
|
# attestation. However, if we skip scheduling attestations, we'll never
|
|
# run the head update - thus we make an attempt now:
|
|
node.updateHead()
|
|
|
|
if node.state.data.slot.slot_to_epoch() + 1 < epoch:
|
|
# We're still behind!
|
|
#
|
|
# There's a few ways this can happen:
|
|
#
|
|
# * we receive no attestations or blocks for an extended period of time
|
|
# * all the attestations we receive are bogus - maybe we're connected to
|
|
# the wrong network?
|
|
# * we just started and still haven't synced
|
|
#
|
|
# TODO make an effort to find other nodes and sync? A worst case scenario
|
|
# here is that the network stalls because nobody is sending out
|
|
# attestations because nobody is scheduling them, in a vicious
|
|
# circle
|
|
# TODO diagnose the various scenarios and do something smart...
|
|
|
|
let
|
|
expectedSlot = node.state.data.getSlotFromTime()
|
|
nextSlot = expectedSlot + 1
|
|
at = node.slotStart(nextSlot)
|
|
|
|
notice "Delaying epoch scheduling, head too old - scheduling new attempt",
|
|
stateSlot = humaneSlotNum(node.state.data.slot),
|
|
expectedEpoch = humaneEpochNum(epoch),
|
|
expectedSlot = humaneSlotNum(expectedSlot),
|
|
fromNow = (at - fastEpochTime()) div 1000
|
|
|
|
addTimer(at) do (p: pointer):
|
|
node.scheduleEpochActions(nextSlot.slot_to_epoch())
|
|
return
|
|
|
|
# TODO: is this necessary with the new shuffling?
|
|
# see get_beacon_proposer_index
|
|
var nextState = node.state.data
|
|
|
|
skipSlots(nextState, node.state.blck.root, epoch.get_epoch_start_slot())
|
|
|
|
# TODO we don't need to do anything at slot 0 - what about slots we missed
|
|
# if we got delayed above?
|
|
let start = if epoch == GENESIS_EPOCH: 1.uint64 else: 0.uint64
|
|
|
|
for i in start ..< SLOTS_PER_EPOCH:
|
|
let slot = epoch * SLOTS_PER_EPOCH + i
|
|
nextState.slot = slot # ugly trick, see get_beacon_proposer_index
|
|
|
|
block: # Schedule block proposals
|
|
let proposerIdx = get_beacon_proposer_index(nextState, slot)
|
|
let validator = node.getAttachedValidator(proposerIdx)
|
|
|
|
if validator != nil:
|
|
# TODO:
|
|
# Warm-up the proposer earlier to try to obtain previous
|
|
# missing blocks if necessary
|
|
scheduleBlockProposal(node, slot, validator)
|
|
|
|
block: # Schedule attestations
|
|
for crosslink_committee in get_crosslink_committees_at_slot(
|
|
nextState, slot):
|
|
for i, validatorIdx in crosslink_committee.committee:
|
|
let validator = node.getAttachedValidator(validatorIdx)
|
|
if validator != nil:
|
|
scheduleAttestation(
|
|
node, validator, slot, crosslink_committee.shard,
|
|
crosslink_committee.committee.len, i)
|
|
|
|
let
|
|
nextEpoch = epoch + 1
|
|
at = node.slotStart(nextEpoch.get_epoch_start_slot())
|
|
|
|
info "Scheduling next epoch update",
|
|
fromNow = (at - fastEpochTime()) div 1000,
|
|
epoch = humaneEpochNum(nextEpoch)
|
|
|
|
addTimer(at) do (p: pointer):
|
|
node.scheduleEpochActions(nextEpoch)
|
|
|
|
proc fetchBlocks(node: BeaconNode, roots: seq[Eth2Digest]) =
|
|
if roots.len == 0: return
|
|
|
|
# TODO shouldn't send to all!
|
|
# TODO should never fail - asyncCheck is wrong here..
|
|
asyncCheck node.network.broadcast(topicfetchBlocks, roots)
|
|
|
|
proc onFetchBlocks(node: BeaconNode, roots: seq[Eth2Digest]) =
|
|
# TODO placeholder logic for block recovery
|
|
debug "fetchBlocks received",
|
|
roots = roots.len
|
|
for root in roots:
|
|
if (let blck = node.db.getBlock(root); blck.isSome()):
|
|
# TODO should never fail - asyncCheck is wrong here..
|
|
# TODO should obviously not spam, but rather send it back to the requester
|
|
asyncCheck node.network.broadcast(topicBeaconBlocks, blck.get())
|
|
|
|
proc scheduleSlotStartActions(node: BeaconNode, slot: Slot) =
|
|
# TODO in this setup, we retry fetching blocks at the beginning of every slot,
|
|
# hoping that we'll get some before it's time to attest or propose - is
|
|
# there a better time to do this?
|
|
let missingBlocks = node.blockPool.checkUnresolved()
|
|
node.fetchBlocks(missingBlocks)
|
|
|
|
let
|
|
nextSlot = slot + 1
|
|
at = node.slotStart(nextSlot)
|
|
|
|
info "Scheduling next slot start action block",
|
|
fromNow = (at - fastEpochTime()) div 1000,
|
|
slot = humaneSlotNum(nextSlot)
|
|
|
|
addTimer(at) do (p: pointer):
|
|
node.scheduleSlotStartActions(nextSlot)
|
|
|
|
proc onAttestation(node: BeaconNode, attestation: Attestation) =
|
|
# We received an attestation from the network but don't know much about it
|
|
# yet - in particular, we haven't verified that it belongs to particular chain
|
|
# we're on, or that it follows the rules of the protocol
|
|
debug "Attestation received",
|
|
slot = humaneSlotNum(attestation.data.slot),
|
|
shard = attestation.data.shard,
|
|
beaconBlockRoot = shortLog(attestation.data.beacon_block_root),
|
|
justifiedEpoch = humaneEpochNum(attestation.data.justified_epoch),
|
|
justifiedBlockRoot = shortLog(attestation.data.justified_block_root),
|
|
signature = shortLog(attestation.aggregate_signature)
|
|
|
|
node.attestationPool.add(node.state.data, attestation)
|
|
|
|
if attestation.data.beacon_block_root notin node.potentialHeads:
|
|
node.potentialHeads.add attestation.data.beacon_block_root
|
|
|
|
proc onBeaconBlock(node: BeaconNode, blck: BeaconBlock) =
|
|
# We received a block but don't know much about it yet - in particular, we
|
|
# don't know if it's part of the chain we're currently building.
|
|
let blockRoot = hash_tree_root_final(blck)
|
|
debug "Block received",
|
|
blockRoot = shortLog(blockRoot),
|
|
slot = humaneSlotNum(blck.slot),
|
|
stateRoot = shortLog(blck.state_root),
|
|
parentRoot = shortLog(blck.parent_root),
|
|
signature = shortLog(blck.signature),
|
|
proposer_slashings = blck.body.proposer_slashings.len,
|
|
attester_slashings = blck.body.attester_slashings.len,
|
|
attestations = blck.body.attestations.len,
|
|
deposits = blck.body.deposits.len,
|
|
voluntary_exits = blck.body.voluntary_exits.len,
|
|
transfers = blck.body.transfers.len
|
|
|
|
if not node.blockPool.add(blockRoot, blck):
|
|
# TODO the fact that add returns a bool that causes the parent block to be
|
|
# pre-emptively fetched is quite ugly - fix.
|
|
node.fetchBlocks(@[blck.parent_root])
|
|
|
|
# Delay updating the head until the latest moment possible - this makes it
|
|
# more likely that we've managed to resolve the block, in case of
|
|
# irregularities
|
|
if blockRoot notin node.potentialHeads:
|
|
node.potentialHeads.add blockRoot
|
|
|
|
# The block we received contains attestations, and we might not yet know about
|
|
# all of them. Let's add them to the attestation pool - in case they block
|
|
# is not yet resolved, neither will the attestations be!
|
|
for attestation in blck.body.attestations:
|
|
# TODO attestation pool needs to be taught to deal with overlapping
|
|
# attestations!
|
|
discard # node.onAttestation(attestation)
|
|
|
|
proc run*(node: BeaconNode) =
|
|
node.network.subscribe(topicBeaconBlocks) do (blck: BeaconBlock):
|
|
node.onBeaconBlock(blck)
|
|
|
|
node.network.subscribe(topicAttestations) do (attestation: Attestation):
|
|
node.onAttestation(attestation)
|
|
|
|
node.network.subscribe(topicfetchBlocks) do (roots: seq[Eth2Digest]):
|
|
node.onFetchBlocks(roots)
|
|
|
|
let nowSlot = node.state.data.getSlotFromTime()
|
|
|
|
node.scheduleEpochActions(nowSlot.slot_to_epoch())
|
|
node.scheduleSlotStartActions(nowSlot)
|
|
|
|
runForever()
|
|
|
|
var gPidFile: string
|
|
proc createPidFile(filename: string) =
|
|
createDir splitFile(filename).dir
|
|
writeFile filename, $getCurrentProcessId()
|
|
gPidFile = filename
|
|
addQuitProc proc {.noconv.} = removeFile gPidFile
|
|
|
|
when isMainModule:
|
|
let config = load BeaconNodeConf
|
|
if config.logLevel != LogLevel.NONE:
|
|
setLogLevel(config.logLevel)
|
|
|
|
case config.cmd
|
|
of createChain:
|
|
createStateSnapshot(
|
|
config.validatorsDir.string, config.numValidators, config.firstValidator,
|
|
config.genesisOffset, config.outputStateFile.string)
|
|
quit 0
|
|
|
|
of noCommand:
|
|
waitFor synchronizeClock()
|
|
createPidFile(config.dataDir.string / "beacon_node.pid")
|
|
|
|
var node = BeaconNode.init config
|
|
|
|
dynamicLogScope(node = node.config.tcpPort - 50000):
|
|
# TODO: while it's nice to cheat by waiting for connections here, we
|
|
# actually need to make this part of normal application flow -
|
|
# losing all connections might happen at any time and we should be
|
|
# prepared to handle it.
|
|
waitFor node.connectToNetwork()
|
|
|
|
if not waitFor node.sync():
|
|
quit 1
|
|
|
|
info "Starting beacon node",
|
|
slotsSinceFinalization = node.state.data.slotDistanceFromNow(),
|
|
stateSlot = humaneSlotNum(node.state.data.slot),
|
|
SHARD_COUNT,
|
|
SLOTS_PER_EPOCH,
|
|
SECONDS_PER_SLOT,
|
|
SPEC_VERSION
|
|
|
|
node.addLocalValidators()
|
|
node.run()
|