mirror of
https://github.com/status-im/nimbus-eth1.git
synced 2025-01-22 18:20:11 +00:00
124 lines
3.7 KiB
Nim
124 lines
3.7 KiB
Nim
# Fluffy
|
|
# Copyright (c) 2023-2024 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.
|
|
|
|
# As per spec:
|
|
# https://github.com/ethereum/portal-network-specs/blob/master/state-network.md#content-keys-and-content-ids
|
|
|
|
{.push raises: [].}
|
|
|
|
import results, ssz_serialization, ../../../common/common_types
|
|
|
|
export ssz_serialization, common_types, hash, results
|
|
|
|
const
|
|
MAX_TRIE_NODE_LEN* = 1024
|
|
MAX_TRIE_PROOF_LEN* = 65
|
|
MAX_BYTECODE_LEN* = 32768
|
|
|
|
type
|
|
TrieNode* = List[byte, MAX_TRIE_NODE_LEN]
|
|
TrieProof* = List[TrieNode, MAX_TRIE_PROOF_LEN]
|
|
Bytecode* = List[byte, MAX_BYTECODE_LEN]
|
|
|
|
AccountTrieNodeOffer* = object
|
|
proof*: TrieProof
|
|
blockHash*: Hash32
|
|
|
|
AccountTrieNodeRetrieval* = object
|
|
node*: TrieNode
|
|
|
|
ContractTrieNodeOffer* = object
|
|
storageProof*: TrieProof
|
|
accountProof*: TrieProof
|
|
blockHash*: Hash32
|
|
|
|
ContractTrieNodeRetrieval* = object
|
|
node*: TrieNode
|
|
|
|
ContractCodeOffer* = object
|
|
code*: Bytecode
|
|
accountProof*: TrieProof
|
|
blockHash*: Hash32
|
|
|
|
ContractCodeRetrieval* = object
|
|
code*: Bytecode
|
|
|
|
ContentOfferType* = AccountTrieNodeOffer | ContractTrieNodeOffer | ContractCodeOffer
|
|
ContentRetrievalType* =
|
|
AccountTrieNodeRetrieval | ContractTrieNodeRetrieval | ContractCodeRetrieval
|
|
ContentValueType* = ContentOfferType | ContentRetrievalType
|
|
|
|
func init*(T: type AccountTrieNodeOffer, proof: TrieProof, blockHash: Hash32): T =
|
|
T(proof: proof, blockHash: blockHash)
|
|
|
|
func init*(T: type AccountTrieNodeRetrieval, node: TrieNode): T =
|
|
T(node: node)
|
|
|
|
func init*(
|
|
T: type ContractTrieNodeOffer,
|
|
storageProof: TrieProof,
|
|
accountProof: TrieProof,
|
|
blockHash: Hash32,
|
|
): T =
|
|
T(storageProof: storageProof, accountProof: accountProof, blockHash: blockHash)
|
|
|
|
func init*(T: type ContractTrieNodeRetrieval, node: TrieNode): T =
|
|
T(node: node)
|
|
|
|
func init*(
|
|
T: type ContractCodeOffer,
|
|
code: Bytecode,
|
|
accountProof: TrieProof,
|
|
blockHash: Hash32,
|
|
): T =
|
|
T(code: code, accountProof: accountProof, blockHash: blockHash)
|
|
|
|
func init*(T: type ContractCodeRetrieval, code: Bytecode): T =
|
|
T(code: code)
|
|
|
|
template toRetrieval*(offer: AccountTrieNodeOffer): AccountTrieNodeRetrieval =
|
|
AccountTrieNodeRetrieval.init(offer.proof[^1])
|
|
|
|
template toRetrieval*(offer: ContractTrieNodeOffer): ContractTrieNodeRetrieval =
|
|
ContractTrieNodeRetrieval.init(offer.storageProof[^1])
|
|
|
|
template toRetrieval*(offer: ContractCodeOffer): ContractCodeRetrieval =
|
|
ContractCodeRetrieval.init(offer.code)
|
|
|
|
func toOffer*(
|
|
retrieval: AccountTrieNodeRetrieval, parent: AccountTrieNodeOffer
|
|
): AccountTrieNodeOffer =
|
|
var proof = parent.proof
|
|
let added = proof.add(retrieval.node)
|
|
doAssert(added)
|
|
AccountTrieNodeOffer.init(proof, parent.blockHash)
|
|
|
|
func toOffer*(
|
|
retrieval: ContractTrieNodeRetrieval, parent: ContractTrieNodeOffer
|
|
): ContractTrieNodeOffer =
|
|
var proof = parent.storageProof
|
|
let added = proof.add(retrieval.node)
|
|
doAssert(added)
|
|
ContractTrieNodeOffer.init(proof, parent.accountProof, parent.blockHash)
|
|
|
|
func toOffer*(
|
|
retrieval: ContractCodeRetrieval, parent: ContractCodeOffer
|
|
): ContractCodeOffer =
|
|
ContractCodeOffer.init(retrieval.code, parent.accountProof, parent.blockHash)
|
|
|
|
template empty*(T: type TrieProof): T =
|
|
T.init(@[])
|
|
|
|
template empty*(T: type Bytecode): T =
|
|
T(@[])
|
|
|
|
func encode*(value: ContentValueType): seq[byte] =
|
|
SSZ.encode(value)
|
|
|
|
func decode*(T: type ContentValueType, bytes: openArray[byte]): Result[T, string] =
|
|
decodeSsz(bytes, T)
|