status-im
/
nim-libp2p-dht
mirror of https://github.com/status-im/nim-libp2p-dht.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

protocol: introduce sendResponse 

This completes the Request/Response semantics.
This commit is contained in:
Csaba Kiraly 2022-02-23 21:11:43 +01:00
parent 70a06b5c44
commit cdd5d6cd34
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -224,13 +224,18 @@ proc updateRecord*(
# TODO: Would it make sense to actively ping ("broadcast") to all the peers # TODO: Would it make sense to actively ping ("broadcast") to all the peers
# we stored a handshake with in order to get that ENR updated? # we stored a handshake with in order to get that ENR updated?
proc sendResponse(d: Protocol, dstId: NodeId, dstAddr: Address,
message: SomeMessage, reqId: RequestId) =
## send Response using the specifid reqId
d.transport.sendMessage(dstId, dstAddr, encodeMessage(message, reqId))
proc sendNodes(d: Protocol, toId: NodeId, toAddr: Address, reqId: RequestId, proc sendNodes(d: Protocol, toId: NodeId, toAddr: Address, reqId: RequestId,
nodes: openArray[Node]) = nodes: openArray[Node]) =
proc sendNodes(d: Protocol, toId: NodeId, toAddr: Address, proc sendNodes(d: Protocol, toId: NodeId, toAddr: Address,
message: NodesMessage, reqId: RequestId) {.nimcall.} = message: NodesMessage, reqId: RequestId) {.nimcall.} =
trace "Respond message packet", dstId = toId, address = toAddr, trace "Respond message packet", dstId = toId, address = toAddr,
kind = MessageKind.nodes kind = MessageKind.nodes
d.transport.sendMessage(toId, toAddr, encodeMessage(message, reqId)) d.sendResponse(toId, toAddr, message, reqId)
if nodes.len == 0: if nodes.len == 0:
# In case of 0 nodes, a reply is still needed # In case of 0 nodes, a reply is still needed
@ -258,7 +263,7 @@ proc handlePing(d: Protocol, fromId: NodeId, fromAddr: Address,
port: fromAddr.port.uint16) port: fromAddr.port.uint16)
trace "Respond message packet", dstId = fromId, address = fromAddr, trace "Respond message packet", dstId = fromId, address = fromAddr,
kind = MessageKind.pong kind = MessageKind.pong
d.transport.sendMessage(fromId, fromAddr, encodeMessage(pong, reqId)) d.sendResponse(fromId, fromAddr, pong, reqId)
proc handleFindNode(d: Protocol, fromId: NodeId, fromAddr: Address, proc handleFindNode(d: Protocol, fromId: NodeId, fromAddr: Address,
fn: FindNodeMessage, reqId: RequestId) = fn: FindNodeMessage, reqId: RequestId) =
@ -294,7 +299,7 @@ proc handleTalkReq(d: Protocol, fromId: NodeId, fromAddr: Address,
trace "Respond message packet", dstId = fromId, address = fromAddr, trace "Respond message packet", dstId = fromId, address = fromAddr,
kind = MessageKind.talkresp kind = MessageKind.talkresp
d.transport.sendMessage(fromId, fromAddr, encodeMessage(talkresp, reqId)) d.sendResponse(fromId, fromAddr, talkresp, reqId)
proc handleMessage(d: Protocol, srcId: NodeId, fromAddr: Address, proc handleMessage(d: Protocol, srcId: NodeId, fromAddr: Address,
message: Message) = message: Message) =