Add parameter checking for MP Services Protocol Thunk driver.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10147 6f19259b-4bc3-4df7-8a09-765794883524

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