diff options
Diffstat (limited to 'EDK/Foundation/Core/Dxe/Misc/DebugMask.c')
| -rw-r--r-- | EDK/Foundation/Core/Dxe/Misc/DebugMask.c | 400 |
1 files changed, 400 insertions, 0 deletions
diff --git a/EDK/Foundation/Core/Dxe/Misc/DebugMask.c b/EDK/Foundation/Core/Dxe/Misc/DebugMask.c new file mode 100644 index 0000000..c2dee47 --- /dev/null +++ b/EDK/Foundation/Core/Dxe/Misc/DebugMask.c @@ -0,0 +1,400 @@ +/*++ + +Copyright (c) 2004 - 2005, 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 (GlobalVariable) + +extern UINTN mErrorLevel; +static VOID *mVariableReadyNotify; + +VOID +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 + // + Status = CoreHandleProtocol ( + ImageHandle, + &gEfiDebugMaskProtocolGuid, + (VOID*)&DebugMaskInterface + ); + if (EFI_ERROR(Status) || (DebugMaskInterface == NULL)) { + return Status; + } + 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 +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(UINT32); + Status = gRT->GetVariable( + L"EFIDebug", + &gEfiGlobalVariableGuid, + 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) || (DebugMaskProtocol == NULL)) { + continue; + } + DebugMaskProtocol->SetDebugMask(DebugMaskProtocol, DebugMask); + } + CoreFreePool(Buffer); +} |