WebUserContentControllerProxy.cpp [plain text]
#include "config.h"
#include "WebUserContentControllerProxy.h"
#include "DataReference.h"
#include "WebProcessProxy.h"
#include "WebScriptMessageHandler.h"
#include "WebUserContentControllerMessages.h"
#include "WebUserContentControllerProxyMessages.h"
#include <WebCore/SerializedScriptValue.h>
namespace WebKit {
static uint64_t generateIdentifier()
{
static uint64_t identifier;
return ++identifier;
}
PassRefPtr<WebUserContentControllerProxy> WebUserContentControllerProxy::create()
{
return adoptRef(new WebUserContentControllerProxy);
}
WebUserContentControllerProxy::WebUserContentControllerProxy()
: m_identifier(generateIdentifier())
{
}
WebUserContentControllerProxy::~WebUserContentControllerProxy()
{
}
void WebUserContentControllerProxy::addProcess(WebProcessProxy& webProcessProxy)
{
ASSERT(webProcessProxy.state() == WebProcessProxy::State::Running);
if (!m_processes.add(&webProcessProxy).isNewEntry)
return;
webProcessProxy.addMessageReceiver(Messages::WebUserContentControllerProxy::messageReceiverName(), m_identifier, *this);
webProcessProxy.connection()->send(Messages::WebUserContentController::AddUserScripts(m_userScripts), m_identifier);
webProcessProxy.connection()->send(Messages::WebUserContentController::AddUserStyleSheets(m_userStyleSheets), m_identifier);
Vector<WebScriptMessageHandlerHandle> messageHandlerHandles;
for (auto& handler : m_scriptMessageHandlers.values())
messageHandlerHandles.append(handler->handle());
webProcessProxy.connection()->send(Messages::WebUserContentController::AddUserScriptMessageHandlers(messageHandlerHandles), m_identifier);
}
void WebUserContentControllerProxy::removeProcess(WebProcessProxy& webProcessProxy)
{
ASSERT(m_processes.contains(&webProcessProxy));
if (m_processes.remove(&webProcessProxy))
webProcessProxy.removeMessageReceiver(Messages::WebUserContentControllerProxy::messageReceiverName(), m_identifier);
}
void WebUserContentControllerProxy::addUserScript(WebCore::UserScript userScript)
{
m_userScripts.append(WTF::move(userScript));
for (auto& processAndCount : m_processes)
processAndCount.key->connection()->send(Messages::WebUserContentController::AddUserScripts({ m_userScripts.last() }), m_identifier);
}
void WebUserContentControllerProxy::removeAllUserScripts()
{
m_userScripts.clear();
for (auto& processAndCount : m_processes)
processAndCount.key->connection()->send(Messages::WebUserContentController::RemoveAllUserScripts(), m_identifier);
}
void WebUserContentControllerProxy::addUserStyleSheet(WebCore::UserStyleSheet userStyleSheet)
{
m_userStyleSheets.append(WTF::move(userStyleSheet));
for (auto& processAndCount : m_processes)
processAndCount.key->connection()->send(Messages::WebUserContentController::AddUserStyleSheets({ m_userStyleSheets.last() }), m_identifier);
}
void WebUserContentControllerProxy::removeAllUserStyleSheets()
{
m_userStyleSheets.clear();
for (auto& processAndCount : m_processes)
processAndCount.key->connection()->send(Messages::WebUserContentController::RemoveAllUserStyleSheets(), m_identifier);
}
bool WebUserContentControllerProxy::addUserScriptMessageHandler(WebScriptMessageHandler* handler)
{
for (auto& existingHandler : m_scriptMessageHandlers.values()) {
if (existingHandler->name() == handler->name())
return false;
}
m_scriptMessageHandlers.add(handler->identifier(), handler);
for (auto& processAndCount : m_processes)
processAndCount.key->connection()->send(Messages::WebUserContentController::AddUserScriptMessageHandlers({ handler->handle() }), m_identifier);
return true;
}
void WebUserContentControllerProxy::removeUserMessageHandlerForName(const String& name)
{
for (auto it = m_scriptMessageHandlers.begin(), end = m_scriptMessageHandlers.end(); it != end; ++it) {
if (it->value->name() == name) {
for (auto& processAndCount : m_processes)
processAndCount.key->connection()->send(Messages::WebUserContentController::RemoveUserScriptMessageHandler(it->value->identifier()), m_identifier);
m_scriptMessageHandlers.remove(it);
return;
}
}
}
void WebUserContentControllerProxy::didPostMessage(IPC::Connection* connection, uint64_t pageID, uint64_t frameID, uint64_t messageHandlerID, const IPC::DataReference& dataReference)
{
WebPageProxy* page = WebProcessProxy::webPage(pageID);
if (!page)
return;
WebProcessProxy* webProcess = WebProcessProxy::fromConnection(connection);
WebFrameProxy* frame = webProcess->webFrame(frameID);
if (!frame)
return;
if (!HashMap<uint64_t, RefPtr<WebScriptMessageHandler>>::isValidKey(messageHandlerID))
return;
RefPtr<WebScriptMessageHandler> handler = m_scriptMessageHandlers.get(messageHandlerID);
if (!handler)
return;
auto buffer = dataReference.vector();
RefPtr<WebCore::SerializedScriptValue> value = WebCore::SerializedScriptValue::adopt(buffer);
handler->client().didPostMessage(*page, *frame, *value);
}
}