如何编写单元测试验证SQL注入防护

秋涛同学_6722

秋涛同学_6722

2026-08-21

994人浏览

原创

真正有效的测试必须让恶意输入穿透校验层、拼接逻辑和dao调用,触发业务层异常(如illegalargumentexception),而非仅验证不报错;需覆盖${}拼接、二次注入等高危路径,并检查审计日志。

如何编写单元测试验证sql注入防护

测试必须传真实恶意字符串,不能 mock DAO 返回值

很多测试失败不是因为防护没写对,而是测试本身绕过了关键路径。比如在 service 层 mock 了 userRepository.findByUsername 的返回值,却没把 "admin'--" 真正送进方法体——这等于测了个假流程。

真正有效的测试要让输入穿透校验层、拼接逻辑(如果有)、DAO 调用,直到触发异常或安全执行。否则你测的不是“防护是否生效”,而是“mock 是否配对成功”。

  • 用 userService.findByUsername("admin'--") 调用,而不是 when(userRepository.findByUsername(any())).thenReturn(...)
  • 如果底层是 MyBatis,确保测试走的是 select * from users where name = #{name} 路径,而非 ${name} 拼接分支
  • 对 JSON 接口,测试 body 原始内容含 %27(URL 编码单引号),确认解码后仍被拦截

断言目标不是“不报错”,而是“抛指定业务异常”

看到测试通过就以为安全?错。很多框架默认允许单引号,仅靠数据库驱动语法校验不可靠——MySQL 可能直接报 MySQLSyntaxErrorException,PostgreSQL 却可能静默截断或返回空结果。

真正要验证的是:防护逻辑是否在 DB 执行前就拦截了。所以 assert 必须明确指向业务层异常,比如 IllegalArgumentException 或自定义的 InvalidInputException。

  • ✅ 正确:assertThatThrownBy(() -> service.login("root' OR '1'='1")) + .isInstanceOf(IllegalArgumentException.class)
  • ❌ 错误:assertThatCode(() -> service.login("...")).doesNotThrowAnyException() —— 这只说明没崩,不代表没进 DB
  • 若实际抛出 PSQLException 或 SQLServerException,说明防护失效,请求已抵达数据库

覆盖 ${} 和字符串拼接等危险写法

MyBatis 的 ${}、JPA 的 createNativeQuery、Java 的 String.format 或 + 拼接,都是高危点。单元测试必须显式构造这些分支并验证其行为。

Cici
Cici

一款AI翻译工具,主要用于Cici 是你的AI聊天助手,可以用于智能对话、写作、翻译、情感支持和编程,是您寻求建议、解答和聊天的首选伙伴,适合需要提升相关任务效率的用户。

下载

尤其注意旧代码里藏得深的拼接逻辑:比如先从 DB 查出模板 SQL,再用用户输入填充字段名或表名——这种场景 #{} 也救不了你。

  • 对使用 ${tableName} 的 mapper,传入 "users; DROP TABLE admins--",验证是否抛出预期异常
  • 对 String sql = "SELECT * FROM " + userInput; 这类代码,测试需触发运行时异常(如 SQLException)或记录非法日志
  • Hibernate 场景下,检查是否用了 @Param 绑定,而非 createNativeQuery("SELECT * FROM users WHERE name = '" + input + "'")

避免用 @Sql 或 H2 执行真实 SQL

用内存数据库跑一条带 ' OR 1=1 的查询,看它返回全表数据——这不是在测你的防护,是在测 H2 的 SQL 解析器是否够严。

真实防护逻辑发生在 Java 层:正则过滤、白名单校验、参数绑定调用。数据库只是最终执行者。把测试焦点放在 DB 层,等于把责任推给方言实现,而不同数据库对非法输入的响应差异极大。

  • 禁用 @Sql(scripts = "malicious-data.sql") 类测试,它无法验证你写的校验逻辑是否触发
  • 不要依赖 H2/hsqldb 的“报错即拦截”行为——它们不模拟 MySQL 的注释解析、PostgreSQL 的类型强转
  • 重点检查日志:当恶意输入进来时,是否记录了 "Blocked SQL injection attempt: ..." 这类审计日志

最常被忽略的一点:二次注入。测试不能只盯着即时接口,还要覆盖“先存后查”路径——比如注册时存入 "admin'; --",登录时再查这个字段触发拼接。这种延迟执行的漏洞,单元测试里最容易漏掉。

相关文章

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

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

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系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

4303

8

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

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

2023.10.27

911

4

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

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

2024.02.23

1109

5

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

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

2024.03.06

6221

10

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

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

2024.03.06

3063

4

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

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

2024.04.07

6200

11

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

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

2024.04.29

8381

6

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

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

2024.04.29

1130

5

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

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

2024.04.29

992

5

热门下载

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

精品课程

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

共10课时 | 2.3万人学习

文心快码功能介绍
文心快码功能介绍

共0课时 | 0人学习

文心快码Agent使用手册
文心快码Agent使用手册

共0课时 | 0人学习