From 1d9ba39660b0672bb580ca979f6e66dc2b87488a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?V=C3=A1clav=20Pavl=C3=ADn?= <vaclav.pavlin@gmail.com>
Date: Mon, 17 Jun 2024 11:18:25 +0200
Subject: [PATCH] spammer is publishing messages at given rats

---
 apps/wakunode2/spammer.nim       | 106 ++++++++++++++++++++-----------
 apps/wakunode2/wakunode2.nim     |   6 +-
 waku/factory/external_config.nim |  18 ++++++
 3 files changed, 87 insertions(+), 43 deletions(-)

diff --git a/apps/wakunode2/spammer.nim b/apps/wakunode2/spammer.nim
index 70e52b9cc..3b12d2383 100644
--- a/apps/wakunode2/spammer.nim
+++ b/apps/wakunode2/spammer.nim
@@ -6,54 +6,84 @@ else:
 import
   chronos,
   chronicles,
-  confutils,
-  confutils/defs,
-  confutils/std/net,
-  confutils/toml/defs as confTomlDefs,
-  confutils/toml/std/net as confTomlNet,
   stew/[byteutils, results],
   std/times,
-  libp2p/protocols/pubsub/gossipsub
+  libp2p/protocols/pubsub/gossipsub,
+  strutils
 
 import
   ../../waku/factory/waku,
+  ../../waku/factory/external_config,
   ../../waku/waku_core,
+  ../../waku/waku_relay,
   ../../waku/node/waku_node,
-  ../../waku/common/confutils/envvar/defs as confEnvvarDefs,
-  ../../waku/common/confutils/envvar/std/net as confEnvvarNet
+  ../../waku/node/peer_manager/peer_manager,
+  ../../waku/waku_rln_relay/rln_relay,
+  ../../tests/waku_rln_relay/rln/waku_rln_relay_utils
 
-type SpammerConfig* = object
-  enable* {.desc: "Enable spammer", defaultValue: false, name: "spammer".}: bool
-  msgRate* {.
-    desc: "Number of messages published per second",
-    defaultValue: 10,
-    name: "spammer-msg-rate"
-  .}: uint64
+proc send(
+    waku: Waku, contentTopic: ContentTopic
+): Future[Result[void, string]] {.async.} =
+  var ephemeral = true
+
+  var message = WakuMessage(
+    payload: toBytes("Hello World!" & intToStr(int(getTime().toUnix()))),
+    contentTopic: contentTopic,
+    #      meta: metaBytes,
+    version: 2,
+    timestamp: getNanosecondTime(getTime().toUnixFloat()),
+    ephemeral: ephemeral,
+  )
+
+  let rlnRes =
+    waku.node.wakuRlnRelay.unsafeAppendRLNProof(message, float64(getTime().toUnix()))
+  if rlnRes.isOk:
+    let pubRes = await waku.node.publish(none(PubsubTopic), message)
+    if pubRes.isErr():
+      error "failed to publish", msg = pubRes.error
+      return err(pubRes.error)
+  else:
+    error "failed to append RLNProof", err = rlnRes.error
+    return err(rlnRes.error)
+
+  return ok()
 
 proc runSpammer*(
-    waku: Waku, contentTopic: ContentTopic = "/spammer/0/test/plain"
+    waku: Waku, conf: WakuNodeConf, contentTopic: ContentTopic = "/spammer/0/test/plain"
 ) {.async.} =
-  try:
-    var conf = SpammerConfig.load()
+  if not conf.enable:
+    return
 
-    if not conf.enable:
-      return
-    var ephemeral = true
-    while true:
-      var message = WakuMessage(
-        payload: toBytes("Hello World!"),
-        contentTopic: contentTopic,
-        #      meta: metaBytes,
-        version: 2,
-        timestamp: getNanosecondTime(getTime().toUnixFloat()),
-        ephemeral: ephemeral,
-      )
-
-      let pubRes = await waku.node.publish(none(PubsubTopic), message)
-      if pubRes.isErr():
-        error "failed to publish", msg = pubRes.error
-      #echo await (waku.node.isReady())
-      await sleepAsync(80)
-  except CatchableError:
-    error "Failed to load config", err = err(getCurrentExceptionMsg())
+  if not conf.rlnRelay:
+    error "RLN not configured!"
     quit(QuitFailure)
+
+  var gotPeers = false
+  while not gotPeers:
+    var (inRelayPeers, outRelayPeers) =
+      waku.node.peerManager.connectedPeers(WakuRelayCodec)
+
+    var numPeers = len(inRelayPeers) + len(outRelayPeers)
+    if numPeers > 0:
+      gotPeers = true
+    info "Waiting for peers", numPeers = numPeers
+    await sleepAsync(1000)
+  #var rate = int(float(1000) / float(conf.msgRate))
+  #var delayBetweenMsg =
+  #  float(conf.rlnEpochSizeSec * 1000) /
+  #  (float(conf.rlnRelayUserMessageLimit) * conf.msgRateMultiplier)
+
+  info "Sending message with delay", delay = conf.delay
+
+  while true:
+    var start = getTime().toUnix()
+
+    (await send(waku, contentTopic)).isOkOr:
+      error "Failed to publish", err = error
+
+    #echo await (waku.node.isReady())
+    var current = getTime().toUnix()
+    var tillNextMsg = int(int64(conf.delay) - (current - start))
+    info "Published messages", sleep = tillNextMsg
+
+    await sleepAsync(tillNextMsg)
diff --git a/apps/wakunode2/wakunode2.nim b/apps/wakunode2/wakunode2.nim
index 59effec13..a4cfafd75 100644
--- a/apps/wakunode2/wakunode2.nim
+++ b/apps/wakunode2/wakunode2.nim
@@ -133,10 +133,6 @@ when isMainModule:
 
     info "Node setup complete"
 
-    if not conf.rlnRelay:
-      error "RLN not configured!"
-      quit(QuitFailure)
-
-    asyncSpawn runSpammer(waku)
+    asyncSpawn runSpammer(waku, conf)
 
     runForever()
diff --git a/waku/factory/external_config.nim b/waku/factory/external_config.nim
index bc3c08bb1..d2b201ed6 100644
--- a/waku/factory/external_config.nim
+++ b/waku/factory/external_config.nim
@@ -595,6 +595,24 @@ type WakuNodeConf* = object
       name: "request-rate-period"
     .}: int64
 
+    enable* {.desc: "Enable spammer", defaultValue: false, name: "spammer".}: bool
+    # msgRate* {.
+    #   desc: "Number of messages published per epoch",
+    #   defaultValue: 10,
+    #   name: "spammer-msg-rate"
+    # .}: int
+
+    # msgRateMultiplier* {.
+    #   desc: "Multiply max message per epoch to publish spam",
+    #   defaultValue: 1,
+    #   name: "spammer-msg-multiplier"
+    # .}: float
+    delay* {.
+      desc: "Delay between spawning a publish method (in miliseconds)",
+      defaultValue: 0,
+      name: "spammer-delay-between-msg"
+    .}: int
+
 ## Parsing
 
 # NOTE: Keys are different in nim-libp2p