From f0d711eb1b73f04ab3be042c984731c0edfe6b28 Mon Sep 17 00:00:00 2001
From: niruiyu <niruiyu@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Tue, 15 Nov 2011 01:48:13 +0000
Subject: Do not call LocateProtocol when the current TPL is above or equal
 TPL_NOTIFY.

Signed-off-by: niruiyu
Reviewed-by: mdkinney

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12696 6f19259b-4bc3-4df7-8a09-765794883524
---
 .../DxeReportStatusCodeLib/ReportStatusCodeLib.c   | 24 ++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

(limited to 'MdeModulePkg/Library/DxeReportStatusCodeLib')

diff --git a/MdeModulePkg/Library/DxeReportStatusCodeLib/ReportStatusCodeLib.c b/MdeModulePkg/Library/DxeReportStatusCodeLib/ReportStatusCodeLib.c
index ce843cc..4ec3a6c 100644
--- a/MdeModulePkg/Library/DxeReportStatusCodeLib/ReportStatusCodeLib.c
+++ b/MdeModulePkg/Library/DxeReportStatusCodeLib/ReportStatusCodeLib.c
@@ -1,7 +1,7 @@
 /** @file
   Report Status Code Library for DXE Phase.
 
-  Copyright (c) 2006 - 2010, Intel Corporation. All rights reserved.<BR>
+  Copyright (c) 2006 - 2011, Intel Corporation. All rights reserved.<BR>
   This program and the accompanying materials
   are licensed and made available under the terms and conditions of the BSD License
   which accompanies this distribution.  The full text of the license may be found at
@@ -45,6 +45,7 @@ InternalGetReportStatusCode (
   )
 {
   EFI_STATUS  Status;
+  EFI_TPL     Tpl;
 
   if (mReportStatusCodeLibStatusCodeProtocol != NULL) {
     return;
@@ -53,10 +54,18 @@ InternalGetReportStatusCode (
   //
   // Check gBS just in case ReportStatusCode is called before gBS is initialized.
   //
-  if (gBS != NULL && gBS->LocateProtocol != NULL) {
-    Status = gBS->LocateProtocol (&gEfiStatusCodeRuntimeProtocolGuid, NULL, (VOID**) &mReportStatusCodeLibStatusCodeProtocol);
-    if (EFI_ERROR (Status)) {
-      mReportStatusCodeLibStatusCodeProtocol = NULL;
+  if ((gBS != NULL) && (gBS->LocateProtocol != NULL) && (gBS->RaiseTPL != NULL) && (gBS->RestoreTPL != NULL)) {
+    //
+    // Retrieve the current TPL
+    //
+    Tpl = gBS->RaiseTPL (TPL_HIGH_LEVEL);
+    gBS->RestoreTPL (Tpl);
+
+    if (Tpl < TPL_NOTIFY) {
+      Status = gBS->LocateProtocol (&gEfiStatusCodeRuntimeProtocolGuid, NULL, (VOID**) &mReportStatusCodeLibStatusCodeProtocol);
+      if (EFI_ERROR (Status)) {
+        mReportStatusCodeLibStatusCodeProtocol = NULL;
+      }
     }
   }
 }
@@ -496,7 +505,10 @@ ReportStatusCodeEx (
   ASSERT (!((ExtendedData == NULL) && (ExtendedDataSize != 0)));
   ASSERT (!((ExtendedData != NULL) && (ExtendedDataSize == 0)));
 
-  if (gBS == NULL || gBS->AllocatePool == NULL || gBS->FreePool == NULL) {
+  if ((gBS == NULL) ||
+      (gBS->AllocatePool == NULL) || (gBS->FreePool == NULL) ||
+      (gBS->RaiseTPL == NULL) || (gBS->RestoreTPL == NULL)
+     ) {
     return EFI_UNSUPPORTED;
   }
 
-- 
cgit v1.1