15c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)/*
25c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * This file is part of the XSL implementation.
35c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) *
45c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Copyright (C) 2004, 2005, 2006, 2007, 2008 Apple, Inc. All rights reserved.
55c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Copyright (C) 2005, 2006 Alexey Proskuryakov <ap@webkit.org>
65c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) *
75c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * This library is free software; you can redistribute it and/or
85c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * modify it under the terms of the GNU Library General Public
95c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * License as published by the Free Software Foundation; either
105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * version 2 of the License, or (at your option) any later version.
115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) *
125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * This library is distributed in the hope that it will be useful,
135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * but WITHOUT ANY WARRANTY; without even the implied warranty of
145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Library General Public License for more details.
165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) *
175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * You should have received a copy of the GNU Library General Public License
185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * along with this library; see the file COPYING.LIB.  If not, write to
195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Boston, MA 02110-1301, USA.
215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */
225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)#include "config.h"
245c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
2553e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "core/xml/XSLTProcessor.h"
2653e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)
2753e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "core/dom/DOMImplementation.h"
2853e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "core/dom/DocumentFragment.h"
2953e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "core/editing/markup.h"
3053e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "core/loader/TextResourceDecoder.h"
3153e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "core/page/ContentSecurityPolicy.h"
325267f701546148b83dfbe1d151cb184385bb5c22Torne (Richard Coles)#include "core/page/DOMWindow.h"
3353e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "core/page/Frame.h"
3453e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "core/page/FrameView.h"
35e52495584422c5edb5b2944981473a2e208da323Torne (Richard Coles)#include "weborigin/SecurityOrigin.h"
365c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
375c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)#include <wtf/Assertions.h>
385c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)#include <wtf/Vector.h>
395c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
405c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)namespace WebCore {
415c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
425c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)static inline void transformTextStringToXHTMLDocumentString(String& text)
435c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){
445c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // Modify the output so that it is a well-formed XHTML document with a <pre> tag enclosing the text.
455c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    text.replaceWithLiteral('&', "&amp;");
465c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    text.replaceWithLiteral('<', "&lt;");
475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    text = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        "<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Strict//EN\" \"http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd\">\n"
495c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        "<html xmlns=\"http://www.w3.org/1999/xhtml\">\n"
505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        "<head><title/></head>\n"
515c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        "<body>\n"
525c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        "<pre>" + text + "</pre>\n"
535c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        "</body>\n"
545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        "</html>\n";
555c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)}
565c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
575c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)XSLTProcessor::~XSLTProcessor()
585c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){
595c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // Stylesheet shouldn't outlive its root node.
605c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    ASSERT(!m_stylesheetRootNode || !m_stylesheet || m_stylesheet->hasOneRef());
615c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)}
625c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
635c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)PassRefPtr<Document> XSLTProcessor::createDocumentFromSource(const String& sourceString,
645c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    const String& sourceEncoding, const String& sourceMIMEType, Node* sourceNode, Frame* frame)
655c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){
665c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    RefPtr<Document> ownerDocument = sourceNode->document();
675c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    bool sourceIsDocument = (sourceNode == ownerDocument.get());
685c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    String documentSource = sourceString;
695c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
705c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    RefPtr<Document> result;
715c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    if (sourceMIMEType == "text/plain") {
72e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch        result = Document::create(DocumentInit(sourceIsDocument ? ownerDocument->url() : KURL(), frame));
735c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        transformTextStringToXHTMLDocumentString(documentSource);
745c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    } else
755c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        result = DOMImplementation::createDocument(sourceMIMEType, frame, sourceIsDocument ? ownerDocument->url() : KURL(), false);
765c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
775c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // Before parsing, we need to save & detach the old document and get the new document
785c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // in place. We have to do this only if we're rendering the result document.
795c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    if (frame) {
805c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        if (FrameView* view = frame->view())
815c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)            view->clear();
825c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
835c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        if (Document* oldDocument = frame->document()) {
845c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)            result->setTransformSourceDocument(oldDocument);
855c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)            result->setSecurityOrigin(oldDocument->securityOrigin());
865c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)            result->setCookieURL(oldDocument->cookieURL());
875c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)            result->contentSecurityPolicy()->copyStateFrom(oldDocument->contentSecurityPolicy());
885c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        }
895c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
905267f701546148b83dfbe1d151cb184385bb5c22Torne (Richard Coles)        frame->domWindow()->setDocument(result);
915c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    }
925c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
935c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    RefPtr<TextResourceDecoder> decoder = TextResourceDecoder::create(sourceMIMEType);
9481a5157921f1d2a7ff6aae115bfe3c139b38a5c8Torne (Richard Coles)    decoder->setEncoding(sourceEncoding.isEmpty() ? UTF8Encoding() : WTF::TextEncoding(sourceEncoding), TextResourceDecoder::EncodingFromXMLHeader);
955c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    result->setDecoder(decoder.release());
965c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
975c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    result->setContent(documentSource);
985c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
995c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    return result.release();
1005c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)}
1015c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1025c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)PassRefPtr<Document> XSLTProcessor::transformToDocument(Node* sourceNode)
1035c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){
1045c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    if (!sourceNode)
1055c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        return 0;
1065c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1075c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    String resultMIMEType;
1085c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    String resultString;
1095c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    String resultEncoding;
1105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    if (!transformToString(sourceNode, resultMIMEType, resultString, resultEncoding))
1115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        return 0;
1125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    return createDocumentFromSource(resultString, resultEncoding, resultMIMEType, sourceNode, 0);
1135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)}
1145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)PassRefPtr<DocumentFragment> XSLTProcessor::transformToFragment(Node* sourceNode, Document* outputDoc)
1165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){
1175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    if (!sourceNode || !outputDoc)
1185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        return 0;
1195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    String resultMIMEType;
1215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    String resultString;
1225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    String resultEncoding;
1235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1245c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // If the output document is HTML, default to HTML method.
1255c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    if (outputDoc->isHTMLDocument())
1265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        resultMIMEType = "text/html";
1275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    if (!transformToString(sourceNode, resultMIMEType, resultString, resultEncoding))
1295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)        return 0;
1305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    return createFragmentForTransformToFragment(resultString, resultMIMEType, outputDoc);
1315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)}
1325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1335c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)void XSLTProcessor::setParameter(const String& /*namespaceURI*/, const String& localName, const String& value)
1345c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){
1355c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // FIXME: namespace support?
1365c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // should make a QualifiedName here but we'd have to expose the impl
1375c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    m_parameters.set(localName, value);
1385c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)}
1395c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1405c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)String XSLTProcessor::getParameter(const String& /*namespaceURI*/, const String& localName) const
1415c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){
1425c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // FIXME: namespace support?
1435c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // should make a QualifiedName here but we'd have to expose the impl
1445c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    return m_parameters.get(localName);
1455c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)}
1465c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)void XSLTProcessor::removeParameter(const String& /*namespaceURI*/, const String& localName)
1485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){
1495c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    // FIXME: namespace support?
1505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    m_parameters.remove(localName);
1515c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)}
1525c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1535c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)void XSLTProcessor::reset()
1545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){
1555c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    m_stylesheet.clear();
1565c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    m_stylesheetRootNode.clear();
1575c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)    m_parameters.clear();
1585c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)}
1595c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)
1605c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} // namespace WebCore
161