Recursively defining custom elements should not lead to redefinition

third_party/WebKit/Source/core/dom/custom/CustomElementsRegistry.cpp

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "core/dom/custom/CustomElementsRegistry.h"
 
 #include "bindings/core/v8/ExceptionState.h"
 #include "bindings/core/v8/ScriptCustomElementDefinitionBuilder.h"
 #include "core/dom/Document.h"
 #include "core/dom/Element.h"
 #include "core/dom/ElementRegistrationOptions.h"
 #include "core/dom/ExceptionCode.h"
 #include "core/dom/custom/CEReactionsScope.h"
 #include "core/dom/custom/CustomElement.h"
 #include "core/dom/custom/CustomElementDefinition.h"
 #include "core/dom/custom/CustomElementDefinitionBuilder.h"
 #include "core/dom/custom/CustomElementDescriptor.h"
 #include "core/dom/custom/CustomElementUpgradeReaction.h"
 #include "core/dom/custom/CustomElementUpgradeSorter.h"
 #include "core/dom/custom/V0CustomElementRegistrationContext.h"
+#include "wtf/Allocator.h"
 
 namespace blink {
 
+class CustomElementsRegistry::NameIsBeingDefined final {
+    STACK_ALLOCATED();
+    DISALLOW_IMPLICIT_CONSTRUCTORS(NameIsBeingDefined);
+public:
+    NameIsBeingDefined(
+        CustomElementsRegistry* registry,
+        const AtomicString& name)
+        : m_registry(registry)
+        , m_name(name)
+    {
+        DCHECK(!m_registry->m_namesBeingDefined.contains(name));
+        m_registry->m_namesBeingDefined.add(name);
+    }
+
+    ~NameIsBeingDefined()
+    {
+        m_registry->m_namesBeingDefined.remove(m_name);
+    }
+
+private:
+    Member<CustomElementsRegistry> m_registry;
+    const AtomicString& m_name;
+};
+
 CustomElementsRegistry* CustomElementsRegistry::create(
     Document* document)
 {
     CustomElementsRegistry* registry = new CustomElementsRegistry(document);
     if (V0CustomElementRegistrationContext* v0Context = registry->v0())
         v0Context->setV1(registry);
     return registry;
 }
 
 CustomElementsRegistry::CustomElementsRegistry(Document* document)
@@ skipping 36 matching lines @@
     if (!builder.checkConstructorIntrinsics())
         return;
 
     if (!CustomElement::isValidName(name)) {
         exceptionState.throwDOMException(
             SyntaxError,
             "\"" + name + "\" is not a valid custom element name");
         return;
     }
 
+    if (m_namesBeingDefined.contains(name)) {
+        exceptionState.throwDOMException(
+            NotSupportedError,
+            "this name is already being defined in this registry");
+        return;
+    }
+    NameIsBeingDefined defining(this, name);
+
     if (nameIsDefined(name) || v0NameIsDefined(name)) {
         exceptionState.throwDOMException(
             NotSupportedError,
             "this name has already been used with this registry");
         return;
     }
 
     if (!builder.checkConstructorNotRegistered())
         return;
 
@@ skipping 100 matching lines @@
         if (!element || !desc.matches(*element))
             continue;
         sorter.add(element);
     }
 
     m_upgradeCandidates->remove(it);
     sorter.sorted(elements, m_document.get());
 }
 
 } // namespace blink