https://git.reactos.org/?p=reactos.git;a=commitdiff;h=1e06829961cb597f7e1927a5e35b66febe674a96
commit 1e06829961cb597f7e1927a5e35b66febe674a96
Author: Thamatip Chitpong <thamatip.chitp...@reactos.org>
AuthorDate: Wed Dec 11 01:50:30 2024 +0700
Commit: Thamatip Chitpong <thamatip.chitp...@reactos.org>
CommitDate: Wed Dec 18 09:53:21 2024 +0700
[NTOS:PS] Add query support for QUOTA_LIMITS_EX
---
ntoskrnl/include/internal/ps_i.h | 4 ++-
ntoskrnl/ps/query.c | 77 ++++++++++++++++++++++++----------------
2 files changed, 50 insertions(+), 31 deletions(-)
diff --git a/ntoskrnl/include/internal/ps_i.h b/ntoskrnl/include/internal/ps_i.h
index 21f8368fa88..04d564bd36a 100644
--- a/ntoskrnl/include/internal/ps_i.h
+++ b/ntoskrnl/include/internal/ps_i.h
@@ -27,7 +27,9 @@ static const INFORMATION_CLASS_INFO PsProcessInfoClass[] =
(
QUOTA_LIMITS,
ULONG,
- ICIF_QUERY | ICIF_SET | ICIF_SET_SIZE_VARIABLE
+
+ /* NOTE: ICIF_SIZE_VARIABLE is for QUOTA_LIMITS_EX support */
+ ICIF_QUERY | ICIF_SET | ICIF_SIZE_VARIABLE
),
/* ProcessIoCounters */
diff --git a/ntoskrnl/ps/query.c b/ntoskrnl/ps/query.c
index 4613d7404b0..01aa0361af8 100644
--- a/ntoskrnl/ps/query.c
+++ b/ntoskrnl/ps/query.c
@@ -153,15 +153,19 @@ NtQueryInformationProcess(
/* Process quota limits */
case ProcessQuotaLimits:
{
- PQUOTA_LIMITS QuotaLimits = (PQUOTA_LIMITS)ProcessInformation;
+ QUOTA_LIMITS_EX QuotaLimits;
+ BOOLEAN Extended;
- if (ProcessInformationLength != sizeof(QUOTA_LIMITS))
+ if (ProcessInformationLength != sizeof(QUOTA_LIMITS) &&
+ ProcessInformationLength != sizeof(QUOTA_LIMITS_EX))
{
Status = STATUS_INFO_LENGTH_MISMATCH;
break;
}
- Length = sizeof(QUOTA_LIMITS);
+ /* Set return length */
+ Length = ProcessInformationLength;
+ Extended = (Length == sizeof(QUOTA_LIMITS_EX));
/* Reference the process */
Status = ObReferenceObjectByHandle(ProcessHandle,
@@ -175,36 +179,49 @@ NtQueryInformationProcess(
/* Indicate success */
Status = STATUS_SUCCESS;
- _SEH2_TRY
+ RtlZeroMemory(&QuotaLimits, sizeof(QuotaLimits));
+
+ /* Get max/min working set sizes */
+ QuotaLimits.MaximumWorkingSetSize =
+ Process->Vm.MaximumWorkingSetSize << PAGE_SHIFT;
+ QuotaLimits.MinimumWorkingSetSize =
+ Process->Vm.MinimumWorkingSetSize << PAGE_SHIFT;
+
+ /* Get default time limits */
+ QuotaLimits.TimeLimit.QuadPart = -1LL;
+
+ /* Is quota block a default one? */
+ if (Process->QuotaBlock == &PspDefaultQuotaBlock)
{
- /* Set max/min working set sizes */
- QuotaLimits->MaximumWorkingSetSize =
- Process->Vm.MaximumWorkingSetSize << PAGE_SHIFT;
- QuotaLimits->MinimumWorkingSetSize =
- Process->Vm.MinimumWorkingSetSize << PAGE_SHIFT;
+ /* Get default pools and pagefile limits */
+ QuotaLimits.PagedPoolLimit = (SIZE_T)-1;
+ QuotaLimits.NonPagedPoolLimit = (SIZE_T)-1;
+ QuotaLimits.PagefileLimit = (SIZE_T)-1;
+ }
+ else
+ {
+ /* Get limits from non-default quota block */
+ QuotaLimits.PagedPoolLimit =
+ Process->QuotaBlock->QuotaEntry[PsPagedPool].Limit;
+ QuotaLimits.NonPagedPoolLimit =
+ Process->QuotaBlock->QuotaEntry[PsNonPagedPool].Limit;
+ QuotaLimits.PagefileLimit =
+ Process->QuotaBlock->QuotaEntry[PsPageFile].Limit;
+ }
- /* Set default time limits */
- QuotaLimits->TimeLimit.LowPart = MAXULONG;
- QuotaLimits->TimeLimit.HighPart = MAXULONG;
+ /* Get additional information, if needed */
+ if (Extended)
+ {
+ /* FIXME: Get the correct information */
+ //QuotaLimits.WorkingSetLimit = (SIZE_T)-1; // Not used on
Win2k3, it is set to 0
+ QuotaLimits.Flags = QUOTA_LIMITS_HARDWS_MIN_DISABLE |
QUOTA_LIMITS_HARDWS_MAX_DISABLE;
+ QuotaLimits.CpuRateLimit.RateData = 0;
+ }
- /* Is quota block a default one? */
- if (Process->QuotaBlock == &PspDefaultQuotaBlock)
- {
- /* Set default pools and pagefile limits */
- QuotaLimits->PagedPoolLimit = (SIZE_T)-1;
- QuotaLimits->NonPagedPoolLimit = (SIZE_T)-1;
- QuotaLimits->PagefileLimit = (SIZE_T)-1;
- }
- else
- {
- /* Get limits from non-default quota block */
- QuotaLimits->PagedPoolLimit =
- Process->QuotaBlock->QuotaEntry[PsPagedPool].Limit;
- QuotaLimits->NonPagedPoolLimit =
- Process->QuotaBlock->QuotaEntry[PsNonPagedPool].Limit;
- QuotaLimits->PagefileLimit =
- Process->QuotaBlock->QuotaEntry[PsPageFile].Limit;
- }
+ /* Protect writes with SEH */
+ _SEH2_TRY
+ {
+ RtlCopyMemory(ProcessInformation, &QuotaLimits, Length);
}
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{
