82580-导入文档拆分详情修改

2024-03-08 18:57:41 +08:00
parent edbb3cdfc2
commit 94900c9430
2 changed files with 69 additions and 24 deletions
@@ -300,4 +300,41 @@ public class FileUnZip {
        return resultlist;
    }

+    /**
+     * 验证文件白名单
+     * @param path
+     * @param suffixs
+     * @return
+     */
+    public static boolean checkFileWhitelist(String path,String suffixs) {
+        boolean result = true;
+
+        String[] suffixArr = suffixs.split(",");
+        File file = new File(path);
+
+        if (file.isDirectory()) {
+            File[] files = file.listFiles();
+            // 对文件进行过滤
+            for (File fi : files) {
+                String name = fi.getName();
+
+                int lastIndexOfDot = name.lastIndexOf('.');
+
+                String fileSuffix = name.substring(lastIndexOfDot);
+
+                boolean flag = false;
+                for (String suffix : suffixArr) {
+                    if (fileSuffix.equals(suffix)) {
+                        flag = true;
+                        break;
+                    }
+                }
+                if (!flag) {
+                    result = flag;
+                    break;
+                }
+            }
+        }
+        return result;
+    }
 }
@@ -1543,32 +1543,40 @@ public class FileSplitItemsEOServiceImpl extends ServiceImpl<FileSplitItemsEOMap
            saveDirectory.mkdir();
        }
        FileUtils.copyInputStreamToFile(file.getInputStream(), new File(path +File.separator+ file.getOriginalFilename()));
-        try{
-            //解压缩
-            String zipEntryName = FileUnZip.unZipFiles2(path +File.separator+ file.getOriginalFilename(), path);
-            // 数据相关处理,
-            // 1.获取其中的Excel,
-            List<File> excelfilelist = FileUnZip.readExcelFile(zipEntryName);
-            if(excelfilelist.size()<1){
-                //删除原上传文件
-                FileUnZip.deleteDir(saveDirectory);
-                if(MessageUtils.getLanguage().getValue().equals(LanguageEnum.CN.getValue())){
-                    resultMsg = "压缩包内没有上传Excel数据";
-                }else{
-                    resultMsg = "In the cabinet did not upload the Excel data";
-                }
-                return Result.error(1, resultMsg);

-            } else if (excelfilelist.size()>1) {
-                //删除原上传文件
-                FileUnZip.deleteDir(saveDirectory);
-                if(MessageUtils.getLanguage().getValue().equals(LanguageEnum.CN.getValue())){
-                    resultMsg = "压缩包根目录下仅能存在一个excel为导入数据";
-                }else{
-                    resultMsg = "Only one Excel file can be imported in the compressed package root directory";
-                }
-                return Result.error(1, resultMsg);
+        //解压缩
+        String zipEntryName = FileUnZip.unZipFiles2(path +File.separator+ file.getOriginalFilename(), path);
+        // 数据相关处理,
+        // 1.获取其中的Excel,
+        List<File> excelfilelist = FileUnZip.readExcelFile(zipEntryName);
+        if(excelfilelist.size()<1){
+            //删除原上传文件
+            FileUnZip.deleteDir(saveDirectory);
+            if(MessageUtils.getLanguage().getValue().equals(LanguageEnum.CN.getValue())){
+                resultMsg = "压缩包内没有上传Excel数据";
+            }else{
+                resultMsg = "In the cabinet did not upload the Excel data";
            }
+            return Result.error(1, resultMsg);
+
+        } else if (excelfilelist.size()>1) {
+            //删除原上传文件
+            FileUnZip.deleteDir(saveDirectory);
+            if(MessageUtils.getLanguage().getValue().equals(LanguageEnum.CN.getValue())){
+                resultMsg = "压缩包根目录下仅能存在一个excel为导入数据";
+            }else{
+                resultMsg = "Only one Excel file can be imported in the compressed package root directory";
+            }
+            return Result.error(1, resultMsg);
+        }
+        String suffixs = ".xls,.xlsx,.jpg,.jpeg,.png,.gif";
+        boolean checkFileWhitelistFlag = FileUnZip.checkFileWhitelist(zipEntryName, suffixs);
+        if (!checkFileWhitelistFlag) {
+            throw new JeroBootException("导入的文件中包含不合法文件，图片只能上传 .jpg、.jpeg、.png、.gif 格式");
+        }
+
+        try{
+
            File excelfile = excelfilelist.get(0);

            try {