24#include <winpr/assert.h>
25#include <winpr/cast.h>
26#include <freerdp/config.h>
28#include <freerdp/codec/interleaved.h>
29#include <freerdp/log.h>
31#define TAG FREERDP_TAG("codec")
33#define UNROLL_BODY(_exp, _count) \
36 for (size_t x = 0; x < (_count); x++) \
45#define UNROLL_MULTIPLE(_condition, _exp, _count) \
48 while ((_condition) >= (_count)) \
50 UNROLL_BODY(_exp, _count); \
51 (_condition) -= (_count); \
55#define UNROLL(_condition, _exp) \
58 UNROLL_MULTIPLE(_condition, _exp, 16); \
59 UNROLL_MULTIPLE(_condition, _exp, 4); \
60 UNROLL_MULTIPLE(_condition, _exp, 1); \
70#define REGULAR_BG_RUN 0x00
71#define MEGA_MEGA_BG_RUN 0xF0
72#define REGULAR_FG_RUN 0x01
73#define MEGA_MEGA_FG_RUN 0xF1
74#define LITE_SET_FG_FG_RUN 0x0C
75#define MEGA_MEGA_SET_FG_RUN 0xF6
76#define LITE_DITHERED_RUN 0x0E
77#define MEGA_MEGA_DITHERED_RUN 0xF8
78#define REGULAR_COLOR_RUN 0x03
79#define MEGA_MEGA_COLOR_RUN 0xF3
80#define REGULAR_FGBG_IMAGE 0x02
81#define MEGA_MEGA_FGBG_IMAGE 0xF2
82#define LITE_SET_FG_FGBG_IMAGE 0x0D
83#define MEGA_MEGA_SET_FGBG_IMAGE 0xF7
84#define REGULAR_COLOR_IMAGE 0x04
85#define MEGA_MEGA_COLOR_IMAGE 0xF4
86#define SPECIAL_FGBG_1 0xF9
87#define SPECIAL_FGBG_2 0xFA
88#define SPECIAL_WHITE 0xFD
89#define SPECIAL_BLACK 0xFE
91#define BLACK_PIXEL 0x000000
95static const BYTE g_MaskSpecialFgBg1 = 0x03;
96static const BYTE g_MaskSpecialFgBg2 = 0x05;
98static const BYTE g_MaskRegularRunLength = 0x1F;
99static const BYTE g_MaskLiteRunLength = 0x0F;
101#if defined(WITH_DEBUG_CODECS)
102static const char* rle_code_str(UINT32 code)
107 return "REGULAR_BG_RUN";
108 case MEGA_MEGA_BG_RUN:
109 return "MEGA_MEGA_BG_RUN";
111 return "REGULAR_FG_RUN";
112 case MEGA_MEGA_FG_RUN:
113 return "MEGA_MEGA_FG_RUN";
114 case LITE_SET_FG_FG_RUN:
115 return "LITE_SET_FG_FG_RUN";
116 case MEGA_MEGA_SET_FG_RUN:
117 return "MEGA_MEGA_SET_FG_RUN";
118 case LITE_DITHERED_RUN:
119 return "LITE_DITHERED_RUN";
120 case MEGA_MEGA_DITHERED_RUN:
121 return "MEGA_MEGA_DITHERED_RUN";
122 case REGULAR_COLOR_RUN:
123 return "REGULAR_COLOR_RUN";
124 case MEGA_MEGA_COLOR_RUN:
125 return "MEGA_MEGA_COLOR_RUN";
126 case REGULAR_FGBG_IMAGE:
127 return "REGULAR_FGBG_IMAGE";
128 case MEGA_MEGA_FGBG_IMAGE:
129 return "MEGA_MEGA_FGBG_IMAGE";
130 case LITE_SET_FG_FGBG_IMAGE:
131 return "LITE_SET_FG_FGBG_IMAGE";
132 case MEGA_MEGA_SET_FGBG_IMAGE:
133 return "MEGA_MEGA_SET_FGBG_IMAGE";
134 case REGULAR_COLOR_IMAGE:
135 return "REGULAR_COLOR_IMAGE";
136 case MEGA_MEGA_COLOR_IMAGE:
137 return "MEGA_MEGA_COLOR_IMAGE";
139 return "SPECIAL_FGBG_1";
141 return "SPECIAL_FGBG_2";
143 return "SPECIAL_WHITE";
145 return "SPECIAL_BLACK";
152#define buffer_within_range(pbSrc, size, pbEnd) \
153 buffer_within_range_((pbSrc), (size), (pbEnd), __func__, __FILE__, __LINE__)
154static inline BOOL buffer_within_range_(
const void* pbSrc,
size_t size,
const void* pbEnd,
155 const char* fkt,
const char* file,
size_t line)
161 if ((
const char*)pbSrc + size > (
const char*)pbEnd)
163 WLog_ERR(TAG,
"[%s:%" PRIuz
"] pbSrc=%p + %" PRIuz
" > pbEnd=%p", fkt, line, pbSrc, size,
174static inline UINT32 ExtractCodeId(BYTE bOrderHdr)
176 if ((bOrderHdr & 0xC0U) != 0xC0U)
181 return bOrderHdr >> 5;
183 else if ((bOrderHdr & 0xF0U) == 0xF0U)
193 return bOrderHdr >> 4;
200static UINT ExtractRunLengthRegularFgBg(
const BYTE* pbOrderHdr,
const BYTE* pbEnd, UINT32* advance)
204 WINPR_ASSERT(pbOrderHdr);
206 WINPR_ASSERT(advance);
208 runLength = (*pbOrderHdr) & g_MaskRegularRunLength;
211 if (!buffer_within_range(pbOrderHdr, 2, pbEnd))
216 runLength = *(pbOrderHdr + 1) + 1;
220 runLength = runLength * 8;
225static UINT ExtractRunLengthLiteFgBg(
const BYTE* pbOrderHdr,
const BYTE* pbEnd, UINT32* advance)
229 WINPR_ASSERT(pbOrderHdr);
231 WINPR_ASSERT(advance);
233 runLength = *pbOrderHdr & g_MaskLiteRunLength;
236 if (!buffer_within_range(pbOrderHdr, 2, pbEnd))
241 runLength = *(pbOrderHdr + 1) + 1;
245 runLength = runLength * 8;
250static UINT ExtractRunLengthRegular(
const BYTE* pbOrderHdr,
const BYTE* pbEnd, UINT32* advance)
254 WINPR_ASSERT(pbOrderHdr);
256 WINPR_ASSERT(advance);
258 runLength = *pbOrderHdr & g_MaskRegularRunLength;
261 if (!buffer_within_range(pbOrderHdr, 2, pbEnd))
266 runLength = *(pbOrderHdr + 1) + 32;
273static UINT ExtractRunLengthMegaMega(
const BYTE* pbOrderHdr,
const BYTE* pbEnd, UINT32* advance)
277 WINPR_ASSERT(pbOrderHdr);
279 WINPR_ASSERT(advance);
281 if (!buffer_within_range(pbOrderHdr, 3, pbEnd))
287 runLength = ((UINT16)pbOrderHdr[1]) | ((((UINT16)pbOrderHdr[2]) << 8) & 0xFF00);
293static UINT ExtractRunLengthLite(
const BYTE* pbOrderHdr,
const BYTE* pbEnd, UINT32* advance)
297 WINPR_ASSERT(pbOrderHdr);
299 WINPR_ASSERT(advance);
301 runLength = *pbOrderHdr & g_MaskLiteRunLength;
304 if (!buffer_within_range(pbOrderHdr, 2, pbEnd))
309 runLength = *(pbOrderHdr + 1) + 16;
315static inline UINT32 ExtractRunLength(UINT32 code,
const BYTE* pbOrderHdr,
const BYTE* pbEnd,
318 UINT32 runLength = 0;
321 WINPR_ASSERT(pbOrderHdr);
323 WINPR_ASSERT(advance);
325#if defined(WITH_DEBUG_CODECS)
326 WLog_VRB(TAG,
"extracting %s", rle_code_str(code));
330 if (!buffer_within_range(pbOrderHdr, 0, pbEnd))
335 case REGULAR_FGBG_IMAGE:
336 runLength = ExtractRunLengthRegularFgBg(pbOrderHdr, pbEnd, &ladvance);
339 case LITE_SET_FG_FGBG_IMAGE:
340 runLength = ExtractRunLengthLiteFgBg(pbOrderHdr, pbEnd, &ladvance);
345 case REGULAR_COLOR_RUN:
346 case REGULAR_COLOR_IMAGE:
347 runLength = ExtractRunLengthRegular(pbOrderHdr, pbEnd, &ladvance);
350 case LITE_SET_FG_FG_RUN:
351 case LITE_DITHERED_RUN:
352 runLength = ExtractRunLengthLite(pbOrderHdr, pbEnd, &ladvance);
355 case MEGA_MEGA_BG_RUN:
356 case MEGA_MEGA_FG_RUN:
357 case MEGA_MEGA_SET_FG_RUN:
358 case MEGA_MEGA_DITHERED_RUN:
359 case MEGA_MEGA_COLOR_RUN:
360 case MEGA_MEGA_FGBG_IMAGE:
361 case MEGA_MEGA_SET_FGBG_IMAGE:
362 case MEGA_MEGA_COLOR_IMAGE:
363 runLength = ExtractRunLengthMegaMega(pbOrderHdr, pbEnd, &ladvance);
376#define ensure_capacity(start, end, size, base) \
377 ensure_capacity_((start), (end), (size), (base), __func__, __FILE__, __LINE__)
378static inline BOOL ensure_capacity_(
const BYTE* start,
const BYTE* end,
size_t size,
size_t base,
379 const char* fkt, WINPR_ATTR_UNUSED
const char* file,
382 const size_t available = (uintptr_t)end - (uintptr_t)start;
383 const BOOL rc = available >= size * base;
384 const BOOL res = rc && (start <= end);
388 "[%s:%" PRIuz
"] failed: start=%p <= end=%p, available=%" PRIuz
" >= size=%" PRIuz
390 fkt, line, WINPR_CXX_COMPAT_CAST(
const void*, start),
391 WINPR_CXX_COMPAT_CAST(
const void*, end), available, size, base);
395static inline void write_pixel_8(BYTE* _buf, BYTE _pix)
401static inline void write_pixel_24(BYTE* _buf, UINT32 _pix)
404 (_buf)[0] = (BYTE)(_pix);
405 (_buf)[1] = (BYTE)((_pix) >> 8);
406 (_buf)[2] = (BYTE)((_pix) >> 16);
409static inline void write_pixel_16(BYTE* _buf, UINT16 _pix)
412 _buf[0] = _pix & 0xFF;
413 _buf[1] = (_pix >> 8) & 0xFF;
420#undef WRITEFIRSTLINEFGBGIMAGE
428#define WHITE_PIXEL 0xFF
429#define DESTWRITEPIXEL(_buf, _pix) \
432 write_pixel_8(_buf, _pix); \
435#define DESTREADPIXEL(_pix, _buf) _pix = (_buf)[0]
436#define SRCREADPIXEL(_pix, _buf) \
439 (_pix) = (_buf)[0]; \
443#define WRITEFGBGIMAGE WriteFgBgImage8to8
444#define WRITEFIRSTLINEFGBGIMAGE WriteFirstLineFgBgImage8to8
445#define RLEDECOMPRESS RleDecompress8to8
447#undef ENSURE_CAPACITY
448#define ENSURE_CAPACITY(_start, _end, _size) ensure_capacity(_start, _end, _size, 1)
449#include "include/bitmap.h"
455#undef WRITEFIRSTLINEFGBGIMAGE
463#define WHITE_PIXEL 0xFFFF
464#define DESTWRITEPIXEL(_buf, _pix) \
467 write_pixel_16(_buf, _pix); \
470#define DESTREADPIXEL(_pix, _buf) _pix = ((UINT16*)(_buf))[0]
471#define SRCREADPIXEL(_pix, _buf) \
474 (_pix) = WINPR_ASSERTING_INT_CAST(UINT16, (_buf)[0] | (((_buf)[1] << 8) & 0xFF00)); \
477#define WRITEFGBGIMAGE WriteFgBgImage16to16
478#define WRITEFIRSTLINEFGBGIMAGE WriteFirstLineFgBgImage16to16
479#define RLEDECOMPRESS RleDecompress16to16
481#undef ENSURE_CAPACITY
482#define ENSURE_CAPACITY(_start, _end, _size) ensure_capacity(_start, _end, _size, 2)
483#include "include/bitmap.h"
489#undef WRITEFIRSTLINEFGBGIMAGE
497#define WHITE_PIXEL 0xffffff
498#define DESTWRITEPIXEL(_buf, _pix) \
501 write_pixel_24(_buf, _pix); \
504#define DESTREADPIXEL(_pix, _buf) \
505 _pix = (_buf)[0] | (((_buf)[1] << 8) & 0xFF00) | (((_buf)[2] << 16) & 0xFF0000)
506#define SRCREADPIXEL(_pix, _buf) \
509 (_pix) = (_buf)[0] | (((_buf)[1] << 8) & 0xFF00) | (((_buf)[2] << 16) & 0xFF0000); \
513#define WRITEFGBGIMAGE WriteFgBgImage24to24
514#define WRITEFIRSTLINEFGBGIMAGE WriteFirstLineFgBgImage24to24
515#define RLEDECOMPRESS RleDecompress24to24
517#undef ENSURE_CAPACITY
518#define ENSURE_CAPACITY(_start, _end, _size) ensure_capacity(_start, _end, _size, 3)
519#include "include/bitmap.h"
521struct S_BITMAP_INTERLEAVED_CONTEXT
531BOOL interleaved_decompress(BITMAP_INTERLEAVED_CONTEXT* WINPR_RESTRICT interleaved,
532 const BYTE* WINPR_RESTRICT pSrcData, UINT32 SrcSize, UINT32 nSrcWidth,
533 UINT32 nSrcHeight, UINT32 bpp, BYTE* WINPR_RESTRICT pDstData,
534 UINT32 DstFormat, UINT32 nDstStep, UINT32 nXDst, UINT32 nYDst,
535 UINT32 nDstWidth, UINT32 nDstHeight,
536 const gdiPalette* WINPR_RESTRICT palette)
539 UINT32 SrcFormat = 0;
540 UINT32 BufferSize = 0;
542 if (!interleaved || !pSrcData || !pDstData)
544 WLog_ERR(TAG,
"invalid arguments: interleaved=%p, pSrcData=%p, pDstData=%p",
545 WINPR_CXX_COMPAT_CAST(
const void*, interleaved),
546 WINPR_CXX_COMPAT_CAST(
const void*, pSrcData),
547 WINPR_CXX_COMPAT_CAST(
const void*, pDstData));
554 scanline = nSrcWidth * 3;
555 SrcFormat = PIXEL_FORMAT_BGR24;
559 scanline = nSrcWidth * 2;
560 SrcFormat = PIXEL_FORMAT_RGB16;
564 scanline = nSrcWidth * 2;
565 SrcFormat = PIXEL_FORMAT_RGB15;
569 scanline = nSrcWidth;
570 SrcFormat = PIXEL_FORMAT_RGB8;
574 WLog_ERR(TAG,
"Invalid color depth %" PRIu32
"", bpp);
578 BufferSize = scanline * nSrcHeight;
580 if (BufferSize > interleaved->TempSize)
582 interleaved->TempBuffer =
583 winpr_aligned_recalloc(interleaved->TempBuffer, BufferSize,
sizeof(BYTE), 16);
584 interleaved->TempSize = BufferSize;
587 if (!interleaved->TempBuffer)
589 WLog_ERR(TAG,
"interleaved->TempBuffer=NULL");
596 if (!RleDecompress24to24(pSrcData, SrcSize, interleaved->TempBuffer, scanline,
597 nSrcWidth, nSrcHeight))
599 WLog_ERR(TAG,
"RleDecompress24to24 failed");
607 if (!RleDecompress16to16(pSrcData, SrcSize, interleaved->TempBuffer, scanline,
608 nSrcWidth, nSrcHeight))
610 WLog_ERR(TAG,
"RleDecompress16to16 failed");
617 if (!RleDecompress8to8(pSrcData, SrcSize, interleaved->TempBuffer, scanline, nSrcWidth,
620 WLog_ERR(TAG,
"RleDecompress8to8 failed");
627 WLog_ERR(TAG,
"Invalid color depth %" PRIu32
"", bpp);
631 if (!freerdp_image_copy_no_overlap(pDstData, DstFormat, nDstStep, nXDst, nYDst, nDstWidth,
632 nDstHeight, interleaved->TempBuffer, SrcFormat, scanline, 0,
633 0, palette, FREERDP_FLIP_VERTICAL | FREERDP_KEEP_DST_ALPHA))
635 WLog_ERR(TAG,
"freerdp_image_copy failed");
641BOOL interleaved_compress(BITMAP_INTERLEAVED_CONTEXT* WINPR_RESTRICT interleaved,
642 BYTE* WINPR_RESTRICT pDstData, UINT32* WINPR_RESTRICT pDstSize,
643 UINT32 nWidth, UINT32 nHeight,
const BYTE* WINPR_RESTRICT pSrcData,
644 UINT32 SrcFormat, UINT32 nSrcStep, UINT32 nXSrc, UINT32 nYSrc,
645 const gdiPalette* WINPR_RESTRICT palette, UINT32 bpp)
649 UINT32 DstFormat = 0;
650 const UINT32 maxSize = 64 * 64 * 4;
652 if (!interleaved || !pDstData || !pSrcData)
655 if ((nWidth == 0) || (nHeight == 0))
660 WLog_ERR(TAG,
"interleaved_compress: width is not a multiple of 4");
664 if ((nWidth > 64) || (nHeight > 64))
667 "interleaved_compress: width (%" PRIu32
") or height (%" PRIu32
668 ") is greater than 64",
676 DstFormat = PIXEL_FORMAT_BGRX32;
680 DstFormat = PIXEL_FORMAT_RGB16;
684 DstFormat = PIXEL_FORMAT_RGB15;
691 if (!freerdp_image_copy_no_overlap(interleaved->TempBuffer, DstFormat, 0, 0, 0, nWidth, nHeight,
692 pSrcData, SrcFormat, nSrcStep, nXSrc, nYSrc, palette,
693 FREERDP_KEEP_DST_ALPHA))
696 s = Stream_New(pDstData, *pDstSize);
701 Stream_SetPosition(interleaved->bts, 0);
703 if (freerdp_bitmap_compress(interleaved->TempBuffer, nWidth, nHeight, s, bpp, maxSize,
704 nHeight - 1, interleaved->bts, 0) < 0)
709 Stream_SealLength(s);
710 *pDstSize = (UINT32)Stream_Length(s);
711 Stream_Free(s, FALSE);
715BOOL bitmap_interleaved_context_reset(BITMAP_INTERLEAVED_CONTEXT* WINPR_RESTRICT interleaved)
723BITMAP_INTERLEAVED_CONTEXT* bitmap_interleaved_context_new(WINPR_ATTR_UNUSED BOOL Compressor)
725 BITMAP_INTERLEAVED_CONTEXT* interleaved = NULL;
726 interleaved = (BITMAP_INTERLEAVED_CONTEXT*)winpr_aligned_recalloc(
727 NULL, 1,
sizeof(BITMAP_INTERLEAVED_CONTEXT), 32);
731 interleaved->TempSize = 64 * 64 * 4;
732 interleaved->TempBuffer = winpr_aligned_calloc(interleaved->TempSize,
sizeof(BYTE), 16);
734 if (!interleaved->TempBuffer)
737 interleaved->bts = Stream_New(NULL, interleaved->TempSize);
739 if (!interleaved->bts)
746 WINPR_PRAGMA_DIAG_PUSH
747 WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
748 bitmap_interleaved_context_free(interleaved);
749 WINPR_PRAGMA_DIAG_POP
753void bitmap_interleaved_context_free(BITMAP_INTERLEAVED_CONTEXT* WINPR_RESTRICT interleaved)
758 winpr_aligned_free(interleaved->TempBuffer);
759 Stream_Free(interleaved->bts, TRUE);
760 winpr_aligned_free(interleaved);