tyler.durden/audk
1
0
Fork 0
mirror of https://github.com/acidanthera/audk.git synced 2025-07-26 23:24:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

CpuExceptionHandlerLib: Add comments to make code more readable

Browse Source 
Today's implementation of handling HOOK_BEFORE and HOOK_AFTER is
a bit complex. More comments is better.

Contributed-under: TianoCore Contribution Agreement 1.1
Signed-off-by: Ruiyu Ni <ruiyu.ni@intel.com>
Cc: Fan Jeff <vanjeff_919@hotmail.com>
Reviewed-by: Jian J Wang <jian.j.wang@intel.com>
This commit is contained in:
Ruiyu Ni 2018-08-31 15:30:26 +08:00
parent 4cf022f2f1
commit 87a9dd0d15
3 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
UefiCpuPkg/Library/CpuExceptionHandlerLib/Ia32/ArchExceptionHandler.c
View File

@ -1,7 +1,7 @@
/** @file /** @file
IA32 CPU Exception Handler functons. IA32 CPU Exception Handler functons.
Copyright (c) 2012 - 2017, Intel Corporation. All rights reserved.<BR> Copyright (c) 2012 - 2018, Intel Corporation. All rights reserved.<BR>
This program and the accompanying materials This program and the accompanying materials
are licensed and made available under the terms and conditions of the BSD License are licensed and made available under the terms and conditions of the BSD License
which accompanies this distribution. The full text of the license may be found at which accompanies this distribution. The full text of the license may be found at
@ -66,7 +66,9 @@ ArchSaveExceptionContext (
ReservedVectors = ExceptionHandlerData->ReservedVectors; ReservedVectors = ExceptionHandlerData->ReservedVectors;
// //
// Save Exception context in global variable // Save Exception context in global variable in first entry of the exception handler.
// So when original exception handler returns to the new exception handler (second entry),
// the Eflags/Cs/Eip/ExceptionData can be used.
// //
ReservedVectors[ExceptionType].OldFlags = SystemContext.SystemContextIa32->Eflags; ReservedVectors[ExceptionType].OldFlags = SystemContext.SystemContextIa32->Eflags;
ReservedVectors[ExceptionType].OldCs = SystemContext.SystemContextIa32->Cs; ReservedVectors[ExceptionType].OldCs = SystemContext.SystemContextIa32->Cs;
@ -79,7 +81,7 @@ ArchSaveExceptionContext (
Eflags.Bits.IF = 0; Eflags.Bits.IF = 0;
SystemContext.SystemContextIa32->Eflags = Eflags.UintN; SystemContext.SystemContextIa32->Eflags = Eflags.UintN;
// //
// Modify the EIP in stack, then old IDT handler will return to the stub code // Modify the EIP in stack, then old IDT handler will return to HookAfterStubBegin.
// //
SystemContext.SystemContextIa32->Eip = (UINTN) ReservedVectors[ExceptionType].HookAfterStubHeaderCode; SystemContext.SystemContextIa32->Eip = (UINTN) ReservedVectors[ExceptionType].HookAfterStubHeaderCode;
} }

12
UefiCpuPkg/Library/CpuExceptionHandlerLib/PeiDxeSmmCpuException.c
View File

@ -40,7 +40,8 @@ CommonExceptionHandlerWorker (
switch (ReservedVectors[ExceptionType].Attribute) { switch (ReservedVectors[ExceptionType].Attribute) {
case EFI_VECTOR_HANDOFF_HOOK_BEFORE: case EFI_VECTOR_HANDOFF_HOOK_BEFORE:
// //
// Need to jmp to old IDT handler after this exception handler // The new exception handler registered by RegisterCpuInterruptHandler() is executed BEFORE original handler.
// Save the original handler to stack so the assembly code can jump to it instead of returning from handler.
// //
ExceptionHandlerContext->ExceptionDataFlag = (mErrorCodeFlag & (1 << ExceptionType)) ? TRUE : FALSE; ExceptionHandlerContext->ExceptionDataFlag = (mErrorCodeFlag & (1 << ExceptionType)) ? TRUE : FALSE;
ExceptionHandlerContext->OldIdtHandler = ReservedVectors[ExceptionType].ExceptonHandler; ExceptionHandlerContext->OldIdtHandler = ReservedVectors[ExceptionType].ExceptonHandler;
@ -48,11 +49,13 @@ CommonExceptionHandlerWorker (
case EFI_VECTOR_HANDOFF_HOOK_AFTER: case EFI_VECTOR_HANDOFF_HOOK_AFTER:
while (TRUE) { while (TRUE) {
// //
// If if anyone has gotten SPIN_LOCK for owner running hook after // If spin-lock can be acquired, it's the first time entering here.
// //
if (AcquireSpinLockOrFail (&ReservedVectors[ExceptionType].SpinLock)) { if (AcquireSpinLockOrFail (&ReservedVectors[ExceptionType].SpinLock)) {
// //
// Need to execute old IDT handler before running this exception handler // The new exception handler registered by RegisterCpuInterruptHandler() is executed AFTER original handler.
// Save the original handler to stack but skip running the new handler so the original handler is executed
// firstly.
// //
ReservedVectors[ExceptionType].ApicId = GetApicId (); ReservedVectors[ExceptionType].ApicId = GetApicId ();
ArchSaveExceptionContext (ExceptionType, SystemContext, ExceptionHandlerData); ArchSaveExceptionContext (ExceptionType, SystemContext, ExceptionHandlerData);
@ -61,7 +64,8 @@ CommonExceptionHandlerWorker (
return; return;
} }
// //
// If failed to acquire SPIN_LOCK, check if it was locked by processor itself // If spin-lock cannot be acquired, it's the second time entering here.
// 'break' instead of 'return' is used so the new exception handler can be executed.
// //
if (ReservedVectors[ExceptionType].ApicId == GetApicId ()) { if (ReservedVectors[ExceptionType].ApicId == GetApicId ()) {
// //

8
UefiCpuPkg/Library/CpuExceptionHandlerLib/X64/ArchExceptionHandler.c
View File

@ -1,7 +1,7 @@
/** @file /** @file
x64 CPU Exception Handler. x64 CPU Exception Handler.
Copyright (c) 2012 - 2017, Intel Corporation. All rights reserved.<BR> Copyright (c) 2012 - 2018, Intel Corporation. All rights reserved.<BR>
This program and the accompanying materials This program and the accompanying materials
are licensed and made available under the terms and conditions of the BSD License are licensed and made available under the terms and conditions of the BSD License
which accompanies this distribution. The full text of the license may be found at which accompanies this distribution. The full text of the license may be found at
@ -67,7 +67,9 @@ ArchSaveExceptionContext (
ReservedVectors = ExceptionHandlerData->ReservedVectors; ReservedVectors = ExceptionHandlerData->ReservedVectors;
// //
// Save Exception context in global variable // Save Exception context in global variable in first entry of the exception handler.
// So when original exception handler returns to the new exception handler (second entry),
// the Eflags/Cs/Eip/ExceptionData can be used.
// //
ReservedVectors[ExceptionType].OldSs = SystemContext.SystemContextX64->Ss; ReservedVectors[ExceptionType].OldSs = SystemContext.SystemContextX64->Ss;
ReservedVectors[ExceptionType].OldSp = SystemContext.SystemContextX64->Rsp; ReservedVectors[ExceptionType].OldSp = SystemContext.SystemContextX64->Rsp;
@ -82,7 +84,7 @@ ArchSaveExceptionContext (
Eflags.Bits.IF = 0; Eflags.Bits.IF = 0;
SystemContext.SystemContextX64->Rflags = Eflags.UintN; SystemContext.SystemContextX64->Rflags = Eflags.UintN;
// //
// Modify the EIP in stack, then old IDT handler will return to the stub code // Modify the EIP in stack, then old IDT handler will return to HookAfterStubBegin.
// //
SystemContext.SystemContextX64->Rip = (UINTN) ReservedVectors[ExceptionType].HookAfterStubHeaderCode; SystemContext.SystemContextX64->Rip = (UINTN) ReservedVectors[ExceptionType].HookAfterStubHeaderCode;
} }