-
Notifications
You must be signed in to change notification settings - Fork 51
/
netcls.go
80 lines (66 loc) · 2.08 KB
/
netcls.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package cgnetcls
import (
"fmt"
"io/ioutil"
"os"
"path/filepath"
"strings"
"go.aporeto.io/trireme-lib/common"
"go.uber.org/zap"
)
// receiver definition.
type netCls struct {
markchan chan uint64
ReleaseAgentPath string
TriremePath string
}
var (
basePath = "/sys/fs/cgroup/net_cls"
markval uint64 = Initialmarkval
)
// GetCgroupList geta list of all cgroup names
// TODO: only used in autoport detection, and a bad usage as well
func GetCgroupList() []string {
var cgroupList []string
// iterate over our different base paths from the different cgroup base paths
for _, baseCgroupPath := range []string{common.TriremeCgroupPath, common.TriremeUIDCgroupPath, common.TriremeDockerHostNetwork} {
filelist, err := ioutil.ReadDir(filepath.Join(basePath, baseCgroupPath))
if err == nil {
for _, file := range filelist {
if file.IsDir() {
cgroupList = append(cgroupList, filepath.Join(baseCgroupPath, file.Name()))
}
}
}
}
return cgroupList
}
// ListCgroupProcesses lists the cgroups that trireme has created
// TODO: only used in autoport detection, and a bad usage as well
func ListCgroupProcesses(cgroupname string) ([]string, error) {
_, err := os.Stat(filepath.Join(basePath, cgroupname))
if os.IsNotExist(err) {
return []string{}, fmt.Errorf("cgroup %s does not exist: %s", cgroupname, err)
}
data, err := ioutil.ReadFile(filepath.Join(basePath, cgroupname, "cgroup.procs"))
if err != nil {
return []string{}, fmt.Errorf("cannot read procs file: %s", err)
}
procs := []string{}
for _, line := range strings.Split(string(data), "\n") {
if len(line) > 0 {
procs = append(procs, line)
}
}
return procs, nil
}
// GetAssignedMarkVal -- returns the mark val assigned to the group
// TODO: looks like dead code
func GetAssignedMarkVal(cgroupName string) string {
mark, err := ioutil.ReadFile(filepath.Join(basePath, cgroupName, markFile))
if err != nil || len(mark) < 1 {
zap.L().Error("Unable to read markval for cgroup", zap.String("Cgroup Name", cgroupName), zap.Error(err))
return ""
}
return string(mark[:len(mark)-1])
}