mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-11-16 01:49:32 +00:00
106 lines
3.8 KiB
C++
106 lines
3.8 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 <stratosphere.hpp>
|
|
|
|
namespace ams::cfg {
|
|
|
|
namespace {
|
|
|
|
std::atomic<u32> g_flag_mount_count;
|
|
|
|
/* Helper. */
|
|
void GetFlagMountName(char *dst) {
|
|
util::SNPrintf(dst, fs::MountNameLengthMax + 1, "#flag%08x", g_flag_mount_count.fetch_add(1));
|
|
}
|
|
|
|
bool HasFlagFile(const char *flag_path) {
|
|
/* All flags are not present until the SD card is. */
|
|
if (!IsSdCardInitialized()) {
|
|
return false;
|
|
}
|
|
|
|
/* Mount the SD card. */
|
|
char mount_name[fs::MountNameLengthMax + 1];
|
|
GetFlagMountName(mount_name);
|
|
if (R_FAILED(fs::MountSdCard(mount_name))) {
|
|
return false;
|
|
}
|
|
ON_SCOPE_EXIT { fs::Unmount(mount_name); };
|
|
|
|
/* Check if the entry exists. */
|
|
char full_path[fs::EntryNameLengthMax + 1];
|
|
util::SNPrintf(full_path, sizeof(full_path), "%s:/%s", mount_name, flag_path[0] == '/' ? flag_path + 1 : flag_path);
|
|
|
|
bool has_file;
|
|
if (R_FAILED(fs::HasFile(std::addressof(has_file), full_path))) {
|
|
return false;
|
|
}
|
|
|
|
return has_file;
|
|
}
|
|
|
|
Result DeleteFlagFile(const char *flag_path) {
|
|
/* We need the SD card to be available to delete anything. */
|
|
AMS_ABORT_UNLESS(IsSdCardInitialized());
|
|
|
|
/* Mount the sd card. */
|
|
char mount_name[fs::MountNameLengthMax + 1];
|
|
GetFlagMountName(mount_name);
|
|
R_TRY(fs::MountSdCard(mount_name));
|
|
ON_SCOPE_EXIT { fs::Unmount(mount_name); };
|
|
|
|
/* Get the flag path. */
|
|
char full_path[fs::EntryNameLengthMax + 1];
|
|
util::SNPrintf(full_path, sizeof(full_path), "%s:/%s", mount_name, flag_path[0] == '/' ? flag_path + 1 : flag_path);
|
|
|
|
/* Delete the file. */
|
|
R_TRY(fs::DeleteFile(full_path));
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
}
|
|
|
|
/* Flag utilities. */
|
|
bool HasFlag(const sm::MitmProcessInfo &process_info, const char *flag) {
|
|
return HasContentSpecificFlag(process_info.program_id, flag) || (process_info.override_status.IsHbl() && HasHblFlag(flag));
|
|
}
|
|
|
|
bool HasContentSpecificFlag(ncm::ProgramId program_id, const char *flag) {
|
|
char content_flag[fs::EntryNameLengthMax + 1];
|
|
util::SNPrintf(content_flag, sizeof(content_flag) - 1, "/atmosphere/contents/%016lx/flags/%s.flag", static_cast<u64>(program_id), flag);
|
|
return HasFlagFile(content_flag);
|
|
}
|
|
|
|
bool HasGlobalFlag(const char *flag) {
|
|
char global_flag[fs::EntryNameLengthMax + 1];
|
|
util::SNPrintf(global_flag, sizeof(global_flag) - 1, "/atmosphere/flags/%s.flag", flag);
|
|
return HasFlagFile(global_flag);
|
|
}
|
|
|
|
bool HasHblFlag(const char *flag) {
|
|
char hbl_flag[0x100];
|
|
util::SNPrintf(hbl_flag, sizeof(hbl_flag) - 1, "hbl_%s", flag);
|
|
return HasGlobalFlag(hbl_flag);
|
|
}
|
|
|
|
Result DeleteGlobalFlag(const char *flag) {
|
|
char global_flag[fs::EntryNameLengthMax + 1];
|
|
util::SNPrintf(global_flag, sizeof(global_flag) - 1, "/atmosphere/flags/%s.flag", flag);
|
|
return DeleteFlagFile(global_flag);
|
|
}
|
|
|
|
}
|