StorageProcessProxy.cpp [plain text]
#include "config.h"
#include "StorageProcessProxy.h"
#include "NetworkProcessMessages.h"
#include "StorageProcessMessages.h"
#include "StorageProcessProxyMessages.h"
#include "WebProcessPool.h"
#include "WebsiteData.h"
#include <WebCore/NotImplemented.h>
using namespace WebCore;
namespace WebKit {
static uint64_t generateCallbackID()
{
static uint64_t callbackID;
return ++callbackID;
}
Ref<StorageProcessProxy> StorageProcessProxy::create(WebProcessPool& processPool)
{
return adoptRef(*new StorageProcessProxy(processPool));
}
StorageProcessProxy::StorageProcessProxy(WebProcessPool& processPool)
: ChildProcessProxy(processPool.alwaysRunsAtBackgroundPriority())
, m_processPool(processPool)
, m_numPendingConnectionRequests(0)
{
connect();
}
StorageProcessProxy::~StorageProcessProxy()
{
ASSERT(m_pendingFetchWebsiteDataCallbacks.isEmpty());
ASSERT(m_pendingDeleteWebsiteDataCallbacks.isEmpty());
ASSERT(m_pendingDeleteWebsiteDataForOriginsCallbacks.isEmpty());
}
void StorageProcessProxy::getLaunchOptions(ProcessLauncher::LaunchOptions& launchOptions)
{
launchOptions.processType = ProcessLauncher::ProcessType::Storage;
ChildProcessProxy::getLaunchOptions(launchOptions);
}
void StorageProcessProxy::processWillShutDown(IPC::Connection& connection)
{
ASSERT_UNUSED(connection, this->connection() == &connection);
}
void StorageProcessProxy::didReceiveMessage(IPC::Connection& connection, IPC::Decoder& decoder)
{
if (decoder.messageReceiverName() == Messages::StorageProcessProxy::messageReceiverName()) {
didReceiveStorageProcessProxyMessage(connection, decoder);
return;
}
}
void StorageProcessProxy::fetchWebsiteData(PAL::SessionID sessionID, OptionSet<WebsiteDataType> dataTypes, WTF::Function<void (WebsiteData)>&& completionHandler)
{
ASSERT(canSendMessage());
uint64_t callbackID = generateCallbackID();
m_pendingFetchWebsiteDataCallbacks.add(callbackID, WTFMove(completionHandler));
send(Messages::StorageProcess::FetchWebsiteData(sessionID, dataTypes, callbackID), 0);
}
void StorageProcessProxy::deleteWebsiteData(PAL::SessionID sessionID, OptionSet<WebsiteDataType> dataTypes, WallTime modifiedSince, WTF::Function<void ()>&& completionHandler)
{
auto callbackID = generateCallbackID();
m_pendingDeleteWebsiteDataCallbacks.add(callbackID, WTFMove(completionHandler));
send(Messages::StorageProcess::DeleteWebsiteData(sessionID, dataTypes, modifiedSince, callbackID), 0);
}
void StorageProcessProxy::deleteWebsiteDataForOrigins(PAL::SessionID sessionID, OptionSet<WebsiteDataType> dataTypes, const Vector<WebCore::SecurityOriginData>& origins, WTF::Function<void()>&& completionHandler)
{
ASSERT(canSendMessage());
uint64_t callbackID = generateCallbackID();
m_pendingDeleteWebsiteDataForOriginsCallbacks.add(callbackID, WTFMove(completionHandler));
send(Messages::StorageProcess::DeleteWebsiteDataForOrigins(sessionID, dataTypes, origins, callbackID), 0);
}
void StorageProcessProxy::getStorageProcessConnection(bool isServiceWorkerProcess, Ref<Messages::WebProcessProxy::GetStorageProcessConnection::DelayedReply>&& reply)
{
m_pendingConnectionReplies.append(WTFMove(reply));
if (state() == State::Launching) {
m_numPendingConnectionRequests++;
return;
}
send(Messages::StorageProcess::CreateStorageToWebProcessConnection(isServiceWorkerProcess), 0, IPC::SendOption::DispatchMessageEvenWhenWaitingForSyncReply);
}
void StorageProcessProxy::didClose(IPC::Connection&)
{
while (!m_pendingConnectionReplies.isEmpty()) {
auto reply = m_pendingConnectionReplies.takeFirst();
#if USE(UNIX_DOMAIN_SOCKETS)
reply->send(IPC::Attachment());
#elif OS(DARWIN)
reply->send(IPC::Attachment(0, MACH_MSG_TYPE_MOVE_SEND));
#else
notImplemented();
#endif
}
for (const auto& callback : m_pendingFetchWebsiteDataCallbacks.values())
callback(WebsiteData());
m_pendingFetchWebsiteDataCallbacks.clear();
for (const auto& callback : m_pendingDeleteWebsiteDataCallbacks.values())
callback();
m_pendingDeleteWebsiteDataCallbacks.clear();
for (const auto& callback : m_pendingDeleteWebsiteDataForOriginsCallbacks.values())
callback();
m_pendingDeleteWebsiteDataForOriginsCallbacks.clear();
m_processPool.storageProcessCrashed(this);
}
void StorageProcessProxy::didReceiveInvalidMessage(IPC::Connection&, IPC::StringReference messageReceiverName, IPC::StringReference messageName)
{
}
void StorageProcessProxy::didCreateStorageToWebProcessConnection(const IPC::Attachment& connectionIdentifier)
{
ASSERT(!m_pendingConnectionReplies.isEmpty());
RefPtr<Messages::WebProcessProxy::GetStorageProcessConnection::DelayedReply> reply = m_pendingConnectionReplies.takeFirst();
#if USE(UNIX_DOMAIN_SOCKETS)
reply->send(connectionIdentifier);
#elif OS(DARWIN)
reply->send(IPC::Attachment(connectionIdentifier.port(), MACH_MSG_TYPE_MOVE_SEND));
#else
notImplemented();
#endif
}
void StorageProcessProxy::didFetchWebsiteData(uint64_t callbackID, const WebsiteData& websiteData)
{
auto callback = m_pendingFetchWebsiteDataCallbacks.take(callbackID);
callback(websiteData);
}
void StorageProcessProxy::didDeleteWebsiteData(uint64_t callbackID)
{
auto callback = m_pendingDeleteWebsiteDataCallbacks.take(callbackID);
callback();
}
void StorageProcessProxy::didDeleteWebsiteDataForOrigins(uint64_t callbackID)
{
auto callback = m_pendingDeleteWebsiteDataForOriginsCallbacks.take(callbackID);
callback();
}
#if ENABLE(SANDBOX_EXTENSIONS)
void StorageProcessProxy::getSandboxExtensionsForBlobFiles(uint64_t requestID, const Vector<String>& paths)
{
SandboxExtension::HandleArray extensions;
extensions.allocate(paths.size());
for (size_t i = 0; i < paths.size(); ++i) {
SandboxExtension::createHandle(paths[i], SandboxExtension::Type::ReadWrite, extensions[i]);
}
send(Messages::StorageProcess::DidGetSandboxExtensionsForBlobFiles(requestID, extensions), 0);
}
#endif
void StorageProcessProxy::didFinishLaunching(ProcessLauncher* launcher, IPC::Connection::Identifier connectionIdentifier)
{
ChildProcessProxy::didFinishLaunching(launcher, connectionIdentifier);
if (IPC::Connection::identifierIsNull(connectionIdentifier)) {
return;
}
for (unsigned i = 0; i < m_numPendingConnectionRequests; ++i)
send(Messages::StorageProcess::CreateStorageToWebProcessConnection(false), 0);
m_numPendingConnectionRequests = 0;
}
#if ENABLE(SERVICE_WORKER)
void StorageProcessProxy::establishWorkerContextConnectionToStorageProcess()
{
m_processPool.establishWorkerContextConnectionToStorageProcess(*this, std::nullopt);
}
void StorageProcessProxy::establishWorkerContextConnectionToStorageProcessForExplicitSession(PAL::SessionID sessionID)
{
m_processPool.establishWorkerContextConnectionToStorageProcess(*this, sessionID);
}
#endif
}