重写HttpServletRequestWrapper

import java.util.HashMap;  
import java.util.Iterator;  
import java.util.Map;  
  
import javax.servlet.http.HttpServletRequest;  
import javax.servlet.http.HttpServletRequestWrapper;  
  
  
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("<", "& lt;").replaceAll(">", "& gt;");
         value = value.replaceAll("\\(", "& #40;").replaceAll("\\)", "& #41;");
         value = value.replaceAll("'", "& #39;");
         value = value.replaceAll("eval\\((.*)\\)", "");
         value = value.replaceAll("[\\\"\\\'][\\s]*javascript:(.*)[\\\"\\\']", "\"\"");
         value = value.replaceAll("script", "");
         return value;    
    }  


}  
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值