; Variables referrenced by C code\r
;\r
\r
+%define MSR_IA32_MISC_ENABLE 0x1A0\r
+%define MSR_EFER 0xc0000080\r
+%define MSR_EFER_XD 0x800\r
+\r
;\r
; Constants relating to PROCESSOR_SMM_DESCRIPTOR\r
;\r
extern ASM_PFX(CpuSmmDebugExit)\r
\r
global ASM_PFX(gSmbase)\r
+global ASM_PFX(mXdSupported)\r
global ASM_PFX(gSmiStack)\r
global ASM_PFX(gSmiCr3)\r
global ASM_PFX(gcSmiHandlerTemplate)\r
mov [cs:bx + 2], eax\r
o32 lgdt [cs:bx] ; lgdt fword ptr cs:[bx]\r
mov ax, PROTECT_MODE_CS\r
- mov [cs:bx-0x2],ax \r
+ mov [cs:bx-0x2],ax\r
DB 0x66, 0xbf ; mov edi, SMBASE\r
ASM_PFX(gSmbase): DD 0\r
lea eax, [edi + (@ProtectedMode - _SmiEntryPoint) + 0x8000]\r
or ebx, 0x23\r
mov cr0, ebx\r
jmp dword 0x0:0x0\r
-_GdtDesc: \r
+_GdtDesc:\r
DW 0\r
DD 0\r
\r
mov eax, TSS_SEGMENT\r
ltr ax\r
\r
+; enable NXE if supported\r
+ DB 0xb0 ; mov al, imm8\r
+ASM_PFX(mXdSupported): DB 1\r
+ cmp al, 0\r
+ jz @SkipXd\r
+;\r
+; Check XD disable bit\r
+;\r
+ mov ecx, MSR_IA32_MISC_ENABLE\r
+ rdmsr\r
+ sub esp, 4\r
+ push rdx ; save MSR_IA32_MISC_ENABLE[63-32]\r
+ test edx, BIT2 ; MSR_IA32_MISC_ENABLE[34]\r
+ jz .0\r
+ and dx, 0xFFFB ; clear XD Disable bit if it is set\r
+ wrmsr\r
+.0:\r
+ mov ecx, MSR_EFER\r
+ rdmsr\r
+ or ax, MSR_EFER_XD ; enable NXE\r
+ wrmsr\r
+ jmp @XdDone\r
+@SkipXd:\r
+ sub esp, 8\r
+@XdDone:\r
+\r
; Switch into @LongMode\r
push LONG_MODE_CS ; push cs hardcore here\r
- call Base ; push reture address for retf later\r
+ call Base ; push return address for retf later\r
Base:\r
add dword [rsp], @LongMode - Base; offset for far retf, seg is the 1st arg\r
- mov ecx, 0xc0000080\r
+\r
+ mov ecx, MSR_EFER\r
rdmsr\r
- or ah, 1\r
+ or ah, 1 ; enable LME\r
wrmsr\r
mov rbx, cr0\r
- or ebx, 080010000h ; enable paging + WP\r
+ or ebx, 0x80010023 ; enable paging + WP + NE + MP + PE\r
mov cr0, rbx\r
retf\r
@LongMode: ; long mode (64-bit code) starts here\r
; jmp _SmiHandler ; instruction is not needed\r
\r
_SmiHandler:\r
- mov rbx, [rsp] ; rbx <- CpuIndex\r
+ mov rbx, [rsp + 0x8] ; rcx <- CpuIndex\r
\r
;\r
; Save FP registers\r
;\r
- sub rsp, 0x208\r
+ sub rsp, 0x200\r
DB 0x48 ; FXSAVE64\r
fxsave [rsp]\r
\r
mov rcx, rbx\r
mov rax, CpuSmmDebugEntry\r
call rax\r
- \r
+\r
mov rcx, rbx\r
mov rax, SmiRendezvous ; rax <- absolute addr of SmiRedezvous\r
call rax\r
- \r
+\r
mov rcx, rbx\r
mov rax, CpuSmmDebugExit\r
call rax\r
- \r
+\r
add rsp, 0x20\r
\r
;\r
DB 0x48 ; FXRSTOR64\r
fxrstor [rsp]\r
\r
+ add rsp, 0x200\r
+\r
+ mov rax, ASM_PFX(mXdSupported)\r
+ mov al, [rax]\r
+ cmp al, 0\r
+ jz .1\r
+ pop rdx ; get saved MSR_IA32_MISC_ENABLE[63-32]\r
+ test edx, BIT2\r
+ jz .1\r
+ mov ecx, MSR_IA32_MISC_ENABLE\r
+ rdmsr\r
+ or dx, BIT2 ; set XD Disable bit if it was set before entering into SMM\r
+ wrmsr\r
+\r
+.1:\r
rsm\r
\r
gcSmiHandlerSize DW $ - _SmiEntryPoint\r