X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=MdePkg%2FLibrary%2FBaseMemoryLibRepStr%2FSetMemWrapper.c;h=0600dc289fb45c4496d3a16d7f56a3d33631d398;hp=294a27a75362e6b9286b93e30c695e4cd93c33ba;hb=1ee3e26b9474444603a36fc740702c80927ca8d1;hpb=508daae71b4c754321cb28e1fe7671a2ec1374f7

diff --git a/MdePkg/Library/BaseMemoryLibRepStr/SetMemWrapper.c b/MdePkg/Library/BaseMemoryLibRepStr/SetMemWrapper.c
index 294a27a753..0600dc289f 100644
--- 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);
 }