Add in ASSERT to check out-of-bound and possible dereference of NULL pointers.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@7391 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 2 insertions, 1 deletions

diff --git a/MdeModulePkg/Universal/SetupBrowserDxe/InputHandler.c b/MdeModulePkg/Universal/SetupBrowserDxe/InputHandler.c
index d122aec569..1bf7704343 100644
--- a/MdeModulePkg/Universal/SetupBrowserDxe/InputHandler.c
+++ b/MdeModulePkg/Universal/SetupBrowserDxe/InputHandler.c
@@ -387,7 +387,7 @@ GetNumericInput (
 

     InputText[0] = LEFT_NUMERIC_DELIMITER;

     SetUnicodeMem (InputText + 1, InputWidth, L' ');

-    ASSERT (InputWidth < MAX_NUMERIC_INPUT_WIDTH); 

+    ASSERT (InputWidth + 2 < MAX_NUMERIC_INPUT_WIDTH); 

     InputText[InputWidth + 1] = RIGHT_NUMERIC_DELIMITER;

     InputText[InputWidth + 2] = L'\0';

 

@@ -649,6 +649,7 @@ EnterCarriageReturn:
         }

 

         Count++;

+        ASSERT (Count < (sizeof (PreviousNumber) / sizeof (PreviousNumber[0])));

         PreviousNumber[Count] = EditValue;

 

         PrintCharAt (Column, Row, Key.UnicodeChar);