logos-storage/nim-libp2p
1
0
Fork 0
mirror of https://github.com/logos-storage/nim-libp2p.git synced 2026-01-04 06:33:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change newXXXX procs to XXXX.new (#585)

Browse Source 
* newBufferStream -> BufferStream.new

* newMultistream -> MultistreamSelect.new

* newSecio -> Secio.new

* newNoise -> Noise.new

* newPlainText -> PlainText.new

* newPubSubPeer -> PubSubPeer.new

* newIdentify -> Identify.new

* newMuxerProvider -> MuxerProvider.new
This commit is contained in:
Tanguy Cizain 2021-06-07 09:32:08 +02:00 committed by GitHub
parent fbe888a3aa
commit caac8191d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
20 changed files with 203 additions and 144 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/GETTING_STARTED.md
View File

@ -59,7 +59,7 @@ proc createSwitch(ma: MultiAddress): (Switch, PeerInfo) =
let mplexProvider = newMuxerProvider(createMplex, MplexCodec) # create multiplexer let mplexProvider = newMuxerProvider(createMplex, MplexCodec) # create multiplexer
let transports = @[Transport(newTransport(TcpTransport))] # add all transports (tcp only for now, but can be anything in the future) let transports = @[Transport(newTransport(TcpTransport))] # add all transports (tcp only for now, but can be anything in the future)
let muxers = {MplexCodec: mplexProvider}.toTable() # add all muxers let muxers = {MplexCodec: mplexProvider}.toTable() # add all muxers
let secureManagers = {SecioCodec: Secure(newSecio(seckey))}.toTable() # setup the secio and any other secure provider let secureManagers = {SecioCodec: Secure(Secio.new(seckey))}.toTable() # setup the secio and any other secure provider
# create the switch # create the switch
let switch = newSwitch(peerInfo, let switch = newSwitch(peerInfo,

8
libp2p/builders.nim
View File

@ -138,7 +138,7 @@ proc build*(b: SwitchBuilder): Switch
var var
secureManagerInstances: seq[Secure] secureManagerInstances: seq[Secure]
if SecureProtocol.Noise in b.secureManagers: if SecureProtocol.Noise in b.secureManagers:
secureManagerInstances.add(newNoise(b.rng, seckey).Secure) secureManagerInstances.add(Noise.new(b.rng, seckey).Secure)
let let
peerInfo = PeerInfo.init( peerInfo = PeerInfo.init(
@ -151,13 +151,13 @@ proc build*(b: SwitchBuilder): Switch
muxers = block: muxers = block:
var muxers: Table[string, MuxerProvider] var muxers: Table[string, MuxerProvider]
if b.mplexOpts.enable: if b.mplexOpts.enable:
muxers[MplexCodec] = newMuxerProvider(b.mplexOpts.newMuxer, MplexCodec) muxers[MplexCodec] = MuxerProvider.new(b.mplexOpts.newMuxer, MplexCodec)
muxers muxers
let let
identify = newIdentify(peerInfo) identify = Identify.new(peerInfo)
connManager = ConnManager.init(b.maxConnsPerPeer, b.maxConnections, b.maxIn, b.maxOut) connManager = ConnManager.init(b.maxConnsPerPeer, b.maxConnections, b.maxIn, b.maxOut)
ms = newMultistream() ms = MultistreamSelect.new()
muxedUpgrade = MuxedUpgrade.init(identify, muxers, secureManagerInstances, connManager, ms) muxedUpgrade = MuxedUpgrade.init(identify, muxers, secureManagerInstances, connManager, ms)
let let

8
libp2p/multistream.nim
View File

@ -40,9 +40,11 @@ type
handlers*: seq[HandlerHolder] handlers*: seq[HandlerHolder]
codec*: string codec*: string
proc newMultistream*(): MultistreamSelect = proc new*(T: typedesc[MultistreamSelect]): T =
new result T(codec: MSCodec)
result.codec = MSCodec
proc newMultistream*(): MultistreamSelect {.deprecated: "use MultistreamSelect.new".} =
MultistreamSelect.new()
template validateSuffix(str: string): untyped = template validateSuffix(str: string): untyped =
if str.endsWith("\n"): if str.endsWith("\n"):

17
libp2p/muxers/muxer.nim
View File

@ -48,11 +48,18 @@ method newStream*(m: Muxer, name: string = "", lazy: bool = false):
method close*(m: Muxer) {.base, async, gcsafe.} = discard method close*(m: Muxer) {.base, async, gcsafe.} = discard
method handle*(m: Muxer): Future[void] {.base, async, gcsafe.} = discard method handle*(m: Muxer): Future[void] {.base, async, gcsafe.} = discard
proc newMuxerProvider*(creator: MuxerConstructor, codec: string): MuxerProvider {.gcsafe.} = proc new*(
new result T: typedesc[MuxerProvider],
result.newMuxer = creator creator: MuxerConstructor,
result.codec = codec codec: string): T {.gcsafe.} =
result.init()
let muxerProvider = T(newMuxer: creator)
muxerProvider.codec = codec
muxerProvider.init()
muxerProvider
proc newMuxerProvider*(creator: MuxerConstructor, codec: string): MuxerProvider {.gcsafe, deprecated: "use MuxerProvider.new".} =
MuxerProvider.new(creator, codec)
method init(c: MuxerProvider) = method init(c: MuxerProvider) =
proc handler(conn: Connection, proto: string) {.async, gcsafe, closure.} = proc handler(conn: Connection, proto: string) {.async, gcsafe, closure.} =

11
libp2p/protocols/identify.nim
View File

@ -109,10 +109,13 @@ proc decodeMsg*(buf: seq[byte]): Option[IdentifyInfo] =
trace "decodeMsg: failed to decode received message" trace "decodeMsg: failed to decode received message"
none[IdentifyInfo]() none[IdentifyInfo]()
proc newIdentify*(peerInfo: PeerInfo): Identify = proc new*(T: typedesc[Identify], peerInfo: PeerInfo): T =
new result let identify = T(peerInfo: peerInfo)
result.peerInfo = peerInfo identify.init()
result.init() identify
proc newIdentify*(peerInfo: PeerInfo): Identify {.deprecated: "use Identify.new".} =
Identify.new(peerInfo)
method init*(p: Identify) = method init*(p: Identify) =
proc handle(conn: Connection, proto: string) {.async, gcsafe, closure.} = proc handle(conn: Connection, proto: string) {.async, gcsafe, closure.} =

2
libp2p/protocols/pubsub/pubsub.nim
View File

@ -283,7 +283,7 @@ proc getOrCreatePeer*(
p.onPubSubPeerEvent(peer, event) p.onPubSubPeerEvent(peer, event)
# create new pubsub peer # create new pubsub peer
let pubSubPeer = newPubSubPeer(peerId, getConn, dropConn, onEvent, protos[0]) let pubSubPeer = PubSubPeer.new(peerId, getConn, dropConn, onEvent, protos[0])
debug "created new pubsub peer", peerId debug "created new pubsub peer", peerId
p.peers[peerId] = pubSubPeer p.peers[peerId] = pubSubPeer

30
libp2p/protocols/pubsub/pubsubpeer.nim
View File

@ -274,15 +274,33 @@ proc send*(p: PubSubPeer, msg: RPCMsg, anonymize: bool) {.raises: [Defect].} =
p.sendEncoded(encoded) p.sendEncoded(encoded)
proc newPubSubPeer*(peerId: PeerID, proc new*(
getConn: GetConn, T: typedesc[PubSubPeer],
dropConn: DropConn, peerId: PeerID,
onEvent: OnEvent, getConn: GetConn,
codec: string): PubSubPeer = dropConn: DropConn,
PubSubPeer( onEvent: OnEvent,
codec: string): T =
T(
getConn: getConn, getConn: getConn,
dropConn: dropConn, dropConn: dropConn,
onEvent: onEvent, onEvent: onEvent,
codec: codec, codec: codec,
peerId: peerId, peerId: peerId,
) )
proc newPubSubPeer*(
peerId: PeerID,
getConn: GetConn,
dropConn: DropConn,
onEvent: OnEvent,
codec: string): PubSubPeer {.deprecated: "use PubSubPeer.new".} =
PubSubPeer.new(
peerId,
getConn,
dropConn,
onEvent,
codec
)

20
libp2p/protocols/secure/noise.nim
View File

@ -595,11 +595,12 @@ method init*(p: Noise) {.gcsafe.} =
procCall Secure(p).init() procCall Secure(p).init()
p.codec = NoiseCodec p.codec = NoiseCodec
proc newNoise*( proc new*(
rng: ref BrHmacDrbgContext, T: typedesc[Noise],
privateKey: PrivateKey, rng: ref BrHmacDrbgContext,
outgoing: bool = true, privateKey: PrivateKey,
commonPrologue: seq[byte] = @[]): Noise = outgoing: bool = true,
commonPrologue: seq[byte] = @[]): T =
let pkBytes = privateKey let pkBytes = privateKey
.getKey() .getKey()
@ -617,4 +618,11 @@ proc newNoise*(
) )
noise.init() noise.init()
return noise noise
proc newNoise*(
rng: ref BrHmacDrbgContext,
privateKey: PrivateKey,
outgoing: bool = true,
commonPrologue: seq[byte] = @[]): Noise {.deprecated: "use Noise.new".}=
Noise.new(rng, privateKey, outgoing, commonPrologue)

10
libp2p/protocols/secure/plaintext.nim
View File

@ -25,6 +25,10 @@ method init(p: PlainText) {.gcsafe.} =
p.codec = PlainTextCodec p.codec = PlainTextCodec
p.handler = handle p.handler = handle
proc newPlainText*(): PlainText = proc new*(T: typedesc[PlainText]): T =
new result let plainText = T()
result.init() plainText.init()
plainText
proc newPlainText*(): PlainText {.deprecated: "use PlainText.new".} =
PlainText.new()

15
libp2p/protocols/secure/secio.nim
View File

@ -36,7 +36,7 @@ const
SecioHashes = "SHA256,SHA512" SecioHashes = "SHA256,SHA512"
type type
Secio = ref object of Secure Secio* = ref object of Secure
rng: ref BrHmacDrbgContext rng: ref BrHmacDrbgContext
localPrivateKey: PrivateKey localPrivateKey: PrivateKey
localPublicKey: PublicKey localPublicKey: PublicKey
@ -431,16 +431,23 @@ method init(s: Secio) {.gcsafe.} =
procCall Secure(s).init() procCall Secure(s).init()
s.codec = SecioCodec s.codec = SecioCodec
proc newSecio*(rng: ref BrHmacDrbgContext, localPrivateKey: PrivateKey): Secio = proc new*(
T: typedesc[Secio],
rng: ref BrHmacDrbgContext,
localPrivateKey: PrivateKey): T =
let pkRes = localPrivateKey.getKey() let pkRes = localPrivateKey.getKey()
if pkRes.isErr: if pkRes.isErr:
raise newException(Defect, "Can't fetch local private key") raise newException(Defect, "Can't fetch local private key")
result = Secio( let secio = Secio(
rng: rng, rng: rng,
localPrivateKey: localPrivateKey, localPrivateKey: localPrivateKey,
localPublicKey: localPrivateKey localPublicKey: localPrivateKey
.getKey() .getKey()
.expect("Can't fetch local private key"), .expect("Can't fetch local private key"),
) )
result.init() secio.init()
secio
proc newSecio*(rng: ref BrHmacDrbgContext, localPrivateKey: PrivateKey): Secio {.deprecated: "use Secio.new".} =
Secio.new(rng, localPrivateKey)

15
libp2p/stream/bufferstream.nim
View File

@ -58,10 +58,17 @@ method initStream*(s: BufferStream) =
trace "BufferStream created", s trace "BufferStream created", s
proc newBufferStream*(timeout: Duration = DefaultConnectionTimeout): BufferStream = proc new*(
new result T: typedesc[BufferStream],
result.timeout = timeout timeout: Duration = DefaultConnectionTimeout): T =
result.initStream()
let bufferStream = T(timeout: timeout)
bufferStream.initStream()
bufferStream
proc newBufferStream*(
timeout: Duration = DefaultConnectionTimeout): BufferStream {.deprecated: "use BufferStream.new".} =
return BufferStream.new(timeout)
method pushData*(s: BufferStream, data: seq[byte]) {.base, async.} = method pushData*(s: BufferStream, data: seq[byte]) {.base, async.} =
## Write bytes to internal read buffer, use this to fill up the ## Write bytes to internal read buffer, use this to fill up the

2
libp2p/upgrademngrs/muxedupgrade.nim
View File

@ -109,7 +109,7 @@ method upgradeIncoming*(
self: MuxedUpgrade, self: MuxedUpgrade,
incomingConn: Connection) {.async, gcsafe.} = # noraises incomingConn: Connection) {.async, gcsafe.} = # noraises
trace "Upgrading incoming connection", incomingConn trace "Upgrading incoming connection", incomingConn
let ms = newMultistream() let ms = MultistreamSelect.new()
# secure incoming connections # secure incoming connections
proc securedHandler(conn: Connection, proc securedHandler(conn: Connection,

11
tests/helpers.nim
View File

@ -74,10 +74,13 @@ type
method write*(s: TestBufferStream, msg: seq[byte]): Future[void] = method write*(s: TestBufferStream, msg: seq[byte]): Future[void] =
s.writeHandler(msg) s.writeHandler(msg)
proc newBufferStream*(writeHandler: WriteHandler): TestBufferStream = proc new*(T: typedesc[TestBufferStream], writeHandler: WriteHandler): T =
new result let testBufferStream = T(writeHandler: writeHandler)
result.writeHandler = writeHandler testBufferStream.initStream()
result.initStream() testBufferStream
proc newBufferStream*(writeHandler: WriteHandler): TestBufferStream {.deprecated: "use TestBufferStream.new".}=
TestBufferStream.new(writeHandler)
proc checkExpiringInternal(cond: proc(): bool {.raises: [Defect].} ): Future[bool] {.async, gcsafe.} = proc checkExpiringInternal(cond: proc(): bool {.raises: [Defect].} ): Future[bool] {.async, gcsafe.} =
{.gcsafe.}: {.gcsafe.}:

58
tests/pubsub/testgossipinternal.nim
View File

@ -25,7 +25,7 @@ proc getPubSubPeer(p: TestGossipSub, peerId: PeerID): PubSubPeer =
proc dropConn(peer: PubSubPeer) = proc dropConn(peer: PubSubPeer) =
discard # we don't care about it here yet discard # we don't care about it here yet
let pubSubPeer = newPubSubPeer(peerId, getConn, dropConn, nil, GossipSubCodec) let pubSubPeer = PubSubPeer.new(peerId, getConn, dropConn, nil, GossipSubCodec)
debug "created new pubsub peer", peerId debug "created new pubsub peer", peerId
p.peers[peerId] = pubSubPeer p.peers[peerId] = pubSubPeer
@ -56,7 +56,7 @@ suite "GossipSub internal":
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
for i in 0..<15: for i in 0..<15:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -97,7 +97,7 @@ suite "GossipSub internal":
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
for i in 0..<15: for i in 0..<15:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -123,7 +123,7 @@ suite "GossipSub internal":
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
var scoreLow = -11'f64 var scoreLow = -11'f64
for i in 0..<15: for i in 0..<15:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -153,7 +153,7 @@ suite "GossipSub internal":
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
for i in 0..<15: for i in 0..<15:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get()) let peerInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get())
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -180,7 +180,7 @@ suite "GossipSub internal":
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
for i in 0..<15: for i in 0..<15:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
var peerInfo = randomPeerInfo() var peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -209,7 +209,7 @@ suite "GossipSub internal":
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
for i in 0..<6: for i in 0..<6:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get()) let peerInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get())
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -243,7 +243,7 @@ suite "GossipSub internal":
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
for i in 0..<6: for i in 0..<6:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -277,7 +277,7 @@ suite "GossipSub internal":
# generate mesh and fanout peers # generate mesh and fanout peers
for i in 0..<30: for i in 0..<30:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -291,7 +291,7 @@ suite "GossipSub internal":
# generate gossipsub (free standing) peers # generate gossipsub (free standing) peers
for i in 0..<15: for i in 0..<15:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -302,7 +302,7 @@ suite "GossipSub internal":
# generate messages # generate messages
var seqno = 0'u64 var seqno = 0'u64
for i in 0..5: for i in 0..5:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -335,7 +335,7 @@ suite "GossipSub internal":
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
for i in 0..<30: for i in 0..<30:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -349,7 +349,7 @@ suite "GossipSub internal":
# generate messages # generate messages
var seqno = 0'u64 var seqno = 0'u64
for i in 0..5: for i in 0..5:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -375,7 +375,7 @@ suite "GossipSub internal":
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
for i in 0..<30: for i in 0..<30:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -390,7 +390,7 @@ suite "GossipSub internal":
# generate messages # generate messages
var seqno = 0'u64 var seqno = 0'u64
for i in 0..5: for i in 0..5:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -416,7 +416,7 @@ suite "GossipSub internal":
gossipSub.fanout[topic] = initHashSet[PubSubPeer]() gossipSub.fanout[topic] = initHashSet[PubSubPeer]()
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
for i in 0..<30: for i in 0..<30:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -431,7 +431,7 @@ suite "GossipSub internal":
# generate messages # generate messages
var seqno = 0'u64 var seqno = 0'u64
for i in 0..5: for i in 0..5:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -454,7 +454,7 @@ suite "GossipSub internal":
let topic = "foobar" let topic = "foobar"
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
for i in 0..<30: for i in 0..<30:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -464,7 +464,7 @@ suite "GossipSub internal":
# generate messages # generate messages
var seqno = 0'u64 var seqno = 0'u64
for i in 0..5: for i in 0..5:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -488,7 +488,7 @@ suite "GossipSub internal":
let topic = "foobar" let topic = "foobar"
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
for i in 0..<30: for i in 0..<30:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -521,7 +521,7 @@ suite "GossipSub internal":
tooManyTopics &= "topic" & $i tooManyTopics &= "topic" & $i
let lotOfSubs = RPCMsg.withSubs(tooManyTopics, true) let lotOfSubs = RPCMsg.withSubs(tooManyTopics, true)
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
let peer = gossipSub.getPubSubPeer(peerInfo.peerId) let peer = gossipSub.getPubSubPeer(peerInfo.peerId)
@ -544,7 +544,7 @@ suite "GossipSub internal":
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
for i in 0..<15: for i in 0..<15:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -575,7 +575,7 @@ suite "GossipSub internal":
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
for i in 0..<15: for i in 0..<15:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -617,7 +617,7 @@ suite "GossipSub internal":
var conns = newSeq[Connection]() var conns = newSeq[Connection]()
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
for i in 0..<6: for i in 0..<6:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conn.transportDir = Direction.In conn.transportDir = Direction.In
conns &= conn conns &= conn
let peerInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get()) let peerInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get())
@ -629,7 +629,7 @@ suite "GossipSub internal":
gossipSub.mesh[topic].incl(peer) gossipSub.mesh[topic].incl(peer)
for i in 0..<7: for i in 0..<7:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conn.transportDir = Direction.Out conn.transportDir = Direction.Out
conns &= conn conns &= conn
let peerInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get()) let peerInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get())
@ -665,7 +665,7 @@ suite "GossipSub internal":
gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]() gossipSub.gossipsub[topic] = initHashSet[PubSubPeer]()
gossipSub.mesh[topic] = initHashSet[PubSubPeer]() gossipSub.mesh[topic] = initHashSet[PubSubPeer]()
for i in 0..<30: for i in 0..<30:
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -676,7 +676,7 @@ suite "GossipSub internal":
block: block:
# should ignore no budget peer # should ignore no budget peer
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -692,7 +692,7 @@ suite "GossipSub internal":
block: block:
# given duplicate ihave should generate only one iwant # given duplicate ihave should generate only one iwant
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo
@ -707,7 +707,7 @@ suite "GossipSub internal":
block: block:
# given duplicate iwant should generate only one message # given duplicate iwant should generate only one message
let conn = newBufferStream(noop) let conn = TestBufferStream.new(noop)
conns &= conn conns &= conn
let peerInfo = randomPeerInfo() let peerInfo = randomPeerInfo()
conn.peerInfo = peerInfo conn.peerInfo = peerInfo

