一、漏洞详情

ComfyUI是一款开源的基于节点工作流的Stable Diffusion和生成式AI图像处理工具。

近日，监测到ComfyUI反序列化远程代码执行漏洞。该漏洞由于系统未对用户上传的pickle文件进行有效校验，攻击者可通过未授权访问的POST /upload/image接口上传构造的恶意shard_*.pkl文件，并通过POST /prompt接口触发工作流执行，使torch.load对攻击者控制的pickle数据进行反序列化。

建议受影响用户做好资产自查以及预防工作，以免遭受黑客攻击。

二、影响范围

ComfyUI <= 0.23.0

三、修复建议

官方已发布安全补丁，请及时更新至安全版本。

ComfyUI > 0.23.0