mirror of
https://github.com/codex-storage/nim-libp2p.git
synced 2025-01-12 12:04:08 +00:00
5819c6a9a7
* mcache fixes * remove timed cache - the window shifting already removes old messages * ref -> object * avoid unnecessary allocations with `[]` operator * simplify init * fix several gossipsub/floodsub issues * floodsub, gossipsub: don't rebroadcast messages that fail validation (!) * floodsub, gossipsub: don't crash when unsubscribing from unknown topics (!) * gossipsub: don't send message to peers that are not interested in the topic, when messages don't share topic list * floodsub: don't repeat all messages for each message when rebroadcasting * floodsub: allow sending empty data * floodsub: fix inefficient unsubscribe * sync floodsub/gossipsub logging * gossipsub: include incoming messages in mcache (!) * gossipsub: don't rebroadcast already-seen messages (!) * pubsubpeer: remove incoming/outgoing seen caches - these are already handled in gossipsub, floodsub and will cause trouble when peers try to resubscribe / regraft topics (because control messages will have same digest) * timedcache: reimplement without timers (fixes timer leaks and extreme inefficiency due to per-message closures, futures etc) * timedcache: ref -> obj
101 lines
3.0 KiB
Nim
101 lines
3.0 KiB
Nim
{.used.}
|
|
|
|
import unittest, options, sets, sequtils, bearssl
|
|
import stew/byteutils
|
|
import ../../libp2p/[peerid,
|
|
crypto/crypto,
|
|
protocols/pubsub/mcache,
|
|
protocols/pubsub/rpc/message,
|
|
protocols/pubsub/rpc/messages]
|
|
|
|
var rng = newRng()
|
|
|
|
proc randomPeerID(): PeerID =
|
|
PeerID.init(PrivateKey.random(ECDSA, rng[]).get()).get()
|
|
|
|
suite "MCache":
|
|
test "put/get":
|
|
var mCache = MCache.init(3, 5)
|
|
var msg = Message(fromPeer: randomPeerID(), seqno: "12345".toBytes())
|
|
let msgId = defaultMsgIdProvider(msg)
|
|
mCache.put(msgId, msg)
|
|
check mCache.get(msgId).isSome and mCache.get(msgId).get() == msg
|
|
|
|
test "window":
|
|
var mCache = MCache.init(3, 5)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: randomPeerID(),
|
|
seqno: "12345".toBytes(),
|
|
topicIDs: @["foo"])
|
|
mCache.put(defaultMsgIdProvider(msg), msg)
|
|
|
|
for i in 0..<5:
|
|
var msg = Message(fromPeer: randomPeerID(),
|
|
seqno: "12345".toBytes(),
|
|
topicIDs: @["bar"])
|
|
mCache.put(defaultMsgIdProvider(msg), msg)
|
|
|
|
var mids = mCache.window("foo")
|
|
check mids.len == 3
|
|
|
|
var id = toSeq(mids)[0]
|
|
check mCache.get(id).get().topicIDs[0] == "foo"
|
|
|
|
test "shift - shift 1 window at a time":
|
|
var mCache = MCache.init(1, 5)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: randomPeerID(),
|
|
seqno: "12345".toBytes(),
|
|
topicIDs: @["foo"])
|
|
mCache.put(defaultMsgIdProvider(msg), msg)
|
|
|
|
mCache.shift()
|
|
check mCache.window("foo").len == 0
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: randomPeerID(),
|
|
seqno: "12345".toBytes(),
|
|
topicIDs: @["bar"])
|
|
mCache.put(defaultMsgIdProvider(msg), msg)
|
|
|
|
mCache.shift()
|
|
check mCache.window("bar").len == 0
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: randomPeerID(),
|
|
seqno: "12345".toBytes(),
|
|
topicIDs: @["baz"])
|
|
mCache.put(defaultMsgIdProvider(msg), msg)
|
|
|
|
mCache.shift()
|
|
check mCache.window("baz").len == 0
|
|
|
|
test "shift - 2 windows at a time":
|
|
var mCache = MCache.init(1, 5)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: randomPeerID(),
|
|
seqno: "12345".toBytes(),
|
|
topicIDs: @["foo"])
|
|
mCache.put(defaultMsgIdProvider(msg), msg)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: randomPeerID(),
|
|
seqno: "12345".toBytes(),
|
|
topicIDs: @["bar"])
|
|
mCache.put(defaultMsgIdProvider(msg), msg)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: randomPeerID(),
|
|
seqno: "12345".toBytes(),
|
|
topicIDs: @["baz"])
|
|
mCache.put(defaultMsgIdProvider(msg), msg)
|
|
|
|
mCache.shift()
|
|
check mCache.window("foo").len == 0
|
|
|
|
mCache.shift()
|
|
check mCache.window("bar").len == 0
|