xref: /llvm-project/compiler-rt/test/xray/TestCases/Posix/arg1-arg0-logging.cpp (revision 643a2080ec028bd7674206e0f97cbc7d0f132f99)
1 // Allow having both the no-arg and arg1 logging implementation live together,
2 // and be called in the correct cases.
3 //
4 // RUN: rm -f arg0-arg1-logging-*
5 // RUN: %clangxx_xray -std=c++11 %s -o %t
6 // RUN: env XRAY_OPTIONS="patch_premain=true verbosity=1 xray_logfile_base=arg0-arg1-logging-" %run %t
7 
8 // REQUIRES: target={{(aarch64|x86_64)-.*}}
9 
10 #include "xray/xray_interface.h"
11 #include <cassert>
12 #include <cstdio>
13 
14 using namespace std;
15 
16 bool arg0loggercalled = false;
17 void arg0logger(int32_t, XRayEntryType) { arg0loggercalled = true; }
18 
19 [[clang::xray_always_instrument]] void arg0fn() { printf("hello, arg0!\n"); }
20 
21 bool arg1loggercalled = false;
22 void arg1logger(int32_t, XRayEntryType, uint64_t) { arg1loggercalled = true; }
23 
24 [[ clang::xray_always_instrument, clang::xray_log_args(1) ]] void
25 arg1fn(uint64_t arg1) {
26   printf("hello, arg1!\n");
27 }
28 
29 int main(int argc, char *argv[]) {
30   __xray_set_handler(arg0logger);
31   __xray_set_handler_arg1(arg1logger);
32   arg0fn();
33   arg1fn(0xcafef00d);
34   __xray_remove_handler_arg1();
35   __xray_remove_handler();
36   assert(arg0loggercalled && arg1loggercalled);
37 }
38