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>