1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
package internal
import (
"testing"
"ior/internal/event"
"ior/internal/types"
)
func TestHandleSocketExitTracksReturnedFd(t *testing.T) {
el := mustNewEventLoop(t, eventLoopConfig{})
enter := &types.SocketEvent{
EventType: types.ENTER_SOCKET_EVENT,
TraceId: types.SYS_ENTER_SOCKET,
Time: 100,
Pid: 42,
Tid: 43,
Family: 1,
Type: 2,
Protocol: 0,
}
exit := &types.RetEvent{
EventType: types.EXIT_SOCKET_EVENT,
TraceId: types.SYS_EXIT_SOCKET,
Time: 200,
Ret: 55,
Pid: 42,
Tid: 43,
}
ep := &event.Pair{EnterEv: enter, ExitEv: exit}
if ok := el.handleSocketExit(ep, enter); !ok {
t.Fatal("handleSocketExit returned false")
}
verifyFileDescriptor(t, el, 55, "socket:1:2:0")
}
func TestHandleSocketpairExitTracksReturnedFds(t *testing.T) {
el := mustNewEventLoop(t, eventLoopConfig{})
enter := &types.SocketpairEvent{
EventType: types.ENTER_SOCKETPAIR_EVENT,
TraceId: types.SYS_ENTER_SOCKETPAIR,
Time: 100,
Pid: 77,
Tid: 78,
Family: 1,
Type: 1,
Protocol: 0,
Sv0: 61,
Sv1: 62,
}
exit := &types.RetEvent{
EventType: types.EXIT_SOCKETPAIR_EVENT,
TraceId: types.SYS_EXIT_SOCKETPAIR,
Time: 200,
Ret: 0,
Pid: 77,
Tid: 78,
}
ep := &event.Pair{EnterEv: enter, ExitEv: exit}
if ok := el.handleSocketpairExit(ep, enter); !ok {
t.Fatal("handleSocketpairExit returned false")
}
verifyFileDescriptor(t, el, 61, "socket:1:1:0")
verifyFileDescriptor(t, el, 62, "socket:1:1:0")
}
func TestInitRawHandlersRegistersSocketEvents(t *testing.T) {
el := mustNewEventLoop(t, eventLoopConfig{})
if _, ok := el.rawHandlers[types.ENTER_SOCKET_EVENT]; !ok {
t.Fatal("ENTER_SOCKET_EVENT handler is not registered")
}
if _, ok := el.rawHandlers[types.ENTER_SOCKETPAIR_EVENT]; !ok {
t.Fatal("ENTER_SOCKETPAIR_EVENT handler is not registered")
}
}
|
