Hide WorkerProcessHost from chrome.

chrome/browser/task_manager/task_manager_worker_resource_provider.cc

 // Copyright (c) 2011 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 "chrome/browser/task_manager/task_manager_worker_resource_provider.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/stl_util.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/debugger/devtools_window.h"
 #include "chrome/browser/profiles/profile_manager.h"
-#include "content/browser/worker_host/worker_process_host.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/child_process_data.h"
 #include "content/public/browser/devtools_agent_host_registry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/worker_service.h"
 #include "grit/generated_resources.h"
 #include "grit/theme_resources_standard.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/resource/resource_bundle.h"
 
 using content::BrowserThread;
 using content::DevToolsAgentHost;
 using content::DevToolsAgentHostRegistry;
 using content::WorkerService;
 
 // Objects of this class are created on the IO thread and then passed to the UI
 // thread where they are passed to the task manager. All methods must be called
 // only on the UI thread. Destructor may be called on any thread.
 class TaskManagerSharedWorkerResource : public TaskManager::Resource {
  public:
-  TaskManagerSharedWorkerResource(const content::ChildProcessData& process_data,
-                                  int routing_id, const GURL& url,
-                                  const string16& name);
+  TaskManagerSharedWorkerResource(const GURL& url,
+                                  const string16& name,
+                                  int process_id,
+                                  int routing_id,
+                                  base::ProcessHandle process_handle);
   virtual ~TaskManagerSharedWorkerResource();
 
   bool Matches(int process_id, int routing_id) const;
 
-  void UpdateProcessData(const content::ChildProcessData& process_data);
-  const content::ChildProcessData& process_data() { return process_data_; }
+  void UpdateProcessHandle(base::ProcessHandle handle);
+  base::ProcessHandle handle() const { return handle_; }
+  int process_id() const { return process_id_; }
 
  private:
   // TaskManager::Resource methods:
   virtual string16 GetTitle() const OVERRIDE;
   virtual string16 GetProfileName() const OVERRIDE;
   virtual SkBitmap GetIcon() const OVERRIDE;
   virtual base::ProcessHandle GetProcess() const OVERRIDE;
   virtual Type GetType() const OVERRIDE;
   virtual bool CanInspect() const OVERRIDE;
   virtual void Inspect() const OVERRIDE;
 
   virtual bool SupportNetworkUsage() const OVERRIDE;
   virtual void SetSupportNetworkUsage() OVERRIDE;
 
-  content::ChildProcessData process_data_;
+  int process_id_;
   int routing_id_;
   string16 title_;
+  base::ProcessHandle handle_;
 
   static SkBitmap* default_icon_;
 
   DISALLOW_COPY_AND_ASSIGN(TaskManagerSharedWorkerResource);
 };
 
 SkBitmap* TaskManagerSharedWorkerResource::default_icon_ = NULL;
 
 TaskManagerSharedWorkerResource::TaskManagerSharedWorkerResource(
-    const content::ChildProcessData& process_data,
+    const GURL& url,
+    const string16& name,
+    int process_id,
     int routing_id,
-    const GURL& url,
-    const string16& name)
-    : process_data_(process_data),
-      routing_id_(routing_id) {
+    base::ProcessHandle process_handle)
+    : process_id_(process_id),
+      routing_id_(routing_id),
+      handle_(process_handle) {
   title_ = UTF8ToUTF16(url.spec());
   if (!name.empty())
     title_ += ASCIIToUTF16(" (") + name + ASCIIToUTF16(")");
 }
 
 TaskManagerSharedWorkerResource::~TaskManagerSharedWorkerResource() {
 }
 
 bool TaskManagerSharedWorkerResource::Matches(int process_id,
                                               int routing_id) const {
-  return process_data_.id == process_id && routing_id_ == routing_id;
+  return process_id_ == process_id && routing_id_ == routing_id;
 }
 
-void TaskManagerSharedWorkerResource::UpdateProcessData(
-    const content::ChildProcessData& process_data) {
-  process_data_ = process_data;
+void TaskManagerSharedWorkerResource::UpdateProcessHandle(
+    base::ProcessHandle handle) {
+  handle_ = handle;
 }
 
 string16 TaskManagerSharedWorkerResource::GetTitle() const {
   return l10n_util::GetStringFUTF16(IDS_TASK_MANAGER_WORKER_PREFIX, title_);
 }
 
 string16 TaskManagerSharedWorkerResource::GetProfileName() const {
   return string16();
 }
 
 SkBitmap TaskManagerSharedWorkerResource::GetIcon() const {
   if (!default_icon_) {
     ResourceBundle& rb = ResourceBundle::GetSharedInstance();
     default_icon_ = rb.GetBitmapNamed(IDR_PLUGIN);
     // TODO(jabdelmalek): use different icon for web workers.
   }
   return *default_icon_;
 }
 
 base::ProcessHandle TaskManagerSharedWorkerResource::GetProcess() const {
-  return process_data_.handle;
+  return handle_;
 }
 
 TaskManager::Resource::Type TaskManagerSharedWorkerResource::GetType() const {
   return WORKER;
 }
 
 bool TaskManagerSharedWorkerResource::CanInspect() const {
   return true;
 }
 
 void TaskManagerSharedWorkerResource::Inspect() const {
   // TODO(yurys): would be better to get profile from one of the tabs connected
   // to the worker.
   Profile* profile = ProfileManager::GetLastUsedProfile();
   if (!profile)
     return;
   DevToolsAgentHost* agent_host =
       DevToolsAgentHostRegistry::GetDevToolsAgentHostForWorker(
-          process_data_.id,
-          routing_id_);
+          process_id_, routing_id_);
   DevToolsWindow::OpenDevToolsWindowForWorker(profile, agent_host);
 }
 
 bool TaskManagerSharedWorkerResource::SupportNetworkUsage() const {
   return false;
 }
 
 void TaskManagerSharedWorkerResource::SetSupportNetworkUsage() {
 }
 
