mirror of
https://github.com/samba-team/samba.git
synced 2025-01-10 01:18:15 +03:00
Add the thread functions to top level lib/util.
Not yet used, that's the next step. Jeremy.
This commit is contained in:
Jeremy Allison
parent
d8a6ea8141
commit
e5702da72c
105
lib/util/smb_threads.c
Normal file
105
lib/util/smb_threads.c
Normal file
@ -0,0 +1,105 @@
|
||||
/*
|
||||
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"
|
||||
|
||||
#define NUM_GLOBAL_LOCKS 1
|
||||
|
||||
/*********************************************************
|
||||
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;
|
||||
|
||||
/* Here we initialize any static locks we're using. */
|
||||
global_lock_array = (void **)SMB_MALLOC_ARRAY(void *, NUM_GLOBAL_LOCKS);
|
||||
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;
|
||||
}
|
||||
global_tfp->create_mutex(name,
|
||||
&global_lock_array[i],
|
||||
__location__);
|
||||
SAFE_FREE(name);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
#if 0
|
||||
/* Test. - pthread implementations. */
|
||||
#include <pthread.h>
|
||||
|
||||
#ifdef malloc
|
||||
#undef malloc
|
||||
#endif
|
||||
|
||||
SMB_THREADS_DEF_PTHREAD_IMPLEMENTATION(tf);
|
||||
|
||||
/* Test function. */
|
||||
int test_threads(void)
|
||||
{
|
||||
int ret;
|
||||
void *plock = NULL;
|
||||
|
||||
smb_thread_set_functions(&tf);
|
||||
|
||||
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);
|
||||
|
||||
return 0;
|
||||
}
|
||||
#endif
|
||||
120
lib/util/smb_threads.h
Normal file
120
lib/util/smb_threads.h
Normal file
@ -0,0 +1,120 @@
|
||||
/*
|
||||
Unix SMB/CIFS implementation.
|
||||
SMB 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/>.
|
||||
*/
|
||||
|
||||
#ifndef _smb_threads_h_
|
||||
#define _smb_threads_h_
|
||||
|
||||
enum smb_thread_lock_type {
|
||||
SMB_THREAD_LOCK = 1,
|
||||
SMB_THREAD_UNLOCK
|
||||
};
|
||||
|
||||
struct smb_thread_functions {
|
||||
/* Mutex and tls functions. */
|
||||
int (*create_mutex)(const char *lockname,
|
||||
void **pplock,
|
||||
const char *location);
|
||||
void (*destroy_mutex)(void *plock,
|
||||
const char *location);
|
||||
int (*lock_mutex)(void *plock, enum smb_thread_lock_type lock_type,
|
||||
const char *location);
|
||||
|
||||
/* Thread local storage. */
|
||||
int (*create_tls)(const char *keyname,
|
||||
void **ppkey,
|
||||
const char *location);
|
||||
void (*destroy_tls)(void *pkey,
|
||||
const char *location);
|
||||
int (*set_tls)(void *pkey, const void *pval, const char *location);
|
||||
void *(*get_tls)(void *pkey, const char *location);
|
||||
};
|
||||
|
||||
extern const struct smb_thread_functions *global_tfp;
|
||||
|
||||
/* Define the pthread version of the functions. */
|
||||
|
||||
#define SMB_THREADS_DEF_PTHREAD_IMPLEMENTATION(tf) \
|
||||
\
|
||||
static int smb_create_mutex_pthread(const char *lockname, void **pplock, const char *location) \
|
||||
{ \
|
||||
pthread_mutex_t *pmut = (pthread_mutex_t *)malloc(sizeof(pthread_mutex_t)); \
|
||||
if (!pmut) { \
|
||||
return ENOMEM; \
|
||||
} \
|
||||
pthread_mutex_init(pmut, NULL); \
|
||||
*pplock = (void *)pmut; \
|
||||
return 0; \
|
||||
} \
|
||||
\
|
||||
static void smb_destroy_mutex_pthread(void *plock, const char *location) \
|
||||
{ \
|
||||
pthread_mutex_destroy((pthread_mutex_t *)plock); \
|
||||
free(plock); \
|
||||
} \
|
||||
\
|
||||
static int smb_lock_pthread(void *plock, enum smb_thread_lock_type lock_type, const char *location) \
|
||||
{ \
|
||||
if (lock_type == SMB_THREAD_UNLOCK) { \
|
||||
return pthread_mutex_unlock((pthread_mutex_t *)plock); \
|
||||
} else { \
|
||||
return pthread_mutex_lock((pthread_mutex_t *)plock); \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
static int smb_create_tls_pthread(const char *keyname, void **ppkey, const char *location) \
|
||||
{ \
|
||||
int ret; \
|
||||
pthread_key_t *pkey = (pthread_key_t *)malloc(sizeof(pthread_key_t)); \
|
||||
if (!pkey) { \
|
||||
return ENOMEM; \
|
||||
} \
|
||||
ret = pthread_key_create(pkey, NULL); \
|
||||
if (ret) { \
|
||||
return ret; \
|
||||
} \
|
||||
*ppkey = (void *)pkey; \
|
||||
return 0; \
|
||||
} \
|
||||
\
|
||||
static void smb_destroy_tls_pthread(void *pkey, const char *location) \
|
||||
{ \
|
||||
pthread_key_delete(*(pthread_key_t *)pkey); \
|
||||
free(pkey); \
|
||||
} \
|
||||
\
|
||||
static int smb_set_tls_pthread(void *pkey, const void *pval, const char *location) \
|
||||
{ \
|
||||
return pthread_setspecific(*(pthread_key_t *)pkey, pval); \
|
||||
} \
|
||||
\
|
||||
static void *smb_get_tls_pthread(void *pkey, const char *location) \
|
||||
{ \
|
||||
return pthread_getspecific(*(pthread_key_t *)pkey); \
|
||||
} \
|
||||
\
|
||||
static const struct smb_thread_functions (tf) = { \
|
||||
smb_create_mutex_pthread, \
|
||||
smb_destroy_mutex_pthread, \
|
||||
smb_lock_pthread, \
|
||||
smb_create_tls_pthread, \
|
||||
smb_destroy_tls_pthread, \
|
||||
smb_set_tls_pthread, \
|
||||
smb_get_tls_pthread }
|
||||
|
||||
#endif
|
||||
52
lib/util/smb_threads_internal.h
Normal file
52
lib/util/smb_threads_internal.h
Normal file
@ -0,0 +1,52 @@
|
||||
/*
|
||||
SMB/CIFS implementation.
|
||||
SMB thread interface internal macros.
|
||||
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/>.
|
||||
*/
|
||||
|
||||
#ifndef _smb_threads_internal_h_
|
||||
#define _smb_threads_internal_h_
|
||||
|
||||
#define SMB_THREAD_CREATE_MUTEX(name, lockvar) \
|
||||
(global_tfp ? global_tfp->create_mutex((name), &(lockvar), __location__) : 0)
|
||||
|
||||
#define SMB_THREAD_DESTROY_MUTEX(plock) \
|
||||
do { \
|
||||
if (global_tfp) { \
|
||||
global_tfp->destroy_mutex(plock, __location__); \
|
||||
}; \
|
||||
} while (0)
|
||||
|
||||
#define SMB_THREAD_LOCK(plock, type) \
|
||||
(global_tfp ? global_tfp->lock_mutex((plock), (type), __location__) : 0)
|
||||
|
||||
#define SMB_THREAD_CREATE_TLS(keyname, key) \
|
||||
(global_tfp ? global_tfp->create_tls((keyname), &(key), __location__) : 0)
|
||||
|
||||
#define SMB_THREAD_DESTROY_TLS(key) \
|
||||
do { \
|
||||
if (global_tfp) { \
|
||||
global_tfp->destroy_tls(key); \
|
||||
}; \
|
||||
} while (0)
|
||||
|
||||
#define SMB_THREAD_SET_TLS(key, val) \
|
||||
(global_tfp ? global_tfp->set_tls((key),(val),__location__) : 0)
|
||||
|
||||
#define SMB_THREAD_GET_TLS(key) \
|
||||
(global_tfp ? global_tfp->get_tls((key), __location__) : NULL)
|
||||
|
||||
#endif
|
||||
Loading…
Reference in New Issue
Block a user