forked from iovisor/bcc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_parse_tracepoint.cc
111 lines (101 loc) · 4.12 KB
/
test_parse_tracepoint.cc
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#include "BPF.h"
#include "catch.hpp"
#include "common.h"
TEST_CASE("test tracepoint parser", "[TracepointParser]") {
std::string format =
"name: sys_enter_read\n"
"ID: 650\n"
"format:\n"
" field:unsigned short common_type; offset:0; size:2; signed:0;\n"
" field:unsigned char common_flags; offset:2; size:1; signed:0;\n"
" field:unsigned char common_preempt_count; offset:3; size:1; signed:0;\n"
" field:int common_pid; offset:4; size:4; signed:1;\n"
"\n"
" field:int __syscall_nr; offset:8; size:4; signed:1;\n"
" field:unsigned int fd; offset:16; size:8; signed:0;\n"
" field:char * buf; offset:24; size:8; signed:0;\n"
" field:size_t count; offset:32; size:8; signed:0;\n"
"\n"
"print fmt: \"fd: 0x%08lx, buf: 0x%08lx, count: 0x%08lx\", ((unsigned long)(REC->fd)), ((unsigned long)(REC->buf)), ((unsigned long)(REC->count))\n";
std::string expected =
"struct tracepoint__syscalls__sys_enter_read {\n"
"\tu64 __do_not_use__;\n"
"\tint __syscall_nr;\n"
"\tchar __pad_12;\n"
"\tchar __pad_13;\n"
"\tchar __pad_14;\n"
"\tchar __pad_15;\n"
"\tu64 fd;\n"
"\tchar * buf;\n"
"\tsize_t count;\n"
"};\n";
{
std::istringstream input(format);
std::string result = ebpf::parse_tracepoint(input, "syscalls", "sys_enter_read");
REQUIRE(expected == result);
}
format =
"name: signal_deliver\n"
"ID: 114\n"
"format:\n"
" field:unsigned short common_type; offset:0; size:2; signed:0;\n"
" field:unsigned char common_flags; offset:2; size:1; signed:0;\n"
" field:unsigned char common_preempt_count; offset:3; size:1; signed:0;\n"
" field:int common_pid; offset:4; size:4; signed:1;\n"
"\n"
" field:int sig; offset:8; size:4; signed:1;\n"
" field:int errno; offset:12; size:4; signed:1;\n"
" field:int code; offset:16; size:4; signed:1;\n"
" field:unsigned long sa_handler; offset:24; size:8; signed:0;\n"
" field:unsigned long sa_flags; offset:32; size:8; signed:0;\n"
"\n"
"print fmt: \"sig=%d errno=%d code=%d sa_handler=%lx sa_flags=%lx\", REC->sig, REC->errno, REC->code, REC->sa_handler, REC->sa_flags\n";
expected =
"struct tracepoint__signal__signal_deliver {\n"
"\tu64 __do_not_use__;\n"
"\tint sig;\n"
"\tint errno;\n"
"\tint code;\n"
"\tchar __pad_20;\n"
"\tchar __pad_21;\n"
"\tchar __pad_22;\n"
"\tchar __pad_23;\n"
"\tunsigned long sa_handler;\n"
"\tunsigned long sa_flags;\n"
"};\n";
{
std::istringstream input(format);
std::string result = ebpf::parse_tracepoint(input, "signal", "signal_deliver");
REQUIRE(expected == result);
}
format =
" field:unsigned short common_type; offset:0; size:2; signed:0;\n"
" field:unsigned char common_flags; offset:2; size:1; signed:0;\n"
" field:unsigned char common_preempt_count; offset:3; size:1; signed:0;\n"
" field:int common_pid; offset:4; size:4; signed:1;\n"
" field:unsigned char common_migrate_disable; offset:8; size:1; signed:0;\n"
" field:unsigned char common_preempt_lazy_count; offset:9; size:1; signed:0;\n"
" field:char comm[16]; offset:12; size:16; signed:1;\n"
" field:pid_t pid; offset:28; size:4; signed:1;\n"
" field:int prio; offset:32; size:4; signed:1;\n"
" field:int success; offset:36; size:4; signed:1;\n"
" field:int target_cpu; offset:40; size:4; signed:1;\n";
expected =
"struct tracepoint__sched__sched_wakeup {\n"
"\tu64 __do_not_use__;\n"
"\tchar __do_not_use__8;\n"
"\tchar __do_not_use__9;\n"
"\tchar __pad_10;\n"
"\tchar __pad_11;\n"
"\tchar comm[16];\n"
"\tpid_t pid;\n"
"\tint prio;\n"
"\tint success;\n"
"\tint target_cpu;\n"
"};\n";
{
std::istringstream input(format);
std::string result = ebpf::parse_tracepoint(input, "sched", "sched_wakeup");
REQUIRE(expected == result);
}
}