| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/task_management/providers/child_process_task_provider.h
" | 5 #include "chrome/browser/task_management/providers/child_process_task_provider.h
" |
| 6 | 6 |
| 7 #include "base/process/process.h" | 7 #include "base/process/process.h" |
| 8 #include "base/stl_util.h" | 8 #include "base/stl_util.h" |
| 9 #include "chrome/browser/task_management/providers/child_process_task.h" | 9 #include "chrome/browser/task_management/providers/child_process_task.h" |
| 10 #include "content/public/browser/browser_child_process_host_iterator.h" | 10 #include "content/public/browser/browser_child_process_host_iterator.h" |
| 11 #include "content/public/browser/browser_thread.h" | 11 #include "content/public/browser/browser_thread.h" |
| 12 #include "content/public/browser/child_process_data.h" | 12 #include "content/public/browser/child_process_data.h" |
| 13 | 13 |
| 14 using content::BrowserChildProcessHostIterator; | 14 using content::BrowserChildProcessHostIterator; |
| 15 using content::BrowserThread; | 15 using content::BrowserThread; |
| 16 using content::ChildProcessData; | 16 using content::ChildProcessData; |
| 17 | 17 |
| 18 namespace task_management { | 18 namespace task_management { |
| 19 | 19 |
| 20 namespace { | 20 namespace { |
| 21 | 21 |
| 22 // Collects and returns the child processes data on the IO thread to get all the | 22 // Collects and returns the child processes data on the IO thread to get all the |
| 23 // pre-existing child process before we start observing | 23 // pre-existing child process before we start observing |
| 24 // |BrowserChildProcessObserver|. | 24 // |BrowserChildProcessObserver|. |
| 25 scoped_ptr<std::vector<ChildProcessData>> CollectChildProcessData() { | 25 scoped_ptr<std::vector<ChildProcessData>> CollectChildProcessData() { |
| 26 // The |BrowserChildProcessHostIterator| must only be used on the IO thread. | 26 // The |BrowserChildProcessHostIterator| must only be used on the IO thread. |
| 27 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 27 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| 28 | 28 |
| 29 scoped_ptr<std::vector<ChildProcessData>> child_processes( | 29 scoped_ptr<std::vector<ChildProcessData>> child_processes( |
| 30 new std::vector<ChildProcessData>()); | 30 new std::vector<ChildProcessData>()); |
| 31 for (BrowserChildProcessHostIterator itr; !itr.Done(); ++itr) { | 31 for (BrowserChildProcessHostIterator itr; !itr.Done(); ++itr) { |
| 32 const ChildProcessData& process_data = itr.GetData(); | 32 const ChildProcessData& process_data = itr.GetData(); |
| 33 | 33 |
| 34 // Only add processes that have already started, i.e. with valid handles. | 34 // Only add processes that have already started, i.e. with valid handles. |
| 35 if (process_data.handle == base::kNullProcessHandle) | 35 if (process_data.handle == base::kNullProcessHandle) |
| 36 continue; | 36 continue; |
| 37 | 37 |
| (...skipping 10 matching lines...) Expand all Loading... |
| 48 tasks_by_pid_(), | 48 tasks_by_pid_(), |
| 49 weak_ptr_factory_(this) { | 49 weak_ptr_factory_(this) { |
| 50 } | 50 } |
| 51 | 51 |
| 52 ChildProcessTaskProvider::~ChildProcessTaskProvider() { | 52 ChildProcessTaskProvider::~ChildProcessTaskProvider() { |
| 53 } | 53 } |
| 54 | 54 |
| 55 Task* ChildProcessTaskProvider::GetTaskOfUrlRequest(int origin_pid, | 55 Task* ChildProcessTaskProvider::GetTaskOfUrlRequest(int origin_pid, |
| 56 int child_id, | 56 int child_id, |
| 57 int route_id) { | 57 int route_id) { |
| 58 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 58 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 59 auto itr = tasks_by_pid_.find(static_cast<base::ProcessId>(origin_pid)); | 59 auto itr = tasks_by_pid_.find(static_cast<base::ProcessId>(origin_pid)); |
| 60 if (itr == tasks_by_pid_.end()) | 60 if (itr == tasks_by_pid_.end()) |
| 61 return nullptr; | 61 return nullptr; |
| 62 | 62 |
| 63 return itr->second; | 63 return itr->second; |
| 64 } | 64 } |
| 65 | 65 |
| 66 void ChildProcessTaskProvider::BrowserChildProcessHostConnected( | 66 void ChildProcessTaskProvider::BrowserChildProcessHostConnected( |
| 67 const content::ChildProcessData& data) { | 67 const content::ChildProcessData& data) { |
| 68 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 68 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 69 if (data.handle == base::kNullProcessHandle) | 69 if (data.handle == base::kNullProcessHandle) |
| 70 return; | 70 return; |
| 71 | 71 |
| 72 CreateTask(data); | 72 CreateTask(data); |
| 73 } | 73 } |
| 74 | 74 |
| 75 void ChildProcessTaskProvider::BrowserChildProcessHostDisconnected( | 75 void ChildProcessTaskProvider::BrowserChildProcessHostDisconnected( |
| 76 const content::ChildProcessData& data) { | 76 const content::ChildProcessData& data) { |
| 77 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 77 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 78 DeleteTask(data.handle); | 78 DeleteTask(data.handle); |
| 79 } | 79 } |
| 80 | 80 |
| 81 void ChildProcessTaskProvider::StartUpdating() { | 81 void ChildProcessTaskProvider::StartUpdating() { |
| 82 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 82 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 83 DCHECK(tasks_by_handle_.empty()); | 83 DCHECK(tasks_by_handle_.empty()); |
| 84 DCHECK(tasks_by_pid_.empty()); | 84 DCHECK(tasks_by_pid_.empty()); |
| 85 | 85 |
| 86 // First, get the pre-existing child processes data. | 86 // First, get the pre-existing child processes data. |
| 87 BrowserThread::PostTaskAndReplyWithResult( | 87 BrowserThread::PostTaskAndReplyWithResult( |
| 88 BrowserThread::IO, | 88 BrowserThread::IO, |
| 89 FROM_HERE, | 89 FROM_HERE, |
| 90 base::Bind(&CollectChildProcessData), | 90 base::Bind(&CollectChildProcessData), |
| 91 base::Bind(&ChildProcessTaskProvider::ChildProcessDataCollected, | 91 base::Bind(&ChildProcessTaskProvider::ChildProcessDataCollected, |
| 92 weak_ptr_factory_.GetWeakPtr())); | 92 weak_ptr_factory_.GetWeakPtr())); |
| 93 } | 93 } |
| 94 | 94 |
| 95 void ChildProcessTaskProvider::StopUpdating() { | 95 void ChildProcessTaskProvider::StopUpdating() { |
| 96 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 96 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 97 | 97 |
| 98 // ChildProcessDataCollected() should never be called after this, and hence | 98 // ChildProcessDataCollected() should never be called after this, and hence |
| 99 // we must invalidate the weak pointers. | 99 // we must invalidate the weak pointers. |
| 100 weak_ptr_factory_.InvalidateWeakPtrs(); | 100 weak_ptr_factory_.InvalidateWeakPtrs(); |
| 101 | 101 |
| 102 // First, stop observing. | 102 // First, stop observing. |
| 103 BrowserChildProcessObserver::Remove(this); | 103 BrowserChildProcessObserver::Remove(this); |
| 104 | 104 |
| 105 // Remember: You can't notify the observer of tasks removal here, | 105 // Remember: You can't notify the observer of tasks removal here, |
| 106 // StopUpdating() is called after the observer has been cleared. | 106 // StopUpdating() is called after the observer has been cleared. |
| 107 | 107 |
| 108 // Then delete all tasks (if any). | 108 // Then delete all tasks (if any). |
| 109 STLDeleteValues(&tasks_by_handle_); // This will clear |tasks_by_handle_|. | 109 STLDeleteValues(&tasks_by_handle_); // This will clear |tasks_by_handle_|. |
| 110 tasks_by_pid_.clear(); | 110 tasks_by_pid_.clear(); |
| 111 } | 111 } |
| 112 | 112 |
| 113 void ChildProcessTaskProvider::ChildProcessDataCollected( | 113 void ChildProcessTaskProvider::ChildProcessDataCollected( |
| 114 scoped_ptr<const std::vector<content::ChildProcessData>> child_processes) { | 114 scoped_ptr<const std::vector<content::ChildProcessData>> child_processes) { |
| 115 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 115 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 116 | 116 |
| 117 for (auto& process_data : *child_processes) | 117 for (auto& process_data : *child_processes) |
| 118 CreateTask(process_data); | 118 CreateTask(process_data); |
| 119 | 119 |
| 120 // Now start observing. | 120 // Now start observing. |
| 121 BrowserChildProcessObserver::Add(this); | 121 BrowserChildProcessObserver::Add(this); |
| 122 } | 122 } |
| 123 | 123 |
| 124 void ChildProcessTaskProvider::CreateTask( | 124 void ChildProcessTaskProvider::CreateTask( |
| 125 const content::ChildProcessData& data) { | 125 const content::ChildProcessData& data) { |
| (...skipping 29 matching lines...) Expand all Loading... |
| 155 | 155 |
| 156 NotifyObserverTaskRemoved(task); | 156 NotifyObserverTaskRemoved(task); |
| 157 | 157 |
| 158 // Finally delete the task. | 158 // Finally delete the task. |
| 159 tasks_by_handle_.erase(itr); | 159 tasks_by_handle_.erase(itr); |
| 160 tasks_by_pid_.erase(task->process_id()); | 160 tasks_by_pid_.erase(task->process_id()); |
| 161 delete task; | 161 delete task; |
| 162 } | 162 } |
| 163 | 163 |
| 164 } // namespace task_management | 164 } // namespace task_management |
| OLD | NEW |