26
tests/testbufferstream.nim
View File

@ -13,7 +13,7 @@ suite "BufferStream":
check getTracker(BufferStreamTrackerName).isLeaked() == false check getTracker(BufferStreamTrackerName).isLeaked() == false
asyncTest "push data to buffer": asyncTest "push data to buffer":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
var data = "12345" var data = "12345"
await buff.pushData(data.toBytes()) await buff.pushData(data.toBytes())
@ -21,7 +21,7 @@ suite "BufferStream":
await buff.close() await buff.close()
asyncTest "push and wait": asyncTest "push and wait":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
let fut0 = buff.pushData("1234".toBytes()) let fut0 = buff.pushData("1234".toBytes())
@ -38,7 +38,7 @@ suite "BufferStream":
await buff.close() await buff.close()
asyncTest "read with size": asyncTest "read with size":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
await buff.pushData("12345".toBytes()) await buff.pushData("12345".toBytes())
@ -48,7 +48,7 @@ suite "BufferStream":
await buff.close() await buff.close()
asyncTest "readExactly": asyncTest "readExactly":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
await buff.pushData("12345".toBytes()) await buff.pushData("12345".toBytes())
@ -59,7 +59,7 @@ suite "BufferStream":
await buff.close() await buff.close()
asyncTest "readExactly raises": asyncTest "readExactly raises":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
await buff.pushData("123".toBytes()) await buff.pushData("123".toBytes())
@ -71,7 +71,7 @@ suite "BufferStream":
await readFut await readFut
asyncTest "readOnce": asyncTest "readOnce":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
var data: array[3, byte] var data: array[3, byte]
@ -84,7 +84,7 @@ suite "BufferStream":
await buff.close() await buff.close()
asyncTest "reads should happen in order": asyncTest "reads should happen in order":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
proc writer1() {.async.} = proc writer1() {.async.} =
@ -126,7 +126,7 @@ suite "BufferStream":
await writerFut2 await writerFut2
asyncTest "small reads": asyncTest "small reads":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
var str: string var str: string
@ -152,7 +152,7 @@ suite "BufferStream":
await buff.close() await buff.close()
asyncTest "read all data after eof": asyncTest "read all data after eof":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
await buff.pushData("12345".toBytes()) await buff.pushData("12345".toBytes())
@ -176,7 +176,7 @@ suite "BufferStream":
await buff.close() # all data should still be read after close await buff.close() # all data should still be read after close
asyncTest "read more data after eof": asyncTest "read more data after eof":
let buff = newBufferStream() let buff = BufferStream.new()
check buff.len == 0 check buff.len == 0
await buff.pushData("12345".toBytes()) await buff.pushData("12345".toBytes())
@ -200,7 +200,7 @@ suite "BufferStream":
await buff.close() # all data should still be read after close await buff.close() # all data should still be read after close
asyncTest "shouldn't get stuck on close": asyncTest "shouldn't get stuck on close":
var stream = newBufferStream() var stream = BufferStream.new()
var var
fut = stream.pushData(toBytes("hello")) fut = stream.pushData(toBytes("hello"))
fut2 = stream.pushData(toBytes("again")) fut2 = stream.pushData(toBytes("again"))
@ -214,7 +214,7 @@ suite "BufferStream":
await stream.close() await stream.close()
asyncTest "no push after close": asyncTest "no push after close":
var stream = newBufferStream() var stream = BufferStream.new()
await stream.pushData("123".toBytes()) await stream.pushData("123".toBytes())
var data: array[3, byte] var data: array[3, byte]
await stream.readExactly(addr data[0], data.len) await stream.readExactly(addr data[0], data.len)
@ -224,7 +224,7 @@ suite "BufferStream":
await stream.pushData("123".toBytes()) await stream.pushData("123".toBytes())
asyncTest "no concurrent pushes": asyncTest "no concurrent pushes":
var stream = newBufferStream() var stream = BufferStream.new()
await stream.pushData("123".toBytes()) await stream.pushData("123".toBytes())
let push = stream.pushData("123".toBytes()) let push = stream.pushData("123".toBytes())

