2019-11-12 15:21:15 +00:00
|
|
|
# beacon_chain
|
2022-01-05 08:42:56 +00:00
|
|
|
# Copyright (c) 2018-2022 Status Research & Development GmbH
|
2019-11-12 15:21:15 +00:00
|
|
|
# Licensed and distributed under either of
|
2019-11-25 15:30:02 +00:00
|
|
|
# * 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).
|
2019-11-12 15:21:15 +00:00
|
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
|
2021-05-28 20:32:26 +00:00
|
|
|
{.used.}
|
|
|
|
|
2019-11-12 15:21:15 +00:00
|
|
|
import
|
|
|
|
# Standard library
|
2021-04-28 16:41:02 +00:00
|
|
|
os, strutils, streams, strformat,
|
2019-11-12 15:21:15 +00:00
|
|
|
macros, sets,
|
|
|
|
# Third-party
|
|
|
|
yaml,
|
|
|
|
# Beacon chain internals
|
2021-05-28 20:32:26 +00:00
|
|
|
../../beacon_chain/spec/datatypes/phase0,
|
2021-05-24 08:42:40 +00:00
|
|
|
# Status libraries
|
|
|
|
snappy,
|
2019-11-12 15:21:15 +00:00
|
|
|
# Test utilities
|
2021-05-28 20:32:26 +00:00
|
|
|
../../testutil, ../fixtures_utils
|
2019-11-12 15:21:15 +00:00
|
|
|
|
|
|
|
# SSZ tests of consensus objects (minimal/mainnet preset specific)
|
|
|
|
|
|
|
|
# Parsing definitions
|
|
|
|
# ----------------------------------------------------------------
|
|
|
|
|
|
|
|
const
|
2021-05-28 20:32:26 +00:00
|
|
|
SSZDir = SszTestsDir/const_preset/"phase0"/"ssz_static"
|
2019-11-12 15:21:15 +00:00
|
|
|
|
|
|
|
type
|
|
|
|
SSZHashTreeRoot = object
|
|
|
|
# The test files have the values at the "root"
|
|
|
|
# so we **must** use "root" as a field name
|
|
|
|
root: string
|
|
|
|
# Some have a signing_root field
|
2020-11-24 18:07:58 +00:00
|
|
|
signing_root {.defaultVal: "".}: string
|
2019-11-12 15:21:15 +00:00
|
|
|
|
2019-12-16 18:08:50 +00:00
|
|
|
# Note this only tracks HashTreeRoot
|
2019-11-12 15:21:15 +00:00
|
|
|
# Checking the values against the yaml file is TODO (require more flexible Yaml parser)
|
|
|
|
|
2021-08-12 13:08:20 +00:00
|
|
|
proc checkSSZ(T: type phase0.SignedBeaconBlock, dir: string, expectedHash: SSZHashTreeRoot) =
|
2020-04-29 20:12:07 +00:00
|
|
|
# Deserialize into a ref object to not fill Nim stack
|
2021-05-24 08:42:40 +00:00
|
|
|
let encoded = snappy.decode(
|
|
|
|
readFileBytes(dir/"serialized.ssz_snappy"), MaxObjectSize)
|
2021-09-27 14:22:58 +00:00
|
|
|
let deserialized = newClone(sszDecodeEntireInput(encoded, T))
|
2020-04-29 20:12:07 +00:00
|
|
|
|
|
|
|
# SignedBeaconBlocks usually not hashed because they're identified by
|
|
|
|
# htr(BeaconBlock), so do it manually
|
2022-04-14 10:47:14 +00:00
|
|
|
check: expectedHash.root == "0x" & toLowerAscii($hash_tree_root(
|
2020-04-29 20:12:07 +00:00
|
|
|
[hash_tree_root(deserialized.message),
|
|
|
|
hash_tree_root(deserialized.signature)]))
|
|
|
|
|
2020-07-16 13:16:51 +00:00
|
|
|
check deserialized.root == hash_tree_root(deserialized.message)
|
2020-05-11 16:27:44 +00:00
|
|
|
check SSZ.encode(deserialized[]) == encoded
|
|
|
|
check sszSize(deserialized[]) == encoded.len
|
|
|
|
|
|
|
|
# TODO check the value (requires YAML loader)
|
2020-04-29 20:12:07 +00:00
|
|
|
|
2020-05-11 16:27:44 +00:00
|
|
|
proc checkSSZ(T: type, dir: string, expectedHash: SSZHashTreeRoot) =
|
2019-11-12 15:21:15 +00:00
|
|
|
# Deserialize into a ref object to not fill Nim stack
|
2021-05-24 08:42:40 +00:00
|
|
|
let encoded = snappy.decode(
|
|
|
|
readFileBytes(dir/"serialized.ssz_snappy"), MaxObjectSize)
|
2021-09-27 14:22:58 +00:00
|
|
|
let deserialized = newClone(sszDecodeEntireInput(encoded, T))
|
2019-11-12 15:21:15 +00:00
|
|
|
|
2022-04-14 10:47:14 +00:00
|
|
|
check: expectedHash.root == "0x" & toLowerAscii($hash_tree_root(deserialized[]))
|
2019-11-12 15:21:15 +00:00
|
|
|
|
2020-05-11 16:27:44 +00:00
|
|
|
check SSZ.encode(deserialized[]) == encoded
|
|
|
|
check sszSize(deserialized[]) == encoded.len
|
|
|
|
|
|
|
|
# TODO check the value (requires YAML loader)
|
2019-11-12 15:21:15 +00:00
|
|
|
|
|
|
|
proc loadExpectedHashTreeRoot(dir: string): SSZHashTreeRoot =
|
2022-02-20 20:13:06 +00:00
|
|
|
let s = openFileStream(dir/"roots.yaml")
|
2019-11-12 15:21:15 +00:00
|
|
|
yaml.load(s, result)
|
|
|
|
s.close()
|
|
|
|
|
|
|
|
# Test runner
|
|
|
|
# ----------------------------------------------------------------
|
|
|
|
|
2022-01-05 08:42:56 +00:00
|
|
|
suite "EF - Phase 0 - SSZ consensus objects " & preset():
|
2022-03-24 00:38:48 +00:00
|
|
|
doAssert dirExists(SSZDir), "You need to run the \"download_test_vectors.sh\" script to retrieve the consensus spec test vectors."
|
2021-07-02 17:33:59 +00:00
|
|
|
for pathKind, sszType in walkDir(SSZDir, relative = true, checkDir = true):
|
2019-11-22 15:47:08 +00:00
|
|
|
doAssert pathKind == pcDir
|
2019-11-12 15:21:15 +00:00
|
|
|
|
2021-04-28 16:41:02 +00:00
|
|
|
test &" Testing {sszType}":
|
2019-11-12 15:21:15 +00:00
|
|
|
let path = SSZDir/sszType
|
2021-07-02 17:33:59 +00:00
|
|
|
for pathKind, sszTestKind in walkDir(
|
|
|
|
path, relative = true, checkDir = true):
|
2019-11-22 15:47:08 +00:00
|
|
|
doAssert pathKind == pcDir
|
2019-11-12 15:21:15 +00:00
|
|
|
let path = SSZDir/sszType/sszTestKind
|
2021-07-02 17:33:59 +00:00
|
|
|
for pathKind, sszTestCase in walkDir(
|
|
|
|
path, relative = true, checkDir = true):
|
2019-11-12 15:21:15 +00:00
|
|
|
let path = SSZDir/sszType/sszTestKind/sszTestCase
|
|
|
|
let hash = loadExpectedHashTreeRoot(path)
|
|
|
|
|
|
|
|
case sszType:
|
2019-12-16 18:08:50 +00:00
|
|
|
of "AggregateAndProof": checkSSZ(AggregateAndProof, path, hash)
|
2019-11-12 15:21:15 +00:00
|
|
|
of "Attestation": checkSSZ(Attestation, path, hash)
|
|
|
|
of "AttestationData": checkSSZ(AttestationData, path, hash)
|
|
|
|
of "AttesterSlashing": checkSSZ(AttesterSlashing, path, hash)
|
2021-08-12 13:08:20 +00:00
|
|
|
of "BeaconBlock": checkSSZ(phase0.BeaconBlock, path, hash)
|
|
|
|
of "BeaconBlockBody": checkSSZ(phase0.BeaconBlockBody, path, hash)
|
2019-11-19 11:04:51 +00:00
|
|
|
of "BeaconBlockHeader": checkSSZ(BeaconBlockHeader, path, hash)
|
2021-08-12 13:08:20 +00:00
|
|
|
of "BeaconState": checkSSZ(phase0.BeaconState, path, hash)
|
2019-11-12 15:21:15 +00:00
|
|
|
of "Checkpoint": checkSSZ(Checkpoint, path, hash)
|
|
|
|
of "Deposit": checkSSZ(Deposit, path, hash)
|
|
|
|
of "DepositData": checkSSZ(DepositData, path, hash)
|
2019-12-16 18:08:50 +00:00
|
|
|
of "DepositMessage": checkSSZ(DepositMessage, path, hash)
|
2020-01-14 17:46:58 +00:00
|
|
|
of "Eth1Block": checkSSZ(Eth1Block, path, hash)
|
2019-11-12 15:21:15 +00:00
|
|
|
of "Eth1Data": checkSSZ(Eth1Data, path, hash)
|
|
|
|
of "Fork": checkSSZ(Fork, path, hash)
|
2020-03-14 21:54:45 +00:00
|
|
|
of "ForkData": checkSSZ(ForkData, path, hash)
|
2019-11-12 15:21:15 +00:00
|
|
|
of "HistoricalBatch": checkSSZ(HistoricalBatch, path, hash)
|
|
|
|
of "IndexedAttestation": checkSSZ(IndexedAttestation, path, hash)
|
|
|
|
of "PendingAttestation": checkSSZ(PendingAttestation, path, hash)
|
|
|
|
of "ProposerSlashing": checkSSZ(ProposerSlashing, path, hash)
|
2020-03-14 21:54:45 +00:00
|
|
|
of "SignedAggregateAndProof":
|
|
|
|
checkSSZ(SignedAggregateAndProof, path, hash)
|
2021-08-12 13:08:20 +00:00
|
|
|
of "SignedBeaconBlock": checkSSZ(phase0.SignedBeaconBlock, path, hash)
|
2019-12-16 18:08:50 +00:00
|
|
|
of "SignedBeaconBlockHeader":
|
|
|
|
checkSSZ(SignedBeaconBlockHeader, path, hash)
|
|
|
|
of "SignedVoluntaryExit": checkSSZ(SignedVoluntaryExit, path, hash)
|
2021-05-28 15:25:58 +00:00
|
|
|
of "SigningData": checkSSZ(SigningData, path, hash)
|
2019-11-19 11:04:51 +00:00
|
|
|
of "Validator": checkSSZ(Validator, path, hash)
|
|
|
|
of "VoluntaryExit": checkSSZ(VoluntaryExit, path, hash)
|
2019-11-12 15:21:15 +00:00
|
|
|
else:
|
|
|
|
raise newException(ValueError, "Unsupported test: " & sszType)
|