Coverage Report

Created: 2024-05-20 06:11

/src/FreeRDP/libfreerdp/codec/interleaved.c
Line
Count
Source (jump to first uncovered line)
1
/**
2
 * FreeRDP: A Remote Desktop Protocol Implementation
3
 * Interleaved RLE Bitmap Codec
4
 *
5
 * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
6
 * Copyright 2015 Thincast Technologies GmbH
7
 * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
8
 * Copyright 2016 Armin Novak <armin.novak@thincast.com>
9
 * Copyright 2016 Thincast Technologies GmbH
10
 *
11
 * Licensed under the Apache License, Version 2.0 (the "License");
12
 * you may not use this file except in compliance with the License.
13
 * You may obtain a copy of the License at
14
 *
15
 *     http://www.apache.org/licenses/LICENSE-2.0
16
 *
17
 * Unless required by applicable law or agreed to in writing, software
18
 * distributed under the License is distributed on an "AS IS" BASIS,
19
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
20
 * See the License for the specific language governing permissions and
21
 * limitations under the License.
22
 */
23
24
#include <winpr/assert.h>
25
#include <freerdp/config.h>
26
27
#include <freerdp/codec/interleaved.h>
28
#include <freerdp/log.h>
29
30
#define TAG FREERDP_TAG("codec")
31
32
#define UNROLL_BODY(_exp, _count)             \
33
2.07M
  do                                        \
34
2.07M
  {                                         \
35
14.5M
    for (size_t x = 0; x < (_count); x++) \
36
12.5M
    {                                     \
37
12.5M
      do                                \
38
12.5M
      {                                 \
39
28.6M
        _exp                          \
40
12.5M
      } while (FALSE);                  \
41
12.5M
    }                                     \
42
2.07M
  } while (FALSE)
43
44
#define UNROLL_MULTIPLE(_condition, _exp, _count) \
45
1.76M
  do                                            \
46
1.76M
  {                                             \
47
3.83M
    while ((_condition) >= _count)            \
48
2.07M
    {                                         \
49
2.07M
      UNROLL_BODY(_exp, _count);            \
50
2.07M
      (_condition) -= _count;               \
51
2.07M
    }                                         \
52
1.76M
  } while (FALSE)
53
54
#define UNROLL(_condition, _exp)               \
55
587k
  do                                         \
56
587k
  {                                          \
57
587k
    UNROLL_MULTIPLE(_condition, _exp, 16); \
58
587k
    UNROLL_MULTIPLE(_condition, _exp, 4);  \
59
587k
    UNROLL_MULTIPLE(_condition, _exp, 1);  \
60
587k
  } while (FALSE)
