Deduplicate listening addresses

Author: jcaesar

src/main.rs

@@ -12,7 +12,7 @@ use netlink::{
 };
 use procfs::process::all_processes;
 use std::{
-    collections::{BTreeMap, HashMap},
+    collections::{BTreeMap, BTreeSet, HashMap},
     env::var_os,
     io::{stdout, BufWriter, Write},
     net::{IpAddr, Ipv4Addr, Ipv6Addr},
@@ -166,7 +166,7 @@ fn interfaces_routes() -> IfaceInfo {
 }
 
 fn sockets_tree<'a>(
-    sockets: impl IntoIterator<Item = impl Deref<Target = SockInfo<'a>>>,
+    sockets: impl IntoIterator<Item = impl Deref<Target = SockInfo<'a>> + std::cmp::Ord>,
     filter: &options::Filters,
 ) -> termtree::Tree {
     let mut pout = termtree::Tree::new();
@@ -184,14 +184,16 @@ fn sockets_tree<'a>(
         {
             sout.leaf("0.0.0.0 + ::".into());
         } else {
+            let socks = socks
+                .iter()
+                .filter(|sock| filter.accept_addr(sock.addr))
+                .collect::<BTreeSet<_>>();
             for sock in socks {
-                if filter.accept_addr(sock.addr) {
-                    match (sock.family, sock.iface) {
-                        (Family::Both, _) => sout.leaf("*".into()),
-                        (_, Some(ifname)) => sout.leaf(format!("{} ({ifname})", sock.addr)),
-                        _ => sout.leaf(format!("{}", sock.addr)),
-                    };
-                }
+                match (sock.family, sock.iface) {
+                    (Family::Both, _) => sout.leaf("*".into()),
+                    (_, Some(ifname)) => sout.leaf(format!("{} ({ifname})", sock.addr)),
+                    _ => sout.leaf(format!("{}", sock.addr)),
+                };
             }
         }
         if filter.accept_port(port) && filter.accept_proto(proto) {