145 lines
3.8 KiB
Nim
145 lines
3.8 KiB
Nim
# nimbus-eth1
|
|
# Copyright (c) 2023-2024 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or
|
|
# http://www.apache.org/licenses/LICENSE-2.0)
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or
|
|
# http://opensource.org/licenses/MIT)
|
|
# at your option. This file may not be copied, modified, or distributed
|
|
# except according to those terms.
|
|
|
|
import stew/arrayops
|
|
|
|
type NibblesBuf* = object
|
|
## Allocation-free type for storing up to 64 4-bit nibbles, as seen in the
|
|
## Ethereum MPT
|
|
bytes: array[32, byte]
|
|
ibegin, iend: int8
|
|
# Where valid nibbles can be found - we use indices here to avoid copies
|
|
# wen slicing - iend not inclusive
|
|
|
|
func high*(T: type NibblesBuf): int =
|
|
63
|
|
|
|
func fromBytes*(T: type NibblesBuf, bytes: openArray[byte]): T =
|
|
result.iend = 2 * (int8 result.bytes.copyFrom(bytes))
|
|
|
|
func nibble*(T: type NibblesBuf, nibble: byte): T =
|
|
result.bytes[0] = nibble shl 4
|
|
result.iend = 1
|
|
|
|
template `[]`*(r: NibblesBuf, i: int): byte =
|
|
let pos = r.ibegin + i
|
|
if (pos and 1) != 0:
|
|
(r.bytes[pos shr 1] and 0xf)
|
|
else:
|
|
(r.bytes[pos shr 1] shr 4)
|
|
|
|
template `[]=`*(r: NibblesBuf, i: int, v: byte) =
|
|
let pos = r.ibegin + i
|
|
r.bytes[pos shr 1] =
|
|
if (pos and 1) != 0:
|
|
(v and 0x0f) or (r.bytes[pos shr 1] and 0xf0)
|
|
else:
|
|
(v shl 4) or (r.bytes[pos shr 1] and 0x0f)
|
|
|
|
func len*(r: NibblesBuf): int =
|
|
r.iend - r.ibegin
|
|
|
|
func `==`*(lhs, rhs: NibblesBuf): bool =
|
|
if lhs.len == rhs.len:
|
|
for i in 0 ..< lhs.len:
|
|
if lhs[i] != rhs[i]:
|
|
return false
|
|
return true
|
|
else:
|
|
return false
|
|
|
|
func `$`*(r: NibblesBuf): string =
|
|
result = newStringOfCap(64)
|
|
for i in 0 ..< r.len:
|
|
const chars = "0123456789abcdef"
|
|
result.add chars[r[i]]
|
|
|
|
func slice*(r: NibblesBuf, ibegin: int, iend = -1): NibblesBuf =
|
|
result.bytes = r.bytes
|
|
result.ibegin = r.ibegin + ibegin.int8
|
|
let e =
|
|
if iend < 0:
|
|
min(64, r.iend + iend + 1)
|
|
else:
|
|
min(64, r.ibegin + iend)
|
|
doAssert ibegin >= 0 and e <= result.bytes.len * 2
|
|
result.iend = e.int8
|
|
|
|
template writeFirstByte(nibbleCountExpr) {.dirty.} =
|
|
let nibbleCount = nibbleCountExpr
|
|
var oddnessFlag = (nibbleCount and 1) != 0
|
|
newSeq(result, (nibbleCount div 2) + 1)
|
|
result[0] = byte((int(isLeaf) * 2 + int(oddnessFlag)) shl 4)
|
|
var writeHead = 0
|
|
|
|
template writeNibbles(r) {.dirty.} =
|
|
for i in 0 ..< r.len:
|
|
let nextNibble = r[i]
|
|
if oddnessFlag:
|
|
result[writeHead] = result[writeHead] or nextNibble
|
|
else:
|
|
inc writeHead
|
|
result[writeHead] = nextNibble shl 4
|
|
oddnessFlag = not oddnessFlag
|
|
|
|
func toHexPrefix*(r: NibblesBuf, isLeaf = false): seq[byte] =
|
|
writeFirstByte(r.len)
|
|
writeNibbles(r)
|
|
|
|
func toHexPrefix*(r1, r2: NibblesBuf, isLeaf = false): seq[byte] =
|
|
writeFirstByte(r1.len + r2.len)
|
|
writeNibbles(r1)
|
|
writeNibbles(r2)
|
|
|
|
func sharedPrefixLen*(lhs, rhs: NibblesBuf): int =
|
|
result = 0
|
|
while result < lhs.len and result < rhs.len:
|
|
if lhs[result] != rhs[result]:
|
|
break
|
|
inc result
|
|
|
|
func startsWith*(lhs, rhs: NibblesBuf): bool =
|
|
sharedPrefixLen(lhs, rhs) == rhs.len
|
|
|
|
func fromHexPrefix*(
|
|
T: type NibblesBuf, r: openArray[byte]
|
|
): tuple[isLeaf: bool, nibbles: NibblesBuf] =
|
|
if r.len > 0:
|
|
result.isLeaf = (r[0] and 0x20) != 0
|
|
let hasOddLen = (r[0] and 0x10) != 0
|
|
|
|
var i = 0'i8
|
|
if hasOddLen:
|
|
result.nibbles[0] = r[0] and 0x0f
|
|
i += 1
|
|
|
|
for j in 1 ..< r.len:
|
|
if i >= 64:
|
|
break
|
|
result.nibbles[i] = r[j] shr 4
|
|
result.nibbles[i + 1] = r[j] and 0x0f
|
|
i += 2
|
|
|
|
result.nibbles.iend = i
|
|
else:
|
|
result.isLeaf = false
|
|
|
|
func `&`*(a, b: NibblesBuf): NibblesBuf =
|
|
for i in 0 ..< a.len:
|
|
result[i] = a[i]
|
|
|
|
for i in 0 ..< b.len:
|
|
result[i + a.len] = b[i]
|
|
|
|
result.iend = int8(min(64, a.len + b.len))
|
|
|
|
template getBytes*(a: NibblesBuf): array[32, byte] =
|
|
a.bytes
|