herolib-mos 0.3.13

Mycelium Operating System (MOS) - Network and VM abstraction layer
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
116
117
118
119
120
121
122
123
124
125
126
127

use super::error::{NetworkError, Result};
use std::process::Command;

pub struct Bridge {
    pub name: String,
}

/// Create a new bridge
pub fn create_bridge(name: &str) -> Result<Bridge> {
    if bridge_exists(name)? {
        return Ok(Bridge {
            name: name.to_string(),
        });
    }

    // Create bridge using ip link
    let output = Command::new("ip")
        .args(["link", "add", "name", name, "type", "bridge"])
        .output()
        .map_err(|e| NetworkError::Bridge(format!("Failed to execute ip command: {}", e)))?;

    if !output.status.success() {
        return Err(NetworkError::Bridge(format!(
            "Failed to create bridge {}: {}",
            name,
            String::from_utf8_lossy(&output.stderr)
        )));
    }

    // Set MTU
    let output = Command::new("ip")
        .args(["link", "set", "dev", name, "mtu", "1500"])
        .output()
        .map_err(|e| NetworkError::Bridge(format!("Failed to set MTU: {}", e)))?;

    if !output.status.success() {
        return Err(NetworkError::Bridge(format!(
            "Failed to set MTU for bridge {}: {}",
            name,
            String::from_utf8_lossy(&output.stderr)
        )));
    }

    // Bring bridge up
    let output = Command::new("ip")
        .args(["link", "set", "dev", name, "up"])
        .output()
        .map_err(|e| NetworkError::Bridge(format!("Failed to bring bridge up: {}", e)))?;

    if !output.status.success() {
        return Err(NetworkError::Bridge(format!(
            "Failed to bring bridge {} up: {}",
            name,
            String::from_utf8_lossy(&output.stderr)
        )));
    }

    Ok(Bridge {
        name: name.to_string(),
    })
}

/// Check if a bridge exists by checking /sys/class/net/{name}/bridge
pub fn bridge_exists(name: &str) -> Result<bool> {
    use std::path::Path;
    
    // Check if the bridge sysfs path exists - this is the most reliable way
    let bridge_path = format!("/sys/class/net/{}/bridge", name);
    Ok(Path::new(&bridge_path).exists())
}

/// Get an existing bridge
pub fn get_bridge(name: &str) -> Result<Bridge> {
    if !bridge_exists(name)? {
        return Err(NetworkError::NotFound(format!("Bridge {} not found", name)));
    }

    Ok(Bridge {
        name: name.to_string(),
    })
}

/// Delete a bridge
pub fn delete_bridge(name: &str) -> Result<()> {
    if !bridge_exists(name)? {
        return Ok(());
    }

    let output = Command::new("ip")
        .args(["link", "delete", name])
        .output()
        .map_err(|e| NetworkError::Bridge(format!("Failed to delete bridge: {}", e)))?;

    if !output.status.success() {
        return Err(NetworkError::Bridge(format!(
            "Failed to delete bridge {}: {}",
            name,
            String::from_utf8_lossy(&output.stderr)
        )));
    }

    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_bridge_struct_creation() {
        let bridge = Bridge {
            name: "test_bridge".to_string(),
        };
        assert_eq!(bridge.name, "test_bridge");
    }

    #[test]
    fn test_bridge_name_validation() {
        let valid_names = vec!["br0", "bridge1", "my-bridge", "test_br"];
        for name in valid_names {
            let bridge = Bridge {
                name: name.to_string(),
            };
            assert!(!bridge.name.is_empty());
        }
    }
}