mirror of
https://github.com/shadow1ng/fscan.git
synced 2025-07-13 21:02:44 +08:00
59 lines
1.7 KiB
Go
59 lines
1.7 KiB
Go
package server
|
||
|
||
import (
|
||
"fmt"
|
||
|
||
"github.com/mark3labs/mcp-go/mcp"
|
||
"github.com/mark3labs/mcp-go/server"
|
||
"github.com/shadow1ng/fscan/Common"
|
||
"github.com/shadow1ng/fscan/Remote/service"
|
||
)
|
||
|
||
func StartMcpServer(transport string) {
|
||
mcpServer := NewFscanMCPServer()
|
||
|
||
if transport == "sse" {
|
||
Common.LogSuccess("🚀 启动 MCP SSE 服务器，监听地址: http://localhost:8080")
|
||
sseServer := server.NewSSEServer(mcpServer, server.WithBaseURL("http://localhost:8080"))
|
||
if err := sseServer.Start(":8080"); err != nil {
|
||
Common.LogError(fmt.Sprintf("❌ 启动 SSE 服务器失败: %v", err))
|
||
panic(err)
|
||
}
|
||
} else {
|
||
Common.LogSuccess("🚀 启动 MCP Stdio 服务器（标准输入输出模式）")
|
||
if err := server.ServeStdio(mcpServer); err != nil {
|
||
Common.LogError(fmt.Sprintf("❌ 启动 Stdio 服务器失败: %v", err))
|
||
panic(err)
|
||
}
|
||
}
|
||
}
|
||
|
||
func NewFscanMCPServer() *server.MCPServer {
|
||
s := server.NewMCPServer(
|
||
"Fscan MCP",
|
||
"1.0.0",
|
||
)
|
||
toolHandler := service.NewFscanMCPTool()
|
||
// 添加提示词
|
||
|
||
// 添加工具处理器
|
||
s.AddTool(
|
||
mcp.NewTool("StartScan",
|
||
mcp.WithDescription("启动端口和服务扫描任务，适用于安全评估或资产排查场景。"),
|
||
mcp.WithString("target",
|
||
mcp.Required(),
|
||
mcp.Description("待扫描的目标地址，支持IP、域名或CIDR格式（如192.168.1.1、example.com、10.0.0.0/24）。"),
|
||
),
|
||
),
|
||
toolHandler.StartScan,
|
||
)
|
||
s.AddTool(
|
||
mcp.NewTool("GetScanResults",
|
||
mcp.WithDescription("获取当前扫描任务的执行进度和已完成部分的结果。若扫描尚未完成，也会返回当前阶段的中间结果，供用户分析或决策使用。"),
|
||
),
|
||
toolHandler.GetScanResults,
|
||
)
|
||
return s
|
||
|
||
}
|