diff --git a/main.go b/main.go index 273b9cf1..b7ea7acd 100644 --- a/main.go +++ b/main.go @@ -163,6 +163,7 @@ func GenerateSpeedtest(path string) error { } req.Header.Set("Accept-Encoding", "gzip") + req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/44.0.2403.157 Safari/537.36") resp, err := http.DefaultClient.Do(req) if err != nil { return err @@ -183,6 +184,10 @@ func GenerateSpeedtest(path string) error { reg := regexp.MustCompile(`host="(.+):[0-9]+"`) matchList := reg.FindAllStringSubmatch(string(body), -1) + if len(matchList) == 0 { + return errors.New("cannot get ookla-speedtest domains") + } + exist := make(map[string]bool) var domainList []string for _, match := range matchList { @@ -204,6 +209,7 @@ func GenerateSpeedtest(path string) error { exist[domain] = true } sort.Strings(domainList) + fmt.Printf("%ddomains added.\n", len(domainList)) fPath := filepath.Join(path, "ookla-speedtest") b := append([]byte("include:ookla-speedtest-ads\n"), []byte(strings.Join(domainList, "\n"))...)