53 lines
1.5 KiB
Nim
53 lines
1.5 KiB
Nim
|
## Nim-Codex
|
||
|
|
## Copyright (c) 2023 Status Research & Development GmbH
|
||
|
|
## Licensed under either of
|
||
|
|
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
||
|
|
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
||
|
|
## at your option.
|
||
|
|
## This file may not be copied, modified, or distributed except according to
|
||
|
|
## those terms.
|
||
|
|
|
||
|
|
import pkg/poseidon2
|
||
|
|
import pkg/questionable/results
|
||
|
|
import pkg/libp2p/multihash
|
||
|
|
|
||
|
|
import ../merkletree
|
||
|
|
|
||
|
|
func digestTree*(
|
||
|
|
_: type Poseidon2Tree,
|
||
|
|
bytes: openArray[byte], chunkSize: int): ?!Poseidon2Tree =
|
||
|
|
## Hashes chunks of data with a sponge of rate 2, and combines the
|
||
|
|
## resulting chunk hashes in a merkle root.
|
||
|
|
##
|
||
|
|
|
||
|
|
var index = 0
|
||
|
|
var leaves: seq[Poseidon2Hash]
|
||
|
|
while index < bytes.len:
|
||
|
|
let start = index
|
||
|
|
let finish = min(index + chunkSize, bytes.len)
|
||
|
|
let digest = Sponge.digest(bytes.toOpenArray(start, finish - 1), rate = 2)
|
||
|
|
leaves.add(digest)
|
||
|
|
index += chunkSize
|
||
|
|
return Poseidon2Tree.init(leaves)
|
||
|
|
|
||
|
|
func digest*(
|
||
|
|
_: type Poseidon2Tree,
|
||
|
|
bytes: openArray[byte], chunkSize: int): ?!Poseidon2Hash =
|
||
|
|
## Hashes chunks of data with a sponge of rate 2, and combines the
|
||
|
|
## resulting chunk hashes in a merkle root.
|
||
|
|
##
|
||
|
|
|
||
|
|
(? Poseidon2Tree.digestTree(bytes, chunkSize)).root
|
||
|
|
|
||
|
|
func digestMhash*(
|
||
|
|
_: type Poseidon2Tree,
|
||
|
|
bytes: openArray[byte], chunkSize: int): ?!MultiHash =
|
||
|
|
## Hashes chunks of data with a sponge of rate 2 and
|
||
|
|
## returns the multihash of the root
|
||
|
|
##
|
||
|
|
|
||
|
|
let
|
||
|
|
hash = ? Poseidon2Tree.digest(bytes, chunkSize)
|
||
|
|
|
||
|
|
? MultiHash.init(Pos2Bn128MrklCodec, hash).mapFailure
|