From: lgao4 <lgao4@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Thu, 22 Jan 2009 14:09:21 +0000 (+0000)
Subject: Add check after allocate pool.
X-Git-Tag: edk2-stable201903~18902
X-Git-Url: https://git.proxmox.com/?a=commitdiff_plain;h=80882266edab7e0f88684c436ed97d773034b480;hp=60c65d37e2437d6a8bbf4bb1d31751e14650c6fa;p=mirror_edk2.git

Add check after allocate pool.

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

diff --git a/MdeModulePkg/Universal/PlatformDriverOverride/PlatOverMngr/PlatOverMngr.c b/MdeModulePkg/Universal/PlatformDriverOverride/PlatOverMngr/PlatOverMngr.c
index a60e5576de..a7e5df4f53 100644
--- a/MdeModulePkg/Universal/PlatformDriverOverride/PlatOverMngr/PlatOverMngr.c
+++ b/MdeModulePkg/Universal/PlatformDriverOverride/PlatOverMngr/PlatOverMngr.c
@@ -789,6 +789,7 @@ UpdateBindingDriverSelectPage (
     DriverName = DevicePathToStr (LoadedImageDevicePath);
 
     NewString = AllocateZeroPool (StrSize (DriverName));
+    ASSERT (NewString != NULL); 
     StrCat (NewString, DriverName);
     NewStringHelpToken = mDriverImageFilePathToken[Index];
     if (NewStringHelpToken == 0) {
@@ -1227,7 +1228,7 @@ PlatOverMngrCallback (
     ASSERT_EFI_ERROR (Status);
   }
 
-  if (((KEY_VALUE_DEVICE_OFFSET <= KeyValue) && (KeyValue < KEY_VALUE_DEVICE_MAX)) || (KeyValue == KEY_VALUE_ORDER_GOTO_PREVIOUS)) {
+  if (((KeyValue >= KEY_VALUE_DEVICE_OFFSET) && (KeyValue < KEY_VALUE_DEVICE_MAX)) || (KeyValue == KEY_VALUE_ORDER_GOTO_PREVIOUS)) {
     if (KeyValue == KEY_VALUE_ORDER_GOTO_PREVIOUS) {
       KeyValue = (EFI_QUESTION_ID) (mSelectedCtrIndex + KEY_VALUE_DEVICE_OFFSET);
     }