欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

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

 更新時(shí)間:2017年02月25日 15:44:10   作者:一只猿  
SQL注入是比較常見(jiàn)的網(wǎng)絡(luò)攻擊方式之一,它不是利用操作系統(tǒng)的BUG來(lái)實(shí)現(xiàn)攻擊,而是針對(duì)程序員編程時(shí)的疏忽,通過(guò)SQL語(yǔ)句,實(shí)現(xiàn)無(wú)帳號(hào)登錄,甚至篡改數(shù)據(jù)庫(kù)。下面這篇文章主要給大家介紹了關(guān)于Python中防止sql注入的方法,需要的朋友可以參考下。

前言

大家應(yīng)該都知道現(xiàn)在web漏洞之首莫過(guò)于sql了,不管使用哪種語(yǔ)言進(jìn)行web后端開發(fā),只要使用了關(guān)系型數(shù)據(jù)庫(kù),可能都會(huì)遇到sql注入攻擊問(wèn)題。那么在Python web開發(fā)的過(guò)程中sql注入是怎么出現(xiàn)的呢,又是怎么去解決這個(gè)問(wèn)題的?

當(dāng)然,我這里并不想討論其他語(yǔ)言是如何避免sql注入的,網(wǎng)上關(guān)于PHP(博主注:據(jù)說(shuō)是世界上最屌的語(yǔ)言)防注入的各種方法都有,Python的方法其實(shí)類似,這里我就舉例來(lái)說(shuō)說(shuō)。

起因

漏洞產(chǎn)生的原因最常見(jiàn)的就是字符串拼接了,當(dāng)然,sql注入并不只是拼接一種情況,還有像寬字節(jié)注入,特殊字符轉(zhuǎn)義等等很多種,這里就說(shuō)說(shuō)最常見(jiàn)的字符串拼接,這也是初級(jí)程序員最容易犯的錯(cuò)誤。

首先咱們定義一個(gè)類來(lái)處理mysql的操作

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()

這段代碼在我之前很多腳本里面都會(huì)看見(jiàn),涉及到Python操作mysql數(shù)據(jù)庫(kù)的腳本我都會(huì)寫進(jìn)去這個(gè)類,那么這個(gè)類有問(wèn)題嗎?
答案是:有!

這個(gè)類是有缺陷的,很容易造成sql注入,下面就說(shuō)說(shuō)為何會(huì)產(chǎn)生sql注入。

為了驗(yàn)證問(wèn)題的真實(shí)性,這里就寫一個(gè)方法來(lái)調(diào)用上面的那個(gè)類里面的方法,如果出現(xiàn)錯(cuò)誤會(huì)直接拋出異常。

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

這個(gè)方法非常簡(jiǎn)單,一個(gè)最常見(jiàn)的select查詢語(yǔ)句,也使用了最簡(jiǎn)單的字符串拼接組成sql語(yǔ)句,很明顯傳入的參數(shù) articleurl 可控,要想進(jìn)行注入測(cè)試,只需要在articleurl的值后面加上單引號(hào)即可進(jìn)行sql注入測(cè)試,這個(gè)不多說(shuō),肯定是存在注入漏洞的,腳本跑一遍,看啥結(jié)果

(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")

回顯報(bào)錯(cuò),很眼熟的錯(cuò)誤,這里我傳入的測(cè)試參數(shù)是

t.tips'

下面再說(shuō)一種導(dǎo)致注入的情況,對(duì)上面的方法進(jìn)行稍微修改后

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

這個(gè)方法里面沒(méi)有直接使用字符串拼接,而是使用了 %s 來(lái)代替要傳入的參數(shù),看起來(lái)是不是非常像預(yù)編譯的sql?那這種寫法能不能防止sql注入呢?測(cè)試一下便知道,回顯如下

(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")

和上面的測(cè)試結(jié)果一樣,所以這種方法也是不行的,而且這種方法并不是預(yù)編譯sql語(yǔ)句,那么怎么做才能防止sql注入呢?

解決

兩種方案

     1> 對(duì)傳入的參數(shù)進(jìn)行編碼轉(zhuǎn)義

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

第一種方案其實(shí)在很多PHP的防注入方法里面都有,對(duì)特殊字符進(jìn)行轉(zhuǎn)義或者過(guò)濾。

第二種方案就是使用內(nèi)部方法,類似于PHP里面的PDO,這里對(duì)上面的數(shù)據(jù)庫(kù)類進(jìn)行簡(jiǎn)單的修改即可。

修改后的代碼

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 執(zhí)行的時(shí)候傳入兩個(gè)參數(shù),第一個(gè)是參數(shù)化的sql語(yǔ)句,第二個(gè)是對(duì)應(yīng)的實(shí)際的參數(shù)值,函數(shù)內(nèi)部會(huì)對(duì)傳入的參數(shù)值進(jìn)行相應(yīng)的處理防止sql注入,實(shí)際使用的方法如下

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

這樣就可以防止sql注入,傳入一個(gè)列表之后,MySQLdb模塊內(nèi)部會(huì)將列表序列化成一個(gè)元組,然后進(jìn)行escape操作。

總結(jié)

我之前的一些腳本中使用了存在sql注入漏洞的代碼會(huì)慢慢改過(guò)來(lái),好了,以上就是這篇文章的全部?jī)?nèi)容了,希望本文的內(nèi)容對(duì)大家的學(xué)習(xí)或者工作能帶來(lái)一定的幫助,如果有疑問(wèn)大家可以留言交流。

相關(guān)文章

最新評(píng)論