OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "config.h" | 5 #include "config.h" |
6 #include "modules/serviceworkers/RespondWithObserver.h" | 6 #include "modules/serviceworkers/RespondWithObserver.h" |
7 | 7 |
8 #include "bindings/core/v8/ScriptFunction.h" | 8 #include "bindings/core/v8/ScriptFunction.h" |
9 #include "bindings/core/v8/ScriptPromise.h" | 9 #include "bindings/core/v8/ScriptPromise.h" |
10 #include "bindings/core/v8/ScriptValue.h" | 10 #include "bindings/core/v8/ScriptValue.h" |
11 #include "bindings/core/v8/V8Binding.h" | 11 #include "bindings/core/v8/V8Binding.h" |
12 #include "bindings/modules/v8/V8Response.h" | 12 #include "bindings/modules/v8/V8Response.h" |
13 #include "core/dom/ExceptionCode.h" | 13 #include "core/dom/ExceptionCode.h" |
14 #include "core/dom/ExecutionContext.h" | 14 #include "core/dom/ExecutionContext.h" |
15 #include "core/inspector/ConsoleMessage.h" | 15 #include "core/inspector/ConsoleMessage.h" |
16 #include "core/streams/Stream.h" | 16 #include "core/streams/Stream.h" |
17 #include "modules/fetch/BodyStreamBuffer.h" | 17 #include "modules/fetch/BodyStreamBuffer.h" |
18 #include "modules/serviceworkers/ServiceWorkerGlobalScopeClient.h" | 18 #include "modules/serviceworkers/ServiceWorkerGlobalScopeClient.h" |
19 #include "platform/RuntimeEnabledFeatures.h" | 19 #include "platform/RuntimeEnabledFeatures.h" |
20 #include "public/platform/WebServiceWorkerResponse.h" | 20 #include "public/platform/WebServiceWorkerResponse.h" |
21 #include "wtf/Assertions.h" | 21 #include "wtf/Assertions.h" |
22 #include "wtf/RefPtr.h" | 22 #include "wtf/RefPtr.h" |
23 #include <v8.h> | 23 #include <v8.h> |
24 | 24 |
25 namespace blink { | 25 namespace blink { |
26 namespace { | |
27 | |
28 class StreamUploader : public BodyStreamBuffer::Observer { | |
29 public: | |
30 StreamUploader(BodyStreamBuffer* buffer, Stream* outStream) | |
31 : m_buffer(buffer), m_outStream(outStream) | |
32 { | |
33 } | |
34 ~StreamUploader() override { } | |
35 void onWrite() override | |
36 { | |
37 bool needToFlush = false; | |
38 while (RefPtr<DOMArrayBuffer> buf = m_buffer->read()) { | |
39 needToFlush = true; | |
40 m_outStream->addData(static_cast<const char*>(buf->data()), buf->byt
eLength()); | |
41 } | |
42 if (needToFlush) | |
43 m_outStream->flush(); | |
44 } | |
45 void onClose() override | |
46 { | |
47 m_outStream->finalize(); | |
48 cleanup(); | |
49 } | |
50 void onError() override | |
51 { | |
52 // If the stream is aborted soon after the stream is registered to the | |
53 // StreamRegistry, ServiceWorkerURLRequestJob may not notice the error | |
54 // and continue waiting forever. | |
55 // FIXME: Add new message to report the error to the browser process. | |
56 m_outStream->abort(); | |
57 cleanup(); | |
58 } | |
59 DEFINE_INLINE_TRACE() | |
60 { | |
61 visitor->trace(m_buffer); | |
62 visitor->trace(m_outStream); | |
63 BodyStreamBuffer::Observer::trace(visitor); | |
64 } | |
65 void start() | |
66 { | |
67 m_buffer->registerObserver(this); | |
68 onWrite(); | |
69 if (m_buffer->hasError()) | |
70 return onError(); | |
71 if (m_buffer->isClosed()) | |
72 return onClose(); | |
73 } | |
74 | |
75 private: | |
76 void cleanup() | |
77 { | |
78 m_buffer->unregisterObserver(); | |
79 m_buffer.clear(); | |
80 m_outStream.clear(); | |
81 } | |
82 Member<BodyStreamBuffer> m_buffer; | |
83 Member<Stream> m_outStream; | |
84 }; | |
85 | |
86 } // namespace | |
87 | 26 |
88 class RespondWithObserver::ThenFunction final : public ScriptFunction { | 27 class RespondWithObserver::ThenFunction final : public ScriptFunction { |
89 public: | 28 public: |
90 enum ResolveType { | 29 enum ResolveType { |
91 Fulfilled, | 30 Fulfilled, |
92 Rejected, | 31 Rejected, |
93 }; | 32 }; |
94 | 33 |
95 static v8::Local<v8::Function> createFunction(ScriptState* scriptState, Resp
ondWithObserver* observer, ResolveType type) | 34 static v8::Local<v8::Function> createFunction(ScriptState* scriptState, Resp
ondWithObserver* observer, ResolveType type) |
96 { | 35 { |
(...skipping 100 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
197 || (m_requestMode != WebURLRequest::FetchRequestModeNoCORS && responseTy
pe == FetchResponseData::OpaqueType) | 136 || (m_requestMode != WebURLRequest::FetchRequestModeNoCORS && responseTy
pe == FetchResponseData::OpaqueType) |
198 || (m_frameType != WebURLRequest::FrameTypeNone && responseType != Fetch
ResponseData::BasicType && responseType != FetchResponseData::DefaultType)) { | 137 || (m_frameType != WebURLRequest::FrameTypeNone && responseType != Fetch
ResponseData::BasicType && responseType != FetchResponseData::DefaultType)) { |
199 responseWasRejected(); | 138 responseWasRejected(); |
200 return; | 139 return; |
201 } | 140 } |
202 if (response->bodyUsed()) { | 141 if (response->bodyUsed()) { |
203 responseWasRejected(); | 142 responseWasRejected(); |
204 return; | 143 return; |
205 } | 144 } |
206 response->lockBody(Body::PassBody); | 145 response->lockBody(Body::PassBody); |
207 if (BodyStreamBuffer* buffer = response->internalBuffer()) { | 146 if (OwnPtr<DrainingBodyStreamBuffer> buffer = response->createInternalDraini
ngStream()) { |
208 if (buffer == response->buffer() && response->isBodyConsumed()) | |
209 buffer = response->createDrainingStream(); | |
210 WebServiceWorkerResponse webResponse; | 147 WebServiceWorkerResponse webResponse; |
211 response->populateWebServiceWorkerResponse(webResponse); | 148 response->populateWebServiceWorkerResponse(webResponse); |
| 149 if (RefPtr<BlobDataHandle> blobDataHandle = buffer->drainAsBlobDataHandl
e(FetchDataConsumerHandle::Reader::AllowBlobWithInvalidSize)) { |
| 150 webResponse.setBlobDataHandle(blobDataHandle); |
| 151 ServiceWorkerGlobalScopeClient::from(executionContext())->didHandleF
etchEvent(m_eventID, webResponse); |
| 152 m_state = Done; |
| 153 return; |
| 154 } |
212 Stream* outStream = Stream::create(executionContext(), ""); | 155 Stream* outStream = Stream::create(executionContext(), ""); |
213 webResponse.setStreamURL(outStream->url()); | 156 webResponse.setStreamURL(outStream->url()); |
214 ServiceWorkerGlobalScopeClient::from(executionContext())->didHandleFetch
Event(m_eventID, webResponse); | 157 ServiceWorkerGlobalScopeClient::from(executionContext())->didHandleFetch
Event(m_eventID, webResponse); |
215 StreamUploader* uploader = new StreamUploader(buffer, outStream); | 158 FetchDataLoader* loader = FetchDataLoader::createLoaderAsStream(outStrea
m); |
216 uploader->start(); | 159 buffer->startLoading(loader, nullptr); |
217 m_state = Done; | 160 m_state = Done; |
218 return; | 161 return; |
219 } | 162 } |
220 WebServiceWorkerResponse webResponse; | 163 WebServiceWorkerResponse webResponse; |
221 response->populateWebServiceWorkerResponse(webResponse); | 164 response->populateWebServiceWorkerResponse(webResponse); |
222 ServiceWorkerGlobalScopeClient::from(executionContext())->didHandleFetchEven
t(m_eventID, webResponse); | 165 ServiceWorkerGlobalScopeClient::from(executionContext())->didHandleFetchEven
t(m_eventID, webResponse); |
223 m_state = Done; | 166 m_state = Done; |
224 } | 167 } |
225 | 168 |
226 RespondWithObserver::RespondWithObserver(ExecutionContext* context, int eventID,
WebURLRequest::FetchRequestMode requestMode, WebURLRequest::FrameType frameType
) | 169 RespondWithObserver::RespondWithObserver(ExecutionContext* context, int eventID,
WebURLRequest::FetchRequestMode requestMode, WebURLRequest::FrameType frameType
) |
227 : ContextLifecycleObserver(context) | 170 : ContextLifecycleObserver(context) |
228 , m_eventID(eventID) | 171 , m_eventID(eventID) |
229 , m_requestMode(requestMode) | 172 , m_requestMode(requestMode) |
230 , m_frameType(frameType) | 173 , m_frameType(frameType) |
231 , m_state(Initial) | 174 , m_state(Initial) |
232 { | 175 { |
233 } | 176 } |
234 | 177 |
235 DEFINE_TRACE(RespondWithObserver) | 178 DEFINE_TRACE(RespondWithObserver) |
236 { | 179 { |
237 ContextLifecycleObserver::trace(visitor); | 180 ContextLifecycleObserver::trace(visitor); |
238 } | 181 } |
239 | 182 |
240 } // namespace blink | 183 } // namespace blink |
OLD | NEW |