Annotation of embedaddon/curl/lib/vtls/mbedtls_threadlock.c, revision 1.1
1.1 ! misho 1: /***************************************************************************
! 2: * _ _ ____ _
! 3: * Project ___| | | | _ \| |
! 4: * / __| | | | |_) | |
! 5: * | (__| |_| | _ <| |___
! 6: * \___|\___/|_| \_\_____|
! 7: *
! 8: * Copyright (C) 2013 - 2020, Daniel Stenberg, <daniel@haxx.se>, et al.
! 9: * Copyright (C) 2010, 2011, Hoi-Ho Chan, <hoiho.chan@gmail.com>
! 10: *
! 11: * This software is licensed as described in the file COPYING, which
! 12: * you should have received as part of this distribution. The terms
! 13: * are also available at https://curl.haxx.se/docs/copyright.html.
! 14: *
! 15: * You may opt to use, copy, modify, merge, publish, distribute and/or sell
! 16: * copies of the Software, and permit persons to whom the Software is
! 17: * furnished to do so, under the terms of the COPYING file.
! 18: *
! 19: * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
! 20: * KIND, either express or implied.
! 21: *
! 22: ***************************************************************************/
! 23: #include "curl_setup.h"
! 24:
! 25: #if defined(USE_MBEDTLS) && \
! 26: ((defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)) || \
! 27: (defined(USE_THREADS_WIN32) && defined(HAVE_PROCESS_H)))
! 28:
! 29: #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
! 30: # include <pthread.h>
! 31: # define MBEDTLS_MUTEX_T pthread_mutex_t
! 32: #elif defined(USE_THREADS_WIN32) && defined(HAVE_PROCESS_H)
! 33: # include <process.h>
! 34: # define MBEDTLS_MUTEX_T HANDLE
! 35: #endif
! 36:
! 37: #include "mbedtls_threadlock.h"
! 38: #include "curl_printf.h"
! 39: #include "curl_memory.h"
! 40: /* The last #include file should be: */
! 41: #include "memdebug.h"
! 42:
! 43: /* number of thread locks */
! 44: #define NUMT 2
! 45:
! 46: /* This array will store all of the mutexes available to Mbedtls. */
! 47: static MBEDTLS_MUTEX_T *mutex_buf = NULL;
! 48:
! 49: int Curl_mbedtlsthreadlock_thread_setup(void)
! 50: {
! 51: int i;
! 52:
! 53: mutex_buf = calloc(NUMT * sizeof(MBEDTLS_MUTEX_T), 1);
! 54: if(!mutex_buf)
! 55: return 0; /* error, no number of threads defined */
! 56:
! 57: for(i = 0; i < NUMT; i++) {
! 58: int ret;
! 59: #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
! 60: ret = pthread_mutex_init(&mutex_buf[i], NULL);
! 61: if(ret)
! 62: return 0; /* pthread_mutex_init failed */
! 63: #elif defined(USE_THREADS_WIN32) && defined(HAVE_PROCESS_H)
! 64: mutex_buf[i] = CreateMutex(0, FALSE, 0);
! 65: if(mutex_buf[i] == 0)
! 66: return 0; /* CreateMutex failed */
! 67: #endif /* USE_THREADS_POSIX && HAVE_PTHREAD_H */
! 68: }
! 69:
! 70: return 1; /* OK */
! 71: }
! 72:
! 73: int Curl_mbedtlsthreadlock_thread_cleanup(void)
! 74: {
! 75: int i;
! 76:
! 77: if(!mutex_buf)
! 78: return 0; /* error, no threads locks defined */
! 79:
! 80: for(i = 0; i < NUMT; i++) {
! 81: int ret;
! 82: #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
! 83: ret = pthread_mutex_destroy(&mutex_buf[i]);
! 84: if(ret)
! 85: return 0; /* pthread_mutex_destroy failed */
! 86: #elif defined(USE_THREADS_WIN32) && defined(HAVE_PROCESS_H)
! 87: ret = CloseHandle(mutex_buf[i]);
! 88: if(!ret)
! 89: return 0; /* CloseHandle failed */
! 90: #endif /* USE_THREADS_POSIX && HAVE_PTHREAD_H */
! 91: }
! 92: free(mutex_buf);
! 93: mutex_buf = NULL;
! 94:
! 95: return 1; /* OK */
! 96: }
! 97:
! 98: int Curl_mbedtlsthreadlock_lock_function(int n)
! 99: {
! 100: if(n < NUMT) {
! 101: int ret;
! 102: #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
! 103: ret = pthread_mutex_lock(&mutex_buf[n]);
! 104: if(ret) {
! 105: DEBUGF(fprintf(stderr,
! 106: "Error: mbedtlsthreadlock_lock_function failed\n"));
! 107: return 0; /* pthread_mutex_lock failed */
! 108: }
! 109: #elif defined(USE_THREADS_WIN32) && defined(HAVE_PROCESS_H)
! 110: ret = (WaitForSingleObject(mutex_buf[n], INFINITE) == WAIT_FAILED?1:0);
! 111: if(ret) {
! 112: DEBUGF(fprintf(stderr,
! 113: "Error: mbedtlsthreadlock_lock_function failed\n"));
! 114: return 0; /* pthread_mutex_lock failed */
! 115: }
! 116: #endif /* USE_THREADS_POSIX && HAVE_PTHREAD_H */
! 117: }
! 118: return 1; /* OK */
! 119: }
! 120:
! 121: int Curl_mbedtlsthreadlock_unlock_function(int n)
! 122: {
! 123: if(n < NUMT) {
! 124: int ret;
! 125: #if defined(USE_THREADS_POSIX) && defined(HAVE_PTHREAD_H)
! 126: ret = pthread_mutex_unlock(&mutex_buf[n]);
! 127: if(ret) {
! 128: DEBUGF(fprintf(stderr,
! 129: "Error: mbedtlsthreadlock_unlock_function failed\n"));
! 130: return 0; /* pthread_mutex_unlock failed */
! 131: }
! 132: #elif defined(USE_THREADS_WIN32) && defined(HAVE_PROCESS_H)
! 133: ret = ReleaseMutex(mutex_buf[n]);
! 134: if(!ret) {
! 135: DEBUGF(fprintf(stderr,
! 136: "Error: mbedtlsthreadlock_unlock_function failed\n"));
! 137: return 0; /* pthread_mutex_lock failed */
! 138: }
! 139: #endif /* USE_THREADS_POSIX && HAVE_PTHREAD_H */
! 140: }
! 141: return 1; /* OK */
! 142: }
! 143:
! 144: #endif /* USE_MBEDTLS */
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>