Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-27 09:34:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

[NTOS:PS] Add query support for QUOTA_LIMITS_EX

Browse source
This commit is contained in:
Thamatip Chitpong 2024-12-11 01:50:30 +07:00
parent bf0dbdbd19
commit 1e06829961
2 changed files with 51 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
ntoskrnl/include/internal/ps_i.h
View file

@ -27,7 +27,9 @@ static const INFORMATION_CLASS_INFO PsProcessInfoClass[] =
( (
QUOTA_LIMITS, QUOTA_LIMITS,
ULONG, 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 */ /* ProcessIoCounters */

79
ntoskrnl/ps/query.c
View file

@ -153,15 +153,19 @@ NtQueryInformationProcess(
/* Process quota limits */ /* Process quota limits */
case ProcessQuotaLimits: 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; Status = STATUS_INFO_LENGTH_MISMATCH;
break; break;
} }
Length = sizeof(QUOTA_LIMITS); /* Set return length */
Length = ProcessInformationLength;
Extended = (Length == sizeof(QUOTA_LIMITS_EX));
/* Reference the process */ /* Reference the process */
Status = ObReferenceObjectByHandle(ProcessHandle, Status = ObReferenceObjectByHandle(ProcessHandle,
@ -175,36 +179,49 @@ NtQueryInformationProcess(
/* Indicate success */ /* Indicate success */
Status = STATUS_SUCCESS; Status = STATUS_SUCCESS;
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)
{
/* 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;
}
/* 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;
}
/* Protect writes with SEH */
_SEH2_TRY _SEH2_TRY
{ {
/* Set max/min working set sizes */ RtlCopyMemory(ProcessInformation, &QuotaLimits, Length);
QuotaLimits->MaximumWorkingSetSize =
Process->Vm.MaximumWorkingSetSize << PAGE_SHIFT;
QuotaLimits->MinimumWorkingSetSize =
Process->Vm.MinimumWorkingSetSize << PAGE_SHIFT;
/* Set default time limits */
QuotaLimits->TimeLimit.LowPart = MAXULONG;
QuotaLimits->TimeLimit.HighPart = MAXULONG;
/* 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;
}
} }
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{ {