Browse Source

新增备案信息,禁止上传类型,删除支持密码功能
新增备案信息,禁止上传类型,删除支持密码功能

Signed-off-by: 高雄 <admin@cxcp.com>

高雄 2 years ago
parent
commit
7b699e08fc
1 changed files with 6 additions and 0 deletions
  1. 6 0
      server/src/main/java/cn/keking/utils/DownloadUtils.java

+ 6 - 0
server/src/main/java/cn/keking/utils/DownloadUtils.java

@@ -49,6 +49,12 @@ public class DownloadUtils {
             logger.error("忽略SSL证书异常:", e);
         }
         ReturnResponse<String> response = new ReturnResponse<>(0, "下载成功!!!", "");
+        if (!KkFileUtils.isAllowedUpload(fileName)) {
+            response.setCode(1);
+            response.setContent(null);
+            response.setMsg("下载失败:不支持的类型!" + urlStr);
+            return response;
+        }
         assert urlStr != null;
         if (urlStr.contains("?fileKey=")) {
             response.setContent(fileDir + urlStrr);