[Fetch API] Response should keep all ResourceResponse information
[WebKit-https.git] / Source / WebCore / Modules / fetch / FetchLoader.cpp
1 /*
2  * Copyright (C) 2016 Canon Inc.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted, provided that the following conditions
6  * are required to be met:
7  *
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  * 3.  Neither the name of Canon Inc. nor the names of
14  *     its contributors may be used to endorse or promote products derived
15  *     from this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY CANON INC. AND ITS CONTRIBUTORS "AS IS" AND ANY
18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20  * DISCLAIMED. IN NO EVENT SHALL CANON INC. AND ITS CONTRIBUTORS BE LIABLE FOR
21  * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
23  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
24  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
25  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
26  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #include "config.h"
30 #include "FetchLoader.h"
31
32 #include "BlobURL.h"
33 #include "CachedResourceRequestInitiators.h"
34 #include "ContentSecurityPolicy.h"
35 #include "FetchBody.h"
36 #include "FetchLoaderClient.h"
37 #include "FetchRequest.h"
38 #include "ResourceRequest.h"
39 #include "ScriptExecutionContext.h"
40 #include "SecurityOrigin.h"
41 #include "SharedBuffer.h"
42 #include "TextResourceDecoder.h"
43 #include "ThreadableBlobRegistry.h"
44
45 namespace WebCore {
46
47 void FetchLoader::start(ScriptExecutionContext& context, const Blob& blob)
48 {
49     auto urlForReading = BlobURL::createPublicURL(context.securityOrigin());
50     if (urlForReading.isEmpty()) {
51         m_client.didFail();
52         return;
53     }
54
55     ThreadableBlobRegistry::registerBlobURL(context.securityOrigin(), urlForReading, blob.url());
56
57     ResourceRequest request(urlForReading);
58     request.setInitiatorIdentifier(context.resourceRequestIdentifier());
59     request.setHTTPMethod("GET");
60
61     ThreadableLoaderOptions options;
62     options.sendLoadCallbacks = SendCallbacks;
63     options.dataBufferingPolicy = DoNotBufferData;
64     options.preflightPolicy = ConsiderPreflight;
65     options.credentials = FetchOptions::Credentials::Include;
66     options.mode = FetchOptions::Mode::SameOrigin;
67     options.contentSecurityPolicyEnforcement = ContentSecurityPolicyEnforcement::DoNotEnforce;
68
69     m_loader = ThreadableLoader::create(context, *this, WTFMove(request), options);
70     m_isStarted = m_loader;
71 }
72
73 void FetchLoader::start(ScriptExecutionContext& context, const FetchRequest& request)
74 {
75     ThreadableLoaderOptions options(request.fetchOptions(), ConsiderPreflight,
76         context.shouldBypassMainWorldContentSecurityPolicy() ? ContentSecurityPolicyEnforcement::DoNotEnforce : ContentSecurityPolicyEnforcement::EnforceConnectSrcDirective,
77         String(cachedResourceRequestInitiators().fetch),
78         ResponseFilteringPolicy::Disable);
79     options.sendLoadCallbacks = SendCallbacks;
80     options.dataBufferingPolicy = DoNotBufferData;
81     options.sameOriginDataURLFlag = SameOriginDataURLFlag::Set;
82
83     ResourceRequest fetchRequest = request.internalRequest();
84
85     ASSERT(context.contentSecurityPolicy());
86     auto& contentSecurityPolicy = *context.contentSecurityPolicy();
87
88     contentSecurityPolicy.upgradeInsecureRequestIfNeeded(fetchRequest, ContentSecurityPolicy::InsecureRequestType::Load);
89
90     if (!context.shouldBypassMainWorldContentSecurityPolicy() && !contentSecurityPolicy.allowConnectToSource(fetchRequest.url())) {
91         m_client.didFail();
92         return;
93     }
94
95     String referrer = request.internalRequestReferrer();
96     if (referrer == "no-referrer") {
97         options.referrerPolicy = ReferrerPolicy::NoReferrer;
98         referrer = String();
99     } else
100         referrer = (referrer == "client") ? context.url().strippedForUseAsReferrer() : URL(context.url(), referrer).strippedForUseAsReferrer();
101
102     m_loader = ThreadableLoader::create(context, *this, WTFMove(fetchRequest), options, WTFMove(referrer));
103     m_isStarted = m_loader;
104 }
105
106 FetchLoader::FetchLoader(FetchLoaderClient& client, FetchBodyConsumer* consumer)
107     : m_client(client)
108     , m_consumer(consumer)
109 {
110 }
111
112 void FetchLoader::stop()
113 {
114     if (m_consumer)
115         m_consumer->clean();
116     if (m_loader)
117         m_loader->cancel();
118 }
119
120 RefPtr<SharedBuffer> FetchLoader::startStreaming()
121 {
122     ASSERT(m_consumer);
123     auto firstChunk = m_consumer->takeData();
124     m_consumer = nullptr;
125     return firstChunk;
126 }
127
128 void FetchLoader::didReceiveResponse(unsigned long, const ResourceResponse& response)
129 {
130     m_client.didReceiveResponse(response);
131 }
132
133 void FetchLoader::didReceiveData(const char* value, int size)
134 {
135     if (!m_consumer) {
136         m_client.didReceiveData(value, size);
137         return;
138     }
139     m_consumer->append(value, size);
140 }
141
142 void FetchLoader::didFinishLoading(unsigned long)
143 {
144     m_client.didSucceed();
145 }
146
147 void FetchLoader::didFail(const ResourceError&)
148 {
149     m_client.didFail();
150 }
151
152 } // namespace WebCore