v0.1.9 - 避免错误文件的使用

oneclickvirt · Jan 27, 2025 · 2f5feac · 2f5feac
1 parent 615ffca
commit 2f5feac
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 4 deletions.
diff --git a/go.mod b/go.mod
@@ -5,7 +5,7 @@ go 1.23.4
 require (
 	github.com/imroc/req/v3 v3.49.0
 	github.com/oneclickvirt/CommonMediaTests v0.0.4-20240704024502
-	github.com/oneclickvirt/UnlockTests v0.0.23-20250118070445
+	github.com/oneclickvirt/UnlockTests v0.0.24-20250127123509
 	github.com/oneclickvirt/backtrace v0.0.4-20240702140722
 	github.com/oneclickvirt/basics v0.0.8-20241108124433
 	github.com/oneclickvirt/cputest v0.0.9-20250103063414

diff --git a/go.sum b/go.sum
@@ -103,8 +103,8 @@ github.com/nxtrace/NTrace-core v1.3.7 h1:ZnTbPrPqpyeraCvUyNbQTNyl4Gz3NRQDh06WdII
 github.com/nxtrace/NTrace-core v1.3.7/go.mod h1:aW2owz9I+W5i+gJEDmnWli75mB+fuO4UTwdOPMcQHpE=
 github.com/oneclickvirt/CommonMediaTests v0.0.4-20240704024502 h1:hRIYJ2uEp2N3AH5bP5X6bwfdwWfZQO/2WoqpUJ8+WsY=
 github.com/oneclickvirt/CommonMediaTests v0.0.4-20240704024502/go.mod h1:DAmFPRjFV5p9fEzUUSml5jJGn2f1NZJQCzTxITHDjc4=
-github.com/oneclickvirt/UnlockTests v0.0.23-20250118070445 h1:lQmYC12cbE6kMNs+onwS+aC+34aXQToSiAf+oOXweBc=
-github.com/oneclickvirt/UnlockTests v0.0.23-20250118070445/go.mod h1:yXWIZB6iLS88pEd9m4QJi1GENn+7I91zA72y5ONz2Oc=
+github.com/oneclickvirt/UnlockTests v0.0.24-20250127123509 h1:QTm/2r+0tKOibglvj/W9RhEqp0sciU916wIZ/xOX6qQ=
+github.com/oneclickvirt/UnlockTests v0.0.24-20250127123509/go.mod h1:yXWIZB6iLS88pEd9m4QJi1GENn+7I91zA72y5ONz2Oc=
 github.com/oneclickvirt/backtrace v0.0.4-20240702140722 h1:UJ/VWf+ZbhGarc9HcHMIyenpmX+b2LxkXu0hlLk3Gxs=
 github.com/oneclickvirt/backtrace v0.0.4-20240702140722/go.mod h1:zvsC7xY/WZqs5KL2JB967OVnuqjNbxu9bW6wXRLo5h8=
 github.com/oneclickvirt/basics v0.0.8-20241108124433 h1:eKZcoNoa9uLWmfisk+qN/QOvSGT4ajCmZdZfIAcHG+o=

diff --git a/goecs.go b/goecs.go
@@ -39,7 +39,7 @@ import (
 )
 
 var (
-	ecsVersion                                                        = "v0.1.8"
+	ecsVersion                                                        = "v0.1.9"
 	menuMode                                                          bool
 	onlyChinaTest                                                     bool
 	input, choice                                                     string

diff --git a/utils/utils.go b/utils/utils.go
@@ -255,11 +255,21 @@ func UploadText(absPath string) (string, string, error) {
 		SetRetryCount(2).
 		SetRetryBackoffInterval(1*time.Second, 5*time.Second).
 		SetRetryFixedInterval(2 * time.Second)
+	// 打开文件
 	file, err := os.Open(absPath)
 	if err != nil {
 		return "", "", fmt.Errorf("failed to open file: %w", err)
 	}
 	defer file.Close()
+	// 获取文件信息并检查大小
+	fileInfo, err := file.Stat()
+	if err != nil {
+		return "", "", fmt.Errorf("failed to get file info: %w", err)
+	}
+	if fileInfo.Size() > 25*1024 { // 25KB
+		return "", "", fmt.Errorf("file size exceeds 25KB limit")
+	}
+	// 上传逻辑
 	upload := func(url string) (string, string, error) {
 		file, err := os.Open(absPath)
 		if err != nil {
@@ -288,10 +298,12 @@ func UploadText(absPath string) (string, string, error) {
 		}
 		return "", "", fmt.Errorf("upload failed for %s with status code: %d", url, resp.StatusCode)
 	}
+	// 尝试上传到主URL
 	httpURL, httpsURL, err := upload(primaryURL)
 	if err == nil {
 		return httpURL, httpsURL, nil
 	}
+	// 尝试上传到备份URL
 	httpURL, httpsURL, err = upload(backupURL)
 	if err != nil {
 		return "", "", fmt.Errorf("failed to upload to both primary and backup URLs: %w", err)