126 lines
3.9 KiB
Nim
126 lines
3.9 KiB
Nim
# Nimbus
|
|
# Copyright (c) 2021 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: [Defect].}
|
|
|
|
import
|
|
nimcrypto/[hash, sha2, keccak], stew/[objects, results], stint,
|
|
ssz_serialization,
|
|
../../common/common_types
|
|
|
|
export ssz_serialization, common_types, hash, results
|
|
|
|
type
|
|
NodeHash* = MDigest[32 * 8] # keccak256
|
|
CodeHash* = MDigest[32 * 8] # keccak256
|
|
Address* = array[20, byte]
|
|
|
|
ContentType* = enum
|
|
accountTrieNode = 0x00
|
|
contractStorageTrieNode = 0x01
|
|
accountTrieProof = 0x02
|
|
contractStorageTrieProof = 0x03
|
|
contractBytecode = 0x04
|
|
|
|
AccountTrieNodeKey* = object
|
|
path*: ByteList
|
|
nodeHash*: NodeHash
|
|
stateRoot*: Bytes32
|
|
|
|
ContractStorageTrieNodeKey* = object
|
|
address*: Address
|
|
path*: ByteList
|
|
nodeHash*: NodeHash
|
|
stateRoot*: Bytes32
|
|
|
|
AccountTrieProofKey* = object
|
|
address*: Address
|
|
stateRoot*: Bytes32
|
|
|
|
ContractStorageTrieProofKey* = object
|
|
address*: Address
|
|
slot*: UInt256
|
|
stateRoot*: Bytes32
|
|
|
|
ContractBytecodeKey* = object
|
|
address*: Address
|
|
codeHash*: CodeHash
|
|
|
|
ContentKey* = object
|
|
case contentType*: ContentType
|
|
of accountTrieNode:
|
|
accountTrieNodeKey*: AccountTrieNodeKey
|
|
of contractStorageTrieNode:
|
|
contractStorageTrieNodeKey*: ContractStorageTrieNodeKey
|
|
of accountTrieProof:
|
|
accountTrieProofKey*: AccountTrieProofKey
|
|
of contractStorageTrieProof:
|
|
contractStorageTrieProofKey*: ContractStorageTrieProofKey
|
|
of contractBytecode:
|
|
contractBytecodeKey*: ContractBytecodeKey
|
|
|
|
func encode*(contentKey: ContentKey): ByteList =
|
|
ByteList.init(SSZ.encode(contentKey))
|
|
|
|
func decode*(contentKey: ByteList): Opt[ContentKey] =
|
|
try:
|
|
Opt.some(SSZ.decode(contentKey.asSeq(), ContentKey))
|
|
except SszError:
|
|
return Opt.none(ContentKey)
|
|
|
|
template computeContentId*(digestCtxType: type, body: untyped): ContentId =
|
|
var h {.inject.}: digestCtxType
|
|
init(h)
|
|
body
|
|
let idHash = finish(h)
|
|
readUintBE[256](idHash.data)
|
|
|
|
func toContentId*(contentKey: ContentKey): ContentId =
|
|
case contentKey.contentType:
|
|
of accountTrieNode: # sha256(path | node_hash)
|
|
let key = contentKey.accountTrieNodeKey
|
|
computeContentId sha256:
|
|
h.update(key.path.asSeq())
|
|
h.update(key.nodeHash.data)
|
|
of contractStorageTrieNode: # sha256(address | path | node_hash)
|
|
let key = contentKey.contractStorageTrieNodeKey
|
|
computeContentId sha256:
|
|
h.update(key.address)
|
|
h.update(key.path.asSeq())
|
|
h.update(key.nodeHash.data)
|
|
of accountTrieProof: # keccak(address)
|
|
let key = contentKey.accountTrieProofKey
|
|
computeContentId keccak256:
|
|
h.update(key.address)
|
|
of contractStorageTrieProof: # (keccak(address) + keccak(slot)) % 2**256
|
|
# TODO: Why is keccak run on slot, when it can be used directly?
|
|
# Also, value to LE or BE? Not mentioned in specification.
|
|
let key = contentKey.contractStorageTrieProofKey
|
|
let n1 =
|
|
block: computeContentId keccak256:
|
|
h.update(key.address)
|
|
let n2 =
|
|
block: computeContentId keccak256:
|
|
h.update(toBytesBE(key.slot))
|
|
|
|
n1 + n2 # uint256 will wrap arround, practically applying the modulo 256
|
|
of contractBytecode: # sha256(address | code_hash)
|
|
let key = contentKey.contractBytecodeKey
|
|
computeContentId sha256:
|
|
h.update(key.address)
|
|
h.update(key.codeHash.data)
|
|
|
|
func toContentId*(contentKey: ByteList): results.Opt[ContentId] =
|
|
let key = decode(contentKey)
|
|
if key.isSome():
|
|
ok(key.get().toContentId())
|
|
else:
|
|
Opt.none(ContentId)
|