3da54ae7821fd4d11799bd41551b721183550d50
[WebKit-https.git] / Source / WebCore / platform / network / curl / CurlDownload.h
1 /*
2  * Copyright (C) 2013 Apple Inc.  All rights reserved.
3  * Copyright (C) 2017 Sony Interactive Entertainment Inc.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
25  */
26
27 #pragma once
28
29 #include "FileSystem.h"
30 #include "ResourceHandle.h"
31 #include "ResourceResponse.h"
32 #include <wtf/Lock.h>
33 #include <wtf/Threading.h>
34
35 #if PLATFORM(WIN)
36 #include <windows.h>
37 #include <winsock2.h>
38 #endif
39
40 #include "CurlContext.h"
41 #include "CurlJobManager.h"
42
43 namespace WebCore {
44
45 class CurlDownloadListener {
46 public:
47     virtual void didReceiveResponse() { }
48     virtual void didReceiveDataOfLength(int) { }
49     virtual void didFinish() { }
50     virtual void didFail() { }
51 };
52
53 class CurlDownload : public ThreadSafeRefCounted<CurlDownload> {
54 public:
55     CurlDownload();
56     ~CurlDownload();
57
58     void init(CurlDownloadListener*, const WebCore::URL&);
59     void init(CurlDownloadListener*, ResourceHandle*, const ResourceRequest&, const ResourceResponse&);
60
61     void setListener(CurlDownloadListener* listener) { m_listener = listener; }
62
63     bool start();
64     bool cancel();
65
66     String getTempPath() const;
67     String getUrl() const;
68     WebCore::ResourceResponse getResponse() const;
69
70     bool deletesFileUponFailure() const { return m_deletesFileUponFailure; }
71     void setDeletesFileUponFailure(bool deletesFileUponFailure) { m_deletesFileUponFailure = deletesFileUponFailure; }
72
73     void setDestination(const String& destination) { m_destination = destination; }
74
75 private:
76     void closeFile();
77     void moveFileToDestination();
78     void writeDataToFile(const char* data, int size);
79
80     void addHeaders(const ResourceRequest&);
81
82     // Called on download thread.
83     void didReceiveHeader(const String& header);
84     void didReceiveData(void* data, int size);
85
86     // Called on main thread.
87     void didReceiveResponse();
88     void didReceiveDataOfLength(int size);
89     void didFinish();
90     void didFail();
91
92     static size_t writeCallback(char* ptr, size_t, size_t nmemb, void* data);
93     static size_t headerCallback(char* ptr, size_t, size_t nmemb, void* data);
94
95     static void downloadFinishedCallback(CurlDownload*);
96     static void downloadFailedCallback(CurlDownload*);
97     static void receivedDataCallback(CurlDownload*, int size);
98     static void receivedResponseCallback(CurlDownload*);
99
100     CurlHandle m_curlHandle;
101     CurlJobTicket m_job;
102     String m_tempPath;
103     String m_destination;
104     WebCore::PlatformFileHandle m_tempHandle { invalidPlatformFileHandle };
105     WebCore::ResourceResponse m_response;
106     bool m_deletesFileUponFailure { false };
107     mutable Lock m_mutex;
108     CurlDownloadListener* m_listener { nullptr };
109 };
110
111 }