diff --git a/EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c b/EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c
index 51240bbdc7..e470da0af0 100644
--- 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;