v6: add KindAccept and wire accept/accept4

1 files changed, 38 insertions, 0 deletions

diff --git a/internal/eventloop_socket_test.go b/internal/eventloop_socket_test.go
index 7712b3c..19c1fa8 100644
--- a/internal/eventloop_socket_test.go
+++ b/internal/eventloop_socket_test.go
@@ -4,6 +4,7 @@ import (
 	"testing"
 
 	"ior/internal/event"
+	"ior/internal/file"
 	"ior/internal/globalfilter"
 	"ior/internal/types"
 )
@@ -107,6 +108,37 @@ func TestHandleSocketpairExitTracksReturnedFdsFromExitEvent(t *testing.T) {
 	verifyFileDescriptor(t, el, 62, "socket:1:1:0")
 }
 
+func TestHandleAcceptExitTracksAcceptedFd(t *testing.T) {
+	el := mustNewEventLoop(t, eventLoopConfig{})
+
+	el.fdState().set(11, file.NewFd(11, "socket:1:1:0", -1))
+
+	enter := &types.AcceptEvent{
+		EventType: types.ENTER_ACCEPT_EVENT,
+		TraceId:   types.SYS_ENTER_ACCEPT4,
+		Time:      100,
+		Pid:       91,
+		Tid:       92,
+		Fd:        11,
+		Ret:       -1,
+	}
+	exit := &types.AcceptEvent{
+		EventType: types.EXIT_ACCEPT_EVENT,
+		TraceId:   types.SYS_EXIT_ACCEPT4,
+		Time:      200,
+		Pid:       91,
+		Tid:       92,
+		Fd:        -1,
+		Ret:       77,
+	}
+	ep := &event.Pair{EnterEv: enter, ExitEv: exit}
+
+	if ok := el.handleAcceptExit(ep, enter); !ok {
+		t.Fatal("handleAcceptExit returned false")
+	}
+	verifyFileDescriptor(t, el, 77, "socket:1:1:0")
+}
+
 func TestInitRawHandlersRegistersSocketEvents(t *testing.T) {
 	el := mustNewEventLoop(t, eventLoopConfig{})
 	if _, ok := el.rawHandlers[types.ENTER_SOCKET_EVENT]; !ok {
@@ -118,4 +150,10 @@ func TestInitRawHandlersRegistersSocketEvents(t *testing.T) {
 	if _, ok := el.rawHandlers[types.EXIT_SOCKETPAIR_EVENT]; !ok {
 		t.Fatal("EXIT_SOCKETPAIR_EVENT handler is not registered")
 	}
+	if _, ok := el.rawHandlers[types.ENTER_ACCEPT_EVENT]; !ok {
+		t.Fatal("ENTER_ACCEPT_EVENT handler is not registered")
+	}
+	if _, ok := el.rawHandlers[types.EXIT_ACCEPT_EVENT]; !ok {
+		t.Fatal("EXIT_ACCEPT_EVENT handler is not registered")
+	}
 }