Remove excessive headers from JavaScriptCore
[WebKit-https.git] / Source / JavaScriptCore / runtime / DisallowScope.h
1 /*
2  * Copyright (C) 2017 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #pragma once
27
28 #include <wtf/Noncopyable.h>
29
30 namespace JSC {
31
32 template<class T>
33 class DisallowScope {
34     WTF_MAKE_NONCOPYABLE(DisallowScope);
35 public:
36 #ifdef NDEBUG
37
38     ALWAYS_INLINE DisallowScope(bool = false) { }
39     ALWAYS_INLINE ~DisallowScope() { }
40     ALWAYS_INLINE static bool isInEffectOnCurrentThread() { return false; }
41     ALWAYS_INLINE void enable() { }
42
43 #else // not NDEBUG
44
45     DisallowScope(bool enabled = true)
46     {
47         m_isEnabled = enabled;
48         if (m_isEnabled)
49             enterScope();
50     }
51
52     ~DisallowScope()
53     {
54         if (m_isEnabled)
55             exitScope();
56     }
57
58     static bool isInEffectOnCurrentThread()
59     {
60         return !!T::scopeReentryCount();
61     }
62
63     void enable()
64     {
65         m_isEnabled = true;
66         enterScope();
67     }
68
69 private:
70     void enterScope()
71     {
72         auto count = T::scopeReentryCount();
73         T::setScopeReentryCount(++count);
74     }
75
76     void exitScope()
77     {
78         auto count = T::scopeReentryCount();
79         ASSERT(count);
80         T::setScopeReentryCount(--count);
81     }
82
83     bool m_isEnabled;
84 #endif // NDEBUG
85 };
86
87 } // namespace JSC