/src/FreeRDP/libfreerdp/gdi/graphics.c
Line | Count | Source (jump to first uncovered line) |
1 | | /** |
2 | | * FreeRDP: A Remote Desktop Protocol Implementation |
3 | | * Graphical Objects |
4 | | * |
5 | | * Copyright 2011 Marc-Andre Moreau <marcandre.moreau@gmail.com> |
6 | | * Copyright 2016 Armin Novak <armin.novak@thincast.com> |
7 | | * Copyright 2016 Thincast Technologies GmbH |
8 | | * |
9 | | * Licensed under the Apache License, Version 2.0 (the "License"); |
10 | | * you may not use this file except in compliance with the License. |
11 | | * You may obtain a copy of the License at |
12 | | * |
13 | | * http://www.apache.org/licenses/LICENSE-2.0 |
14 | | * |
15 | | * Unless required by applicable law or agreed to in writing, software |
16 | | * distributed under the License is distributed on an "AS IS" BASIS, |
17 | | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
18 | | * See the License for the specific language governing permissions and |
19 | | * limitations under the License. |
20 | | */ |
21 | | |
22 | | #include <freerdp/config.h> |
23 | | |
24 | | #include <winpr/crt.h> |
25 | | |
26 | | #include <freerdp/log.h> |
27 | | #include <freerdp/freerdp.h> |
28 | | #include <freerdp/gdi/dc.h> |
29 | | #include <freerdp/gdi/shape.h> |
30 | | #include <freerdp/gdi/region.h> |
31 | | #include <freerdp/gdi/bitmap.h> |
32 | | |
33 | | #include "clipping.h" |
34 | | #include "drawing.h" |
35 | | #include "brush.h" |
36 | | #include "graphics.h" |
37 | | |
38 | | #define TAG FREERDP_TAG("gdi") |
39 | | /* Bitmap Class */ |
40 | | |
41 | | HGDI_BITMAP gdi_create_bitmap(rdpGdi* gdi, UINT32 nWidth, UINT32 nHeight, UINT32 SrcFormat, |
42 | | BYTE* data) |
43 | 0 | { |
44 | 0 | UINT32 nSrcStep = 0; |
45 | 0 | UINT32 nDstStep = 0; |
46 | 0 | BYTE* pSrcData = NULL; |
47 | 0 | BYTE* pDstData = NULL; |
48 | 0 | HGDI_BITMAP bitmap = NULL; |
49 | |
|
50 | 0 | if (!gdi) |
51 | 0 | return NULL; |
52 | | |
53 | 0 | nDstStep = nWidth * FreeRDPGetBytesPerPixel(gdi->dstFormat); |
54 | 0 | pDstData = winpr_aligned_malloc(1ull * nHeight * nDstStep, 16); |
55 | |
|
56 | 0 | if (!pDstData) |
57 | 0 | return NULL; |
58 | | |
59 | 0 | pSrcData = data; |
60 | 0 | nSrcStep = nWidth * FreeRDPGetBytesPerPixel(SrcFormat); |
61 | |
|
62 | 0 | if (!freerdp_image_copy(pDstData, gdi->dstFormat, nDstStep, 0, 0, nWidth, nHeight, pSrcData, |
63 | 0 | SrcFormat, nSrcStep, 0, 0, &gdi->palette, FREERDP_FLIP_NONE)) |
64 | 0 | { |
65 | 0 | winpr_aligned_free(pDstData); |
66 | 0 | return NULL; |
67 | 0 | } |
68 | | |
69 | 0 | bitmap = gdi_CreateBitmap(nWidth, nHeight, gdi->dstFormat, pDstData); |
70 | 0 | return bitmap; |
71 | 0 | } |
72 | | |
73 | | static BOOL gdi_Bitmap_New(rdpContext* context, rdpBitmap* bitmap) |
74 | 0 | { |
75 | 0 | gdiBitmap* gdi_bitmap = NULL; |
76 | 0 | rdpGdi* gdi = context->gdi; |
77 | 0 | gdi_bitmap = (gdiBitmap*)bitmap; |
78 | 0 | gdi_bitmap->hdc = gdi_CreateCompatibleDC(gdi->hdc); |
79 | |
|
80 | 0 | if (!gdi_bitmap->hdc) |
81 | 0 | return FALSE; |
82 | | |
83 | 0 | if (!bitmap->data) |
84 | 0 | gdi_bitmap->bitmap = gdi_CreateCompatibleBitmap(gdi->hdc, bitmap->width, bitmap->height); |
85 | 0 | else |
86 | 0 | { |
87 | 0 | UINT32 format = bitmap->format; |
88 | 0 | gdi_bitmap->bitmap = |
89 | 0 | gdi_create_bitmap(gdi, bitmap->width, bitmap->height, format, bitmap->data); |
90 | 0 | } |
91 | |
|
92 | 0 | if (!gdi_bitmap->bitmap) |
93 | 0 | { |
94 | 0 | gdi_DeleteDC(gdi_bitmap->hdc); |
95 | 0 | gdi_bitmap->hdc = NULL; |
96 | 0 | return FALSE; |
97 | 0 | } |
98 | | |
99 | 0 | gdi_bitmap->hdc->format = gdi_bitmap->bitmap->format; |
100 | 0 | gdi_SelectObject(gdi_bitmap->hdc, (HGDIOBJECT)gdi_bitmap->bitmap); |
101 | 0 | gdi_bitmap->org_bitmap = NULL; |
102 | 0 | return TRUE; |
103 | 0 | } |
104 | | |
105 | | static void gdi_Bitmap_Free(rdpContext* context, rdpBitmap* bitmap) |
106 | 0 | { |
107 | 0 | gdiBitmap* gdi_bitmap = (gdiBitmap*)bitmap; |
108 | |
|
109 | 0 | if (gdi_bitmap) |
110 | 0 | { |
111 | 0 | if (gdi_bitmap->hdc) |
112 | 0 | gdi_SelectObject(gdi_bitmap->hdc, (HGDIOBJECT)gdi_bitmap->org_bitmap); |
113 | |
|
114 | 0 | gdi_DeleteObject((HGDIOBJECT)gdi_bitmap->bitmap); |
115 | 0 | gdi_DeleteDC(gdi_bitmap->hdc); |
116 | 0 | winpr_aligned_free(bitmap->data); |
117 | 0 | } |
118 | |
|
119 | 0 | free(bitmap); |
120 | 0 | } |
121 | | |
122 | | static BOOL gdi_Bitmap_Paint(rdpContext* context, rdpBitmap* bitmap) |
123 | 0 | { |
124 | 0 | gdiBitmap* gdi_bitmap = (gdiBitmap*)bitmap; |
125 | 0 | UINT32 width = bitmap->right - bitmap->left + 1; |
126 | 0 | UINT32 height = bitmap->bottom - bitmap->top + 1; |
127 | 0 | return gdi_BitBlt(context->gdi->primary->hdc, bitmap->left, bitmap->top, width, height, |
128 | 0 | gdi_bitmap->hdc, 0, 0, GDI_SRCCOPY, &context->gdi->palette); |
129 | 0 | } |
130 | | |
131 | | static BOOL gdi_Bitmap_Decompress(rdpContext* context, rdpBitmap* bitmap, const BYTE* pSrcData, |
132 | | UINT32 DstWidth, UINT32 DstHeight, UINT32 bpp, UINT32 length, |
133 | | BOOL compressed, UINT32 codecId) |
134 | 0 | { |
135 | 0 | int status = 0; |
136 | 0 | UINT32 SrcSize = length; |
137 | 0 | rdpGdi* gdi = context->gdi; |
138 | 0 | UINT32 size = DstWidth * DstHeight; |
139 | 0 | bitmap->compressed = FALSE; |
140 | 0 | bitmap->format = gdi->dstFormat; |
141 | |
|
142 | 0 | if ((FreeRDPGetBytesPerPixel(bitmap->format) == 0) || (DstWidth == 0) || (DstHeight == 0) || |
143 | 0 | (DstWidth > UINT32_MAX / DstHeight) || |
144 | 0 | (size > (UINT32_MAX / FreeRDPGetBytesPerPixel(bitmap->format)))) |
145 | 0 | { |
146 | 0 | WLog_ERR(TAG, "invalid input data"); |
147 | 0 | return FALSE; |
148 | 0 | } |
149 | | |
150 | 0 | size *= FreeRDPGetBytesPerPixel(bitmap->format); |
151 | 0 | bitmap->length = size; |
152 | 0 | bitmap->data = (BYTE*)winpr_aligned_malloc(bitmap->length, 16); |
153 | |
|
154 | 0 | if (!bitmap->data) |
155 | 0 | return FALSE; |
156 | | |
157 | 0 | if (compressed) |
158 | 0 | { |
159 | 0 | if ((codecId == RDP_CODEC_ID_REMOTEFX) || (codecId == RDP_CODEC_ID_IMAGE_REMOTEFX)) |
160 | 0 | { |
161 | 0 | REGION16 invalidRegion; |
162 | 0 | region16_init(&invalidRegion); |
163 | |
|
164 | 0 | if (!rfx_process_message(context->codecs->rfx, pSrcData, SrcSize, bitmap->left, |
165 | 0 | bitmap->top, bitmap->data, bitmap->format, gdi->stride, |
166 | 0 | gdi->height, &invalidRegion)) |
167 | 0 | { |
168 | 0 | WLog_ERR(TAG, "rfx_process_message failed"); |
169 | 0 | return FALSE; |
170 | 0 | } |
171 | | |
172 | 0 | status = 1; |
173 | 0 | } |
174 | 0 | else if (codecId == RDP_CODEC_ID_NSCODEC) |
175 | 0 | { |
176 | 0 | status = nsc_process_message(context->codecs->nsc, 32, DstWidth, DstHeight, pSrcData, |
177 | 0 | SrcSize, bitmap->data, bitmap->format, 0, 0, 0, DstWidth, |
178 | 0 | DstHeight, FREERDP_FLIP_VERTICAL); |
179 | |
|
180 | 0 | if (status < 1) |
181 | 0 | { |
182 | 0 | WLog_ERR(TAG, "nsc_process_message failed"); |
183 | 0 | return FALSE; |
184 | 0 | } |
185 | | |
186 | 0 | return freerdp_image_copy(bitmap->data, bitmap->format, 0, 0, 0, DstWidth, DstHeight, |
187 | 0 | pSrcData, PIXEL_FORMAT_XRGB32, 0, 0, 0, &gdi->palette, |
188 | 0 | FREERDP_FLIP_VERTICAL); |
189 | 0 | } |
190 | 0 | else if (bpp < 32) |
191 | 0 | { |
192 | 0 | if (!interleaved_decompress(context->codecs->interleaved, pSrcData, SrcSize, DstWidth, |
193 | 0 | DstHeight, bpp, bitmap->data, bitmap->format, 0, 0, 0, |
194 | 0 | DstWidth, DstHeight, &gdi->palette)) |
195 | 0 | { |
196 | 0 | WLog_ERR(TAG, "interleaved_decompress failed"); |
197 | 0 | return FALSE; |
198 | 0 | } |
199 | 0 | } |
200 | 0 | else |
201 | 0 | { |
202 | 0 | const BOOL fidelity = |
203 | 0 | freerdp_settings_get_bool(context->settings, FreeRDP_DrawAllowDynamicColorFidelity); |
204 | 0 | freerdp_planar_switch_bgr(context->codecs->planar, fidelity); |
205 | 0 | if (!planar_decompress(context->codecs->planar, pSrcData, SrcSize, DstWidth, DstHeight, |
206 | 0 | bitmap->data, bitmap->format, 0, 0, 0, DstWidth, DstHeight, |
207 | 0 | TRUE)) |
208 | 0 | { |
209 | 0 | WLog_ERR(TAG, "planar_decompress failed"); |
210 | 0 | return FALSE; |
211 | 0 | } |
212 | 0 | } |
213 | 0 | } |
214 | 0 | else |
215 | 0 | { |
216 | 0 | const UINT32 SrcFormat = gdi_get_pixel_format(bpp); |
217 | 0 | const size_t sbpp = FreeRDPGetBytesPerPixel(SrcFormat); |
218 | 0 | const size_t dbpp = FreeRDPGetBytesPerPixel(bitmap->format); |
219 | |
|
220 | 0 | if ((sbpp == 0) || (dbpp == 0)) |
221 | 0 | return FALSE; |
222 | 0 | else |
223 | 0 | { |
224 | 0 | const size_t dstSize = SrcSize * dbpp / sbpp; |
225 | |
|
226 | 0 | if (dstSize < bitmap->length) |
227 | 0 | { |
228 | 0 | WLog_ERR(TAG, "dstSize %" PRIuz " < bitmap->length %" PRIu32, dstSize, |
229 | 0 | bitmap->length); |
230 | 0 | return FALSE; |
231 | 0 | } |
232 | 0 | } |
233 | | |
234 | 0 | if (!freerdp_image_copy(bitmap->data, bitmap->format, 0, 0, 0, DstWidth, DstHeight, |
235 | 0 | pSrcData, SrcFormat, 0, 0, 0, &gdi->palette, FREERDP_FLIP_VERTICAL)) |
236 | 0 | { |
237 | 0 | WLog_ERR(TAG, "freerdp_image_copy failed"); |
238 | 0 | return FALSE; |
239 | 0 | } |
240 | 0 | } |
241 | | |
242 | 0 | return TRUE; |
243 | 0 | } |
244 | | |
245 | | static BOOL gdi_Bitmap_SetSurface(rdpContext* context, rdpBitmap* bitmap, BOOL primary) |
246 | 0 | { |
247 | 0 | rdpGdi* gdi = NULL; |
248 | |
|
249 | 0 | if (!context) |
250 | 0 | return FALSE; |
251 | | |
252 | 0 | gdi = context->gdi; |
253 | |
|
254 | 0 | if (!gdi) |
255 | 0 | return FALSE; |
256 | | |
257 | 0 | if (primary) |
258 | 0 | gdi->drawing = gdi->primary; |
259 | 0 | else |
260 | 0 | gdi->drawing = (gdiBitmap*)bitmap; |
261 | |
|
262 | 0 | return TRUE; |
263 | 0 | } |
264 | | |
265 | | /* Glyph Class */ |
266 | | static BOOL gdi_Glyph_New(rdpContext* context, rdpGlyph* glyph) |
267 | 0 | { |
268 | 0 | BYTE* data = NULL; |
269 | 0 | gdiGlyph* gdi_glyph = NULL; |
270 | |
|
271 | 0 | if (!context || !glyph) |
272 | 0 | return FALSE; |
273 | | |
274 | 0 | gdi_glyph = (gdiGlyph*)glyph; |
275 | 0 | gdi_glyph->hdc = gdi_GetDC(); |
276 | |
|
277 | 0 | if (!gdi_glyph->hdc) |
278 | 0 | return FALSE; |
279 | | |
280 | 0 | gdi_glyph->hdc->format = PIXEL_FORMAT_MONO; |
281 | 0 | data = freerdp_glyph_convert(glyph->cx, glyph->cy, glyph->aj); |
282 | |
|
283 | 0 | if (!data) |
284 | 0 | { |
285 | 0 | gdi_DeleteDC(gdi_glyph->hdc); |
286 | 0 | return FALSE; |
287 | 0 | } |
288 | | |
289 | 0 | gdi_glyph->bitmap = gdi_CreateBitmap(glyph->cx, glyph->cy, PIXEL_FORMAT_MONO, data); |
290 | |
|
291 | 0 | if (!gdi_glyph->bitmap) |
292 | 0 | { |
293 | 0 | gdi_DeleteDC(gdi_glyph->hdc); |
294 | 0 | winpr_aligned_free(data); |
295 | 0 | return FALSE; |
296 | 0 | } |
297 | | |
298 | 0 | gdi_SelectObject(gdi_glyph->hdc, (HGDIOBJECT)gdi_glyph->bitmap); |
299 | 0 | gdi_glyph->org_bitmap = NULL; |
300 | 0 | return TRUE; |
301 | 0 | } |
302 | | |
303 | | static void gdi_Glyph_Free(rdpContext* context, rdpGlyph* glyph) |
304 | 0 | { |
305 | 0 | gdiGlyph* gdi_glyph = NULL; |
306 | 0 | gdi_glyph = (gdiGlyph*)glyph; |
307 | |
|
308 | 0 | if (gdi_glyph) |
309 | 0 | { |
310 | 0 | gdi_SelectObject(gdi_glyph->hdc, (HGDIOBJECT)gdi_glyph->org_bitmap); |
311 | 0 | gdi_DeleteObject((HGDIOBJECT)gdi_glyph->bitmap); |
312 | 0 | gdi_DeleteDC(gdi_glyph->hdc); |
313 | 0 | free(glyph->aj); |
314 | 0 | free(glyph); |
315 | 0 | } |
316 | 0 | } |
317 | | |
318 | | static BOOL gdi_Glyph_Draw(rdpContext* context, const rdpGlyph* glyph, INT32 x, INT32 y, INT32 w, |
319 | | INT32 h, INT32 sx, INT32 sy, BOOL fOpRedundant) |
320 | 0 | { |
321 | 0 | const gdiGlyph* gdi_glyph = NULL; |
322 | 0 | rdpGdi* gdi = NULL; |
323 | 0 | HGDI_BRUSH brush = NULL; |
324 | 0 | BOOL rc = FALSE; |
325 | |
|
326 | 0 | if (!context || !glyph) |
327 | 0 | return FALSE; |
328 | | |
329 | 0 | gdi = context->gdi; |
330 | 0 | gdi_glyph = (const gdiGlyph*)glyph; |
331 | |
|
332 | 0 | if (!fOpRedundant) |
333 | 0 | { |
334 | 0 | GDI_RECT rect = { 0 }; |
335 | |
|
336 | 0 | if (x > 0) |
337 | 0 | rect.left = x; |
338 | |
|
339 | 0 | if (y > 0) |
340 | 0 | rect.top = y; |
341 | |
|
342 | 0 | if (x + w > 0) |
343 | 0 | rect.right = x + w - 1; |
344 | |
|
345 | 0 | if (y + h > 0) |
346 | 0 | rect.bottom = y + h - 1; |
347 | |
|
348 | 0 | if ((rect.left < rect.right) && (rect.top < rect.bottom)) |
349 | 0 | { |
350 | 0 | brush = gdi_CreateSolidBrush(gdi->drawing->hdc->bkColor); |
351 | |
|
352 | 0 | if (!brush) |
353 | 0 | return FALSE; |
354 | | |
355 | 0 | gdi_FillRect(gdi->drawing->hdc, &rect, brush); |
356 | 0 | gdi_DeleteObject((HGDIOBJECT)brush); |
357 | 0 | } |
358 | 0 | } |
359 | | |
360 | 0 | brush = gdi_CreateSolidBrush(gdi->drawing->hdc->textColor); |
361 | |
|
362 | 0 | if (!brush) |
363 | 0 | return FALSE; |
364 | | |
365 | 0 | gdi_SelectObject(gdi->drawing->hdc, (HGDIOBJECT)brush); |
366 | 0 | rc = gdi_BitBlt(gdi->drawing->hdc, x, y, w, h, gdi_glyph->hdc, sx, sy, GDI_GLYPH_ORDER, |
367 | 0 | &context->gdi->palette); |
368 | 0 | gdi_DeleteObject((HGDIOBJECT)brush); |
369 | 0 | return rc; |
370 | 0 | } |
371 | | |
372 | | static BOOL gdi_Glyph_BeginDraw(rdpContext* context, INT32 x, INT32 y, INT32 width, INT32 height, |
373 | | UINT32 bgcolor, UINT32 fgcolor, BOOL fOpRedundant) |
374 | 0 | { |
375 | 0 | rdpGdi* gdi = NULL; |
376 | |
|
377 | 0 | if (!context || !context->gdi) |
378 | 0 | return FALSE; |
379 | | |
380 | 0 | gdi = context->gdi; |
381 | |
|
382 | 0 | if (!gdi->drawing || !gdi->drawing->hdc) |
383 | 0 | return FALSE; |
384 | | |
385 | 0 | if (!fOpRedundant) |
386 | 0 | { |
387 | 0 | if (!gdi_decode_color(gdi, bgcolor, &bgcolor, NULL)) |
388 | 0 | return FALSE; |
389 | | |
390 | 0 | if (!gdi_decode_color(gdi, fgcolor, &fgcolor, NULL)) |
391 | 0 | return FALSE; |
392 | | |
393 | 0 | gdi_SetClipRgn(gdi->drawing->hdc, x, y, width, height); |
394 | 0 | gdi_SetTextColor(gdi->drawing->hdc, bgcolor); |
395 | 0 | gdi_SetBkColor(gdi->drawing->hdc, fgcolor); |
396 | |
|
397 | 0 | if (1) |
398 | 0 | { |
399 | 0 | GDI_RECT rect = { 0 }; |
400 | 0 | HGDI_BRUSH brush = gdi_CreateSolidBrush(fgcolor); |
401 | |
|
402 | 0 | if (!brush) |
403 | 0 | return FALSE; |
404 | | |
405 | 0 | if (x > 0) |
406 | 0 | rect.left = x; |
407 | |
|
408 | 0 | if (y > 0) |
409 | 0 | rect.top = y; |
410 | |
|
411 | 0 | rect.right = x + width - 1; |
412 | 0 | rect.bottom = y + height - 1; |
413 | |
|
414 | 0 | if ((x + width > rect.left) && (y + height > rect.top)) |
415 | 0 | gdi_FillRect(gdi->drawing->hdc, &rect, brush); |
416 | |
|
417 | 0 | gdi_DeleteObject((HGDIOBJECT)brush); |
418 | 0 | } |
419 | | |
420 | 0 | return gdi_SetNullClipRgn(gdi->drawing->hdc); |
421 | 0 | } |
422 | | |
423 | 0 | return TRUE; |
424 | 0 | } |
425 | | |
426 | | static BOOL gdi_Glyph_EndDraw(rdpContext* context, INT32 x, INT32 y, INT32 width, INT32 height, |
427 | | UINT32 bgcolor, UINT32 fgcolor) |
428 | 0 | { |
429 | 0 | rdpGdi* gdi = NULL; |
430 | |
|
431 | 0 | if (!context || !context->gdi) |
432 | 0 | return FALSE; |
433 | | |
434 | 0 | gdi = context->gdi; |
435 | |
|
436 | 0 | if (!gdi->drawing || !gdi->drawing->hdc) |
437 | 0 | return FALSE; |
438 | | |
439 | 0 | gdi_SetNullClipRgn(gdi->drawing->hdc); |
440 | 0 | return TRUE; |
441 | 0 | } |
442 | | |
443 | | /* Graphics Module */ |
444 | | BOOL gdi_register_graphics(rdpGraphics* graphics) |
445 | 0 | { |
446 | 0 | rdpBitmap bitmap = { 0 }; |
447 | 0 | rdpGlyph glyph = { 0 }; |
448 | 0 | bitmap.size = sizeof(gdiBitmap); |
449 | 0 | bitmap.New = gdi_Bitmap_New; |
450 | 0 | bitmap.Free = gdi_Bitmap_Free; |
451 | 0 | bitmap.Paint = gdi_Bitmap_Paint; |
452 | 0 | bitmap.Decompress = gdi_Bitmap_Decompress; |
453 | 0 | bitmap.SetSurface = gdi_Bitmap_SetSurface; |
454 | 0 | graphics_register_bitmap(graphics, &bitmap); |
455 | 0 | glyph.size = sizeof(gdiGlyph); |
456 | 0 | glyph.New = gdi_Glyph_New; |
457 | 0 | glyph.Free = gdi_Glyph_Free; |
458 | 0 | glyph.Draw = gdi_Glyph_Draw; |
459 | 0 | glyph.BeginDraw = gdi_Glyph_BeginDraw; |
460 | 0 | glyph.EndDraw = gdi_Glyph_EndDraw; |
461 | 0 | graphics_register_glyph(graphics, &glyph); |
462 | 0 | return TRUE; |
463 | 0 | } |