netem_trace/

lib.rs

//! This crate provides a set of tools to generate traces for network emulation.
//!
//! ## Examples
//!
//! If you want to use the pre-defined models, please enable the `model` or `bw-model` feature.
//!
//! And if you want read configuration from file, `serde` feature should also be enabled.
//! We else recommend you to enable `human` feature to make the configuration file more human-readable.
//!
//! An example to build model from configuration:
//!
//! ```
//! # use netem_trace::model::StaticBwConfig;
//! # use netem_trace::{Bandwidth, Duration, BwTrace};
//! let mut static_bw = StaticBwConfig::new()
//!     .bw(Bandwidth::from_mbps(24))
//!     .duration(Duration::from_secs(1))
//!     .build();
//! assert_eq!(static_bw.next_bw(), Some((Bandwidth::from_mbps(24), Duration::from_secs(1))));
//! assert_eq!(static_bw.next_bw(), None);
//! ```
//!
//! A more common use case is to build model from a configuration file (e.g. json file):
//!
//! ```
//! # use netem_trace::model::{StaticBwConfig, BwTraceConfig};
//! # use netem_trace::{Bandwidth, Duration, BwTrace};
//! # #[cfg(feature = "human")]
//! # let config_file_content = "{\"RepeatedBwPatternConfig\":{\"pattern\":[{\"StaticBwConfig\":{\"bw\":\"12Mbps\",\"duration\":\"1s\"}},{\"StaticBwConfig\":{\"bw\":\"24Mbps\",\"duration\":\"1s\"}}],\"count\":2}}";
//! // The content would be "{\"RepeatedBwPatternConfig\":{\"pattern\":[{\"StaticBwConfig\":{\"bw\":{\"gbps\":0,\"bps\":12000000},\"duration\":{\"secs\":1,\"nanos\":0}}},{\"StaticBwConfig\":{\"bw\":{\"gbps\":0,\"bps\":24000000},\"duration\":{\"secs\":1,\"nanos\":0}}}],\"count\":2}}"
//! // if the `human` feature is not enabled.
//! # #[cfg(not(feature = "human"))]
//! let config_file_content = "{\"RepeatedBwPatternConfig\":{\"pattern\":[{\"StaticBwConfig\":{\"bw\":{\"gbps\":0,\"bps\":12000000},\"duration\":{\"secs\":1,\"nanos\":0}}},{\"StaticBwConfig\":{\"bw\":{\"gbps\":0,\"bps\":24000000},\"duration\":{\"secs\":1,\"nanos\":0}}}],\"count\":2}}";
//! let des: Box<dyn BwTraceConfig> = serde_json::from_str(config_file_content).unwrap();
//! let mut model = des.into_model();
//! assert_eq!(
//!     model.next_bw(),
//!     Some((Bandwidth::from_mbps(12), Duration::from_secs(1)))
//! );
//! assert_eq!(
//!     model.next_bw(),
//!     Some((Bandwidth::from_mbps(24), Duration::from_secs(1)))
//! );
//! assert_eq!(
//!     model.next_bw(),
//!     Some((Bandwidth::from_mbps(12), Duration::from_secs(1)))
//! );
//! assert_eq!(
//!     model.next_bw(),
//!     Some((Bandwidth::from_mbps(24), Duration::from_secs(1)))
//! );
//! assert_eq!(model.next_bw(), None);
//! ```
//!
//! ## Make your own model
//!
//! Here is an simple example of how to do this. For more complicated examples, please refer to our pre-defined models.
//!
//! ```
//! use netem_trace::BwTrace;
//! use netem_trace::{Bandwidth, Duration};
//!
//! struct MyStaticBw {
//!    bw: Bandwidth,
//!    duration: Option<Duration>,
//! }
//!
//! impl BwTrace for MyStaticBw {
//!     fn next_bw(&mut self) -> Option<(Bandwidth, Duration)> {
//!         if let Some(duration) = self.duration.take() {
//!             if duration.is_zero() {
//!                 None
//!             } else {
//!                 Some((self.bw, duration))
//!             }
//!         } else {
//!             None
//!         }
//!     }
//! }
//! ```
//!
//! This is almost the same as how this library implements the [`model::StaticBw`] model.
//!
//! ## Features
//!
//! ### Model Features
//!
//! - `model`: Enable this feature if you want to use all pre-defined models.
//!     - `bw-model`: Enable this feature if you want to use the pre-defined [`BwTrace`] models.
//!     - `truncated-normal`: Enable this feature if you want to use truncated normal distribution in [`model::NormalizedBw`] models.
//!
//! ### Trace Format Features
//!
//! - `mahimahi`: Enable this feature if you want to load or output traces in [mahimahi](https://github.com/ravinet/mahimahi) format.
//!
//! ### Other Features
//!
//! - `serde`: Enable this features if you want some structs to be serializable/deserializable. Often used with model features.
//! - `human`: Enable this feature if you want to use human-readable format in configuration files. Often used with model features.

