From 6882a04fb36642862b11efe514251d32070c3d65 Mon Sep 17 00:00:00 2001 From: Konstantin Tokarev Date: Thu, 25 Aug 2016 19:20:41 +0300 Subject: Imported QtWebKit TP3 (git b57bc6801f1876c3220d5a4bfea33d620d477443) Change-Id: I3b1d8a2808782c9f34d50240000e20cb38d3680f Reviewed-by: Konstantin Tokarev --- .../UserContent/WebUserContentController.cpp | 217 +++++++++++++++++++++ 1 file changed, 217 insertions(+) create mode 100644 Source/WebKit2/WebProcess/UserContent/WebUserContentController.cpp (limited to 'Source/WebKit2/WebProcess/UserContent/WebUserContentController.cpp') diff --git a/Source/WebKit2/WebProcess/UserContent/WebUserContentController.cpp b/Source/WebKit2/WebProcess/UserContent/WebUserContentController.cpp new file mode 100644 index 000000000..67a5422a4 --- /dev/null +++ b/Source/WebKit2/WebProcess/UserContent/WebUserContentController.cpp @@ -0,0 +1,217 @@ +/* + * Copyright (C) 2014 Apple Inc. 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. + * + * THIS SOFTWARE IS PROVIDED BY APPLE INC. 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 INC. 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 "WebUserContentController.h" + +#include "DataReference.h" +#include "WebCompiledContentExtension.h" +#include "WebFrame.h" +#include "WebPage.h" +#include "WebProcess.h" +#include "WebUserContentControllerMessages.h" +#include "WebUserContentControllerProxyMessages.h" +#include +#include +#include +#include +#include +#include + +#if ENABLE(USER_MESSAGE_HANDLERS) +#include +#include +#endif + +using namespace WebCore; + +namespace WebKit { + +static HashMap& userContentControllers() +{ + static NeverDestroyed> userContentControllers; + + return userContentControllers; +} + +PassRefPtr WebUserContentController::getOrCreate(uint64_t identifier) +{ + auto& userContentControllerPtr = userContentControllers().add(identifier, nullptr).iterator->value; + if (userContentControllerPtr) + return userContentControllerPtr; + + RefPtr userContentController = adoptRef(new WebUserContentController(identifier)); + userContentControllerPtr = userContentController.get(); + + return userContentController.release(); +} + +WebUserContentController::WebUserContentController(uint64_t identifier) + : m_identifier(identifier) + , m_userContentController(UserContentController::create()) +{ + WebProcess::singleton().addMessageReceiver(Messages::WebUserContentController::messageReceiverName(), m_identifier, *this); +} + +WebUserContentController::~WebUserContentController() +{ + ASSERT(userContentControllers().contains(m_identifier)); + + WebProcess::singleton().removeMessageReceiver(Messages::WebUserContentController::messageReceiverName(), m_identifier); + + userContentControllers().remove(m_identifier); +} + +void WebUserContentController::addUserScripts(const Vector& userScripts) +{ + for (const auto& userScript : userScripts) + m_userContentController->addUserScript(mainThreadNormalWorld(), std::make_unique(userScript)); +} + +void WebUserContentController::removeUserScript(const String& urlString) +{ + m_userContentController->removeUserScript(mainThreadNormalWorld(), URL(URL(), urlString)); +} + +void WebUserContentController::removeAllUserScripts() +{ + m_userContentController->removeUserScripts(mainThreadNormalWorld()); +} + +void WebUserContentController::addUserStyleSheets(const Vector& userStyleSheets) +{ + for (const auto& userStyleSheet : userStyleSheets) { + m_userContentController->addUserStyleSheet(mainThreadNormalWorld(), + std::make_unique(userStyleSheet), InjectInExistingDocuments); + } +} + +void WebUserContentController::removeUserStyleSheet(const String& urlString) +{ + m_userContentController->removeUserStyleSheet(mainThreadNormalWorld(), URL(URL(), urlString)); +} + +void WebUserContentController::removeAllUserStyleSheets() +{ + m_userContentController->removeUserStyleSheets(mainThreadNormalWorld()); +} + +#if ENABLE(USER_MESSAGE_HANDLERS) +class WebUserMessageHandlerDescriptorProxy : public RefCounted, public WebCore::UserMessageHandlerDescriptor::Client { +public: + static PassRefPtr create(WebUserContentController* controller, const String& name, uint64_t identifier) + { + return adoptRef(new WebUserMessageHandlerDescriptorProxy(controller, name, identifier)); + } + + virtual ~WebUserMessageHandlerDescriptorProxy() + { + m_descriptor->invalidateClient(); + } + + // WebCore::UserMessageHandlerDescriptor::Client + virtual void didPostMessage(WebCore::UserMessageHandler& handler, WebCore::SerializedScriptValue* value) + { + WebCore::Frame* frame = handler.frame(); + if (!frame) + return; + + WebFrame* webFrame = WebFrame::fromCoreFrame(*frame); + if (!webFrame) + return; + + WebPage* webPage = webFrame->page(); + if (!webPage) + return; + + WebProcess::singleton().parentProcessConnection()->send(Messages::WebUserContentControllerProxy::DidPostMessage(webPage->pageID(), webFrame->frameID(), SecurityOriginData::fromFrame(frame), m_identifier, IPC::DataReference(value->data())), m_controller->identifier()); + } + + WebCore::UserMessageHandlerDescriptor& descriptor() { return *m_descriptor; } + uint64_t identifier() { return m_identifier; } + +private: + WebUserMessageHandlerDescriptorProxy(WebUserContentController* controller, const String& name, uint64_t identifier) + : m_controller(controller) + , m_descriptor(UserMessageHandlerDescriptor::create(name, mainThreadNormalWorld(), *this)) + , m_identifier(identifier) + { + } + + RefPtr m_controller; + RefPtr m_descriptor; + uint64_t m_identifier; +}; +#endif + +void WebUserContentController::addUserScriptMessageHandlers(const Vector& scriptMessageHandlers) +{ +#if ENABLE(USER_MESSAGE_HANDLERS) + for (auto& handle : scriptMessageHandlers) { + RefPtr descriptor = WebUserMessageHandlerDescriptorProxy::create(this, handle.name, handle.identifier); + + m_userMessageHandlerDescriptors.add(descriptor->identifier(), descriptor); + m_userContentController->addUserMessageHandlerDescriptor(descriptor->descriptor()); + } +#else + UNUSED_PARAM(scriptMessageHandlers); +#endif +} + +void WebUserContentController::removeUserScriptMessageHandler(uint64_t identifier) +{ +#if ENABLE(USER_MESSAGE_HANDLERS) + auto it = m_userMessageHandlerDescriptors.find(identifier); + ASSERT(it != m_userMessageHandlerDescriptors.end()); + + m_userContentController->removeUserMessageHandlerDescriptor(it->value->descriptor()); + m_userMessageHandlerDescriptors.remove(it); +#else + UNUSED_PARAM(identifier); +#endif +} + +#if ENABLE(CONTENT_EXTENSIONS) +void WebUserContentController::addUserContentExtensions(const Vector>& userContentExtensions) +{ + for (const auto& userContentExtension : userContentExtensions) { + WebCompiledContentExtensionData contentExtensionData = userContentExtension.second; + RefPtr compiledContentExtension = WebCompiledContentExtension::create(WTFMove(contentExtensionData)); + m_userContentController->addUserContentExtension(userContentExtension.first, compiledContentExtension); + } +} + +void WebUserContentController::removeUserContentExtension(const String& name) +{ + m_userContentController->removeUserContentExtension(name); +} + +void WebUserContentController::removeAllUserContentExtensions() +{ + m_userContentController->removeAllUserContentExtensions(); +} +#endif + +} // namespace WebKit -- cgit v1.2.1