third_party/WebKit/Source/core/xml/XSLTProcessor.cpp

   1 /*
   2  * This file is part of the XSL implementation.
   3  *
   4  * Copyright (C) 2004, 2005, 2006, 2007, 2008 Apple, Inc. All rights reserved.
   5  * Copyright (C) 2005, 2006 Alexey Proskuryakov <ap@webkit.org>
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Library General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Library General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Library General Public License
  18  * along with this library; see the file COPYING.LIB.  If not, write to
  19  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  20  * Boston, MA 02110-1301, USA.
  21  */
  22
  23 #include "config.h"
  24 #include "core/xml/XSLTProcessor.h"
  25
  26 #include "core/dom/DOMImplementation.h"
  27 #include "core/dom/DocumentEncodingData.h"
  28 #include "core/dom/DocumentFragment.h"
  29 #include "core/editing/serializers/Serialization.h"
  30 #include "core/frame/FrameView.h"
  31 #include "core/frame/LocalDOMWindow.h"
  32 #include "core/frame/LocalFrame.h"
  33 #include "core/frame/csp/ContentSecurityPolicy.h"
  34 #include "core/loader/FrameLoaderClient.h"
  35 #include "core/xml/DocumentXSLT.h"
  36 #include "platform/weborigin/SecurityOrigin.h"
  37 #include "wtf/Assertions.h"
  38
  39 namespace blink {
  40
  41 static inline void transformTextStringToXHTMLDocumentString(String& text)
  42 {
  43     // Modify the output so that it is a well-formed XHTML document with a <pre> tag enclosing the text.
  44     text.replaceWithLiteral('&', "&amp;");
  45     text.replaceWithLiteral('<', "&lt;");
  46     text = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
  47         "<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Strict//EN\" \"http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd\">\n"
  48         "<html xmlns=\"http://www.w3.org/1999/xhtml\">\n"
  49         "<head><title/></head>\n"
  50         "<body>\n"
  51         "<pre>" + text + "</pre>\n"
  52         "</body>\n"
  53         "</html>\n";
  54 }
  55
  56 XSLTProcessor::~XSLTProcessor()
  57 {
  58 #if !ENABLE(OILPAN)
  59     // Stylesheet shouldn't outlive its root node.
  60     ASSERT(!m_stylesheetRootNode || !m_stylesheet || m_stylesheet->hasOneRef());
  61 #endif
  62 }
  63
  64 PassRefPtrWillBeRawPtr<Document> XSLTProcessor::createDocumentFromSource(const String& sourceString,
  65     const String& sourceEncoding, const String& sourceMIMEType, Node* sourceNode, LocalFrame* frame)
  66 {
  67     RefPtrWillBeRawPtr<Document> ownerDocument(sourceNode->document());
  68     bool sourceIsDocument = (sourceNode == ownerDocument.get());
  69     String documentSource = sourceString;
  70
  71     RefPtrWillBeRawPtr<Document> result = nullptr;
  72     DocumentInit init(sourceIsDocument ? ownerDocument->url() : KURL(), frame);
  73
  74     bool forceXHTML = sourceMIMEType == "text/plain";
  75     if (forceXHTML)
  76         transformTextStringToXHTMLDocumentString(documentSource);
  77
  78     if (frame) {
  79         RefPtrWillBeRawPtr<Document> oldDocument = frame->document();
  80         // Before parsing, we need to save & detach the old document and get the new document
  81         // in place. Document::detach() tears down the FrameView, so remember whether or not
  82         // there was one.
  83         bool hasView = frame->view();
  84         oldDocument->detach();
  85         // Re-create the FrameView if needed.
  86         if (hasView)
  87             frame->loader().client()->transitionToCommittedForNewPage();
  88         result = frame->localDOMWindow()->installNewDocument(sourceMIMEType, init, forceXHTML);
  89
  90         if (oldDocument) {
  91             DocumentXSLT::from(*result).setTransformSourceDocument(oldDocument.get());
  92             result->updateSecurityOrigin(oldDocument->securityOrigin());
  93             result->setCookieURL(oldDocument->cookieURL());
  94
  95             RefPtrWillBeRawPtr<ContentSecurityPolicy> csp = ContentSecurityPolicy::create();
  96             csp->copyStateFrom(oldDocument->contentSecurityPolicy());
  97             result->initContentSecurityPolicy(csp);
  98         }
  99     } else {
 100         result = LocalDOMWindow::createDocument(sourceMIMEType, init, forceXHTML);
 101     }
 102
 103     DocumentEncodingData data;
 104     data.setEncoding(sourceEncoding.isEmpty() ? UTF8Encoding() : WTF::TextEncoding(sourceEncoding));
 105     result->setEncodingData(data);
 106     result->setContent(documentSource);
 107
 108     return result.release();
 109 }
 110
 111 PassRefPtrWillBeRawPtr<Document> XSLTProcessor::transformToDocument(Node* sourceNode)
 112 {
 113     String resultMIMEType;
 114     String resultString;
 115     String resultEncoding;
 116     if (!transformToString(sourceNode, resultMIMEType, resultString, resultEncoding))
 117         return nullptr;
 118     return createDocumentFromSource(resultString, resultEncoding, resultMIMEType, sourceNode, 0);
 119 }
 120
 121 PassRefPtrWillBeRawPtr<DocumentFragment> XSLTProcessor::transformToFragment(Node* sourceNode, Document* outputDoc)
 122 {
 123     String resultMIMEType;
 124     String resultString;
 125     String resultEncoding;
 126
 127     // If the output document is HTML, default to HTML method.
 128     if (outputDoc->isHTMLDocument())
 129         resultMIMEType = "text/html";
 130
 131     if (!transformToString(sourceNode, resultMIMEType, resultString, resultEncoding))
 132         return nullptr;
 133     return createFragmentForTransformToFragment(resultString, resultMIMEType, *outputDoc);
 134 }
 135
 136 void XSLTProcessor::setParameter(const String& /*namespaceURI*/, const String& localName, const String& value)
 137 {
 138     // FIXME: namespace support?
 139     // should make a QualifiedName here but we'd have to expose the impl
 140     m_parameters.set(localName, value);
 141 }
 142
 143 String XSLTProcessor::getParameter(const String& /*namespaceURI*/, const String& localName) const
 144 {
 145     // FIXME: namespace support?
 146     // should make a QualifiedName here but we'd have to expose the impl
 147     return m_parameters.get(localName);
 148 }
 149
 150 void XSLTProcessor::removeParameter(const String& /*namespaceURI*/, const String& localName)
 151 {
 152     // FIXME: namespace support?
 153     m_parameters.remove(localName);
 154 }
 155
 156 void XSLTProcessor::reset()
 157 {
 158     m_stylesheet.clear();
 159     m_stylesheetRootNode.clear();
 160     m_parameters.clear();
 161 }
 162
 163 DEFINE_TRACE(XSLTProcessor)
 164 {
 165     visitor->trace(m_stylesheet);
 166     visitor->trace(m_stylesheetRootNode);
 167     visitor->trace(m_document);
 168 }
 169
 170 } // namespace blink