Passing correct dir to Yamux during dcutr

libp2p/builders.nim

@@ -111,7 +111,7 @@ proc withMplex*(
     maxChannCount = 200): SwitchBuilder {.public.} =
   ## | Uses `Mplex <https://docs.libp2p.io/concepts/stream-multiplexing/#mplex>`_ as a multiplexer
   ## | `Timeout` is the duration after which a inactive connection will be closed
-  proc newMuxer(conn: Connection): Muxer =
+  proc newMuxer(conn: Connection, direction: Opt[Direction] = Opt.none(Direction)): Muxer =
     Mplex.new(
       conn,
       inTimeout,
@@ -123,7 +123,7 @@ proc withMplex*(
   b
 
 proc withYamux*(b: SwitchBuilder): SwitchBuilder =
-  proc newMuxer(conn: Connection): Muxer = Yamux.new(conn)
+  proc newMuxer(conn: Connection, dir: Opt[Direction] = Opt.none(Direction)): Muxer = Yamux.new(conn, dir)
 
   assert b.muxers.countIt(it.codec == YamuxCodec) == 0, "Yamux build multiple times"
   b.muxers.add(MuxerProvider.new(newMuxer, YamuxCodec))

libp2p/dialer.nim

@@ -75,9 +75,7 @@ proc dialAndUpgrade(
 
       let mux =
         try:
-          let m = await transport.upgrade(dialed, upgradeDir, peerId)
-          m.connection.dir = upgradeDir
-          m
+          await transport.upgrade(dialed, upgradeDir, peerId)
         except CatchableError as exc:
           # If we failed to establish the connection through one transport,
           # we won't succeeded through another - no use in trying again

libp2p/muxers/muxer.nim

@@ -10,6 +10,7 @@
 {.push raises: [].}
 
 import chronos, chronicles
+import stew/results
 import ../stream/connection,
        ../errors
 
@@ -32,7 +33,7 @@ type
     connection*: Connection
 
   # user provider proc that returns a constructed Muxer
-  MuxerConstructor* = proc(conn: Connection): Muxer {.gcsafe, closure, raises: [].}
+  MuxerConstructor* = proc(conn: Connection, direction: Opt[Direction] = Opt.none(Direction)): Muxer {.gcsafe, closure, raises: [].}
 
   # this wraps a creator proc that knows how to make muxers
   MuxerProvider* = object

libp2p/muxers/yamux/yamux.nim

@@ -10,7 +10,7 @@
 {.push raises: [].}
 
 import sequtils, std/[tables]
-import chronos, chronicles, metrics, stew/[endians2, byteutils, objects]
+import chronos, chronicles, metrics, stew/[endians2, byteutils, objects, results]
 import ../muxer,
        ../../stream/connection
 
@@ -389,14 +389,7 @@ proc createStream(m: Yamux, id: uint32, isSrc: bool): YamuxChannel =
     closedRemotely: newFuture[void]()
   )
   result.objName = "YamuxStream"
-  result.dir =
-    if isSrc:
-      if m.connection.dir == Direction.In:
-        Direction.In
-      else:
-        Direction.Out
-    else:
-      Direction.In
+  result.dir = if isSrc: Direction.Out else: Direction.In
   result.timeoutHandler = proc(): Future[void] {.gcsafe.} =
     trace "Idle timeout expired, resetting YamuxChannel"
     result.reset()
@@ -531,9 +524,14 @@ method newStream*(
     await stream.open()
   return stream
 
-proc new*(T: type[Yamux], conn: Connection, maxChannCount: int = MaxChannelCount): T =
+proc new*(T: type[Yamux], conn: Connection, direction: Opt[Direction] = Opt.none(Direction), maxChannCount: int = MaxChannelCount): T =
+  let dir =
+    block:
+      direction.withValue(d):
+        d
+      else: conn.dir
   T(
     connection: conn,
-    currentId: if conn.dir == Out: 1 else: 2,
+    currentId: if dir == Out: 1 else: 2,
     maxChannCount: maxChannCount
   )

libp2p/upgrademngrs/muxedupgrade.nim

@@ -52,7 +52,7 @@ proc mux*(
   trace "Found a muxer", conn, muxerName
 
   # create new muxer for connection
-  let muxer = self.getMuxerByCodec(muxerName).newMuxer(conn)
+  let muxer = self.getMuxerByCodec(muxerName).newMuxer(conn, Opt.some(direction))
 
   # install stream handler
   muxer.streamHandler = self.streamHandler