如何在单元测试中针对动态数据层编写SQL注入防御用例

酷强酱_7810

酷强酱_7810

2026-10-03

281人浏览

原创

sql注入测试需绕过orm自动转义,重点检测原始字符串拼接、动态标识符白名单校验、多数据库语法差异及mock时真实sql捕获。

如何在单元测试中针对动态数据层编写sql注入防御用例

SQL注入测试必须绕过ORM的自动转义

很多团队误以为用了 SQLAlchemy 或 Django ORM 就天然防注入,其实不然——只要代码里拼接了原始字符串再交给数据库驱动执行,风险就存在。比如用 text() + execute()、或直接调用 cursor.execute(sql, params) 但把用户输入塞进 sql 字符串里(而非 params),就等于主动拆掉防护。

实操建议:

  • 在测试中构造典型恶意输入:"' OR 1=1 -- "、"; DROP TABLE users; --、"' UNION SELECT password FROM users -- "
  • 不要只测返回结果是否为空,要检查实际执行的 SQL 是否被完整传入(可通过 mock cursor.execute 或启用 DB 日志捕获)
  • 若使用 psycopg2,注意 %s 占位符必须配合参数元组/字典传入;写成 f"WHERE name = '{name}'" 就是硬伤

动态表名/列名无法用参数化,必须白名单校验

参数化查询对表名、列名、排序字段(ORDER BY xxx)、分页偏移(LIMIT ?, ? 中的第一个 ?)完全无效。这类动态部分必须提前约束,否则测试再全也拦不住注入。

实操建议:

  • 提取可接受的表名列名列表,例如:ALLOWED_TABLES = {"users", "orders", "products"},在拼接前强制校验 if table_name not in ALLOWED_TABLES: raise ValueError
  • 测试用例要覆盖边界:传入 "users; DROP TABLE admins"、"users--"、"USERS"(大小写绕过)、"user\x00s"(空字节截断)
  • 避免用 str.lower() 后比对——某些数据库(如 PostgreSQL)对标识符大小写敏感,而校验逻辑可能漏掉引号包裹的双引号标识符("UsErS")

测试需覆盖不同数据库的语法差异

同一段“看似安全”的拼接逻辑,在 MySQL、PostgreSQL、SQLite 上可能表现不同。比如 MySQL 支持 /* */ 注释、PostgreSQL 支持 || 字符串拼接、SQLite 对单引号转义更宽松——测试若只跑 SQLite,很可能漏掉真实环境的风险。

左手医生开放平台
左手医生开放平台

左手医生开放平台是一款面向智慧医疗服务建设的 AI 医疗开放平台。

下载

实操建议:

  • 单元测试中启动轻量级内存 DB 实例(如 sqlite:///:memory:)仅够验证逻辑;真正防注入必须在 CI 中并行跑 PostgreSQL 和 MySQL 容器
  • 重点验证注释绕过:"user' -- comment" 在 PostgreSQL 中会注释掉后续条件,但在某些 MySQL 模式下可能报错或被忽略
  • 留意数据库驱动对特殊字符的预处理——例如 pyodbc 对 ; 的默认行为可能和 psycopg2 不同,测试时得用真实 driver

Mock 数据层时别绕过 SQL 构建逻辑

为提速而 mock session.execute 或 cursor.execute 是常见做法,但如果 mock 直接返回假数据、跳过 SQL 拼接过程,那就等于没测防御逻辑本身。

实操建议:

  • mock 应拦截并记录实际传入的 sql 字符串和 params,再断言 sql 中不包含用户输入原文(例如 assert "' OR 1=1" not in captured_sql)
  • 对使用 text() 的场景,mock text() 构造函数,检查其参数是否含未过滤变量(如 text(f"SELECT * FROM {table}"))
  • 避免 patch 错层级:patch sqlalchemy.engine.Engine.execute 比 patch your_module.query_users 更接近真实执行路径

真正的难点不在写几个 payload,而在于确认 SQL 字符串在到达数据库驱动前的那一刻,是否已被净化或隔离——所有中间环节(ORM 封装、自定义 query builder、DB wrapper 类)都得暴露出来被观测。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

sql注入

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

2023.10.12

3943

8

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.27

851

4

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

2024.02.23

1029

5

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

5781

10

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

2723

4

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

2024.04.07

5780

11

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

2024.04.29

7641

6

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

1050

5

sql中删除一列的命令是什么
sql中删除一列的命令是什么

在sql中,使用alter table语句可以删除一列,语法为:alter table table_name drop column column_name。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

912

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
PHP开发基础之数据库篇(PDO)
PHP开发基础之数据库篇(PDO)

共10课时 | 2.3万人学习