gitee
/
file-online-preview
mirror of https://gitee.com/kekingcn/file-online-preview
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

feat: 新增预览文件 host 黑名单机制

pull/275/head
kl 2024-03-08 10:12:05 +08:00
parent d1d8ffef7a
commit d407b88b67
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
server/src/main/java/cn/keking/web/filter/TrustHostFilter.java
View File

@ -2,6 +2,7 @@ package cn.keking.web.filter;
import cn.keking.config.ConfigConstants; import cn.keking.config.ConfigConstants;
import cn.keking.utils.WebUtils; import cn.keking.utils.WebUtils;
import java.io.IOException; import java.io.IOException;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
import javax.servlet.Filter; import javax.servlet.Filter;
@ -10,6 +11,8 @@ import javax.servlet.FilterConfig;
import javax.servlet.ServletException; import javax.servlet.ServletException;
import javax.servlet.ServletRequest; import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse; import javax.servlet.ServletResponse;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.core.io.ClassPathResource; import org.springframework.core.io.ClassPathResource;
import org.springframework.util.FileCopyUtils; import org.springframework.util.FileCopyUtils;
@ -38,7 +41,7 @@ public class TrustHostFilter implements Filter {
String url = WebUtils.getSourceUrl(request); String url = WebUtils.getSourceUrl(request);
String host = WebUtils.getHost(url); String host = WebUtils.getHost(url);
assert host != null; assert host != null;
if (!isTrustHost(host) || isNotTrustHost(host)) { if (isNotTrustHost(host)) {
String html = this.notTrustHostHtmlView.replace("${current_host}", host); String html = this.notTrustHostHtmlView.replace("${current_host}", host);
response.getWriter().write(html); response.getWriter().write(html);
response.getWriter().close(); response.getWriter().close();
@ -47,12 +50,14 @@ public class TrustHostFilter implements Filter {
} }
} }
public boolean isTrustHost(String host) {
return !ConfigConstants.getTrustHostSet().isEmpty() && ConfigConstants.getTrustHostSet().contains(host);
}
public boolean isNotTrustHost(String host) { public boolean isNotTrustHost(String host) {
return !ConfigConstants.getNotTrustHostSet().isEmpty() && ConfigConstants.getNotTrustHostSet().contains(host); if (CollectionUtils.isNotEmpty(ConfigConstants.getNotTrustHostSet())) {
return ConfigConstants.getNotTrustHostSet().contains(host);
}
if (CollectionUtils.isNotEmpty(ConfigConstants.getTrustHostSet())) {
return !ConfigConstants.getTrustHostSet().contains(host);
}
return false;
} }
@Override @Override