network/netadaptercx/netvadapterlibrary/code/rtl/KCriticalRegion.h


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

// Copyright (C) Microsoft Corporation. All rights reserved.
#pragma once

#include <KMacros.h>

class KCriticalRegion
{
public:

    PAGED KCriticalRegion() : m_Entered(false) { }

    PAGED ~KCriticalRegion() { if (m_Entered) Leave(); }

    KCriticalRegion(KCriticalRegion &) = delete;
    KCriticalRegion &operator=(KCriticalRegion &) = delete;

    PAGED void Enter()
    {
        ASSERT(m_Entered == false);
        UnbalancedEnter();
        m_Entered = true;
    }

    PAGED void Leave()
    {
        ASSERT(m_Entered == true);
        m_Entered = false;
        UnbalancedLeave();
    }

    static PAGED void UnbalancedEnter()
    {
#if _KERNEL_MODE
        KeEnterCriticalRegion();
#endif
    }

    static PAGED void UnbalancedLeave()
    {
#if _KERNEL_MODE
        KeLeaveCriticalRegion();
#endif
    }

private:

    bool m_Entered;
};

struct KDefaultRegion
{
    void Enter() { }
    void Leave() { }
};

struct KIrqlRegion
{
    KIrqlRegion() { }
    ~KIrqlRegion() { }

    void Enter() { }
    void Leave() { }

    KIRQL m_OldIrql;
};