长亭百川云 - 技术讨论

MCP Server 牛刀小试之雷池 MCP

发布于 2 天前

# 雷池 WAF
# 雷池技术博客投稿
技术资料

发布于 2 天前

头像

熊大,光头强又来砍树了

更新于 2 天前

1

0

本文作者:小谈谈

什么是 MCP

MCP(Model Context Protocol)可以算是当下 AI 编程圈子里最火爆的话题之一了。实际上 MCP 早在 2024 年 11 月就已经由 Anthropic 提出,目前我们广泛接触到的 MCP 规范是 2025-03-26 版本

MCP 是一种用于标准化 AI 模型与本地和远程资源进行交互的标准协议。MCP 就像是 AI 应用程序的 USB-C 接口,为 AI 模型提供了一种标准化的方式来连接不同的数据源和工具。

Figure_1

MCP 采用的是 C/S 架构,一个 MCP Host 可以连接到多个 MCP 服务器,以扩展 AI 的能力。

Figure_3

如上图,在 MCP 中,有这样几个角色。

  • MCP Hosts: 像 Claude Desktop、IDEs 或 AI 工具这样的程序,它们希望通过 MCP 访问资源
  • MCP Clients: 维护与服务器 1:1 连接的协议客户端
  • MCP Servers: 轻量级程序,通过标准化的 Model Context Protocol 暴露特定功能
  • Local Resources: 你的计算机资源(数据库、文件、服务),MCP 服务器可以安全地访问这些资源
  • Remote Resources: 通过互联网可用的资源(例如,通过 APIs),MCP 服务器可以连接到这些资源

MCP 为 LLM 提供的能力不仅仅是工具,但目前大家重点关注到的是 Tool 能力。今天,我们要做的就是为雷池 WAF 创建一个 MCP Server,以实现通过 AI 对 WAF 进行自动化管理。

牛刀小试

雷池自 6.x 开始,对用户公开了 API 能力,用户可以通过管理面创建 API Token,这为我们实践 MCP Server 创造了一个有利条件。如下图,我们创建一个 API Token,后面 MCP Server 中需要进行对应的配置。

image.png

对于 MCP 协议,官方提供了 Python、TypeScript、Java 等语言的 SDK,社区提供了 Golang 版本的 SDK。我们使用的是 github.com/mark3labs/mcp-go 这个 SDK。

功能实现

项目的组织结构比较简单,main.go 创建了一个 SSE 形式的 MCP SERVER。在 utils 中,定义了和雷池 API 通信的协议,还有一些辅助函数。

具体的 MCP 工具定义在 tools 中。

目前已经实现的能力:

  • 创建受保护的应用
  • 获取证书
  • 获取攻击事件,获取攻击记录
  • 攻击事件统计
  • WAF 防护网站响应码统计
  • 常用时间计算

项目地址:https://cnb.cool/hex/go-mcp-safeline

开始使用

编辑 .env 启动服务。

TRANSPORT=sse
MCPS_ADDR=http://127.0.0.1:8099
SAFELINE_APISERVER=雷池管理端地址,例如 https://1.1.1.1:9443
SAFELINE_APITOKEN=雷池管理端 Token
DEBUG=true

在 Cursor 中使用。

{
    "mcpServers": {
      "mcp-safeline": {
        "url": "http://127.0.0.1:8099/sse",
        "env": {
        }
      }
    }
  }

演示

我做了两个示例,演示通过 MCP 协议来对 WAF 进行自动化管理。

完整的视频演示:https://www.bilibili.com/video/BV1aDZWYuEBu/

问题 1:今天是三月最后一天了,帮我看看近一个月,waf 都拦截了哪些攻击事件?

LLM 分析了用户的诉求,自主调用工具计算当前时间和一个月前是什么时候,获取了所有的攻击记录,进行汇总并返回给用户。

image.png

问题 2:在 waf 上配置了哪些证书?

LLM 通过相关工具获取了 WAF 上的证书列表和证书信息,并返回给了用户。

image.png

后记

在实现 MCP Server 的过程中,这个开发的工作量还是比较大的,我们需要了解 API,做好工具规划,给每个变量和函数写好描述信息。甚至给 MCP 写描述,比项目中写注释还累。在我的理解下,MCP Server 编程更像是给 LLM 讲述工具怎么用的过程,你既要给够 LLM 工具,还要给他讲明白工具使用背景和使用方法。

image.png

虽然 MCP 协议让 LLM 和工具解耦,开发工具时不用关心具体应用在什么 LLM 上,但“好刀更要有一位好厨子”。LLM 作为大脑,不仅需要强大的推理能力,还需要能够根据具体场景选择和规划调用 MCP Server,从而高效调度工具,最大化发挥工具的能力。

合理的工具调用策略,结合 LLM 的智能调度,才能真正提升应用的智能化水平,使其具备更强的泛化能力和实用性。

使用 邀请链接 或 推荐码: GVWYWCFHJ6MOHASCEPDTNJV37CSEAU7V 购买雷池专业版、商业版 WAF,享受专享优惠。

扩展

头像

齐天大圣孙悟空

管理员

更新于 2 天前

0

0

牛逼,这两天我们自己也在做雷池的 mcp server

头像
关注或联系我们
添加百川云公众号,移动管理云安全产品
咨询热线:
4000-327-707
百川公众号
百川公众号
百川云客服
百川云客服

Copyright ©2024 北京长亭科技有限公司
icon
京ICP备2024055124号-2