#[cfg(feature = "mahimahi")]
pub mod mahimahi;
#[cfg(feature = "mahimahi")]
pub use mahimahi::{load_mahimahi_trace, Mahimahi, MahimahiExt};

#[cfg(any(
    feature = "bw-model",
    feature = "delay-model",
    feature = "loss-model",
    feature = "duplicate-model",
    feature = "model",
))]
pub mod model;

pub use bandwidth::Bandwidth;
pub use std::time::Duration;

/// The delay describes how long a packet is delayed when going through.
pub type Delay = std::time::Duration;

/// The loss_pattern describes how the packets are dropped when going through.
///
/// The loss_pattern is a sequence of conditional probabilities describing how packets are dropped.
/// The probability is a f64 between 0 and 1.
///
/// The meaning of the loss_pattern sequence is as follows:
///
/// - The probability on index 0 describes how likely a packet will be dropped **if the previous packet was not lost**.
/// - The probability on index 1 describes how likely a packet will be dropped **if the previous packet was lost**.
/// - The probability on index 2 describes how likely a packet will be dropped **if the previous 2 packet was lost**.
/// - ...
///
/// For example, if the loss_pattern is [0.1, 0.2], and packet 100 is not lost,
/// then the probability of packet 101 being lost is 0.1.
///
/// If the packet 101 is lost, then the probability of packet 102 being lost is 0.2.
/// If the packet 101 is not lost, then the probability of packet 102 being lost is still 0.1.
pub type LossPattern = Vec<f64>;

/// The duplicate_pattern describes how the packets are duplicated.
///
/// The duplicate_pattern is a sequence of conditional probabilities describing how packets are duplicated.
/// The probability is a f64 between 0 and 1.
///
/// The meaning of the duplicate_pattern sequence is:
///
/// - The probability on index 0 describes how likely a packet will be duplicated
///   **if the previous packet was transmitted normally**.
/// - The probability on index 1 describes how likely a packet will be duplicated
///   **if the previous packet was duplicated**.
/// - ...
///
/// For example, if the duplicate_pattern is [0.8, 0.1], and packet 100 is not duplicated, then the
/// probability of packet 101 being duplicated is 0.8.
///
/// If the packet 101 is duplicated, the the probability of packet 102 being duplicated is 0.1; if
/// the packet 101 is not duplicated, then the probability of packet 102 being duplicated is still 0.8.
///
/// If both packet 101 and 102 were duplicated, then the probability of packet 103 being duplicated
/// is still 0.1, and as long as the packets were duplicated, the probability of the next packet
/// being duplicated is always the last element - in this case, 0.1.
pub type DuplicatePattern = Vec<f64>;

