MdeModulePkg/DisplayEngine: Add Debug message to show mismatch menu info

[mirror_edk2.git] / MdeModulePkg / Universal / EsrtDxe / EsrtImpl.c

diff --git a/MdeModulePkg/Universal/EsrtDxe/EsrtImpl.c b/MdeModulePkg/Universal/EsrtDxe/EsrtImpl.c
index 1ed0250090588b6d1905df0fdde15d7ff46fd2c2..fff17b98fa3debc6872783dcc06e35639836cf69 100644 (file)
--- a/MdeModulePkg/Universal/EsrtDxe/EsrtImpl.c
+++ b/MdeModulePkg/Universal/EsrtDxe/EsrtImpl.c
@@ -2,13 +2,7 @@
   Esrt management implementation.\r
 \r
 Copyright (c) 2015 - 2018, Intel Corporation. All rights reserved.<BR>\r
-This program and the accompanying materials\r
-are licensed and made available under the terms and conditions of the BSD License\r
-which accompanies this distribution.  The full text of the license may be found at\r
-http://opensource.org/licenses/bsd-license.php\r
-\r
-THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 \r
@@ -245,6 +239,11 @@ DeleteEsrtEntry(
     goto EXIT;\r
   }\r
 \r
+  if (EsrtRepository == NULL) {\r
+    Status = EFI_OUT_OF_RESOURCES;\r
+    goto EXIT;\r
+  }\r
+\r
   if ((RepositorySize % sizeof(EFI_SYSTEM_RESOURCE_ENTRY)) != 0) {\r
     DEBUG((EFI_D_ERROR, "Repository Corrupt. Need to rebuild Repository.\n"));\r
     //\r
@@ -338,6 +337,11 @@ UpdateEsrtEntry(
              &RepositorySize\r
              );\r
 \r
+  if (EsrtRepository == NULL) {\r
+    Status = EFI_OUT_OF_RESOURCES;\r
+    goto EXIT;\r
+  }\r
+\r
   if (!EFI_ERROR(Status)) {\r
     //\r
     // if exist, update Esrt cache repository\r