contrib/compiler-rt/lib/sanitizer_common/sanitizer_linux.h

   1 //===-- sanitizer_linux.h ---------------------------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // Linux-specific syscall wrappers and classes.
  11 //
  12 //===----------------------------------------------------------------------===//
  13 #ifndef SANITIZER_LINUX_H
  14 #define SANITIZER_LINUX_H
  15
  16 #include "sanitizer_platform.h"
  17 #if SANITIZER_FREEBSD || SANITIZER_LINUX
  18 #include "sanitizer_common.h"
  19 #include "sanitizer_internal_defs.h"
  20 #include "sanitizer_posix.h"
  21 #include "sanitizer_platform_limits_posix.h"
  22
  23 struct link_map;  // Opaque type returned by dlopen().
  24 struct sigaltstack;
  25
  26 namespace __sanitizer {
  27 // Dirent structure for getdents(). Note that this structure is different from
  28 // the one in <dirent.h>, which is used by readdir().
  29 struct linux_dirent;
  30
  31 // Syscall wrappers.
  32 uptr internal_getdents(fd_t fd, struct linux_dirent *dirp, unsigned int count);
  33 uptr internal_sigaltstack(const struct sigaltstack* ss,
  34                           struct sigaltstack* oss);
  35 uptr internal_sigprocmask(int how, __sanitizer_sigset_t *set,
  36     __sanitizer_sigset_t *oldset);
  37
  38 // Linux-only syscalls.
  39 #if SANITIZER_LINUX
  40 uptr internal_prctl(int option, uptr arg2, uptr arg3, uptr arg4, uptr arg5);
  41 // Used only by sanitizer_stoptheworld. Signal handlers that are actually used
  42 // (like the process-wide error reporting SEGV handler) must use
  43 // internal_sigaction instead.
  44 int internal_sigaction_norestorer(int signum, const void *act, void *oldact);
  45 #if (defined(__x86_64__) || SANITIZER_MIPS64) && !SANITIZER_GO
  46 // Uses a raw system call to avoid interceptors.
  47 int internal_sigaction_syscall(int signum, const void *act, void *oldact);
  48 #endif
  49 void internal_sigdelset(__sanitizer_sigset_t *set, int signum);
  50 #if defined(__x86_64__) || defined(__mips__) || defined(__aarch64__) \
  51   || defined(__powerpc64__) || defined(__s390__) || defined(__i386__) \
  52   || defined(__arm__)
  53 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
  54                     int *parent_tidptr, void *newtls, int *child_tidptr);
  55 #endif
  56 #endif  // SANITIZER_LINUX
  57
  58 // This class reads thread IDs from /proc/<pid>/task using only syscalls.
  59 class ThreadLister {
  60  public:
  61   explicit ThreadLister(int pid);
  62   ~ThreadLister();
  63   // GetNextTID returns -1 if the list of threads is exhausted, or if there has
  64   // been an error.
  65   int GetNextTID();
  66   void Reset();
  67   bool error();
  68
  69  private:
  70   bool GetDirectoryEntries();
  71
  72   int pid_;
  73   int descriptor_;
  74   InternalScopedBuffer<char> buffer_;
  75   bool error_;
  76   struct linux_dirent* entry_;
  77   int bytes_read_;
  78 };
  79
  80 // Exposed for testing.
  81 uptr ThreadDescriptorSize();
  82 uptr ThreadSelf();
  83 uptr ThreadSelfOffset();
  84
  85 // Matches a library's file name against a base name (stripping path and version
  86 // information).
  87 bool LibraryNameIs(const char *full_name, const char *base_name);
  88
  89 // Call cb for each region mapped by map.
  90 void ForEachMappedRegion(link_map *map, void (*cb)(const void *, uptr));
  91
  92 #if SANITIZER_ANDROID
  93
  94 #if defined(__aarch64__)
  95 # define __get_tls() \
  96     ({ void** __v; __asm__("mrs %0, tpidr_el0" : "=r"(__v)); __v; })
  97 #elif defined(__arm__)
  98 # define __get_tls() \
  99     ({ void** __v; __asm__("mrc p15, 0, %0, c13, c0, 3" : "=r"(__v)); __v; })
 100 #elif defined(__mips__)
 101 // On mips32r1, this goes via a kernel illegal instruction trap that's
 102 // optimized for v1.
 103 # define __get_tls() \
 104     ({ register void** __v asm("v1"); \
 105        __asm__(".set    push\n" \
 106                ".set    mips32r2\n" \
 107                "rdhwr   %0,$29\n" \
 108                ".set    pop\n" : "=r"(__v)); \
 109        __v; })
 110 #elif defined(__i386__)
 111 # define __get_tls() \
 112     ({ void** __v; __asm__("movl %%gs:0, %0" : "=r"(__v)); __v; })
 113 #elif defined(__x86_64__)
 114 # define __get_tls() \
 115     ({ void** __v; __asm__("mov %%fs:0, %0" : "=r"(__v)); __v; })
 116 #else
 117 #error "Unsupported architecture."
 118 #endif
 119
 120 // The Android Bionic team has allocated a TLS slot for TSan starting with N,
 121 // given that Android currently doesn't support ELF TLS. It is used to store
 122 // Sanitizers thread specific data.
 123 static const int TLS_SLOT_TSAN = 8;
 124
 125 ALWAYS_INLINE uptr *get_android_tls_ptr() {
 126   return reinterpret_cast<uptr *>(&__get_tls()[TLS_SLOT_TSAN]);
 127 }
 128
 129 #endif  // SANITIZER_ANDROID
 130
 131 }  // namespace __sanitizer
 132
 133 #endif  // SANITIZER_FREEBSD || SANITIZER_LINUX
 134 #endif  // SANITIZER_LINUX_H