asan_mac.cc revision e0cff0bc20ae51790c8edfbceb817e18ebf5355e
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 391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanynamespace __asan { 401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 413f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanyvoid GetPcSpBp(void *context, uptr *pc, uptr *sp, uptr *bp) { 429107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany ucontext_t *ucontext = (ucontext_t*)context; 439107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# if __WORDSIZE == 64 449107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *pc = ucontext->uc_mcontext->__ss.__rip; 459107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *bp = ucontext->uc_mcontext->__ss.__rbp; 469107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *sp = ucontext->uc_mcontext->__ss.__rsp; 479107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# else 489107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *pc = ucontext->uc_mcontext->__ss.__eip; 499107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *bp = ucontext->uc_mcontext->__ss.__ebp; 509107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *sp = ucontext->uc_mcontext->__ss.__esp; 519107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# endif // __WORDSIZE 529107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany} 539107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany 5438dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonovenum { 5538dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov MACOS_VERSION_UNKNOWN = 0, 5638dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov MACOS_VERSION_LEOPARD, 5738dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov MACOS_VERSION_SNOW_LEOPARD, 5838dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov MACOS_VERSION_LION, 5938dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov}; 6038dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov 6138dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonovstatic int GetMacosVersion() { 6259dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko int mib[2] = { CTL_KERN, KERN_OSRELEASE }; 6359dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko char version[100]; 643f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonov uptr len = 0, maxlen = sizeof(version) / sizeof(version[0]); 6559dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko for (int i = 0; i < maxlen; i++) version[i] = '\0'; 6659dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko // Get the version length. 673f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany CHECK(sysctl(mib, 2, 0, &len, 0, 0) != -1); 6859dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko CHECK(len < maxlen); 693f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany CHECK(sysctl(mib, 2, version, &len, 0, 0) != -1); 7059dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko switch (version[0]) { 7159dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko case '9': return MACOS_VERSION_LEOPARD; 7259dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko case '1': { 7359dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko switch (version[1]) { 7459dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko case '0': return MACOS_VERSION_SNOW_LEOPARD; 7559dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko case '1': return MACOS_VERSION_LION; 7659dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko default: return MACOS_VERSION_UNKNOWN; 7759dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko } 7859dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko } 7959dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko default: return MACOS_VERSION_UNKNOWN; 8059dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko } 8159dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko} 8259dc578df0de177b44c8c78f69d73735e38e5c14Alexander Potapenko 8338dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonovbool PlatformHasDifferentMemcpyAndMemmove() { 8438dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov // On OS X 10.7 memcpy() and memmove() are both resolved 8538dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov // into memmove$VARIANT$sse42. 8638dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov // See also http://code.google.com/p/address-sanitizer/issues/detail?id=34. 8738dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov // TODO(glider): need to check dynamically that memcpy() and memmove() are 8838dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov // actually the same function. 8938dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov return GetMacosVersion() == MACOS_VERSION_SNOW_LEOPARD; 9038dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov} 9138dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov 921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// No-op. Mac does not support static linkage anyway. 931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *AsanDoesNotSupportStaticLinkage() { 943f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany return 0; 951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 974803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryanybool AsanInterceptsSignal(int signum) { 984803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany return (signum == SIGSEGV || signum == SIGBUS) && FLAG_handle_segv; 994803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany} 1004803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany 1013f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonovvoid *AsanMmapSomewhereOrDie(uptr size, const char *mem_type) { 102de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany size = RoundUpTo(size, kPageSize); 103ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov void *res = internal_mmap(0, size, 104ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov PROT_READ | PROT_WRITE, 105ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov MAP_PRIVATE | MAP_ANON, -1, 0); 106de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany if (res == (void*)-1) { 107de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany OutOfMemoryMessageAndDie(mem_type, size); 108de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany } 109de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return res; 110de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 111de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 1123f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonovvoid *AsanMmapFixedNoReserve(uptr fixed_addr, uptr size) { 113ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov return internal_mmap((void*)fixed_addr, size, 114ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov PROT_READ | PROT_WRITE, 115ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE, 116ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov 0, 0); 117a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany} 118a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 1193f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonovvoid *AsanMprotect(uptr fixed_addr, uptr size) { 120ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov return internal_mmap((void*)fixed_addr, size, 121ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov PROT_NONE, 122ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE, 123ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov 0, 0); 124a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany} 125a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 1263f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonovvoid AsanUnmapOrDie(void *addr, uptr size) { 127de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany if (!addr || !size) return; 1281f11d31faa5ed89b74f7d543b1182fe8de198be5Alexey Samsonov int res = internal_munmap(addr, size); 129de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany if (res != 0) { 130de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany Report("Failed to unmap\n"); 13147657ce6cbac2fa93d0fd765c5d2872443b50e87Alexey Samsonov Die(); 132de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany } 133de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 134de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 1351e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenkoconst char *AsanGetEnv(const char *name) { 1361e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko char ***env_ptr = _NSGetEnviron(); 1371e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko CHECK(env_ptr); 1381e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko char **environ = *env_ptr; 1391e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko CHECK(environ); 1403f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonov uptr name_len = internal_strlen(name); 1413f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany while (*environ != 0) { 1423f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonov uptr len = internal_strlen(*environ); 1431e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko if (len > name_len) { 1441e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko const char *p = *environ; 1451e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko if (!internal_memcmp(p, name, name_len) && 1461e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko p[name_len] == '=') { // Match. 1474dd8ba8238b1b698953628affe6e5b2edf3b3e3fAlexey Samsonov return *environ + name_len + 1; // String starting after =. 1481e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko } 1491e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko } 1501e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko environ++; 1511e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko } 1523f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany return 0; 1531e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko} 1541e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko 1558a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander PotapenkoAsanProcMaps::AsanProcMaps() { 1568a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko Reset(); 1578a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 1588a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 1598a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander PotapenkoAsanProcMaps::~AsanProcMaps() { 1608a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 1618a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 1623feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// More information about Mach-O headers can be found in mach-o/loader.h 1633feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Each Mach-O image has a header (mach_header or mach_header_64) starting with 1643feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// a magic number, and a list of linker load commands directly following the 1653feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// header. 1663feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// A load command is at least two 32-bit words: the command type and the 1673feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// command size in bytes. We're interested only in segment load commands 1683feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// (LC_SEGMENT and LC_SEGMENT_64), which tell that a part of the file is mapped 1693feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// into the task's address space. 1703feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// The |vmaddr|, |vmsize| and |fileoff| fields of segment_command or 1713feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// segment_command_64 correspond to the memory address, memory size and the 1723feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// file offset of the current memory segment. 1733feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Because these fields are taken from the images as is, one needs to add 1743feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// _dyld_get_image_vmaddr_slide() to get the actual addresses at runtime. 1753feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko 1768a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenkovoid AsanProcMaps::Reset() { 1778a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko // Count down from the top. 1788a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko // TODO(glider): as per man 3 dyld, iterating over the headers with 1793feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko // _dyld_image_count is thread-unsafe. We need to register callbacks for 1803feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko // adding and removing images which will invalidate the AsanProcMaps state. 1818a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko current_image_ = _dyld_image_count(); 1823feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_count_ = -1; 1833f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany current_load_cmd_addr_ = 0; 184ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko current_magic_ = 0; 1853feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko} 1863feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko 1873feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Next and NextSegmentLoad were inspired by base/sysinfo.cc in 1883feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Google Perftools, http://code.google.com/p/google-perftools. 1893feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko 1903feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// NextSegmentLoad scans the current image for the next segment load command 1913feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// and returns the start and end addresses and file offset of the corresponding 1923feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// segment. 1933feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Note that the segment addresses are not necessarily sorted. 194ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryanytemplate<u32 kLCSegment, typename SegmentCommand> 1953feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenkobool AsanProcMaps::NextSegmentLoad( 1963f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany uptr *start, uptr *end, uptr *offset, 1973f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonov char filename[], uptr filename_size) { 1983feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko const char* lc = current_load_cmd_addr_; 1993feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_addr_ += ((const load_command *)lc)->cmdsize; 2008a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (((const load_command *)lc)->cmd == kLCSegment) { 201ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryany const sptr dlloff = _dyld_get_image_vmaddr_slide(current_image_); 2028a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko const SegmentCommand* sc = (const SegmentCommand *)lc; 2038a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (start) *start = sc->vmaddr + dlloff; 2048a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (end) *end = sc->vmaddr + sc->vmsize + dlloff; 2058a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (offset) *offset = sc->fileoff; 2068a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (filename) { 20709672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov REAL(strncpy)(filename, _dyld_get_image_name(current_image_), 20809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov filename_size); 2098a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko } 210ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko if (FLAG_v >= 4) 211675293d458c9f6c797a7211c11438eb9bfcfe9bbAlexey Samsonov Report("LC_SEGMENT: %p--%p %s+%p\n", (void*)*start, (void*)*end, 212675293d458c9f6c797a7211c11438eb9bfcfe9bbAlexey Samsonov filename, (void*)*offset); 2138a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return true; 2148a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko } 2158a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return false; 2168a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 2178a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 2183f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanybool AsanProcMaps::Next(uptr *start, uptr *end, 2193f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany uptr *offset, char filename[], 2203f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonov uptr filename_size) { 2218a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko for (; current_image_ >= 0; current_image_--) { 2228a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko const mach_header* hdr = _dyld_get_image_header(current_image_); 2238a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (!hdr) continue; 2243feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko if (current_load_cmd_count_ < 0) { 2253feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko // Set up for this image; 2263feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_count_ = hdr->ncmds; 227ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko current_magic_ = hdr->magic; 228ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko switch (current_magic_) { 2293feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko#ifdef MH_MAGIC_64 2303feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko case MH_MAGIC_64: { 2313feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_addr_ = (char*)hdr + sizeof(mach_header_64); 2323feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko break; 2333feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2343feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko#endif 2353feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko case MH_MAGIC: { 2363feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_addr_ = (char*)hdr + sizeof(mach_header); 2373feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko break; 2383feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2393feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko default: { 2403feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko continue; 2413feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2423feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2433feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2448a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 245ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko for (; current_load_cmd_count_ >= 0; current_load_cmd_count_--) { 246ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko switch (current_magic_) { 247ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko // current_magic_ may be only one of MH_MAGIC, MH_MAGIC_64. 2488a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#ifdef MH_MAGIC_64 249ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko case MH_MAGIC_64: { 250ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko if (NextSegmentLoad<LC_SEGMENT_64, struct segment_command_64>( 251ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko start, end, offset, filename, filename_size)) 252ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko return true; 253ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko break; 254ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko } 2558a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#endif 256ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko case MH_MAGIC: { 257ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko if (NextSegmentLoad<LC_SEGMENT, struct segment_command>( 258ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko start, end, offset, filename, filename_size)) 259ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko return true; 260ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko break; 261ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko } 262ddf50a3e7a28c2ab12243e72b35ccf8c04f15b49Alexander Potapenko } 2638a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko } 2648a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko // If we get here, no more load_cmd's in this image talk about 2658a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko // segments. Go on to the next image. 2668a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko } 2678a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return false; 2688a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 2698a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 2703f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanybool AsanProcMaps::GetObjectNameAndOffset(uptr addr, uptr *offset, 2718a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko char filename[], 2723f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonov uptr filename_size) { 2738a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return IterateForObjectNameAndOffset(addr, offset, filename, filename_size); 2748a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 2758a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 276c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryanyvoid AsanThread::SetThreadStackTopAndBottom() { 2773f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonov uptr stacksize = pthread_get_stacksize_np(pthread_self()); 278c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany void *stackaddr = pthread_get_stackaddr_np(pthread_self()); 2793f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany stack_top_ = (uptr)stackaddr; 280c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany stack_bottom_ = stack_top_ - stacksize; 281c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany int local; 2823f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany CHECK(AddrIsInStack((uptr)&local)); 283c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany} 284c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany 285d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya SerebryanyAsanLock::AsanLock(LinkerInitialized) { 286d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany // We assume that OS_SPINLOCK_INIT is zero 287d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany} 288d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany 289d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryanyvoid AsanLock::Lock() { 290d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany CHECK(sizeof(OSSpinLock) <= sizeof(opaque_storage_)); 291d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany CHECK(OS_SPINLOCK_INIT == 0); 2923f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany CHECK(owner_ != (uptr)pthread_self()); 293d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany OSSpinLockLock((OSSpinLock*)&opaque_storage_); 294d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany CHECK(!owner_); 2953f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany owner_ = (uptr)pthread_self(); 296d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany} 297d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany 298d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryanyvoid AsanLock::Unlock() { 2993f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany CHECK(owner_ == (uptr)pthread_self()); 300d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany owner_ = 0; 301d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany OSSpinLockUnlock((OSSpinLock*)&opaque_storage_); 302d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany} 303d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany 3043f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonovvoid AsanStackTrace::GetStackTrace(uptr max_s, uptr pc, uptr bp) { 3059cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov size = 0; 3069cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov trace[0] = pc; 3079cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov if ((max_s) > 1) { 3089cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov max_size = max_s; 3099cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov FastUnwindStack(pc, bp); 3109cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov } 3119cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov} 3129cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov 3135b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov// The range of pages to be used for escape islands. 3143281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// TODO(glider): instead of mapping a fixed range we must find a range of 3153281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// unmapped pages in vmmap and take them. 3161346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// These constants were chosen empirically and may not work if the shadow 3171346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// memory layout changes. Unfortunately they do necessarily depend on 3181346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// kHighMemBeg or kHighMemEnd. 3193f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanystatic void *island_allocator_pos = 0; 3205b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov 3211346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#if __WORDSIZE == 32 3225b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandEnd (0xffdf0000 - kPageSize) 3235b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandBeg (kIslandEnd - 256 * kPageSize) 3241346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#else 3255b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandEnd (0x7fffffdf0000 - kPageSize) 3265b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov# define kIslandBeg (kIslandEnd - 256 * kPageSize) 3271346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#endif 3283281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko 3293281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkoextern "C" 3305b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonovmach_error_t __interception_allocate_island(void **ptr, 3313f46cf42b0f8e0667a3bea88cf871ebd4dc0ecdfAlexey Samsonov uptr unused_size, 3325b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov void *unused_hint) { 3333281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko if (!island_allocator_pos) { 3341346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko island_allocator_pos = 335ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov internal_mmap((void*)kIslandBeg, kIslandEnd - kIslandBeg, 336ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov PROT_READ | PROT_WRITE | PROT_EXEC, 337ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov MAP_PRIVATE | MAP_ANON | MAP_FIXED, 338ae4d9caa4f47fa6abcd641719e9f520622940c17Alexey Samsonov -1, 0); 3391346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko if (island_allocator_pos != (void*)kIslandBeg) { 3403281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko return KERN_NO_SPACE; 3413281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko } 342ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko if (FLAG_v) { 343ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko Report("Mapped pages %p--%p for branch islands.\n", 344675293d458c9f6c797a7211c11438eb9bfcfe9bbAlexey Samsonov (void*)kIslandBeg, (void*)kIslandEnd); 345ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko } 346ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko // Should not be very performance-critical. 347ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko internal_memset(island_allocator_pos, 0xCC, kIslandEnd - kIslandBeg); 3483281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko }; 3493281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko *ptr = island_allocator_pos; 3503281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko island_allocator_pos = (char*)island_allocator_pos + kPageSize; 351ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko if (FLAG_v) { 352ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko Report("Branch island allocated at %p\n", *ptr); 353ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko } 3543281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko return err_none; 3553281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko} 3563281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko 3573281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkoextern "C" 3585b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonovmach_error_t __interception_deallocate_island(void *ptr) { 3593281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko // Do nothing. 3603281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko // TODO(glider): allow to free and reuse the island memory. 3613281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko return err_none; 3623281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko} 363d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany 3641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Support for the following functions from libdispatch on Mac OS: 3651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_async_f() 3661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_async() 3671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync_f() 3681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() 3691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_after_f() 3701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_after() 3711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_group_async_f() 3721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_group_async() 3731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): libdispatch API contains other functions that we don't support 3741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// yet. 3751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 3761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() and dispatch_sync_f() are synchronous, although chances are 3771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// they can cause jobs to run on a thread different from the current one. 3781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): if so, we need a test for this (otherwise we should remove 3791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// them). 3801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 3811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following functions use dispatch_barrier_async_f() (which isn't a library 3821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// function but is exported) and are thus supported: 3831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_cancel_handler_f() 3841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_cancel_handler() 3851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_event_handler_f() 3861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_event_handler() 3871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 3881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The reference manual for Grand Central Dispatch is available at 3891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// http://developer.apple.com/library/mac/#documentation/Performance/Reference/GCD_libdispatch_Ref/Reference/reference.html 3901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The implementation details are at 3911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// http://libdispatch.macosforge.org/trac/browser/trunk/src/queue.c 3921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3935cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef void* pthread_workqueue_t; 3945cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef void* pthread_workitem_handle_t; 395f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov 396f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef void* dispatch_group_t; 397f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef void* dispatch_queue_t; 398ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryanytypedef u64 dispatch_time_t; 399f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovtypedef void (*dispatch_function_t)(void *block); 4005cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef void* (*worker_t)(void *block); 4015cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov 4025cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov// A wrapper for the ObjC blocks used to support libdispatch. 4035cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovtypedef struct { 4045cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov void *block; 4055cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov dispatch_function_t func; 406e0cff0bc20ae51790c8edfbceb817e18ebf5355eKostya Serebryany u32 parent_tid; 4075cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov} asan_block_context_t; 4085cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov 409f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// We use extern declarations of libdispatch functions here instead 410f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// of including <dispatch/dispatch.h>. This header is not present on 411f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// Mac OS X Leopard and eariler, and although we don't expect ASan to 412f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// work on legacy systems, it's bad to break the build of 413f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov// LLVM compiler-rt there. 4145cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovextern "C" { 415f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_async_f(dispatch_queue_t dq, void *ctxt, 416f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov dispatch_function_t func); 417f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_sync_f(dispatch_queue_t dq, void *ctxt, 418f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov dispatch_function_t func); 419f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_after_f(dispatch_time_t when, dispatch_queue_t dq, void *ctxt, 420f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov dispatch_function_t func); 4215cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovvoid dispatch_barrier_async_f(dispatch_queue_t dq, void *ctxt, 4225cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov dispatch_function_t func); 423f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonovvoid dispatch_group_async_f(dispatch_group_t group, dispatch_queue_t dq, 424f7ceaad2919d2e26e9edea29232bc9dd8f145c42Alexey Samsonov void *ctxt, dispatch_function_t func); 4255cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonovint pthread_workqueue_additem_np(pthread_workqueue_t workq, 4265cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov void *(*workitem_func)(void *), void * workitem_arg, 4275cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov pthread_workitem_handle_t * itemhandlep, unsigned int *gencountp); 4285cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov} // extern "C" 4295cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov 4301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 4311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid asan_dispatch_call_block_and_release(void *block) { 4329cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *context = (asan_block_context_t*)block; 4341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 4351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("asan_dispatch_call_block_and_release(): " 4361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "context: %p, pthread_self: %p\n", 4371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany block, pthread_self()); 4381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanThread *t = asanThreadRegistry().GetCurrent(); 440af3441580555ceed092170232cd5f2cc180f19f4Kostya Serebryany if (!t) { 4413f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany t = AsanThread::Create(context->parent_tid, 0, 0, &stack); 44255cdfc6c5af92560bc0623b5a0d70af71511c3c8Alexey Samsonov asanThreadRegistry().RegisterThread(t); 44369eca73ac96688c8bfe1f23ee006af29c7858c40Kostya Serebryany t->Init(); 4441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asanThreadRegistry().SetCurrent(t); 4451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Call the original dispatcher for the block. 4471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany context->func(context->block); 4481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_free(context, &stack); 4491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} // namespace __asan 4521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyusing namespace __asan; // NOLINT 4541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Wrap |ctxt| and |func| into an asan_block_context_t. 4561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The caller retains control of the allocated context. 4571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 4581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyasan_block_context_t *alloc_asan_context(void *ctxt, dispatch_function_t func, 4591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanStackTrace *stack) { 4601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = 4611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), stack); 4621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->block = ctxt; 4631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->func = func; 464e0cff0bc20ae51790c8edfbceb817e18ebf5355eKostya Serebryany asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrInvalid(); 4651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return asan_ctxt; 4661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): can we reduce code duplication by introducing a macro? 469f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_async_f, dispatch_queue_t dq, void *ctxt, 470f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov dispatch_function_t func) { 4719cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 4731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 4741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_async_f(): context: %p, pthread_self: %p\n", 4751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 4761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 4771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 47809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov return REAL(dispatch_async_f)(dq, (void*)asan_ctxt, 47909672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov asan_dispatch_call_block_and_release); 4801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 482f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_sync_f, dispatch_queue_t dq, void *ctxt, 483f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov dispatch_function_t func) { 4849cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 4861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 4871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_sync_f(): context: %p, pthread_self: %p\n", 4881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 4891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 4901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 49109672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov return REAL(dispatch_sync_f)(dq, (void*)asan_ctxt, 49209672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov asan_dispatch_call_block_and_release); 4931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 495f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_after_f, dispatch_time_t when, 496f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov dispatch_queue_t dq, void *ctxt, 497f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov dispatch_function_t func) { 4989cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 5001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 5011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_after_f: %p\n", asan_ctxt); 5021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 5031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 50409672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov return REAL(dispatch_after_f)(when, dq, (void*)asan_ctxt, 50509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov asan_dispatch_call_block_and_release); 5061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 508f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_barrier_async_f, dispatch_queue_t dq, void *ctxt, 509f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov dispatch_function_t func) { 5109cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 5111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 5121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 5131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_barrier_async_f(): context: %p, pthread_self: %p\n", 5141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 5151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 5161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 51709672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov REAL(dispatch_barrier_async_f)(dq, (void*)asan_ctxt, 51809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov asan_dispatch_call_block_and_release); 5191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 521f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, dispatch_group_async_f, dispatch_group_t group, 522f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov dispatch_queue_t dq, void *ctxt, 523f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov dispatch_function_t func) { 5249cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 5251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 5261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 5271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_group_async_f(): context: %p, pthread_self: %p\n", 5281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 5291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 5301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 53109672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov REAL(dispatch_group_async_f)(group, dq, (void*)asan_ctxt, 53209672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov asan_dispatch_call_block_and_release); 5331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following stuff has been extremely helpful while looking for the 5361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// unhandled functions that spawned jobs on Chromium shutdown. If the verbosity 5371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// level is 2 or greater, we wrap pthread_workqueue_additem_np() in order to 5381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// find the points of worker thread creation (each of such threads may be used 5391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// to run several tasks, that's why this is not enough to support the whole 5401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// libdispatch API. 5411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 5421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *wrap_workitem_func(void *arg) { 5431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 5441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("wrap_workitem_func: %p, pthread_self: %p\n", arg, pthread_self()); 5451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *ctxt = (asan_block_context_t*)arg; 5471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany worker_t fn = (worker_t)(ctxt->func); 5481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *result = fn(ctxt->block); 5499cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 5501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_free(arg, &stack); 5511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return result; 5521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 554f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(int, pthread_workqueue_additem_np, pthread_workqueue_t workq, 5551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *(*workitem_func)(void *), void * workitem_arg, 5561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_workitem_handle_t * itemhandlep, unsigned int *gencountp) { 5579cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 5581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = 5591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), &stack); 5601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->block = workitem_arg; 5611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->func = (dispatch_function_t)workitem_func; 562e0cff0bc20ae51790c8edfbceb817e18ebf5355eKostya Serebryany asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrInvalid(); 5631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 5641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("pthread_workqueue_additem_np: %p\n", asan_ctxt); 5651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 5661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 56709672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov return REAL(pthread_workqueue_additem_np)(workq, wrap_workitem_func, 56809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov asan_ctxt, itemhandlep, 56909672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov gencountp); 5701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 571d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany 572431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// CF_RC_BITS, the layout of CFRuntimeBase and __CFStrIsConstant are internal 573431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// and subject to change in further CoreFoundation versions. Apple does not 574431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// guarantee any binary compatibility from release to release. 575431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 576431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFInternal.h 577431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if defined(__BIG_ENDIAN__) 578431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#define CF_RC_BITS 0 579431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif 580431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 581431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if defined(__LITTLE_ENDIAN__) 582431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#define CF_RC_BITS 3 583431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif 584431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 585431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFRuntime.h 586431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkotypedef struct __CFRuntimeBase { 5873f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany uptr _cfisa; 588ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryany u8 _cfinfo[4]; 589431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if __LP64__ 590ee3925515e4c7966f3ef489f687aa7e5692806a9Kostya Serebryany u32 _rc; 591431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif 592431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko} CFRuntimeBase; 593431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 594431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFString.c 595431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkoint __CFStrIsConstant(CFStringRef str) { 596431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko CFRuntimeBase *base = (CFRuntimeBase*)str; 597431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if __LP64__ 598431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko return base->_rc == 0; 599431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#else 600431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko return (base->_cfinfo[CF_RC_BITS]) == 0; 601431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif 602431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko} 603431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 604f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(CFStringRef, CFStringCreateCopy, CFAllocatorRef alloc, 605f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov CFStringRef str) { 606431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko if (__CFStrIsConstant(str)) { 607431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko return str; 608431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko } else { 60909672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov return REAL(CFStringCreateCopy)(alloc, str); 610431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko } 611431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko} 612431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 61364ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonovnamespace __asan { 614beba6448539095b67cab266d09cd7b7d313b8c3dAlexey Samsonov 615beba6448539095b67cab266d09cd7b7d313b8c3dAlexey Samsonovvoid InitializeMacInterceptors() { 6165cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov CHECK(INTERCEPT_FUNCTION(dispatch_async_f)); 6175cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov CHECK(INTERCEPT_FUNCTION(dispatch_sync_f)); 6185cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov CHECK(INTERCEPT_FUNCTION(dispatch_after_f)); 6195cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov CHECK(INTERCEPT_FUNCTION(dispatch_barrier_async_f)); 6205cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov CHECK(INTERCEPT_FUNCTION(dispatch_group_async_f)); 6215cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov // We don't need to intercept pthread_workqueue_additem_np() to support the 6225cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov // libdispatch API, but it helps us to debug the unsupported functions. Let's 6235cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov // intercept it only during verbose runs. 6245cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov if (FLAG_v >= 2) { 6255cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov CHECK(INTERCEPT_FUNCTION(pthread_workqueue_additem_np)); 6265cf832dc0a6566ae4bb8d48b1f41da623d2c2c1aAlexey Samsonov } 62764ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov // Normally CFStringCreateCopy should not copy constant CF strings. 62864ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov // Replacing the default CFAllocator causes constant strings to be copied 62964ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov // rather than just returned, which leads to bugs in big applications like 63064ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov // Chromium and WebKit, see 63164ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov // http://code.google.com/p/address-sanitizer/issues/detail?id=10 63264ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov // Until this problem is fixed we need to check that the string is 63364ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov // non-constant before calling CFStringCreateCopy. 63464ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov CHECK(INTERCEPT_FUNCTION(CFStringCreateCopy)); 63564ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov} 636beba6448539095b67cab266d09cd7b7d313b8c3dAlexey Samsonov 63764ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov} // namespace __asan 63864ce2db7c838cd95315f7a4428e8a628eaa3e2fcAlexey Samsonov 639d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany#endif // __APPLE__ 640