|
@ -0,0 +1,55 @@
|
|
|
package com.yihu.jw.base.filter;
|
|
|
|
|
|
|
|
|
import com.yihu.jw.util.XssUtil;
|
|
|
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
|
import javax.servlet.http.HttpServletRequestWrapper;
|
|
|
|
|
|
/**
|
|
|
* Created by yeshijie on 2020/8/27.
|
|
|
*/
|
|
|
public class XssAndHttpServletRequestWrapper extends HttpServletRequestWrapper {
|
|
|
HttpServletRequest request;
|
|
|
|
|
|
public XssAndHttpServletRequestWrapper(HttpServletRequest request) {
|
|
|
super(request);
|
|
|
this.request = request;
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public String getParameter(String name) {
|
|
|
String value = super.getParameter(name);
|
|
|
// System.out.println("name:" + name + "," + value);
|
|
|
if (value != null) {
|
|
|
// 转换Html
|
|
|
value = XssUtil.cleanXss(value);
|
|
|
}
|
|
|
return value;
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public String getHeader(String name) {
|
|
|
return XssUtil.cleanXss(super.getHeader(name));
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public String getQueryString() {
|
|
|
return XssUtil.cleanXss(super.getQueryString());
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public String[] getParameterValues(String parameter) {
|
|
|
String[] values = super.getParameterValues(parameter);
|
|
|
if (values == null) {
|
|
|
return null;
|
|
|
}
|
|
|
int count = values.length;
|
|
|
String[] encodedValues = new String[count];
|
|
|
for (int i = 0; i < count; i++) {
|
|
|
encodedValues[i] = XssUtil.cleanXss(values[i]);
|
|
|
}
|
|
|
return encodedValues;
|
|
|
}
|
|
|
|
|
|
}
|