content/child/service_worker/service_worker_network_provider.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "content/child/service_worker/service_worker_network_provider.h"
   6
   7 #include "base/atomic_sequence_num.h"
   8 #include "content/child/child_thread_impl.h"
   9 #include "content/child/service_worker/service_worker_provider_context.h"
  10 #include "content/common/service_worker/service_worker_messages.h"
  11
  12 namespace content {
  13
  14 namespace {
  15
  16 const char kUserDataKey[] = "SWProviderKey";
  17
  18 // Must be unique in the child process.
  19 int GetNextProviderId() {
  20   static base::StaticAtomicSequenceNumber sequence;
  21   return sequence.GetNext();  // We start at zero.
  22 }
  23
  24 // When the provider is for a sandboxed iframe we use
  25 // kInvalidServiceWorkerProviderId as the provider type and we don't create
  26 // ServiceWorkerProviderContext and ServiceWorkerProviderHost.
  27 int GenerateProviderIdForType(const ServiceWorkerProviderType provider_type) {
  28   if (provider_type == SERVICE_WORKER_PROVIDER_FOR_SANDBOXED_FRAME)
  29     return kInvalidServiceWorkerProviderId;
  30   return GetNextProviderId();
  31 }
  32
  33 }  // namespace
  34
  35 void ServiceWorkerNetworkProvider::AttachToDocumentState(
  36     base::SupportsUserData* datasource_userdata,
  37     scoped_ptr<ServiceWorkerNetworkProvider> network_provider) {
  38   datasource_userdata->SetUserData(&kUserDataKey, network_provider.release());
  39 }
  40
  41 ServiceWorkerNetworkProvider* ServiceWorkerNetworkProvider::FromDocumentState(
  42     base::SupportsUserData* datasource_userdata) {
  43   return static_cast<ServiceWorkerNetworkProvider*>(
  44       datasource_userdata->GetUserData(&kUserDataKey));
  45 }
  46
  47 ServiceWorkerNetworkProvider::ServiceWorkerNetworkProvider(
  48     int route_id,
  49     ServiceWorkerProviderType provider_type)
  50     : provider_id_(GenerateProviderIdForType(provider_type)) {
  51   if (provider_id_ == kInvalidServiceWorkerProviderId)
  52     return;
  53   context_ = new ServiceWorkerProviderContext(provider_id_);
  54   if (!ChildThreadImpl::current())
  55     return;  // May be null in some tests.
  56   ChildThreadImpl::current()->Send(new ServiceWorkerHostMsg_ProviderCreated(
  57       provider_id_, route_id, provider_type));
  58 }
  59
  60 ServiceWorkerNetworkProvider::~ServiceWorkerNetworkProvider() {
  61   if (provider_id_ == kInvalidServiceWorkerProviderId)
  62     return;
  63   if (!ChildThreadImpl::current())
  64     return;  // May be null in some tests.
  65   ChildThreadImpl::current()->Send(
  66       new ServiceWorkerHostMsg_ProviderDestroyed(provider_id_));
  67 }
  68
  69 void ServiceWorkerNetworkProvider::SetServiceWorkerVersionId(
  70     int64 version_id) {
  71   DCHECK_NE(kInvalidServiceWorkerProviderId, provider_id_);
  72   if (!ChildThreadImpl::current())
  73     return;  // May be null in some tests.
  74   ChildThreadImpl::current()->Send(
  75       new ServiceWorkerHostMsg_SetVersionId(provider_id_, version_id));
  76 }
  77
  78 }  // namespace content