mirror of
https://github.com/XIU2/CloudflareSpeedTest.git
synced 2026-03-04 13:45:40 +08:00
207 lines
7.5 KiB
Go
207 lines
7.5 KiB
Go
package task
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"io"
|
||
"net"
|
||
"net/http"
|
||
"sort"
|
||
"strconv"
|
||
"time"
|
||
|
||
"github.com/XIU2/CloudflareSpeedTest/utils"
|
||
|
||
"github.com/VividCortex/ewma"
|
||
)
|
||
|
||
const (
|
||
bufferSize = 1024
|
||
defaultURL = "https://cf.xiu2.xyz/url"
|
||
defaultTimeout = 10 * time.Second
|
||
defaultDisableDownload = false
|
||
defaultTestNum = 10
|
||
defaultMinSpeed float64 = 0.0
|
||
)
|
||
|
||
var (
|
||
URL = defaultURL
|
||
Timeout = defaultTimeout
|
||
Disable = defaultDisableDownload
|
||
|
||
TestCount = defaultTestNum
|
||
MinSpeed = defaultMinSpeed
|
||
)
|
||
|
||
func checkDownloadDefault() {
|
||
if URL == "" {
|
||
URL = defaultURL
|
||
}
|
||
if Timeout <= 0 {
|
||
Timeout = defaultTimeout
|
||
}
|
||
if TestCount <= 0 {
|
||
TestCount = defaultTestNum
|
||
}
|
||
if MinSpeed <= 0.0 {
|
||
MinSpeed = defaultMinSpeed
|
||
}
|
||
}
|
||
|
||
func TestDownloadSpeed(ipSet utils.PingDelaySet) (speedSet utils.DownloadSpeedSet) {
|
||
checkDownloadDefault()
|
||
if Disable {
|
||
return utils.DownloadSpeedSet(ipSet)
|
||
}
|
||
if len(ipSet) <= 0 { // IP数组长度(IP数量) 大于 0 时才会继续下载测速
|
||
fmt.Println("\n\033[33m[信息] 延迟测速结果 IP 数量为 0，跳过下载测速。\033[0m")
|
||
return
|
||
}
|
||
testNum := TestCount
|
||
if len(ipSet) < TestCount || MinSpeed > 0 { // 如果IP数组长度(IP数量) 小于下载测速数量（-dn），则次数修正为IP数
|
||
testNum = len(ipSet)
|
||
}
|
||
if testNum < TestCount {
|
||
TestCount = testNum
|
||
}
|
||
|
||
fmt.Printf("\033[34m开始下载测速（下限：%.2f MB/s, 数量：%d, 队列：%d）\033[0m\n", MinSpeed, TestCount, testNum)
|
||
// 控制 下载测速进度条 与 延迟测速进度条 长度一致（强迫症）
|
||
bar_a := len(strconv.Itoa(len(ipSet)))
|
||
bar_b := " "
|
||
for i := 0; i < bar_a; i++ {
|
||
bar_b += " "
|
||
}
|
||
bar := utils.NewBar(TestCount, bar_b, "")
|
||
for i := 0; i < testNum; i++ {
|
||
speed, colo := downloadHandler(ipSet[i].IP)
|
||
ipSet[i].DownloadSpeed = speed
|
||
if ipSet[i].Colo == "" { // 只有当 Colo 是空的时候，才写入，否则代表之前是 httping 测速并获取过了
|
||
ipSet[i].Colo = colo
|
||
}
|
||
// 在每个 IP 下载测速后，以 [下载速度下限] 条件过滤结果
|
||
if speed >= MinSpeed*1024*1024 {
|
||
bar.Grow(1, "")
|
||
speedSet = append(speedSet, ipSet[i]) // 高于下载速度下限时，添加到新数组中
|
||
if len(speedSet) == TestCount { // 凑够满足条件的 IP 时（下载测速数量 -dn），就跳出循环
|
||
break
|
||
}
|
||
}
|
||
}
|
||
bar.Done()
|
||
if utils.Debug && len(speedSet) == 0 { // 调试模式下，没有满足速度限制的数据，返回所有测速数据供用户查看当前的测速结果，以便适当调低预期测速条件
|
||
fmt.Println("\033[33m[调试] 没有满足 下载速度下限 条件的 IP，忽略条件返回所有测速数据（方便下次测速时调整条件）。\033[0m")
|
||
speedSet = utils.DownloadSpeedSet(ipSet)
|
||
}
|
||
// 按速度排序
|
||
sort.Sort(speedSet)
|
||
return
|
||
}
|
||
|
||
func getDialContext(ip *net.IPAddr) func(ctx context.Context, network, address string) (net.Conn, error) {
|
||
var fakeSourceAddr string
|
||
if isIPv4(ip.String()) {
|
||
fakeSourceAddr = fmt.Sprintf("%s:%d", ip.String(), TCPPort)
|
||
} else {
|
||
fakeSourceAddr = fmt.Sprintf("[%s]:%d", ip.String(), TCPPort)
|
||
}
|
||
return func(ctx context.Context, network, address string) (net.Conn, error) {
|
||
return (&net.Dialer{}).DialContext(ctx, network, fakeSourceAddr)
|
||
}
|
||
}
|
||
|
||
// return download Speed
|
||
func downloadHandler(ip *net.IPAddr) (float64, string) {
|
||
client := &http.Client{
|
||
Transport: &http.Transport{DialContext: getDialContext(ip)},
|
||
Timeout: Timeout,
|
||
CheckRedirect: func(req *http.Request, via []*http.Request) error {
|
||
if len(via) > 10 { // 限制最多重定向 10 次
|
||
if utils.Debug { // 调试模式下，输出更多信息
|
||
fmt.Printf("\033[31m[调试] IP: %s, 下载测速地址重定向次数过多，终止测速，下载测速地址: %s\033[0m\n", ip.String(), req.URL.String())
|
||
}
|
||
return http.ErrUseLastResponse
|
||
}
|
||
if req.Header.Get("Referer") == defaultURL { // 当使用默认下载测速地址时，重定向不携带 Referer
|
||
req.Header.Del("Referer")
|
||
}
|
||
return nil
|
||
},
|
||
}
|
||
req, err := http.NewRequest("GET", URL, nil)
|
||
if err != nil {
|
||
if utils.Debug { // 调试模式下，输出更多信息
|
||
fmt.Printf("\033[31m[调试] IP: %s, 下载测速请求创建失败，错误信息: %v, 下载测速地址: %s\033[0m\n", ip.String(), err, URL)
|
||
}
|
||
return 0.0, ""
|
||
}
|
||
|
||
req.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")
|
||
|
||
response, err := client.Do(req)
|
||
if err != nil {
|
||
if utils.Debug { // 调试模式下，输出更多信息
|
||
finalURL := URL // 默认的最终 URL，这样当 response 为空时也能输出
|
||
if response != nil && response.Request != nil && response.Request.URL != nil { // 如果 response 和 URL 存在，则获取最终 URL
|
||
finalURL = response.Request.URL.String()
|
||
}
|
||
fmt.Printf("\033[31m[调试] IP: %s, 下载测速失败，错误信息: %v, 下载测速地址: %s, 最终地址(如有重定向): %s\033[0m\n", ip.String(), err, URL, finalURL)
|
||
}
|
||
return 0.0, ""
|
||
}
|
||
defer response.Body.Close()
|
||
if response.StatusCode != 200 {
|
||
if utils.Debug { // 调试模式下，输出更多信息
|
||
fmt.Printf("\033[31m[调试] IP: %s, 下载测速终止，HTTP 状态码: %d, 下载测速地址: %s, 最终地址(如有重定向): %s\033[0m\n", ip.String(), response.StatusCode, URL, response.Request.URL.String())
|
||
}
|
||
return 0.0, ""
|
||
}
|
||
// 通过头部 Server 值判断是 Cloudflare 还是 AWS CloudFront 并设置 cfRay 为各自的机场地区码完整内容
|
||
colo := getHeaderColo(response.Header)
|
||
|
||
timeStart := time.Now() // 开始时间（当前）
|
||
timeEnd := timeStart.Add(Timeout) // 加上下载测速时间得到的结束时间
|
||
|
||
contentLength := response.ContentLength // 文件大小
|
||
buffer := make([]byte, bufferSize)
|
||
|
||
var (
|
||
contentRead int64 = 0
|
||
timeSlice = Timeout / 100
|
||
timeCounter = 1
|
||
lastContentRead int64 = 0
|
||
)
|
||
|
||
var nextTime = timeStart.Add(timeSlice * time.Duration(timeCounter))
|
||
e := ewma.NewMovingAverage()
|
||
|
||
// 循环计算，如果文件下载完了（两者相等），则退出循环（终止测速）
|
||
for contentLength != contentRead {
|
||
currentTime := time.Now()
|
||
if currentTime.After(nextTime) {
|
||
timeCounter++
|
||
nextTime = timeStart.Add(timeSlice * time.Duration(timeCounter))
|
||
e.Add(float64(contentRead - lastContentRead))
|
||
lastContentRead = contentRead
|
||
}
|
||
// 如果超出下载测速时间，则退出循环（终止测速）
|
||
if currentTime.After(timeEnd) {
|
||
break
|
||
}
|
||
bufferRead, err := response.Body.Read(buffer)
|
||
if err != nil {
|
||
if err != io.EOF { // 如果文件下载过程中遇到报错（如 Timeout），且并不是因为文件下载完了，则退出循环（终止测速）
|
||
break
|
||
} else if contentLength == -1 { // 文件下载完成 且 文件大小未知，则退出循环（终止测速），例如：https://speed.cloudflare.com/__down?bytes=200000000 这样的，如果在 10 秒内就下载完成了，会导致测速结果明显偏低甚至显示为 0.00（下载速度太快时）
|
||
break
|
||
}
|
||
// 获取上个时间片
|
||
last_time_slice := timeStart.Add(timeSlice * time.Duration(timeCounter-1))
|
||
// 下载数据量 / (用当前时间 - 上个时间片/ 时间片)
|
||
e.Add(float64(contentRead-lastContentRead) / (float64(currentTime.Sub(last_time_slice)) / float64(timeSlice)))
|
||
}
|
||
contentRead += int64(bufferRead)
|
||
}
|
||
return e.Value() / (Timeout.Seconds() / 120), colo
|
||
}
|