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
|
package main
import (
"fmt"
"os"
"os/signal"
"syscall"
"github.com/evilsocket/ftrace"
)
func setupSignals(cb func()) {
sigChan := make(chan os.Signal, 1)
signal.Notify(sigChan,
syscall.SIGHUP,
syscall.SIGINT,
syscall.SIGTERM,
syscall.SIGQUIT)
go func() {
_ = <-sigChan
cb()
os.Exit(0)
}()
}
func main() {
subEvents := []string{
"sched/sched_process_fork",
"sched/sched_process_exec",
"sched/sched_process_exit",
}
probe := ftrace.NewProbe("test_probe", "sys_execve", subEvents)
// start from a clean status
probe.Reset()
if err := probe.Enable(); err != nil {
fmt.Printf("%s\n", err)
return
}
setupSignals(func() {
if err := probe.Disable(); err != nil {
fmt.Printf("%s\n", err)
} else {
fmt.Printf("Probe disabled.\n")
}
})
fmt.Printf("Probe is running ...\n")
for e := range probe.Events() {
if e.IsSyscall {
fmt.Printf("SYSCALL %s\n", e)
} else {
fmt.Printf(" %s\n", e)
}
}
}
|