Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 //===- subzero/src/IceTLS.h - thread_local workaround -----------*- C++ -*-===// | |
| 2 // | |
| 3 // The Subzero Code Generator | |
| 4 // | |
| 5 // This file is distributed under the University of Illinois Open Source | |
| 6 // License. See LICENSE.TXT for details. | |
| 7 // | |
| 8 //===----------------------------------------------------------------------===// | |
| 9 // | |
| 10 // This file defines macros for working around the lack of support for | |
| 11 // thread_local in MacOS 10.6. It assumes std::thread is written in | |
| 12 // terms of pthread. Define ICE_THREAD_LOCAL_HACK to enable the | |
| 13 // pthread workarounds. | |
| 14 // | |
| 15 //===----------------------------------------------------------------------===// | |
| 16 | |
| 17 #ifndef SUBZERO_SRC_ICETLS_H | |
| 18 #define SUBZERO_SRC_ICETLS_H | |
| 19 | |
| 20 #if defined(_MSC_VER) | |
| 21 #define ICE_ATTRIBUTE_TLS __declspec(thread) | |
| 22 #else // !_MSC_VER | |
| 23 #define ICE_ATTRIBUTE_TLS thread_local | |
| 24 #endif // !_MSC_VER | |
| 25 | |
| 26 // Defines 4 macros for unifying thread_local and pthread: | |
| 27 // | |
| 28 // ICE_TLS_DECLARE_FIELD(Type, FieldName): Declare a static | |
| 29 // thread_local field inside the current class definition. "Type" | |
| 30 // needs to be a pointer type, such as int* or class Foo*. | |
| 31 // | |
| 32 // ICE_TLS_DEFINE_FIELD(Type, ClassName, FieldName): Define a static | |
| 33 // thread_local field outside of its class definition. The field will | |
| 34 // ultimately be initialized to nullptr. | |
| 35 // | |
| 36 // ICE_TLS_INIT_FIELD(FieldName): Ensure the thread_local field is | |
| 37 // properly initialized. This is intended to be called from within a | |
| 38 // static method of the field's class after main() starts (to ensure | |
| 39 // that the pthread library is fully initialized) but before any uses | |
| 40 // of ICE_TLS_GET_FIELD or ICE_TLS_SET_FIELD. | |
| 41 // | |
| 42 // ICE_TLS_GET_FIELD(Type, FieldName): Read the value of the static | |
| 43 // thread_local field. Must be done within the context of its class. | |
| 44 // | |
| 45 // ICE_TLS_SET_FIELD(FieldName, Value): Write a value into the static | |
| 46 // thread_local field. Must be done within the context of its class. | |
| 47 | |
| 48 // TODO(stichnot): Limit this define to only the platforms that | |
| 49 // absolutely require it. And ideally, eventually remove this hack | |
| 50 // altogether. | |
| 51 #define ICE_THREAD_LOCAL_HACK | |
| 52 #ifdef ICE_THREAD_LOCAL_HACK | |
| 53 | |
| 54 // For a static thread_local field F of a class C, instead of | |
| 55 // declaring and defining C::F, we create two static fields: | |
| 56 // static pthread_key_t F__key; | |
| 57 // static int F__initStatus; | |
| 58 // | |
| 59 // The F__initStatus field is used to hold the result of the | |
| 60 // pthread_key_create() call, where a zero value indicates success, | |
| 61 // and a nonzero value indicates failure or that ICE_TLS_INIT_FIELD() | |
| 62 // was never called. | |
| 63 // The F__key field is used as the argument to | |
| 64 // pthread_getspecific() and pthread_setspecific(). | |
| 65 | |
| 66 #include <pthread.h> | |
| 67 | |
| 68 #define ICE_TLS_DECLARE_FIELD(Type, FieldName) \ | |
| 69 typedef Type FieldName##__type; \ | |
| 70 static pthread_key_t FieldName##__key; \ | |
| 71 static int FieldName##__initStatus | |
| 72 #define ICE_TLS_DEFINE_FIELD(Type, ClassName, FieldName) \ | |
| 73 pthread_key_t ClassName::FieldName##__key; \ | |
| 74 int ClassName::FieldName##__initStatus = 1; | |
| 75 #define ICE_TLS_INIT_FIELD(FieldName) \ | |
| 76 if (FieldName##__initStatus) { \ | |
| 77 FieldName##__initStatus = pthread_key_create(&FieldName##__key, nullptr); \ | |
| 78 if (FieldName##__initStatus) \ | |
| 79 llvm_unreachable("Failed to create pthread key"); \ | |
|
JF
2015/01/26 18:48:33
This should be a fatal error instead, unreachable
Jim Stichnoth
2015/01/26 19:06:19
Oops, that's what I meant... Done
| |
| 80 } | |
| 81 #define ICE_TLS_GET_FIELD(FieldName) \ | |
| 82 (assert(FieldName##__initStatus == 0), \ | |
| 83 static_cast<FieldName##__type>(pthread_getspecific(FieldName##__key))) | |
| 84 #define ICE_TLS_SET_FIELD(FieldName, Value) \ | |
| 85 (assert(FieldName##__initStatus == 0), \ | |
| 86 pthread_setspecific(FieldName##__key, (Value))) | |
| 87 | |
| 88 #else // !ICE_THREAD_LOCAL_HACK | |
| 89 | |
| 90 #define ICE_TLS_DECLARE_FIELD(Type, FieldName) \ | |
| 91 static ICE_ATTRIBUTE_TLS Type FieldName | |
| 92 #define ICE_TLS_DEFINE_FIELD(Type, ClassName, FieldName) \ | |
| 93 ICE_ATTRIBUTE_TLS Type ClassName::FieldName = nullptr | |
| 94 #define ICE_TLS_INIT_FIELD(FieldName) | |
| 95 #define ICE_TLS_GET_FIELD(FieldName) (FieldName) | |
| 96 #define ICE_TLS_SET_FIELD(FieldName, Value) (FieldName = (Value)) | |
| 97 | |
| 98 #endif // !ICE_THREAD_LOCAL_HACK | |
| 99 | |
| 100 #endif // SUBZERO_SRC_ICETLS_H | |
| OLD | NEW |