1 files changed, 10 insertions, 2 deletions
diff --git a/ArmPlatformPkg/Drivers/LcdGraphicsOutputDxe/LcdGraphicsOutputDxe.c b/ArmPlatformPkg/Drivers/LcdGraphicsOutputDxe/LcdGraphicsOutputDxe.c
index f4165d0003..b42336446c 100644
--- a/ArmPlatformPkg/Drivers/LcdGraphicsOutputDxe/LcdGraphicsOutputDxe.c
+++ b/ArmPlatformPkg/Drivers/LcdGraphicsOutputDxe/LcdGraphicsOutputDxe.c
@@ -133,7 +133,6 @@ InitializeDisplay (
   // Setup all the relevant mode information
   Instance->Gop.Mode->SizeOfInfo      = sizeof(EFI_GRAPHICS_OUTPUT_MODE_INFORMATION);
   Instance->Gop.Mode->FrameBufferBase = VramBaseAddress;
-  Instance->Gop.Mode->FrameBufferSize = VramSize;
 
   // Set the flag before changing the mode, to avoid infinite loops
   mDisplayInitialized = TRUE;
@@ -297,7 +296,8 @@ LcdGraphicsSetMode (
 {
 	EFI_STATUS                      Status               = EFI_SUCCESS;
 	EFI_GRAPHICS_OUTPUT_BLT_PIXEL   FillColour;
-	LCD_INSTANCE* Instance;
+	LCD_INSTANCE*                   Instance;
+	LCD_BPP                         Bpp;
 
 	Instance = LCD_INSTANCE_FROM_GOP_THIS (This);
 
@@ -326,6 +326,14 @@ LcdGraphicsSetMode (
 	// Update the UEFI mode information
 	This->Mode->Mode = ModeNumber;
 	LcdPlatformQueryMode (ModeNumber,&Instance->ModeInfo);
+	Status = LcdPlatformGetBpp(ModeNumber, &Bpp);
+	if (EFI_ERROR(Status)) {
+	  DEBUG ((DEBUG_ERROR, "LcdGraphicsSetMode: ERROR - Couldn't get bytes per pixel, status: %r\n", Status));
+	  goto EXIT;
+	}
+	This->Mode->FrameBufferSize =  Instance->ModeInfo.VerticalResolution
+	                             * Instance->ModeInfo.PixelsPerScanLine
+	                             * GetBytesPerPixel(Bpp);
 
   // Set the hardware to the new mode
 	Status = LcdSetMode (ModeNumber);