From 07d97279a370e6e66dc1ae9b848fbd5f4eca202e Mon Sep 17 00:00:00 2001
From: lgao4 <lgao4@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Tue, 10 Mar 2009 07:54:11 +0000
Subject: Add more checks for Dxe Report status code library to access boot
 service.

Add measure to GPT boot table.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@7846 6f19259b-4bc3-4df7-8a09-765794883524
---
 .../Library/DxeReportStatusCodeLibFramework/DxeSupport.c              | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'IntelFrameworkModulePkg/Library/DxeReportStatusCodeLibFramework')

diff --git a/IntelFrameworkModulePkg/Library/DxeReportStatusCodeLibFramework/DxeSupport.c b/IntelFrameworkModulePkg/Library/DxeReportStatusCodeLibFramework/DxeSupport.c
index 42f7699e22..602e42b182 100644
--- a/IntelFrameworkModulePkg/Library/DxeReportStatusCodeLibFramework/DxeSupport.c
+++ b/IntelFrameworkModulePkg/Library/DxeReportStatusCodeLibFramework/DxeSupport.c
@@ -30,7 +30,7 @@ InternalGetReportStatusCode (
 
   if (gRT->Hdr.Revision < 0x20000) {
     return ((FRAMEWORK_EFI_RUNTIME_SERVICES*)gRT)->ReportStatusCode;
-  } else if (gBS != NULL) {
+  } else if (gBS != NULL && gBS->LocateProtocol != NULL) {
     Status = gBS->LocateProtocol (&gEfiStatusCodeRuntimeProtocolGuid, NULL, (VOID**)&StatusCodeProtocol);
     if (!EFI_ERROR (Status) && StatusCodeProtocol != NULL) {
       return StatusCodeProtocol->ReportStatusCode;
@@ -99,7 +99,7 @@ InternalReportStatusCodeEx (
   ASSERT (!((ExtendedData == NULL) && (ExtendedDataSize != 0)));
   ASSERT (!((ExtendedData != NULL) && (ExtendedDataSize == 0)));
 
-  if (gBS == NULL) {
+  if (gBS == NULL || gBS->AllocatePool == NULL || gBS->FreePool == NULL) {
     return EFI_UNSUPPORTED;
   }
 
-- 
cgit v1.2.3