feat(perf): add nim-libp2p v1.1

perf/impl/Makefile

@@ -3,6 +3,7 @@ RUST_SUBDIRS := $(wildcard rust-libp2p/*/.)
 RUST_QUINN_SUBDIRS := $(wildcard rust-libp2p-quinn/*/.)
 HTTPS_SUBDIRS := $(wildcard https/*/.)
 QUIC_GO_SUBDIRS := $(wildcard quic-go/*/.)
+NIM_SUBDIRS := $(wildcard nim-libp2p/*/.)
 
 all: $(RUST_SUBDIRS) $(RUST_QUINN_SUBDIRS) $(GO_SUBDIRS) $(HTTPS_SUBDIRS) $(QUIC_GO_SUBDIRS)
 $(RUST_SUBDIRS):
@@ -15,10 +16,12 @@ $(HTTPS_SUBDIRS):
 	$(MAKE) -C $@
 $(QUIC_GO_SUBDIRS):
 	$(MAKE) -C $@
+$(NIM_SUBDIRS):
+	$(MAKE) -C $@
 
-clean: $(RUST_SUBDIRS:%=%clean) $(RUST_QUINN_SUBDIRS:%=%clean) $(GO_SUBDIRS:%=%clean) $(HTTPS_SUBDIRS:%=%clean) $(QUIC_GO_SUBDIRS:%=%clean)
+clean: $(RUST_SUBDIRS:%=%clean) $(RUST_QUINN_SUBDIRS:%=%clean) $(GO_SUBDIRS:%=%clean) $(HTTPS_SUBDIRS:%=%clean) $(QUIC_GO_SUBDIRS:%=%clean) $(NIM_SUBDIRS:%=%clean)
 
 %clean:
 	$(MAKE) -C $* clean
 
-.PHONY: $(RUST_SUBDIRS) $(RUST_QUINN_SUBDIRS) $(GO_SUBDIRS) $(HTTPS_SUBDIRS) $(QUIC_GO_SUBDIRS) all clean
+.PHONY: $(RUST_SUBDIRS) $(RUST_QUINN_SUBDIRS) $(GO_SUBDIRS) $(HTTPS_SUBDIRS) $(QUIC_GO_SUBDIRS) $(QUIC_GO_SUBDIRS) $(NIM_SUBDIRS) all clean

perf/impl/nim-libp2p/v1.0/Makefile

@@ -0,0 +1,23 @@
+commitSha := e03547ea3e2f0372c540c586e993803299d3c4b6
+
+all: perf
+
+perf: perf.nim nim-libp2p
+	docker run --rm --user "$(shell id -u):$(shell id -g)" -v "$(shell pwd)":/usr/src/myapp -w /usr/src/myapp nimlang/nim:1.6.10 sh -c "cd nim-libp2p && nimble install_pinned && cd - && nim c --NimblePath:nim-libp2p/nimbledeps/pkgs -p:nim-libp2p -d:chronicles_log_level=WARN --threads:off -d:release perf.nim"
+
+nim-libp2p: nim-libp2p-${commitSha}
+	rm -rf nim-libp2p
+	ln -s nim-libp2p-${commitSha} nim-libp2p
+
+nim-libp2p-${commitSha}: nim-libp2p-${commitSha}.zip
+	unzip -o nim-libp2p-${commitSha}.zip
+
+nim-libp2p-${commitSha}.zip:
+	wget -O $@ "https://github.com/status-im/nim-libp2p/archive/${commitSha}.zip"
+
+clean:
+	rm -f perf
+	rm -f nim-libp2p
+	rm -rf nim-libp2p-*
+
+.PHONY: all clean

perf/impl/nim-libp2p/v1.0/perf.nim

@@ -0,0 +1,87 @@
+import os, strutils, strformat
+import chronos, bearssl/rand, bearssl/hash
+import ./nim-libp2p/libp2p,
+       ./nim-libp2p/libp2p/protocols/perf/client,
+       ./nim-libp2p/libp2p/protocols/perf/server,
+       ./nim-libp2p/libp2p/protocols/perf/core
+
+const fixedPeerId = "12D3KooWPnQpbXGqzgESFrkaFh1xvCrB64ADnLQQRYfMhnbSuFHF"
+
+type
+  Flags = object
+    runServer: bool
+    serverIpAddress: TransportAddress
+    transport: string
+    uploadBytes: uint
+    downloadBytes: uint
+
+proc seededRng(): ref HmacDrbgContext =
+  var seed: cint = 0
+  var rng = (ref HmacDrbgContext)()
+  hmacDrbgInit(rng[], addr sha256Vtable, cast[pointer](addr seed), sizeof(seed).uint)
+  return rng
+
+proc runServer(address: TransportAddress) {.async.} =
+  let endlessFut = newFuture[void]()
+  var switch = SwitchBuilder.new()
+    .withRng(seededRng())
+    .withAddresses(@[ MultiAddress.init(address).tryGet() ])
+    .withTcpTransport()
+    # .withQuicTransport() TODO: Remove comment when quic transport is done
+    .withMplex()
+    .withNoise()
+    .build()
+  switch.mount(Perf.new())
+  await switch.start()
+  await endlessFut # Await forever, exit on interrupt
+
+proc runClient(f: Flags) {.async.} =
+  let switchBuilder = SwitchBuilder.new()
+    .withRng(newRng())
+    .withAddress(MultiAddress.init("/ip4/127.0.0.1/tcp/0").tryGet())
+    .withMplex()
+    .withNoise()
+  let switch =
+    case f.transport:
+    of "tcp": switchBuilder.withTcpTransport().build()
+    # TODO: Remove comment when quic transport is done
+    # of "quic": switchBuilder.withQuicTransport().build()
+    else: raise (ref Defect)()
+  await switch.start()
+  let start = Moment.now()
+  let conn = await switch.dial(PeerId.init(fixedPeerId).tryGet(),
+                               @[ MultiAddress.init(f.serverIpAddress).tryGet() ],
+                               PerfCodec)
+  var dur = Moment.now() - start
+  dur = dur + (await PerfClient.perf(conn, f.uploadBytes, f.downloadBytes))
+  let ns = dur.nanos
+  let s = Second.nanos
+  echo "{\"latency\": ", fmt"{ns div s}.{ns mod s:09}", "}"
+
+proc main() {.async.} =
+  var i = 1
+  var flags = Flags(transport: "tcp")
+  while i < paramCount():
+    case paramStr(i)
+    of "--run-server": flags.runServer = true
+    of "--server-ip-address":
+      flags.serverIpAddress = initTAddress(paramStr(i + 1))
+      i += 1
+    of "--transport":
+      flags.transport = paramStr(i + 1)
+      i += 1
+    of "--upload-bytes":
+      flags.uploadBytes = parseUInt(paramStr(i + 1))
+      i += 1
+    of "--download-bytes":
+      flags.downloadBytes = parseUInt(paramStr(i + 1))
+      i += 1
+    else: discard
+    i += 1
+
+  if flags.runServer:
+    await runServer(flags.serverIpAddress)
+  else:
+    await runClient(flags)
+
+waitFor(main())