sjero.net Git - iperf/blob - include/Mutex.h

   1 /*---------------------------------------------------------------
   2  * Copyright (c) 1999,2000,2001,2002,2003
   3  * The Board of Trustees of the University of Illinois
   4  * All Rights Reserved.
   5  *---------------------------------------------------------------
   6  * Permission is hereby granted, free of charge, to any person
   7  * obtaining a copy of this software (Iperf) and associated
   8  * documentation files (the "Software"), to deal in the Software
   9  * without restriction, including without limitation the
  10  * rights to use, copy, modify, merge, publish, distribute,
  11  * sublicense, and/or sell copies of the Software, and to permit
  12  * persons to whom the Software is furnished to do
  13  * so, subject to the following conditions:
  14  *
  15  *
  16  * Redistributions of source code must retain the above
  17  * copyright notice, this list of conditions and
  18  * the following disclaimers.
  19  *
  20  *
  21  * Redistributions in binary form must reproduce the above
  22  * copyright notice, this list of conditions and the following
  23  * disclaimers in the documentation and/or other materials
  24  * provided with the distribution.
  25  *
  26  *
  27  * Neither the names of the University of Illinois, NCSA,
  28  * nor the names of its contributors may be used to endorse
  29  * or promote products derived from this Software without
  30  * specific prior written permission.
  31  *
  32  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  33  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
  34  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  35  * NONINFRINGEMENT. IN NO EVENT SHALL THE CONTIBUTORS OR COPYRIGHT
  36  * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  37  * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  38  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  39  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  40  * ________________________________________________________________
  41  * National Laboratory for Applied Network Research
  42  * National Center for Supercomputing Applications
  43  * University of Illinois at Urbana-Champaign
  44  * http://www.ncsa.uiuc.edu
  45  * ________________________________________________________________
  46  *
  47  * Mutex.h
  48  * by Mark Gates <mgates@nlanr.net>
  49  * -------------------------------------------------------------------
  50  * An abstract class for locking a mutex (mutual exclusion). If
  51  * threads are not available, this does nothing.
  52  * ------------------------------------------------------------------- */
  53 #ifndef MUTEX_H
  54 #define MUTEX_H
  55
  56 #include "headers.h"
  57
  58 #if   defined( HAVE_POSIX_THREAD )
  59     typedef pthread_mutex_t Mutex;
  60 #elif defined( HAVE_WIN32_THREAD )
  61     typedef HANDLE Mutex;
  62 #else
  63     typedef int Mutex;
  64 #endif
  65
  66 /* ------------------------------------------------------------------- *
  67 class Mutex {
  68 public:*/
  69
  70     // initialize mutex
  71 #if   defined( HAVE_POSIX_THREAD )
  72     #define Mutex_Initialize( MutexPtr ) pthread_mutex_init( MutexPtr, NULL )
  73 #elif defined( HAVE_WIN32_THREAD )
  74     #define Mutex_Initialize( MutexPtr ) *MutexPtr = CreateMutex( NULL, false, NULL )
  75 #else
  76     #define Mutex_Initialize( MutexPtr )
  77 #endif
  78
  79     // lock the mutex variable
  80 #if   defined( HAVE_POSIX_THREAD )
  81     #define Mutex_Lock( MutexPtr ) pthread_mutex_lock( MutexPtr )
  82 #elif defined( HAVE_WIN32_THREAD )
  83     #define Mutex_Lock( MutexPtr ) WaitForSingleObject( *MutexPtr, INFINITE )
  84 #else
  85     #define Mutex_Lock( MutexPtr )
  86 #endif
  87
  88     // unlock the mutex variable
  89 #if   defined( HAVE_POSIX_THREAD )
  90     #define Mutex_Unlock( MutexPtr ) pthread_mutex_unlock( MutexPtr )
  91 #elif defined( HAVE_WIN32_THREAD )
  92     #define Mutex_Unlock( MutexPtr ) ReleaseMutex( *MutexPtr )
  93 #else
  94     #define Mutex_Unlock( MutexPtr )
  95 #endif
  96
  97     // destroy, making sure mutex is unlocked
  98 #if   defined( HAVE_POSIX_THREAD )
  99     #define Mutex_Destroy( MutexPtr )  do {         \
 100         int rc = pthread_mutex_destroy( MutexPtr ); \
 101         if ( rc == EBUSY ) {                        \
 102             Mutex_Unlock( MutexPtr );               \
 103             pthread_mutex_destroy( MutexPtr );      \
 104         }                                           \
 105     } while ( 0 )
 106 #elif defined( HAVE_WIN32_THREAD )
 107     #define Mutex_Destroy( MutexPtr ) CloseHandle( *MutexPtr )
 108 #else
 109     #define Mutex_Destroy( MutexPtr )
 110 #endif
 111
 112 #endif // MUTEX_H