61
62
/*
63
   RLE Compressed Bitmap Stream (RLE_BITMAP_STREAM)
64
   http://msdn.microsoft.com/en-us/library/cc240895%28v=prot.10%29.aspx
65
   pseudo-code
66
   http://msdn.microsoft.com/en-us/library/dd240593%28v=prot.10%29.aspx
67
*/
68
69
773k
#define REGULAR_BG_RUN 0x00
70
230k
#define MEGA_MEGA_BG_RUN 0xF0
71
366k
#define REGULAR_FG_RUN 0x01
72
52.7k
#define MEGA_MEGA_FG_RUN 0xF1
73
196k
#define LITE_SET_FG_FG_RUN 0x0C
74
117k
#define MEGA_MEGA_SET_FG_RUN 0xF6
75
37.9k
#define LITE_DITHERED_RUN 0x0E
76
20.3k
#define MEGA_MEGA_DITHERED_RUN 0xF8
77
512k
#define REGULAR_COLOR_RUN 0x03
78
104k
#define MEGA_MEGA_COLOR_RUN 0xF3
79
28.1k
#define REGULAR_FGBG_IMAGE 0x02
80
25.1k
#define MEGA_MEGA_FGBG_IMAGE 0xF2
81
68.0k
#define LITE_SET_FG_FGBG_IMAGE 0x0D
82
48.7k
#define MEGA_MEGA_SET_FGBG_IMAGE 0xF7
83
440k
#define REGULAR_COLOR_IMAGE 0x04
84
26.7k
#define MEGA_MEGA_COLOR_IMAGE 0xF4
85
2.31k
#define SPECIAL_FGBG_1 0xF9
86
3.75k
#define SPECIAL_FGBG_2 0xFA
87
2.14k
#define SPECIAL_WHITE 0xFD
88
2.88k
#define SPECIAL_BLACK 0xFE
89
90
#define BLACK_PIXEL 0x000000
91
92
typedef UINT32 PIXEL;
93
94
static const BYTE g_MaskSpecialFgBg1 = 0x03;
95
static const BYTE g_MaskSpecialFgBg2 = 0x05;
96
97
static const BYTE g_MaskRegularRunLength = 0x1F;
98
static const BYTE g_MaskLiteRunLength = 0x0F;
99
100
static const char* rle_code_str(UINT32 code)
101
0
{
102
0
  switch (code)
103
0
  {
104
0
    case REGULAR_BG_RUN:
105
0
      return "REGULAR_BG_RUN";
106
0
    case MEGA_MEGA_BG_RUN:
107
0
      return "MEGA_MEGA_BG_RUN";
108
0
    case REGULAR_FG_RUN:
109
0
      return "REGULAR_FG_RUN";
110
0
    case MEGA_MEGA_FG_RUN:
111
0
      return "MEGA_MEGA_FG_RUN";
112
0
    case LITE_SET_FG_FG_RUN:
113
0
      return "LITE_SET_FG_FG_RUN";
114
0
    case MEGA_MEGA_SET_FG_RUN:
115
0
      return "MEGA_MEGA_SET_FG_RUN";
116
0
    case LITE_DITHERED_RUN:
117
0
      return "LITE_DITHERED_RUN";
118
0
    case MEGA_MEGA_DITHERED_RUN:
119
0
      return "MEGA_MEGA_DITHERED_RUN";
120
0
    case REGULAR_COLOR_RUN:
121
0
      return "REGULAR_COLOR_RUN";
122
0
    case MEGA_MEGA_COLOR_RUN:
123
0
      return "MEGA_MEGA_COLOR_RUN";
124
0
    case REGULAR_FGBG_IMAGE:
125
0
      return "REGULAR_FGBG_IMAGE";
126
0
    case MEGA_MEGA_FGBG_IMAGE:
127
0
      return "MEGA_MEGA_FGBG_IMAGE";
128
0
    case LITE_SET_FG_FGBG_IMAGE:
129
0
      return "LITE_SET_FG_FGBG_IMAGE";
130
0
    case MEGA_MEGA_SET_FGBG_IMAGE:
131
0
      return "MEGA_MEGA_SET_FGBG_IMAGE";
132
0
    case REGULAR_COLOR_IMAGE:
133
0
      return "REGULAR_COLOR_IMAGE";
134
0
    case MEGA_MEGA_COLOR_IMAGE:
135
0
      return "MEGA_MEGA_COLOR_IMAGE";
136
0
    case SPECIAL_FGBG_1:
137
0
      return "SPECIAL_FGBG_1";
138
0
    case SPECIAL_FGBG_2:
139
0
      return "SPECIAL_FGBG_2";
140
0
    case SPECIAL_WHITE:
141
0
      return "SPECIAL_WHITE";
142
0
    case SPECIAL_BLACK:
143
0
      return "SPECIAL_BLACK";
144
0
    default:
145
0
      return "UNKNOWN";
146
0
  }
147
0
}
148
149
static const char* rle_code_str_buffer(UINT32 code, char* buffer, size_t size)
150
0
{
151
0
  const char* str = rle_code_str(code);
152
0
  _snprintf(buffer, size, "%s [0x%08" PRIx32 "]", str, code);
153
0
  return buffer;
154
0
}
155
156
#define buffer_within_range(pbSrc, size, pbEnd) \
157
845k
  buffer_within_range_((pbSrc), (size), (pbEnd), __func__, __FILE__, __LINE__)
158
static INLINE BOOL buffer_within_range_(const void* pbSrc, size_t size, const void* pbEnd,
159
                                        const char* fkt, const char* file, size_t line)
160
845k
{
161
845k
  WINPR_UNUSED(file);
162
845k
  WINPR_ASSERT(pbSrc);
163
845k
  WINPR_ASSERT(pbEnd);
164
165
845k
  if ((const char*)pbSrc + size > (const char*)pbEnd)
166
3.76k
  {
167
3.76k
    WLog_ERR(TAG, "[%s:%" PRIuz "] pbSrc=%p + %" PRIuz " > pbEnd=%p", fkt, line, pbSrc, size,
168
3.76k
             pbEnd);
169
3.76k
    return FALSE;
170
3.76k
  }
171
841k
  return TRUE;
172
845k
}
173
174
/**
175
 * Reads the supplied order header and extracts the compression
176
 * order code ID.
177
 */
