constructArray() should always allocate the requested length.
[WebKit-https.git] / Source / JavaScriptCore / runtime / DisallowScope.h
1 /*
2  * Copyright (C) 2017-2018 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     void disable()
70     {
71         m_isEnabled = false;
72         exitScope();
73     }
74
75 private:
76     void enterScope()
77     {
78         auto count = T::scopeReentryCount();
79         T::setScopeReentryCount(++count);
80     }
81
82     void exitScope()
83     {
84         auto count = T::scopeReentryCount();
85         ASSERT(count);
86         T::setScopeReentryCount(--count);
87     }
88
89     bool m_isEnabled;
90 #endif // NDEBUG
91 };
92
93 } // namespace JSC