Buriburizaemon/nezha_domains
1
0
Fork 0
mirror of https://github.com/Buriburizaem0n/nezha_domains.git synced 2026-02-05 05:00:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

🚸 [agent] get geoip with UA

Browse Source
This commit is contained in:
naiba
2023-03-30 17:17:06 +08:00
parent cda39cac27
commit 3e9705d6f4
3 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
cmd/agent/monitor/myip.go
View File

@@ -7,6 +7,8 @@ import (
"strings"
"time"
fakeUa "github.com/EDDYCJY/fake-useragent"
"github.com/naiba/nezha/pkg/utils"
)
@@ -78,12 +80,13 @@ func fetchGeoIP(servers []string, isV6 bool) geoIP {
var ip geoIP
var resp *http.Response
var err error
// 双栈支持参差不齐，不能随机请求，有些 IPv6 取不到 IP
for i := 0; i < len(servers); i++ {
if isV6 {
resp, err = httpClientV6.Get(servers[i])
resp, err = httpGetWithUA(httpClientV6, servers[i])
} else {
resp, err = httpClientV4.Get(servers[i])
resp, err = httpGetWithUA(httpClientV4, servers[i])
}
if err == nil {
body, err := io.ReadAll(resp.Body)
@@ -117,6 +120,6 @@ func httpGetWithUA(client *http.Client, url string) (*http.Response, error) {
if err != nil {
return nil, err
}
req.Header.Add("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.84 Safari/537.36")
req.Header.Add("User-Agent", fakeUa.Random())
return client.Do(req)
}