178
static INLINE UINT32 ExtractCodeId(BYTE bOrderHdr)
179
501k
{
180
501k
  if ((bOrderHdr & 0xC0U) != 0xC0U)
181
443k
  {
182
    /* REGULAR orders
183
     * (000x xxxx, 001x xxxx, 010x xxxx, 011x xxxx, 100x xxxx)
184
     */
185
443k
    return bOrderHdr >> 5;
186
443k
  }
187
57.9k
  else if ((bOrderHdr & 0xF0U) == 0xF0U)
188
28.5k
  {
189
    /* MEGA and SPECIAL orders (0xF*) */
190
28.5k
    return bOrderHdr;
191
28.5k
  }
192
29.4k
  else
193
29.4k
  {
194
    /* LITE orders
195
     * 1100 xxxx, 1101 xxxx, 1110 xxxx)
196
     */
197
29.4k
    return bOrderHdr >> 4;
198
29.4k
  }
199
501k
}
200
201
/**
202
 * Extract the run length of a compression order.
203
 */
204
static UINT ExtractRunLengthRegularFgBg(const BYTE* pbOrderHdr, const BYTE* pbEnd, UINT32* advance)
205
14.0k
{
206
14.0k
  UINT runLength = 0;
207
208
14.0k
  WINPR_ASSERT(pbOrderHdr);
209
14.0k
  WINPR_ASSERT(pbEnd);
210
14.0k
  WINPR_ASSERT(advance);
211
212
14.0k
  runLength = (*pbOrderHdr) & g_MaskRegularRunLength;
213
14.0k
  if (runLength == 0)
214
5.67k
  {
215
5.67k
    if (!buffer_within_range(pbOrderHdr, 2, pbEnd))
216
36
    {
217
36
      *advance = 0;
218
36
      return 0;
219
36
    }
220
5.63k
    runLength = *(pbOrderHdr + 1) + 1;
221
5.63k
    (*advance)++;
222
5.63k
  }
223
8.38k
  else
224
8.38k
    runLength = runLength * 8;
225
226
14.0k
  return runLength;
227
14.0k
}
228
229
static UINT ExtractRunLengthLiteFgBg(const BYTE* pbOrderHdr, const BYTE* pbEnd, UINT32* advance)
230
4.58k
{
231
4.58k
  UINT runLength = 0;
232
233
4.58k
  WINPR_ASSERT(pbOrderHdr);
234
4.58k
  WINPR_ASSERT(pbEnd);
235
4.58k
  WINPR_ASSERT(advance);
236
237
4.58k
  runLength = *pbOrderHdr & g_MaskLiteRunLength;
238
4.58k
  if (runLength == 0)
239
614
  {
240
614
    if (!buffer_within_range(pbOrderHdr, 2, pbEnd))
241
12
    {
242
12
      *advance = 0;
243
12
      return 0;
244
12
    }
245
602
    runLength = *(pbOrderHdr + 1) + 1;
246
602
    (*advance)++;
247
602
  }
248
3.97k
  else
249
3.97k
    runLength = runLength * 8;
250
251
4.57k
  return runLength;
252
4.58k
}
253
254
static UINT ExtractRunLengthRegular(const BYTE* pbOrderHdr, const BYTE* pbEnd, UINT32* advance)
255
427k
{
256
427k
  UINT runLength = 0;
257
258
427k
  WINPR_ASSERT(pbOrderHdr);
259
427k
  WINPR_ASSERT(pbEnd);
260
427k
  WINPR_ASSERT(advance);
261
262
427k
  runLength = *pbOrderHdr & g_MaskRegularRunLength;
263
427k
  if (runLength == 0)
264
144k
  {
265
144k
    if (!buffer_within_range(pbOrderHdr, 2, pbEnd))
266
646
    {
267
646
      *advance = 0;
268
646
      return 0;
269
646
    }
270
143k
    runLength = *(pbOrderHdr + 1) + 32;
271
143k
    (*advance)++;
272
143k
  }
273
274
427k
  return runLength;
275
427k
}
276
277
static UINT ExtractRunLengthMegaMega(const BYTE* pbOrderHdr, const BYTE* pbEnd, UINT32* advance)
278
12.4k
{
279
12.4k
  UINT runLength = 0;
280
281
12.4k
  WINPR_ASSERT(pbOrderHdr);
282
12.4k
  WINPR_ASSERT(pbEnd);
283
12.4k
  WINPR_ASSERT(advance);
284
285
12.4k
  if (!buffer_within_range(pbOrderHdr, 3, pbEnd))
286
372
  {
287
372
    *advance = 0;
288
372
    return 0;
289
372
  }
290
291
12.0k
  runLength = ((UINT16)pbOrderHdr[1]) | (((UINT16)pbOrderHdr[2]) << 8);
292
12.0k
  (*advance) += 2;
293
294
12.0k
  return runLength;
295
12.4k
}
296
297
static UINT ExtractRunLengthLite(const BYTE* pbOrderHdr, const BYTE* pbEnd, UINT32* advance)
298
24.8k
{
299
24.8k
  UINT runLength = 0;
300
301
24.8k
  WINPR_ASSERT(pbOrderHdr);
302
24.8k
  WINPR_ASSERT(pbEnd);
303
24.8k
  WINPR_ASSERT(advance);
304
305
24.8k
  runLength = *pbOrderHdr & g_MaskLiteRunLength;
306
24.8k
  if (runLength == 0)
307
3.86k
  {
308
3.86k
    if (!buffer_within_range(pbOrderHdr, 2, pbEnd))
309
33
    {
310
33
      *advance = 0;
311
33
      return 0;
312
33
    }
313
3.83k
    runLength = *(pbOrderHdr + 1) + 16;
314
3.83k
    (*advance)++;
315
3.83k
  }
316
24.8k
  return runLength;
317
24.8k
}
318
319
static INLINE UINT32 ExtractRunLength(UINT32 code, const BYTE* pbOrderHdr, const BYTE* pbEnd,
320
                                      UINT32* advance)
