tetragon: Add kprobe stats timer

Signed-off-by: Jiri Olsa <[email protected]>

pkg/sensors/tracing/generickprobe.go

@@ -13,8 +13,10 @@ import (
 	"net/http"
 	"path"
 	"strings"
+	"time"
 
 	"github.com/cilium/ebpf"
+	"github.com/cilium/ebpf/link"
 	"github.com/cilium/tetragon/pkg/api/ops"
 	"github.com/cilium/tetragon/pkg/api/processapi"
 	api "github.com/cilium/tetragon/pkg/api/tracingapi"
@@ -28,12 +30,14 @@ import (
 	"github.com/cilium/tetragon/pkg/kernels"
 	"github.com/cilium/tetragon/pkg/logger"
 	"github.com/cilium/tetragon/pkg/metrics/kprobemetrics"
+	"github.com/cilium/tetragon/pkg/metrics/probemetrics"
 	"github.com/cilium/tetragon/pkg/observer"
 	"github.com/cilium/tetragon/pkg/option"
 	"github.com/cilium/tetragon/pkg/policyfilter"
 	"github.com/cilium/tetragon/pkg/selectors"
 	"github.com/cilium/tetragon/pkg/sensors"
 	"github.com/cilium/tetragon/pkg/sensors/program"
+	"github.com/cilium/tetragon/pkg/timer"
 	lru "github.com/hashicorp/golang-lru/v2"
 	"github.com/sirupsen/logrus"
 
@@ -513,6 +517,34 @@ func getKprobeSymbols(symbol string, syscall bool, lists []v1alpha1.ListSpec) ([
 	return []string{symbol}, syscall, nil
 }
 
+func getKprobeStats(policyID policyfilter.PolicyID, policyName string, progs []*program.Program) {
+	for _, prog := range progs {
+		info, err := prog.Link.Info()
+		if err != nil {
+			continue
+		}
+
+		missed := uint64(0)
+
+		switch info.Type {
+		case link.PerfEventType:
+			pevent := info.PerfEvent()
+			switch pevent.PerfEventType {
+			case link.KprobePerfEventInfoType, link.KretprobePerfEventInfoType:
+				kprobe := pevent.Kprobe()
+				missed = kprobe.Missed
+			default:
+			}
+		case link.KprobeMultiType:
+			kmulti := info.KprobeMulti()
+			missed = kmulti.Missed
+		default:
+		}
+
+		probemetrics.Store(uint32(policyID), policyName, prog.Attach, float64(missed))
+	}
+}
+
 func createGenericKprobeSensor(
 	spec *v1alpha1.TracingPolicySpec,
 	name string,
@@ -583,10 +615,24 @@ func createGenericKprobeSensor(
 		return nil, err
 	}
 
+	statsTimer := timer.NewPeriodicTimer("generickprobe stats timer",
+		func() { getKprobeStats(policyID, policyName, progs) }, true)
+
 	return &sensors.Sensor{
 		Name:  name,
 		Progs: progs,
 		Maps:  maps,
+		PostLoadHook: func() error {
+			statsTimer.Start(2 * time.Second)
+			return nil
+		},
+		PostUnloadHook: func() error {
+			statsTimer.Stop()
+			for _, prog := range progs {
+				probemetrics.Remove(uint32(policyID), prog.Attach)
+			}
+			return nil
+		},
 		DestroyHook: func() error {
 			var errs error
 			for _, id := range ids {