當然，我這裏並不想討論其他語言是如何避免sql註入的，網上關於PHP防註入的各種方法都有，Python的方法其實類似，這裏我就舉例來說說。
起因漏洞產生的原因最常見的就是字符串拼接了，當然，sql註入並不只是拼接壹種情況，還有像寬字節註入，特殊字符轉義等等很多種，這裏就說說最常見的字符串拼接，這也是初級程序員最容易犯的錯誤。
首先咱們定義壹個類來處理mysql的操作
class Database:
hostname = '127.0.0.1'
user = 'root'
password = 'root'
db = 'pythontab'
charset = 'utf8'
def init(self):
self.connection = MySQLdb.connect(self.hostname, 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()這個類有問題嗎？
答案是：有！
這個類是有缺陷的，很容易造成sql註入，下面就說說為何會產生sql註入。
為了驗證問題的真實性，這裏就寫壹個方法來調用上面的那個類裏面的方法，如果出現錯誤會直接拋出異常。
def test_query(testUrl):
mysql = Database()
try:
querySql = "SELECT * FROM `article` WHERE url='" + testUrl + "'"
chanels = mysql.query(querySql)
return chanels
except Exception, e:
print e這個方法非常簡單，壹個最常見的select查詢語句，也使用了最簡單的字符串拼接組成sql語句，很明顯傳入的參數 testUrl 可控，要想進行註入測試，只需要在testUrl的值後面加上單引號即可進行sql註入測試，這個不多說，肯定是存在註入漏洞的，腳本跑壹遍，看啥結果
(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")
回顯報錯，很眼熟的錯誤，這裏我傳入的測試參數是
t.tips'
下面再說壹種導致註入的情況，對上面的方法進行稍微修改後
def test_query(testUrl):
mysql = Database()
try:
querySql = ("SELECT * FROM `article` WHERE url='%s'" % testUrl)
chanels = mysql.query(querySql)
return chanels
except Exception, e:
print e這個方法裏面沒有直接使用字符串拼接，而是使用了 %s 來代替要傳入的參數，看起來是不是非常像預編譯的sql？那這種寫法能不能防止sql註入呢？測試壹下便知道，回顯如下
(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，這裏對上面的數據庫類進行簡單的修改即可。
修改後的代碼
class Database:
hostname = '127.0.0.1'
user = 'root'
password = 'root'
db = 'pythontab'
charset = 'utf8'
def init(self):
self.connection = MySQLdb.connect(self.hostname, 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註入，實際使用的方法如下
preUpdateSql = "UPDATE `article` SET title=%s,date=%s,mainbody=%s WHERE id=%s"
mysql.insert(preUpdateSql, [title, date, content, aid])
這樣就可以防止sql註入，傳入壹個列表之後，MySQLdb模塊內部會將列表序列化成壹個元組，然後進行escape操作。