cfd 0.1.0

CFD is a tool that allows you to check one or more domains to see if they are protected by CloudFlare or not. The check is carried out based on five criteria: 3 headers in the HTTP response, IP, and SSL certificate issuer. The check result can be displayed on the screen or saved to a file.
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

//! This structure joins domains and cf_ips together to execute checking tasks concurrently.

use crate::cf_ips::CFIPs;
use crate::domain::Domain;
use std::{error::Error, sync::Arc};
use tokio::sync::Mutex;

#[derive(Debug)]
pub struct Checker {
    pub domains: Vec<Arc<Mutex<Domain>>>,
    pub cf_ips: Arc<CFIPs>,
}

impl Checker {
    /// Build a new checker instance.
    /// The function takes one or several domains separated by newline characters as input
    /// #Example:
    /// ```
    /// use cfd::checker::Checker;
    /// #[tokio::main]
    /// async fn main(){
    ///     let solo_target = "example.com";
    ///     let multi_target = "example.com\nexample2.com\nexample3.com";
    ///     let solo_checker = Checker::build(solo_target.to_string()).await;
    ///     let multi_checker = Checker::build(solo_target.to_string()).await;
    ///     assert!(solo_checker.is_ok());
    ///     assert!(multi_checker.is_ok());
    /// }
    /// ```
    pub async fn build(target: String) -> Result<Self, Box<dyn Error>> {
        let target = target.split("\n").collect::<Vec<&str>>();
        let mut domains = vec![];
        target.iter().for_each(|domain| {
            if let Ok(domain) = Domain::build(domain.to_string()) {
                domains.push(Arc::new(Mutex::new(domain)));
            }
        });
        let cf_ips = CFIPs::load().await?;
        Ok(Self {
            domains,
            cf_ips: Arc::new(cf_ips),
        })
    }
}

impl Checker {
    /// Starts a check to determine if domains are behind CF.
    /// #Example:
    /// ```
    /// use cfd::checker::Checker;
    /// #[tokio::main]
    /// async fn main(){
    ///    let target = "cloudflare.com";
    ///    let mut checker = Checker::build(target.to_string()).await.unwrap();
    ///    checker.check().await.unwrap();
    ///    assert_eq!(checker.cf_detected_domains().await.len() > 0, true);
    /// }
    /// ```
    pub async fn check(&mut self) -> Result<(), Box<dyn Error>> {
        let mut handles = vec![];
        for domain in self.domains.iter_mut() {
            let cf_ips = self.cf_ips.clone();
            let domain = domain.clone();
            let handle = tokio::spawn(async move {
                domain.lock().await.verify_domain(cf_ips).await.unwrap();
            });
            handles.push(handle);
        }
        for handle in handles {
            handle.await?;
        }
        Ok(())
    }
}

impl Checker {
    /// Returns a vector of domains that are behind CF.
    /// #Example:
    /// ```
    /// use cfd::checker::Checker;
    /// #[tokio::main]
    /// async fn main(){
    ///    let target = "example.com\ncloudflare.com";
    ///    let mut checker = Checker::build(target.to_string()).await.unwrap();
    ///    checker.check().await.unwrap();
    ///    assert_eq!(checker.cf_detected_domains().await.len() == 1, true);
    /// }
    /// ```
    pub async fn cf_detected_domains(&self) -> Vec<Arc<Mutex<Domain>>> {
        let mut v = vec![];
        let iter = self.domains.iter();
        for domain in iter {
            let domain = domain.clone();
            if domain.lock().await.check_result != 0 {
                v.push(domain.clone());
            }
        }
        v
    }
}