DownloadProxyMap.cpp [plain text]
#include "config.h"
#include "DownloadProxyMap.h"
#include "ChildProcessProxy.h"
#include "DownloadProxy.h"
#include "DownloadProxyMessages.h"
#include "MessageReceiverMap.h"
#include <wtf/StdLibExtras.h>
namespace WebKit {
DownloadProxyMap::DownloadProxyMap(ChildProcessProxy* process)
: m_process(process)
{
}
DownloadProxyMap::~DownloadProxyMap()
{
ASSERT(m_downloads.isEmpty());
}
DownloadProxy* DownloadProxyMap::createDownloadProxy(WebProcessPool& processPool, const WebCore::ResourceRequest& resourceRequest)
{
RefPtr<DownloadProxy> downloadProxy = DownloadProxy::create(*this, processPool, resourceRequest);
m_downloads.set(downloadProxy->downloadID(), downloadProxy);
m_process->addMessageReceiver(Messages::DownloadProxy::messageReceiverName(), downloadProxy->downloadID().downloadID(), *downloadProxy);
return downloadProxy.get();
}
void DownloadProxyMap::downloadFinished(DownloadProxy* downloadProxy)
{
auto downloadID = downloadProxy->downloadID();
ASSERT(m_downloads.contains(downloadID));
m_process->removeMessageReceiver(Messages::DownloadProxy::messageReceiverName(), downloadID.downloadID());
downloadProxy->invalidate();
m_downloads.remove(downloadID);
}
void DownloadProxyMap::processDidClose()
{
for (const auto& download : m_downloads.values()) {
download->processDidClose();
download->invalidate();
m_process->removeMessageReceiver(Messages::DownloadProxy::messageReceiverName(), download->downloadID().downloadID());
}
m_downloads.clear();
m_process = nullptr;
}
}