Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-28 01:55:19 +00:00
Code Issues Projects Releases Packages Wiki Activity

Setup quota blocks for processes.

Browse source 
svn path=/trunk/; revision=16942
This commit is contained in:
Filip Navara 2005-08-01 11:20:44 +00:00
parent 50398b7a03
commit f28b5f7302
4 changed files with 59 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
reactos/ntoskrnl/ps/kill.c
View file

@ -509,6 +509,8 @@ PspExitProcess(PEPROCESS Process)
PspRunCreateProcessNotifyRoutines(Process, FALSE); PspRunCreateProcessNotifyRoutines(Process, FALSE);
PspDestroyQuotaBlock(Process);
/* close all handles associated with our process, this needs to be done /* close all handles associated with our process, this needs to be done
when the last thread still runs */ when the last thread still runs */
ObKillProcess(Process); ObKillProcess(Process);

8
reactos/ntoskrnl/ps/process.c
View file

@ -20,6 +20,8 @@ PEPROCESS EXPORTED PsInitialSystemProcess = NULL;
PEPROCESS PsIdleProcess = NULL; PEPROCESS PsIdleProcess = NULL;
POBJECT_TYPE EXPORTED PsProcessType = NULL; POBJECT_TYPE EXPORTED PsProcessType = NULL;
EPROCESS_QUOTA_BLOCK PspDefaultQuotaBlock;
LIST_ENTRY PsActiveProcessHead; LIST_ENTRY PsActiveProcessHead;
FAST_MUTEX PspActiveProcessMutex; FAST_MUTEX PspActiveProcessMutex;
LARGE_INTEGER ShortPsLockDelay, PsLockTimeout; LARGE_INTEGER ShortPsLockDelay, PsLockTimeout;
@ -295,8 +297,8 @@ PspCreateProcess(OUT PHANDLE ProcessHandle,
Process->Session = pParentProcess->Session; Process->Session = pParentProcess->Session;
} }
/* FIXME: Set up the Quota Block from the Parent /* Set up the Quota Block from the Parent */
PspInheritQuota(Parent, Process); */ PspInheritQuota(Process, pParentProcess);
/* FIXME: Set up Dos Device Map from the Parent /* FIXME: Set up Dos Device Map from the Parent
ObInheritDeviceMap(Parent, Process) */ ObInheritDeviceMap(Parent, Process) */
@ -369,7 +371,7 @@ PspCreateProcess(OUT PHANDLE ProcessHandle,
DPRINT1("Failed to create CID handle (unique process ID)! Status: 0x%x\n", Status); DPRINT1("Failed to create CID handle (unique process ID)! Status: 0x%x\n", Status);
ObDereferenceObject(Process); ObDereferenceObject(Process);
goto exitdereferenceobjects; goto exitdereferenceobjects;
} }
/* FIXME: Insert into Job Object */ /* FIXME: Insert into Job Object */

33
reactos/ntoskrnl/ps/psmgr.c
View file

