OMC/be.ems
1
0
Fork 0
Code Issues Releases 3 Activity

fix: 获取网络连接进程参数调整

Browse Source
This commit is contained in:
TsMask
2024-09-06 16:56:35 +08:00
parent d5ff11d0ef
commit 9760e95ee1
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/modules/ws/model/net_connect.go
View File

@@ -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"`
Name string `json:"name"`
PID int32 `json:"pid"`
}

9
src/modules/ws/processor/net_connect.go
View File

@@ -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) {