CloverBootloader/rEFIt_UEFI/libeg/AppleImageCodec.cpp
SergeySlice d54c3b9601 embed AppleImageCodec protocol
Signed-off-by: SergeySlice <sergey.slice@gmail.com>
2022-08-16 21:30:49 +03:00

235 lines
6.0 KiB
C++

/** @file
Driver with Apple image decode protocol implementation
for PNG.
**/
#include <Uefi.h>
#include <Library/BaseLib.h>
#include <Library/UefiLib.h>
#include <Library/BaseMemoryLib.h>
#include <Library/MemoryAllocationLib.h>
#include <Library/DebugLib.h>
#include <Library/UefiBootServicesTableLib.h>
#include <Protocol/UgaDraw.h>
#include <Protocol/AppleImageCodecProtocol.h>
#include "XImage.h"
//#include "picopng.h"
#include "lodepng.h"
//#define DBG(...) AsciiPrint(__VA_ARGS__);
#define DBG(...)
struct EFI_RES_ENTRY {
CHAR8 Name[64];
UINT32 DataOffset;
UINT32 DataLength;
};
struct EFI_RES {
UINT16 Magic; // 0x200 (BigEndian) or 0x02 (LE)
UINT16 Num; // LE
struct EFI_RES_ENTRY Entries[1]; //NUM - dynamic array
};
//Fool Proof
/*
02 00 14 00 61 76 61 74 61 72 2E 70 6E 67 00 00
00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
00 00 00 00 EC 05 00 00 AF 2D 00 00
61 76 61 74 61 72 40 32 78 2E 70 6E 67 00 00 00
00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
9B 33 00 00 32 84 00 00
Num=20
1. avatar.png Offset=0x05EC (LittleEndian) Size=0x2DAF Offset+Size=0x339B
2. avatar@2x.png Offset=0x339C (!) Aligned Size=0x8432 End=0xB7CE
3. Offset=0xB7CD = 0x339B+0x8432 Not alinged(!!!) File=0xB7CE Aligned
*/
XImage egDecodeAny(IN UINT8 *FileData, IN UINTN FileDataLength)
{
XImage NewXImage;
NewXImage.FromPNG(FileData, FileDataLength);
if (NewXImage.isEmpty()) {
DBG(" ..png is wrong try to decode icns\n");
NewXImage.FromICNS(FileData, FileDataLength, 128);
}
return NewXImage;
}
//
// PNG Image codec protocol instance implementation
//
EFI_STATUS
EFIAPI
RecognizeImageData (//IN APPLE_IMAGE_CODEC_PROTOCOL* This,
VOID *ImageBuffer,
UINTN ImageSize,
OUT VOID **OutBuffer
)
{
XImage Image;
if (!ImageBuffer) {
return EFI_INVALID_PARAMETER;
}
if (*(UINT16*)ImageBuffer == 0x02) {
return EFI_SUCCESS; //this is efires image
}
DBG("AppleImageCodec RecognizeImageData: Status = ");
Image = egDecodeAny((UINT8*)ImageBuffer, ImageSize);
if (Image.isEmpty()) {
DBG("EFI_UNSUPPORTED\n");
return EFI_UNSUPPORTED;
}
DBG("EFI_SUCCESS\n");
DBG(" ImageSize=%d\n", ImageSize);
// DBG("Decoded: W=%d, H=%d\n", Image.GetWidth(), Image.GetHeight());
return EFI_SUCCESS;
}
EFI_STATUS
EFIAPI
GetImageDims (//IN APPLE_IMAGE_CODEC_PROTOCOL* This,
VOID *ImageBuffer,
UINTN ImageSize,
UINT32 *ImageWidth,
UINT32 *ImageHeight
)
{
XImage Image;
DBG("AppleImageCodec GetImageDims: Status = ");
Image = egDecodeAny((UINT8*)ImageBuffer, ImageSize);
if (Image.isEmpty()) {
DBG("EFI_UNSUPPORTED\n");
return EFI_UNSUPPORTED;
}
*ImageWidth = (UINT32)Image.GetWidth();
*ImageHeight = (UINT32)Image.GetHeight();
DBG("EFI_SUCCESS, Width=%d, Height=%d\n", *ImageWidth, *ImageHeight);
DBG("ImageSize=%d\n", ImageSize);
return EFI_SUCCESS;
}
EFI_STATUS
EFIAPI
DecodeImageData (//IN APPLE_IMAGE_CODEC_PROTOCOL* This,
VOID *ImageBuffer,
UINTN ImageSize,
EFI_UGA_PIXEL **RawImageData,
UINT32 *RawImageDataSize
)
{
EFI_STATUS Status;
XImage Image;
//automatic choose format
if (!RawImageData || !RawImageDataSize) {
return EFI_INVALID_PARAMETER;
}
DBG("AppleImageCodec DecodeImageData: Status = ");
Image = egDecodeAny((UINT8*)ImageBuffer, ImageSize);
if (Image.isEmpty()) {
DBG("EFI_UNSUPPORTED\n");
return EFI_UNSUPPORTED;
}
*RawImageDataSize = (UINT32)(Image.GetWidth() * Image.GetHeight() * sizeof(EFI_UGA_PIXEL));
Status = gBS->AllocatePool(EfiBootServicesData, *RawImageDataSize, (VOID **)RawImageData);
if (!EFI_ERROR(Status)) {
CopyMem(*RawImageData, (VOID*)Image.GetPixelPtr(0,0), *RawImageDataSize);
}
DBG("EFI_SUCCESS, RawImageDataSize=%d\n", *RawImageDataSize);
// DBG("ImageBuffer=%p, ImageSize=%d\n", ImageBuffer, ImageSize);
// DBG("Decoded: W=%d, H=%d\n", Image->Width, Image->Height);
// for (int Index=0; Index<10; Index++) {
// DBG("P%d: r,g,b,a= %x, %x, %x, %x\n", Index, (*RawImageData)[Index].Red, (*RawImageData)[Index].Green, (*RawImageData)[Index].Blue, (*RawImageData)[Index].Reserved);
// }
// egFreeImage(Image);
return EFI_SUCCESS;
}
//more explanations are in OC but the functions are useless
EFI_STATUS
EFIAPI
Unknown1 (VOID* ImageBuffer, UINTN Param1, UINTN Param2, UINTN Param3)
{
DBG("AppleImageCodec unk1 called\n");
return EFI_SUCCESS;
}
EFI_STATUS
EFIAPI
Unknown2 (VOID* ImageBuffer, UINTN Param1, UINTN Param2, UINTN Param3)
{
DBG("AppleImageCodec unk2 called\n");
return EFI_SUCCESS;
}
/** Image codec protocol instance. */
APPLE_IMAGE_CODEC_PROTOCOL gAppleImageCodec = {
// Version
1,
// FileExt
0,
RecognizeImageData,
GetImageDims,
DecodeImageData,
Unknown1,
Unknown2,
};
/** Driver's entry point. Installs our StartImage to detect boot loader start. */
EFI_STATUS
EFIAPI
AppleImageCodecEntrypoint (
IN EFI_HANDLE ImageHandle,
IN EFI_SYSTEM_TABLE *SystemTable
)
{
EFI_STATUS Status;
// EFI_HANDLE NewHandle;
//
// Install instance of Apple image codec protocol for
// PNG files
//
// NewHandle = NULL; // install to a new handle
// Status = gBS->InstallMultipleProtocolInterfaces(&NewHandle, &gAppleImageCodecProtocolGuid, &gAppleImageCodec, NULL);
Status = gBS->InstallProtocolInterface (
&ImageHandle,
&gAppleImageCodecProtocolGuid,
EFI_NATIVE_INTERFACE,
(VOID *)&gAppleImageCodec
);
if (EFI_ERROR(Status)) {
DBG("AppleImageCodec: error installing protocol, Status = %s\n", efiStrError(Status));
}
return Status;
}