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