IntelFsp2Pkg: Fixed potentially NULL pointer accessing

[mirror_edk2.git] / IntelFsp2Pkg / FspSecCore / SecMain.c

diff --git a/IntelFsp2Pkg/FspSecCore/SecMain.c b/IntelFsp2Pkg/FspSecCore/SecMain.c
index f319c68cc56067a3d265a30ba9edd1249367b2f4..70460a3c8befc298546949be0ddab4d05349cd0b 100644 (file)
--- a/IntelFsp2Pkg/FspSecCore/SecMain.c
+++ b/IntelFsp2Pkg/FspSecCore/SecMain.c
@@ -100,7 +100,7 @@ SecStartup (
   // |-------------------|---->  TempRamBase\r
   IdtTableInStack.PeiService  = NULL;\r
   AsmReadIdtr (&IdtDescriptor);\r
-  if ((IdtDescriptor.Base == 0) && (IdtDescriptor.Limit == 0xFFFF)) {\r
+  if (IdtDescriptor.Base == 0) {\r
     ExceptionHandler = FspGetExceptionHandler(mIdtEntryTemplate);\r
     for (Index = 0; Index < FixedPcdGet8(PcdFspMaxInterruptSupported); Index ++) {\r
       CopyMem ((VOID*)&IdtTableInStack.IdtTable[Index], (VOID*)&ExceptionHandler, sizeof (UINT64));\r
@@ -113,8 +113,9 @@ SecStartup (
       // ERROR: IDT table size from boot loader is larger than FSP can support, DeadLoop here!\r
       //\r
       CpuDeadLoop();\r
+    } else {\r
+      CopyMem ((VOID *) (UINTN) &IdtTableInStack.IdtTable, (VOID *) IdtDescriptor.Base, IdtSize);\r
     }\r
-    CopyMem ((VOID *) (UINTN) &IdtTableInStack.IdtTable, (VOID *) IdtDescriptor.Base, IdtSize);\r
   }\r
   IdtDescriptor.Base  = (UINTN) &IdtTableInStack.IdtTable;\r
   IdtDescriptor.Limit = (UINT16)(IdtSize - 1);\r