2025 年初,DeepSeek R1 开源发布,因低成本、性能好受关注。由于官网常繁忙,大家多用 Ollama + OpenWebUI、LM Studio 等工具本地部署,将其引入企业内网和个人 PC。但腾讯朱雀实验室发现这些 AI 工具存在安全漏洞,使用不当可能被攻击者窃取数据、滥用算力资源,甚至控制设备。
二、常见 AI 工具安全问题
(一)Ollama
功能与风险:Ollama 可在多设备本地部署 LLM,简化部署过程,但默认启动开放 11434 端口,docker 中默认以 root 权限启动并开放到公网,且接口无鉴权。
攻击手段:攻击者可进行模型删除、窃取、算力窃取、模型投毒等操作,还存在远程命令执行漏洞 CVE-2024 - 37032,影响 Ollama 0.1.34 之前版本。
缓解方案:升级到最新版 Ollama,运行 ollama serve 时确认环境变量 OLLAMA_HOST 为本地地址,避免公网运行,建议本地运行 Ollama 后用反向代理工具(如 Nginx)为服务端增访问保护。
(二)OpenWebUI
功能与风险:OpenWebUI 是流行的大模型对话 webui,含多种功能且方便与 Ollama 集成,但存在文件上传漏洞【CVE-2024 - 6707】,可被攻击者利用上传恶意文件执行任意代码或实现远程命令执行。
缓解方案:升级到最新版 OpenWebUI,避免开启用户系统。
(三)ComfyUI
功能与风险:ComfyUI 是流行 diffusion 模型应用,插件生态丰富,但无鉴权方式,存在多个漏洞,影响包括远程命令执行、任意文件读取 / 写入,数据窃取等。
缓解方案:因漏洞修复缓慢,最新版本仍存漏洞,不建议将其暴露公网使用。
三、AI-Infra-Guard 工具
AI-Infra-Guard 是高效、轻量、易用的 AI 基础设施安全评估工具,可检测 30 种 AI 组件,涵盖常见 AI 应用和开发训练框架。利用大模型自动积累漏洞规则,使用友好,零依赖,开箱即用,内存占用小,跨平台兼容。
使用方法
一键检测
./ai-infra-guard -localscan
单个目标:
./ai-infra-guard -target [IP:PORT/域名]
多个目标:
./ai-infra-guard -target [IP:PORT/域名] -target [IP:PORT/域名]
扫描网段寻找 AI 服务:
./ai-infra-guard -target 192.168.1.0/24
从文件读取目标扫描:
./ai-infra-guard -file target.txt