From 3e0c2839b544a8e345239afdb918392b6028f037 Mon Sep 17 00:00:00 2001 From: Csaba Kiraly Date: Wed, 10 May 2023 22:12:56 +0200 Subject: [PATCH] add recvFrom to fake DatagramTransport Signed-off-by: Csaba Kiraly --- libp2pdht/private/eth/p2p/discoveryv5/transport.nim | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/libp2pdht/private/eth/p2p/discoveryv5/transport.nim b/libp2pdht/private/eth/p2p/discoveryv5/transport.nim index 3662044..3e26fa0 100644 --- a/libp2pdht/private/eth/p2p/discoveryv5/transport.nim +++ b/libp2pdht/private/eth/p2p/discoveryv5/transport.nim @@ -37,14 +37,19 @@ when(true): #enable network emulator proc `$`*(transp: DatagramTransport): string = $transp.local + proc recvFrom[T](transp: DatagramTransport, remote: TransportAddress, + msg: sink seq[T], msglen = -1) = + #echo "recv from ", remote + {.gcsafe.}: + transp.ingress.addLast(msg) + # call the callback on remote + asyncCheck transp.callback(transp, remote) + proc sendTo*[T](transp: DatagramTransport, remote: TransportAddress, msg: sink seq[T], msglen = -1) {.async.} = #echo "sending to ", remote {.gcsafe.}: - network[remote.port].ingress.addLast(msg) - # call the callback on remote - asyncCheck network[remote.port].function(network[remote.port], transp.local) - + network[remote.port].recvFrom(transp.local, msg) proc getMessage*(t: DatagramTransport,): seq[byte] {. raises: [Defect, CatchableError].} =