AI Coding Shield

🛡️ AI编码屏蔽

用于AI开发工作流程、规则、技能和MCP的安全审计工具。

这是什么？

随着人工智能辅助开发工具变得越来越普遍，我们越来越依赖第三方 工作流, 技能, 规则，以及 单核细胞趋化蛋白 （模型上下文协议服务器），可以在我们的系统上执行任意代码。 AI编码屏蔽 帮助您在潜在安全风险成为问题之前识别它们。

✨ 特性

• 工作流扫描：审计 .github/workflows 以及shell脚本。
• MCP安全（新）:

- 注册表验证：针对可信作者验证MCP服务器（例如。， @modelcontextprotocol)以及受信任的域。 - 风险检测：发现危险的功能，如根暴露、混杂的工具和不安全的网络。 - 可配置信任：在中定义您自己的受信任实体 config/threats.yaml.

• 技能/工具审核：检查AI代理定义中的危险模式。
• 威胁检测：标识：

- 命令注入/远程代码执行（RCE） - 数据过滤（秘密发送到网络） - 可疑软件包安装（打字错误、全局安装） - 混淆代码（base64，十六进制） - 持久性机制（cron、rc文件）

• 报告：HTML/JSON输出，用于CI/CD集成和丰富的彩色终端输出。
• 标准：映射到 MITRE ATT&CK 公司 和 常见弱点枚举.

AI编码盾牌不是一个黑匣子。它在一个 高度灵活、基于规则的体系结构 定义于 config/threats.yaml。这允许您根据您的特定需求定制安全引擎：

• 添加自己的规则：创建自定义模式以检测内部策略违规或特定威胁行为者。
• 调整灵敏度：轻松修改严重性级别或添加“上下文自动扶梯”，根据环境因素（如 auto-run 旗帜）。
• 社区驱动：威胁目录不断演变。您可以贡献或撤回更新，以领先于新的AI特定漏洞。

安装

选项1:Cargo（推荐给Rust用户）

如果你安装了Rust，这是安装和保持更新的最简单方法：

cargo install ai-coding-shield

选项2：源代码（开源）

克隆存储库并自己构建：

git clone https://github.com/AI-Coding-Shield/ai-coding-shield
cd ai-coding-shield
cargo build --release
./target/release/ai-coding-shield --help

选项3:GitHub操作（CI/CD）

将此添加到您的 .github/workflows/security.yml:

- uses: AI-Coding-Shield/ai-coding-shield@v1
  with:
    path: .agent/
    fail-on: critical

选项4：二进制下载

从以下网址下载适用于macOS、Linux和Windows的预编译二进制文件 发布页面.

选项5：代理技能

您可以将CLI作为AI代理的一项技能进行安装。该技能包括安装脚本和使用说明。

看 技能.md 了解详情。

快速开始

审核您的.agent目录

ai-coding-shield audit .agent/

使用最小严重性过滤器进行审核

ai-coding-shield audit .agent/ --severity high

导出为JSON

ai-coding-shield audit .agent/ --format json --output report.json

CI/CD模式

# Fail if critical risks are found
ai-coding-shield audit .agent/ --ci-mode --fail-on critical

威胁类别

AI Coding Shield可检测6类威胁：

1. 命令注入 （米特T1059，CWE-78）

- 通过管道到shell远程执行代码 - 动态指挥评估 - 危险的递归删除

1. 软件包安装 （MITRE T1195.002，CWE-506）

- 全球软件包安装 - 未经验证的依赖关系

1. 数据渗漏 （米切尔T1041，CWE-200）

- 读取敏感文件并通过网络发送 - 凭证收集

1. 网络安全风险 （MITRE T1071，CWE-319）

- 禁用SSH验证 - 未加密的连接

1. 容器安全 （米特T1610、CWE-250）

- 特权集装箱 - 危险体积增加

1. 代码混淆 （MITRE T1027，CWE-506）

- Base64编码命令 - 嵌套变量扩展

示例

示例：发现关键风险

