Add parameter checking for MP Services Protocol Thunk driver.

[mirror_edk2.git] / EdkCompatibilityPkg / Compatibility / MpServicesOnFrameworkMpServicesThunk / MpServicesOnFrameworkMpServicesThunk.c

diff --git a/EdkCompatibilityPkg/Compatibility/MpServicesOnFrameworkMpServicesThunk/MpServicesOnFrameworkMpServicesThunk.c b/EdkCompatibilityPkg/Compatibility/MpServicesOnFrameworkMpServicesThunk/MpServicesOnFrameworkMpServicesThunk.c
index b31d17c47f420b1f894298e5d68646799d36dd46..9005670baebd8839ac6948d96c5b697a6ad5f045 100644 (file)
--- a/EdkCompatibilityPkg/Compatibility/MpServicesOnFrameworkMpServicesThunk/MpServicesOnFrameworkMpServicesThunk.c
+++ b/EdkCompatibilityPkg/Compatibility/MpServicesOnFrameworkMpServicesThunk/MpServicesOnFrameworkMpServicesThunk.c
@@ -899,6 +899,7 @@ CheckThisAP (
   CPU_STATE       CpuState;\r
 \r
   ASSERT (ProcessorNumber < mNumberOfProcessors);\r
+  ASSERT (ProcessorNumber < MAX_CPU_NUMBER);\r
 \r
   CpuData = &mMPSystemData.CpuData[ProcessorNumber];\r
 \r
@@ -1220,6 +1221,7 @@ WakeUpAp (
   EFI_PROCESSOR_INFORMATION    ProcessorInfoBuffer;\r
 \r
   ASSERT (ProcessorNumber < mNumberOfProcessors);\r
+  ASSERT (ProcessorNumber < MAX_CPU_NUMBER);\r
 \r
   CpuData = &mMPSystemData.CpuData[ProcessorNumber];\r
 \r
@@ -1300,6 +1302,7 @@ ChangeCpuState (
   CPU_DATA_BLOCK  *CpuData;\r
 \r
   ASSERT (ProcessorNumber < mNumberOfProcessors);\r
+  ASSERT (ProcessorNumber < MAX_CPU_NUMBER);\r
 \r
   CpuData = &mMPSystemData.CpuData[ProcessorNumber];\r
 \r