(feat/webrtc-sniffer): capture the traffic using pcap, parse network headers

Author: vlad9486

Cargo.lock

@@ -31,6 +31,8 @@ members = [
     "tools/fuzzing",
     "tools/archive-breadcrumb-compare",
     "tools/heartbeats-processor",
+    "tools/webrtc-sniffer",
+
     "producer-dashboard",
 
     "fuzzer",
@@ -54,8 +56,8 @@ mina-curves = { git = "https://github.com/openmina/proof-systems", rev = "dec49a
 # UNCOMMENTED_IN_CI mina-curves = { git = "https://github.com/openmina/proof-systems", rev = "dec49a9", features = [ "32x9" ] }
 o1-utils = { git = "https://github.com/openmina/proof-systems", rev = "dec49a9" }
 kimchi = { git = "https://github.com/openmina/proof-systems", rev = "dec49a9" }
-mina-poseidon = {git = "https://github.com/openmina/proof-systems", rev = "dec49a9" }
-poly-commitment = {git = "https://github.com/openmina/proof-systems", rev = "dec49a9" }
+mina-poseidon = { git = "https://github.com/openmina/proof-systems", rev = "dec49a9" }
+poly-commitment = { git = "https://github.com/openmina/proof-systems", rev = "dec49a9" }
 
 libp2p = { git = "https://github.com/openmina/rust-libp2p", rev = "5c44c7d9", default-features = false }
 vrf = { path = "vrf" }
@@ -90,9 +92,9 @@ incremental = false
 codegen-units = 1
 
 [patch.crates-io]
-ark-ff = { git = "https://github.com/openmina/algebra", rev = "aea157a" } # branch: fix-openmina-webnode
-ark-ec = { git = "https://github.com/openmina/algebra", rev = "aea157a" } # branch: fix-openmina-webnode
-ark-poly = { git = "https://github.com/openmina/algebra", rev = "aea157a" } # branch: fix-openmina-webnode
+ark-ff = { git = "https://github.com/openmina/algebra", rev = "aea157a" }        # branch: fix-openmina-webnode
+ark-ec = { git = "https://github.com/openmina/algebra", rev = "aea157a" }        # branch: fix-openmina-webnode
+ark-poly = { git = "https://github.com/openmina/algebra", rev = "aea157a" }      # branch: fix-openmina-webnode
 ark-serialize = { git = "https://github.com/openmina/algebra", rev = "aea157a" } # branch: fix-openmina-webnode
 
 num-bigint = { git = "https://github.com/openmina/num-bigint", rev = "8bb5ee4" }     # branch: on-stack

Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "webrtc-sniffer"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+clap = { version = "4.5", features = ["derive"] }
+env_logger = { version = "0.11.6" }
+log = { version = "0.4.25" }
+pcap = { version = "2.2" }
+ctrlc = { version = "3.4" }
+sudo = { version = "0.6.0" }
+hex = { version = "0.4.3" }
+etherparse = { version = "0.17.0" }
+thiserror = { version = "2.0" }

tools/webrtc-sniffer/Cargo.toml

@@ -0,0 +1,97 @@
+use std::path::PathBuf;
+
+use clap::Parser;
+use pcap::{Capture, ConnectionStatus, Device, IfFlags};
+
+// cargo run --release --bin sniffer -- --interface auto --port 443 --path target/test.pcap
+
+#[derive(Parser)]
+struct Cli {
+    #[arg(
+        long,
+        help = "name of the interface, use `auto` to determine automatically"
+    )]
+    interface: Option<String>,
+    #[arg(long, help = "filter by the port")]
+    port: u16,
+    #[arg(
+        long,
+        help = "if `interface` is set, the packets will be written to the `pcap` file, \
+                otherwise the file will be a source of packets"
+    )]
+    path: PathBuf,
+}
+
+fn init_logger_std() -> Box<dyn log::Log> {
+    use env_logger::{Builder, Env};
+
+    let env = Env::new().filter_or("RUST_LOG", "debug");
+    let logger = Builder::default().parse_env(env).build();
+    Box::new(logger) as Box<dyn log::Log>
+}
+
+fn main() {
+    log::set_boxed_logger(init_logger_std()).unwrap_or_default();
+    log::set_max_level(log::LevelFilter::max());
+
+    let Cli {
+        interface,
+        port,
+        path,
+    } = Cli::parse();
+    if let Some(name) = interface {
+        sudo::escalate_if_needed().unwrap();
+
+        log::info!("try to choose device");
+        let mut selected = None;
+        match Device::list() {
+            Ok(list) => {
+                for device in list {
+                    if name != "auto" {
+                        if device.name.eq(&name) {
+                            selected = Some(device);
+                            break;
+                        }
+                    } else {
+                        log::debug!("candidate: {device:?}");
+                        if !device.addresses.is_empty()
+                            && device.flags.contains(IfFlags::UP | IfFlags::RUNNING)
+                            && matches!(device.flags.connection_status, ConnectionStatus::Connected)
+                        {
+                            selected = Some(device);
+                        }
+                    }
+                }
+            }
+            Err(err) => log::error!("{err}"),
+        }
+
+        if let Some(device) = selected {
+            log::info!("will use: {device:?}");
+            let res = Ok(()).and_then(|()| {
+                let mut capture = Capture::from_device(device)?.open()?;
+                let filter = format!("udp port {port}");
+                capture
+                    .filter(&filter, true)
+                    .expect("Failed to apply filter");
+                let savefile = capture.savefile(&path)?;
+
+                webrtc_sniffer::run(capture, Some(savefile))
+            });
+            if let Err(err) = res {
+                log::error!("{err}");
+            }
+        } else {
+            log::error!("cannot find a device: {name}");
+        }
+    } else {
+        log::info!("use file");
+        let res = Ok(()).and_then(|()| {
+            let capture = Capture::from_file(&path)?;
+            webrtc_sniffer::run(capture, None)
+        });
+        if let Err(err) = res {
+            log::error!("{err}");
+        }
+    }
+}