nimbus-eth2/beacon_chain/fork_choice.nim

68 lines
2.1 KiB
Nim
Raw Normal View History

2018-11-26 13:33:06 +00:00
import
2018-11-29 01:08:34 +00:00
deques,
spec/[datatypes, crypto]
2018-11-26 13:33:06 +00:00
type
Attestation* = object
validator*: int
data*: AttestationData
2018-11-29 01:08:34 +00:00
signature*: ValidatorSig
2018-11-26 13:33:06 +00:00
AttestationPool* = object
2018-11-29 01:08:34 +00:00
attestations: Deque[seq[Attestation]]
2018-11-26 13:33:06 +00:00
startingSlot: int
proc init*(T: type AttestationPool, startingSlot: int): T =
result.attestationsPerSlot = initDeque[seq[Attestation]]()
result.startingSlot = startingSlot
2018-11-29 01:08:34 +00:00
proc setLen*[T](d: var Deque[T], len: int) =
# TODO: The upstream `Deque` type should gain a proper resize API
let delta = len - d.len
if delta > 0:
for i in 0 ..< delta:
var defaultVal: T
d.addLast(defaultVal)
else:
d.shrink(fromLast = delta)
2018-11-26 13:33:06 +00:00
proc add*(pool: var AttestationPool,
attestation: Attestation,
beaconState: BeaconState) =
# The caller of this function is responsible for ensuring that
# the attestations will be given in a strictly slot increasing order:
2018-11-29 01:08:34 +00:00
doAssert attestation.data.slot.int >= pool.startingSlot
2018-11-26 13:33:06 +00:00
2018-11-29 01:08:34 +00:00
let slotIdxInPool = attestation.data.slot.int - pool.startingSlot
2018-11-26 13:33:06 +00:00
if slotIdxInPool >= pool.attestations.len:
pool.attestations.setLen(slotIdxInPool + 1)
2018-11-29 01:08:34 +00:00
2018-11-26 13:33:06 +00:00
pool.attestations[slotIdxInPool].add attestation
iterator each*(pool: AttestationPool,
firstSlot, lastSlot: int): Attestation =
## Both indices are treated inclusively
## TODO: this should return a lent value
doAssert firstSlot <= lastSlot
for idx in countup(max(0, firstSlot - pool.startingSlot),
2018-11-29 01:08:34 +00:00
min(pool.attestations.len - 1, lastSlot - pool.startingSlot)):
2018-11-26 13:33:06 +00:00
for attestation in pool.attestations[idx]:
yield attestation
proc discardHistoryToSlot*(pool: var AttestationPool, slot: int) =
## The index is treated inclusively
let slotIdx = slot - pool.startingSlot
if slotIdx < 0: return
2018-11-29 01:08:34 +00:00
pool.attestations.shrink(fromFirst = slotIdx + 1)
2018-11-26 13:33:06 +00:00
proc getLatestAttestation*(pool: AttestationPool, validator: ValidatorRecord) =
discard
proc getLatestAttestationTarget*() =
discard
proc forkChoice*(pool: AttestationPool, oldHead, newBlock: BeaconBlock): bool =
# This will return true if the new block is accepted over the old head block
discard