321
483k
{
322
483k
  UINT32 runLength = 0;
323
483k
  UINT32 ladvance = 1;
324
325
483k
  WINPR_ASSERT(pbOrderHdr);
326
483k
  WINPR_ASSERT(pbEnd);
327
483k
  WINPR_ASSERT(advance);
328
329
483k
  *advance = 0;
330
483k
  if (!buffer_within_range(pbOrderHdr, 0, pbEnd))
331
0
    return 0;
332
333
483k
  switch (code)
334
483k
  {
335
14.0k
    case REGULAR_FGBG_IMAGE:
336
14.0k
      runLength = ExtractRunLengthRegularFgBg(pbOrderHdr, pbEnd, &ladvance);
337
14.0k
      break;
338
339
4.58k
    case LITE_SET_FG_FGBG_IMAGE:
340
4.58k
      runLength = ExtractRunLengthLiteFgBg(pbOrderHdr, pbEnd, &ladvance);
341
4.58k
      break;
342
343
272k
    case REGULAR_BG_RUN:
344
319k
    case REGULAR_FG_RUN:
345
415k
    case REGULAR_COLOR_RUN:
346
427k
    case REGULAR_COLOR_IMAGE:
347
427k
      runLength = ExtractRunLengthRegular(pbOrderHdr, pbEnd, &ladvance);
348
427k
      break;
349
350
11.8k
    case LITE_SET_FG_FG_RUN:
351
24.8k
    case LITE_DITHERED_RUN:
352
24.8k
      runLength = ExtractRunLengthLite(pbOrderHdr, pbEnd, &ladvance);
353
24.8k
      break;
354
355
1.40k
    case MEGA_MEGA_BG_RUN:
356
3.64k
    case MEGA_MEGA_FG_RUN:
357
4.97k
    case MEGA_MEGA_SET_FG_RUN:
358
6.14k
    case MEGA_MEGA_DITHERED_RUN:
359
7.31k
    case MEGA_MEGA_COLOR_RUN:
360
9.21k
    case MEGA_MEGA_FGBG_IMAGE:
361
10.2k
    case MEGA_MEGA_SET_FGBG_IMAGE:
362
12.4k
    case MEGA_MEGA_COLOR_IMAGE:
363
12.4k
      runLength = ExtractRunLengthMegaMega(pbOrderHdr, pbEnd, &ladvance);
364
12.4k
      break;
365
366
0
    default:
367
0
      runLength = 0;
368
0
      ladvance = 0;
369
0
      break;
370
483k
  }
371
372
483k
  *advance = ladvance;
373
483k
  return runLength;
374
483k
}
375
376
#define ensure_capacity(start, end, size, base) \
377
846k
  ensure_capacity_((start), (end), (size), (base), __func__, __FILE__, __LINE__)
