讲师中心 微信公众号
AI工具推荐 视频效率加速

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

秋涛同学_6722

秋涛同学_6722

发布时间:2026-08-21 08:15:22

|

1014人浏览过

|

来源于php中文网

原创

真正有效的测试必须让恶意输入穿透校验层、拼接逻辑和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 或 + 拼接,都是高危点。单元测试必须显式构造这些分支并验证其行为。

尤其注意旧代码里藏得深的拼接逻辑:比如先从 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'; --",登录时再查这个字段触发拼接。这种延迟执行的漏洞,单元测试里最容易漏掉。

热门AI工具

更多
SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

相关专题

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

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

4323

2023.10.12

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

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

911

2023.10.27

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

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

1129

2024.02.23

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

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

6241

2024.03.06

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

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

3083

2024.03.06

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

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

6220

2024.04.07

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

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

8421

2024.04.29

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

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

1150

2024.04.29

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

20

2026.10.16

热门下载

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

精品课程

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

共10课时 | 2.3万人学习

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

共0课时 | 0人学习

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

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn