Merge 8e984033dd into 176271d5bb

README.md

@@ -24,7 +24,7 @@
 
 ### 下载运行
 
-1. 下载编译好的可执行文件（ [Github Releases](https://github.com/XIU2/CloudflareSpeedTest/releases) / [蓝奏云](https://pan.lanzouj.com/b0742hkxe) ）并解压。  
+1. 下载编译好的可执行文件（ [Github Releases](https://github.com/XIU2/CloudflareSpeedTest/releases) / [蓝奏云](https://pan.lanpw.com/b0742hkxe) ）并解压。  
 2. 双击运行 `CloudflareST.exe` 文件（Windows 系统），等待测速完成...
 
 <details>

main.go

@@ -92,7 +92,7 @@ https://github.com/XIU2/CloudflareSpeedTest
 	flag.Float64Var(&task.MinSpeed, "sl", 0, "下载速度下限")
 
 	flag.IntVar(&utils.PrintNum, "p", 10, "显示结果数量")
-	flag.StringVar(&task.IPFile, "f", "ip.txt", "IP段数据文件")
+	flag.StringVar(&task.IPFile, "f", "", "IP段数据文件")
 	flag.StringVar(&task.IPText, "ip", "", "指定IP段数据")
 	flag.StringVar(&utils.Output, "o", "result.csv", "输出结果文件")
 

task/ip.go

@@ -9,15 +9,28 @@ import (
 	"strconv"
 	"strings"
 	"time"
+	"path/filepath"
 )
 
 const defaultInputFile = "ip.txt"
 
+func getDefaultInputFile() string {
+	exe, err := os.Executable()
+	if err != nil {
+		log.Fatal(err)
+	}
+	sym, err := filepath.EvalSymlinks(exe)
+	if err != nil {
+		log.Fatal(err)
+	}
+	return filepath.Join(filepath.Dir(sym), defaultInputFile)
+}
+
 var (
 	// TestAll test all ip
 	TestAll = false
 	// IPFile is the filename of IP Rangs
-	IPFile = defaultInputFile
+	IPFile = getDefaultInputFile()
 	IPText string
 )
 
@@ -165,7 +178,11 @@ func loadIPRanges() []*net.IPAddr {
 		}
 	} else { // 从文件中获取 IP 段数据
 		if IPFile == "" {
-			IPFile = defaultInputFile
+			if _, err := os.Stat(defaultInputFile); err != nil {
+				IPFile = getDefaultInputFile()
+			} else {
+				IPFile = defaultInputFile
+			}
 		}
 		file, err := os.Open(IPFile)
 		if err != nil {