[sir_richard] 45008: Trap handlers in C Patch 4 of X: [NTOS]: Implement trap 6 (invalid opcode) in C.

Author: sir_richard
Date: Fri Jan  8 19:45:04 2010
New Revision: 45008

URL: http://svn.reactos.org/svn/reactos?rev=45008&view=rev
Log:
Trap handlers in C Patch 4 of X:
    [NTOS]: Implement trap 6 (invalid opcode) in C.

Modified:
    trunk/reactos/ntoskrnl/ke/i386/trap.s
    trunk/reactos/ntoskrnl/ke/i386/traphdlr.c

Modified: trunk/reactos/ntoskrnl/ke/i386/trap.s
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/ke/i386/trap.s?re…
==============================================================================
--- trunk/reactos/ntoskrnl/ke/i386/trap.s [iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/ke/i386/trap.s [iso-8859-1] Fri Jan  8 19:45:04 2010
@@ -664,13 +664,6 @@
 
 /* HARDWARE TRAP HANDLERS ****************************************************/
 
-.func KiFixupFrame
-_KiFixupFrame:
-
-    /* TODO: Routine to fixup a KTRAP_FRAME when faulting from a syscall. */
-    UNHANDLED_PATH "Trap Frame Fixup"
-.endfunc
-
 GENERATE_TRAP_HANDLER KiTrap0, 1
 GENERATE_TRAP_HANDLER KiTrap1, 1
 
@@ -699,122 +692,7 @@
 GENERATE_TRAP_HANDLER KiTrap3, 1
 GENERATE_TRAP_HANDLER KiTrap4, 1
 GENERATE_TRAP_HANDLER KiTrap5, 1
-
-.func KiTrap6
-TRAP_FIXUPS kit6_a, kit6_t, DoFixupV86, DoNotFixupAbios
-_KiTrap6:
-
-    /* It this a V86 GPF? */
-    test dword ptr [esp+8], EFLAGS_V86_MASK
-    jz NotV86UD
-
-    /* Enter V86 Trap */
-    V86_TRAP_PROLOG kit6_a, kit6_v
-
-VdmOpCodeFault:
-    /* Not yet supported (Invalid OPCODE from V86) */
-    UNHANDLED_V86_PATH
-
-NotV86UD:
-    /* Push error code */
-    push 0
-
-    /* Enter trap */
-    TRAP_PROLOG kit6_a, kit6_t
-
-DispatchLockErrata:
-    /* Check if this happened in kernel mode */
-    test byte ptr [ebp+KTRAP_FRAME_CS], MODE_MASK
-    jz KmodeOpcode
-
-    /* Check for VDM */
-    cmp word ptr [ebp+KTRAP_FRAME_CS], KGDT_R3_CODE + RPL_MASK
-    jz UmodeOpcode
-
-    /* Check if the process is vDM */
-    mov ebx, PCR[KPCR_CURRENT_THREAD]
-    mov ebx, [ebx+KTHREAD_APCSTATE_PROCESS]
-    cmp dword ptr [ebx+EPROCESS_VDM_OBJECTS], 0
-    jnz IsVdmOpcode
-
-UmodeOpcode:
-    /* Get EIP and enable interrupts at this point */
-    mov esi, [ebp+KTRAP_FRAME_EIP]
-    sti
-
-    /* Set intruction prefix length */
-    mov ecx, 4
-
-    /* Setup a SEH frame */
-    push ebp
-    push OpcodeSEH
-    push PCR[KPCR_EXCEPTION_LIST]
-    mov PCR[KPCR_EXCEPTION_LIST], esp
-
-OpcodeLoop:
-    /* Get the instruction and check if it's LOCK */
-    mov al, [esi]
-    cmp al, 0xF0
-    jz LockCrash
-
-    /* Keep moving */
-    add esi, 1
-    loop OpcodeLoop
-
-    /* Undo SEH frame */
-    pop PCR[KPCR_EXCEPTION_LIST]
-    add esp, 8
-
-KmodeOpcode:
-
-    /* Re-enable interrupts */
-    sti
-
-    /* Setup illegal instruction exception and dispatch it */
-    mov ebx, [ebp+KTRAP_FRAME_EIP]
-    mov eax, STATUS_ILLEGAL_INSTRUCTION
-    jmp _DispatchNoParam
-
-LockCrash:
-
-    /* Undo SEH Frame */
-    pop PCR[KPCR_EXCEPTION_LIST]
-    add esp, 8
-
-    /* Setup invalid lock exception and dispatch it */
-    mov ebx, [ebp+KTRAP_FRAME_EIP]
-    mov eax, STATUS_INVALID_LOCK_SEQUENCE
-    jmp _DispatchNoParam
-
-IsVdmOpcode:
-
-    /* Unhandled yet */
-    UNHANDLED_V86_PATH
-
-    /* Return to caller */
-    jmp _Kei386EoiHelper@0
-
-OpcodeSEH:
-
-    /* Get SEH frame */
-    mov esp, [esp+8]
-    pop PCR[KPCR_EXCEPTION_LIST]
-    add esp, 4
-    pop ebp
-
-    /* Check if this was user mode */
-    test dword ptr [ebp+KTRAP_FRAME_CS], MODE_MASK
-    jnz KmodeOpcode
-
-    /* Do a bugcheck */
-    push ebp
-    push 0
-    push 0
-    push 0
-    push 0
-    push KMODE_EXCEPTION_NOT_HANDLED
-    call _KeBugCheckWithTf@24
-.endfunc
+GENERATE_TRAP_HANDLER KiTrap6, 1
 
 .func KiTrap7
 TRAP_FIXUPS kit7_a, kit7_t, DoFixupV86, DoNotFixupAbios
@@ -1473,8 +1351,13 @@
     pop PCR[KPCR_EXCEPTION_LIST]
     add esp, 8
 
-    /* Illegal instruction */
-    jmp KmodeOpcode
+    /* Re-enable interrupts */
+    sti
+
+    /* Setup illegal instruction exception and dispatch it */
+    mov ebx, [ebp+KTRAP_FRAME_EIP]
+    mov eax, STATUS_ILLEGAL_INSTRUCTION
+    jmp _DispatchNoParam
 
 NotPrefixByte:
     /* Check if it's a HLT */

Modified: trunk/reactos/ntoskrnl/ke/i386/traphdlr.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/ke/i386/traphdlr.…
==============================================================================
--- trunk/reactos/ntoskrnl/ke/i386/traphdlr.c [iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/ke/i386/traphdlr.c [iso-8859-1] Fri Jan  8 19:45:04 2010
@@ -340,6 +340,51 @@
     KiDispatchException0Args(STATUS_ARRAY_BOUNDS_EXCEEDED,
                              TrapFrame->Eip,
                              TrapFrame);
+}
+
+VOID
+FASTCALL
+KiTrap6Handler(IN PKTRAP_FRAME TrapFrame)
+{
+    PUCHAR Instruction;
+    ULONG i;
+    
+    /* Save trap frame */
+    KiEnterTrap(TrapFrame);
+    
+    /* Check for VDM trap */
+    ASSERT((KiVdmTrap(TrapFrame)) == FALSE);
+    
+    /* Enable interrupts */
+    Instruction = (PUCHAR)TrapFrame->Eip;
+    _enable();
+        
+    /* Check for user trap */
+    if (KiUserTrap(TrapFrame))
+    {
+        /* FIXME: Use SEH */
+        
+        /* Scan next 4 opcodes */
+        for (i = 0; i < 4; i++)
+        {
+            /* Check for LOCK instruction */
+            if (Instruction[i] == 0xF0)
+            {
+                /* Send invalid lock sequence exception */
+                KiDispatchException0Args(STATUS_INVALID_LOCK_SEQUENCE,
+                                         TrapFrame->Eip,
+                                         TrapFrame);
+            }
+        }
+        
+        /* FIXME: SEH ends here */
+    }
+    
+    /* Kernel-mode or user-mode fault (but not LOCK) */
+    KiDispatchException0Args(STATUS_ILLEGAL_INSTRUCTION,
+                             TrapFrame->Eip,
+                             TrapFrame);
+    
 }
 
 VOID