minilsof 0.1.4

minilsof
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

#[cfg(feature = "async")]
use tokio::task;
use crate::{Fdinfo, LsofData, Result};
use std::collections::HashMap;
use std::path::Path;
use std::sync::{Arc, Mutex};

/// Asynchronous wrapper functions for LsofData
/// 
/// This implementation uses tokio to run the blocking operations
/// in a separate thread pool to avoid blocking the async runtime.
#[cfg(feature = "async")]
pub struct LsofAsync {
    inner: Arc<Mutex<LsofData>>,
}

#[cfg(feature = "async")]
impl LsofAsync {
    /// Create a new LsofAsync instance
    pub fn new() -> Self {
        Self {
            inner: Arc::new(Mutex::new(LsofData::new())),
        }
    }

    /// Get information about all open files by all processes
    pub async fn file_ls(&self) -> Result<HashMap<String, Fdinfo>> {
        let inner = self.inner.clone();
        
        task::spawn_blocking(move || {
            let mut data = inner.lock().map_err(|_| 
                crate::Error::Other("Failed to acquire lock".to_string())
            )?;
            
            let result = data.file_ls().ok_or_else(|| 
                crate::Error::Other("Failed to list all files".to_string())
            )?;
            
            // Clone the result to avoid returning a reference to data inside the lock
            Ok(result.clone())
        }).await.map_err(|_| crate::Error::Other("Task join error".to_string()))?
    }

    /// Get information about processes using a specific file
    pub async fn target_file_ls(&self, path: impl AsRef<Path> + Send + 'static) -> Result<Vec<Fdinfo>> {
        let path_str = path.as_ref().to_string_lossy().to_string();
        let inner = self.inner.clone();
        
        task::spawn_blocking(move || {
            let mut data = inner.lock().map_err(|_| 
                crate::Error::Other("Failed to acquire lock".to_string())
            )?;
            
            data.target_file_ls(path_str).ok_or_else(|| 
                crate::Error::Other(format!("Failed to list file: {}", path.as_ref().display()))
            )
        }).await.map_err(|_| crate::Error::Other("Task join error".to_string()))?
    }

    /// Get information about processes using a specific port
    pub async fn port_ls(&self, port: impl AsRef<str> + Send + 'static) -> Result<Vec<Fdinfo>> {
        let port_str = port.as_ref().to_string();
        let inner = self.inner.clone();
        
        task::spawn_blocking(move || {
            let mut data = inner.lock().map_err(|_| 
                crate::Error::Other("Failed to acquire lock".to_string())
            )?;
            
            data.port_ls(port_str).ok_or_else(|| 
                crate::Error::Other(format!("Failed to list port: {}", port.as_ref()))
            )
        }).await.map_err(|_| crate::Error::Other("Task join error".to_string()))?
    }
}

#[cfg(feature = "async")]
impl Default for LsofAsync {
    fn default() -> Self {
        Self::new()
    }
}

#[cfg(all(test, feature = "async"))]
mod tests {
    use super::*;
    
    #[tokio::test]
    async fn test_async_file_ls() {
        let lsof = LsofAsync::new();
        let result = lsof.file_ls().await;
        assert!(result.is_ok());
    }
    
    #[tokio::test]
    async fn test_async_target_file() {
        let lsof = LsofAsync::new();
        // Use a common file that should exist on most systems
        let result = lsof.target_file_ls("/etc/passwd").await;
        // The test passes whether or not processes are using the file
        if let Ok(processes) = &result {
            println!("Found {} processes using /etc/passwd", processes.len());
        }
    }
    
    #[tokio::test]
    async fn test_async_port() {
        let lsof = LsofAsync::new();
        let result = lsof.port_ls("80").await;
        // The test passes whether or not processes are using the port
        if let Ok(processes) = &result {
            println!("Found {} processes using port 80", processes.len());
        }
    }
}