/// This is a trait that represents a trace of bandwidths.
///
/// The trace is a sequence of `(bandwidth, duration)` pairs.
/// The bandwidth describes how many bits can be sent per second.
/// The duration is the time that the bandwidth lasts.
///
/// For example, if the sequence is [(1Mbps, 1s), (2Mbps, 2s), (3Mbps, 3s)],
/// then the bandwidth will be 1Mbps for 1s, then 2Mbps for 2s, then 3Mbps for 3s.
///
/// The next_bw function either returns **the next bandwidth and its duration**
/// in the sequence, or **None** if the trace goes to end.
pub trait BwTrace: Send {
    fn next_bw(&mut self) -> Option<(Bandwidth, Duration)>;
}

/// This is a trait that represents a trace of delays.
///
/// The trace is a sequence of `(delay, duration)` pairs.
/// The delay describes how long a packet is delayed when going through.
/// The duration is the time that the delay lasts.
///
/// For example, if the sequence is [(10ms, 1s), (20ms, 2s), (30ms, 3s)],
/// then the delay will be 10ms for 1s, then 20ms for 2s, then 30ms for 3s.
///
/// The next_delay function either returns **the next delay and its duration**
/// in the sequence, or **None** if the trace goes to end.
pub trait DelayTrace: Send {
    fn next_delay(&mut self) -> Option<(Delay, Duration)>;
}

/// This is a trait that represents a trace of per-packet delays.
///
/// The trace is a sequence of `delay`.
/// The delay describes how long the packet is delayed when going through.
///
/// For example, if the sequence is [10ms, 20ms, 30ms],
/// then the delay will be 10ms for the first packet, then 20ms for second, then 30ms for third.
///
/// The next_delay function either returns **the next delay**
/// in the sequence, or **None** if the trace goes to end.
pub trait DelayPerPacketTrace: Send {
    fn next_delay(&mut self) -> Option<Delay>;
}

/// This is a trait that represents a trace of loss patterns.
///
/// The trace is a sequence of `(loss_pattern, duration)` pairs.
/// The loss_pattern describes how packets are dropped when going through.
/// The duration is the time that the loss_pattern lasts.
///
/// The next_loss function either returns **the next loss_pattern and its duration**
/// in the sequence, or **None** if the trace goes to end.
pub trait LossTrace: Send {
    fn next_loss(&mut self) -> Option<(LossPattern, Duration)>;
}

/// This is a trait that represents a trace of duplicate patterns.
///
/// The trace is a sequence of `(duplicate_pattern, duration)` pairs.
/// The duplicate_pattern describes how packets are duplicated when going through.
/// The duration is the time that the duplicate_pattern lasts.
///
/// The next_duplicate function either returns **the next duplicate_pattern and its duration** in
/// the sequence, or **None** if the trace goes to end.
pub trait DuplicateTrace: Send {
    fn next_duplicate(&mut self) -> Option<(DuplicatePattern, Duration)>;
}

#[cfg(test)]
mod test {
    use model::TraceBwConfig;

    use self::model::bw::Forever;

    use super::*;
    #[cfg(feature = "serde")]
    use crate::model::RepeatedBwPatternConfig;
    use crate::model::{BwTraceConfig, NormalizedBwConfig, SawtoothBwConfig, StaticBwConfig};

    #[test]
    fn test_static_bw_model() {
        let mut static_bw = StaticBwConfig::new()
            .bw(Bandwidth::from_mbps(24))
            .duration(Duration::from_secs(1))
            .build();
        assert_eq!(
            static_bw.next_bw(),
            Some((Bandwidth::from_mbps(24), Duration::from_secs(1)))
        );
    }

