-/*
+/* $Id: kill.c,v 1.46 2001/07/28 07:57:24 ea Exp $
+ *
* COPYRIGHT: See COPYING in the top level directory
* PROJECT: ReactOS kernel
* FILE: ntoskrnl/ps/kill.c
#include <internal/ke.h>
#include <internal/mm.h>
#include <internal/ob.h>
+#include <internal/port.h>
+#include <internal/pool.h>
#define NDEBUG
#include <internal/debug.h>
/* GLOBALS *******************************************************************/
extern ULONG PiNrThreads;
+extern ULONG PiNrRunnableThreads;
+extern KSPIN_LOCK PiThreadListLock;
+extern LIST_ENTRY PiThreadListHead;
+extern KSPIN_LOCK PiApcLock;
+
+VOID PsTerminateCurrentThread(NTSTATUS ExitStatus);
+
+#define TAG_TERMINATE_APC TAG('T', 'A', 'P', 'C')
/* FUNCTIONS *****************************************************************/
+VOID
+PiTerminateProcessThreads(PEPROCESS Process, NTSTATUS ExitStatus)
+{
+ KIRQL oldlvl;
+ PLIST_ENTRY current_entry;
+ PETHREAD current;
+
+ DPRINT("PiTerminateProcessThreads(Process %x, ExitStatus %x)\n",
+ Process, ExitStatus);
+
+ KeAcquireSpinLock(&PiThreadListLock, &oldlvl);
+
+ current_entry = Process->ThreadListHead.Flink;
+ while (current_entry != &Process->ThreadListHead)
+ {
+ current = CONTAINING_RECORD(current_entry, ETHREAD,
+ Tcb.ProcessThreadListEntry);
+ if (current != PsGetCurrentThread() &&
+ current->DeadThread == 0)
+ {
+ DPRINT("Terminating %x, current thread: %x, "
+ "thread's process: %x\n", current, PsGetCurrentThread(),
+ current->ThreadsProcess);
+ KeReleaseSpinLock(&PiThreadListLock, oldlvl);
+ PsTerminateOtherThread(current, ExitStatus);
+ KeAcquireSpinLock(&PiThreadListLock, &oldlvl);
+ current_entry = Process->ThreadListHead.Flink;
+ }
+ else
+ {
+ current_entry = current_entry->Flink;
+ }
+ }
+ KeReleaseSpinLock(&PiThreadListLock, oldlvl);
+ DPRINT("Finished PiTerminateProcessThreads()\n");
+}
+
+VOID
+PsReapThreads(VOID)
+{
+ PLIST_ENTRY current_entry;
+ PETHREAD current;
+ KIRQL oldIrql;
+
+// DPRINT1("PsReapThreads()\n");
+
+ KeAcquireSpinLock(&PiThreadListLock, &oldIrql);
+
+ current_entry = PiThreadListHead.Flink;
+
+ while (current_entry != &PiThreadListHead)
+ {
+ current = CONTAINING_RECORD(current_entry, ETHREAD,
+ Tcb.ThreadListEntry);
+
+ current_entry = current_entry->Flink;
+
+ if (current->Tcb.State == THREAD_STATE_TERMINATED_1)
+ {
+ PEPROCESS Process = current->ThreadsProcess;
+ NTSTATUS Status = current->ExitStatus;
+
+ DPRINT("PsProcessType %x\n", PsProcessType);
+ DPRINT("Reaping thread %x\n", current);
+ DPRINT("Ref count %d\n", ObGetReferenceCount(Process));
+ current->Tcb.State = THREAD_STATE_TERMINATED_2;
+ RemoveEntryList(¤t->Tcb.ProcessThreadListEntry);
+ if (IsListEmpty(&Process->ThreadListHead))
+ {
+ DPRINT("Last thread terminated, terminating process\n");
+ KeReleaseSpinLock( &PiThreadListLock, oldIrql );
+ PiTerminateProcess(Process, Status);
+ KeAcquireSpinLock( &PiThreadListLock, &oldIrql );
+ }
+ DPRINT("Ref count %d\n", ObGetReferenceCount(Process));
+ KeReleaseSpinLock(&PiThreadListLock, oldIrql);
+ ObDereferenceObject(current);
+ KeAcquireSpinLock(&PiThreadListLock, &oldIrql);
+ current_entry = PiThreadListHead.Flink;
+ }
+ }
+ KeReleaseSpinLock(&PiThreadListLock, oldIrql);
+}
+
VOID PsTerminateCurrentThread(NTSTATUS ExitStatus)
/*
* FUNCTION: Terminates the current thread
*/
{
- KIRQL oldlvl;
+ KIRQL oldIrql;
PETHREAD CurrentThread;
- PiNrThreads--;
-
CurrentThread = PsGetCurrentThread();
- CurrentThread->ExitStatus = ExitStatus;
-
DPRINT("terminating %x\n",CurrentThread);
- ObDereferenceObject(CurrentThread->ThreadsProcess);
- CurrentThread->ThreadsProcess = NULL;
- KeRaiseIrql(DISPATCH_LEVEL,&oldlvl);
- CurrentThread->Tcb.State = THREAD_STATE_TERMINATED;
+ KeAcquireSpinLock(&PiThreadListLock, &oldIrql);
+
+ CurrentThread->ExitStatus = ExitStatus;
+ KeCancelTimer(&KeGetCurrentThread()->Timer);
+ KeAcquireDispatcherDatabaseLock(FALSE);
CurrentThread->Tcb.DispatcherHeader.SignalState = TRUE;
KeDispatcherObjectWake(&CurrentThread->Tcb.DispatcherHeader);
- ZwYieldExecution();
- for(;;);
+ KeReleaseDispatcherDatabaseLock(FALSE);
+
+ PsDispatchThreadNoLock(THREAD_STATE_TERMINATED_1);
+ KeBugCheck(0);
}
-VOID PsTerminateOtherThread(PETHREAD Thread, NTSTATUS ExitStatus)
+VOID
+PiTerminateThreadRundownRoutine(PKAPC Apc)
+{
+ ExFreePool(Apc);
+}
+
+VOID
+PiTerminateThreadKernelRoutine(PKAPC Apc,
+ PKNORMAL_ROUTINE* NormalRoutine,
+ PVOID* NormalContext,
+ PVOID* SystemArgument1,
+ PVOID* SystemArguemnt2)
+{
+ ExFreePool(Apc);
+}
+
+VOID
+PiTerminateThreadNormalRoutine(PVOID NormalContext,
+ PVOID SystemArgument1,
+ PVOID SystemArgument2)
+{
+ PsTerminateCurrentThread(PsGetCurrentThread()->ExitStatus);
+}
+
+VOID
+PsTerminateOtherThread(PETHREAD Thread, NTSTATUS ExitStatus)
/*
- * FUNCTION: Terminate a thread when calling from that thread's context
+ * FUNCTION: Terminate a thread when calling from another thread's context
+ * NOTES: This function must be called with PiThreadListLock held
*/
{
- KIRQL oldlvl;
-
- PiNrThreads--;
- KeRaiseIrql(DISPATCH_LEVEL, &oldlvl);
- Thread->Tcb.State = THREAD_STATE_TERMINATED;
- Thread->Tcb.DispatcherHeader.SignalState = TRUE;
- KeDispatcherObjectWake(&Thread->Tcb.DispatcherHeader);
- ObDereferenceObject(Thread->ThreadsProcess);
- Thread->ThreadsProcess = NULL;
- KeLowerIrql(oldlvl);
+ PKAPC Apc;
+
+ DPRINT("PsTerminateOtherThread(Thread %x, ExitStatus %x)\n",
+ Thread, ExitStatus);
+
+ Thread->DeadThread = 1;
+ Thread->ExitStatus = ExitStatus;
+ Apc = ExAllocatePoolWithTag(NonPagedPool, sizeof(KAPC), TAG_TERMINATE_APC);
+ KeInitializeApc(Apc,
+ &Thread->Tcb,
+ 0,
+ PiTerminateThreadKernelRoutine,
+ PiTerminateThreadRundownRoutine,
+ PiTerminateThreadNormalRoutine,
+ KernelMode,
+ NULL);
+ KeInsertQueueApc(Apc,
+ NULL,
+ NULL,
+ KernelMode);
}
+NTSTATUS STDCALL
+PiTerminateProcess(PEPROCESS Process, NTSTATUS ExitStatus)
+{
+ DPRINT("PiTerminateProcess(Process %x, ExitStatus %x) RC %d HC %d\n",
+ Process, ExitStatus, ObGetReferenceCount(Process),
+ ObGetHandleCount(Process));
+
+ if (InterlockedExchange((PLONG)&Process->Pcb.State,
+ PROCESS_STATE_TERMINATED) ==
+ PROCESS_STATE_TERMINATED)
+ {
+ return(STATUS_SUCCESS);
+ }
+ KeAttachProcess( Process );
+ ObCloseAllHandles(Process);
+ KeDetachProcess();
+ KeAcquireDispatcherDatabaseLock(FALSE);
+ Process->Pcb.DispatcherHeader.SignalState = TRUE;
+ KeDispatcherObjectWake(&Process->Pcb.DispatcherHeader);
+ KeReleaseDispatcherDatabaseLock(FALSE);
+ return(STATUS_SUCCESS);
+}
-NTSTATUS
-STDCALL
-NtTerminateProcess (
- IN HANDLE ProcessHandle,
- IN NTSTATUS ExitStatus
- )
+NTSTATUS STDCALL NtTerminateProcess(IN HANDLE ProcessHandle,
+ IN NTSTATUS ExitStatus)
{
NTSTATUS Status;
PEPROCESS Process;
- KIRQL oldlvl;
DPRINT("NtTerminateProcess(ProcessHandle %x, ExitStatus %x)\n",
- ProcessHandle, ExitStatus);
+ ProcessHandle, ExitStatus);
Status = ObReferenceObjectByHandle(ProcessHandle,
PROCESS_TERMINATE,
UserMode,
(PVOID*)&Process,
NULL);
- if (Status != STATUS_SUCCESS)
- {
- return(Status);
- }
+ if (!NT_SUCCESS(Status))
+ {
+ return(Status);
+ }
PiTerminateProcessThreads(Process, ExitStatus);
- KeRaiseIrql(DISPATCH_LEVEL, &oldlvl);
- Process->Pcb.ProcessState = PROCESS_STATE_TERMINATED;
- Process->Pcb.DispatcherHeader.SignalState = TRUE;
- KeDispatcherObjectWake(&Process->Pcb.DispatcherHeader);
if (PsGetCurrentThread()->ThreadsProcess == Process)
- {
- KeLowerIrql(oldlvl);
- ObDereferenceObject(Process);
- PsTerminateCurrentThread(ExitStatus);
- }
- KeLowerIrql(oldlvl);
+ {
+ ObDereferenceObject(Process);
+ PsTerminateCurrentThread(ExitStatus);
+ }
ObDereferenceObject(Process);
return(STATUS_SUCCESS);
}
-NTSTATUS
-STDCALL
-NtTerminateThread (
- IN HANDLE ThreadHandle,
- IN NTSTATUS ExitStatus
- )
+NTSTATUS STDCALL NtTerminateThread(IN HANDLE ThreadHandle,
+ IN NTSTATUS ExitStatus)
{
PETHREAD Thread;
NTSTATUS Status;
{
return(Status);
}
-
+
+ ObDereferenceObject(Thread);
+
if (Thread == PsGetCurrentThread())
{
PsTerminateCurrentThread(ExitStatus);
return(STATUS_SUCCESS);
}
-VOID PsReleaseThread(PETHREAD Thread)
-/*
- * FUNCTION: Called internally to release a thread's resources from another
- * thread's context
- * ARGUMENTS:
- * Thread = Thread to release
- */
-{
- DPRINT("PsReleaseThread(Thread %x)\n",Thread);
-
- RemoveEntryList(&Thread->Tcb.Entry);
- HalReleaseTask(Thread);
- ObDereferenceObject(Thread);
-}
-
-NTSTATUS PsTerminateSystemThread(NTSTATUS ExitStatus)
+NTSTATUS STDCALL PsTerminateSystemThread(NTSTATUS ExitStatus)
/*
* FUNCTION: Terminates the current thread
* ARGUMENTS:
return(STATUS_SUCCESS);
}
+NTSTATUS STDCALL
+NtCallTerminatePorts(PETHREAD Thread)
+{
+ KIRQL oldIrql;
+ PLIST_ENTRY current_entry;
+ PEPORT_TERMINATION_REQUEST current;
+
+ KeAcquireSpinLock(&Thread->ActiveTimerListLock, &oldIrql);
+ while ((current_entry = RemoveHeadList(&Thread->TerminationPortList)) !=
+ &Thread->TerminationPortList);
+ {
+ current = CONTAINING_RECORD(current_entry,
+ EPORT_TERMINATION_REQUEST,
+ ThreadListEntry);
+ KeReleaseSpinLock(&Thread->ActiveTimerListLock, oldIrql);
+ LpcSendTerminationPort(current->Port,
+ Thread->CreateTime);
+ KeAcquireSpinLock(&Thread->ActiveTimerListLock, &oldIrql);
+ }
+ KeReleaseSpinLock(&Thread->ActiveTimerListLock, oldIrql);
+ return(STATUS_SUCCESS);
+}
-NTSTATUS
-STDCALL
-NtRegisterThreadTerminatePort (
- HANDLE TerminationPort
- )
+NTSTATUS STDCALL
+NtRegisterThreadTerminatePort(HANDLE TerminationPortHandle)
{
- UNIMPLEMENTED;
+ NTSTATUS Status;
+ PEPORT_TERMINATION_REQUEST Request;
+ PEPORT TerminationPort;
+ KIRQL oldIrql;
+ PETHREAD Thread;
+
+ Status = ObReferenceObjectByHandle(TerminationPortHandle,
+ PORT_ALL_ACCESS,
+ ExPortType,
+ UserMode,
+ (PVOID*)&TerminationPort,
+ NULL);
+ if (!NT_SUCCESS(Status))
+ {
+ return(Status);
+ }
+
+ Request = ExAllocatePool(NonPagedPool, sizeof(Request));
+ Request->Port = TerminationPort;
+ Thread = PsGetCurrentThread();
+ KeAcquireSpinLock(&Thread->ActiveTimerListLock, &oldIrql);
+ InsertTailList(&Thread->TerminationPortList, &Request->ThreadListEntry);
+ KeReleaseSpinLock(&Thread->ActiveTimerListLock, oldIrql);
+
+ return(STATUS_SUCCESS);
}
projects / reactos.git / blobdiff