Function getNetConnections

agent/utils/websocket/process_data.go:323–384

(config NetConfig)

}

func getNetConnections(config NetConfig) (res []byte, err error) {
	result := make([]ProcessConnect, 0, 1024)
	ctx, cancel := context.WithTimeout(context.Background(), defaultTimeout)
	defer cancel()

	connections, err := net.ConnectionsMaxWithContext(ctx, "all", 32768)
	if err != nil {
		res, _ = json.Marshal(result)
		return
	}

	pidConnectionsMap := make(map[int32][]net.ConnectionStat, 256)
	pidNameMap := make(map[int32]string, 256)

	for _, conn := range connections {
		if conn.Family != 2 && conn.Family != 10 {
			continue
		}

		if conn.Pid == 0 {
			continue
		}

		if config.ProcessID > 0 && conn.Pid != config.ProcessID {
			continue
		}

		if config.Port > 0 && conn.Laddr.Port != config.Port && conn.Raddr.Port != config.Port {
			continue
		}

		if _, exists := pidNameMap[conn.Pid]; !exists {
			pName, _ := getProcessNameWithContext(ctx, conn.Pid)
			if pName == "" {
				pName = "<UNKNOWN>"
			}
			pidNameMap[conn.Pid] = pName
		}

		pidConnectionsMap[conn.Pid] = append(pidConnectionsMap[conn.Pid], conn)
	}

	for pid, connections := range pidConnectionsMap {
		pName := pidNameMap[pid]
		if config.ProcessName != "" && !strings.Contains(pName, config.ProcessName) {
			continue
		}
		for _, conn := range connections {
			result = append(result, ProcessConnect{
				Type:   getConnectionType(conn.Type, conn.Family),
				Status: conn.Status,
				Laddr:  conn.Laddr,
				Raddr:  conn.Raddr,
				PID:    conn.Pid,
				Name:   pName,
			})
		}
	}