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