1. 程式人生 > >使用過濾器防止簡單的頁面SQL注入

使用過濾器防止簡單的頁面SQL注入

在之前寫的一個群博系統中,在上線一段時間後,被自己人發現了一個天大的漏洞——SQL注入,自己也是第一次遇到,真是難以置信,後來看到這樣一篇文章:java類過濾器,防止頁面SQL注入。自己修改了一下,可以解決一般的SQL注入了(最起碼使用sqlmap這種工具沒有造成注入)。

首先說一下在我的系統中造成SQL注入的原因:

  1. 大量的SQL語句拼接——為了能夠根據不同的請求引數在同一個頁面查詢後展示,所以在主頁存在著大量的SQL語句拼接,由於請求引數是通過get方法傳送,在瀏覽器位址列很容易被發現,就有可能直接在位址列新增SQL語句造成SQL注入
  2. 專案使用的資料庫連線的使用者沒有設定許可權——由於直接使用的是root使用者,所以在別人在進行SQL注入的時候寫許可權是有的,這就非常的可怕,同時,由於使用的是root使用者,整個資料庫的資訊都能夠被拿到,也包括mysql庫中的user表,這真是。。。

好了,不多說了,先說這個通過過濾器來防止SQL注入是如何做到的!

這裡是通過過濾器取得所有的請求引數,判斷請求引數中是否含有SQL中的關鍵字,而達到防止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; public class AntiSqlInjectionfilter implements Filter { 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; HttpServletRequest res=(HttpServletRequest)args1; //獲得所有請求引數名 Enumeration params = req.getParameterNames(); String sql = ""; while (params.hasMoreElements()) { //得到引數名 String name = params.nextElement().toString(); //得到引數對應值 String[] value = req.getParameterValues(name); for (int i = 0; i < value.length; i++) { sql = sql + value[i]; } } //有sql關鍵字,跳轉到error.html if (sqlValidate(sql)) { throw new IOException("您傳送請求中的引數中含有非法字元"); //String ip = req.getRemoteAddr(); } 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|" + "information_schema.columns|table_schema|union|where|select|delete|update|order|by|count|*|" + "chr|mid|master|truncate|char|declare|or|;|-|--|+|,|like|//|/|%|#";//過濾掉的sql關鍵字,可以手動新增 String[] badStrs = badStr.split("\\|"); for (int i = 0; i < badStrs.length; i++) { //迴圈檢測,判斷在請求引數當中是否包含SQL關鍵字 if (str.indexOf(badStrs[i]) >= 0) { return true; } } return false; } }

如上,在過濾器當中首先獲取到所有的請求引數的名,然後遍歷這些請求引數得到請求引數的值,拼接成一個字串,然後在sqlValidate方法中檢驗是否含有SQL欄位,當檢驗包含 SQL欄位就可以做自己的處理了,在這個例子中是丟擲了一個IO異常,我們也可通過在web.xml中配置異常的頁面達到類似與捕獲異常的效果,如下:

  <error-page>
    <exception-type>java.io.IOException</exception-type>
    <location>/500.html</location>
  </error-page>

這裡使用sql字串涵蓋了較多的關鍵字,可以根據自己的需要刪除(說不定我的合法請求裡面本身就含有某些關鍵字呢!)

還有一種解決SQL注入的方法,就是通過PreparedStatement來實現,由於PreparedStatement在執行前設定的引數都會被當成欄位的值,所以即便欄位值包含SQL語句,也只會被當做一個欄位的只,而不會產生SQL語句拼接的情況!