blob: aceef4daaa9d6e80462791fd613c8aadeba5a733 (
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
|
#include "Semaphore.h"
#include <cassert>
#ifndef CRU_WINDOWS
#include <errno.h>
#endif
namespace cru {
Semaphore::Semaphore(unsigned init_value) {
#ifdef CRU_WINDOWS
handle_ = CreateSemaphoreW(nullptr, init_value, 100, nullptr);
assert(handle_);
#else
semaphore_ = std::make_unique<sem_t>();
auto c = sem_init(semaphore_.get(), 0, init_value);
assert(c == 0);
#endif
}
Semaphore::Semaphore(Semaphore &&other)
#ifdef CRU_WINDOWS
: handle_(other.handle_)
#else
: semaphore_(std::move(other.semaphore_))
#endif
{
#ifdef CRU_WINDOWS
other.handle_ = nullptr;
#endif
}
Semaphore &Semaphore::operator=(Semaphore &&other) {
if (this != &other) {
Destroy();
#ifdef CRU_WINDOWS
handle_ = other.handle_;
other.handle_ = nullptr;
#else
semaphore_ = std::move(other.semaphore_);
#endif
}
return *this;
}
Semaphore::~Semaphore() { Destroy(); }
void Semaphore::P() { Acquire(); }
void Semaphore::V() { Release(); }
void Semaphore::Acquire() {
#ifdef CRU_WINDOWS
auto c = WaitForSingleObject(handle_, INFINITE);
assert(c == WAIT_OBJECT_0);
#else
auto c = sem_wait(semaphore_.get());
assert(c == 0);
#endif
}
bool Semaphore::TryAcquire() {
#ifdef CRU_WINDOWS
auto c = WaitForSingleObject(handle_, 0);
assert(c == WAIT_OBJECT_0 || c == WAIT_TIMEOUT);
return c == WAIT_OBJECT_0 ? true : false;
#else
auto c = sem_trywait(semaphore_.get());
assert((c == 0) || (c == -1 && errno == EAGAIN));
return c == 0 ? true : false;
#endif
}
void Semaphore::Release() {
#ifdef CRU_WINDOWS
auto c = ReleaseSemaphore(handle_, 1, nullptr);
assert(c);
#else
auto c = sem_post(semaphore_.get());
assert(c == 0);
#endif
}
void Semaphore::Destroy() {
#ifdef CRU_WINDOWS
if (handle_ != nullptr) {
auto c = CloseHandle(handle_);
assert(c);
handle_ = nullptr;
}
#else
if (semaphore_ != nullptr) {
auto c = sem_destroy(semaphore_.get());
assert(c == 0);
semaphore_ = nullptr;
}
#endif
}
} // namespace cru
|
