Import the UefiRuntimeLib in MdePkg.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2787 6f19259b-4bc3-4df7-8a09-765794883524
2007-06-27 07:22:18 +00:00
parent 7c6d55db88
commit ebcc8fb7ba
4 changed files with 720 additions and 0 deletions
--- a/MdePkg/Library/UefiRuntimeLib/RuntimeLib.c
+++ b/MdePkg/Library/UefiRuntimeLib/RuntimeLib.c
@@ -0,0 +1,173 @@
+/**@file
+  Library utility functions for Runtime driver.
+
+Copyright (c) 2006 Intel Corporation. <BR>
+All rights reserved. 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
+http://opensource.org/licenses/bsd-license.php
+
+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+
+**/
+
+#include <RuntimeLibInternal.h>
+
+///
+/// Driver Lib Module Globals
+///
+
+STATIC EFI_EVENT              mEfiVirtualNotifyEvent;
+STATIC BOOLEAN                mEfiGoneVirtual         = FALSE;
+STATIC BOOLEAN                mEfiAtRuntime           = FALSE;
+EFI_RUNTIME_SERVICES          *mRT;
+
+/**
+  Set AtRuntime flag as TRUE after ExitBootServices
+
+  @param[in]  Event   The Event that is being processed
+  @param[in]  Context Event Context
+**/
+VOID
+EFIAPI
+RuntimeDriverExitBootServices (
+  IN EFI_EVENT        Event,
+  IN VOID             *Context
+  )
+{
+  //
+  // Clear out BootService globals
+  //
+  gBS             = NULL;
+
+  mEfiAtRuntime = TRUE;
+}
+
+/**
+  Fixup internal data so that EFI can be call in virtual mode.
+  Call the passed in Child Notify event and convert any pointers in
+  lib to virtual mode.
+
+  @param[in]    Event   The Event that is being processed
+  @param[in]    Context Event Context
+**/
+STATIC
+VOID
+EFIAPI
+RuntimeLibVirtualNotifyEvent (
+  IN EFI_EVENT        Event,
+  IN VOID             *Context
+  )
+{
+  UINTN Index;
+  EFI_EVENT_NOTIFY  ChildNotifyEventHandler;
+
+  for (Index = 0;
+       _gDriverSetVirtualAddressMapEvent[Index] != NULL;
+       Index++) {
+    ChildNotifyEventHandler = _gDriverSetVirtualAddressMapEvent[Index];
+    ChildNotifyEventHandler (Event, NULL);
+  }
+
+  //
+  // Update global for Runtime Services Table and IO
+  //
+  EfiConvertPointer (0, (VOID **) &mRT);
+
+  mEfiGoneVirtual = TRUE;
+}
+
+/**
+  Intialize runtime Driver Lib if it has not yet been initialized.
+
+  @param[in]  ImageHandle   The firmware allocated handle for the EFI image.
+  @param[in]  SystemTable   A pointer to the EFI System Table.
+
+  @return     EFI_STATUS    always returns EFI_SUCCESS except EFI_ALREADY_STARTED if already started.
+**/
+EFI_STATUS
+EFIAPI
+RuntimeDriverLibConstruct (
+  IN EFI_HANDLE           ImageHandle,
+  IN EFI_SYSTEM_TABLE     *SystemTable
+  )
+{
+  EFI_STATUS  Status;
+
+  ASSERT (SystemTable != NULL);
+  mRT = SystemTable->RuntimeServices;
+  ASSERT (mRT != NULL);
+  
+  //
+  // Register SetVirtualAddressMap () notify function
+  //
+  ASSERT (gBS != NULL);
+  Status = gBS->CreateEvent (
+                  EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE,
+                  TPL_NOTIFY,
+                  RuntimeLibVirtualNotifyEvent,
+                  NULL,
+                  &mEfiVirtualNotifyEvent
+                  );
+
+  ASSERT_EFI_ERROR (Status);
+
+  return Status;
+}
+
+/**
+  This routine will free some resources which have been allocated in
+  EfiInitializeRuntimeDriverLib(). If a runtime driver exits with an error,
+  it must call this routine to free the allocated resource before the exiting.
+
+  @retval     EFI_SUCCESS       Shutdown the Runtime Driver Lib successfully
+  @retval     EFI_UNSUPPORTED   Runtime Driver lib was not initialized at all
+**/
+EFI_STATUS
+EFIAPI
+RuntimeDriverLibDeconstruct (
+  IN EFI_HANDLE        ImageHandle,
+  IN EFI_SYSTEM_TABLE  *SystemTable
+  )
+{
+  EFI_STATUS  Status;
+
+  //
+  // Close SetVirtualAddressMap () notify function
+  //
+  ASSERT (gBS != NULL);
+  Status = gBS->CloseEvent (mEfiVirtualNotifyEvent);
+  ASSERT_EFI_ERROR (Status);
+
+  return Status;
+}
+
+/**
+  Return TRUE if ExitBootServices () has been called
+
+  @retval TRUE If ExitBootServices () has been called
+**/
+BOOLEAN
+EFIAPI
+EfiAtRuntime (
+  VOID
+  )
+{
+  return mEfiAtRuntime;
+}
+
+/**
+  Return TRUE if SetVirtualAddressMap () has been called
+
+  @retval TRUE  If SetVirtualAddressMap () has been called
+**/
+BOOLEAN
+EFIAPI
+EfiGoneVirtual (
+  VOID
+  )
+{
+  return mEfiGoneVirtual;
+}
+