| /* |
| * Copyright (C) 2017 Apple Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * |
| * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' |
| * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, |
| * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
| * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS |
| * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
| * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
| * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
| * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
| * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF |
| * THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include "config.h" |
| #include "ServiceWorkerFetch.h" |
| |
| #if ENABLE(SERVICE_WORKER) |
| |
| #include "CrossOriginAccessControl.h" |
| #include "EventNames.h" |
| #include "FetchEvent.h" |
| #include "FetchRequest.h" |
| #include "FetchResponse.h" |
| #include "ReadableStreamChunk.h" |
| #include "ResourceRequest.h" |
| #include "ServiceWorker.h" |
| #include "ServiceWorkerClientIdentifier.h" |
| #include "WorkerGlobalScope.h" |
| |
| namespace WebCore { |
| |
| namespace ServiceWorkerFetch { |
| |
| static void processResponse(Ref<Client>&& client, FetchResponse* response) |
| { |
| if (!response) { |
| client->didFail(); |
| return; |
| } |
| auto protectedResponse = makeRef(*response); |
| |
| client->didReceiveResponse(response->resourceResponse()); |
| |
| if (response->loadingError()) { |
| client->didFail(); |
| return; |
| } |
| |
| if (response->isBodyReceivedByChunk()) { |
| response->consumeBodyReceivedByChunk([client = WTFMove(client)] (auto&& result) mutable { |
| if (result.hasException()) { |
| client->didFail(); |
| return; |
| } |
| |
| if (auto chunk = result.returnValue()) |
| client->didReceiveData(SharedBuffer::create(reinterpret_cast<const char*>(chunk->data), chunk->size)); |
| else |
| client->didFinish(); |
| }); |
| return; |
| } |
| |
| auto body = response->consumeBody(); |
| WTF::switchOn(body, [&] (Ref<FormData>& formData) { |
| client->didReceiveFormDataAndFinish(WTFMove(formData)); |
| }, [&] (Ref<SharedBuffer>& buffer) { |
| client->didReceiveData(WTFMove(buffer)); |
| client->didFinish(); |
| }, [&] (std::nullptr_t&) { |
| client->didFinish(); |
| }); |
| } |
| |
| void dispatchFetchEvent(Ref<Client>&& client, ServiceWorkerGlobalScope& globalScope, std::optional<ServiceWorkerClientIdentifier> clientId, ResourceRequest&& request, String&& referrer, FetchOptions&& options) |
| { |
| auto requestHeaders = FetchHeaders::create(FetchHeaders::Guard::Immutable, HTTPHeaderMap { request.httpHeaderFields() }); |
| |
| bool isNavigation = options.mode == FetchOptions::Mode::Navigate; |
| bool isNonSubresourceRequest = WebCore::isNonSubresourceRequest(options.destination); |
| |
| ASSERT(globalScope.registration().active()); |
| ASSERT(globalScope.registration().active()->identifier() == globalScope.thread().identifier()); |
| ASSERT(globalScope.registration().active()->state() == ServiceWorkerState::Activated); |
| |
| auto* formData = request.httpBody(); |
| std::optional<FetchBody> body; |
| if (formData && !formData->isEmpty()) { |
| body = FetchBody::fromFormData(*formData); |
| if (!body) { |
| client->didNotHandle(); |
| return; |
| } |
| } |
| // FIXME: loading code should set redirect mode to manual. |
| if (isNavigation) |
| options.redirect = FetchOptions::Redirect::Manual; |
| |
| auto fetchRequest = FetchRequest::create(globalScope, WTFMove(body), WTFMove(requestHeaders), WTFMove(request), WTFMove(options), WTFMove(referrer)); |
| |
| FetchEvent::Init init; |
| init.request = WTFMove(fetchRequest); |
| if (isNavigation) { |
| // FIXME: Set reservedClientId. |
| if (clientId) |
| init.targetClientId = clientId->toString(); |
| } else if (clientId) |
| init.clientId = clientId->toString(); |
| init.cancelable = true; |
| auto event = FetchEvent::create(eventNames().fetchEvent, WTFMove(init), Event::IsTrusted::Yes); |
| |
| event->onResponse([client = client.copyRef()] (FetchResponse* response) mutable { |
| processResponse(WTFMove(client), response); |
| }); |
| |
| globalScope.dispatchEvent(event); |
| |
| if (!event->respondWithEntered()) { |
| if (event->defaultPrevented()) { |
| client->didFail(); |
| return; |
| } |
| client->didNotHandle(); |
| } |
| |
| globalScope.updateExtendedEventsSet(event.ptr()); |
| |
| auto& registration = globalScope.registration(); |
| if (isNonSubresourceRequest || registration.needsUpdate()) |
| registration.softUpdate(); |
| } |
| |
| } // namespace ServiceWorkerFetch |
| |
| } // namespace WebCore |
| |
| #endif // ENABLE(SERVICE_WORKER) |