compiler-rt/test/xray/TestCases/Posix/profiling-multi-threaded.cpp

   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
  23 [[clang::xray_always_instrument]] void f2() { return; }
  24 [[clang::xray_always_instrument]] void f1() { f2(); }
  25 [[clang::xray_always_instrument]] void f0() { f1(); }
  26
  27 using namespace std;
  28
  29 volatile int buffer_counter = 0;
  30
  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
  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 }