| 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-single-* |
|---|
| 7 | // RUN: XRAY_OPTIONS=verbosity=1 \ |
|---|
| 8 | // RUN: XRAY_PROFILING_OPTIONS=no_flush=true %run %t |
|---|
| 9 | // RUN: XRAY_OPTIONS=verbosity=1 %run %t |
|---|
| 10 | // RUN: PROFILES=`ls xray-log.profiling-single-* | wc -l` |
|---|
| 11 | // RUN: [ $PROFILES -eq 2 ] |
|---|
| 12 | // RUN: rm -f xray-log.profiling-single-* |
|---|
| 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 | |
|---|
| 22 | [[clang::xray_always_instrument]] void f2() { return; } |
|---|
| 23 | [[clang::xray_always_instrument]] void f1() { f2(); } |
|---|
| 24 | [[clang::xray_always_instrument]] void f0() { f1(); } |
|---|
| 25 | |
|---|
| 26 | using namespace std; |
|---|
| 27 | |
|---|
| 28 | volatile int buffer_counter = 0; |
|---|
| 29 | |
|---|
| 30 | [[clang::xray_never_instrument]] void process_buffer(const char *, XRayBuffer) { |
|---|
| 31 | // FIXME: Actually assert the contents of the buffer. |
|---|
| 32 | ++buffer_counter; |
|---|
| 33 | } |
|---|
| 34 | |
|---|
| 35 | [[clang::xray_always_instrument]] int main(int, char **) { |
|---|
| 36 | assert(__xray_log_select_mode("xray-profiling" ) == |
|---|
| 37 | XRayLogRegisterStatus::XRAY_REGISTRATION_OK); |
|---|
| 38 | assert(__xray_log_get_current_mode() != nullptr); |
|---|
| 39 | std::string current_mode = __xray_log_get_current_mode(); |
|---|
| 40 | assert(current_mode == "xray-profiling" ); |
|---|
| 41 | assert(__xray_patch() == XRayPatchingStatus::SUCCESS); |
|---|
| 42 | assert(__xray_log_init_mode("xray-profiling" , "" ) == |
|---|
| 43 | XRayLogInitStatus::XRAY_LOG_INITIALIZED); |
|---|
| 44 | f0(); |
|---|
| 45 | assert(__xray_log_finalize() == XRayLogInitStatus::XRAY_LOG_FINALIZED); |
|---|
| 46 | f0(); |
|---|
| 47 | assert(__xray_log_process_buffers(process_buffer) == |
|---|
| 48 | XRayLogFlushStatus::XRAY_LOG_FLUSHED); |
|---|
| 49 | // There's always at least one buffer, containing the profile file header. We |
|---|
| 50 | // assert that we have two, to indicate that we're expecting exactly one |
|---|
| 51 | // thread's worth of data. |
|---|
| 52 | assert(buffer_counter == 2); |
|---|
| 53 | assert(__xray_log_flushLog() == XRayLogFlushStatus::XRAY_LOG_FLUSHED); |
|---|
| 54 | |
|---|
| 55 | // Let's reset the counter. |
|---|
| 56 | buffer_counter = 0; |
|---|
| 57 | |
|---|
| 58 | assert(__xray_log_init_mode("xray-profiling" , "" ) == |
|---|
| 59 | XRayLogInitStatus::XRAY_LOG_INITIALIZED); |
|---|
| 60 | f0(); |
|---|
| 61 | assert(__xray_log_finalize() == XRayLogInitStatus::XRAY_LOG_FINALIZED); |
|---|
| 62 | f0(); |
|---|
| 63 | assert(__xray_log_process_buffers(process_buffer) == |
|---|
| 64 | XRayLogFlushStatus::XRAY_LOG_FLUSHED); |
|---|
| 65 | assert(buffer_counter == 2); |
|---|
| 66 | assert(__xray_log_flushLog() == XRayLogFlushStatus::XRAY_LOG_FLUSHED); |
|---|
| 67 | } |
|---|
| 68 | |
|---|
