cef/tests/cefclient/browser/task_manager_test.cc

129 lines
3.8 KiB
C++
Raw Normal View History

// Copyright (c) 2024 The Chromium Embedded Framework 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 "tests/cefclient/browser/task_manager_test.h"
#include "include/cef_task_manager.h"
namespace {
constexpr auto kTestUrlPath = "/task_manager";
struct TaskInfo {
int64_t task_id;
CefString title;
cef_task_type_t type;
bool is_killable;
};
std::string TaskTypeToString(cef_task_type_t type) {
switch (type) {
case CEF_TASK_TYPE_UNKNOWN:
return "Unknown";
case CEF_TASK_TYPE_BROWSER:
return "Browser";
case CEF_TASK_TYPE_GPU:
return "GPU";
case CEF_TASK_TYPE_ZYGOTE:
return "Zygote";
case CEF_TASK_TYPE_UTILITY:
return "Utility";
case CEF_TASK_TYPE_RENDERER:
return "Renderer";
case CEF_TASK_TYPE_EXTENSION:
return "Extension";
case CEF_TASK_TYPE_GUEST:
return "Guest";
case CEF_TASK_TYPE_PLUGIN:
return "Plugin";
case CEF_TASK_TYPE_SANDBOX_HELPER:
return "Sandbox Helper";
case CEF_TASK_TYPE_DEDICATED_WORKER:
return "Dedicated Worker";
case CEF_TASK_TYPE_SHARED_WORKER:
return "Shared Worker";
case CEF_TASK_TYPE_SERVICE_WORKER:
return "Service Worker";
}
return "Unknown";
}
std::string TasksToJsonString(const std::vector<CefTaskInfo>& tasks) {
std::string json = "[";
for (size_t i = 0; i < tasks.size(); ++i) {
const auto& task = tasks[i];
if (i > 0) {
json += ",";
}
json += "{";
json += "\"id\":" + std::to_string(task.id) + ",";
json += "\"type\":\"" + TaskTypeToString(task.type) + "\",";
json += "\"is_killable\":";
json += (task.is_killable ? "true," : "false,");
json += "\"title\":\"" + CefString(&task.title).ToString() + "\",";
json += "\"cpu_usage\":" + std::to_string(task.cpu_usage) + ",";
json += "\"number_of_processors\":" +
std::to_string(task.number_of_processors) + ",";
json += "\"memory\":" + std::to_string(task.memory) + ",";
json += "\"gpu_memory\":" + std::to_string(task.gpu_memory) + ",";
json += "\"is_gpu_memory_inflated\":";
json += (task.is_gpu_memory_inflated ? "true" : "false");
json += "}";
}
json += "]";
return json;
}
// Handle messages in the browser process.
class Handler final : public CefMessageRouterBrowserSide::Handler {
public:
explicit Handler() : task_manager_(CefTaskManager::GetTaskManager()) {}
// Called due to cefQuery execution in binary_transfer.html.
bool OnQuery(CefRefPtr<CefBrowser> browser,
CefRefPtr<CefFrame> frame,
int64_t query_id,
const CefString& request,
bool persistent,
CefRefPtr<Callback> callback) override {
// Only handle messages from the test URL.
const std::string& url = frame->GetURL();
if (!client::test_runner::IsTestURL(url, kTestUrlPath)) {
return false;
}
if (request == "get_tasks") {
CefTaskManager::TaskIdList task_ids;
task_manager_->GetTaskIdsList(task_ids);
std::vector<CefTaskInfo> tasks;
for (auto task_id : task_ids) {
CefTaskInfo info;
task_manager_->GetTaskInfo(task_id, info);
tasks.push_back(info);
}
callback->Success(TasksToJsonString(tasks));
return true;
}
// Otherwise assume that task id was passed as string to kill it
int64_t task_id = std::stoll(request.ToString());
task_manager_->KillTask(task_id);
callback->Success("");
return true;
}
private:
CefRefPtr<CefTaskManager> task_manager_;
};
} // namespace
namespace client::task_manager_test {
void CreateMessageHandlers(test_runner::MessageHandlerSet& handlers) {
handlers.insert(new Handler());
}
} // namespace client::task_manager_test