asan_mac.cc revision d6567c5166412f6acdde851e767c26f332d51d3d
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 <algorithm> 251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <sys/mman.h> 271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <unistd.h> 281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanynamespace __asan { 301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_async_f_f real_dispatch_async_f; 321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_sync_f_f real_dispatch_sync_f; 331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_after_f_f real_dispatch_after_f; 341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_barrier_async_f_f real_dispatch_barrier_async_f; 351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern dispatch_group_async_f_f real_dispatch_group_async_f; 361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern pthread_workqueue_additem_np_f real_pthread_workqueue_additem_np; 371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// No-op. Mac does not support static linkage anyway. 391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *AsanDoesNotSupportStaticLinkage() { 401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return NULL; 411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *asan_mmap(void *addr, size_t length, int prot, int flags, 441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int fd, uint64_t offset) { 451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return mmap(addr, length, prot, flags, fd, offset); 461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyssize_t asan_write(int fd, const void *buf, size_t count) { 491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return write(fd, buf, count); 501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Support for the following functions from libdispatch on Mac OS: 531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_async_f() 541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_async() 551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync_f() 561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() 571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_after_f() 581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_after() 591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_group_async_f() 601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_group_async() 611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): libdispatch API contains other functions that we don't support 621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// yet. 631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_sync() and dispatch_sync_f() are synchronous, although chances are 651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// they can cause jobs to run on a thread different from the current one. 661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): if so, we need a test for this (otherwise we should remove 671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// them). 681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following functions use dispatch_barrier_async_f() (which isn't a library 701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// function but is exported) and are thus supported: 711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_cancel_handler_f() 721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_cancel_handler() 731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_event_handler_f() 741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// dispatch_source_set_event_handler() 751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The reference manual for Grand Central Dispatch is available at 771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// http://developer.apple.com/library/mac/#documentation/Performance/Reference/GCD_libdispatch_Ref/Reference/reference.html 781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The implementation details are at 791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// http://libdispatch.macosforge.org/trac/browser/trunk/src/queue.c 801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid asan_dispatch_call_block_and_release(void *block) { 831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *context = (asan_block_context_t*)block; 851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("asan_dispatch_call_block_and_release(): " 871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "context: %p, pthread_self: %p\n", 881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany block, pthread_self()); 891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanThread *t = asanThreadRegistry().GetCurrent(); 911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (t) { 921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // We've already executed a job on this worker thread. Let's reuse the 931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // AsanThread object. 941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (t != asanThreadRegistry().GetMain()) { 951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Flush the statistics and update the current thread's tid. 961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asanThreadRegistry().UnregisterThread(t); 971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asanThreadRegistry().RegisterThread(t, context->parent_tid, &stack); 981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Otherwise the worker is being executed on the main thread -- we are 1001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // draining the dispatch queue. 1011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // TODO(glider): any checks for that? 1021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 1031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // It's incorrect to assert that the current thread is not dying: at least 1041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // the callbacks from dispatch_sync() are sometimes called after the TSD is 1051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // destroyed. 1061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany t = (AsanThread*)asan_malloc(sizeof(AsanThread), &stack); 1071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany new(t) AsanThread(context->parent_tid, 1081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany /*start_routine*/NULL, /*arg*/NULL, &stack); 1091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asanThreadRegistry().SetCurrent(t); 1101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Call the original dispatcher for the block. 1121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany context->func(context->block); 1131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_free(context, &stack); 1141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} // namespace __asan 1171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyusing namespace __asan; // NOLINT 1191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Wrap |ctxt| and |func| into an asan_block_context_t. 1211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The caller retains control of the allocated context. 1221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 1231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyasan_block_context_t *alloc_asan_context(void *ctxt, dispatch_function_t func, 1241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanStackTrace *stack) { 1251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = 1261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), stack); 1271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->block = ctxt; 1281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->func = func; 1291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany AsanThread *curr_thread = asanThreadRegistry().GetCurrent(); 1301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_debug) { 1311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Sometimes at Chromium teardown this assertion is violated: 1321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // -- a task is created via dispatch_async() on the "CFMachPort" 1331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // thread while doing _dispatch_queue_drain(); 1341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // -- a task is created via dispatch_async_f() on the 1351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // "com.apple.root.default-overcommit-priority" thread while doing 1361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // _dispatch_dispose(). 1371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // TODO(glider): find out what's going on. 1381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany CHECK(curr_thread || asanThreadRegistry().IsCurrentThreadDying()); 1391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne(); 1411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return asan_ctxt; 1421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): can we reduce code duplication by introducing a macro? 1451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 1461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(dispatch_async_f)(dispatch_queue_t dq, 1471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, 1481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_function_t func) { 1491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 1501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 1511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 1521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_async_f(): context: %p, pthread_self: %p\n", 1531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 1541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 1551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_dispatch_async_f(dq, (void*)asan_ctxt, 1571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 1581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 1611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(dispatch_sync_f)(dispatch_queue_t dq, 1621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, 1631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_function_t func) { 1641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 1651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 1661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 1671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_sync_f(): context: %p, pthread_self: %p\n", 1681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 1691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 1701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_dispatch_sync_f(dq, (void*)asan_ctxt, 1721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 1731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 1761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(dispatch_after_f)(dispatch_time_t when, 1771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_queue_t dq, 1781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, 1791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_function_t func) { 1801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 1811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 1821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 1831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_after_f: %p\n", asan_ctxt); 1841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 1851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_dispatch_after_f(when, dq, (void*)asan_ctxt, 1871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 1881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 1911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid WRAP(dispatch_barrier_async_f)(dispatch_queue_t dq, 1921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, dispatch_function_t func) { 1931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 1941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 1951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 1961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_barrier_async_f(): context: %p, pthread_self: %p\n", 1971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 1981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 1991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany real_dispatch_barrier_async_f(dq, (void*)asan_ctxt, 2011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 2021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 2051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid WRAP(dispatch_group_async_f)(dispatch_group_t group, 2061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany dispatch_queue_t dq, 2071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ctxt, dispatch_function_t func) { 2081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 2091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = alloc_asan_context(ctxt, func, &stack); 2101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 2111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("dispatch_group_async_f(): context: %p, pthread_self: %p\n", 2121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt, pthread_self()); 2131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 2141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany real_dispatch_group_async_f(group, dq, (void*)asan_ctxt, 2161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_dispatch_call_block_and_release); 2171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The following stuff has been extremely helpful while looking for the 2201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// unhandled functions that spawned jobs on Chromium shutdown. If the verbosity 2211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// level is 2 or greater, we wrap pthread_workqueue_additem_np() in order to 2221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// find the points of worker thread creation (each of such threads may be used 2231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// to run several tasks, that's why this is not enough to support the whole 2241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// libdispatch API. 2251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 2261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *wrap_workitem_func(void *arg) { 2271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 2281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("wrap_workitem_func: %p, pthread_self: %p\n", arg, pthread_self()); 2291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *ctxt = (asan_block_context_t*)arg; 2311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany worker_t fn = (worker_t)(ctxt->func); 2321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *result = fn(ctxt->block); 2331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 2341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_free(arg, &stack); 2351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return result; 2361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" 2391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint WRAP(pthread_workqueue_additem_np)(pthread_workqueue_t workq, 2401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *(*workitem_func)(void *), void * workitem_arg, 2411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_workitem_handle_t * itemhandlep, unsigned int *gencountp) { 2421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GET_STACK_TRACE_HERE(kStackTraceMax, /*fast_unwind*/false); 2431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_block_context_t *asan_ctxt = 2441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (asan_block_context_t*) asan_malloc(sizeof(asan_block_context_t), &stack); 2451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->block = workitem_arg; 2461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->func = (dispatch_function_t)workitem_func; 2471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_ctxt->parent_tid = asanThreadRegistry().GetCurrentTidOrMinusOne(); 2481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (FLAG_v >= 2) { 2491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Report("pthread_workqueue_additem_np: %p\n", asan_ctxt); 2501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany PRINT_CURRENT_STACK(); 2511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return real_pthread_workqueue_additem_np(workq, wrap_workitem_func, asan_ctxt, 2531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany itemhandlep, gencountp); 2541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 255d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany 256d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany#endif // __APPLE__ 257