PluginDocument.cpp [plain text]
#include "config.h"
#include "PluginDocument.h"
#include "DocumentLoader.h"
#include "Frame.h"
#include "FrameLoader.h"
#include "FrameLoaderClient.h"
#include "FrameView.h"
#include "HTMLBodyElement.h"
#include "HTMLEmbedElement.h"
#include "HTMLHtmlElement.h"
#include "HTMLNames.h"
#include "Page.h"
#include "RawDataDocumentParser.h"
#include "RenderEmbeddedObject.h"
namespace WebCore {
using namespace HTMLNames;
class PluginDocumentParser final : public RawDataDocumentParser {
public:
static Ref<PluginDocumentParser> create(PluginDocument& document)
{
return adoptRef(*new PluginDocumentParser(document));
}
private:
PluginDocumentParser(Document& document)
: RawDataDocumentParser(document)
, m_embedElement(0)
{
}
void appendBytes(DocumentWriter&, const char*, size_t) override;
void createDocumentStructure();
HTMLEmbedElement* m_embedElement;
};
void PluginDocumentParser::createDocumentStructure()
{
auto& document = downcast<PluginDocument>(*this->document());
auto rootElement = HTMLHtmlElement::create(document);
document.appendChild(rootElement);
rootElement->insertedByParser();
if (document.frame())
document.frame()->injectUserScripts(InjectAtDocumentStart);
#if PLATFORM(IOS)
document.processViewport(ASCIILiteral("user-scalable=no"), ViewportArguments::PluginDocument);
#endif
auto body = HTMLBodyElement::create(document);
body->setAttributeWithoutSynchronization(marginwidthAttr, AtomicString("0", AtomicString::ConstructFromLiteral));
body->setAttributeWithoutSynchronization(marginheightAttr, AtomicString("0", AtomicString::ConstructFromLiteral));
#if PLATFORM(IOS)
body->setAttribute(styleAttr, AtomicString("background-color: rgb(217,224,233)", AtomicString::ConstructFromLiteral));
#else
body->setAttribute(styleAttr, AtomicString("background-color: rgb(38,38,38)", AtomicString::ConstructFromLiteral));
#endif
rootElement->appendChild(body);
auto embedElement = HTMLEmbedElement::create(document);
m_embedElement = embedElement.ptr();
embedElement->setAttributeWithoutSynchronization(widthAttr, AtomicString("100%", AtomicString::ConstructFromLiteral));
embedElement->setAttributeWithoutSynchronization(heightAttr, AtomicString("100%", AtomicString::ConstructFromLiteral));
embedElement->setAttributeWithoutSynchronization(nameAttr, AtomicString("plugin", AtomicString::ConstructFromLiteral));
embedElement->setAttributeWithoutSynchronization(srcAttr, document.url().string());
ASSERT(document.loader());
if (auto* loader = document.loader())
m_embedElement->setAttributeWithoutSynchronization(typeAttr, loader->writer().mimeType());
document.setPluginElement(m_embedElement);
body->appendChild(embedElement);
}
void PluginDocumentParser::appendBytes(DocumentWriter&, const char*, size_t)
{
if (m_embedElement)
return;
createDocumentStructure();
Frame* frame = document()->frame();
if (!frame)
return;
document()->updateLayout();
frame->view()->flushAnyPendingPostLayoutTasks();
if (RenderWidget* renderer = m_embedElement->renderWidget()) {
if (Widget* widget = renderer->widget()) {
frame->loader().client().redirectDataToPlugin(widget);
frame->loader().activeDocumentLoader()->setMainResourceDataBufferingPolicy(DoNotBufferData);
}
}
}
PluginDocument::PluginDocument(Frame* frame, const URL& url)
: HTMLDocument(frame, url, PluginDocumentClass)
, m_shouldLoadPluginManually(true)
{
setCompatibilityMode(DocumentCompatibilityMode::QuirksMode);
lockCompatibilityMode();
}
Ref<DocumentParser> PluginDocument::createParser()
{
return PluginDocumentParser::create(*this);
}
Widget* PluginDocument::pluginWidget()
{
if (m_pluginElement && m_pluginElement->renderer())
return downcast<RenderEmbeddedObject>(*m_pluginElement->renderer()).widget();
return nullptr;
}
void PluginDocument::setPluginElement(PassRefPtr<HTMLPlugInElement> element)
{
m_pluginElement = element;
}
void PluginDocument::detachFromPluginElement()
{
m_pluginElement = nullptr;
frame()->loader().client().redirectDataToPlugin(nullptr);
}
void PluginDocument::cancelManualPluginLoad()
{
if (!shouldLoadPluginManually())
return;
DocumentLoader* documentLoader = frame()->loader().activeDocumentLoader();
documentLoader->cancelMainResourceLoad(frame()->loader().cancelledError(documentLoader->request()));
setShouldLoadPluginManually(false);
}
}