12d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// Test that dynamically allocated TLS space is included in the root set. 22d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN: LSAN_BASE="report_objects=1:use_stacks=0:use_registers=0" 32d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN: %clangxx %s -DBUILD_DSO -fPIC -shared -o %t-so.so 42d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN: %clangxx_lsan %s -o %t 52d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN: LSAN_OPTIONS=$LSAN_BASE:"use_tls=0" not %run %t 2>&1 | FileCheck %s 62d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN: LSAN_OPTIONS=$LSAN_BASE:"use_tls=1" %run %t 2>&1 72d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN: LSAN_OPTIONS="" %run %t 2>&1 82d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines 92d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#ifndef BUILD_DSO 102d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#include <assert.h> 112d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#include <dlfcn.h> 122d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#include <stdio.h> 132d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#include <stdlib.h> 142d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#include <string> 152d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines 162d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesint main(int argc, char *argv[]) { 172d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines std::string path = std::string(argv[0]) + "-so.so"; 182d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines 192d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines void *handle = dlopen(path.c_str(), RTLD_LAZY); 202d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines assert(handle != 0); 212d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines typedef void **(* store_t)(void *p); 222d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines store_t StoreToTLS = (store_t)dlsym(handle, "StoreToTLS"); 232d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines assert(dlerror() == 0); 242d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines 252d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines void *p = malloc(1337); 262d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines // If we don't know about dynamic TLS, we will return a false leak above. 272d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines void **p_in_tls = StoreToTLS(p); 282d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines assert(*p_in_tls == p); 292d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines fprintf(stderr, "Test alloc: %p.\n", p); 302d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines return 0; 312d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines} 322d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// CHECK: Test alloc: [[ADDR:.*]]. 332d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// CHECK: LeakSanitizer: detected memory leaks 342d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// CHECK: [[ADDR]] (1337 bytes) 352d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// CHECK: SUMMARY: {{(Leak|Address)}}Sanitizer: 362d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines 372d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#else // BUILD_DSO 382d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// A loadable module with a large thread local section, which would require 392d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// allocation of a new TLS storage chunk when loaded with dlopen(). We use it 402d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// to test the reachability of such chunks in LSan tests. 412d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines 422d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// This must be large enough that it doesn't fit into preallocated static TLS 432d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// space (see STATIC_TLS_SURPLUS in glibc). 442d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines__thread void *huge_thread_local_array[(1 << 20) / sizeof(void *)]; // NOLINT 452d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines 462d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesextern "C" void **StoreToTLS(void *p) { 472d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines huge_thread_local_array[0] = p; 482d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines return &huge_thread_local_array[0]; 492d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines} 502d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#endif // BUILD_DSO 51