MdeModulePkg: Put report status code at the end of RuntimeDriverSetVirtualAddressMap.

[mirror_edk2.git] / MdeModulePkg / Core / RuntimeDxe / Runtime.c

diff --git a/MdeModulePkg/Core/RuntimeDxe/Runtime.c b/MdeModulePkg/Core/RuntimeDxe/Runtime.c
index 66c02b2f5b79716930b4e3ca12f2b9763824b18b..8e4c75691273b4fce498a590e0d144715449623f 100644 (file)
--- a/MdeModulePkg/Core/RuntimeDxe/Runtime.c
+++ b/MdeModulePkg/Core/RuntimeDxe/Runtime.c
@@ -35,8 +35,8 @@ Revision History:
   Table now contains an item named CalculateCrc32.\r
 \r
 \r
-Copyright (c) 2006 - 2008, Intel Corporation. <BR>\r
-All rights reserved. This program and the accompanying materials\r
+Copyright (c) 2006 - 2015, 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
@@ -275,11 +275,9 @@ RuntimeDriverSetVirtualAddressMap (
   mVirtualMap               = VirtualMap;\r
 \r
   //\r
-  // Currently the bug in StatusCode/RuntimeLib has been fixed, it will\r
-  // check whether in Runtime or not (this is judged by looking at\r
-  // mEfiAtRuntime global So this ReportStatusCode will work\r
+  // ReporstStatusCodeLib will check and make sure this service can be called in runtime mode.\r
   //\r
-  REPORT_STATUS_CODE (EFI_PROGRESS_CODE, PcdGet32 (PcdStatusCodeValueSetVirtualAddressMap));\r
+  REPORT_STATUS_CODE (EFI_PROGRESS_CODE, (EFI_SOFTWARE_EFI_RUNTIME_SERVICE | EFI_SW_RS_PC_SET_VIRTUAL_ADDRESS_MAP));\r
 \r
   //\r
   // Signal all the EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE events.\r
@@ -359,6 +357,10 @@ RuntimeDriverSetVirtualAddressMap (
   //\r
   mVirtualMap = NULL;\r
 \r
+  //\r
+  // Report Status Code here since EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE event was signalled.\r
+  //\r
+  REPORT_STATUS_CODE (EFI_PROGRESS_CODE, (EFI_SOFTWARE_DXE_BS_DRIVER | EFI_SW_DXE_BS_PC_VIRTUAL_ADDRESS_CHANGE_EVENT));\r
   return EFI_SUCCESS;\r
 }\r
 \r