Skip to content

Commit

Permalink
pkg/sensors: add TestKprobeMatchBinariesPrefixLargePath
Browse files Browse the repository at this point in the history
Adding test that has Prefix operator in matchBinaries selector. The file
path of the test binary (true) being executed is larger than 256 bytes:
it should be around 3900 chars.

Co-authored-by: Mahe Tardy <[email protected]>
Signed-off-by: Andrei Fedotov <[email protected]>
  • Loading branch information
anfedotoff and mtardy committed Aug 2, 2024
1 parent b28ad1e commit d5ad471
Showing 1 changed file with 49 additions and 0 deletions.
49 changes: 49 additions & 0 deletions pkg/sensors/tracing/kprobe_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -3886,6 +3886,55 @@ func TestKprobeMatchBinaries(t *testing.T) {
})
}

func TestKprobeMatchBinariesPrefixLargePath(t *testing.T) {
if !kernels.EnableLargeProgs() {
t.Skip()
}

// create a large temporary directory path
tmpDir := t.TempDir()
targetBinLargePath := tmpDir
// add (255 + 1) * 15 = 3840 chars to the path
// max is 4096 and we want to leave some space for the tmpdir + others
for range 15 {
targetBinLargePath += "/" + strings.Repeat("a", unix.NAME_MAX)
}
err := os.MkdirAll(targetBinLargePath, 0755)
require.NoError(t, err)

// copy the binary into it
targetBinLargePath += "/true"
fileExec, err := exec.LookPath("true")
require.NoError(t, err)
err = exec.Command("cp", fileExec, targetBinLargePath).Run()
require.NoError(t, err)

var doneWG, readyWG sync.WaitGroup
defer doneWG.Wait()

ctx, cancel := context.WithTimeout(context.Background(), tus.Conf().CmdWaitTime)
defer cancel()

createCrdFile(t, getMatchBinariesCrd("Prefix", []string{tmpDir}))

obs, err := observertesthelper.GetDefaultObserverWithFile(t, ctx, testConfigFile, tus.Conf().TetragonLib, observertesthelper.WithMyPid())
if err != nil {
t.Fatalf("GetDefaultObserverWithFile error: %s", err)
}
observertesthelper.LoopEvents(ctx, t, &doneWG, &readyWG, obs)
readyWG.Wait()

if err := exec.Command(targetBinLargePath).Run(); err != nil {
t.Fatalf("failed to run true: %s", err)
}

checker := ec.NewUnorderedEventChecker(ec.NewProcessKprobeChecker("").
WithProcess(ec.NewProcessChecker().WithBinary(sm.Full(targetBinLargePath))).
WithFunctionName(sm.Full("fd_install")))
err = jsonchecker.JsonTestCheck(t, checker)
assert.NoError(t, err)
}

// matchBinariesPerfringTest checks that the matchBinaries do correctly
// filter the events i.e. it checks that no other events appear.
func matchBinariesPerfringTest(t *testing.T, operator string, values []string) {
Expand Down

0 comments on commit d5ad471

Please sign in to comment.