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 "RawDataDocumentParser.h"
#include "RenderEmbeddedObject.h"
#include <wtf/IsoMallocInlines.h>
namespace WebCore {
WTF_MAKE_ISO_ALLOCATED_IMPL(PluginDocument);
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)
{
}
void appendBytes(DocumentWriter&, const char*, size_t) final;
void createDocumentStructure();
HTMLEmbedElement* m_embedElement { nullptr };
};
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(UserScriptInjectionTime::DocumentStart);
#if PLATFORM(IOS_FAMILY)
document.processViewport("user-scalable=no"_s, ViewportArguments::PluginDocument);
#endif
auto body = HTMLBodyElement::create(document);
body->setAttributeWithoutSynchronization(marginwidthAttr, AtomString("0", AtomString::ConstructFromLiteral));
body->setAttributeWithoutSynchronization(marginheightAttr, AtomString("0", AtomString::ConstructFromLiteral));
#if PLATFORM(IOS_FAMILY)
body->setAttribute(styleAttr, AtomString("background-color: rgb(217,224,233)", AtomString::ConstructFromLiteral));
#else
body->setAttribute(styleAttr, AtomString("background-color: rgb(38,38,38)", AtomString::ConstructFromLiteral));
#endif
rootElement->appendChild(body);
auto embedElement = HTMLEmbedElement::create(document);
m_embedElement = embedElement.ptr();
embedElement->setAttributeWithoutSynchronization(widthAttr, AtomString("100%", AtomString::ConstructFromLiteral));
embedElement->setAttributeWithoutSynchronization(heightAttr, AtomString("100%", AtomString::ConstructFromLiteral));
embedElement->setAttributeWithoutSynchronization(nameAttr, AtomString("plugin", AtomString::ConstructFromLiteral));
embedElement->setAttributeWithoutSynchronization(srcAttr, document.url().string());
ASSERT(document.loader());
if (auto loader = makeRefPtr(document.loader()))
m_embedElement->setAttributeWithoutSynchronization(typeAttr, loader->writer().mimeType());
document.setPluginElement(*m_embedElement);
body->appendChild(embedElement);
document.setHasVisuallyNonEmptyCustomContent();
}
void PluginDocumentParser::appendBytes(DocumentWriter&, const char*, size_t)
{
if (m_embedElement)
return;
createDocumentStructure();
auto frame = makeRefPtr(document()->frame());
if (!frame)
return;
document()->updateLayout();
frame->view()->flushAnyPendingPostLayoutTasks();
if (RenderWidget* renderer = m_embedElement->renderWidget()) {
if (RefPtr<Widget> widget = renderer->widget()) {
frame->loader().client().redirectDataToPlugin(*widget);
frame->loader().activeDocumentLoader()->setMainResourceDataBufferingPolicy(DataBufferingPolicy::DoNotBufferData);
}
}
}
PluginDocument::PluginDocument(Frame& frame, const URL& url)
: HTMLDocument(&frame, url, PluginDocumentClass)
{
setCompatibilityMode(DocumentCompatibilityMode::QuirksMode);
lockCompatibilityMode();
}
Ref<DocumentParser> PluginDocument::createParser()
{
return PluginDocumentParser::create(*this);
}
Widget* PluginDocument::pluginWidget()
{
if (!m_pluginElement)
return nullptr;
auto* renderer = m_pluginElement->renderer();
if (!renderer)
return nullptr;
return downcast<RenderEmbeddedObject>(*m_pluginElement->renderer()).widget();
}
void PluginDocument::setPluginElement(HTMLPlugInElement& element)
{
m_pluginElement = &element;
}
void PluginDocument::detachFromPluginElement()
{
m_pluginElement = nullptr;
}
void PluginDocument::cancelManualPluginLoad()
{
if (!shouldLoadPluginManually())
return;
auto& frameLoader = frame()->loader();
auto& documentLoader = *frameLoader.activeDocumentLoader();
documentLoader.cancelMainResourceLoad(frameLoader.cancelledError(documentLoader.request()));
m_shouldLoadPluginManually = false;
}
}