mirror of
https://github.com/logos-messaging/logos-delivery.git
synced 2026-08-04 02:43:36 +00:00
6837ae0c1f
* bump nim-libp2p pin to v2.0.0 tag * bump json_rpc to v0.6.1, lsquic to v0.5.1, boringssl to v0.0.8 (latest tags) * add libp2p_mix dep; repoint libp2p/protocols/mix -> libp2p_mix * pin nimble.lock: websock / protobuf_serialization / npeg / jwt * Makefile: add -d:libp2p_quic_support * regenerate nix/deps.nix (adds libp2p_mix, refreshes pins) * migrate rng ref HmacDrbgContext -> libp2p Rng across prod/channels/tests (interface-only; same DRBG) * waku_switch: TransportConfig factory; unified 2.0.0 connection limits (withMaxInOut, withMaxConnections); local MaxConnections * waku_relay/rendezvous/discv5/kademlia: v2.0.0 API (rng, config, ServiceDiscovery rename) * call Service.setup() on post-build switch services (2.0.0 split setup/start) * drop libp2p/utils/semaphore -> chronos AsyncSemaphore * add logos_delivery/waku/compat/option_valueor shim (Option[T] valueOr/withValue, dropped upstream) * add std/options where a transitive re-export was removed * add newStandardSwitch shim (libp2p removed it in 2.0.0); mounts yamux+mplex to match prod muxer * PeerId.random(rng); common.rng()/crypto.newRng(); hoist shared rng (instantiation cleanup) * update expectations for 2.0.0 defaults: DEFAULT_PROTOCOLS += /ipfs/id/push/1.0.0; agent "nim-libp2p" * drop relay reboot/reconnect test (asserted a Switch restart capability that is simply not supported) * fix up a few tests that were flaking on MacOS (libp2p upgrade may have exposed these)
86 lines
2.7 KiB
Nim
86 lines
2.7 KiB
Nim
{.used.}
|
|
|
|
import
|
|
std/[sequtils, tempfiles, options],
|
|
stew/byteutils,
|
|
chronos,
|
|
chronicles,
|
|
libp2p/switch,
|
|
libp2p/protocols/pubsub/pubsub
|
|
|
|
from std/times import epochTime
|
|
|
|
import
|
|
../../../logos_delivery/waku/
|
|
[node/waku_node, node/peer_manager, waku_core, waku_node, waku_rln_relay],
|
|
../waku_store/store_utils,
|
|
../waku_archive/archive_utils,
|
|
../testlib/[wakucore, futures, assertions]
|
|
|
|
proc setupStaticRln*(
|
|
node: WakuNode,
|
|
identifier: uint,
|
|
rlnRelayEthContractAddress: Option[string] = none(string),
|
|
) {.async.} =
|
|
await node.mountRlnRelay(
|
|
WakuRlnConfig(dynamic: false, credIndex: some(identifier), epochSizeSec: 1)
|
|
)
|
|
|
|
proc setupRelayWithStaticRln*(
|
|
node: WakuNode, identifier: uint, shards: seq[RelayShard]
|
|
) {.async.} =
|
|
await node.mountRelay(shards)
|
|
await setupStaticRln(node, identifier)
|
|
|
|
proc subscribeCompletionHandler*(node: WakuNode, pubsubTopic: string): Future[bool] =
|
|
var completionFut = newFuture[bool]()
|
|
proc relayHandler(
|
|
topic: PubsubTopic, msg: WakuMessage
|
|
): Future[void] {.async, gcsafe.} =
|
|
if topic == pubsubTopic:
|
|
completionFut.complete(true)
|
|
|
|
node.subscribe((kind: PubsubSub, topic: pubsubTopic), some(relayHandler)).isOkOr:
|
|
error "failed to subscribe to relay", topic = pubsubTopic, error = error
|
|
completionFut.complete(false)
|
|
|
|
return completionFut
|
|
|
|
proc sendRlnMessage*(
|
|
client: WakuNode,
|
|
pubsubTopic: string,
|
|
contentTopic: string,
|
|
completionFuture: Future[bool],
|
|
payload: seq[byte] = "Hello".toBytes(),
|
|
): Future[bool] {.async.} =
|
|
var message = WakuMessage(payload: payload, contentTopic: contentTopic)
|
|
message.proof = (
|
|
await client.wakuRlnRelay.generateRLNProof(message.toRLNSignal(), epochTime())
|
|
).valueOr:
|
|
raiseAssert "generateRLNProof failed: " & error
|
|
discard await client.publish(some(pubsubTopic), message)
|
|
let isCompleted = await completionFuture.withTimeout(FUTURE_TIMEOUT)
|
|
return isCompleted
|
|
|
|
proc sendRlnMessageWithInvalidProof*(
|
|
client: WakuNode,
|
|
pubsubTopic: string,
|
|
contentTopic: string,
|
|
completionFuture: Future[bool],
|
|
payload: seq[byte] = "Hello".toBytes(),
|
|
): Future[bool] {.async.} =
|
|
let extraBytes: seq[byte] = @[byte(1), 2, 3]
|
|
let rateLimitProofRes = await client.wakuRlnRelay.groupManager.generateProof(
|
|
concat(payload, extraBytes),
|
|
# we add extra bytes to invalidate proof verification against original payload
|
|
client.wakuRlnRelay.getCurrentEpoch(),
|
|
)
|
|
let
|
|
rateLimitProof = rateLimitProofRes.get().encode().buffer
|
|
message =
|
|
WakuMessage(payload: @payload, contentTopic: contentTopic, proof: rateLimitProof)
|
|
|
|
discard await client.publish(some(pubsubTopic), message)
|
|
let isCompleted = await completionFuture.withTimeout(FUTURE_TIMEOUT)
|
|
return isCompleted
|