src/shims/tsd.h

   1 /*
   2  * Copyright (c) 2008-2013 Apple Inc. All rights reserved.
   3  *
   4  * @APPLE_APACHE_LICENSE_HEADER_START@
   5  *
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *
  18  * @APPLE_APACHE_LICENSE_HEADER_END@
  19  */
  20
  21 /*
  22  * IMPORTANT: This header file describes INTERNAL interfaces to libdispatch
  23  * which are subject to change in future releases of Mac OS X. Any applications
  24  * relying on these interfaces WILL break.
  25  */
  26
  27 #ifndef __DISPATCH_SHIMS_TSD__
  28 #define __DISPATCH_SHIMS_TSD__
  29
  30 #if HAVE_PTHREAD_MACHDEP_H
  31 #include <pthread_machdep.h>
  32 #endif
  33
  34 #define DISPATCH_TSD_INLINE DISPATCH_ALWAYS_INLINE_NDEBUG
  35
  36 #if USE_APPLE_TSD_OPTIMIZATIONS && HAVE_PTHREAD_KEY_INIT_NP && \
  37         !defined(DISPATCH_USE_DIRECT_TSD)
  38 #define DISPATCH_USE_DIRECT_TSD 1
  39 #if __has_include(<os/tsd.h>)
  40 #include <os/tsd.h>
  41 #endif
  42 #endif
  43
  44 #if DISPATCH_USE_DIRECT_TSD
  45 static const unsigned long dispatch_queue_key           = __PTK_LIBDISPATCH_KEY0;
  46 #if DISPATCH_USE_OS_SEMAPHORE_CACHE
  47 static const unsigned long dispatch_sema4_key           = __TSD_SEMAPHORE_CACHE;
  48 #else
  49 static const unsigned long dispatch_sema4_key           = __PTK_LIBDISPATCH_KEY1;
  50 #endif
  51 static const unsigned long dispatch_cache_key           = __PTK_LIBDISPATCH_KEY2;
  52 static const unsigned long dispatch_io_key                      = __PTK_LIBDISPATCH_KEY3;
  53 static const unsigned long dispatch_apply_key           = __PTK_LIBDISPATCH_KEY4;
  54 #if DISPATCH_INTROSPECTION
  55 static const unsigned long dispatch_introspection_key = __PTK_LIBDISPATCH_KEY5;
  56 #elif DISPATCH_PERF_MON
  57 static const unsigned long dispatch_bcounter_key        = __PTK_LIBDISPATCH_KEY5;
  58 #endif
  59
  60 DISPATCH_TSD_INLINE
  61 static inline void
  62 _dispatch_thread_key_create(const unsigned long *k, void (*d)(void *))
  63 {
  64         dispatch_assert_zero(pthread_key_init_np((int)*k, d));
  65 }
  66 #else
  67 extern pthread_key_t dispatch_queue_key;
  68 #if DISPATCH_USE_OS_SEMAPHORE_CACHE
  69 #error "Invalid DISPATCH_USE_OS_SEMAPHORE_CACHE configuration"
  70 #else
  71 extern pthread_key_t dispatch_sema4_key;
  72 #endif
  73 extern pthread_key_t dispatch_cache_key;
  74 extern pthread_key_t dispatch_io_key;
  75 extern pthread_key_t dispatch_apply_key;
  76 #if DISPATCH_INTROSPECTION
  77 extern pthread_key_t dispatch_introspection_key;
  78 #elif DISPATCH_PERF_MON
  79 extern pthread_key_t dispatch_bcounter_key;
  80 #endif
  81
  82
  83 DISPATCH_TSD_INLINE
  84 static inline void
  85 _dispatch_thread_key_create(pthread_key_t *k, void (*d)(void *))
  86 {
  87         dispatch_assert_zero(pthread_key_create(k, d));
  88 }
  89 #endif
  90
  91 #if DISPATCH_USE_TSD_BASE && !DISPATCH_DEBUG
  92 #else // DISPATCH_USE_TSD_BASE
  93 DISPATCH_TSD_INLINE
  94 static inline void
  95 _dispatch_thread_setspecific(pthread_key_t k, void *v)
  96 {
  97 #if DISPATCH_USE_DIRECT_TSD
  98         if (_pthread_has_direct_tsd()) {
  99                 (void)_pthread_setspecific_direct(k, v);
 100                 return;
 101         }
 102 #endif
 103         dispatch_assert_zero(pthread_setspecific(k, v));
 104 }
 105
 106 DISPATCH_TSD_INLINE
 107 static inline void *
 108 _dispatch_thread_getspecific(pthread_key_t k)
 109 {
 110 #if DISPATCH_USE_DIRECT_TSD
 111         if (_pthread_has_direct_tsd()) {
 112                 return _pthread_getspecific_direct(k);
 113         }
 114 #endif
 115         return pthread_getspecific(k);
 116 }
 117 #endif // DISPATCH_USE_TSD_BASE
 118
 119 #if TARGET_OS_WIN32
 120 #define _dispatch_thread_self() ((uintptr_t)GetCurrentThreadId())
 121 #else
 122 #if DISPATCH_USE_DIRECT_TSD
 123 #define _dispatch_thread_self() ((uintptr_t)_dispatch_thread_getspecific( \
 124                 _PTHREAD_TSD_SLOT_PTHREAD_SELF))
 125 #else
 126 #define _dispatch_thread_self() ((uintptr_t)pthread_self())
 127 #endif
 128 #endif
 129
 130 DISPATCH_TSD_INLINE DISPATCH_CONST
 131 static inline unsigned int
 132 _dispatch_cpu_number(void)
 133 {
 134 #if TARGET_IPHONE_SIMULATOR && IPHONE_SIMULATOR_HOST_MIN_VERSION_REQUIRED < 1090
 135         return 0;
 136 #elif __has_include(<os/tsd.h>)
 137         return _os_cpu_number();
 138 #elif defined(__x86_64__) || defined(__i386__)
 139         struct { uintptr_t p1, p2; } p;
 140         __asm__("sidt %[p]" : [p] "=&m" (p));
 141         return (unsigned int)(p.p1 & 0xfff);
 142 #else
 143         // Not yet implemented.
 144         return 0;
 145 #endif
 146 }
 147
 148 #undef DISPATCH_TSD_INLINE
 149
 150 #endif