123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134 |
- #include "curl_setup.h"
- #if defined(USE_MBEDTLS) && \
- ((defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)) || \
- defined(_WIN32))
- #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
- # include <pthread.h>
- # define MBEDTLS_MUTEX_T pthread_mutex_t
- #elif defined(_WIN32)
- # define MBEDTLS_MUTEX_T HANDLE
- #endif
- #include "mbedtls_threadlock.h"
- #include "curl_printf.h"
- #include "curl_memory.h"
- #include "memdebug.h"
- #define NUMT 2
- static MBEDTLS_MUTEX_T *mutex_buf = NULL;
- int Curl_mbedtlsthreadlock_thread_setup(void)
- {
- int i;
- mutex_buf = calloc(1, NUMT * sizeof(MBEDTLS_MUTEX_T));
- if(!mutex_buf)
- return 0;
- for(i = 0; i < NUMT; i++) {
- #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
- if(pthread_mutex_init(&mutex_buf[i], NULL))
- return 0;
- #elif defined(_WIN32)
- mutex_buf[i] = CreateMutex(0, FALSE, 0);
- if(mutex_buf[i] == 0)
- return 0;
- #endif
- }
- return 1;
- }
- int Curl_mbedtlsthreadlock_thread_cleanup(void)
- {
- int i;
- if(!mutex_buf)
- return 0;
- for(i = 0; i < NUMT; i++) {
- #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
- if(pthread_mutex_destroy(&mutex_buf[i]))
- return 0;
- #elif defined(_WIN32)
- if(!CloseHandle(mutex_buf[i]))
- return 0;
- #endif
- }
- free(mutex_buf);
- mutex_buf = NULL;
- return 1;
- }
- int Curl_mbedtlsthreadlock_lock_function(int n)
- {
- if(n < NUMT) {
- #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
- if(pthread_mutex_lock(&mutex_buf[n])) {
- DEBUGF(fprintf(stderr,
- "Error: mbedtlsthreadlock_lock_function failed\n"));
- return 0;
- }
- #elif defined(_WIN32)
- if(WaitForSingleObject(mutex_buf[n], INFINITE) == WAIT_FAILED) {
- DEBUGF(fprintf(stderr,
- "Error: mbedtlsthreadlock_lock_function failed\n"));
- return 0;
- }
- #endif
- }
- return 1;
- }
- int Curl_mbedtlsthreadlock_unlock_function(int n)
- {
- if(n < NUMT) {
- #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
- if(pthread_mutex_unlock(&mutex_buf[n])) {
- DEBUGF(fprintf(stderr,
- "Error: mbedtlsthreadlock_unlock_function failed\n"));
- return 0;
- }
- #elif defined(_WIN32)
- if(!ReleaseMutex(mutex_buf[n])) {
- DEBUGF(fprintf(stderr,
- "Error: mbedtlsthreadlock_unlock_function failed\n"));
- return 0;
- }
- #endif
- }
- return 1;
- }
- #endif
|