cf-mach 0.2.2

Network quality measurement CLI for latency, throughput, packet loss, and responsiveness
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

// Copyright (c) 2023-2024 Cloudflare, Inc.
// Licensed under the BSD-3-Clause license found in the LICENSE file or at https://opensource.org/licenses/BSD-3-Clause

//! Arguments for running a network saturation test.

use crate::args::ConnType;

/// Download data (GET) from an endpoint, reporting latency measurements and
/// total throughput.
#[derive(Debug, clap::Args)]
pub struct SaturateArgs {
    /// The connection type to use.
    #[clap(short = 't', long, default_value = "h1-clear-text")]
    pub(crate) conn_type: ConnType,
    /// Which direction to saturate: `up`, `down` or `both`.
    #[clap(subcommand)]
    pub(crate) direction: Direction,
    /// The duration in seconds to saturate the network for. Defaults
    /// to 20s.
    #[clap(long, default_value = "20")]
    pub(crate) duration: u64,
}

#[derive(Debug, Clone, clap::Subcommand)]
pub enum Direction {
    /// Saturate the download (ingress) side of the network.
    Down {
        /// The URL to upload data to.
        #[clap(
            short,
            long,
            default_value = "http://h3.speed.cloudflare.com/__down?bytes=10000000000"
        )]
        download_url: String,
    },
    /// Saturate the upload (egress) side of the network.
    Up {
        /// The URL to upload data to.
        #[clap(short, long, default_value = "http://h3.speed.cloudflare.com/__up")]
        upload_url: String,
    },
    /// Saturate both the download (ingress) and upload (egress) side of the network.
    Both {
        /// The URL to download data from.
        #[clap(
            short,
            long,
            default_value = "http://h3.speed.cloudflare.com/__down?bytes=10000000000"
        )]
        download_url: String,
        /// The URL to upload data to.
        #[clap(short, long, default_value = "http://h3.speed.cloudflare.com/__up")]
        upload_url: String,
    },
}

impl Default for Direction {
    fn default() -> Self {
        Self::Down {
            download_url: "http://h3.speed.cloudflare.com/__down?bytes=10000000000".into(),
        }
    }
}

pub(crate) struct Urls {
    pub(crate) upload: String,
    pub(crate) download: String,
}

impl Direction {
    pub(crate) fn urls(&self) -> Urls {
        match self {
            Direction::Up { upload_url: url } | Direction::Down { download_url: url } => Urls {
                upload: url.clone(),
                download: url.clone(),
            },
            Direction::Both {
                download_url,
                upload_url,
            } => Urls {
                download: download_url.clone(),
                upload: upload_url.clone(),
            },
        }
    }
}