1 files changed, 17 insertions, 20 deletions
diff --git a/MdePkg/Library/UefiMemoryLib/ScanMem8Wrapper.c b/MdePkg/Library/UefiMemoryLib/ScanMem8Wrapper.c
index 9b817b1b0e..72a8078b81 100644
--- a/MdePkg/Library/UefiMemoryLib/ScanMem8Wrapper.c
+++ b/MdePkg/Library/UefiMemoryLib/ScanMem8Wrapper.c
@@ -26,24 +26,21 @@
 #include "MemLibInternals.h"
 
 /**
-  Scans a target buffer for an 8-bit value, and returns a pointer to the
-  matching 8-bit value in the target buffer.
+  Scans a target buffer for an 8-bit value, and returns a pointer to the matching 8-bit value
+  in the target buffer.
 
-  This function searches target the buffer specified by Buffer and Length from
-  the lowest address to the highest address for an 8-bit value that matches
-  Value. If a match is found, then a pointer to the matching byte in the target
-  buffer is returned. If no match is found, then NULL is returned. If Length is
-  0, then NULL is returned.
+  This function searches target the buffer specified by Buffer and Length from the lowest
+  address to the highest address for an 8-bit value that matches Value.  If a match is found,
+  then a pointer to the matching byte in the target buffer is returned.  If no match is found,
+  then NULL is returned.  If Length is 0, then NULL is returned.
+  If Length > 0 and Buffer is NULL, then ASSERT().
+  If Length is greater than (MAX_ADDRESS � Buffer + 1), then ASSERT(). 
 
-  If Buffer is NULL, then ASSERT().
-  If Length is greater than (MAX_ADDRESS - Buffer + 1), then ASSERT().
+  @param  Buffer      Pointer to the target buffer to scan.
+  @param  Length      Number of bytes in Buffer to scan.
+  @param  Value       Value to search for in the target buffer.
 
-  @param  Buffer  Pointer to the target buffer to scan.
-  @param  Length  Number of bytes in Buffer to scan.
-  @param  Value   Value to search for in the target buffer.
-
-  @return Pointer to the first occurrence or NULL if not found.
-  @retval NULL  if Length == 0 or Value was not found.
+  @return A pointer to the matching byte in the target buffer or NULL otherwise.
 
 **/
 VOID *
@@ -54,11 +51,11 @@ ScanMem8 (
   IN      UINT8                     Value
   )
 {
-  ASSERT (Buffer != NULL);
-  ASSERT (Length <= MAX_ADDRESS + (UINTN)Buffer + 1);
-
-  if ((Length /= sizeof (Value)) == 0) {
+  if (Length == 0) {
     return NULL;
   }
-  return (VOID*)InternalMemScanMem8 (Buffer, Length, Value);
+  ASSERT (Buffer != NULL);
+  ASSERT ((Length - 1) <= (MAX_ADDRESS - (UINTN)Buffer));
+ 
+  return (VOID*)InternalMemScanMem8 (Buffer, Length / sizeof (Value), Value);
 }