-
Notifications
You must be signed in to change notification settings - Fork 12
/
main.go
56 lines (44 loc) · 1.09 KB
/
main.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
package main
import (
"bytes"
_ "embed"
"github.com/sirupsen/logrus"
manager "github.com/DataDog/ebpf-manager"
)
//go:embed ebpf/bin/main.o
var Probe []byte
var m = &manager.Manager{
Probes: []*manager.Probe{
{
ProbeIdentificationPair: manager.ProbeIdentificationPair{
EBPFFuncName: "sock_filter",
},
},
},
}
func main() {
// Create a socket pair that will be used to trigger the socket filter
sockPair, err := newSocketPair()
if err != nil {
logrus.Fatal(err)
}
// Set the socket file descriptor on which the socket filter should trigger
m.Probes[0].SocketFD = sockPair[0]
// Initialize the manager
if err := m.Init(bytes.NewReader(Probe)); err != nil {
logrus.Fatal(err)
}
// Start the manager
if err := m.Start(); err != nil {
logrus.Fatal(err)
}
logrus.Println("successfully started, head over to /sys/kernel/debug/tracing/trace_pipe")
// Send a message through the socket pair to trigger the probe
if err := trigger(sockPair); err != nil {
logrus.Error(err)
}
// Close the manager
if err := m.Stop(manager.CleanAll); err != nil {
logrus.Fatal(err)
}
}