Update dangling detection

melbadry9 · Sep 10, 2021 · a28925e · a28925e
1 parent cfd4138
commit a28925e
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 15 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "sslenum"
-version = "1.0.0"
+version = "1.0.1"
 authors = ["melbadry9 <me@melbadry9.xyz>"]
 edition = "2018"
 description = "SSL certificate Extractor"

diff --git a/src/main.rs b/src/main.rs
@@ -24,25 +24,37 @@ struct DomainData {
 impl DomainData {
     fn check_dangling(self: &mut Self) {
         let domain = List.parse_dns_name(self.hostname.as_str()).unwrap();
-        let host = List.parse_dns_name(self.cn[0].as_str());
-        match host {
-            Ok(host) => {
-                let host_root = host.root();
-                match host_root {
-                    Some(host_root) => {
-                        if !(domain.root().unwrap() == host_root) {
+        let mut dns_names: Vec<String> = Vec::new();
+
+        dns_names.extend(self.cn.clone());
+        dns_names.extend(self.alt_names.clone());
+
+        for cand in dns_names {
+            let host = List.parse_dns_name(cand.as_str());
+            match host {
+                Ok(host) => {
+                    let host_root = host.root();
+                    match host_root {
+                        Some(host_root) => {
+                            if !(domain.root().unwrap() == host_root) {
+                                self.dangling = true;
+                            } else {
+                                self.dangling = false;
+                                break;
+                            }
+                        }
+                        None => {
                             self.dangling = true;
                         }
                     }
-                    None => {
-                        self.dangling = true;
-                    }
+                }
+                Err(_) => {
+                    self.dangling = false;
                 }
             }
-            Err(_) => {
-                self.dangling = false;
-            }
+
         }
+
     }
 }