OpenControl.c

Go to the documentation of this file.

 
#include <Uefi.h>
#include <Library/BaseLib.h>
#include <Library/BaseMemoryLib.h>
#include <Library/UefiApplicationEntryPoint.h>
#include <Library/UefiBootServicesTableLib.h>
#include <Library/OcMiscLib.h>
#include <Library/OcStringLib.h>
#include <Library/UefiLib.h>
#include <Protocol/OcFirmwareRuntime.h>
 
EFI_STATUS
EFIAPI
UefiMain (
  IN EFI_HANDLE        ImageHandle,
  IN EFI_SYSTEM_TABLE  *SystemTable
  )
{
  EFI_STATUS                    Status;
  UINTN                         Argc;
  CHAR16                        **Argv;
  OC_FIRMWARE_RUNTIME_PROTOCOL  *FwRuntime;
  OC_FWRT_CONFIG                Config;
 
  Status = GetArguments (&Argc, &Argv);
  if (!EFI_ERROR (Status) && (Argc == 2)) {
    Status = gBS->LocateProtocol (
                    &gOcFirmwareRuntimeProtocolGuid,
                    NULL,
                    (VOID **)&FwRuntime
                    );
 
    if (EFI_ERROR (Status)) {
      Print (L"FwRuntime protocol is unavailable - %r\n", Status);
      return EFI_SUCCESS;
    }
 
    if (OcStriCmp (Argv[1], L"disable") == 0) {
      ZeroMem (&Config, sizeof (Config));
      FwRuntime->SetOverride (&Config);
      return EFI_SUCCESS;
    }
 
    if (OcStriCmp (Argv[1], L"restore") == 0) {
      FwRuntime->SetOverride (NULL);
      return EFI_SUCCESS;
    }
  }
 
  Print (L"Usage: OpenControl <disable|restore>\n");
  return EFI_SUCCESS;
}