mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-12-05 04:28:04 +01:00
151 lines
6.7 KiB
C++
151 lines
6.7 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 <mesosphere.hpp>
|
|
|
|
namespace ams::kern::svc {
|
|
|
|
/* ============================= Common ============================= */
|
|
|
|
namespace {
|
|
|
|
constexpr bool IsValidLimitableResource(ams::svc::LimitableResource which) {
|
|
return which < ams::svc::LimitableResource_Count;
|
|
}
|
|
|
|
Result GetResourceLimitLimitValue(int64_t *out_limit_value, ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which) {
|
|
/* Validate the resource. */
|
|
R_UNLESS(IsValidLimitableResource(which), svc::ResultInvalidEnumValue());
|
|
|
|
/* Get the resource limit. */
|
|
KScopedAutoObject resource_limit = GetCurrentProcess().GetHandleTable().GetObject<KResourceLimit>(resource_limit_handle);
|
|
R_UNLESS(resource_limit.IsNotNull(), svc::ResultInvalidHandle());
|
|
|
|
/* Get the limit value. */
|
|
*out_limit_value = resource_limit->GetLimitValue(which);
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result GetResourceLimitCurrentValue(int64_t *out_current_value, ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which) {
|
|
/* Validate the resource. */
|
|
R_UNLESS(IsValidLimitableResource(which), svc::ResultInvalidEnumValue());
|
|
|
|
/* Get the resource limit. */
|
|
KScopedAutoObject resource_limit = GetCurrentProcess().GetHandleTable().GetObject<KResourceLimit>(resource_limit_handle);
|
|
R_UNLESS(resource_limit.IsNotNull(), svc::ResultInvalidHandle());
|
|
|
|
/* Get the current value. */
|
|
*out_current_value = resource_limit->GetCurrentValue(which);
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result GetResourceLimitPeakValue(int64_t *out_peak_value, ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which) {
|
|
/* Validate the resource. */
|
|
R_UNLESS(IsValidLimitableResource(which), svc::ResultInvalidEnumValue());
|
|
|
|
/* Get the resource limit. */
|
|
KScopedAutoObject resource_limit = GetCurrentProcess().GetHandleTable().GetObject<KResourceLimit>(resource_limit_handle);
|
|
R_UNLESS(resource_limit.IsNotNull(), svc::ResultInvalidHandle());
|
|
|
|
/* Get the peak value. */
|
|
*out_peak_value = resource_limit->GetPeakValue(which);
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result CreateResourceLimit(ams::svc::Handle *out_handle) {
|
|
/* Create a new resource limit. */
|
|
KResourceLimit *resource_limit = KResourceLimit::Create();
|
|
R_UNLESS(resource_limit != nullptr, svc::ResultOutOfResource());
|
|
|
|
/* Ensure we don't leak a reference to the limit. */
|
|
ON_SCOPE_EXIT { resource_limit->Close(); };
|
|
|
|
/* Initialize the resource limit. */
|
|
resource_limit->Initialize();
|
|
|
|
/* Register the limit. */
|
|
KResourceLimit::Register(resource_limit);
|
|
|
|
/* Add the limit to the handle table. */
|
|
R_TRY(GetCurrentProcess().GetHandleTable().Add(out_handle, resource_limit));
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result SetResourceLimitLimitValue(ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which, int64_t limit_value) {
|
|
/* Validate the resource. */
|
|
R_UNLESS(IsValidLimitableResource(which), svc::ResultInvalidEnumValue());
|
|
|
|
/* Get the resource limit. */
|
|
KScopedAutoObject resource_limit = GetCurrentProcess().GetHandleTable().GetObject<KResourceLimit>(resource_limit_handle);
|
|
R_UNLESS(resource_limit.IsNotNull(), svc::ResultInvalidHandle());
|
|
|
|
/* Set the limit value. */
|
|
R_TRY(resource_limit->SetLimitValue(which, limit_value));
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
}
|
|
|
|
/* ============================= 64 ABI ============================= */
|
|
|
|
Result GetResourceLimitLimitValue64(int64_t *out_limit_value, ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which) {
|
|
return GetResourceLimitLimitValue(out_limit_value, resource_limit_handle, which);
|
|
}
|
|
|
|
Result GetResourceLimitCurrentValue64(int64_t *out_current_value, ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which) {
|
|
return GetResourceLimitCurrentValue(out_current_value, resource_limit_handle, which);
|
|
}
|
|
|
|
Result GetResourceLimitPeakValue64(int64_t *out_peak_value, ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which) {
|
|
return GetResourceLimitPeakValue(out_peak_value, resource_limit_handle, which);
|
|
}
|
|
|
|
Result CreateResourceLimit64(ams::svc::Handle *out_handle) {
|
|
return CreateResourceLimit(out_handle);
|
|
}
|
|
|
|
Result SetResourceLimitLimitValue64(ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which, int64_t limit_value) {
|
|
return SetResourceLimitLimitValue(resource_limit_handle, which, limit_value);
|
|
}
|
|
|
|
/* ============================= 64From32 ABI ============================= */
|
|
|
|
Result GetResourceLimitLimitValue64From32(int64_t *out_limit_value, ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which) {
|
|
return GetResourceLimitLimitValue(out_limit_value, resource_limit_handle, which);
|
|
}
|
|
|
|
Result GetResourceLimitCurrentValue64From32(int64_t *out_current_value, ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which) {
|
|
return GetResourceLimitCurrentValue(out_current_value, resource_limit_handle, which);
|
|
}
|
|
|
|
Result GetResourceLimitPeakValue64From32(int64_t *out_peak_value, ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which) {
|
|
return GetResourceLimitPeakValue(out_peak_value, resource_limit_handle, which);
|
|
}
|
|
|
|
Result CreateResourceLimit64From32(ams::svc::Handle *out_handle) {
|
|
return CreateResourceLimit(out_handle);
|
|
}
|
|
|
|
Result SetResourceLimitLimitValue64From32(ams::svc::Handle resource_limit_handle, ams::svc::LimitableResource which, int64_t limit_value) {
|
|
return SetResourceLimitLimitValue(resource_limit_handle, which, limit_value);
|
|
}
|
|
|
|
}
|