Add custom port support for --mcp parameter

- Change --mcp from boolean to string flag
- Support formats: :8080, http://localhost:8080
- Falls back to config.yaml if not specified
- Update all IsMCPServer references to MCPServer string check

Co-authored-by: keac <[email protected]>

Author: Copilot

.gitignore

@@ -4,6 +4,7 @@
 *.dll
 *.so
 *.dylib
+enscan
 
 .idea
 # Test binary, built with `go test -c`
@@ -12,6 +13,7 @@
 # Output of the go coverage tool, specifically when used with LiteIDE
 *.out
 config.yaml
+*.gob
 /res
 /out
 /Report

common/config.go

@@ -46,8 +46,8 @@ type ENOptions struct {
 	IsNoMerge        bool   //聚合
 	OutPutType       string // 导出文件类型
 	IsApiMode        bool
-	IsMCPServer      bool
-	IsPlugins        bool // 是否作为后置插件查询
+	MCPServer        string // MCP模式运行地址，如 :8080 或 http://localhost:8080
+	IsPlugins        bool   // 是否作为后置插件查询
 	IsFast           bool // 是否快速查询
 	ENConfig         *ENConfig
 	BranchFilter     string

common/flag.go

@@ -49,7 +49,7 @@ func Flag(Info *ENOptions) {
 	//其他设定
 	flag.BoolVar(&Info.IsGroup, "is-group", false, "查询关键词为集团")
 	flag.BoolVar(&Info.IsApiMode, "api", false, "API模式运行")
-	flag.BoolVar(&Info.IsMCPServer, "mcp", false, "MCP模式运行")
+	flag.StringVar(&Info.MCPServer, "mcp", "", "MCP模式运行，可指定监听地址，如 :8080 或 http://localhost:8080")
 	flag.BoolVar(&Info.ISKeyPid, "is-pid", false, "批量查询文件是否为公司PID")
 	flag.IntVar(&Info.DelayTime, "delay", 0, "每个请求延迟（S）-1为随机延迟1-5S")
 	flag.StringVar(&Info.Proxy, "proxy", "", "设置代理")

common/parse.go

@@ -54,7 +54,7 @@ func (op *ENOptions) Parse() {
 		gologger.Fatal().Msgf("配置文件当前[V%.1f] 程序需要[V%.1f] 不匹配，请备份配置文件重新运行-v\n", conf.Version, cfgYV)
 	}
 
-	if op.KeyWord == "" && op.CompanyID == "" && op.InputFile == "" && !op.IsApiMode && !op.IsMCPServer {
+	if op.KeyWord == "" && op.CompanyID == "" && op.InputFile == "" && !op.IsApiMode && op.MCPServer == "" {
 		flag.PrintDefaults()
 		os.Exit(0)
 	}

runner/mcpServer.go

@@ -180,8 +180,20 @@ func McpServer(options *common.ENOptions) {
 		),
 	), getInfoByKeyword(enTask))
 	enTask.StartENWorkers()
-	sseServer := server.NewSSEServer(s, server.WithBaseURL(options.ENConfig.Api.Mcp))
-	port, err := utils.ExtractPortString(options.ENConfig.Api.Mcp)
+	
+	// 优先使用命令行参数，否则使用配置文件
+	mcpAddr := options.MCPServer
+	if mcpAddr == "" {
+		mcpAddr = options.ENConfig.Api.Mcp
+	}
+	
+	// 如果只提供了端口（如 :8080），则补全为完整URL
+	if len(mcpAddr) > 0 && mcpAddr[0] == ':' {
+		mcpAddr = "http://localhost" + mcpAddr
+	}
+	
+	sseServer := server.NewSSEServer(s, server.WithBaseURL(mcpAddr))
+	port, err := utils.ExtractPortString(mcpAddr)
 	if err != nil {
 		gologger.Error().Msgf("MCP服务启动失败！")
 		gologger.Fatal().Msgf(err.Error())

runner/runner.go

@@ -116,7 +116,7 @@ func RunEnumeration(options *common.ENOptions) {
 
 	} else if options.IsApiMode {
 		api(options)
-	} else if options.IsMCPServer {
+	} else if options.MCPServer != "" {
 		McpServer(options)
 	} else {
 		esjob := NewENTaskQueue(1, options)
@@ -330,7 +330,7 @@ func (q *ESJob) StartENWorkers() {
 					gologger.Error().Msgf("保存缓存失败: %v", err)
 				}
 			case <-quitSig:
-				if !q.op.IsApiMode && !q.op.IsMCPServer {
+				if !q.op.IsApiMode && q.op.MCPServer == "" {
 					gologger.Error().Msgf("任务未完成退出，将自动保存过程文件！")
 					_ = q.OutFileByEnInfo("未完成任务结果")
 				}