FramebufferInfo.c

Go to the documentation of this file.

 
#include "OcConsoleLibInternal.h"
#include "ConsoleGopInternal.h"
 
#include <Protocol/AppleFramebufferInfo.h>
#include <Protocol/GraphicsOutput.h>
 
#include <Library/BaseLib.h>
#include <Library/DebugLib.h>
#include <Library/OcConsoleLib.h>
#include <Library/OcMiscLib.h>
#include <Library/UefiBootServicesTableLib.h>
#include <Library/UefiRuntimeServicesTableLib.h>
 
STATIC
EFI_STATUS
EFIAPI
AppleFramebufferGetInfo (
  IN   APPLE_FRAMEBUFFER_INFO_PROTOCOL  *This,
  OUT  EFI_PHYSICAL_ADDRESS             *FramebufferBase,
  OUT  UINT32                           *FramebufferSize,
  OUT  UINT32                           *ScreenRowBytes,
  OUT  UINT32                           *ScreenWidth,
  OUT  UINT32                           *ScreenHeight,
  OUT  UINT32                           *ScreenDepth
  )
{
  EFI_STATUS                            Status;
  EFI_GRAPHICS_OUTPUT_PROTOCOL          *GraphicsOutput;
  EFI_GRAPHICS_OUTPUT_PROTOCOL_MODE     *Mode;
  EFI_GRAPHICS_OUTPUT_MODE_INFORMATION  *Info;
  CONST CONSOLE_GOP_CONTEXT             *DirectConsole;
 
  if (  (This == NULL)
     || (FramebufferBase == NULL)
     || (FramebufferSize == NULL)
     || (ScreenRowBytes == NULL)
     || (ScreenWidth == NULL)
     || (ScreenHeight == NULL)
     || (ScreenDepth == NULL))
  {
    return EFI_INVALID_PARAMETER;
  }
 
  DirectConsole = InternalGetDirectGopContext ();
  if (DirectConsole != NULL) {
    *FramebufferBase = DirectConsole->OriginalFrameBufferBase;
    *FramebufferSize = (UINT32)DirectConsole->OriginalFrameBufferSize;
    *ScreenRowBytes  = (UINT32)(DirectConsole->OriginalModeInfo.PixelsPerScanLine * sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL));
    *ScreenWidth     = DirectConsole->CustomModeInfo.HorizontalResolution;
    *ScreenHeight    = DirectConsole->CustomModeInfo.VerticalResolution;
    *ScreenDepth     = DEFAULT_COLOUR_DEPTH;
    return EFI_SUCCESS;
  }
 
  Status = gBS->HandleProtocol (
                  gST->ConsoleOutHandle,
                  &gEfiGraphicsOutputProtocolGuid,
                  (VOID **)&GraphicsOutput
                  );
 
  if (EFI_ERROR (Status)) {
    return EFI_UNSUPPORTED;
  }
 
  Mode = GraphicsOutput->Mode;
  Info = Mode->Info;
 
  if (Info == NULL) {
    return EFI_UNSUPPORTED;
  }
 
  //
  // This is a bit inaccurate as it assumes 32-bit BPP, but will do for most cases.
  //
  *FramebufferBase = Mode->FrameBufferBase;
  *FramebufferSize = (UINT32)Mode->FrameBufferSize;
  *ScreenRowBytes  = (UINT32)(Info->PixelsPerScanLine * sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL));
  *ScreenWidth     = Info->HorizontalResolution;
  *ScreenHeight    = Info->VerticalResolution;
  *ScreenDepth     = DEFAULT_COLOUR_DEPTH;
 
  return EFI_SUCCESS;
}
 
STATIC
APPLE_FRAMEBUFFER_INFO_PROTOCOL
  mAppleFramebufferInfo = {
  AppleFramebufferGetInfo
};
 
APPLE_FRAMEBUFFER_INFO_PROTOCOL *
OcAppleFbInfoInstallProtocol (
  IN BOOLEAN  Reinstall
  )
{
  EFI_STATUS                       Status;
  APPLE_FRAMEBUFFER_INFO_PROTOCOL  *Protocol;
  EFI_GRAPHICS_OUTPUT_PROTOCOL     *GraphicsOutput;
  EFI_HANDLE                       NewHandle;
 
  DEBUG ((DEBUG_VERBOSE, "OcAppleFbInfoInstallProtocol\n"));
 
  Status = gBS->LocateProtocol (
                  &gEfiGraphicsOutputProtocolGuid,
                  NULL,
                  (VOID *)&GraphicsOutput
                  );
 
  if (EFI_ERROR (Status)) {
    DEBUG ((DEBUG_INFO, "OCFB: No GOP protocols for FB info, ignoring\n"));
    return NULL;
  }
 
  if (Reinstall) {
    Status = OcUninstallAllProtocolInstances (&gAppleFramebufferInfoProtocolGuid);
    if (EFI_ERROR (Status)) {
      DEBUG ((DEBUG_ERROR, "OCFB: Uninstall failed: %r\n", Status));
      return NULL;
    }
  } else {
    Status = gBS->LocateProtocol (
                    &gAppleFramebufferInfoProtocolGuid,
                    NULL,
                    (VOID *)&Protocol
                    );
 
    if (!EFI_ERROR (Status)) {
      return Protocol;
    }
  }
 
  NewHandle = NULL;
  Status    = gBS->InstallMultipleProtocolInterfaces (
                     &NewHandle,
                     &gAppleFramebufferInfoProtocolGuid,
                     (VOID *)&mAppleFramebufferInfo,
                     NULL
                     );
 
  if (EFI_ERROR (Status)) {
    return NULL;
  }
 
  return &mAppleFramebufferInfo;
}