Add Yamux experimental support (#5080)

status-im · Jun 23, 2023 · 11d18ac · 11d18ac
1 parent 8a853fc
commit 11d18ac
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/beacon_chain/conf.nim b/beacon_chain/conf.nim
@@ -305,6 +305,12 @@ type
         defaultValue: false
         name: "enr-auto-update" .}: bool
 
+      enableYamux* {.
+        hidden
+        desc: "Enable the Yamux multiplexer"
+        defaultValue: false
+        name: "enable-yamux" .}: bool
+
       weakSubjectivityCheckpoint* {.
         desc: "Weak subjectivity checkpoint in the format block_root:epoch_number"
         name: "weak-subjectivity-checkpoint" .}: Option[Checkpoint]

diff --git a/beacon_chain/conf_light_client.nim b/beacon_chain/conf_light_client.nim
@@ -102,6 +102,12 @@ type LightClientConf* = object
     defaultValue: false
     name: "enr-auto-update" .}: bool
 
+  enableYamux* {.
+    hidden
+    desc: "Enable the Yamux multiplexer"
+    defaultValue: false
+    name: "enable-yamux" .}: bool
+
   agentString* {.
     defaultValue: "nimbus",
     desc: "Node agent string which is used as identifier in network"

diff --git a/beacon_chain/networking/eth2_network.nim b/beacon_chain/networking/eth2_network.nim
@@ -2271,8 +2271,14 @@ func gossipId(
 proc newBeaconSwitch(config: BeaconNodeConf | LightClientConf,
                      seckey: PrivateKey, address: MultiAddress,
                      rng: ref HmacDrbgContext): Switch {.raises: [Defect, CatchableError].} =
-  SwitchBuilder
-    .new()
+  var sb =
+    if config.enableYamux:
+      SwitchBuilder.new().withYamux()
+    else:
+      SwitchBuilder.new()
+  # Order of multiplexers matters, the first will be default
+
+  sb
     .withPrivateKey(seckey)
     .withAddress(address)
     .withRng(rng)