From 9760e95ee1202e51b5f2e304bac1aeef5d870b7f Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Fri, 6 Sep 2024 16:56:35 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E8=8E=B7=E5=8F=96=E7=BD=91=E7=BB=9C?=
 =?UTF-8?q?=E8=BF=9E=E6=8E=A5=E8=BF=9B=E7=A8=8B=E5=8F=82=E6=95=B0=E8=B0=83?=
 =?UTF-8?q?=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/ws/model/net_connect.go     | 8 ++++----
 src/modules/ws/processor/net_connect.go | 9 ++++++---
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/modules/ws/model/net_connect.go b/src/modules/ws/model/net_connect.go
index 945ede5d..6b67edb6 100644
--- a/src/modules/ws/model/net_connect.go
+++ b/src/modules/ws/model/net_connect.go
@@ -8,13 +8,13 @@ type NetConnectData struct {
 	Status string   `json:"status"`
 	Laddr  net.Addr `json:"localaddr"`
 	Raddr  net.Addr `json:"remoteaddr"`
-	PID    int32    `json:"PID"`
+	PID    int32    `json:"pid"`
 	Name   string   `json:"name"`
 }
 
 // NetConnectQuery 网络连接进程查询
 type NetConnectQuery struct {
-	Port        int32  `json:"port"`
-	ProcessName string `json:"processName"`
-	ProcessID   int32  `json:"processID"`
+	Port int32  `json:"port"`
+	Name string `json:"name"`
+	PID  int32  `json:"pid"`
 }
diff --git a/src/modules/ws/processor/net_connect.go b/src/modules/ws/processor/net_connect.go
index 187de88b..5e242435 100644
--- a/src/modules/ws/processor/net_connect.go
+++ b/src/modules/ws/processor/net_connect.go
@@ -29,13 +29,16 @@ func GetNetConnections(requestID string, data any) ([]byte, error) {
 			continue
 		}
 		for _, conn := range connections {
-			if query.ProcessID > 0 && query.ProcessID != conn.Pid {
+			if query.PID > 0 && query.PID != conn.Pid {
 				continue
 			}
 			proc, err := process.NewProcess(conn.Pid)
 			if err == nil {
-				name, _ := proc.Name()
-				if name != "" && query.ProcessName != "" && !strings.Contains(name, query.ProcessName) {
+				name, err := proc.Name()
+				if err != nil {
+					continue
+				}
+				if query.Name != "" && !strings.Contains(name, query.Name) {
 					continue
 				}
 				if query.Port > 0 && query.Port != int32(conn.Laddr.Port) && query.Port != int32(conn.Raddr.Port) {