OcAfterBootCompatLib.c

Go to the documentation of this file.

 
#include <Uefi.h>
 
#include <Library/BaseLib.h>
#include <Library/BaseMemoryLib.h>
#include <Library/OcDebugLogLib.h>
#include <Library/DevicePathLib.h>
#include <Library/MemoryAllocationLib.h>
#include <Library/OcAfterBootCompatLib.h>
#include <Library/OcDevicePathLib.h>
#include <Library/OcMiscLib.h>
#include <Library/PrintLib.h>
#include <Library/UefiRuntimeServicesTableLib.h>
#include <Library/UefiBootServicesTableLib.h>
 
#include <Protocol/OcAfterBootCompat.h>
 
#include "BootCompatInternal.h"
 
STATIC OC_AFTER_BOOT_COMPAT_PROTOCOL  mOcAfterBootCompatProtocol = {
  OC_AFTER_BOOT_COMPAT_PROTOCOL_REVISION
};
 
STATIC BOOT_COMPAT_CONTEXT  mOcAfterBootCompatContext;
 
STATIC
EFI_STATUS
InstallAbcProtocol (
  VOID
  )
{
  EFI_STATUS  Status;
  VOID        *Interface;
  EFI_HANDLE  Handle;
 
  Status = gBS->LocateProtocol (
                  &gOcAfterBootCompatProtocolGuid,
                  NULL,
                  &Interface
                  );
 
  if (!EFI_ERROR (Status)) {
    //
    // Ensure we do not run with AptioMemoryFix.
    // It also checks for attempts to install this protocol twice.
    //
    DEBUG ((DEBUG_WARN, "OCABC: Found legacy AptioMemoryFix driver!\n"));
    return EFI_ALREADY_STARTED;
  }
 
  Handle = NULL;
  Status = gBS->InstallMultipleProtocolInterfaces (
                  &Handle,
                  &gOcAfterBootCompatProtocolGuid,
                  &mOcAfterBootCompatProtocol,
                  NULL
                  );
 
  if (EFI_ERROR (Status)) {
    DEBUG ((DEBUG_WARN, "OCABC: protocol install failure - %r\n", Status));
    return Status;
  }
 
  return EFI_SUCCESS;
}
 
BOOT_COMPAT_CONTEXT *
GetBootCompatContext (
  VOID
  )
{
  return &mOcAfterBootCompatContext;
}
 
EFI_STATUS
OcAbcInitialize (
  IN OC_ABC_SETTINGS  *Settings,
  IN OC_CPU_INFO      *CpuInfo
  )
{
  EFI_STATUS           Status;
  BOOT_COMPAT_CONTEXT  *BootCompat;
  UINTN                LowMemory;
  UINTN                TotalMemory;
 
  Status = InstallAbcProtocol ();
  if (EFI_ERROR (Status)) {
    return Status;
  }
 
  DEBUG ((
    DEBUG_INFO,
    "OCABC: ALRBL %d RTDFRG %d DEVMMIO %d NOSU %d NOVRWR %d NOSB %d FBSIG %d NOHBMAP %d SMSLIDE %d WRUNPROT %d CLRTS %d\n",
    Settings->AllowRelocationBlock,
    Settings->AvoidRuntimeDefrag,
    Settings->DevirtualiseMmio,
    Settings->DisableSingleUser,
    Settings->DisableVariableWrite,
    Settings->ProtectSecureBoot,
    Settings->ForceBooterSignature,
    Settings->DiscardHibernateMap,
    Settings->EnableSafeModeSlide,
    Settings->EnableWriteUnprotector,
    Settings->ClearTaskSwitchBit
    ));
 
  DEBUG ((
    DEBUG_INFO,
    "OCABC: FEXITBS %d PRMRG %d CSLIDE %d MSLIDE %d PRSRV %d RBMAP %d VMAP %d APPLOS %d RTPERMS %d ARBAR %d RBIO %d\n",
    Settings->ForceExitBootServices,
    Settings->ProtectMemoryRegions,
    Settings->ProvideCustomSlide,
    Settings->ProvideMaxSlide,
    Settings->ProtectUefiServices,
    Settings->RebuildAppleMemoryMap,
    Settings->SetupVirtualMap,
    Settings->SignalAppleOS,
    Settings->SyncRuntimePermissions,
    Settings->ResizeAppleGpuBars,
    Settings->ResizeUsePciRbIo
    ));
 
  DEBUG_CODE_BEGIN ();
  TotalMemory = OcCountFreePages (&LowMemory);
  DEBUG ((
    DEBUG_INFO,
    "OCABC: Firmware has %Lu free pages (%Lu in lower 4 GB)\n",
    (UINT64)TotalMemory,
    (UINT64)LowMemory
    ));
  DEBUG_CODE_END ();
 
  BootCompat = GetBootCompatContext ();
 
  CopyMem (
    &BootCompat->Settings,
    Settings,
    sizeof (BootCompat->Settings)
    );
 
  BootCompat->CpuInfo = CpuInfo;
 
  InstallServiceOverrides (BootCompat);
 
  return EFI_SUCCESS;
}