378
static INLINE BOOL ensure_capacity_(const BYTE* start, const BYTE* end, size_t size, size_t base,
379
                                    const char* fkt, const char* file, size_t line)
380
846k
{
381
846k
  const size_t available = (uintptr_t)end - (uintptr_t)start;
382
846k
  const BOOL rc = available >= size * base;
383
846k
  const BOOL res = rc && (start <= end);
384
385
846k
  if (!res)
386
846k
    WLog_ERR(TAG,
387
846k
             "[%s:%" PRIuz "] failed: start=%p <= end=%p, available=%" PRIuz " >= size=%" PRIuz
388
846k
             " * base=%" PRIuz,
389
846k
             fkt, line, start, end, available, size, base);
390
846k
  return res;
391
846k
}
392
393
static INLINE void write_pixel_8(BYTE* _buf, BYTE _pix)
394
0
{
395
0
  WINPR_ASSERT(_buf);
396
0
  *_buf = _pix;
397
0
}
398
399
static INLINE void write_pixel_24(BYTE* _buf, UINT32 _pix)
400
4.10M
{
401
4.10M
  WINPR_ASSERT(_buf);
402
4.10M
  (_buf)[0] = (BYTE)(_pix);
403
4.10M
  (_buf)[1] = (BYTE)((_pix) >> 8);
404
4.10M
  (_buf)[2] = (BYTE)((_pix) >> 16);
405
4.10M
}
406
407
static INLINE void write_pixel_16(BYTE* _buf, UINT16 _pix)
408
9.25M
{
409
9.25M
  WINPR_ASSERT(_buf);
410
9.25M
  _buf[0] = _pix & 0xFF;
411
9.25M
  _buf[1] = (_pix >> 8) & 0xFF;
412
9.25M
}
413
414
#undef DESTWRITEPIXEL
415
#undef DESTREADPIXEL
416
#undef SRCREADPIXEL
417
#undef WRITEFGBGIMAGE
418
#undef WRITEFIRSTLINEFGBGIMAGE
419
#undef RLEDECOMPRESS
420
#undef RLEEXTRA
421
#undef WHITE_PIXEL
422
#undef PIXEL_SIZE
423
#undef PIXEL
424
#define PIXEL_SIZE 1
425
0
#define PIXEL BYTE
426
0
#define WHITE_PIXEL 0xFF
427
#define DESTWRITEPIXEL(_buf, _pix) \
428
0
  do                             \
429
0
  {                              \
430
0
    write_pixel_8(_buf, _pix); \
431
0
    _buf += 1;                 \
432
0
  } while (0)
433
0
#define DESTREADPIXEL(_pix, _buf) _pix = (_buf)[0]
434
#define SRCREADPIXEL(_pix, _buf) \
435
0
  do                           \
436
0
  {                            \
437
0
    _pix = (_buf)[0];        \
438
0
    _buf += 1;               \
439
0
  } while (0)
440
441
0
#define WRITEFGBGIMAGE WriteFgBgImage8to8
442
0
#define WRITEFIRSTLINEFGBGIMAGE WriteFirstLineFgBgImage8to8
443
#define RLEDECOMPRESS RleDecompress8to8
444
#define RLEEXTRA
445
#undef ENSURE_CAPACITY
446
0
#define ENSURE_CAPACITY(_start, _end, _size) ensure_capacity(_start, _end, _size, 1)
447
#include "include/bitmap.c"
448
449
#undef DESTWRITEPIXEL
450
#undef DESTREADPIXEL
451
#undef SRCREADPIXEL
452
#undef WRITEFGBGIMAGE
453
#undef WRITEFIRSTLINEFGBGIMAGE
454
#undef RLEDECOMPRESS
455
#undef RLEEXTRA
456
#undef WHITE_PIXEL
457
#undef PIXEL_SIZE
458
#undef PIXEL
459
#define PIXEL_SIZE 2
460
105k
#define PIXEL UINT16
461
11.4k
#define WHITE_PIXEL 0xFFFF
462
#define DESTWRITEPIXEL(_buf, _pix)  \
463
162k
  do                              \
464
162k
  {                               \
465
162k
    write_pixel_16(_buf, _pix); \
466
162k
    _buf += 2;                  \
467
162k
  } while (0)
