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

前言

创新互联是专业的徐州网站建设公司，徐州接单;提供成都网站设计、成都网站建设、外贸网站建设,网页设计,网站设计,建网站,PHP网站建设等专业做网站服务;采用PHP框架,可快速的进行徐州网站开发网页制作和功能扩展;专业做搜索引擎喜爱的网站,专业的做网站团队,希望更多企业前来合作!

大家应该都知道现在web漏洞之首莫过于sql了，不管使用哪种语言进行web后端开发，只要使用了关系型数据库，可能都会遇到sql注入攻击问题。那么在Python web开发的过程中sql注入是怎么出现的呢，又是怎么去解决这个问题的?

当然，我这里并不想讨论其他语言是如何避免sql注入的，网上关于PHP(博主注：据说是世界上最屌的语言)防注入的各种方法都有，Python的方法其实类似，这里我就举例来说说。

起因

漏洞产生的原因最常见的就是字符串拼接了，当然，sql注入并不只是拼接一种情况，还有像宽字节注入，特殊字符转义等等很多种，这里就说说最常见的字符串拼接，这也是初级程序员最容易犯的错误。

首先咱们定义一个类来处理mysql的操作

 
 
 
      
  
  
  1. class Database: 
    2.   
  
  
  2.  aurl = '127.0.0.1' 
    3.   
  
  
  3.  user = 'root' 
    4.   
  
  
  4.  password = 'root' 
    5.   
  
  
  5.  db = 'testdb' 
    6.   
  
  
  6.  charset = 'utf8' 
    7.   
  
  
  7.   
    8.   
  
  
  8.  def __init__(self): 
    9.   
  
  
  9.   self.connection = MySQLdb.connect(self.aurl, self.user, self.password, self.db, charset=self.charset) 
    10.   
  
  
  10.   selfself.cursor = self.connection.cursor() 
    11.   
  
  
  11.   
    12.   
  
  
  12.  def insert(self, query): 
    13.   
  
  
  13.   try: 
    14.   
  
  
  14.    self.cursor.execute(query) 
    15.   
  
  
  15.    self.connection.commit() 
    16.   
  
  
  16.   except Exception, e: 
    17.   
  
  
  17.    print e 
    18.   
  
  
  18.    self.connection.rollback() 
    19.   
  
  
  19.   
    20.   
  
  
  20.  def query(self, query): 
    21.   
  
  
  21.   cursor = self.connection.cursor(MySQLdb.cursors.DictCursor) 
    22.   
  
  
  22.   cursor.execute(query) 
    23.   
  
  
  23.   return cursor.fetchall() 
    24.   
  
  
  24.   
    25.   
  
  
  25.  def __del__(self): 
    26.   
  
  
  26.   self.connection.close() 
    27.

这段代码在我之前很多脚本里面都会看见，涉及到Python操作mysql数据库的脚本我都会写进去这个类，那么这个类有问题吗?

答案是：有!

这个类是有缺陷的，很容易造成sql注入，下面就说说为何会产生sql注入。

为了验证问题的真实性，这里就写一个方法来调用上面的那个类里面的方法，如果出现错误会直接抛出异常。

 
 
 
      
  
  
  1. def test_query(articleurl): 
    2.   
  
  
  2.  mysql = Database() 
    3.   
  
  
  3.  try: 
    4.   
  
  
  4.   querySql = "SELECT * FROM `article` WHERE url='" + articleurl + "'" 
    5.   
  
  
  5.   chanels = mysql.query(querySql) 
    6.   
  
  
  6.   return chanels 
    7.   
  
  
  7.  except Exception, e: 
    8.   
  
  
  8.   print e 
    9.

这个方法非常简单，一个最常见的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") 
    2.

回显报错，很眼熟的错误，这里我传入的测试参数是

 
 
 
      
  
  
  1. t.tips' 
    2.

