Fixed remainingcalls to ObGetReferenceCount() and ObGetHandleCount().
[reactos.git] / reactos / ntoskrnl / ps / kill.c
index 129705d..b73b576 100644 (file)
@@ -1,4 +1,5 @@
-/*
+/* $Id: kill.c,v 1.51 2002/03/05 11:33:12 ekohl Exp $
+ *
  * COPYRIGHT:       See COPYING in the top level directory
  * PROJECT:         ReactOS kernel
  * FILE:            ntoskrnl/ps/kill.c
@@ -15,6 +16,8 @@
 #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>
@@ -25,36 +28,54 @@ 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)
+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 = PiThreadListHead.Flink;
-   while (current_entry != &PiThreadListHead)
+   current_entry = Process->ThreadListHead.Flink;
+   while (current_entry != &Process->ThreadListHead)
      {
-       current = CONTAINING_RECORD(current_entry,ETHREAD,Tcb.QueueListEntry);
-       if (current->ThreadsProcess == Process &&
-           current != PsGetCurrentThread())
+       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 = PiThreadListHead.Flink;
+            current_entry = Process->ThreadListHead.Flink;
+         }
+       else
+         {
+            current_entry = current_entry->Flink;
          }
-       current_entry = current_entry->Flink;
      }
-
    KeReleaseSpinLock(&PiThreadListLock, oldlvl);
+   DPRINT("Finished PiTerminateProcessThreads()\n");
 }
 
-VOID PsReapThreads(VOID)
+VOID
+PsReapThreads(VOID)
 {
    PLIST_ENTRY current_entry;
    PETHREAD current;
@@ -78,41 +99,39 @@ VOID PsReapThreads(VOID)
             PEPROCESS Process = current->ThreadsProcess; 
             NTSTATUS Status = current->ExitStatus;
             
-            ObReferenceObjectByPointer(Process, 
-                                       0, 
-                                       PsProcessType, 
-                                       KernelMode );
+            DPRINT("PsProcessType %x\n", PsProcessType);
             DPRINT("Reaping thread %x\n", current);
+            DPRINT("Pointer count %d\n", ObGetObjectPointerCount(Process));
             current->Tcb.State = THREAD_STATE_TERMINATED_2;
             RemoveEntryList(&current->Tcb.ProcessThreadListEntry);
-            KeReleaseSpinLock(&PiThreadListLock, oldIrql);
-            ObDereferenceObject(current);
-            KeAcquireSpinLock(&PiThreadListLock, &oldIrql);
-            if(IsListEmpty( &Process->Pcb.ThreadListHead))
+            if (IsListEmpty(&Process->ThreadListHead))
               {
-                 /* 
-                  * TODO: Optimize this so it doesnt jerk the IRQL around so 
-                  * much :)
-                  */
                  DPRINT("Last thread terminated, terminating process\n");
-                 KeReleaseSpinLock(&PiThreadListLock, oldIrql);
+                 KeReleaseSpinLock( &PiThreadListLock, oldIrql );
                  PiTerminateProcess(Process, Status);
-                 KeAcquireSpinLock(&PiThreadListLock, &oldIrql);
+                 KeAcquireSpinLock( &PiThreadListLock, &oldIrql );
               }
-            ObDereferenceObject(Process);
+            DPRINT("Pointer count %d\n", ObGetObjectPointerCount(Process));
+            KeReleaseSpinLock(&PiThreadListLock, oldIrql);
+            ObDereferenceObject(current);
+            KeAcquireSpinLock(&PiThreadListLock, &oldIrql);
             current_entry = PiThreadListHead.Flink;
          }
      }
    KeReleaseSpinLock(&PiThreadListLock, oldIrql);
 }
 
-VOID PsTerminateCurrentThread(NTSTATUS ExitStatus)
+VOID
+PsTerminateCurrentThread(NTSTATUS ExitStatus)
 /*
  * FUNCTION: Terminates the current thread
  */
 {
    KIRQL oldIrql;
    PETHREAD CurrentThread;
+   PKTHREAD Thread;
+   PLIST_ENTRY current_entry;
+   PKMUTANT Mutant;
    
    CurrentThread = PsGetCurrentThread();
    
@@ -120,63 +139,120 @@ VOID PsTerminateCurrentThread(NTSTATUS ExitStatus)
    KeAcquireSpinLock(&PiThreadListLock, &oldIrql);
    
    CurrentThread->ExitStatus = ExitStatus;
+   Thread = KeGetCurrentThread();
+   KeCancelTimer(&Thread->Timer);
+   KeReleaseSpinLock(&PiThreadListLock, oldIrql);
    
-   DPRINT("ObGetReferenceCount(CurrentThread) %d\n",
-         ObGetReferenceCount(CurrentThread));
-   DPRINT("ObGetHandleCount(CurrentThread) %x\n",
-         ObGetHandleCount(CurrentThread));
+   /* abandon all owned mutants */
+   current_entry = Thread->MutantListHead.Flink;
+   while (current_entry != &Thread->MutantListHead)
+     {
+       Mutant = CONTAINING_RECORD(current_entry, KMUTANT,
+                                  MutantListEntry);
+       KeReleaseMutant(Mutant,
+                       MUTANT_INCREMENT,
+                       TRUE,
+                       FALSE);
+       current_entry = Thread->MutantListHead.Flink;
+     }
    
+   KeAcquireDispatcherDatabaseLock(FALSE);
    CurrentThread->Tcb.DispatcherHeader.SignalState = TRUE;
    KeDispatcherObjectWake(&CurrentThread->Tcb.DispatcherHeader);
-   
+   KeReleaseDispatcherDatabaseLock(FALSE);
+
+   KeAcquireSpinLock(&PiThreadListLock, &oldIrql);
    PsDispatchThreadNoLock(THREAD_STATE_TERMINATED_1);
    KeBugCheck(0);
 }
 
