mirror of
https://github.com/status-im/nim-eth-p2p.git
synced 2025-01-19 19:28:46 +00:00
87 lines
3.2 KiB
Nim
87 lines
3.2 KiB
Nim
|
import
|
||
|
|
macros,
|
||
|
|
chronicles, serialization, serialization/streams, json_serialization/writer,
|
||
|
|
private/types
|
||
|
|
|
||
|
|
export
|
||
|
|
# XXX: Nim visibility rules get in the way here.
|
||
|
|
# It would be nice if the users of this module don't have to
|
||
|
|
# import json_serializer, but this won't work at the moment,
|
||
|
|
# because the `encode` call inside `logMsgEvent` has its symbols
|
||
|
|
# mixed in from the module where `logMsgEvent` is called
|
||
|
|
# (instead of from this module, which will be more logical).
|
||
|
|
init, writeValue, getOutput
|
||
|
|
# TODO: File this as an issue
|
||
|
|
|
||
|
|
const tracingEnabled* = defined(p2pdump)
|
||
|
|
|
||
|
|
when tracingEnabled:
|
||
|
|
logStream p2pMessages[json[file(p2p_messages.json,truncate)]]
|
||
|
|
|
||
|
|
proc logMsgEventImpl(eventName: static[string],
|
||
|
|
peer: Peer,
|
||
|
|
protocol: ProtocolInfo,
|
||
|
|
msgId: int,
|
||
|
|
json: string) =
|
||
|
|
# this is kept as a separate proc to reduce the code bloat
|
||
|
|
p2pMessages.log LogLevel.NONE, eventName, port = int(peer.network.address.tcpPort),
|
||
|
|
peer = $peer.remote,
|
||
|
|
protocol = protocol.name,
|
||
|
|
msgId, data = JsonString(json)
|
||
|
|
|
||
|
|
proc logMsgEvent[Msg](eventName: static[string], peer: Peer, msg: Msg) =
|
||
|
|
mixin msgProtocol, protocolInfo, msgId
|
||
|
|
|
||
|
|
logMsgEventImpl(eventName, peer,
|
||
|
|
Msg.msgProtocol.protocolInfo,
|
||
|
|
Msg.msgId,
|
||
|
|
StringJsonWriter.encode(msg))
|
||
|
|
|
||
|
|
proc logSentMsgFields*(peer: NimNode,
|
||
|
|
protocolInfo: NimNode,
|
||
|
|
msgId: int,
|
||
|
|
fields: openarray[NimNode]): NimNode =
|
||
|
|
## This generates the tracing code inserted in the message sending procs
|
||
|
|
## `fields` contains all the params that were serialized in the message
|
||
|
|
var tracer = ident("tracer")
|
||
|
|
|
||
|
|
result = quote do:
|
||
|
|
var `tracer` = init StringJsonWriter
|
||
|
|
beginRecord(`tracer`)
|
||
|
|
|
||
|
|
for f in fields:
|
||
|
|
result.add newCall(bindSym"writeField", tracer, newLit($f), f)
|
||
|
|
|
||
|
|
result.add quote do:
|
||
|
|
endRecord(`tracer`)
|
||
|
|
logMsgEventImpl("outgoing_msg", `peer`, `protocolInfo`, `msgId`, getOutput(`tracer`))
|
||
|
|
|
||
|
|
template logSentMsg*(peer: Peer, msg: auto) =
|
||
|
|
logMsgEvent("outgoing_msg", peer, msg)
|
||
|
|
|
||
|
|
template logReceivedMsg*(peer: Peer, msg: auto) =
|
||
|
|
logMsgEvent("incoming_msg", peer, msg)
|
||
|
|
|
||
|
|
template logConnectedPeer*(peer: Peer) =
|
||
|
|
p2pMessages.log LogLevel.NONE, "peer_connected",
|
||
|
|
port = int(peer.network.address.tcpPort),
|
||
|
|
peer = $peer.remote
|
||
|
|
|
||
|
|
template logAcceptedPeer*(peer: Peer) =
|
||
|
|
p2pMessages.log LogLevel.NONE, "peer_accepted",
|
||
|
|
port = int(peer.network.address.tcpPort),
|
||
|
|
peer = $peer.remote
|
||
|
|
|
||
|
|
template logDisconnectedPeer*(peer: Peer) =
|
||
|
|
p2pMessages.log LogLevel.NONE, "peer_disconnected",
|
||
|
|
port = int(peer.network.address.tcpPort),
|
||
|
|
peer = $peer.remote
|
||
|
|
|
||
|
|
else:
|
||
|
|
template logSentMsg*(peer: Peer, msg: auto) = discard
|
||
|
|
template logReceivedMsg*(peer: Peer, msg: auto) = discard
|
||
|
|
template logConnectedPeer*(peer: Peer) = discard
|
||
|
|
template logAcceptedPeer*(peer: Peer) = discard
|
||
|
|
template logDisconnectedPeer*(peer: Peer) = discard
|
||
|
|
|