468
147k
#define DESTREADPIXEL(_pix, _buf) _pix = ((UINT16*)(_buf))[0]
469
#define SRCREADPIXEL(_pix, _buf)             \
470
98.8k
  do                                       \
471
98.8k
  {                                        \
472
98.8k
    _pix = (_buf)[0] | ((_buf)[1] << 8); \
473
98.8k
    _buf += 2;                           \
474
98.8k
  } while (0)
475
71.6k
#define WRITEFGBGIMAGE WriteFgBgImage16to16
476
18.1k
#define WRITEFIRSTLINEFGBGIMAGE WriteFirstLineFgBgImage16to16
477
#define RLEDECOMPRESS RleDecompress16to16
478
#define RLEEXTRA
479
#undef ENSURE_CAPACITY
480
575k
#define ENSURE_CAPACITY(_start, _end, _size) ensure_capacity(_start, _end, _size, 2)
481
#include "include/bitmap.c"
482
483
#undef DESTWRITEPIXEL
484
#undef DESTREADPIXEL
485
#undef SRCREADPIXEL
486
#undef WRITEFGBGIMAGE
487
#undef WRITEFIRSTLINEFGBGIMAGE
488
#undef RLEDECOMPRESS
489
#undef RLEEXTRA
490
#undef WHITE_PIXEL
491
#undef PIXEL_SIZE
492
#undef PIXEL
493
#define PIXEL_SIZE 3
494
46.3k
#define PIXEL UINT32
495
5.71k
#define WHITE_PIXEL 0xffffff
496
#define DESTWRITEPIXEL(_buf, _pix)  \
497
79.3k
  do                              \
498
79.3k
  {                               \
499
79.3k
    write_pixel_24(_buf, _pix); \
500
79.3k
    _buf += 3;                  \
501
79.3k
  } while (0)
502
71.4k
#define DESTREADPIXEL(_pix, _buf) _pix = (_buf)[0] | ((_buf)[1] << 8) | ((_buf)[2] << 16)
503
#define SRCREADPIXEL(_pix, _buf)                                 \
504
43.6k
  do                                                           \
505
43.6k
  {                                                            \
506
43.6k
    _pix = (_buf)[0] | ((_buf)[1] << 8) | ((_buf)[2] << 16); \
507
43.6k
    _buf += 3;                                               \
508
43.6k
  } while (0)
509
510
29.1k
#define WRITEFGBGIMAGE WriteFgBgImage24to24
511
10.9k
#define WRITEFIRSTLINEFGBGIMAGE WriteFirstLineFgBgImage24to24
512
#define RLEDECOMPRESS RleDecompress24to24
513
#define RLEEXTRA
514
#undef ENSURE_CAPACITY
515
270k
#define ENSURE_CAPACITY(_start, _end, _size) ensure_capacity(_start, _end, _size, 3)
516
#include "include/bitmap.c"
517
518
struct S_BITMAP_INTERLEAVED_CONTEXT
519
{
520
  BOOL Compressor;
521
522
  UINT32 TempSize;
523
  BYTE* TempBuffer;
524
525
  wStream* bts;
526
};
527
528
BOOL interleaved_decompress(BITMAP_INTERLEAVED_CONTEXT* interleaved, const BYTE* pSrcData,
529
                            UINT32 SrcSize, UINT32 nSrcWidth, UINT32 nSrcHeight, UINT32 bpp,
530
                            BYTE* pDstData, UINT32 DstFormat, UINT32 nDstStep, UINT32 nXDst,
531
                            UINT32 nYDst, UINT32 nDstWidth, UINT32 nDstHeight,
532
                            const gdiPalette* palette)
