/** @file
  Provide OEM to switch the CSM16 function.

;******************************************************************************
;* Copyright (c) 2018, 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 <PiPei.h>
#include <Library/CmosLib.h>
#include <Library/HobLib.h>
#include <Library/DxeOemSvcKernelLib.h>


/**
   Provide OEM to switch the CSM16 function. Current functions of CSM16 as following:

    Bit         Function	                                Disable bit (set to 0)	               Enable bit (set to 1)
   --------------------------------------------------------------------------------------------------------------
    0,          Prepare LegacyTable(BDA/EDBA/SMBIOS data/   0: Disable                                       1: Enable
                ACPI table) Before UEFI Boot
    1,          POST_TIME_DISABLE_KEYBOARD                0: Enable KeyboardReset                          1: Disable KeyboardReset
    2,          POST_TIME_DISABLE_KEYBOARD_BEEP           0: Enable KeyboardBeep                           1: Disable KeyboardBeep
    3,          LEGACY_AUTO_BOOT_RETRY                    0: Disable AutoBootRetry                         1: Enable AutoBootRetry
    4,          ODD_IMAGE_DETECT                          0: Detect ODD image before INT19                 1: Detect ODD image after INT19
    5,          SET_VGAMODE_BEFORE_INT19                  0: Disable                                       1: Set Vga mode before INT19
    6,          ENABLE_A20_BEFORE_INT19                   0: Disable                                       1: Enable A20 before INT19
    7,          A20_GATE_SUPPORT                          0: supported on keyboard controller              1: supported with bit 1 of I/O port 92h
    8,          MONITORKEY_SUPPORT_IN_OPROM               0: Disable MonitorKey during dispatch OPROM      1: Enable MonitorKey during dispatch OPROM
    9           PAUSE_KEY_SUPPORT                         0: Enable Pause Key                              1: Disable Pause Key
    10,         SMBIOS_DATA_LOCATE                        0: Locate at E segment                           1: Locate at F segment
    11 ~ 13     For PMM high memory size
                  if the value is 0-> 4 MB (default)
                  if the value is 1-> 2 to the (4 + 1) = 32 MB
                  if the value is 2-> 2 to the (4 + 2) = 64 MB
                  if the value is 3-> 2 to the (4 + 3) = 128 MB
                  if the value is 4-> 2 to the (4 + 4) = 256 MB
                  if the value is 5-> 2 to the (4 + 5) = 512 MB
                  if the value is 6-> 2 to the (4 + 6) = 1024 MB
                  if the value is 7-> 2 to the (4 + 7) = 2048 MB
    14 ~ 15     RESERVED
   --------------------------------------------------------------------------------------------------------------

  @param  *CSM16OemSwitchPtr    Point to the CSM16OemSwitch of H2O_TABLE in LegacyBios.

  @retval EFI_MEDIA_CHANGED     Set the CSM16 OEM switch success.
  @retval Others                Base on OEM design.
**/
EFI_STATUS
OemSvcCsm16ReferSwitch (
  IN OUT UINT16                 *CSM16OemSwitchPtr
  )
{
  EFI_BOOT_MODE                         BootMode;
  CSM16_OEM_CONFIG                      LocalConfig;

  LocalConfig.CONFIG = (*CSM16OemSwitchPtr);
  BootMode = BOOT_WITH_FULL_CONFIGURATION;

  BootMode = GetBootModeHob ();

  switch (BootMode) {

  case BOOT_WITH_MINIMAL_CONFIGURATION:
  case BOOT_ASSUMING_NO_CONFIGURATION_CHANGES:
    LocalConfig.Config.SET_VGAMODE_BEFORE_INT19 = 0;
    break;
  case BOOT_ON_FLASH_UPDATE:
  case BOOT_IN_RECOVERY_MODE:
    LocalConfig.Config.DisableKeyboardReset = 1;
    break;
  case BOOT_WITH_FULL_CONFIGURATION:
  case BOOT_WITH_FULL_CONFIGURATION_PLUS_DIAGNOSTICS:
  case BOOT_WITH_DEFAULT_SETTINGS:
  default:
    LocalConfig.Config.SET_VGAMODE_BEFORE_INT19 = 1;
    break;
  }

  (*CSM16OemSwitchPtr) = LocalConfig.CONFIG;

  return EFI_SUCCESS;
}