sql非法字符集过滤

 

import java.io.IOException;
import java.util.Enumeration;

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.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class AntiSqlInjectionfilter implements Filter {
    private static Logger logger = LoggerFactory
            .getLogger(AntiSqlInjectionfilter.class);

    public void destroy() {
        // TODO Auto-generated method stub
    }

    public void init(FilterConfig arg0) throws ServletException {
        // TODO Auto-generated method stub
    }

    public void doFilter(ServletRequest args0, ServletResponse args1,
            FilterChain chain) throws IOException, ServletException {
        HttpServletRequest req = (HttpServletRequest) args0;
        String method = req.getMethod();
        String getServletPath = req.getServletPath();
        // HttpServletReponse res=(HttpServletReponse)args1;
        if (getServletPath != null && getServletPath.endsWith(".do")) {
            String sql = "";
            // 获得所有请求参数名
            Enumeration params = req.getParameterNames();
            while (params.hasMoreElements()) {
                // 得到参数名
                String name = String.valueOf(params.nextElement());
                // logger.info("UUID:" + uuids.get() + "; 方法名：" + method +
                // "; 参数名:" + name);
                if ("password".equals(name) || "ticket".equals(name)) {
                    continue;
                }
                // 得到参数对应值
                String[] value = req.getParameterValues(name);
                for (int i = 0; i < value.length; i++) {
                    sql = sql + value[i];
                }
            }
            try {
                if (sqlValidate(sql)) {
                    throw new IOException("您发送请求中的参数中含有非法字符");
                } else {
                    chain.doFilter(args0, args1);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else {
            chain.doFilter(args0, args1);
        }

    }

    // 效验
    protected static boolean sqlValidate(String str) {
        str = str.toLowerCase();// 统一转为小写
        String badStr = "\'|\"|and|exec|execute|insert|select|delete|update|count|drop|*|%|chr|mid|master|truncate|"
                + "char|declare|sitename|net user|xp_cmdshell|;|or|,|like'|and|exec|execute|insert|create|drop|"
                + "table|from|grant|use|group_concat|column_name|xp_shell|(|)|"
                + "information_schema.columns|table_schema|union|where|select|delete|update|order|count|*|"
                + "chr|mid|master|truncate|char|declare|--|+|,|like|//|/|%|#";// 过滤掉的sql关键字，可以手动添加
        String[] badStrs = badStr.split("\\|");
        for (int i = 0; i < badStrs.length; i++) {
            // 循环检测，判断在请求参数当中是否包含SQL关键字
            if (str.indexOf(badStrs[i]) >= 0 || str.indexOf("\\|\\|") >= 0) {
                return true;
            }
        }
        return false;
    }
}