2020-07-05 13:19:53 +08:00
|
|
|
|
package main
|
|
|
|
|
|
|
|
|
|
import (
|
2020-08-31 00:44:45 +08:00
|
|
|
|
"flag"
|
2020-07-05 13:19:53 +08:00
|
|
|
|
"fmt"
|
2020-08-31 00:44:45 +08:00
|
|
|
|
"os"
|
2020-07-09 10:52:26 +08:00
|
|
|
|
"sort"
|
2020-07-05 13:19:53 +08:00
|
|
|
|
"sync"
|
2020-07-09 10:52:26 +08:00
|
|
|
|
"time"
|
2020-08-31 00:44:45 +08:00
|
|
|
|
|
|
|
|
|
"github.com/cheggaaa/pb/v3"
|
2020-07-05 13:19:53 +08:00
|
|
|
|
)
|
|
|
|
|
|
2020-08-31 00:44:45 +08:00
|
|
|
|
var version string
|
2020-09-01 02:21:16 +08:00
|
|
|
|
var disableDownload bool
|
|
|
|
|
var ipFile string
|
2020-08-31 00:44:45 +08:00
|
|
|
|
|
|
|
|
|
func init() {
|
2020-07-09 10:52:26 +08:00
|
|
|
|
var downloadSecond int64
|
2020-08-31 00:44:45 +08:00
|
|
|
|
var printVersion bool
|
2020-08-31 18:51:45 +08:00
|
|
|
|
const help = `
|
|
|
|
|
CloudflareSpeedTest
|
|
|
|
|
测试 Cloudflare CDN 所有 IP 的延迟和速度,获取最快 IP!
|
2020-08-31 11:31:31 +08:00
|
|
|
|
https://github.com/XIU2/CloudflareSpeedTest
|
2020-08-31 00:44:45 +08:00
|
|
|
|
|
|
|
|
|
参数:
|
|
|
|
|
-n 500
|
2020-09-01 02:21:16 +08:00
|
|
|
|
测速线程数量;数值越大速度越快,请勿超过1000(结果误差大);(默认 500)
|
2020-08-31 00:44:45 +08:00
|
|
|
|
-t 4
|
2020-09-01 02:21:16 +08:00
|
|
|
|
延迟测速次数;单个 IP 测速次数,为 1 时将过滤丢包的IP,TCP协议;(默认 4)
|
2020-08-31 11:31:31 +08:00
|
|
|
|
-dn 20
|
2020-09-01 02:21:16 +08:00
|
|
|
|
下载测速数量;延迟测速并排序后,从最低延迟起测试下载速度的数量,请勿太多(速度慢);(默认 20)
|
2020-08-31 00:44:45 +08:00
|
|
|
|
-dt 10
|
2020-09-01 02:21:16 +08:00
|
|
|
|
下载测速时间;单个 IP 测速最长时间,单位:秒;(默认 10)
|
|
|
|
|
-f ip.txt
|
|
|
|
|
IP 数据文件;支持相对路径和绝对路径,如果包含空格请前后加上引号;(默认 ip.txt)
|
|
|
|
|
-dd
|
|
|
|
|
禁用下载测速;如果带上该参数就是禁用下载测速;(默认 启用)
|
2020-08-31 00:44:45 +08:00
|
|
|
|
-v
|
|
|
|
|
打印程序版本
|
|
|
|
|
-h
|
|
|
|
|
打印帮助说明
|
|
|
|
|
|
|
|
|
|
示例:
|
2020-09-01 02:21:16 +08:00
|
|
|
|
CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10
|
2020-09-01 02:26:38 +08:00
|
|
|
|
CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -f "C:\abc\ip.txt" -dd`
|
2020-08-31 00:44:45 +08:00
|
|
|
|
|
2020-09-01 02:21:16 +08:00
|
|
|
|
flag.IntVar(&pingRoutine, "n", 500, "测速线程数量")
|
|
|
|
|
flag.IntVar(&pingTime, "t", 4, "延迟测速次数")
|
|
|
|
|
flag.IntVar(&downloadTestCount, "dn", 20, "下载测速数量")
|
2020-08-31 17:31:48 +08:00
|
|
|
|
flag.Int64Var(&downloadSecond, "dt", 10, "下载测速时间")
|
2020-09-01 02:21:16 +08:00
|
|
|
|
flag.BoolVar(&disableDownload, "dd", false, "禁用下载测速")
|
|
|
|
|
flag.StringVar(&ipFile, "f", "ip.txt", "IP 数据文件")
|
2020-08-31 00:44:45 +08:00
|
|
|
|
flag.BoolVar(&printVersion, "v", false, "打印程序版本")
|
|
|
|
|
|
2020-07-09 10:52:26 +08:00
|
|
|
|
downloadTestTime = time.Duration(downloadSecond) * time.Second
|
2020-08-31 00:44:45 +08:00
|
|
|
|
|
|
|
|
|
flag.Usage = func() { fmt.Print(help) }
|
|
|
|
|
flag.Parse()
|
|
|
|
|
if printVersion {
|
|
|
|
|
println(version)
|
|
|
|
|
os.Exit(0)
|
|
|
|
|
}
|
2020-09-01 12:43:51 +08:00
|
|
|
|
if pingRoutine <= 0 {
|
|
|
|
|
pingRoutine = 500
|
|
|
|
|
}
|
|
|
|
|
if pingTime <= 0 {
|
|
|
|
|
pingTime = 4
|
|
|
|
|
}
|
|
|
|
|
if downloadTestCount <= 0 {
|
|
|
|
|
downloadTestCount = 20
|
|
|
|
|
}
|
|
|
|
|
if downloadSecond <= 0 {
|
|
|
|
|
downloadSecond = 10
|
|
|
|
|
}
|
|
|
|
|
if ipFile == "" {
|
|
|
|
|
ipFile = "ip.txt"
|
|
|
|
|
}
|
2020-07-05 13:19:53 +08:00
|
|
|
|
}
|
|
|
|
|
|
2020-07-09 10:52:26 +08:00
|
|
|
|
func main() {
|
2020-09-01 02:21:16 +08:00
|
|
|
|
initipEndWith() // 随机数
|
|
|
|
|
failTime = pingTime // 设置接收次数
|
|
|
|
|
ips := loadFirstIPOfRangeFromFile(ipFile) // 读入IP
|
|
|
|
|
pingCount := len(ips) * pingTime // 计算进度条总数(IP*测试次数)
|
|
|
|
|
bar := pb.Full.Start(pingCount) // 进度条总数
|
2020-07-05 13:19:53 +08:00
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
|
var mu sync.Mutex
|
2020-07-09 10:52:26 +08:00
|
|
|
|
var data = make([]CloudflareIPData, 0)
|
|
|
|
|
|
2020-08-31 00:44:45 +08:00
|
|
|
|
fmt.Println("开始延迟测速(TCP):")
|
2020-07-09 10:52:26 +08:00
|
|
|
|
control := make(chan bool, pingRoutine)
|
|
|
|
|
for _, ip := range ips {
|
2020-07-05 13:19:53 +08:00
|
|
|
|
wg.Add(1)
|
2020-07-09 10:52:26 +08:00
|
|
|
|
control <- false
|
2020-09-01 12:43:51 +08:00
|
|
|
|
handleProgress := handleProgressGenerator(bar) // 多线程进度条
|
2020-07-09 10:52:26 +08:00
|
|
|
|
go tcpingGoroutine(&wg, &mu, ip, pingTime, &data, control, handleProgress)
|
2020-07-05 13:19:53 +08:00
|
|
|
|
}
|
|
|
|
|
wg.Wait()
|
|
|
|
|
bar.Finish()
|
2020-09-01 02:21:16 +08:00
|
|
|
|
|
|
|
|
|
sort.Sort(CloudflareIPDataSet(data)) // 排序
|
|
|
|
|
if !disableDownload { // 如果禁用下载测速就跳过
|
2020-09-01 12:43:51 +08:00
|
|
|
|
if len(data) > 0 { // IP数组长度(IP数量) 大于 0 时继续
|
|
|
|
|
if len(data) < downloadTestCount { // 如果IP数组长度(IP数量) 小于 下载测速次数,则次数改为IP数
|
|
|
|
|
downloadTestCount = len(data)
|
|
|
|
|
fmt.Println("\n[信息] IP数量小于下载测速次数,下载测速次数改为IP数。\n")
|
|
|
|
|
}
|
|
|
|
|
bar = pb.Simple.Start(downloadTestCount)
|
|
|
|
|
fmt.Println("开始下载测速:")
|
|
|
|
|
for i := 0; i < downloadTestCount; i++ {
|
|
|
|
|
_, speed := DownloadSpeedHandler(data[i].ip)
|
|
|
|
|
data[i].downloadSpeed = speed
|
|
|
|
|
bar.Add(1)
|
|
|
|
|
}
|
|
|
|
|
bar.Finish()
|
|
|
|
|
} else {
|
|
|
|
|
fmt.Println("\n[信息] IP数量为 0,跳过下载测速。")
|
2020-09-01 02:21:16 +08:00
|
|
|
|
}
|
2020-07-09 10:52:26 +08:00
|
|
|
|
}
|
2020-09-01 02:21:16 +08:00
|
|
|
|
ExportCsv("./result.csv", data) // 输出结果
|
2020-07-05 13:19:53 +08:00
|
|
|
|
}
|