6
tests/testconnection.nim
View File

@ -6,13 +6,13 @@ import ./helpers
suite "Connection": suite "Connection":
asyncTest "close": asyncTest "close":
var conn = newBufferStream() var conn = BufferStream.new()
await conn.close() await conn.close()
check: check:
conn.closed == true conn.closed == true
asyncTest "parent close": asyncTest "parent close":
var buf = newBufferStream() var buf = BufferStream.new()
var conn = buf var conn = buf
await conn.close() await conn.close()
@ -21,7 +21,7 @@ suite "Connection":
buf.closed == true buf.closed == true
asyncTest "child close": asyncTest "child close":
var buf = newBufferStream() var buf = BufferStream.new()
var conn = buf var conn = buf
await buf.close() await buf.close()

8
tests/testidentify.nim
View File

@ -42,11 +42,11 @@ suite "Identify":
transport1 = TcpTransport.init(upgrade = Upgrade()) transport1 = TcpTransport.init(upgrade = Upgrade())
transport2 = TcpTransport.init(upgrade = Upgrade()) transport2 = TcpTransport.init(upgrade = Upgrade())
identifyProto1 = newIdentify(remotePeerInfo) identifyProto1 = Identify.new(remotePeerInfo)
identifyProto2 = newIdentify(remotePeerInfo) identifyProto2 = Identify.new(remotePeerInfo)
msListen = newMultistream() msListen = MultistreamSelect.new()
msDial = newMultistream() msDial = MultistreamSelect.new()
asyncTeardown: asyncTeardown:
await conn.close() await conn.close()

