1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use net_file::translator::{global_header::GlobalHeader, packet::Packet};
use std::fs::File;
use uuid::Uuid;
use std::io::prelude::*;
use std::fmt::Debug;
use thiserror::Error;

fn write_to_file(buf: &[u8], directory: &str) -> std::io::Result<()> {
    let file_name = format!("{}/{}.pcap", directory, Uuid::new_v4());
    let mut file = File::create(file_name)?;
    file.write_all(buf)?;
    Ok(())
}

pub struct PacketHandler { 
    pub directory: String,
}

impl crate::core::poller::Handler for PacketHandler {
    fn decode(&self, packet: Packet) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
        let global_header = GlobalHeader::default();

        let mut buf = global_header.to_bytes();
        buf.append(&mut packet.to_bytes());

        match write_to_file(&buf, &self.directory) {
            Err(err) => Err(Box::new(PacketHandlerError::WriteToFile(self.directory.clone(), format!("{}", err)))), 
            Ok(_) => Ok(())
        }
    }
}

#[derive(Debug, Error)]
pub enum PacketHandlerError {
    #[error("Coulnd't write to '{0}' directory: {1}")]
    WriteToFile(String, String),
}