blob: a3aea5e13b2ec7af2ec9b7ca52d2604915f55b80 (
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
|
#pragma once
#ifdef __serenity__
#include <AK/Assertions.h>
#include <AK/Types.h>
#include <unistd.h>
#define memory_barrier() asm volatile("" :: \
: "memory")
static inline u32 CAS(volatile u32* mem, u32 newval, u32 oldval)
{
u32 ret;
asm volatile(
"cmpxchgl %2, %1"
: "=a"(ret), "+m"(*mem)
: "r"(newval), "0"(oldval)
: "cc", "memory");
return ret;
}
namespace LibThread {
class Lock {
public:
Lock() {}
~Lock() {}
void lock();
void unlock();
private:
volatile u32 m_lock { 0 };
u32 m_level { 0 };
int m_holder { -1 };
};
class Locker {
public:
[[gnu::always_inline]] inline explicit Locker(Lock& l)
: m_lock(l)
{
lock();
}
[[gnu::always_inline]] inline ~Locker() { unlock(); }
[[gnu::always_inline]] inline void unlock() { m_lock.unlock(); }
[[gnu::always_inline]] inline void lock() { m_lock.lock(); }
private:
Lock& m_lock;
};
[[gnu::always_inline]] inline void Lock::lock()
{
int tid = gettid();
for (;;) {
if (CAS(&m_lock, 1, 0) == 0) {
if (m_holder == -1 || m_holder == tid) {
m_holder = tid;
++m_level;
memory_barrier();
m_lock = 0;
return;
}
m_lock = 0;
}
donate(m_holder);
}
}
inline void Lock::unlock()
{
for (;;) {
if (CAS(&m_lock, 1, 0) == 0) {
ASSERT(m_holder == gettid());
ASSERT(m_level);
--m_level;
if (m_level) {
memory_barrier();
m_lock = 0;
return;
}
m_holder = -1;
memory_barrier();
m_lock = 0;
return;
}
donate(m_holder);
}
}
#define LOCKER(lock) LibThread::Locker locker(lock)
template<typename T>
class Lockable {
public:
Lockable() {}
Lockable(T&& resource)
: m_resource(move(resource))
{
}
Lock& lock() { return m_lock; }
T& resource() { return m_resource; }
T lock_and_copy()
{
LOCKER(m_lock);
return m_resource;
}
private:
T m_resource;
Lock m_lock;
};
}
#else
namespace LibThread {
class Lock {
public:
Lock() { }
~Lock() { }
};
}
#define LOCKER(x)
#endif
|
