1 /*
2 * Copyright (c) 2016 GitHub, Inc.
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16 #include <iostream>
17 #include <string>
18
19 #include "catch.hpp"
20 #include "usdt.h"
21
22 using std::experimental::optional;
23 using std::experimental::nullopt;
24
verify_register(USDT::ArgumentParser & parser,int arg_size,long long constant)25 static void verify_register(USDT::ArgumentParser &parser, int arg_size,
26 long long constant) {
27 USDT::Argument arg;
28 REQUIRE(parser.parse(&arg));
29 REQUIRE(arg.arg_size() == arg_size);
30
31 REQUIRE(arg.constant());
32 REQUIRE(arg.constant() == constant);
33 }
34
verify_register(USDT::ArgumentParser & parser,int arg_size,const std::string & regname,optional<int> deref_offset=nullopt,optional<std::string> deref_ident=nullopt,optional<std::string> index_regname=nullopt,optional<int> scale=nullopt)35 static void verify_register(USDT::ArgumentParser &parser, int arg_size,
36 const std::string ®name,
37 optional<int> deref_offset = nullopt,
38 optional<std::string> deref_ident = nullopt,
39 optional<std::string> index_regname = nullopt,
40 optional<int> scale = nullopt) {
41 USDT::Argument arg;
42 REQUIRE(parser.parse(&arg));
43 REQUIRE(arg.arg_size() == arg_size);
44
45 REQUIRE(arg.base_register_name());
46 REQUIRE(arg.base_register_name() == regname);
47
48 REQUIRE(arg.deref_offset() == deref_offset);
49 REQUIRE(arg.deref_ident() == deref_ident);
50
51 REQUIRE(arg.index_register_name() == index_regname);
52 REQUIRE(arg.scale() == scale);
53 }
54
55 /* supported arches only */
56 #if defined(__aarch64__) || defined(__powerpc64__) || \
57 defined(__s390x__) || defined(__x86_64__)
58
59 TEST_CASE("test usdt argument parsing", "[usdt]") {
60 SECTION("parse failure") {
61 #ifdef __aarch64__
62 USDT::ArgumentParser_aarch64 parser("4@[x32,200]");
63 #elif __powerpc64__
64 USDT::ArgumentParser_powerpc64 parser("4@-12(42)");
65 #elif __s390x__
66 USDT::ArgumentParser_s390x parser("4@-12(%r42)");
67 #elif defined(__x86_64__)
68 USDT::ArgumentParser_x64 parser("4@i%ra+1r");
69 #endif
70 USDT::Argument arg;
71 REQUIRE(!parser.parse(&arg));
72 int i;
73 for (i = 0; i < 10 && !parser.done(); ++i) {
74 parser.parse(&arg);
75 }
76 // Make sure we reach termination
77 REQUIRE(i < 10);
78 }
79 SECTION("argument examples from the Python implementation") {
80 #ifdef __aarch64__
81 USDT::ArgumentParser_aarch64 parser(
82 "-1@x0 4@5 8@[x12] -4@[x30,-40] -4@[x31,-40] 8@[sp, 120]");
83 verify_register(parser, -1, "regs[0]");
84 verify_register(parser, 4, 5);
85 verify_register(parser, 8, "regs[12]", 0);
86 verify_register(parser, -4, "regs[30]", -40);
87 verify_register(parser, -4, "sp", -40);
88 verify_register(parser, 8, "sp", 120);
89 #elif __powerpc64__
90 USDT::ArgumentParser_powerpc64 parser(
91 "-4@0 8@%r0 8@i0 4@0(%r0) -2@0(0) "
92 "1@0 -2@%r3 -8@i9 -1@0(%r4) -4@16(6) "
93 "2@7 4@%r11 4@i-67 8@-16(%r17) 1@-52(11) "
94 "-8@13 -8@%r25 2@i-11 -2@14(%r26) -8@-32(24) "
95 "4@29 2@%r17 -8@i-693 -1@-23(%r31) 4@28(30) "
96 "-2@31 -4@%r30 2@i1097 4@108(%r30) -2@-4(31)");
97
98 verify_register(parser, -4, "gpr[0]");
99 verify_register(parser, 8, "gpr[0]");
100 verify_register(parser, 8, 0);
101 verify_register(parser, 4, "gpr[0]", 0);
102 verify_register(parser, -2, "gpr[0]", 0);
103
104 verify_register(parser, 1, "gpr[0]");
105 verify_register(parser, -2, "gpr[3]");
106 verify_register(parser, -8, 9);
107 verify_register(parser, -1, "gpr[4]", 0);
108 verify_register(parser, -4, "gpr[6]", 16);
109
110 verify_register(parser, 2, "gpr[7]");
111 verify_register(parser, 4, "gpr[11]");
112 verify_register(parser, 4, -67);
113 verify_register(parser, 8, "gpr[17]", -16);
114 verify_register(parser, 1, "gpr[11]", -52);
115
116 verify_register(parser, -8, "gpr[13]");
117 verify_register(parser, -8, "gpr[25]");
118 verify_register(parser, 2, -11);
119 verify_register(parser, -2, "gpr[26]", 14);
120 verify_register(parser, -8, "gpr[24]", -32);
121
122 verify_register(parser, 4, "gpr[29]");
123 verify_register(parser, 2, "gpr[17]");
124 verify_register(parser, -8, -693);
125 verify_register(parser, -1, "gpr[31]", -23);
126 verify_register(parser, 4, "gpr[30]", 28);
127
128 verify_register(parser, -2, "gpr[31]");
129 verify_register(parser, -4, "gpr[30]");
130 verify_register(parser, 2, 1097);
131 verify_register(parser, 4, "gpr[30]", 108);
132 verify_register(parser, -2, "gpr[31]", -4);
133 #elif __s390x__
134 USDT::ArgumentParser_s390x parser(
135 "-4@%r0 8@%r0 8@0 4@0(%r0) -2@0(%r0) "
136 "1@%r0 -2@%r3 -8@9 -1@0(%r4) -4@16(%r6) "
137 "2@%r7 4@%r11 4@-67 8@-16(%r15) 1@-52(%r11) "
138 "-8@%r4 -8@%r14 2@-11 -2@14(%r13) -8@-32(%r12) "
139 "4@%r5 2@%r11 -8@-693 -1@-23(%r10) 4@28(%r9) "
140 "-2@%r3 -4@%r8 2@1097 4@108(%r7) -2@-4(%r6)");
141
142 verify_register(parser, -4, "gprs[0]");
143 verify_register(parser, 8, "gprs[0]");
144 verify_register(parser, 8, 0);
145 verify_register(parser, 4, "gprs[0]", 0);
146 verify_register(parser, -2, "gprs[0]", 0);
147
148 verify_register(parser, 1, "gprs[0]");
149 verify_register(parser, -2, "gprs[3]");
150 verify_register(parser, -8, 9);
151 verify_register(parser, -1, "gprs[4]", 0);
152 verify_register(parser, -4, "gprs[6]", 16);
153
154 verify_register(parser, 2, "gprs[7]");
155 verify_register(parser, 4, "gprs[11]");
156 verify_register(parser, 4, -67);
157 verify_register(parser, 8, "gprs[15]", -16);
158 verify_register(parser, 1, "gprs[11]", -52);
159
160 verify_register(parser, -8, "gprs[4]");
161 verify_register(parser, -8, "gprs[14]");
162 verify_register(parser, 2, -11);
163 verify_register(parser, -2, "gprs[13]", 14);
164 verify_register(parser, -8, "gprs[12]", -32);
165
166 verify_register(parser, 4, "gprs[5]");
167 verify_register(parser, 2, "gprs[11]");
168 verify_register(parser, -8, -693);
169 verify_register(parser, -1, "gprs[10]", -23);
170 verify_register(parser, 4, "gprs[9]", 28);
171
172 verify_register(parser, -2, "gprs[3]");
173 verify_register(parser, -4, "gprs[8]");
174 verify_register(parser, 2, 1097);
175 verify_register(parser, 4, "gprs[7]", 108);
176 verify_register(parser, -2, "gprs[6]", -4);
177 #elif defined(__x86_64__)
178 USDT::ArgumentParser_x64 parser(
179 "-4@$0 8@$1234 %rdi %rax %rsi "
180 "8@$9223372036854775806 8@$18446744073709551614 "
181 "-8@$-1 "
182 "-8@%rbx 4@%r12 8@-8(%rbp) 4@(%rax) "
183 "-4@global_max_action(%rip) "
184 "8@24+mp_(%rip) "
185 "-4@CheckpointStats+40(%rip) "
186 "4@glob-2(%rip) "
187 "8@(%rax,%rdx,8) "
188 "4@(%rbx,%rcx)");
189
190 verify_register(parser, -4, 0);
191 verify_register(parser, 8, 1234);
192
193 verify_register(parser, 8, "di");
194 verify_register(parser, 8, "ax");
195 verify_register(parser, 8, "si");
196 verify_register(parser, 8, 9223372036854775806ll);
197 verify_register(parser, 8, (long long)18446744073709551614ull);
198 verify_register(parser, -8, -1);
199 verify_register(parser, -8, "bx");
200 verify_register(parser, 4, "r12");
201
202 verify_register(parser, 8, "bp", -8);
203 verify_register(parser, 4, "ax", 0);
204
205 verify_register(parser, -4, "ip", 0, std::string("global_max_action"));
206 verify_register(parser, 8, "ip", 24, std::string("mp_"));
207 verify_register(parser, -4, "ip", 40, std::string("CheckpointStats"));
208 verify_register(parser, 4, "ip", -2, std::string("glob"));
209
210 verify_register(parser, 8, "ax", 0, nullopt, std::string("dx"), 8);
211 verify_register(parser, 4, "bx", 0, nullopt, std::string("cx"));
212 #endif
213
214 REQUIRE(parser.done());
215 }
216 }
217
218 #endif /* supported arches only */
219