AI代码审计实战

原文首发在：奇安信攻防社区

https://forum.butian.net/ai_security/98

AI代码审计不仅能节省时间，还能发现一些人工容易忽略的漏洞。不过需要注意的是，AI有时候会产生幻觉，所以必须实际验证才能确认漏洞是否真实存在。

不花一分钱，用AI挖洞！三款免费工具横向对比，看看谁更强？

前言

AI代码审计不仅能节省时间，还能发现一些人工容易忽略的漏洞。不过需要注意的是，AI有时候会产生幻觉，所以必须实际验证才能确认漏洞是否真实存在。

费用说明：三款工具全部免费！

一、为什么用AI做代码审计？

传统代码审计就像大海捞针——逐行阅读源码，眼睛看花了还容易漏掉关键点。

AI可以：

• 快速理解代码逻辑，比人工快N倍
• 自动追踪数据流，不放过任何可疑点
• 识别危险函数调用，精准定位漏洞
• 深度规划分析，给出专业建议

需要注意的是：AI会产生幻觉，所以必须实际验证！

二、源代码环境采用

java语言

https://gitee.com/project_team/Tmall_demo

微信公众号代码审计解析

这个源代码环境的漏洞解析，在B站上也有视频教学。

代码审计之路之白盒挖掘机

三、AI代码审计执行框架

统一提示词

四、工具选型对比（全部免费！）

重点：以下三款工具全部免费使用，无需付费！

4.1 Yak MCP（免费）

一句话总结：免费+验证能力强，能看到真实的HTTP请求，避免AI幻觉！

Yak操作界面：

Yak审计过程：

Yak深度规划：

Yak详细步骤：

Yak漏洞审计结果：

4.2 Trae IDE + 智能体（免费）

一句话总结：免费+深度审计，报告详细，但速度慢！

Trae智能体架构：

Trae审计过程：

Trae审计进度：

4.3 MonkeyCode AI（免费）

一句话总结：免费+快速扫描，适合开发，但不够深入！

MonkeyCode操作界面：

MonkeyCode执行过程：

MonkeyCode漏洞验证：

MonkeyCode自动修复：

五、Yak技能库配置

可以尝试制作几个yak的技能库，来提升效率。

Yak技能配置界面：

六、Trae智能体配置

6.1 智能体solo 模式分工

┌─────────────┐    ┌─────────────┐    ┌─────────────┐│  s1-s2      │ -> │  s3-s9       │ -> │  s10-s11     ││  信息收集    │    │  漏洞审计    │    │  验证复现    │└─────────────┘    └─────────────┘    └─────────────┘

核心原则：先审计 → 再写脚本验证 → 最后写报告

6.2 Trae提示词

在当前项目进行代码审计，按顺序调用智能体s1到s11完成代码审计工作。## 执行顺序严格按照 s1 → s2 → s3 → s4 → s5 → s6 → s7 → s8 → s9 → s10 → s11 顺序执行。## 智能体职责- s1-s2: 信息收集阶段- s3-s9: 漏洞审计阶段  - s10-s11: 验证复现阶段## 报告要求1. 所有审计结果写入 `代码审计.md` 文件2. 每个阶段末尾必须标注：`*本阶段由 sX 智能体完成*`3. 每个漏洞必须提供完整的POC（Raw格式HTTP数据包、curl命令、实际测试结果）## 工具使用根据审计需要，灵活选择以下工具：**Skills工具**:- java-sql-audit、java-route-mapper、java-route-tracer、java-auth-audit、java-vuln-scanner- vuln-analysis-expert、logic-vulnerability-hunter- Penetration-Testing-Expert、wsl-pentest-suite**命令行工具**:- SQLMap、Nuclei、curl、httpx、fscan**验证工具**:- WebFetch、MCP浏览器工具## 注意事项- 根据项目实际情况灵活调整审计策略- 发现漏洞后必须实际验证，确保漏洞真实存在- 区分演示项目和真实漏洞，准确评估风险等级

缺点：提示词字数上限6000，有时会限制表达。

七、项目架构分析

7.1 技术架构图

