diff --git a/server/src/main/java/cn/keking/utils/KkFileUtils.java b/server/src/main/java/cn/keking/utils/KkFileUtils.java index aa7a8bfe..b4785565 100644 --- a/server/src/main/java/cn/keking/utils/KkFileUtils.java +++ b/server/src/main/java/cn/keking/utils/KkFileUtils.java @@ -1,7 +1,9 @@ package cn.keking.utils; +import cn.keking.config.ConfigConstants; import org.slf4j.Logger; import org.slf4j.LoggerFactory; +import org.springframework.util.ObjectUtils; import org.springframework.util.StringUtils; import org.springframework.web.util.HtmlUtils; @@ -174,4 +176,19 @@ public class KkFileUtils { return true; } + /** + * 判断文件是否允许上传 + * + * @param file 文件扩展名 + * @return 是否允许上传 + */ + public static boolean isAllowedUpload(String file) { + String fileType = suffixFromFileName(file); + for (String type : ConfigConstants.getprohibit()) { + if (type.equals(fileType)) + return false; + } + return !ObjectUtils.isEmpty(fileType); + } + }