aboutsummaryrefslogtreecommitdiff
path: root/lib/scudo/scudo_tls_linux.cpp
blob: 5a9cc998bccf39135a489407ce1b45ac4a8133fa (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
//===-- scudo_tls_linux.cpp -------------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
///
/// Scudo thread local structure implementation for platforms supporting
/// thread_local.
///
//===----------------------------------------------------------------------===//

#include "sanitizer_common/sanitizer_platform.h"

#if SANITIZER_LINUX && !SANITIZER_ANDROID

#include "scudo_tls.h"

#include <limits.h>
#include <pthread.h>

namespace __scudo {

static pthread_once_t GlobalInitialized = PTHREAD_ONCE_INIT;
static pthread_key_t PThreadKey;

__attribute__((tls_model("initial-exec")))
THREADLOCAL ThreadState ScudoThreadState = ThreadNotInitialized;
__attribute__((tls_model("initial-exec")))
THREADLOCAL ScudoThreadContext ThreadLocalContext;

static void teardownThread(void *Ptr) {
  uptr Iteration = reinterpret_cast<uptr>(Ptr);
  // The glibc POSIX thread-local-storage deallocation routine calls user
  // provided destructors in a loop of PTHREAD_DESTRUCTOR_ITERATIONS.
  // We want to be called last since other destructors might call free and the
  // like, so we wait until PTHREAD_DESTRUCTOR_ITERATIONS before draining the
  // quarantine and swallowing the cache.
  if (Iteration < PTHREAD_DESTRUCTOR_ITERATIONS) {
    pthread_setspecific(PThreadKey, reinterpret_cast<void *>(Iteration + 1));
    return;
  }
  ThreadLocalContext.commitBack();
  ScudoThreadState = ThreadTornDown;
}


static void initOnce() {
  CHECK_EQ(pthread_key_create(&PThreadKey, teardownThread), 0);
  initScudo();
}

void initThread() {
  pthread_once(&GlobalInitialized, initOnce);
  pthread_setspecific(PThreadKey, reinterpret_cast<void *>(1));
  ThreadLocalContext.init();
  ScudoThreadState = ThreadInitialized;
}

}  // namespace __scudo

#endif  // SANITIZER_LINUX && !SANITIZER_ANDROID