Delete Cargo.lock

.gitignore

@@ -0,0 +1,24 @@
+# Generated by Cargo
+# will have compiled files and executables
+debug
+target
+
+# These are backup files generated by rustfmt
+**/*.rs.bk
+
+# MSVC Windows builds of rustc generate these, which store debugging information
+*.pdb
+
+# Generated by cargo mutants
+# Contains mutation testing data
+**/mutants.out*/
+
+# rustc will dump stack traces when hitting an internal compiler error to PWD
+rustc-ice-*.txt
+
+# RustRover
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/

Cargo.toml

@@ -8,4 +8,6 @@ edition = "2021"
 [dependencies]
 clap = { version = "4.5.4", features = ["derive"] }
 dns-lookup = "2.0.4"
+rayon = "1.12.0"
 reqwest = { version = "0.12.4", features = ["blocking"] }
+trust-dns-resolver = "0.23.2"

src/main.rs

@@ -1,9 +1,23 @@
-use std::fs;
-use reqwest::{self, StatusCode};
 use clap::Parser;
+use rayon::iter::{IntoParallelRefIterator, ParallelIterator};
+use rayon::ThreadPoolBuilder;
+use reqwest::{self, StatusCode};
+use std::fs::File;
+use std::io::BufWriter;
+use std::io::Write;
+use std::path::PathBuf;
+use std::process::exit;
+use std::sync::mpsc::{channel, Sender};
 use std::thread;
-use dns_lookup::lookup_host;
+use std::{collections::HashMap, fs};
+use trust_dns_resolver::config::*;
+use trust_dns_resolver::Resolver;
 
+enum OutputMessage {
+    UrlResult(String),
+    SubDomainResult(String),
+    Shutdown,
+}
 
 /// Subdomain Bruteforcer, and directory bruteforcer written in rust, and multi threaded!!!
 #[derive(Parser, Debug)]
