You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"bytes"
|
|
|
|
"github.com/cilium/ebpf"
|
|
"github.com/cilium/ebpf/link"
|
|
"github.com/cilium/ebpf/ringbuf"
|
|
|
|
_ "embed"
|
|
)
|
|
|
|
const progName = "{{crate_name}}"
|
|
|
|
//go:embed .ebpf/{{project-name}}
|
|
var ebpfBytes []byte
|
|
|
|
func main() {
|
|
if len(os.Args) < 2 {
|
|
fmt.Printf("Usage: %s <interface>\n", os.Args[0])
|
|
os.Exit(1)
|
|
}
|
|
ifaceName := os.Args[1]
|
|
|
|
spec, err := ebpf.LoadCollectionSpecFromReader(bytes.NewReader(ebpfBytes))
|
|
if err != nil {
|
|
log.Fatalf("LoadCollectionSpec failed: %v", err)
|
|
}
|
|
|
|
coll, err := ebpf.NewCollection(spec)
|
|
if err != nil {
|
|
log.Fatalf("NewCollection failed: %v", err)
|
|
}
|
|
defer coll.Close()
|
|
|
|
prog := coll.Programs[progName]
|
|
if prog == nil {
|
|
log.Fatalf("Program %s not found", progName)
|
|
}
|
|
|
|
iface, err := net.InterfaceByName(ifaceName)
|
|
if err != nil {
|
|
log.Fatalf("Interface not found: %v", err)
|
|
}
|
|
|
|
l, err := link.AttachXDP(link.XDPOptions{
|
|
Program: prog,
|
|
Interface: iface.Index,
|
|
})
|
|
if err != nil {
|
|
log.Fatalf("AttachXDP failed: %v", err)
|
|
}
|
|
defer l.Close()
|
|
fmt.Printf("✅ Program '%s' attached to %s\n", progName, ifaceName)
|
|
|
|
logMap, ok := coll.Maps["AYA_LOGS"]
|
|
if !ok {
|
|
log.Fatal("AYA_LOGS map not found")
|
|
}
|
|
|
|
reader, err := ringbuf.NewReader(logMap)
|
|
if err != nil {
|
|
log.Fatalf("failed to create ringbuf reader: %v", err)
|
|
}
|
|
defer reader.Close()
|
|
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
go func() {
|
|
fmt.Println("Listening to Aya logs...")
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
default:
|
|
record, err := reader.Read()
|
|
if err != nil {
|
|
continue
|
|
}
|
|
fmt.Printf("Aya log: %s\n", string(record.RawSample))
|
|
}
|
|
}
|
|
}()
|
|
|
|
<-ctx.Done()
|
|
fmt.Println("Shutting down...")
|
|
}
|