init. 1AQQW051HEADmaster

1 files changed, 399 insertions, 0 deletions

diff --git a/Core/CORE_DXE/DebugMask.c b/Core/CORE_DXE/DebugMask.c
new file mode 100644
index 0000000..f9b0cf4
--- /dev/null
+++ b/Core/CORE_DXE/DebugMask.c
@@ -0,0 +1,399 @@
+/*++
+
+Copyright (c) 2004 - 2007, Intel Corporation                                                         
+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.             
+
+Module Name:
+
+  DebugMask.c
+    
+Abstract:
+
+
+--*/
+
+#include "Tiano.h"
+#include "DxeCore.h"
+#include "DebugMask.h"
+#include EFI_GUID_DEFINITION (GenericVariable)
+
+extern UINTN                     mErrorLevel;
+static VOID                      *mVariableReadyNotify;
+
+VOID
+EFIAPI
+UpdateDebugMask (
+  EFI_EVENT Event,
+  VOID      *Context
+  );
+  
+EFI_STATUS
+EFIAPI
+GetDebugMask (
+  IN EFI_DEBUG_MASK_PROTOCOL      *This,             // Calling context
+  IN OUT UINTN                    *CurrentDebugMask  // Ptr to store current debug mask
+  )
+/*++
+
+Routine Description:
+  DebugMask protocol member function.
+  Gets the current debug mask for an image, on which this protocol has been installed.
+  
+Arguments:
+
+  This              - Indicates calling context 
+  CurrentDebugMask  - Ptr to store current debug mask
+
+Returns:
+  EFI_SUCCESS - Debug mask is retrieved successfully
+  EFI_INVALID_PARAMETER - CurrentDebugMask is NULL.
+  EFI_UNSUPPORTED - The handle on which this protocol is installed is not an image handle.
+
+--*/
+{
+  DEBUG_MASK_PRIVATE_DATA        *Private;
+  //
+  // Check Parameter
+  //
+  if (CurrentDebugMask == NULL){
+    return EFI_INVALID_PARAMETER;
+  }
+  
+  //
+  // Get Private Data
+  //
+  Private = DEBUG_MASK_PRIVATE_DATA_FROM_THIS(This);
+  *CurrentDebugMask = Private->ImageDebugMask;
+  return EFI_SUCCESS;
+}
+  
+
+EFI_STATUS
+EFIAPI
+SetDebugMask (
+  IN  EFI_DEBUG_MASK_PROTOCOL     *This,             // Calling context
+  IN  UINTN                       NewDebugMask       // New Debug Mask value to set
+  )
+/*++
+
+Routine Description:
+  DebugMask protocol member function.
+  Updates the current debug mask for an image, on which this protocol has been installed.
+  
+Arguments:
+
+  This          - Calling context
+  NewDebugMask  - New Debug Mask value to set
+
+Returns:
+  EFI_SUCCESS - Debug mask is updated with the new value successfully
+  EFI_UNSUPPORTED - The handle on which this protocol is installed is not an image handle.
+
+--*/
+{
+  DEBUG_MASK_PRIVATE_DATA        *Private;
+  //
+  // Set Private Data
+  //
+  Private = DEBUG_MASK_PRIVATE_DATA_FROM_THIS(This);
+  Private->ImageDebugMask = NewDebugMask;
+  return EFI_SUCCESS;
+}  
+
+EFI_STATUS
+EFIAPI
+SetCoreDebugMask (
+  IN  EFI_DEBUG_MASK_PROTOCOL     *This,             // Calling context
+  IN  UINTN                       NewDebugMask       // New Debug Mask value to set
+  )
+/*++
+
+Routine Description:
+  DebugMask protocol member function.
+  Updates the current debug mask for core.
+  
+Arguments:
+
+  This          - Calling context
+  NewDebugMask  - New Debug Mask value to set
+
+Returns:
+  EFI_SUCCESS - Debug mask is updated with the new value successfully
+  EFI_UNSUPPORTED - The handle on which this protocol is installed is not an image handle.
+
+--*/
+{
+  mErrorLevel = NewDebugMask;
+  return SetDebugMask(This, NewDebugMask);
+}  
+
+EFI_STATUS
+InstallDebugMaskProtocol(
+  IN EFI_HANDLE ImageHandle
+  )
+/*++
+
+Routine Description:
+
+  Install debug mask protocol on an image handle.
+
+Arguments:
+
+  ImageHandle     - Image handle which debug mask protocol will install on
+
+Returns:
+
+  EFI_INVALID_PARAMETER   - Invalid image handle
+  
+  EFI_OUT_OF_RESOURCES    - No enough buffer could be allocated
+  
+  EFI_SUCCESS             - Debug mask protocol successfully installed
+
+--*/
+{
+  EFI_STATUS                 Status;
+  EFI_LOADED_IMAGE_PROTOCOL  *LoadImageInterface;
+  DEBUG_MASK_PRIVATE_DATA    *DebugMaskPrivate;
+
+  if (ImageHandle == NULL) {
+    return EFI_INVALID_PARAMETER;
+  }
+  //
+  // Check Image Handle
+  //
+  Status = CoreHandleProtocol (
+                  ImageHandle, 
+                  &gEfiLoadedImageProtocolGuid, 
+                  (VOID*)&LoadImageInterface
+                  );
+                  
+  if (EFI_ERROR (Status)) {
+    return EFI_INVALID_PARAMETER;
+  }
+  //
+  // Create Pool for Private Data
+  //
+  DebugMaskPrivate = CoreAllocateZeroBootServicesPool (sizeof (DEBUG_MASK_PRIVATE_DATA));
+  
+  if (DebugMaskPrivate == NULL) {
+    return EFI_OUT_OF_RESOURCES;
+  }
+  //
+  // Fill in private data structure
+  //
+  DebugMaskPrivate->Signature  =  DEBUGMASK_PRIVATE_DATA_SIGNATURE;
+  DebugMaskPrivate->ImageDebugMask  =  mErrorLevel;
+  DebugMaskPrivate->DebugMaskInterface.Revision  =  EFI_DEBUG_MASK_REVISION;
+  DebugMaskPrivate->DebugMaskInterface.GetDebugMask  =  GetDebugMask;
+  DebugMaskPrivate->DebugMaskInterface.SetDebugMask  =  SetDebugMask;
+  //
+  // Install Debug Mask Protocol in Image Handle
+  //
+  Status = CoreInstallProtocolInterface (
+             &ImageHandle,
+             &gEfiDebugMaskProtocolGuid,
+             EFI_NATIVE_INTERFACE,
+             &(DebugMaskPrivate->DebugMaskInterface)
+             );
+
+  return Status;
+}
+
+EFI_STATUS
+UninstallDebugMaskProtocol(
+  IN EFI_HANDLE ImageHandle
+  )
+/*++
+
+Routine Description:
+
+  Uninstall debug mask protocol on an image handle.
+
+Arguments:
+
+  ImageHandle     - Image handle which debug mask protocol will uninstall on
+
+Returns:
+
+  EFI_INVALID_PARAMETER   - Invalid image handle
+  
+  EFI_SUCCESS             - Debug mask protocol successfully uninstalled
+
+--*/
+{
+  EFI_STATUS                 Status;
+  EFI_DEBUG_MASK_PROTOCOL    *DebugMaskInterface;
+  DEBUG_MASK_PRIVATE_DATA    *DebugMaskPrivate;
+  
+  if (ImageHandle == NULL) {
+    return EFI_INVALID_PARAMETER;
+  }
+  //
+  // Get Protocol from ImageHandle
+  //
+  Status = CoreHandleProtocol (
+                  ImageHandle, 
+                  &gEfiDebugMaskProtocolGuid, 
+                  (VOID*)&DebugMaskInterface
+                  );
+  if (EFI_ERROR (Status)) {
+    return Status;
+  }
+  
+  DebugMaskPrivate = DEBUG_MASK_PRIVATE_DATA_FROM_THIS(DebugMaskInterface);
+  //
+  // Remove Protocol from ImageHandle
+  //                    
+  Status = CoreUninstallProtocolInterface (
+             ImageHandle,
+             &gEfiDebugMaskProtocolGuid,
+             (VOID*)DebugMaskInterface
+             );
+  if (EFI_ERROR (Status)) {
+    return Status;
+  }
+  //
+  // Free Private Data Pool
+  //
+  Status = CoreFreePool(DebugMaskPrivate);
+  return Status;
+}
+
+EFI_STATUS
+InstallCoreDebugMaskProtocol(
+  IN EFI_HANDLE ImageHandle
+  )
+/*++
+
+Routine Description:
+
+  Install debug mask protocol on core.
+
+Arguments:
+
+  ImageHandle     - Core handle
+
+Returns:
+
+  EFI_INVALID_PARAMETER   - Invalid image handle
+  
+  EFI_OUT_OF_RESOURCES    - No enough buffer could be allocated
+  
+  EFI_SUCCESS             - Debug mask protocol successfully installed
+
+--*/
+{
+  EFI_DEBUG_MASK_PROTOCOL    *DebugMaskInterface;
+  EFI_STATUS                 Status;
+  EFI_EVENT                  Event;
+
+  if (ImageHandle == NULL) {
+    return EFI_INVALID_PARAMETER;
+  }
+
+  Status = InstallDebugMaskProtocol(ImageHandle);
+  if (EFI_ERROR(Status)) {
+    return Status;
+  }
+
+  //
+  // Check Image Handle
+  //
+  CoreHandleProtocol (
+    ImageHandle, 
+    &gEfiDebugMaskProtocolGuid, 
+    (VOID*)&DebugMaskInterface
+    );
+  DebugMaskInterface->SetDebugMask = SetCoreDebugMask;
+
+  Status = CoreCreateEvent (
+             EFI_EVENT_NOTIFY_SIGNAL,
+             EFI_TPL_CALLBACK,
+             UpdateDebugMask,
+             mVariableReadyNotify,
+             &Event
+             );
+
+  if (!EFI_ERROR (Status)) {
+    Status = CoreRegisterProtocolNotify (
+               &gEfiVariableArchProtocolGuid,
+               Event,
+               &mVariableReadyNotify
+               );
+  }
+                  
+  return Status;
+}
+
+VOID
+EFIAPI
+UpdateDebugMask (
+  EFI_EVENT Event,
+  VOID      *Context
+  )
+/*++
+
+Routine Description:
+
+  Event callback function to update the debug mask when the variable service is ready.
+
+Arguments:
+
+  Event   - The Event
+  Context - The event's context
+
+Returns:
+
+  None
+
+--*/
+{
+  UINTN                   NoHandles;
+  EFI_STATUS              Status;
+  UINTN                   DebugMask;
+  UINTN                   Index;
+  UINTN                   DataSize;
+  EFI_HANDLE              *Buffer;
+  EFI_DEBUG_MASK_PROTOCOL *DebugMaskProtocol;
+
+  DataSize = sizeof(UINTN);
+  Status = gRT->GetVariable(
+                  L"EFIDebug",
+                  &gEfiGenericVariableGuid,
+                  NULL,
+                  &DataSize,
+                  &DebugMask
+                  );
+  if (EFI_ERROR(Status)) {
+    return;
+  }
+ 
+  Status = CoreLocateHandleBuffer (
+             AllHandles,
+             &gEfiDebugMaskProtocolGuid,
+             NULL,
+             &NoHandles,
+             &Buffer
+             );
+  if (EFI_ERROR(Status)) {
+    return;
+  }
+  for (Index = 0; Index < NoHandles; Index ++) {
+    Status = CoreHandleProtocol (
+               Buffer[Index],
+               &gEfiDebugMaskProtocolGuid,
+               &DebugMaskProtocol
+               );
+    if (EFI_ERROR(Status)) {
+      continue;
+    }
+    DebugMaskProtocol->SetDebugMask(DebugMaskProtocol, DebugMask);
+  }
+  CoreFreePool(Buffer);
+}