snortal 1.0.18

Detect captive portal URLs on a Linux network
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

use crate::types::{PortalUrl, Source};

const DEFAULT_LEASE_FILES: &[&str] = &[
    "/var/lib/dhcp/dhclient.leases",         // Debian/Ubuntu
    "/var/lib/dhclient/dhclient.leases",     // RHEL/Fedora/CentOS
    "/var/lib/dhclient/dhclient6.leases",
    "/var/lib/NetworkManager/dhclient.conf", // NM managed
];

/// `files` overrides the built-in lease file list when non-empty.
pub async fn detect(files: &[String]) -> Vec<PortalUrl> {
    detect_inner(files).await.unwrap_or_default()
}

async fn detect_inner(
    files: &[String],
) -> Result<Vec<PortalUrl>, Box<dyn std::error::Error + Send + Sync>> {
    let candidates: Vec<&str> = if files.is_empty() {
        DEFAULT_LEASE_FILES.to_vec()
    } else {
        files.iter().map(String::as_str).collect()
    };

    let mut results = Vec::new();
    for path in &candidates {
        if let Ok(content) = tokio::fs::read_to_string(path).await {
            for line in content.lines() {
                let trimmed = line.trim();
                if trimmed.starts_with("option dhcp-captive-portal-uri") {
                    if let Some(url) = extract_quoted(trimmed) {
                        if url::Url::parse(url).is_ok() {
                            results.push(PortalUrl {
                                url: url.to_owned(),
                                source: Source::DhcpLease,
                                confidence: 95,
                            });
                        }
                    }
                }
            }
        }
    }

    // De-duplicate, keep last occurrence (most recent lease)
    results.dedup_by(|a, b| a.url == b.url);
    Ok(results)
}

fn extract_quoted(line: &str) -> Option<&str> {
    let start = line.find('"')? + 1;
    let end = line.rfind('"')?;
    if start < end { Some(&line[start..end]) } else { None }
}

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

    #[test]
    fn parses_quoted_uri() {
        let line = r#"  option dhcp-captive-portal-uri "http://192.168.1.1/portal";"#;
        assert_eq!(extract_quoted(line), Some("http://192.168.1.1/portal"));
    }

    #[test]
    fn empty_string_returns_none() {
        assert_eq!(extract_quoted("option dhcp-captive-portal-uri"), None);
    }
}