mirrors/reactos
0
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-11-27 13:33:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

[NTOS:KE/x64] Improve KiInitializeTss

Browse Source
This commit is contained in:
Timo Kreuzer 2023-12-02 13:24:28 +02:00
parent bbe16859b5
commit e1497d43d7
2 changed files with 17 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ntoskrnl/include/internal/amd64/ke.h
View File

@ -385,8 +385,6 @@ Ki386PerfEnd(VOID)
struct _KPCR;
//VOID KiInitializeTss(IN PKTSS Tss, IN UINT64 Stack);
DECLSPEC_NORETURN VOID KiSwitchToBootStack(IN ULONG_PTR InitialStack);
VOID KiDivideErrorFault(VOID);
VOID KiDebugTrapOrFault(VOID);

26
ntoskrnl/ke/amd64/kiinit.c
View File

@ -242,15 +242,19 @@ KiInitializeCpu(PKIPCR Pcr)
_mm_setcsr(INITIAL_MXCSR);
}
static
VOID
FASTCALL
KiInitializeTss(IN PKTSS64 Tss,
IN UINT64 Stack)
KiInitializeTss(
_In_ PKIPCR Pcr,
_Out_ PKTSS64 Tss,
_In_ PVOID InitialStack,
_In_ PVOID DoubleFaultStack,
_In_ PVOID NmiStack)
{
PKGDTENTRY64 TssEntry;
/* Get pointer to the GDT entry */
TssEntry = KiGetGdtEntry(KeGetPcr()->GdtBase, KGDT64_SYS_TSS);
TssEntry = KiGetGdtEntry(Pcr->GdtBase, KGDT64_SYS_TSS);
/* Initialize the GDT entry */
KiInitGdtEntry(TssEntry, (ULONG64)Tss, sizeof(KTSS64), AMD64_TSS, 0);
@ -262,16 +266,16 @@ KiInitializeTss(IN PKTSS64 Tss,
Tss->IoMapBase = 0x68;
/* Setup ring 0 stack pointer */
Tss->Rsp0 = Stack;
Tss->Rsp0 = (ULONG64)InitialStack;
/* Setup a stack for Double Fault Traps */
Tss->Ist[1] = (ULONG64)KiP0DoubleFaultStack;
Tss->Ist[1] = (ULONG64)DoubleFaultStack;
/* Setup a stack for CheckAbort Traps */
Tss->Ist[2] = (ULONG64)KiP0DoubleFaultStack;
Tss->Ist[2] = (ULONG64)DoubleFaultStack;
/* Setup a stack for NMI Traps */
Tss->Ist[3] = (ULONG64)KiP0DoubleFaultStack;
Tss->Ist[3] = (ULONG64)NmiStack;
/* Load the task register */
__ltr(KGDT64_SYS_TSS);
@ -470,7 +474,11 @@ KiSystemStartup(IN PLOADER_PARAMETER_BLOCK LoaderBlock)
KiInitModuleList(LoaderBlock);
/* Setup the TSS descriptors and entries */
KiInitializeTss(Pcr->TssBase, InitialStack);
KiInitializeTss(Pcr,
Pcr->TssBase,
(PVOID)InitialStack,
KiP0DoubleFaultStack,
KiP0DoubleFaultStack);
/* Setup the IDT */
KeInitExceptions();