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
38using namespace WebCore;
39
40namespace WebKit {
41
42PassOwnPtr<Download> Download::create(uint64_t downloadID, const ResourceRequest& request)
43{
44    return adoptPtr(new Download(downloadID, request));
45}
46
47Download::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
59Download::~Download()
60{
61    platformInvalidate();
62
63    WebProcess::shared().enableTermination();
64}
65
66CoreIPC::Connection* Download::connection() const
67{
68    return WebProcess::shared().connection();
69}
70
71void Download::didStart()
72{
73    send(Messages::DownloadProxy::DidStart(m_request));
74}
75
76void Download::didReceiveAuthenticationChallenge(const AuthenticationChallenge& authenticationChallenge)
77{
78    AuthenticationManager::shared().didReceiveAuthenticationChallenge(this, authenticationChallenge);
79}
80
81void Download::didReceiveResponse(const ResourceResponse& response)
82{
83    send(Messages::DownloadProxy::DidReceiveResponse(response));
84}
85
86void Download::didReceiveData(uint64_t length)
87{
88    send(Messages::DownloadProxy::DidReceiveData(length));
89}
90
91bool 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
100String 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
114String 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
123void Download::didCreateDestination(const String& path)
124{
125    send(Messages::DownloadProxy::DidCreateDestination(path));
126}
127
128void 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
139void 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
148void 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