48
tests/testmplex.nim
View File

@ -26,7 +26,7 @@ suite "Mplex":
proc encHandler(msg: seq[byte]) {.async.} = proc encHandler(msg: seq[byte]) {.async.} =
check msg == fromHex("000873747265616d2031") check msg == fromHex("000873747265616d2031")
let conn = newBufferStream(encHandler) let conn = TestBufferStream.new(encHandler)
await conn.writeMsg(0, MessageType.New, ("stream 1").toBytes) await conn.writeMsg(0, MessageType.New, ("stream 1").toBytes)
await conn.close() await conn.close()
@ -34,7 +34,7 @@ suite "Mplex":
proc encHandler(msg: seq[byte]) {.async.} = proc encHandler(msg: seq[byte]) {.async.} =
check msg == fromHex("88010873747265616d2031") check msg == fromHex("88010873747265616d2031")
let conn = newBufferStream(encHandler) let conn = TestBufferStream.new(encHandler)
await conn.writeMsg(17, MessageType.New, ("stream 1").toBytes) await conn.writeMsg(17, MessageType.New, ("stream 1").toBytes)
await conn.close() await conn.close()
@ -42,7 +42,7 @@ suite "Mplex":
proc encHandler(msg: seq[byte]) {.async.} = proc encHandler(msg: seq[byte]) {.async.} =
check msg == fromHex("020873747265616d2031") check msg == fromHex("020873747265616d2031")
let conn = newBufferStream(encHandler) let conn = TestBufferStream.new(encHandler)
await conn.writeMsg(0, MessageType.MsgOut, ("stream 1").toBytes) await conn.writeMsg(0, MessageType.MsgOut, ("stream 1").toBytes)
await conn.close() await conn.close()
@ -50,12 +50,12 @@ suite "Mplex":
proc encHandler(msg: seq[byte]) {.async.} = proc encHandler(msg: seq[byte]) {.async.} =
check msg == fromHex("8a010873747265616d2031") check msg == fromHex("8a010873747265616d2031")
let conn = newBufferStream(encHandler) let conn = TestBufferStream.new(encHandler)
await conn.writeMsg(17, MessageType.MsgOut, ("stream 1").toBytes) await conn.writeMsg(17, MessageType.MsgOut, ("stream 1").toBytes)
await conn.close() await conn.close()
asyncTest "decode header with channel id 0": asyncTest "decode header with channel id 0":
let stream = newBufferStream() let stream = BufferStream.new()
let conn = stream let conn = stream
await stream.pushData(fromHex("000873747265616d2031")) await stream.pushData(fromHex("000873747265616d2031"))
let msg = await conn.readMsg() let msg = await conn.readMsg()
@ -65,7 +65,7 @@ suite "Mplex":
await conn.close() await conn.close()
asyncTest "decode header and body with channel id 0": asyncTest "decode header and body with channel id 0":
let stream = newBufferStream() let stream = BufferStream.new()
let conn = stream let conn = stream
await stream.pushData(fromHex("021668656C6C6F2066726F6D206368616E6E656C20302121")) await stream.pushData(fromHex("021668656C6C6F2066726F6D206368616E6E656C20302121"))
let msg = await conn.readMsg() let msg = await conn.readMsg()
@ -76,7 +76,7 @@ suite "Mplex":
await conn.close() await conn.close()
asyncTest "decode header and body with channel id other than 0": asyncTest "decode header and body with channel id other than 0":
let stream = newBufferStream() let stream = BufferStream.new()
let conn = stream let conn = stream
await stream.pushData(fromHex("8a011668656C6C6F2066726F6D206368616E6E656C20302121")) await stream.pushData(fromHex("8a011668656C6C6F2066726F6D206368616E6E656C20302121"))
let msg = await conn.readMsg() let msg = await conn.readMsg()
@ -90,7 +90,7 @@ suite "Mplex":
asyncTest "(local close) - should close for write": asyncTest "(local close) - should close for write":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
await chann.close() await chann.close()
@ -102,7 +102,7 @@ suite "Mplex":
asyncTest "(local close) - should allow reads until remote closes": asyncTest "(local close) - should allow reads until remote closes":
let let
conn = newBufferStream( conn = TestBufferStream.new(
proc (data: seq[byte]) {.gcsafe, async.} = proc (data: seq[byte]) {.gcsafe, async.} =
discard, discard,
) )
@ -129,7 +129,7 @@ suite "Mplex":
asyncTest "(remote close) - channel should close for reading by remote": asyncTest "(remote close) - channel should close for reading by remote":
let let
conn = newBufferStream( conn = TestBufferStream.new(
proc (data: seq[byte]) {.gcsafe, async.} = proc (data: seq[byte]) {.gcsafe, async.} =
discard, discard,
) )
@ -152,7 +152,7 @@ suite "Mplex":
asyncTest "(remote close) - channel should allow writing on remote close": asyncTest "(remote close) - channel should allow writing on remote close":
let let
testData = "Hello!".toBytes testData = "Hello!".toBytes
conn = newBufferStream( conn = TestBufferStream.new(
proc (data: seq[byte]) {.gcsafe, async.} = proc (data: seq[byte]) {.gcsafe, async.} =
discard discard
) )
@ -168,7 +168,7 @@ suite "Mplex":
asyncTest "should not allow pushing data to channel when remote end closed": asyncTest "should not allow pushing data to channel when remote end closed":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
await chann.pushEof() await chann.pushEof()
var buf: array[1, byte] var buf: array[1, byte]
@ -185,7 +185,7 @@ suite "Mplex":
asyncTest "channel should fail reading": asyncTest "channel should fail reading":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
await chann.reset() await chann.reset()
@ -198,7 +198,7 @@ suite "Mplex":
asyncTest "reset should complete read": asyncTest "reset should complete read":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
var data = newSeq[byte](1) var data = newSeq[byte](1)
@ -213,7 +213,7 @@ suite "Mplex":
asyncTest "reset should complete pushData": asyncTest "reset should complete pushData":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
proc pushes() {.async.} = # pushes don't hang on reset proc pushes() {.async.} = # pushes don't hang on reset
@ -232,7 +232,7 @@ suite "Mplex":
asyncTest "reset should complete both read and push": asyncTest "reset should complete both read and push":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
var data = newSeq[byte](1) var data = newSeq[byte](1)
@ -247,7 +247,7 @@ suite "Mplex":
asyncTest "reset should complete both read and pushes": asyncTest "reset should complete both read and pushes":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
var data = newSeq[byte](1) var data = newSeq[byte](1)
@ -272,7 +272,7 @@ suite "Mplex":
asyncTest "reset should complete both read and push with cancel": asyncTest "reset should complete both read and push with cancel":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
var data = newSeq[byte](1) var data = newSeq[byte](1)
@ -286,7 +286,7 @@ suite "Mplex":
asyncTest "should complete both read and push after reset": asyncTest "should complete both read and push after reset":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
var data = newSeq[byte](1) var data = newSeq[byte](1)
@ -304,7 +304,7 @@ suite "Mplex":
asyncTest "reset should complete ongoing push without reader": asyncTest "reset should complete ongoing push without reader":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
await chann.pushData(@[0'u8]) await chann.pushData(@[0'u8])
@ -316,7 +316,7 @@ suite "Mplex":
asyncTest "reset should complete ongoing read without a push": asyncTest "reset should complete ongoing read without a push":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
var data = newSeq[byte](1) var data = newSeq[byte](1)
@ -328,7 +328,7 @@ suite "Mplex":
asyncTest "reset should allow all reads and pushes to complete": asyncTest "reset should allow all reads and pushes to complete":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
var data = newSeq[byte](1) var data = newSeq[byte](1)
@ -357,7 +357,7 @@ suite "Mplex":
asyncTest "channel should fail writing": asyncTest "channel should fail writing":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init(1, conn, true) chann = LPChannel.init(1, conn, true)
await chann.reset() await chann.reset()
@ -369,7 +369,7 @@ suite "Mplex":
asyncTest "channel should reset on timeout": asyncTest "channel should reset on timeout":
proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard proc writeHandler(data: seq[byte]) {.async, gcsafe.} = discard
let let
conn = newBufferStream(writeHandler) conn = TestBufferStream.new(writeHandler)
chann = LPChannel.init( chann = LPChannel.init(
1, conn, true, timeout = 100.millis) 1, conn, true, timeout = 100.millis)

