reactos/ntoskrnl/kd64/kdtrap.c

   1 /*
   2  * PROJECT:         ReactOS Kernel
   3  * LICENSE:         GPL - See COPYING in the top level directory
   4  * FILE:            ntoskrnl/kd64/kdtrap.c
   5  * PURPOSE:         KD64 Trap Handlers
   6  * PROGRAMMERS:     Alex Ionescu (alex.ionescu@reactos.org)
   7  */
   8
   9 /* INCLUDES ******************************************************************/
  10
  11 #include <ntoskrnl.h>
  12 #define NDEBUG
  13 #include <debug.h>
  14
  15 /* FUNCTIONS *****************************************************************/
  16
  17 BOOLEAN
  18 NTAPI
  19 KdpReport(IN PKTRAP_FRAME TrapFrame,
  20           IN PKEXCEPTION_FRAME ExceptionFrame,
  21           IN PEXCEPTION_RECORD ExceptionRecord,
  22           IN PCONTEXT ContextRecord,
  23           IN KPROCESSOR_MODE PreviousMode,
  24           IN BOOLEAN SecondChanceException)
  25 {
  26     BOOLEAN Entered, Status;
  27     PKPRCB Prcb;
  28     NTSTATUS ExceptionCode = ExceptionRecord->ExceptionCode;
  29
  30     /* Check if this is INT1 or 3, or if we're forced to handle it */
  31     if ((ExceptionCode == STATUS_BREAKPOINT) ||
  32         (ExceptionCode == STATUS_SINGLE_STEP) ||
  33         //(ExceptionCode == STATUS_ASSERTION_FAILURE) ||
  34         (NtGlobalFlag & FLG_STOP_ON_EXCEPTION))
  35     {
  36         /* Check if we can't really handle this */
  37         if ((SecondChanceException) ||
  38             (ExceptionCode == STATUS_PORT_DISCONNECTED) ||
  39             (NT_SUCCESS(ExceptionCode)))
  40         {
  41             /* Return false to have someone else take care of the exception */
  42             return FALSE;
  43         }
  44     }
  45     else if (SecondChanceException)
  46     {
  47         /* We won't bother unless this is second chance */
  48         return FALSE;
  49     }
  50
  51     /* Enter the debugger */
  52     Entered = KdEnterDebugger(TrapFrame, ExceptionFrame);
  53
  54     /*
  55      * Get the KPRCB and save the CPU Control State manually instead of
  56      * using KiSaveProcessorState, since we already have a valid CONTEXT.
  57      */
  58     Prcb = KeGetCurrentPrcb();
  59     KiSaveProcessorControlState(&Prcb->ProcessorState);
  60     RtlCopyMemory(&Prcb->ProcessorState.ContextFrame,
  61                   ContextRecord,
  62                   sizeof(CONTEXT));
  63
  64     /* Report the new state */
  65     Status = KdpReportExceptionStateChange(ExceptionRecord,
  66                                            &Prcb->ProcessorState.
  67                                            ContextFrame,
  68                                            SecondChanceException);
  69
  70     /* Now restore the processor state, manually again. */
  71     RtlCopyMemory(ContextRecord,
  72                   &Prcb->ProcessorState.ContextFrame,
  73                   sizeof(CONTEXT));
  74     //KiRestoreProcessorControlState(&Prcb->ProcessorState);
  75
  76     /* Exit the debugger and clear the CTRL-C state */
  77     KdExitDebugger(Entered);
  78     KdpControlCPressed = FALSE;
  79     return Status;
  80 }
  81
  82 BOOLEAN
  83 NTAPI
  84 KdpStub(IN PKTRAP_FRAME TrapFrame,
  85         IN PKEXCEPTION_FRAME ExceptionFrame,
  86         IN PEXCEPTION_RECORD ExceptionRecord,
  87         IN PCONTEXT ContextRecord,
  88         IN KPROCESSOR_MODE PreviousMode,
  89         IN BOOLEAN SecondChanceException)
  90 {
  91     ULONG ExceptionCommand = ExceptionRecord->ExceptionInformation[0];
  92
  93     /* Check if this was a breakpoint due to DbgPrint or Load/UnloadSymbols */
  94     if ((ExceptionRecord->ExceptionCode == STATUS_BREAKPOINT) &&
  95         (ExceptionRecord->NumberParameters > 0) &&
  96         ((ExceptionCommand == BREAKPOINT_LOAD_SYMBOLS) ||
  97          (ExceptionCommand == BREAKPOINT_UNLOAD_SYMBOLS) ||
  98          (ExceptionCommand == BREAKPOINT_COMMAND_STRING) ||
  99          (ExceptionCommand == BREAKPOINT_PRINT)))
 100     {
 101         /* This we can handle: simply bump EIP */
 102 #if defined (_M_X86)
 103         ContextRecord->Eip++;
 104 #elif defined (_M_AMD64)
 105         ContextRecord->Rip++;
 106 #else
 107 #error Unknown platform
 108 #endif
 109         return TRUE;
 110     }
 111     else if (KdPitchDebugger)
 112     {
 113         /* There's no debugger, fail. */
 114         return FALSE;
 115     }
 116     else if ((KdAutoEnableOnEvent) &&
 117              (KdPreviouslyEnabled) &&
 118              !(KdDebuggerEnabled) &&
 119              (KdEnableDebugger()) &&
 120              (KdDebuggerEnabled))
 121     {
 122         /* Debugging was Auto-Enabled. We can now send this to KD. */
 123         return KdpTrap(TrapFrame,
 124                        ExceptionFrame,
 125                        ExceptionRecord,
 126                        ContextRecord,
 127                        PreviousMode,
 128                        SecondChanceException);
 129     }
 130     else
 131     {
 132         /* FIXME: All we can do in this case is trace this exception */
 133         return FALSE;
 134     }
 135 }