55 lines
2.0 KiB
Nim
55 lines
2.0 KiB
Nim
|
from std/sequtils import all
|
||
|
|
from stew/objects import isZeroMemory
|
||
|
|
|
||
|
|
import ./eth2_merkleization
|
||
|
|
from ./datatypes/base import Eth1Data, DepositContractState
|
||
|
|
from ./digest import Eth2Digest
|
||
|
|
|
||
|
|
export
|
||
|
|
depositCountBytes, depositCountU64
|
||
|
|
|
||
|
|
type
|
||
|
|
OldDepositContractSnapshot* = object
|
||
|
|
eth1Block*: Eth2Digest
|
||
|
|
depositContractState*: DepositContractState
|
||
|
|
|
||
|
|
DepositTreeSnapshot* = object
|
||
|
|
## https://eips.ethereum.org/EIPS/eip-4881
|
||
|
|
eth1Block*: Eth2Digest
|
||
|
|
depositContractState*: DepositContractState
|
||
|
|
blockHeight*: uint64
|
||
|
|
|
||
|
|
func toDepositTreeSnapshot*(d: OldDepositContractSnapshot,
|
||
|
|
blockHeight: uint64): DepositTreeSnapshot =
|
||
|
|
DepositTreeSnapshot(
|
||
|
|
eth1Block: d.eth1Block,
|
||
|
|
depositContractState: d.depositContractState,
|
||
|
|
blockHeight: blockHeight)
|
||
|
|
|
||
|
|
func toOldDepositContractSnapshot*(d: DepositTreeSnapshot): OldDepositContractSnapshot =
|
||
|
|
OldDepositContractSnapshot(eth1Block: d.eth1Block,
|
||
|
|
depositContractState: d.depositContractState)
|
||
|
|
|
||
|
|
template getDepositCountU64*(d: OldDepositContractSnapshot |
|
||
|
|
DepositTreeSnapshot): uint64 =
|
||
|
|
depositCountU64(d.depositContractState.deposit_count)
|
||
|
|
|
||
|
|
func getDepositRoot*(d: OldDepositContractSnapshot |
|
||
|
|
DepositTreeSnapshot): Eth2Digest =
|
||
|
|
let merk = DepositsMerkleizer.init(d.depositContractState)
|
||
|
|
let hash = merk.getFinalHash()
|
||
|
|
# TODO: mixInLength should accept unsigned int instead of int as
|
||
|
|
# this right now cuts in half the theoretical number of deposits.
|
||
|
|
return mixInLength(hash, int(merk.totalChunks))
|
||
|
|
|
||
|
|
func isValid*(d: DepositTreeSnapshot, wantedDepositRoot: Eth2Digest): bool =
|
||
|
|
## `isValid` requires the snapshot to be self-consistent and
|
||
|
|
## to point to a specific Ethereum block
|
||
|
|
return not (d.eth1Block.isZeroMemory or
|
||
|
|
d.blockHeight == 0 or
|
||
|
|
d.getDepositRoot() != wantedDepositRoot)
|
||
|
|
|
||
|
|
func matches*(snapshot: DepositTreeSnapshot, eth1_data: Eth1Data): bool =
|
||
|
|
snapshot.getDepositCountU64() == eth1_data.deposit_count and
|
||
|
|
snapshot.getDepositRoot() == eth1_data.deposit_root
|