package com.what21.filter.xss;
import java.io.IOException;
import java.util.LinkedHashMap;
import java.util.Map;
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
public class XSSFilter implements Filter {
// XSS處理Map
private static Map<String,String> xssMap = new LinkedHashMap<String,String>();
public void init(FilterConfig filterConfig) throws ServletException {
// 含有腳本： script
xssMap.put("[s|S][c|C][r|R][i|C][p|P][t|T]", "");
// 含有腳本 javascript
xssMap.put("[\\\"\\\'][\\s]*[j|J][a|A][v|V][a|A][s|S][c|C][r|R][i|I][p|P][t|T]:(.*)[\\\"\\\']", "\"\"");
// 含有函數(shù)： eval
xssMap.put("[e|E][v|V][a|A][l|L]\\((.*)\\)", "");
// 含有符號(hào) <
xssMap.put("<", "<");
// 含有符號(hào) >
xssMap.put(">", ">");
// 含有符號(hào) (
xssMap.put("\\(", "(");
// 含有符號(hào) )
xssMap.put("\\)", ")");
// 含有符號(hào) '
xssMap.put("'", "'");
// 含有符號(hào) "
xssMap.put("\"", "\"");
}
public void doFilter(ServletRequest request, ServletResponse response,
FilterChain chain) throws IOException, ServletException {
// 強(qiáng)制類(lèi)型轉(zhuǎn)換 HttpServletRequest
HttpServletRequest httpReq = (HttpServletRequest)request;
// 構(gòu)造HttpRequestWrapper對(duì)象處理XSS
HttpRequestWrapper httpReqWarp = new HttpRequestWrapper(httpReq,xssMap);
//
chain.doFilter(httpReqWarp, response);
}
public void destroy() {
}
}
=============================
package com.what21.filter.xss;
import java.util.Map;
import java.util.Set;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
public final class HttpRequestWrapper extends HttpServletRequestWrapper {
private Map<String, String> xssMap;
public HttpRequestWrapper(HttpServletRequest request) {
super(request);
}
public HttpRequestWrapper(HttpServletRequest request,
Map<String, String> xssMap) {
super(request);
this.xssMap = xssMap;
}
@Override
public String[] getParameterValues(String parameter) {
String[] values = super.getParameterValues(parameter);
if (values == null) {
return null;
}
int count = values.length;
// 遍歷每一個(gè)參數(shù)尊勿，檢查是否含有
String[] encodedValues = new String[count];
for (int i = 0; i < count; i++) {
encodedValues[i] = cleanXSS(values[i]);
}
return encodedValues;
}
@Override
public String getParameter(String parameter) {
String value = super.getParameter(parameter);
if (value == null) {
return null;
}
return cleanXSS(value);
}
public String getHeader(String name) {
String value = super.getHeader(name);
if (value == null)
return null;
return cleanXSS(value);
}
/**
* 清除惡意的XSS腳本
*
* @param value
* @return
*/
private String cleanXSS(String value) {
Set<String> keySet = xssMap.keySet();
for(String key : keySet){
String v = xssMap.get(key);
value = value.replaceAll(key,v);
}
return value;
}
}
=============================================================================
<filter>
<filter-name>XSSFilter</filter-name>
<filter-class>com.what21.filter.xss.XSSFilter</filter-class>
</filter>
<filter-mapping>
<filter-name>XSSFilter</filter-name>
<url-pattern>/*</url-pattern>
</filter-mapping>
