/** @file
  This module is a sample driver to locate PlatformBinaryProtocol, installing ACPI Windows Platform Binary Table (WPBT).

;******************************************************************************
;* Copyright (c) 2013 - 2019, Insyde Software Corp. All Rights Reserved.
;*
;* You may not reproduce, distribute, publish, display, perform, modify, adapt,
;* transmit, broadcast, present, recite, release, license or otherwise exploit
;* any part of this publication in any form, by any means, without the prior
;* written permission of Insyde Software Corporation.
;*
;******************************************************************************
*/

#include <Uefi.h>
#include <Library/UefiLib.h>
#include <Library/BaseLib.h>
#include <Library/UefiBootServicesTableLib.h>
#include <Library/PcdLib.h>
#include <Library/DxeOemSvcKernelLib.h>
#include <Library/MemoryAllocationLib.h>
#include <Library/BaseMemoryLib.h>
#include <Protocol/AcpiWpbt.h>
#include <Library/DebugLib.h>

/**
 Sample Dxe driver entry point

 @param [in]   ImageHandle      EFI_HANDLE
 @param [in]   SystemTable      EFI_SYSTEM_TABLE

 @return Status code

**/
EFI_STATUS
EFIAPI
InstallWpbtDxeEntryPoint (
  IN EFI_HANDLE          ImageHandle,
  IN EFI_SYSTEM_TABLE    *SystemTable
  )
{
  EFI_STATUS                      Status;
  EFI_ACPI_WPBT_PROTOCOL          *Pb;
  EFI_GUID                        *DefaultFileGuid;
  CHAR16                          *DefaultInputArg;
  CHAR16                          *DefaultInputArgBuf;
  UINTN                           ArgLen;
  UINTN                           TotalStringBufSize;

  Status = gBS->LocateProtocol (
                  &gEfiAcpiWpbtProtocolGuid,
                  NULL,
                  (VOID **) &Pb
                  );
  if (EFI_ERROR (Status)) {
    return Status;
  }

  //
  // Get default file GUID and input argument
  //
  DefaultFileGuid = (EFI_GUID *)PcdGetPtr(PcdWpbtFileGuid);
  DefaultInputArg = (CHAR16 *)PcdGetPtr(PcdWpbtInputArg);

  //
  // Base on WPBT_INPUT_ARG_RESERVED_CHARS and default PCD 
  // string size to allocate the string buffer for OemServices
  //
  TotalStringBufSize = (StrSize(DefaultInputArg) + (sizeof (CHAR16) * WPBT_INPUT_ARG_RESERVED_CHARS));
  DefaultInputArgBuf = (CHAR16 *) AllocateZeroPool (TotalStringBufSize);
  if (DefaultInputArgBuf == NULL) {
    return EFI_OUT_OF_RESOURCES;
  }
  CopyMem (DefaultInputArgBuf, DefaultInputArg, StrSize(DefaultInputArg));
  DefaultInputArg = DefaultInputArgBuf;
  
  //
  // OemServices
  //
  DEBUG_OEM_SVC ((DEBUG_INFO, "OemKernelServices Call: OemSvcCustomizeWpbtTable \n"));
  Status = OemSvcCustomizeWpbtTable (DefaultFileGuid, DefaultInputArg);
  DEBUG_OEM_SVC ((DEBUG_INFO, "OemKernelServices OemSvcCustomizeWpbtTable Status: %r\n", Status));
  if (!EFI_ERROR (Status)) {
    FreePool(DefaultInputArgBuf);
    return Status;
  }
  
  ArgLen = StrSize(DefaultInputArg);

  //
  // Pass the file guid of binary, the input string and the input length to LocatePlatformBinary
  //
  Status = Pb->LocatePlatformBinary (Pb, DefaultFileGuid, (UINT16)ArgLen, DefaultInputArg);
  
  FreePool(DefaultInputArgBuf);

  return Status;
}