809673ab544b245afe6c653061862525decab4a7
[ira/wip.git] / lib / util / smb_threads.h
1 /*
2         Unix SMB/CIFS implementation.
3         SMB thread interface functions.
4         Copyright (C) Jeremy Allison, 2009.
5
6         This program is free software; you can redistribute it and/or modify
7         it under the terms of the GNU General Public License as published by
8         the Free Software Foundation; either version 3 of the License, or
9         (at your option) any later version.
10
11         This program is distributed in the hope that it will be useful,
12         but WITHOUT ANY WARRANTY; without even the implied warranty of
13         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14         GNU General Public License for more details.
15
16         You should have received a copy of the GNU General Public License
17         along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #ifndef _smb_threads_h_
21 #define _smb_threads_h_
22
23 typedef bool smb_thread_once_t;
24 #define SMB_THREAD_ONCE_INIT false
25 #define SMB_THREAD_ONCE_IS_INITIALIZED(val) ((val) == true)
26 #define SMB_THREAD_ONCE_INITIALIZE(val) ((val) = true)
27
28 enum smb_thread_lock_type {
29         SMB_THREAD_LOCK = 1,
30         SMB_THREAD_UNLOCK
31 };
32
33 struct smb_thread_functions {
34         /* Mutex and tls functions. */
35         int (*create_mutex)(const char *lockname,
36                         void **pplock,
37                         const char *location);
38         void (*destroy_mutex)(void *plock,
39                         const char *location);
40         int (*lock_mutex)(void *plock,
41                           int lock_type,
42                           const char *location);
43
44         /* Thread local storage. */
45         int (*create_tls)(const char *keyname,
46                         void **ppkey,
47                         const char *location);
48         void (*destroy_tls)(void **pkey,
49                         const char *location);
50         int (*set_tls)(void *pkey, const void *pval, const char *location);
51         void *(*get_tls)(void *pkey, const char *location);
52 };
53
54 int smb_thread_set_functions(const struct smb_thread_functions *tf);
55 int smb_thread_once(smb_thread_once_t *ponce,
56                     void (*init_fn)(void *pdata),
57                     void *pdata);
58
59 extern const struct smb_thread_functions *global_tfp;
60
61 /* Define the pthread version of the functions. */
62
63 #define SMB_THREADS_DEF_PTHREAD_IMPLEMENTATION(tf) \
64  \
65 static int smb_create_mutex_pthread(const char *lockname, void **pplock, const char *location) \
66 { \
67         pthread_mutex_t *pmut = (pthread_mutex_t *)malloc(sizeof(pthread_mutex_t)); \
68         if (!pmut) { \
69                 return ENOMEM; \
70         } \
71         pthread_mutex_init(pmut, NULL); \
72         *pplock = (void *)pmut; \
73         return 0; \
74 } \
75  \
76 static void smb_destroy_mutex_pthread(void *plock, const char *location) \
77 { \
78         pthread_mutex_destroy((pthread_mutex_t *)plock); \
79         free(plock); \
80 } \
81  \
82 static int smb_lock_pthread(void *plock, int lock_type, const char *location) \
83 { \
84         if (lock_type == SMB_THREAD_UNLOCK) { \
85                 return pthread_mutex_unlock((pthread_mutex_t *)plock); \
86         } else { \
87                 return pthread_mutex_lock((pthread_mutex_t *)plock); \
88         } \
89 } \
90  \
91 static int smb_create_tls_pthread(const char *keyname, void **ppkey, const char *location) \
92 { \
93         int ret; \
94         pthread_key_t *pkey; \
95         pkey = (pthread_key_t *)malloc(sizeof(pthread_key_t)); \
96         if (!pkey) { \
97                 return ENOMEM; \
98         } \
99         ret = pthread_key_create(pkey, NULL); \
100         if (ret) { \
101                 free(pkey); \
102                 return ret; \
103         } \
104         *ppkey = (void *)pkey; \
105         return 0; \
106 } \
107  \
108 static void smb_destroy_tls_pthread(void **ppkey, const char *location) \
109 { \
110         if (*ppkey) { \
111                 pthread_key_delete(*(pthread_key_t *)ppkey); \
112                 free(*ppkey); \
113                 *ppkey = NULL; \
114         } \
115 } \
116  \
117 static int smb_set_tls_pthread(void *pkey, const void *pval, const char *location) \
118 { \
119         return pthread_setspecific(*(pthread_key_t *)pkey, pval); \
120 } \
121  \
122 static void *smb_get_tls_pthread(void *pkey, const char *location) \
123 { \
124         return pthread_getspecific(*(pthread_key_t *)pkey); \
125 } \
126  \
127 static const struct smb_thread_functions (tf) = { \
128                         smb_create_mutex_pthread, \
129                         smb_destroy_mutex_pthread, \
130                         smb_lock_pthread, \
131                         smb_create_tls_pthread, \
132                         smb_destroy_tls_pthread, \
133                         smb_set_tls_pthread, \
134                         smb_get_tls_pthread }
135
136 #endif