From 278ef8da0408c1bb42be48ace010f7e75a6c3ac2 Mon Sep 17 00:00:00 2001
From: Jeremy Wall <jeremy@marzhillstudios.com>
Date: Sat, 20 Feb 2021 19:36:25 -0500
Subject: [PATCH] Allow ip addresses in the pingHosts list

---
 src/util.rs | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/util.rs b/src/util.rs
index d042da0..cab81ec 100644
--- a/src/util.rs
+++ b/src/util.rs
@@ -32,9 +32,16 @@ pub fn resolve_hosts<'a>(servers: &'a Vec<&str>) -> io::Result<Vec<Option<IpAddr
     config.use_inet6 = ALLOWIPV6.flag;
     let resolver = DnsResolver::new(config)?;
     for name in servers.iter().cloned() {
-        // TODO for resolution errors return a more valid error with the domain name.
-        let mut iter = resolver.resolve_host(name)?;
-        results.push(iter.next());
+        match name.parse::<IpAddr>() {
+            Ok(ip) => {
+                results.push(Some(ip));
+            }
+            Err(_) => {
+                // TODO for resolution errors return a more valid error with the domain name.
+                let mut iter = resolver.resolve_host(name)?;
+                results.push(iter.next());
+            }
+        }
     }
     return Ok(results);
 }