/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 | } |