You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
gopsutil/net/net_netlink_linux.go

122 lines
2.7 KiB
Go

// +build linux
// +build netlink
package net
import (
"context"
"fmt"
"strconv"
"sync"
"syscall"
"github.com/shirou/gopsutil/internal/common"
"github.com/shirou/gopsutil/net/netlink"
)
func connectionsPidMaxWithoutUidsWithContext(ctx context.Context, kind string, pid int32, max int, skipUids bool) ([]ConnectionStat, error) {
tmap, ok := netConnectionKindMap[kind]
if !ok {
return nil, fmt.Errorf("invalid kind, %s", kind)
}
root := common.HostProc()
var err error
var inodes map[string][]inodeMap
if pid == 0 {
inodes, err = getProcInodesAll(root, max)
} else {
inodes, err = getProcInodes(root, pid, max)
}
if len(inodes) == 0 {
// no inode for the pid
return []ConnectionStat{}, nil
}
if err != nil {
return nil, fmt.Errorf("cound not get pid(s), %d: %s", pid, err)
}
return connectionsNetLink(tmap, inodes)
}
func connectionsNetLink(kinds []netConnectionKindType, inodes map[string][]inodeMap) ([]ConnectionStat, error) {
wait := &sync.WaitGroup{}
reply := make([][]ConnectionStat, len(kinds))
var retErrr error
for i, kind := range kinds {
wait.Add(1)
if kind.family == syscall.AF_UNIX {
go getUnixConnections(wait, i, reply)
} else {
go getInetConnections(wait, i, kind, reply)
}
}
wait.Wait()
ret := make([]ConnectionStat, 0)
for i := range kinds {
for _, cs := range reply[i] {
iss, exists := inodes[strconv.Itoa(int(cs.Fd))]
if exists {
for _, is := range iss {
cs.Pid = is.pid
cs.Fd = is.fd
}
} else {
cs.Fd = 0
}
ret = append(ret, cs)
}
}
return ret, retErrr
}
func getInetConnections(wait *sync.WaitGroup, i int, k netConnectionKindType, reply [][]ConnectionStat) {
defer wait.Done()
msgs, err := netlink.InetConnections(k.family, k.netlinkProto)
if err != nil {
// just ignored
return
}
t := make([]ConnectionStat, len(msgs))
for i, msg := range msgs {
conn := ConnectionStat{
Family: uint32(msg.Family),
Type: uint32(k.sockType),
Laddr: Addr{msg.SrcIP().String(), uint32(msg.SrcPort())},
Raddr: Addr{msg.DstIP().String(), uint32(msg.DstPort())},
Status: netlink.TCPState(msg.State).String(),
Uids: []int32{int32(msg.UID)},
// Pid: msg.Pid,
Fd: msg.Inode,
}
t[i] = conn
}
reply[i] = t
}
func getUnixConnections(wait *sync.WaitGroup, i int, reply [][]ConnectionStat) {
defer wait.Done()
msgs, err := netlink.UnixConnections()
if err != nil {
// just ignored
return
}
t := make([]ConnectionStat, len(msgs))
for i, msg := range msgs {
conn := ConnectionStat{
Family: uint32(msg.Family),
Status: netlink.TCPState(msg.State).String(),
Uids: []int32{},
Laddr: Addr{
IP: msg.Path,
},
Fd: msg.Inode,
}
t[i] = conn
}
reply[i] = t
}