WebURLSchemeHandler.cpp [plain text]
#include "config.h"
#include "WebURLSchemeHandler.h"
#include "URLSchemeTaskParameters.h"
#include "WebPageProxy.h"
#include "WebURLSchemeTask.h"
namespace WebKit {
using namespace WebCore;
static uint64_t generateWebURLSchemeHandlerIdentifier()
{
static uint64_t nextIdentifier = 1;
return nextIdentifier++;
}
WebURLSchemeHandler::WebURLSchemeHandler()
: m_identifier(generateWebURLSchemeHandlerIdentifier())
{
}
WebURLSchemeHandler::~WebURLSchemeHandler()
{
ASSERT(m_tasks.isEmpty());
}
void WebURLSchemeHandler::startTask(WebPageProxy& page, WebProcessProxy& process, PageIdentifier webPageID, URLSchemeTaskParameters&& parameters, SyncLoadCompletionHandler&& completionHandler)
{
auto taskIdentifier = parameters.taskIdentifier;
auto result = m_tasks.add(taskIdentifier, WebURLSchemeTask::create(*this, page, process, webPageID, WTFMove(parameters), WTFMove(completionHandler)));
ASSERT(result.isNewEntry);
auto pageEntry = m_tasksByPageIdentifier.add(page.identifier(), HashSet<uint64_t>());
ASSERT(!pageEntry.iterator->value.contains(taskIdentifier));
pageEntry.iterator->value.add(taskIdentifier);
platformStartTask(page, result.iterator->value);
}
WebProcessProxy* WebURLSchemeHandler::processForTaskIdentifier(uint64_t taskIdentifier) const
{
if (!decltype(m_tasks)::isValidKey(taskIdentifier))
return nullptr;
auto iterator = m_tasks.find(taskIdentifier);
if (iterator == m_tasks.end())
return nullptr;
return iterator->value->process();
}
void WebURLSchemeHandler::stopAllTasksForPage(WebPageProxy& page, WebProcessProxy* process)
{
auto iterator = m_tasksByPageIdentifier.find(page.identifier());
if (iterator == m_tasksByPageIdentifier.end())
return;
auto& tasksByPage = iterator->value;
Vector<uint64_t> taskIdentifiersToStop;
taskIdentifiersToStop.reserveInitialCapacity(tasksByPage.size());
for (auto taskIdentifier : tasksByPage) {
if (!process || processForTaskIdentifier(taskIdentifier) == process)
taskIdentifiersToStop.uncheckedAppend(taskIdentifier);
}
for (auto& taskIdentifier : taskIdentifiersToStop)
stopTask(page, taskIdentifier);
}
void WebURLSchemeHandler::stopTask(WebPageProxy& page, uint64_t taskIdentifier)
{
if (!decltype(m_tasks)::isValidKey(taskIdentifier))
return;
auto iterator = m_tasks.find(taskIdentifier);
if (iterator == m_tasks.end())
return;
iterator->value->stop();
platformStopTask(page, iterator->value);
removeTaskFromPageMap(page.identifier(), taskIdentifier);
m_tasks.remove(iterator);
}
void WebURLSchemeHandler::taskCompleted(WebURLSchemeTask& task)
{
auto takenTask = m_tasks.take(task.identifier());
ASSERT_UNUSED(takenTask, takenTask == &task);
removeTaskFromPageMap(task.pageProxyID(), task.identifier());
platformTaskCompleted(task);
}
void WebURLSchemeHandler::removeTaskFromPageMap(WebPageProxyIdentifier pageID, uint64_t taskID)
{
auto iterator = m_tasksByPageIdentifier.find(pageID);
ASSERT(iterator != m_tasksByPageIdentifier.end());
ASSERT(iterator->value.contains(taskID));
if (!decltype(iterator->value)::isValidValue(taskID))
return;
iterator->value.remove(taskID);
if (iterator->value.isEmpty())
m_tasksByPageIdentifier.remove(iterator);
}
}