Implement document.currentScript

Source/core/dom/ScriptElement.cpp

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
  *           (C) 2001 Dirk Mueller (mueller@kde.org)
  * Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008 Apple Inc. All rights reserved.
  * Copyright (C) 2008 Nikolas Zimmermann <zimmermann@kde.org>
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Library General Public
  * License as published by the Free Software Foundation; either
@@ skipping 264 matching lines @@
     }
 
     if (m_cachedScript) {
         return true;
     }
 
     dispatchErrorEvent();
     return false;
 }
 
+bool isHTMLScriptElement(Element* element)
+{
+    return element->isHTMLElement() && element->hasTagName(HTMLNames::scriptTag);
+}
+
 void ScriptElement::executeScript(const ScriptSourceCode& sourceCode)
 {
     ASSERT(m_alreadyStarted);
 
     if (sourceCode.isEmpty())
         return;
 
     RefPtr<Document> document = m_element->document();
     Frame* frame = document->frame();
 
     bool shouldBypassMainWorldContentSecurityPolicy = (frame && frame->script()->shouldBypassMainWorldContentSecurityPolicy());
     if (!shouldBypassMainWorldContentSecurityPolicy && !document->contentSecurityPolicy()->allowScriptNonce(m_element->fastGetAttribute(HTMLNames::nonceAttr), document->url(), m_startLineNumber))
         return;
 
     if (!m_isExternalScript && (!shouldBypassMainWorldContentSecurityPolicy && !document->contentSecurityPolicy()->allowInlineScript(document->url(), m_startLineNumber)))
         return;
 
     if (m_isExternalScript && m_cachedScript && !m_cachedScript->mimeTypeAllowedByNosniff()) {
         document->addConsoleMessage(SecurityMessageSource, ErrorMessageLevel, "Refused to execute script from '" + m_cachedScript->url().elidedString() + "' because its MIME type ('" + m_cachedScript->mimeType() + "') is not executable, and strict MIME type checking is enabled.");
         return;
     }
 
     if (frame) {
         {
             IgnoreDestructiveWriteCountIncrementer ignoreDesctructiveWriteCountIncrementer(m_isExternalScript ? document.get() : 0);
+
+            if (isHTMLScriptElement(m_element))
+                document->setCurrentScript(static_cast<HTMLScriptElement*>(m_element));
+
             // Create a script from the script element node, using the script
             // block's source and the script block's type.
             // Note: This is where the script is compiled and actually executed.
             frame->script()->evaluate(sourceCode);
+
+            document->clearCurrentScript();

[esprehn, 2013/05/15 16:08:16]

You're sure this is safe without checking if (document->currentScript() == this)
before it?

[arv (Not doing code reviews), 2013/05/15 16:18:11]

There is a case where this will fail. When you doc.write inline script it id
evaluated synchronously. I'll have to add tests and a stack to handle this case
correctly.

         }
     }
 }
 
 void ScriptElement::stopLoadRequest()
 {
     if (m_cachedScript) {
         if (!m_willBeParserExecuted)
             m_cachedScript->removeClient(this);
         m_cachedScript = 0;
@@ skipping 86 matching lines @@
     }
 
     if (firstTextNode && !foundMultipleTextNodes)
         return firstTextNode->data();
 
     return content.toString();
 }
 
 ScriptElement* toScriptElementIfPossible(Element* element)
 {
-    if (element->isHTMLElement() && element->hasTagName(HTMLNames::scriptTag))
+    if (isHTMLScriptElement(element))
         return static_cast<HTMLScriptElement*>(element);
 
 #if ENABLE(SVG)
     if (element->isSVGElement() && element->hasTagName(SVGNames::scriptTag))
         return static_cast<SVGScriptElement*>(element);
 #endif
 
     return 0;
 }
 
 }