    #[test]
    fn test_normalized_bw_model() {
        let mut normal_bw = NormalizedBwConfig::new()
            .mean(Bandwidth::from_mbps(12))
            .std_dev(Bandwidth::from_mbps(1))
            .duration(Duration::from_secs(1))
            .step(Duration::from_millis(100))
            .seed(42)
            .build();
        assert_eq!(
            normal_bw.next_bw(),
            Some((Bandwidth::from_bps(12069427), Duration::from_millis(100)))
        );
        assert_eq!(
            normal_bw.next_bw(),
            Some((Bandwidth::from_bps(12132938), Duration::from_millis(100)))
        );
        let mut normal_bw = NormalizedBwConfig::new()
            .mean(Bandwidth::from_mbps(12))
            .std_dev(Bandwidth::from_mbps(1))
            .duration(Duration::from_secs(1))
            .step(Duration::from_millis(100))
            .seed(42)
            .upper_bound(Bandwidth::from_kbps(12100))
            .lower_bound(Bandwidth::from_kbps(11900))
            .build();
        assert_eq!(
            normal_bw.next_bw(),
            Some((Bandwidth::from_bps(12069427), Duration::from_millis(100)))
        );
        assert_eq!(
            normal_bw.next_bw(),
            Some((Bandwidth::from_bps(12100000), Duration::from_millis(100)))
        );
    }

