asan_mac.cc revision 3f4c3875c42078e22c7e5356c5746fd18756d958
11e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===-- asan_mac.cc -------------------------------------------------------===//
21e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
31e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//                     The LLVM Compiler Infrastructure
41e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
51e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This file is distributed under the University of Illinois Open Source
61e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// License. See LICENSE.TXT for details.
71e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
81e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===----------------------------------------------------------------------===//
91e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This file is a part of AddressSanitizer, an address sanity checker.
111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Mac-specific details.
131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===----------------------------------------------------------------------===//
141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
15d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany#ifdef __APPLE__
161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
1764ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov#include "asan_interceptors.h"
181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_internal.h"
19895b3872acb5bcccb1769ea69d37dd33c722f99dAlexander Potapenko#include "asan_mapping.h"
208a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#include "asan_procmaps.h"
211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_stack.h"
221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_thread.h"
231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_thread_registry.h"
241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
251e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko#include <crt_externs.h>  // for _NSGetEnviron
268a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#include <mach-o/dyld.h>
279b993e8cd0f8964782ee93524603d0c53adc2249Kostya Serebryany#include <mach-o/loader.h>
281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <sys/mman.h>
29ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany#include <sys/resource.h>
3059dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko#include <sys/sysctl.h>
319107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany#include <sys/ucontext.h>
32c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany#include <pthread.h>
33a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany#include <fcntl.h>
341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <unistd.h>
35d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany#include <libkern/OSAtomic.h>
3664ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov#include <CoreFoundation/CFString.h>
371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanynamespace __asan {
391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
403f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanyvoid GetPcSpBp(void *context, uptr *pc, uptr *sp, uptr *bp) {
419107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  ucontext_t *ucontext = (ucontext_t*)context;
429107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# if __WORDSIZE == 64
439107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *pc = ucontext->uc_mcontext->__ss.__rip;
449107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *bp = ucontext->uc_mcontext->__ss.__rbp;
459107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *sp = ucontext->uc_mcontext->__ss.__rsp;
469107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# else
479107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *pc = ucontext->uc_mcontext->__ss.__eip;
489107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *bp = ucontext->uc_mcontext->__ss.__ebp;
499107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *sp = ucontext->uc_mcontext->__ss.__esp;
509107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# endif  // __WORDSIZE
519107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany}
529107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany
5338dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonovenum {
5438dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  MACOS_VERSION_UNKNOWN = 0,
5538dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  MACOS_VERSION_LEOPARD,
5638dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  MACOS_VERSION_SNOW_LEOPARD,
5738dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  MACOS_VERSION_LION,
5838dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov};
5938dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov
6038dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonovstatic int GetMacosVersion() {
6159dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  int mib[2] = { CTL_KERN, KERN_OSRELEASE };
6259dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  char version[100];
6359dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  size_t len = 0, maxlen = sizeof(version) / sizeof(version[0]);
6459dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  for (int i = 0; i < maxlen; i++) version[i] = '\0';
6559dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  // Get the version length.
663f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(sysctl(mib, 2, 0, &len, 0, 0) != -1);
6759dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  CHECK(len < maxlen);
683f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(sysctl(mib, 2, version, &len, 0, 0) != -1);
6959dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  switch (version[0]) {
7059dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko    case '9': return MACOS_VERSION_LEOPARD;
7159dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko    case '1': {
7259dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko      switch (version[1]) {
7359dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko        case '0': return MACOS_VERSION_SNOW_LEOPARD;
7459dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko        case '1': return MACOS_VERSION_LION;
7559dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko        default: return MACOS_VERSION_UNKNOWN;
7659dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko      }
7759dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko    }
7859dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko    default: return MACOS_VERSION_UNKNOWN;
7959dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  }
8059dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko}
8159dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko
8238dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonovbool PlatformHasDifferentMemcpyAndMemmove() {
8338dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // On OS X 10.7 memcpy() and memmove() are both resolved
8438dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // into memmove$VARIANT$sse42.
8538dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // See also http://code.google.com/p/address-sanitizer/issues/detail?id=34.
8638dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // TODO(glider): need to check dynamically that memcpy() and memmove() are
8738dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // actually the same function.
8838dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  return GetMacosVersion() == MACOS_VERSION_SNOW_LEOPARD;
8938dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov}
9038dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov
911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// No-op. Mac does not support static linkage anyway.
921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *AsanDoesNotSupportStaticLinkage() {
933f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  return 0;
941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
964803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryanybool AsanInterceptsSignal(int signum) {
974803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  return (signum == SIGSEGV || signum == SIGBUS) && FLAG_handle_segv;
984803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
994803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
100a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanystatic void *asan_mmap(void *addr, size_t length, int prot, int flags,
1011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany                int fd, uint64_t offset) {
1021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return mmap(addr, length, prot, flags, fd, offset);
1031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
1041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
1050ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanWrite(int fd, const void *buf, size_t count) {
1061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return write(fd, buf, count);
1071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
1081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
109de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid *AsanMmapSomewhereOrDie(size_t size, const char *mem_type) {
110de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  size = RoundUpTo(size, kPageSize);
111de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  void *res = asan_mmap(0, size,
112de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany                        PROT_READ | PROT_WRITE,
113de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany                        MAP_PRIVATE | MAP_ANON, -1, 0);
114de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  if (res == (void*)-1) {
115de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany    OutOfMemoryMessageAndDie(mem_type, size);
116de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  }
117de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  return res;
118de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
119de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
1203f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanyvoid *AsanMmapFixedNoReserve(uptr fixed_addr, size_t size) {
121a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany  return asan_mmap((void*)fixed_addr, size,
122a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany                   PROT_READ | PROT_WRITE,
123a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany                   MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE,
124a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany                   0, 0);
125a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany}
126a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany
1273f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanyvoid *AsanMprotect(uptr fixed_addr, size_t size) {
128a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany  return asan_mmap((void*)fixed_addr, size,
129a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany                   PROT_NONE,
130a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany                   MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE,
131a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany                   0, 0);
132a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany}
133a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany
134de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid AsanUnmapOrDie(void *addr, size_t size) {
135de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  if (!addr || !size) return;
136de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  int res = munmap(addr, size);
137de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  if (res != 0) {
138de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany    Report("Failed to unmap\n");
1390ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryany    AsanDie();
140de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  }
141de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
142de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
143de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanOpenReadonly(const char* filename) {
144de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  return open(filename, O_RDONLY);
145de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
146de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
1471e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenkoconst char *AsanGetEnv(const char *name) {
1481e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  char ***env_ptr = _NSGetEnviron();
1491e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  CHECK(env_ptr);
1501e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  char **environ = *env_ptr;
1511e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  CHECK(environ);
1521e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  size_t name_len = internal_strlen(name);
1533f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  while (*environ != 0) {
1541e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko    size_t len = internal_strlen(*environ);
1551e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko    if (len > name_len) {
1561e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko      const char *p = *environ;
1571e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko      if (!internal_memcmp(p, name, name_len) &&
1581e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko          p[name_len] == '=') {  // Match.
1594dd8ba8238b1b698953628affe6e5b2edf3b3e3fAlexey Samsonov        return *environ + name_len + 1;  // String starting after =.
1601e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko      }
1611e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko    }
1621e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko    environ++;
1631e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  }
1643f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  return 0;
1651e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko}
1661e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko
1670ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanRead(int fd, void *buf, size_t count) {
168de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  return read(fd, buf, count);
169de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
170de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
171de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanClose(int fd) {
172de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  return close(fd);
173de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
174de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
1758a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander PotapenkoAsanProcMaps::AsanProcMaps() {
1768a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  Reset();
1778a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
1788a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
1798a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander PotapenkoAsanProcMaps::~AsanProcMaps() {
1808a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
1818a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
1823feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// More information about Mach-O headers can be found in mach-o/loader.h
1833feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Each Mach-O image has a header (mach_header or mach_header_64) starting with
1843feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// a magic number, and a list of linker load commands directly following the
1853feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// header.
1863feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// A load command is at least two 32-bit words: the command type and the
1873feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// command size in bytes. We're interested only in segment load commands
1883feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// (LC_SEGMENT and LC_SEGMENT_64), which tell that a part of the file is mapped
1893feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// into the task's address space.
1903feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// The |vmaddr|, |vmsize| and |fileoff| fields of segment_command or
1913feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// segment_command_64 correspond to the memory address, memory size and the
1923feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// file offset of the current memory segment.
1933feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Because these fields are taken from the images as is, one needs to add
1943feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// _dyld_get_image_vmaddr_slide() to get the actual addresses at runtime.
1953feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko
1968a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenkovoid AsanProcMaps::Reset() {
1978a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  // Count down from the top.
1988a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  // TODO(glider): as per man 3 dyld, iterating over the headers with
1993feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  // _dyld_image_count is thread-unsafe. We need to register callbacks for
2003feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  // adding and removing images which will invalidate the AsanProcMaps state.
2018a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  current_image_ = _dyld_image_count();
2023feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  current_load_cmd_count_ = -1;
2033f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  current_load_cmd_addr_ = 0;
204ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko  current_magic_ = 0;
2053feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko}
2063feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko
2073feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Next and NextSegmentLoad were inspired by base/sysinfo.cc in
2083feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Google Perftools, http://code.google.com/p/google-perftools.
2093feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko
2103feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// NextSegmentLoad scans the current image for the next segment load command
2113feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// and returns the start and end addresses and file offset of the corresponding
2123feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// segment.
2133feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Note that the segment addresses are not necessarily sorted.
2143feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenkotemplate<uint32_t kLCSegment, typename SegmentCommand>
2153feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenkobool AsanProcMaps::NextSegmentLoad(
2163f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany    uptr *start, uptr *end, uptr *offset,
2173feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko    char filename[], size_t filename_size) {
2183feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  const char* lc = current_load_cmd_addr_;
2193feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  current_load_cmd_addr_ += ((const load_command *)lc)->cmdsize;
2208a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  if (((const load_command *)lc)->cmd == kLCSegment) {
2213feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko    const intptr_t dlloff = _dyld_get_image_vmaddr_slide(current_image_);
2228a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    const SegmentCommand* sc = (const SegmentCommand *)lc;
2238a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (start) *start = sc->vmaddr + dlloff;
2248a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (end) *end = sc->vmaddr + sc->vmsize + dlloff;
2258a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (offset) *offset = sc->fileoff;
2268a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (filename) {
22709672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov      REAL(strncpy)(filename, _dyld_get_image_name(current_image_),
22809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                    filename_size);
2298a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    }
230ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko    if (FLAG_v >= 4)
231ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      Report("LC_SEGMENT: %p--%p %s+%p\n", *start, *end, filename, *offset);
2328a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    return true;
2338a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  }
2348a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  return false;
2358a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
2368a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
2373f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanybool AsanProcMaps::Next(uptr *start, uptr *end,
2383f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany                        uptr *offset, char filename[],
2398a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko                        size_t filename_size) {
2408a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  for (; current_image_ >= 0; current_image_--) {
2418a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    const mach_header* hdr = _dyld_get_image_header(current_image_);
2428a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (!hdr) continue;
2433feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko    if (current_load_cmd_count_ < 0) {
2443feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko      // Set up for this image;
2453feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko      current_load_cmd_count_ = hdr->ncmds;
246ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      current_magic_ = hdr->magic;
247ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      switch (current_magic_) {
2483feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko#ifdef MH_MAGIC_64
2493feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        case MH_MAGIC_64: {
2503feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          current_load_cmd_addr_ = (char*)hdr + sizeof(mach_header_64);
2513feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          break;
2523feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        }
2533feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko#endif
2543feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        case MH_MAGIC: {
2553feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          current_load_cmd_addr_ = (char*)hdr + sizeof(mach_header);
2563feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          break;
2573feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        }
2583feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        default: {
2593feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          continue;
2603feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        }
2613feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko      }
2623feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko    }
2638a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
264ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko    for (; current_load_cmd_count_ >= 0; current_load_cmd_count_--) {
265ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      switch (current_magic_) {
266ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        // current_magic_ may be only one of MH_MAGIC, MH_MAGIC_64.
2678a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#ifdef MH_MAGIC_64
268ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        case MH_MAGIC_64: {
269ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko          if (NextSegmentLoad<LC_SEGMENT_64, struct segment_command_64>(
270ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko                  start, end, offset, filename, filename_size))
271ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko            return true;
272ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko          break;
273ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        }
2748a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#endif
275ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        case MH_MAGIC: {
276ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko          if (NextSegmentLoad<LC_SEGMENT, struct segment_command>(
277ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko                  start, end, offset, filename, filename_size))
278ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko            return true;
279ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko          break;
280ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        }
281ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      }
2828a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    }
2838a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    // If we get here, no more load_cmd's in this image talk about
2848a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    // segments.  Go on to the next image.
2858a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  }
2868a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  return false;
2878a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
2888a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
2893f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanybool AsanProcMaps::GetObjectNameAndOffset(uptr addr, uptr *offset,
2908a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko                                          char filename[],
2918a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko                                          size_t filename_size) {
2928a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  return IterateForObjectNameAndOffset(addr, offset, filename, filename_size);
2938a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
2948a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
295c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryanyvoid AsanThread::SetThreadStackTopAndBottom() {
296c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany  size_t stacksize = pthread_get_stacksize_np(pthread_self());
297c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany  void *stackaddr = pthread_get_stackaddr_np(pthread_self());
2983f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  stack_top_ = (uptr)stackaddr;
299c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany  stack_bottom_ = stack_top_ - stacksize;
300c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany  int local;
3013f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(AddrIsInStack((uptr)&local));
302c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany}
303c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany
304d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya SerebryanyAsanLock::AsanLock(LinkerInitialized) {
305d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  // We assume that OS_SPINLOCK_INIT is zero
306d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany}
307d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany
308d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryanyvoid AsanLock::Lock() {
309d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  CHECK(sizeof(OSSpinLock) <= sizeof(opaque_storage_));
310d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  CHECK(OS_SPINLOCK_INIT == 0);
3113f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(owner_ != (uptr)pthread_self());
312d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  OSSpinLockLock((OSSpinLock*)&opaque_storage_);
313d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  CHECK(!owner_);
3143f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  owner_ = (uptr)pthread_self();
315d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany}
316d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany
317d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryanyvoid AsanLock::Unlock() {
3183f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(owner_ == (uptr)pthread_self());
319d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  owner_ = 0;
320d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  OSSpinLockUnlock((OSSpinLock*)&opaque_storage_);
321d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany}
322d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany
3233f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanyvoid AsanStackTrace::GetStackTrace(size_t max_s, uptr pc, uptr bp) {
3249cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  size = 0;
3259cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  trace[0] = pc;
3269cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  if ((max_s) > 1) {
3279cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov    max_size = max_s;
3289cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov    FastUnwindStack(pc, bp);
3299cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  }
3309cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov}
3319cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov
3325b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov// The range of pages to be used for escape islands.
3333281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// TODO(glider): instead of mapping a fixed range we must find a range of
3343281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// unmapped pages in vmmap and take them.
3351346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// These constants were chosen empirically and may not work if the shadow
3361346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// memory layout changes. Unfortunately they do necessarily depend on
3371346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// kHighMemBeg or kHighMemEnd.
3383f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanystatic void *island_allocator_pos = 0;
3395b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov
3401346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#if __WORDSIZE == 32
3415b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandEnd (0xffdf0000 - kPageSize)
3425b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandBeg (kIslandEnd - 256 * kPageSize)
3431346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#else
3445b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandEnd (0x7fffffdf0000 - kPageSize)
3455b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandBeg (kIslandEnd - 256 * kPageSize)
3461346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#endif
3473281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko
3483281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkoextern "C"
3495b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonovmach_error_t __interception_allocate_island(void **ptr,
3505b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov                                            size_t unused_size,
3515b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov                                            void *unused_hint) {
3523281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  if (!island_allocator_pos) {
3531346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko    island_allocator_pos =
3541346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko        asan_mmap((void*)kIslandBeg, kIslandEnd - kIslandBeg,
3551346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko                  PROT_READ | PROT_WRITE | PROT_EXEC,
3561346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko                  MAP_PRIVATE | MAP_ANON | MAP_FIXED,
3571346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko                 -1, 0);
3581346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko    if (island_allocator_pos != (void*)kIslandBeg) {
3593281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko      return KERN_NO_SPACE;
3603281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko    }
361ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    if (FLAG_v) {
362ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko      Report("Mapped pages %p--%p for branch islands.\n",
363ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko             kIslandBeg, kIslandEnd);
364ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    }
365ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    // Should not be very performance-critical.
366ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    internal_memset(island_allocator_pos, 0xCC, kIslandEnd - kIslandBeg);
3673281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  };
3683281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  *ptr = island_allocator_pos;
3693281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  island_allocator_pos = (char*)island_allocator_pos + kPageSize;
370ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko  if (FLAG_v) {
371ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    Report("Branch island allocated at %p\n", *ptr);
372ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko  }
3733281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  return err_none;
3743281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko}
3753281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko
3763281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkoextern "C"
3775b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonovmach_error_t __interception_deallocate_island(void *ptr) {
3783281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  // Do nothing.
3793281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  // TODO(glider): allow to free and reuse the island memory.
3803281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  return err_none;
3813281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko}
382d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany
3831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Support for the following functions from libdispatch on Mac OS:
3841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_async_f()
3851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_async()
3861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_sync_f()
3871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_sync()
3881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_after_f()
3891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_after()
3901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_group_async_f()
3911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_group_async()
3921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): libdispatch API contains other functions that we don't support
3931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// yet.
3941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
3951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() and dispatch_sync_f() are synchronous, although chances are
3961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// they can cause jobs to run on a thread different from the current one.
3971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): if so, we need a test for this (otherwise we should remove
3981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// them).
3991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
4001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following functions use dispatch_barrier_async_f() (which isn't a library
4011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// function but is exported) and are thus supported:
4021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_source_set_cancel_handler_f()
4031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_source_set_cancel_handler()
4041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_source_set_event_handler_f()
4051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_source_set_event_handler()
4061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
4071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The reference manual for Grand Central Dispatch is available at
4081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   http://developer.apple.com/library/mac/#documentation/Performance/Reference/GCD_libdispatch_Ref/Reference/reference.html
4091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The implementation details are at
4101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   http://libdispatch.macosforge.org/trac/browser/trunk/src/queue.c
4111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4125cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef void* pthread_workqueue_t;
4135cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef void* pthread_workitem_handle_t;
414f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov
415f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef void* dispatch_group_t;
416f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef void* dispatch_queue_t;
417f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef uint64_t dispatch_time_t;
418f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef void (*dispatch_function_t)(void *block);
4195cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef void* (*worker_t)(void *block);
4205cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov
4215cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov// A wrapper for the ObjC blocks used to support libdispatch.
4225cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef struct {
4235cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  void *block;
4245cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  dispatch_function_t func;
4255cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  int parent_tid;
4265cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov} asan_block_context_t;
4275cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov
428f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// We use extern declarations of libdispatch functions here instead
429f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// of including <dispatch/dispatch.h>. This header is not present on
430f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// Mac OS X Leopard and eariler, and although we don't expect ASan to
431f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// work on legacy systems, it's bad to break the build of
432f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// LLVM compiler-rt there.
4335cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovextern "C" {
434f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_async_f(dispatch_queue_t dq, void *ctxt,
435f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov                      dispatch_function_t func);
436f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_sync_f(dispatch_queue_t dq, void *ctxt,
437f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov                     dispatch_function_t func);
438f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_after_f(dispatch_time_t when, dispatch_queue_t dq, void *ctxt,
439f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov                      dispatch_function_t func);
4405cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovvoid dispatch_barrier_async_f(dispatch_queue_t dq, void *ctxt,
4415cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov                              dispatch_function_t func);
442f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_group_async_f(dispatch_group_t group, dispatch_queue_t dq,
443f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov                            void *ctxt, dispatch_function_t func);
4445cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovint pthread_workqueue_additem_np(pthread_workqueue_t workq,
4455cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov    void *(*workitem_func)(void *), void * workitem_arg,
4465cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov    pthread_workitem_handle_t * itemhandlep, unsigned int *gencountp);
4475cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov}  // extern "C"
4485cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov
4491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C"
4501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid asan_dispatch_call_block_and_release(void *block) {
4519cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
4521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *context = (asan_block_context_t*)block;
4531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
4541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("asan_dispatch_call_block_and_release(): "
4551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany           "context: %p, pthread_self: %p\n",
4561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany           block, pthread_self());
4571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
4581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  AsanThread *t = asanThreadRegistry().GetCurrent();
459af3441580555ceed092170232cd5f2cc180f19f4Kostya Serebryany  if (!t) {
4603f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany    t = AsanThread::Create(context->parent_tid, 0, 0, &stack);
46155cdfc6c5af92560bc0623b5a0d70af71511c3c8Alexey Samsonov    asanThreadRegistry().RegisterThread(t);
46269eca73ac96688c8bfe1f23ee006af29c7858c40Kostya Serebryany    t->Init();
4631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    asanThreadRegistry().SetCurrent(t);
4641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
4651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // Call the original dispatcher for the block.
4661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  context->func(context->block);
4671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_free(context, &stack);
4681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
4691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}  // namespace __asan
4711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyusing namespace __asan;  // NOLINT
4731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Wrap |ctxt| and |func| into an asan_block_context_t.
4751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The caller retains control of the allocated context.
4761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C"
4771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyasan_block_context_t *alloc_asan_context(void *ctxt, dispatch_function_t func,
4781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany                                         AsanStackTrace *stack) {
4791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt =
4801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany      (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), stack);
4811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->block = ctxt;
4821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->func = func;
4831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne();
4841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return asan_ctxt;
4851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
4861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): can we reduce code duplication by introducing a macro?
488f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_async_f, dispatch_queue_t dq, void *ctxt,
489f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                    dispatch_function_t func) {
4909cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
4911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
4921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
4931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_async_f(): context: %p, pthread_self: %p\n",
4941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany        asan_ctxt, pthread_self());
4951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
4961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
49709672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(dispatch_async_f)(dq, (void*)asan_ctxt,
49809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                asan_dispatch_call_block_and_release);
4991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
501f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_sync_f, dispatch_queue_t dq, void *ctxt,
502f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                   dispatch_function_t func) {
5039cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
5051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_sync_f(): context: %p, pthread_self: %p\n",
5071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany        asan_ctxt, pthread_self());
5081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
51009672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(dispatch_sync_f)(dq, (void*)asan_ctxt,
51109672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                               asan_dispatch_call_block_and_release);
5121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
514f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_after_f, dispatch_time_t when,
515f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                    dispatch_queue_t dq, void *ctxt,
516f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                    dispatch_function_t func) {
5179cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
5191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_after_f: %p\n", asan_ctxt);
5211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
52309672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(dispatch_after_f)(when, dq, (void*)asan_ctxt,
52409672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                asan_dispatch_call_block_and_release);
5251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
527f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_barrier_async_f, dispatch_queue_t dq, void *ctxt,
528f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                            dispatch_function_t func) {
5299cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
5311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_barrier_async_f(): context: %p, pthread_self: %p\n",
5331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany           asan_ctxt, pthread_self());
5341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
53609672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  REAL(dispatch_barrier_async_f)(dq, (void*)asan_ctxt,
53709672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                 asan_dispatch_call_block_and_release);
5381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
540f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_group_async_f, dispatch_group_t group,
541f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                          dispatch_queue_t dq, void *ctxt,
542f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                          dispatch_function_t func) {
5439cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
5451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_group_async_f(): context: %p, pthread_self: %p\n",
5471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany           asan_ctxt, pthread_self());
5481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
55009672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  REAL(dispatch_group_async_f)(group, dq, (void*)asan_ctxt,
55109672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                               asan_dispatch_call_block_and_release);
5521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
5541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following stuff has been extremely helpful while looking for the
5551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// unhandled functions that spawned jobs on Chromium shutdown. If the verbosity
5561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// level is 2 or greater, we wrap pthread_workqueue_additem_np() in order to
5571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// find the points of worker thread creation (each of such threads may be used
5581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// to run several tasks, that's why this is not enough to support the whole
5591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// libdispatch API.
5601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C"
5611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *wrap_workitem_func(void *arg) {
5621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("wrap_workitem_func: %p, pthread_self: %p\n", arg, pthread_self());
5641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
5651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *ctxt = (asan_block_context_t*)arg;
5661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  worker_t fn = (worker_t)(ctxt->func);
5671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  void *result =  fn(ctxt->block);
5689cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_free(arg, &stack);
5701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return result;
5711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
573f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(int, pthread_workqueue_additem_np, pthread_workqueue_t workq,
5741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    void *(*workitem_func)(void *), void * workitem_arg,
5751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    pthread_workitem_handle_t * itemhandlep, unsigned int *gencountp) {
5769cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt =
5781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany      (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), &stack);
5791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->block = workitem_arg;
5801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->func = (dispatch_function_t)workitem_func;
5811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne();
5821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("pthread_workqueue_additem_np: %p\n", asan_ctxt);
5841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
58609672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(pthread_workqueue_additem_np)(workq, wrap_workitem_func,
58709672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                            asan_ctxt, itemhandlep,
58809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                            gencountp);
5891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
590d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany
591431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// CF_RC_BITS, the layout of CFRuntimeBase and __CFStrIsConstant are internal
592431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// and subject to change in further CoreFoundation versions. Apple does not
593431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// guarantee any binary compatibility from release to release.
594431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
595431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFInternal.h
596431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if defined(__BIG_ENDIAN__)
597431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#define CF_RC_BITS 0
598431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif
599431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
600431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if defined(__LITTLE_ENDIAN__)
601431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#define CF_RC_BITS 3
602431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif
603431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
604431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFRuntime.h
605431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkotypedef struct __CFRuntimeBase {
6063f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  uptr _cfisa;
607431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  uint8_t _cfinfo[4];
608431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if __LP64__
609431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  uint32_t _rc;
610431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif
611431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko} CFRuntimeBase;
612431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
613431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFString.c
614431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkoint __CFStrIsConstant(CFStringRef str) {
615431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  CFRuntimeBase *base = (CFRuntimeBase*)str;
616431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if __LP64__
617431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  return base->_rc == 0;
618431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#else
619431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  return (base->_cfinfo[CF_RC_BITS]) == 0;
620431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif
621431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko}
622431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
623f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(CFStringRef, CFStringCreateCopy, CFAllocatorRef alloc,
624f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                             CFStringRef str) {
625431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  if (__CFStrIsConstant(str)) {
626431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko    return str;
627431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  } else {
62809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov    return REAL(CFStringCreateCopy)(alloc, str);
629431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  }
630431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko}
631431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
63264ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonovnamespace __asan {
633beba6448539095b67cab266d09cd7b7d313b8c3dAlexey Samsonov
634beba6448539095b67cab266d09cd7b7d313b8c3dAlexey Samsonovvoid InitializeMacInterceptors() {
6355cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_async_f));
6365cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_sync_f));
6375cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_after_f));
6385cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_barrier_async_f));
6395cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_group_async_f));
6405cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  // We don't need to intercept pthread_workqueue_additem_np() to support the
6415cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  // libdispatch API, but it helps us to debug the unsupported functions. Let's
6425cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  // intercept it only during verbose runs.
6435cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  if (FLAG_v >= 2) {
6445cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov    CHECK(INTERCEPT_FUNCTION(pthread_workqueue_additem_np));
6455cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  }
64664ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // Normally CFStringCreateCopy should not copy constant CF strings.
64764ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // Replacing the default CFAllocator causes constant strings to be copied
64864ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // rather than just returned, which leads to bugs in big applications like
64964ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // Chromium and WebKit, see
65064ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // http://code.google.com/p/address-sanitizer/issues/detail?id=10
65164ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // Until this problem is fixed we need to check that the string is
65264ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // non-constant before calling CFStringCreateCopy.
65364ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(CFStringCreateCopy));
65464ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov}
655beba6448539095b67cab266d09cd7b7d313b8c3dAlexey Samsonov
65664ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov}  // namespace __asan
65764ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov
658d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany#endif  // __APPLE__
659