UefiCpuPkg/CpuExceptionHandlerLib: Setup single step in #PF handler

[mirror_edk2.git] / UefiCpuPkg / Library / CpuExceptionHandlerLib / X64 / ExceptionHandlerAsm.nasm

diff --git a/UefiCpuPkg/Library/CpuExceptionHandlerLib/X64/ExceptionHandlerAsm.nasm b/UefiCpuPkg/Library/CpuExceptionHandlerLib/X64/ExceptionHandlerAsm.nasm
index 7b97810d10976a164688804a81519111a899e7f7..f842af23365d806ac9f920b7b4cbc030dd7fc425 100644 (file)
--- a/UefiCpuPkg/Library/CpuExceptionHandlerLib/X64/ExceptionHandlerAsm.nasm
+++ b/UefiCpuPkg/Library/CpuExceptionHandlerLib/X64/ExceptionHandlerAsm.nasm
@@ -336,6 +336,10 @@ HasErrorCode:
     pop     r15\r
 \r
     mov     rsp, rbp\r
+    cmp     qword [rbp + 8], 14 ; #PF?\r
+    jne     .1\r
+    bts     qword [rsp + 40], 8 ; RFLAGS.TF\r
+.1:\r
     pop     rbp\r
     add     rsp, 16\r
     cmp     qword [rsp - 32], 0  ; check EXCEPTION_HANDLER_CONTEXT.OldIdtHandler\r