Add parameter check in SmmBase.Communicate().

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10289 6f19259b-4bc3-4df7-8a09-765794883524
2010-03-19 08:39:40 +00:00
parent 3b1464d5ac
commit 1329da4417
1 changed files with 4 additions and 0 deletions
--- a/EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c
+++ b/EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c
@@ -202,6 +202,10 @@ SmmBaseCommunicate (
  /// Note this is a runtime interface
  ///

+  if (CommunicationBuffer == NULL || BufferSize == NULL) {
+    return EFI_INVALID_PARAMETER;
+  }
+
  mCommunicationData.FunctionData.Function = SmmBaseFunctionCommunicate;
  mCommunicationData.FunctionData.Args.Communicate.ImageHandle = ImageHandle;
  mCommunicationData.FunctionData.Args.Communicate.CommunicationBuffer = CommunicationBuffer;