68 lines · cpp
1// Check that we can get a profile from a single-threaded application, on2// 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-profiling6// RUN: rm -f xray-log.profiling-single-*7// RUN: env XRAY_OPTIONS=verbosity=1 \8// RUN: XRAY_PROFILING_OPTIONS=no_flush=true %run %t9// RUN: env XRAY_OPTIONS=verbosity=1 %run %t10// RUN: ls xray-log.profiling-single-* | wc -l | tr -d '\n' > %t.profiles11// RUN: %python -c "import sys; sys.exit(int(sys.argv[1]) - 2)" %{readfile:%t.profiles} 12// RUN: rm -f xray-log.profiling-single-*13//14// REQUIRES: built-in-llvm-tree15 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 26using namespace std;27 28volatile 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. We50 // assert that we have two, to indicate that we're expecting exactly one51 // 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