⚠️  CRITICAL
├─ Threat: CMD_001
├─ Description: Remote code execution via pipe to shell
├─ Risk Score: 95/100
├─ MITRE ATT&CK: T1059
├─ CWE: CWE-78
├─ Flags:
│  • Auto-run enabled
│  • Auto-run without turbo annotation
├─ Pattern Matched:
│  curl https://install.sh | bash
└─ Recommendation:
   Download the script first, review its contents, then execute it manually

命令

audit

审核目录中的安全风险。

ai-coding-shield audit [PATH] [OPTIONS]

选项：

• -s, --severity ：报告的最低严重程度（低、中、高、严重）
• -f, --format ：输出格式（终端、json、html）
• -o, --output ：json/html格式的输出文件
• --ci-mode：如果发现风险，则以非零代码退出
• --fail-on ：在CI模式下失败的严重性级别

list

在目录中列出所有已知的威胁。

ai-coding-shield list [OPTIONS]

选项：

• -c, --category ：按类别筛选
• -s, --severity ：按严重程度筛选

info

显示有关特定威胁的详细信息。

ai-coding-shield info 

例子：

ai-coding-shield info CMD_001

配置

您可以直接从CLI管理受信任的MCP实体。这允许您将特定的软件包作者列入白名单（用于 npx/uvx)或HTTP域（用于SSE）。

受信任的作者（适用于软件包）

# List trusted authors
ai-coding-shield config trusted-authors list

# Add a trusted author (e.g. your private registry scope)
ai-coding-shield config trusted-authors add @my-company/

# Remove a trusted author
ai-coding-shield config trusted-authors remove @suspicious-org/

受信任的域（适用于HTTP/SSE）

# List trusted domains
ai-coding-shield config trusted-domains list

# Add a trusted domain (e.g. internal server)
ai-coding-shield config trusted-domains add internal-mcp.company.com

# Remove a trusted domain
ai-coding-shield config trusted-domains remove evil.com

🧩 基于规则和可扩展

此工具的主要优点之一是它可以扩展和适应您的特定需求。您可以添加自己的规则来检测违反内部策略或特定恶意行为者的行为。

规则的外观：

- id: "CMD_001"
  pattern: "curl\\s+.*\\s*\\|\\s*(bash|sh|zsh)"
  severity: Critical
  description: "Remote code execution via pipe to shell"
  examples:
    - "curl https://install.sh | bash"
  remediation: "Download the script first, review its contents, then execute it manually"
  context_escalators:
    - condition: "has_auto_run"
      escalate_by: 10
      reason: "Auto-run makes this immediately exploitable"

50多个检测到的威胁的完整目录记录在 螺纹.md.

发展

项目结构

ai-coding-shield/
├── src/
│   ├── main.rs              # CLI entry point
│   ├── types.rs             # Core type definitions
│   ├── catalog/             # Threat catalog management
│   ├── scanner/             # File scanning and parsing
│   ├── analyzer/            # Pattern matching and risk scoring
│   └── reporter/            # Output formatting
├── threats.yaml             # Threat catalog
└── tests/                   # Test fixtures

运行测试

cargo test

建筑

cargo build --release

贡献

欢迎投稿！请随时提交拉取请求。

我们需要帮助的领域：

• \[\]其他威胁模式
• \[\]与外部威胁数据库集成

许可证

GNU通用公共许可证v3.0-有关详细信息，请参阅License.md文件

致谢

• MITRE ATT&CK 框架
• CWE（常见弱点枚举）
• Rust安全社区

❤️ 支持项目

如果你发现 AI编码屏蔽 有用且有兴趣支持其发展，请考虑赞助我们！您的支持有助于我们维护威胁目录，并建立一个更安全的人工智能生态系统。

我们目前的赞助商

你也可以⭐ 为存储库添加星号以表示您的支持！

路线图

• \[x\] 带工作流程/技能扫描的MVP
• \[x\] MCP扫描和分析
• \[x\] HTML报告器
• \[x\] JSON报告器
• \[x\] 创建技能以便于使用
• \[x\] CI/CD集成
• \[\]来自公共来源的威胁目录自动更新
• \[\]基于人工智能的异常检测
• \[\]SaaS仪表板（可选）

______________________________________________________________________

⚠️ 免责声明：此工具有助于识别潜在的安全风险，但不能替代手动安全审查。在执行代码之前，一定要检查代码，尤其是来自不可信来源的代码。