import std/[strutils, sequtils, tables], confutils, chronos, stew/shims/net, chronicles/topics_registry import libp2p/protocols/ping, libp2p/crypto/[crypto, secp], libp2p/nameresolving/dnsresolver, libp2p/multicodec import ../../waku/v2/node/peer_manager, ../../waku/v2/waku_core, ../../waku/v2/waku_node # protocols and their tag const ProtocolsTable = { "store": "/vac/waku/store/", "relay": "/vac/waku/relay/", "lightpush": "/vac/waku/lightpush/", "filter": "/vac/waku/filter/", }.toTable const WebSocketPortOffset = 1000 # cli flags type WakuCanaryConf* = object address* {. desc: "Multiaddress of the peer node to attempt to dial", defaultValue: "", name: "address", abbr: "a".}: string timeout* {. desc: "Timeout to consider that the connection failed", defaultValue: chronos.seconds(10), name: "timeout", abbr: "t".}: chronos.Duration protocols* {. desc: "Protocol required to be supported: store,relay,lightpush,filter (can be used multiple times)", name: "protocol", abbr: "p".}: seq[string] logLevel* {. desc: "Sets the log level", defaultValue: LogLevel.DEBUG, name: "log-level", abbr: "l".}: LogLevel nodePort* {. desc: "Listening port for waku node", defaultValue: 60000, name: "node-port", abbr: "np".}: uint16 ## websocket secure config websocketSecureKeyPath* {. desc: "Secure websocket key path: '/path/to/key.txt' ", defaultValue: "" name: "websocket-secure-key-path".}: string websocketSecureCertPath* {. desc: "Secure websocket Certificate path: '/path/to/cert.txt' ", defaultValue: "" name: "websocket-secure-cert-path".}: string proc parseCmdArg*(T: type chronos.Duration, p: string): T = try: result = chronos.seconds(parseInt(p)) except CatchableError: raise newException(ConfigurationError, "Invalid timeout value") proc completeCmdArg*(T: type chronos.Duration, val: string): seq[string] = return @[] # checks if rawProtocols (skipping version) are supported in nodeProtocols proc areProtocolsSupported( rawProtocols: seq[string], nodeProtocols: seq[string]): bool = var numOfSupportedProt: int = 0 for nodeProtocol in nodeProtocols: for rawProtocol in rawProtocols: let protocolTag = ProtocolsTable[rawProtocol] if nodeProtocol.startsWith(protocolTag): info "Supported protocol ok", expected = protocolTag, supported = nodeProtocol numOfSupportedProt += 1 break if numOfSupportedProt == rawProtocols.len: return true return false proc main(rng: ref HmacDrbgContext): Future[int] {.async.} = let conf: WakuCanaryConf = WakuCanaryConf.load() # create dns resolver let nameServers = @[ initTAddress(ValidIpAddress.init("1.1.1.1"), Port(53)), initTAddress(ValidIpAddress.init("1.0.0.1"), Port(53))] resolver: DnsResolver = DnsResolver.new(nameServers) if conf.logLevel != LogLevel.NONE: setLogLevel(conf.logLevel) # ensure input protocols are valid for p in conf.protocols: if p notin ProtocolsTable: error "invalid protocol", protocol = p, valid = ProtocolsTable raise newException(ConfigurationError, "Invalid cli flag values" & p) info "Cli flags", address = conf.address, timeout = conf.timeout, protocols = conf.protocols, logLevel = conf.logLevel let peerRes = parsePeerInfo(conf.address) if peerRes.isErr(): error "Couldn't parse 'conf.address'", error = peerRes.error return 1 let peer = peerRes.value let nodeKey = crypto.PrivateKey.random(Secp256k1, rng[])[] bindIp = ValidIpAddress.init("0.0.0.0") wsBindPort = Port(conf.nodePort + WebSocketPortOffset) nodeTcpPort = Port(conf.nodePort) isWs = peer.addrs[0].contains(multiCodec("ws")).get() isWss = peer.addrs[0].contains(multiCodec("wss")).get() var builder = WakuNodeBuilder.init() builder.withNodeKey(nodeKey) let netConfig = NetConfig.init( bindIp = bindIp, bindPort = nodeTcpPort, wsBindPort = wsBindPort, wsEnabled = isWs, wssEnabled = isWss, ) if isWss and (conf.websocketSecureKeyPath.len == 0 or conf.websocketSecureCertPath.len == 0): error "WebSocket Secure requires key and certificate, see --help" return 1 builder.withNetworkConfiguration(netConfig.tryGet()) builder.withSwitchConfiguration( secureKey = some(conf.websocketSecureKeyPath), secureCert = some(conf.websocketSecureCertPath), nameResolver = resolver, ) let node = builder.build().tryGet() await node.start() let timedOut = not await node.connectToNodes(@[peer]).withTimeout(conf.timeout) if timedOut: error "Timedout after", timeout = conf.timeout let lp2pPeerStore = node.switch.peerStore let conStatus = node.peerManager.peerStore[ConnectionBook][peer.peerId] if conStatus in [Connected, CanConnect]: let nodeProtocols = lp2pPeerStore[ProtoBook][peer.peerId] if not areProtocolsSupported(conf.protocols, nodeProtocols): error "Not all protocols are supported", expected = conf.protocols, supported = nodeProtocols return 1 elif conStatus == CannotConnect: error "Could not connect", peerId = peer.peerId return 1 return 0 when isMainModule: let rng = crypto.newRng() let status = waitFor main(rng) if status == 0: info "The node is reachable and supports all specified protocols" else: error "The node has some problems (see logs)" quit status