Mcptrace

🚀 bpftrace MCP Server：生成eBPF以跟蹤Linux內核

bpftrace MCP Server是一個輕量級的MCP（模型上下文協議）服務器，它為AI助手提供了訪問bpftrace內核跟蹤功能的途徑，助力開發者更高效地進行Linux內核的跟蹤和分析。

目前該項目採用Rust語言實現，藉助rmcp crate提升了性能和類型安全性。Python版本的實現仍可在git歷史記錄中找到。

✨ 主要特性

• list_probes：列出可用的bpftrace探測點，並支持可選的過濾功能。
• list_helpers：獲取bpftrace輔助函數的相關信息。
• exec_program：執行bpftrace程序，並提供緩衝輸出。
• get_result：異步檢索執行結果。

📦 安裝指南

前提條件

1. 安裝Rust（若尚未安裝）：

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

2. 確保已安裝bpftrace：

sudo apt-get install bpftrace  # Ubuntu/Debian
# 或者
sudo dnf install bpftrace      # Fedora

3. 構建服務器：

cargo build --release

快速設置

使用我們提供的自動化設置腳本：

• Claude Desktop：./setup/setup_claude.sh
• Claude Code：./setup/setup_claude_code.sh

如需詳細的設置說明和手動配置方法，請參閱 setup/SETUP.md。

🚀 快速開始

直接執行

./target/release/bpftrace-mcp-server

通過Cargo執行

cargo run --release

手動配置

若需瞭解Claude Desktop或Claude Code的手動設置說明，請參閱 setup/SETUP.md。

💻 使用示例

基礎用法

列出系統調用探測點

await list_probes(filter="syscalls:*read*")

獲取BPF系統信息

info = await bpf_info()
# 返回系統信息、內核輔助函數、特性、映射類型和探測點類型

高級用法

執行簡單跟蹤

result = await exec_program(
    'tracepoint:syscalls:sys_enter_open { printf("%s\\n", comm); }',
    timeout=10
)
exec_id = result["execution_id"]

獲取結果

output = await get_result(exec_id)
print(output["output"])

⚠️ 安全提示

• 服務器運行bpftrace需要sudo權限。
• 密碼處理：創建一個.env文件並存儲你的sudo密碼：

  echo "BPFTRACE_PASSWD=your_sudo_password" > .env
  

• 替代方法：為bpftrace配置無密碼sudo：

  sudo visudo
  # 添加：your_username ALL=(ALL) NOPASSWD: /usr/bin/bpftrace
  

• 本服務器不進行腳本驗證，需信任AI客戶端生成安全的腳本。
• 資源限制：最大執行時間為60秒，緩衝區最大為10000行。
• 詳細的安全配置請參閱 SECURITY.md。

🔧 技術細節

該Rust服務器採用了以下技術：

• Tokio異步運行時，實現併發操作。
• 子進程管理，用於執行bpftrace程序。
• DashMap，實現線程安全的內存緩衝。
• 自動清理舊緩衝區。
• rmcp crate，用於實現MCP協議。

📚 詳細文檔

• 設置指南 — 詳細的安裝和配置說明。
• Claude Code設置 — 針對Claude Code的特定說明。
• CLAUDE.md — AI助手的開發指南。
• 設計文檔 — 架構和設計細節。

📄 侷限性

• 不支持實時流傳輸（可使用get_result進行輪詢）。
• 密碼處理方式較為簡單（生產環境中需改進）。
• 未對執行記錄進行持久化存儲。
• 錯誤處理功能較為基礎。

📈 未來改進方向

• 增加SSE傳輸，實現實時流傳輸。
• 實現適當的身份驗證機制。
• 增加腳本驗證和沙箱功能。
• 支持保存/加載跟蹤會話。
• 與eBPF程序進行集成。