/** @file
  TraceHubStatusCodeHandlerSmm implementation.

@copyright
 Copyright (c) 2013 - 2014 Intel Corporation. All rights reserved
 This software and associated documentation (if any) is furnished
 under a license and may only be used or copied in accordance
 with the terms of the license. Except as permitted by the
 license, no part of this software or documentation may be
 reproduced, stored in a retrieval system, or transmitted in any
 form or by any means without the express written consent of
 Intel Corporation.
 This file contains 'Framework Code' and is licensed as such
 under the terms of your license agreement with Intel or your
 vendor. This file may not be modified, except as allowed by
 additional terms of your license agreement.

@par Specification Reference:
**/

#include <TraceHubStatusCodeHandlerSmm.h>

EFI_SMM_RSC_HANDLER_PROTOCOL    *mRscHandlerProtocol = NULL;
psven_handle_t                  TraceHubHandle            = NULL;
sven_header_t                   TraceHubHandleHeader;
extern EFI_GUID                 gEfiTraceHubStatusCodeHandleSmmGuid;

/**
  Entry point of SMM TraceHub Status Code Driver.

  This function is the entry point of SMM TraceHub Status Code Driver.

  @param  ImageHandle       The firmware allocated handle for the EFI image.
  @param  SystemTable       A pointer to the EFI System Table.

  @retval EFI_SUCCESS       The entry point is executed successfully.

**/
EFI_STATUS
EFIAPI
TraceHubStatusCodeHandlerSmmEntry (
  IN EFI_HANDLE         ImageHandle,
  IN EFI_SYSTEM_TABLE   *SystemTable
  )
{
  EFI_STATUS                Status;
  TRACE_HUB_PLATFORM_DATA   mTraceHubPlatformData;

  Status = gSmst->SmmLocateProtocol (
                    &gEfiSmmRscHandlerProtocolGuid,
                    NULL,
                    (VOID **) &mRscHandlerProtocol
                    );
  ASSERT_EFI_ERROR (Status);

  if (FeaturePcdGet (PcdStatusCodeUseTraceHub)) {
    mTraceHubPlatformData.Master = FixedPcdGet8 (PcdTraceHubStatusCodeMaster);
    mTraceHubPlatformData.Channel = FixedPcdGet8 (PcdTraceHubStatusCodeChannel);

    SVEN_INIT_STATE (&TraceHubHandleHeader, sventx_PlatformInit, NULL);
    TraceHubHandle = SVEN_ALLOC_HANDLE_STATE (&TraceHubHandleHeader, &mTraceHubPlatformData);

    SVEN_SET_HANDLE_GUID_UNIT (
      TraceHubHandle,
      *((sven_guid_t *)(VOID *) &gEfiTraceHubStatusCodeHandleSmmGuid),
      0
      );

    mRscHandlerProtocol->Register (TraceHubStatusCodeReportWorkerSmm);
  }

  return EFI_SUCCESS;
}