Peter Collingbourne 6beb6f34bc dfsan: Fix test with gcc 15.
With gcc 15 we end up emitting a reference to the
std::__glibcxx_assert_fail function because of this change:
361d230fd7
combined with assertion checks in the std::atomic implementation.

This reference is undefined with dfsan causing the test to fail. Fix it
by defining the macro that disables assertions.

Pull Request: https://github.com/llvm/llvm-project/pull/153873
2025-08-15 14:44:27 -07:00

63 lines
1.8 KiB
C++

// RUN: %clangxx_dfsan %s -fno-exceptions -D_GLIBCXX_NO_ASSERTIONS -o %t && %run %t
// RUN: %clangxx_dfsan -DORIGIN_TRACKING -mllvm -dfsan-track-origins=1 %s -fno-exceptions -D_GLIBCXX_NO_ASSERTIONS -o %t && %run %t
//
// Use -fno-exceptions to turn off exceptions to avoid instrumenting
// __cxa_begin_catch, std::terminate and __gxx_personality_v0.
//
// Use -D_GLIBCXX_NO_ASSERTIONS to avoid depending on
// std::__glibcxx_assert_fail with gcc >= 15.
//
// TODO: Support builtin atomics. For example, https://gcc.gnu.org/onlinedocs/gcc/_005f_005fatomic-Builtins.html
// DFSan instrumentation pass cannot identify builtin callsites yet.
#include <sanitizer/dfsan_interface.h>
#include <assert.h>
#include <atomic>
#include <pthread.h>
std::atomic<int> atomic_i{0};
struct arg_struct {
size_t index;
dfsan_origin origin;
};
static void *ThreadFn(void *arg) {
if (((arg_struct *)arg)->index % 2) {
int i = 10;
dfsan_set_label(8, (void *)&i, sizeof(i));
atomic_i.store(i, std::memory_order_relaxed);
return 0;
}
int j = atomic_i.load();
assert(dfsan_get_label(j) == 0 || dfsan_get_label(j) == 2);
#ifdef ORIGIN_TRACKING
if (dfsan_get_label(j) == 2)
assert(dfsan_get_init_origin(&j) == ((arg_struct *)arg)->origin);
#endif
return 0;
}
int main(void) {
int i = 10;
dfsan_set_label(2, (void *)&i, sizeof(i));
#ifdef ORIGIN_TRACKING
dfsan_origin origin = dfsan_get_origin(i);
#endif
atomic_i.store(i, std::memory_order_relaxed);
const int kNumThreads = 24;
pthread_t t[kNumThreads];
arg_struct args[kNumThreads];
for (int i = 0; i < kNumThreads; ++i) {
args[i].index = i;
#ifdef ORIGIN_TRACKING
args[i].origin = origin;
#endif
pthread_create(&t[i], 0, ThreadFn, (void *)(args + i));
}
for (int i = 0; i < kNumThreads; ++i)
pthread_join(t[i], 0);
return 0;
}