Forcibly clear worker ref counts on shutdown.

content/browser/renderer_host/render_process_host_impl.cc

 // Copyright 2012 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.
 
 // Represents the browser side of the browser <--> renderer communication
 // channel. There will be one RenderProcessHost per renderer process.
 
 #include "content/browser/renderer_host/render_process_host_impl.h"
 
 #include <algorithm>
@@ skipping 657 matching lines @@
     bool is_for_guests_only)
     : fast_shutdown_started_(false),
       deleting_soon_(false),
 #ifndef NDEBUG
       is_self_deleted_(false),
 #endif
       pending_views_(0),
       child_token_(mojo::edk::GenerateRandomToken()),
       service_worker_ref_count_(0),
       shared_worker_ref_count_(0),
+      is_worker_ref_count_disabled_(false),
       route_provider_binding_(this),
       associated_interface_provider_bindings_(
           mojo::BindingSetDispatchMode::WITH_CONTEXT),
       visible_widgets_(0),
       is_process_backgrounded_(false),
       is_initialized_(false),
       id_(ChildProcessHostImpl::GenerateChildProcessUniqueId()),
       browser_context_(browser_context),
       storage_partition_impl_(storage_partition_impl),
       sudden_termination_allowed_(true),
@@ skipping 662 matching lines @@
   return manager->GetCdm(render_frame_id, cdm_id);
 }
 #endif
 
 bool RenderProcessHostImpl::IsProcessBackgrounded() const {
   return is_process_backgrounded_;
 }
 
 void RenderProcessHostImpl::IncrementServiceWorkerRefCount() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(!is_worker_ref_count_disabled_);
   ++service_worker_ref_count_;
   if (worker_ref_count() > max_worker_count_)
     max_worker_count_ = worker_ref_count();
 }
 
 void RenderProcessHostImpl::DecrementServiceWorkerRefCount() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
-  DCHECK_GT(worker_ref_count(), 0UL);
+  DCHECK(!is_worker_ref_count_disabled_);
+  DCHECK_GT(worker_ref_count(), 0U);
   --service_worker_ref_count_;
   if (worker_ref_count() == 0)
     Cleanup();
 }
 
 void RenderProcessHostImpl::IncrementSharedWorkerRefCount() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(!is_worker_ref_count_disabled_);
   ++shared_worker_ref_count_;
   if (worker_ref_count() > max_worker_count_)
     max_worker_count_ = worker_ref_count();
 }
 
 void RenderProcessHostImpl::DecrementSharedWorkerRefCount() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
-  DCHECK_GT(worker_ref_count(), 0UL);
+  DCHECK(!is_worker_ref_count_disabled_);
+  DCHECK_GT(worker_ref_count(), 0U);
   --shared_worker_ref_count_;
   if (worker_ref_count() == 0)
     Cleanup();
 }
 
+void RenderProcessHostImpl::ForceReleaseWorkerRefCounts() {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(!is_worker_ref_count_disabled_);
+  is_worker_ref_count_disabled_ = true;
+  if (!worker_ref_count())
+    return;
+  service_worker_ref_count_ = 0;
+  shared_worker_ref_count_ = 0;
+  Cleanup();
+}
+
+bool RenderProcessHostImpl::IsWorkerRefCountDisabled() {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  return is_worker_ref_count_disabled_;
+}
+
 void RenderProcessHostImpl::PurgeAndSuspend() {
   Send(new ChildProcessMsg_PurgeAndSuspend());
 }
 
 mojom::RouteProvider* RenderProcessHostImpl::GetRemoteRouteProvider() {
   if (!remote_route_provider_) {
     DCHECK(channel_);
     channel_->GetRemoteAssociatedInterface(&remote_route_provider_);
   }
   return remote_route_provider_.get();
@@ skipping 1596 matching lines @@
   LOG(ERROR) << "Terminating render process for bad Mojo message: " << error;
 
   // The ReceivedBadMessage call below will trigger a DumpWithoutCrashing. Alias
   // enough information here so that we can determine what the bad message was.
   base::debug::Alias(&error);
   bad_message::ReceivedBadMessage(process.get(),
                                   bad_message::RPH_MOJO_PROCESS_ERROR);
 }
 
 }  // namespace content