From 0ceebd99022c81551242984753e0e88aa8fdf720 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= <iwind.liu@gmail.com>
Date: Tue, 30 Nov 2021 15:03:46 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AB=AF=E5=8F=A3=E6=8F=90=E7=A4=BA=E8=A2=AB?=
 =?UTF-8?q?=E5=8D=A0=E7=94=A8=E6=97=B6=E6=8F=90=E7=A4=BA=E8=AF=AD=E4=B8=AD?=
 =?UTF-8?q?=E5=8A=A0=E5=85=A5=E5=BD=93=E5=89=8D=E5=8D=A0=E7=94=A8=E7=AB=AF?=
 =?UTF-8?q?=E5=8F=A3=E7=9A=84=E8=BF=9B=E7=A8=8B=E5=90=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/nodes/listener_manager.go | 47 ++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/internal/nodes/listener_manager.go b/internal/nodes/listener_manager.go
index abfa7d0..319581d 100644
--- a/internal/nodes/listener_manager.go
+++ b/internal/nodes/listener_manager.go
@@ -1,13 +1,18 @@
 package nodes
 
 import (
+	"bytes"
+	"errors"
 	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
 	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
 	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/lists"
 	"net/url"
+	"os/exec"
 	"regexp"
+	"runtime"
+	"strings"
 	"sync"
 	"time"
 )
@@ -113,10 +118,25 @@ func (this *ListenerManager) Start(node *nodeconfigs.NodeConfig) error {
 				if firstServer == nil {
 					remotelogs.Error("LISTENER_MANAGER", err.Error())
 				} else {
+					// 当前占用的进程名
+					if strings.Contains(err.Error(), "in use") {
+						portIndex := strings.LastIndex(addr, ":")
+						if portIndex > 0 {
+							var port = addr[portIndex+1:]
+							var processName = this.findProcessNameWithPort(port)
+							if len(processName) > 0 {
+								err = errors.New(err.Error() + " (the process using port: '" + processName + "')")
+							}
+						}
+					}
+
 					remotelogs.ServerError(firstServer.Id, "LISTENER_MANAGER", err.Error())
 				}
 
 				continue
+			} else {
+				// TODO 是否是从错误中恢复
+
 			}
 			this.listenersMap[addr] = listener
 		}
@@ -160,6 +180,33 @@ func (this *ListenerManager) retryListeners() {
 			delete(this.retryListenerMap, addr)
 			this.listenersMap[addr] = listener
 			remotelogs.ServerSuccess(listener.group.FirstServer().Id, "LISTENER_MANAGER", "retry to listen '"+addr+"' successfully")
+
+			// TODO 删除失败记录
 		}
 	}
 }
+
+func (this *ListenerManager) findProcessNameWithPort(port string) string {
+	if runtime.GOOS != "linux" {
+		return ""
+	}
+
+	path, err := exec.LookPath("ss")
+	if err != nil {
+		return ""
+	}
+
+	var cmd = exec.Command(path, "-tlpn", "sport = :"+port)
+	var output = &bytes.Buffer{}
+	cmd.Stdout = output
+	err = cmd.Run()
+	if err != nil {
+		return ""
+	}
+
+	var matches = regexp.MustCompile(`(?U)\(\("(.+)",pid=\d+,fd=\d+\)\)`).FindStringSubmatch(output.String())
+	if len(matches) > 1 {
+		return matches[1]
+	}
+	return ""
+}