third_party/WebKit/Source/platform/ScriptForbiddenScope.cpp

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "config.h"
   6 #include "platform/ScriptForbiddenScope.h"
   7
   8 #include "wtf/Assertions.h"
   9 #include "wtf/MainThread.h"
  10
  11 namespace blink {
  12
  13 static unsigned s_scriptForbiddenCount = 0;
  14
  15 ScriptForbiddenScope::ScriptForbiddenScope()
  16 {
  17     ASSERT(isMainThread());
  18     ++s_scriptForbiddenCount;
  19 }
  20
  21 ScriptForbiddenScope::~ScriptForbiddenScope()
  22 {
  23     ASSERT(isMainThread());
  24     ASSERT(s_scriptForbiddenCount);
  25     --s_scriptForbiddenCount;
  26 }
  27
  28 void ScriptForbiddenScope::enter()
  29 {
  30     ASSERT(isMainThread());
  31     ++s_scriptForbiddenCount;
  32 }
  33
  34 void ScriptForbiddenScope::exit()
  35 {
  36     ASSERT(isMainThread());
  37     ASSERT(s_scriptForbiddenCount);
  38     --s_scriptForbiddenCount;
  39 }
  40
  41 bool ScriptForbiddenScope::isScriptForbidden()
  42 {
  43     return isMainThread() && s_scriptForbiddenCount;
  44 }
  45
  46 ScriptForbiddenScope::AllowUserAgentScript::AllowUserAgentScript()
  47 {
  48     if (isMainThread())
  49         m_change.emplace(s_scriptForbiddenCount, 0);
  50 }
  51
  52 ScriptForbiddenScope::AllowUserAgentScript::~AllowUserAgentScript()
  53 {
  54     ASSERT(!isMainThread() || !s_scriptForbiddenCount);
  55 }
  56
  57 } // namespace blink