# 检查当前内核配置
zcat /proc/config.gz 2>/dev/null || cat /boot/config-$(uname -r)

CONFIG_BPF=y
CONFIG_BPF_SYSCALL=y
CONFIG_BPF_LSM=y
CONFIG_DEBUG_INFO_BTF=y

# 检查 BPF LSM 是否激活
cat /sys/kernel/security/lsm
# 列表中应包含 "bpf"

# 如果没有，请添加到内核启动参数：
# 编辑 /etc/default/grub，在 GRUB_CMDLINE_LINUX 中添加：
#   lsm=lockdown,capability,landlock,yama,apparmor,bpf

# 然后更新 grub 并重启：
sudo update-grub
sudo reboot

# 创建启用 BPF LSM 的脚本
cat > /tmp/enable_bpf_lsm.sh << 'EOF'
#!/bin/bash
GRUB_FILE="/etc/default/grub"
if grep -q "lsm=" "$GRUB_FILE"; then
    sudo sed -i 's/lsm=[^""]*/lsm=lockdown,capability,landlock,yama,apparmor,bpf/' "$GRUB_FILE"
else
    sudo sed -i 's/GRUB_CMDLINE_LINUX="\(.*\)"/GRUB_CMDLINE_LINUX="\1 lsm=lockdown,capability,landlock,yama,apparmor,bpf"/' "$GRUB_FILE"
fi
sudo update-grub
echo "BPF LSM 已启用。请重启。"
EOF
chmod +x /tmp/enable_bpf_lsm.sh
sudo /tmp/enable_bpf_lsm.sh

# 安装 Rust
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh

# 安装构建依赖（Ubuntu/Debian）
sudo apt-get install -y clang llvm libelf-dev linux-headers-$(uname -r)

# 安装 Rust 的 BPF 目标
rustup target add bpfel-unknown-none

cd bpfjail

# 构建 BPF 程序
cd bpfjailer-bpf && cargo build --release && cd ..

# 构建守护进程和客户端
cargo build --release

# 以 root 身份运行（如果存在则加载 config/policy.json）
sudo RUST_LOG=info ./target/release/bpfjailer-daemon

[INFO] BpfJailer daemon starting...
[INFO] Loading BpfJailer eBPF programs with libbpf-rs...
[INFO] ✓ pending_enrollments map available for enrollment
[INFO] ✓ network_rules map available for port/protocol filtering
[INFO] ✓ task_storage map created successfully
[INFO] ✓ Program task_alloc attached
[INFO] ✓ Program file_open attached
[INFO] ✓ Program socket_bind attached
[INFO] ✓ Program socket_connect attached
[INFO] ✓ Program bprm_check_security attached
[INFO] Initialized with default roles: restricted (1), permissive (2)
[INFO] Loaded policy from config/policy.json
[INFO] Loaded 5 roles
[INFO] Enrollment server listening on /run/bpfjailer/enrollment.sock

# 在没有 jail 的情况下运行漏洞测试（显示攻击成功）
sudo python3 tests/vulnerable_apps/run_tests.py

# 使用受限角色运行漏洞测试（显示攻击被阻止）
sudo python3 tests/vulnerable_apps/run_tests.py --role 1

# 运行特定测试
sudo python3 tests/vulnerable_apps/run_tests.py --role 1 --test path

# 列出可用测试和角色
sudo python3 tests/vulnerable_apps/run_tests.py --list

============================================================
TEST: Path Traversal / Arbitrary File Read
============================================================
Attempting to read /etc/passwd via path traversal...
BLOCKED - Permission denied (BpfJailer blocked file access)

============================================================
TEST: Command Injection
============================================================
Attempting command injection...
BLOCKED - Permission denied (BpfJailer blocked exec)

============================================================
TEST: Reverse Shell / Data Exfiltration
============================================================
Test 1: Reverse shell connection to 127.0.0.1:4444
BLOCKED - Permission denied (BpfJailer blocked connect)

#!/usr/bin/env python3
import socket
import json
import os

# 连接到守护进程
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
sock.connect("/run/bpfjailer/enrollment.sock")

# 使用受限角色（ID 1）注册
request = {"Enroll": {"pod_id": 1, "role_id": 1}}
sock.send((json.dumps(request) + "\n").encode())
response = sock.recv(4096).decode()
print(f"Enrollment: {response}")
sock.close()

# 尝试读取文件（应被阻止）
try:
    open("/etc/passwd").read()
    print("File access: ALLOWED")
except PermissionError:
    print("File access: BLOCKED")

{
  "roles": {
    "restricted": {
      "id": 1,
      "name": "restricted",
      "flags": {
        "allow_file_access": false,
        "allow_network": false,
        "allow_exec": false
      },
      "network_rules": []
    },
    "webserver": {
      "id": 3,
      "name": "webserver",
      "flags": {
        "allow_file_access": true,
        "allow_network": true,
        "allow_exec": false
      },
      "network_rules": [
        {"protocol": "tcp", "port": 80, "allow": true},
        {"protocol": "tcp", "port": 443, "allow": true}
      ]
    }
  },
  "pods": []
}

