From 13734288dbc63c48af0b8f75f0453f0d30b750a7 Mon Sep 17 00:00:00 2001
From: Max Audron <audron@cocaine.farm>
Date: Sat, 2 Jan 2021 19:57:06 +0100
Subject: rework handshakemessage parsing

---
 src/message/handshake/init.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'src/message/handshake/init.rs')

diff --git a/src/message/handshake/init.rs b/src/message/handshake/init.rs
index e4c0fa9..860df8a 100644
--- a/src/message/handshake/init.rs
+++ b/src/message/handshake/init.rs
@@ -37,7 +37,15 @@ impl Init {
             handshake |= 0x02;
         }
 
-        return handshake.serialize().unwrap();
+        // Select Protocol 2: Datastream
+
+        let mut init: Vec<u8> = vec![];
+
+        // Add handshake and protocol to our buffer
+        init.extend(handshake.serialize().unwrap());
+        init.extend(crate::message::Protocol::Datastream.serialize());
+
+        return init;
     }
 
     pub fn parse(buf: &[u8]) -> Self {
-- 
cgit v1.2.3