CVE-2025-32434：PyTorch远程代码执行漏洞 - PoC

什么是CVE-2025-32434？

CVE-2025-32434 是PyTorch中的一个严重远程代码执行（RCE）漏洞，PyTorch是深度学习和神经网络中广泛使用的Python库。

核心问题

该漏洞存在于torch.load()函数中，即使使用了看似安全的weights_only=True参数。weights_only=True选项旨在通过仅允许加载模型参数来防止代码执行。然而，研究人员找到了一种绕过此保护的方法。

受影响版本

严重性

• CVSS评分：10分中的9.8分（严重）
• 发现者：Ji'an Zhou（在Black Hat USA 2025上展示）
• 影响：未认证的攻击者可在加载恶意模型时执行任意代码
• 用户交互：无需

技术影响

未认证的攻击者可以创建一个恶意模型文件，当受害者加载该文件时，会执行任意代码。攻击无需用户交互，可导致系统完全受损。该漏洞利用了PyTorch使用Python的pickle序列化的方式，允许攻击者在模型文件中嵌入恶意代码。

危险

此漏洞尤其危险，因为许多开发者依赖weights_only=True作为安全措施，但此漏洞证明即使是“安全”选项也不足够。

PoC工作原理

提供的full.py脚本演示了如何创建一个在加载时写入文件系统的恶意PyTorch模型。

攻击流程

1. 载荷创建：脚本生成包含反弹shell载荷的cron条目
2. ASCII转换：将载荷字符串转换为ASCII码列表
3. 内存映射：使用torch.from_file()创建指向目标cron文件的内存映射张量
4. 数据写入：将ASCII值复制到内存映射区域，从而写入文件
5. 模型序列化：将操作保存为TorchScript模型
6. 执行：模型在加载并执行时，将载荷写入目标文件

代码示例

# Memory-mapped file writing
t = torch.from_file("/etc/cron.d/rev", 
                    shared=True, 
                    size=len(asciis), 
                    dtype=torch.uint8)
msg = torch.tensor(asciis, dtype=torch.uint8)
t.copy_(msg)  # Writes to /etc/cron.d/rev

CVE-2025-32434 PoC - 用户指南

如何使用此概念验证

本指南提供如何使用CVE-2025-32434概念验证（PoC）脚本的分步说明。

先决条件

开始之前，请确保您具备：

• 已安装PyTorch 2.5.1或更早版本（有漏洞版本）

安装PyTorch

# Install PyTorch (vulnerable version for testing)
pip install torch==2.5.1

或安装最新版本（用于测试修复）

pip install torch