MdePkg/Library/UefiRuntimeLib/RuntimeLib.c

   1 /**@file
   2   Library utility functions for Runtime driver.
   3
   4 Copyright (c) 2006 Intel Corporation. <BR>
   5 All rights reserved. This program and the accompanying materials
   6 are licensed and made available under the terms and conditions of the BSD License
   7 which accompanies this distribution.  The full text of the license may be found at
   8 http://opensource.org/licenses/bsd-license.php
   9
  10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  12
  13 **/
  14
  15 #include <RuntimeLibInternal.h>
  16
  17 ///
  18 /// Driver Lib Module Globals
  19 ///
  20
  21 STATIC EFI_EVENT              mEfiVirtualNotifyEvent;
  22 STATIC EFI_EVENT              mEfiExitBootServicesEvent;
  23 STATIC BOOLEAN                mEfiGoneVirtual         = FALSE;
  24 STATIC BOOLEAN                mEfiAtRuntime           = FALSE;
  25 EFI_RUNTIME_SERVICES          *mRT;
  26
  27 /**
  28   Set AtRuntime flag as TRUE after ExitBootServices
  29
  30   @param[in]  Event   The Event that is being processed
  31   @param[in]  Context Event Context
  32 **/
  33 STATIC
  34 VOID
  35 EFIAPI
  36 RuntimeLibExitBootServicesEvent (
  37   IN EFI_EVENT        Event,
  38   IN VOID             *Context
  39   )
  40 {
  41   //
  42   // Clear out BootService globals
  43   //
  44   gBS             = NULL;
  45
  46   mEfiAtRuntime = TRUE;
  47 }
  48
  49 /**
  50   Fixup internal data so that EFI can be call in virtual mode.
  51   Call the passed in Child Notify event and convert any pointers in
  52   lib to virtual mode.
  53
  54   @param[in]    Event   The Event that is being processed
  55   @param[in]    Context Event Context
  56 **/
  57 STATIC
  58 VOID
  59 EFIAPI
  60 RuntimeLibVirtualNotifyEvent (
  61   IN EFI_EVENT        Event,
  62   IN VOID             *Context
  63   )
  64 {
  65   //
  66   // Update global for Runtime Services Table and IO
  67   //
  68   EfiConvertPointer (0, (VOID **) &mRT);
  69
  70   mEfiGoneVirtual = TRUE;
  71 }
  72
  73 /**
  74   Intialize runtime Driver Lib if it has not yet been initialized.
  75
  76   @param[in]  ImageHandle   The firmware allocated handle for the EFI image.
  77   @param[in]  SystemTable   A pointer to the EFI System Table.
  78
  79   @return     EFI_STATUS    always returns EFI_SUCCESS except EFI_ALREADY_STARTED if already started.
  80 **/
  81 EFI_STATUS
  82 EFIAPI
  83 RuntimeDriverLibConstruct (
  84   IN EFI_HANDLE           ImageHandle,
  85   IN EFI_SYSTEM_TABLE     *SystemTable
  86   )
  87 {
  88   EFI_STATUS  Status;
  89
  90   ASSERT (SystemTable != NULL);
  91   mRT = SystemTable->RuntimeServices;
  92   ASSERT (mRT != NULL);
  93
  94   //
  95   // Register SetVirtualAddressMap () notify function
  96   //
  97   ASSERT (gBS != NULL);
  98   Status = gBS->CreateEvent (
  99                   EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE,
 100                   TPL_NOTIFY,
 101                   RuntimeLibVirtualNotifyEvent,
 102                   NULL,
 103                   &mEfiVirtualNotifyEvent
 104                   );
 105
 106   ASSERT_EFI_ERROR (Status);
 107
 108   Status = gBS->CreateEvent (
 109                   EVT_SIGNAL_EXIT_BOOT_SERVICES,
 110                   TPL_NOTIFY,
 111                   RuntimeLibExitBootServicesEvent,
 112                   NULL,
 113                   &mEfiExitBootServicesEvent
 114                   );
 115
 116   ASSERT_EFI_ERROR (Status);
 117
 118   return Status;
 119 }
 120
 121 /**
 122   This routine will free some resources which have been allocated in
 123   EfiInitializeRuntimeDriverLib(). If a runtime driver exits with an error,
 124   it must call this routine to free the allocated resource before the exiting.
 125
 126   @retval     EFI_SUCCESS       Shutdown the Runtime Driver Lib successfully
 127   @retval     EFI_UNSUPPORTED   Runtime Driver lib was not initialized at all
 128 **/
 129 EFI_STATUS
 130 EFIAPI
 131 RuntimeDriverLibDeconstruct (
 132   IN EFI_HANDLE        ImageHandle,
 133   IN EFI_SYSTEM_TABLE  *SystemTable
 134   )
 135 {
 136   EFI_STATUS  Status;
 137
 138   //
 139   // Close SetVirtualAddressMap () notify function
 140   //
 141   ASSERT (gBS != NULL);
 142   Status = gBS->CloseEvent (mEfiVirtualNotifyEvent);
 143   ASSERT_EFI_ERROR (Status);
 144
 145   Status = gBS->CloseEvent (mEfiExitBootServicesEvent);
 146   ASSERT_EFI_ERROR (Status);
 147
 148   return Status;
 149 }
 150
 151 /**
 152   Return TRUE if ExitBootServices () has been called
 153
 154   @retval TRUE If ExitBootServices () has been called
 155 **/
 156 BOOLEAN
 157 EFIAPI
 158 EfiAtRuntime (
 159   VOID
 160   )
 161 {
 162   return mEfiAtRuntime;
 163 }
 164
 165 /**
 166   Return TRUE if SetVirtualAddressMap () has been called
 167
 168   @retval TRUE  If SetVirtualAddressMap () has been called
 169 **/
 170 BOOLEAN
 171 EFIAPI
 172 EfiGoneVirtual (
 173   VOID
 174   )
 175 {
 176   return mEfiGoneVirtual;
 177 }
 178