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
Curl_mbedtlsthreadlock_thread_setup(void)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
Curl_mbedtlsthreadlock_thread_cleanup(void)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
Curl_mbedtlsthreadlock_lock_function(int n)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
Curl_mbedtlsthreadlock_unlock_function(int n)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 */
145