[jcatena] 45392: [NTOS] msc trap stub

Author: jcatena
Date: Wed Feb  3 14:25:03 2010
New Revision: 45392
URL: http://svn.reactos.org/svn/reactos?rev=45392&view=rev
Log:
[NTOS] msc trap stub
Modified:
    branches/jcatena-branch/ntoskrnl/trap/i386/TrapStub.h
    branches/jcatena-branch/ntoskrnl/trap/trap.c
    branches/jcatena-branch/ntoskrnl/trap/trap.h
Modified: branches/jcatena-branch/ntoskrnl/trap/i386/TrapStub.h
URL:
http://svn.reactos.org/svn/reactos/branches/jcatena-branch/ntoskrnl/trap/i3…
==============================================================================
--- branches/jcatena-branch/ntoskrnl/trap/i386/TrapStub.h [iso-8859-1] (original)
+++ branches/jcatena-branch/ntoskrnl/trap/i386/TrapStub.h [iso-8859-1] Wed Feb  3 14:25:03
2010
@@ -6,17 +6,20 @@
 #error
 #endif
-#define TRAP_STUB_NAMEH0(x) x##Handler
-#define TRAP_STUB_NAMEH1(x) TRAP_STUB_NAMEH0(x)
-#define TRAP_STUB_NAMEH TRAP_STUB_NAMEH1(TRAP_STUB_NAME)
+#define TRAP_STUB_NAMEH tokenpaste(TRAP_STUB_NAME, Handler)
+#if (TRAP_STUB_FLAGS & TRAPF_INTERRUPT)
+#define TRAP_STUB_PARAM2 tokenpaste(TRAP_STUB_NAME, Interrupt)
+PKINTERRUPT TRAP_STUB_PARAM2;
+#else
 VOID _FASTCALL TRAP_STUB_NAMEH(KTRAP_FRAME *TrapFrame);
+#endif
 _NAKED VOID TRAP_STUB_NAME(VOID)
 {
        _ASM_BEGIN
                // setup frame
-#if (TRAP_STUB_FLAGS & TRAPF_SYSENTER)
+#if (TRAP_STUB_FLAGS & TRAPF_FASTSYSCALL)
                mov esp, ss:[KIP0PCRADDRESS + offset KPCR.TSS]
                mov esp, KTSS.Esp0[esp]
                // sub esp, dword ptr offset KTRAP_FRAME.V86Es
@@ -37,7 +40,7 @@
 #if !(TRAP_STUB_FLAGS & TRAPF_NOSAVESEG)
                mov KTRAP_FRAME.SegDs[esp], ds
                mov KTRAP_FRAME.SegEs[esp], es
-#if !(TRAP_STUB_FLAGS & TRAPF_NOSAVEFS)
+#if (TRAP_STUB_FLAGS & TRAPF_SAVEFS)
                mov KTRAP_FRAME.SegFs[esp], fs
 #endif
 #if !(TRAP_STUB_FLAGS & TRAPF_NOLOADDS)
@@ -57,23 +60,16 @@
                mov KTRAP_FRAME.Edi[esp], edi
 #endif
-#if (TRAP_STUB_FLAGS & TRAPF_VECTOR)
-:scadr:
-               mov edx, 0
-               call edx
-#endif
                // call handler
                mov ecx, esp
+#if (TRAP_STUB_FLAGS & TRAPF_INTERRUPT)
+               mov edx, TRAP_STUB_PARAM2
+               call PKINTERRUPT.DispatchAddress[edx]
+#else
                call TRAP_STUB_NAMEH
+#endif
-       _ASM_END
-
-               // asmcall(TRAP_STUB_NAMEH);
-               // call TRAP_STUB_NAMEH
-               // call HandlerName
-
-       _ASM_BEGIN
-               // return
+               // restore regs
 #if (TRAP_STUB_FLAGS & TRAPF_SAVENOVOL)
                mov ebp, KTRAP_FRAME.Ebp[esp]
                mov ebx, KTRAP_FRAME.Ebx[esp]
@@ -84,12 +80,21 @@
 #if !(TRAP_STUB_FLAGS & TRAPF_NOSAVESEG)
                mov ds, KTRAP_FRAME.SegDs[esp]
                mov es, KTRAP_FRAME.SegEs[esp]
-#if !(TRAP_STUB_FLAGS & TRAPF_NOSAVEFS)
+#if (TRAP_STUB_FLAGS & TRAPF_SAVEFS)
                mov fs, KTRAP_FRAME.SegFs[esp]
 #endif
 #endif
+               // restore volatle regs and return
                mov eax, KTRAP_FRAME.Eax[esp]
+
+#if (TRAP_STUB_FLAGS & TRAPF_FASTSYSCALL)
+               mov ecx, KTRAP_FRAME.HardwareEsp[esp]
+               mov edx, KTRAP_FRAME.Eip[esp]
+               add esp, dword ptr offset KTRAP_FRAME.V86Es
+               sti
+               sysexit
+#endif
                mov ecx, KTRAP_FRAME.Ecx[esp]
                mov edx, KTRAP_FRAME.Edx[esp]
                iretd
Modified: branches/jcatena-branch/ntoskrnl/trap/trap.c
URL:
http://svn.reactos.org/svn/reactos/branches/jcatena-branch/ntoskrnl/trap/tr…
==============================================================================
--- branches/jcatena-branch/ntoskrnl/trap/trap.c [iso-8859-1] (original)
+++ branches/jcatena-branch/ntoskrnl/trap/trap.c [iso-8859-1] Wed Feb  3 14:25:03 2010
@@ -129,7 +129,7 @@
 //
 #define TRAP_STUB_NAME KiFastCallEntry
-#define TRAP_STUB_FLAGS TRAPF_SYSENTER
+#define TRAP_STUB_FLAGS TRAPF_FASTSYSCALL
 #include <TrapStub.h>
 #define TRAP_STUB_NAME KiInterruptTemplate
@@ -140,6 +140,5 @@
 VOID _FASTCALL init(void)
 {
-       void *t = scadr;
 }
Modified: branches/jcatena-branch/ntoskrnl/trap/trap.h
URL:
http://svn.reactos.org/svn/reactos/branches/jcatena-branch/ntoskrnl/trap/tr…
==============================================================================
--- branches/jcatena-branch/ntoskrnl/trap/trap.h [iso-8859-1] (original)
+++ branches/jcatena-branch/ntoskrnl/trap/trap.h [iso-8859-1] Wed Feb  3 14:25:03 2010
@@ -1,14 +1,14 @@
 _ONCE
-#define TRAPF_NOSAVESEG                1
-#define TRAPF_NOSAVEFS         2
-#define TRAPF_SAVENOVOL                4
-#define TRAPF_VECTOR           0x100
-#define TRAPF_ERRORCODE                0x200
-#define TRAPF_SYSENTER         0x400
-
-extern KPCR const *TrapPcr;
+// TRAP_STUB_FLAGS TrapStub x-macro flags
+// trap type
+#define TRAPF_ERRORCODE                1
+#define TRAPF_INTERRUPT                2
+#define TRAPF_FASTSYSCALL      4
+// options
+#define TRAPF_NOSAVESEG                0x100
+#define TRAPF_SAVEFS           0x200
+#define TRAPF_SAVENOVOL                0x400
+#define TRAPF_NOLOADDS         0x800
 #include <trap_asm.h>
-
-