高位漏洞修改
zhangshengli
2 months ago
parent
0cc0f6d9c1
commit
4b33b0695f
@ -1,101 +1,118 @@
|
||||
package com.fuint.common.filter;
|
||||
|
||||
|
||||
import cn.hutool.core.util.ObjectUtil;
|
||||
import com.alibaba.fastjson2.JSONObject;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import com.fuint.framework.exception.BusinessRuntimeException;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.springframework.web.multipart.support.StandardServletMultipartResolver;
|
||||
|
||||
import javax.servlet.*;
|
||||
import javax.servlet.annotation.WebFilter;
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.io.BufferedReader;
|
||||
import java.io.IOException;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import java.util.TreeMap;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
|
||||
@WebFilter(filterName = "CharsetFilter",urlPatterns = "/*")
|
||||
public class SpecialCharFilter implements Filter {
|
||||
|
||||
private static final String SQL_REGX = "[\\\\^$*+?{}()=&;%+\\[\\].|]";
|
||||
|
||||
@Override
|
||||
public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain chain) throws IOException, ServletException, IOException {
|
||||
HttpServletRequest req = (HttpServletRequest) servletRequest;
|
||||
|
||||
// 防止流读取一次后就没有了, 所以需要将流继续写出去
|
||||
MyRequestWrapper requestWrapper = new MyRequestWrapper(req);
|
||||
|
||||
// 获取请求参数
|
||||
Map<String, Object> paramsMaps = new TreeMap<>();
|
||||
if ("POST".equals(req.getMethod().toUpperCase())) {
|
||||
String body = requestWrapper.getBody();
|
||||
paramsMaps = JSONObject.parseObject(body, TreeMap.class);
|
||||
} else {
|
||||
Map<String, String[]> parameterMap = requestWrapper.getParameterMap();
|
||||
Set<Map.Entry<String, String[]>> entries = parameterMap.entrySet();
|
||||
for (Map.Entry<String, String[]> next : entries) {
|
||||
paramsMaps.put(next.getKey(), next.getValue()[0]);
|
||||
}
|
||||
}
|
||||
|
||||
// 校验SQL注入
|
||||
if (ObjectUtil.isNotEmpty(paramsMaps)) {
|
||||
for (Object o : paramsMaps.entrySet()) {
|
||||
Map.Entry entry = (Map.Entry) o;
|
||||
Object value = entry.getValue();
|
||||
if (value != null) {
|
||||
boolean isValid = checkSqlInject(value.toString(), servletResponse);
|
||||
if (!isValid) {
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
chain.doFilter(requestWrapper, servletResponse);
|
||||
|
||||
|
||||
}
|
||||
|
||||
//获取request请求body中参数
|
||||
public static String getBodyString(BufferedReader br) {
|
||||
String inputLine;
|
||||
String str = "";
|
||||
try {
|
||||
while ((inputLine = br.readLine()) != null) {
|
||||
str += inputLine;
|
||||
}
|
||||
br.close();
|
||||
} catch (IOException e) {
|
||||
System.out.println("IOException: " + e);
|
||||
}
|
||||
return str;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查SQL注入
|
||||
*
|
||||
* @param value 参数值
|
||||
* @param servletResponse 相应实例
|
||||
* @throws IOException IO异常
|
||||
*/
|
||||
private boolean checkSqlInject(String value, ServletResponse servletResponse) throws IOException {
|
||||
if (null != value) {
|
||||
String output = value.replaceAll(SQL_REGX, "");
|
||||
if (output.length()<value.length()) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
|
||||
}
|
||||
//package com.fuint.common.filter;
|
||||
//
|
||||
//
|
||||
//import cn.hutool.core.util.ObjectUtil;
|
||||
//import com.alibaba.fastjson2.JSONObject;
|
||||
//import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
//import com.fuint.framework.exception.BusinessRuntimeException;
|
||||
//import org.apache.commons.lang3.StringUtils;
|
||||
//import org.springframework.beans.factory.annotation.Autowired;
|
||||
//import org.springframework.beans.factory.annotation.Qualifier;
|
||||
//import org.springframework.stereotype.Component;
|
||||
//import org.springframework.web.multipart.support.StandardServletMultipartResolver;
|
||||
//import org.springframework.web.servlet.HandlerExceptionResolver;
|
||||
//
|
||||
//import javax.servlet.*;
|
||||
//import javax.servlet.annotation.WebFilter;
|
||||
//import javax.servlet.http.HttpServletRequest;
|
||||
//import javax.servlet.http.HttpServletResponse;
|
||||
//import java.io.BufferedReader;
|
||||
//import java.io.IOException;
|
||||
//import java.util.Map;
|
||||
//import java.util.Set;
|
||||
//import java.util.TreeMap;
|
||||
//import java.util.stream.Collectors;
|
||||
//
|
||||
//
|
||||
//@WebFilter(filterName = "CharsetFilter", urlPatterns = "/*")
|
||||
//@Component
|
||||
//public class SpecialCharFilter implements Filter {
|
||||
//
|
||||
// private static final String SQL_REGX = "[=]";
|
||||
// @Qualifier("handlerExceptionResolver")
|
||||
// @Autowired
|
||||
// private HandlerExceptionResolver resolver;
|
||||
//
|
||||
// @Override
|
||||
// public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain chain) throws IOException, ServletException, IOException {
|
||||
// HttpServletRequest req = (HttpServletRequest) servletRequest;
|
||||
// String servletPath = req.getServletPath();
|
||||
// HttpServletResponse rep = (HttpServletResponse) servletResponse;
|
||||
// MyRequestWrapper requestWrapper = new MyRequestWrapper(req);
|
||||
// if (ObjectUtil.notEqual(servletPath,"/backendApi/login/doLogin")){
|
||||
// // 防止流读取一次后就没有了, 所以需要将流继续写出去
|
||||
//
|
||||
// // 获取请求参数
|
||||
// Map<String, Object> paramsMaps = new TreeMap<>();
|
||||
// if ("POST".equals(req.getMethod().toUpperCase())) {
|
||||
// String body = requestWrapper.getBody();
|
||||
// paramsMaps = JSONObject.parseObject(body, TreeMap.class);
|
||||
// } else {
|
||||
// Map<String, String[]> parameterMap = requestWrapper.getParameterMap();
|
||||
// Set<Map.Entry<String, String[]>> entries = parameterMap.entrySet();
|
||||
// for (Map.Entry<String, String[]> next : entries) {
|
||||
// paramsMaps.put(next.getKey(), next.getValue()[0]);
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// // 校验SQL注入
|
||||
// if (ObjectUtil.isNotEmpty(paramsMaps)) {
|
||||
// for (Object o : paramsMaps.entrySet()) {
|
||||
// Map.Entry entry = (Map.Entry) o;
|
||||
// Object value = entry.getValue();
|
||||
// if (value != null) {
|
||||
//
|
||||
// boolean isValid = checkSqlInject(value.toString(), servletResponse);
|
||||
// if (!isValid) {
|
||||
// resolver.resolveException(req, rep,
|
||||
// null, FileterException("客户端信息非法！！,存在特殊字符请重新输入"));
|
||||
// return;
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// chain.doFilter(requestWrapper, servletResponse);
|
||||
//
|
||||
//
|
||||
// }
|
||||
//
|
||||
// //获取request请求body中参数
|
||||
// public static String getBodyString(BufferedReader br) {
|
||||
// String inputLine;
|
||||
// String str = "";
|
||||
// try {
|
||||
// while ((inputLine = br.readLine()) != null) {
|
||||
// str += inputLine;
|
||||
// }
|
||||
// br.close();
|
||||
// } catch (IOException e) {
|
||||
// System.out.println("IOException: " + e);
|
||||
// }
|
||||
// return str;
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 检查SQL注入
|
||||
// *
|
||||
// * @param value 参数值
|
||||
// * @param servletResponse 相应实例
|
||||
// * @throws IOException IO异常
|
||||
// */
|
||||
// private boolean checkSqlInject(String value, ServletResponse servletResponse) throws IOException {
|
||||
// if (null != value) {
|
||||
// String output = value.replaceAll(SQL_REGX, "");
|
||||
// if (output.length() < value.length()) {
|
||||
// return false;
|
||||
// }
|
||||
// }
|
||||
// return true;
|
||||
// }
|
||||
//
|
||||
// public Exception FileterException(String cause) {
|
||||
// return new Exception(cause);
|
||||
// }
|
||||
//
|
||||
//
|
||||
//}
|
||||
|
||||
@ -0,0 +1,23 @@
|
||||
package com.fuint.framework.exception;
|
||||
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.EqualsAndHashCode;
|
||||
import org.springframework.http.HttpStatus;
|
||||
import org.springframework.web.bind.annotation.ResponseStatus;
|
||||
|
||||
@EqualsAndHashCode(callSuper = true)
|
||||
@Data
|
||||
@ResponseStatus(HttpStatus.CREATED)
|
||||
public class FileterException extends RuntimeException {
|
||||
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
||||
private String reason;
|
||||
|
||||
public FileterException(String cause) {
|
||||
super(cause);
|
||||
this.reason = cause;
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue