hpz937/zurg
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix ipv6 stuffs 8

Browse Source
This commit is contained in:
Ben Sarmiento
2024-01-11 03:43:31 +01:00
parent c4fd7358b9
commit e9e4a7dfc2
1 changed files with 6 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
pkg/http/client.go
View File

@@ -208,31 +208,23 @@ func (r *HTTPClient) replaceHostIfNeeded(req *http.Request) {
host := req.URL.Host
if strings.HasSuffix(host, ".com") {
newHost := strings.Replace(host, ".com", ".cloud", 1)
newURL := *req.URL
newURL.Host = newHost
req.URL = &newURL
req.Host = newHost
req.URL.Host = newHost
}
return
}
// if hosts are found, ensure the host is an IPv6 host
host, _, err := net.SplitHostPort(req.URL.Host)
if err != nil {
host = req.URL.Host // Use the host without port
}
found := false
for _, h := range r.ipv6Hosts {
if h == host {
if h == req.URL.Host {
found = true
break
}
}
// if host is not an IPv6 host, replace it with a random IPv6 host
if !found {
r.log.Warnf("Host %s is not an IPv6 host (ensure you have preferred_hosts properly set in your config.yml, if unset, run `zurg network-test -t ipv6`)", host)
r.log.Warnf("Host %s is not an IPv6 host (ensure you have preferred_hosts properly set in your config.yml, if unset, run `zurg network-test -t ipv6`)", req.URL.Host)
newHost := r.ipv6Hosts[rand.Intn(len(r.ipv6Hosts))]
newURL := *req.URL
newURL.Host = newHost
req.URL = &newURL
r.log.Warnf("Replaced host %s with %s", host, newHost)
req.Host = newHost
req.URL.Host = newHost
}
}