解决xss漏洞

小咪咪 2022-02-13 13:15 427阅读 0赞

web.xml

<filter>
    		<filter-name>xssFilter</filter-name>
    		<filter-class>com.aoi.selfhelp.filters.XSSFilter</filter-class>
    	</filter>
    
    	
    	<filter-mapping>
    		<filter-name>xssFilter</filter-name>
    		<url-pattern>/*</url-pattern>
    	</filter-mapping>

package com.aoi.selfhelp.filters;
    import java.io.IOException;
    
    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 com.cmcc.aoi.selfhelp.util.XssHttpServletRequestWrapper;
      
    public class XSSFilter implements Filter {  
      
        @Override  
        public void init(FilterConfig filterConfig) throws ServletException {  
        }  
      
        @Override  
        public void doFilter(ServletRequest request, ServletResponse response,  
                FilterChain chain) throws IOException, ServletException {  
      
            XssHttpServletRequestWrapper xssRequest = new XssHttpServletRequestWrapper(  
                    (HttpServletRequest) request);  
            chain.doFilter(xssRequest, response);  
        }  
      
        @Override  
        public void destroy() {  
        }  
      
    }

package com.aoi.selfhelp.util;
    
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletRequestWrapper;
    
    public class XssHttpServletRequestWrapper extends HttpServletRequestWrapper
    {
    	HttpServletRequest orgRequest = null;
    
    	public XssHttpServletRequestWrapper(HttpServletRequest request)
    	{
    		super(request);
    		orgRequest = request;
    	}
    
    	@Override
    	public String[] getParameterValues(String name)
    	{
    		String[] parameterValues = super.getParameterValues(name);
    		if (parameterValues == null)
    			parameterValues = new String[] {};
    		for (int i = 0; i < parameterValues.length; i++)
    		{
    			String value = parameterValues[i];
    			if (value != null)
    			{
    				parameterValues[i] = xssEncode(value);
    			}
    		}
    		return parameterValues;
    	}
    
    	/**
    	 * 覆盖getParameter方法，将参数名和参数值都做xss过滤。<br/>
    	 * 如果需要获得原始的值，则通过super.getParameterValues(name)来获取<br/>
    	 * getParameterNames,getParameterValues和getParameterMap也可能需要覆盖
    	 */
    	@Override
    	public String getParameter(String name)
    	{
    		String value = super.getParameter(xssEncode(name));
    		if (value != null)
    		{
    			value = xssEncode(value);
    		}
    		return value;
    	}
    
    	/**
    	 * 覆盖getHeader方法，将参数名和参数值都做xss过滤。<br/>
    	 * 如果需要获得原始的值，则通过super.getHeaders(name)来获取<br/>
    	 * getHeaderNames 也可能需要覆盖
    	 */
    	@Override
    	public String getHeader(String name)
    	{
    
    		String value = super.getHeader(xssEncode(name));
    		if (value != null)
    		{
    			value = xssEncode(value);
    		}
    		return value;
    	}
    
    	/**
    	 * 将容易引起xss漏洞的半角字符直接替换成全角字符
    	 * 
    	 * @param s
    	 * @return
    	 */
    	private static String xssEncode(String s)
    	{
    		if (s == null || s.isEmpty())
    		{
    			return s;
    		}
    		StringBuilder sb = new StringBuilder(s.length() + 16);
    		for (int i = 0; i < s.length(); i++)
    		{
    			char c = s.charAt(i);
    			switch (c)
    			{
    				case '>':
    					sb.append("＞");// 转义大于号
    					break;
    				case '<':
    					sb.append("＜");// 转义小于号
    					break;
    				case '\'':
    					sb.append("＇");// 转义单引号
    					break;
    				case '\"':
    					sb.append("＂");// 转义双引号
    					break;
    				// case '&':
    				//	sb.append("＆");// 转义&
    				//	break;
    				default:
    					sb.append(c);
    					break;
    			}
    		}
    		return sb.toString();
    	}
    
    	/**
    	 * 获取最原始的request
    	 * 
    	 * @return
    	 */
    	public HttpServletRequest getOrgRequest()
    	{
    		return orgRequest;
    	}
    
    	/**
    	 * 获取最原始的request的静态方法
    	 * 
    	 * @return
    	 */
    	public static HttpServletRequest getOrgRequest(HttpServletRequest req)
    	{
    		if (req instanceof XssHttpServletRequestWrapper)
    		{
    			return ((XssHttpServletRequestWrapper) req).getOrgRequest();
    		}
    
    		return req;
    	}
    }

## 捐助开发者 ##

在兴趣的驱动下,写一个`免费`的东西，有欣喜，也还有汗水，希望你喜欢我的作品，同时也能支持一下。 当然，有钱捧个钱场（右上角的爱心标志，支持支付宝和PayPal捐助），没钱捧个人场，谢谢各位。

![img_81e700ed90042f6a809cc76caff4b5a2.png][]![img_74a44b81a2d1f6112585a18c54f87eb5.jpeg][]![img_4979b56769b1a2715573f5ace62e9135.png][]  
  
  
谢谢您的赞助，我会做的更好！

[img_81e700ed90042f6a809cc76caff4b5a2.png]: https://yqfile.alicdn.com/img_81e700ed90042f6a809cc76caff4b5a2.png
[img_74a44b81a2d1f6112585a18c54f87eb5.jpeg]: https://yqfile.alicdn.com/img_74a44b81a2d1f6112585a18c54f87eb5.jpeg
[img_4979b56769b1a2715573f5ace62e9135.png]: /images/20220213/ff898ede33c84e3f89c1481eb1dfce9a.png