From 349f73d88307b37c7eda4d444c0ffda1ad8f6306 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Sat, 30 Mar 2024 20:31:59 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E6=96=87=E4=BB=B6=E4=B8=8A=E4=BC=A0?=
 =?UTF-8?q?=E5=90=8D=E7=A7=B0=E5=8E=BB=E9=99=A4=E7=A9=BA=E6=A0=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/framework/utils/file/file.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/framework/utils/file/file.go b/src/framework/utils/file/file.go
index 4077816c..d0a602ad 100644
--- a/src/framework/utils/file/file.go
+++ b/src/framework/utils/file/file.go
@@ -60,9 +60,12 @@ func uploadWhiteList() []string {
 // fileName 原始文件名称含后缀，如：logo.png
 func generateFileName(fileName string) string {
 	fileExt := filepath.Ext(fileName)
-	// 替换掉后缀和特殊字符保留文件名
+	// 去除后缀
 	newFileName := regular.Replace(fileName, fileExt, "")
-	newFileName = regular.Replace(newFileName, `[<>:"\\|?*]+`, "")
+	// 去除非法字符
+	newFileName = regular.Replace(newFileName, `[\\/:*?"<>|]`, "")
+	// 去除空格
+	newFileName = regular.Replace(newFileName, `\s`, "_")
 	newFileName = strings.TrimSpace(newFileName)
 	return fmt.Sprintf("%s_%s%s", newFileName, generate.Code(6), fileExt)
 }