| /* |
| * Copyright (c) 2016 GitHub, Inc. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| #include <iostream> |
| #include <string> |
| |
| #include "catch.hpp" |
| #include "usdt.h" |
| |
| using std::experimental::optional; |
| using std::experimental::nullopt; |
| |
| static void verify_register(USDT::ArgumentParser &parser, int arg_size, |
| int constant) { |
| USDT::Argument arg; |
| REQUIRE(parser.parse(&arg)); |
| REQUIRE(arg.arg_size() == arg_size); |
| |
| REQUIRE(arg.constant()); |
| REQUIRE(arg.constant() == constant); |
| } |
| |
| static void verify_register(USDT::ArgumentParser &parser, int arg_size, |
| const std::string ®name, |
| optional<int> deref_offset = nullopt, |
| optional<std::string> deref_ident = nullopt, |
| optional<std::string> index_regname = nullopt, |
| optional<int> scale = nullopt) { |
| USDT::Argument arg; |
| REQUIRE(parser.parse(&arg)); |
| REQUIRE(arg.arg_size() == arg_size); |
| |
| REQUIRE(arg.base_register_name()); |
| REQUIRE(arg.base_register_name() == regname); |
| |
| REQUIRE(arg.deref_offset() == deref_offset); |
| REQUIRE(arg.deref_ident() == deref_ident); |
| |
| REQUIRE(arg.index_register_name() == index_regname); |
| REQUIRE(arg.scale() == scale); |
| } |
| |
| TEST_CASE("test usdt argument parsing", "[usdt]") { |
| SECTION("parse failure") { |
| #ifdef __aarch64__ |
| USDT::ArgumentParser_aarch64 parser("4@[x32,200]"); |
| #elif __powerpc64__ |
| USDT::ArgumentParser_powerpc64 parser("4@-12(42)"); |
| #elif defined(__x86_64__) |
| USDT::ArgumentParser_x64 parser("4@i%ra+1r"); |
| #endif |
| USDT::Argument arg; |
| REQUIRE(!parser.parse(&arg)); |
| int i; |
| for (i = 0; i < 10 && !parser.done(); ++i) { |
| parser.parse(&arg); |
| } |
| // Make sure we reach termination |
| REQUIRE(i < 10); |
| } |
| SECTION("argument examples from the Python implementation") { |
| #ifdef __aarch64__ |
| USDT::ArgumentParser_aarch64 parser("-1@x0 4@5 8@[x12] -4@[x31,-40]"); |
| verify_register(parser, -1, "regs[0]"); |
| verify_register(parser, 4, 5); |
| verify_register(parser, 8, "regs[12]", 0); |
| verify_register(parser, -4, "regs[31]", -40); |
| #elif __powerpc64__ |
| USDT::ArgumentParser_powerpc64 parser( |
| "-4@0 8@%r0 8@i0 4@0(%r0) -2@0(0) " |
| "1@0 -2@%r3 -8@i9 -1@0(%r4) -4@16(6) " |
| "2@7 4@%r11 4@i-67 8@-16(%r17) 1@-52(11) " |
| "-8@13 -8@%r25 2@i-11 -2@14(%r26) -8@-32(24) " |
| "4@29 2@%r17 -8@i-693 -1@-23(%r31) 4@28(30) " |
| "-2@31 -4@%r30 2@i1097 4@108(%r30) -2@-4(31)"); |
| |
| verify_register(parser, -4, "gpr[0]"); |
| verify_register(parser, 8, "gpr[0]"); |
| verify_register(parser, 8, 0); |
| verify_register(parser, 4, "gpr[0]", 0); |
| verify_register(parser, -2, "gpr[0]", 0); |
| |
| verify_register(parser, 1, "gpr[0]"); |
| verify_register(parser, -2, "gpr[3]"); |
| verify_register(parser, -8, 9); |
| verify_register(parser, -1, "gpr[4]", 0); |
| verify_register(parser, -4, "gpr[6]", 16); |
| |
| verify_register(parser, 2, "gpr[7]"); |
| verify_register(parser, 4, "gpr[11]"); |
| verify_register(parser, 4, -67); |
| verify_register(parser, 8, "gpr[17]", -16); |
| verify_register(parser, 1, "gpr[11]", -52); |
| |
| verify_register(parser, -8, "gpr[13]"); |
| verify_register(parser, -8, "gpr[25]"); |
| verify_register(parser, 2, -11); |
| verify_register(parser, -2, "gpr[26]", 14); |
| verify_register(parser, -8, "gpr[24]", -32); |
| |
| verify_register(parser, 4, "gpr[29]"); |
| verify_register(parser, 2, "gpr[17]"); |
| verify_register(parser, -8, -693); |
| verify_register(parser, -1, "gpr[31]", -23); |
| verify_register(parser, 4, "gpr[30]", 28); |
| |
| verify_register(parser, -2, "gpr[31]"); |
| verify_register(parser, -4, "gpr[30]"); |
| verify_register(parser, 2, 1097); |
| verify_register(parser, 4, "gpr[30]", 108); |
| verify_register(parser, -2, "gpr[31]", -4); |
| #elif defined(__x86_64__) |
| USDT::ArgumentParser_x64 parser( |
| "-4@$0 8@$1234 %rdi %rax %rsi " |
| "-8@%rbx 4@%r12 8@-8(%rbp) 4@(%rax) " |
| "-4@global_max_action(%rip) " |
| "8@24+mp_(%rip) " |
| "-4@CheckpointStats+40(%rip) " |
| "4@glob-2(%rip) " |
| "8@(%rax,%rdx,8) " |
| "4@(%rbx,%rcx)"); |
| |
| verify_register(parser, -4, 0); |
| verify_register(parser, 8, 1234); |
| |
| verify_register(parser, 8, "di"); |
| verify_register(parser, 8, "ax"); |
| verify_register(parser, 8, "si"); |
| verify_register(parser, -8, "bx"); |
| verify_register(parser, 4, "r12"); |
| |
| verify_register(parser, 8, "bp", -8); |
| verify_register(parser, 4, "ax", 0); |
| |
| verify_register(parser, -4, "ip", 0, std::string("global_max_action")); |
| verify_register(parser, 8, "ip", 24, std::string("mp_")); |
| verify_register(parser, -4, "ip", 40, std::string("CheckpointStats")); |
| verify_register(parser, 4, "ip", -2, std::string("glob")); |
| |
| verify_register(parser, 8, "ax", 0, nullopt, std::string("dx"), 8); |
| verify_register(parser, 4, "bx", 0, nullopt, std::string("cx")); |
| #endif |
| |
| REQUIRE(parser.done()); |
| } |
| } |