4545| 🔎 ** Slips行为分析** | 端口扫描、DDoS、C2通信行为检测 |
4646| 🎮 ** RL安全响应** | DQN/PPO智能安全响应决策 |
4747| 📊 ** 统一管道** | 多算法级联融合检测 |
48+ | 🛡️ ** 一键防护** | VPN风格一键开启防护,支持多级别安全策略 |
4849| 🌐 ** Web界面** | 现代化响应式Web操作界面 |
4950| 🔌 ** RESTful API** | 完整的API接口,支持批量预测 |
5051| 📈 ** 实时监控** | 系统统计与实时状态监控 |
52+ | 🚀 ** 流量模拟** | 高并发流量模拟器,支持百万级压测 |
5153| 🐳 ** 容器化部署** | Docker + Kubernetes生产级部署 |
5254
5355## 🔬 集成算法
@@ -131,6 +133,11 @@ Network-Security-Based-On-ML/
131133│ │ ├── rules.py # 规则定义
132134│ │ └── api.py # 防火墙API
133135│ │
136+ │ ├── protection/ # 一键防护模块
137+ │ │ ├── service.py # 防护服务(单例模式)
138+ │ │ ├── api.py # 防护API
139+ │ │ └── __init__.py # 模块导出
140+ │ │
134141│ ├── components/ # 核心组件层
135142│ ├── pipeline/ # 训练与预测管道
136143│ ├── entity/ # 配置与产物实体类
@@ -150,8 +157,12 @@ Network-Security-Based-On-ML/
150157│ ├── predict.html # 预测页面
151158│ ├── model_select.html # 模型选择
152159│ ├── dashboard.html # 统计仪表盘
160+ │ ├── protection.html # 一键防护页面
153161│ └── training.html # 训练控制台
154162│
163+ ├── scripts/ # 脚本工具
164+ │ └── traffic_simulator.py # 高并发流量模拟器
165+ │
155166├── tests/ # 测试文件 (298 passed)
156167│ ├── test_github_algorithms.py # GitHub算法测试
157168│ ├── test_ml_models.py # ML模型测试
@@ -212,7 +223,9 @@ python app.py
212223| 页面 | 地址 | 描述 |
213224| ------| ------| ------|
214225| 首页 | ` / ` | 系统概览 |
226+ | 一键防护 | ` /protection ` | VPN风格一键开启防护 |
215227| 威胁预测 | ` /predict ` | 实时威胁检测 |
228+ | 仪表盘 | ` /dashboard ` | 实时流量监控 |
216229| 模型训练 | ` /train ` | 训练控制台 |
217230| 模型解释 | ` /explanation ` | SHAP可解释性分析 |
218231| 使用教程 | ` /tutorial ` | 操作指南 |
@@ -231,6 +244,11 @@ python app.py
231244| ` GET ` | ` /api/v1/stats/overview ` | 流量统计概览 |
232245| ` GET ` | ` /api/v1/stats/threats ` | 威胁分布统计 |
233246| ` POST ` | ` /api/v1/firewall/analyze ` | 防火墙流量分析 |
247+ | ` GET ` | ` /api/v1/protection/state ` | 获取防护状态 |
248+ | ` POST ` | ` /api/v1/protection/start ` | 启动防护服务 |
249+ | ` POST ` | ` /api/v1/protection/stop ` | 停止防护服务 |
250+ | ` POST ` | ` /api/v1/protection/toggle ` | 切换防护状态 |
251+ | ` POST ` | ` /api/v1/protection/level ` | 设置防护级别 |
234252
235253### 算法演示
236254
@@ -239,6 +257,38 @@ python app.py
239257python demo_algorithms.py
240258```
241259
260+ ### 一键防护
261+
262+ 系统提供VPN风格的一键防护功能,支持四种防护级别:
263+
264+ | 级别 | 描述 |
265+ | ------| ------|
266+ | ** 低级** | 仅监控记录,不拦截 |
267+ | ** 中级** | 拦截高风险威胁 |
268+ | ** 高级** | 拦截中高风险威胁 |
269+ | ** 严格** | 拦截所有可疑流量 |
270+
271+ 访问 ` /protection ` 页面,点击电源按钮即可一键开启防护。
272+
273+ ### 流量模拟器
274+
275+ 使用高并发流量模拟器进行压力测试:
276+
277+ ``` bash
278+ # 模拟100万请求
279+ python scripts/traffic_simulator.py -n 1000000 -c 500
280+
281+ # 低强度长时间测试
282+ python scripts/traffic_simulator.py -n 5000000 -c 50 -b 20
283+
284+ # 参数说明
285+ # -n: 总请求数
286+ # -c: 并发数
287+ # -b: 批次大小
288+ ```
289+
290+ 流量分布:70%正常用户、12%爬虫、8%机器人、10%攻击者
291+
242292输出示例:
243293```
244294============================================================
0 commit comments