检测PyTorch pickle后门的扫描工具

torch.load执行pickle可触发任意代码，作者构建了Model-Supply-Chain-Auditor来反汇编检测恶意字节码。

- PyTorch模型文件使用pickle序列化，通过reduce可注入os.system等调用，实现RCE，攻击代码仅需4行。
- 工具解析pickle opcode，标记STACK_GLOBAL+REDUCE模式（如os.system、subprocess），支持协议0-5，包括Windows（nt.system）和Linux（posix.system）变体。
- 防御建议：使用SafeTensors格式替代pickle；加载前验证SHA-256签名；用本工具扫描后再加载。

GitHub
原文链接

#开发者 #工具
📢 频道：@DevToolboxHub