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 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(())
} }
} }