WebOriginDataManagerProxy.cpp [plain text]
#include "config.h"
#include "WebOriginDataManagerProxy.h"
#include "SecurityOriginData.h"
#include "WebContext.h"
#include "WebOriginDataManagerMessages.h"
#include "WebOriginDataManagerProxyMessages.h"
#include "WebSecurityOrigin.h"
#include <WebCore/SecurityOrigin.h>
#include <wtf/NeverDestroyed.h>
using namespace WebCore;
namespace WebKit {
const char* WebOriginDataManagerProxy::supplementName()
{
return "WebOriginDataManagerProxy";
}
PassRefPtr<WebOriginDataManagerProxy> WebOriginDataManagerProxy::create(WebContext* context)
{
return adoptRef(new WebOriginDataManagerProxy(context));
}
WebOriginDataManagerProxy::WebOriginDataManagerProxy(WebContext* context)
: WebContextSupplement(context)
{
context->addMessageReceiver(Messages::WebOriginDataManagerProxy::messageReceiverName(), *this);
}
WebOriginDataManagerProxy::~WebOriginDataManagerProxy()
{
}
void WebOriginDataManagerProxy::contextDestroyed()
{
invalidateCallbackMap(m_arrayCallbacks, CallbackBase::Error::OwnerWasInvalidated);
}
void WebOriginDataManagerProxy::processDidClose(WebProcessProxy*)
{
invalidateCallbackMap(m_arrayCallbacks, CallbackBase::Error::ProcessExited);
}
bool WebOriginDataManagerProxy::shouldTerminate(WebProcessProxy*) const
{
return m_arrayCallbacks.isEmpty();
}
void WebOriginDataManagerProxy::refWebContextSupplement()
{
API::Object::ref();
}
void WebOriginDataManagerProxy::derefWebContextSupplement()
{
API::Object::deref();
}
void WebOriginDataManagerProxy::getOrigins(WKOriginDataTypes types, std::function<void (API::Array*, CallbackBase::Error)> callbackFunction)
{
if (!(types & kWKIndexedDatabaseData)) {
callbackFunction(nullptr, CallbackBase::Error::None);
return;
}
RefPtr<ArrayCallback> callback = ArrayCallback::create(WTF::move(callbackFunction));
if (!context()) {
callback->invalidate();
return;
}
uint64_t callbackID = callback->callbackID();
m_arrayCallbacks.set(callbackID, callback.release());
context()->sendToDatabaseProcessRelaunchingIfNecessary(Messages::WebOriginDataManager::GetOrigins(types, callbackID));
}
void WebOriginDataManagerProxy::didGetOrigins(IPC::Connection* connection, const Vector<SecurityOriginData>& originIdentifiers, uint64_t callbackID)
{
RefPtr<ArrayCallback> callback = m_arrayCallbacks.take(callbackID);
MESSAGE_CHECK_BASE(callback, connection);
Vector<RefPtr<API::Object>> securityOrigins;
securityOrigins.reserveInitialCapacity(originIdentifiers.size());
for (const auto& originIdentifier : originIdentifiers)
securityOrigins.uncheckedAppend(WebSecurityOrigin::create(originIdentifier.securityOrigin()));
callback->performCallbackWithReturnValue(API::Array::create(WTF::move(securityOrigins)).get());
}
void WebOriginDataManagerProxy::deleteEntriesForOrigin(WKOriginDataTypes types, WebSecurityOrigin* origin, std::function<void (CallbackBase::Error)> callbackFunction)
{
if (!(types & kWKIndexedDatabaseData)) {
callbackFunction(CallbackBase::Error::None);
return;
}
RefPtr<VoidCallback> callback = VoidCallback::create(WTF::move(callbackFunction));
if (!context()) {
callback->invalidate();
return;
}
uint64_t callbackID = callback->callbackID();
m_voidCallbacks.set(callbackID, callback.release());
SecurityOriginData securityOriginData;
securityOriginData.protocol = origin->securityOrigin().protocol();
securityOriginData.host = origin->securityOrigin().host();
securityOriginData.port = origin->securityOrigin().port();
context()->sendToDatabaseProcessRelaunchingIfNecessary(Messages::WebOriginDataManager::DeleteEntriesForOrigin(types, securityOriginData, callbackID));
}
void WebOriginDataManagerProxy::deleteEntriesModifiedBetweenDates(WKOriginDataTypes types, double startDate, double endDate, std::function<void (CallbackBase::Error)> callbackFunction)
{
if (!(types & kWKIndexedDatabaseData)) {
callbackFunction(CallbackBase::Error::None);
return;
}
RefPtr<VoidCallback> callback = VoidCallback::create(WTF::move(callbackFunction));
if (!context()) {
callback->invalidate();
return;
}
uint64_t callbackID = callback->callbackID();
m_voidCallbacks.set(callbackID, callback.release());
context()->sendToDatabaseProcessRelaunchingIfNecessary(Messages::WebOriginDataManager::DeleteEntriesModifiedBetweenDates(types, startDate, endDate, callbackID));
}
void WebOriginDataManagerProxy::didDeleteEntries(IPC::Connection* connection, uint64_t callbackID)
{
RefPtr<VoidCallback> callback = m_voidCallbacks.take(callbackID);
MESSAGE_CHECK_BASE(callback, connection);
callback->performCallback();
}
void WebOriginDataManagerProxy::deleteAllEntries(WKOriginDataTypes types, std::function<void (CallbackBase::Error)> callbackFunction)
{
if (!(types & kWKIndexedDatabaseData)) {
callbackFunction(CallbackBase::Error::None);
return;
}
RefPtr<VoidCallback> callback = VoidCallback::create(WTF::move(callbackFunction));
if (!context()) {
callback->invalidate();
return;
}
uint64_t callbackID = callback->callbackID();
m_voidCallbacks.set(callbackID, callback.release());
context()->sendToDatabaseProcessRelaunchingIfNecessary(Messages::WebOriginDataManager::DeleteAllEntries(types, callbackID));
}
void WebOriginDataManagerProxy::didDeleteAllEntries(IPC::Connection* connection, uint64_t callbackID)
{
RefPtr<VoidCallback> callback = m_voidCallbacks.take(callbackID);
MESSAGE_CHECK_BASE(callback, connection);
callback->performCallback();
}
}