format code.
This commit is contained in:
39
src/main.rs
39
src/main.rs
@@ -1,12 +1,11 @@
|
||||
use std::net::{SocketAddr, IpAddr, ToSocketAddrs};
|
||||
use std::net::TcpStream;
|
||||
use std::{process, thread};
|
||||
use std::fmt::{Debug};
|
||||
use std::time::Duration;
|
||||
use chrono::Local;
|
||||
use clap::{Parser};
|
||||
use clap::Parser;
|
||||
use dns_lookup::lookup_host;
|
||||
|
||||
use std::fmt::Debug;
|
||||
use std::net::TcpStream;
|
||||
use std::net::{IpAddr, SocketAddr, ToSocketAddrs};
|
||||
use std::time::Duration;
|
||||
use std::{process, thread};
|
||||
|
||||
/// Simple program to greet a person
|
||||
#[derive(Parser, Debug)]
|
||||
@@ -73,16 +72,16 @@ fn get_target_socket(args: &Args) -> Result<SocketAddr, String> {
|
||||
if let Ok(sock_iter) = args.host.to_socket_addrs() {
|
||||
for sock in sock_iter {
|
||||
if !args.v4 && !args.v6 {
|
||||
return Ok(sock)
|
||||
return Ok(sock);
|
||||
} else if args.v4 && sock.is_ipv4() {
|
||||
return Ok(sock)
|
||||
return Ok(sock);
|
||||
} else if args.v6 && sock.is_ipv6() {
|
||||
return Ok(sock);
|
||||
}
|
||||
}
|
||||
Err(format!("Error: Cannot parse host"))
|
||||
} else if let Ok(target_result) = get_target_ip(&args.host, &args.v4, &args.v6) {
|
||||
return Ok(SocketAddr::new(target_result, args.port))
|
||||
return Ok(SocketAddr::new(target_result, args.port));
|
||||
} else {
|
||||
Err(format!("Error: Unable to resolve Host"))
|
||||
}
|
||||
@@ -101,22 +100,26 @@ fn get_target_ip(s: &str, v4: &bool, v6: &bool) -> Result<IpAddr, String> {
|
||||
return Ok(ip as IpAddr);
|
||||
}
|
||||
}
|
||||
Err(format!(
|
||||
"Failed to Lookup IP",
|
||||
))
|
||||
Err(format!("Failed to Lookup IP",))
|
||||
} else {
|
||||
Err(format!(
|
||||
"Unable to resolve Address",
|
||||
))
|
||||
Err(format!("Unable to resolve Address",))
|
||||
}
|
||||
}
|
||||
|
||||
fn ping(target: &SocketAddr, timeout: &Duration) -> Result<(), ()> {
|
||||
if let Ok(_stream) = TcpStream::connect_timeout(&target, *timeout) {
|
||||
println!("{} - {} - Connection Successful", Local::now().format("%Y-%m-%d %H:%M:%S"), target);
|
||||
println!(
|
||||
"{} - {} - Connection Successful",
|
||||
Local::now().format("%Y-%m-%d %H:%M:%S"),
|
||||
target
|
||||
);
|
||||
Ok(())
|
||||
} else {
|
||||
println!("{} - {} - Connection Failed", Local::now().format("%Y-%m-%d %H:%M:%S"), target);
|
||||
println!(
|
||||
"{} - {} - Connection Failed",
|
||||
Local::now().format("%Y-%m-%d %H:%M:%S"),
|
||||
target
|
||||
);
|
||||
Err(())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user