test for retrieving exit code

Signed-off-by: Haiyu Zuo <958474674@qq.com>

contrib/tester-progs/.gitignore

@@ -8,6 +8,7 @@ nop
 sigkill-unprivileged-user-ns-tester
 exit-leader
 exit-tester
+exit-code
 libuprobe.so
 uprobe-test-1
 uprobe-test-2

contrib/tester-progs/Makefile

@@ -11,6 +11,7 @@ PROGS = sigkill-tester \
 	nop \
 	exit-leader \
 	exit-tester \
+	exit-code \
 	uprobe-test-1 \
 	uprobe-test-2
 
@@ -28,9 +29,15 @@ exit-tester: exit-tester.c
 sigkill-unprivileged-user-ns-tester: sigkill-unprivileged-user-ns-tester.c
 	$(GCC) -Wall $< -o $@ -lcap
 
+nop: nop.c
+	$(GCC) -Wall $< -o $@ -lpthread
+
 exit-leader: exit-leader.c
 	$(GCC) -Wall $< -o $@ -lpthread
 
+exit-code: exit-code.c
+	$(GCC) -Wall $< -o $@ -lpthread
+
 libuprobe.so: uprobe-lib.c
 	$(GCC) -Wall -fPIC $< -o $@ -shared
 

contrib/tester-progs/exit-code.c

@@ -0,0 +1,4 @@
+int main(int argc, char **argv)
+{
+	return 6;
+}

pkg/sensors/exec/exit_test.go

@@ -173,3 +173,70 @@ func TestExitZombie(t *testing.T) {
 	err = jsonchecker.JsonTestCheck(t, checker)
 	assert.NoError(t, err)
 }
+
+// TestExitCode tests whether we properly return the exit code of the process.
+// see: tester-progs/exit-code.c for the program we use to test this.
+//
+// The program will:
+//   - create a thread
+//   - return a exit code
+//
+// In our test we check whether the observed exit code equals the real exit code.
+func TestExitCode(t *testing.T) {
+	var doneWG, readyWG sync.WaitGroup
+	defer doneWG.Wait()
+
+	ctx, cancel := context.WithTimeout(context.Background(), tus.Conf().CmdWaitTime)
+	defer cancel()
+
+	obs, err := observer.GetDefaultObserver(t, ctx, tus.Conf().TetragonLib)
+	if err != nil {
+		t.Fatalf("Failed to run observer: %s", err)
+	}
+	observer.LoopEvents(ctx, t, &doneWG, &readyWG, obs)
+	readyWG.Wait()
+
+	testExitCode := testutils.RepoRootPath("contrib/tester-progs/exit-code")
+
+	var exitCode, realCode uint32
+
+	if err := exec.Command(testExitCode).Run(); err != nil {
+		if exitErr, ok := err.(*exec.ExitError); ok {
+			// handle ExitError
+			realCode = uint32(exitErr.ExitCode())
+		} else {
+			t.Fatalf("Failed to execute test binary: %s\n", err)
+		}
+	}
+
+	// The test executes 'exit-code' benary which return a exit code 6
+	nextCheck := func(event ec.Event, l *logrus.Logger) (bool, error) {
+		switch ev := event.(type) {
+		case *tetragon.ProcessExit:
+			if ev.Process.Binary == testExitCode {
+				exitCode = ev.Status
+			}
+			return false, nil
+		}
+		return false, nil
+	}
+
+	finalCheck := func(l *logrus.Logger) error {
+		fmt.Printf("exitCode %v\n", exitCode)
+
+		if exitCode == realCode {
+			return nil
+		}
+		return fmt.Errorf("tetragon returns the exit code of the process uncorrectly")
+	}
+
+	checker := &ec.FnEventChecker{
+		NextCheckFn:  nextCheck,
+		FinalCheckFn: finalCheck,
+	}
+
+	if err := jsonchecker.JsonTestCheck(t, checker); err != nil {
+		t.Logf("error: %s", err)
+		t.Fail()
+	}
+}