cannam@148: // Copyright (c) 2014, Jason Choy <jjwchoy@gmail.com>
cannam@148: // Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors
cannam@148: // Licensed under the MIT License:
cannam@148: //
cannam@148: // Permission is hereby granted, free of charge, to any person obtaining a copy
cannam@148: // of this software and associated documentation files (the "Software"), to deal
cannam@148: // in the Software without restriction, including without limitation the rights
cannam@148: // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
cannam@148: // copies of the Software, and to permit persons to whom the Software is
cannam@148: // furnished to do so, subject to the following conditions:
cannam@148: //
cannam@148: // The above copyright notice and this permission notice shall be included in
cannam@148: // all copies or substantial portions of the Software.
cannam@148: //
cannam@148: // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
cannam@148: // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
cannam@148: // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
cannam@148: // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
cannam@148: // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
cannam@148: // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
cannam@148: // THE SOFTWARE.
cannam@148: 
cannam@148: #ifndef KJ_THREADLOCAL_H_
cannam@148: #define KJ_THREADLOCAL_H_
cannam@148: 
cannam@148: #if defined(__GNUC__) && !KJ_HEADER_WARNINGS
cannam@148: #pragma GCC system_header
cannam@148: #endif
cannam@148: // This file declares a macro `KJ_THREADLOCAL_PTR` for declaring thread-local pointer-typed
cannam@148: // variables.  Use like:
cannam@148: //     KJ_THREADLOCAL_PTR(MyType) foo = nullptr;
cannam@148: // This is equivalent to:
cannam@148: //     thread_local MyType* foo = nullptr;
cannam@148: // This can only be used at the global scope.
cannam@148: //
cannam@148: // AVOID USING THIS.  Use of thread-locals is discouraged because they often have many of the same
cannam@148: // properties as singletons: http://www.object-oriented-security.org/lets-argue/singletons
cannam@148: //
cannam@148: // Also, thread-locals tend to be hostile to event-driven code, which can be particularly
cannam@148: // surprising when using fibers (all fibers in the same thread will share the same threadlocals,
cannam@148: // even though they do not share a stack).
cannam@148: //
cannam@148: // That said, thread-locals are sometimes needed for runtime logistics in the KJ framework.  For
cannam@148: // example, the current exception callback and current EventLoop are stored as thread-local
cannam@148: // pointers.  Since KJ only ever needs to store pointers, not values, we avoid the question of
cannam@148: // whether these values' destructors need to be run, and we avoid the need for heap allocation.
cannam@148: 
cannam@148: #include "common.h"
cannam@148: 
cannam@148: #if !defined(KJ_USE_PTHREAD_THREADLOCAL) && defined(__APPLE__)
cannam@148: #include "TargetConditionals.h"
cannam@148: #if TARGET_OS_IPHONE
cannam@148: // iOS apparently does not support __thread (nor C++11 thread_local).
cannam@148: #define KJ_USE_PTHREAD_TLS 1
cannam@148: #endif
cannam@148: #endif
cannam@148: 
cannam@148: #if KJ_USE_PTHREAD_TLS
cannam@148: #include <pthread.h>
cannam@148: #endif
cannam@148: 
cannam@148: namespace kj {
cannam@148: 
cannam@148: #if KJ_USE_PTHREAD_TLS
cannam@148: // If __thread is unavailable, we'll fall back to pthreads.
cannam@148: 
cannam@148: #define KJ_THREADLOCAL_PTR(type) \
cannam@148:   namespace { struct KJ_UNIQUE_NAME(_kj_TlpTag); } \
cannam@148:   static ::kj::_::ThreadLocalPtr< type, KJ_UNIQUE_NAME(_kj_TlpTag)>
cannam@148: // Hack:  In order to ensure each thread-local results in a unique template instance, we declare
cannam@148: //   a one-off dummy type to use as the second type parameter.
cannam@148: 
cannam@148: namespace _ {  // private
cannam@148: 
cannam@148: template <typename T, typename>
cannam@148: class ThreadLocalPtr {
cannam@148:   // Hacky type to emulate __thread T*.  We need a separate instance of the ThreadLocalPtr template
cannam@148:   // for every thread-local variable, because we don't want to require a global constructor, and in
cannam@148:   // order to initialize the TLS on first use we need to use a local static variable (in getKey()).
cannam@148:   // Each template instance will get a separate such local static variable, fulfilling our need.
cannam@148: 
cannam@148: public:
cannam@148:   ThreadLocalPtr() = default;
cannam@148:   constexpr ThreadLocalPtr(decltype(nullptr)) {}
cannam@148:   // Allow initialization to nullptr without a global constructor.
cannam@148: 
cannam@148:   inline ThreadLocalPtr& operator=(T* val) {
cannam@148:     pthread_setspecific(getKey(), val);
cannam@148:     return *this;
cannam@148:   }
cannam@148: 
cannam@148:   inline operator T*() const {
cannam@148:     return get();
cannam@148:   }
cannam@148: 
cannam@148:   inline T& operator*() const {
cannam@148:     return *get();
cannam@148:   }
cannam@148: 
cannam@148:   inline T* operator->() const {
cannam@148:     return get();
cannam@148:   }
cannam@148: 
cannam@148: private:
cannam@148:   inline T* get() const {
cannam@148:     return reinterpret_cast<T*>(pthread_getspecific(getKey()));
cannam@148:   }
cannam@148: 
cannam@148:   inline static pthread_key_t getKey() {
cannam@148:     static pthread_key_t key = createKey();
cannam@148:     return key;
cannam@148:   }
cannam@148: 
cannam@148:   static pthread_key_t createKey() {
cannam@148:     pthread_key_t key;
cannam@148:     pthread_key_create(&key, 0);
cannam@148:     return key;
cannam@148:   }
cannam@148: };
cannam@148: 
cannam@148: }  // namespace _ (private)
cannam@148: 
cannam@148: #elif __GNUC__
cannam@148: 
cannam@148: #define KJ_THREADLOCAL_PTR(type) static __thread type*
cannam@148: // GCC's __thread is lighter-weight than thread_local and is good enough for our purposes.
cannam@148: 
cannam@148: #else
cannam@148: 
cannam@148: #define KJ_THREADLOCAL_PTR(type) static thread_local type*
cannam@148: 
cannam@148: #endif // KJ_USE_PTHREAD_TLS
cannam@148: 
cannam@148: }  // namespace kj
cannam@148: 
cannam@148: #endif  // KJ_THREADLOCAL_H_