mirror of https://github.com/status-im/consul.git
97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
// +build freebsd darwin
|
|
|
|
package net
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/shirou/gopsutil/internal/common"
|
|
)
|
|
|
|
// Return a list of network connections opened.
|
|
func Connections(kind string) ([]ConnectionStat, error) {
|
|
return ConnectionsWithContext(context.Background(), kind)
|
|
}
|
|
|
|
func ConnectionsWithContext(ctx context.Context, kind string) ([]ConnectionStat, error) {
|
|
return ConnectionsPid(kind, 0)
|
|
}
|
|
|
|
// Return a list of network connections opened returning at most `max`
|
|
// connections for each running process.
|
|
func ConnectionsMax(kind string, max int) ([]ConnectionStat, error) {
|
|
return ConnectionsMaxWithContext(context.Background(), kind, max)
|
|
}
|
|
|
|
func ConnectionsMaxWithContext(ctx context.Context, kind string, max int) ([]ConnectionStat, error) {
|
|
return []ConnectionStat{}, common.ErrNotImplementedError
|
|
}
|
|
|
|
// Return a list of network connections opened by a process.
|
|
func ConnectionsPid(kind string, pid int32) ([]ConnectionStat, error) {
|
|
return ConnectionsPidWithContext(context.Background(), kind, pid)
|
|
}
|
|
|
|
func ConnectionsPidWithContext(ctx context.Context, kind string, pid int32) ([]ConnectionStat, error) {
|
|
var ret []ConnectionStat
|
|
|
|
args := []string{"-i"}
|
|
switch strings.ToLower(kind) {
|
|
default:
|
|
fallthrough
|
|
case "":
|
|
fallthrough
|
|
case "all":
|
|
fallthrough
|
|
case "inet":
|
|
args = append(args, "tcp", "-i", "udp")
|
|
case "inet4":
|
|
args = append(args, "4")
|
|
case "inet6":
|
|
args = append(args, "6")
|
|
case "tcp":
|
|
args = append(args, "tcp")
|
|
case "tcp4":
|
|
args = append(args, "4tcp")
|
|
case "tcp6":
|
|
args = append(args, "6tcp")
|
|
case "udp":
|
|
args = append(args, "udp")
|
|
case "udp4":
|
|
args = append(args, "6udp")
|
|
case "udp6":
|
|
args = append(args, "6udp")
|
|
case "unix":
|
|
args = []string{"-U"}
|
|
}
|
|
|
|
r, err := common.CallLsofWithContext(ctx, invoke, pid, args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, rr := range r {
|
|
if strings.HasPrefix(rr, "COMMAND") {
|
|
continue
|
|
}
|
|
n, err := parseNetLine(rr)
|
|
if err != nil {
|
|
|
|
continue
|
|
}
|
|
|
|
ret = append(ret, n)
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
// Return up to `max` network connections opened by a process.
|
|
func ConnectionsPidMax(kind string, pid int32, max int) ([]ConnectionStat, error) {
|
|
return ConnectionsPidMaxWithContext(context.Background(), kind, pid, max)
|
|
}
|
|
|
|
func ConnectionsPidMaxWithContext(ctx context.Context, kind string, pid int32, max int) ([]ConnectionStat, error) {
|
|
return []ConnectionStat{}, common.ErrNotImplementedError
|
|
}
|