24
tests/testmultistream.nim
View File

@ -172,13 +172,13 @@ suite "Multistream select":
checkTrackers() checkTrackers()
asyncTest "test select custom proto": asyncTest "test select custom proto":
let ms = newMultistream() let ms = MultistreamSelect.new()
let conn = newTestSelectStream() let conn = newTestSelectStream()
check (await ms.select(conn, @["/test/proto/1.0.0"])) == "/test/proto/1.0.0" check (await ms.select(conn, @["/test/proto/1.0.0"])) == "/test/proto/1.0.0"
await conn.close() await conn.close()
asyncTest "test handle custom proto": asyncTest "test handle custom proto":
let ms = newMultistream() let ms = MultistreamSelect.new()
let conn = newTestSelectStream() let conn = newTestSelectStream()
var protocol: LPProtocol = new LPProtocol var protocol: LPProtocol = new LPProtocol
@ -193,7 +193,7 @@ suite "Multistream select":
await ms.handle(conn) await ms.handle(conn)
asyncTest "test handle `ls`": asyncTest "test handle `ls`":
let ms = newMultistream() let ms = MultistreamSelect.new()
proc testLsHandler(proto: seq[byte]) {.async, gcsafe.} # forward declaration proc testLsHandler(proto: seq[byte]) {.async, gcsafe.} # forward declaration
let conn = Connection(newTestLsStream(testLsHandler)) let conn = Connection(newTestLsStream(testLsHandler))
@ -214,7 +214,7 @@ suite "Multistream select":
await done.wait(5.seconds) await done.wait(5.seconds)
asyncTest "test handle `na`": asyncTest "test handle `na`":
let ms = newMultistream() let ms = MultistreamSelect.new()
proc testNaHandler(msg: string): Future[void] {.async, gcsafe.} proc testNaHandler(msg: string): Future[void] {.async, gcsafe.}
let conn = newTestNaStream(testNaHandler) let conn = newTestNaStream(testNaHandler)
@ -245,7 +245,7 @@ suite "Multistream select":
await conn.close() await conn.close()
protocol.handler = testHandler protocol.handler = testHandler
let msListen = newMultistream() let msListen = MultistreamSelect.new()
msListen.addHandler("/test/proto/1.0.0", protocol) msListen.addHandler("/test/proto/1.0.0", protocol)
let transport1 = TcpTransport.init(upgrade = Upgrade()) let transport1 = TcpTransport.init(upgrade = Upgrade())
@ -258,7 +258,7 @@ suite "Multistream select":
let handlerWait = acceptHandler() let handlerWait = acceptHandler()
let msDial = newMultistream() let msDial = MultistreamSelect.new()
let transport2 = TcpTransport.init(upgrade = Upgrade()) let transport2 = TcpTransport.init(upgrade = Upgrade())
let conn = await transport2.dial(transport1.ma) let conn = await transport2.dial(transport1.ma)
@ -279,7 +279,7 @@ suite "Multistream select":
let let
handlerWait = newFuture[void]() handlerWait = newFuture[void]()
let msListen = newMultistream() let msListen = MultistreamSelect.new()
var protocol: LPProtocol = new LPProtocol var protocol: LPProtocol = new LPProtocol
protocol.handler = proc(conn: Connection, proto: string) {.async, gcsafe.} = protocol.handler = proc(conn: Connection, proto: string) {.async, gcsafe.} =
# never reached # never reached
@ -310,7 +310,7 @@ suite "Multistream select":
await conn.close() await conn.close()
let acceptFut = acceptHandler() let acceptFut = acceptHandler()
let msDial = newMultistream() let msDial = MultistreamSelect.new()
let transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade()) let transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade())
let conn = await transport2.dial(transport1.ma) let conn = await transport2.dial(transport1.ma)
@ -337,7 +337,7 @@ suite "Multistream select":
await conn.close() await conn.close()
protocol.handler = testHandler protocol.handler = testHandler
let msListen = newMultistream() let msListen = MultistreamSelect.new()
msListen.addHandler("/test/proto/1.0.0", protocol) msListen.addHandler("/test/proto/1.0.0", protocol)
let transport1: TcpTransport = TcpTransport.init(upgrade = Upgrade()) let transport1: TcpTransport = TcpTransport.init(upgrade = Upgrade())
@ -348,7 +348,7 @@ suite "Multistream select":
await msListen.handle(conn) await msListen.handle(conn)
let acceptFut = acceptHandler() let acceptFut = acceptHandler()
let msDial = newMultistream() let msDial = MultistreamSelect.new()
let transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade()) let transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade())
let conn = await transport2.dial(transport1.ma) let conn = await transport2.dial(transport1.ma)
@ -374,7 +374,7 @@ suite "Multistream select":
await conn.close() await conn.close()
protocol.handler = testHandler protocol.handler = testHandler
let msListen = newMultistream() let msListen = MultistreamSelect.new()
msListen.addHandler("/test/proto1/1.0.0", protocol) msListen.addHandler("/test/proto1/1.0.0", protocol)
msListen.addHandler("/test/proto2/1.0.0", protocol) msListen.addHandler("/test/proto2/1.0.0", protocol)
@ -386,7 +386,7 @@ suite "Multistream select":
await msListen.handle(conn) await msListen.handle(conn)
let acceptFut = acceptHandler() let acceptFut = acceptHandler()
let msDial = newMultistream() let msDial = MultistreamSelect.new()
let transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade()) let transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade())
let conn = await transport2.dial(transport1.ma) let conn = await transport2.dial(transport1.ma)

