1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-23 17:34:34 +03:00
samba-mirror/lib/util/smb_threads.c
Jeremy Allison 8e06e945b6 Ensure a tls key is only generated once - wrap create & destroy in a mutex.
Change the function names to add _once to the tls_create & tls_destroy to
make this obvious.
Jeremy.
2009-04-14 15:19:39 -07:00

122 lines
3.1 KiB
C

/*
Unix SMB/CIFS implementation.
SMB client library implementation (thread interface functions).
Copyright (C) Jeremy Allison, 2009.
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
/*
* This code is based in the ideas in openssl
* but somewhat simpler and expended to include
* thread local storage.
*/
#include "includes.h"
/*********************************************************
Functions to vector the locking primitives used internally
by libsmbclient.
*********************************************************/
const struct smb_thread_functions *global_tfp;
/*********************************************************
Dynamic lock array.
*********************************************************/
void **global_lock_array;
/*********************************************************
Function to set the locking primitives used by libsmbclient.
*********************************************************/
int smb_thread_set_functions(const struct smb_thread_functions *tf)
{
int i;
global_tfp = tf;
#if defined(PARANOID_MALLOC_CHECKER)
#ifdef malloc
#undef malloc
#endif
#endif
/* Here we initialize any static locks we're using. */
global_lock_array = (void **)malloc(sizeof(void *) *NUM_GLOBAL_LOCKS);
#if defined(PARANOID_MALLOC_CHECKER)
#define malloc(s) __ERROR_DONT_USE_MALLOC_DIRECTLY
#endif
if (global_lock_array == NULL) {
return ENOMEM;
}
for (i = 0; i < NUM_GLOBAL_LOCKS; i++) {
char *name = NULL;
if (asprintf(&name, "global_lock_%d", i) == -1) {
SAFE_FREE(global_lock_array);
return ENOMEM;
}
if (global_tfp->create_mutex(name,
&global_lock_array[i],
__location__)) {
smb_panic("smb_thread_set_functions: create mutexes failed");
}
SAFE_FREE(name);
}
return 0;
}
#if 0
/* Test. - pthread implementations. */
#include <pthread.h>
#ifdef malloc
#undef malloc
#endif
SMB_THREADS_DEF_PTHREAD_IMPLEMENTATION(tf);
void *pkey = NULL;
/* Test function. */
int test_threads(void)
{
int ret;
void *plock = NULL;
smb_thread_set_functions(&tf);
if ((ret = SMB_THREAD_CREATE_TLS_ONCE("test_tls", pkey)) != 0) {
printf("Create tls once error: %d\n", ret);
}
if ((ret = SMB_THREAD_CREATE_MUTEX("test", plock)) != 0) {
printf("Create lock error: %d\n", ret);
}
if ((ret = SMB_THREAD_LOCK(plock, SMB_THREAD_LOCK)) != 0) {
printf("lock error: %d\n", ret);
}
if ((SMB_THREAD_LOCK(plock, SMB_THREAD_UNLOCK)) != 0) {
printf("unlock error: %d\n", ret);
}
SMB_THREAD_DESTROY_MUTEX(plock);
SMB_THREAD_DESTROY_TLS_ONCE(pkey);
return 0;
}
#endif