asan_mac.cc revision 431e51782d62d1257348e41e24da6b544fe70507
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 171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_mac.h" 181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_internal.h" 208a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#include "asan_procmaps.h" 211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_stack.h" 221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_thread.h" 231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_thread_registry.h" 241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 251e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko#include <crt_externs.h> // for _NSGetEnviron 268a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#include <mach-o/dyld.h> 271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <sys/mman.h> 28ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany#include <sys/resource.h> 299107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany#include <sys/ucontext.h> 30c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany#include <pthread.h> 31a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany#include <fcntl.h> 321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <unistd.h> 33d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany#include <libkern/OSAtomic.h> 341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanynamespace __asan { 361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 373281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkovoid *island_allocator_pos = NULL; 383281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko 391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_async_f_f real_dispatch_async_f; 401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_sync_f_f real_dispatch_sync_f; 411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_after_f_f real_dispatch_after_f; 421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_barrier_async_f_f real_dispatch_barrier_async_f; 431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_group_async_f_f real_dispatch_group_async_f; 441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern pthread_workqueue_additem_np_f real_pthread_workqueue_additem_np; 45431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkoextern CFStringCreateCopy_f real_CFStringCreateCopy; 461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 479107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryanyvoid GetPcSpBp(void *context, uintptr_t *pc, uintptr_t *sp, uintptr_t *bp) { 489107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany ucontext_t *ucontext = (ucontext_t*)context; 499107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# if __WORDSIZE == 64 509107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *pc = ucontext->uc_mcontext->__ss.__rip; 519107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *bp = ucontext->uc_mcontext->__ss.__rbp; 529107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *sp = ucontext->uc_mcontext->__ss.__rsp; 539107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# else 549107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *pc = ucontext->uc_mcontext->__ss.__eip; 559107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *bp = ucontext->uc_mcontext->__ss.__ebp; 569107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *sp = ucontext->uc_mcontext->__ss.__esp; 579107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# endif // __WORDSIZE 589107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany} 599107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany 601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// No-op. Mac does not support static linkage anyway. 611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *AsanDoesNotSupportStaticLinkage() { 621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return NULL; 631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 654803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryanybool AsanInterceptsSignal(int signum) { 664803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany return (signum == SIGSEGV || signum == SIGBUS) && FLAG_handle_segv; 674803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany} 684803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany 69a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanystatic void *asan_mmap(void *addr, size_t length, int prot, int flags, 701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int fd, uint64_t offset) { 711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return mmap(addr, length, prot, flags, fd, offset); 721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 740ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanWrite(int fd, const void *buf, size_t count) { 751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return write(fd, buf, count); 761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 78de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid *AsanMmapSomewhereOrDie(size_t size, const char *mem_type) { 79de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany size = RoundUpTo(size, kPageSize); 80de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany void *res = asan_mmap(0, size, 81de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany PROT_READ | PROT_WRITE, 82de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany MAP_PRIVATE | MAP_ANON, -1, 0); 83de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany if (res == (void*)-1) { 84de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany OutOfMemoryMessageAndDie(mem_type, size); 85de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany } 86de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return res; 87de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 88de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 89a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMmapFixedNoReserve(uintptr_t fixed_addr, size_t size) { 90a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany return asan_mmap((void*)fixed_addr, size, 91a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany PROT_READ | PROT_WRITE, 92a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE, 93a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 0, 0); 94a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany} 95a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 96a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMmapFixedReserve(uintptr_t fixed_addr, size_t size) { 97a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany return asan_mmap((void*)fixed_addr, size, 98a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany PROT_READ | PROT_WRITE, 99a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany MAP_PRIVATE | MAP_ANON | MAP_FIXED, 100a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 0, 0); 101a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany} 102a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 103a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMprotect(uintptr_t fixed_addr, size_t size) { 104a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany return asan_mmap((void*)fixed_addr, size, 105a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany PROT_NONE, 106a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE, 107a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 0, 0); 108a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany} 109a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 110de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid AsanUnmapOrDie(void *addr, size_t size) { 111de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany if (!addr || !size) return; 112de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany int res = munmap(addr, size); 113de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany if (res != 0) { 114de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany Report("Failed to unmap\n"); 1150ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryany AsanDie(); 116de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany } 117de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 118de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 119de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanOpenReadonly(const char* filename) { 120de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return open(filename, O_RDONLY); 121de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 122de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 1231e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenkoconst char *AsanGetEnv(const char *name) { 1241e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko char ***env_ptr = _NSGetEnviron(); 1251e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko CHECK(env_ptr); 1261e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko char **environ = *env_ptr; 1271e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko CHECK(environ); 1281e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko size_t name_len = internal_strlen(name); 1291e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko while (*environ != NULL) { 1301e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko size_t len = internal_strlen(*environ); 1311e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko if (len > name_len) { 1321e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko const char *p = *environ; 1331e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko if (!internal_memcmp(p, name, name_len) && 1341e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko p[name_len] == '=') { // Match. 1354dd8ba8238b1b698953628affe6e5b2edf3b3e3fAlexey Samsonov return *environ + name_len + 1; // String starting after =. 1361e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko } 1371e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko } 1381e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko environ++; 1391e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko } 1401e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko return NULL; 1411e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko} 1421e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenko 1430ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanRead(int fd, void *buf, size_t count) { 144de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return read(fd, buf, count); 145de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 146de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 147de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanClose(int fd) { 148de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return close(fd); 149de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 150de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 1518a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander PotapenkoAsanProcMaps::AsanProcMaps() { 1528a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko Reset(); 1538a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 1548a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 1558a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander PotapenkoAsanProcMaps::~AsanProcMaps() { 1568a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 1578a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 1583feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// More information about Mach-O headers can be found in mach-o/loader.h 1593feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Each Mach-O image has a header (mach_header or mach_header_64) starting with 1603feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// a magic number, and a list of linker load commands directly following the 1613feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// header. 1623feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// A load command is at least two 32-bit words: the command type and the 1633feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// command size in bytes. We're interested only in segment load commands 1643feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// (LC_SEGMENT and LC_SEGMENT_64), which tell that a part of the file is mapped 1653feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// into the task's address space. 1663feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// The |vmaddr|, |vmsize| and |fileoff| fields of segment_command or 1673feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// segment_command_64 correspond to the memory address, memory size and the 1683feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// file offset of the current memory segment. 1693feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Because these fields are taken from the images as is, one needs to add 1703feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// _dyld_get_image_vmaddr_slide() to get the actual addresses at runtime. 1713feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko 1728a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenkovoid AsanProcMaps::Reset() { 1738a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko // Count down from the top. 1748a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko // TODO(glider): as per man 3 dyld, iterating over the headers with 1753feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko // _dyld_image_count is thread-unsafe. We need to register callbacks for 1763feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko // adding and removing images which will invalidate the AsanProcMaps state. 1778a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko current_image_ = _dyld_image_count(); 1783feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_count_ = -1; 1793feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_addr_ = NULL; 1803feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko} 1813feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko 1823feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Next and NextSegmentLoad were inspired by base/sysinfo.cc in 1833feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Google Perftools, http://code.google.com/p/google-perftools. 1843feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko 1853feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// NextSegmentLoad scans the current image for the next segment load command 1863feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// and returns the start and end addresses and file offset of the corresponding 1873feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// segment. 1883feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko// Note that the segment addresses are not necessarily sorted. 1893feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenkotemplate<uint32_t kLCSegment, typename SegmentCommand> 1903feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenkobool AsanProcMaps::NextSegmentLoad( 1913feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko uintptr_t *start, uintptr_t *end, uintptr_t *offset, 1923feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko char filename[], size_t filename_size) { 1933feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko const char* lc = current_load_cmd_addr_; 1943feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_addr_ += ((const load_command *)lc)->cmdsize; 1958a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (((const load_command *)lc)->cmd == kLCSegment) { 1963feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko const intptr_t dlloff = _dyld_get_image_vmaddr_slide(current_image_); 1978a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko const SegmentCommand* sc = (const SegmentCommand *)lc; 1988a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (start) *start = sc->vmaddr + dlloff; 1998a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (end) *end = sc->vmaddr + sc->vmsize + dlloff; 2008a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (offset) *offset = sc->fileoff; 2018a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (filename) { 2023feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko real_strncpy(filename, _dyld_get_image_name(current_image_), 2038a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko filename_size); 2048a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko } 2058a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return true; 2068a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko } 2078a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return false; 2088a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 2098a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 2108a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenkobool AsanProcMaps::Next(uintptr_t *start, uintptr_t *end, 2118a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko uintptr_t *offset, char filename[], 2128a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko size_t filename_size) { 2138a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko for (; current_image_ >= 0; current_image_--) { 2148a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko const mach_header* hdr = _dyld_get_image_header(current_image_); 2158a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko if (!hdr) continue; 2163feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko if (current_load_cmd_count_ < 0) { 2173feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko // Set up for this image; 2183feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_count_ = hdr->ncmds; 2193feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko switch (hdr->magic) { 2203feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko#ifdef MH_MAGIC_64 2213feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko case MH_MAGIC_64: { 2223feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_addr_ = (char*)hdr + sizeof(mach_header_64); 2233feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko break; 2243feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2253feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko#endif 2263feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko case MH_MAGIC: { 2273feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_addr_ = (char*)hdr + sizeof(mach_header); 2283feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko break; 2293feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2303feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko default: { 2313feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko continue; 2323feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2333feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2343feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko } 2358a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 2368a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko // We start with the next load command (we've already looked at this one). 2373feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko for (current_load_cmd_count_--; 2383feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_count_ >= 0; 2393feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko current_load_cmd_count_--) { 2408a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#ifdef MH_MAGIC_64 2413feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko if (NextSegmentLoad<LC_SEGMENT_64, struct segment_command_64>( 2423feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko start, end, offset, filename, filename_size)) 2438a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return true; 2448a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko#endif 2453feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko if (NextSegmentLoad<LC_SEGMENT, struct segment_command>( 2463feef82eab6d5f0ace384a670bfb5988e0c165a0Alexander Potapenko start, end, offset, filename, filename_size)) 2478a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return true; 2488a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko } 2498a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko // If we get here, no more load_cmd's in this image talk about 2508a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko // segments. Go on to the next image. 2518a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko } 2528a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return false; 2538a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 2548a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 2558a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenkobool AsanProcMaps::GetObjectNameAndOffset(uintptr_t addr, uintptr_t *offset, 2568a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko char filename[], 2578a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko size_t filename_size) { 2588a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko return IterateForObjectNameAndOffset(addr, offset, filename, filename_size); 2598a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko} 2608a34d384255f9bf4c2a9b03a4df81b9af57124d8Alexander Potapenko 261c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryanyvoid AsanThread::SetThreadStackTopAndBottom() { 262c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany size_t stacksize = pthread_get_stacksize_np(pthread_self()); 263c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany void *stackaddr = pthread_get_stackaddr_np(pthread_self()); 264c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany stack_top_ = (uintptr_t)stackaddr; 265c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany stack_bottom_ = stack_top_ - stacksize; 266c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany int local; 267c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany CHECK(AddrIsInStack((uintptr_t)&local)); 268c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany} 269c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany 270d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya SerebryanyAsanLock::AsanLock(LinkerInitialized) { 271d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany // We assume that OS_SPINLOCK_INIT is zero 272d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany} 273d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany 274d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryanyvoid AsanLock::Lock() { 275d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany CHECK(sizeof(OSSpinLock) <= sizeof(opaque_storage_)); 276d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany CHECK(OS_SPINLOCK_INIT == 0); 277d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany CHECK(owner_ != (uintptr_t)pthread_self()); 278d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany OSSpinLockLock((OSSpinLock*)&opaque_storage_); 279d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany CHECK(!owner_); 280d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany owner_ = (uintptr_t)pthread_self(); 281d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany} 282d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany 283d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryanyvoid AsanLock::Unlock() { 284d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany CHECK(owner_ == (uintptr_t)pthread_self()); 285d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany owner_ = 0; 286d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany OSSpinLockUnlock((OSSpinLock*)&opaque_storage_); 287d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany} 288d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany 2899cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanovvoid AsanStackTrace::GetStackTrace(size_t max_s, uintptr_t pc, uintptr_t bp) { 2909cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov size = 0; 2919cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov trace[0] = pc; 2929cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov if ((max_s) > 1) { 2939cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov max_size = max_s; 2949cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov FastUnwindStack(pc, bp); 2959cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov } 2969cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov} 2979cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov 2983281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// The range of pages to be used by __asan_mach_override_ptr for escape 2993281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// islands. 3003281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// TODO(glider): instead of mapping a fixed range we must find a range of 3013281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko// unmapped pages in vmmap and take them. 3021346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// These constants were chosen empirically and may not work if the shadow 3031346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// memory layout changes. Unfortunately they do necessarily depend on 3041346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko// kHighMemBeg or kHighMemEnd. 3051346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#if __WORDSIZE == 32 3061346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#define kIslandEnd (0xffdf0000 - kPageSize) 3071346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#define kIslandBeg (kIslandEnd - 256 * kPageSize) 3081346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#else 3093281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko#define kIslandEnd (0x7fffffdf0000 - kPageSize) 3103281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko#define kIslandBeg (kIslandEnd - 256 * kPageSize) 3111346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko#endif 3123281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko 3133281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkoextern "C" 3143281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkomach_error_t __asan_allocate_island(void **ptr, 3153281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko size_t unused_size, 3163281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko void *unused_hint) { 3173281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko if (!island_allocator_pos) { 3181346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko island_allocator_pos = 3191346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko asan_mmap((void*)kIslandBeg, kIslandEnd - kIslandBeg, 3201346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko PROT_READ | PROT_WRITE | PROT_EXEC, 3211346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko MAP_PRIVATE | MAP_ANON | MAP_FIXED, 3221346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko -1, 0); 3231346ced2eb8d10305e8d98496d9006cfbbad1548Alexander Potapenko if (island_allocator_pos != (void*)kIslandBeg) { 3243281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko return KERN_NO_SPACE; 3253281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko } 3263281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko }; 3273281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko *ptr = island_allocator_pos; 3283281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko island_allocator_pos = (char*)island_allocator_pos + kPageSize; 3293281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko return err_none; 3303281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko} 3313281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko 3323281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkoextern "C" 3333281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenkomach_error_t __asan_deallocate_island(void *ptr) { 3343281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko // Do nothing. 3353281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko // TODO(glider): allow to free and reuse the island memory. 3363281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko return err_none; 3373281209790b5e543c79acb2f5008d1df77fb76d9Alexander Potapenko} 338d55f5f8c413622db4bd28b5cca9bfeb4d61564e0Kostya Serebryany 3391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Support for the following functions from libdispatch on Mac OS: 3401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_async_f() 3411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_async() 3421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync_f() 3431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() 3441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_after_f() 3451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_after() 3461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_group_async_f() 3471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_group_async() 3481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): libdispatch API contains other functions that we don't support 3491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// yet. 3501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 3511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() and dispatch_sync_f() are synchronous, although chances are 3521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// they can cause jobs to run on a thread different from the current one. 3531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): if so, we need a test for this (otherwise we should remove 3541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// them). 3551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 3561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following functions use dispatch_barrier_async_f() (which isn't a library 3571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// function but is exported) and are thus supported: 3581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_cancel_handler_f() 3591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_cancel_handler() 3601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_event_handler_f() 3611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_event_handler() 3621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 3631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The reference manual for Grand Central Dispatch is available at 3641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// http://developer.apple.com/library/mac/#documentation/Performance/Reference/GCD_libdispatch_Ref/Reference/reference.html 3651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The implementation details are at 3661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// http://libdispatch.macosforge.org/trac/browser/trunk/src/queue.c 3671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 3691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid asan_dispatch_call_block_and_release(void *block) { 3709cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 3711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *context = (asan_block_context_t*)block; 3721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 3731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("asan_dispatch_call_block_and_release(): " 3741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "context: %p, pthread_self: %p\n", 3751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany block, pthread_self()); 3761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 3771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanThread *t = asanThreadRegistry().GetCurrent(); 378af3441580555ceed092170232cd5f2cc180f19f4Kostya Serebryany if (!t) { 37955cdfc6c5af92560bc0623b5a0d70af71511c3c8Alexey Samsonov t = AsanThread::Create(context->parent_tid, NULL, NULL, &stack); 38055cdfc6c5af92560bc0623b5a0d70af71511c3c8Alexey Samsonov asanThreadRegistry().RegisterThread(t); 38169eca73ac96688c8bfe1f23ee006af29c7858c40Kostya Serebryany t->Init(); 3821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asanThreadRegistry().SetCurrent(t); 3831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 3841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Call the original dispatcher for the block. 3851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany context->func(context->block); 3861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_free(context, &stack); 3871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} // namespace __asan 3901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyusing namespace __asan; // NOLINT 3921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Wrap |ctxt| and |func| into an asan_block_context_t. 3941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The caller retains control of the allocated context. 3951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 3961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyasan_block_context_t *alloc_asan_context(void *ctxt, dispatch_function_t func, 3971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanStackTrace *stack) { 3981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = 3991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), stack); 4001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->block = ctxt; 4011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->func = func; 4021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne(); 4031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return asan_ctxt; 4041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): can we reduce code duplication by introducing a macro? 4071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 4081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(dispatch_async_f)(dispatch_queue_t dq, 4091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, 4101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_function_t func) { 4119cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 4131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 4141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_async_f(): context: %p, pthread_self: %p\n", 4151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 4161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 4171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_dispatch_async_f(dq, (void*)asan_ctxt, 4191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 4201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 4231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(dispatch_sync_f)(dispatch_queue_t dq, 4241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, 4251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_function_t func) { 4269cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 4281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 4291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_sync_f(): context: %p, pthread_self: %p\n", 4301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 4311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 4321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_dispatch_sync_f(dq, (void*)asan_ctxt, 4341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 4351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 4381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(dispatch_after_f)(dispatch_time_t when, 4391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_queue_t dq, 4401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, 4411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_function_t func) { 4429cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 4441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 4451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_after_f: %p\n", asan_ctxt); 4461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 4471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_dispatch_after_f(when, dq, (void*)asan_ctxt, 4491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 4501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 4531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid WRAP(dispatch_barrier_async_f)(dispatch_queue_t dq, 4541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, dispatch_function_t func) { 4559cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 4571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 4581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_barrier_async_f(): context: %p, pthread_self: %p\n", 4591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 4601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 4611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany real_dispatch_barrier_async_f(dq, (void*)asan_ctxt, 4631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 4641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 4671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid WRAP(dispatch_group_async_f)(dispatch_group_t group, 4681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_queue_t dq, 4691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, dispatch_function_t func) { 4709cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 4721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 4731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_group_async_f(): context: %p, pthread_self: %p\n", 4741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 4751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 4761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany real_dispatch_group_async_f(group, dq, (void*)asan_ctxt, 4781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 4791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following stuff has been extremely helpful while looking for the 4821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// unhandled functions that spawned jobs on Chromium shutdown. If the verbosity 4831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// level is 2 or greater, we wrap pthread_workqueue_additem_np() in order to 4841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// find the points of worker thread creation (each of such threads may be used 4851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// to run several tasks, that's why this is not enough to support the whole 4861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// libdispatch API. 4871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 4881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *wrap_workitem_func(void *arg) { 4891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 4901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("wrap_workitem_func: %p, pthread_self: %p\n", arg, pthread_self()); 4911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *ctxt = (asan_block_context_t*)arg; 4931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany worker_t fn = (worker_t)(ctxt->func); 4941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *result = fn(ctxt->block); 4959cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 4961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_free(arg, &stack); 4971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return result; 4981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 5011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(pthread_workqueue_additem_np)(pthread_workqueue_t workq, 5021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *(*workitem_func)(void *), void * workitem_arg, 5031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_workitem_handle_t * itemhandlep, unsigned int *gencountp) { 5049cfa194cc62026fc7c6e82f7303eee8ad4d10cf4Evgeniy Stepanov GET_STACK_TRACE_HERE(kStackTraceMax); 5051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = 5061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), &stack); 5071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->block = workitem_arg; 5081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->func = (dispatch_function_t)workitem_func; 5091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne(); 5101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 5111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("pthread_workqueue_additem_np: %p\n", asan_ctxt); 5121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 5131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_pthread_workqueue_additem_np(workq, wrap_workitem_func, asan_ctxt, 5151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany itemhandlep, gencountp); 5161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 517d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany 518431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// CF_RC_BITS, the layout of CFRuntimeBase and __CFStrIsConstant are internal 519431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// and subject to change in further CoreFoundation versions. Apple does not 520431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// guarantee any binary compatibility from release to release. 521431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 522431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFInternal.h 523431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if defined(__BIG_ENDIAN__) 524431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#define CF_RC_BITS 0 525431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif 526431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 527431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if defined(__LITTLE_ENDIAN__) 528431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#define CF_RC_BITS 3 529431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif 530431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 531431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFRuntime.h 532431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkotypedef struct __CFRuntimeBase { 533431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko uintptr_t _cfisa; 534431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko uint8_t _cfinfo[4]; 535431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if __LP64__ 536431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko uint32_t _rc; 537431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif 538431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko} CFRuntimeBase; 539431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 540431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko// See http://opensource.apple.com/source/CF/CF-635.15/CFString.c 541431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkoint __CFStrIsConstant(CFStringRef str) { 542431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko CFRuntimeBase *base = (CFRuntimeBase*)str; 543431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#if __LP64__ 544431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko return base->_rc == 0; 545431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#else 546431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko return (base->_cfinfo[CF_RC_BITS]) == 0; 547431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko#endif 548431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko} 549431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 550431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenkoextern "C" 551431e51782d62d1257348e41e24da6b544fe70507Alexander PotapenkoCFStringRef WRAP(CFStringCreateCopy)(CFAllocatorRef alloc, CFStringRef str) { 552431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko if (__CFStrIsConstant(str)) { 553431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko return str; 554431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko } else { 555431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko return real_CFStringCreateCopy(alloc, str); 556431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko } 557431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko} 558431e51782d62d1257348e41e24da6b544fe70507Alexander Potapenko 559d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany#endif // __APPLE__ 560