新增 HTTPing 延迟测速模式(beta); 新增 IP 机场三字码 colo 筛选功能(beta) (#282)

这两个功能目前仅为测试版，后续会大幅改动，以最终成品为准~

task/httping.go

@@ -0,0 +1,140 @@
+package task
+
+import (
+	"crypto/tls"
+	"fmt"
+	"io"
+	"log"
+	"net"
+	"net/http"
+	"strings"
+	"sync"
+	"time"
+
+	"CloudflareSpeedTest/utils"
+)
+
+var (
+	Httping        bool   //是否启用httping
+	HttpingTimeout int    //设置超时时间，单位毫秒
+	HttpingColo    string //有值代表筛选机场三字码区域
+)
+
+var (
+	HttpingColomap *sync.Map
+	HttpingRequest *http.Request
+)
+
+// pingReceived pingTotalTime
+func (p *Ping) httping(ip *net.IPAddr) (int, time.Duration) {
+	var fullAddress string
+	if isIPv4(ip.String()) {
+		fullAddress = fmt.Sprintf("%s", ip.String())
+	} else {
+		fullAddress = fmt.Sprintf("[%s]", ip.String())
+	}
+	hc := http.Client{
+		Timeout: time.Duration(HttpingTimeout) * time.Millisecond,
+		Transport: &http.Transport{
+			TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
+		},
+	} // #nosec
+
+	traceURL := fmt.Sprintf("http://%s/cdn-cgi/trace",
+		fullAddress)
+
+	// for connect and get colo
+	{
+		requ, err := http.NewRequest(http.MethodHead, traceURL, nil)
+		if err != nil {
+			return 0, 0
+		}
+		requ.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.80 Safari/537.36")
+		resp, err := hc.Do(requ)
+		if err != nil {
+			return 0, 0
+		}
+		defer resp.Body.Close()
+		io.Copy(io.Discard, resp.Body)
+
+		cfRay := resp.Header.Get("CF-RAY")
+
+		colo := p.getColo(cfRay)
+		if colo == "" {
+			return 0, 0
+		}
+
+	}
+
+	// for test delay
+	success := 0
+	var delay time.Duration
+	for i := 0; i < PingTimes; i++ {
+		requ, err := http.NewRequest(http.MethodHead, traceURL, nil)
+		if err != nil {
+			log.Fatal("意外的错误，情报告：", err)
+			return 0, 0
+		}
+		requ.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.80 Safari/537.36")
+		if i == PingTimes-1 {
+			requ.Header.Set("Connection", "close")
+		}
+		startTime := time.Now()
+		resp, err := hc.Do(requ)
+		if err != nil {
+			continue
+		}
+		success++
+		io.Copy(io.Discard, resp.Body)
+		_ = resp.Body.Close()
+		duration := time.Since(startTime)
+		delay += duration
+
+	}
+
+	return success, delay
+
+}
+
+func MapColoMap() *sync.Map {
+	if HttpingColo == "" {
+		return nil
+	}
+
+	colos := strings.Split(HttpingColo, ",")
+	colomap := &sync.Map{}
+	for _, colo := range colos {
+		colomap.Store(colo, colo)
+	}
+	return colomap
+}
+
+func GetRequest() *http.Request {
+	req, err := http.NewRequest("GET", URL, nil)
+	if err != nil {
+		log.Fatal(err)
+	}
+	return req
+}
+
+func (p *Ping) getColo(b string) string {
+	if b == "" {
+		return ""
+	}
+	idColo := strings.Split(b, "-")
+
+	out := idColo[1]
+
+	utils.ColoAble.Store(out, out)
+
+	if HttpingColomap == nil {
+		return out
+	}
+
+	_, ok := HttpingColomap.Load(out)
+	if ok {
+		return out
+	}
+
+	return ""
+}