pgl: begin skeletoning shell service, implement two commands.

2020-04-16 02:33:36 -07:00
parent b1aee64b81
commit 6ce038acad
11 changed files with 390 additions and 21 deletions
--- a/libraries/libstratosphere/source/pgl/srv/pgl_srv_shell.cpp
+++ b/libraries/libstratosphere/source/pgl/srv/pgl_srv_shell.cpp
@@ -0,0 +1,114 @@
+/*
+ * Copyright (c) 2018-2020 Atmosphère-NX
+ *
+ * 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 <stratosphere.hpp>
+#include "pgl_srv_shell.hpp"
+
+namespace ams::pgl::srv {
+
+    namespace {
+
+        constexpr inline size_t ProcessDataCount = 0x20;
+
+        struct ProcessData {
+            os::ProcessId process_id;
+            u32 flags;
+        };
+        static_assert(std::is_pod<ProcessData>::value);
+
+        enum ProcessDataFlag : u32 {
+            ProcessDataFlag_None                        = 0,
+            ProcessDataFlag_DetailedCrashReportAllowed  = (1 << 0),
+            ProcessDataFlag_DetailedCrashReportEnabled  = (1 << 1),
+            ProcessDataFlag_HasLogOption                = (1 << 2),
+            ProcessDataFlag_OutputAllLog                = (1 << 3),
+            ProcessDataFlag_EnableCrashReportScreenShot = (1 << 4),
+        };
+
+        bool g_is_production                  = true;
+        bool g_enable_crash_report_screenshot = true;
+        bool g_enable_jit_debug               = false;
+
+        os::SdkMutex g_process_data_mutex;
+        ProcessData g_process_data[ProcessDataCount];
+
+        ProcessData *FindProcessData(os::ProcessId process_id) {
+            for (auto &data : g_process_data) {
+                if (data.process_id == process_id) {
+                    return std::addressof(data);
+                }
+            }
+            return nullptr;
+        }
+
+        u32 ConvertToProcessDataFlags(u8 pgl_flags) {
+            if ((pgl_flags & pgl::LaunchFlags_EnableDetailedCrashReport) == 0) {
+                /* If we shouldn't generate detailed crash reports, set no flags. */
+                return ProcessDataFlag_None;
+            } else {
+                /* We can and should generate detailed crash reports. */
+                u32 data_flags = ProcessDataFlag_DetailedCrashReportAllowed | ProcessDataFlag_DetailedCrashReportEnabled;
+
+                /* If we should enable crash report screenshots, check the correct flag. */
+                if (g_enable_crash_report_screenshot) {
+                    const u32 test_flag = g_is_production ? pgl::LaunchFlags_EnableCrashReportScreenShotForProduction : pgl::LaunchFlags_EnableCrashReportScreenShotForDevelop;
+                    if ((pgl_flags & test_flag) != 0) {
+                            data_flags |= ProcessDataFlag_EnableCrashReportScreenShot;
+                    }
+                }
+
+                return data_flags;
+            }
+        }
+
+    }
+
+    void InitializeProcessControlTask() {
+        /* TODO */
+    }
+
+    Result LaunchProgram(os::ProcessId *out, const ncm::ProgramLocation &loc, u32 pm_flags, u8 pgl_flags) {
+        /* Convert the input flags to the internal format. */
+        const u32 data_flags = ConvertToProcessDataFlags(pgl_flags);
+
+        /* If jit debug is enabled, we want to be signaled on crash. */
+        if (g_enable_jit_debug) {
+            pm_flags |= pm::LaunchFlags_SignalOnException;
+        }
+
+        /* Launch the process. */
+        os::ProcessId process_id;
+        R_TRY(pm::shell::LaunchProgram(std::addressof(process_id), loc, pm_flags & pm::LaunchFlagsMask));
+
+        /* Create a ProcessData for the process. */
+        {
+            std::scoped_lock lk(g_process_data_mutex);
+            ProcessData *new_data = FindProcessData(os::InvalidProcessId);
+            AMS_ABORT_UNLESS(new_data != nullptr);
+
+            new_data->process_id = process_id;
+            new_data->flags      = data_flags;
+        }
+
+        /* We succeeded. */
+        *out = process_id;
+        return ResultSuccess();
+    }
+
+    Result TerminateProcess(os::ProcessId process_id) {
+        return pm::shell::TerminateProcess(process_id);
+    }
+
+}