Skip to content

Commit

Permalink
tetragon: Remove MapDir from LoadProbeArgs
Browse files Browse the repository at this point in the history
Removing MapDir from LoadProbeArgs and from its related users.

Signed-off-by: Jiri Olsa <jolsa@kernel.org>
  • Loading branch information
olsajiri committed Feb 21, 2024
1 parent 74e820c commit fe527e8
Show file tree
Hide file tree
Showing 5 changed files with 13 additions and 14 deletions.
1 change: 0 additions & 1 deletion pkg/sensors/load.go
Original file line number Diff line number Diff line change
Expand Up @@ -346,7 +346,6 @@ func loadInstance(bpfDir, mapDir string, load *program.Program, version, verbose
// Registered probes need extra setup
return probe.LoadProbe(LoadProbeArgs{
BPFDir: bpfDir,
MapDir: mapDir,
Load: load,
Version: version,
Verbose: verbose,
Expand Down
2 changes: 1 addition & 1 deletion pkg/sensors/sensors.go
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ func RegisterProbeType(probeType string, s probeLoader) {

// LoadProbeArgs are the args to the LoadProbe function.
type LoadProbeArgs struct {
BPFDir, MapDir string
BPFDir string
Load *program.Program
Version, Verbose int
}
Expand Down
12 changes: 6 additions & 6 deletions pkg/sensors/tracing/generickprobe.go
Original file line number Diff line number Diff line change
Expand Up @@ -933,7 +933,7 @@ func getMapLoad(load *program.Program, kprobeEntry *genericKprobe, index uint32)
return selectorsMaploads(state, kprobeEntry.pinPathPrefix, index)
}

func loadSingleKprobeSensor(id idtable.EntryID, bpfDir, mapDir string, load *program.Program, verbose int) error {
func loadSingleKprobeSensor(id idtable.EntryID, bpfDir string, load *program.Program, verbose int) error {
gk, err := genericKprobeTableGet(id)
if err != nil {
return err
Expand Down Expand Up @@ -961,7 +961,7 @@ func loadSingleKprobeSensor(id idtable.EntryID, bpfDir, mapDir string, load *pro
return err
}

func loadMultiKprobeSensor(ids []idtable.EntryID, bpfDir, mapDir string, load *program.Program, verbose int) error {
func loadMultiKprobeSensor(ids []idtable.EntryID, bpfDir string, load *program.Program, verbose int) error {
bin_buf := make([]bytes.Buffer, len(ids))

data := &program.MultiKprobeAttachData{}
Expand Down Expand Up @@ -1005,12 +1005,12 @@ func loadMultiKprobeSensor(ids []idtable.EntryID, bpfDir, mapDir string, load *p
return nil
}

func loadGenericKprobeSensor(bpfDir, mapDir string, load *program.Program, verbose int) error {
func loadGenericKprobeSensor(bpfDir string, load *program.Program, verbose int) error {
if id, ok := load.LoaderData.(idtable.EntryID); ok {
return loadSingleKprobeSensor(id, bpfDir, mapDir, load, verbose)
return loadSingleKprobeSensor(id, bpfDir, load, verbose)
}
if ids, ok := load.LoaderData.([]idtable.EntryID); ok {
return loadMultiKprobeSensor(ids, bpfDir, mapDir, load, verbose)
return loadMultiKprobeSensor(ids, bpfDir, load, verbose)
}
return fmt.Errorf("invalid loadData type: expecting idtable.EntryID/[] and got: %T (%v)",
load.LoaderData, load.LoaderData)
Expand Down Expand Up @@ -1234,5 +1234,5 @@ func retprobeMerge(prev pendingEvent, curr pendingEvent) *tracing.MsgGenericKpro
}

func (k *observerKprobeSensor) LoadProbe(args sensors.LoadProbeArgs) error {
return loadGenericKprobeSensor(args.BPFDir, args.MapDir, args.Load, args.Verbose)
return loadGenericKprobeSensor(args.BPFDir, args.Load, args.Verbose)
}
4 changes: 2 additions & 2 deletions pkg/sensors/tracing/generictracepoint.go
Original file line number Diff line number Diff line change
Expand Up @@ -567,7 +567,7 @@ func (tp *genericTracepoint) EventConfig() (api.EventConfig, error) {
return config, nil
}

func LoadGenericTracepointSensor(bpfDir, mapDir string, load *program.Program, verbose int) error {
func LoadGenericTracepointSensor(bpfDir string, load *program.Program, verbose int) error {

tracepointLog = logger.GetLogger()

Expand Down Expand Up @@ -760,5 +760,5 @@ func handleMsgGenericTracepoint(
}

func (t *observerTracepointSensor) LoadProbe(args sensors.LoadProbeArgs) error {
return LoadGenericTracepointSensor(args.BPFDir, args.MapDir, args.Load, args.Verbose)
return LoadGenericTracepointSensor(args.BPFDir, args.Load, args.Verbose)
}
8 changes: 4 additions & 4 deletions pkg/sensors/tracing/killer.go
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ func (kp *killerPolicy) PolicyHandler(

func (kp *killerPolicy) loadSingleKillerSensor(
kh *killerHandler,
bpfDir, mapDir string, load *program.Program, verbose int,
bpfDir string, load *program.Program, verbose int,
) error {
if err := program.LoadKprobeProgramAttachMany(bpfDir, load, kh.syscallsSyms, verbose); err == nil {
logger.GetLogger().Infof("Loaded killer sensor: %s", load.Attach)
Expand All @@ -117,7 +117,7 @@ func (kp *killerPolicy) loadSingleKillerSensor(

func (kp *killerPolicy) loadMultiKillerSensor(
kh *killerHandler,
bpfDir, mapDir string, load *program.Program, verbose int,
bpfDir string, load *program.Program, verbose int,
) error {
data := &program.MultiKprobeAttachData{}

Expand All @@ -144,10 +144,10 @@ func (kp *killerPolicy) LoadProbe(args sensors.LoadProbeArgs) error {
return fmt.Errorf("failed to get killer handler for '%s'", name)
}
if args.Load.Label == "kprobe.multi/killer" {
return kp.loadMultiKillerSensor(kh, args.BPFDir, args.MapDir, args.Load, args.Verbose)
return kp.loadMultiKillerSensor(kh, args.BPFDir, args.Load, args.Verbose)
}
if args.Load.Label == "kprobe/killer" {
return kp.loadSingleKillerSensor(kh, args.BPFDir, args.MapDir, args.Load, args.Verbose)
return kp.loadSingleKillerSensor(kh, args.BPFDir, args.Load, args.Verbose)
}

if strings.HasPrefix(args.Load.Label, "fmod_ret/") {
Expand Down

0 comments on commit fe527e8

Please sign in to comment.