#include "config.h"
#include "Download.h"
#include "AuthenticationManager.h"
#include "Connection.h"
#include "DataReference.h"
#include "DownloadManager.h"
#include "DownloadProxyMessages.h"
#include "Logging.h"
#include "NetworkDataTask.h"
#include "NetworkProcess.h"
#include "NetworkSession.h"
#include "SandboxExtension.h"
#include "WebCoreArgumentCoders.h"
#include <WebCore/NotImplemented.h>
#if PLATFORM(COCOA)
#include "NetworkDataTaskCocoa.h"
#endif
using namespace WebCore;
#define RELEASE_LOG_IF_ALLOWED(fmt, ...) RELEASE_LOG_IF(isAlwaysOnLoggingAllowed(), Network, "%p - Download::" fmt, this, ##__VA_ARGS__)
namespace WebKit {
Download::Download(DownloadManager& downloadManager, DownloadID downloadID, NetworkDataTask& download, const PAL::SessionID& sessionID, const String& suggestedName)
: m_downloadManager(downloadManager)
, m_downloadID(downloadID)
, m_download(&download)
, m_sessionID(sessionID)
, m_suggestedName(suggestedName)
{
ASSERT(m_downloadID.downloadID());
m_downloadManager.didCreateDownload();
}
#if PLATFORM(COCOA)
Download::Download(DownloadManager& downloadManager, DownloadID downloadID, NSURLSessionDownloadTask* download, const PAL::SessionID& sessionID, const String& suggestedName)
: m_downloadManager(downloadManager)
, m_downloadID(downloadID)
, m_downloadTask(download)
, m_sessionID(sessionID)
, m_suggestedName(suggestedName)
{
ASSERT(m_downloadID.downloadID());
m_downloadManager.didCreateDownload();
}
#endif
Download::~Download()
{
m_downloadManager.didDestroyDownload();
}
void Download::cancel()
{
if (m_download) {
m_download->cancel();
didCancel({ });
return;
}
platformCancelNetworkLoad();
}
void Download::didReceiveChallenge(const WebCore::AuthenticationChallenge& challenge, ChallengeCompletionHandler&& completionHandler)
{
if (challenge.protectionSpace().isPasswordBased() && !challenge.proposedCredential().isEmpty() && !challenge.previousFailureCount()) {
completionHandler(AuthenticationChallengeDisposition::UseCredential, challenge.proposedCredential());
return;
}
NetworkProcess::singleton().authenticationManager().didReceiveAuthenticationChallenge(*this, challenge, WTFMove(completionHandler));
}
void Download::didCreateDestination(const String& path)
{
send(Messages::DownloadProxy::DidCreateDestination(path));
}
void Download::didReceiveData(uint64_t length)
{
if (!m_hasReceivedData) {
RELEASE_LOG_IF_ALLOWED("didReceiveData: Started receiving data (id = %" PRIu64 ")", downloadID().downloadID());
m_hasReceivedData = true;
}
send(Messages::DownloadProxy::DidReceiveData(length));
}
void Download::didFinish()
{
RELEASE_LOG_IF_ALLOWED("didFinish: (id = %" PRIu64 ")", downloadID().downloadID());
send(Messages::DownloadProxy::DidFinish());
if (m_sandboxExtension) {
m_sandboxExtension->revoke();
m_sandboxExtension = nullptr;
}
m_downloadManager.downloadFinished(this);
}
void Download::didFail(const ResourceError& error, const IPC::DataReference& resumeData)
{
RELEASE_LOG_IF_ALLOWED("didFail: (id = %" PRIu64 ", isTimeout = %d, isCancellation = %d, errCode = %d)",
downloadID().downloadID(), error.isTimeout(), error.isCancellation(), error.errorCode());
send(Messages::DownloadProxy::DidFail(error, resumeData));
if (m_sandboxExtension) {
m_sandboxExtension->revoke();
m_sandboxExtension = nullptr;
}
m_downloadManager.downloadFinished(this);
}
void Download::didCancel(const IPC::DataReference& resumeData)
{
RELEASE_LOG_IF_ALLOWED("didCancel: (id = %" PRIu64 ")", downloadID().downloadID());
send(Messages::DownloadProxy::DidCancel(resumeData));
if (m_sandboxExtension) {
m_sandboxExtension->revoke();
m_sandboxExtension = nullptr;
}
m_downloadManager.downloadFinished(this);
}
IPC::Connection* Download::messageSenderConnection()
{
return m_downloadManager.downloadProxyConnection();
}
uint64_t Download::messageSenderDestinationID()
{
return m_downloadID.downloadID();
}
bool Download::isAlwaysOnLoggingAllowed() const
{
#if PLATFORM(COCOA)
return m_sessionID.isAlwaysOnLoggingAllowed();
#else
return false;
#endif
}
#if !PLATFORM(COCOA)
void Download::platformCancelNetworkLoad()
{
}
#endif
}