asan_mac.cc revision ae4d9caa4f47fa6abcd641719e9f520622940c17
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"
24ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov#include "sanitizer_common/sanitizer_libc.h"
251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
261e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko#include <crt_externs.h>  // for _NSGetEnviron
278a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#include <mach-o/dyld.h>
289b993e8cd0f8964782ee93524603d0c53adc2249Kostya Serebryany#include <mach-o/loader.h>
291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <sys/mman.h>
30ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany#include <sys/resource.h>
3159dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko#include <sys/sysctl.h>
329107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany#include <sys/ucontext.h>
33c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany#include <pthread.h>
34a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany#include <fcntl.h>
351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <unistd.h>
36d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany#include <libkern/OSAtomic.h>
3764ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov#include <CoreFoundation/CFString.h>
381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
39ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonovusing namespace __sanitizer;  // NOLINT
40ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov
411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanynamespace __asan {
421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
433f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanyvoid GetPcSpBp(void *context, uptr *pc, uptr *sp, uptr *bp) {
449107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  ucontext_t *ucontext = (ucontext_t*)context;
459107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# if __WORDSIZE == 64
469107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *pc = ucontext->uc_mcontext->__ss.__rip;
479107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *bp = ucontext->uc_mcontext->__ss.__rbp;
489107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *sp = ucontext->uc_mcontext->__ss.__rsp;
499107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# else
509107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *pc = ucontext->uc_mcontext->__ss.__eip;
519107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *bp = ucontext->uc_mcontext->__ss.__ebp;
529107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany  *sp = ucontext->uc_mcontext->__ss.__esp;
539107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# endif  // __WORDSIZE
549107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany}
559107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany
5638dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonovenum {
5738dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  MACOS_VERSION_UNKNOWN = 0,
5838dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  MACOS_VERSION_LEOPARD,
5938dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  MACOS_VERSION_SNOW_LEOPARD,
6038dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  MACOS_VERSION_LION,
6138dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov};
6238dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov
6338dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonovstatic int GetMacosVersion() {
6459dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  int mib[2] = { CTL_KERN, KERN_OSRELEASE };
6559dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  char version[100];
6659dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  size_t len = 0, maxlen = sizeof(version) / sizeof(version[0]);
6759dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  for (int i = 0; i < maxlen; i++) version[i] = '\0';
6859dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  // Get the version length.
693f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(sysctl(mib, 2, 0, &len, 0, 0) != -1);
7059dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  CHECK(len < maxlen);
713f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(sysctl(mib, 2, version, &len, 0, 0) != -1);
7259dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  switch (version[0]) {
7359dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko    case '9': return MACOS_VERSION_LEOPARD;
7459dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko    case '1': {
7559dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko      switch (version[1]) {
7659dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko        case '0': return MACOS_VERSION_SNOW_LEOPARD;
7759dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko        case '1': return MACOS_VERSION_LION;
7859dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko        default: return MACOS_VERSION_UNKNOWN;
7959dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko      }
8059dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko    }
8159dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko    default: return MACOS_VERSION_UNKNOWN;
8259dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko  }
8359dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko}
8459dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko
8538dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonovbool PlatformHasDifferentMemcpyAndMemmove() {
8638dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // On OS X 10.7 memcpy() and memmove() are both resolved
8738dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // into memmove$VARIANT$sse42.
8838dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // See also http://code.google.com/p/address-sanitizer/issues/detail?id=34.
8938dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // TODO(glider): need to check dynamically that memcpy() and memmove() are
9038dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  // actually the same function.
9138dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  return GetMacosVersion() == MACOS_VERSION_SNOW_LEOPARD;
9238dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov}
9338dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov
941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// No-op. Mac does not support static linkage anyway.
951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *AsanDoesNotSupportStaticLinkage() {
963f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  return 0;
971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
994803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryanybool AsanInterceptsSignal(int signum) {
1004803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  return (signum == SIGSEGV || signum == SIGBUS) && FLAG_handle_segv;
1014803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
1024803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
1030ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanWrite(int fd, const void *buf, size_t count) {
1041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return write(fd, buf, count);
1051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
1061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
107de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid *AsanMmapSomewhereOrDie(size_t size, const char *mem_type) {
108de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  size = RoundUpTo(size, kPageSize);
109ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov  void *res = internal_mmap(0, size,
110ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                            PROT_READ | PROT_WRITE,
111ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                            MAP_PRIVATE | MAP_ANON, -1, 0);
112de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  if (res == (void*)-1) {
113de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany    OutOfMemoryMessageAndDie(mem_type, size);
114de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  }
115de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  return res;
116de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
117de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
1183f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanyvoid *AsanMmapFixedNoReserve(uptr fixed_addr, size_t size) {
119ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov  return internal_mmap((void*)fixed_addr, size,
120ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                      PROT_READ | PROT_WRITE,
121ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                      MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE,
122ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                      0, 0);
123a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany}
124a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany
1253f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanyvoid *AsanMprotect(uptr fixed_addr, size_t size) {
126ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov  return internal_mmap((void*)fixed_addr, size,
127ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                       PROT_NONE,
128ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                       MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE,
129ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                       0, 0);
130a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany}
131a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany
132de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid AsanUnmapOrDie(void *addr, size_t size) {
133de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  if (!addr || !size) return;
134de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  int res = munmap(addr, size);
135de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  if (res != 0) {
136de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany    Report("Failed to unmap\n");
1370ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryany    AsanDie();
138de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  }
139de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
140de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
141de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanOpenReadonly(const char* filename) {
142de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  return open(filename, O_RDONLY);
143de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
144de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
1451e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenkoconst char *AsanGetEnv(const char *name) {
1461e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  char ***env_ptr = _NSGetEnviron();
1471e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  CHECK(env_ptr);
1481e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  char **environ = *env_ptr;
1491e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  CHECK(environ);
1501e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  size_t name_len = internal_strlen(name);
1513f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  while (*environ != 0) {
1521e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko    size_t len = internal_strlen(*environ);
1531e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko    if (len > name_len) {
1541e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko      const char *p = *environ;
1551e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko      if (!internal_memcmp(p, name, name_len) &&
1561e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko          p[name_len] == '=') {  // Match.
1574dd8ba8238b1b698953628affe6e5b2edf3b3e3fAlexey Samsonov        return *environ + name_len + 1;  // String starting after =.
1581e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko      }
1591e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko    }
1601e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko    environ++;
1611e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko  }
1623f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  return 0;
1631e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko}
1641e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko
1650ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanRead(int fd, void *buf, size_t count) {
166de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  return read(fd, buf, count);
167de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
168de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
169de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanClose(int fd) {
170de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany  return close(fd);
171de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany}
172de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany
1738a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander PotapenkoAsanProcMaps::AsanProcMaps() {
1748a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  Reset();
1758a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
1768a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
1778a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander PotapenkoAsanProcMaps::~AsanProcMaps() {
1788a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
1798a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
1803feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// More information about Mach-O headers can be found in mach-o/loader.h
1813feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Each Mach-O image has a header (mach_header or mach_header_64) starting with
1823feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// a magic number, and a list of linker load commands directly following the
1833feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// header.
1843feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// A load command is at least two 32-bit words: the command type and the
1853feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// command size in bytes. We're interested only in segment load commands
1863feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// (LC_SEGMENT and LC_SEGMENT_64), which tell that a part of the file is mapped
1873feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// into the task's address space.
1883feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// The |vmaddr|, |vmsize| and |fileoff| fields of segment_command or
1893feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// segment_command_64 correspond to the memory address, memory size and the
1903feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// file offset of the current memory segment.
1913feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Because these fields are taken from the images as is, one needs to add
1923feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// _dyld_get_image_vmaddr_slide() to get the actual addresses at runtime.
1933feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko
1948a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenkovoid AsanProcMaps::Reset() {
1958a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  // Count down from the top.
1968a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  // TODO(glider): as per man 3 dyld, iterating over the headers with
1973feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  // _dyld_image_count is thread-unsafe. We need to register callbacks for
1983feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  // adding and removing images which will invalidate the AsanProcMaps state.
1998a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  current_image_ = _dyld_image_count();
2003feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  current_load_cmd_count_ = -1;
2013f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  current_load_cmd_addr_ = 0;
202ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko  current_magic_ = 0;
2033feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko}
2043feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko
2053feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Next and NextSegmentLoad were inspired by base/sysinfo.cc in
2063feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Google Perftools, http://code.google.com/p/google-perftools.
2073feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko
2083feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// NextSegmentLoad scans the current image for the next segment load command
2093feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// and returns the start and end addresses and file offset of the corresponding
2103feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// segment.
2113feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Note that the segment addresses are not necessarily sorted.
212ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryanytemplate<u32 kLCSegment, typename SegmentCommand>
2133feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenkobool AsanProcMaps::NextSegmentLoad(
2143f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany    uptr *start, uptr *end, uptr *offset,
2153feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko    char filename[], size_t filename_size) {
2163feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  const char* lc = current_load_cmd_addr_;
2173feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko  current_load_cmd_addr_ += ((const load_command *)lc)->cmdsize;
2188a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  if (((const load_command *)lc)->cmd == kLCSegment) {
219ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryany    const sptr dlloff = _dyld_get_image_vmaddr_slide(current_image_);
2208a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    const SegmentCommand* sc = (const SegmentCommand *)lc;
2218a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (start) *start = sc->vmaddr + dlloff;
2228a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (end) *end = sc->vmaddr + sc->vmsize + dlloff;
2238a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (offset) *offset = sc->fileoff;
2248a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (filename) {
22509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov      REAL(strncpy)(filename, _dyld_get_image_name(current_image_),
22609672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                    filename_size);
2278a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    }
228ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko    if (FLAG_v >= 4)
229ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      Report("LC_SEGMENT: %p--%p %s+%p\n", *start, *end, filename, *offset);
2308a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    return true;
2318a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  }
2328a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  return false;
2338a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
2348a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
2353f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanybool AsanProcMaps::Next(uptr *start, uptr *end,
2363f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany                        uptr *offset, char filename[],
2378a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko                        size_t filename_size) {
2388a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  for (; current_image_ >= 0; current_image_--) {
2398a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    const mach_header* hdr = _dyld_get_image_header(current_image_);
2408a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    if (!hdr) continue;
2413feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko    if (current_load_cmd_count_ < 0) {
2423feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko      // Set up for this image;
2433feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko      current_load_cmd_count_ = hdr->ncmds;
244ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      current_magic_ = hdr->magic;
245ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      switch (current_magic_) {
2463feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko#ifdef MH_MAGIC_64
2473feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        case MH_MAGIC_64: {
2483feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          current_load_cmd_addr_ = (char*)hdr + sizeof(mach_header_64);
2493feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          break;
2503feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        }
2513feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko#endif
2523feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        case MH_MAGIC: {
2533feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          current_load_cmd_addr_ = (char*)hdr + sizeof(mach_header);
2543feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          break;
2553feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        }
2563feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        default: {
2573feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko          continue;
2583feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko        }
2593feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko      }
2603feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko    }
2618a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
262ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko    for (; current_load_cmd_count_ >= 0; current_load_cmd_count_--) {
263ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      switch (current_magic_) {
264ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        // current_magic_ may be only one of MH_MAGIC, MH_MAGIC_64.
2658a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#ifdef MH_MAGIC_64
266ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        case MH_MAGIC_64: {
267ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko          if (NextSegmentLoad<LC_SEGMENT_64, struct segment_command_64>(
268ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko                  start, end, offset, filename, filename_size))
269ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko            return true;
270ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko          break;
271ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        }
2728a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#endif
273ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        case MH_MAGIC: {
274ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko          if (NextSegmentLoad<LC_SEGMENT, struct segment_command>(
275ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko                  start, end, offset, filename, filename_size))
276ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko            return true;
277ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko          break;
278ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko        }
279ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko      }
2808a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    }
2818a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    // If we get here, no more load_cmd's in this image talk about
2828a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko    // segments.  Go on to the next image.
2838a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  }
2848a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  return false;
2858a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
2868a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
2873f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanybool AsanProcMaps::GetObjectNameAndOffset(uptr addr, uptr *offset,
2888a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko                                          char filename[],
2898a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko                                          size_t filename_size) {
2908a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko  return IterateForObjectNameAndOffset(addr, offset, filename, filename_size);
2918a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko}
2928a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko
293c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryanyvoid AsanThread::SetThreadStackTopAndBottom() {
294c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany  size_t stacksize = pthread_get_stacksize_np(pthread_self());
295c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany  void *stackaddr = pthread_get_stackaddr_np(pthread_self());
2963f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  stack_top_ = (uptr)stackaddr;
297c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany  stack_bottom_ = stack_top_ - stacksize;
298c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany  int local;
2993f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(AddrIsInStack((uptr)&local));
300c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany}
301c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany
302d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya SerebryanyAsanLock::AsanLock(LinkerInitialized) {
303d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  // We assume that OS_SPINLOCK_INIT is zero
304d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany}
305d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany
306d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryanyvoid AsanLock::Lock() {
307d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  CHECK(sizeof(OSSpinLock) <= sizeof(opaque_storage_));
308d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  CHECK(OS_SPINLOCK_INIT == 0);
3093f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(owner_ != (uptr)pthread_self());
310d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  OSSpinLockLock((OSSpinLock*)&opaque_storage_);
311d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  CHECK(!owner_);
3123f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  owner_ = (uptr)pthread_self();
313d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany}
314d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany
315d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryanyvoid AsanLock::Unlock() {
3163f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(owner_ == (uptr)pthread_self());
317d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  owner_ = 0;
318d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany  OSSpinLockUnlock((OSSpinLock*)&opaque_storage_);
319d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany}
320d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany
3213f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanyvoid AsanStackTrace::GetStackTrace(size_t max_s, uptr pc, uptr bp) {
3229cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  size = 0;
3239cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  trace[0] = pc;
3249cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  if ((max_s) > 1) {
3259cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov    max_size = max_s;
3269cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov    FastUnwindStack(pc, bp);
3279cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  }
3289cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov}
3299cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov
3305b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov// The range of pages to be used for escape islands.
3313281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// TODO(glider): instead of mapping a fixed range we must find a range of
3323281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// unmapped pages in vmmap and take them.
3331346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// These constants were chosen empirically and may not work if the shadow
3341346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// memory layout changes. Unfortunately they do necessarily depend on
3351346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// kHighMemBeg or kHighMemEnd.
3363f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanystatic void *island_allocator_pos = 0;
3375b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov
3381346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#if __WORDSIZE == 32
3395b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandEnd (0xffdf0000 - kPageSize)
3405b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandBeg (kIslandEnd - 256 * kPageSize)
3411346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#else
3425b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandEnd (0x7fffffdf0000 - kPageSize)
3435b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandBeg (kIslandEnd - 256 * kPageSize)
3441346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#endif
3453281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko
3463281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkoextern "C"
3475b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonovmach_error_t __interception_allocate_island(void **ptr,
3485b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov                                            size_t unused_size,
3495b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov                                            void *unused_hint) {
3503281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  if (!island_allocator_pos) {
3511346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko    island_allocator_pos =
352ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov        internal_mmap((void*)kIslandBeg, kIslandEnd - kIslandBeg,
353ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                      PROT_READ | PROT_WRITE | PROT_EXEC,
354ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                      MAP_PRIVATE | MAP_ANON | MAP_FIXED,
355ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov                      -1, 0);
3561346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko    if (island_allocator_pos != (void*)kIslandBeg) {
3573281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko      return KERN_NO_SPACE;
3583281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko    }
359ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    if (FLAG_v) {
360ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko      Report("Mapped pages %p--%p for branch islands.\n",
361ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko             kIslandBeg, kIslandEnd);
362ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    }
363ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    // Should not be very performance-critical.
364ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    internal_memset(island_allocator_pos, 0xCC, kIslandEnd - kIslandBeg);
3653281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  };
3663281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  *ptr = island_allocator_pos;
3673281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  island_allocator_pos = (char*)island_allocator_pos + kPageSize;
368ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko  if (FLAG_v) {
369ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko    Report("Branch island allocated at %p\n", *ptr);
370ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko  }
3713281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  return err_none;
3723281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko}
3733281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko
3743281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkoextern "C"
3755b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonovmach_error_t __interception_deallocate_island(void *ptr) {
3763281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  // Do nothing.
3773281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  // TODO(glider): allow to free and reuse the island memory.
3783281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko  return err_none;
3793281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko}
380d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany
3811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Support for the following functions from libdispatch on Mac OS:
3821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_async_f()
3831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_async()
3841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_sync_f()
3851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_sync()
3861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_after_f()
3871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_after()
3881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_group_async_f()
3891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_group_async()
3901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): libdispatch API contains other functions that we don't support
3911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// yet.
3921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
3931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() and dispatch_sync_f() are synchronous, although chances are
3941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// they can cause jobs to run on a thread different from the current one.
3951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): if so, we need a test for this (otherwise we should remove
3961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// them).
3971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
3981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following functions use dispatch_barrier_async_f() (which isn't a library
3991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// function but is exported) and are thus supported:
4001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_source_set_cancel_handler_f()
4011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_source_set_cancel_handler()
4021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_source_set_event_handler_f()
4031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   dispatch_source_set_event_handler()
4041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
4051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The reference manual for Grand Central Dispatch is available at
4061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   http://developer.apple.com/library/mac/#documentation/Performance/Reference/GCD_libdispatch_Ref/Reference/reference.html
4071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The implementation details are at
4081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//   http://libdispatch.macosforge.org/trac/browser/trunk/src/queue.c
4091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4105cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef void* pthread_workqueue_t;
4115cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef void* pthread_workitem_handle_t;
412f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov
413f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef void* dispatch_group_t;
414f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef void* dispatch_queue_t;
415ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryanytypedef u64 dispatch_time_t;
416f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef void (*dispatch_function_t)(void *block);
4175cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef void* (*worker_t)(void *block);
4185cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov
4195cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov// A wrapper for the ObjC blocks used to support libdispatch.
4205cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef struct {
4215cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  void *block;
4225cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  dispatch_function_t func;
4235cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  int parent_tid;
4245cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov} asan_block_context_t;
4255cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov
426f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// We use extern declarations of libdispatch functions here instead
427f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// of including <dispatch/dispatch.h>. This header is not present on
428f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// Mac OS X Leopard and eariler, and although we don't expect ASan to
429f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// work on legacy systems, it's bad to break the build of
430f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// LLVM compiler-rt there.
4315cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovextern "C" {
432f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_async_f(dispatch_queue_t dq, void *ctxt,
433f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov                      dispatch_function_t func);
434f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_sync_f(dispatch_queue_t dq, void *ctxt,
435f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov                     dispatch_function_t func);
436f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_after_f(dispatch_time_t when, dispatch_queue_t dq, void *ctxt,
437f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov                      dispatch_function_t func);
4385cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovvoid dispatch_barrier_async_f(dispatch_queue_t dq, void *ctxt,
4395cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov                              dispatch_function_t func);
440f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_group_async_f(dispatch_group_t group, dispatch_queue_t dq,
441f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov                            void *ctxt, dispatch_function_t func);
4425cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovint pthread_workqueue_additem_np(pthread_workqueue_t workq,
4435cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov    void *(*workitem_func)(void *), void * workitem_arg,
4445cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov    pthread_workitem_handle_t * itemhandlep, unsigned int *gencountp);
4455cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov}  // extern "C"
4465cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov
4471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C"
4481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid asan_dispatch_call_block_and_release(void *block) {
4499cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
4501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *context = (asan_block_context_t*)block;
4511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
4521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("asan_dispatch_call_block_and_release(): "
4531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany           "context: %p, pthread_self: %p\n",
4541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany           block, pthread_self());
4551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
4561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  AsanThread *t = asanThreadRegistry().GetCurrent();
457af3441580555ceed092170232cd5f2cc180f19f4Kostya Serebryany  if (!t) {
4583f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany    t = AsanThread::Create(context->parent_tid, 0, 0, &stack);
45955cdfc6c5af92560bc0623b5a0d70af71511c3c8Alexey Samsonov    asanThreadRegistry().RegisterThread(t);
46069eca73ac96688c8bfe1f23ee006af29c7858c40Kostya Serebryany    t->Init();
4611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    asanThreadRegistry().SetCurrent(t);
4621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
4631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // Call the original dispatcher for the block.
4641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  context->func(context->block);
4651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_free(context, &stack);
4661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
4671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}  // namespace __asan
4691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyusing namespace __asan;  // NOLINT
4711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Wrap |ctxt| and |func| into an asan_block_context_t.
4731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The caller retains control of the allocated context.
4741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C"
4751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyasan_block_context_t *alloc_asan_context(void *ctxt, dispatch_function_t func,
4761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany                                         AsanStackTrace *stack) {
4771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt =
4781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany      (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), stack);
4791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->block = ctxt;
4801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->func = func;
4811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne();
4821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return asan_ctxt;
4831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
4841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): can we reduce code duplication by introducing a macro?
486f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_async_f, dispatch_queue_t dq, void *ctxt,
487f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                    dispatch_function_t func) {
4889cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
4891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
4901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
4911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_async_f(): context: %p, pthread_self: %p\n",
4921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany        asan_ctxt, pthread_self());
4931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
4941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
49509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(dispatch_async_f)(dq, (void*)asan_ctxt,
49609672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                asan_dispatch_call_block_and_release);
4971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
4981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
499f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_sync_f, dispatch_queue_t dq, void *ctxt,
500f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                   dispatch_function_t func) {
5019cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
5031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_sync_f(): context: %p, pthread_self: %p\n",
5051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany        asan_ctxt, pthread_self());
5061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
50809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(dispatch_sync_f)(dq, (void*)asan_ctxt,
50909672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                               asan_dispatch_call_block_and_release);
5101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
512f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_after_f, dispatch_time_t when,
513f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                    dispatch_queue_t dq, void *ctxt,
514f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                    dispatch_function_t func) {
5159cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
5171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_after_f: %p\n", asan_ctxt);
5191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
52109672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(dispatch_after_f)(when, dq, (void*)asan_ctxt,
52209672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                asan_dispatch_call_block_and_release);
5231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
525f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_barrier_async_f, dispatch_queue_t dq, void *ctxt,
526f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                            dispatch_function_t func) {
5279cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
5291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_barrier_async_f(): context: %p, pthread_self: %p\n",
5311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany           asan_ctxt, pthread_self());
5321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
53409672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  REAL(dispatch_barrier_async_f)(dq, (void*)asan_ctxt,
53509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                 asan_dispatch_call_block_and_release);
5361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
538f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_group_async_f, dispatch_group_t group,
539f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                          dispatch_queue_t dq, void *ctxt,
540f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                          dispatch_function_t func) {
5419cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack);
5431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("dispatch_group_async_f(): context: %p, pthread_self: %p\n",
5451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany           asan_ctxt, pthread_self());
5461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
54809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  REAL(dispatch_group_async_f)(group, dq, (void*)asan_ctxt,
54909672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                               asan_dispatch_call_block_and_release);
5501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
5521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following stuff has been extremely helpful while looking for the
5531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// unhandled functions that spawned jobs on Chromium shutdown. If the verbosity
5541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// level is 2 or greater, we wrap pthread_workqueue_additem_np() in order to
5551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// find the points of worker thread creation (each of such threads may be used
5561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// to run several tasks, that's why this is not enough to support the whole
5571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// libdispatch API.
5581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C"
5591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *wrap_workitem_func(void *arg) {
5601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("wrap_workitem_func: %p, pthread_self: %p\n", arg, pthread_self());
5621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
5631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *ctxt = (asan_block_context_t*)arg;
5641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  worker_t fn = (worker_t)(ctxt->func);
5651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  void *result =  fn(ctxt->block);
5669cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_free(arg, &stack);
5681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return result;
5691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
571f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(int, pthread_workqueue_additem_np, pthread_workqueue_t workq,
5721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    void *(*workitem_func)(void *), void * workitem_arg,
5731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    pthread_workitem_handle_t * itemhandlep, unsigned int *gencountp) {
5749cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov  GET_STACK_TRACE_HERE(kStackTraceMax);
5751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_block_context_t *asan_ctxt =
5761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany      (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), &stack);
5771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->block = workitem_arg;
5781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->func = (dispatch_function_t)workitem_func;
5791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne();
5801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (FLAG_v >= 2) {
5811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    Report("pthread_workqueue_additem_np: %p\n", asan_ctxt);
5821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    PRINT_CURRENT_STACK();
5831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
58409672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(pthread_workqueue_additem_np)(workq, wrap_workitem_func,
58509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                            asan_ctxt, itemhandlep,
58609672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov                                            gencountp);
5871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
588d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany
589431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// CF_RC_BITS, the layout of CFRuntimeBase and __CFStrIsConstant are internal
590431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// and subject to change in further CoreFoundation versions. Apple does not
591431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// guarantee any binary compatibility from release to release.
592431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
593431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFInternal.h
594431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if defined(__BIG_ENDIAN__)
595431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#define CF_RC_BITS 0
596431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif
597431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
598431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if defined(__LITTLE_ENDIAN__)
599431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#define CF_RC_BITS 3
600431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif
601431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
602431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFRuntime.h
603431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkotypedef struct __CFRuntimeBase {
6043f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  uptr _cfisa;
605ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryany  u8 _cfinfo[4];
606431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if __LP64__
607ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryany  u32 _rc;
608431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif
609431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko} CFRuntimeBase;
610431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
611431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFString.c
612431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkoint __CFStrIsConstant(CFStringRef str) {
613431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  CFRuntimeBase *base = (CFRuntimeBase*)str;
614431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if __LP64__
615431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  return base->_rc == 0;
616431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#else
617431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  return (base->_cfinfo[CF_RC_BITS]) == 0;
618431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif
619431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko}
620431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
621f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(CFStringRef, CFStringCreateCopy, CFAllocatorRef alloc,
622f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov                                             CFStringRef str) {
623431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  if (__CFStrIsConstant(str)) {
624431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko    return str;
625431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  } else {
62609672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov    return REAL(CFStringCreateCopy)(alloc, str);
627431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko  }
628431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko}
629431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko
63064ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonovnamespace __asan {
631beba6448539095b67cab266d09cd7b7d313b8c3dAlexey Samsonov
632beba6448539095b67cab266d09cd7b7d313b8c3dAlexey Samsonovvoid InitializeMacInterceptors() {
6335cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_async_f));
6345cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_sync_f));
6355cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_after_f));
6365cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_barrier_async_f));
6375cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(dispatch_group_async_f));
6385cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  // We don't need to intercept pthread_workqueue_additem_np() to support the
6395cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  // libdispatch API, but it helps us to debug the unsupported functions. Let's
6405cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  // intercept it only during verbose runs.
6415cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  if (FLAG_v >= 2) {
6425cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov    CHECK(INTERCEPT_FUNCTION(pthread_workqueue_additem_np));
6435cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov  }
64464ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // Normally CFStringCreateCopy should not copy constant CF strings.
64564ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // Replacing the default CFAllocator causes constant strings to be copied
64664ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // rather than just returned, which leads to bugs in big applications like
64764ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // Chromium and WebKit, see
64864ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // http://code.google.com/p/address-sanitizer/issues/detail?id=10
64964ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // Until this problem is fixed we need to check that the string is
65064ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  // non-constant before calling CFStringCreateCopy.
65164ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov  CHECK(INTERCEPT_FUNCTION(CFStringCreateCopy));
65264ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov}
653beba6448539095b67cab266d09cd7b7d313b8c3dAlexey Samsonov
65464ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov}  // namespace __asan
65564ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov
656d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany#endif  // __APPLE__
657