rebuild main

IPRangeLoader.go

@@ -35,8 +35,7 @@ func getCidrIPRange(cidr string) (uint8, uint8) {
 	seg4MinIP, seg4MaxIP := getIPSeg4Range(ipSegs, maskLen)
 	//ipPrefix := ipSegs[0] + "." + ipSegs[1] + "." + ipSegs[2] + "."
 
-	return seg4MinIP,
-		seg4MaxIP
+	return seg4MinIP, seg4MaxIP
 }
 
 // 获取 IP 最后一段的区间
@@ -59,6 +58,7 @@ func loadFirstIPOfRangeFromFile(ipFile string) []net.IPAddr {
 	if err != nil {
 		log.Fatal(err)
 	}
+	defer file.Close()
 	firstIPs := make([]net.IPAddr, 0)
 	scanner := bufio.NewScanner(file)
 	scanner.Split(bufio.ScanLines)
@@ -83,8 +83,8 @@ func loadFirstIPOfRangeFromFile(ipFile string) []net.IPAddr {
 			MaxIPNum := getCidrHostNum(Mask)                         // 根据子网掩码获取主机数量
 			for IPRange.Contains(firstIP) {
 				if allip { // 如果是测速全部 IP
-					for i := int(minIP); i <= int(maxIP); i++ { // 遍历 IP 最后一段最小值到最大值
-						firstIP[15] = uint8(i)
+					for i := minIP; i <= maxIP; i++ { // 遍历 IP 最后一段最小值到最大值
+						firstIP[15] = i
 						firstIPCopy := make([]byte, len(firstIP))
 						copy(firstIPCopy, firstIP)
 						firstIPs = append(firstIPs, net.IPAddr{IP: firstIPCopy})
@@ -103,60 +103,60 @@ func loadFirstIPOfRangeFromFile(ipFile string) []net.IPAddr {
 					}
 				}
 			}
-		} else { //IPv6
-			var tempIP uint8
-			for IPRange.Contains(firstIP) {
-				//fmt.Println(firstIP)
-				//fmt.Println(firstIP[0], firstIP[1], firstIP[2], firstIP[3], firstIP[4], firstIP[5], firstIP[6], firstIP[7], firstIP[8], firstIP[9], firstIP[10], firstIP[11], firstIP[12], firstIP[13], firstIP[14], firstIP[15])
-				if !strings.Contains(IPString, "/128") {
-					firstIP[15] = randipEndWith(255) // 随机 IP 的最后一段
-					firstIP[14] = randipEndWith(255) // 随机 IP 的最后一段
-				}
-				firstIPCopy := make([]byte, len(firstIP))
-				copy(firstIPCopy, firstIP)
-				firstIPs = append(firstIPs, net.IPAddr{IP: firstIPCopy})
-				tempIP = firstIP[13]
-				firstIP[13] += randipEndWith(255)
-				if firstIP[13] < tempIP {
-					tempIP = firstIP[12]
-					firstIP[12] += randipEndWith(255)
-					if firstIP[12] < tempIP {
-						tempIP = firstIP[11]
-						firstIP[11] += randipEndWith(255)
-						if firstIP[11] < tempIP {
-							tempIP = firstIP[10]
-							firstIP[10] += randipEndWith(255)
-							if firstIP[10] < tempIP {
-								tempIP = firstIP[9]
-								firstIP[9] += randipEndWith(255)
-								if firstIP[9] < tempIP {
-									tempIP = firstIP[8]
-									firstIP[8] += randipEndWith(255)
-									if firstIP[8] < tempIP {
-										tempIP = firstIP[7]
-										firstIP[7] += randipEndWith(255)
-										if firstIP[7] < tempIP {
-											tempIP = firstIP[6]
-											firstIP[6] += randipEndWith(255)
-											if firstIP[6] < tempIP {
-												tempIP = firstIP[5]
-												firstIP[5] += randipEndWith(255)
-												if firstIP[5] < tempIP {
-													tempIP = firstIP[4]
-													firstIP[4] += randipEndWith(255)
-													if firstIP[4] < tempIP {
-														tempIP = firstIP[3]
-														firstIP[3] += randipEndWith(255)
-														if firstIP[3] < tempIP {
-															tempIP = firstIP[2]
-															firstIP[2] += randipEndWith(255)
-															if firstIP[2] < tempIP {
-																tempIP = firstIP[1]
-																firstIP[1] += randipEndWith(255)
-																if firstIP[1] < tempIP {
-																	tempIP = firstIP[0]
-																	firstIP[0] += randipEndWith(255)
-																}
+		}
+		//IPv6
+		var tempIP uint8
+		for IPRange.Contains(firstIP) {
+			//fmt.Println(firstIP)
+			//fmt.Println(firstIP[0], firstIP[1], firstIP[2], firstIP[3], firstIP[4], firstIP[5], firstIP[6], firstIP[7], firstIP[8], firstIP[9], firstIP[10], firstIP[11], firstIP[12], firstIP[13], firstIP[14], firstIP[15])
+			if !strings.Contains(IPString, "/128") {
+				firstIP[15] = randipEndWith(255) // 随机 IP 的最后一段
+				firstIP[14] = randipEndWith(255) // 随机 IP 的最后一段
+			}
+			firstIPCopy := make([]byte, len(firstIP))
+			copy(firstIPCopy, firstIP)
+			firstIPs = append(firstIPs, net.IPAddr{IP: firstIPCopy})
+			tempIP = firstIP[13]
+			firstIP[13] += randipEndWith(255)
+			if firstIP[13] < tempIP {
+				tempIP = firstIP[12]
+				firstIP[12] += randipEndWith(255)
+				if firstIP[12] < tempIP {
+					tempIP = firstIP[11]
+					firstIP[11] += randipEndWith(255)
+					if firstIP[11] < tempIP {
+						tempIP = firstIP[10]
+						firstIP[10] += randipEndWith(255)
+						if firstIP[10] < tempIP {
+							tempIP = firstIP[9]
+							firstIP[9] += randipEndWith(255)
+							if firstIP[9] < tempIP {
+								tempIP = firstIP[8]
+								firstIP[8] += randipEndWith(255)
+								if firstIP[8] < tempIP {
+									tempIP = firstIP[7]
+									firstIP[7] += randipEndWith(255)
+									if firstIP[7] < tempIP {
+										tempIP = firstIP[6]
+										firstIP[6] += randipEndWith(255)
+										if firstIP[6] < tempIP {
+											tempIP = firstIP[5]
+											firstIP[5] += randipEndWith(255)
+											if firstIP[5] < tempIP {
+												tempIP = firstIP[4]
+												firstIP[4] += randipEndWith(255)
+												if firstIP[4] < tempIP {
+													tempIP = firstIP[3]
+													firstIP[3] += randipEndWith(255)
+													if firstIP[3] < tempIP {
+														tempIP = firstIP[2]
+														firstIP[2] += randipEndWith(255)
+														if firstIP[2] < tempIP {
+															tempIP = firstIP[1]
+															firstIP[1] += randipEndWith(255)
+															if firstIP[1] < tempIP {
+																tempIP = firstIP[0]
+																firstIP[0] += randipEndWith(255)
 															}
 														}
 													}

main.go

@@ -8,17 +8,18 @@ import (
 	"os"
 	"runtime"
 	"sort"
-	"strconv"
 	"sync"
 	"time"
 
 	"github.com/cheggaaa/pb/v3"
 )
 
-var version, ipFile, outputFile, versionNew string
-var disableDownload, ipv6Mode, allip bool
-var tcpPort, printResultNum, downloadSecond int
-var timeLimit, timeLimitLow, speedLimit float64
+var (
+	version, ipFile, outputFile, versionNew string
+	disableDownload, ipv6Mode, allip        bool
+	tcpPort, printResultNum, downloadSecond int
+	timeLimit, timeLimitLow, speedLimit     float64
+)
 
 func init() {
 	var printVersion bool
@@ -139,19 +140,21 @@ func main() {
 	ips := loadFirstIPOfRangeFromFile(ipFile) // 读入IP
 	pingCount := len(ips) * pingTime          // 计算进度条总数（IP*测试次数）
 	bar := pb.Simple.Start(pingCount)         // 进度条总数
-	var wg sync.WaitGroup
-	var mu sync.Mutex
-	var data = make([]CloudflareIPData, 0)
-	var data2 = make([]CloudflareIPData, 0)
+	var (
+		wg    sync.WaitGroup
+		mu    sync.Mutex
+		data  = make([]CloudflareIPData, 0)
+		data2 = make([]CloudflareIPData, 0)
+	)
 	downloadTestTime = time.Duration(downloadSecond) * time.Second
 
 	// 开始延迟测速
-	fmt.Println("# XIU2/CloudflareSpeedTest " + version + "\n")
+	fmt.Printf("# XIU2/CloudflareSpeedTest %s \n", version)
+	ipVersion := "IPv4"
 	if ipv6Mode { // IPv6 模式判断
-		fmt.Println("开始延迟测速（模式：TCP IPv6，端口：" + strconv.Itoa(tcpPort) + "，平均延迟上限：" + fmt.Sprintf("%.2f", timeLimit) + " ms" + "，平均延迟下限：" + fmt.Sprintf("%.2f", timeLimitLow) + " ms）：")
-	} else {
-		fmt.Println("开始延迟测速（模式：TCP IPv4，端口：" + strconv.Itoa(tcpPort) + "，平均延迟上限：" + fmt.Sprintf("%.2f", timeLimit) + " ms" + "，平均延迟下限：" + fmt.Sprintf("%.2f", timeLimitLow) + " ms）：")
+		ipVersion = "IPv6"
 	}
+	fmt.Printf("开始延迟测速（模式：TCP %s，端口：%d ，平均延迟上限：%.2f ms，平均延迟下限：%.2f ms）：\n", ipVersion, tcpPort, timeLimit, timeLimitLow)
 	control := make(chan bool, pingRoutine)
 	for _, ip := range ips {
 		wg.Add(1)
@@ -167,15 +170,13 @@ func main() {
 	// 延迟测速完毕后，以 [平均延迟上限] + [平均延迟下限] 条件过滤结果
 	if timeLimit != 9999 || timeLimitLow != 0 {
 		for i := 0; i < len(data); i++ {
-			if float64(data[i].pingTime) <= timeLimit { // 平均延迟上限
-				if float64(data[i].pingTime) > timeLimitLow { // 平均延迟下限
-					data2 = append(data2, data[i]) // 延迟满足条件时，添加到新数组中
-				} else {
-					continue
-				}
-			} else {
+			if float64(data[i].pingTime) > timeLimit { // 平均延迟上限
 				break
 			}
+			if float64(data[i].pingTime) <= timeLimitLow { // 平均延迟下限
+				continue
+			}
+			data2 = append(data2, data[i]) // 延迟满足条件时，添加到新数组中
 		}
 		data = data2
 		data2 = []CloudflareIPData{}
@@ -183,34 +184,7 @@ func main() {
 
 	// 开始下载测速
 	if !disableDownload { // 如果禁用下载测速就跳过
-		if len(data) > 0 { // IP数组长度(IP数量) 大于 0 时才会继续下载测速
-			if len(data) < downloadTestCount { // 如果IP数组长度(IP数量) 小于下载测速数量（-dn），则次数修正为IP数
-				downloadTestCount = len(data)
-			}
-			var downloadTestCount2 int // 临时的下载测速次数，即实际的下载测速数量
-			if speedLimit > 0 {
-				downloadTestCount2 = len(data) // 如果指定了 [下载速度下限] 条件，则临时变量改为总数量（即一直测速下去，直到凑够下载测速数量 -dn）
-			} else {
-				downloadTestCount2 = downloadTestCount // 如果没有指定 [下载速度下限] 条件，则临时变量为下载测速数量（-dn）
-			}
-			fmt.Println("开始下载测速（下载速度下限：" + fmt.Sprintf("%.2f", speedLimit) + " MB/s，下载测速数量：" + strconv.Itoa(downloadTestCount) + "，下载测速队列：" + strconv.Itoa(downloadTestCount2) + "）：")
-			bar = pb.Simple.Start(downloadTestCount)
-			for i := 0; i < downloadTestCount2; i++ {
-				_, speed := DownloadSpeedHandler(data[i].ip)
-				data[i].downloadSpeed = speed
-				// 在每个 IP 下载测速后，以 [下载速度下限] 条件过滤结果
-				if float64(speed)/1024/1024 >= speedLimit {
-					data2 = append(data2, data[i]) // 高于下载速度下限时，添加到新数组中
-					bar.Add(1)
-					if len(data2) == downloadTestCount { // 凑够满足条件的 IP 时（下载测速数量 -dn），就跳出循环
-						break
-					}
-				}
-			}
-			bar.Finish()
-		} else {
-			fmt.Println("\n[信息] 延迟测速结果 IP 数量为 0，跳过下载测速。")
-		}
+		testDownloadSpeed(data, data2, bar)
 	}
 
 	if len(data2) > 0 { // 如果该数组有内容，说明指定了 [下载测速下限] 条件，且最少有 1 个满足条件的 IP
@@ -223,49 +197,77 @@ func main() {
 	printResult(data) // 显示最快结果
 }
 
+func testDownloadSpeed(data, data2 []CloudflareIPData, bar *pb.ProgressBar) {
+	if len(data) <= 0 { // IP数组长度(IP数量) 大于 0 时才会继续下载测速
+		fmt.Println("\n[信息] 延迟测速结果 IP 数量为 0，跳过下载测速。")
+		return
+	}
+	if len(data) < downloadTestCount { // 如果IP数组长度(IP数量) 小于下载测速数量（-dn），则次数修正为IP数
+		downloadTestCount = len(data)
+	}
+	// 临时的下载测速次数，即实际的下载测速数量
+	downloadTestCount2 := downloadTestCount
+	// 如果没有指定 [下载速度下限] 条件，则临时变量为下载测速数量（-dn）
+	if speedLimit > 0 {
+		// 如果指定了 [下载速度下限] 条件，则临时变量改为总数量（即一直测速下去，直到凑够下载测速数量 -dn）
+		downloadTestCount2 = len(data)
+	}
+	fmt.Printf("开始下载测速（下载速度下限：%.2f MB/s，下载测速数量：%d，下载测速队列：%d）：\n", speedLimit, downloadTestCount, downloadTestCount2)
+	bar = pb.Simple.Start(downloadTestCount)
+	for i := 0; i < downloadTestCount2; i++ {
+		_, speed := DownloadSpeedHandler(data[i].ip)
+		data[i].downloadSpeed = speed
+		// 在每个 IP 下载测速后，以 [下载速度下限] 条件过滤结果
+		if float64(speed) >= speedLimit*1024*1024 {
+			data2 = append(data2, data[i]) // 高于下载速度下限时，添加到新数组中
+			bar.Add(1)
+			if len(data2) == downloadTestCount { // 凑够满足条件的 IP 时（下载测速数量 -dn），就跳出循环
+				break
+			}
+		}
+	}
+	bar.Finish()
+}
+
 // 显示最快结果
 func printResult(data []CloudflareIPData) {
 	sysType := runtime.GOOS
-	if printResultNum > 0 { // 如果禁止直接输出结果就跳过
-		dateString := convertToString(data) // 转为多维数组 [][]String
-		if len(dateString) > 0 {            // IP数组长度(IP数量) 大于 0 时继续
-			if len(dateString) < printResultNum { // 如果IP数组长度(IP数量) 小于  打印次数，则次数改为IP数量
-				printResultNum = len(dateString)
-			}
-			if ipv6Mode { // IPv6 太长了，所以需要调整一下间隔
-				fmt.Printf("%-40s%-5s%-5s%-5s%-6s%-11s\n", "IP 地址", "已发送", "已接收", "丢包率", "平均延迟", "下载速度 (MB/s)")
-				for i := 0; i < printResultNum; i++ {
-					fmt.Printf("%-42s%-8s%-8s%-8s%-10s%-15s\n", ipPadding(dateString[i][0]), dateString[i][1], dateString[i][2], dateString[i][3], dateString[i][4], dateString[i][5])
-				}
-			} else {
-				fmt.Printf("%-16s%-5s%-5s%-5s%-6s%-11s\n", "IP 地址", "已发送", "已接收", "丢包率", "平均延迟", "下载速度 (MB/s)")
-				for i := 0; i < printResultNum; i++ {
-					fmt.Printf("%-18s%-8s%-8s%-8s%-10s%-15s\n", ipPadding(dateString[i][0]), dateString[i][1], dateString[i][2], dateString[i][3], dateString[i][4], dateString[i][5])
-				}
-			}
-
-			if versionNew != "" {
-				fmt.Println("\n*** 发现新版本 [" + versionNew + "]！请前往 [https://github.com/XIU2/CloudflareSpeedTest] 更新！ ***")
-			}
-
-			if sysType == "windows" { // 如果是 Windows 系统，则需要按下 回车键 或 Ctrl+C 退出（避免通过双击运行时，测速完毕后直接关闭）
-				if outputFile != "" {
-					fmt.Printf("\n完整测速结果已写入 %v 文件，请使用记事本/表格软件查看。\n按下 回车键 或 Ctrl+C 退出。", outputFile)
-				} else {
-					fmt.Printf("\n按下 回车键 或 Ctrl+C 退出。")
-				}
-				var pause int
-				fmt.Scanln(&pause)
-			} else { // 其它系统直接退出
-				if outputFile != "" {
-					fmt.Println("\n完整测速结果已写入 " + outputFile + " 文件，请使用记事本/表格软件查看。")
-				}
-			}
-		} else {
-			fmt.Println("\n[信息] 完整测速结果 IP 数量为 0，跳过输出结果。")
+	if printResultNum <= 0 { // 如果禁止直接输出结果就跳过
+		fmt.Printf("完整测速结果已写入 %v 文件，请使用记事本/表格软件查看。\n", outputFile)
+		return
+	}
+	dateString := convertToString(data) // 转为多维数组 [][]String
+	if len(dateString) <= 0 {           // IP数组长度(IP数量) 大于 0 时继续
+		fmt.Println("\n[信息] 完整测速结果 IP 数量为 0，跳过输出结果。")
+		return
+	}
+	if len(dateString) < printResultNum { // 如果IP数组长度(IP数量) 小于  打印次数，则次数改为IP数量
+		printResultNum = len(dateString)
+	}
+	resHeader := []interface{}{"IP 地址", "已发送", "已接收", "丢包率", "平均延迟", "下载速度 (MB/s)"}
+	if ipv6Mode { // IPv6 太长了，所以需要调整一下间隔
+		fmt.Printf("%-40s%-5s%-5s%-5s%-6s%-11s\n", resHeader...)
+		for i := 0; i < printResultNum; i++ {
+			fmt.Printf("%-42s%-8s%-8s%-8s%-10s%-15s\n", ipPadding(dateString[i][0]), dateString[i][1], dateString[i][2], dateString[i][3], dateString[i][4], dateString[i][5])
 		}
 	} else {
-		fmt.Println("\n完整测速结果已写入 " + outputFile + " 文件，请使用记事本/表格软件查看。")
+		fmt.Printf("%-16s%-5s%-5s%-5s%-6s%-11s\n", resHeader...)
+		for i := 0; i < printResultNum; i++ {
+			fmt.Printf("%-18s%-8s%-8s%-8s%-10s%-15s\n", ipPadding(dateString[i][0]), dateString[i][1], dateString[i][2], dateString[i][3], dateString[i][4], dateString[i][5])
+		}
+	}
+
+	if versionNew != "" {
+		fmt.Printf("\n*** 发现新版本 [%s]！请前往 [https://github.com/XIU2/CloudflareSpeedTest] 更新！ ***\n", versionNew)
+	}
+
+	if outputFile != "" {
+		fmt.Printf("完整测速结果已写入 %v 文件，请使用记事本/表格软件查看。\n", outputFile)
+	}
+	if sysType == "windows" { // 如果是 Windows 系统，则需要按下 回车键 或 Ctrl+C 退出（避免通过双击运行时，测速完毕后直接关闭）
+		fmt.Printf("按下 回车键 或 Ctrl+C 退出。\n")
+		var pause int
+		fmt.Scanln(&pause)
 	}
 }
 
@@ -274,15 +276,17 @@ func checkUpdate() {
 	timeout := time.Duration(10 * time.Second)
 	client := http.Client{Timeout: timeout}
 	res, err := client.Get("https://api.xiuer.pw/ver/cloudflarespeedtest.txt")
-	if err == nil {
-		// 读取资源数据 body: []byte
-		body, err := ioutil.ReadAll(res.Body)
-		// 关闭资源流
-		res.Body.Close()
-		if err == nil {
-			if string(body) != version {
-				versionNew = string(body)
-			}
-		}
+	if err != nil {
+		return
+	}
+	// 读取资源数据 body: []byte
+	body, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return
+	}
+	// 关闭资源流
+	defer res.Body.Close()
+	if string(body) != version {
+		versionNew = string(body)
 	}
 }