From d67ebba6fca7f96b3788a2b1c4da356f6736fa4f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E5=A4=A9=E5=B0=A7?=
 <10594804+zhangtianyao1578@user.noreply.gitee.com>
Date: Thu, 6 Feb 2025 10:22:27 +0800
Subject: [PATCH] =?UTF-8?q?1.=E5=A2=9E=E5=8A=A04503=E9=94=99=E8=AF=AF?=
 =?UTF-8?q?=E6=8F=90=E7=A4=BA=202.=E4=B8=8A=E4=BC=A0=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E6=94=B9=E4=B8=BA=E6=8C=87=E5=AE=9A=E5=90=8E=E7=BC=80=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/xqopen/kehui/exception/ErrorCode.java   |  2 +-
 .../java/com/xqopen/kehui/util/FileUploadUtil.java   | 12 +++++++++---
 src/main/resources/i18n/messages_zh_CN.properties    |  1 +
 3 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/xqopen/kehui/exception/ErrorCode.java b/src/main/java/com/xqopen/kehui/exception/ErrorCode.java
index 4c109dd..679a09c 100644
--- a/src/main/java/com/xqopen/kehui/exception/ErrorCode.java
+++ b/src/main/java/com/xqopen/kehui/exception/ErrorCode.java
@@ -103,7 +103,7 @@ public class ErrorCode {
     public static final int UPLOAD_FILE_EMPTY = 4501;
 
     public static final int UPLOAD_FILE_ERORR = 4502;
-
+    public static final int UPLOAD_FILE_ERORR_FORMAT = 4503;
 
     //后勤管理[5000,5500)
 
diff --git a/src/main/java/com/xqopen/kehui/util/FileUploadUtil.java b/src/main/java/com/xqopen/kehui/util/FileUploadUtil.java
index 8889695..2b582fd 100644
--- a/src/main/java/com/xqopen/kehui/util/FileUploadUtil.java
+++ b/src/main/java/com/xqopen/kehui/util/FileUploadUtil.java
@@ -20,11 +20,17 @@ public class FileUploadUtil {
         if (file.isEmpty()) {
             throw new XQException(ErrorCode.UPLOAD_FILE_EMPTY, null);
         }
-        String filename = file.getOriginalFilename();
+        String fileName = file.getOriginalFilename();
+//        String[] fileNameList={".jpg",".png",".jpeg",".pdf",".docx",".doc",".txt",".xlsx",".xls",".rar",".zip",".7z"};
         //后缀名固定
-        if(!(filename.endsWith(".jpg")||filename.endsWith(".png")||filename.endsWith(".jpeg")||filename.endsWith(".pdf")||filename.endsWith(".docx")||filename.endsWith(".doc"))){
-            throw new XQException(ErrorCode.UPLOAD_FILE_ERORR, null);
+        if(!(fileName.endsWith(".jpg")||fileName.endsWith(".png")||fileName.endsWith(".jpeg")
+                ||fileName.endsWith(".docx")||fileName.endsWith(".doc")||fileName.endsWith(".txt")||fileName.endsWith(".pdf")
+                ||fileName.endsWith(".wps")||fileName.endsWith(".wpt")||fileName.endsWith(".dot")||fileName.endsWith(".rtf")
+                ||fileName.endsWith(".xlsx")|fileName.endsWith(".xls"))
+                ||fileName.endsWith(".rar")||fileName.endsWith("zip")||fileName.endsWith(".7z")){
+            throw new XQException(ErrorCode.UPLOAD_FILE_ERORR_FORMAT, null);
         }
+
         StringBuffer sb = new StringBuffer();
 
         String originalFileName = file.getOriginalFilename();
diff --git a/src/main/resources/i18n/messages_zh_CN.properties b/src/main/resources/i18n/messages_zh_CN.properties
index d479005..a6d5bd5 100644
--- a/src/main/resources/i18n/messages_zh_CN.properties
+++ b/src/main/resources/i18n/messages_zh_CN.properties
@@ -44,6 +44,7 @@
 #\u6587\u4EF6\u4E0A\u4F20[4500,5000)
 4501=\u4E0A\u4F20\u6587\u4EF6\u4E3A\u7A7A
 4502=\u4E0A\u4F20\u6587\u4EF6\u51FA\u9519
+4503=\u6587\u4ef6\u7c7b\u578b\u4e0d\u88ab\u5141\u8bb8
 #\u540E\u52E4:\u63A5\u5F85\u7BA1\u7406 [5000,5100)
 5000=\u5185\u5BB9\u4E0D\u80FD\u4E3A\u7A7A
 5001=\u63A5\u5F85\u6E05\u5355\u4E0D\u80FD\u4E3A\u7A7A