[sginsberg] 38768: - DbgkpSendApiMessage/Lpc: Modification of EPROCESS flags must be interlocked, use PspSetProcessFlag. - DbgkMap/UnmapViewOfSection: Don't check for the DeadThread flag, doing so is only necessary during early thread creation. - DbgkCreateThread: Set the StartAddress pointer in the case where we only send the Create Thread message. Also, we should receive the target thread as the first parameter. - NtDebugActiveProcess: Don't allow debugging the current process. - NtDebugContin

Author: sginsberg
Date: Thu Jan 15 07:14:37 2009
New Revision: 38768

URL: http://svn.reactos.org/svn/reactos?rev=38768&view=rev
Log:
- DbgkpSendApiMessage/Lpc: Modification of EPROCESS flags must be interlocked, use PspSetProcessFlag.
- DbgkMap/UnmapViewOfSection: Don't check for the DeadThread flag, doing so is only necessary during early thread creation.
- DbgkCreateThread: Set the StartAddress pointer in the case where we only send the Create Thread message. Also, we should receive the target thread as the first parameter.
- NtDebugActiveProcess: Don't allow debugging the current process.
- NtDebugContinue: We should signal the Debug Object, not the Debug Event. Also make sure we pass the correct event to DbgkpWakeTarget.
- NtWaitForDebugEvent: Cleanup probe and capture of the input timeout.

Modified:
    trunk/reactos/ntoskrnl/dbgk/dbgkobj.c
    trunk/reactos/ntoskrnl/dbgk/dbgkutil.c
    trunk/reactos/ntoskrnl/include/internal/dbgk.h
    trunk/reactos/ntoskrnl/ps/thread.c

