1 files changed, 38 insertions, 0 deletions
diff --git a/src/server/packet_parser.rs b/src/server/packet_parser.rs
new file mode 100644
index 0000000..d8fd93d
--- /dev/null
+++ b/src/server/packet_parser.rs
@@ -0,0 +1,38 @@
+// Copyleft 2021-2021 Whirlsplash
+// SPDX-License-Identifier: GPL-3.0-only
+
+use bytes::BytesMut;
+
+/// Read all commands from the given buffer.
+///
+/// # Process
+/// 1. Get a command from `buffer` based on first byte.
+/// 2. Push command to `commands`.
+/// 3. Remove command from `buffer`.
+/// 4. Iterate and do this for all commands within `buffer`.
+pub fn parse_commands_from_packet(mut buffer: BytesMut) -> Vec<BytesMut> {
+  let mut commands: Vec<BytesMut> = Vec::new();
+  debug!("initial buffer: {:?}, length: {}", buffer, buffer.len());
+
+  let data_length = buffer.get(0).unwrap().to_owned() as usize;
+  if buffer.len() > data_length {
+    loop {
+      debug!("loop: {:?}, length: {}", buffer, buffer.len());
+      let command_length = buffer.get(0).unwrap().to_owned() as usize;
+      commands.push(BytesMut::from(buffer.get(0..command_length).unwrap()));
+
+      // Remove command from buffer
+      buffer = buffer.split_off(command_length);
+
+      // Check if any more commands are present
+      if buffer.is_empty() {
+        break;
+      }
+    }
+  } else {
+    // There will always be at least one command, push it.
+    commands.push(BytesMut::from(buffer.get(0..data_length).unwrap()));
+  }
+
+  commands // Return command (s)
+}