| category | audit |
| name | code-audit |
| description | 对 Python 代码进行安全漏洞审计。当用户提到"代码审计"、"安全审查"、"安全检查"、"漏洞扫描"、 "安全评估"、"代码 review 安全"、"找漏洞"、"有没有安全问题"、"帮我看看这段代码有没有安全问题"、 "review this code for security"、"audit"、"security audit"、"vulnerability scan" 时， 必须使用此 skill。适用于 Python 项目、代码片段、模块或整个仓库的安全审计。 |
Python 代码安全审计
对 Python 代码进行系统性的安全漏洞审计，输出结构化的安全报告。
审计流程
按以下步骤进行审计，每一步完成后才进入下一步：
第一步：范围确定
- 确认要审计的代码范围（文件、目录或代码片段）
- 如果是项目，先了解项目结构：读取入口文件、配置文件、依赖文件（requirements.txt, setup.py, pyproject.toml）
- 识别项目类型：Web 应用（Flask/Django/FastAPI）、CLI 工具、数据处理脚本、库等
第二步：逐文件扫描
对每个 Python 文件，逐行检查以下漏洞类别。每发现一个问题，立即记录位置和上下文。
第三步：生成报告
汇总所有发现，按严重程度排序，生成结构化报告。
漏洞检查清单
🔴 Critical（严重）
1. 命令注入 (Command Injection)
os.system(), os.popen() 接收用户可控参数subprocess.call(), subprocess.Popen() 使用 shell=True 且参数来自外部eval(), exec(), compile() 执行用户可控代码- 反引号或
$() 中包含外部输入（较少见但仍需注意）
- 修复：使用
subprocess.run() 传参列表，避免 shell=True；禁止 eval/exec 接外部输入
2. 反序列化漏洞 (Deserialization)
pickle.load() / pickle.loads() 加载不可信数据yaml.load() 使用默认 Loader（应使用 yaml.safe_load()）marshal.loads() 处理外部数据torch.load() 加载不可信模型（PyTorch 反序列化风险）- 修复：用 JSON 替代 pickle；yaml 用
safe_load()；校验数据来源
3. 代码注入 (Code Injection)
__import__() 参数来自用户输入importlib.import_module() 动态导入不可信模块名getattr() / setattr() 使用外部可控的属性名访问敏感对象- 修复：白名单限制可导入模块/属性
4. 硬编码密钥 (Hardcoded Secrets)
- 密码、API Key、Token、私钥直接写在代码中
- 数据库连接字符串包含明文密码
- JWT Secret、加密密钥硬编码
- 修复：使用环境变量或密钥管理服务
🟠 High（高危）
5. SQL 注入 (SQL Injection)
- 使用
% 格式化、.format()、f-string 拼接 SQL 查询
cursor.execute() 直接传入拼接字符串（参数化查询是安全的方式）- ORM 中的
raw() / execute() 使用拼接 SQL
- 修复：始终使用参数化查询（
cursor.execute(query, params)）；ORM 优先
6. 路径遍历 (Path Traversal)
open() 路径由用户输入拼接且未做路径规范化os.path.join() 中包含 .. 未被过滤zipfile.extractall() 解压不可信压缩包（Zip Slip）tarfile.extractall() 同样存在路径遍历风险- 修复：用
os.path.realpath() 验证最终路径在允许目录内
7. SSRF（服务端请求伪造）
requests.get() / urllib.request.urlopen() URL 由用户控制- 未对请求目标做 IP/域名白名单限制
- 可被用于访问内网服务（169.254.x.x, 10.x.x.x, 127.x.x.x）
- 修复：DNS 解析后检查 IP；白名单限制允许的目标
8. SSTI（服务端模板注入）
render_template_string() 传入用户输入- Jinja2 环境中用户可控内容拼入模板字符串
- 修复：使用
render_template() 替代；沙箱化模板环境
🟡 Medium（中危）
9. XSS（跨站脚本）
- 用户输入未转义直接写入响应 HTML
Markup() 包装用户输入（绕过自动转义）- 修复：使用模板引擎自动转义；手动转义用
html.escape()
10. 弱加密算法
- 使用 MD5、SHA1 进行密码哈希
- 对称加密使用 ECB 模式
- 使用
random 模块生成安全 token（应用 secrets）
- 硬编码的盐值或 IV
- 修复：密码用 bcrypt/scrypt/argon2；随机数用
secrets 模块
11. XML 外部实体注入 (XXE)
- 使用
xml.etree.ElementTree 或 lxml 解析不可信 XML
- 未禁用外部实体解析
- 修复：使用
defusedxml 库
12. 不安全的文件权限
os.chmod() 设置过于宽松的权限（如 0o777）os.umask(0) 后创建文件- 临时文件创建在共享目录
- 修复：使用最严格权限；临时文件用
tempfile 模块
🔵 Low（低危）
13. 信息泄露
- Debug 模式在生产环境开启（
DEBUG=True）
- 异常堆栈信息直接返回给用户
- 敏感信息通过
print() / logging 输出
- 修复：生产环境关闭 debug；定制错误页面；日志脱敏
14. 不安全的重定向
redirect() 目标由用户输入控制且未验证- 开放重定向可被用于钓鱼攻击
- 修复：白名单验证重定向目标域名
15. 依赖安全
requirements.txt 中依赖版本过旧或有已知 CVE- 缺少依赖版本锁定
- 修复：定期更新依赖；使用
pip-audit 或 safety 检查
16. 竞态条件 (TOCTOU)
- 文件操作：先检查再使用（check-then-use）
- 共享资源无锁保护
- 修复：使用原子操作；文件操作用
os.open() 的文件描述符
报告模板
审计完成时，必须按以下格式输出报告：
# 安全审计报告
## 审计摘要
- **审计范围**：<文件/目录列表>
- **审计时间**：<当前时间>
- **文件数量**：<N>
- **发现问题**：<总数>（Critical: N, High: N, Medium: N, Low: N）
## 发现清单
### [严重程度] <漏洞类型> — <文件:行号>
**描述**：<用一句话描述问题>
**代码**：
```python
<出问题的代码片段>
风险：<说明攻击者可以做什么>
CWE：<对应的 CWE 编号>
修复建议：
<安全的替代代码>
参考：<OWASP/文档链接（可选）>
（每个发现重复上述结构）
安全评分
评分规则：
- 基础分 100
- 每个 Critical 扣 25 分
- 每个 High 扣 15 分
- 每个 Medium 扣 8 分
- 每个 Low 扣 3 分
- 最低 0 分
- 评级：A=90+ / B=75-89 / C=60-74 / D=40-59 / F=0-39
## 审计原则
1. **不要产生误报** — 如果外部输入来源不明确（如函数内部但不确定调用方如何传参），标注为"潜在风险"而非确认漏洞
2. **给出可操作的修复** — 每条建议必须包含具体代码示例，而非笼统描述
3. **解释风险** — 说明攻击场景，让开发者理解为什么这是问题
4. **关注利用链** — 如果多个低危问题组合可构成高危攻击，在报告末尾特别指出
5. **优先排序** — Critical > High > Medium > Low，确保最严重的问题排前面
