- oc_loader: major refactoring

Source/Atmosphere/stratosphere/loader/source/oc/ptm/ptm.cpp

@@ -0,0 +1,136 @@
+/*
+ * Copyright (C) Switch-OC-Suite
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms and conditions of the GNU General Public License,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "ptm.hpp"
+
+namespace ams::ldr::oc::ptm {
+
+Result CpuPtmBoost(perf_conf_entry* entry) {
+    if (!C.marikoCpuBoostClock)
+        R_SUCCEED();
+
+    u32 cpuPtmBoostNew = C.marikoCpuBoostClock * 1000;
+
+    PatchOffset(&(entry->cpu_freq_1), cpuPtmBoostNew);
+    PatchOffset(&(entry->cpu_freq_2), cpuPtmBoostNew);
+
+    R_SUCCEED();
+}
+
+Result MemPtmMax(perf_conf_entry* entry) {
+    if (!C.marikoEmcMaxClock)
+        R_SUCCEED();
+
+    u32 memPtmMax = C.marikoEmcMaxClock * 1000;
+
+    PatchOffset(&(entry->emc_freq_1), memPtmMax);
+    PatchOffset(&(entry->emc_freq_2), memPtmMax);
+
+    R_SUCCEED();
+}
+
+Result MemPtmAlt(perf_conf_entry* entry) {
+    PatchOffset(&(entry->emc_freq_1), memPtmLimit);
+    PatchOffset(&(entry->emc_freq_2), memPtmLimit);
+
+    R_SUCCEED();
+}
+
+bool PtmEntryIsValid(perf_conf_entry* entry) {
+    return (entry->cpu_freq_1 == entry->cpu_freq_2 &&
+            entry->gpu_freq_1 == entry->gpu_freq_2 &&
+            entry->emc_freq_1 == entry->emc_freq_2);
+}
+
+bool PtmTablePatternFn(u32* ptr) {
+    perf_conf_entry* entry = reinterpret_cast<perf_conf_entry *>(ptr);
+    if (!PtmEntryIsValid(entry))
+        return false;
+
+    return entry->cpu_freq_1 == cpuPtmDefault;
+}
+
+void Patch(uintptr_t mapped_nso, size_t nso_size) {
+    #ifdef ATMOSPHERE_IS_STRATOSPHERE
+    bool isMariko = (spl::GetSocType() == spl::SocType_Mariko);
+    if (!isMariko)
+        return;
+    #endif
+
+    perf_conf_entry* confTable = nullptr;
+    for (uintptr_t ptr = mapped_nso;
+         ptr <= mapped_nso + nso_size - sizeof(perf_conf_entry) * entryCnt;
+         ptr += sizeof(u32))
+    {
+        u32* ptr32 = reinterpret_cast<u32 *>(ptr);
+        if (PtmTablePatternFn(ptr32)) {
+            confTable = reinterpret_cast<perf_conf_entry *>(ptr);
+            break;
+        }
+    }
+
+    if (!confTable) {
+        CRASH("confTable not found!");
+    }
+
+    PatcherEntry<perf_conf_entry> patches[] = {
+        { "CPU Ptm Boost",  &CpuPtmBoost,   2, },
+        { "MEM Ptm Max",    &MemPtmMax,     9, },
+        { "MEM Ptm Alt",    &MemPtmAlt,     7, },
+    };
+
+    for (u32 i = 0; i < entryCnt; i++) {
+        perf_conf_entry* entry = confTable + i;
+
+        if (!PtmEntryIsValid(entry)) {
+            LOGGING("@%p", &entry);
+            CRASH("Invalid ptm confTable entry");
+        }
+
+        switch (entry->cpu_freq_1) {
+            case cpuPtmBoost:
+                patches[0].Apply(entry);
+                break;
+            case cpuPtmDefault:
+            case cpuPtmDevOC:
+                break;
+            default:
+                LOGGING("%u (0x%08x) @%p", entry->cpu_freq_1, entry->conf_id, &(entry->cpu_freq_1));
+                CRASH("Unknown CPU Freq");
+        }
+
+        switch (entry->emc_freq_1) {
+            case memPtmLimit:
+                patches[1].Apply(entry);
+                break;
+            case memPtmAlt:
+            case memPtmClamp:
+                patches[2].Apply(entry);
+                break;
+            default:
+                LOGGING("%u (0x%08x) @%p", entry->emc_freq_1, entry->conf_id, &(entry->emc_freq_2));
+                CRASH("Unknown MEM Freq");
+        }
+    }
+
+    for (auto& patch : patches) {
+        LOGGING("%s Count: %zu", patch.description, patch.patched_count);
+        if (R_FAILED(patch.CheckResult()))
+            CRASH(patch.description);
+    }
+}
+
+}