新增 自定义TCP端口 功能。

IPRangeLoader.go

@@ -7,8 +7,8 @@ import (
 	"os"
 )
 
-func loadFirstIPOfRangeFromFile() []net.IPAddr {
-	file, err := os.Open("ip.txt")
+func loadFirstIPOfRangeFromFile(ipFile string) []net.IPAddr {
+	file, err := os.Open(ipFile)
 	if err != nil {
 		log.Fatal(err)
 	}

README.md

@@ -1,4 +1,10 @@
-# CloudflareSpeedTest
+# XIU2/CloudflareSpeedTest
+
+[![Go Version](https://img.shields.io/github/go-mod/go-version/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=Go&color=00ADD8)](https://github.com/XIU2/CloudflareSpeedTest/blob/master/go.mod)
+[![Release Version](https://img.shields.io/github/v/release/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=Release&color=1784ff)](https://github.com/XIU2/CloudflareSpeedTest/releases/latest)
+[![GitHub license](https://img.shields.io/github/license/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=License&color=f38020)](https://github.com/XIU2/CloudflareSpeedTest/blob/master/LICENSE)
+[![GitHub Star](https://img.shields.io/github/stars/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=Star&color=f38020)](https://github.com/XIU2/CloudflareSpeedTest/stargazers)
+[![GitHub Fork](https://img.shields.io/github/forks/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=Fork&color=f38020)](https://github.com/XIU2/CloudflareSpeedTest/network/members)
 
 国外很多网站都在使用 Cloudflare CDN，但分配给中国访客的 IP 并不友好。  
 虽然 Cloudflare 公开了所有 [IP 段](https://www.cloudflare.com/ips/) ，但想要在这么多 IP 中找到适合自己的，怕是要累死，所以就有了这个软件。  
@@ -12,10 +18,10 @@
 1. 下载编译好的可执行文件 [蓝奏云](https://www.lanzoux.com/b0742hkxe) / [Github](https://github.com/XIU2/CloudflareSpeedTest/releases) 并解压。  
 2. 双击运行 `CloudflareST.exe`文件（Windows系统），等待测速...  
 
-测速完毕后，会把结果保存在当前目录下的 `result.csv` 文件中（只输出丢包率 50% 以下的），用记事本打开，排序为**延迟由低到高**，每一列用逗号分隔，分别是：  
+测速完毕后，会显示最快的 20 个 IP，完整结果则保存在当前目录下的 `result.csv` 文件中，用记事本打开，排序为**延迟由低到高**，每一列用逗号分隔，分别是：  
 ```
-IP 地址, Ping 发送次数, Ping 接收次数, Ping 接收率, 平均延迟, 下载速度 (MB/s)
-104.27.70.18, 4, 4, 1.0000, 150.7948, 12.8951
+IP 地址, 测试次数, 成功次数, 成功比率, 平均延迟, 下载速度 (MB/s)
+104.27.70.18, 4, 4, 1.00, 150.79, 12.89
 ```
 选择一个平均延迟与下载速度都不错的 IP 放到 `Hosts` 文件中（指向域名）。  
 
@@ -32,37 +38,54 @@ https://github.com/XIU2/CloudflareSpeedTest
 
 参数：
     -n 500
-        测速线程数量；请勿超过1000 (默认 500)
+        测速线程数量；数值越大速度越快，请勿超过 1000(结果误差大)；(默认 500)
     -t 4
-        延迟测速次数；单个 IP 测速次数，TCP协议 (默认 4)
+        延迟测速次数；单个 IP 测速次数，为 1 时将过滤丢包的IP，TCP协议；(默认 4)
+    -tp 443
+        延迟测速端口；延迟测速 TCP 协议的端口；(默认 443)
     -dn 20
-        下载测速数量；延迟测速后，从最低延迟起测试下载速度的数量，请勿太多 (默认 20)
+        下载测速数量；延迟测速并排序后，从最低延迟起下载测速数量，请勿太多(速度慢)；(默认 20)
     -dt 10
-        下载测速时间；单个 IP 测速最长时间，单位：秒 (默认 10)
+        下载测速时间；单个 IP 测速最长时间，单位：秒；(默认 10)
+    -p 20
+        直接显示结果；测速后直接显示指定数量的结果，为 -1 时不显示结果直接退出；(默认 20)
+    -f ip.txt
+        IP 数据文件；相对/绝对路径，如包含空格请加上引号；支持其他 CDN IP段，记得禁用下载测速；(默认 ip.txt)
+    -o result.csv
+        输出结果文件；相对/绝对路径，如包含空格请加上引号；允许 .txt 等后缀；(默认 result.csv)
+    -dd
+        禁用下载测速；如果带上该参数就是禁用下载测速；(默认 启用)
     -v
         打印程序版本
     -h
         打印帮助说明
 
 示例：
-    Windows：CloudflareST.exe -n 800 -t 4 -dn 20 -dt 10
-    Linux：CloudflareST -n 800 -t 4 -dn 20 -dt 10
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -f "ip.txt" -o "result.csv" -dd
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -f "C:\abc\ip.txt" -o "C:\abc\result.csv" -dd
 ```
 
 #### 使用示例
 
 在 CMD 中运行，或者把启动参数添加到快捷方式中。  
-> **注意：** 不需要四个参数都加上，如果你认为某个参数默认就很合适，那就跳过。  
+> **注意：** 不需要加上所有参数，按需选择，参数前后顺序随意。  
 
 ``` cmd
 # CMD 示例
-CloudflareST.exe -n 800 -t 4 -dn 20 -dt 10
+CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10
+
+# 指定 IP数据文件 及 输出结果文件（相对路径，即当前目录下）
+CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -f "ip.txt" -o "result.csv" -dd
+
+# 指定 IP数据文件 及 输出结果文件（绝对路径，即 C:\abc\ 目录下）
+CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -f "C:\abc\ip.txt" -o "C:\abc\result.csv" -dd
 ```
 
 ``` cmd
 # 快捷方式示例（右键快捷方式 - 目标）
 ## 如果有引号就放在引号外面，记得引号和 - 之间有空格。
-"D:\Program Files\CloudflareST\CloudflareST.exe" -n 800 -t 4 -dn 20 -dt 10
+"D:\Program Files\CloudflareST\CloudflareST.exe" -n 500 -t 4 -dn 20 -dt 10
 ```
 
 ****

go.mod

@@ -1,4 +1,4 @@
-module CloudflareIPScanner
+module CloudflareSpeedTest
 
 go 1.14
 

main.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"os"
 	"sort"
+	"strconv"
 	"sync"
 	"time"
 
@@ -12,6 +13,11 @@ import (
 )
 
 var version string
+var disableDownload bool
+var tcpPort int
+var ipFile string
+var outputFile string
+var printResult int
 
 func init() {
 	var downloadSecond int64
@@ -23,27 +29,42 @@ https://github.com/XIU2/CloudflareSpeedTest
 
 参数：
     -n 500
-        测速线程数量；请勿超过1000 (默认 500)
+        测速线程数量；数值越大速度越快，请勿超过 1000(结果误差大)；(默认 500)
     -t 4
-        延迟测速次数；单个 IP 测速次数，TCP协议 (默认 4)
+        延迟测速次数；单个 IP 测速次数，为 1 时将过滤丢包的IP，TCP协议；(默认 4)
+    -tp 443
+        延迟测速端口；延迟测速 TCP 协议的端口；(默认 443)
     -dn 20
-        下载测速数量；延迟测速后，从最低延迟起测试下载速度的数量，请勿太多 (默认 20)
+        下载测速数量；延迟测速并排序后，从最低延迟起下载测速数量，请勿太多(速度慢)；(默认 20)
     -dt 10
-        下载测速时间；单个 IP 测速最长时间，单位：秒 (默认 10)
+        下载测速时间；单个 IP 测速最长时间，单位：秒；(默认 10)
+    -p 20
+        直接显示结果；测速后直接显示指定数量的结果，为 -1 时不显示结果直接退出；(默认 20)
+    -f ip.txt
+        IP 数据文件；相对/绝对路径，如包含空格请加上引号；支持其他 CDN IP段，记得禁用下载测速；(默认 ip.txt)
+    -o result.csv
+        输出结果文件；相对/绝对路径，如包含空格请加上引号；允许 .txt 等后缀；(默认 result.csv)
+    -dd
+        禁用下载测速；如果带上该参数就是禁用下载测速；(默认 启用)
     -v
         打印程序版本
     -h
         打印帮助说明
 
 示例：
-    Windows：CloudflareST.exe -n 800 -t 4 -dn 20 -dt 10
-    Linux：CloudflareST -n 800 -t 4 -dn 20 -dt 10
-`
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -f "ip.txt" -o "result.csv" -dd
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -f "C:\abc\ip.txt" -o "C:\abc\result.csv" -dd`
 
-	pingRoutine = *flag.Int("n", 500, "测速线程数量")
-	pingTime = *flag.Int("t", 4, "延迟测速次数")
-	downloadTestCount = *flag.Int("dn", 20, "下载测速数量")
+	flag.IntVar(&pingRoutine, "n", 500, "测速线程数量")
+	flag.IntVar(&pingTime, "t", 4, "延迟测速次数")
+	flag.IntVar(&tcpPort, "tp", 443, "延迟测速端口")
+	flag.IntVar(&downloadTestCount, "dn", 20, "下载测速数量")
 	flag.Int64Var(&downloadSecond, "dt", 10, "下载测速时间")
+	flag.IntVar(&printResult, "p", 20, "直接显示结果")
+	flag.BoolVar(&disableDownload, "dd", false, "禁用下载测速")
+	flag.StringVar(&ipFile, "f", "ip.txt", "IP 数据文件")
+	flag.StringVar(&outputFile, "o", "result.csv", "输出结果文件")
 	flag.BoolVar(&printVersion, "v", false, "打印程序版本")
 
 	downloadTestTime = time.Duration(downloadSecond) * time.Second
@@ -54,36 +75,95 @@ https://github.com/XIU2/CloudflareSpeedTest
 		println(version)
 		os.Exit(0)
 	}
+	if pingRoutine <= 0 {
+		pingRoutine = 500
+	}
+	if pingTime <= 0 {
+		pingTime = 4
+	}
+	if tcpPort < 1 || tcpPort > 65535 {
+		tcpPort = 443
+	}
+	if downloadTestCount <= 0 {
+		downloadTestCount = 20
+	}
+	if downloadSecond <= 0 {
+		downloadSecond = 10
+	}
+	if printResult == 0 {
+		printResult = 20
+	}
+	if ipFile == "" {
+		ipFile = "ip.txt"
+	}
+	if outputFile == "" {
+		outputFile = "result.csv"
+	}
 }
 
 func main() {
-	initipEndWith()
-	ips := loadFirstIPOfRangeFromFile()
-	pingCount := len(ips) * pingTime
-	bar := pb.StartNew(pingCount)
+	initipEndWith()                           // 随机数
+	failTime = pingTime                       // 设置接收次数
+	ips := loadFirstIPOfRangeFromFile(ipFile) // 读入IP
+	pingCount := len(ips) * pingTime          // 计算进度条总数（IP*测试次数）
+	bar := pb.Full.Start(pingCount)           // 进度条总数
 	var wg sync.WaitGroup
 	var mu sync.Mutex
 	var data = make([]CloudflareIPData, 0)
 
-	fmt.Println("开始延迟测速(TCP)：")
-
+	fmt.Println("开始延迟测速（模式：TCP，端口：" + strconv.Itoa(tcpPort) + "）：")
 	control := make(chan bool, pingRoutine)
 	for _, ip := range ips {
 		wg.Add(1)
 		control <- false
-		handleProgress := handleProgressGenerator(bar)
-		go tcpingGoroutine(&wg, &mu, ip, pingTime, &data, control, handleProgress)
+		handleProgress := handleProgressGenerator(bar) // 多线程进度条
+		go tcpingGoroutine(&wg, &mu, ip, tcpPort, pingTime, &data, control, handleProgress)
 	}
 	wg.Wait()
 	bar.Finish()
-	bar = pb.StartNew(downloadTestCount)
-	sort.Sort(CloudflareIPDataSet(data))
-	fmt.Println("开始下载测速：")
-	for i := 0; i < downloadTestCount; i++ {
-		_, speed := DownloadSpeedHandler(data[i].ip)
-		data[i].downloadSpeed = speed
-		bar.Add(1)
+
+	sort.Sort(CloudflareIPDataSet(data)) // 排序
+
+	// 下载测速
+	if !disableDownload { // 如果禁用下载测速就跳过
+		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，跳过下载测速。")
+		}
 	}
-	bar.Finish()
-	ExportCsv("./result.csv", data)
+
+	// 直接输出结果
+	if printResult > 0 { // 如果禁用下载测速就跳过
+		dateString := convertToString(data) // 转为多维数组 [][]String
+		if len(dateString) > 0 {            // IP数组长度(IP数量) 大于 0 时继续
+			if len(dateString) < printResult { // 如果IP数组长度(IP数量) 小于  打印次数，则次数改为IP数量
+				printResult = len(dateString)
+				fmt.Println("\n[信息] IP数量小于显示结果数量，显示结果数量改为IP数量。\n")
+			}
+			fmt.Println("\nIP 地址    \t", "测试次数\t", "成功次数\t", "成功比率\t", "平均延迟\t", "下载速度 (MB/s)")
+			for i := 0; i < printResult; i++ {
+				fmt.Println(dateString[i][0], "\t", dateString[i][1], "\t\t", dateString[i][2], "\t\t", dateString[i][3], "\t\t", dateString[i][4], "\t", dateString[i][5])
+			}
+			fmt.Printf("\n完整内容请查看 %v 文件。请按 回车键 或 Ctrl+C 退出。", outputFile)
+			var pause int
+			fmt.Scanln(&pause)
+		} else {
+			fmt.Println("\n[信息] IP数量为 0，跳过输出结果。")
+		}
+	}
+
+	// 输出结果到文件
+	ExportCsv(outputFile, data)
 }

tcping.go

@@ -2,19 +2,20 @@ package main
 
 import (
 	"context"
-	"github.com/VividCortex/ewma"
 	"io"
 	"net"
 	"net/http"
 	"strconv"
 	"sync"
 	"time"
+
+	"github.com/VividCortex/ewma"
 )
 
 //bool connectionSucceed float32 time
-func tcping(ip net.IPAddr) (bool, float32) {
+func tcping(ip net.IPAddr, tcpPort int) (bool, float32) {
 	startTime := time.Now()
-	conn, err := net.DialTimeout("tcp", ip.String()+":"+strconv.Itoa(defaultTcpPort), tcpConnectTimeout)
+	conn, err := net.DialTimeout("tcp", ip.String()+":"+strconv.Itoa(tcpPort), tcpConnectTimeout)
 	if err != nil {
 		return false, 0
 	} else {
@@ -26,11 +27,11 @@ func tcping(ip net.IPAddr) (bool, float32) {
 }
 
 //pingReceived pingTotalTime
-func checkConnection(ip net.IPAddr) (int, float32) {
+func checkConnection(ip net.IPAddr, tcpPort int) (int, float32) {
 	pingRecv := 0
 	var pingTime float32 = 0.0
 	for i := 1; i <= failTime; i++ {
-		pingSucceed, pingTimeCurrent := tcping(ip)
+		pingSucceed, pingTimeCurrent := tcping(ip, tcpPort)
 		if pingSucceed {
 			pingRecv++
 			pingTime += pingTimeCurrent
@@ -40,12 +41,12 @@ func checkConnection(ip net.IPAddr) (int, float32) {
 }
 
 //return Success packetRecv averagePingTime specificIPAddr
-func tcpingHandler(ip net.IPAddr, pingCount int, progressHandler func(e progressEvent)) (bool, int, float32, net.IPAddr) {
+func tcpingHandler(ip net.IPAddr, tcpPort int, pingCount int, progressHandler func(e progressEvent)) (bool, int, float32, net.IPAddr) {
 	ipCanConnect := false
 	pingRecv := 0
 	var pingTime float32 = 0.0
 	for !ipCanConnect {
-		pingRecvCurrent, pingTimeCurrent := checkConnection(ip)
+		pingRecvCurrent, pingTimeCurrent := checkConnection(ip, tcpPort)
 		if pingRecvCurrent != 0 {
 			ipCanConnect = true
 			pingRecv = pingRecvCurrent
@@ -61,7 +62,7 @@ func tcpingHandler(ip net.IPAddr, pingCount int, progressHandler func(e progress
 	if ipCanConnect {
 		progressHandler(AvailableIPFound)
 		for i := failTime; i < pingCount; i++ {
-			pingSuccess, pingTimeCurrent := tcping(ip)
+			pingSuccess, pingTimeCurrent := tcping(ip, tcpPort)
 			progressHandler(NormalPing)
 			if pingSuccess {
 				pingRecv++
@@ -75,9 +76,9 @@ func tcpingHandler(ip net.IPAddr, pingCount int, progressHandler func(e progress
 	}
 }
 
-func tcpingGoroutine(wg *sync.WaitGroup, mutex *sync.Mutex, ip net.IPAddr, pingCount int, csv *[]CloudflareIPData, control chan bool, progressHandler func(e progressEvent)) {
+func tcpingGoroutine(wg *sync.WaitGroup, mutex *sync.Mutex, ip net.IPAddr, tcpPort int, pingCount int, csv *[]CloudflareIPData, control chan bool, progressHandler func(e progressEvent)) {
 	defer wg.Done()
-	success, pingRecv, pingTimeAvg, currentIP := tcpingHandler(ip, pingCount, progressHandler)
+	success, pingRecv, pingTimeAvg, currentIP := tcpingHandler(ip, tcpPort, pingCount, progressHandler)
 	if success {
 		mutex.Lock()
 		var cfdata CloudflareIPData
@@ -130,7 +131,7 @@ func DownloadSpeedHandler(ip net.IPAddr) (bool, float32) {
 			var nextTime = timeStart.Add(timeSlice * time.Duration(timeCounter))
 			e := ewma.NewMovingAverage()
 
-			for ; contentLength != contentRead; {
+			for contentLength != contentRead {
 				var currentTime = time.Now()
 				if currentTime.After(nextTime) {
 					timeCounter += 1

util.go

@@ -36,21 +36,19 @@ func ExportCsv(filePath string, data []CloudflareIPData) {
 	}
 	defer fp.Close()
 	w := csv.NewWriter(fp) //创建一个新的写入文件流
-	w.Write([]string{"IP 地址", "Ping 发送次数", "Ping 接收次数", "Ping 接收率", "平均延迟", "下载速度 (MB/s)"})
+	w.Write([]string{"IP 地址", "测试次数", "成功次数", "成功比率", "平均延迟", "下载速度 (MB/s)"})
 	w.WriteAll(convertToString(data))
 	w.Flush()
 }
 
-//"IP Address","Ping Count","Ping received","Ping received rate","Ping time","Download speed"
-
 func (cf *CloudflareIPData) toString() []string {
 	result := make([]string, 6)
 	result[0] = cf.ip.String()
 	result[1] = strconv.Itoa(cf.pingCount)
 	result[2] = strconv.Itoa(cf.pingReceived)
-	result[3] = strconv.FormatFloat(float64(cf.getRecvRate()), 'f', 4, 32)
-	result[4] = strconv.FormatFloat(float64(cf.pingTime), 'f', 4, 32)
-	result[5] = strconv.FormatFloat(float64(cf.downloadSpeed)/1024/1024, 'f', 4, 32)
+	result[3] = strconv.FormatFloat(float64(cf.getRecvRate()), 'f', 2, 32)
+	result[4] = strconv.FormatFloat(float64(cf.pingTime), 'f', 2, 32)
+	result[5] = strconv.FormatFloat(float64(cf.downloadSpeed)/1024/1024, 'f', 2, 32)
 	return result
 }
 
@@ -83,9 +81,10 @@ const downloadBufferSize = 1024
 
 var downloadTestCount int
 
-const defaultTcpPort = 443
+//const defaultTcpPort = 443
 const tcpConnectTimeout = time.Second * 1
-const failTime = 4
+
+var failTime int
 
 type CloudflareIPDataSet []CloudflareIPData