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
|
//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef TEST_SUPPORT_CHECKING_MUTEX_H
#define TEST_SUPPORT_CHECKING_MUTEX_H
#include <cassert>
#include <chrono>
struct checking_mutex {
enum state {
locked_via_lock,
locked_via_try_lock,
locked_via_try_lock_for,
locked_via_try_lock_until,
unlocked,
none,
};
state current_state = unlocked;
state last_try = none;
bool reject = false;
checking_mutex() = default;
checking_mutex(const checking_mutex&) = delete;
~checking_mutex() { assert(current_state == unlocked); }
void lock() {
assert(current_state == unlocked);
assert(!reject);
current_state = locked_via_lock;
last_try = locked_via_lock;
reject = true;
}
void unlock() {
assert(current_state != unlocked && current_state != none);
last_try = unlocked;
current_state = unlocked;
reject = false;
}
bool try_lock() {
last_try = locked_via_try_lock;
if (reject)
return false;
current_state = locked_via_try_lock;
return true;
}
template <class Rep, class Period>
bool try_lock_for(const std::chrono::duration<Rep, Period>&) {
last_try = locked_via_try_lock_for;
if (reject)
return false;
current_state = locked_via_try_lock_for;
return true;
}
template <class Clock, class Duration>
bool try_lock_until(const std::chrono::time_point<Clock, Duration>&) {
last_try = locked_via_try_lock_until;
if (reject)
return false;
current_state = locked_via_try_lock_until;
return true;
}
checking_mutex* operator&() = delete;
template <class T>
void operator,(const T&) = delete;
};
#endif // TEST_SUPPORT_CHECKING_MUTEX_H
|
