From 5fb01149ef1cecc36b261094b304ff0654bb7ae3 Mon Sep 17 00:00:00 2001
From: kavsrf <kavsrf@gmail.com>
Date: Sat, 17 Sep 2016 01:00:44 +0300
Subject: DcsRe select volume is added to boot from usb

Notes (7.2, 7.4)
---
 Library/CommonLib/EfiBio.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'Library/CommonLib')

diff --git a/Library/CommonLib/EfiBio.c b/Library/CommonLib/EfiBio.c
index 9577c08..e6509f9 100644
--- a/Library/CommonLib/EfiBio.c
+++ b/Library/CommonLib/EfiBio.c
@@ -71,6 +71,7 @@ EfiGetHandles(
 {
    EFI_STATUS res = EFI_BUFFER_TOO_SMALL;
    UINTN      BufferSize;
+	if ((Buffer == NULL) || (Count == NULL)) return EFI_INVALID_PARAMETER;
    if(*Buffer != NULL) MEM_FREE(*Buffer);
    *Count = 0;
    *Buffer = (EFI_HANDLE*) MEM_ALLOC(sizeof(EFI_HANDLE));
@@ -80,7 +81,7 @@ EfiGetHandles(
       if (res == RETURN_BUFFER_TOO_SMALL) {
          MEM_FREE(*Buffer);
          *Buffer = (EFI_HANDLE*)MEM_ALLOC(BufferSize);
-         if (!Buffer) {
+         if (*Buffer == NULL) {
             return EFI_OUT_OF_RESOURCES;
          }
          res = gBS->LocateHandle(SearchType, Protocol, SearchKey, &BufferSize, *Buffer);
-- 
cgit v1.2.3