ProcessStatisticsReader.cpp 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  1. /*
  2. * Copyright (c) 2018-2021, Andreas Kling <kling@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/ByteBuffer.h>
  7. #include <AK/JsonArray.h>
  8. #include <AK/JsonObject.h>
  9. #include <AK/JsonValue.h>
  10. #include <LibCore/ProcessStatisticsReader.h>
  11. #include <pwd.h>
  12. namespace Core {
  13. HashMap<uid_t, DeprecatedString> ProcessStatisticsReader::s_usernames;
  14. ErrorOr<AllProcessesStatistics> ProcessStatisticsReader::get_all(Core::Stream::SeekableStream& proc_all_file, bool include_usernames)
  15. {
  16. TRY(proc_all_file.seek(0, Core::Stream::SeekMode::SetPosition));
  17. AllProcessesStatistics all_processes_statistics;
  18. auto file_contents = TRY(proc_all_file.read_until_eof());
  19. auto json_obj = TRY(JsonValue::from_string(file_contents)).as_object();
  20. json_obj.get_array("processes"sv)->for_each([&](auto& value) {
  21. const JsonObject& process_object = value.as_object();
  22. Core::ProcessStatistics process;
  23. // kernel data first
  24. process.pid = process_object.get_u32("pid"sv).value_or(0);
  25. process.pgid = process_object.get_u32("pgid"sv).value_or(0);
  26. process.pgp = process_object.get_u32("pgp"sv).value_or(0);
  27. process.sid = process_object.get_u32("sid"sv).value_or(0);
  28. process.uid = process_object.get_u32("uid"sv).value_or(0);
  29. process.gid = process_object.get_u32("gid"sv).value_or(0);
  30. process.ppid = process_object.get_u32("ppid"sv).value_or(0);
  31. process.nfds = process_object.get_u32("nfds"sv).value_or(0);
  32. process.kernel = process_object.get_bool("kernel"sv).value_or(false);
  33. process.name = process_object.get_deprecated_string("name"sv).value_or("");
  34. process.executable = process_object.get_deprecated_string("executable"sv).value_or("");
  35. process.tty = process_object.get_deprecated_string("tty"sv).value_or("");
  36. process.pledge = process_object.get_deprecated_string("pledge"sv).value_or("");
  37. process.veil = process_object.get_deprecated_string("veil"sv).value_or("");
  38. process.amount_virtual = process_object.get_u32("amount_virtual"sv).value_or(0);
  39. process.amount_resident = process_object.get_u32("amount_resident"sv).value_or(0);
  40. process.amount_shared = process_object.get_u32("amount_shared"sv).value_or(0);
  41. process.amount_dirty_private = process_object.get_u32("amount_dirty_private"sv).value_or(0);
  42. process.amount_clean_inode = process_object.get_u32("amount_clean_inode"sv).value_or(0);
  43. process.amount_purgeable_volatile = process_object.get_u32("amount_purgeable_volatile"sv).value_or(0);
  44. process.amount_purgeable_nonvolatile = process_object.get_u32("amount_purgeable_nonvolatile"sv).value_or(0);
  45. auto& thread_array = process_object.get_array("threads"sv).value();
  46. process.threads.ensure_capacity(thread_array.size());
  47. thread_array.for_each([&](auto& value) {
  48. auto& thread_object = value.as_object();
  49. Core::ThreadStatistics thread;
  50. thread.tid = thread_object.get_u32("tid"sv).value_or(0);
  51. thread.times_scheduled = thread_object.get_u32("times_scheduled"sv).value_or(0);
  52. thread.name = thread_object.get_deprecated_string("name"sv).value_or("");
  53. thread.state = thread_object.get_deprecated_string("state"sv).value_or("");
  54. thread.time_user = thread_object.get_u64("time_user"sv).value_or(0);
  55. thread.time_kernel = thread_object.get_u64("time_kernel"sv).value_or(0);
  56. thread.cpu = thread_object.get_u32("cpu"sv).value_or(0);
  57. thread.priority = thread_object.get_u32("priority"sv).value_or(0);
  58. thread.syscall_count = thread_object.get_u32("syscall_count"sv).value_or(0);
  59. thread.inode_faults = thread_object.get_u32("inode_faults"sv).value_or(0);
  60. thread.zero_faults = thread_object.get_u32("zero_faults"sv).value_or(0);
  61. thread.cow_faults = thread_object.get_u32("cow_faults"sv).value_or(0);
  62. thread.unix_socket_read_bytes = thread_object.get_u32("unix_socket_read_bytes"sv).value_or(0);
  63. thread.unix_socket_write_bytes = thread_object.get_u32("unix_socket_write_bytes"sv).value_or(0);
  64. thread.ipv4_socket_read_bytes = thread_object.get_u32("ipv4_socket_read_bytes"sv).value_or(0);
  65. thread.ipv4_socket_write_bytes = thread_object.get_u32("ipv4_socket_write_bytes"sv).value_or(0);
  66. thread.file_read_bytes = thread_object.get_u32("file_read_bytes"sv).value_or(0);
  67. thread.file_write_bytes = thread_object.get_u32("file_write_bytes"sv).value_or(0);
  68. process.threads.append(move(thread));
  69. });
  70. // and synthetic data last
  71. if (include_usernames) {
  72. process.username = username_from_uid(process.uid);
  73. }
  74. all_processes_statistics.processes.append(move(process));
  75. });
  76. all_processes_statistics.total_time_scheduled = json_obj.get_u64("total_time"sv).value_or(0);
  77. all_processes_statistics.total_time_scheduled_kernel = json_obj.get_u64("total_time_kernel"sv).value_or(0);
  78. return all_processes_statistics;
  79. }
  80. ErrorOr<AllProcessesStatistics> ProcessStatisticsReader::get_all(bool include_usernames)
  81. {
  82. auto proc_all_file = TRY(Core::Stream::File::open("/sys/kernel/processes"sv, Core::Stream::OpenMode::Read));
  83. return get_all(*proc_all_file, include_usernames);
  84. }
  85. DeprecatedString ProcessStatisticsReader::username_from_uid(uid_t uid)
  86. {
  87. if (s_usernames.is_empty()) {
  88. setpwent();
  89. while (auto* passwd = getpwent())
  90. s_usernames.set(passwd->pw_uid, passwd->pw_name);
  91. endpwent();
  92. }
  93. auto it = s_usernames.find(uid);
  94. if (it != s_usernames.end())
  95. return (*it).value;
  96. return DeprecatedString::number(uid);
  97. }
  98. }