# 現在のカーネル設定を確認
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

# 隔離なしで脆弱性テストを実行（攻撃が成功することを確認）
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"登録: {response}")
sock.close()

# ファイルを読み取ろうとする（ブロックされるはず）
try:
    open("/etc/passwd").read()
    print("ファイルアクセス: 許可")
except PermissionError:
    print("ファイルアクセス: ブロック")

{
  "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 }
  値: allowed (1) または denied (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 → info)   │  │ (role→flags)│  │(task→info)   │   │
│  └──────────────────┘  └─────────────┘  └──────────────┘   │
├─────────────────────────────────────────────────────────────┤
│                     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

# ソケットのパーミッション確認
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