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 "WebProcess.h"
37
38 using namespace WebCore;
39
40 namespace WebKit {
41
create(uint64_t downloadID,const ResourceRequest & request)42 PassOwnPtr<Download> Download::create(uint64_t downloadID, const ResourceRequest& request)
43 {
44 return adoptPtr(new Download(downloadID, request));
45 }
46
Download(uint64_t downloadID,const ResourceRequest & request)47 Download::Download(uint64_t downloadID, const ResourceRequest& request)
48 : m_downloadID(downloadID)
49 , m_request(request)
50 #if USE(CFNETWORK)
51 , m_allowOverwrite(false)
52 #endif
53 {
54 ASSERT(m_downloadID);
55
56 WebProcess::shared().disableTermination();
57 }
58
~Download()59 Download::~Download()
60 {
61 platformInvalidate();
62
63 WebProcess::shared().enableTermination();
64 }
65
connection() const66 CoreIPC::Connection* Download::connection() const
67 {
68 return WebProcess::shared().connection();
69 }
70
didStart()71 void Download::didStart()
72 {
73 send(Messages::DownloadProxy::DidStart(m_request));
74 }
75
didReceiveAuthenticationChallenge(const AuthenticationChallenge & authenticationChallenge)76 void Download::didReceiveAuthenticationChallenge(const AuthenticationChallenge& authenticationChallenge)
77 {
78 AuthenticationManager::shared().didReceiveAuthenticationChallenge(this, authenticationChallenge);
79 }
80
didReceiveResponse(const ResourceResponse & response)81 void Download::didReceiveResponse(const ResourceResponse& response)
82 {
83 send(Messages::DownloadProxy::DidReceiveResponse(response));
84 }
85
didReceiveData(uint64_t length)86 void Download::didReceiveData(uint64_t length)
87 {
88 send(Messages::DownloadProxy::DidReceiveData(length));
89 }
90
shouldDecodeSourceDataOfMIMEType(const String & mimeType)91 bool Download::shouldDecodeSourceDataOfMIMEType(const String& mimeType)
92 {
93 bool result;
94 if (!sendSync(Messages::DownloadProxy::ShouldDecodeSourceDataOfMIMEType(mimeType), Messages::DownloadProxy::ShouldDecodeSourceDataOfMIMEType::Reply(result)))
95 return true;
96
97 return result;
98 }
99
retrieveDestinationWithSuggestedFilename(const String & filename,bool & allowOverwrite)100 String Download::retrieveDestinationWithSuggestedFilename(const String& filename, bool& allowOverwrite)
101 {
102 String destination;
103 SandboxExtension::Handle sandboxExtensionHandle;
104 if (!sendSync(Messages::DownloadProxy::DecideDestinationWithSuggestedFilename(filename), Messages::DownloadProxy::DecideDestinationWithSuggestedFilename::Reply(destination, allowOverwrite, sandboxExtensionHandle)))
105 return String();
106
107 m_sandboxExtension = SandboxExtension::create(sandboxExtensionHandle);
108 if (m_sandboxExtension)
109 m_sandboxExtension->consume();
110
111 return destination;
112 }
113
decideDestinationWithSuggestedFilename(const String & filename,bool & allowOverwrite)114 String Download::decideDestinationWithSuggestedFilename(const String& filename, bool& allowOverwrite)
115 {
116 String destination = retrieveDestinationWithSuggestedFilename(filename, allowOverwrite);
117
118 didDecideDestination(destination, allowOverwrite);
119
120 return destination;
121 }
122
didCreateDestination(const String & path)123 void Download::didCreateDestination(const String& path)
124 {
125 send(Messages::DownloadProxy::DidCreateDestination(path));
126 }
127
didFinish()128 void Download::didFinish()
129 {
130 platformDidFinish();
131
132 send(Messages::DownloadProxy::DidFinish());
133
134 if (m_sandboxExtension)
135 m_sandboxExtension->invalidate();
136 DownloadManager::shared().downloadFinished(this);
137 }
138
didFail(const ResourceError & error,const CoreIPC::DataReference & resumeData)139 void Download::didFail(const ResourceError& error, const CoreIPC::DataReference& resumeData)
140 {
141 send(Messages::DownloadProxy::DidFail(error, resumeData));
142
143 if (m_sandboxExtension)
144 m_sandboxExtension->invalidate();
145 DownloadManager::shared().downloadFinished(this);
146 }
147
didCancel(const CoreIPC::DataReference & resumeData)148 void Download::didCancel(const CoreIPC::DataReference& resumeData)
149 {
150 send(Messages::DownloadProxy::DidCancel(resumeData));
151
152 if (m_sandboxExtension)
153 m_sandboxExtension->invalidate();
154 DownloadManager::shared().downloadFinished(this);
155 }
156
157 } // namespace WebKit
158