0x01 工具介绍
AI Infra Guard(AI Infrastructure Guard) 是一个高效、轻量、易用的AI基础设施安全评估工具，专为发现和检测AI系统潜在安全风险而设计。
下载地址 https://github.com/Tencent/AI-Infra-Guard
0x02 功能简介
🚀 快速预览
WEBUI![AI基础设施安全评估系统]( "AI基础设施安全评估系统|漏洞探测")
命令行
命令行
🚀 项目亮点
-
高效扫描
-
支持 28 种 AI 框架指纹识别 -
涵盖 200+ 安全漏洞数据库
-
-
易于使用
-
开箱即用，无复杂配置 -
指纹、漏洞YAML规则定义 -
灵活的匹配语法
-
-
轻量级
-
核心组件简洁高效 -
二进制体积小，资源占用低 -
跨平台支持
-
📊 AI组件覆盖情况
|
|
|
|---|---|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
0x03更新说明
5672e24 update version to 0.1devcaa95c8 更新对version latest的处理e4b2a8e auto-update from HunYuan:2025-03-07 10:56:21 | Updated 2 vulnerability fingerprints29dcf15 update serverc5dd937 auto-update from HunYuan:2025-03-05 19:11:31 | Updated 2 vulnerability fingerprintsfe46d98 新增xinference指纹3a8b2e9 update readme_cnacd9d17 update readme
0x04 使用介绍
使用
WEBUI 可视化操作
./ai-infra-guard -ws本地一键检测
./ai-infra-guard -localscan单个目标
./ai-infra-guard -target [IP/域名] ./ai-infra-guard -target [IP/域名] -target [IP/域名]./ai-infra-guard -file target.txt# hunyuan token./ai-infra-guard -target [IP/Domain] -ai -hunyuan-token [Hunyuan token]# deepseek token./ai-infra-guard -target [IP/Domain] -ai -deepseek-token [Deepseek token]
🔍 指纹匹配规则
AI Infra Guard 基于WEB指纹识别组件，指纹规则在data/fingerprints目录中，漏洞匹配规则在data/vuln目录中。
示例：Gradio 指纹规则
info:name: gradioauthor: Security Teamseverity: infometadata:product: gradiovendor: gradiohttp:- method: GETpath: '/'matchers:- body="<script>window.gradio_config = {" || body="document.getElementsByTagName("gradio-app");"
🛠️ 指纹匹配语法
匹配位置标题（title）正文（body）请求头（header）图标哈希（icon）
逻辑运算符
= 模糊匹配== 全等!= 不等~= 正则匹配&& 与|| 或() 括号分组
0x05 下载：
https://github.com/Tencent/AI-Infra-Guard
原文始发于微信公众号（渗透安全HackTwo）：AI基础设施安全评估系统|漏洞探测
免责声明:文章中涉及的程序(方法)可能带有攻击性，仅供安全研究与教学之用，读者将其信息做其他用途，由读者承担全部法律及连带责任，本站不承担任何法律及连带责任；如有问题可邮件联系(建议使用企业邮箱或有效邮箱,避免邮件被拦截，联系方式见首页)，望知悉。
- 左青龙
- 微信扫一扫
-
- 右白虎
- 微信扫一扫
-
评论