BaseMemoryLib: Add missing ASSERT()s for some interfaces.

[mirror_edk2.git] / MdePkg / Library / BaseMemoryLibSse2 / MemLibGuid.c

diff --git a/MdePkg/Library/BaseMemoryLibSse2/MemLibGuid.c b/MdePkg/Library/BaseMemoryLibSse2/MemLibGuid.c
index 06b2721c64f407a6ce20c477c6005b8e9ce83f63..c54983c77b5436f21f95faf458bb3d8bb09aba7c 100644 (file)
--- a/MdePkg/Library/BaseMemoryLibSse2/MemLibGuid.c
+++ b/MdePkg/Library/BaseMemoryLibSse2/MemLibGuid.c
@@ -119,6 +119,14 @@ ScanGuid (
 {\r
   CONST GUID                        *GuidPtr;\r
 \r
+  ASSERT (Buffer != NULL);\r
+  //\r
+  // Make sure Buffer is aligned on a 64-bit boundary.\r
+  //\r
+  ASSERT (((UINTN) Buffer & 7) == 0);\r
+\r
+  ASSERT (Length <= (MAX_ADDRESS - (UINTN)Buffer + 1));\r
+\r
   GuidPtr = (GUID*)Buffer;\r
   Buffer = GuidPtr + Length / sizeof (*GuidPtr);\r
   while (GuidPtr < (CONST GUID*)Buffer) {\r