blob: 661d084800092951ead26754261c438991b94b9a (
plain)
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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
/*
* Copyright (C) 2013-2019 Argonne National Laboratory, Department of Energy,
* UChicago Argonne, LLC and The HDF Group.
* All rights reserved.
*
* The full copyright notice, including terms governing use, modification,
* and redistribution, is contained in the COPYING file that can be
* found at the root of the source code distribution tree.
*/
#ifndef MERCURY_THREAD_SPIN_H
#define MERCURY_THREAD_SPIN_H
#include "mercury_util_config.h"
#if defined(_WIN32)
# include <windows.h>
typedef volatile LONG hg_thread_spin_t;
#elif defined(HG_UTIL_HAS_PTHREAD_SPINLOCK_T)
# include <pthread.h>
typedef pthread_spinlock_t hg_thread_spin_t;
#else
/* Default to hg_thread_mutex_t if pthread_spinlock_t is not supported */
# include "mercury_thread_mutex.h"
typedef hg_thread_mutex_t hg_thread_spin_t;
#endif
#ifdef __cplusplus
extern "C" {
#endif
/**
* Initialize the spin lock.
*
* \param lock [IN/OUT] pointer to lock object
*
* \return Non-negative on success or negative on failure
*/
HG_UTIL_PUBLIC int
hg_thread_spin_init(hg_thread_spin_t *lock);
/**
* Destroy the spin lock.
*
* \param lock [IN/OUT] pointer to lock object
*
* \return Non-negative on success or negative on failure
*/
HG_UTIL_PUBLIC int
hg_thread_spin_destroy(hg_thread_spin_t *lock);
/**
* Lock the spin lock.
*
* \param lock [IN/OUT] pointer to lock object
*
* \return Non-negative on success or negative on failure
*/
static HG_UTIL_INLINE int
hg_thread_spin_lock(hg_thread_spin_t *lock);
/**
* Try locking the spin lock.
*
* \param mutex [IN/OUT] pointer to lock object
*
* \return Non-negative on success or negative on failure
*/
static HG_UTIL_INLINE int
hg_thread_spin_try_lock(hg_thread_spin_t *lock);
/**
* Unlock the spin lock.
*
* \param mutex [IN/OUT] pointer to lock object
*
* \return Non-negative on success or negative on failure
*/
static HG_UTIL_INLINE int
hg_thread_spin_unlock(hg_thread_spin_t *lock);
/*---------------------------------------------------------------------------*/
static HG_UTIL_INLINE int
hg_thread_spin_lock(hg_thread_spin_t *lock)
{
#if defined(_WIN32)
while (InterlockedExchange(lock, EBUSY)) {
/* Don't lock while waiting */
while (*lock) {
YieldProcessor();
/* Compiler barrier. Prevent caching of *lock */
MemoryBarrier();
}
}
return HG_UTIL_SUCCESS;
#elif defined(HG_UTIL_HAS_PTHREAD_SPINLOCK_T)
if (pthread_spin_lock(lock))
return HG_UTIL_FAIL;
return HG_UTIL_SUCCESS;
#else
return hg_thread_mutex_lock(lock);
#endif
}
/*---------------------------------------------------------------------------*/
static HG_UTIL_INLINE int
hg_thread_spin_try_lock(hg_thread_spin_t *lock)
{
#if defined(_WIN32)
return InterlockedExchange(lock, EBUSY);
#elif defined(HG_UTIL_HAS_PTHREAD_SPINLOCK_T)
if (pthread_spin_trylock(lock))
return HG_UTIL_FAIL;
return HG_UTIL_SUCCESS;
#else
return hg_thread_mutex_try_lock(lock);
#endif
}
/*---------------------------------------------------------------------------*/
static HG_UTIL_INLINE int
hg_thread_spin_unlock(hg_thread_spin_t *lock)
{
#if defined(_WIN32)
/* Compiler barrier. The store below acts with release semantics */
MemoryBarrier();
*lock = 0;
return HG_UTIL_SUCCESS;
#elif defined(HG_UTIL_HAS_PTHREAD_SPINLOCK_T)
if (pthread_spin_unlock(lock))
return HG_UTIL_FAIL;
return HG_UTIL_SUCCESS;
#else
return hg_thread_mutex_unlock(lock);
#endif
}
#ifdef __cplusplus
}
#endif
#endif /* MERCURY_THREAD_SPIN_H */
|
