status-im/nim-eth
2
0
Fork 0
mirror of https://github.com/status-im/nim-eth.git synced 2025-02-16 16:06:35 +00:00
Code Issues Projects Releases Wiki Activity

Expose id and address of talkreq sender (#393)

Browse Source
This commit is contained in:
KonradStaniec 2021-09-07 10:49:18 +02:00 committed by GitHub
parent bfadcfbfaf
commit c078f85e48
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
eth/p2p/discoveryv5/protocol.nim
View File

@ -140,7 +140,7 @@ type
node: Node node: Node
message: seq[byte] message: seq[byte]
TalkProtocolHandler* = proc(p: TalkProtocol, request: seq[byte]): seq[byte] TalkProtocolHandler* = proc(p: TalkProtocol, request: seq[byte], fromId: NodeId, fromUdpAddress: Address): seq[byte]
{.gcsafe, raises: [Defect].} {.gcsafe, raises: [Defect].}
TalkProtocol* = ref object of RootObj TalkProtocol* = ref object of RootObj
@ -318,7 +318,7 @@ proc handleTalkReq(d: Protocol, fromId: NodeId, fromAddr: Address,
TalkRespMessage(response: @[]) TalkRespMessage(response: @[])
else: else:
TalkRespMessage(response: talkProtocol.protocolHandler(talkProtocol, TalkRespMessage(response: talkProtocol.protocolHandler(talkProtocol,
talkreq.request)) talkreq.request, fromId, fromAddr))
let (data, _) = encodeMessagePacket(d.rng[], d.codec, fromId, fromAddr, let (data, _) = encodeMessagePacket(d.rng[], d.codec, fromId, fromAddr,
encodeMessage(talkresp, reqId)) encodeMessage(talkresp, reqId))

4
tests/p2p/test_discoveryv5.nim
View File

@ -645,7 +645,7 @@ procSuite "Discovery v5 Tests":
rng, PrivateKey.random(rng[]), localAddress(20303)) rng, PrivateKey.random(rng[]), localAddress(20303))
talkProtocol = "echo".toBytes() talkProtocol = "echo".toBytes()
proc handler(protocol: TalkProtocol, request: seq[byte]): seq[byte] proc handler(protocol: TalkProtocol, request: seq[byte], fromId: NodeId, fromUdpAddress: Address): seq[byte]
{.gcsafe, raises: [Defect].} = {.gcsafe, raises: [Defect].} =
request request
@ -670,7 +670,7 @@ procSuite "Discovery v5 Tests":
rng, PrivateKey.random(rng[]), localAddress(20303)) rng, PrivateKey.random(rng[]), localAddress(20303))
talkProtocol = "echo".toBytes() talkProtocol = "echo".toBytes()
proc handler(protocol: TalkProtocol, request: seq[byte]): seq[byte] proc handler(protocol: TalkProtocol, request: seq[byte], fromId: NodeId, fromUdpAddress: Address): seq[byte]
{.gcsafe, raises: [Defect].} = {.gcsafe, raises: [Defect].} =
request request