add processes_info::active_executables for linux

Cargo.toml

@@ -84,6 +84,7 @@ sysinfo = "0.29.10"
 
 [target.'cfg(target_os = "linux")'.dependencies]
 tun = { version = "0.7.5", features = ["async"] }
+sysinfo = "0.33.0"
 
 [dev-dependencies]
 env_logger = "0.11"

mitmproxy-rs/src/process_info.rs

@@ -3,7 +3,7 @@ use std::path::{Path, PathBuf};
 use anyhow::Result;
 use pyo3::prelude::*;
 
-#[cfg(any(windows, target_os = "macos"))]
+#[cfg(any(windows, target_os = "macos", target_os = "linux"))]
 use mitmproxy::processes;
 
 #[pyclass(module = "mitmproxy_rs.process_info", frozen)]
@@ -47,18 +47,18 @@ impl Process {
 /// Return a list of all running executables.
 /// Note that this groups multiple processes by executable name.
 ///
-/// *Availability: Windows, macOS*
+/// *Availability: Windows, macOS, Linux*
 #[pyfunction]
 pub fn active_executables() -> PyResult<Vec<Process>> {
-    #[cfg(any(windows, target_os = "macos"))]
+    #[cfg(any(windows, target_os = "macos", target_os = "linux"))]
     {
         processes::active_executables()
             .map(|p| p.into_iter().map(Process).collect())
             .map_err(|e| pyo3::exceptions::PyRuntimeError::new_err(format!("{}", e)))
     }
-    #[cfg(not(any(windows, target_os = "macos")))]
+    #[cfg(not(any(windows, target_os = "macos", target_os = "linux")))]
     Err(pyo3::exceptions::PyNotImplementedError::new_err(
-        "active_executables is only available on Windows",
+        "active_executables not supported on the current OS",
     ))
 }
 

src/processes/linux_list.rs

@@ -0,0 +1,51 @@
+use crate::processes::{ProcessInfo, ProcessList};
+use anyhow::Result;
+use std::collections::hash_map::Entry;
+use std::collections::HashMap;
+use std::path::PathBuf;
+use sysinfo::{ProcessRefreshKind, ProcessesToUpdate, System, UpdateKind};
+
+pub fn active_executables() -> Result<ProcessList> {
+    let mut executables: HashMap<PathBuf, ProcessInfo> = HashMap::new();
+    let mut sys = System::new();
+
+    sys.refresh_processes_specifics(
+        ProcessesToUpdate::All,
+        true,
+        ProcessRefreshKind::nothing().with_exe(UpdateKind::OnlyIfNotSet),
+    );
+
+    for process in sys.processes().values() {
+        // process.exe() will return empty path if there was an error while trying to read /proc/<pid>/exe.
+        if let Some(exec) = process.exe() {
+            let exec_buf = exec.to_path_buf();
+
+            match executables.entry(exec_buf) {
+                Entry::Occupied(_) => {}
+                Entry::Vacant(e) => {
+                    let exec_buf = e.key().clone();
+                    // process display name can contain non-UTF-8 characters, forcing us to use to_string_lossy
+                    let display_name = process.name().to_string_lossy().to_string();
+                    e.insert(ProcessInfo {
+                        executable: exec_buf,
+                        display_name,
+                        is_visible: false,
+                        is_system: false,
+                    });
+                }
+            }
+        }
+    }
+    Ok(executables.into_values().collect())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn process_list() {
+        let lst = active_executables().unwrap();
+        assert!(!lst.is_empty());
+    }
+}

src/processes/mod.rs

@@ -13,6 +13,11 @@ pub use self::windows_list::active_executables;
 #[cfg(windows)]
 pub use self::windows_list::get_process_name;
 
+#[cfg(target_os = "linux")]
+mod linux_list;
+#[cfg(target_os = "linux")]
+pub use self::linux_list::active_executables;
+
 #[cfg(target_os = "macos")]
 mod macos_icons;
 #[cfg(target_os = "macos")]