fumble 0.6.1

an oxidized implementation of the original clumsy tool, designed to simulate adverse network conditions on Windows systems.
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
39
40

use crate::network::modules::stats::util::ewma::Ewma;

pub struct DropStats {
    pub total_packets: usize,
    pub total_dropped: usize,
    ewma: Ewma,
}

impl DropStats {
    pub fn new(alpha: f64) -> Self {
        Self {
            total_packets: 0,
            total_dropped: 0,
            ewma: Ewma::new(alpha),
        }
    }

    pub fn record(&mut self, dropped: bool) {
        self.total_packets += 1;
        if dropped {
            self.total_dropped += 1;
        }

        // Update the EWMA with the new drop status (1.0 if dropped, 0.0 if not)
        let current_drop_rate = if dropped { 1.0 } else { 0.0 };
        self.ewma.update(current_drop_rate);
    }

    pub fn total_drop_rate(&self) -> f64 {
        if self.total_packets == 0 {
            0.0
        } else {
            self.total_dropped as f64 / self.total_packets as f64
        }
    }

    pub fn recent_drop_rate(&self) -> f64 {
        self.ewma.get().unwrap_or(0.0)
    }
}