-VOID PsTerminateOtherThread(PETHREAD Thread, NTSTATUS ExitStatus)
+VOID STDCALL
+PiTerminateThreadRundownRoutine(PKAPC Apc)
+{
+  ExFreePool(Apc);
+}
+
+VOID STDCALL
+PiTerminateThreadKernelRoutine(PKAPC Apc,
+                              PKNORMAL_ROUTINE* NormalRoutine,
+                              PVOID* NormalContext,
+                              PVOID* SystemArgument1,
+                              PVOID* SystemArguemnt2)
+{
+  ExFreePool(Apc);
+}
+
+VOID STDCALL
+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 oldIrql;
-   
-   KeAcquireSpinLock(&PiThreadListLock, &oldIrql);
-   if (Thread->Tcb.State == THREAD_STATE_RUNNABLE)
-     {
-       RemoveEntryList(&Thread->Tcb.QueueListEntry);
-     }
-   Thread->Tcb.State = THREAD_STATE_TERMINATED_2;
-   Thread->Tcb.DispatcherHeader.SignalState = TRUE;
-   KeDispatcherObjectWake(&Thread->Tcb.DispatcherHeader);
-   KeReleaseSpinLock(&PiThreadListLock, oldIrql);
-   ObDereferenceObject(Thread);
+  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)
+NTSTATUS STDCALL
+PiTerminateProcess(PEPROCESS Process,
+                  NTSTATUS ExitStatus)
 {
-   KIRQL oldlvl;
+   DPRINT("PiTerminateProcess(Process %x, ExitStatus %x) PC %d HC %d\n",
+          Process, ExitStatus, ObGetObjectPointerCount(Process),
+          ObGetObjectHandleCount(Process));
    
-   DPRINT("PsTerminateProcess(Process %x, ExitStatus %x)\n",
-          Process, ExitStatus);
-   
-   PiTerminateProcessThreads(Process, ExitStatus);
-   KeRaiseIrql(DISPATCH_LEVEL, &oldlvl);
-   Process->Pcb.ProcessState = PROCESS_STATE_TERMINATED;
+   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);
-   KeLowerIrql(oldlvl);
+   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;
    
    DPRINT("NtTerminateProcess(ProcessHandle %x, ExitStatus %x)\n",
-          ProcessHandle, ExitStatus);
+          ProcessHandle, ExitStatus);
    
    Status = ObReferenceObjectByHandle(ProcessHandle,
                                       PROCESS_TERMINATE,
@@ -184,24 +260,25 @@ NTSTATUS STDCALL NtTerminateProcess(IN    HANDLE          ProcessHandle,
                                      UserMode,
                                       (PVOID*)&Process,
                                      NULL);
-   if (Status != STATUS_SUCCESS)
-   {
-        return(Status);
-   }
+   if (!NT_SUCCESS(Status))
+     {
+       return(Status);
+     }
    
-   PiTerminateProcess(Process, ExitStatus);
+   PiTerminateProcessThreads(Process, ExitStatus);
    if (PsGetCurrentThread()->ThreadsProcess == Process)
-   {
-      ObDereferenceObject(Process);
-      PsTerminateCurrentThread(ExitStatus);
-   }
+     {
+       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;
@@ -231,7 +308,8 @@ NTSTATUS STDCALL NtTerminateThread(IN       HANDLE          ThreadHandle,
 }
 
 
-NTSTATUS PsTerminateSystemThread(NTSTATUS ExitStatus)
+NTSTATUS STDCALL
+PsTerminateSystemThread(NTSTATUS ExitStatus)
 /*
  * FUNCTION: Terminates the current thread
  * ARGUMENTS:
@@ -243,8 +321,55 @@ NTSTATUS PsTerminateSystemThread(NTSTATUS ExitStatus)
    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);
 }