【add】上传文件类型的黑名单限制

2021-04-14 11:36:29 +08:00
parent c3c919fd7e
commit 7eeb542b89
4 changed files with 32 additions and 0 deletions
@@ -57,6 +57,11 @@ public class CommonController {
     */
    @Value(value="${jero.uploadType}")
    private String uploadType;
+    /**
+     * 文件后缀黑名单
+     */
+    @Value(value="${jero.fileSuffixLimits}")
+    private String[] fileSuffixLimits;

    /**
     * @Author 政辉
@@ -81,6 +86,13 @@ public class CommonController {
        MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
        // 获取上传文件对象
        MultipartFile file = multipartRequest.getFile("file");
+        // 文件类型是否处于黑名单
+        if(CommonUtils.limitFileSuffix(file.getOriginalFilename(),fileSuffixLimits)){
+            result.setMessage("该文件类型不允许上传");
+            result.setSuccess(false);
+            result.setCode(0);
+            return result;
+        }
        if(oConvertUtils.isEmpty(bizPath)){
                bizPath = "";
        }
@@ -94,6 +106,7 @@ public class CommonController {
        if(oConvertUtils.isNotEmpty(savePath)){
            //上传成功 进行数据库存储
            OSSFile ossFile = new OSSFile();
+            // 文件名
            String fileName = file.getOriginalFilename();
            fileName = CommonUtils.getFileName(fileName);
            ossFile.setFileName(fileName);