git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@8602 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			276 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			276 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
  Library constructor & destructor, event handlers, and other internal worker functions.
 | 
						|
 | 
						|
  Copyright (c) 2006 - 2009, 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 "ReportStatusCodeLibInternal.h"
 | 
						|
 | 
						|
EFI_EVENT               mVirtualAddressChangeEvent;
 | 
						|
EFI_EVENT               mExitBootServicesEvent;
 | 
						|
EFI_STATUS_CODE_DATA    *mStatusCodeData;
 | 
						|
BOOLEAN                 mInSmm;
 | 
						|
EFI_SMM_BASE_PROTOCOL   *mSmmBase;
 | 
						|
EFI_RUNTIME_SERVICES    *mInternalRT;
 | 
						|
BOOLEAN                 mHaveExitedBootServices = FALSE;
 | 
						|
EFI_REPORT_STATUS_CODE  mReportStatusCode = NULL;
 | 
						|
 | 
						|
/**
 | 
						|
  Locate the report status code service.
 | 
						|
 | 
						|
  In SMM, it retrieves OemHookStatusCodeReport() from customized OEM Hook Status Code Lib.
 | 
						|
  Otherwise, it first tries to retrieve ReportStatusCode() in Runtime Services Table.
 | 
						|
  If not found, it then tries to retrieve ReportStatusCode() API of Report Status Code Protocol.
 | 
						|
 | 
						|
  @return   Function pointer to the report status code service.
 | 
						|
            NULL is returned if no status code service is available.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_REPORT_STATUS_CODE
 | 
						|
InternalGetReportStatusCode (
 | 
						|
  VOID
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS_CODE_PROTOCOL  *StatusCodeProtocol;
 | 
						|
  EFI_STATUS                Status;
 | 
						|
 | 
						|
  if (mInSmm) {
 | 
						|
    return (EFI_REPORT_STATUS_CODE) OemHookStatusCodeReport;
 | 
						|
  } else if (mInternalRT != NULL && mInternalRT->Hdr.Revision < 0x20000) {
 | 
						|
    return ((FRAMEWORK_EFI_RUNTIME_SERVICES*)mInternalRT)->ReportStatusCode;
 | 
						|
  } else if (!mHaveExitedBootServices) {
 | 
						|
  	//
 | 
						|
  	// Check gBS just in case. ReportStatusCode is called before gBS is initialized.
 | 
						|
  	//
 | 
						|
    if (gBS != NULL) {
 | 
						|
      Status = gBS->LocateProtocol (&gEfiStatusCodeRuntimeProtocolGuid, NULL, (VOID**)&StatusCodeProtocol);
 | 
						|
      if (!EFI_ERROR (Status) && StatusCodeProtocol != NULL) {
 | 
						|
        return StatusCodeProtocol->ReportStatusCode;
 | 
						|
      }
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  return NULL;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
  Internal worker function that reports a status code through the status code service.
 | 
						|
 | 
						|
  If status code service is not cached, then this function checks if status code service is
 | 
						|
  available in system.  If status code service is not available, then EFI_UNSUPPORTED is
 | 
						|
  returned.  If status code service is present, then it is cached in mReportStatusCode.
 | 
						|
  Finally this function reports status code through the status code service.
 | 
						|
 | 
						|
  @param  Type              Status code type.
 | 
						|
  @param  Value             Status code value.
 | 
						|
  @param  Instance          Status code instance number.
 | 
						|
  @param  CallerId          Pointer to a GUID that identifies the caller of this
 | 
						|
                            function.  This is an optional parameter that may be
 | 
						|
                            NULL.
 | 
						|
  @param  Data              Pointer to the extended data buffer.  This is an
 | 
						|
                            optional parameter that may be NULL.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS       The status code was reported.
 | 
						|
  @retval EFI_UNSUPPORTED   Status code service is not available.
 | 
						|
  @retval EFI_UNSUPPORTED   Status code type is not supported.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
InternalReportStatusCode (
 | 
						|
  IN EFI_STATUS_CODE_TYPE     Type,
 | 
						|
  IN EFI_STATUS_CODE_VALUE    Value,
 | 
						|
  IN UINT32                   Instance,
 | 
						|
  IN CONST EFI_GUID           *CallerId OPTIONAL,
 | 
						|
  IN EFI_STATUS_CODE_DATA     *Data     OPTIONAL
 | 
						|
  )
 | 
						|
{
 | 
						|
  if ((ReportProgressCodeEnabled() && ((Type) & EFI_STATUS_CODE_TYPE_MASK) == EFI_PROGRESS_CODE) ||
 | 
						|
      (ReportErrorCodeEnabled() && ((Type) & EFI_STATUS_CODE_TYPE_MASK) == EFI_ERROR_CODE) ||
 | 
						|
      (ReportDebugCodeEnabled() && ((Type) & EFI_STATUS_CODE_TYPE_MASK) == EFI_DEBUG_CODE)) {
 | 
						|
    //
 | 
						|
    // If mReportStatusCode is NULL, then check if status code service is available in system.
 | 
						|
    //
 | 
						|
    if (mReportStatusCode == NULL) {
 | 
						|
      mReportStatusCode = InternalGetReportStatusCode ();
 | 
						|
      if (mReportStatusCode == NULL) {
 | 
						|
        return EFI_UNSUPPORTED;
 | 
						|
      }
 | 
						|
    }
 | 
						|
  
 | 
						|
    //
 | 
						|
    // A status code service is present in system, so pass in all the parameters to the service.
 | 
						|
    //
 | 
						|
    return (*mReportStatusCode) (Type, Value, Instance, (EFI_GUID *)CallerId, Data);
 | 
						|
  }
 | 
						|
  
 | 
						|
  return EFI_UNSUPPORTED;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
  Notification function of EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE.
 | 
						|
 | 
						|
  @param  Event        Event whose notification function is being invoked.
 | 
						|
  @param  Context      Pointer to the notification function's context
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
ReportStatusCodeLibVirtualAddressChange (
 | 
						|
  IN EFI_EVENT        Event,
 | 
						|
  IN VOID             *Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  if (mReportStatusCode != NULL) {
 | 
						|
    mInternalRT->ConvertPointer (0, (VOID **) &mReportStatusCode);
 | 
						|
  }
 | 
						|
  mInternalRT->ConvertPointer (0, (VOID **) &mStatusCodeData);
 | 
						|
  mInternalRT->ConvertPointer (0, (VOID **) &mInternalRT);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
  Notification function of EVT_SIGNAL_EXIT_BOOT_SERVICES.
 | 
						|
 | 
						|
  @param  Event        Event whose notification function is being invoked.
 | 
						|
  @param  Context      Pointer to the notification function's context
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
ReportStatusCodeLibExitBootServices (
 | 
						|
  IN EFI_EVENT        Event,
 | 
						|
  IN VOID             *Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  mHaveExitedBootServices = TRUE;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
  The constructor function of SMM Runtime DXE Report Status Code Lib.
 | 
						|
 | 
						|
  This function allocates memory for extended status code data, caches
 | 
						|
  the report status code service, and registers events.
 | 
						|
 | 
						|
  @param  ImageHandle   The firmware allocated handle for the EFI image.
 | 
						|
  @param  SystemTable   A pointer to the EFI System Table.
 | 
						|
  
 | 
						|
  @retval EFI_SUCCESS   The constructor always returns EFI_SUCCESS.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
ReportStatusCodeLibConstruct (
 | 
						|
  IN EFI_HANDLE           ImageHandle,
 | 
						|
  IN EFI_SYSTEM_TABLE     *SystemTable
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS     Status;
 | 
						|
 | 
						|
  //
 | 
						|
  // If in SMM mode, then allocates memory from SMRAM for extended status code data.
 | 
						|
  //
 | 
						|
  Status = gBS->LocateProtocol (&gEfiSmmBaseProtocolGuid, NULL, (VOID **) &mSmmBase);
 | 
						|
  if (!EFI_ERROR (Status)) {
 | 
						|
    mSmmBase->InSmm (mSmmBase, &mInSmm);
 | 
						|
    if (mInSmm) {
 | 
						|
      Status = mSmmBase->SmmAllocatePool (
 | 
						|
                           mSmmBase,
 | 
						|
                           EfiRuntimeServicesData, 
 | 
						|
                           sizeof (EFI_STATUS_CODE_DATA) + EFI_STATUS_CODE_DATA_MAX_SIZE, 
 | 
						|
                           (VOID **) &mStatusCodeData
 | 
						|
                           );
 | 
						|
      ASSERT_EFI_ERROR (Status);
 | 
						|
      OemHookStatusCodeInitialize ();
 | 
						|
      return EFI_SUCCESS;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
 | 
						|
  //
 | 
						|
  // If not in SMM mode, then allocate runtime memory for extended status code data.
 | 
						|
  //
 | 
						|
  // Library should not use the gRT directly, for it may be converted by other library instance.
 | 
						|
  // 
 | 
						|
  mInternalRT = gRT;
 | 
						|
  mInSmm      = FALSE;
 | 
						|
 | 
						|
  mStatusCodeData = AllocateRuntimePool (sizeof (EFI_STATUS_CODE_DATA) + EFI_STATUS_CODE_DATA_MAX_SIZE);
 | 
						|
  ASSERT (mStatusCodeData != NULL);
 | 
						|
  //
 | 
						|
  // Cache the report status code service
 | 
						|
  // 
 | 
						|
  mReportStatusCode = InternalGetReportStatusCode ();
 | 
						|
 | 
						|
  //
 | 
						|
  // Register notify function for EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE
 | 
						|
  // 
 | 
						|
  Status = gBS->CreateEventEx (
 | 
						|
                  EVT_NOTIFY_SIGNAL,
 | 
						|
                  TPL_NOTIFY,
 | 
						|
                  ReportStatusCodeLibVirtualAddressChange,
 | 
						|
                  NULL,
 | 
						|
                  &gEfiEventVirtualAddressChangeGuid,
 | 
						|
                  &mVirtualAddressChangeEvent
 | 
						|
                  );
 | 
						|
  ASSERT_EFI_ERROR (Status);
 | 
						|
 | 
						|
  //
 | 
						|
  // Register notify function for EVT_SIGNAL_EXIT_BOOT_SERVICES
 | 
						|
  // 
 | 
						|
  Status = gBS->CreateEventEx (
 | 
						|
                  EVT_NOTIFY_SIGNAL,
 | 
						|
                  TPL_NOTIFY,
 | 
						|
                  ReportStatusCodeLibExitBootServices,
 | 
						|
                  NULL,
 | 
						|
                  &gEfiEventExitBootServicesGuid,
 | 
						|
                  &mExitBootServicesEvent
 | 
						|
                  );
 | 
						|
  ASSERT_EFI_ERROR (Status);
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
  The destructor function of SMM Runtime DXE Report Status Code Lib.
 | 
						|
  
 | 
						|
  The destructor function frees memory allocated by constructor, and closes related events.
 | 
						|
  It will ASSERT() if that related operation fails and it will always return EFI_SUCCESS. 
 | 
						|
 | 
						|
  @param  ImageHandle   The firmware allocated handle for the EFI image.
 | 
						|
  @param  SystemTable   A pointer to the EFI System Table.
 | 
						|
  
 | 
						|
  @retval EFI_SUCCESS   The constructor always returns EFI_SUCCESS.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
ReportStatusCodeLibDestruct (
 | 
						|
  IN EFI_HANDLE        ImageHandle,
 | 
						|
  IN EFI_SYSTEM_TABLE  *SystemTable
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
 | 
						|
  if (!mInSmm) {
 | 
						|
    ASSERT (gBS != NULL);
 | 
						|
    Status = gBS->CloseEvent (mVirtualAddressChangeEvent);
 | 
						|
    ASSERT_EFI_ERROR (Status);
 | 
						|
    Status = gBS->CloseEvent (mExitBootServicesEvent);
 | 
						|
    ASSERT_EFI_ERROR (Status);
 | 
						|
 | 
						|
    FreePool (mStatusCodeData);
 | 
						|
  } else {
 | 
						|
    mSmmBase->SmmFreePool (mSmmBase, mStatusCodeData);
 | 
						|
  }
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 |