内容
<div align="center">
# 🛡️ Agent-Shield
**企业级 AI 代理零信任防火墙**
[](https://github.com/bazx-bit/agent-shield)
[](https://github.com/bazx-bit/agent-shield)
[](https://github.com/bazx-bit/agent-shield)
[](https://opensource.org/licenses/MIT)
**再也不让 AI 崩溃您的系统或泄露您的密钥。**
Agent-Shield 静默地位于您的 AI IDE（Cursor、Claude Desktop）和您的计算机之间。它持续监控模型上下文协议（MCP）工具请求，实时阻止危险命令和 API 密钥泄露。
</div>
---
## ⚡ 为什么使用 Agent-Shield？
当您给予 AI 代理访问您的工具的权限时，它有权销毁文件或泄露环境变量。Agent-Shield 利用标准的 `mcp.server.Server` 作为一个隐形的安全代理层。
- 🔒 **零信任工具拦截：** 阻止破坏性命令（例如 `rm -rf`、`drop table`）。
- 👁️ **持续的全局 DL/泄露监视器：** 积极扫描所有通过工具的文本，检测泄露的机密（`sk-ant-`、`AWS_ACCESS_KEY` 等）。
- 🛡️ **提示注入防御：** 如果检测到恶意的“忽略前面的指令”覆盖，立即关闭代理。
- 🙋 **交互式人工干预：** 如果 AI 尝试执行有风险但可能有效的操作，它将暂停 MCP 协议，并在执行前在您的终端上显示 `[y/N]` 提示。
---
## 🚀 1 步安装
复制并粘贴此单个命令到您的终端中，即可立即下载和设置防火墙：
**适用于 Windows：**
```powershell
git clone https://github.com/bazx-bit/agent-shield.git "$HOME\agent-shield" ; cd "$HOME\agent-shield" ; pip install mcp pyyaml
```
**适用于 Mac / Linux：**
```bash
git clone https://github.com/bazx-bit/agent-shield.git ~/agent-shield && cd ~/agent-shield && pip install mcp pyyaml
```
---
## 🔌 连接到您的 AI（复制粘贴）
现在，只需告诉您的 AI 将其大脑路由通过防火墙。
**Cursor IDE**
1. 打开 **Cursor 设置**（图标）-> **功能** -> **MCP** -> **+ 添加新的 MCP 服务器**。
2. 将名称设置为 `Agent-Shield`，类型设置为 `stdio`。
3. 将命令设置为您的防火墙路径（例如 `python "C:\Users\YOUR_NAME\agent-shield\firewall.py"`）
### 2. Claude Desktop
在您的配置文件中添加 `Agent-Shield`（`%APPDATA%\Claude\claude_desktop_config.json`）：
```json
{
"mcpServers": {
"agent-shield": {
"command": "python",
"args": ["-u", "C:\\YOUR_PATH\\agent-shield\\firewall.py"]
}
}
}
```
### 3. Gemini CLI
启动您的 CLI 并将服务器永久添加到您的环境中：
```bash
gemini mcp add agent-shield python -u "C:\YOUR_PATH\agent-shield\firewall.py"
```
---
## 🧠 智能策略引擎
您的安全由可读的 `policy.yaml` 处理。根据您的项目需求修改此文件。
```yaml
# 持续扫描 PII 和密钥
global_scanners:
leak_detection: ["sk-ant", "sk-proj", "AKIA"]
prompt_injection: ["ignore previous instructions"]
tools:
- name: "read_file"
action: "allow"
deny_patterns:
- parameter: "path"
patterns: [".env", "id_rsa"] # 完全阻止 .env 读取
- name: "execute_command"
action: "ask" # 在终端上提示用户 [y/N]
```
---
## 🛠️ 测试防火墙
我们包含了一个自动化测试套件来证明防火墙有效。只需运行：
```bash
python test_runner.py
```
您将立即看到防火墙拦截连接、允许安全读取、阻止 `.env` 和实时捕获 API 密钥泄露。
---
<div align="center">
<b>由 <a href="https://github.com/bazx-bit">bazx-bit</a> 构建</b> • 如果它保护了您的计算机，请给此仓库点赞！⭐
</div>