@ -126,21 +126,30 @@ PsInitProcessManagment(VOID)
DPRINT("Creating Process Object Type\n"); DPRINT("Creating Process Object Type\n");
/* Initialize the Thread type */ /* Initialize the Process type */
RtlZeroMemory(&ObjectTypeInitializer, sizeof(ObjectTypeInitializer)); RtlZeroMemory(&ObjectTypeInitializer, sizeof(ObjectTypeInitializer));
RtlInitUnicodeString(&Name, L"Process"); RtlInitUnicodeString(&Name, L"Process");
ObjectTypeInitializer.Length = sizeof(ObjectTypeInitializer); ObjectTypeInitializer.Length = sizeof(ObjectTypeInitializer);
ObjectTypeInitializer.DefaultNonPagedPoolCharge = sizeof(EPROCESS); ObjectTypeInitializer.DefaultNonPagedPoolCharge = sizeof(EPROCESS);
ObjectTypeInitializer.GenericMapping = PiProcessMapping; ObjectTypeInitializer.GenericMapping = PiProcessMapping;
ObjectTypeInitializer.PoolType = NonPagedPool; ObjectTypeInitializer.PoolType = NonPagedPool;
ObjectTypeInitializer.ValidAccessMask = PROCESS_ALL_ACCESS; ObjectTypeInitializer.ValidAccessMask = PROCESS_ALL_ACCESS;
ObjectTypeInitializer.UseDefaultObject = TRUE; ObjectTypeInitializer.UseDefaultObject = TRUE;
ObjectTypeInitializer.DeleteProcedure = PspDeleteProcess; ObjectTypeInitializer.DeleteProcedure = PspDeleteProcess;
ObpCreateTypeObject(&ObjectTypeInitializer, &Name, &PsProcessType); ObpCreateTypeObject(&ObjectTypeInitializer, &Name, &PsProcessType);
InitializeListHead(&PsActiveProcessHead); InitializeListHead(&PsActiveProcessHead);
ExInitializeFastMutex(&PspActiveProcessMutex); ExInitializeFastMutex(&PspActiveProcessMutex);
/*
* Initialize the default quota block.
*/
RtlZeroMemory(&PspDefaultQuotaBlock, sizeof(PspDefaultQuotaBlock));
PspDefaultQuotaBlock.QuotaEntry[PagedPool].Limit = (SIZE_T)-1;
PspDefaultQuotaBlock.QuotaEntry[NonPagedPool].Limit = (SIZE_T)-1;
PspDefaultQuotaBlock.QuotaEntry[2].Limit = (SIZE_T)-1; /* Page file */
/* /*
* Initialize the idle process * Initialize the idle process
*/ */
@ -175,6 +184,7 @@ PsInitProcessManagment(VOID)
FALSE); FALSE);
PsIdleProcess->Pcb.DirectoryTableBase.QuadPart = (ULONG_PTR)MmGetPageDirectory(); PsIdleProcess->Pcb.DirectoryTableBase.QuadPart = (ULONG_PTR)MmGetPageDirectory();
strcpy(PsIdleProcess->ImageFileName, "Idle"); strcpy(PsIdleProcess->ImageFileName, "Idle");
PspInheritQuota(PsIdleProcess, NULL);
/* /*
* Initialize the system process * Initialize the system process
@ -207,6 +217,7 @@ PsInitProcessManagment(VOID)
sizeof(EPROCESS), sizeof(EPROCESS),
FALSE); FALSE);
KProcess = &PsInitialSystemProcess->Pcb; KProcess = &PsInitialSystemProcess->Pcb;
PspInheritQuota(PsInitialSystemProcess, NULL);
MmInitializeAddressSpace(PsInitialSystemProcess, MmInitializeAddressSpace(PsInitialSystemProcess,
&PsInitialSystemProcess->AddressSpace); &PsInitialSystemProcess->AddressSpace);

30
reactos/ntoskrnl/ps/quota.c
View file

@ -15,6 +15,36 @@
/* FUNCTIONS ***************************************************************/ /* FUNCTIONS ***************************************************************/
VOID
STDCALL
PspInheritQuota(PEPROCESS Process, PEPROCESS ParentProcess)
{
PEPROCESS_QUOTA_BLOCK QuotaBlock;
if (ParentProcess != NULL)
QuotaBlock = ParentProcess->QuotaBlock;
else
QuotaBlock = &PspDefaultQuotaBlock;
ASSERT(QuotaBlock != NULL);
InterlockedIncrement(&QuotaBlock->ReferenceCount);
Process->QuotaBlock = QuotaBlock;
}
VOID
STDCALL
PspDestroyQuotaBlock(PEPROCESS Process)
{
PEPROCESS_QUOTA_BLOCK QuotaBlock = Process->QuotaBlock;
if (InterlockedDecrement(&QuotaBlock->ReferenceCount) == 0)
{
if (QuotaBlock != &PspDefaultQuotaBlock)
ExFreePool(QuotaBlock);
}
}
/* /*
* @implemented * @implemented
*/ */