Annotation of embedaddon/curl/lib/vtls/mbedtls_threadlock.c, revision 1.1.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>