mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-11-25 00:50:22 +00:00
78 lines
3.2 KiB
C++
78 lines
3.2 KiB
C++
/*
|
|
* Copyright (c) 2024, Andrew Kaster <akaster@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/String.h>
|
|
#include <LibCore/File.h>
|
|
#include <LibCore/Platform/ProcessStatistics.h>
|
|
#include <unistd.h>
|
|
|
|
namespace Core::Platform {
|
|
|
|
static auto user_hz = sysconf(_SC_CLK_TCK);
|
|
static auto page_size = sysconf(_SC_PAGESIZE);
|
|
static auto ncpu_online = sysconf(_SC_NPROCESSORS_ONLN);
|
|
|
|
ErrorOr<void> update_process_statistics(ProcessStatistics& statistics)
|
|
{
|
|
// Read the total time scheduled from /proc/stat, and each process's usage from /proc/pid/stat
|
|
// Calculate the CPU percentage for each process based on the total time scheduled and the time spent in the process
|
|
|
|
static auto proc_stat = TRY(Core::File::open("/proc/stat"sv, Core::File::OpenMode::Read));
|
|
TRY(proc_stat->seek(0, SeekMode::SetPosition));
|
|
|
|
char buf[1024] = {};
|
|
auto buffer = Bytes { buf, sizeof(buf) };
|
|
auto line = TRY(proc_stat->read_some(buffer));
|
|
|
|
int user_time = 0;
|
|
int system_time = 0;
|
|
int idle_time = 0;
|
|
int irq_time = 0;
|
|
int softirq_time = 0;
|
|
|
|
// user, nice (ignored), system, idle, iowait (ignored), irq, softirq, steal (ignored), guest (ignored), guest_nice (ignored)
|
|
int res = sscanf(reinterpret_cast<char const*>(line.data()), "cpu %d %*d %d %d %*d %d %d", &user_time, &system_time, &idle_time, &irq_time, &softirq_time);
|
|
if (res != 5)
|
|
return Error::from_string_literal("Failed to parse /proc/stat");
|
|
|
|
u64 const total_time_scheduled = user_time + system_time + idle_time + irq_time + softirq_time;
|
|
float const total_time_scheduled_diff = total_time_scheduled - statistics.total_time_scheduled;
|
|
statistics.total_time_scheduled = total_time_scheduled;
|
|
|
|
for (auto& process : statistics.processes) {
|
|
auto proc_pid_stat_or_error = Core::File::open(MUST(String::formatted("/proc/{}/stat", process->pid)), Core::File::OpenMode::Read);
|
|
if (proc_pid_stat_or_error.is_error()) {
|
|
// FIXME: Remove stale process from process list?
|
|
continue;
|
|
}
|
|
auto proc_pid_stat = proc_pid_stat_or_error.release_value();
|
|
line = TRY(proc_pid_stat->read_some(buffer));
|
|
|
|
// We only care about fields 14 (utime), 15 (stime), and 24 (rss)
|
|
unsigned long utime = 0;
|
|
unsigned long stime = 0;
|
|
long rss = 0;
|
|
// 1 2 3 4 5 6 7 8 9 10 11 12 13 14* 15* 16 17 18 19 20 21 22 23 24*
|
|
res = sscanf(reinterpret_cast<char const*>(line.data()), "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %lu %lu %*d %*d %*d %*d %*d %*d %*u %*u %ld", &utime, &stime, &rss);
|
|
if (res != 3)
|
|
return Error::from_string_literal("Failed to parse /proc/pid/stat");
|
|
|
|
process->memory_usage_bytes = rss * page_size;
|
|
|
|
u64 const time_process = utime + stime;
|
|
float const time_scheduled_diff = time_process - process->time_spent_in_process;
|
|
process->time_spent_in_process = time_process;
|
|
|
|
process->cpu_percent = 0.0;
|
|
if (total_time_scheduled_diff > 0) {
|
|
process->cpu_percent = time_scheduled_diff / (total_time_scheduled_diff / ncpu_online) * 100.0f;
|
|
}
|
|
}
|
|
|
|
return {};
|
|
}
|
|
|
|
}
|