network_rules 映射：
  键：{ role_id, port, protocol, direction }
  值：允许 (1) 或拒绝 (0)

// 在守护进程代码中：
use bpfjailer_daemon::process_tracker::{PROTO_TCP, DIR_CONNECT};

// 仅允许 TCP 连接到端口 80 和 443
process_tracker.add_network_rule(RoleId(1), 80, PROTO_TCP, DIR_CONNECT, true)?;
process_tracker.add_network_rule(RoleId(1), 443, PROTO_TCP, DIR_CONNECT, true)?;

{
  "network_rules": [
    {"protocol": "tcp", "port": 443, "allow": true},
    {"protocol": "tcp", "port_start": 8000, "port_end": 8100, "allow": true}
  ]
}

┌─────────────────────────────────────────────────────────────┐
│                     用户空间                                │
├─────────────────────────────────────────────────────────────┤
│  ┌─────────────┐    ┌──────────────────┐                    │
│  │   客户端     │───▶│  bpfjailer-daemon │                    │
│  │  (注册)     │    │                  │                    │
│  └─────────────┘    │  - PolicyManager │                    │
│                     │  - ProcessTracker│                    │
│                     │  - EnrollmentSvr │                    │
│                     └────────┬─────────┘                    │
│                              │ 写入                         │
│                              ▼                              │
├─────────────────────────────────────────────────────────────┤
│                     BPF 映射                                │
│  ┌──────────────────┐  ┌─────────────┐  ┌──────────────┐   │
│  │pending_enrollments│  │ role_flags  │  │ task_storage │   │
│  │   (PID → 信息)   │  │ (角色→标志) │  │(任务→信息)   │   │
│  └──────────────────┘  └─────────────┘  └──────────────┘   │
├─────────────────────────────────────────────────────────────┤
│                     BPF LSM 钩子                             │
│  ┌────────────┐ ┌───────────┐ ┌─────────────┐ ┌──────────┐ │
│  │ task_alloc │ │ file_open │ │socket_bind/ │ │bprm_check│ │
│  │(继承 + 初始│ │(迁移 + 检 │ │  connect    │ │_security │ │
│  │化)         │ │查)        │ │ (检查)      │ │ (检查)   │ │
│  └────────────┘ └───────────┘ └─────────────┘ └──────────┘ │
└─────────────────────────────────────────────────────────────┘

# 安装 systemd 服务
sudo cp config/bpfjailer-daemon.service /etc/systemd/system/
sudo cp target/release/bpfjailer-daemon /usr/sbin/
sudo mkdir -p /etc/bpfjailer
sudo cp config/policy.json /etc/bpfjailer/

# 启用并启动
sudo systemctl daemon-reload
sudo systemctl enable bpfjailer-daemon
sudo systemctl start bpfjailer-daemon

# 安装引导服务
sudo cp config/bpfjailer-bootstrap.service /etc/systemd/system/
sudo cp target/release/bpfjailer-bootstrap /usr/sbin/
sudo mkdir -p /etc/bpfjailer
sudo cp config/policy.json /etc/bpfjailer/

# 启用（将在下次启动时运行）
sudo systemctl daemon-reload
sudo systemctl enable bpfjailer-bootstrap

# 或者立即手动运行
sudo bpfjailer-bootstrap

ls -la /sys/fs/bpf/bpfjailer/
ls -la /sys/fs/bpf/bpfjailer/maps/
ls -la /sys/fs/bpf/bpfjailer/progs/

# 事件被发送到 perf 缓冲区，由 journald 接收
journalctl -f | grep bpfjailer

# 检查内核版本（需要 5.11+）
uname -r

# 检查 BPF LSM 是否激活
cat /sys/kernel/security/lsm | grep bpf

# 检查 BTF 是否可用
ls -la /sys/kernel/btf/vmlinux

# 检查 BPF 程序是否可以加载
sudo bpftool prog list

# 检查能力
sudo capsh --print | grep cap_bpf

# 确保守护进程以 root 身份运行
ps aux | grep bpfjailer

# 检查 socket 权限
ls -la /run/bpfjailer/enrollment.sock

{
  "exec_enrollments": [
    {
      "executable_path": "/usr/bin/nginx",
      "pod_id": 1000,
      "role": "webserver"
    }
  ]
}

{
  "cgroup_enrollments": [
    {
      "cgroup_path": "/sys/fs/cgroup/bpfjailer/sandbox",
      "pod_id": 2000,
      "role": "sandbox"
    }
  ]
}

# 创建 cgroup
sudo mkdir -p /sys/fs/cgroup/bpfjailer/sandbox

# 将进程移动到 cgroup
echo $$ | sudo tee /sys/fs/cgroup/bpfjailer/sandbox/cgroup.procs

# 进程现在自动注册为 sandbox 角色

# 设置注册 xattr
sudo setfattr -n user.bpfjailer.pod_id -v $(printf '\x01\x00\x00\x00\x00\x00\x00\x00') /path/to/binary
sudo setfattr -n user.bpfjailer.role_id -v $(printf '\x03\x00\x00\x00') /path/to/binary

# 检查 xattr
getfattr -d /path/to/binary