高位漏洞修改
zhangshengli
6 months ago
parent
2a1078a9ae
commit
bce113666c
@ -0,0 +1,109 @@
|
|||||||
|
package com.fuint.common.filter;
|
||||||
|
|
||||||
|
|
||||||
|
import com.alipay.api.internal.util.file.Charsets;
|
||||||
|
|
||||||
|
import javax.servlet.ReadListener;
|
||||||
|
import javax.servlet.ServletInputStream;
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
import javax.servlet.http.HttpServletRequestWrapper;
|
||||||
|
import java.io.*;
|
||||||
|
import java.util.Enumeration;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <p> @Title MyRequestWrapper
|
||||||
|
* <p> @Description 用于过滤器中获取POST请求参数
|
||||||
|
*
|
||||||
|
* @author zhj
|
||||||
|
* @date 2021/10/28 9:52
|
||||||
|
*/
|
||||||
|
public class MyRequestWrapper extends HttpServletRequestWrapper {
|
||||||
|
private String body;
|
||||||
|
public MyRequestWrapper(HttpServletRequest request) throws IOException {
|
||||||
|
super(request);
|
||||||
|
StringBuilder stringBuilder = new StringBuilder();
|
||||||
|
BufferedReader bufferedReader = null;
|
||||||
|
try {
|
||||||
|
InputStream inputStream = request.getInputStream();
|
||||||
|
if (inputStream != null) {
|
||||||
|
bufferedReader = new BufferedReader(new InputStreamReader(inputStream,"UTF-8"));
|
||||||
|
char[] charBuffer = new char[128];
|
||||||
|
int bytesRead = -1;
|
||||||
|
while ((bytesRead = bufferedReader.read(charBuffer)) > 0) {
|
||||||
|
stringBuilder.append(charBuffer, 0, bytesRead);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
stringBuilder.append("");
|
||||||
|
}
|
||||||
|
} catch (IOException ex) {
|
||||||
|
throw ex;
|
||||||
|
} finally {
|
||||||
|
if (bufferedReader != null) {
|
||||||
|
try {
|
||||||
|
bufferedReader.close();
|
||||||
|
} catch (IOException ex) {
|
||||||
|
throw ex;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
body = stringBuilder.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ServletInputStream getInputStream() throws IOException {
|
||||||
|
final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes("UTF-8"));
|
||||||
|
ServletInputStream servletInputStream = new ServletInputStream() {
|
||||||
|
@Override
|
||||||
|
public boolean isFinished() {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isReady() {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void setReadListener(ReadListener readListener) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int read() {
|
||||||
|
return byteArrayInputStream.read();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
return servletInputStream;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BufferedReader getReader() throws IOException {
|
||||||
|
return new BufferedReader(new InputStreamReader(this.getInputStream(), Charsets.UTF_8));
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getBody() {
|
||||||
|
return this.body;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getParameter(String name) {
|
||||||
|
return super.getParameter(name);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Map<String, String[]> getParameterMap() {
|
||||||
|
return super.getParameterMap();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Enumeration<String> getParameterNames() {
|
||||||
|
return super.getParameterNames();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String[] getParameterValues(String name) {
|
||||||
|
return super.getParameterValues(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@ -0,0 +1,101 @@
|
|||||||
|
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;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue