BaseMemoryLib:

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

diff --git a/MdePkg/Library/BaseMemoryLib/SetMem64Wrapper.c b/MdePkg/Library/BaseMemoryLib/SetMem64Wrapper.c
index 846d1c2c33a1920a104f876afd922015693cb9f6..0128a2d6f13fe2b6057c37edbcd60fa4f8a9cac5 100644 (file)
--- a/MdePkg/Library/BaseMemoryLib/SetMem64Wrapper.c
+++ b/MdePkg/Library/BaseMemoryLib/SetMem64Wrapper.c
@@ -23,7 +23,7 @@
 \r
 **/\r
 \r
-#include "MemLibWrappers.h"\r
+#include "MemLibInternals.h"\r
 \r
 /**\r
   Fills a target buffer with a 64-bit value, and returns the target buffer.\r
@@ -52,10 +52,10 @@ SetMem64 (
   IN      UINT64                    Value\r
   )\r
 {\r
-  ASSERT (Buffer != NULL);\r
+  ASSERT (!(Buffer == NULL && Length > 0));\r
   ASSERT (Length <= MAX_ADDRESS - (UINTN)Buffer + 1);\r
-  ASSERT ((((UINTN)Buffer) & 1) != 0);\r
-  ASSERT ((Length & 1) != 0);\r
+  ASSERT ((((UINTN)Buffer) & (sizeof (Value) - 1)) == 0);\r
+  ASSERT ((Length & (sizeof (Value) - 1)) == 0);\r
 \r
   if ((Length /= sizeof (Value)) == 0) {\r
     return Buffer;\r