使用拦截器Filter修补输入框的xss漏洞

package cmcc.jx.ict.xsgj.filter.xss;


import java.io.IOException;
import java.util.Iterator;
import java.util.List;

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;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileItemFactory;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

public class XssFilter implements Filter
{
	FilterConfig	filterConfig	= null;

	public void init(FilterConfig filterConfig) throws ServletException
	{
		this.filterConfig = filterConfig;
	}

	public void destroy()
	{
		this.filterConfig = null;
	}

	public void doFilter(ServletRequest request, ServletResponse response,
			FilterChain chain) throws IOException, ServletException {
		chain.doFilter(new XssHttpServletRequestWrapper(
				(HttpServletRequest) request), response);
	}
}

 

package cmcc.jx.ict.xsgj.filter.xss;

import java.io.ObjectInputStream.GetField;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;

import org.apache.ibatis.reflection.invoker.GetFieldInvoker;

public class XssHttpServletRequestWrapper extends HttpServletRequestWrapper
{
	public XssHttpServletRequestWrapper(HttpServletRequest servletRequest)
	{
		super(servletRequest);
	}
	
    public Map getParameterMap() {

        HashMap paramMap = (HashMap) super.getParameterMap();
        paramMap = (HashMap) paramMap.clone();

        for (Iterator iterator = paramMap.entrySet().iterator(); iterator.hasNext(); ) 
        {
            Map.Entry entry = (Map.Entry) iterator.next();
            String [] values = (String[])entry.getValue();
            for (int i = 0; i < values.length; i++) {
                if(values[i] instanceof String){
                    values[i] = cleanXSS(values[i]);
                }
            }
            entry.setValue(values);
        }
        return paramMap;
    }
    
    @Override  
    public String[] getParameterValues(String name) {  
        String[] values = super.getParameterValues(name);  
        if (values == null) {  
            return null;  
        }  
        String[] newValues = new String[values.length];  
  
        for (int i = 0; i < values.length; i++) {  
            newValues[i] = cleanXSS(values[i]);  
        }  
  
        return newValues;  
    }  
	
	public String getParameter(String parameter)
	{
		super.getParameterMap();
		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);
	}
	
	public String getQueryString()
	{
		String str = super.getQueryString();
		if(str == null)
			return null;
		return cleanXSS(str);
	}

	private String cleanXSS(String value)
	{
		 if (value == null || "".equals(value)) {  
	            return value;  
	        }  
		 	value = value.replaceAll("<", "<").replaceAll(">", ">");
	        value = value.replaceAll("\\(", "(").replaceAll("\\)", ")");
	        value = value.replaceAll("'", "'");
	        value = value.replaceAll("\"", """);
	        return value;  
	}

}

 

  <filter>
  	<filter-name>xssFilter</filter-name>
  	<filter-class>cmcc.jx.ict.xsgj.filter.xss.XssFilter</filter-class>
  </filter>
  
  <filter-mapping>
  	<filter-name>xssFilter</filter-name>
  	<url-pattern>/*</url-pattern>
  </filter-mapping>