5fde38534dafa231c38465186f03249f299f0eb8
[WebKit-https.git] / 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, uint64_t downloadID, const ResourceRequest& request)
48 #else
49 Download::Download(DownloadManager& downloadManager, uint64_t 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);
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;
166 }
167
168 } // namespace WebKit