format code.
This commit is contained in:
43
src/main.rs
43
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 chrono::Local;
|
||||||
use clap::{Parser};
|
use clap::Parser;
|
||||||
use dns_lookup::lookup_host;
|
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
|
/// Simple program to greet a person
|
||||||
#[derive(Parser, Debug)]
|
#[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() {
|
if let Ok(sock_iter) = args.host.to_socket_addrs() {
|
||||||
for sock in sock_iter {
|
for sock in sock_iter {
|
||||||
if !args.v4 && !args.v6 {
|
if !args.v4 && !args.v6 {
|
||||||
return Ok(sock)
|
return Ok(sock);
|
||||||
} else if args.v4 && sock.is_ipv4() {
|
} else if args.v4 && sock.is_ipv4() {
|
||||||
return Ok(sock)
|
return Ok(sock);
|
||||||
} else if args.v6 && sock.is_ipv6() {
|
} else if args.v6 && sock.is_ipv6() {
|
||||||
return Ok(sock);
|
return Ok(sock);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Err(format!("Error: Cannot parse host"))
|
Err(format!("Error: Cannot parse host"))
|
||||||
} else if let Ok(target_result) = get_target_ip(&args.host, &args.v4, &args.v6){
|
} 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 {
|
} else {
|
||||||
Err(format!("Error: Unable to resolve Host"))
|
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);
|
return Ok(ip as IpAddr);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Err(format!(
|
Err(format!("Failed to Lookup IP",))
|
||||||
"Failed to Lookup IP",
|
|
||||||
))
|
|
||||||
} else {
|
} else {
|
||||||
Err(format!(
|
Err(format!("Unable to resolve Address",))
|
||||||
"Unable to resolve Address",
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn ping(target: &SocketAddr, timeout: &Duration) -> Result<(), ()> {
|
fn ping(target: &SocketAddr, timeout: &Duration) -> Result<(), ()> {
|
||||||
if let Ok(_stream) = TcpStream::connect_timeout(&target, *timeout) {
|
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(())
|
Ok(())
|
||||||
} else {
|
} 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(())
|
Err(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user