fatt 0.1.1

Find All The Things - A high-performance, distributed security scanning tool
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

use fatt::resolver::DnsResolver;
use tempfile::tempdir;

#[test]
fn test_resolver_creation() -> anyhow::Result<()> {
    // Create a temporary directory for DNS cache
    let _temp_dir = tempdir()?;
    // We're using a temporary directory now, no need for cache_path variable

    // Test for testing resolver
    let resolver = DnsResolver::new_for_testing()?;
    assert!(resolver.is_test_resolver());

    // Clean up
    _temp_dir.close()?;

    Ok(())
}

#[tokio::test]
async fn test_resolver_lookup() -> anyhow::Result<()> {
    // Create test resolver (should use mock responses)
    let resolver = DnsResolver::new_for_testing()?;

    // Test some domain lookups with the test resolver
    let first_result = resolver.lookup("example.com").await?;
    let second_result = resolver.lookup("test.example.org").await?;

    // Both lookups should succeed and have results
    assert!(first_result.is_some());
    assert!(second_result.is_some());

    // Make sure we got valid IP addresses
    if let Some(ip) = &first_result {
        assert!(!ip.is_empty(), "First lookup returned empty IP");
        assert_eq!(ip, "192.0.2.1"); // TEST-NET-1 address
    }

    if let Some(ip) = &second_result {
        assert!(!ip.is_empty(), "Second lookup returned empty IP");
        assert_eq!(ip, "192.0.2.1");
    }

    Ok(())
}

#[tokio::test]
async fn test_resolver_concurrency() -> anyhow::Result<()> {
    // Create test resolver with limited concurrency
    let resolver = DnsResolver::new_for_testing()?;

    // Perform multiple concurrent lookups
    let futures = vec![
        resolver.lookup("domain1.com"),
        resolver.lookup("domain2.com"),
        resolver.lookup("domain3.com"),
        resolver.lookup("domain4.com"),
        resolver.lookup("domain5.com"),
    ];

    // Run all lookups concurrently
    let results = futures::future::join_all(futures).await;

    // Verify all completed successfully
    for result in results {
        assert!(result.is_ok());
        let ips = result?;
        assert!(ips.is_some());
    }

    Ok(())
}

#[tokio::test]
async fn test_resolver_cache() -> anyhow::Result<()> {
    // Create a temporary directory for DNS cache
    let _temp_dir = tempdir()?;

    // Create resolver with caching
    let resolver = DnsResolver::new_for_testing()?;

    // Perform the same lookup multiple times to test caching
    let domain = "cached-example.com";

    // First lookup should not be cached
    let first_result = resolver.lookup(domain).await?;

    // Second lookup should use cache
    let second_result = resolver.lookup(domain).await?;

    // Both lookups should succeed and have results
    assert!(first_result.is_some());
    assert!(second_result.is_some());

    // Make sure we got valid IP addresses
    if let Some(ip) = &first_result {
        assert!(!ip.is_empty(), "First lookup returned empty IP");
    }

    if let Some(ip) = &second_result {
        assert!(!ip.is_empty(), "Second lookup returned empty IP");
    }

    // Results should be the same
    assert_eq!(first_result, second_result);

    Ok(())
}