diff --git a/server/src/main/java/cn/keking/utils/DownloadUtils.java b/server/src/main/java/cn/keking/utils/DownloadUtils.java index 15af9ee8..58f2da7d 100644 --- a/server/src/main/java/cn/keking/utils/DownloadUtils.java +++ b/server/src/main/java/cn/keking/utils/DownloadUtils.java @@ -49,6 +49,12 @@ public class DownloadUtils { } ReturnResponse response = new ReturnResponse<>(0, "下载成功!!!", ""); String realPath = getRelFilePath(fileName, fileAttribute); + if(!StringUtils.hasText(realPath)){ + response.setCode(1); + response.setContent(null); + response.setMsg("下载失败:文件名不合法!" + urlStr); + return response; + } if (!KkFileUtils.isAllowedUpload(realPath)) { response.setCode(1); response.setContent(null); @@ -61,12 +67,6 @@ public class DownloadUtils { response.setMsg(fileName); return response; } - if(!StringUtils.hasText(realPath)){ - response.setCode(1); - response.setContent(null); - response.setMsg("下载失败:文件名不合法!" + urlStr); - return response; - } if(realPath.equals("cunzhai")){ response.setContent(fileDir + fileName); response.setMsg(fileName);