Atmosphere/stratosphere/ams_mitm/source/fs_mitm/fsmitm_save_utils.cpp

114 lines
4.3 KiB
C++

/*
* 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 "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();
}
}