git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@520 6f19259b...

[mirror_edk2.git] / MdePkg / Library / BaseMemoryLib / SetMemWrapper.c

diff --git a/MdePkg/Library/BaseMemoryLib/SetMemWrapper.c b/MdePkg/Library/BaseMemoryLib/SetMemWrapper.c
index 255b84d836006148d155b5dabaa8f7b567a7d873..0600dc289fb45c4496d3a16d7f56a3d33631d398 100644 (file)
--- a/MdePkg/Library/BaseMemoryLib/SetMemWrapper.c
+++ b/MdePkg/Library/BaseMemoryLib/SetMemWrapper.c
@@ -23,7 +23,7 @@
 \r
 **/\r
 \r
-#include "MemLibWrappers.h"\r
+#include "MemLibInternals.h"\r
 \r
 /**\r
   Set Buffer to Value for Size bytes.\r
@@ -47,6 +47,9 @@ SetMem (
   IN      UINT8                     Value\r
   )\r
 {\r
-  ASSERT (Size <= MAX_ADDRESS - (UINTN)Buffer + 1);\r
+  if (Size == 0) {\r
+    return Buffer;\r
+  }\r
+  ASSERT (Size - 1 <= MAX_ADDRESS - (UINTN)Buffer);\r
   return InternalMemSetMem (Buffer, Size, Value);\r
 }\r