1dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block/* 2dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * Copyright (C) 2010. Adam Barth. All rights reserved. 3dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * 4dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * Redistribution and use in source and binary forms, with or without 5dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * modification, are permitted provided that the following conditions 6dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * are met: 7dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * 8dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * 1. Redistributions of source code must retain the above copyright 9dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * notice, this list of conditions and the following disclaimer. 10dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * 2. Redistributions in binary form must reproduce the above copyright 11dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * notice, this list of conditions and the following disclaimer in the 12dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * documentation and/or other materials provided with the distribution. 13dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of 14dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * its contributors may be used to endorse or promote products derived 15dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * from this software without specific prior written permission. 16dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * 17dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 18dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 19dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 20dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 21dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 22dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 23dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 24dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block */ 28dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 29dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "config.h" 30dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "DocumentWriter.h" 31dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 32dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "DOMImplementation.h" 33dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "DOMWindow.h" 34dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "Frame.h" 35dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "FrameLoader.h" 36dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "FrameLoaderClient.h" 3706ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen#include "FrameLoaderStateMachine.h" 38dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "FrameView.h" 39dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "PlaceholderDocument.h" 40dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "PluginDocument.h" 41ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block#include "RawDataDocumentParser.h" 42ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block#include "ScriptableDocumentParser.h" 43dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "SecurityOrigin.h" 44dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "SegmentedString.h" 45dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "Settings.h" 46e78cbe89e6f337f2f1fe40315be88f742b547151Steve Block#include "SinkDocument.h" 47dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block#include "TextResourceDecoder.h" 48ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block 49dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 50dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blocknamespace WebCore { 51dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 52dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockstatic inline bool canReferToParentFrameEncoding(const Frame* frame, const Frame* parentFrame) 53dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 54dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block return parentFrame && parentFrame->document()->securityOrigin()->canAccess(frame->document()->securityOrigin()); 55dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 56dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 57dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve BlockDocumentWriter::DocumentWriter(Frame* frame) 58dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block : m_frame(frame) 59dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block , m_receivedData(false) 60dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block , m_encodingWasChosenByUser(false) 61dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 62dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 63dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 6406ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen// This is only called by ScriptController::executeIfJavaScriptURL 6506ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen// and always contains the result of evaluating a javascript: url. 6606ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen// This is the <iframe src="javascript:'html'"> case. 6706ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsenvoid DocumentWriter::replaceDocument(const String& source) 68dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 69dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->loader()->stopAllLoaders(); 702fc2651226baac27029e38c9d6ef883fa32084dbSteve Block begin(m_frame->document()->url(), true, m_frame->document()->securityOrigin()); 7106ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen 7206ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen if (!source.isNull()) { 7306ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen if (!m_receivedData) { 7406ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen m_receivedData = true; 755ddde30071f639962dd557c453f2ad01f8f0fd00Kristian Monsen m_frame->document()->setCompatibilityMode(Document::NoQuirksMode); 7606ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen } 7706ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen 78ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block // FIXME: This should call DocumentParser::appendBytes instead of append 79ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block // to support RawDataDocumentParsers. 8006ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen if (DocumentParser* parser = m_frame->document()->parser()) 81ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block parser->append(source); 8206ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen } 8306ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen 84dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block end(); 85dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 86dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 87dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockvoid DocumentWriter::clear() 88dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 89dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_decoder = 0; 90dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_receivedData = false; 91dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (!m_encodingWasChosenByUser) 92dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_encoding = String(); 93dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 94dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 95dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockvoid DocumentWriter::begin() 96dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 97dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block begin(KURL()); 98dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 99dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 10006ea8e899e48f1f2f396b70e63fae369f2f23232Kristian MonsenPassRefPtr<Document> DocumentWriter::createDocument(const KURL& url) 101dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 10206ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen if (!m_frame->loader()->stateMachine()->isDisplayingInitialEmptyDocument() && m_frame->loader()->client()->shouldUsePluginDocument(m_mimeType)) 10306ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen return PluginDocument::create(m_frame, url); 104dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (!m_frame->loader()->client()->hasHTMLView()) 10506ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen return PlaceholderDocument::create(m_frame, url); 10606ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen return DOMImplementation::createDocument(m_mimeType, m_frame, url, m_frame->inViewSourceMode()); 107dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 108dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 109dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockvoid DocumentWriter::begin(const KURL& url, bool dispatch, SecurityOrigin* origin) 110dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 111dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // We need to take a reference to the security origin because |clear| 112dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // might destroy the document that owns it. 113dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block RefPtr<SecurityOrigin> forcedSecurityOrigin = origin; 114dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 115dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // Create a new document before clearing the frame, because it may need to 116dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // inherit an aliased security context. 1174cdf1c24e9a5622cded19588d9821f6c41b4d5a8Steve Block#if PLATFORM(ANDROID) 1184cdf1c24e9a5622cded19588d9821f6c41b4d5a8Steve Block // Temporary hack for http://b/5188895 1194cdf1c24e9a5622cded19588d9821f6c41b4d5a8Steve Block m_frame->setDocumentIsNotUpToDate(); 1204cdf1c24e9a5622cded19588d9821f6c41b4d5a8Steve Block#endif 12106ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen RefPtr<Document> document = createDocument(url); 122e78cbe89e6f337f2f1fe40315be88f742b547151Steve Block 123e78cbe89e6f337f2f1fe40315be88f742b547151Steve Block // If the new document is for a Plugin but we're supposed to be sandboxed from Plugins, 124545e470e52f0ac6a3a072bf559c796b42c6066b6Ben Murdoch // then replace the document with one whose parser will ignore the incoming data (bug 39323) 125e78cbe89e6f337f2f1fe40315be88f742b547151Steve Block if (document->isPluginDocument() && m_frame->loader()->isSandboxed(SandboxPlugins)) 12606ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen document = SinkDocument::create(m_frame, url); 127dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 1282daae5fd11344eaa88a0d92b0f6d65f8d2255c00Ben Murdoch // FIXME: Do we need to consult the content security policy here about blocked plug-ins? 1292daae5fd11344eaa88a0d92b0f6d65f8d2255c00Ben Murdoch 13006ea8e899e48f1f2f396b70e63fae369f2f23232Kristian Monsen bool resetScripting = !(m_frame->loader()->stateMachine()->isDisplayingInitialEmptyDocument() && m_frame->document()->securityOrigin()->isSecureTransitionTo(url)); 131dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->loader()->clear(resetScripting, resetScripting); 1322fc2651226baac27029e38c9d6ef883fa32084dbSteve Block clear(); 133dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (resetScripting) 134dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->script()->updatePlatformScriptObjects(); 135dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 1362fc2651226baac27029e38c9d6ef883fa32084dbSteve Block m_frame->loader()->setOutgoingReferrer(url); 137dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->setDocument(document); 138dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 139dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (m_decoder) 140dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block document->setDecoder(m_decoder.get()); 141dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (forcedSecurityOrigin) 142dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block document->setSecurityOrigin(forcedSecurityOrigin.get()); 143dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 144dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->domWindow()->setURL(document->url()); 145dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->domWindow()->setSecurityOrigin(document->securityOrigin()); 146dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 147dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->loader()->didBeginDocument(dispatch); 148dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 149dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block document->implicitOpen(); 150dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 151dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (m_frame->view() && m_frame->loader()->client()->hasHTMLView()) 152dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->view()->setContentsSize(IntSize()); 153dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 154dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 155ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve BlockTextResourceDecoder* DocumentWriter::createDecoderIfNeeded() 156dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 157dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (!m_decoder) { 158dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (Settings* settings = m_frame->settings()) { 159dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_decoder = TextResourceDecoder::create(m_mimeType, 160dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block settings->defaultTextEncodingName(), 161dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block settings->usesEncodingDetector()); 162dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block Frame* parentFrame = m_frame->tree()->parent(); 163dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // Set the hint encoding to the parent frame encoding only if 164dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // the parent and the current frames share the security origin. 165dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // We impose this condition because somebody can make a child frame 166dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // containing a carefully crafted html/javascript in one encoding 167dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // that can be mistaken for hintEncoding (or related encoding) by 168dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // an auto detector. When interpreted in the latter, it could be 169dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // an attack vector. 170dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // FIXME: This might be too cautious for non-7bit-encodings and 171dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // we may consider relaxing this later after testing. 172dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (canReferToParentFrameEncoding(m_frame, parentFrame)) 173dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_decoder->setHintEncoding(parentFrame->document()->decoder()); 174dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block } else 175dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_decoder = TextResourceDecoder::create(m_mimeType, String()); 176dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block Frame* parentFrame = m_frame->tree()->parent(); 177dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (m_encoding.isEmpty()) { 178dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (canReferToParentFrameEncoding(m_frame, parentFrame)) 179dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_decoder->setEncoding(parentFrame->document()->inputEncoding(), TextResourceDecoder::EncodingFromParentFrame); 180dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block } else { 181dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_decoder->setEncoding(m_encoding, 182dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_encodingWasChosenByUser ? TextResourceDecoder::UserChosenEncoding : TextResourceDecoder::EncodingFromHTTPHeader); 183dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block } 184dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->document()->setDecoder(m_decoder.get()); 185dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block } 186ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block return m_decoder.get(); 187ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block} 188dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 1895ddde30071f639962dd557c453f2ad01f8f0fd00Kristian Monsenvoid DocumentWriter::reportDataReceived() 190ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block{ 191ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block ASSERT(m_decoder); 192dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (!m_receivedData) { 193dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_receivedData = true; 194dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (m_decoder->encoding().usesVisualOrdering()) 195dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->document()->setVisuallyOrdered(); 196dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->document()->recalcStyle(Node::Force); 197dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block } 198ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block} 199dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 200ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Blockvoid DocumentWriter::addData(const char* str, int len, bool flush) 201ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block{ 202ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block if (len == -1) 203ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block len = strlen(str); 204ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block 205ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block DocumentParser* parser = m_frame->document()->parser(); 206ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block if (parser) 207ca9cb53ed1119a3fd98fafa0972ffeb56dee1c24Steve Block parser->appendBytes(this, str, len, flush); 208dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 209dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 210dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockvoid DocumentWriter::end() 211dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 212dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->loader()->didEndDocument(); 213dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block endIfNotLoadingMainResource(); 214dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 215dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 216dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockvoid DocumentWriter::endIfNotLoadingMainResource() 217dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 218dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (m_frame->loader()->isLoadingMainResource() || !m_frame->page() || !m_frame->document()) 219dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block return; 220dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 221dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // http://bugs.webkit.org/show_bug.cgi?id=10854 222dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // The frame's last ref may be removed and it can be deleted by checkCompleted(), 223dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // so we'll add a protective refcount 224dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block RefPtr<Frame> protector(m_frame); 225dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 226dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block // make sure nothing's left in there 227dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block addData(0, 0, true); 228dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->document()->finishParsing(); 229dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 230dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 231dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve BlockString DocumentWriter::encoding() const 232dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 233dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (m_encodingWasChosenByUser && !m_encoding.isEmpty()) 234dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block return m_encoding; 235dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block if (m_decoder && m_decoder->encoding().isValid()) 236dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block return m_decoder->encoding().name(); 237dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block Settings* settings = m_frame->settings(); 238dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block return settings ? settings->defaultTextEncodingName() : String(); 239dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 240dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 241dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockvoid DocumentWriter::setEncoding(const String& name, bool userChosen) 242dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 243dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_frame->loader()->willSetEncoding(); 244dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_encoding = name; 245dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_encodingWasChosenByUser = userChosen; 246dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 247dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 248dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockvoid DocumentWriter::setDecoder(TextResourceDecoder* decoder) 249dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 250dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block m_decoder = decoder; 251dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 252dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 253dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve BlockString DocumentWriter::deprecatedFrameEncoding() const 254dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block{ 2552fc2651226baac27029e38c9d6ef883fa32084dbSteve Block return m_frame->document()->url().isEmpty() ? m_encoding : encoding(); 256dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} 257dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 258a94275402997c11dd2e778633dacf4b7e630a35dBen Murdochvoid DocumentWriter::setDocumentWasLoadedAsPartOfNavigation() 259a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch{ 260a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch m_frame->document()->parser()->setDocumentWasLoadedAsPartOfNavigation(); 261a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch} 262a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch 263dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block} // namespace WebCore 264