@@ -11,11 +25,7 @@ use dns_lookup::lookup_host;
 struct Args {
     /// target to attack, can be a URL or a dns name, you can specify more than one by comma separating the targets
     #[arg(short, long)]
-    target: String,
-
-    ///Threads to use, there is not upper limit so if you lock up your comptuer that's on you
-    #[arg(long)]
-    threads: usize,
+    target: Option<String>,
 
     /// wordlist to use for subdomains
     #[arg(short, long, default_value_t = String::from("none"))]
@@ -24,42 +34,158 @@ struct Args{
     ///wordlist to use for directories
     #[arg(short, long, default_value_t = String::from("none"))]
     dirwordlist: String,
+
+    ///output files to write, will create two files yourinput_urls.txt and yourinput_subs.txt
+    #[arg(short, long)]
+    outfile: Option<String>,
+
+    ///input file full of urls formatted to one url per line.
+    #[arg(short, long)]
+    urlfile: Option<PathBuf>,
+
+    ///input file full of domains formatted to one domain per line.
+    #[arg(short, long)]
+    domainfile: Option<PathBuf>,
+
+    ///number of threads to use by default it will use the rayon global pool default.
+    #[arg(long)]
+    threads: Option<usize>,
 }
 
-
-fn try_sub(domain: String){
-    let ips = lookup_host(&domain);
-    if ips.is_ok(){
-        let mut ip_string = String::new();
-        for ip in ips.unwrap(){
-            ip_string = format!("{},{}",ip, ip_string);
+fn try_sub(
+    domain: String,
+    wildcard_reses: HashMap<&String, Vec<String>>,
+    tx: Sender<OutputMessage>,
+    output: bool,
+) {
+    let mut opts = ResolverOpts::default();
+    opts.edns0 = true;
+    let resolver = Resolver::new(ResolverConfig::default(), opts).unwrap();
+    let mut ips = Vec::new();
+    if let Ok(ipv4s) = resolver.ipv4_lookup(&domain) {
+        ipv4s.iter().for_each(|ip| ips.push(ip.0.to_string()));
     }
-        println!("{} {}", domain, ip_string);
+    if let Ok(ipv6s) = resolver.ipv6_lookup(&domain) {
+        ipv6s.iter().for_each(|ip| ips.push(ip.0.to_string()));
+    }
+    let mut wild_card = None;
+    for target in wildcard_reses.keys() {
+        if domain.contains(&target.to_string()) {
+            wild_card = wildcard_reses.get(target);
+        }
+    }
+    if !ips.is_empty() {
+        if let Some(wild_ips) = wild_card {
+            if ips.len() == wild_ips.len() {
+                let mut wild_only = true;
+                for ip in &ips {
+                    if !wild_ips.contains(ip) {
+                        wild_only = false;
+                    }
+                }
+                if wild_only {
+                    return;
+                }
+            }
+        }
+        if output {
+            if let Err(e) = tx.send(OutputMessage::SubDomainResult(format!(
+                "{}: {}",
+                &domain,
+                ips.join(", ")
+            ))) {
+                eprintln!("error sending output! {}", e);
+            }
+        }
+        println!("{} {}", domain, ips.join(", "));
     }
 }
 
-
-fn try_dir(url: String){
+fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
     let resp_stat = reqwest::blocking::get(&url);
     if resp_stat.is_ok() {
         let resp = resp_stat.unwrap().status();
         match resp {
-            StatusCode::OK => println!("{} {}",resp, url),
-            StatusCode::ACCEPTED => println!("{} {}", resp, url),
-            StatusCode::CONTINUE => println!("{} {}", resp, url),
-            StatusCode::CREATED => println!("{} {}", resp, url),
-            StatusCode::FOUND => println!("{} {}", resp, url),
-            StatusCode::IM_A_TEAPOT => println!("And what a beautiful teapot you are {}", url),
-            StatusCode::MOVED_PERMANENTLY => println!("{} {}", resp, url),
-            StatusCode::PERMANENT_REDIRECT => println!("{} {}", resp, url),
-            StatusCode::TEMPORARY_REDIRECT => println!("{} {}", resp, url),
-            _ => ()
+            StatusCode::OK => {
+                if output {
+                    if let Err(e) = tx.send(OutputMessage::UrlResult(format!("{} {}", resp, url))) {
+                        eprintln!("error sending output! {}", e);
+                    }
+                }
+                println!("{} {}", resp, url);
+            }
+            StatusCode::ACCEPTED => {
+                if output {
+                    if let Err(e) = tx.send(OutputMessage::UrlResult(format!("{} {}", resp, url))) {
+                        eprintln!("error sending output! {}", e);
+                    }
+                }
+                println!("{} {}", resp, url);
+            }
+            StatusCode::CONTINUE => {
+                if output {
+                    if let Err(e) = tx.send(OutputMessage::UrlResult(format!("{} {}", resp, url))) {
+                        eprintln!("error sending output! {}", e);
+                    }
+                }
+                println!("{} {}", resp, url);
+            }
+            StatusCode::CREATED => {
+                if output {
+                    if let Err(e) = tx.send(OutputMessage::UrlResult(format!("{} {}", resp, url))) {
+                        eprintln!("error sending output! {}", e);
+                    }
+                }
+                println!("{} {}", resp, url);
+            }
+            StatusCode::FOUND => {
+                if output {
+                    if let Err(e) = tx.send(OutputMessage::UrlResult(format!("{} {}", resp, url))) {
+                        eprintln!("error sending output! {}", e);
+                    }
+                }
+                println!("{} {}", resp, url);
+            }
+            StatusCode::IM_A_TEAPOT => {
+                if output {
+                    if let Err(e) = tx.send(OutputMessage::UrlResult(format!("{} {}", resp, url))) {
+                        eprintln!("error sending output! {}", e);
+                    }
+                }
+                println!("And what a beautiful teapot you are {}", url);
+            }
+            StatusCode::MOVED_PERMANENTLY => {
+                if output {
+                    if let Err(e) = tx.send(OutputMessage::UrlResult(format!("{} {}", resp, url))) {
+                        eprintln!("error sending output! {}", e);
+                    }
+                }
+                println!("{} {}", resp, url);
+            }
+            StatusCode::PERMANENT_REDIRECT => {
+                if output {
+                    if let Err(e) = tx.send(OutputMessage::UrlResult(format!("{} {}", resp, url))) {
+                        eprintln!("error sending output! {}", e);
+                    }
+                }
+                println!("{} {}", resp, url);
+            }
+            StatusCode::TEMPORARY_REDIRECT => {
+                if output {
+                    if let Err(e) = tx.send(OutputMessage::UrlResult(format!("{} {}", resp, url))) {
+                        eprintln!("error sending output! {}", e);
+                    }
+                }
+                println!("{} {}", resp, url);
+            }
+            _ => (),
         }
     }
 }
 
 fn main() {
-    print!("
+    print!(
+        "
 
 
 
@@ -129,74 +255,189 @@ fn main() {
      ╙█▄▄▄▀▀▀▓▄▄▄▄▄▄▄▓▀▀
 
 
-");
+"
+    );
     let mut dirs_to_try = Vec::new();
     let mut subs_to_try = Vec::new();
     let args = Args::parse();
     let mut targets = Vec::new();
-    for target in args.target.split(",").collect::<Vec<&str>>(){
+    let (tx, rx) = channel();
+    if let Some(given_targets) = args.target {
+        for target in given_targets.split(",").collect::<Vec<&str>>() {
             targets.push(target.to_owned());
         }
+    }
+    if let Some(url_file) = args.urlfile {
+        if let Ok(url_string) = fs::read_to_string(url_file) {
+            url_string.lines().into_iter().for_each(|url| {
+                targets.push(url.to_string());
+            });
+        }
+    }
+    if let Some(domain_file) = args.domainfile {
+        if let Ok(domain_string) = fs::read_to_string(domain_file) {
+            domain_string.lines().into_iter().for_each(|domain| {
+                targets.push(domain.to_string());
+            });
+        }
+    }
+    if targets.is_empty() {
+        println!("no targets provided!");
+        exit(1);
+    }
+    if let Some(thread) = args.threads {
+        ThreadPoolBuilder::new()
+            .num_threads(thread)
+            .build_global()
+            .unwrap();
+    }
+    println!(
+        "INFO:::{} threads will be used:::",
+        rayon::current_num_threads()
+    );
+    println!("detecting if a wild card record exists...");
+    let mut opts = ResolverOpts::default();
+    opts.edns0 = true;
+    let resolver = Resolver::new(ResolverConfig::default(), opts).unwrap();
+    let mut wild_card_results = HashMap::new();
+    for target in &targets {
+        if !target.contains("http") {
+            subs_to_try.push(target.clone());
+            let mut ips = Vec::new();
+            if let Ok(ipv4s) = resolver.ipv4_lookup(&format!("burstpyrofoo.{}", target)) {
+                ipv4s.iter().for_each(|ip| {
+                    ips.push(ip.0.to_string());
+                });
+                println!("INFO:::wildcard found: {} {}:::", target, ips.join(", "));
+            }
+            if let Ok(ipv6s) = resolver.ipv6_lookup(&format!("burstpyrofoo.{}", target)) {
+                println!("wildcard found!");
+                ipv6s.iter().for_each(|ip| {
+                    ips.push(ip.0.to_string());
+                });
+            }
+            if !ips.is_empty() {
+                wild_card_results.insert(target, ips);
+            }
+        } else {
+            dirs_to_try.push(target.clone());
+        }
+    }
     if args.dirwordlist != String::from("none") {
-        let dirwordlist = fs::read_to_string(args.dirwordlist).expect("error reading directory wordlist");
+        let dirwordlist =
+            fs::read_to_string(args.dirwordlist).expect("error reading directory wordlist");
         for dir in dirwordlist.split("\n").collect::<Vec<&str>>() {
-            dirs_to_try.push(dir.trim_end().trim_start().to_owned());
+            for target in &targets {
+                if target.contains("http") {
+                    dirs_to_try.push(format!("{}/{}", target, dir.trim()));
+                }
+            }
         }
     }
     if args.subwordlist != String::from("none") {
-        let subwordlist = fs::read_to_string(args.subwordlist).expect("error reading subdomain word list");
+        let subwordlist =
+            fs::read_to_string(args.subwordlist).expect("error reading subdomain word list");
         for sub in subwordlist.split("\n").collect::<Vec<&str>>() {
-            subs_to_try.push(sub.trim_end().trim_start().to_owned())
-        }
-    }
-    let mut domains_to_try = Vec::new();
-    let mut urls_to_try = Vec::new();
-    println!("loading targets, and wordlists...");
-    for target in targets{
-        if target.contains("/"){
-            for dir in &dirs_to_try{
-                let url = format!("{}/{}", target, dir);
-                urls_to_try.push(url.to_owned());
-            }
-        }
-        else if target.contains(".") == true{
-            for sub in &subs_to_try{
-                let domain = format!("{}.{}", sub, target);
-                domains_to_try.push(domain.to_owned());
-            }
-        }
-        else{
-            println!("{} is not a valid target, please supply either a domain or url", target);
-        }
-    }
-    let mut threads = Vec::new();
-    let urls_len = urls_to_try.len();
-    let domains_len = domains_to_try.len();
-    println!("DONE!, {} URLS to try, and {} subdomains to try", &urls_len, &domains_len);
-    if urls_len > 0{
-        let dir_chunk_size = urls_len / &args.threads;
-        let url_vecs: Vec<Vec<String>> = urls_to_try.chunks(dir_chunk_size).map(|x| x.to_vec()).collect();
-        for urlvec in url_vecs{
-            for url in urlvec{
-                threads.push(thread::spawn(move || {
-                    try_dir(url)
-                }));
+            for target in &targets {
+                if !target.contains("http") {
+                    subs_to_try.push(format!("{}.{}", sub.trim(), target));
                 }
             }
         }
-    if domains_len > 0 {
-        let domain_chunk_size = domains_len /&args.threads;
-        let domain_vecs: Vec<Vec<String>> = domains_to_try.chunks(domain_chunk_size).map(|x| x.to_vec()).collect();
-        for domainvec in domain_vecs{
-            for domain in domainvec{
-                threads.push(thread::spawn(move || {
-                    try_sub(domain);
-                }))
+    }
+    println!("INFO:::loading targets, and wordlists...:::");
+    let urls_len = dirs_to_try.len();
+    let domains_len = subs_to_try.len();
+    println!(
+        "INFO:::DONE!, {} URLS to try, and {} subdomains to try:::",
+        &urls_len, &domains_len
+    );
+    let mut output = false;
+    let mut urlpath = PathBuf::new();
+    let mut subpath = PathBuf::new();
+    if args.outfile.is_some() {
+        output = true;
+        let path = args.outfile.clone().unwrap();
+        urlpath = PathBuf::from(format!("{}_urls.txt", &path));
+        subpath = PathBuf::from(format!("{}_subs.txt", &path));
+        println!(
+            "INFO:::results will be saved to {}_urls.txt and {}_subs.txt:::",
+            path, path
+        );
+    }
+    let write_handler = thread::spawn(move || {
+        if output {
+            let urlfile = File::create(urlpath).unwrap();
+            let subfile = File::create(subpath).unwrap();
+            let mut urlwrite = BufWriter::new(urlfile);
+            let mut subwrite = BufWriter::new(subfile);
+            const FLUSHEVERY: usize = 10;
+            let mut url_counter = 0;
+            let mut sub_counter = 0;
+            loop {
+                let msg = rx.recv().unwrap();
+                match msg {
+                    OutputMessage::UrlResult(res) => {
+                        if let Err(e) = writeln!(urlwrite, "{}", res) {
+                            println!("error writing url file! {}", e);
+                        } else {
+                            url_counter += 1;
                         }
                     }
-        for thread in threads{
-            let _ = thread.join();
+                    OutputMessage::SubDomainResult(res) => {
+                        if let Err(e) = writeln!(subwrite, "{}", res) {
+                            println!("error writing sub file! {}", e);
+                        } else {
+                            sub_counter += 1;
                         }
                     }
-    println!("done bruteforcing, happy hunting!");
+                    OutputMessage::Shutdown => {
+                        urlwrite.flush().unwrap();
+                        subwrite.flush().unwrap();
+                        break;
+                    }
+                }
+                if url_counter == FLUSHEVERY {
+                    if let Err(e) = urlwrite.flush() {
+                        println!("error writing url file! {e}");
+                    }
+                }
+                if sub_counter == FLUSHEVERY {
+                    if let Err(e) = subwrite.flush() {
+                        println!("error writing sub file! {e}");
+                    }
+                }
+            }
+        } else {
+            println!("INFO:::results will be printed to console and not saved...:::");
+        }
+    });
+    rayon::scope(|s| {
+        if !dirs_to_try.is_empty() {
+            s.spawn(|_| {
+                dirs_to_try.par_iter().for_each(|url| {
+                    try_dir(url.to_string(), tx.clone(), output);
+                });
+            });
+        }
+        if !subs_to_try.is_empty() {
+            s.spawn(|_| {
+                subs_to_try.par_iter().for_each(|sub| {
+                    try_sub(
+                        sub.to_string(),
+                        wild_card_results.clone(),
+                        tx.clone(),
+                        output,
+                    );
+                });
+            })
+        }
+    });
+    println!("INFO:::enumeration finished!:::");
+    if output {
+        println!("INFO:::waiting for output writer to finish...:::");
+        tx.send(OutputMessage::Shutdown).unwrap();
+    }
+    write_handler.join().unwrap();
+    println!("INFO:::done bruteforcing, happy hunting!:::");
 }