fix: filter for Ipv4 addresses when resolving hostnames

fujiapple852 · May 9, 2022 · 44554fc · 44554fc
1 parent a02b1c1
commit 44554fc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main.rs b/src/main.rs
@@ -80,7 +80,10 @@ fn main() -> anyhow::Result<()> {
         .map(|target| {
             let target_addr: IpAddr = resolver
                 .lookup(target)
-                .map_err(|e| anyhow!("failed to resolve target: {} ({})", target, e))?[0];
+                .map_err(|e| anyhow!("failed to resolve target: {} ({})", target, e))?
+                .into_iter()
+                .find(|addr| matches!(addr, IpAddr::V4(_)))
+                .unwrap();
             let trace_data = Arc::new(RwLock::new(Trace::new(args.tui_max_samples)));
             Ok(TuiTraceInfo::new(
                 trace_data,