test all

content/renderer/service_worker/worker_fetch_context_impl.cc

+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "content/renderer/service_worker/worker_fetch_context_impl.h"
+
+#include "content/child/child_thread_impl.h"
+#include "content/child/request_extra_data.h"
+#include "content/child/resource_dispatcher.h"
+#include "content/child/thread_safe_sender.h"
+#include "content/child/web_url_loader_impl.h"
+#include "content/common/frame_messages.h"
+#include "mojo/public/cpp/bindings/associated_binding.h"
+#include "third_party/WebKit/public/platform/WebWorkerFetchContext.h"
+
+namespace content {
+
+WorkerFetchContextImpl::WorkerFetchContextImpl(
+    mojom::WorkerURLLoaderFactoryProviderPtrInfo provider_info)
+    : provider_info_(std::move(provider_info)),
+      thread_safe_sender_(ChildThreadImpl::current()->thread_safe_sender()) {}
+
+WorkerFetchContextImpl::~WorkerFetchContextImpl() {}
+
+void WorkerFetchContextImpl::InitializeOnWorkerThread(
+    base::SingleThreadTaskRunner* loading_task_runner) {
+  DCHECK(loading_task_runner->RunsTasksOnCurrentThread());
+  DCHECK(!resource_dispatcher_);
+  DCHECK(!binding_);
+  resource_dispatcher_ =
+      base::MakeUnique<ResourceDispatcher>(nullptr, loading_task_runner);
+  binding_ = base::MakeUnique<
+      mojo::AssociatedBinding<mojom::ServiceWorkerWorkerClient>>(this);
+  DCHECK(provider_info_.is_valid());
+  provider_.Bind(std::move(provider_info_));
+  mojom::ServiceWorkerWorkerClientAssociatedPtrInfo ptr_info;
+  binding_->Bind(&ptr_info);
+  provider_->CreateWorkerFetchContext(mojo::MakeRequest(&url_loader_factory_),
+                                      std::move(ptr_info),
+                                      service_worker_provider_id_);
+}
+
+blink::WebURLLoader* WorkerFetchContextImpl::CreateURLLoader() {
+  return new content::WebURLLoaderImpl(resource_dispatcher_.get(),
+                                       url_loader_factory_.get());
+}
+
+void WorkerFetchContextImpl::WillSendRequest(blink::WebURLRequest& request) {
+  RequestExtraData* extra_data = new RequestExtraData();
+  extra_data->set_service_worker_provider_id(service_worker_provider_id_);
+  extra_data->set_render_frame_id(parent_frame_id_);
+  extra_data->set_initiated_in_secure_context(is_secure_context_);
+  request.SetExtraData(extra_data);
+
+  if (data_saver_enabled_)
+    request.SetHTTPHeaderField("Save-Data", "on");
+
+  if (!IsControlledByServiceWorker() &&
+      request.GetServiceWorkerMode() !=
+          blink::WebURLRequest::ServiceWorkerMode::kNone) {
+    request.SetServiceWorkerMode(
+        blink::WebURLRequest::ServiceWorkerMode::kForeign);
+  }
+}
+
+bool WorkerFetchContextImpl::IsControlledByServiceWorker() const {
+  return is_controlled_by_service_worker_ ||
+         (controller_version_id_ != kInvalidServiceWorkerVersionId);
+}
+
+int64_t WorkerFetchContextImpl::ServiceWorkerID() const {
+  return controller_version_id_;
+}
+
+void WorkerFetchContextImpl::DidRunContentWithCertificateErrors(
+    const blink::WebURL& url) {
+  Send(new FrameHostMsg_DidRunContentWithCertificateErrors(parent_frame_id_,
+                                                           url));
+}
+
+void WorkerFetchContextImpl::DidDisplayContentWithCertificateErrors(
+    const blink::WebURL& url) {
+  Send(new FrameHostMsg_DidDisplayContentWithCertificateErrors(parent_frame_id_,
+                                                               url));
+}
+
+bool WorkerFetchContextImpl::Send(IPC::Message* message) {
+  return thread_safe_sender_->Send(message);
+}
+
+void WorkerFetchContextImpl::SetAppCacheHostID(int id) {
+  appcache_host_id_ = id;
+}
+
+void WorkerFetchContextImpl::SetParentFrameID(int id) {
+  parent_frame_id_ = id;
+}
+
+void WorkerFetchContextImpl::SetServiceWorkerProviderID(int id) {
+  service_worker_provider_id_ = id;
+}
+
+void WorkerFetchContextImpl::SetIsSecureContext(bool flag) {
+  is_secure_context_ = flag;
+}
+
+void WorkerFetchContextImpl::SetIsControlledByServiceWorker(bool flag) {
+  is_controlled_by_service_worker_ = flag;
+}
+
+void WorkerFetchContextImpl::SetDataSaverEnabled(bool flag) {
+  data_saver_enabled_ = flag;
+}
+
+void WorkerFetchContextImpl::SetStrictMixedContentCheckingEnabled(
+    bool enabled) {
+  strict_mixed_content_checking_enabled_ = enabled;
+}
+
+void WorkerFetchContextImpl::SetControllerServiceWorker(
+    int64_t controller_version_id) {
+  controller_version_id_ = controller_version_id;
+}
+
+}  // namespace content