1/* 2 * Copyright (C) 2012 Apple Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' 14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, 15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS 17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 23 * THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26#include "config.h" 27#include "DownloadProxyMap.h" 28 29#include "ChildProcessProxy.h" 30#include "DownloadProxy.h" 31#include "DownloadProxyMessages.h" 32#include "MessageReceiverMap.h" 33#include <wtf/StdLibExtras.h> 34 35namespace WebKit { 36 37DownloadProxyMap::DownloadProxyMap(ChildProcessProxy* process) 38 : m_process(process) 39{ 40} 41 42DownloadProxyMap::~DownloadProxyMap() 43{ 44 ASSERT(m_downloads.isEmpty()); 45} 46 47DownloadProxy* DownloadProxyMap::createDownloadProxy(WebContext& webContext) 48{ 49 RefPtr<DownloadProxy> downloadProxy = DownloadProxy::create(*this, webContext); 50 m_downloads.set(downloadProxy->downloadID(), downloadProxy); 51 52 m_process->addMessageReceiver(Messages::DownloadProxy::messageReceiverName(), downloadProxy->downloadID(), *downloadProxy); 53 54 return downloadProxy.get(); 55} 56 57void DownloadProxyMap::downloadFinished(DownloadProxy* downloadProxy) 58{ 59 uint64_t downloadID = downloadProxy->downloadID(); 60 61 ASSERT(m_downloads.contains(downloadID)); 62 63 downloadProxy->invalidate(); 64 m_downloads.remove(downloadID); 65 66 m_process->removeMessageReceiver(Messages::DownloadProxy::messageReceiverName(), downloadID); 67} 68 69void DownloadProxyMap::processDidClose() 70{ 71 // Invalidate all outstanding downloads. 72 for (HashMap<uint64_t, RefPtr<DownloadProxy>>::iterator::Values it = m_downloads.begin().values(), end = m_downloads.end().values(); it != end; ++it) { 73 (*it)->processDidClose(); 74 (*it)->invalidate(); 75 } 76 77 m_downloads.clear(); 78 m_process = 0; 79} 80 81} // namespace WebKit 82