netdig 0.0.5

Utilities for analyzing and aggregating CIDR blocks
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
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use anyhow::Result;
#[cfg(feature = "tracing")]
use clap::ArgAction;
use clap::Parser;
#[cfg(feature = "json")]
use netdig::Json;
use netdig::{IpSet, Tabular};
use std::io;
use strum::EnumString;
#[cfg(feature = "tracing")]
use tracing::{debug, trace};

#[derive(Debug, Parser)]
struct Cli {
    #[cfg(feature = "tracing")]
    #[clap(short, long, action = ArgAction::Count, global = true)]
    verbose: u8,

    /// Format the output for display.
    #[clap(short, long, default_value = "text", global = true)]
    format: Format,

    /// Aggregate subnets together into the minimum number of blocks that
    /// include all subnets. Prints the aggregated list and doesn't run WHOIS
    /// queries.
    #[clap(short, long)]
    aggregate: bool,

    /// Truncate any addresses to the cananonical address for the subnet. Prints
    /// the truncated list and doesn't run WHOIS queries.
    #[clap(short, long)]
    truncate: bool,

    /// Skip running the WHOIS query, and just summarize/validate the subnets.
    #[clap(long)]
    no_whois: bool,

    ips: Vec<String>,
}

#[derive(Debug, Clone, Copy, EnumString)]
#[strum(serialize_all = "snake_case")]
enum Format {
    Text,
    #[cfg(feature = "json")]
    Json,
}

#[allow(clippy::too_many_lines)]
fn main() -> Result<()> {
    let cli = Cli::parse();

    #[cfg(feature = "tracing")]
    {
        jacklog::from_level!(cli.verbose + 2)?;
        debug!(?cli);
    }

    // If the ips argument was empty, then we need to read the IPs from STDIN.
    let mut ips = IpSet::default();
    if cli.ips.is_empty() {
        let lines = io::stdin().lines();
        for line in lines {
            let line = line?;
            let line = line.trim();
            #[cfg(feature = "tracing")]
            trace!(?line);

            if line.is_empty() {
                continue;
            }

            ips.insert(line.to_string())?;
        }
    } else {
        for ip in cli.ips {
            ips.insert(ip)?;
        }
    };

    let outputs = ips.check_ips(!cli.no_whois)?;

    // If the user just wants the aggregated subnets, display those.
    if cli.aggregate {
        for out in outputs {
            let Some(agg) = out.contained_by else {
                continue;
            };

            println!("{agg}");
        }

        return Ok(());
    }

    // If the user just wants the truncated, canonical form of the subnet, print
    // those and return.
    if cli.truncate {
        for out in outputs {
            println!("{}", out.canonical);
        }

        return Ok(());
    }

    match cli.format {
        #[cfg(feature = "json")]
        Format::Json => {
            println!("{}", outputs.to_json()?);
        },

        Format::Text => {
            for line in outputs.table(atty::is(atty::Stream::Stdout)).lines() {
                println!("{}", line.trim());
            }
        },
    }

    Ok(())
}