fs.mitm: Intercept qlaunch sysver request

stratosphere/fs_mitm/source/setsys_mitm_service.cpp

@@ -0,0 +1,104 @@
+/*
+ * Copyright (c) 2018 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 <mutex>
+#include <switch.h>
+#include "setsys_mitm_service.hpp"
+
+#include "mitm_query_service.hpp"
+#include "debug.hpp"
+
+static HosMutex g_version_mutex;
+static bool g_got_version = false;
+static SetSysFirmwareVersion g_fw_version = {0};
+
+static Result _GetFirmwareVersion(SetSysFirmwareVersion *out) {
+    std::lock_guard<HosMutex> lock(g_version_mutex);
+    if (!g_got_version) {
+        Result rc = setsysGetFirmwareVersion(&g_fw_version);
+        if (R_FAILED(rc)) {
+            return rc;
+        }
+        
+        /* Modify the output firmware version. */
+        {
+            u32 major, minor, micro;
+            char display_version[sizeof(g_fw_version.display_version)] = {0};
+            
+            GetAtmosphereApiVersion(&major, &minor, &micro, nullptr, nullptr);
+            snprintf(display_version, sizeof(display_version), "%s (AMS %u.%u.%u)", g_fw_version.display_version, major, minor, micro);
+            
+            memcpy(g_fw_version.display_version, display_version, sizeof(g_fw_version.display_version));
+        }
+        
+        g_got_version = true;
+    }
+    
+    *out = g_fw_version;
+    return 0;
+}
+
+Result SetSysMitMService::dispatch(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size) {
+    Result rc = 0xF601;
+
+    switch (static_cast<SetSysCmd>(cmd_id)) {
+        case SetSysCmd::GetFirmwareVersion:
+            rc = WrapIpcCommandImpl<&SetSysMitMService::get_firmware_version>(this, r, out_c, pointer_buffer, pointer_buffer_size);
+            break;
+        case SetSysCmd::GetFirmwareVersion2:
+            if (kernelAbove300()) {
+                rc = WrapIpcCommandImpl<&SetSysMitMService::get_firmware_version2>(this, r, out_c, pointer_buffer, pointer_buffer_size);
+            }
+            break;
+        default:
+            break;
+    }
+    
+    return rc;
+}
+
+void SetSysMitMService::postprocess(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size) {
+    /* No commands need postprocessing. */    
+}
+
+Result SetSysMitMService::handle_deferred() {
+    /* This service is never deferrable. */
+    return 0;
+}
+
+std::tuple<Result> SetSysMitMService::get_firmware_version(OutPointerWithServerSize<SetSysFirmwareVersion, 0x1> out) {
+    if (out.num_elements != 1) {
+        return {0xF601};
+    }
+    
+    Result rc = _GetFirmwareVersion(out.pointer);
+    
+    /* GetFirmwareVersion sanitizes these fields. */
+    out.pointer->revision_major = 0;
+    out.pointer->revision_minor = 0;
+        
+    return {rc};
+}
+
+std::tuple<Result> SetSysMitMService::get_firmware_version2(OutPointerWithServerSize<SetSysFirmwareVersion, 0x1> out) {
+    if (out.num_elements != 1) {
+        return {0xF601};
+    }
+    
+    Result rc = _GetFirmwareVersion(out.pointer);
+        
+    return {rc};
+}