asan_mac.cc revision 9107c26bd88fc9cf44a2cd7d6967eb830ac63be3
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" 201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_stack.h" 211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_thread.h" 221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_thread_registry.h" 231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <sys/mman.h> 25ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany#include <sys/resource.h> 269107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany#include <sys/ucontext.h> 27c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany#include <pthread.h> 28a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany#include <fcntl.h> 291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <unistd.h> 301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 311957399c5d44802939021d5b89b8171719d96b75Kostya Serebryany#include <new> 324f24d3efb98fcf2b4d667bd42a4d840dbab0ce82Kostya Serebryany 331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanynamespace __asan { 341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_async_f_f real_dispatch_async_f; 361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_sync_f_f real_dispatch_sync_f; 371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_after_f_f real_dispatch_after_f; 381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_barrier_async_f_f real_dispatch_barrier_async_f; 391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_group_async_f_f real_dispatch_group_async_f; 401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern pthread_workqueue_additem_np_f real_pthread_workqueue_additem_np; 411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 429107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryanyvoid GetPcSpBp(void *context, uintptr_t *pc, uintptr_t *sp, uintptr_t *bp) { 439107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany ucontext_t *ucontext = (ucontext_t*)context; 449107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# if __WORDSIZE == 64 459107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *pc = ucontext->uc_mcontext->__ss.__rip; 469107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *bp = ucontext->uc_mcontext->__ss.__rbp; 479107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *sp = ucontext->uc_mcontext->__ss.__rsp; 489107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# else 499107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *pc = ucontext->uc_mcontext->__ss.__eip; 509107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *bp = ucontext->uc_mcontext->__ss.__ebp; 519107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany *sp = ucontext->uc_mcontext->__ss.__esp; 529107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany# endif // __WORDSIZE 539107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany} 549107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany 559107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryany 561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// No-op. Mac does not support static linkage anyway. 571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *AsanDoesNotSupportStaticLinkage() { 581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return NULL; 591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 61a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanystatic void *asan_mmap(void *addr, size_t length, int prot, int flags, 621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int fd, uint64_t offset) { 631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return mmap(addr, length, prot, flags, fd, offset); 641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 66a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyssize_t AsanWrite(int fd, const void *buf, size_t count) { 671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return write(fd, buf, count); 681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 70de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid *AsanMmapSomewhereOrDie(size_t size, const char *mem_type) { 71de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany size = RoundUpTo(size, kPageSize); 72de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany void *res = asan_mmap(0, size, 73de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany PROT_READ | PROT_WRITE, 74de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany MAP_PRIVATE | MAP_ANON, -1, 0); 75de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany if (res == (void*)-1) { 76de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany OutOfMemoryMessageAndDie(mem_type, size); 77de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany } 78de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return res; 79de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 80de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 81a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMmapFixedNoReserve(uintptr_t fixed_addr, size_t size) { 82a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany return asan_mmap((void*)fixed_addr, size, 83a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany PROT_READ | PROT_WRITE, 84a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE, 85a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 0, 0); 86a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany} 87a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 88a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMmapFixedReserve(uintptr_t fixed_addr, size_t size) { 89a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany return asan_mmap((void*)fixed_addr, size, 90a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany PROT_READ | PROT_WRITE, 91a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany MAP_PRIVATE | MAP_ANON | MAP_FIXED, 92a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 0, 0); 93a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany} 94a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 95a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMprotect(uintptr_t fixed_addr, size_t size) { 96a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany return asan_mmap((void*)fixed_addr, size, 97a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany PROT_NONE, 98a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany MAP_PRIVATE | MAP_ANON | MAP_FIXED | MAP_NORESERVE, 99a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 0, 0); 100a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany} 101a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryany 102de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid AsanUnmapOrDie(void *addr, size_t size) { 103de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany if (!addr || !size) return; 104de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany int res = munmap(addr, size); 105de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany if (res != 0) { 106de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany Report("Failed to unmap\n"); 107de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany ASAN_DIE; 108de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany } 109de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 110de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 111de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanOpenReadonly(const char* filename) { 112de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return open(filename, O_RDONLY); 113de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 114de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 115de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyssize_t AsanRead(int fd, void *buf, size_t count) { 116de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return read(fd, buf, count); 117de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 118de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 119de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanClose(int fd) { 120de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return close(fd); 121de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 122de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 123c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryanyvoid AsanThread::SetThreadStackTopAndBottom() { 124c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany size_t stacksize = pthread_get_stacksize_np(pthread_self()); 125c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany void *stackaddr = pthread_get_stackaddr_np(pthread_self()); 126c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany stack_top_ = (uintptr_t)stackaddr; 127c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany stack_bottom_ = stack_top_ - stacksize; 128c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany int local; 129c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany CHECK(AddrIsInStack((uintptr_t)&local)); 130c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany} 131c549dd7b5fa5fb97270f57067797224cee0429f2Kostya Serebryany 132ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryanyvoid AsanDisableCoreDumper() { 133ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany struct rlimit nocore; 134ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany nocore.rlim_cur = 0; 135ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany nocore.rlim_max = 0; 136ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany setrlimit(RLIMIT_CORE, &nocore); 137ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany} 138ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany 1391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Support for the following functions from libdispatch on Mac OS: 1401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_async_f() 1411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_async() 1421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync_f() 1431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() 1441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_after_f() 1451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_after() 1461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_group_async_f() 1471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_group_async() 1481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): libdispatch API contains other functions that we don't support 1491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// yet. 1501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 1511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() and dispatch_sync_f() are synchronous, although chances are 1521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// they can cause jobs to run on a thread different from the current one. 1531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): if so, we need a test for this (otherwise we should remove 1541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// them). 1551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 1561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following functions use dispatch_barrier_async_f() (which isn't a library 1571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// function but is exported) and are thus supported: 1581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_cancel_handler_f() 1591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_cancel_handler() 1601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_event_handler_f() 1611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_event_handler() 1621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 1631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The reference manual for Grand Central Dispatch is available at 1641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// http://developer.apple.com/library/mac/#documentation/Performance/Reference/GCD_libdispatch_Ref/Reference/reference.html 1651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The implementation details are at 1661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// http://libdispatch.macosforge.org/trac/browser/trunk/src/queue.c 1671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 1691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid asan_dispatch_call_block_and_release(void *block) { 1701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 1711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *context = (asan_block_context_t*)block; 1721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 1731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("asan_dispatch_call_block_and_release(): " 1741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "context: %p, pthread_self: %p\n", 1751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany block, pthread_self()); 1761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanThread *t = asanThreadRegistry().GetCurrent(); 1781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (t) { 1791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // We've already executed a job on this worker thread. Let's reuse the 1801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // AsanThread object. 1811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (t != asanThreadRegistry().GetMain()) { 1821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Flush the statistics and update the current thread's tid. 1831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asanThreadRegistry().UnregisterThread(t); 1841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asanThreadRegistry().RegisterThread(t, context->parent_tid, &stack); 1851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Otherwise the worker is being executed on the main thread -- we are 1871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // draining the dispatch queue. 1881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // TODO(glider): any checks for that? 1891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 1901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // It's incorrect to assert that the current thread is not dying: at least 1911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // the callbacks from dispatch_sync() are sometimes called after the TSD is 1921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // destroyed. 1931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany t = (AsanThread*)asan_malloc(sizeof(AsanThread), &stack); 1941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany new(t) AsanThread(context->parent_tid, 1951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany /*start_routine*/NULL, /*arg*/NULL, &stack); 19669eca73ac96688c8bfe1f23ee006af29c7858c40Kostya Serebryany t->Init(); 1971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asanThreadRegistry().SetCurrent(t); 1981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Call the original dispatcher for the block. 2001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany context->func(context->block); 2011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_free(context, &stack); 2021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} // namespace __asan 2051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyusing namespace __asan; // NOLINT 2071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Wrap |ctxt| and |func| into an asan_block_context_t. 2091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The caller retains control of the allocated context. 2101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 2111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyasan_block_context_t *alloc_asan_context(void *ctxt, dispatch_function_t func, 2121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanStackTrace *stack) { 2131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = 2141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), stack); 2151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->block = ctxt; 2161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->func = func; 2171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanThread *curr_thread = asanThreadRegistry().GetCurrent(); 2181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_debug) { 2191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Sometimes at Chromium teardown this assertion is violated: 2201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // -- a task is created via dispatch_async() on the "CFMachPort" 2211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // thread while doing _dispatch_queue_drain(); 2221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // -- a task is created via dispatch_async_f() on the 2231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // "com.apple.root.default-overcommit-priority" thread while doing 2241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // _dispatch_dispose(). 2251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // TODO(glider): find out what's going on. 2261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany CHECK(curr_thread || asanThreadRegistry().IsCurrentThreadDying()); 2271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne(); 2291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return asan_ctxt; 2301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): can we reduce code duplication by introducing a macro? 2331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 2341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(dispatch_async_f)(dispatch_queue_t dq, 2351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, 2361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_function_t func) { 2371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 2381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 2391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 2401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_async_f(): context: %p, pthread_self: %p\n", 2411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 2421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 2431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_dispatch_async_f(dq, (void*)asan_ctxt, 2451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 2461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 2491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(dispatch_sync_f)(dispatch_queue_t dq, 2501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, 2511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_function_t func) { 2521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 2531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 2541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 2551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_sync_f(): context: %p, pthread_self: %p\n", 2561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 2571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 2581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_dispatch_sync_f(dq, (void*)asan_ctxt, 2601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 2611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 2641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(dispatch_after_f)(dispatch_time_t when, 2651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_queue_t dq, 2661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, 2671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_function_t func) { 2681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 2691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 2701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 2711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_after_f: %p\n", asan_ctxt); 2721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 2731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_dispatch_after_f(when, dq, (void*)asan_ctxt, 2751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 2761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 2791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid WRAP(dispatch_barrier_async_f)(dispatch_queue_t dq, 2801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, dispatch_function_t func) { 2811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 2821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 2831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 2841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_barrier_async_f(): context: %p, pthread_self: %p\n", 2851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 2861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 2871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany real_dispatch_barrier_async_f(dq, (void*)asan_ctxt, 2891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 2901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 2931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid WRAP(dispatch_group_async_f)(dispatch_group_t group, 2941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_queue_t dq, 2951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, dispatch_function_t func) { 2961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 2971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 2981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 2991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_group_async_f(): context: %p, pthread_self: %p\n", 3001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 3011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 3021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 3031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany real_dispatch_group_async_f(group, dq, (void*)asan_ctxt, 3041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 3051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following stuff has been extremely helpful while looking for the 3081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// unhandled functions that spawned jobs on Chromium shutdown. If the verbosity 3091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// level is 2 or greater, we wrap pthread_workqueue_additem_np() in order to 3101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// find the points of worker thread creation (each of such threads may be used 3111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// to run several tasks, that's why this is not enough to support the whole 3121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// libdispatch API. 3131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 3141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *wrap_workitem_func(void *arg) { 3151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 3161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("wrap_workitem_func: %p, pthread_self: %p\n", arg, pthread_self()); 3171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 3181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *ctxt = (asan_block_context_t*)arg; 3191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany worker_t fn = (worker_t)(ctxt->func); 3201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *result = fn(ctxt->block); 3211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 3221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_free(arg, &stack); 3231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return result; 3241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 3271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(pthread_workqueue_additem_np)(pthread_workqueue_t workq, 3281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *(*workitem_func)(void *), void * workitem_arg, 3291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_workitem_handle_t * itemhandlep, unsigned int *gencountp) { 3301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 3311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = 3321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), &stack); 3331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->block = workitem_arg; 3341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->func = (dispatch_function_t)workitem_func; 3351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne(); 3361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 3371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("pthread_workqueue_additem_np: %p\n", asan_ctxt); 3381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 3391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 3401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_pthread_workqueue_additem_np(workq, wrap_workitem_func, asan_ctxt, 3411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany itemhandlep, gencountp); 3421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 343d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany 344d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany#endif // __APPLE__ 345