dmnt: Add TargetIO_Open/Close/Read/Write funcs.

stratosphere/dmnt/source/dmnt_main.cpp

@@ -23,7 +23,7 @@
 #include <atmosphere.h>
 #include <stratosphere.hpp>
 
-#include "dmnt_debug_monitor.hpp"
+#include "dmnt_service.hpp"
 
 extern "C" {
     extern u32 __start__;

stratosphere/dmnt/source/dmnt_service.hpp

@@ -81,6 +81,11 @@ class DebugMonitorService final : public IServiceObject {
         Result GetProcessId(Out<u64> out_pid, Handle hnd);
         Result GetProcessHandle(Out<Handle> out_hnd, u64 pid);
         Result WaitSynchronization(Handle hnd, u64 ns);
+        
+        Result TargetIO_FileOpen(OutBuffer<u64> out_hnd, InBuffer<char> path, int open_mode, u32 create_mode);
+        Result TargetIO_FileClose(InBuffer<u64> hnd);
+        Result TargetIO_FileRead(InBuffer<u64> hnd, OutBuffer<u8> out_data, Out<u32> out_read, u64 offset);
+        Result TargetIO_FileWrite(InBuffer<u64> hnd, InBuffer<u8> data, Out<u32> out_written, u64 offset);
     public:
         DEFINE_SERVICE_DISPATCH_TABLE {
             MakeServiceCommandMeta<DebugMonitor_Cmd_BreakDebugProcess, &DebugMonitorService::BreakDebugProcess>(),
@@ -110,10 +115,10 @@ class DebugMonitorService final : public IServiceObject {
             // MakeServiceCommandMeta<DebugMonitor_Cmd_GetJitDebugProcessList, &DebugMonitorService::GetJitDebugProcessList>(),
             // MakeServiceCommandMeta<DebugMonitor_Cmd_CreateCoreDump, &DebugMonitorService::CreateCoreDump>(),
             // MakeServiceCommandMeta<DebugMonitor_Cmd_GetAllDebugThreadInfo, &DebugMonitorService::GetAllDebugThreadInfo>(),
-            // MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileOpen, &DebugMonitorService::TargetIO_FileOpen>(),
+            MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileOpen, &DebugMonitorService::TargetIO_FileOpen>(),
-            // MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileClose, &DebugMonitorService::TargetIO_FileClose>(),
+            MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileClose, &DebugMonitorService::TargetIO_FileClose>(),
-            // MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileRead, &DebugMonitorService::TargetIO_FileRead>(),
+            MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileRead, &DebugMonitorService::TargetIO_FileRead>(),
-            // MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileWrite, &DebugMonitorService::TargetIO_FileWrite>(),
+            MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileWrite, &DebugMonitorService::TargetIO_FileWrite>(),
             // MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileSetAttributes, &DebugMonitorService::TargetIO_FileSetAttributes>(),
             // MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileGetInformation, &DebugMonitorService::TargetIO_FileGetInformation>(),
             // MakeServiceCommandMeta<DebugMonitor_Cmd_TargetIO_FileSetTime, &DebugMonitorService::TargetIO_FileSetTime>(),

stratosphere/dmnt/source/dmnt_service_debug.cpp

@@ -15,7 +15,7 @@
  */
  
 #include <switch.h>
-#include "dmnt_debug_monitor.hpp"
+#include "dmnt_service.hpp"
 
 Result DebugMonitorService::BreakDebugProcess(Handle debug_hnd) {
     /* Nintendo discards the output of this command, but we will return it. */

stratosphere/dmnt/source/dmnt_service_target_io.cpp

@@ -0,0 +1,195 @@
+/*
+ * 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 <unordered_map>
+#include <switch.h>
+#include "dmnt_service.hpp"
+
+enum TIOCreateOption : u32 {
+    TIOCreateOption_CreateNew = 1,
+    TIOCreateOption_CreateAlways = 2,
+    TIOCreateOption_OpenExisting = 3,
+    TIOCreateOption_OpenAlways = 4,
+    TIOCreateOption_ResetSize = 5,
+};
+
+/* Nintendo uses actual pointers as file handles. We'll add a layer of indirection... */
+static bool g_sd_initialized = false;
+static HosMutex g_sd_lock;
+static FsFileSystem g_sd_fs;
+
+static HosMutex g_file_handle_lock;
+static u64 g_cur_fd = 0;
+static std::unordered_map<u64, FsFile> g_file_handles;
+
+static Result EnsureSdInitialized() {
+    std::scoped_lock<HosMutex> lk(g_sd_lock);
+    if (g_sd_initialized) {
+        return 0;
+    }
+    
+    Result rc = fsMountSdcard(&g_sd_fs);
+    if (R_SUCCEEDED(rc)) {
+        g_sd_initialized = true;
+    }
+    return rc;
+}
+
+static u64 GetFileHandle(FsFile f) {
+    std::scoped_lock<HosMutex> lk(g_file_handle_lock);
+    
+    u64 fd = g_cur_fd++;
+    g_file_handles[fd] = f;
+    return fd;
+}
+
+static Result GetFileByHandle(FsFile *out, u64 handle) {
+    std::scoped_lock<HosMutex> lk(g_file_handle_lock);
+    if (g_file_handles.find(handle) != g_file_handles.end()) {
+        *out = g_file_handles[handle];
+        return 0;
+    }
+    return 0x2EE202;
+}
+
+static Result CloseFileByHandle(u64 handle) {
+    std::scoped_lock<HosMutex> lk(g_file_handle_lock);
+    if (g_file_handles.find(handle) != g_file_handles.end()) {
+        fsFileClose(&g_file_handles[handle]);
+        g_file_handles.erase(handle);
+        return 0;
+    }
+    return 0x2EE202;
+}
+
+static void FixPath(char *dst, size_t dst_size, InBuffer<char> &path) {
+    dst[dst_size - 1] = 0;
+    strncpy(dst, "/", dst_size - 1);
+    
+    const char *src = path.buffer;
+    size_t src_idx = 0;
+    size_t dst_idx = 1;
+    while (src_idx < path.num_elements && (src[src_idx] == '/' || src[src_idx] == '\\')) {
+        src_idx++;
+    }
+    
+    while (src_idx < path.num_elements && dst_idx < dst_size - 1 && src[src_idx] != 0) {
+        if (src[src_idx] == '\\') {
+            dst[dst_idx] = '/';
+        } else {
+            dst[dst_idx] = src[src_idx];
+        }
+        
+        src_idx++;
+        dst_idx++;
+    }
+    
+    if (dst_idx < dst_size) {
+        dst[dst_idx] = 0;
+    }
+}
+
+Result DebugMonitorService::TargetIO_FileOpen(OutBuffer<u64> out_hnd, InBuffer<char> path, int open_mode, u32 create_mode) {
+    if (out_hnd.num_elements != 1) {
+        return 0xF601;
+    }
+    
+    Result rc = EnsureSdInitialized();
+    if (R_FAILED(rc)) {
+        return rc;
+    }
+    
+    char fs_path[FS_MAX_PATH];
+    
+    FixPath(fs_path, sizeof(fs_path), path);
+    
+    if (create_mode == TIOCreateOption_CreateAlways) {
+        fsFsDeleteFile(&g_sd_fs, fs_path);
+        rc = fsFsCreateFile(&g_sd_fs, fs_path, 0, 0);
+    } else if (create_mode == TIOCreateOption_CreateNew) {
+        rc = fsFsCreateFile(&g_sd_fs, fs_path, 0, 0);
+    }
+    
+    if (R_FAILED(rc)) {
+        return rc;
+    }
+    
+    FsFile f;
+    rc = fsFsOpenFile(&g_sd_fs, fs_path, open_mode, &f);
+    if (R_FAILED(rc)) {
+        if (create_mode == TIOCreateOption_OpenAlways) {
+            fsFsCreateFile(&g_sd_fs, fs_path, 0, 0);
+            rc = fsFsOpenFile(&g_sd_fs, fs_path, open_mode, &f);
+        }
+    }
+    
+    if (R_SUCCEEDED(rc)) {
+        if (create_mode == TIOCreateOption_ResetSize) {
+            rc = fsFileSetSize(&f, 0);
+        }
+        if (R_SUCCEEDED(rc)) {
+            out_hnd[0] = GetFileHandle(f);
+        } else {
+            fsFileClose(&f);
+        }
+    }
+    
+    return rc;
+}
+
+Result DebugMonitorService::TargetIO_FileClose(InBuffer<u64> hnd) {
+    if (hnd.num_elements != 1) {
+        return 0xF601;
+    }
+    
+    return CloseFileByHandle(hnd[0]);
+}
+
+Result DebugMonitorService::TargetIO_FileRead(InBuffer<u64> hnd, OutBuffer<u8> out_data, Out<u32> out_read, u64 offset) {
+    if (hnd.num_elements != 1) {
+        return 0xF601;
+    }
+    
+    FsFile f;
+    Result rc = GetFileByHandle(&f, hnd[0]);
+    if (R_FAILED(rc)) {
+        return rc;
+    }
+    
+    size_t read = 0;
+    rc = fsFileRead(&f, offset, out_data.buffer, out_data.num_elements, &read);
+    out_read.SetValue(static_cast<u32>(read));
+    return rc;
+}
+
+Result DebugMonitorService::TargetIO_FileWrite(InBuffer<u64> hnd, InBuffer<u8> data, Out<u32> out_written, u64 offset) {
+    if (hnd.num_elements != 1) {
+        return 0xF601;
+    }
+    
+    FsFile f;
+    Result rc = GetFileByHandle(&f, hnd[0]);
+    if (R_FAILED(rc)) {
+        return rc;
+    }
+    
+    rc = fsFileWrite(&f, offset, data.buffer, data.num_elements);
+    if (R_SUCCEEDED(rc)) {
+        out_written.SetValue(data.num_elements);
+    }
+    
+    return rc;
+}