MCP 服务器

code-audit-mcp

Enables automated security code auditing using LLM and MCP, including AST parsing, taint analysis, dataflow tracing, and automated PoC generation for multi-language codebases.

README

Code Audit MCP Server

基于大语言模型（LLM）与模型上下文协议（MCP）的自动化代码审计工具。

功能特性

代码语义解析：AST解析、类型推导、污点标记
调用链追踪：跨文件/跨模块的数据流追踪
漏洞验证：自动生成PoC并执行验证
多语言支持：PHP、Python、Java、JavaScript/TypeScript

快速开始

环境要求

Node.js >= 18.0
TypeScript >= 5.0
Cursor / Windsurf（支持MCP的IDE）

安装

git clone https://github.com/kkk1259/code-audit-mcp.git
cd code-audit-mcp
npm install
npm run build

配置

在Cursor中配置MCP Server：

// .cursor/mcp.json
{
  "mcpServers": {
    "code-audit": {
      "command": "node",
      "args": ["./dist/server.js"],
      "env": {
        "PROJECT_ROOT": "/path/to/your/project"
      }
    }
  }
}

使用示例

SQL注入审计

用户: 请审计这个项目的SQL注入漏洞

AI: 我将使用MCP工具进行系统性审计。
    首先，识别所有用户输入源和SQL执行点...

反序列化漏洞挖掘

用户: 检查是否存在反序列化漏洞

AI: 我将分析项目中的unserialize调用，并追踪POP链...

工具列表

工具名	功能	参数
`parse_ast`	解析源码AST	`file`: 文件路径
`find_callers`	查找函数调用者	`funcName`: 函数名, `maxDepth`: 最大深度
`trace_dataflow`	追踪数据流	`source`: 源, `sink`: 汇
`generate_poc`	生成漏洞PoC	`vulnType`: 漏洞类型, `params`: 参数

项目结构

code-audit-mcp/
├── src/
│   ├── server.ts          # MCP Server入口
│   ├── tools/
│   │   ├── parser.ts      # AST解析
│   │   ├── callgraph.ts   # 调用链追踪
│   │   ├── verifier.ts    # 漏洞验证
│   │   └── poc.ts         # PoC生成
│   ├── rules/
│   │   ├── php.json       # PHP安全规则
│   │   ├── python.json    # Python安全规则
│   │   └── java.json      # Java安全规则
│   └── utils/
│       ├── taint.ts       # 污点分析
│       └── sanitizer.ts   # 过滤函数识别
├── prompts/
│   ├── audit-system.md    # 系统Prompt
│   └── vuln-types/        # 各漏洞类型Prompt
├── tests/
├── package.json
└── README.md