@@ skipping 62 matching lines @@
   registrar_.Remove(
       this, content::NOTIFICATION_CHILD_PROCESS_HOST_DISCONNECTED,
       content::NotificationService::AllBrowserContextsAndSources());
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE, base::Bind(
           &TaskManagerWorkerResourceProvider::StopObservingWorkers,
           this));
 }
 
 void TaskManagerWorkerResourceProvider::WorkerCreated(
-      WorkerProcessHost* process,
-      const WorkerProcessHost::WorkerInstance& instance) {
+    const GURL& url,
+    const string16& name,
+    int process_id,
+    int route_id) {
   TaskManagerSharedWorkerResource* resource =
-      new TaskManagerSharedWorkerResource(process->GetData(),
-                                          instance.worker_route_id(),
-                                          instance.url(), instance.name());
+      new TaskManagerSharedWorkerResource(
+          url, name, process_id, route_id, base::kNullProcessHandle);
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
       base::Bind(&TaskManagerWorkerResourceProvider::NotifyWorkerCreated,
                  this, base::Owned(new WorkerResourceHolder(resource))));
 }
 
-void TaskManagerWorkerResourceProvider::WorkerDestroyed(
-      WorkerProcessHost* process,
-      int worker_route_id) {
+void TaskManagerWorkerResourceProvider::WorkerDestroyed(int process_id,
+                                                        int route_id) {
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE, base::Bind(
           &TaskManagerWorkerResourceProvider::NotifyWorkerDestroyed,
-          this, process->GetData().id, worker_route_id));
+          this, process_id, route_id));
 }
 
 void TaskManagerWorkerResourceProvider::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   content::ChildProcessData* process_data =
       content::Details<content::ChildProcessData>(details).ptr();
   if (process_data->type != content::PROCESS_TYPE_WORKER)
     return;
   if (type == content::NOTIFICATION_CHILD_PROCESS_HOST_CONNECTED) {
     ProcessIdToWorkerResources::iterator it =
         launching_workers_.find(process_data->id);
     if (it == launching_workers_.end())
       return;
     WorkerResourceList& resources = it->second;
     for (WorkerResourceList::iterator r = resources.begin();
          r !=resources.end(); ++r) {
-      (*r)->UpdateProcessData(*process_data);
+      (*r)->UpdateProcessHandle(process_data->handle);
       task_manager_->AddResource(*r);
     }
     launching_workers_.erase(it);
   } else if (type == content::NOTIFICATION_CHILD_PROCESS_HOST_DISCONNECTED) {
     // Worker process may be destroyed before WorkerMsg_TerminateWorkerContex
     // message is handled and WorkerDestroyed is fired. In this case we won't
     // get WorkerDestroyed notification and have to clear resources for such
     // workers here when the worker process has been destroyed.
     for (WorkerResourceList::iterator it = resources_.begin();
          it !=resources_.end();) {
-      if ((*it)->process_data().id == process_data->id) {
+      if ((*it)->process_id() == process_data->id) {
         task_manager_->RemoveResource(*it);
         delete *it;
         it = resources_.erase(it);
       } else {
         ++it;
       }
     }
     DCHECK(launching_workers_.find(process_data->id) ==
            launching_workers_.end());
   }
@@ skipping 20 matching lines @@
       resources_.erase(it);
       return;
     }
   }
 }
 
 void TaskManagerWorkerResourceProvider::StartObservingWorkers() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
 
   scoped_ptr<WorkerResourceListHolder> holder(new WorkerResourceListHolder);
-  for (WorkerProcessHostIterator iter; !iter.Done(); ++iter) {
-    const WorkerProcessHost::Instances& instances = (*iter)->instances();
-    for (WorkerProcessHost::Instances::const_iterator i = instances.begin();
-         i != instances.end(); ++i) {
-       holder->resources()->push_back(new TaskManagerSharedWorkerResource(
-           iter.GetData(), i->worker_route_id(), i->url(), i->name()));
-    }
+  std::vector<WorkerService::WorkerInfo> worker_info =
+      WorkerService::GetInstance()->GetWorkers();
+
+  for (size_t i = 0; i < worker_info.size(); ++i) {
+    holder->resources()->push_back(new TaskManagerSharedWorkerResource(
+        worker_info[i].url, worker_info[i].name, worker_info[i].process_id,
+        worker_info[i].route_id, worker_info[i].handle));
   }
 
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
       base::Bind(
           &TaskManagerWorkerResourceProvider::AddWorkerResourceList,
           this, base::Owned(holder.release())));
 
   WorkerService::GetInstance()->AddObserver(this);
 }
@@ skipping 12 matching lines @@
        it !=resources->end(); ++it) {
     AddResource(*it);
   }
   resources->clear();
 }
 
 void TaskManagerWorkerResourceProvider::AddResource(
     TaskManagerSharedWorkerResource* resource) {
   DCHECK(updating_);
   resources_.push_back(resource);
-  if (resource->process_data().handle == base::kNullProcessHandle) {
-    int process_id = resource->process_data().id;
+  if (resource->handle() == base::kNullProcessHandle) {
+    int process_id = resource->process_id();
     launching_workers_[process_id].push_back(resource);
   } else {
     task_manager_->AddResource(resource);
   }
 }
 
 void TaskManagerWorkerResourceProvider::DeleteAllResources() {
   STLDeleteElements(&resources_);
 }