From 8057fbc9cf26178a76859754fc83e8991c3d78b4 Mon Sep 17 00:00:00 2001
From: Edu4rdSHL <edu4rdshl@protonmail.com>
Date: Thu, 20 May 2021 00:46:54 -0500
Subject: [PATCH] Improve wildcard handling.

Signed-off-by: Edu4rdSHL <edu4rdshl@protonmail.com>
---
 src/main.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 6d34caa..c5cb0f7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -148,7 +148,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     let hosts: Vec<String> = if matches.is_present("domain") {
         let domain = value_t!(matches, "domain", String).unwrap();
-        wildcard_ips = detect_wildcards(&domain, &resolvers, quiet_flag).await;
+        wildcard_ips = detect_wildcards(&domain, &trustable_resolver, quiet_flag).await;
         buffer
             .lines()
             .map(|word| format!("{}.{}", word, domain))
@@ -253,9 +253,9 @@ async fn detect_wildcards(
         println!("Running wildcards detection for {}...\n", target)
     }
     let mut generated_wilcards: HashSet<String> = HashSet::new();
-    for _ in 1..10 {
+    for _ in 1..20 {
         generated_wilcards.insert(format!(
-            "{}.{}",
+            "{}.{}.",
             rng()
                 .sample_iter(Alphanumeric)
                 .take(15)