本文目录导读：
PyYAML安全加载模式详解：如何避免反序列化漏洞
目录导读
PyYAML安全加载模式概述
在Python生态中，PyYAML是最流行的YAML解析库之一，开发者经常问：“PyYAML安全加载模式有吗？”答案是有，PyYAML从5.1版本开始，默认的加载函数yaml.load()就不再安全，并会发出警告，官方推荐使用yaml.safe_load()来实现安全的反序列化，安全加载模式的核心目的是防止YAML文件中的恶意代码被执行，从而避免反序列化漏洞（例如任意代码执行）。
为什么需要安全加载模式？
YAML格式本身支持自定义标签（如!!python/object），这些标签可以实例化任意Python对象，如果加载不受信任的YAML数据，攻击者可以通过构造恶意YAML文件，让PyYAML执行系统命令（例如!!python/object/apply:os.system）。
!!python/object/apply:os.system ["rm -rf /"]
如果使用yaml.load()，该代码会被执行，而yaml.safe_load()会拒绝加载任何非标准YAML标签,从而阻止攻击。
PyYAML支持的三种加载模式
PyYAML提供了三种主要加载模式,理解它们的区别至关重要：
| 模式 | 函数 | 安全性 | 适用场景 |
|---|---|---|---|
| 完全加载 | yaml.load(data, Loader=yaml.FullLoader) |
中等 | 需要加载自定义标签，且数据完全可信 |
| 不安全加载 | yaml.load(data, Loader=yaml.UnsafeLoader) |
极低 | 保留旧行为，强烈不推荐 |
| 安全加载 | yaml.safe_load(data) |
高 | 处理不受信任或用户输入的YAML |
注意：yaml.safe_load()默认使用SafeLoader，它只支持YAML标准类型（如列表、字典、字符串、数字等），如果YAML中包含!!python/或其他非标准标签，safe_load会抛出异常。
如何正确使用安全加载模式？
1 基本用法
import yaml
# 安全加载 - 推荐
with open('config.yaml', 'r') as f:
data = yaml.safe_load(f)
# 如果必须使用完全加载（仅限非常可信的数据）
with open('trusted_config.yaml', 'r') as f:
data = yaml.load(f, Loader=yaml.FullLoader)
2 何时不能使用safe_load？
如果你确实需要加载自定义Python对象（例如从配置文件恢复类实例），且数据完全由你自己生成，你可以使用yaml.load()配合FullLoader,但需要确保：
- YAML文件不会被外部用户修改
- 文件来源绝对可信
3 新版PyYAML的默认行为
从PyYAML 5.1开始，直接调用yaml.load(data)而不指定Loader会引发警告，并在更早版本（如6.0+）中直接报错,显式指定加载器已成为最佳实践。
常见问题与问答
Q1：使用safe_load后，能否加载自定义Python类？
不能。safe_load严格限制为标准YAML类型，如果需要加载自定义类，需使用yaml.load()并指定Loader=yaml.FullLoader,但请先确认数据来源安全。
Q2：如何检查YAML文件是否安全？
- 用
yaml.safe_load()尝试加载，如果抛出yaml.constructor.ConstructorError,说明文件包含非标准标签。 - 也可手动扫描文件是否包含
!!python/或!<tag:yaml.org,2002:python>等关键字。
Q3：yaml.safe_load和yaml.load(Loader=yaml.SafeLoader)有区别吗？
没有区别。yaml.safe_load(data)等价于yaml.load(data, Loader=yaml.SafeLoader)，后者写法更显式，但前者更简洁,官方推荐使用。
Q4：为什么有的教程还直接用yaml.load()？
那些教程可能基于旧版本PyYAML（5.1之前），或者作者未强调安全性，在2025年的今天，任何涉及用户输入的YAML解析都应使用safe_load。
最佳实践总结
- 默认使用
safe_load()– 安全第一,适用于绝大多数场景。 - 永远不要用
UnsafeLoader– 除非你是安全研究员做测试。 - 显式声明加载器 – 即使使用
FullLoader，也应明确写出yaml.load(data, Loader=yaml.FullLoader),避免依赖遗留行为。 - 对用户输入永远保持警惕 – 如果YAML来自上传文件、API请求或外部配置,务必使用安全模式。
- 升级PyYAML版本 – 确保使用5.1以上版本,以获得安全警告和默认行为改进。
延伸阅读：如果你需要更灵活的安全与功能平衡，可以考虑使用yaml.CSafeLoader（C语言实现，性能更高，但同样安全），对于要求极高的场景，推荐使用yaml.RoundTripLoader配合自定义安全检查。
