bmalloc
[WebKit-https.git] / Source / bmalloc / bmalloc / PerThread.h
1 /*
2  * Copyright (C) 2014 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #ifndef PerThread_h
27 #define PerThread_h
28
29 #include "Inline.h"
30 #include <mutex>
31 #include <pthread.h>
32 #if defined(__has_include) && __has_include(<System/pthread_machdep.h>)
33 #include <System/pthread_machdep.h>
34 #endif
35
36 namespace bmalloc {
37
38 // Usage:
39 //     Object* object = PerThread<Object>::get();
40
41 template<typename T>
42 class PerThread {
43 public:
44     static T* get();
45     static T* getFastCase();
46     static T* getSlowCase();
47
48 private:
49     static void destructor(void*);
50 };
51
52 class Cache;
53
54 template<typename T> struct PerThreadStorage;
55
56 #if defined(__has_include) && __has_include(<System/pthread_machdep.h>)
57
58 // For now, we only support PerThread<Cache>. We can expand to other types by
59 // using more keys.
60
61 template<> struct PerThreadStorage<Cache> {
62     static const pthread_key_t key = __PTK_FRAMEWORK_JAVASCRIPTCORE_KEY0;
63     static void* get() { return _pthread_getspecific_direct(key); }
64     static void init(void* object, void (*destructor)(void*))
65     {
66         _pthread_setspecific_direct(key, object);
67         pthread_key_init_np(key, destructor);
68     }
69 };
70
71 #else
72
73 template<typename T> struct PerThreadStorage {
74     static __thread void* object;
75     static pthread_key_t key;
76     static std::once_flag onceFlag;
77
78     static void* get() { return object; }
79     static void init(void* object, void (*destructor)(void*))
80     {
81         std::call_once(onceFlag, [destructor]() {
82             pthread_key_create(&key, destructor);
83         });
84         pthread_setspecific(key, object);
85         PerThreadStorage<Cache>::object = object;
86     }
87 };
88
89 template<typename T> __thread void* PerThreadStorage<T>::object;
90 template<typename T> pthread_key_t PerThreadStorage<T>::key;
91 template<typename T> std::once_flag PerThreadStorage<T>::onceFlag;
92
93 #endif // defined(__has_include) && __has_include(<System/pthread_machdep.h>)
94
95 template<typename T>
96 INLINE T* PerThread<T>::getFastCase()
97 {
98     return static_cast<T*>(PerThreadStorage<T>::get());
99 }
100
101 template<typename T>
102 inline T* PerThread<T>::get()
103 {
104     T* t = getFastCase();
105     if (!t)
106         return getSlowCase();
107     return t;
108 }
109
110 template<typename T>
111 void PerThread<T>::destructor(void* p)
112 {
113     T* t = static_cast<T*>(p);
114     delete t;
115 }
116
117 template<typename T>
118 T* PerThread<T>::getSlowCase()
119 {
120     ASSERT(!getFastCase());
121     T* t = new T;
122     PerThreadStorage<T>::init(t, destructor);
123     return t;
124 }
125
126 } // namespace bmalloc
127
128 #endif // PerThread_h