643a2080ec
When running tests in compiler-rt using the lit internal shell with the following command: ``` LIT_USE_INTERNAL_SHELL=1 ninja check-compiler-rt ``` one common error that occurs is: ``` 'XRAY_OPTIONS=patch_premain=false:verbosity=1': command not found ``` This error, along with over 50 similar "environment variable not found" errors, appears across 35 files in complier-rt. These errors happen because the environment variables are not being set correctly when the internal shell is used, leading to commands failing due to unrecognized environment variables. This patch addresses the issue by using the `env` command to properly set the environment variables before running the tests. By explicitly setting the environment variables through `env`, the internal shell can correctly interpret and apply them, allowing the tests to pass. fixes: #102395 [link to RFC](https://discourse.llvm.org/t/rfc-enabling-the-lit-internal-shell-by-default/80179)
94 lines
2.7 KiB
C++
94 lines
2.7 KiB
C++
// Check that we can patch and unpatch specific function ids.
|
|
//
|
|
// RUN: %clangxx_xray -std=c++11 %s -o %t
|
|
// RUN: env XRAY_OPTIONS="patch_premain=false" %run %t | FileCheck %s
|
|
// RUN: %clangxx_xray -fno-xray-function-index -std=c++11 %s -o %t
|
|
// RUN: env XRAY_OPTIONS="patch_premain=false" %run %t | FileCheck %s
|
|
|
|
// UNSUPPORTED: target-is-mips64,target-is-mips64el
|
|
|
|
#include "xray/xray_interface.h"
|
|
|
|
#include <set>
|
|
#include <cstdio>
|
|
#include <cassert>
|
|
|
|
std::set<int32_t> function_ids;
|
|
|
|
[[clang::xray_never_instrument]] void coverage_handler(int32_t fid,
|
|
XRayEntryType) {
|
|
thread_local bool patching = false;
|
|
if (patching) return;
|
|
patching = true;
|
|
function_ids.insert(fid);
|
|
__xray_unpatch_function(fid);
|
|
patching = false;
|
|
}
|
|
|
|
[[clang::xray_always_instrument]] void baz() {
|
|
// do nothing!
|
|
}
|
|
|
|
[[clang::xray_always_instrument]] void bar() {
|
|
baz();
|
|
}
|
|
|
|
[[clang::xray_always_instrument]] void foo() {
|
|
bar();
|
|
}
|
|
|
|
[[clang::xray_always_instrument]] int main(int argc, char *argv[]) {
|
|
__xray_set_handler(coverage_handler);
|
|
assert(__xray_patch() == XRayPatchingStatus::SUCCESS);
|
|
foo();
|
|
assert(__xray_unpatch() == XRayPatchingStatus::SUCCESS);
|
|
|
|
// print out the function_ids.
|
|
printf("first pass.\n");
|
|
for (const auto id : function_ids)
|
|
printf("patched: %d\n", id);
|
|
|
|
// CHECK-LABEL: first pass.
|
|
// CHECK-DAG: patched: [[F1:.*]]
|
|
// CHECK-DAG: patched: [[F2:.*]]
|
|
// CHECK-DAG: patched: [[F3:.*]]
|
|
|
|
// make a copy of the function_ids, then patch them later.
|
|
auto called_fns = function_ids;
|
|
|
|
// clear the function_ids.
|
|
function_ids.clear();
|
|
|
|
// patch the functions we've called before.
|
|
for (const auto id : called_fns)
|
|
assert(__xray_patch_function(id) == XRayPatchingStatus::SUCCESS);
|
|
|
|
// then call them again.
|
|
foo();
|
|
assert(__xray_unpatch() == XRayPatchingStatus::SUCCESS);
|
|
|
|
// confirm that we've seen the same functions again.
|
|
printf("second pass.\n");
|
|
for (const auto id : function_ids)
|
|
printf("patched: %d\n", id);
|
|
// CHECK-LABEL: second pass.
|
|
// CHECK-DAG: patched: [[F1]]
|
|
// CHECK-DAG: patched: [[F2]]
|
|
// CHECK-DAG: patched: [[F3]]
|
|
|
|
// Now we want to make sure that if we unpatch one, that we're only going to
|
|
// see two calls of the coverage_handler.
|
|
function_ids.clear();
|
|
assert(__xray_patch() == XRayPatchingStatus::SUCCESS);
|
|
assert(__xray_unpatch_function(1) == XRayPatchingStatus::SUCCESS);
|
|
foo();
|
|
assert(__xray_unpatch() == XRayPatchingStatus::SUCCESS);
|
|
|
|
// confirm that we don't see function id one called anymore.
|
|
printf("missing 1.\n");
|
|
for (const auto id : function_ids)
|
|
printf("patched: %d\n", id);
|
|
// CHECK-LABEL: missing 1.
|
|
// CHECK-NOT: patched: 1
|
|
}
|