00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047 #ifndef ASTERISK_THREADSTORAGE_H
00048 #define ASTERISK_THREADSTORAGE_H
00049
00050 #include "asterisk/utils.h"
00051 #include "asterisk/inline_api.h"
00052
00053
00054
00055
00056 struct ast_threadstorage {
00057 pthread_once_t once;
00058 pthread_key_t key;
00059 void (*key_init)(void);
00060 int (*custom_init)(void *);
00061 };
00062
00063 #if defined(DEBUG_THREADLOCALS)
00064 void __ast_threadstorage_object_add(void *key, size_t len, const char *file, const char *function, unsigned int line);
00065 void __ast_threadstorage_object_remove(void *key);
00066 void __ast_threadstorage_object_replace(void *key_old, void *key_new, size_t len);
00067 #endif
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081 #define AST_THREADSTORAGE(name) \
00082 AST_THREADSTORAGE_CUSTOM_SCOPE(name, NULL, ast_free_ptr, static)
00083 #define AST_THREADSTORAGE_PUBLIC(name) \
00084 AST_THREADSTORAGE_CUSTOM_SCOPE(name, NULL, ast_free_ptr,)
00085 #define AST_THREADSTORAGE_EXTERNAL(name) \
00086 extern struct ast_threadstorage name
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101
00102
00103
00104 #define AST_THREADSTORAGE_CUSTOM(a,b,c) AST_THREADSTORAGE_CUSTOM_SCOPE(a,b,c,static)
00105
00106 #if defined(PTHREAD_ONCE_INIT_NEEDS_BRACES)
00107 # define AST_PTHREAD_ONCE_INIT { PTHREAD_ONCE_INIT }
00108 #else
00109 # define AST_PTHREAD_ONCE_INIT PTHREAD_ONCE_INIT
00110 #endif
00111
00112 #if !defined(DEBUG_THREADLOCALS)
00113 #define AST_THREADSTORAGE_CUSTOM_SCOPE(name, c_init, c_cleanup, scope) \
00114 static void __init_##name(void); \
00115 scope struct ast_threadstorage name = { \
00116 .once = AST_PTHREAD_ONCE_INIT, \
00117 .key_init = __init_##name, \
00118 .custom_init = c_init, \
00119 }; \
00120 static void __init_##name(void) \
00121 { \
00122 pthread_key_create(&(name).key, c_cleanup); \
00123 }
00124 #else
00125 #define AST_THREADSTORAGE_CUSTOM_SCOPE(name, c_init, c_cleanup, scope) \
00126 static void __init_##name(void); \
00127 scope struct ast_threadstorage name = { \
00128 .once = AST_PTHREAD_ONCE_INIT, \
00129 .key_init = __init_##name, \
00130 .custom_init = c_init, \
00131 }; \
00132 static void __cleanup_##name(void *data) \
00133 { \
00134 __ast_threadstorage_object_remove(data); \
00135 c_cleanup(data); \
00136 } \
00137 static void __init_##name(void) \
00138 { \
00139 pthread_key_create(&(name).key, __cleanup_##name); \
00140 }
00141 #endif
00142
00143
00144
00145
00146
00147
00148
00149
00150
00151
00152
00153
00154
00155
00156
00157
00158
00159
00160
00161
00162
00163
00164
00165
00166
00167
00168
00169
00170
00171
00172 #if !defined(DEBUG_THREADLOCALS)
00173 AST_INLINE_API(
00174 void *ast_threadstorage_get(struct ast_threadstorage *ts, size_t init_size),
00175 {
00176 void *buf;
00177
00178 pthread_once(&ts->once, ts->key_init);
00179 if (!(buf = pthread_getspecific(ts->key))) {
00180 if (!(buf = ast_calloc(1, init_size)))
00181 return NULL;
00182 if (ts->custom_init && ts->custom_init(buf)) {
00183 free(buf);
00184 return NULL;
00185 }
00186 pthread_setspecific(ts->key, buf);
00187 }
00188
00189 return buf;
00190 }
00191 )
00192 #else
00193 AST_INLINE_API(
00194 void *__ast_threadstorage_get(struct ast_threadstorage *ts, size_t init_size, const char *file, const char *function, unsigned int line),
00195 {
00196 void *buf;
00197
00198 pthread_once(&ts->once, ts->key_init);
00199 if (!(buf = pthread_getspecific(ts->key))) {
00200 if (!(buf = ast_calloc(1, init_size)))
00201 return NULL;
00202 if (ts->custom_init && ts->custom_init(buf)) {
00203 free(buf);
00204 return NULL;
00205 }
00206 pthread_setspecific(ts->key, buf);
00207 __ast_threadstorage_object_add(buf, init_size, file, function, line);
00208 }
00209
00210 return buf;
00211 }
00212 )
00213
00214 #define ast_threadstorage_get(ts, init_size) __ast_threadstorage_get(ts, init_size, __FILE__, __PRETTY_FUNCTION__, __LINE__)
00215 #endif
00216
00217 #endif