aboutsummaryrefslogtreecommitdiff
path: root/test/xray/TestCases/Linux/func-id-utils.cc
blob: 412753666019aac3087ebbaa382c8e38211416c4 (plain) (blame)
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
// Check that we can turn a function id to a function address, and also get the
// maximum function id for the current binary.
//
// RUN: %clangxx_xray -std=c++11 %s -o %t
// RUN: XRAY_OPTIONS="patch_premain=false xray_naive_log=false" %run %t

// UNSUPPORTED: target-is-mips64,target-is-mips64el

#include "xray/xray_interface.h"
#include <algorithm>
#include <cassert>
#include <cstdio>
#include <iterator>
#include <set>

[[clang::xray_always_instrument]] void bar(){}

[[clang::xray_always_instrument]] void foo() {
  bar();
}

[[clang::xray_always_instrument]] int main(int argc, char *argv[]) {
  assert(__xray_max_function_id() != 0 && "we need xray instrumentation!");
  std::set<void *> must_be_instrumented = {reinterpret_cast<void *>(&foo),
                                           reinterpret_cast<void *>(&bar),
                                           reinterpret_cast<void *>(&main)};
  std::set<void *> all_instrumented;
  for (auto i = __xray_max_function_id(); i != 0; --i) {
    auto addr = __xray_function_address(i);
    all_instrumented.insert(reinterpret_cast<void *>(addr));
  }
  assert(all_instrumented.size() == __xray_max_function_id() &&
         "each function id must be assigned to a unique function");

  std::set<void *> not_instrumented;
  std::set_difference(
      must_be_instrumented.begin(), must_be_instrumented.end(),
      all_instrumented.begin(), all_instrumented.end(),
      std::inserter(not_instrumented, not_instrumented.begin()));
  assert(
      not_instrumented.empty() &&
      "we should see all explicitly instrumented functions with function ids");
  return not_instrumented.empty() ? 0 : 1;
}