123 lines
4.4 KiB
PowerShell
123 lines
4.4 KiB
PowerShell
#Requires -Version 5.1
|
||
<#
|
||
Seep 工作台 —— IDA Pro 配置
|
||
----------------------------------------------------------------
|
||
IDA Pro 是商业软件，不随包分发。本脚本：
|
||
1. 探测本机 IDA 安装位置
|
||
2. 把 ida-pro-mcp 装进 IDA 自带的 Python
|
||
3. 把 IDA 路径写入 ~/.pi/agent/mcp.json
|
||
未装 IDA 时：不报错，仅提示，并说明免费替代路线。
|
||
#>
|
||
[CmdletBinding()]
|
||
param(
|
||
[string]$IdaRoot = '', # 手工指定 IDA 安装目录
|
||
[switch]$SkipPip # 只探测+写配置，不装 ida-pro-mcp
|
||
)
|
||
|
||
$ErrorActionPreference = 'Continue'
|
||
$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
|
||
$Root = Split-Path -Parent $ScriptDir
|
||
$AgentDir = Join-Path $env:USERPROFILE '.pi\agent'
|
||
$McpJson = Join-Path $AgentDir 'mcp.json'
|
||
|
||
function Write-Ok($m) { Write-Host " [OK] $m" -ForegroundColor Green }
|
||
function Write-Warn($m) { Write-Host " [!!] $m" -ForegroundColor Yellow }
|
||
function Write-Info($m) { Write-Host " [..] $m" -ForegroundColor Gray }
|
||
|
||
Write-Host "`n === IDA Pro 配置 ===" -ForegroundColor Cyan
|
||
|
||
# ---------------------------------------------------------------- 1. 探测
|
||
$candidates = @()
|
||
if ($IdaRoot -ne '') { $candidates += $IdaRoot }
|
||
$candidates += @(
|
||
'D:\Tool\IDA Pro',
|
||
'C:\Program Files\IDA Pro',
|
||
'C:\Program Files\IDA Professional 9.0',
|
||
'C:\Program Files\IDA Free',
|
||
'C:\IDA Pro',
|
||
'C:\IDA'
|
||
)
|
||
|
||
$found = $null
|
||
foreach ($c in $candidates) {
|
||
if (Test-Path (Join-Path $c 'ida.exe')) { $found = $c; break }
|
||
}
|
||
|
||
if (-not $found) {
|
||
Write-Warn '未探测到 IDA Pro 安装'
|
||
Write-Host @'
|
||
|
||
说明：IDA Pro 为商业软件，本包不包含，需你自备授权。
|
||
· 购买/试用：https://hex-rays.com/ida-pro
|
||
· 免费替代：见 MANUAL\IDA-PRO.md（seep MCP 的 radare2 八件套已覆盖大部分场景）
|
||
· 已装但未探测到？手工指定：
|
||
powershell -File .\install-ida.ps1 -IdaRoot "你的IDA目录"
|
||
|
||
mcp.json 中 ida 条目将保留占位符，不影响 seep MCP 使用。
|
||
'@ -ForegroundColor Gray
|
||
exit 0
|
||
}
|
||
|
||
Write-Ok "探测到 IDA: $found"
|
||
|
||
# ---------------------------------------------------------------- 2. 定位 Python
|
||
$idaPy = $null
|
||
foreach ($sub in @('python311\python.exe', 'python3\python.exe', 'python\python.exe')) {
|
||
$p = Join-Path $found $sub
|
||
if (Test-Path $p) { $idaPy = $p; break }
|
||
}
|
||
if (-not $idaPy) {
|
||
# 兜底：全局 python
|
||
$g = Get-Command python -ErrorAction SilentlyContinue
|
||
if ($g) { $idaPy = $g.Source; Write-Warn "IDA 内置 Python 未找到，改用全局 python: $idaPy" }
|
||
}
|
||
if ($idaPy) { Write-Ok "IDA Python: $idaPy" } else { Write-Warn '未找到可用的 Python' }
|
||
|
||
# ---------------------------------------------------------------- 3. 装 ida-pro-mcp
|
||
if (-not $SkipPip -and $idaPy) {
|
||
Write-Info 'pip install ida-pro-mcp ...'
|
||
& $idaPy -m pip install --quiet --upgrade ida-pro-mcp 2>&1 | Out-Null
|
||
if ($LASTEXITCODE -eq 0) { Write-Ok 'ida-pro-mcp 已装入 IDA Python' }
|
||
else { Write-Warn 'ida-pro-mcp 安装失败（可手动：pip install ida-pro-mcp）' }
|
||
}
|
||
|
||
# ---------------------------------------------------------------- 4. 写 mcp.json
|
||
$serverPy = Join-Path $found 'python311\Lib\site-packages\ida_pro_mcp\server.py'
|
||
if (-not (Test-Path $serverPy)) {
|
||
# 尝试常见路径
|
||
foreach ($sp in @(
|
||
(Join-Path $found 'python311\Lib\site-packages\ida_pro_mcp\server.py'),
|
||
(Join-Path $found 'Lib\site-packages\ida_pro_mcp\server.py')
|
||
)) {
|
||
if (Test-Path $sp) { $serverPy = $sp; break }
|
||
}
|
||
}
|
||
|
||
if (-not (Test-Path $McpJson)) {
|
||
Write-Warn "mcp.json 不存在，请先运行 install-pi.ps1"
|
||
exit 0
|
||
}
|
||
|
||
$raw = Get-Content $McpJson -Raw -Encoding UTF8
|
||
$changed = $false
|
||
|
||
if ($raw -match '<IDA_ROOT>') { $raw = $raw.Replace('<IDA_ROOT>', $found); $changed = $true }
|
||
if ($raw -match '<IDA_PYTHON>') { $raw = $raw.Replace('<IDA_PYTHON>', $idaPy); $changed = $true }
|
||
|
||
if ($changed) {
|
||
Set-Content -Path $McpJson -Value $raw -Encoding UTF8
|
||
Write-Ok "mcp.json 已更新 IDA 路径"
|
||
Write-Info " IDA_ROOT = $found"
|
||
Write-Info " IDA_PYTHON = $idaPy"
|
||
} else {
|
||
Write-Ok 'mcp.json 中 IDA 路径已配置（无占位符）'
|
||
}
|
||
|
||
if (-not (Test-Path $serverPy)) {
|
||
Write-Warn "未找到 ida_pro_mcp\server.py（预期: $serverPy）"
|
||
Write-Info '请确认已执行 pip install ida-pro-mcp'
|
||
}
|
||
|
||
Write-Host "`n 提示：重启 pi 使 IDA MCP 生效" -ForegroundColor Yellow
|
||
exit 0
|