ebpf-challenge/GetRuntimeAddresses/main.go

57 lines
1.1 KiB
Go
Raw Permalink Normal View History

2020-01-14 15:32:06 +01:00
package main
import (
2020-01-16 00:02:59 +01:00
"flag"
2020-01-14 15:32:06 +01:00
"fmt"
"os"
"time"
2020-01-14 15:32:06 +01:00
2022-11-17 14:32:57 +01:00
"kesim.org/ebpf-challenge/GetRuntimeAddresses/ebpf"
"kesim.org/ebpf-challenge/GetRuntimeAddresses/symbolyze"
2020-01-14 15:32:06 +01:00
)
2020-01-16 00:02:59 +01:00
var (
symbol = flag.String("symbol", "_PyRuntime", "Symbol to search for")
glob = flag.String("glob", "*python3*", "pattern an ELF-file must match (see filepath.Match)")
debug = flag.Bool("debug", false, "run in debug mode")
)
2020-01-14 15:32:06 +01:00
func main() {
2020-01-16 00:02:59 +01:00
flag.Parse()
2020-01-14 15:32:06 +01:00
mapFD, err := ebpf.CreateMap()
if err != nil {
fmt.Printf("Failed to create eBPF map: %s\n", err)
os.Exit(1)
}
fmt.Printf("Created eBPF map (FD: %d)\n", mapFD)
scanner := symbolyze.NewScanner(*symbol, *glob)
scanner.OnFound(mapFD.Set)
scanner.Debug(*debug)
go scanner.RunEvery(time.Second)
time.Sleep(10 * time.Second)
scanner.Stop()
2020-01-16 00:02:59 +01:00
err = scanner.Errors()
if err != nil {
fmt.Printf("Failed to run the symbolyze scanner: %s", err)
os.Exit(1)
}
2020-01-15 12:48:36 +01:00
mapContents, err := mapFD.GetMap()
2020-01-14 15:32:06 +01:00
if err != nil {
fmt.Printf("Failed to get the map contents: %s", err)
os.Exit(1)
}
fmt.Printf("Printing contents of map %d\n", mapFD)
for k, v := range mapContents {
fmt.Printf("\t%d -> 0x%x\n", k, v)
}
os.Exit(0)
}