BaseMemoryLib: Add missing ASSERT()s for some interfaces.

[mirror_edk2.git] / MdePkg / Library / PeiMemoryLib / ZeroMemWrapper.c

diff --git a/MdePkg/Library/PeiMemoryLib/ZeroMemWrapper.c b/MdePkg/Library/PeiMemoryLib/ZeroMemWrapper.c
index d7d7919ba26a46aa5dabfb8ff71eb0eb9bb0271b..4a3d615ce50839c44ce64259ee4f2314193c74cc 100644 (file)
--- a/MdePkg/Library/PeiMemoryLib/ZeroMemWrapper.c
+++ b/MdePkg/Library/PeiMemoryLib/ZeroMemWrapper.c
@@ -43,9 +43,10 @@ VOID *
 EFIAPI\r
 ZeroMem (\r
   IN      VOID                      *Buffer,\r
-  IN      UINTN                     Size\r
+  IN      UINTN                     Length\r
   )\r
 {\r
-  ASSERT (Buffer != NULL);\r
-  return InternalMemSetMem (Buffer, Size, 0);\r
+  ASSERT (!(Buffer == NULL && Length > 0));\r
+  ASSERT (Length <= (MAX_ADDRESS - (UINTN)Buffer + 1));\r
+  return InternalMemSetMem (Buffer, Length, 0);\r
 }\r