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 |
|
---|
26 | enum smb_thread_lock_type {
|
---|
27 | SMB_THREAD_LOCK = 1,
|
---|
28 | SMB_THREAD_UNLOCK
|
---|
29 | };
|
---|
30 |
|
---|
31 | struct smb_thread_functions {
|
---|
32 | /* Mutex and tls functions. */
|
---|
33 | int (*create_mutex)(const char *lockname,
|
---|
34 | void **pplock,
|
---|
35 | const char *location);
|
---|
36 | void (*destroy_mutex)(void *plock,
|
---|
37 | const char *location);
|
---|
38 | int (*lock_mutex)(void *plock,
|
---|
39 | int lock_type,
|
---|
40 | const char *location);
|
---|
41 |
|
---|
42 | /* Thread local storage. */
|
---|
43 | int (*create_tls)(const char *keyname,
|
---|
44 | void **ppkey,
|
---|
45 | const char *location);
|
---|
46 | void (*destroy_tls)(void **pkey,
|
---|
47 | const char *location);
|
---|
48 | int (*set_tls)(void *pkey, const void *pval, const char *location);
|
---|
49 | void *(*get_tls)(void *pkey, const char *location);
|
---|
50 | };
|
---|
51 |
|
---|
52 | int smb_thread_set_functions(const struct smb_thread_functions *tf);
|
---|
53 | int smb_thread_once(smb_thread_once_t *ponce,
|
---|
54 | void (*init_fn)(void *pdata),
|
---|
55 | void *pdata);
|
---|
56 |
|
---|
57 | extern const struct smb_thread_functions *global_tfp;
|
---|
58 |
|
---|
59 | /* Define the pthread version of the functions. */
|
---|
60 |
|
---|
61 | #define SMB_THREADS_DEF_PTHREAD_IMPLEMENTATION(tf) \
|
---|
62 | \
|
---|
63 | static int smb_create_mutex_pthread(const char *lockname, void **pplock, const char *location) \
|
---|
64 | { \
|
---|
65 | pthread_mutex_t *pmut = (pthread_mutex_t *)malloc(sizeof(pthread_mutex_t)); \
|
---|
66 | if (!pmut) { \
|
---|
67 | return ENOMEM; \
|
---|
68 | } \
|
---|
69 | pthread_mutex_init(pmut, NULL); \
|
---|
70 | *pplock = (void *)pmut; \
|
---|
71 | return 0; \
|
---|
72 | } \
|
---|
73 | \
|
---|
74 | static void smb_destroy_mutex_pthread(void *plock, const char *location) \
|
---|
75 | { \
|
---|
76 | pthread_mutex_destroy((pthread_mutex_t *)plock); \
|
---|
77 | free(plock); \
|
---|
78 | } \
|
---|
79 | \
|
---|
80 | static int smb_lock_pthread(void *plock, int lock_type, const char *location) \
|
---|
81 | { \
|
---|
82 | if (lock_type == SMB_THREAD_UNLOCK) { \
|
---|
83 | return pthread_mutex_unlock((pthread_mutex_t *)plock); \
|
---|
84 | } else { \
|
---|
85 | return pthread_mutex_lock((pthread_mutex_t *)plock); \
|
---|
86 | } \
|
---|
87 | } \
|
---|
88 | \
|
---|
89 | static int smb_create_tls_pthread(const char *keyname, void **ppkey, const char *location) \
|
---|
90 | { \
|
---|
91 | int ret; \
|
---|
92 | pthread_key_t *pkey; \
|
---|
93 | pkey = (pthread_key_t *)malloc(sizeof(pthread_key_t)); \
|
---|
94 | if (!pkey) { \
|
---|
95 | return ENOMEM; \
|
---|
96 | } \
|
---|
97 | ret = pthread_key_create(pkey, NULL); \
|
---|
98 | if (ret) { \
|
---|
99 | free(pkey); \
|
---|
100 | return ret; \
|
---|
101 | } \
|
---|
102 | *ppkey = (void *)pkey; \
|
---|
103 | return 0; \
|
---|
104 | } \
|
---|
105 | \
|
---|
106 | static void smb_destroy_tls_pthread(void **ppkey, const char *location) \
|
---|
107 | { \
|
---|
108 | if (*ppkey) { \
|
---|
109 | pthread_key_delete(*(pthread_key_t *)ppkey); \
|
---|
110 | free(*ppkey); \
|
---|
111 | *ppkey = NULL; \
|
---|
112 | } \
|
---|
113 | } \
|
---|
114 | \
|
---|
115 | static int smb_set_tls_pthread(void *pkey, const void *pval, const char *location) \
|
---|
116 | { \
|
---|
117 | return pthread_setspecific(*(pthread_key_t *)pkey, pval); \
|
---|
118 | } \
|
---|
119 | \
|
---|
120 | static void *smb_get_tls_pthread(void *pkey, const char *location) \
|
---|
121 | { \
|
---|
122 | return pthread_getspecific(*(pthread_key_t *)pkey); \
|
---|
123 | } \
|
---|
124 | \
|
---|
125 | static const struct smb_thread_functions (tf) = { \
|
---|
126 | smb_create_mutex_pthread, \
|
---|
127 | smb_destroy_mutex_pthread, \
|
---|
128 | smb_lock_pthread, \
|
---|
129 | smb_create_tls_pthread, \
|
---|
130 | smb_destroy_tls_pthread, \
|
---|
131 | smb_set_tls_pthread, \
|
---|
132 | smb_get_tls_pthread }
|
---|
133 |
|
---|
134 | #endif
|
---|