blob: d398f137c84f1464492e721109abcc75e32e8d4d (
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
|
// Copyright (C) Microsoft Corporation. All rights reserved.
#pragma once
#include <KCriticalRegion.h>
#include <KPushLock.h>
class KRTL_CLASS KLockHolder
{
private:
enum { Unlocked, Shared, Exclusive } m_State;
public:
PAGED KLockHolder(KPushLockBase &lock) : m_Lock(lock), m_State(Unlocked) { }
PAGED ~KLockHolder()
{
switch (m_State)
{
case Shared:
ReleaseShared();
break;
case Exclusive:
ReleaseExclusive();
break;
}
}
KLockHolder(KLockHolder &) = delete;
KLockHolder &operator=(KLockHolder &) = delete;
_IRQL_requires_(PASSIVE_LEVEL)
PAGED void AcquireShared()
{
m_Region.Enter();
ASSERT(m_State == Unlocked);
m_Lock.AcquireShared();
m_State = Shared;
}
_IRQL_requires_(PASSIVE_LEVEL)
PAGED void ReleaseShared()
{
ASSERT(m_State == Shared);
m_Lock.ReleaseShared();
m_State = Unlocked;
m_Region.Leave();
}
_IRQL_requires_(PASSIVE_LEVEL)
PAGED void AcquireExclusive()
{
m_Region.Enter();
ASSERT(m_State == Unlocked);
m_Lock.AcquireExclusive();
m_State = Exclusive;
}
_IRQL_requires_(PASSIVE_LEVEL)
PAGED void ReleaseExclusive()
{
ASSERT(m_State == Exclusive);
m_Lock.ReleaseExclusive();
m_State = Unlocked;
m_Region.Leave();
}
private:
KPushLockBase &m_Lock;
KCriticalRegion m_Region;
};
class KRTL_CLASS KLockThisShared : protected KLockHolder
{
public:
PAGED KLockThisShared(KPushLockBase &lock) : KLockHolder(lock)
{
AcquireShared();
}
PAGED void Acquire()
{
AcquireShared();
}
PAGED void Release()
{
ReleaseShared();
}
};
class KRTL_CLASS KLockThisExclusive : protected KLockHolder
{
public:
PAGED KLockThisExclusive(KPushLockBase &lock) : KLockHolder(lock)
{
AcquireExclusive();
}
PAGED void Acquire()
{
AcquireExclusive();
}
PAGED void Release()
{
ReleaseExclusive();
}
};
|
