x6: add pipe/eventfd fd-from-air syscall support

1 files changed, 120 insertions, 0 deletions

diff --git a/internal/eventloop_ipc_test.go b/internal/eventloop_ipc_test.go
new file mode 100644
index 0000000..eec819b
--- /dev/null
+++ b/internal/eventloop_ipc_test.go
@@ -0,0 +1,120 @@
+package internal
+
+import (
+	"testing"
+
+	"ior/internal/event"
+	"ior/internal/globalfilter"
+	"ior/internal/types"
+)
+
+func TestHandlePipeExitTracksReturnedFds(t *testing.T) {
+	el := mustNewEventLoop(t, eventLoopConfig{})
+
+	enter := &types.PipeEvent{
+		EventType: types.ENTER_PIPE_EVENT,
+		TraceId:   types.SYS_ENTER_PIPE2,
+		Time:      100,
+		Pid:       70,
+		Tid:       71,
+		Flags:     0x80000,
+		Fd0:       -1,
+		Fd1:       -1,
+		Ret:       0,
+	}
+	exit := &types.PipeEvent{
+		EventType: types.EXIT_PIPE_EVENT,
+		TraceId:   types.SYS_EXIT_PIPE2,
+		Time:      200,
+		Pid:       70,
+		Tid:       71,
+		Flags:     0x80000,
+		Fd0:       52,
+		Fd1:       53,
+		Ret:       0,
+	}
+	ep := &event.Pair{EnterEv: enter, ExitEv: exit}
+
+	if ok := el.handlePipeExit(ep, enter); !ok {
+		t.Fatal("handlePipeExit returned false")
+	}
+	verifyFileDescriptor(t, el, 52, "pipe:524288:52:53")
+	verifyFileDescriptor(t, el, 53, "pipe:524288:52:53")
+}
+
+func TestHandleEventfdExitTracksReturnedFd(t *testing.T) {
+	el := mustNewEventLoop(t, eventLoopConfig{})
+
+	enter := &types.EventfdEvent{
+		EventType: types.ENTER_EVENTFD_EVENT,
+		TraceId:   types.SYS_ENTER_EVENTFD2,
+		Time:      100,
+		Pid:       80,
+		Tid:       81,
+		Flags:     0x800,
+		Ret:       -1,
+	}
+	exit := &types.EventfdEvent{
+		EventType: types.EXIT_EVENTFD_EVENT,
+		TraceId:   types.SYS_EXIT_EVENTFD2,
+		Time:      200,
+		Pid:       80,
+		Tid:       81,
+		Flags:     0x800,
+		Ret:       61,
+	}
+	ep := &event.Pair{EnterEv: enter, ExitEv: exit}
+
+	if ok := el.handleEventfdExit(ep, enter); !ok {
+		t.Fatal("handleEventfdExit returned false")
+	}
+	verifyFileDescriptor(t, el, 61, "eventfd:2048")
+}
+
+func TestHandleEventfdExitAppliesPairFilter(t *testing.T) {
+	el := mustNewEventLoop(t, eventLoopConfig{
+		filter: globalfilter.Filter{
+			Syscall: &globalfilter.StringFilter{Pattern: "openat"},
+		},
+	})
+
+	enter := &types.EventfdEvent{
+		EventType: types.ENTER_EVENTFD_EVENT,
+		TraceId:   types.SYS_ENTER_EVENTFD,
+		Time:      100,
+		Pid:       82,
+		Tid:       83,
+		Flags:     0,
+		Ret:       -1,
+	}
+	exit := &types.EventfdEvent{
+		EventType: types.EXIT_EVENTFD_EVENT,
+		TraceId:   types.SYS_EXIT_EVENTFD,
+		Time:      200,
+		Pid:       82,
+		Tid:       83,
+		Flags:     0,
+		Ret:       44,
+	}
+	ep := &event.Pair{EnterEv: enter, ExitEv: exit}
+
+	if ok := el.handleEventfdExit(ep, enter); ok {
+		t.Fatal("handleEventfdExit should reject pair due to filter mismatch")
+	}
+}
+
+func TestInitRawHandlersRegistersIPCEvents(t *testing.T) {
+	el := mustNewEventLoop(t, eventLoopConfig{})
+	if _, ok := el.rawHandlers[types.ENTER_PIPE_EVENT]; !ok {
+		t.Fatal("ENTER_PIPE_EVENT handler is not registered")
+	}
+	if _, ok := el.rawHandlers[types.EXIT_PIPE_EVENT]; !ok {
+		t.Fatal("EXIT_PIPE_EVENT handler is not registered")
+	}
+	if _, ok := el.rawHandlers[types.ENTER_EVENTFD_EVENT]; !ok {
+		t.Fatal("ENTER_EVENTFD_EVENT handler is not registered")
+	}
+	if _, ok := el.rawHandlers[types.EXIT_EVENTFD_EVENT]; !ok {
+		t.Fatal("EXIT_EVENTFD_EVENT handler is not registered")
+	}
+}