tetragon

1
// SPDX-License-Identifier: Apache-2.0
2
// Copyright Authors of Tetragon
3

4
package test
5

6
import (
7
	"github.com/cilium/tetragon/pkg/sensors"
8
	"github.com/cilium/tetragon/pkg/sensors/program"
9
)
10

11
var (
12
	CgroupMkdir = program.Builder(
13
		"bpf_cgroup_mkdir.o",
14
		"cgroup/cgroup_mkdir",
15
		"raw_tracepoint/cgroup_mkdir",
16
		"tg_tp_cgrp_mkdir",
17
		"raw_tracepoint",
18
	)
19

20
	CgroupRmdir = program.Builder(
21
		"bpf_cgroup_rmdir.o",
22
		"cgroup/cgroup_rmdir",
23
		"raw_tracepoint/cgroup_rmdir",
24
		"tg_tp_cgrp_rmdir",
25
		"raw_tracepoint",
26
	)
27

28
	CgroupRelease = program.Builder(
29
		"bpf_cgroup_release.o",
30
		"cgroup/cgroup_release",
31
		"raw_tracepoint/cgroup_release",
32
		"tg_tp_cgrp_release",
33
		"raw_tracepoint",
34
	)
35

36
	/* Cgroup tracking maps */
37
	CgroupsTrackingMap = program.MapBuilder("tg_cgrps_tracking_map", CgroupMkdir)
38
)
39

40
func GetCgroupsTrackingMap() *program.Map {
41
	return CgroupsTrackingMap
42
}
43

44
func getCgroupPrograms() []*program.Program {
45
	progs := []*program.Program{
46
		CgroupMkdir,
47
		CgroupRmdir,
48
		CgroupRelease,
49
	}
50
	return progs
51
}
52

53
func getCgroupMaps() []*program.Map {
54
	maps := []*program.Map{
55
		GetCgroupsTrackingMap(),
56
	}
57
	return maps
58
}
59

60
// GetCgroupSensor returns the Cgroups base sensor
61
func GetCgroupSensor() *sensors.Sensor {
62
	return &sensors.Sensor{
63
		Name:  "test-sensor-cgroups",
64
		Progs: getCgroupPrograms(),
65
		Maps:  getCgroupMaps(),
66
	}
67
}
68