| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 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 "chrome/browser/ui/webui/task_manager_handler.h" | |
| 6 | |
| 7 #include <algorithm> | |
| 8 #include <functional> | |
| 9 #include "base/bind.h" | |
| 10 #include "base/bind_helpers.h" | |
| 11 #include "base/string_number_conversions.h" | |
| 12 #include "base/values.h" | |
| 13 #include "chrome/browser/profiles/profile.h" | |
| 14 #include "chrome/browser/task_manager/task_manager.h" | |
| 15 #include "chrome/browser/ui/webui/web_ui_util.h" | |
| 16 #include "chrome/common/chrome_notification_types.h" | |
| 17 #include "content/browser/renderer_host/render_view_host.h" | |
| 18 #include "content/public/browser/notification_service.h" | |
| 19 #include "content/public/browser/notification_source.h" | |
| 20 #include "content/public/browser/render_view_host_delegate.h" | |
| 21 #include "content/public/browser/web_contents.h" | |
| 22 #include "content/public/browser/web_ui.h" | |
| 23 #include "third_party/skia/include/core/SkBitmap.h" | |
| 24 #include "webkit/glue/webpreferences.h" | |
| 25 | |
| 26 namespace { | |
| 27 | |
| 28 Value* CreateColumnValue(const TaskManagerModel* tm, | |
| 29 const std::string column_name, | |
| 30 const int i) { | |
| 31 if (column_name == "uniqueId") | |
| 32 return Value::CreateIntegerValue(tm->GetResourceUniqueId(i)); | |
| 33 if (column_name == "type") | |
| 34 return Value::CreateStringValue( | |
| 35 TaskManager::Resource::GetResourceTypeAsString( | |
| 36 tm->GetResourceType(i))); | |
| 37 if (column_name == "processId") | |
| 38 return Value::CreateStringValue(tm->GetResourceProcessId(i)); | |
| 39 if (column_name == "processIdValue") | |
| 40 return Value::CreateIntegerValue(tm->GetProcessId(i)); | |
| 41 if (column_name == "cpuUsage") | |
| 42 return Value::CreateStringValue(tm->GetResourceCPUUsage(i)); | |
| 43 if (column_name == "cpuUsageValue") | |
| 44 return Value::CreateDoubleValue(tm->GetCPUUsage(i)); | |
| 45 if (column_name == "privateMemory") | |
| 46 return Value::CreateStringValue(tm->GetResourcePrivateMemory(i)); | |
| 47 if (column_name == "privateMemoryValue") { | |
| 48 size_t private_memory; | |
| 49 tm->GetPrivateMemory(i, &private_memory); | |
| 50 return Value::CreateDoubleValue(private_memory); | |
| 51 } | |
| 52 if (column_name == "sharedMemory") | |
| 53 return Value::CreateStringValue(tm->GetResourceSharedMemory(i)); | |
| 54 if (column_name == "sharedMemoryValue") { | |
| 55 size_t shared_memory; | |
| 56 tm->GetSharedMemory(i, &shared_memory); | |
| 57 return Value::CreateDoubleValue(shared_memory); | |
| 58 } | |
| 59 if (column_name == "physicalMemory") | |
| 60 return Value::CreateStringValue(tm->GetResourcePhysicalMemory(i)); | |
| 61 if (column_name == "physicalMemoryValue") { | |
| 62 size_t physical_memory; | |
| 63 tm->GetPhysicalMemory(i, &physical_memory); | |
| 64 return Value::CreateDoubleValue(physical_memory); | |
| 65 } | |
| 66 if (column_name == "icon") | |
| 67 return Value::CreateStringValue( | |
| 68 web_ui_util::GetImageDataUrl(tm->GetResourceIcon(i))); | |
| 69 if (column_name == "title") | |
| 70 return Value::CreateStringValue(tm->GetResourceTitle(i)); | |
| 71 if (column_name == "profileName") | |
| 72 return Value::CreateStringValue(tm->GetResourceProfileName(i)); | |
| 73 if (column_name == "networkUsage") | |
| 74 return Value::CreateStringValue(tm->GetResourceNetworkUsage(i)); | |
| 75 if (column_name == "networkUsageValue") | |
| 76 return Value::CreateDoubleValue(tm->GetNetworkUsage(i)); | |
| 77 if (column_name == "webCoreImageCacheSize") | |
| 78 return Value::CreateStringValue(tm->GetResourceWebCoreImageCacheSize(i)); | |
| 79 if (column_name == "webCoreImageCacheSizeValue") { | |
| 80 WebKit::WebCache::ResourceTypeStats resource_stats; | |
| 81 tm->GetWebCoreCacheStats(i, &resource_stats); | |
| 82 return Value::CreateDoubleValue(resource_stats.images.size); | |
| 83 } | |
| 84 if (column_name == "webCoreScriptsCacheSize") | |
| 85 return Value::CreateStringValue(tm->GetResourceWebCoreScriptsCacheSize(i)); | |
| 86 if (column_name == "webCoreScriptsCacheSizeValue") { | |
| 87 WebKit::WebCache::ResourceTypeStats resource_stats; | |
| 88 tm->GetWebCoreCacheStats(i, &resource_stats); | |
| 89 return Value::CreateDoubleValue(resource_stats.scripts.size); | |
| 90 } | |
| 91 if (column_name == "webCoreCSSCacheSize") | |
| 92 return Value::CreateStringValue(tm->GetResourceWebCoreCSSCacheSize(i)); | |
| 93 if (column_name == "webCoreCSSCacheSizeValue") { | |
| 94 WebKit::WebCache::ResourceTypeStats resource_stats; | |
| 95 tm->GetWebCoreCacheStats(i, &resource_stats); | |
| 96 return Value::CreateDoubleValue(resource_stats.cssStyleSheets.size); | |
| 97 } | |
| 98 if (column_name == "fps") | |
| 99 return Value::CreateStringValue(tm->GetResourceFPS(i)); | |
| 100 if (column_name == "fpsValue") { | |
| 101 float fps; | |
| 102 tm->GetFPS(i, &fps); | |
| 103 return Value::CreateDoubleValue(fps); | |
| 104 } | |
| 105 if (column_name == "sqliteMemoryUsed") | |
| 106 return Value::CreateStringValue(tm->GetResourceSqliteMemoryUsed(i)); | |
| 107 if (column_name == "sqliteMemoryUsedValue") { | |
| 108 size_t sqlite_memory; | |
| 109 tm->GetSqliteMemoryUsedBytes(i, &sqlite_memory); | |
| 110 return Value::CreateDoubleValue(sqlite_memory); | |
| 111 } | |
| 112 if (column_name == "goatsTeleported") | |
| 113 return Value::CreateStringValue(tm->GetResourceGoatsTeleported(i)); | |
| 114 if (column_name == "goatsTeleportedValue") | |
| 115 return Value::CreateIntegerValue(tm->GetGoatsTeleported(i)); | |
| 116 if (column_name == "v8MemoryAllocatedSize") | |
| 117 return Value::CreateStringValue(tm->GetResourceV8MemoryAllocatedSize(i)); | |
| 118 if (column_name == "v8MemoryAllocatedSizeValue") { | |
| 119 size_t v8_memory; | |
| 120 tm->GetV8Memory(i, &v8_memory); | |
| 121 return Value::CreateDoubleValue(v8_memory); | |
| 122 } | |
| 123 if (column_name == "canInspect") | |
| 124 return Value::CreateBooleanValue(tm->CanInspect(i)); | |
| 125 if (column_name == "canActivate") | |
| 126 return Value::CreateBooleanValue(tm->CanActivate(i)); | |
| 127 | |
| 128 NOTREACHED(); | |
| 129 return NULL; | |
| 130 } | |
| 131 | |
| 132 void CreateGroupColumnList(const TaskManagerModel* tm, | |
| 133 const std::string& column_name, | |
| 134 const int index, | |
| 135 const int length, | |
| 136 DictionaryValue* val) { | |
| 137 ListValue *list = new ListValue(); | |
| 138 for (int i = index; i < (index + length); ++i) { | |
| 139 list->Append(CreateColumnValue(tm, column_name, i)); | |
| 140 } | |
| 141 val->Set(column_name, list); | |
| 142 } | |
| 143 | |
| 144 struct ColumnType { | |
| 145 const char* column_id; | |
| 146 // Whether the column has the real value separately or not, instead of the | |
| 147 // formatted value to display. | |
| 148 const bool has_real_value; | |
| 149 // Whether the column has single datum or multiple data in each group. | |
| 150 const bool has_multiple_data; | |
| 151 }; | |
| 152 | |
| 153 const ColumnType kColumnsList[] = { | |
| 154 {"type", false, false}, | |
| 155 {"processId", true, false}, | |
| 156 {"cpuUsage", true, false}, | |
| 157 {"physicalMemory", true, false}, | |
| 158 {"sharedMemory", true, false}, | |
| 159 {"privateMemory", true, false}, | |
| 160 {"webCoreImageCacheSize", true, false}, | |
| 161 {"webCoreImageCacheSize", true, false}, | |
| 162 {"webCoreScriptsCacheSize", true, false}, | |
| 163 {"webCoreCSSCacheSize", true, false}, | |
| 164 {"sqliteMemoryUsed", true, false}, | |
| 165 {"v8MemoryAllocatedSize", true, false}, | |
| 166 {"icon", false, true}, | |
| 167 {"title", false, true}, | |
| 168 {"profileName", false, true}, | |
| 169 {"networkUsage", true, true}, | |
| 170 {"fps", true, true}, | |
| 171 {"goatsTeleported", true, true}, | |
| 172 {"canInspect", false, true}, | |
| 173 {"canActivate", false, true} | |
| 174 }; | |
| 175 | |
| 176 DictionaryValue* CreateTaskGroupValue( | |
| 177 const TaskManagerModel* tm, | |
| 178 const int group_index, | |
| 179 const std::set<std::string>& columns) { | |
| 180 DictionaryValue* val = new DictionaryValue(); | |
| 181 | |
| 182 const int group_count = tm->GroupCount(); | |
| 183 if (group_index >= group_count) | |
| 184 return val; | |
| 185 | |
| 186 int index = tm->GetResourceIndexForGroup(group_index, 0); | |
| 187 std::pair<int, int> group_range; | |
| 188 group_range = tm->GetGroupRangeForResource(index); | |
| 189 int length = group_range.second; | |
| 190 | |
| 191 // Forces to set following 3 columns regardless of |enable_columns|. | |
| 192 val->SetInteger("index", index); | |
| 193 val->SetBoolean("isBackgroundResource", | |
| 194 tm->IsBackgroundResource(index)); | |
| 195 CreateGroupColumnList(tm, "uniqueId", index, length, val); | |
| 196 CreateGroupColumnList(tm, "processId", index, 1, val); | |
| 197 | |
| 198 for (size_t i = 0; i < arraysize(kColumnsList); ++i) { | |
| 199 const std::string column_id = kColumnsList[i].column_id; | |
| 200 | |
| 201 if (columns.find(column_id) == columns.end()) | |
| 202 continue; | |
| 203 | |
| 204 int column_length = kColumnsList[i].has_multiple_data ? length : 1; | |
| 205 CreateGroupColumnList(tm, column_id, index, column_length, val); | |
| 206 | |
| 207 if (kColumnsList[i].has_real_value) | |
| 208 CreateGroupColumnList(tm, column_id + "Value", index, column_length, val); | |
| 209 } | |
| 210 | |
| 211 return val; | |
| 212 } | |
| 213 | |
| 214 } // namespace | |
| 215 | |
| 216 TaskManagerHandler::TaskManagerHandler(TaskManager* tm) | |
| 217 : task_manager_(tm), | |
| 218 model_(tm->model()), | |
| 219 is_enabled_(false) { | |
| 220 } | |
| 221 | |
| 222 TaskManagerHandler::~TaskManagerHandler() { | |
| 223 DisableTaskManager(NULL); | |
| 224 } | |
| 225 | |
| 226 // TaskManagerHandler, public: ----------------------------------------------- | |
| 227 | |
| 228 void TaskManagerHandler::OnModelChanged() { | |
| 229 OnGroupChanged(0, model_->GroupCount()); | |
| 230 } | |
| 231 | |
| 232 void TaskManagerHandler::OnItemsChanged(const int start, const int length) { | |
| 233 OnGroupChanged(0, model_->GroupCount()); | |
| 234 } | |
| 235 | |
| 236 void TaskManagerHandler::OnItemsAdded(const int start, const int length) { | |
| 237 } | |
| 238 | |
| 239 void TaskManagerHandler::OnItemsRemoved(const int start, const int length) { | |
| 240 } | |
| 241 | |
| 242 void TaskManagerHandler::RegisterMessages() { | |
| 243 web_ui()->RegisterMessageCallback("killProcesses", | |
| 244 base::Bind(&TaskManagerHandler::HandleKillProcesses, | |
| 245 base::Unretained(this))); | |
| 246 web_ui()->RegisterMessageCallback("inspect", | |
| 247 base::Bind(&TaskManagerHandler::HandleInspect, | |
| 248 base::Unretained(this))); | |
| 249 web_ui()->RegisterMessageCallback("activatePage", | |
| 250 base::Bind(&TaskManagerHandler::HandleActivatePage, | |
| 251 base::Unretained(this))); | |
| 252 web_ui()->RegisterMessageCallback("openAboutMemory", | |
| 253 base::Bind(&TaskManagerHandler::OpenAboutMemory, | |
| 254 base::Unretained(this))); | |
| 255 web_ui()->RegisterMessageCallback("enableTaskManager", | |
| 256 base::Bind(&TaskManagerHandler::EnableTaskManager, | |
| 257 base::Unretained(this))); | |
| 258 web_ui()->RegisterMessageCallback("disableTaskManager", | |
| 259 base::Bind(&TaskManagerHandler::DisableTaskManager, | |
| 260 base::Unretained(this))); | |
| 261 web_ui()->RegisterMessageCallback("setUpdateColumn", | |
| 262 base::Bind(&TaskManagerHandler::HandleSetUpdateColumn, | |
| 263 base::Unretained(this))); | |
| 264 } | |
| 265 | |
| 266 static int parseIndex(const Value* value) { | |
| 267 int index = -1; | |
| 268 string16 string16_index; | |
| 269 double double_index; | |
| 270 if (value->GetAsString(&string16_index)) { | |
| 271 bool converted = base::StringToInt(string16_index, &index); | |
| 272 DCHECK(converted); | |
| 273 } else if (value->GetAsDouble(&double_index)) { | |
| 274 index = static_cast<int>(double_index); | |
| 275 } else { | |
| 276 value->GetAsInteger(&index); | |
| 277 } | |
| 278 return index; | |
| 279 } | |
| 280 | |
| 281 void TaskManagerHandler::HandleKillProcesses(const ListValue* unique_ids) { | |
| 282 for (ListValue::const_iterator i = unique_ids->begin(); | |
| 283 i != unique_ids->end(); ++i) { | |
| 284 int unique_id = parseIndex(*i); | |
| 285 int resource_index = model_->GetResourceIndexByUniqueId(unique_id); | |
| 286 if (resource_index == -1) | |
| 287 continue; | |
| 288 | |
| 289 task_manager_->KillProcess(resource_index); | |
| 290 } | |
| 291 } | |
| 292 | |
| 293 void TaskManagerHandler::HandleActivatePage(const ListValue* unique_ids) { | |
| 294 for (ListValue::const_iterator i = unique_ids->begin(); | |
| 295 i != unique_ids->end(); ++i) { | |
| 296 int unique_id = parseIndex(*i); | |
| 297 int resource_index = model_->GetResourceIndexByUniqueId(unique_id); | |
| 298 if (resource_index == -1) | |
| 299 continue; | |
| 300 | |
| 301 task_manager_->ActivateProcess(resource_index); | |
| 302 break; | |
| 303 } | |
| 304 } | |
| 305 | |
| 306 void TaskManagerHandler::HandleInspect(const ListValue* unique_ids) { | |
| 307 for (ListValue::const_iterator i = unique_ids->begin(); | |
| 308 i != unique_ids->end(); ++i) { | |
| 309 int unique_id = parseIndex(*i); | |
| 310 int resource_index = model_->GetResourceIndexByUniqueId(unique_id); | |
| 311 if (resource_index == -1) | |
| 312 continue; | |
| 313 | |
| 314 if (model_->CanInspect(resource_index)) | |
| 315 model_->Inspect(resource_index); | |
| 316 break; | |
| 317 } | |
| 318 } | |
| 319 | |
| 320 void TaskManagerHandler::DisableTaskManager(const ListValue* indexes) { | |
| 321 if (!is_enabled_) | |
| 322 return; | |
| 323 | |
| 324 is_enabled_ = false; | |
| 325 model_->StopUpdating(); | |
| 326 model_->RemoveObserver(this); | |
| 327 } | |
| 328 | |
| 329 void TaskManagerHandler::EnableTaskManager(const ListValue* indexes) { | |
| 330 if (is_enabled_) | |
| 331 return; | |
| 332 | |
| 333 is_enabled_ = true; | |
| 334 | |
| 335 OnGroupChanged(0, model_->GroupCount()); | |
| 336 | |
| 337 model_->AddObserver(this); | |
| 338 model_->StartUpdating(); | |
| 339 | |
| 340 content::NotificationService::current()->Notify( | |
| 341 chrome::NOTIFICATION_TASK_MANAGER_WINDOW_READY, | |
| 342 content::Source<TaskManagerModel>(model_), | |
| 343 content::NotificationService::NoDetails()); | |
| 344 } | |
| 345 | |
| 346 void TaskManagerHandler::OpenAboutMemory(const ListValue* indexes) { | |
| 347 RenderViewHost* rvh = web_ui()->GetWebContents()->GetRenderViewHost(); | |
| 348 if (rvh && rvh->delegate()) { | |
| 349 WebPreferences webkit_prefs = rvh->delegate()->GetWebkitPrefs(); | |
| 350 webkit_prefs.allow_scripts_to_close_windows = true; | |
| 351 rvh->UpdateWebkitPreferences(webkit_prefs); | |
| 352 } else { | |
| 353 DCHECK(false); | |
| 354 } | |
| 355 | |
| 356 task_manager_->OpenAboutMemory(); | |
| 357 } | |
| 358 | |
| 359 void TaskManagerHandler::HandleSetUpdateColumn(const ListValue* args) { | |
| 360 DCHECK_EQ(2U, args->GetSize()); | |
| 361 | |
| 362 bool ret = true; | |
| 363 std::string column_id; | |
| 364 ret &= args->GetString(0, &column_id); | |
| 365 bool is_enabled; | |
| 366 ret &= args->GetBoolean(1, &is_enabled); | |
| 367 DCHECK(ret); | |
| 368 | |
| 369 if (is_enabled) | |
| 370 enabled_columns_.insert(column_id); | |
| 371 else | |
| 372 enabled_columns_.erase(column_id); | |
| 373 } | |
| 374 | |
| 375 // TaskManagerHandler, private: ----------------------------------------------- | |
| 376 | |
| 377 bool TaskManagerHandler::is_alive() { | |
| 378 return web_ui()->GetWebContents()->GetRenderViewHost() != NULL; | |
| 379 } | |
| 380 | |
| 381 void TaskManagerHandler::OnGroupChanged(const int group_start, | |
| 382 const int group_length) { | |
| 383 base::FundamentalValue start_value(group_start); | |
| 384 base::FundamentalValue length_value(group_length); | |
| 385 base::ListValue tasks_value; | |
| 386 | |
| 387 for (int i = 0; i < group_length; ++i) | |
| 388 tasks_value.Append( | |
| 389 CreateTaskGroupValue(model_, group_start + i, enabled_columns_)); | |
| 390 | |
| 391 if (is_enabled_ && is_alive()) { | |
| 392 web_ui()->CallJavascriptFunction("taskChanged", | |
| 393 start_value, length_value, tasks_value); | |
| 394 } | |
| 395 } | |
| 396 | |
| 397 void TaskManagerHandler::OnGroupAdded(const int group_start, | |
| 398 const int group_length) { | |
| 399 } | |
| 400 | |
| 401 void TaskManagerHandler::OnGroupRemoved(const int group_start, | |
| 402 const int group_length) { | |
| 403 } | |
| 404 | |
| 405 void TaskManagerHandler::OnReadyPeriodicalUpdate() { | |
| 406 } | |
| OLD | NEW |