/* * Copyright (c) 2018-2021, Andreas Kling * Copyright (c) 2021, Undefine * Copyright (c) 2022, the SerenityOS developers. * * SPDX-License-Identifier: BSD-2-Clause */ #include "GraphWidget.h" #include "MemoryStatsWidget.h" #include "NetworkStatisticsWidget.h" #include "ProcessFileDescriptorMapWidget.h" #include "ProcessMemoryMapWidget.h" #include "ProcessModel.h" #include "ProcessStateWidget.h" #include "ProcessUnveiledPathsWidget.h" #include "ThreadStackWidget.h" #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include static ErrorOr> build_process_window(pid_t); static void build_performance_tab(GUI::Widget&); static RefPtr statusbar; namespace SystemMonitor { class ProgressbarPaintingDelegate final : public GUI::TableCellPaintingDelegate { public: virtual ~ProgressbarPaintingDelegate() override = default; virtual void paint(GUI::Painter& painter, Gfx::IntRect const& a_rect, Palette const& palette, GUI::ModelIndex const& index) override { auto rect = a_rect.shrunken(2, 2); auto percentage = index.data(GUI::ModelRole::Custom).to_i32(); auto data = index.data(); DeprecatedString text; if (data.is_string()) text = data.as_string(); Gfx::StylePainter::paint_progressbar(painter, rect, palette, 0, 100, percentage, text); painter.draw_rect(rect, Color::Black); } }; class UnavailableProcessWidget final : public GUI::Frame { C_OBJECT(UnavailableProcessWidget) public: virtual ~UnavailableProcessWidget() override = default; DeprecatedString const& text() const { return m_text; } void set_text(DeprecatedString text) { m_text = move(text); update(); } private: UnavailableProcessWidget() { REGISTER_DEPRECATED_STRING_PROPERTY("text", text, set_text); } virtual void paint_event(GUI::PaintEvent& event) override { Frame::paint_event(event); if (text().is_empty()) return; GUI::Painter painter(*this); painter.add_clip_rect(event.rect()); painter.draw_text(frame_inner_rect(), text(), Gfx::TextAlignment::Center, palette().window_text(), Gfx::TextElision::Right); } DeprecatedString m_text; }; class StorageTabWidget final : public GUI::LazyWidget { C_OBJECT(StorageTabWidget) public: StorageTabWidget() { this->on_first_show = [](GUI::LazyWidget& self) { auto& fs_table_view = *self.find_child_of_type_named("storage_table"); Vector df_fields; df_fields.empend("mount_point", "Mount point", Gfx::TextAlignment::CenterLeft); df_fields.empend("class_name", "Class", Gfx::TextAlignment::CenterLeft); df_fields.empend("source", "Source", Gfx::TextAlignment::CenterLeft); df_fields.empend( "Size", Gfx::TextAlignment::CenterRight, [](JsonObject const& object) { StringBuilder size_builder; size_builder.append(' '); size_builder.append(human_readable_size(object.get_u64("total_block_count"sv).value_or(0) * object.get_u64("block_size"sv).value_or(0))); size_builder.append(' '); return size_builder.to_deprecated_string(); }, [](JsonObject const& object) { return object.get_u64("total_block_count"sv).value_or(0) * object.get_u64("block_size"sv).value_or(0); }, [](JsonObject const& object) { auto total_blocks = object.get_u64("total_block_count"sv).value_or(0); if (total_blocks == 0) return 0; auto free_blocks = object.get_u64("free_block_count"sv).value_or(0); auto used_blocks = total_blocks - free_blocks; int percentage = (static_cast(used_blocks) / static_cast(total_blocks) * 100.0); return percentage; }); df_fields.empend( "Used", Gfx::TextAlignment::CenterRight, [](JsonObject const& object) { auto total_blocks = object.get_u64("total_block_count"sv).value_or(0); auto free_blocks = object.get_u64("free_block_count"sv).value_or(0); auto used_blocks = total_blocks - free_blocks; return human_readable_size(used_blocks * object.get_u64("block_size"sv).value_or(0)); }, [](JsonObject const& object) { auto total_blocks = object.get_u64("total_block_count"sv).value_or(0); auto free_blocks = object.get_u64("free_block_count"sv).value_or(0); auto used_blocks = total_blocks - free_blocks; return used_blocks * object.get_u64("block_size"sv).value_or(0); }); df_fields.empend( "Available", Gfx::TextAlignment::CenterRight, [](JsonObject const& object) { return human_readable_size(object.get_u64("free_block_count"sv).value_or(0) * object.get_u64("block_size"sv).value_or(0)); }, [](JsonObject const& object) { return object.get_u64("free_block_count"sv).value_or(0) * object.get_u64("block_size"sv).value_or(0); }); df_fields.empend("Access", Gfx::TextAlignment::CenterLeft, [](JsonObject const& object) { bool readonly = object.get_bool("readonly"sv).value_or(false); int mount_flags = object.get_i32("mount_flags"sv).value_or(0); return readonly || (mount_flags & MS_RDONLY) ? "Read-only" : "Read/Write"; }); df_fields.empend("Mount flags", Gfx::TextAlignment::CenterLeft, [](JsonObject const& object) { int mount_flags = object.get_i32("mount_flags"sv).value_or(0); StringBuilder builder; bool first = true; auto check = [&](int flag, StringView name) { if (!(mount_flags & flag)) return; if (!first) builder.append(','); builder.append(name); first = false; }; check(MS_NODEV, "nodev"sv); check(MS_NOEXEC, "noexec"sv); check(MS_NOSUID, "nosuid"sv); check(MS_BIND, "bind"sv); check(MS_RDONLY, "ro"sv); check(MS_WXALLOWED, "wxallowed"sv); check(MS_AXALLOWED, "axallowed"sv); check(MS_NOREGULAR, "noregular"sv); if (builder.string_view().is_empty()) return DeprecatedString("defaults"); return builder.to_deprecated_string(); }); df_fields.empend("free_block_count", "Free blocks", Gfx::TextAlignment::CenterRight); df_fields.empend("total_block_count", "Total blocks", Gfx::TextAlignment::CenterRight); df_fields.empend("free_inode_count", "Free inodes", Gfx::TextAlignment::CenterRight); df_fields.empend("total_inode_count", "Total inodes", Gfx::TextAlignment::CenterRight); df_fields.empend("block_size", "Block size", Gfx::TextAlignment::CenterRight); fs_table_view.set_model(MUST(GUI::SortingProxyModel::create(GUI::JsonArrayModel::create("/sys/kernel/df", move(df_fields))))); fs_table_view.set_column_painting_delegate(3, make()); fs_table_view.model()->invalidate(); }; } }; } REGISTER_WIDGET(SystemMonitor, StorageTabWidget) REGISTER_WIDGET(SystemMonitor, UnavailableProcessWidget) static bool can_access_pid(pid_t pid) { int rc = kill(pid, 0); return rc == 0; } ErrorOr serenity_main(Main::Arguments arguments) { { // Before we do anything else, boost our process priority to the maximum allowed. // It's very frustrating when the system is bogged down under load and you just want // System Monitor to work. sched_param param { .sched_priority = THREAD_PRIORITY_MAX, }; sched_setparam(0, ¶m); } TRY(Core::System::pledge("stdio thread proc recvfd sendfd rpath exec unix")); auto app = TRY(GUI::Application::try_create(arguments)); Config::pledge_domain("SystemMonitor"); TRY(Core::System::unveil("/etc/passwd", "r")); TRY(Core::System::unveil("/res", "r")); TRY(Core::System::unveil("/proc", "r")); TRY(Core::System::unveil("/sys/kernel", "r")); TRY(Core::System::unveil("/dev", "r")); TRY(Core::System::unveil("/bin", "r")); TRY(Core::System::unveil("/bin/Escalator", "x")); TRY(Core::System::unveil("/usr/lib", "r")); // This directory only exists if ports are installed if (auto result = Core::System::unveil("/usr/local/bin", "r"); result.is_error() && result.error().code() != ENOENT) return result.release_error(); if (auto result = Core::System::unveil("/usr/local/lib", "r"); result.is_error() && result.error().code() != ENOENT) return result.release_error(); // This file is only accessible when running as root if it is available on the disk image. // It might be possible to not have this file on the disk image, if the user decided to not // include kernel symbols for debug purposes so don't fail if the error is ENOENT. if (auto result = Core::System::unveil("/boot/Kernel.debug", "r"); result.is_error() && (result.error().code() != EACCES && result.error().code() != ENOENT)) return result.release_error(); TRY(Core::System::unveil("/bin/Profiler", "rx")); TRY(Core::System::unveil("/bin/Inspector", "rx")); TRY(Core::System::unveil("/bin/HackStudio", "rx")); TRY(Core::System::unveil(nullptr, nullptr)); StringView args_tab = "processes"sv; Core::ArgsParser parser; parser.add_option(args_tab, "Tab, one of 'processes', 'graphs', 'fs', 'hardware', or 'network'", "open-tab", 't', "tab"); parser.parse(arguments); StringView args_tab_view = args_tab; auto app_icon = GUI::Icon::default_icon("app-system-monitor"sv); auto window = GUI::Window::construct(); window->set_title("System Monitor"); window->resize(560, 430); auto main_widget = TRY(window->set_main_widget()); TRY(main_widget->load_from_gml(system_monitor_gml)); auto& tabwidget = *main_widget->find_descendant_of_type_named("main_tabs"); statusbar = main_widget->find_descendant_of_type_named("statusbar"); auto& process_table_container = *tabwidget.find_descendant_of_type_named("processes"); auto process_model = ProcessModel::create(); process_model->on_state_update = [&](int process_count, int thread_count) { statusbar->set_text(0, DeprecatedString::formatted("Processes: {}", process_count)); statusbar->set_text(1, DeprecatedString::formatted("Threads: {}", thread_count)); }; auto& performance_widget = *tabwidget.find_descendant_of_type_named("performance"); build_performance_tab(performance_widget); auto& process_table_view = *process_table_container.find_child_of_type_named("process_table"); process_table_view.set_model(TRY(GUI::SortingProxyModel::create(process_model))); for (auto column = 0; column < ProcessModel::Column::__Count; ++column) { process_table_view.set_column_visible(column, Config::read_bool("SystemMonitor"sv, "ProcessTableColumns"sv, process_model->column_name(column), process_model->is_default_column(column))); } process_table_view.set_key_column_and_sort_order(ProcessModel::Column::CPU, GUI::SortOrder::Descending); process_model->update(); i32 frequency = Config::read_i32("SystemMonitor"sv, "Monitor"sv, "Frequency"sv, 3); if (frequency != 1 && frequency != 3 && frequency != 5) { frequency = 3; Config::write_i32("SystemMonitor"sv, "Monitor"sv, "Frequency"sv, frequency); } auto update_stats = [&] { // FIXME: remove the primitive re-toggling code once persistent model indices work. auto toggled_indices = process_table_view.selection().indices(); toggled_indices.remove_all_matching([&](auto const& index) { return !process_table_view.is_toggled(index); }); process_model->update(); if (!process_table_view.selection().is_empty()) process_table_view.selection().for_each_index([&](auto& selection) { if (toggled_indices.contains_slow(selection)) process_table_view.expand_all_parents_of(selection); }); if (auto* memory_stats_widget = SystemMonitor::MemoryStatsWidget::the()) memory_stats_widget->refresh(); }; update_stats(); auto& refresh_timer = window->add(frequency * 1000, move(update_stats)); refresh_timer.start(); auto selected_id = [&](ProcessModel::Column column) -> pid_t { if (process_table_view.selection().is_empty()) return -1; auto pid_index = process_table_view.model()->index(process_table_view.selection().first().row(), column, process_table_view.selection().first().parent()); return pid_index.data().to_i32(); }; auto selected_name = [&](ProcessModel::Column column) -> DeprecatedString { if (process_table_view.selection().is_empty()) return {}; auto pid_index = process_table_view.model()->index(process_table_view.selection().first().row(), column, process_table_view.selection().first().parent()); return pid_index.data().to_deprecated_string(); }; auto kill_action = GUI::Action::create( "&Kill Process", { Mod_Ctrl, Key_K }, { Key_Delete }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/kill.png"sv)), [&](const GUI::Action&) { pid_t pid = selected_id(ProcessModel::Column::PID); if (pid == -1) return; auto rc = GUI::MessageBox::show(window, DeprecatedString::formatted("Do you really want to kill \"{}\" (PID {})?", selected_name(ProcessModel::Column::Name), pid), "System Monitor"sv, GUI::MessageBox::Type::Question, GUI::MessageBox::InputType::YesNo); if (rc == GUI::Dialog::ExecResult::Yes) kill(pid, SIGKILL); }, &process_table_view); auto stop_action = GUI::Action::create( "&Stop Process", { Mod_Ctrl, Key_S }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/stop-hand.png"sv)), [&](const GUI::Action&) { pid_t pid = selected_id(ProcessModel::Column::PID); if (pid == -1) return; auto rc = GUI::MessageBox::show(window, DeprecatedString::formatted("Do you really want to stop \"{}\" (PID {})?", selected_name(ProcessModel::Column::Name), pid), "System Monitor"sv, GUI::MessageBox::Type::Question, GUI::MessageBox::InputType::YesNo); if (rc == GUI::Dialog::ExecResult::Yes) kill(pid, SIGSTOP); }, &process_table_view); auto continue_action = GUI::Action::create( "&Continue Process", { Mod_Ctrl, Key_C }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/continue.png"sv)), [&](const GUI::Action&) { pid_t pid = selected_id(ProcessModel::Column::PID); if (pid != -1) kill(pid, SIGCONT); }, &process_table_view); auto profile_action = GUI::Action::create( "&Profile Process", { Mod_Ctrl, Key_P }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/app-profiler.png"sv)), [&](auto&) { pid_t pid = selected_id(ProcessModel::Column::PID); if (pid == -1) return; auto pid_string = DeprecatedString::number(pid); GUI::Process::spawn_or_show_error(window, "/bin/Profiler"sv, Array { "--pid", pid_string.characters() }); }, &process_table_view); auto debug_action = GUI::Action::create( "Debug in HackStudio", { Mod_Ctrl, Key_D }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/app-hack-studio.png"sv)), [&](const GUI::Action&) { pid_t pid = selected_id(ProcessModel::Column::PID); if (pid == -1) return; auto pid_string = DeprecatedString::number(pid); GUI::Process::spawn_or_show_error(window, "/bin/HackStudio"sv, Array { "--pid", pid_string.characters() }); }, &process_table_view); HashMap> process_windows; auto process_properties_action = GUI::CommonActions::make_properties_action( [&](auto&) { auto pid = selected_id(ProcessModel::Column::PID); if (pid == -1) return; RefPtr process_window; auto it = process_windows.find(pid); if (it == process_windows.end()) { auto process_window_or_error = build_process_window(pid); if (process_window_or_error.is_error()) return; process_window = process_window_or_error.release_value(); process_window->on_close_request = [pid, &process_windows] { process_windows.remove(pid); return GUI::Window::CloseRequestDecision::Close; }; process_windows.set(pid, *process_window); } else { process_window = it->value; } process_window->show(); process_window->move_to_front(); }, &process_table_view); auto& file_menu = window->add_menu("&File"); file_menu.add_action(GUI::CommonActions::make_quit_action([](auto&) { GUI::Application::the()->quit(); })); auto process_context_menu = GUI::Menu::construct(); process_context_menu->add_action(kill_action); process_context_menu->add_action(stop_action); process_context_menu->add_action(continue_action); process_context_menu->add_separator(); process_context_menu->add_action(profile_action); process_context_menu->add_action(debug_action); process_context_menu->add_separator(); process_context_menu->add_action(process_properties_action); process_table_view.on_context_menu_request = [&]([[maybe_unused]] const GUI::ModelIndex& index, const GUI::ContextMenuEvent& event) { if (index.is_valid()) process_context_menu->popup(event.screen_position(), process_properties_action); }; auto& frequency_menu = window->add_menu("F&requency"); GUI::ActionGroup frequency_action_group; frequency_action_group.set_exclusive(true); auto make_frequency_action = [&](int seconds) { auto action = GUI::Action::create_checkable(DeprecatedString::formatted("&{} Sec", seconds), [&refresh_timer, seconds](auto&) { Config::write_i32("SystemMonitor"sv, "Monitor"sv, "Frequency"sv, seconds); refresh_timer.restart(seconds * 1000); }); action->set_status_tip(DeprecatedString::formatted("Refresh every {} seconds", seconds)); action->set_checked(frequency == seconds); frequency_action_group.add_action(*action); frequency_menu.add_action(*action); }; make_frequency_action(1); make_frequency_action(3); make_frequency_action(5); auto& help_menu = window->add_menu("&Help"); help_menu.add_action(GUI::CommonActions::make_command_palette_action(window)); help_menu.add_action(GUI::CommonActions::make_about_action("System Monitor", app_icon, window)); process_table_view.on_activation = [&](auto&) { if (process_properties_action->is_enabled()) process_properties_action->activate(); }; static pid_t last_selected_pid; process_table_view.on_selection_change = [&] { pid_t pid = selected_id(ProcessModel::Column::PID); if (pid == last_selected_pid || pid < 1) return; last_selected_pid = pid; bool has_access = can_access_pid(pid); kill_action->set_enabled(has_access); stop_action->set_enabled(has_access); continue_action->set_enabled(has_access); profile_action->set_enabled(has_access); debug_action->set_enabled(has_access); process_properties_action->set_enabled(has_access); }; app->on_action_enter = [](GUI::Action const& action) { statusbar->set_override_text(action.status_tip()); }; app->on_action_leave = [](GUI::Action const&) { statusbar->set_override_text({}); }; window->show(); window->set_icon(app_icon.bitmap_for_size(16)); if (args_tab_view == "processes") tabwidget.set_active_widget(&process_table_container); else if (args_tab_view == "graphs") tabwidget.set_active_widget(&performance_widget); else if (args_tab_view == "fs") tabwidget.set_active_widget(tabwidget.find_descendant_of_type_named("storage")); else if (args_tab_view == "network") tabwidget.set_active_widget(tabwidget.find_descendant_of_type_named("network")); int exec = app->exec(); // When exiting the application, save the configuration of the columns // to be loaded the next time the application is opened. auto& process_table_header = process_table_view.column_header(); for (auto column = 0; column < ProcessModel::Column::__Count; ++column) Config::write_bool("SystemMonitor"sv, "ProcessTableColumns"sv, process_model->column_name(column), process_table_header.is_section_visible(column)); return exec; } ErrorOr> build_process_window(pid_t pid) { auto window = GUI::Window::construct(); window->resize(480, 360); window->set_title(DeprecatedString::formatted("PID {} - System Monitor", pid)); auto app_icon = GUI::Icon::default_icon("app-system-monitor"sv); window->set_icon(app_icon.bitmap_for_size(16)); auto main_widget = TRY(window->set_main_widget()); TRY(main_widget->load_from_gml(process_window_gml)); GUI::ModelIndex process_index; for (int row = 0; row < ProcessModel::the().row_count({}); ++row) { auto index = ProcessModel::the().index(row, ProcessModel::Column::PID); if (index.data().to_i32() == pid) { process_index = index; break; } } VERIFY(process_index.is_valid()); if (auto icon_data = process_index.sibling_at_column(ProcessModel::Column::Icon).data(); icon_data.is_icon()) { main_widget->find_descendant_of_type_named("icon_label")->set_icon(icon_data.as_icon().bitmap_for_size(32)); } main_widget->find_descendant_of_type_named("process_name")->set_text(DeprecatedString::formatted("{} (PID {})", process_index.sibling_at_column(ProcessModel::Column::Name).data().to_deprecated_string(), pid)); main_widget->find_descendant_of_type_named("process_state")->set_pid(pid); main_widget->find_descendant_of_type_named("open_files")->set_pid(pid); main_widget->find_descendant_of_type_named("thread_stack")->set_ids(pid, pid); main_widget->find_descendant_of_type_named("memory_map")->set_pid(pid); main_widget->find_descendant_of_type_named("unveiled_paths")->set_pid(pid); auto& widget_stack = *main_widget->find_descendant_of_type_named("widget_stack"); auto& unavailable_process_widget = *widget_stack.find_descendant_of_type_named("unavailable_process"); unavailable_process_widget.set_text(DeprecatedString::formatted("Unable to access PID {}", pid)); if (can_access_pid(pid)) widget_stack.set_active_widget(widget_stack.find_descendant_of_type_named("available_process")); else widget_stack.set_active_widget(&unavailable_process_widget); return window; } void build_performance_tab(GUI::Widget& graphs_container) { auto& cpu_graph_group_box = *graphs_container.find_descendant_of_type_named("cpu_graph"); size_t cpu_graphs_per_row = min(4, ProcessModel::the().cpus().size()); auto cpu_graph_rows = ceil_div(ProcessModel::the().cpus().size(), cpu_graphs_per_row); cpu_graph_group_box.set_fixed_height(120u * cpu_graph_rows); Vector cpu_graphs; for (auto row = 0u; row < cpu_graph_rows; ++row) { auto& cpu_graph_row = cpu_graph_group_box.add(); cpu_graph_row.set_layout(6); cpu_graph_row.set_fixed_height(108); for (auto i = 0u; i < cpu_graphs_per_row; ++i) { auto& cpu_graph = cpu_graph_row.add(); cpu_graph.set_max(100); cpu_graph.set_value_format(0, { .graph_color_role = ColorRole::SyntaxPreprocessorStatement, .text_formatter = [](u64 value) { return DeprecatedString::formatted("Total: {}%", value); }, }); cpu_graph.set_value_format(1, { .graph_color_role = ColorRole::SyntaxPreprocessorValue, .text_formatter = [](u64 value) { return DeprecatedString::formatted("Kernel: {}%", value); }, }); cpu_graphs.append(cpu_graph); } } ProcessModel::the().on_cpu_info_change = [cpu_graphs](Vector> const& cpus) mutable { float sum_cpu = 0; for (size_t i = 0; i < cpus.size(); ++i) { cpu_graphs[i].add_value({ static_cast(cpus[i]->total_cpu_percent), static_cast(cpus[i]->total_cpu_percent_kernel) }); sum_cpu += cpus[i]->total_cpu_percent; } float cpu_usage = sum_cpu / (float)cpus.size(); statusbar->set_text(2, DeprecatedString::formatted("CPU usage: {}%", (int)roundf(cpu_usage))); }; auto& memory_graph = *graphs_container.find_descendant_of_type_named("memory_graph"); memory_graph.set_value_format(0, { .graph_color_role = ColorRole::SyntaxComment, .text_formatter = [](u64 bytes) { return DeprecatedString::formatted("Committed: {}", human_readable_size(bytes)); }, }); memory_graph.set_value_format(1, { .graph_color_role = ColorRole::SyntaxPreprocessorStatement, .text_formatter = [](u64 bytes) { return DeprecatedString::formatted("Allocated: {}", human_readable_size(bytes)); }, }); memory_graph.set_value_format(2, { .graph_color_role = ColorRole::SyntaxPreprocessorValue, .text_formatter = [](u64 bytes) { return DeprecatedString::formatted("Kernel heap: {}", human_readable_size(bytes)); }, }); }