87 lines
2.5 KiB
Nim
87 lines
2.5 KiB
Nim
# Nimbus
|
|
# Copyright (c) 2021-2023 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.
|
|
|
|
{.push raises: [].}
|
|
|
|
import
|
|
json_rpc/jsonmarshal,
|
|
stew/results,
|
|
eth/p2p/discoveryv5/[routing_table, enr, node]
|
|
|
|
export jsonmarshal, routing_table, enr, node
|
|
|
|
type
|
|
NodeInfo* = object
|
|
enr*: Record
|
|
nodeId*: NodeId
|
|
|
|
RoutingTableInfo* = object
|
|
localNodeId*: NodeId
|
|
buckets*: seq[seq[NodeId]]
|
|
|
|
func getNodeInfo*(r: RoutingTable): NodeInfo =
|
|
NodeInfo(enr: r.localNode.record, nodeId: r.localNode.id)
|
|
|
|
func getRoutingTableInfo*(r: RoutingTable): RoutingTableInfo =
|
|
var info: RoutingTableInfo
|
|
for b in r.buckets:
|
|
var bucket: seq[NodeId]
|
|
for n in b.nodes:
|
|
bucket.add(n.id)
|
|
|
|
info.buckets.add(bucket)
|
|
|
|
info.localNodeId = r.localNode.id
|
|
|
|
info
|
|
|
|
func toNodeWithAddress*(enr: Record): Node {.raises: [ValueError].} =
|
|
let nodeRes = newNode(enr)
|
|
if nodeRes.isErr():
|
|
raise newException(ValueError, $nodeRes.error)
|
|
|
|
let node = nodeRes.get()
|
|
if node.address.isNone():
|
|
raise newException(ValueError, "ENR without address")
|
|
else:
|
|
node
|
|
|
|
func `%`*(value: Record): JsonNode =
|
|
newJString(value.toURI())
|
|
|
|
func fromJson*(n: JsonNode, argName: string, result: var Record)
|
|
{.raises: [ValueError].} =
|
|
n.kind.expect(JString, argName)
|
|
if not fromURI(result, n.getStr()):
|
|
raise newException(ValueError, "Invalid ENR")
|
|
|
|
func `%`*(value: NodeId): JsonNode =
|
|
%("0x" & value.toHex())
|
|
|
|
func fromJson*(n: JsonNode, argName: string, result: var NodeId)
|
|
{.raises: [ValueError].} =
|
|
n.kind.expect(JString, argName)
|
|
|
|
# TODO: fromHex (and thus parse) call seems to let pass several invalid
|
|
# UInt256.
|
|
result = UInt256.fromHex(n.getStr())
|
|
|
|
# TODO: This one should go to nim-json-rpc but before we can do that we will
|
|
# have to update the vendor module to the current latest.
|
|
func fromJson*(n: JsonNode, argName: string, result: var uint16)
|
|
{.raises: [ValueError].} =
|
|
n.kind.expect(JInt, argName)
|
|
let asInt = n.getBiggestInt()
|
|
if asInt < 0:
|
|
raise newException(
|
|
ValueError, "JSON-RPC input is an unexpected negative value")
|
|
if asInt > BiggestInt(uint16.high()):
|
|
raise newException(
|
|
ValueError, "JSON-RPC input is too large for uint32")
|
|
|
|
result = uint16(asInt)
|