Remove bincode; Add set_packet_id_raw for setting packet_id in buffer…

…; Add message headers before run;

Cargo.toml

@@ -6,7 +6,6 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-bincode = "1.3.3"
 clap = { version = "4.4.14", features = ["derive"] }
 clap-markdown = "0.1.3"
 env_logger = "0.10.1"

src/net/mod.rs

@@ -47,6 +47,13 @@ impl MessageHeader {
         }
     }
 
+    pub fn set_packet_id_raw(buffer: &mut [u8], packet_id: u64) {
+        unsafe {
+            let header = std::mem::transmute::<&mut [u8], &mut [u64]>(buffer);
+            header[2] = packet_id;
+        }
+    }
+
     pub fn get_test_id(buffer: &[u8]) -> u64 {
         unsafe {
             let header = std::mem::transmute::<&[u8], &[u64]>(buffer);

src/node/client.rs

@@ -113,7 +113,7 @@ impl Client {
         let mut total_amount_used_packet_ids: u64 = 0;
 
         for packet_buffer in self.packet_buffer.iter_mut() {
-            let amount_used_packet_ids = packet_buffer.add_message_header(self.test_id, self.next_packet_id)?;
+            let amount_used_packet_ids = packet_buffer.add_packet_ids(self.next_packet_id)?;
             self.next_packet_id += amount_used_packet_ids;
             total_amount_used_packet_ids += amount_used_packet_ids;
         }
@@ -123,6 +123,11 @@ impl Client {
         Ok(total_amount_used_packet_ids)
     }
 
+    fn add_message_headers(&mut self) {
+        for packet_buffer in self.packet_buffer.iter_mut() {
+            packet_buffer.add_message_header(self.test_id, 0).expect("Error adding message header");
+        }
+    }
 
     fn fill_packet_buffers_with_repeating_pattern(&mut self) {
         for packet_buffer in self.packet_buffer.iter_mut() {
@@ -135,6 +140,7 @@ impl Client {
 impl Node for Client {
     fn run(&mut self, io_model: IOModel) -> Result<Statistic, &'static str> {
         self.fill_packet_buffers_with_repeating_pattern(); 
+        self.add_message_headers();
         self.socket.connect().expect("Error connecting to remote host"); 
 
         if let Ok(mss) = self.socket.get_mss() {

src/util/packet_buffer.rs

@@ -79,7 +79,7 @@ impl PacketBuffer {
         debug!("Filled buffer of size {} with repeating pattern", self.buffer.len());
     }
 
-    pub fn add_message_header(&mut self, test_id: u64, packet_id: u64) -> Result<u64, &'static str>{
+    pub fn add_message_header(&mut self, test_id: u64, packet_id: u64) -> Result<u64, &'static str> {
         let mut amount_used_packet_ids: u64 = 0;
         let mut header = MessageHeader::new(MessageType::MEASUREMENT, test_id, packet_id);
 
@@ -95,6 +95,20 @@ impl PacketBuffer {
         Ok(amount_used_packet_ids)
     }
 
+    pub fn add_packet_ids(&mut self, packet_id: u64) -> Result<u64, &'static str> {
+        let mut amount_used_packet_ids: u64 = 0;
+
+        for i in 0..self.packets_amount {
+            let start_of_packet = i * self.datagram_size as usize;
+            MessageHeader::set_packet_id_raw(&mut self.buffer[start_of_packet..], packet_id + amount_used_packet_ids);
+            amount_used_packet_ids += 1;
+        }
+
+        debug!("Added packet IDs to buffer! Used packet IDs: {}, Next packet ID: {}", amount_used_packet_ids, packet_id + amount_used_packet_ids);
+        // Return amount of used packet IDs
+        Ok(amount_used_packet_ids)
+    }
+
 
     pub fn get_buffer_pointer(&mut self) -> &mut [u8] {
         &mut self.buffer