533
17.1k
{
534
17.1k
  UINT32 scanline = 0;
535
17.1k
  UINT32 SrcFormat = 0;
536
17.1k
  UINT32 BufferSize = 0;
537
538
17.1k
  if (!interleaved || !pSrcData || !pDstData)
539
0
  {
540
0
    WLog_ERR(TAG, "invalid arguments: interleaved=%p, pSrcData=%p, pDstData=%p", interleaved,
541
0
             pSrcData, pDstData);
542
0
    return FALSE;
543
0
  }
544
545
17.1k
  switch (bpp)
546
17.1k
  {
547
5.71k
    case 24:
548
5.71k
      scanline = nSrcWidth * 3;
549
5.71k
      SrcFormat = PIXEL_FORMAT_BGR24;
550
5.71k
      break;
551
552
5.71k
    case 16:
553
5.71k
      scanline = nSrcWidth * 2;
554
5.71k
      SrcFormat = PIXEL_FORMAT_RGB16;
555
5.71k
      break;
556
557
5.71k
    case 15:
558
5.71k
      scanline = nSrcWidth * 2;
559
5.71k
      SrcFormat = PIXEL_FORMAT_RGB15;
560
5.71k
      break;
561
562
0
    case 8:
563
0
      scanline = nSrcWidth;
564
0
      SrcFormat = PIXEL_FORMAT_RGB8;
565
0
      break;
566
567
0
    default:
568
0
      WLog_ERR(TAG, "Invalid color depth %" PRIu32 "", bpp);
569
0
      return FALSE;
570
17.1k
  }
571
572
17.1k
  BufferSize = scanline * nSrcHeight;
573
574
17.1k
  if (BufferSize > interleaved->TempSize)
575
0
  {
576
0
    interleaved->TempBuffer =
577
0
        winpr_aligned_recalloc(interleaved->TempBuffer, BufferSize, sizeof(BYTE), 16);
578
0
    interleaved->TempSize = BufferSize;
579
0
  }
580
581
17.1k
  if (!interleaved->TempBuffer)
582
0
  {
583
0
    WLog_ERR(TAG, "interleaved->TempBuffer=%p", interleaved->TempBuffer);
584
0
    return FALSE;
585
0
  }
586
587
17.1k
  switch (bpp)
588
17.1k
  {
589
5.71k
    case 24:
590
5.71k
      if (!RleDecompress24to24(pSrcData, SrcSize, interleaved->TempBuffer, scanline,
591
5.71k
                               nSrcWidth, nSrcHeight))
592
4.29k
      {
593
4.29k
        WLog_ERR(TAG, "RleDecompress24to24 failed");
594
4.29k
        return FALSE;
595
4.29k
      }
596
597
1.41k
      break;
598
599
5.71k
    case 16:
600
11.4k
    case 15:
601
11.4k
      if (!RleDecompress16to16(pSrcData, SrcSize, interleaved->TempBuffer, scanline,
602
11.4k
                               nSrcWidth, nSrcHeight))
603
8.65k
      {
604
8.65k
        WLog_ERR(TAG, "RleDecompress16to16 failed");
605
8.65k
        return FALSE;
606
8.65k
      }
607
608
2.77k
      break;
609
610
2.77k
    case 8:
611
0
      if (!RleDecompress8to8(pSrcData, SrcSize, interleaved->TempBuffer, scanline, nSrcWidth,
612
0
                             nSrcHeight))
613
0
      {
614
0
        WLog_ERR(TAG, "RleDecompress8to8 failed");
615
0
        return FALSE;
616
0
      }
617
618
0
      break;
619
620
0
    default:
621
0
      WLog_ERR(TAG, "Invalid color depth %" PRIu32 "", bpp);
622
0
      return FALSE;
623
17.1k
  }
624
625
4.18k
  if (!freerdp_image_copy(pDstData, DstFormat, nDstStep, nXDst, nYDst, nDstWidth, nDstHeight,
626
4.18k
                          interleaved->TempBuffer, SrcFormat, scanline, 0, 0, palette,
627
4.18k
                          FREERDP_FLIP_VERTICAL | FREERDP_KEEP_DST_ALPHA))
628
0
  {
629
0
    WLog_ERR(TAG, "freerdp_image_copy failed");
630
0
    return FALSE;
631
0
  }
632
4.18k
  return TRUE;
633
4.18k
}
634
635
BOOL interleaved_compress(BITMAP_INTERLEAVED_CONTEXT* interleaved, BYTE* pDstData, UINT32* pDstSize,
636
                          UINT32 nWidth, UINT32 nHeight, const BYTE* pSrcData, UINT32 SrcFormat,
637
                          UINT32 nSrcStep, UINT32 nXSrc, UINT32 nYSrc, const gdiPalette* palette,
638
                          UINT32 bpp)
