241 lines
6.2 KiB
C++
241 lines
6.2 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 "../Platform/Utils.h"
|
|
|
|
//#include "picopng.h"
|
|
#include "lodepng.h"
|
|
|
|
#define DBG(...) DebugLog(1, __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 aligned(!!!) 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);
|
|
}
|
|
|
|
if (NewXImage.isEmpty()) {
|
|
DBG(" ..png is wrong try to decode bmp\n");
|
|
NewXImage.FromBMP(FileData, FileDataLength);
|
|
}
|
|
|
|
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=%lld\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=%lld\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=%lld\n", ImageBuffer, ImageSize);
|
|
DBG("Decoded: W=%lld, H=%lld\n", Image.GetWidth(), Image.GetHeight());
|
|
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;
|
|
}
|