git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@520 6f19259b-4bc3-4df7-8a09-765794883524

2006-06-15 02:54:27 +00:00
parent 508daae71b
commit 1ee3e26b94
6 changed files with 24 additions and 6 deletions
--- a/MdePkg/Library/BaseMemoryLibRepStr/SetMemWrapper.c
+++ b/MdePkg/Library/BaseMemoryLibRepStr/SetMemWrapper.c
@@ -47,6 +47,9 @@ SetMem (
  IN      UINT8                     Value
  )
 {
-  ASSERT (Size <= MAX_ADDRESS - (UINTN)Buffer + 1);
+  if (Size == 0) {
+    return Buffer;
+  }
+  ASSERT (Size - 1 <= MAX_ADDRESS - (UINTN)Buffer);
  return InternalMemSetMem (Buffer, Size, Value);
 }