95 lines
3.1 KiB
Nim
95 lines
3.1 KiB
Nim
# beacon_chain
|
|
# Copyright (c) 2018-Present 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.
|
|
|
|
import
|
|
# Standard library
|
|
os, strutils, typetraits,
|
|
# Internals
|
|
../../beacon_chain/ssz,
|
|
../../beacon_chain/spec/datatypes/[phase0, altair],
|
|
../../beacon_chain/spec/[state_transition_epoch],
|
|
# Status libs,
|
|
snappy,
|
|
stew/byteutils,
|
|
serialization, json_serialization
|
|
|
|
export # Workaround:
|
|
# - https://github.com/status-im/nim-serialization/issues/4
|
|
# - https://github.com/status-im/nim-serialization/issues/5
|
|
# - https://github.com/nim-lang/Nim/issues/11225
|
|
serialization.readValue,
|
|
Json, ssz, phase0, altair
|
|
|
|
# Process current EF test format
|
|
# ---------------------------------------------
|
|
|
|
# #######################
|
|
# JSON deserialization
|
|
|
|
func readValue*(r: var JsonReader, a: var seq[byte]) =
|
|
## Custom deserializer for seq[byte]
|
|
a = hexToSeqByte(r.readValue(string))
|
|
|
|
# #######################
|
|
# Test helpers
|
|
|
|
type
|
|
UnconsumedInput* = object of CatchableError
|
|
TestSizeError* = object of ValueError
|
|
|
|
const
|
|
FixturesDir* =
|
|
currentSourcePath.rsplit(DirSep, 1)[0] / ".." / ".." / "vendor" / "nim-eth2-scenarios"
|
|
SszTestsDir* = FixturesDir / "tests-v1.1.0-beta.2"
|
|
MaxObjectSize* = 3_000_000
|
|
|
|
proc parseTest*(path: string, Format: typedesc[Json], T: typedesc): T =
|
|
try:
|
|
# debugEcho " [Debug] Loading file: \"", path, '\"'
|
|
result = Format.loadFile(path, T)
|
|
except SerializationError as err:
|
|
writeStackTrace()
|
|
stderr.write $Format & " load issue for file \"", path, "\"\n"
|
|
stderr.write err.formatMsg(path), "\n"
|
|
quit 1
|
|
|
|
template readFileBytes*(path: string): seq[byte] =
|
|
cast[seq[byte]](readFile(path))
|
|
|
|
proc sszDecodeEntireInput*(input: openArray[byte], Decoded: type): Decoded =
|
|
var stream = unsafeMemoryInput(input)
|
|
var reader = init(SszReader, stream)
|
|
reader.readValue(result)
|
|
|
|
if stream.readable:
|
|
raise newException(UnconsumedInput, "Remaining bytes in the input")
|
|
|
|
proc parseTest*(path: string, Format: typedesc[SSZ], T: typedesc): T =
|
|
try:
|
|
# debugEcho " [Debug] Loading file: \"", path, '\"'
|
|
sszDecodeEntireInput(snappy.decode(readFileBytes(path), MaxObjectSize), T)
|
|
except SerializationError as err:
|
|
writeStackTrace()
|
|
stderr.write $Format & " load issue for file \"", path, "\"\n"
|
|
stderr.write err.formatMsg(path), "\n"
|
|
quit 1
|
|
|
|
proc process_justification_and_finalization*(state: var phase0.BeaconState) =
|
|
var cache = StateCache()
|
|
|
|
var rewards: RewardInfo
|
|
rewards.init(state)
|
|
rewards.process_attestations(state, cache)
|
|
process_justification_and_finalization(state, rewards.total_balances)
|
|
|
|
func process_slashings*(state: var phase0.BeaconState) =
|
|
var cache = StateCache()
|
|
var rewards: RewardInfo
|
|
rewards.init(state)
|
|
rewards.process_attestations(state, cache)
|
|
process_slashings(state, rewards.total_balances.current_epoch)
|