Modified: trunk/reactos/ntoskrnl/dbgk/dbgkobj.c
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/dbgk/dbgkobj.c?rev...
==============================================================================
--- trunk/reactos/ntoskrnl/dbgk/dbgkobj.c [iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/dbgk/dbgkobj.c [iso-8859-1] Thu Jan 15 07:14:37 2009
@@ -219,7 +219,7 @@
     Message->ReturnedStatus = STATUS_PENDING;

/* Set create process reported state */
-    PsGetCurrentProcess()->CreateReported = TRUE;
+    PspSetProcessFlag(PsGetCurrentProcess(), PSF_CREATE_REPORTED_BIT);

/* Send the LPC command */
     Status = LpcRequestWaitReplyPort(Port,
@@ -254,7 +254,7 @@
     ApiMsg->ReturnedStatus = STATUS_PENDING;

/* Set create process reported state */
-    PsGetCurrentProcess()->CreateReported = TRUE;
+    PspSetProcessFlag(PsGetCurrentProcess(), PSF_CREATE_REPORTED_BIT);

/* Send the LPC command */
     Status = DbgkpQueueMessage(PsGetCurrentProcess(),
@@ -1081,6 +1081,7 @@
     BOOLEAN DebugPortCleared = FALSE;
     PLIST_ENTRY DebugEventList;
     PDEBUG_EVENT DebugEvent;
+    PAGED_CODE();
     DBGKTRACE(DBGK_OBJECT_DEBUG, "OwnerProcess: %p DebugObject: %p\n",
               OwnerProcess, DebugObject);

@@ -1260,8 +1261,9 @@
         else
         {
             /* Set the process flags */
-            InterlockedOr((PLONG)&Process->Flags,
-                          PSF_NO_DEBUG_INHERIT_BIT | PSF_CREATE_REPORTED_BIT);
+            PspSetProcessFlag(Process,
+                              PSF_NO_DEBUG_INHERIT_BIT |
+                              PSF_CREATE_REPORTED_BIT);

/* Reference the debug object */
             ObReferenceObject(DebugObject);
@@ -1474,12 +1476,15 @@
                                  &hDebug);
         if (NT_SUCCESS(Status))
         {
+            /* Enter SEH to protect the write */
             _SEH2_TRY
             {
+                /* Return the handle */
                 *DebugHandle = hDebug;
             }
-            _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
-            {
+            _SEH2_EXCEPT(ExSystemExceptionFilter())
+            {
+                /* Get the exception code */
                 Status = _SEH2_GetExceptionCode();
             } _SEH2_END;
         }
@@ -1570,7 +1575,7 @@
                     {
                         /* Wake it up and break out */
                         DebugEvent->Flags &= ~DEBUG_EVENT_INACTIVE;
-                        KeSetEvent(&DebugEvent->ContinueEvent,
+                        KeSetEvent(&DebugObject->EventsPresent,
                                    IO_NO_INCREMENT,
                                    FALSE);
                         break;
@@ -1603,11 +1608,11 @@
             if (NeedsWake)
             {
                 /* Set the continue status */
-                DebugEvent->ApiMsg.ReturnedStatus = ContinueStatus;
-                DebugEvent->Status = STATUS_SUCCESS;
+                DebugEventToWake->ApiMsg.ReturnedStatus = ContinueStatus;
+                DebugEventToWake->Status = STATUS_SUCCESS;

/* Wake the target */
-                DbgkpWakeTarget(DebugEvent);
+                DbgkpWakeTarget(DebugEventToWake);
             }
             else
             {
@@ -1628,7 +1633,7 @@
 {
     PEPROCESS Process;
     PDEBUG_OBJECT DebugObject;
-    KPROCESSOR_MODE PreviousMode = KeGetPreviousMode();
+    KPROCESSOR_MODE PreviousMode = ExGetPreviousMode();
     PETHREAD LastThread;
     NTSTATUS Status;
     PAGED_CODE();
@@ -1644,8 +1649,9 @@
                                        NULL);
     if (!NT_SUCCESS(Status)) return Status;

-    /* Don't allow debugging the initial system process */
-    if (Process == PsInitialSystemProcess)
+    /* Don't allow debugging the current process or the system process */
+    if ((Process == PsGetCurrentProcess()) ||
+         (Process == PsInitialSystemProcess))
     {
         /* Dereference and fail */
         ObDereferenceObject(Process);
@@ -1700,7 +1706,7 @@
 {
     PEPROCESS Process;
     PDEBUG_OBJECT DebugObject;
-    KPROCESSOR_MODE PreviousMode = KeGetPreviousMode();
+    KPROCESSOR_MODE PreviousMode = ExGetPreviousMode();
     NTSTATUS Status;
     PAGED_CODE();
     DBGKTRACE(DBGK_PROCESS_DEBUG, "Process: %p Handle: %p\n",
@@ -1823,7 +1829,7 @@
                     OUT PDBGUI_WAIT_STATE_CHANGE StateChange)
 {
     KPROCESSOR_MODE PreviousMode = ExGetPreviousMode();
-    LARGE_INTEGER SafeTimeOut;
+    LARGE_INTEGER LocalTimeOut;
     PEPROCESS Process;
     LARGE_INTEGER StartTime;
     PETHREAD Thread;
@@ -1837,44 +1843,46 @@
     PAGED_CODE();
     DBGKTRACE(DBGK_OBJECT_DEBUG, "Handle: %p\n", DebugHandle);

-    /* Clear the initial wait state change structure */
+    /* Clear the initial wait state change structure and the timeout */
     RtlZeroMemory(&WaitStateChange, sizeof(WaitStateChange));
-
-    /* Protect probe in SEH */
-    _SEH2_TRY
-    {
-        /* Check if we came with a timeout */
-        if (Timeout)
-        {
-            /* Check if the call was from user mode */
-            if (PreviousMode != KernelMode)
+    LocalTimeOut.QuadPart = 0;
+
+    /* Check if we were called from user mode */
+    if (PreviousMode != KernelMode)
+    {
+        /* Protect probe in SEH */
+        _SEH2_TRY
+        {
+            /* Check if we came with a timeout */
+            if (Timeout)
             {
                 /* Probe it */
                 ProbeForReadLargeInteger(Timeout);
-            }
-
-            /* Make a local copy */
-            SafeTimeOut = *Timeout;
-            Timeout = &SafeTimeOut;
-
-            /* Query the current time */
-            KeQuerySystemTime(&StartTime);
-        }
-
-        /* Check if the call was from user mode */
-        if (PreviousMode != KernelMode)
-        {
+
+                /* Make a local copy */
+                LocalTimeOut = *Timeout;
+                Timeout = &LocalTimeOut;
+            }
+
             /* Probe the state change structure */
             ProbeForWrite(StateChange, sizeof(*StateChange), sizeof(ULONG));
         }
-    }
-    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
-    {
-        /* Get the exception code */
-        Status = _SEH2_GetExceptionCode();
-    }
-    _SEH2_END;
-    if (!NT_SUCCESS(Status)) return Status;
+        _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
+        {
+            /* Get the exception code */
+            Status = _SEH2_GetExceptionCode();
+        }
+        _SEH2_END;
+        if (!NT_SUCCESS(Status)) return Status;
+    }
+    else
+    {
+        /* Copy directly */
+        if (Timeout) LocalTimeOut = *Timeout;
+    }
+
+    /* If we were passed a timeout, query the current time */
+    if (Timeout) KeQuerySystemTime(&StartTime);

/* Get the debug object */
     Status = ObReferenceObjectByHandle(DebugHandle,
@@ -1892,7 +1900,7 @@
     /* Wait on the debug object given to us */
     while (TRUE)
     {
-        Status = KeWaitForSingleObject(DebugObject,
+        Status = KeWaitForSingleObject(&DebugObject->EventsPresent,
                                        Executive,
                                        PreviousMode,
                                        Alertable,
@@ -2002,17 +2010,17 @@
         if (!GotEvent)
         {
             /* Check if we can wait again */
-            if (SafeTimeOut.QuadPart < 0)
+            if (LocalTimeOut.QuadPart < 0)
             {
                 /* Query the new time */
                 KeQuerySystemTime(&NewTime);

/* Substract times */
-                SafeTimeOut.QuadPart += (NewTime.QuadPart - StartTime.QuadPart);
+                LocalTimeOut.QuadPart += (NewTime.QuadPart - StartTime.QuadPart);
                 StartTime = NewTime;

/* Check if we've timed out */
-                if (SafeTimeOut.QuadPart >= 0)
+                if (LocalTimeOut.QuadPart >= 0)
                 {
                     /* We have, break out of the loop */
                     Status = STATUS_TIMEOUT;

Modified: trunk/reactos/ntoskrnl/dbgk/dbgkutil.c
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/dbgk/dbgkutil.c?re...
==============================================================================
--- trunk/reactos/ntoskrnl/dbgk/dbgkutil.c [iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/dbgk/dbgkutil.c [iso-8859-1] Thu Jan 15 07:14:37 2009
@@ -11,8 +11,6 @@
 #include <ntoskrnl.h>
 #define NDEBUG
 #include <debug.h>
-
-extern ULONG DbgkpTraceLevel;

/* FUNCTIONS *****************************************************************/

@@ -86,9 +84,9 @@

VOID
 NTAPI
-DbgkCreateThread(PVOID StartAddress)
-{
-    PETHREAD Thread = PsGetCurrentThread();
+DbgkCreateThread(IN PETHREAD Thread,
+                 IN PVOID StartAddress)
+{
     PEPROCESS Process = PsGetCurrentProcess();
     ULONG ProcessFlags;
     IMAGE_INFO ImageInfo;
@@ -106,10 +104,13 @@
     PTEB Teb;
     PAGED_CODE();

+    /* Sanity check */
+    ASSERT(Thread == PsGetCurrentThread());
+
     /* Try ORing in the create reported and image notify flags */
-    ProcessFlags = InterlockedOr((PLONG)&Process->Flags,
-                                 PSF_CREATE_REPORTED_BIT |
-                                 PSF_IMAGE_NOTIFY_DONE_BIT);
+    ProcessFlags = PspSetProcessFlag(Process,
+                                     PSF_CREATE_REPORTED_BIT |
+                                     PSF_IMAGE_NOTIFY_DONE_BIT);

/* Check if we were the first to set them or if another thread raced us */
     if (!(ProcessFlags & PSF_IMAGE_NOTIFY_DONE_BIT) && (PsImageNotifyEnabled))
@@ -282,7 +283,7 @@
     {
         /* Otherwise, do it just for the thread */
         CreateThread->SubSystemKey = 0;
-        CreateThread->StartAddress = NULL;
+        CreateThread->StartAddress = StartAddress;

/* Setup the API Message */
         ApiMessage.h.u1.Length = sizeof(DBGKM_MSG) << 16 |
@@ -385,11 +386,10 @@
     DBGKTRACE(DBGK_PROCESS_DEBUG,
               "Section: %p. Base: %p\n", Section, BaseAddress);

-    /* Check if this thread is hidden, doesn't have a debug port, or died */
-    if ((Thread->HideFromDebugger) ||
-        !(Process->DebugPort) ||
-        (Thread->DeadThread) ||
-        (KeGetPreviousMode() == KernelMode))
+    /* Check if this thread is kernel, hidden or doesn't have a debug port */
+    if ((ExGetPreviousMode() == KernelMode) ||
+        (Thread->HideFromDebugger) ||
+        !(Process->DebugPort))
     {
         /* Don't notify the debugger */
         return;
@@ -434,11 +434,10 @@
     PETHREAD Thread = PsGetCurrentThread();
     PAGED_CODE();

-    /* Check if this thread is hidden, doesn't have a debug port, or died */
-    if ((Thread->HideFromDebugger) ||
-        !(Process->DebugPort) ||
-        (Thread->DeadThread) ||
-        (KeGetPreviousMode() == KernelMode))
+    /* Check if this thread is kernel, hidden or doesn't have a debug port */
+    if ((ExGetPreviousMode() == KernelMode) ||
+        (Thread->HideFromDebugger) ||
+        !(Process->DebugPort))
     {
         /* Don't notify the debugger */
         return;
@@ -456,5 +455,3 @@
     /* Send the message */
     DbgkpSendApiMessage(&ApiMessage, TRUE);
 }
-
-/* EOF */

Modified: trunk/reactos/ntoskrnl/include/internal/dbgk.h
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/include/internal/d...
==============================================================================
--- trunk/reactos/ntoskrnl/include/internal/dbgk.h [iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/include/internal/dbgk.h [iso-8859-1] Thu Jan 15 07:14:37 2009
@@ -56,6 +56,7 @@
 VOID
 NTAPI
 DbgkCreateThread(
+    IN PETHREAD Thread,
     IN PVOID StartAddress
 );

@@ -133,6 +134,7 @@
     IN PDEBUG_OBJECT SourceDebugObject
 );

+extern ULONG DbgkpTraceLevel;
 extern POBJECT_TYPE DbgkDebugObjectType;

/* EOF */

Modified: trunk/reactos/ntoskrnl/ps/thread.c
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/ps/thread.c?rev=38...
==============================================================================
--- trunk/reactos/ntoskrnl/ps/thread.c [iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/ps/thread.c [iso-8859-1] Thu Jan 15 07:14:37 2009
@@ -56,7 +56,7 @@
     if (!(Thread->DeadThread) && !(Thread->HideFromDebugger))
     {
         /* We're not, so notify the debugger */
-        DbgkCreateThread(StartContext);
+        DbgkCreateThread(Thread, StartContext);
     }

/* Make sure we're not already dead */