|
|
@@ -153,9 +153,9 @@ ResultVal<s64> RetrieveResourceLimitValue(Core::System& system, Handle resource_
|
|
|
const auto* const current_process = system.Kernel().CurrentProcess();
|
|
|
ASSERT(current_process != nullptr);
|
|
|
|
|
|
- const auto resource_limit_object =
|
|
|
- current_process->GetHandleTable().Get<KResourceLimit>(resource_limit);
|
|
|
- if (!resource_limit_object) {
|
|
|
+ auto resource_limit_object =
|
|
|
+ current_process->GetHandleTable().GetObject<KResourceLimit>(resource_limit);
|
|
|
+ if (resource_limit_object.IsNull()) {
|
|
|
LOG_ERROR(Kernel_SVC, "Handle to non-existent resource limit instance used. Handle={:08X}",
|
|
|
resource_limit);
|
|
|
return ResultInvalidHandle;
|
|
|
@@ -843,12 +843,10 @@ static ResultCode GetInfo(Core::System& system, u64* result, u64 info_id, Handle
|
|
|
return RESULT_SUCCESS;
|
|
|
}
|
|
|
|
|
|
- const auto table_result = handle_table.Create(resource_limit.get());
|
|
|
- if (table_result.Failed()) {
|
|
|
- return table_result.Code();
|
|
|
- }
|
|
|
+ Handle handle{};
|
|
|
+ R_TRY(handle_table.Add(&handle, resource_limit));
|
|
|
|
|
|
- *result = *table_result;
|
|
|
+ *result = handle;
|
|
|
return RESULT_SUCCESS;
|
|
|
}
|
|
|
|
|
|
@@ -2093,83 +2091,86 @@ static ResultCode GetProcessInfo(Core::System& system, u64* out, Handle process_
|
|
|
}
|
|
|
|
|
|
static ResultCode CreateResourceLimit(Core::System& system, Handle* out_handle) {
|
|
|
- std::lock_guard lock{HLE::g_hle_lock};
|
|
|
LOG_DEBUG(Kernel_SVC, "called");
|
|
|
|
|
|
+ // Create a new resource limit.
|
|
|
auto& kernel = system.Kernel();
|
|
|
- auto resource_limit = std::make_shared<KResourceLimit>(kernel, system.CoreTiming());
|
|
|
+ KResourceLimit* resource_limit = KResourceLimit::Create(kernel);
|
|
|
+ R_UNLESS(resource_limit != nullptr, ResultOutOfResource);
|
|
|
|
|
|
- auto* const current_process = kernel.CurrentProcess();
|
|
|
- ASSERT(current_process != nullptr);
|
|
|
+ // Ensure we don't leak a reference to the limit.
|
|
|
+ SCOPE_EXIT({ resource_limit->Close(); });
|
|
|
|
|
|
- const auto handle = current_process->GetHandleTable().Create(resource_limit.get());
|
|
|
- if (handle.Failed()) {
|
|
|
- return handle.Code();
|
|
|
- }
|
|
|
+ // Initialize the resource limit.
|
|
|
+ resource_limit->Initialize(&system.CoreTiming());
|
|
|
+
|
|
|
+ // Register the limit.
|
|
|
+ KResourceLimit::Register(kernel, resource_limit);
|
|
|
+
|
|
|
+ // Add the limit to the handle table.
|
|
|
+ R_TRY(kernel.CurrentProcess()->GetHandleTable().Add(out_handle, resource_limit));
|
|
|
|
|
|
- *out_handle = *handle;
|
|
|
return RESULT_SUCCESS;
|
|
|
}
|
|
|
|
|
|
-static ResultCode GetResourceLimitLimitValue(Core::System& system, u64* out_value,
|
|
|
- Handle resource_limit, u32 resource_type) {
|
|
|
- LOG_DEBUG(Kernel_SVC, "called. Handle={:08X}, Resource type={}", resource_limit, resource_type);
|
|
|
+static ResultCode GetResourceLimitLimitValue(Core::System& system, u64* out_limit_value,
|
|
|
+ Handle resource_limit_handle,
|
|
|
+ LimitableResource which) {
|
|
|
+ LOG_DEBUG(Kernel_SVC, "called, resource_limit_handle={:08X}, which={}", resource_limit_handle,
|
|
|
+ which);
|
|
|
|
|
|
- const auto limit_value = RetrieveResourceLimitValue(system, resource_limit, resource_type,
|
|
|
- ResourceLimitValueType::LimitValue);
|
|
|
- if (limit_value.Failed()) {
|
|
|
- return limit_value.Code();
|
|
|
- }
|
|
|
+ // Validate the resource.
|
|
|
+ R_UNLESS(IsValidResourceType(which), ResultInvalidEnumValue);
|
|
|
+
|
|
|
+ // Get the resource limit.
|
|
|
+ auto& kernel = system.Kernel();
|
|
|
+ KScopedAutoObject resource_limit =
|
|
|
+ kernel.CurrentProcess()->GetHandleTable().GetObject<KResourceLimit>(resource_limit_handle);
|
|
|
+ R_UNLESS(resource_limit.IsNotNull(), ResultInvalidHandle);
|
|
|
+
|
|
|
+ // Get the limit value.
|
|
|
+ *out_limit_value = resource_limit->GetLimitValue(which);
|
|
|
|
|
|
- *out_value = static_cast<u64>(*limit_value);
|
|
|
return RESULT_SUCCESS;
|
|
|
}
|
|
|
|
|
|
-static ResultCode GetResourceLimitCurrentValue(Core::System& system, u64* out_value,
|
|
|
- Handle resource_limit, u32 resource_type) {
|
|
|
- LOG_DEBUG(Kernel_SVC, "called. Handle={:08X}, Resource type={}", resource_limit, resource_type);
|
|
|
+static ResultCode GetResourceLimitCurrentValue(Core::System& system, u64* out_current_value,
|
|
|
+ Handle resource_limit_handle,
|
|
|
+ LimitableResource which) {
|
|
|
+ LOG_DEBUG(Kernel_SVC, "called, resource_limit_handle={:08X}, which={}", resource_limit_handle,
|
|
|
+ which);
|
|
|
|
|
|
- const auto current_value = RetrieveResourceLimitValue(system, resource_limit, resource_type,
|
|
|
- ResourceLimitValueType::CurrentValue);
|
|
|
- if (current_value.Failed()) {
|
|
|
- return current_value.Code();
|
|
|
- }
|
|
|
+ // Validate the resource.
|
|
|
+ R_UNLESS(IsValidResourceType(which), ResultInvalidEnumValue);
|
|
|
+
|
|
|
+ // Get the resource limit.
|
|
|
+ auto& kernel = system.Kernel();
|
|
|
+ KScopedAutoObject resource_limit =
|
|
|
+ kernel.CurrentProcess()->GetHandleTable().GetObject<KResourceLimit>(resource_limit_handle);
|
|
|
+ R_UNLESS(resource_limit.IsNotNull(), ResultInvalidHandle);
|
|
|
+
|
|
|
+ // Get the current value.
|
|
|
+ *out_current_value = resource_limit->GetCurrentValue(which);
|
|
|
|
|
|
- *out_value = static_cast<u64>(*current_value);
|
|
|
return RESULT_SUCCESS;
|
|
|
}
|
|
|
|
|
|
-static ResultCode SetResourceLimitLimitValue(Core::System& system, Handle resource_limit,
|
|
|
- u32 resource_type, u64 value) {
|
|
|
- LOG_DEBUG(Kernel_SVC, "called. Handle={:08X}, Resource type={}, Value={}", resource_limit,
|
|
|
- resource_type, value);
|
|
|
+static ResultCode SetResourceLimitLimitValue(Core::System& system, Handle resource_limit_handle,
|
|
|
+ LimitableResource which, u64 limit_value) {
|
|
|
+ LOG_DEBUG(Kernel_SVC, "called, resource_limit_handle={:08X}, which={}, limit_value={}",
|
|
|
+ resource_limit_handle, which, limit_value);
|
|
|
|
|
|
- const auto type = static_cast<LimitableResource>(resource_type);
|
|
|
- if (!IsValidResourceType(type)) {
|
|
|
- LOG_ERROR(Kernel_SVC, "Invalid resource limit type: '{}'", resource_type);
|
|
|
- return ResultInvalidEnumValue;
|
|
|
- }
|
|
|
-
|
|
|
- auto* const current_process = system.Kernel().CurrentProcess();
|
|
|
- ASSERT(current_process != nullptr);
|
|
|
+ // Validate the resource.
|
|
|
+ R_UNLESS(IsValidResourceType(which), ResultInvalidEnumValue);
|
|
|
|
|
|
- auto resource_limit_object =
|
|
|
- current_process->GetHandleTable().Get<KResourceLimit>(resource_limit);
|
|
|
- if (!resource_limit_object) {
|
|
|
- LOG_ERROR(Kernel_SVC, "Handle to non-existent resource limit instance used. Handle={:08X}",
|
|
|
- resource_limit);
|
|
|
- return ResultInvalidHandle;
|
|
|
- }
|
|
|
+ // Get the resource limit.
|
|
|
+ auto& kernel = system.Kernel();
|
|
|
+ KScopedAutoObject resource_limit =
|
|
|
+ kernel.CurrentProcess()->GetHandleTable().GetObject<KResourceLimit>(resource_limit_handle);
|
|
|
+ R_UNLESS(resource_limit.IsNotNull(), ResultInvalidHandle);
|
|
|
|
|
|
- const auto set_result = resource_limit_object->SetLimitValue(type, static_cast<s64>(value));
|
|
|
- if (set_result.IsError()) {
|
|
|
- LOG_ERROR(Kernel_SVC,
|
|
|
- "Attempted to lower resource limit ({}) for category '{}' below its current "
|
|
|
- "value ({})",
|
|
|
- resource_limit_object->GetLimitValue(type), resource_type,
|
|
|
- resource_limit_object->GetCurrentValue(type));
|
|
|
- return set_result;
|
|
|
- }
|
|
|
+ // Set the limit value.
|
|
|
+ R_TRY(resource_limit->SetLimitValue(which, limit_value));
|
|
|
|
|
|
return RESULT_SUCCESS;
|
|
|
}
|