mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-07-08 13:37:57 +02:00
Add files via upload
This commit is contained in:
公明
@@ -0,0 +1,319 @@
|
||||
# CyberStrikeAI
|
||||
|
||||
基于Golang和Gin框架的AI驱动自主渗透测试平台，使用MCP协议集成安全工具。
|
||||
|
||||
## 功能特性
|
||||
|
||||
- 🤖 **AI代理连接** - 支持Claude、GPT等兼容MCP的AI代理通过FastMCP协议连接
|
||||
- 🧠 **智能分析** - 决策引擎分析目标并选择最佳测试策略
|
||||
- ⚡ **自主执行** - AI代理执行全面的安全评估
|
||||
- 🔄 **实时适应** - 系统根据结果和发现的漏洞进行调整
|
||||
- 📊 **高级报告** - 可视化方式输出漏洞卡片和风险分析
|
||||
- 💬 **对话式交互** - 前端以对话形式调用后端agent-loop
|
||||
- 📈 **实时监控** - 监控安全工具的执行状态、结果、调用次数等
|
||||
|
||||
## 项目结构
|
||||
|
||||
```
|
||||
CyberStrikeAI/
|
||||
├── cmd/
|
||||
│ └── server/
|
||||
│ └── main.go # 程序入口
|
||||
├── internal/
|
||||
│ ├── agent/ # AI代理模块
|
||||
│ ├── app/ # 应用初始化
|
||||
│ ├── config/ # 配置管理
|
||||
│ ├── handler/ # HTTP处理器
|
||||
│ ├── logger/ # 日志系统
|
||||
│ ├── mcp/ # MCP协议实现
|
||||
│ └── security/ # 安全工具执行器
|
||||
├── web/
|
||||
│ ├── static/ # 静态资源
|
||||
│ │ ├── css/
|
||||
│ │ └── js/
|
||||
│ └── templates/ # HTML模板
|
||||
├── config.yaml # 配置文件
|
||||
├── go.mod # Go模块文件
|
||||
└── README.md # 说明文档
|
||||
```
|
||||
|
||||
## 快速开始
|
||||
|
||||
### 前置要求
|
||||
|
||||
- Go 1.21 或更高版本
|
||||
- OpenAI API Key（或其他兼容OpenAI协议的API）
|
||||
- 安全工具（可选）：nmap, sqlmap, nikto, dirb
|
||||
|
||||
### 安装步骤
|
||||
|
||||
1. **克隆项目**
|
||||
```bash
|
||||
cd /Users/temp/Desktop/wenjian/tools/CyberStrikeAI
|
||||
```
|
||||
|
||||
2. **安装依赖**
|
||||
```bash
|
||||
go mod download
|
||||
```
|
||||
|
||||
3. **配置**
|
||||
编辑 `config.yaml` 文件，设置您的OpenAI API Key：
|
||||
```yaml
|
||||
openai:
|
||||
api_key: "sk-your-api-key-here"
|
||||
base_url: "https://api.openai.com/v1"
|
||||
model: "gpt-4"
|
||||
```
|
||||
|
||||
4. **启动服务器**
|
||||
|
||||
#### 方式一：使用启动脚本
|
||||
```bash
|
||||
./run.sh
|
||||
```
|
||||
|
||||
#### 方式二：直接运行
|
||||
```bash
|
||||
go run cmd/server/main.go
|
||||
```
|
||||
|
||||
#### 方式三：编译后运行
|
||||
```bash
|
||||
go build -o cyberstrike-ai cmd/server/main.go
|
||||
./cyberstrike-ai
|
||||
```
|
||||
|
||||
5. **访问应用**
|
||||
打开浏览器访问：http://localhost:8080
|
||||
|
||||
## 配置说明
|
||||
|
||||
### 服务器配置
|
||||
```yaml
|
||||
server:
|
||||
host: "0.0.0.0"
|
||||
port: 8080
|
||||
```
|
||||
|
||||
### MCP配置
|
||||
```yaml
|
||||
mcp:
|
||||
enabled: true
|
||||
host: "0.0.0.0"
|
||||
port: 8081
|
||||
```
|
||||
|
||||
### 安全工具配置
|
||||
```yaml
|
||||
security:
|
||||
tools:
|
||||
- name: "nmap"
|
||||
command: "nmap"
|
||||
args: ["-sV", "-sC"]
|
||||
description: "网络扫描工具"
|
||||
enabled: true
|
||||
```
|
||||
|
||||
## 使用示例
|
||||
|
||||
### 对话式渗透测试
|
||||
|
||||
在"对话测试"标签页中，您可以：
|
||||
|
||||
1. **网络扫描**
|
||||
```
|
||||
扫描 192.168.1.1 的开放端口
|
||||
```
|
||||
|
||||
2. **SQL注入检测**
|
||||
```
|
||||
检测 https://example.com 的SQL注入漏洞
|
||||
```
|
||||
|
||||
3. **Web漏洞扫描**
|
||||
```
|
||||
扫描 https://example.com 的Web服务器漏洞
|
||||
```
|
||||
|
||||
4. **目录扫描**
|
||||
```
|
||||
扫描 https://example.com 的隐藏目录
|
||||
```
|
||||
|
||||
### 监控工具执行
|
||||
|
||||
在"工具监控"标签页中，您可以：
|
||||
|
||||
- 查看所有工具的执行统计
|
||||
- 查看详细的执行记录
|
||||
- 查看发现的漏洞列表
|
||||
- 实时监控工具状态
|
||||
|
||||
## API接口
|
||||
|
||||
### Agent Loop API
|
||||
|
||||
**POST** `/api/agent-loop`
|
||||
|
||||
请求体：
|
||||
```json
|
||||
{
|
||||
"message": "扫描 192.168.1.1"
|
||||
}
|
||||
```
|
||||
|
||||
使用示例：
|
||||
```bash
|
||||
curl -X POST http://localhost:8080/api/agent-loop \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{"message": "扫描 192.168.1.1"}'
|
||||
```
|
||||
|
||||
### 监控API
|
||||
|
||||
- **GET** `/api/monitor` - 获取所有监控信息
|
||||
- **GET** `/api/monitor/execution/:id` - 获取特定执行记录
|
||||
- **GET** `/api/monitor/stats` - 获取统计信息
|
||||
- **GET** `/api/monitor/vulnerabilities` - 获取漏洞列表
|
||||
|
||||
使用示例：
|
||||
```bash
|
||||
# 获取所有监控信息
|
||||
curl http://localhost:8080/api/monitor
|
||||
|
||||
# 获取统计信息
|
||||
curl http://localhost:8080/api/monitor/stats
|
||||
|
||||
# 获取漏洞列表
|
||||
curl http://localhost:8080/api/monitor/vulnerabilities
|
||||
```
|
||||
|
||||
### MCP接口
|
||||
|
||||
**POST** `/api/mcp` - MCP协议端点
|
||||
|
||||
## MCP协议
|
||||
|
||||
本项目实现了MCP（Model Context Protocol）协议，支持：
|
||||
|
||||
- `initialize` - 初始化连接
|
||||
- `tools/list` - 列出可用工具
|
||||
- `tools/call` - 调用工具
|
||||
|
||||
工具调用是异步执行的，系统会跟踪每个工具的执行状态和结果。
|
||||
|
||||
### MCP协议使用示例
|
||||
|
||||
#### 初始化连接
|
||||
|
||||
```bash
|
||||
curl -X POST http://localhost:8080/api/mcp \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{
|
||||
"jsonrpc": "2.0",
|
||||
"id": "1",
|
||||
"method": "initialize",
|
||||
"params": {
|
||||
"protocolVersion": "2024-11-05",
|
||||
"capabilities": {},
|
||||
"clientInfo": {
|
||||
"name": "test-client",
|
||||
"version": "1.0.0"
|
||||
}
|
||||
}
|
||||
}'
|
||||
```
|
||||
|
||||
#### 列出工具
|
||||
|
||||
```bash
|
||||
curl -X POST http://localhost:8080/api/mcp \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{
|
||||
"jsonrpc": "2.0",
|
||||
"id": "2",
|
||||
"method": "tools/list"
|
||||
}'
|
||||
```
|
||||
|
||||
#### 调用工具
|
||||
|
||||
```bash
|
||||
curl -X POST http://localhost:8080/api/mcp \
|
||||
-H "Content-Type: application/json" \
|
||||
-d '{
|
||||
"jsonrpc": "2.0",
|
||||
"id": "3",
|
||||
"method": "tools/call",
|
||||
"params": {
|
||||
"name": "nmap",
|
||||
"arguments": {
|
||||
"target": "192.168.1.1",
|
||||
"ports": "1-1000"
|
||||
}
|
||||
}
|
||||
}'
|
||||
```
|
||||
|
||||
## 安全工具支持
|
||||
|
||||
当前支持的安全工具：
|
||||
- **nmap** - 网络扫描
|
||||
- **sqlmap** - SQL注入检测
|
||||
- **nikto** - Web服务器扫描
|
||||
- **dirb** - 目录扫描
|
||||
|
||||
可以通过修改 `config.yaml` 添加更多工具。
|
||||
|
||||
## 故障排除
|
||||
|
||||
### 问题：无法连接到OpenAI API
|
||||
|
||||
- 检查API Key是否正确
|
||||
- 检查网络连接
|
||||
- 检查base_url配置
|
||||
|
||||
### 问题：工具执行失败
|
||||
|
||||
- 确保已安装相应的安全工具（nmap, sqlmap等）
|
||||
- 检查工具是否在PATH中
|
||||
- 某些工具可能需要root权限
|
||||
|
||||
### 问题：前端无法加载
|
||||
|
||||
- 检查服务器是否正常运行
|
||||
- 检查端口8080是否被占用
|
||||
- 查看浏览器控制台错误信息
|
||||
|
||||
## 安全注意事项
|
||||
|
||||
⚠️ **重要提示**：
|
||||
|
||||
- 仅对您拥有或已获得授权的系统进行测试
|
||||
- 遵守相关法律法规
|
||||
- 建议在隔离的测试环境中使用
|
||||
- 不要在生产环境中使用
|
||||
- 某些安全工具可能需要root权限
|
||||
|
||||
## 开发
|
||||
|
||||
### 添加新工具
|
||||
|
||||
1. 在 `config.yaml` 中添加工具配置
|
||||
2. 在 `internal/security/executor.go` 的 `buildCommandArgs` 方法中添加参数构建逻辑
|
||||
3. 在 `internal/agent/agent.go` 的 `getAvailableTools` 方法中添加工具定义
|
||||
|
||||
### 构建
|
||||
|
||||
```bash
|
||||
go build -o cyberstrike-ai cmd/server/main.go
|
||||
```
|
||||
|
||||
## 许可证
|
||||
|
||||
本项目仅供学习和研究使用。
|
||||
|
||||
## 贡献
|
||||
|
||||
欢迎提交Issue和Pull Request！
|
||||
Reference in New Issue
Block a user