portls 0.1.0

Modern cross-platform port inspector - ls for network ports
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

use std::collections::HashSet;
use std::io::{self, Write};
use std::thread;
use std::time::Duration;

use anyhow::Result;

use crate::cli::{ProtocolFilter, SortField};
use crate::output::{json, table};
use crate::platform;
use crate::types::PortInfo;

pub struct WatchOptions {
    pub interval: Duration,
    pub json: bool,
    pub filter: Option<String>,
    pub connections: bool,
    pub sort: Option<SortField>,
    pub protocol: Option<ProtocolFilter>,
}

pub fn run(options: WatchOptions) -> Result<()> {
    let mut previous: HashSet<PortInfo> = HashSet::new();

    loop {
        clear_screen();

        let ports = if options.connections {
            platform::get_connections()?
        } else {
            platform::get_listening_ports()?
        };
        let ports = PortInfo::filter_protocol(ports, options.protocol);
        // Enrich docker-proxy entries with container names
        let ports = PortInfo::enrich_with_docker(ports);
        let mut filtered = filter_ports(ports, &options.filter);
        PortInfo::sort_vec(&mut filtered, options.sort);

        if options.json {
            json::print_ports(&filtered);
        } else {
            let new_ports: HashSet<&PortInfo> = filtered
                .iter()
                .filter(|p| !previous.contains(*p))
                .collect();

            table::print_ports_watch(&filtered, &new_ports);
        }

        print_watch_status(&options);
        io::stdout().flush()?;

        previous = filtered.into_iter().collect();
        thread::sleep(options.interval);
    }
}

fn filter_ports(ports: Vec<PortInfo>, filter: &Option<String>) -> Vec<PortInfo> {
    match filter {
        None => ports,
        Some(query) => {
            let query_lower = query.to_lowercase();
            if let Ok(port_num) = query.parse::<u16>() {
                ports.into_iter().filter(|p| p.port == port_num).collect()
            } else {
                ports
                    .into_iter()
                    .filter(|p| {
                        // Match process name or container name
                        p.process_name.to_lowercase().contains(&query_lower)
                            || p.container
                                .as_ref()
                                .map(|c| c.to_lowercase().contains(&query_lower))
                                .unwrap_or(false)
                    })
                    .collect()
            }
        }
    }
}

fn clear_screen() {
    print!("\x1B[2J\x1B[1;1H");
}

fn print_watch_status(options: &WatchOptions) {
    use colored::Colorize;
    let mode = if options.connections {
        "connections"
    } else {
        "listening"
    };
    println!(
        "\n{} {} (every {:.1}s, Ctrl+C to exit)",
        "Watching".dimmed(),
        mode.dimmed(),
        options.interval.as_secs_f64()
    );
}