┌─────────────────────────────────────────────────────────────────────────────┐│                              客户端层 (Client)                              ││                    浏览器 → JSP页面 + AJAX请求                               │└─────────────────────────────────────────────────────────────────────────────┘                                      ↓┌─────────────────────────────────────────────────────────────────────────────┐│                           Web层 (Spring MVC)                              ││  ┌─────────────────────────────────────────────────────────────────────┐    ││  │                     Filter过滤器层                                    │    ││  │           AdminPermissionFilter (后台权限过滤 /admin/*)              │    ││  └─────────────────────────────────────────────────────────────────────┘    ││                                      ↓                                       ││  ┌─────────────────────────────────────────────────────────────────────┐    ││  │                     Controller控制器层                                │    ││  │  ┌──────────────────┐    ┌──────────────────┐                       │    ││  │  │   后台管理模块    │    │    前台用户模块    │                       │    ││  │  │  admin包         │    │     fore包        │                       │    ││  │  │  - AdminLogin    │    │  - ForeLogin      │                       │    ││  │  │  - AdminHome     │    │  - ForeRegister   │                       │    ││  │  │  - Account       │    │  - ForeHome       │                       │    ││  │  │  - Category      │    │  - ForeProduct    │                       │    ││  │  │  - Product       │    │  - ForeOrder      │                       │    ││  │  │  - Order         │    │  - ForeUser       │                       │    ││  │  │  - User          │    │  - ForeReview     │                       │    ││  │  │  - Reward        │    │  - ForeReward     │                       │    ││  │  └──────────────────┘    └──────────────────┘                       │    ││  └─────────────────────────────────────────────────────────────────────┘    │└─────────────────────────────────────────────────────────────────────────────┘                                      ↓┌─────────────────────────────────────────────────────────────────────────────┐│                          业务逻辑层 (Service)                                ││  Service接口 + ServiceImpl实现类                                             ││  UserService, AdminService, ProductService, OrderService...                 │└─────────────────────────────────────────────────────────────────────────────┘                                      ↓┌─────────────────────────────────────────────────────────────────────────────┐│                        数据访问层 (DAO/Mapper)                                     ││  Mapper接口 + XML映射文件                                                    ││  UserMapper, AdminMapper, ProductMapper, OrderMapper...                    │└─────────────────────────────────────────────────────────────────────────────┘                                      ↓┌─────────────────────────────────────────────────────────────────────────────┐│                           数据库层 (Database)                                  ││                    MySQL数据库: tmalldemodb                                  │└─────────────────────────────────────────────────────────────────────────────┘

7.2 技术栈识别

7.3 项目目录结构

八、漏洞审计结果

8.1 漏洞统计对比

8.2 Yak MCP验证结果详情

8.3 最终确认漏洞统计

九、详细漏洞报告

9.1 SQL注入漏洞 (5处)

SQL注入详细分析（SQL-003 – 无需认证）：

漏洞位置: ProductMapper.xml:134

触发端点: GET /product/{index}/{count}

Controller 代码:

MyBatis XML:

POC – HTTP攻击数据包:

POC – curl命令:

SQLMap 自动化利用:

实际测试结果:

9.2 XSS漏洞 (11处)

存储型XSS (7处)

XSS-001 详细分析（用户评论功能）：

漏洞位置：

• 输入点：ForeReviewController.java:89
• 输出点：J_review.jsp:54

漏洞代码：

${review.review_content}

漏洞分析：用户提交的评论内容直接存储到数据库，在显示时使用 ${} EL 表达式直接输出，未经过任何转义处理。

POC – HTTP攻击数据包：

POC – curl命令：

反射型XSS (3处)

DOM型XSS (1处)

9.3 IDOR/越权漏洞 (5处)

IDOR-001 详细分析（管理员账户越权修改）：

漏洞位置：AccountController.java:92-141

漏洞代码：

漏洞分析：路径参数 admin_id 可由攻击者任意指定，未验证是否属于当前登录管理员。

POC – HTTP攻击数据包：

攻击场景：

1. 管理员A（ID=2）登录后台
2. 发送请求修改 （超级管理员）
3. 成功修改超级管理员密码

9.4 文件上传RCE漏洞 (4处)

文件上传详细分析：

漏洞等级: 高危 (High)

漏洞描述:项目的文件上传功能存在严重安全缺陷，未对上传文件类型进行有效验证，攻击者可以上传 JSP WebShell 获取服务器控制权。

受影响位置:

漏洞代码分析:

安全缺陷:

1. 仅验证文件扩展名，未验证 MIME 类型
2. 未验证文件内容（文件头）
3. 上传路径在 webapp 目录内，可直接访问执行
4. 文件名未重命名，保留原始文件名

POC – HTTP攻击数据包:

POC – curl命令:

WebShell内容示例:

9.5 FastJSON反序列化RCE (5处)

FastJSON反序列化详细分析：

漏洞等级: 严重 (Critical)

漏洞描述:项目使用 FastJSON 1.2.83 版本，存在已知的反序列化远程代码执行漏洞。FastJSON 在处理 JSON 反序列化时，攻击者可以通过 @type 字段指定恶意类，利用 JNDI 注入实现远程代码执行。

受影响位置:

漏洞代码示例:

POC – HTTP攻击数据包:

POC – curl命令:

利用步骤:

1. 启动 JNDI 注入服务器 (JNDIExploit 或 JNDI-Injection-Exploit)
2. 构造恶意 JSON Payload
3. 发送请求到受影响端点
4. 触发 JNDI 查询，加载远程恶意类
5. 执行任意命令

修复建议:

9.6 认证安全漏洞 (6处)

AUTH-001 详细分析（明文密码存储）：

漏洞位置：

• UserMapper.xml:82-86
• AdminMapper.xml:49-53

漏洞代码：

数据库存储：

风险：

• 数据库泄露导致所有用户密码直接暴露
• 内部人员可直接查看用户密码
• 不符合安全合规要求

POC验证：

AUTH-002 详细分析（会话固定攻击）：

漏洞位置：

• ForeLoginController.java:36-48
• AdminLoginController.java:34-48

问题代码：

攻击场景：

1. 攻击者访问网站，获取Session ID: JSESSIONID=ATTACKER_SESSION2. 攻击者诱导受害者使用此Session ID访问登录页面3. 受害者登录成功，但Session ID未改变4. 攻击者使用相同的Session ID即可访问受害者账户

POC – curl命令：

AUTH-003 详细分析（暴力破解无防护）：

漏洞位置：

• ForeLoginController.java:35-49
• AdminLoginController.java:33-48

问题分析：

• 无登录失败次数限制
• 无账户锁定机制
• 无验证码保护
• 无响应延迟

POC – Python暴力破解脚本：

9.7 配置安全漏洞 (4处)

CONF-001 详细分析（数据库凭证硬编码）：

漏洞位置：application.properties:6-8

漏洞代码：

风险分析：

• 数据库密码明文存储在配置文件中
• 使用弱密码 
• 配置文件可能被提交到版本控制系统

修复建议：

# 使用环境变量或配置中心spring.datasource.username=${DB_USERNAME}spring.datasource.password=${DB_PASSWORD}

9.8 业务逻辑漏洞 (4处)

BIZ-001 详细分析（订单状态机绕过）：

漏洞位置：ForeOrderController.java:706-739

问题分析：

• 用户可自行将订单标记为”已发货”
• 绕过商家发货环节
• 可能导致退款欺诈

POC – HTTP攻击数据包：

GET /order/delivery/2023031712345601 HTTP/1.1Host: target.comCookie: JSESSIONID=<用户Session>

BIZ-002 详细分析（缺少库存管理）：

问题分析：

• Product实体类缺少库存字段
• 下单时不检查库存
• 可导致超卖问题

POC – 并发下单脚本：

BIZ-003 详细分析（竞争条件漏洞）：

漏洞位置：ForeOrderController.java:586-614

问题分析：

• 支付操作未使用数据库锁
• 并发支付导致销量重复累加

POC – 并发支付脚本：

十、工具特点对比

十一、对比结论

1. Yak MCP 验证能力最强，实际确认4个漏洞，剔除2个误报/无法验证
2. Trae IDE 发现漏洞最多（45个），审计最详细，但需要实际验证
3. MonkeyCode AI 速度最快（2-3分钟），但漏洞发现较少（6个），且未验证

十二、AI验证漏洞

Trae验证漏洞

验证漏洞的AI提示词：

验证漏洞完成的报告：

Yak验证漏洞

但是还是太慢了这个速度，得需要我调教调教才行。

漏洞验证，可以实时看见http流量请求内容，比其他的方便很多，真实可靠。

十三、心得总结

13.1 AI不是万能的

审计结果需要实际验证，避免幻觉！

13.2 工具组合使用

Trae深度审计 + Yak验证 + MonkeyCode快速扫描 = 完美组合

13.3 速度问题

目前AI审计速度较慢，需要优化：

• 制作Yak技能库提升效率
• 优化智能体提示词
• 调整模型配置

13.4 报告规范

每个漏洞必须提供完整POC：

• Raw格式HTTP数据包
• curl命令
• 实际测试结果

十四、改进方向

1. 制作Yak技能库 – 提升效率
2. 优化智能体提示词 – 更精准的审计
3. 调整模型配置 – 加快速度
4. 完善验证脚本 – 减少误报

十五、总结

AI辅助代码审计是未来趋势，但记住：

AI是工具，不是替代品。验证是必须的，幻觉是危险的。

最佳实践：

1. 使用Trae进行深度审计
2. 使用Yak进行实际验证
3. 使用MonkeyCode AI进行快速扫描
4. 形成完整的审计闭环
5. 持续优化工具链
6. 规范化报告输出

相关资源：

• B站视频教学：搜索”代码审计之路之白盒挖掘机”
• 项目地址：https://gitee.com/project_team/Tmall_demo
• Yak工具：https://github.com/yaklang/yak
• 微信公众号：代码审计解析
• *

AI代码审计工具链接

Yak MCP（免费）

• GitHub：https://github.com/yaklang/yak
• 文档：https://yaklang.io/
• 社区：https://github.com/yaklang/yak/discussions

Trae IDE（免费）

• 官网：https://trae.ai/
• 文档：https://docs.trae.ai/
• 社区：https://github.com/trae-ai

MonkeyCode AI（免费）

• 官网：https://monkeycode.ai/
• 文档：https://docs.monkeycode.ai/

其他AI代码审计工具推荐

• CodeQL（免费）：https://codeql.github.com/
• Semgrep（免费）：https://semgrep.dev/
• SonarQube（社区版免费）：https://www.sonarqube.org/

作者：Sha0_s1发布时间：2026年3月本文基于实际代码审计实践总结

文章来源：

黑白之道发布、转载的文章中所涉及的技术、思路和工具仅供以安全为目的的学习交流使用，任何人不得将其用于非法用途及盈利等目的，否则后果自行承担！

END