24#include <winpr/config.h>
26#include <winpr/wtypes.h>
28#include <winpr/file.h>
29#include <winpr/cast.h>
31#include <winpr/image.h>
33#if defined(WINPR_UTILS_IMAGE_PNG)
37#if defined(WINPR_UTILS_IMAGE_JPEG)
38#define INT32 INT32_WINPR
43#if defined(WINPR_UTILS_IMAGE_WEBP)
44#include <webp/encode.h>
45#include <webp/decode.h>
48#if defined(WITH_LODEPNG)
51#include <winpr/stream.h>
55#define TAG WINPR_TAG("utils.image")
57static SSIZE_T winpr_convert_from_jpeg(
const BYTE* comp_data,
size_t comp_data_bytes, UINT32* width,
58 UINT32* height, UINT32* bpp, BYTE** ppdecomp_data);
59static SSIZE_T winpr_convert_from_png(
const BYTE* comp_data,
size_t comp_data_bytes, UINT32* width,
60 UINT32* height, UINT32* bpp, BYTE** ppdecomp_data);
61static SSIZE_T winpr_convert_from_webp(
const BYTE* comp_data,
size_t comp_data_bytes, UINT32* width,
62 UINT32* height, UINT32* bpp, BYTE** ppdecomp_data);
69 Stream_Write_UINT8(s, bf->bfType[0]);
70 Stream_Write_UINT8(s, bf->bfType[1]);
71 Stream_Write_UINT32(s, bf->bfSize);
72 Stream_Write_UINT16(s, bf->bfReserved1);
73 Stream_Write_UINT16(s, bf->bfReserved2);
74 Stream_Write_UINT32(s, bf->bfOffBits);
80 static wLog* log = NULL;
88 Stream_Read_UINT8(s, bf->bfType[0]);
89 Stream_Read_UINT8(s, bf->bfType[1]);
90 Stream_Read_UINT32(s, bf->bfSize);
91 Stream_Read_UINT16(s, bf->bfReserved1);
92 Stream_Read_UINT16(s, bf->bfReserved2);
93 Stream_Read_UINT32(s, bf->bfOffBits);
97 WLog_Print(log, WLOG_ERROR,
"Invalid bitmap::bfSize=%" PRIu32
", require at least %" PRIuz,
102 if ((bf->bfType[0] !=
'B') || (bf->bfType[1] !=
'M'))
104 WLog_Print(log, WLOG_ERROR,
"Invalid bitmap header [%c%c], expected [BM]", bf->bfType[0],
108 return Stream_CheckAndLogRequiredCapacityWLog(log, s,
117 Stream_Write_UINT32(s, bi->biSize);
118 Stream_Write_INT32(s, bi->biWidth);
119 Stream_Write_INT32(s, bi->biHeight);
120 Stream_Write_UINT16(s, bi->biPlanes);
121 Stream_Write_UINT16(s, bi->biBitCount);
122 Stream_Write_UINT32(s, bi->biCompression);
123 Stream_Write_UINT32(s, bi->biSizeImage);
124 Stream_Write_INT32(s, bi->biXPelsPerMeter);
125 Stream_Write_INT32(s, bi->biYPelsPerMeter);
126 Stream_Write_UINT32(s, bi->biClrUsed);
127 Stream_Write_UINT32(s, bi->biClrImportant);
136 const size_t start = Stream_GetPosition(s);
137 Stream_Read_UINT32(s, bi->biSize);
138 Stream_Read_INT32(s, bi->biWidth);
139 Stream_Read_INT32(s, bi->biHeight);
140 Stream_Read_UINT16(s, bi->biPlanes);
141 Stream_Read_UINT16(s, bi->biBitCount);
142 Stream_Read_UINT32(s, bi->biCompression);
143 Stream_Read_UINT32(s, bi->biSizeImage);
144 Stream_Read_INT32(s, bi->biXPelsPerMeter);
145 Stream_Read_INT32(s, bi->biYPelsPerMeter);
146 Stream_Read_UINT32(s, bi->biClrUsed);
147 Stream_Read_UINT32(s, bi->biClrImportant);
149 if ((bi->biBitCount < 1) || (bi->biBitCount > 32))
151 WLog_WARN(TAG,
"invalid biBitCount=%" PRIu32, bi->biBitCount);
157 switch (bi->biCompression)
160 if (bi->biBitCount <= 8)
162 DWORD used = bi->biClrUsed;
164 used = (1 << bi->biBitCount) / 8;
165 offset +=
sizeof(
RGBQUAD) * used;
167 if (bi->biSizeImage == 0)
169 UINT32 stride = WINPR_ASSERTING_INT_CAST(
170 uint32_t, ((((bi->biWidth * bi->biBitCount) + 31) & ~31) >> 3));
171 bi->biSizeImage = WINPR_ASSERTING_INT_CAST(uint32_t, abs(bi->biHeight)) * stride;
175 offset +=
sizeof(DWORD) * 3;
178 WLog_ERR(TAG,
"unsupported biCompression %" PRIu32, bi->biCompression);
182 if (bi->biSizeImage == 0)
184 WLog_ERR(TAG,
"invalid biSizeImage %" PRIu32, bi->biSizeImage);
188 const size_t pos = Stream_GetPosition(s) - start;
189 if (bi->biSize < pos)
191 WLog_ERR(TAG,
"invalid biSize %" PRIu32
" < (actual) offset %" PRIuz, bi->biSize, pos);
196 return Stream_SafeSeek(s, bi->biSize - pos);
199BYTE* winpr_bitmap_construct_header(
size_t width,
size_t height,
size_t bpp)
205 size_t stride = (width * bpp + 7) / 8;
206 if ((stride % 4) != 0)
207 stride += 4 - (stride % 4);
209 size_t imgSize = stride * height;
210 if ((width > INT32_MAX) || (height > INT32_MAX) || (bpp > UINT16_MAX) || (imgSize > UINT32_MAX))
213 wStream* s = Stream_New(NULL, WINPR_IMAGE_BMP_HEADER_LEN);
223 bi.biSizeImage = (UINT32)imgSize;
224 bf.bfSize = bf.bfOffBits + bi.biSizeImage;
225 bi.biWidth = (INT32)width;
226 bi.biHeight = -1 * (INT32)height;
228 bi.biBitCount = (UINT16)bpp;
229 bi.biCompression = BI_RGB;
230 bi.biXPelsPerMeter = (INT32)width;
231 bi.biYPelsPerMeter = (INT32)height;
233 bi.biClrImportant = 0;
236 switch (bi.biCompression)
239 if (bi.biBitCount <= 8)
241 DWORD used = bi.biClrUsed;
243 used = (1 << bi.biBitCount) / 8;
244 offset +=
sizeof(
RGBQUAD) * used;
248 offset +=
sizeof(DWORD) * 3;
254 if (!writeBitmapFileHeader(s, &bf))
257 if (!writeBitmapInfoHeader(s, &bi))
260 if (!Stream_EnsureRemainingCapacity(s, offset))
263 Stream_Zero(s, offset);
264 result = Stream_Buffer(s);
266 Stream_Free(s, result == 0);
274WINPR_ATTR_MALLOC(free, 1)
276static
void* winpr_bitmap_write_buffer(const BYTE* data, WINPR_ATTR_UNUSED
size_t size,
277 UINT32 width, UINT32 height, UINT32 stride, UINT32 bpp,
280 WINPR_ASSERT(data || (size == 0));
283 size_t bpp_stride = 1ull * width * (bpp / 8);
284 if ((bpp_stride % 4) != 0)
285 bpp_stride += 4 - (bpp_stride % 4);
287 if (bpp_stride > UINT32_MAX)
290 wStream* s = Stream_New(NULL, 1024);
293 stride = (UINT32)bpp_stride;
295 BYTE* bmp_header = winpr_bitmap_construct_header(width, height, bpp);
298 if (!Stream_EnsureRemainingCapacity(s, WINPR_IMAGE_BMP_HEADER_LEN))
300 Stream_Write(s, bmp_header, WINPR_IMAGE_BMP_HEADER_LEN);
302 if (!Stream_EnsureRemainingCapacity(s, 1ULL * bpp_stride * height))
305 for (
size_t y = 0; y < height; y++)
307 const BYTE* line = &data[stride * y];
309 Stream_Write(s, line, stride);
310 Stream_Zero(s, bpp_stride - stride);
313 result = Stream_Buffer(s);
315 const size_t pos = Stream_GetPosition(s);
316 if (pos > UINT32_MAX)
318 *pSize = (UINT32)pos;
321 Stream_Free(s, result == NULL);
326int winpr_bitmap_write(
const char* filename,
const BYTE* data,
size_t width,
size_t height,
329 return winpr_bitmap_write_ex(filename, data, 0, width, height, bpp);
332int winpr_bitmap_write_ex(
const char* filename,
const BYTE* data,
size_t stride,
size_t width,
333 size_t height,
size_t bpp)
337 void* bmpdata = NULL;
338 const size_t bpp_stride = ((((width * bpp) + 31) & (size_t)~31) >> 3);
340 if ((stride > UINT32_MAX) || (width > UINT32_MAX) || (height > UINT32_MAX) ||
350 const size_t size = stride * 1ull * height;
351 bmpdata = winpr_bitmap_write_buffer(data, size, (UINT32)width, (UINT32)height,
352 (UINT32)stride, (UINT32)bpp, &bmpsize);
357 fp = winpr_fopen(filename,
"w+b");
360 WLog_ERR(TAG,
"failed to open file %s", filename);
364 if (fwrite(bmpdata, bmpsize, 1, fp) != 1)
376static int write_and_free(
const char* filename,
void* data,
size_t size)
383 FILE* fp = winpr_fopen(filename,
"w+b");
388 const size_t w = fwrite(data, 1, size, fp);
390 status = (w == size) ? 1 : -1;
398int winpr_image_write(
wImage* image,
const char* filename)
401 return winpr_image_write_ex(image, WINPR_ASSERTING_INT_CAST(uint32_t, image->type), filename);
404int winpr_image_write_ex(
wImage* image, UINT32 format,
const char* filename)
409 void* data = winpr_image_write_buffer(image, format, &size);
412 return write_and_free(filename, data, size);
415static int winpr_image_bitmap_read_buffer(
wImage* image,
const BYTE* buffer,
size_t size)
422 wStream* s = Stream_StaticConstInit(&sbuffer, buffer, size);
427 size_t bmpoffset = 0;
428 if (!readBitmapFileHeader(s, &bf) || !readBitmapInfoHeader(s, &bi, &bmpoffset))
431 if ((bf.bfType[0] !=
'B') || (bf.bfType[1] !=
'M'))
433 WLog_WARN(TAG,
"Invalid bitmap header %c%c", bf.bfType[0], bf.bfType[1]);
437 image->type = WINPR_IMAGE_BITMAP;
440 const size_t pos = Stream_GetPosition(s);
441 const size_t expect = bf.bfOffBits;
444 WLog_WARN(TAG,
"pos=%" PRIuz
", expected %" PRIuz
", offset=%" PRIuz, pos, expect,
450 if (!Stream_CheckAndLogRequiredCapacity(TAG, s, bi.biSizeImage))
455 WLog_WARN(TAG,
"bi.biWidth=%" PRId32, bi.biWidth);
459 image->width = (UINT32)bi.biWidth;
464 image->height = (UINT32)(-1 * bi.biHeight);
469 image->height = (UINT32)bi.biHeight;
472 if (image->height <= 0)
474 WLog_WARN(TAG,
"image->height=%" PRIu32, image->height);
478 image->bitsPerPixel = bi.biBitCount;
480 const size_t bpp = (bi.biBitCount + 7UL) / 8UL;
481 image->bytesPerPixel = WINPR_ASSERTING_INT_CAST(uint32_t, bpp);
483 image->scanline = WINPR_ASSERTING_INT_CAST(uint32_t, bi.biWidth) * image->bytesPerPixel;
484 if ((image->scanline % 4) != 0)
485 image->scanline += 4 - image->scanline % 4;
488 const size_t bmpsize = 1ULL * image->scanline * image->height;
489 if (bmpsize != bi.biSizeImage)
490 WLog_WARN(TAG,
"bmpsize=%" PRIuz
" != bi.biSizeImage=%" PRIu32, bmpsize,
494 size_t scanline = image->scanline;
495 if (bi.biSizeImage < bmpsize)
498 const size_t uscanline = image->width * bpp;
499 const size_t unaligned = image->height * uscanline;
500 if (bi.biSizeImage != unaligned)
502 scanline = uscanline;
507 const size_t asize = 1ULL * image->scanline * image->height;
509 image->data = (BYTE*)malloc(asize);
517 BYTE* pDstData = image->data;
519 for (
size_t index = 0; index < image->height; index++)
521 Stream_Read(s, pDstData, scanline);
522 Stream_Seek(s, image->scanline - scanline);
523 pDstData += scanline;
528 BYTE* pDstData = &(image->data[(image->height - 1ull) * image->scanline]);
530 for (
size_t index = 0; index < image->height; index++)
532 Stream_Read(s, pDstData, scanline);
533 Stream_Seek(s, image->scanline - scanline);
534 pDstData -= scanline;
553int winpr_image_read(
wImage* image,
const char* filename)
557 FILE* fp = winpr_fopen(filename,
"rb");
560 WLog_ERR(TAG,
"failed to open file %s", filename);
564 (void)fseek(fp, 0, SEEK_END);
565 INT64 pos = _ftelli64(fp);
566 (void)fseek(fp, 0, SEEK_SET);
570 BYTE* buffer = malloc((
size_t)pos);
573 size_t r = fread(buffer, 1, (
size_t)pos, fp);
574 if (r == (
size_t)pos)
576 status = winpr_image_read_buffer(image, buffer, (
size_t)pos);
585int winpr_image_read_buffer(
wImage* image,
const BYTE* buffer,
size_t size)
587 BYTE sig[12] = { 0 };
590 if (size <
sizeof(sig))
593 CopyMemory(sig, buffer,
sizeof(sig));
595 if ((sig[0] ==
'B') && (sig[1] ==
'M'))
597 image->type = WINPR_IMAGE_BITMAP;
598 status = winpr_image_bitmap_read_buffer(image, buffer, size);
600 else if ((sig[0] ==
'R') && (sig[1] ==
'I') && (sig[2] ==
'F') && (sig[3] ==
'F') &&
601 (sig[8] ==
'W') && (sig[9] ==
'E') && (sig[10] ==
'B') && (sig[11] ==
'P'))
603 image->type = WINPR_IMAGE_WEBP;
604 const SSIZE_T rc = winpr_convert_from_webp(buffer, size, &image->width, &image->height,
605 &image->bitsPerPixel, &image->data);
608 image->bytesPerPixel = (image->bitsPerPixel + 7) / 8;
609 image->scanline = image->width * image->bytesPerPixel;
613 else if ((sig[0] == 0xFF) && (sig[1] == 0xD8) && (sig[2] == 0xFF) && (sig[3] == 0xE0) &&
614 (sig[6] == 0x4A) && (sig[7] == 0x46) && (sig[8] == 0x49) && (sig[9] == 0x46) &&
617 image->type = WINPR_IMAGE_JPEG;
618 const SSIZE_T rc = winpr_convert_from_jpeg(buffer, size, &image->width, &image->height,
619 &image->bitsPerPixel, &image->data);
622 image->bytesPerPixel = (image->bitsPerPixel + 7) / 8;
623 image->scanline = image->width * image->bytesPerPixel;
627 else if ((sig[0] == 0x89) && (sig[1] ==
'P') && (sig[2] ==
'N') && (sig[3] ==
'G') &&
628 (sig[4] ==
'\r') && (sig[5] ==
'\n') && (sig[6] == 0x1A) && (sig[7] ==
'\n'))
630 image->type = WINPR_IMAGE_PNG;
631 const SSIZE_T rc = winpr_convert_from_png(buffer, size, &image->width, &image->height,
632 &image->bitsPerPixel, &image->data);
635 image->bytesPerPixel = (image->bitsPerPixel + 7) / 8;
636 image->scanline = image->width * image->bytesPerPixel;
644wImage* winpr_image_new(
void)
654void winpr_image_free(
wImage* image, BOOL bFreeBuffer)
665static void* winpr_convert_to_jpeg(WINPR_ATTR_UNUSED
const void* data,
666 WINPR_ATTR_UNUSED
size_t size, WINPR_ATTR_UNUSED UINT32 width,
667 WINPR_ATTR_UNUSED UINT32 height, WINPR_ATTR_UNUSED UINT32 stride,
668 WINPR_ATTR_UNUSED UINT32 bpp, WINPR_ATTR_UNUSED UINT32* pSize)
670 WINPR_ASSERT(data || (size == 0));
675#if !defined(WINPR_UTILS_IMAGE_JPEG)
676 WLog_WARN(TAG,
"JPEG not supported in this build");
679 BYTE* outbuffer = NULL;
680 unsigned long outsize = 0;
681 struct jpeg_compress_struct cinfo = { 0 };
683 const size_t expect1 = 1ull * stride * height;
684 const size_t bytes = (bpp + 7) / 8;
685 const size_t expect2 = 1ull * width * height * bytes;
686 if (expect1 < expect2)
692 struct jpeg_error_mgr jerr = { 0 };
693 cinfo.err = jpeg_std_error(&jerr);
695 jpeg_create_compress(&cinfo);
696 jpeg_mem_dest(&cinfo, &outbuffer, &outsize);
698 cinfo.image_width = width;
699 cinfo.image_height = height;
700 WINPR_ASSERT(bpp <= INT32_MAX / 8);
701 cinfo.input_components = (int)(bpp + 7) / 8;
702 cinfo.in_color_space = (bpp > 24) ? JCS_EXT_BGRA : JCS_EXT_BGR;
703 cinfo.data_precision = 8;
705 jpeg_set_defaults(&cinfo);
706 jpeg_set_quality(&cinfo, 100, TRUE);
708 cinfo.comp_info[0].h_samp_factor = cinfo.comp_info[0].v_samp_factor = 1;
710 jpeg_start_compress(&cinfo, TRUE);
712 const JSAMPLE* cdata = data;
713 for (
size_t x = 0; x < height; x++)
715 WINPR_ASSERT(x * stride <= UINT32_MAX);
716 const JDIMENSION offset = (JDIMENSION)x * stride;
720 JSAMPLE* coffset = WINPR_CAST_CONST_PTR_AWAY(&cdata[offset], JSAMPLE*);
721 if (jpeg_write_scanlines(&cinfo, &coffset, 1) != 1)
726 jpeg_finish_compress(&cinfo);
727 jpeg_destroy_compress(&cinfo);
729 WINPR_ASSERT(outsize <= UINT32_MAX);
730 *pSize = (UINT32)outsize;
736SSIZE_T winpr_convert_from_jpeg(WINPR_ATTR_UNUSED
const BYTE* comp_data,
737 WINPR_ATTR_UNUSED
size_t comp_data_bytes,
738 WINPR_ATTR_UNUSED UINT32* width, WINPR_ATTR_UNUSED UINT32* height,
739 WINPR_ATTR_UNUSED UINT32* bpp,
740 WINPR_ATTR_UNUSED BYTE** ppdecomp_data)
743 WINPR_ASSERT(comp_data || (comp_data_bytes == 0));
745 WINPR_ASSERT(height);
747 WINPR_ASSERT(ppdecomp_data);
749#if !defined(WINPR_UTILS_IMAGE_JPEG)
750 WLog_WARN(TAG,
"JPEG not supported in this build");
753 struct jpeg_decompress_struct cinfo = { 0 };
754 struct jpeg_error_mgr jerr;
756 BYTE* decomp_data = NULL;
758 cinfo.err = jpeg_std_error(&jerr);
759 jpeg_create_decompress(&cinfo);
760 jpeg_mem_src(&cinfo, comp_data, comp_data_bytes);
762 if (jpeg_read_header(&cinfo, 1) != JPEG_HEADER_OK)
765 cinfo.out_color_space = cinfo.num_components > 3 ? JCS_EXT_RGBA : JCS_EXT_BGR;
767 *width = WINPR_ASSERTING_INT_CAST(uint32_t, cinfo.image_width);
768 *height = WINPR_ASSERTING_INT_CAST(uint32_t, cinfo.image_height);
769 *bpp = WINPR_ASSERTING_INT_CAST(uint32_t, cinfo.num_components * 8);
771 if (!jpeg_start_decompress(&cinfo))
775 1ULL * cinfo.image_width * WINPR_ASSERTING_INT_CAST(uint32_t, cinfo.num_components);
777 if ((stride == 0) || (cinfo.image_height == 0))
780 decomp_data = calloc(stride, cinfo.image_height);
783 while (cinfo.output_scanline < cinfo.image_height)
785 JSAMPROW row = &decomp_data[cinfo.output_scanline * stride];
786 if (jpeg_read_scanlines(&cinfo, &row, 1) != 1)
789 const size_t ssize = stride * cinfo.image_height;
790 WINPR_ASSERT(ssize < SSIZE_MAX);
791 size = (SSIZE_T)ssize;
793 jpeg_finish_decompress(&cinfo);
796 jpeg_destroy_decompress(&cinfo);
797 *ppdecomp_data = decomp_data;
802static void* winpr_convert_to_webp(WINPR_ATTR_UNUSED
const void* data,
803 WINPR_ATTR_UNUSED
size_t size, WINPR_ATTR_UNUSED UINT32 width,
804 WINPR_ATTR_UNUSED UINT32 height, WINPR_ATTR_UNUSED UINT32 stride,
805 WINPR_ATTR_UNUSED UINT32 bpp, UINT32* pSize)
807 WINPR_ASSERT(data || (size == 0));
812#if !defined(WINPR_UTILS_IMAGE_WEBP)
813 WLog_WARN(TAG,
"WEBP not supported in this build");
817 uint8_t* pDstData = NULL;
818 WINPR_ASSERT(width <= INT32_MAX);
819 WINPR_ASSERT(height <= INT32_MAX);
820 WINPR_ASSERT(stride <= INT32_MAX);
824 dstSize = WebPEncodeLosslessBGRA(data, (
int)width, (
int)height, (
int)stride, &pDstData);
827 dstSize = WebPEncodeLosslessBGR(data, (
int)width, (
int)height, (
int)stride, &pDstData);
833 void* rc = malloc(dstSize);
836 memcpy(rc, pDstData, dstSize);
838 WINPR_ASSERT(dstSize <= UINT32_MAX);
839 *pSize = (UINT32)dstSize;
846SSIZE_T winpr_convert_from_webp(WINPR_ATTR_UNUSED
const BYTE* comp_data,
847 WINPR_ATTR_UNUSED
size_t comp_data_bytes, UINT32* width,
848 UINT32* height, UINT32* bpp, BYTE** ppdecomp_data)
850 WINPR_ASSERT(comp_data || (comp_data_bytes == 0));
852 WINPR_ASSERT(height);
854 WINPR_ASSERT(ppdecomp_data);
859 *ppdecomp_data = NULL;
860#if !defined(WINPR_UTILS_IMAGE_WEBP)
861 WLog_WARN(TAG,
"WEBP not supported in this build");
867 uint8_t* dst = WebPDecodeBGRA(comp_data, comp_data_bytes, &w, &h);
868 if (!dst || (w < 0) || (h < 0))
874 *width = WINPR_ASSERTING_INT_CAST(uint32_t, w);
875 *height = WINPR_ASSERTING_INT_CAST(uint32_t, h);
877 *ppdecomp_data = dst;
882#if defined(WINPR_UTILS_IMAGE_PNG)
889static void png_write_data(png_structp png_ptr, png_bytep data, png_size_t length)
892 struct png_mem_encode* p =
893 (
struct png_mem_encode*)png_get_io_ptr(png_ptr);
894 size_t nsize = p->size + length;
899 char* tmp = realloc(p->buffer, nsize);
904 p->buffer = malloc(nsize);
907 png_error(png_ptr,
"Write Error");
910 memcpy(p->buffer + p->size, data, length);
915static void png_flush(WINPR_ATTR_UNUSED png_structp png_ptr)
919static SSIZE_T save_png_to_buffer(UINT32 bpp, UINT32 width, uint32_t stride, UINT32 height,
920 const uint8_t* data,
size_t size,
void** pDstData)
923 png_structp png_ptr = NULL;
924 png_infop info_ptr = NULL;
925 png_byte** row_pointers = NULL;
926 struct png_mem_encode state = { 0 };
930 if (!data || (size == 0))
933 WINPR_ASSERT(pDstData);
935 if (size < (1ULL * stride * height))
939 png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
944 info_ptr = png_create_info_struct(png_ptr);
945 if (info_ptr == NULL)
949 if (setjmp(png_jmpbuf(png_ptr)))
953 int colorType = PNG_COLOR_TYPE_PALETTE;
955 colorType = PNG_COLOR_TYPE_RGB;
957 colorType = PNG_COLOR_TYPE_RGB;
959 colorType = PNG_COLOR_TYPE_RGBA;
961 png_set_IHDR(png_ptr, info_ptr, width, height, 8, colorType, PNG_INTERLACE_NONE,
962 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
965 row_pointers = (png_byte**)png_malloc(png_ptr, height *
sizeof(png_byte*));
966 for (
size_t y = 0; y < height; ++y)
968 const uint8_t* line = &data[y * stride];
969 uint8_t* row = png_malloc(png_ptr,
sizeof(uint8_t) * stride);
970 row_pointers[y] = (png_byte*)row;
971 for (
size_t x = 0; x < width; ++x)
985 png_set_write_fn(png_ptr, &state, png_write_data, png_flush);
986 png_set_rows(png_ptr, info_ptr, row_pointers);
987 png_write_png(png_ptr, info_ptr, PNG_TRANSFORM_BGR, NULL);
990 for (
size_t y = 0; y < height; y++)
991 png_free(png_ptr, row_pointers[y]);
992 png_free(png_ptr, (
void*)row_pointers);
995 if (state.size > SSIZE_MAX)
997 rc = (SSIZE_T)state.size;
998 *pDstData = state.buffer;
1000 png_destroy_write_struct(&png_ptr, &info_ptr);
1009 png_uint_32 bufsize;
1010 png_uint_32 current_pos;
1011} MEMORY_READER_STATE;
1013static void read_data_memory(png_structp png_ptr, png_bytep data,
size_t length)
1015 MEMORY_READER_STATE* f = png_get_io_ptr(png_ptr);
1016 if (length > (f->bufsize - f->current_pos))
1017 png_error(png_ptr,
"read error in read_data_memory (loadpng)");
1020 memcpy(data, f->buffer + f->current_pos, length);
1021 f->current_pos += length;
1025static void* winpr_read_png_from_buffer(
const void* data,
size_t SrcSize,
size_t* pSize,
1026 UINT32* pWidth, UINT32* pHeight, UINT32* pBpp)
1029 png_uint_32 width = 0;
1030 png_uint_32 height = 0;
1033 int interlace_type = 0;
1034 int transforms = PNG_TRANSFORM_IDENTITY;
1035 MEMORY_READER_STATE memory_reader_state = { 0 };
1036 png_bytepp row_pointers = NULL;
1037 png_infop info_ptr = NULL;
1038 if (SrcSize > UINT32_MAX)
1041 png_structp png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
1044 info_ptr = png_create_info_struct(png_ptr);
1048 memory_reader_state.buffer = WINPR_CAST_CONST_PTR_AWAY(data, png_bytep);
1049 memory_reader_state.bufsize = (UINT32)SrcSize;
1050 memory_reader_state.current_pos = 0;
1052 png_set_read_fn(png_ptr, &memory_reader_state, read_data_memory);
1054 transforms |= PNG_TRANSFORM_BGR;
1055 png_read_png(png_ptr, info_ptr, transforms, NULL);
1057 if (png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type, &interlace_type,
1061 WINPR_ASSERT(bit_depth >= 0);
1062 const png_byte channelcount = png_get_channels(png_ptr, info_ptr);
1063 const size_t bpp = channelcount * (size_t)bit_depth;
1065 row_pointers = png_get_rows(png_ptr, info_ptr);
1068 const size_t stride = 1ULL * width * bpp / 8ull;
1069 const size_t png_stride = png_get_rowbytes(png_ptr, info_ptr);
1070 const size_t size = 1ULL * width * height * bpp / 8ull;
1071 const size_t copybytes = stride > png_stride ? png_stride : stride;
1077 for (png_uint_32 i = 0; i < height; i++)
1079 memcpy(cur, row_pointers[i], copybytes);
1085 WINPR_ASSERT(bpp <= UINT32_MAX);
1086 *pBpp = (UINT32)bpp;
1091 png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
1096static void* winpr_convert_to_png(WINPR_ATTR_UNUSED
const void* data, WINPR_ATTR_UNUSED
size_t size,
1097 WINPR_ATTR_UNUSED UINT32 width, WINPR_ATTR_UNUSED UINT32 height,
1098 WINPR_ATTR_UNUSED UINT32 stride, WINPR_ATTR_UNUSED UINT32 bpp,
1101 WINPR_ASSERT(data || (size == 0));
1102 WINPR_ASSERT(pSize);
1106#if defined(WINPR_UTILS_IMAGE_PNG)
1108 SSIZE_T rc = save_png_to_buffer(bpp, width, stride, height, data, size, &dst);
1111 *pSize = (UINT32)rc;
1113#elif defined(WITH_LODEPNG)
1122 rc = lodepng_encode32(&dst, &dstsize, data, width, height);
1125 rc = lodepng_encode24(&dst, &dstsize, data, width, height);
1132 *pSize = (UINT32)dstsize;
1136 WLog_WARN(TAG,
"PNG not supported in this build");
1141SSIZE_T winpr_convert_from_png(WINPR_ATTR_UNUSED
const BYTE* comp_data,
1142 WINPR_ATTR_UNUSED
size_t comp_data_bytes,
1143 WINPR_ATTR_UNUSED UINT32* width, WINPR_ATTR_UNUSED UINT32* height,
1144 WINPR_ATTR_UNUSED UINT32* bpp,
1145 WINPR_ATTR_UNUSED BYTE** ppdecomp_data)
1147#if defined(WINPR_UTILS_IMAGE_PNG)
1150 winpr_read_png_from_buffer(comp_data, comp_data_bytes, &len, width, height, bpp);
1151 if (!*ppdecomp_data)
1153 return (SSIZE_T)len;
1154#elif defined(WITH_LODEPNG)
1156 return lodepng_decode32((
unsigned char**)ppdecomp_data, width, height, comp_data,
1159 WLog_WARN(TAG,
"PNG not supported in this build");
1164BOOL winpr_image_format_is_supported(UINT32 format)
1168 case WINPR_IMAGE_BITMAP:
1169#if defined(WINPR_UTILS_IMAGE_PNG) || defined(WITH_LODEPNG)
1170 case WINPR_IMAGE_PNG:
1172#if defined(WINPR_UTILS_IMAGE_JPEG)
1173 case WINPR_IMAGE_JPEG:
1175#if defined(WINPR_UTILS_IMAGE_WEBP)
1176 case WINPR_IMAGE_WEBP:
1184static BYTE* convert(
const wImage* image,
size_t* pstride, WINPR_ATTR_UNUSED UINT32 flags)
1186 WINPR_ASSERT(image);
1187 WINPR_ASSERT(pstride);
1190 if (image->bitsPerPixel < 24)
1193 const size_t stride = image->width * 4ull;
1194 BYTE* data = calloc(stride, image->height);
1197 for (
size_t y = 0; y < image->height; y++)
1199 const BYTE* srcLine = &image->data[image->scanline * y];
1200 BYTE* dstLine = &data[stride * y];
1201 if (image->bitsPerPixel == 32)
1202 memcpy(dstLine, srcLine, stride);
1205 for (
size_t x = 0; x < image->width; x++)
1207 const BYTE* src = &srcLine[image->bytesPerPixel * x];
1208 BYTE* dst = &dstLine[4ull * x];
1225static BOOL compare_byte_relaxed(BYTE a, BYTE b, UINT32 flags)
1229 if ((flags & WINPR_IMAGE_CMP_FUZZY) != 0)
1231 const int diff = abs((
int)a) - abs((
int)b);
1244static BOOL compare_pixel(
const BYTE* pa,
const BYTE* pb, UINT32 flags)
1249 if (!compare_byte_relaxed(*pa++, *pb++, flags))
1251 if (!compare_byte_relaxed(*pa++, *pb++, flags))
1253 if (!compare_byte_relaxed(*pa++, *pb++, flags))
1255 if ((flags & WINPR_IMAGE_CMP_IGNORE_ALPHA) == 0)
1257 if (!compare_byte_relaxed(*pa++, *pb++, flags))
1263BOOL winpr_image_equal(
const wImage* imageA,
const wImage* imageB, UINT32 flags)
1265 if (imageA == imageB)
1267 if (!imageA || !imageB)
1270 if (imageA->height != imageB->height)
1272 if (imageA->width != imageB->width)
1275 if ((flags & WINPR_IMAGE_CMP_IGNORE_DEPTH) == 0)
1277 if (imageA->bitsPerPixel != imageB->bitsPerPixel)
1279 if (imageA->bytesPerPixel != imageB->bytesPerPixel)
1286 BYTE* dataA = convert(imageA, &astride, flags);
1287 BYTE* dataB = convert(imageA, &bstride, flags);
1288 if (dataA && dataB && (astride == bstride))
1291 for (
size_t y = 0; y < imageA->height; y++)
1293 const BYTE* lineA = &dataA[astride * y];
1294 const BYTE* lineB = &dataB[bstride * y];
1296 for (
size_t x = 0; x < imageA->width; x++)
1298 const BYTE* pa = &lineA[x * 4ull];
1299 const BYTE* pb = &lineB[x * 4ull];
1301 if (!compare_pixel(pa, pb, flags))
1311const char* winpr_image_format_mime(UINT32 format)
1315 case WINPR_IMAGE_BITMAP:
1317 case WINPR_IMAGE_PNG:
1319 case WINPR_IMAGE_WEBP:
1320 return "image/webp";
1321 case WINPR_IMAGE_JPEG:
1322 return "image/jpeg";
1328const char* winpr_image_format_extension(UINT32 format)
1332 case WINPR_IMAGE_BITMAP:
1334 case WINPR_IMAGE_PNG:
1336 case WINPR_IMAGE_WEBP:
1338 case WINPR_IMAGE_JPEG:
1345void* winpr_image_write_buffer(
wImage* image, UINT32 format,
size_t* psize)
1347 WINPR_ASSERT(image);
1350 case WINPR_IMAGE_BITMAP:
1353 size_t size = 1ull * image->height * image->scanline;
1354 void* data = winpr_bitmap_write_buffer(image->data, size, image->width, image->height,
1355 image->scanline, image->bitsPerPixel, &outsize);
1359 case WINPR_IMAGE_WEBP:
1362 size_t size = 1ull * image->height * image->scanline;
1363 void* data = winpr_convert_to_webp(image->data, size, image->width, image->height,
1364 image->scanline, image->bitsPerPixel, &outsize);
1368 case WINPR_IMAGE_JPEG:
1371 size_t size = 1ull * image->height * image->scanline;
1372 void* data = winpr_convert_to_jpeg(image->data, size, image->width, image->height,
1373 image->scanline, image->bitsPerPixel, &outsize);
1377 case WINPR_IMAGE_PNG:
1380 size_t size = 1ull * image->height * image->scanline;
1381 void* data = winpr_convert_to_png(image->data, size, image->width, image->height,
1382 image->scanline, image->bitsPerPixel, &outsize);