639
0
{
640
0
  BOOL status = 0;
641
0
  wStream* s = NULL;
642
0
  UINT32 DstFormat = 0;
643
0
  const UINT32 maxSize = 64 * 64 * 4;
644
645
0
  if (!interleaved || !pDstData || !pSrcData)
646
0
    return FALSE;
647
648
0
  if ((nWidth == 0) || (nHeight == 0))
649
0
    return FALSE;
650
651
0
  if (nWidth % 4)
652
0
  {
653
0
    WLog_ERR(TAG, "interleaved_compress: width is not a multiple of 4");
654
0
    return FALSE;
655
0
  }
656
657
0
  if ((nWidth > 64) || (nHeight > 64))
658
0
  {
659
0
    WLog_ERR(TAG,
660
0
             "interleaved_compress: width (%" PRIu32 ") or height (%" PRIu32
661
0
             ") is greater than 64",
662
0
             nWidth, nHeight);
663
0
    return FALSE;
664
0
  }
665
666
0
  switch (bpp)
667
0
  {
668
0
    case 24:
669
0
      DstFormat = PIXEL_FORMAT_BGRX32;
670
0
      break;
671
672
0
    case 16:
673
0
      DstFormat = PIXEL_FORMAT_RGB16;
674
0
      break;
675
676
0
    case 15:
677
0
      DstFormat = PIXEL_FORMAT_RGB15;
678
0
      break;
679
680
0
    default:
681
0
      return FALSE;
682
0
  }
683
684
0
  if (!freerdp_image_copy(interleaved->TempBuffer, DstFormat, 0, 0, 0, nWidth, nHeight, pSrcData,
685
0
                          SrcFormat, nSrcStep, nXSrc, nYSrc, palette, FREERDP_KEEP_DST_ALPHA))
686
0
    return FALSE;
687
688
0
  s = Stream_New(pDstData, *pDstSize);
689
690
0
  if (!s)
691
0
    return FALSE;
692
693
0
  Stream_SetPosition(interleaved->bts, 0);
694
695
0
  if (freerdp_bitmap_compress(interleaved->TempBuffer, nWidth, nHeight, s, bpp, maxSize,
696
0
                              nHeight - 1, interleaved->bts, 0) < 0)
697
0
    status = FALSE;
698
0
  else
699
0
    status = TRUE;
700
701
0
  Stream_SealLength(s);
702
0
  *pDstSize = (UINT32)Stream_Length(s);
703
0
  Stream_Free(s, FALSE);
704
0
  return status;
705
0
}
706
707
BOOL bitmap_interleaved_context_reset(BITMAP_INTERLEAVED_CONTEXT* interleaved)
708
17.1k
{
709
17.1k
  if (!interleaved)
710
0
    return FALSE;
711
712
17.1k
  return TRUE;
713
17.1k
}
714
715
BITMAP_INTERLEAVED_CONTEXT* bitmap_interleaved_context_new(BOOL Compressor)
716
5.71k
{
717
5.71k
  BITMAP_INTERLEAVED_CONTEXT* interleaved = NULL;
718
5.71k
  interleaved = (BITMAP_INTERLEAVED_CONTEXT*)winpr_aligned_recalloc(
719
5.71k
      NULL, 1, sizeof(BITMAP_INTERLEAVED_CONTEXT), 32);
720
721
5.71k
  if (interleaved)
722
5.71k
  {
723
5.71k
    interleaved->TempSize = 64 * 64 * 4;
724
5.71k
    interleaved->TempBuffer = winpr_aligned_calloc(interleaved->TempSize, sizeof(BYTE), 16);
725
726
5.71k
    if (!interleaved->TempBuffer)
727
0
      goto fail;
728
729
5.71k
    interleaved->bts = Stream_New(NULL, interleaved->TempSize);
730
731
5.71k
    if (!interleaved->bts)
732
0
      goto fail;
733
5.71k
  }
734
735
5.71k
  return interleaved;
736
737
0
fail:
738
0
  WINPR_PRAGMA_DIAG_PUSH
739
0
  WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
740
0
  bitmap_interleaved_context_free(interleaved);
741
0
  WINPR_PRAGMA_DIAG_POP
742
0
  return NULL;
743
5.71k
}
744
745
void bitmap_interleaved_context_free(BITMAP_INTERLEAVED_CONTEXT* interleaved)
746
5.71k
{
747
5.71k
  if (!interleaved)
748
0
    return;
749
750
5.71k
  winpr_aligned_free(interleaved->TempBuffer);
751
5.71k
  Stream_Free(interleaved->bts, TRUE);
752
5.71k
  winpr_aligned_free(interleaved);
753
5.71k
}