mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-11-14 17:16:36 +00:00
130 lines
5 KiB
C++
130 lines
5 KiB
C++
/*
|
|
* Copyright (c) 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>
|
|
#include "set_mitm_service.hpp"
|
|
#include "set_shim.h"
|
|
|
|
namespace ams::mitm::settings {
|
|
|
|
using namespace ams::settings;
|
|
|
|
namespace {
|
|
|
|
constinit os::ProcessId g_application_process_id = os::InvalidProcessId;
|
|
constinit cfg::OverrideLocale g_application_locale;
|
|
constinit bool g_valid_language;
|
|
constinit bool g_valid_region;
|
|
}
|
|
|
|
SetMitmService::SetMitmService(std::shared_ptr<::Service> &&s, const sm::MitmProcessInfo &c) : sf::MitmServiceImplBase(std::forward<std::shared_ptr<::Service>>(s), c) {
|
|
if (m_client_info.program_id == ncm::SystemProgramId::Ns) {
|
|
os::ProcessId application_process_id;
|
|
if (R_SUCCEEDED(pm::dmnt::GetApplicationProcessId(std::addressof(application_process_id))) && g_application_process_id == application_process_id) {
|
|
m_locale = g_application_locale;
|
|
m_is_valid_language = g_valid_language;
|
|
m_is_valid_region = g_valid_region;
|
|
m_got_locale = true;
|
|
} else {
|
|
this->InvalidateLocale();
|
|
}
|
|
} else {
|
|
this->InvalidateLocale();
|
|
}
|
|
}
|
|
|
|
Result SetMitmService::EnsureLocale() {
|
|
/* Optimization: if locale has already been gotten, we can stop. */
|
|
if (AMS_LIKELY(m_got_locale)) {
|
|
return ResultSuccess();
|
|
}
|
|
|
|
std::scoped_lock lk(m_lock);
|
|
|
|
const bool is_ns = m_client_info.program_id == ncm::SystemProgramId::Ns;
|
|
|
|
if (!m_got_locale) {
|
|
ncm::ProgramId program_id = m_client_info.program_id;
|
|
os::ProcessId application_process_id = os::InvalidProcessId;
|
|
|
|
if (is_ns) {
|
|
/* When NS asks for a locale, refresh to get the current application locale. */
|
|
R_TRY(pm::dmnt::GetApplicationProcessId(std::addressof(application_process_id)));
|
|
R_TRY(pm::info::GetProgramId(std::addressof(program_id), application_process_id));
|
|
}
|
|
m_locale = cfg::GetOverrideLocale(program_id);
|
|
m_is_valid_language = settings::IsValidLanguageCode(m_locale.language_code);
|
|
m_is_valid_region = settings::IsValidRegionCode(m_locale.region_code);
|
|
m_got_locale = true;
|
|
|
|
if (is_ns) {
|
|
g_application_locale = m_locale;
|
|
g_valid_language = m_is_valid_language;
|
|
g_valid_region = m_is_valid_region;
|
|
g_application_process_id = application_process_id;
|
|
}
|
|
}
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
void SetMitmService::InvalidateLocale() {
|
|
std::scoped_lock lk(m_lock);
|
|
|
|
std::memset(std::addressof(m_locale), 0xCC, sizeof(m_locale));
|
|
m_is_valid_language = false;
|
|
m_is_valid_region = false;
|
|
m_got_locale = false;
|
|
}
|
|
|
|
Result SetMitmService::GetLanguageCode(sf::Out<settings::LanguageCode> out) {
|
|
this->EnsureLocale();
|
|
|
|
/* If there's no override locale, just use the actual one. */
|
|
if (AMS_UNLIKELY(!m_is_valid_language)) {
|
|
static_assert(sizeof(u64) == sizeof(settings::LanguageCode));
|
|
R_TRY(setGetLanguageCodeFwd(m_forward_service.get(), reinterpret_cast<u64 *>(std::addressof(m_locale.language_code))));
|
|
|
|
m_is_valid_language = true;
|
|
if (m_client_info.program_id == ncm::SystemProgramId::Ns) {
|
|
g_application_locale.language_code = m_locale.language_code;
|
|
g_valid_language = true;
|
|
}
|
|
}
|
|
|
|
out.SetValue(m_locale.language_code);
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result SetMitmService::GetRegionCode(sf::Out<settings::RegionCode> out) {
|
|
this->EnsureLocale();
|
|
|
|
/* If there's no override locale, just use the actual one. */
|
|
if (AMS_UNLIKELY(!m_is_valid_region)) {
|
|
static_assert(sizeof(::SetRegion) == sizeof(settings::RegionCode));
|
|
R_TRY(setGetRegionCodeFwd(m_forward_service.get(), reinterpret_cast<::SetRegion *>(std::addressof(m_locale.region_code))));
|
|
|
|
m_is_valid_region = true;
|
|
if (m_client_info.program_id == ncm::SystemProgramId::Ns) {
|
|
g_application_locale.region_code = m_locale.region_code;
|
|
g_valid_region = true;
|
|
}
|
|
}
|
|
|
|
out.SetValue(m_locale.region_code);
|
|
return ResultSuccess();
|
|
}
|
|
|
|
}
|