Source/WebKit2/Shared/Downloads/Download.cpp

   1 /*
   2  * Copyright (C) 2010, 2011 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 "Download.h"
  28
  29 #include "AuthenticationManager.h"
  30 #include "Connection.h"
  31 #include "DataReference.h"
  32 #include "DownloadProxyMessages.h"
  33 #include "DownloadManager.h"
  34 #include "SandboxExtension.h"
  35 #include "WebCoreArgumentCoders.h"
  36 #include <WebCore/NotImplemented.h>
  37
  38 #if !USE(NETWORK_SESSION)
  39 #include "DownloadAuthenticationClient.h"
  40 #endif
  41
  42 using namespace WebCore;
  43
  44 namespace WebKit {
  45
  46 #if USE(NETWORK_SESSION)
  47 Download::Download(DownloadManager& downloadManager, const NetworkSession& session, DownloadID downloadID, const ResourceRequest& request)
  48 #else
  49 Download::Download(DownloadManager& downloadManager, DownloadID downloadID, const ResourceRequest& request)
  50 #endif
  51     : m_downloadManager(downloadManager)
  52     , m_downloadID(downloadID)
  53     , m_request(request)
  54 #if USE(NETWORK_SESSION)
  55     , m_session(session)
  56 #endif
  57 {
  58     ASSERT(m_downloadID.downloadID());
  59
  60     m_downloadManager.didCreateDownload();
  61 }
  62
  63 Download::~Download()
  64 {
  65     platformInvalidate();
  66
  67     m_downloadManager.didDestroyDownload();
  68 }
  69
  70 void Download::didStart()
  71 {
  72     send(Messages::DownloadProxy::DidStart(m_request));
  73 }
  74
  75 void Download::didReceiveAuthenticationChallenge(const AuthenticationChallenge& authenticationChallenge)
  76 {
  77 #if USE(NETWORK_SESSION)
  78     notImplemented();
  79 #else
  80     m_downloadManager.downloadsAuthenticationManager().didReceiveAuthenticationChallenge(*this, authenticationChallenge);
  81 #endif
  82 }
  83
  84 void Download::didReceiveResponse(const ResourceResponse& response)
  85 {
  86     send(Messages::DownloadProxy::DidReceiveResponse(response));
  87 }
  88
  89 void Download::didReceiveData(uint64_t length)
  90 {
  91     send(Messages::DownloadProxy::DidReceiveData(length));
  92 }
  93
  94 bool Download::shouldDecodeSourceDataOfMIMEType(const String& mimeType)
  95 {
  96     bool result;
  97     if (!sendSync(Messages::DownloadProxy::ShouldDecodeSourceDataOfMIMEType(mimeType), Messages::DownloadProxy::ShouldDecodeSourceDataOfMIMEType::Reply(result)))
  98         return true;
  99
 100     return result;
 101 }
 102
 103 String Download::decideDestinationWithSuggestedFilename(const String& filename, bool& allowOverwrite)
 104 {
 105     String destination;
 106     SandboxExtension::Handle sandboxExtensionHandle;
 107     if (!sendSync(Messages::DownloadProxy::DecideDestinationWithSuggestedFilename(filename), Messages::DownloadProxy::DecideDestinationWithSuggestedFilename::Reply(destination, allowOverwrite, sandboxExtensionHandle)))
 108         return String();
 109
 110     m_sandboxExtension = SandboxExtension::create(sandboxExtensionHandle);
 111     if (m_sandboxExtension)
 112         m_sandboxExtension->consume();
 113
 114     return destination;
 115 }
 116
 117 void Download::didCreateDestination(const String& path)
 118 {
 119     send(Messages::DownloadProxy::DidCreateDestination(path));
 120 }
 121
 122 void Download::didFinish()
 123 {
 124     platformDidFinish();
 125
 126     send(Messages::DownloadProxy::DidFinish());
 127
 128     if (m_sandboxExtension) {
 129         m_sandboxExtension->revoke();
 130         m_sandboxExtension = nullptr;
 131     }
 132
 133     m_downloadManager.downloadFinished(this);
 134 }
 135
 136 void Download::didFail(const ResourceError& error, const IPC::DataReference& resumeData)
 137 {
 138     send(Messages::DownloadProxy::DidFail(error, resumeData));
 139
 140     if (m_sandboxExtension) {
 141         m_sandboxExtension->revoke();
 142         m_sandboxExtension = nullptr;
 143     }
 144     m_downloadManager.downloadFinished(this);
 145 }
 146
 147 void Download::didCancel(const IPC::DataReference& resumeData)
 148 {
 149     send(Messages::DownloadProxy::DidCancel(resumeData));
 150
 151     if (m_sandboxExtension) {
 152         m_sandboxExtension->revoke();
 153         m_sandboxExtension = nullptr;
 154     }
 155     m_downloadManager.downloadFinished(this);
 156 }
 157
 158 IPC::Connection* Download::messageSenderConnection()
 159 {
 160     return m_downloadManager.downloadProxyConnection();
 161 }
 162
 163 uint64_t Download::messageSenderDestinationID()
 164 {
 165     return m_downloadID.downloadID();
 166 }
 167
 168 } // namespace WebKit