Python中防止sql注入的方法詳解

SQL注入是比較常見的網絡攻擊方式之一，它不是利用操作系統的BUG來實現攻擊，而是針對程序員編程時的疏忽，通過SQL語句，實現無帳號登錄，甚至篡改數據庫。下面這篇文章主要給大家介紹了關於Python中防止sql注入的方法，需要的朋友可以參考下。

 

前言

大家應該都知道現在web漏洞之首莫過於sql了，不管使用哪種語言進行web后端開發，只要使用了關系型數據庫，可能都會遇到sql注入攻擊問題。那么在Python web開發的過程中sql注入是怎么出現的呢，又是怎么去解決這個問題的？

當然，我這里並不想討論其他語言是如何避免sql注入的，網上關於PHP（博主注：據說是世界上最屌的語言）防注入的各種方法都有，Python的方法其實類似，這里我就舉例來說說。

起因

漏洞產生的原因最常見的就是字符串拼接了，當然，sql注入並不只是拼接一種情況，還有像寬字節注入，特殊字符轉義等等很多種，這里就說說最常見的字符串拼接，這也是初級程序員最容易犯的錯誤。

首先咱們定義一個類來處理mysql的操作

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26

class Database:   aurl = '127.0.0.1'   user = 'root'   password = 'root'   db = 'testdb'   charset = 'utf8'     def __init__( self ):    self .connection = MySQLdb.connect( self .aurl, self .user, self .password, self .db, charset = self .charset)    self .cursor = self .connection.cursor()     def insert( self , query):    try :     self .cursor.execute(query)     self .connection.commit()    except Exception, e:     print e     self .connection.rollback()     def query( self , query):    cursor = self .connection.cursor(MySQLdb.cursors.DictCursor)    cursor.execute(query)    return cursor.fetchall()     def __del__( self ):    self .connection.close()

這段代碼在我之前很多腳本里面都會看見，涉及到Python操作mysql數據庫的腳本我都會寫進去這個類，那么這個類有問題嗎？
答案是：有！

這個類是有缺陷的，很容易造成sql注入，下面就說說為何會產生sql注入。

為了驗證問題的真實性，這里就寫一個方法來調用上面的那個類里面的方法，如果出現錯誤會直接拋出異常。

?

1 2 3 4 5 6 7 8

def test_query(articleurl):   mysql = Database()   try :    querySql = "SELECT * FROM `article` WHERE url='" + articleurl + "'"    chanels = mysql.query(querySql)    return chanels   except Exception, e:    print e

這個方法非常簡單，一個最常見的select查詢語句，也使用了最簡單的字符串拼接組成sql語句，很明顯傳入的參數 articleurl 可控，要想進行注入測試，只需要在articleurl的值后面加上單引號即可進行sql注入測試，這個不多說，肯定是存在注入漏洞的，腳本跑一遍，看啥結果

?

1	( 1064 , "You have an error in your SQL syntax; check the manual that corresponds to your MariaDB server version for the right syntax to use near ''t.tips''' at line 1" )

回顯報錯，很眼熟的錯誤，這里我傳入的測試參數是

?

1

t.tips'

下面再說一種導致注入的情況，對上面的方法進行稍微修改后

?

1 2 3 4 5 6 7 8

def test_query(articleurl):   mysql = Database()   try :    querySql = ( "SELECT * FROM `article` WHERE url='%s'" % articleurl)    chanels = mysql.query(querySql)    return chanels   except Exception, e:    print e

這個方法里面沒有直接使用字符串拼接，而是使用了 %s 來代替要傳入的參數，看起來是不是非常像預編譯的sql？那這種寫法能不能防止sql注入呢？測試一下便知道，回顯如下

?

1	( 1064 , "You have an error in your SQL syntax; check the manual that corresponds to your MariaDB server version for the right syntax to use near ''t.tips''' at line 1" )

和上面的測試結果一樣，所以這種方法也是不行的，而且這種方法並不是預編譯sql語句，那么怎么做才能防止sql注入呢？

解決

兩種方案

     1> 對傳入的參數進行編碼轉義

     2> 使用Python的MySQLdb模塊自帶的方法

第一種方案其實在很多PHP的防注入方法里面都有，對特殊字符進行轉義或者過濾。

第二種方案就是使用內部方法，類似於PHP里面的PDO，這里對上面的數據庫類進行簡單的修改即可。

修改后的代碼

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26

class Database:   aurl = '127.0.0.1'   user = 'root'   password = 'root'   db = 'testdb'   charset = 'utf8'     def __init__( self ):    self .connection = MySQLdb.connect( self .aurl, self .user, self .password, self .db, charset = self .charset)    self .cursor = self .connection.cursor()     def insert( self , query, params):    try :     self .cursor.execute(query, params)     self .connection.commit()    except Exception, e:     print e     self .connection.rollback()     def query( self , query, params):    cursor = self .connection.cursor(MySQLdb.cursors.DictCursor)    cursor.execute(query, params)    return cursor.fetchall()     def __del__( self ):    self .connection.close()

這里 execute 執行的時候傳入兩個參數，第一個是參數化的sql語句，第二個是對應的實際的參數值，函數內部會對傳入的參數值進行相應的處理防止sql注入，實際使用的方法如下

?

1 2	preUpdateSql = "UPDATE `article` SET title=%s,date=%s,mainbody=%s WHERE id=%s" mysql.insert(preUpdateSql, [title, date, content, aid])

這樣就可以防止sql注入，傳入一個列表之后，MySQLdb模塊內部會將列表序列化成一個元組，然后進行escape操作。