pyro/rustbuster
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 5 Wiki Activity

added a progress bar its a real tool now!

Browse Source
This commit is contained in:
pyro
2026-05-18 12:39:01 -05:00
parent 9c3f4b2950
commit 0ea3ddea2b
2 changed files with 37 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Cargo.toml
+1
View File
@@ -9,6 +9,7 @@ edition = "2021"
clap = { version = "4.5.4", features = ["derive"] } clap = { version = "4.5.4", features = ["derive"] }
dns-lookup = "2.0.4" dns-lookup = "2.0.4"
futures = "0.3.32" futures = "0.3.32"
indicatif = "0.18.4"
reqwest = { version = "0.12.4", features = ["blocking"] } reqwest = { version = "0.12.4", features = ["blocking"] }
tokio = { version = "1.52.3", features = ["full"] } tokio = { version = "1.52.3", features = ["full"] }
trust-dns-resolver = "0.23.2" trust-dns-resolver = "0.23.2"
src/main.rs
+36 -20
View File
@@ -1,5 +1,7 @@
use clap::Parser; use clap::Parser;
use futures::{stream, StreamExt}; use futures::{stream, StreamExt};
use indicatif::ProgressBar;
use indicatif::ProgressStyle;
use reqwest::{self, StatusCode}; use reqwest::{self, StatusCode};
use std::fs::File; use std::fs::File;
use std::io::BufWriter; use std::io::BufWriter;
@@ -44,7 +46,7 @@ struct Args {
///input file full of domains formatted to one domain per line. ///input file full of domains formatted to one domain per line.
#[arg(short, long)] #[arg(short, long)]
domainfile: Option<PathBuf>, indomainfile: Option<PathBuf>,
} }
async fn try_sub( async fn try_sub(
@@ -53,6 +55,7 @@ async fn try_sub(
tx: Sender<OutputMessage>, tx: Sender<OutputMessage>,
output: bool, output: bool,
resolver: Arc<TokioAsyncResolver>, resolver: Arc<TokioAsyncResolver>,
pb: Arc<ProgressBar>,
) { ) {
let mut ips = Vec::new(); let mut ips = Vec::new();
if let Ok(ipv4s) = resolver.ipv4_lookup(&domain).await { if let Ok(ipv4s) = resolver.ipv4_lookup(&domain).await {
@@ -93,11 +96,12 @@ async fn try_sub(
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("{} {}", domain, ips.join(", ")); pb.println(&format!("{} {}", domain, ips.join(", ")));
} }
pb.inc(1);
} }
async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) { async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool, pb: Arc<ProgressBar>) {
let resp_stat = reqwest::get(&encode(&url).to_string()).await; let resp_stat = reqwest::get(&encode(&url).to_string()).await;
if resp_stat.is_ok() { if resp_stat.is_ok() {
let resp = resp_stat.unwrap().status(); let resp = resp_stat.unwrap().status();
@@ -111,7 +115,7 @@ async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("{} {}", resp, url); pb.println(&format!("{} {}", resp, url));
} }
StatusCode::ACCEPTED => { StatusCode::ACCEPTED => {
if output { if output {
@@ -122,7 +126,7 @@ async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("{} {}", resp, url); pb.println(&format!("{} {}", resp, url));
} }
StatusCode::CONTINUE => { StatusCode::CONTINUE => {
if output { if output {
@@ -133,7 +137,7 @@ async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("{} {}", resp, url); pb.println(&format!("{} {}", resp, url));
} }
StatusCode::CREATED => { StatusCode::CREATED => {
if output { if output {
@@ -144,7 +148,7 @@ async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("{} {}", resp, url); pb.println(&format!("{} {}", resp, url));
} }
StatusCode::FOUND => { StatusCode::FOUND => {
if output { if output {
@@ -155,7 +159,7 @@ async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("{} {}", resp, url); pb.println(&format!("{} {}", resp, url));
} }
StatusCode::IM_A_TEAPOT => { StatusCode::IM_A_TEAPOT => {
if output { if output {
@@ -166,7 +170,7 @@ async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("And what a beautiful teapot you are {}", url); pb.println(&format!("{} {}", resp, url));
} }
StatusCode::MOVED_PERMANENTLY => { StatusCode::MOVED_PERMANENTLY => {
if output { if output {
@@ -177,7 +181,7 @@ async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("{} {}", resp, url); pb.println(&format!("{} {}", resp, url));
} }
StatusCode::PERMANENT_REDIRECT => { StatusCode::PERMANENT_REDIRECT => {
if output { if output {
@@ -188,7 +192,7 @@ async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("{} {}", resp, url); pb.println(&format!("{} {}", resp, url));
} }
StatusCode::TEMPORARY_REDIRECT => { StatusCode::TEMPORARY_REDIRECT => {
if output { if output {
@@ -199,11 +203,12 @@ async fn try_dir(url: String, tx: Sender<OutputMessage>, output: bool) {
eprintln!("error sending output! {}", e); eprintln!("error sending output! {}", e);
} }
} }
println!("{} {}", resp, url); pb.println(&format!("{} {}", resp, url));
} }
_ => (), _ => (),
} }
} }
pb.inc(1);
} }
#[tokio::main] #[tokio::main]
@@ -298,7 +303,7 @@ async fn main() {
}); });
} }
} }
if let Some(domain_file) = args.domainfile { if let Some(domain_file) = args.indomainfile {
if let Ok(domain_string) = fs::read_to_string(domain_file) { if let Ok(domain_string) = fs::read_to_string(domain_file) {
domain_string.lines().into_iter().for_each(|domain| { domain_string.lines().into_iter().for_each(|domain| {
targets.push(domain.to_string()); targets.push(domain.to_string());
@@ -366,8 +371,8 @@ async fn main() {
} }
} }
println!("INFO:::loading targets, and wordlists...:::"); println!("INFO:::loading targets, and wordlists...:::");
let urls_len = dirs_to_try.len(); let urls_len = dirs_to_try.len() as u64;
let domains_len = subs_to_try.len(); let domains_len = subs_to_try.len() as u64;
println!( println!(
"INFO:::DONE!, {} URLS to try, and {} subdomains to try:::", "INFO:::DONE!, {} URLS to try, and {} subdomains to try:::",
&urls_len, &domains_len &urls_len, &domains_len
@@ -432,13 +437,22 @@ async fn main() {
println!("INFO:::results will be printed to console and not saved...:::"); println!("INFO:::results will be printed to console and not saved...:::");
} }
}); });
let total = urls_len + domains_len;
let pb = Arc::new(ProgressBar::new(total));
pb.set_style(
ProgressStyle::with_template("[{elapsed_precise}] {bar:40.orange/red} {pos}/{len} ({eta})")
.unwrap(),
);
let resolver = TokioAsyncResolver::tokio(ResolverConfig::default(), opts); let resolver = TokioAsyncResolver::tokio(ResolverConfig::default(), opts);
let resolver = Arc::new(resolver); let resolver = Arc::new(resolver);
pb.println("starting...");
pb.force_draw();
let dir_stream = stream::iter(dirs_to_try.into_iter().map(|url| { let dir_stream = stream::iter(dirs_to_try.into_iter().map(|url| {
let pb = pb.clone();
let tx = tx.clone(); let tx = tx.clone();
let output = output.clone(); let output = output.clone();
async move { async move {
try_dir(url, tx, output).await; try_dir(url, tx, output, pb).await;
} }
})); }));
let sub_stream = stream::iter(subs_to_try.into_iter().map(|sub| { let sub_stream = stream::iter(subs_to_try.into_iter().map(|sub| {
@@ -446,8 +460,9 @@ async fn main() {
let output = output.clone(); let output = output.clone();
let wc = wild_card_results.clone(); let wc = wild_card_results.clone();
let resolver = resolver.clone(); let resolver = resolver.clone();
let pb = pb.clone();
async move { async move {
try_sub(sub, wc, tx, output, resolver.clone()).await; try_sub(sub, wc, tx, output, resolver.clone(), pb).await;
} }
})); }));
@@ -461,11 +476,12 @@ async fn main() {
.buffer_unordered(concurrency) .buffer_unordered(concurrency)
.collect::<Vec<_>>() .collect::<Vec<_>>()
.await; .await;
println!("INFO:::enumeration finished!:::"); pb.println("INFO:::enumeration finished!:::");
if output { if output {
println!("INFO:::waiting for output writer to finish...:::"); pb.println("INFO:::waiting for output writer to finish...:::");
tx.send(OutputMessage::Shutdown).await.unwrap(); tx.send(OutputMessage::Shutdown).await.unwrap();
} }
write_handler.await.unwrap(); write_handler.await.unwrap();
println!("INFO:::done bruteforcing, happy hunting!:::"); pb.println("INFO:::done bruteforcing, happy hunting!:::");
pb.finish_with_message("Done!");
} }