format code.

This commit is contained in:
Peter Dwyer
2022-07-06 10:24:34 +01:00
parent 2b4bb6a49d
commit 5dfc991d93

View File

@@ -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))
} else if let Ok(target_result) = get_target_ip(&args.host, &args.v4, &args.v6) {
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(())
}
}
}