IPSec-VPN-Classification

This project focuses on IPSec VPN encrypted traffic classification.

文档说明

项目已包含基础开源文档和使用说明文件，包括 README.md、requirements.txt、environment.yml、QUICK_START_OPENSOURCE.md、CONTRIBUTING.md、SECURITY.md、CHANGELOG.md、LICENSE 等。

安装步骤如下：

1. 克隆或下载项目代码。

2. 创建 Python 环境并安装依赖：

pip install -r requirements.txt

也可以使用 Conda 环境文件：

conda env create -f environment.yml
conda activate ipsec-vpn-classification

3. 准备模型和标准化参数。项目已包含训练模型文件 model/length_prediction.keras。在线推理流程还需要由预处理流程生成或提供 model/processed_data.npz，其中保存输入和输出标准化参数。

4. 启动 Web 服务：

python app.py

服务默认监听 http://127.0.0.1:5000 或 http://0.0.0.0:5000。用户打开浏览器后上传 .pcap 文件，即可查看提取特征和预测结果，并下载 CSV 文件。

5. 训练与复现实验流程如下：

python model/traffic_decryption.py
python model/feature_extraction.py
python model/data_preprocess.py
python model/model_training.py
python test.py

其中 traffic_decryption.py 用于实验数据解密与加密/明文包对齐，feature_extraction.py 用于生成训练特征，data_preprocess.py 用于生成标准化后的训练数据，model_training.py 用于训练 Transformer 模型，test.py 用于计算 MAE、RMSE 并导出结果。