From 3bdae21716d10032f70a0a889070766bbbe4d141 Mon Sep 17 00:00:00 2001
From: Max Audron <audron@cocaine.farm>
Date: Sat, 26 Sep 2020 12:01:27 +0200
Subject: split handshake.rs

---
 src/message/handshake/protocol.rs | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 src/message/handshake/protocol.rs

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

diff --git a/src/message/handshake/protocol.rs b/src/message/handshake/protocol.rs
new file mode 100644
index 0000000..d020f33
--- /dev/null
+++ b/src/message/handshake/protocol.rs
@@ -0,0 +1,36 @@
+use crate::Deserialize;
+use crate::Serialize;
+
+pub enum Protocol {
+    Legacy = 0x00000001,
+    Datastream = 0x00000002,
+}
+
+impl Protocol {
+    pub fn new() -> Self {
+        Protocol::Datastream
+    }
+
+    pub fn serialize(self) -> Vec<u8> {
+        let proto: u32 = 0x80000002;
+
+        proto.serialize().unwrap()
+    }
+
+    pub fn parse(buf: &[u8]) -> Self {
+        let mut protolist: Vec<u32> = Vec::new();
+        let mut pos = 0;
+        loop {
+            let (_, proto) = u32::parse(&buf[pos..(pos + 4)]).unwrap();
+            if (proto & 0x80000000) >= 1 {
+                protolist.push(proto - 0x80000000);
+                break;
+            } else {
+                protolist.push(proto);
+                pos += 4;
+            }
+        }
+
+        Protocol::Datastream
+    }
+}
-- 
cgit v1.2.3