| // Test that registers of running threads are included in the root set. |
| // RUN: LSAN_BASE="report_objects=1:use_stacks=0" |
| // RUN: %clangxx_lsan -pthread %s -o %t |
| // RUN: %env_lsan_opts=$LSAN_BASE:"use_registers=0" not %run %t 2>&1 | FileCheck %s |
| // RUN: %env_lsan_opts=$LSAN_BASE:"use_registers=1" %run %t 2>&1 |
| // RUN: %env_lsan_opts="" %run %t 2>&1 |
| |
| #include "sanitizer_common/print_address.h" |
| #include <assert.h> |
| #include <pthread.h> |
| #include <sched.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| |
| extern "C" void *registers_thread_func(void *arg) { |
| int *sync = reinterpret_cast<int *>(arg); |
| void *p = malloc(1337); |
| print_address("Test alloc: ", 1, p); |
| fflush(stderr); |
| |
| // To store the pointer, choose a register which is unlikely to be reused by |
| // a function call. |
| #if defined(__i386__) || defined(__i686__) |
| asm("mov %0, %%edi" |
| : |
| : "r"(p)); |
| #elif defined(__x86_64__) |
| asm("mov %0, %%r15" |
| : |
| : "r"(p)); |
| #elif defined(__mips__) |
| asm("move $16, %0" |
| : |
| : "r"(p)); |
| #elif defined(__arm__) |
| asm("mov r5, %0" |
| : |
| : "r"(p)); |
| #elif defined(__aarch64__) |
| // x9-10are used. x11-12 are probably used. |
| // So we pick x13 to be safe. |
| asm("mov x13, %0" |
| : |
| : "r"(p)); |
| #elif defined(__powerpc__) |
| asm("mr 30, %0" |
| : |
| : "r"(p)); |
| #elif defined(__s390x__) |
| asm("lgr %%r10, %0" |
| : |
| : "r"(p)); |
| #elif defined(__riscv) |
| asm("mv s11, %0" |
| : |
| : "r"(p)); |
| #else |
| #error "Test is not supported on this architecture." |
| #endif |
| __sync_fetch_and_xor(sync, 1); |
| while (true) |
| sched_yield(); |
| } |
| |
| int main() { |
| int sync = 0; |
| pthread_t thread_id; |
| int res = pthread_create(&thread_id, 0, registers_thread_func, &sync); |
| assert(res == 0); |
| while (!__sync_fetch_and_xor(&sync, 0)) |
| sched_yield(); |
| return 0; |
| } |
| // CHECK: Test alloc: [[ADDR:0x[0-9,a-f]+]] |
| // CHECK: LeakSanitizer: detected memory leaks |
| // CHECK: [[ADDR]] (1337 bytes) |
| // CHECK: SUMMARY: {{(Leak|Address)}}Sanitizer: |