blob: 45447b25aa903f3afc498800091c0635def39e75 [file] [log] [blame]
/*
* Copyright (C) 2010. Adam Barth. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of
* its contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "config.h"
#include "core/loader/DocumentWriter.h"
#include "core/dom/Document.h"
#include "core/dom/ScriptableDocumentParser.h"
#include "core/frame/FrameView.h"
#include "core/frame/LocalDOMWindow.h"
#include "core/frame/LocalFrame.h"
#include "core/frame/Settings.h"
#include "core/html/parser/TextResourceDecoder.h"
#include "core/loader/FrameLoader.h"
#include "core/loader/FrameLoaderStateMachine.h"
#include "platform/weborigin/KURL.h"
#include "platform/weborigin/SecurityOrigin.h"
#include "wtf/PassOwnPtr.h"
namespace blink {
PassRefPtrWillBeRawPtr<DocumentWriter> DocumentWriter::create(Document* document, ParserSynchronizationPolicy parsingPolicy, const AtomicString& mimeType, const AtomicString& encoding)
{
return adoptRefWillBeNoop(new DocumentWriter(document, parsingPolicy, mimeType, encoding));
}
DocumentWriter::DocumentWriter(Document* document, ParserSynchronizationPolicy parserSyncPolicy, const AtomicString& mimeType, const AtomicString& encoding)
: m_document(document)
, m_decoderBuilder(mimeType, encoding)
// We grab a reference to the parser so that we'll always send data to the
// original parser, even if the document acquires a new parser (e.g., via
// document.open).
, m_parser(m_document->implicitOpen(parserSyncPolicy))
{
if (m_document->frame()) {
if (FrameView* view = m_document->frame()->view())
view->setContentsSize(IntSize());
}
}
DocumentWriter::~DocumentWriter()
{
}
DEFINE_TRACE(DocumentWriter)
{
visitor->trace(m_document);
visitor->trace(m_parser);
}
void DocumentWriter::appendReplacingData(const String& source)
{
m_document->setCompatibilityMode(Document::NoQuirksMode);
// FIXME: This should call DocumentParser::appendBytes instead of append
// to support RawDataDocumentParsers.
if (DocumentParser* parser = m_document->parser())
parser->append(source);
}
void DocumentWriter::addData(const char* bytes, size_t length)
{
ASSERT(m_parser);
if (m_parser->needsDecoder() && 0 < length) {
OwnPtr<TextResourceDecoder> decoder = m_decoderBuilder.buildFor(m_document);
m_parser->setDecoder(decoder.release());
}
// appendBytes() can result replacing DocumentLoader::m_writer.
m_parser->appendBytes(bytes, length);
}
void DocumentWriter::end()
{
ASSERT(m_document);
// http://bugs.webkit.org/show_bug.cgi?id=10854
// The frame's last ref may be removed and it can be deleted by checkCompleted(),
// so we'll add a protective refcount
RefPtrWillBeRawPtr<LocalFrame> protect(m_document->frame());
if (!m_parser)
return;
if (m_parser->needsDecoder()) {
OwnPtr<TextResourceDecoder> decoder = m_decoderBuilder.buildFor(m_document);
m_parser->setDecoder(decoder.release());
}
// finish() can result replacing DocumentLoader::m_writer.
RefPtrWillBeRawPtr<DocumentWriter> protectingThis(this);
m_parser->finish();
m_parser = nullptr;
m_document = nullptr;
}
void DocumentWriter::setDocumentWasLoadedAsPartOfNavigation()
{
ASSERT(m_parser && !m_parser->isStopped());
m_parser->setDocumentWasLoadedAsPartOfNavigation();
}
} // namespace blink