| // 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: LSAN_OPTIONS=$LSAN_BASE:"use_registers=0" not %run %t 2>&1 | FileCheck %s |
| // RUN: LSAN_OPTIONS=$LSAN_BASE:"use_registers=1" %run %t 2>&1 |
| // RUN: LSAN_OPTIONS="" %run %t 2>&1 |
| |
| #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); |
| // To store the pointer, choose a register which is unlikely to be reused by |
| // a function call. |
| #if defined(__i386__) |
| asm ( "mov %0, %%esi" |
| : |
| : "r" (p) |
| ); |
| #elif defined(__x86_64__) |
| asm ( "mov %0, %%r15" |
| : |
| : "r" (p) |
| ); |
| #elif defined(__mips__) |
| asm ( "move $16, %0" |
| : |
| : "r" (p) |
| ); |
| #else |
| #error "Test is not supported on this architecture." |
| #endif |
| fprintf(stderr, "Test alloc: %p.\n", p); |
| fflush(stderr); |
| __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:.*]]. |
| // CHECK: LeakSanitizer: detected memory leaks |
| // CHECK: [[ADDR]] (1337 bytes) |
| // CHECK: SUMMARY: {{(Leak|Address)}}Sanitizer: |