cf-mach 0.2.1

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

// 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

mod access;
mod aim_report;
pub(crate) mod args;
mod latency;
mod nq_core;
mod nq_latency;
mod nq_load_generator;
mod nq_packetloss;
mod nq_rpm;
mod nq_stats;
mod nq_tokio_network;
mod packet_loss;
mod report;
mod rpm;
mod saturate;
mod up_down;
mod util;

use clap::Parser;
use clap_verbosity_flag::LevelFilter;

use crate::args::Command;
use crate::args::rpm::RpmArgs;

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let args = args::Cli::parse();

    setup_logging(args.verbosity)?;

    // default to RPM
    let command = args
        .command
        .unwrap_or_else(|| Command::Rpm(RpmArgs::default()));

    match command {
        Command::Rpm(config) => rpm::run(config).await?,
        Command::Download(config) => up_down::download(config).await?,
        Command::Upload(config) => up_down::upload(config).await?,
        Command::Rtt { url, runs } => latency::run(url, runs).await?,
        Command::PacketLoss(config) => packet_loss::run(config).await?,
        Command::Saturate(config) => saturate::run(config).await?,
    }

    Ok(())
}

fn setup_logging(verbosity: clap_verbosity_flag::Verbosity) -> anyhow::Result<()> {
    let filter = if let Ok(log) = std::env::var("RUST_LOG") {
        log
    } else {
        match verbosity.log_level_filter() {
            LevelFilter::Off => "off",
            LevelFilter::Error => "error",
            LevelFilter::Warn => "warn",
            LevelFilter::Info => "info",
            LevelFilter::Debug => "debug",
            LevelFilter::Trace => "trace",
        }
        .to_string()
    };

    tracing_subscriber::fmt()
        .with_env_filter(filter)
        .with_writer(std::io::stderr)
        .init();

    Ok(())
}