67 lines
2.3 KiB
Nim
67 lines
2.3 KiB
Nim
import
|
|
tables,
|
|
chronos,
|
|
spec/[datatypes, crypto, helpers], ssz,
|
|
beacon_node_types
|
|
|
|
|
|
proc init*(T: type ValidatorPool): T =
|
|
result.validators = initTable[ValidatorPubKey, AttachedValidator]()
|
|
|
|
template count*(pool: ValidatorPool): int =
|
|
pool.validators.len
|
|
|
|
proc addLocalValidator*(pool: var ValidatorPool,
|
|
idx: int,
|
|
pubKey: ValidatorPubKey,
|
|
privKey: ValidatorPrivKey) =
|
|
let v = AttachedValidator(idx: idx,
|
|
kind: inProcess,
|
|
privKey: privKey)
|
|
pool.validators[pubKey] = v
|
|
|
|
proc getValidator*(pool: ValidatorPool,
|
|
validatorKey: ValidatorPubKey): AttachedValidator =
|
|
pool.validators.getOrDefault(validatorKey)
|
|
|
|
proc signBlockProposal*(v: AttachedValidator, fork: Fork,
|
|
blck: BeaconBlock): Future[ValidatorSig] {.async.} =
|
|
if v.kind == inProcess:
|
|
await sleepAsync(chronos.milliseconds(1))
|
|
result = bls_sign(v.privKey, signed_root(blck).data,
|
|
get_domain(fork, slot_to_epoch(blck.slot), DOMAIN_BEACON_BLOCK))
|
|
else:
|
|
# TODO:
|
|
# send RPC
|
|
discard
|
|
|
|
proc signAttestation*(v: AttachedValidator,
|
|
attestation: AttestationData): Future[ValidatorSig] {.async.} =
|
|
# TODO: implement this
|
|
if v.kind == inProcess:
|
|
await sleepAsync(chronos.milliseconds(1))
|
|
|
|
let attestationRoot = hash_tree_root(attestation)
|
|
# TODO: Avoid the allocations belows
|
|
var dataToSign = @(attestationRoot.data) & @[0'u8]
|
|
# TODO: Use `domain` here
|
|
let domain = 0'u64
|
|
result = bls_sign(v.privKey, dataToSign, domain)
|
|
else:
|
|
# TODO:
|
|
# send RPC
|
|
discard
|
|
|
|
func genRandaoReveal*(k: ValidatorPrivKey, state: BeaconState, slot: Slot):
|
|
ValidatorSig =
|
|
# https://github.com/ethereum/eth2.0-specs/blob/v0.2.0/specs/core/0_beacon-chain.md#randao
|
|
|
|
# Off-by-one? I often get slot == state.slot but the check was "doAssert slot > state.slot" (Mamy)
|
|
doAssert slot >= state.slot, "input slot: " & $humaneSlotNum(slot) & " - beacon state slot: " & $humaneSlotNum(state.slot)
|
|
bls_sign(k, hash_tree_root(slot_to_epoch(slot).uint64).data,
|
|
get_domain(state.fork, slot_to_epoch(slot), DOMAIN_RANDAO))
|
|
|
|
func genRandaoReveal*(v: AttachedValidator, state: BeaconState, slot: Slot):
|
|
ValidatorSig =
|
|
genRandaoReveal(v.privKey, state, slot)
|