From 49183383be0a5d76fd5ca98376dff489eec47453 Mon Sep 17 00:00:00 2001
From: Mimir <mimirz@proton.me>
Date: Tue, 16 Jul 2024 20:48:13 +0200
Subject: [PATCH] Added handshake for Yamux token

---
 p2p/src/network/select/p2p_network_select_effects.rs | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/p2p/src/network/select/p2p_network_select_effects.rs b/p2p/src/network/select/p2p_network_select_effects.rs
index 910585bba8..5b54787365 100644
--- a/p2p/src/network/select/p2p_network_select_effects.rs
+++ b/p2p/src/network/select/p2p_network_select_effects.rs
@@ -297,11 +297,13 @@ impl P2pNetworkSelectAction {
                         error: error.clone(),
                     });
                 } else if let Some(token) = &state.to_send {
-                    store.dispatch(P2pNetworkSelectAction::OutgoingTokens {
-                        addr,
-                        kind,
-                        tokens: vec![token.clone()],
-                    });
+                    let tokens = match token.clone() {
+                        Token::Protocol(Protocol::Mux(token)) => {
+                            vec![Token::Handshake, Token::Protocol(Protocol::Mux(token))]
+                        }
+                        token => vec![token],
+                    };
+                    store.dispatch(P2pNetworkSelectAction::OutgoingTokens { addr, kind, tokens });
                 }
             }
             P2pNetworkSelectAction::OutgoingTokens { addr, kind, tokens } => {