脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Python防止sql注入

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

 更新时间：2017年02月25日 15:44:10   作者：一只猿  
SQL注入是比较常见的网络攻击方式之一，它不是利用操作系统的BUG来实现攻击，而是针对程序员编程时的疏忽，通过SQL语句，实现无帐号登录，甚至篡改数据库。下面这篇文章主要给大家介绍了关于Python中防止sql注入的方法，需要的朋友可以参考下。

前言

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

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

起因

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

首先咱们定义一个类来处理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()

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

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

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

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注入测试，这个不多说，肯定是存在注入漏洞的，脚本跑一遍，看啥结果

(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(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注入呢？测试一下便知道，回显如下

(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:
 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注入，实际使用的方法如下

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

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

总结

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

您可能感兴趣的文章:

相关文章

  • Python实现注册登录功能

    Python实现注册登录功能

    这篇文章主要为大家详细介绍了Python实现注册登录功能，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2022-02-02
  • 深入了解python中元类的相关知识

    深入了解python中元类的相关知识

    这篇文章主要介绍了深入了解python中的元类,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08
  • Python  inspect 模块的简单使用及常用方法

    Python  inspect 模块的简单使用及常用方法

    Python的inspect模块是一个强大的内省工具，允许开发者检查（inspect）活动对象和源代码，这篇文章主要介绍了Python inspect 模块的简单使用及常用方法,需要的朋友可以参考下
    2024-03-03
  • 详解django中url路由配置及渲染方式

    详解django中url路由配置及渲染方式

    这篇文章主要介绍了详解django中url路由配置及渲染方式，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2019-02-02
  • Python DataFrame使用drop_duplicates()函数去重(保留重复值，取重复值)

    Python DataFrame使用drop_duplicates()函数去重(保留重复值，取重复值)

    这篇文章主要介绍了Python DataFrame使用drop_duplicates()函数去重(保留重复值，取重复值)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-07-07
  • Python调用系统命令os.system()和os.popen()的实现

    Python调用系统命令os.system()和os.popen()的实现

    这篇文章主要介绍了Python调用系统命令os.system()和os.popen()的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12
  • 使用NumPy进行数组数据处理的示例详解

    使用NumPy进行数组数据处理的示例详解

    NumPy是Python中用于数值计算的核心包之一，它提供了大量的高效数组操作函数和数学函数，可以支持多维数组和矩阵运算。本文主要为大家介绍了NumPy进行数组数据处理的具体方法，需要的可以参考一下
    2023-03-03
  • win10系统下Anaconda3安装配置方法图文教程

    win10系统下Anaconda3安装配置方法图文教程

    这篇文章主要为大家详细介绍了win10系统下Anaconda3安装配置方法图文教程，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2018-09-09
  • Python-VTK批量读取二维切片并显示三维模型

    Python-VTK批量读取二维切片并显示三维模型

    这篇文章主要介绍了Python-VTK批量读取二维切片并显示三维模型，文章基于python的相关资料展开对主题的详细介绍，具有一定的参考价值，需要的小伙伴可以参考一下
    2022-04-04
  • Python格式化字符串f-string的使用教程

    Python格式化字符串f-string的使用教程

    这篇文章主要为大家详细介绍了Python中格式化字符串f-string的使用教程，文中通过示例为大家进行了详细的介绍，需要的可以参考一下
    2022-07-07

最新评论