edu4rdshl/fhc
1
0
Fork 0
mirror of https://github.com/edu4rdshl/fhc.git synced 2026-07-17 23:24:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add --retries support.

Browse source 
Fixes #1

Signed-off-by: Edu4rdSHL <edu4rdshl@protonmail.com>
This commit is contained in:
Edu4rdSHL 2021-05-09 05:12:56 -05:00
parent 4439d45a10
commit cd220f74ea
No known key found for this signature in database
GPG key ID: 3A574A4009F553E5
1 changed files with 33 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

34
src/main.rs
View file

@ -44,6 +44,13 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
.takes_value(true)
.help("Target domain."),
)
.arg(
Arg::with_name("retries")
.short("r")
.long("retries")
.takes_value(true)
.help("Max number of http probes per target."),
)
.arg(
Arg::with_name("1xx")
.short("1")
@ -96,6 +103,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
0
};
let threads = value_t!(matches.value_of("threads"), usize).unwrap_or_else(|_| 100);
let retries = value_t!(matches.value_of("retries"), usize).unwrap_or_else(|_| 1);
let timeout = value_t!(matches.value_of("timeout"), u64).unwrap_or_else(|_| 5);
let user_agents_list = utils::user_agents();
let show_status_codes = matches.is_present("show-codes");
@ -134,7 +142,31 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
async move {
let mut response_code = 0;
let mut valid_url = String::new();
if let Ok(resp) = https_send_fut.send().await {
if retries != 1 {
let mut counter = 0;
while counter < retries {
if let Ok(resp) = https_send_fut
.try_clone()
.expect("Failed to clone https future")
.send()
.await
{
valid_url = https_url.clone();
response_code = resp.status().as_u16();
drop(resp)
} else if let Ok(resp) = http_send_fut
.try_clone()
.expect("Failed to clone http future")
.send()
.await
{
valid_url = http_url.clone();
response_code = resp.status().as_u16();
drop(resp)
}
counter += 1
}
} else if let Ok(resp) = https_send_fut.send().await {
valid_url = https_url;
response_code = resp.status().as_u16();
drop(resp)