net_agent/
packet_handler.rs

1use net_file::translator::{global_header::GlobalHeader, packet::Packet};
2use std::fs::File;
3use uuid::Uuid;
4use std::io::prelude::*;
5use std::fmt::Debug;
6use thiserror::Error;
7
8fn write_to_file(buf: &[u8], directory: &str) -> std::io::Result<()> {
9    let file_name = format!("{}/{}.pcap", directory, Uuid::new_v4());
10    let mut file = File::create(file_name)?;
11    file.write_all(buf)?;
12    Ok(())
13}
14
15pub struct PacketHandler { 
16    pub directory: String,
17}
18
19impl crate::core::poller::Handler for PacketHandler {
20    fn decode(&self, packet: Packet) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
21        let global_header = GlobalHeader::default();
22
23        let mut buf = global_header.to_bytes();
24        buf.append(&mut packet.to_bytes());
25
26        match write_to_file(&buf, &self.directory) {
27            Err(err) => Err(Box::new(PacketHandlerError::WriteToFile(self.directory.clone(), format!("{}", err)))), 
28            Ok(_) => Ok(())
29        }
30    }
31}
32
33#[derive(Debug, Error)]
34pub enum PacketHandlerError {
35    #[error("Coulnd't write to '{0}' directory: {1}")]
36    WriteToFile(String, String),
37}
net_agent/packet_handler.rs

net_agent/
packet_handler.rs
