From 4b71e3a87ead7cf47a988522201d7d58b07d4003 Mon Sep 17 00:00:00 2001
From: Michael Scire <SciresM@gmail.com>
Date: Sat, 12 Dec 2020 05:34:31 -0800
Subject: [PATCH] kern: implement process cpu utilization debug

---
 .../include/mesosphere/kern_k_process.hpp     |   1 +
 .../source/kern_k_dump_object.cpp             | 136 +++++++++++++++++-
 2 files changed, 134 insertions(+), 3 deletions(-)

diff --git a/libraries/libmesosphere/include/mesosphere/kern_k_process.hpp b/libraries/libmesosphere/include/mesosphere/kern_k_process.hpp
index f58fe89ce..b8dc58b0b 100644
--- a/libraries/libmesosphere/include/mesosphere/kern_k_process.hpp
+++ b/libraries/libmesosphere/include/mesosphere/kern_k_process.hpp
@@ -279,6 +279,7 @@ namespace ams::kern {
             constexpr KProcessAddress GetProcessLocalRegionAddress() const { return this->plr_address; }
 
             void AddCpuTime(s64 diff) { this->cpu_time += diff; }
+            s64 GetCpuTime() { return this->cpu_time; }
 
             constexpr s64 GetScheduledCount() const { return this->schedule_count; }
             void IncrementScheduledCount() { ++this->schedule_count; }
diff --git a/libraries/libmesosphere/source/kern_k_dump_object.cpp b/libraries/libmesosphere/source/kern_k_dump_object.cpp
index 6582a8143..f1ecf1279 100644
--- a/libraries/libmesosphere/source/kern_k_dump_object.cpp
+++ b/libraries/libmesosphere/source/kern_k_dump_object.cpp
@@ -725,12 +725,142 @@ namespace ams::kern::KDumpObject {
     }
 
     void DumpCpuUtilization() {
-        /* TODO */
+        MESOSPHERE_RELEASE_LOG("Dump Cpu Utilization\n");
+
+        constexpr size_t MaxLogDepth = 64;
+        {
+            /* Create tracking arrays. */
+            KAutoObject *objects[MaxLogDepth];
+            u32 cpu_time[MaxLogDepth];
+
+            s64 start_tick;
+            size_t i, n;
+            KDpcManager::Sync();
+            {
+                /* Lock the process list. */
+                KProcess::ListAccessor accessor;
+
+                /* Begin tracking. */
+                start_tick = GetTickOrdered();
+
+                /* Iterate, finding processes. */
+                const auto end = accessor.end();
+                i = 0;
+                for (auto it = accessor.begin(); it != end; ++it) {
+                    KProcess *process = static_cast<KProcess *>(std::addressof(*it));
+                    if (AMS_LIKELY(i < MaxLogDepth)) {
+                        if (AMS_LIKELY(process->Open())) {
+                            cpu_time[i] = process->GetCpuTime();
+                            objects[i]  = process;
+                            ++i;
+                        }
+                    }
+                }
+
+                /* Keep track of how many processes we found. */
+                n = i;
+            }
+
+            /* Wait one second. */
+            const s64 timeout = KHardwareTimer::GetTick() + ams::svc::Tick(TimeSpan::FromSeconds(1));
+            GetCurrentThread().Sleep(timeout);
+            KDpcManager::Sync();
+
+            /* Update our metrics. */
+            for (i = 0; i < n; ++i) {
+                KProcess *process = static_cast<KProcess *>(objects[i]);
+                cpu_time[i] = process->GetCpuTime() - cpu_time[i];
+            }
+
+            /* End tracking. */
+            const s64 end_tick = GetTickOrdered();
+
+            /* Log process utilization. */
+            for (i = 0; i < n; ++i) {
+                KProcess *process = static_cast<KProcess *>(objects[i]);
+                const s64 t = static_cast<u64>(cpu_time[i]) * 1000 / (end_tick - start_tick);
+
+                MESOSPHERE_RELEASE_LOG("pid=%3lu %-11s %3lu.%lu%%\n", process->GetId(), process->GetName(), t / 10, t % 10);
+            }
+
+            /* Close all objects. */
+            for (i = 0; i < n; ++i) {
+                objects[i]->Close();
+            }
+        }
+
+        MESOSPHERE_RELEASE_LOG("\n");
     }
 
     void DumpCpuUtilization(u64 process_id) {
-        /* TODO */
-        MESOSPHERE_UNUSED(process_id);
+        MESOSPHERE_RELEASE_LOG("Dump Kernel Cpu Utilization\n");
+
+        constexpr size_t MaxLogDepth = 64;
+        {
+            /* Create tracking arrays. */
+            KAutoObject *objects[MaxLogDepth];
+            u32 cpu_time[MaxLogDepth];
+
+            s64 start_tick;
+            size_t i, n;
+            KDpcManager::Sync();
+            {
+                /* Lock the thread list. */
+                KThread::ListAccessor accessor;
+
+                /* Begin tracking. */
+                start_tick = GetTickOrdered();
+
+                /* Iterate, finding process threads. */
+                const auto end = accessor.end();
+                i = 0;
+                for (auto it = accessor.begin(); it != end; ++it) {
+                    KThread *thread = static_cast<KThread *>(std::addressof(*it));
+                    if (KProcess *process = thread->GetOwnerProcess(); process != nullptr && process->GetId() == process_id) {
+                        if (AMS_LIKELY(i < MaxLogDepth)) {
+                            if (AMS_LIKELY(thread->Open())) {
+                                cpu_time[i] = thread->GetCpuTime();
+                                objects[i]  = thread;
+                                ++i;
+                            }
+                        }
+                    }
+                }
+
+                /* Keep track of how many process threads we found. */
+                n = i;
+            }
+
+            /* Wait one second. */
+            const s64 timeout = KHardwareTimer::GetTick() + ams::svc::Tick(TimeSpan::FromSeconds(1));
+            GetCurrentThread().Sleep(timeout);
+            KDpcManager::Sync();
+
+            /* Update our metrics. */
+            for (i = 0; i < n; ++i) {
+                KThread *thread = static_cast<KThread *>(objects[i]);
+                cpu_time[i] = thread->GetCpuTime() - cpu_time[i];
+            }
+
+            /* End tracking. */
+            const s64 end_tick = GetTickOrdered();
+
+            /* Log thread utilization. */
+            for (i = 0; i < n; ++i) {
+                KThread *thread   = static_cast<KThread *>(objects[i]);
+                KProcess *process = thread->GetOwnerProcess();
+                const s64 t = static_cast<u64>(cpu_time[i]) * 1000 / (end_tick - start_tick);
+
+                MESOSPHERE_RELEASE_LOG("tid=%3lu pid=%3lu %-11s %3lu.%lu%% pri=%2d af=%lx\n", thread->GetId(), process->GetId(), process->GetName(), t / 10, t % 10, thread->GetPriority(), thread->GetAffinityMask().GetAffinityMask());
+            }
+
+            /* Close all objects. */
+            for (i = 0; i < n; ++i) {
+                objects[i]->Close();
+            }
+        }
+
+        MESOSPHERE_RELEASE_LOG("\n");
     }
 
     void DumpProcess(u64 process_id) {