26
tests/testnoise.nim
View File

@ -58,15 +58,15 @@ proc createSwitch(ma: MultiAddress; outgoing: bool, secio: bool = false): (Switc
result = Mplex.init(conn) result = Mplex.init(conn)
let let
identify = newIdentify(peerInfo) identify = Identify.new(peerInfo)
mplexProvider = newMuxerProvider(createMplex, MplexCodec) mplexProvider = MuxerProvider.new(createMplex, MplexCodec)
muxers = [(MplexCodec, mplexProvider)].toTable() muxers = [(MplexCodec, mplexProvider)].toTable()
secureManagers = if secio: secureManagers = if secio:
[Secure(newSecio(rng, peerInfo.privateKey))] [Secure(Secio.new(rng, peerInfo.privateKey))]
else: else:
[Secure(newNoise(rng, peerInfo.privateKey, outgoing = outgoing))] [Secure(Noise.new(rng, peerInfo.privateKey, outgoing = outgoing))]
connManager = ConnManager.init() connManager = ConnManager.init()
ms = newMultistream() ms = MultistreamSelect.new()
muxedUpgrade = MuxedUpgrade.init(identify, muxers, secureManagers, connManager, ms) muxedUpgrade = MuxedUpgrade.init(identify, muxers, secureManagers, connManager, ms)
transports = @[Transport(TcpTransport.init(upgrade = muxedUpgrade))] transports = @[Transport(TcpTransport.init(upgrade = muxedUpgrade))]
@ -88,7 +88,7 @@ suite "Noise":
let let
server = Multiaddress.init("/ip4/0.0.0.0/tcp/0").tryGet() server = Multiaddress.init("/ip4/0.0.0.0/tcp/0").tryGet()
serverInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [server]) serverInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [server])
serverNoise = newNoise(rng, serverInfo.privateKey, outgoing = false) serverNoise = Noise.new(rng, serverInfo.privateKey, outgoing = false)
let transport1: TcpTransport = TcpTransport.init(upgrade = Upgrade()) let transport1: TcpTransport = TcpTransport.init(upgrade = Upgrade())
asyncCheck transport1.start(server) asyncCheck transport1.start(server)
@ -106,7 +106,7 @@ suite "Noise":
acceptFut = acceptHandler() acceptFut = acceptHandler()
transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade()) transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade())
clientInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [transport1.ma]) clientInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [transport1.ma])
clientNoise = newNoise(rng, clientInfo.privateKey, outgoing = true) clientNoise = Noise.new(rng, clientInfo.privateKey, outgoing = true)
conn = await transport2.dial(transport1.ma) conn = await transport2.dial(transport1.ma)
sconn = await clientNoise.secure(conn, true) sconn = await clientNoise.secure(conn, true)
@ -125,7 +125,7 @@ suite "Noise":
let let
server = Multiaddress.init("/ip4/0.0.0.0/tcp/0").tryGet() server = Multiaddress.init("/ip4/0.0.0.0/tcp/0").tryGet()
serverInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [server]) serverInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [server])
serverNoise = newNoise(rng, serverInfo.privateKey, outgoing = false) serverNoise = Noise.new(rng, serverInfo.privateKey, outgoing = false)
let let
transport1: TcpTransport = TcpTransport.init(upgrade = Upgrade()) transport1: TcpTransport = TcpTransport.init(upgrade = Upgrade())
@ -146,7 +146,7 @@ suite "Noise":
handlerWait = acceptHandler() handlerWait = acceptHandler()
transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade()) transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade())
clientInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [transport1.ma]) clientInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [transport1.ma])
clientNoise = newNoise(rng, clientInfo.privateKey, outgoing = true, commonPrologue = @[1'u8, 2'u8, 3'u8]) clientNoise = Noise.new(rng, clientInfo.privateKey, outgoing = true, commonPrologue = @[1'u8, 2'u8, 3'u8])
conn = await transport2.dial(transport1.ma) conn = await transport2.dial(transport1.ma)
var sconn: Connection = nil var sconn: Connection = nil
expect(NoiseDecryptTagError): expect(NoiseDecryptTagError):
@ -161,7 +161,7 @@ suite "Noise":
let let
server = Multiaddress.init("/ip4/0.0.0.0/tcp/0").tryGet() server = Multiaddress.init("/ip4/0.0.0.0/tcp/0").tryGet()
serverInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [server]) serverInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [server])
serverNoise = newNoise(rng, serverInfo.privateKey, outgoing = false) serverNoise = Noise.new(rng, serverInfo.privateKey, outgoing = false)
readTask = newFuture[void]() readTask = newFuture[void]()
let transport1: TcpTransport = TcpTransport.init(upgrade = Upgrade()) let transport1: TcpTransport = TcpTransport.init(upgrade = Upgrade())
@ -182,7 +182,7 @@ suite "Noise":
acceptFut = acceptHandler() acceptFut = acceptHandler()
transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade()) transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade())
clientInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [transport1.ma]) clientInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [transport1.ma])
clientNoise = newNoise(rng, clientInfo.privateKey, outgoing = true) clientNoise = Noise.new(rng, clientInfo.privateKey, outgoing = true)
conn = await transport2.dial(transport1.ma) conn = await transport2.dial(transport1.ma)
sconn = await clientNoise.secure(conn, true) sconn = await clientNoise.secure(conn, true)
@ -197,7 +197,7 @@ suite "Noise":
let let
server = Multiaddress.init("/ip4/0.0.0.0/tcp/0").tryGet() server = Multiaddress.init("/ip4/0.0.0.0/tcp/0").tryGet()
serverInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [server]) serverInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [server])
serverNoise = newNoise(rng, serverInfo.privateKey, outgoing = false) serverNoise = Noise.new(rng, serverInfo.privateKey, outgoing = false)
readTask = newFuture[void]() readTask = newFuture[void]()
var hugePayload = newSeq[byte](0xFFFFF) var hugePayload = newSeq[byte](0xFFFFF)
@ -221,7 +221,7 @@ suite "Noise":
acceptFut = acceptHandler() acceptFut = acceptHandler()
transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade()) transport2: TcpTransport = TcpTransport.init(upgrade = Upgrade())
clientInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [transport1.ma]) clientInfo = PeerInfo.init(PrivateKey.random(ECDSA, rng[]).get(), [transport1.ma])
clientNoise = newNoise(rng, clientInfo.privateKey, outgoing = true) clientNoise = Noise.new(rng, clientInfo.privateKey, outgoing = true)
conn = await transport2.dial(transport1.ma) conn = await transport2.dial(transport1.ma)
sconn = await clientNoise.secure(conn, true) sconn = await clientNoise.secure(conn, true)