tftpd 1.0.0

Multithreaded TFTP server daemon
Documentation 
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
38

use std::error::Error;
use std::sync::Mutex;

use crate::log::*;
use crate::Packet;

static TX_DROP: Mutex<Vec<i32>> = Mutex::new(Vec::new());

pub fn drop_set(opt: Option<String>) -> Result<(), Box<dyn Error>> {
    if let Some(arg) = opt {
        let mut tx_drop = TX_DROP.lock().unwrap();
        for val in arg.split(',') {
            let val_num = val.parse::<i32>()?;
            tx_drop.push(val_num);
        }
        Ok(())
    } else {
        Err("Missing argument".into())
    }
}

fn check_seq_num(num: u16) -> bool {
    let mut tx_drop = TX_DROP.lock().unwrap();
    if !tx_drop.is_empty() && tx_drop[0] == num as i32 {
        log_dbg!("Dropping pkt {} as requested", num);
        tx_drop.remove(0);
        return true;
    }
    false
}

pub fn drop_check(packet: &Packet) -> bool {
    match packet {
        Packet::Data { block_num, data: _ } => check_seq_num(*block_num),
        Packet::Ack(block_num) => check_seq_num(*block_num),
        _ => false,
    }
}