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

1 files changed, 3 insertions, 0 deletions

diff --git a/EdkCompatibilityPkg/Compatibility/MpServicesOnFrameworkMpServicesThunk/MpServicesOnFrameworkMpServicesThunk.c b/EdkCompatibilityPkg/Compatibility/MpServicesOnFrameworkMpServicesThunk/MpServicesOnFrameworkMpServicesThunk.c
index b31d17c47f..9005670bae 100644
--- a/EdkCompatibilityPkg/Compatibility/MpServicesOnFrameworkMpServicesThunk/MpServicesOnFrameworkMpServicesThunk.c
+++ b/EdkCompatibilityPkg/Compatibility/MpServicesOnFrameworkMpServicesThunk/MpServicesOnFrameworkMpServicesThunk.c
@@ -899,6 +899,7 @@ CheckThisAP (
   CPU_STATE       CpuState;

 

   ASSERT (ProcessorNumber < mNumberOfProcessors);

+  ASSERT (ProcessorNumber < MAX_CPU_NUMBER);

 

   CpuData = &mMPSystemData.CpuData[ProcessorNumber];

 

@@ -1220,6 +1221,7 @@ WakeUpAp (
   EFI_PROCESSOR_INFORMATION    ProcessorInfoBuffer;

 

   ASSERT (ProcessorNumber < mNumberOfProcessors);

+  ASSERT (ProcessorNumber < MAX_CPU_NUMBER);

 

   CpuData = &mMPSystemData.CpuData[ProcessorNumber];

 

@@ -1300,6 +1302,7 @@ ChangeCpuState (
   CPU_DATA_BLOCK  *CpuData;

 

   ASSERT (ProcessorNumber < mNumberOfProcessors);

+  ASSERT (ProcessorNumber < MAX_CPU_NUMBER);

 

   CpuData = &mMPSystemData.CpuData[ProcessorNumber];