下面再说一种导致注入的情况，对上面的方法进行稍微修改后

 
 
 
      
  
  
  1. def test_query(articleurl): 
    2.   
  
  
  2.  mysql = Database() 
    3.   
  
  
  3.  try: 
    4.   
  
  
  4.   querySql = ("SELECT * FROM `article` WHERE url='%s'" % articleurl) 
    5.   
  
  
  5.   chanels = mysql.query(querySql) 
    6.   
  
  
  6.   return chanels 
    7.   
  
  
  7.  except Exception, e: 
    8.   
  
  
  8.   print e 
    9.

这个方法里面没有直接使用字符串拼接，而是使用了 %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") 
    2.

和上面的测试结果一样，所以这种方法也是不行的，而且这种方法并不是预编译sql语句，那么怎么做才能防止sql注入呢?

解决

两种方案

1> 对传入的参数进行编码转义

2> 使用Python的MySQLdb模块自带的方法

第一种方案其实在很多PHP的防注入方法里面都有，对特殊字符进行转义或者过滤。

第二种方案就是使用内部方法，类似于PHP里面的PDO，这里对上面的数据库类进行简单的修改即可。

修改后的代码

 
 
 
      
  
  
  1. class Database: 
    2.   
  
  
  2.  aurl = '127.0.0.1' 
    3.   
  
  
  3.  user = 'root' 
    4.   
  
  
  4.  password = 'root' 
    5.   
  
  
  5.  db = 'testdb' 
    6.   
  
  
  6.  charset = 'utf8' 
    7.   
  
  
  7.   
    8.   
  
  
  8.  def __init__(self): 
    9.   
  
  
  9.   self.connection = MySQLdb.connect(self.aurl, self.user, self.password, self.db, charset=self.charset) 
    10.   
  
  
  10.   selfself.cursor = self.connection.cursor() 
    11.   
  
  
  11.   
    12.   
  
  
  12.  def insert(self, query, params): 
    13.   
  
  
  13.   try: 
    14.   
  
  
  14.    self.cursor.execute(query, params) 
    15.   
  
  
  15.    self.connection.commit() 
    16.   
  
  
  16.   except Exception, e: 
    17.   
  
  
  17.    print e 
    18.   
  
  
  18.    self.connection.rollback() 
    19.   
  
  
  19.   
    20.   
  
  
  20.  def query(self, query, params): 
    21.   
  
  
  21.   cursor = self.connection.cursor(MySQLdb.cursors.DictCursor) 
    22.   
  
  
  22.   cursor.execute(query, params) 
    23.   
  
  
  23.   return cursor.fetchall() 
    24.   
  
  
  24.   
    25.   
  
  
  25.  def __del__(self): 
    26.   
  
  
  26.   self.connection.close() 
    27.

这里 execute 执行的时候传入两个参数，第一个是参数化的sql语句，第二个是对应的实际的参数值，函数内部会对传入的参数值进行相应的处理防止sql注入，实际使用的方法如下

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

这样就可以防止sql注入，传入一个列表之后，MySQLdb模块内部会将列表序列化成一个元组，然后进行escape操作。

总结

我之前的一些脚本中使用了存在sql注入漏洞的代码会慢慢改过来，好了，以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作能带来一定的帮助。

新闻名称：Python中防止sql注入的方法详解
标题URL：http://www.shufengxianlan.com/qtweb/news25/83925.html

网站建设、网络推广公司-创新互联，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

广告

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源： 创新互联

猜你还喜欢下面的内容

App设计知识

行业网站建设

成都婚庆策划　　 　绵阳电信机房　　 　迷你发光字　　 　四川电信机房托管　　 　网站解决方案　　 　成都企业网站建设　　 　APP定制　　 　广安网站建设　　 　成都网站制作　　 　时嘉做网站　　 　网站推广营销　　 　校园网站建设方案　　 　成都小程序制作　　 　企业网站设计　　 　南充服务器托管　　 　网站推广优化　　 　主机租用　　 　盐亭做网站　　 　江油网站建设　　 　温江电信机房