Fixed DNS resolving A records when -ports is set (#1347)

* Fixed DNS resolving A records when -ports is set * fixed ineffectual assignment * fmt --------- Co-authored-by: kzoltan <[email protected]> Co-authored-by: mzack <[email protected]>
projectdiscovery · Sep 13, 2023 · cb522ce · cb522ce
1 parent 9e26204
commit cb522ce
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/runner/runner.go b/runner/runner.go
@@ -1630,7 +1630,12 @@ retry:
 		builder.WriteString(fmt.Sprintf(" [%s]", ip))
 	}
 
-	ips, cnames, err := getDNSData(hp, URL.Host)
+	var onlyHost string
+	onlyHost, _, err = net.SplitHostPort(URL.Host)
+	if err != nil {
+		onlyHost = URL.Host
+	}
+	ips, cnames, err := getDNSData(hp, onlyHost)
 	if err != nil {
 		ips = append(ips, ip)
 	}