X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=MdePkg%2FLibrary%2FBaseMemoryLib%2FCopyMemWrapper.c;h=5ca62f50478428426398688c405bb399231eaa7b;hp=155877fdde2b9b55665382eba605959e3785e662;hb=24de7645e857287d096b058f7a64fb7f618c0194;hpb=608fd3fd6159e8bcbf6f55926fc805e175d168eb

diff --git a/MdePkg/Library/BaseMemoryLib/CopyMemWrapper.c b/MdePkg/Library/BaseMemoryLib/CopyMemWrapper.c
index 155877fdde..5ca62f5047 100644
--- a/MdePkg/Library/BaseMemoryLib/CopyMemWrapper.c
+++ b/MdePkg/Library/BaseMemoryLib/CopyMemWrapper.c
@@ -23,7 +23,7 @@
 
 **/
 
-#include "MemLibWrappers.h"
+#include "MemLibInternals.h"
 
 /**
   Copy Length bytes from Source to Destination.
@@ -51,8 +51,14 @@ CopyMem (
   IN      UINTN                     Length
   )
 {
-  ASSERT (Length <= MAX_ADDRESS - (UINTN)Destination + 1);
-  ASSERT (Length <= MAX_ADDRESS - (UINTN)Source + 1);
+  ASSERT (
+    Destination == NULL ||
+    Length <= MAX_ADDRESS - (UINTN)Destination + 1
+    );
+  ASSERT (
+    Source == NULL ||
+    Length <= MAX_ADDRESS - (UINTN)Source + 1
+    );
   if (Destination == Source || Length == 0) {
     return Destination;
   }