xref: /llvm-project/compiler-rt/test/xray/TestCases/Posix/profiling-multi-threaded.cpp (revision f7624b080a6d1f379042261a61b4934f000fd2b1)
1 // Check that we can get a profile from a single-threaded application, on
2 // demand through the XRay logging implementation API.
3 //
4 // FIXME: Make -fxray-modes=xray-profiling part of the default?
5 // RUN: %clangxx_xray -std=c++11 %s -o %t -fxray-modes=xray-profiling
6 // RUN: rm -f xray-log.profiling-multi-*
7 // RUN: XRAY_OPTIONS=verbosity=1 \
8 // RUN:     XRAY_PROFILING_OPTIONS=no_flush=1 %run %t
9 // RUN: XRAY_OPTIONS=verbosity=1 %run %t
10 // RUN: PROFILES=`ls xray-log.profiling-multi-* | wc -l`
11 // RUN: [ $PROFILES -eq 1 ]
12 // RUN: rm -f xray-log.profiling-multi-*
13 //
14 // REQUIRES: built-in-llvm-tree
15 
16 #include "xray/xray_interface.h"
17 #include "xray/xray_log_interface.h"
18 #include <cassert>
19 #include <cstdio>
20 #include <string>
21 #include <thread>
22 
f2()23 [[clang::xray_always_instrument]] void f2() { return; }
f1()24 [[clang::xray_always_instrument]] void f1() { f2(); }
f0()25 [[clang::xray_always_instrument]] void f0() { f1(); }
26 
27 using namespace std;
28 
29 volatile int buffer_counter = 0;
30 
process_buffer(const char *,XRayBuffer)31 [[clang::xray_never_instrument]] void process_buffer(const char *, XRayBuffer) {
32   // FIXME: Actually assert the contents of the buffer.
33   ++buffer_counter;
34 }
35 
main(int,char **)36 [[clang::xray_always_instrument]] int main(int, char **) {
37   assert(__xray_log_select_mode("xray-profiling") ==
38          XRayLogRegisterStatus::XRAY_REGISTRATION_OK);
39   assert(__xray_log_get_current_mode() != nullptr);
40   std::string current_mode = __xray_log_get_current_mode();
41   assert(current_mode == "xray-profiling");
42   assert(__xray_patch() == XRayPatchingStatus::SUCCESS);
43   assert(__xray_log_init_mode("xray-profiling", "") ==
44          XRayLogInitStatus::XRAY_LOG_INITIALIZED);
45   std::thread t0([] { f0(); });
46   std::thread t1([] { f0(); });
47   f0();
48   t0.join();
49   t1.join();
50   assert(__xray_log_finalize() == XRayLogInitStatus::XRAY_LOG_FINALIZED);
51   assert(__xray_log_process_buffers(process_buffer) ==
52          XRayLogFlushStatus::XRAY_LOG_FLUSHED);
53   // We're running three threads, so we expect four buffers (including the file
54   // header buffer).
55   assert(buffer_counter == 4);
56   assert(__xray_log_flushLog() == XRayLogFlushStatus::XRAY_LOG_FLUSHED);
57 }
58