    #[test]
    fn test_sawtooth_bw_model() {
        let mut sawtooth_bw = SawtoothBwConfig::new()
            .bottom(Bandwidth::from_mbps(12))
            .top(Bandwidth::from_mbps(16))
            .duration(Duration::from_secs(1))
            .step(Duration::from_millis(100))
            .interval(Duration::from_millis(500))
            .duty_ratio(0.8)
            .build();
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(12), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(13), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(14), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(15), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(16), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(12), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(13), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(14), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(15), Duration::from_millis(100)))
        );
        let mut sawtooth_bw = SawtoothBwConfig::new()
            .bottom(Bandwidth::from_mbps(12))
            .top(Bandwidth::from_mbps(16))
            .duration(Duration::from_secs(1))
            .step(Duration::from_millis(100))
            .interval(Duration::from_millis(500))
            .duty_ratio(0.8)
            .std_dev(Bandwidth::from_mbps(5))
            .upper_noise_bound(Bandwidth::from_mbps(1))
            .lower_noise_bound(Bandwidth::from_kbps(500))
            .build();
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_bps(12347139), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_bps(13664690), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_mbps(15), Duration::from_millis(100)))
        );
        assert_eq!(
            sawtooth_bw.next_bw(),
            Some((Bandwidth::from_bps(14500000), Duration::from_millis(100)))
        );
    }

    #[test]
    fn test_trace_bw() {
        let mut trace_bw = TraceBwConfig::new()
            .pattern(vec![
                (
                    Duration::from_millis(1),
                    vec![
                        Bandwidth::from_kbps(29123),
                        Bandwidth::from_kbps(41242),
                        Bandwidth::from_kbps(7395),
                    ],
                ),
                (
                    Duration::from_millis(2),
                    vec![Bandwidth::from_mbps(1), Bandwidth::from_kbps(8542)],
                ),
            ])
            .build();

        assert_eq!(
            trace_bw.next_bw(),
            Some((Bandwidth::from_bps(29123000), Duration::from_millis(1)))
        );
        assert_eq!(
            trace_bw.next_bw(),
            Some((Bandwidth::from_bps(41242000), Duration::from_millis(1)))
        );
        assert_eq!(
            trace_bw.next_bw(),
            Some((Bandwidth::from_bps(7395000), Duration::from_millis(1)))
        );
        assert_eq!(
            trace_bw.next_bw(),
            Some((Bandwidth::from_bps(1000000), Duration::from_millis(2)))
        );
        assert_eq!(
            trace_bw.next_bw(),
            Some((Bandwidth::from_bps(8542000), Duration::from_millis(2)))
        );
        assert_eq!(trace_bw.next_bw(), None);
    }

    #[test]
    #[cfg(feature = "serde")]
    fn test_model_serde() {
        let a = vec![
            Box::new(
                StaticBwConfig::new()
                    .bw(Bandwidth::from_mbps(12))
                    .duration(Duration::from_secs(1)),
            ) as Box<dyn BwTraceConfig>,
            Box::new(
                StaticBwConfig::new()
                    .bw(Bandwidth::from_mbps(24))
                    .duration(Duration::from_secs(1)),
            ) as Box<dyn BwTraceConfig>,
        ];
        let ser =
            Box::new(RepeatedBwPatternConfig::new().pattern(a).count(2)) as Box<dyn BwTraceConfig>;
        let ser_str = serde_json::to_string(&ser).unwrap();
        #[cfg(not(feature = "human"))]
        let des_str = "{\"RepeatedBwPatternConfig\":{\"pattern\":[{\"StaticBwConfig\":{\"bw\":{\"gbps\":0,\"bps\":12000000},\"duration\":{\"secs\":1,\"nanos\":0}}},{\"StaticBwConfig\":{\"bw\":{\"gbps\":0,\"bps\":24000000},\"duration\":{\"secs\":1,\"nanos\":0}}}],\"count\":2}}";
        #[cfg(feature = "human")]
        let des_str = "{\"RepeatedBwPatternConfig\":{\"pattern\":[{\"StaticBwConfig\":{\"bw\":\"12Mbps\",\"duration\":\"1s\"}},{\"StaticBwConfig\":{\"bw\":\"24Mbps\",\"duration\":\"1s\"}}],\"count\":2}}";
        assert_eq!(ser_str, des_str);
        let des: Box<dyn BwTraceConfig> = serde_json::from_str(des_str).unwrap();
        let mut model = des.into_model();
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(12), Duration::from_secs(1)))
        );
    }

    #[test]
    fn test_forever() {
        let mut normal_bw = NormalizedBwConfig::new()
            .mean(Bandwidth::from_mbps(12))
            .std_dev(Bandwidth::from_mbps(1))
            .duration(Duration::from_millis(200))
            .step(Duration::from_millis(100))
            .seed(42)
            .build();
        assert_eq!(
            normal_bw.next_bw(),
            Some((Bandwidth::from_bps(12069427), Duration::from_millis(100)))
        );
        assert_eq!(
            normal_bw.next_bw(),
            Some((Bandwidth::from_bps(12132938), Duration::from_millis(100)))
        );
        assert_eq!(normal_bw.next_bw(), None);
        let normal_bw_config = NormalizedBwConfig::new()
            .mean(Bandwidth::from_mbps(12))
            .std_dev(Bandwidth::from_mbps(1))
            .duration(Duration::from_millis(200))
            .step(Duration::from_millis(100))
            .seed(42);
        let normal_bw_repeated = normal_bw_config.forever();
        let mut model = Box::new(normal_bw_repeated).into_model();
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_bps(12069427), Duration::from_millis(100)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_bps(12132938), Duration::from_millis(100)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_bps(12069427), Duration::from_millis(100)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_bps(12132938), Duration::from_millis(100)))
        );
    }

    #[test]
    #[cfg(feature = "human")]
    fn test_compatibility_with_figment() {
        use figment::{
            providers::{Format, Json},
            Figment,
        };

        let config = r##"
{
   "RepeatedBwPatternConfig":{
      "pattern":[
         {
            "TraceBwConfig":{
               "pattern":[
                  [
                     "25ms",
                     [
                        "10Mbps",
                        "20Mbps"
                     ]
                  ],
                  [
                     "2ms",
                     [
                        "11Mbps",
                        "23Mbps"
                     ]
                  ]
               ]
            }
         },
         {
            "SawtoothBwConfig":{
               "bottom":"10Mbps",
               "top":"20Mbps",
               "step":"1ms",
               "interval":"10ms",
               "duty_ratio":0.5
            }
         }
      ],
      "count":0
   }
}"##;

        let trace: Box<dyn BwTraceConfig> = Figment::new()
            .merge(Json::string(config))
            .extract()
            .unwrap();

        let mut model = trace.into_model();

        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(10), Duration::from_millis(25)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(20), Duration::from_millis(25)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(11), Duration::from_millis(2)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(23), Duration::from_millis(2)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(10), Duration::from_millis(1)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(12), Duration::from_millis(1)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(14), Duration::from_millis(1)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(16), Duration::from_millis(1)))
        );
        assert_eq!(
            model.next_bw(),
            Some((Bandwidth::from_mbps(18), Duration::from_millis(1)))
        );
    }
}