Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-02 12:55:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

-call process notify routines below DISPATCH_LVL

Browse source 
-enable removal of process notify routines

svn path=/trunk/; revision=4845
This commit is contained in:
Gunnar Dalsnes 2003-06-05 22:45:22 +00:00
parent 5320d2eb3f
commit 1f8cbb88a0
1 changed files with 75 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,4 +1,4 @@
/* $Id: process.c,v 1.105 2003/05/24 17:14:41 hbirr Exp $ /* $Id: process.c,v 1.106 2003/06/05 22:45:22 gdalsnes Exp $
* *
* COPYRIGHT: See COPYING in the top level directory * COPYRIGHT: See COPYING in the top level directory
* PROJECT: ReactOS kernel * PROJECT: ReactOS kernel
@ -52,7 +52,6 @@ static GENERIC_MAPPING PiProcessMapping = {PROCESS_READ,
#define MAX_PROCESS_NOTIFY_ROUTINE_COUNT 8 #define MAX_PROCESS_NOTIFY_ROUTINE_COUNT 8
static ULONG PiProcessNotifyRoutineCount = 0;
static PCREATE_PROCESS_NOTIFY_ROUTINE static PCREATE_PROCESS_NOTIFY_ROUTINE
PiProcessNotifyRoutine[MAX_PROCESS_NOTIFY_ROUTINE_COUNT]; PiProcessNotifyRoutine[MAX_PROCESS_NOTIFY_ROUTINE_COUNT];
@ -286,20 +285,32 @@ PiDeleteProcess(PVOID ObjectBody)
KIRQL oldIrql; KIRQL oldIrql;
PEPROCESS Process; PEPROCESS Process;
ULONG i; ULONG i;
PCREATE_PROCESS_NOTIFY_ROUTINE NotifyRoutine[MAX_PROCESS_NOTIFY_ROUTINE_COUNT];
ULONG NotifyRoutineCount;
DPRINT("PiDeleteProcess(ObjectBody %x)\n",ObjectBody); DPRINT("PiDeleteProcess(ObjectBody %x)\n",ObjectBody);
Process = (PEPROCESS)ObjectBody; Process = (PEPROCESS)ObjectBody;
KeAcquireSpinLock(&PsProcessListLock, &oldIrql); KeAcquireSpinLock(&PsProcessListLock, &oldIrql);
for (i = 0; i < PiProcessNotifyRoutineCount; i++) NotifyRoutineCount = 0;
{ for (i = 0; i < MAX_PROCESS_NOTIFY_ROUTINE_COUNT; i++)
PiProcessNotifyRoutine[i](Process->InheritedFromUniqueProcessId, {
(HANDLE)Process->UniqueProcessId, if (PiProcessNotifyRoutine[i])
FALSE); {
} NotifyRoutine[NotifyRoutineCount++] = PiProcessNotifyRoutine[i];
}
}
RemoveEntryList(&Process->ProcessListEntry); RemoveEntryList(&Process->ProcessListEntry);
KeReleaseSpinLock(&PsProcessListLock, oldIrql); KeReleaseSpinLock(&PsProcessListLock, oldIrql);
for (i = 0;i < NotifyRoutineCount; i++)
{
//must be called below DISPATCH_LVL
NotifyRoutine[i](Process->InheritedFromUniqueProcessId,
(HANDLE)Process->UniqueProcessId,
FALSE);
}
/* KDB hook */ /* KDB hook */
KDB_DELETEPROCESS_HOOK(Process); KDB_DELETEPROCESS_HOOK(Process);
@ -472,7 +483,9 @@ NtCreateProcess(OUT PHANDLE ProcessHandle,
PVOID BaseAddress; PVOID BaseAddress;
PMEMORY_AREA MemoryArea; PMEMORY_AREA MemoryArea;
ULONG i; ULONG i;
PCREATE_PROCESS_NOTIFY_ROUTINE NotifyRoutine[MAX_PROCESS_NOTIFY_ROUTINE_COUNT];
ULONG NotifyRoutineCount;
DPRINT("NtCreateProcess(ObjectAttributes %x)\n",ObjectAttributes); DPRINT("NtCreateProcess(ObjectAttributes %x)\n",ObjectAttributes);
Status = ObReferenceObjectByHandle(ParentProcessHandle, Status = ObReferenceObjectByHandle(ParentProcessHandle,
@ -558,16 +571,26 @@ NtCreateProcess(OUT PHANDLE ProcessHandle,
} }
KeAcquireSpinLock(&PsProcessListLock, &oldIrql); KeAcquireSpinLock(&PsProcessListLock, &oldIrql);
for (i = 0; i < PiProcessNotifyRoutineCount; i++) NotifyRoutineCount = 0;
{ for (i = 0; i < MAX_PROCESS_NOTIFY_ROUTINE_COUNT; i++)
PiProcessNotifyRoutine[i](Process->InheritedFromUniqueProcessId, {
(HANDLE)Process->UniqueProcessId, if (PiProcessNotifyRoutine[i])
TRUE); {
} NotifyRoutine[NotifyRoutineCount++] = PiProcessNotifyRoutine[i];
}
}
InsertHeadList(&PsProcessListHead, &Process->ProcessListEntry); InsertHeadList(&PsProcessListHead, &Process->ProcessListEntry);
InitializeListHead(&Process->ThreadListHead); InitializeListHead(&Process->ThreadListHead);
KeReleaseSpinLock(&PsProcessListLock, oldIrql); KeReleaseSpinLock(&PsProcessListLock, oldIrql);
for (i = 0;i < NotifyRoutineCount; i++)
{
//must be called below DISPATCH_LVL
NotifyRoutine[i](Process->InheritedFromUniqueProcessId,
(HANDLE)Process->UniqueProcessId,
TRUE);
}
Process->Pcb.State = PROCESS_STATE_ACTIVE; Process->Pcb.State = PROCESS_STATE_ACTIVE;
/* /*
@ -1318,13 +1341,44 @@ NTSTATUS STDCALL
PsSetCreateProcessNotifyRoutine(IN PCREATE_PROCESS_NOTIFY_ROUTINE NotifyRoutine, PsSetCreateProcessNotifyRoutine(IN PCREATE_PROCESS_NOTIFY_ROUTINE NotifyRoutine,
IN BOOLEAN Remove) IN BOOLEAN Remove)
{ {
if (PiProcessNotifyRoutineCount >= MAX_PROCESS_NOTIFY_ROUTINE_COUNT) KIRQL oldIrql;
return(STATUS_INSUFFICIENT_RESOURCES); ULONG i;
PiProcessNotifyRoutine[PiProcessNotifyRoutineCount] = NotifyRoutine; KeAcquireSpinLock(&PsProcessListLock, &oldIrql);
PiProcessNotifyRoutineCount++;
return(STATUS_SUCCESS); if (Remove)
{
for(i=0;i<MAX_PROCESS_NOTIFY_ROUTINE_COUNT;i++)
{
if ((PVOID)PiProcessNotifyRoutine[i] == (PVOID)NotifyRoutine)
{
PiProcessNotifyRoutine[i] = NULL;
break;
}
}
KeReleaseSpinLock(&PsProcessListLock, oldIrql);
return(STATUS_SUCCESS);
}
/*insert*/
for(i=0;i<MAX_PROCESS_NOTIFY_ROUTINE_COUNT;i++)
{
if (PiProcessNotifyRoutine[i] == NULL)
{
PiProcessNotifyRoutine[i] = NotifyRoutine;
break;
}
}
KeReleaseSpinLock(&PsProcessListLock, oldIrql);
if (i == MAX_PROCESS_NOTIFY_ROUTINE_COUNT)
{
return STATUS_INSUFFICIENT_RESOURCES;
}
return STATUS_SUCCESS;
} }
/* EOF */ /* EOF */