ams_mitm: Implement savedata redirection

2019-12-06 23:19:48 -08:00
parent 90367aea0d
commit 7835486a4d
23 changed files with 899 additions and 47 deletions
--- a/stratosphere/ams_mitm/source/fs_mitm/fsmitm_save_utils.cpp
+++ b/stratosphere/ams_mitm/source/fs_mitm/fsmitm_save_utils.cpp
@@ -0,0 +1,114 @@
+/*
+ * Copyright (c) 2018-2019 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 "fsmitm_save_utils.hpp"
+
+namespace ams::mitm::fs {
+
+    using namespace ams::fs;
+
+    namespace {
+
+        Result GetSaveDataSpaceIdString(const char **out_str, u8 space_id) {
+            switch (space_id) {
+                case FsSaveDataSpaceId_System:
+                case FsSaveDataSpaceId_ProperSystem:
+                    *out_str = "sys";
+                    break;
+                case FsSaveDataSpaceId_User:
+                    *out_str = "user";
+                    break;
+                case FsSaveDataSpaceId_SdSystem:
+                    *out_str = "sd_sys";
+                    break;
+                case FsSaveDataSpaceId_Temporary:
+                    *out_str = "temp";
+                    break;
+                case FsSaveDataSpaceId_SdUser:
+                    *out_str = "sd_user";
+                    break;
+                case FsSaveDataSpaceId_SafeMode:
+                    *out_str = "safe";
+                    break;
+                default:
+                    return fs::ResultInvalidSaveDataSpaceId();
+            }
+
+            return ResultSuccess();
+        }
+
+        Result GetSaveDataTypeString(const char **out_str, u8 save_data_type) {
+            switch (save_data_type) {
+                case FsSaveDataType_System:
+                    *out_str = "system";
+                    break;
+                case FsSaveDataType_Account:
+                    *out_str = "account";
+                    break;
+                case FsSaveDataType_Bcat:
+                    *out_str = "bcat";
+                    break;
+                case FsSaveDataType_Device:
+                    *out_str = "device";
+                    break;
+                case FsSaveDataType_Temporary:
+                    *out_str = "temp";
+                    break;
+                case FsSaveDataType_Cache:
+                    *out_str = "cache";
+                    break;
+                case FsSaveDataType_SystemBcat:
+                    *out_str = "system_bcat";
+                    break;
+                default:
+                    /* TODO: Better result? */
+                    return fs::ResultInvalidArgument();
+            }
+
+            return ResultSuccess();
+        }
+
+        constexpr inline bool IsEmptyAccountId(const AccountUid &uid) {
+            constexpr AccountUid empty_uid = {};
+            return std::memcmp(&uid, &empty_uid, sizeof(uid)) == 0;
+        }
+
+
+    }
+
+    Result SaveUtil::GetDirectorySaveDataPath(char *dst, size_t dst_size, ncm::ProgramId program_id, u8 space_id, const FsSaveDataAttribute &attribute) {
+        /* Saves should be separate for emunand vs sysnand. */
+        const char *emummc_str = emummc::IsActive() ? "emummc" : "sysmmc";
+
+        /* Get space_id, save_data_type strings. */
+        const char *space_id_str, *save_type_str;
+        R_TRY(GetSaveDataSpaceIdString(&space_id_str, space_id));
+        R_TRY(GetSaveDataTypeString(&save_type_str, attribute.save_data_type));
+
+        /* Initialize the path. */
+        const bool is_system = attribute.system_save_data_id != 0 && IsEmptyAccountId(attribute.uid);
+        size_t out_path_len;
+        if (is_system) {
+            out_path_len = static_cast<size_t>(std::snprintf(dst, dst_size, "/atmosphere/saves/%s/%s/%s/%016lx", emummc_str, space_id_str, save_type_str, attribute.system_save_data_id));
+        } else {
+            out_path_len = static_cast<size_t>(std::snprintf(dst, dst_size, "/atmosphere/saves/%s/%s/%s/%016lx/%016lx%016lx", emummc_str, space_id_str, save_type_str, static_cast<u64>(program_id), attribute.uid.uid[1], attribute.uid.uid[0]));
+        }
+
+        R_UNLESS(out_path_len < dst_size, fs::ResultTooLongPath());
+
+        return ResultSuccess();
+    }
+
+}