mutex.h
3.8 KB
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
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
//
// mutex.h
// BBAAPIRequest
//
// Created by cairui01 on 2020/05/11.
//
#ifndef MUTEX_H_
#define MUTEX_H_
#include <errno.h>
#include <stdint.h>
#include <pthread.h>
#include <sys/time.h>
#include "time_utils.h"
class Mutex {
public:
typedef pthread_mutex_t handle_type;
Mutex(bool _recursive = false)
: magic_(reinterpret_cast<uintptr_t>(this)), mutex_(), attr_() {
//禁止重复加锁
int ret = pthread_mutexattr_init(&attr_);
ret = pthread_mutexattr_settype(&attr_, _recursive ? PTHREAD_MUTEX_RECURSIVE : PTHREAD_MUTEX_ERRORCHECK);
ret = pthread_mutex_init(&mutex_, &attr_);
}
~Mutex() {
magic_ = 0;
int ret = pthread_mutex_destroy(&mutex_);
ret = pthread_mutexattr_destroy(&attr_);
}
bool lock() {
// 成功返回0，失败返回错误码
if (reinterpret_cast<uintptr_t>(this) != magic_) return false;
int ret = pthread_mutex_lock(&mutex_);
return 0 == ret;
}
bool unlock() {
int ret = pthread_mutex_unlock(&mutex_);
return 0 == ret;
}
bool trylock() {
if (reinterpret_cast<uintptr_t>(this) != magic_) return false;
int ret = pthread_mutex_trylock(&mutex_);
if (EBUSY == ret) return false;
return 0 == ret;
}
#ifdef ANDROID
bool timedlock(long _millisecond) {
ASSERT2(reinterpret_cast<uintptr_t>(this) == magic_ && 0 != magic_, "this:%p != mageic:%p", this, (void*)magic_);
if (reinterpret_cast<uintptr_t>(this) != magic_) return false;
int ret = 0;
#if defined(ANDROID) && __ANDROID_API__ < 21 && !defined(__LP64__)
ret = pthread_mutex_lock_timeout_np(&mutex_, (unsigned)_millisecond);
#else
struct timespec ts;
MakeTimeout(&ts, _millisecond);
ret = pthread_mutex_timedlock(&mutex_, &ts);
#endif
switch (ret) {
case 0: return true;
case ETIMEDOUT: return false;
case EBUSY: return false;
case EAGAIN: ASSERT2(false, "EAGAIN"); return false;
case EDEADLK: ASSERT2(false, "EDEADLK"); return false;
case EINVAL: ASSERT2(false, "EINVAL"); return false;
default: ASSERT2(false, "%d", ret); return false;
}
return false;
}
#elif defined(__linux__)
bool timedlock(long _millisecond) {
ASSERT2(reinterpret_cast<uintptr_t>(this) == magic_ && 0 != magic_, "this:%p != mageic:%p", this, (void*)magic_);
if (reinterpret_cast<uintptr_t>(this) != magic_) return false;
struct timespec ts;
MakeTimeout(&ts, _millisecond);
int ret = pthread_mutex_timedlock(&mutex_, &ts);
switch (ret) {
case 0: return true;
case ETIMEDOUT: return false;
case EAGAIN: ASSERT2(false, "EAGAIN"); return false;
case EDEADLK: ASSERT2(false, "EDEADLK"); return false;
case EINVAL: ASSERT2(false, "EINVAL"); return false;
default: ASSERT2(false, "%d", ret); return false;
}
return false;
}
#endif
bool islocked() {
int ret = pthread_mutex_trylock(&mutex_);
if (0 == ret) unlock();
return 0 != ret;
}
handle_type& internal() { return mutex_; }
private:
Mutex(const Mutex&);
Mutex& operator = (const Mutex&);
private:
static void MakeTimeout(struct timespec* pts, long millisecond) {
struct timeval tv;
gettimeofday(&tv, 0);
pts->tv_sec = millisecond / 1000 + tv.tv_sec;
pts->tv_nsec = (millisecond % 1000) * 1000 * 1000 + tv.tv_usec * 1000;
pts->tv_sec += pts->tv_nsec / (1000 * 1000 * 1000);
pts->tv_nsec = pts->tv_nsec % (1000 * 1000 * 1000);
}
private:
uintptr_t magic_; // Dangling pointer will dead lock, so check it!!!
pthread_mutex_t mutex_;
pthread_mutexattr_t attr_;
};
#endif