blob: eb43f5988c811bb57db1e83d729b7aea793839ac [file] [log] [blame]
Hardeninga83409c2013-04-01 23:43:58 +02001/*
2 * Copyright © 2013 Hardening <rdp.effort@gmail.com>
3 *
4 * Permission to use, copy, modify, distribute, and sell this software and
5 * its documentation for any purpose is hereby granted without fee, provided
6 * that the above copyright notice appear in all copies and that both that
7 * copyright notice and this permission notice appear in supporting
8 * documentation, and that the name of the copyright holders not be used in
9 * advertising or publicity pertaining to distribution of the software
10 * without specific, written prior permission. The copyright holders make
11 * no representations about the suitability of this software for any
12 * purpose. It is provided "as is" without express or implied warranty.
13 *
14 * THE COPYRIGHT HOLDERS DISCLAIM ALL WARRANTIES WITH REGARD TO THIS
15 * SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
16 * FITNESS, IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
17 * SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
18 * RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
19 * CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
20 * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
21 */
22
23#ifdef HAVE_CONFIG_H
24#include <config.h>
25#endif
26
27#include <stdlib.h>
28#include <string.h>
29#include <errno.h>
30#include <linux/input.h>
31
32#include <freerdp/freerdp.h>
33#include <freerdp/listener.h>
34#include <freerdp/update.h>
35#include <freerdp/input.h>
36#include <freerdp/codec/color.h>
37#include <freerdp/codec/rfx.h>
38#include <freerdp/codec/nsc.h>
Hardening4a3c1502013-04-06 23:39:26 +020039#include <winpr/input.h>
Hardeninga83409c2013-04-01 23:43:58 +020040
41#include "compositor.h"
42#include "pixman-renderer.h"
43
44#define MAX_FREERDP_FDS 32
45
46struct rdp_compositor_config {
47 int width;
48 int height;
49 char *bind_address;
50 int port;
51 char *rdp_key;
52 char *server_cert;
53 char *server_key;
54 char *extra_modes;
55 int env_socket;
56};
57
58struct rdp_output;
59
60struct rdp_compositor {
61 struct weston_compositor base;
Hardeninga83409c2013-04-01 23:43:58 +020062
63 freerdp_listener *listener;
64 struct wl_event_source *listener_events[MAX_FREERDP_FDS];
65 struct rdp_output *output;
66
67 char *server_cert;
68 char *server_key;
69 char *rdp_key;
70 int tls_enabled;
71};
72
73enum peer_item_flags {
74 RDP_PEER_ACTIVATED = (1 << 0),
75 RDP_PEER_OUTPUT_ENABLED = (1 << 1),
76};
77
78struct rdp_peers_item {
79 int flags;
80 freerdp_peer *peer;
81 struct weston_seat seat;
82
83 struct wl_list link;
84};
85
86struct rdp_output {
87 struct weston_output base;
88 struct wl_event_source *finish_frame_timer;
89 pixman_image_t *shadow_surface;
90
91 struct wl_list peers;
92};
93
94struct rdp_peer_context {
95 rdpContext _p;
96 struct rdp_compositor *rdpCompositor;
97
98 /* file descriptors and associated events */
99 int fds[MAX_FREERDP_FDS];
100 struct wl_event_source *events[MAX_FREERDP_FDS];
101
102 RFX_CONTEXT *rfx_context;
103 wStream *encode_stream;
104 RFX_RECT *rfx_rects;
105 NSC_CONTEXT *nsc_context;
106
107 struct rdp_peers_item item;
108};
109typedef struct rdp_peer_context RdpPeerContext;
110
111static void
112rdp_compositor_config_init(struct rdp_compositor_config *config) {
113 config->width = 640;
114 config->height = 480;
115 config->bind_address = NULL;
116 config->port = 3389;
117 config->rdp_key = NULL;
118 config->server_cert = NULL;
119 config->server_key = NULL;
120 config->extra_modes = NULL;
121 config->env_socket = 0;
122}
123
124static void
125rdp_peer_refresh_rfx(pixman_region32_t *damage, pixman_image_t *image, freerdp_peer *peer)
126{
127 int width, height, nrects, i;
128 pixman_box32_t *region, *rects;
129 uint32_t *ptr;
130 RFX_RECT *rfxRect;
131 rdpUpdate *update = peer->update;
132 SURFACE_BITS_COMMAND *cmd = &update->surface_bits_command;
133 RdpPeerContext *context = (RdpPeerContext *)peer->context;
134
Hardeninga1ce6cd2013-05-22 23:40:16 +0200135 Stream_Clear(context->encode_stream);
136 Stream_SetPosition(context->encode_stream, 0);
Hardeninga83409c2013-04-01 23:43:58 +0200137
138 width = (damage->extents.x2 - damage->extents.x1);
139 height = (damage->extents.y2 - damage->extents.y1);
140
141 cmd->destLeft = damage->extents.x1;
142 cmd->destTop = damage->extents.y1;
143 cmd->destRight = damage->extents.x2;
144 cmd->destBottom = damage->extents.y2;
145 cmd->bpp = 32;
146 cmd->codecID = peer->settings->RemoteFxCodecId;
147 cmd->width = width;
148 cmd->height = height;
149
150 ptr = pixman_image_get_data(image) + damage->extents.x1 +
151 damage->extents.y1 * (pixman_image_get_stride(image) / sizeof(uint32_t));
152
153 rects = pixman_region32_rectangles(damage, &nrects);
154 context->rfx_rects = realloc(context->rfx_rects, nrects * sizeof *rfxRect);
155
156 for (i = 0; i < nrects; i++) {
157 region = &rects[i];
158 rfxRect = &context->rfx_rects[i];
159
160 rfxRect->x = (region->x1 - damage->extents.x1);
161 rfxRect->y = (region->y1 - damage->extents.y1);
162 rfxRect->width = (region->x2 - region->x1);
163 rfxRect->height = (region->y2 - region->y1);
164 }
165
166 rfx_compose_message(context->rfx_context, context->encode_stream, context->rfx_rects, nrects,
167 (BYTE *)ptr, width, height,
168 pixman_image_get_stride(image)
169 );
170
Hardeninga1ce6cd2013-05-22 23:40:16 +0200171 cmd->bitmapDataLength = Stream_GetPosition(context->encode_stream);
172 cmd->bitmapData = Stream_Buffer(context->encode_stream);
Hardeninga83409c2013-04-01 23:43:58 +0200173
174 update->SurfaceBits(update->context, cmd);
175}
176
177
178static void
179rdp_peer_refresh_nsc(pixman_region32_t *damage, pixman_image_t *image, freerdp_peer *peer)
180{
181 int width, height;
182 uint32_t *ptr;
183 rdpUpdate *update = peer->update;
184 SURFACE_BITS_COMMAND *cmd = &update->surface_bits_command;
185 RdpPeerContext *context = (RdpPeerContext *)peer->context;
186
Hardeninga1ce6cd2013-05-22 23:40:16 +0200187 Stream_Clear(context->encode_stream);
188 Stream_SetPosition(context->encode_stream, 0);
Hardeninga83409c2013-04-01 23:43:58 +0200189
190 width = (damage->extents.x2 - damage->extents.x1);
191 height = (damage->extents.y2 - damage->extents.y1);
192
193 cmd->destLeft = damage->extents.x1;
194 cmd->destTop = damage->extents.y1;
195 cmd->destRight = damage->extents.x2;
196 cmd->destBottom = damage->extents.y2;
197 cmd->bpp = 32;
198 cmd->codecID = peer->settings->NSCodecId;
199 cmd->width = width;
200 cmd->height = height;
201
202 ptr = pixman_image_get_data(image) + damage->extents.x1 +
203 damage->extents.y1 * (pixman_image_get_stride(image) / sizeof(uint32_t));
204
205 nsc_compose_message(context->nsc_context, context->encode_stream, (BYTE *)ptr,
206 cmd->width, cmd->height,
207 pixman_image_get_stride(image));
Hardeninga1ce6cd2013-05-22 23:40:16 +0200208 cmd->bitmapDataLength = Stream_GetPosition(context->encode_stream);
209 cmd->bitmapData = Stream_Buffer(context->encode_stream);
Hardeninga83409c2013-04-01 23:43:58 +0200210 update->SurfaceBits(update->context, cmd);
211}
212
213static void
214rdp_peer_refresh_raw(pixman_region32_t *region, pixman_image_t *image, freerdp_peer *peer)
215{
216 pixman_image_t *tile;
217 rdpUpdate *update = peer->update;
218 SURFACE_BITS_COMMAND *cmd = &update->surface_bits_command;
219 pixman_box32_t *extends = pixman_region32_extents(region);
220
221 cmd->bpp = 32;
222 cmd->codecID = 0;
223 cmd->width = (extends->x2 - extends->x1);
224 cmd->height = (extends->y2 - extends->y1);;
225 cmd->bitmapDataLength = cmd->width * cmd->height * 4;
226 tile = pixman_image_create_bits(PIXMAN_x8r8g8b8, cmd->width, cmd->height, 0, cmd->width * 4);
227 pixman_image_composite32(PIXMAN_OP_SRC, image, NULL, /* op, src, mask */
228 tile, extends->x1, extends->y1, /* dest, src_x, src_y */
229 0, 0, /* mask_x, mask_y */
230 0, 0, /* dest_x, dest_y */
231 cmd->width, cmd->height /* width, height */
232 );
233 freerdp_image_flip((BYTE *)pixman_image_get_data(tile),
234 (BYTE *)pixman_image_get_data(tile),
235 cmd->width, cmd->height, cmd->bpp
236 );
237 cmd->bitmapData = (BYTE *)pixman_image_get_data(tile);
238 cmd->destLeft = extends->x1;
239 cmd->destTop = extends->y1;
240 cmd->destRight = extends->x2;
241 cmd->destBottom = extends->y2;
242 update->SurfaceBits(peer->context, cmd);
243 pixman_image_unref(tile);
244}
245
246static void
247rdp_peer_refresh_region(pixman_region32_t *region, freerdp_peer *peer)
248{
249 RdpPeerContext *context = (RdpPeerContext *)peer->context;
250 struct rdp_output *output = context->rdpCompositor->output;
251 rdpSettings *settings = peer->settings;
252 pixman_box32_t *extents = pixman_region32_extents(region);
253
254 int regionSz = (extents->x2 - extents->x1) * (extents->y2 - extents->y1);
255
256 if(regionSz > 64 * 64) {
257 if(settings->RemoteFxCodec)
258 rdp_peer_refresh_rfx(region, output->shadow_surface, peer);
259 else if(settings->NSCodec)
260 rdp_peer_refresh_nsc(region, output->shadow_surface, peer);
261 else
262 rdp_peer_refresh_raw(region, output->shadow_surface, peer);
263 } else {
264 rdp_peer_refresh_raw(region, output->shadow_surface, peer);
265 }
266}
267
Jonas Ådahle5a12252013-04-05 23:07:11 +0200268static void
269rdp_output_start_repaint_loop(struct weston_output *output)
270{
271 uint32_t msec;
272 struct timeval tv;
273
274 gettimeofday(&tv, NULL);
275 msec = tv.tv_sec * 1000 + tv.tv_usec / 1000;
276 weston_output_finish_frame(output, msec);
277}
Hardeninga83409c2013-04-01 23:43:58 +0200278
279static void
280rdp_output_repaint(struct weston_output *output_base, pixman_region32_t *damage)
281{
282 struct rdp_output *output = container_of(output_base, struct rdp_output, base);
283 struct weston_compositor *ec = output->base.compositor;
284 struct rdp_peers_item *outputPeer;
285
286 pixman_renderer_output_set_buffer(output_base, output->shadow_surface);
287 ec->renderer->repaint_output(&output->base, damage);
288
289 wl_list_for_each(outputPeer, &output->peers, link) {
290 if ((outputPeer->flags & RDP_PEER_ACTIVATED) &&
291 (outputPeer->flags & RDP_PEER_OUTPUT_ENABLED))
292 {
293 rdp_peer_refresh_region(damage, outputPeer->peer);
294 }
295 }
296
297 pixman_region32_subtract(&ec->primary_plane.damage,
298 &ec->primary_plane.damage, damage);
299
300 wl_event_source_timer_update(output->finish_frame_timer, 16);
301}
302
303static void
304rdp_output_destroy(struct weston_output *output_base)
305{
306 struct rdp_output *output = (struct rdp_output *)output_base;
307
308 wl_event_source_remove(output->finish_frame_timer);
309 free(output);
310}
311
312static int
313finish_frame_handler(void *data)
314{
Jonas Ådahle5a12252013-04-05 23:07:11 +0200315 rdp_output_start_repaint_loop(data);
Hardeninga83409c2013-04-01 23:43:58 +0200316
317 return 1;
318}
319
320
321static struct weston_mode *
322find_matching_mode(struct weston_output *output, struct weston_mode *target) {
323 struct weston_mode *local;
324
325 wl_list_for_each(local, &output->mode_list, link) {
326 if((local->width == target->width) && (local->height == target->height))
327 return local;
328 }
329 return 0;
330}
331
332static int
333rdp_switch_mode(struct weston_output *output, struct weston_mode *target_mode) {
334 struct rdp_output *rdpOutput = container_of(output, struct rdp_output, base);
335 struct rdp_peers_item *rdpPeer;
336 rdpSettings *settings;
337 pixman_image_t *new_shadow_buffer;
338 struct weston_mode *local_mode;
339
340 local_mode = find_matching_mode(output, target_mode);
341 if(!local_mode) {
342 weston_log("mode %dx%d not available\n", target_mode->width, target_mode->height);
343 return -ENOENT;
344 }
345
346 if(local_mode == output->current)
347 return 0;
348
349 output->current->flags = 0;
350 output->current = local_mode;
351 output->current->flags = WL_OUTPUT_MODE_CURRENT | WL_OUTPUT_MODE_PREFERRED;
352
353 pixman_renderer_output_destroy(output);
354 pixman_renderer_output_create(output);
355
356 new_shadow_buffer = pixman_image_create_bits(PIXMAN_x8r8g8b8, target_mode->width,
357 target_mode->height, 0, target_mode->width * 4);
358 pixman_image_composite32(PIXMAN_OP_SRC, rdpOutput->shadow_surface, 0, new_shadow_buffer,
359 0, 0, 0, 0, 0, 0, target_mode->width, target_mode->height);
360 pixman_image_unref(rdpOutput->shadow_surface);
361 rdpOutput->shadow_surface = new_shadow_buffer;
362
363 wl_list_for_each(rdpPeer, &rdpOutput->peers, link) {
364 settings = rdpPeer->peer->settings;
365 if(!settings->DesktopResize) {
366 /* too bad this peer does not support desktop resize */
367 rdpPeer->peer->Close(rdpPeer->peer);
368 } else {
369 settings->DesktopWidth = target_mode->width;
370 settings->DesktopHeight = target_mode->height;
371 rdpPeer->peer->update->DesktopResize(rdpPeer->peer->context);
372 }
373 }
374 return 0;
375}
376
377static int
378parse_extra_modes(const char *modes_str, struct rdp_output *output) {
379 const char *startAt = modes_str;
380 const char *nextPos;
381 int w, h;
382 struct weston_mode *mode;
383
384 while(startAt && *startAt) {
385 nextPos = strchr(startAt, 'x');
386 if(!nextPos)
387 return -1;
388
389 w = strtoul(startAt, NULL, 0);
390 startAt = nextPos + 1;
391 if(!*startAt)
392 return -1;
393
394 h = strtoul(startAt, NULL, 0);
395
396 if(!w || (w > 3000) || !h || (h > 3000))
397 return -1;
398 mode = malloc(sizeof *mode);
399 if(!mode)
400 return -1;
401
402 mode->width = w;
403 mode->height = h;
404 mode->refresh = 5;
405 mode->flags = 0;
406 wl_list_insert(&output->base.mode_list, &mode->link);
407
408 startAt = strchr(startAt, ',');
409 if(startAt && *startAt == ',')
410 startAt++;
411 }
412 return 0;
413}
414static int
415rdp_compositor_create_output(struct rdp_compositor *c, int width, int height,
416 const char *extraModes)
417{
418 struct rdp_output *output;
419 struct wl_event_loop *loop;
420 struct weston_mode *currentMode, *next;
421
422 output = malloc(sizeof *output);
423 if (output == NULL)
424 return -1;
425 memset(output, 0, sizeof *output);
426
427 wl_list_init(&output->peers);
428 wl_list_init(&output->base.mode_list);
429
430 currentMode = malloc(sizeof *currentMode);
431 if(!currentMode)
432 goto out_free_output;
433 currentMode->flags = WL_OUTPUT_MODE_CURRENT | WL_OUTPUT_MODE_PREFERRED;
434 currentMode->width = width;
435 currentMode->height = height;
436 currentMode->refresh = 5;
437 wl_list_insert(&output->base.mode_list, &currentMode->link);
438
439 if(parse_extra_modes(extraModes, output) < 0) {
440 weston_log("invalid extra modes\n");
441 goto out_free_output_and_modes;
442 }
443
444 output->base.current = currentMode;
445 weston_output_init(&output->base, &c->base, 0, 0, width, height,
446 WL_OUTPUT_TRANSFORM_NORMAL);
447
448 output->base.make = "weston";
449 output->base.model = "rdp";
450 output->shadow_surface = pixman_image_create_bits(PIXMAN_x8r8g8b8,
451 width, height,
452 NULL,
453 width * 4);
454 if (output->shadow_surface == NULL) {
455 weston_log("Failed to create surface for frame buffer.\n");
456 goto out_output;
457 }
458
459 if (pixman_renderer_output_create(&output->base) < 0)
460 goto out_shadow_surface;
461
462 weston_output_move(&output->base, 0, 0);
463
464 loop = wl_display_get_event_loop(c->base.wl_display);
465 output->finish_frame_timer = wl_event_loop_add_timer(loop, finish_frame_handler, output);
466
467 output->base.origin = output->base.current;
Jonas Ådahle5a12252013-04-05 23:07:11 +0200468 output->base.start_repaint_loop = rdp_output_start_repaint_loop;
Hardeninga83409c2013-04-01 23:43:58 +0200469 output->base.repaint = rdp_output_repaint;
470 output->base.destroy = rdp_output_destroy;
471 output->base.assign_planes = NULL;
472 output->base.set_backlight = NULL;
473 output->base.set_dpms = NULL;
474 output->base.switch_mode = rdp_switch_mode;
475 c->output = output;
476
477 wl_list_insert(c->base.output_list.prev, &output->base.link);
478 return 0;
479
480out_shadow_surface:
481 pixman_image_unref(output->shadow_surface);
482out_output:
483 weston_output_destroy(&output->base);
484out_free_output_and_modes:
485 wl_list_for_each_safe(currentMode, next, &output->base.mode_list, link)
486 free(currentMode);
487out_free_output:
488 free(output);
489 return -1;
490}
491
492static void
493rdp_restore(struct weston_compositor *ec)
494{
495}
496
497static void
498rdp_destroy(struct weston_compositor *ec)
499{
Hardeninga83409c2013-04-01 23:43:58 +0200500 ec->renderer->destroy(ec);
501 weston_compositor_shutdown(ec);
502
503 free(ec);
504}
505
506static
507int rdp_listener_activity(int fd, uint32_t mask, void *data) {
508 freerdp_listener* instance = (freerdp_listener *)data;
509
510 if (!(mask & WL_EVENT_READABLE))
511 return 0;
512 if (!instance->CheckFileDescriptor(instance))
513 {
514 weston_log("failed to check FreeRDP file descriptor\n");
515 return -1;
516 }
517 return 0;
518}
519
520static
521int rdp_implant_listener(struct rdp_compositor *c, freerdp_listener* instance) {
522 int i, fd;
523 int rcount = 0;
524 void* rfds[MAX_FREERDP_FDS];
525 struct wl_event_loop *loop;
526
527 if (!instance->GetFileDescriptor(instance, rfds, &rcount)) {
528 weston_log("Failed to get FreeRDP file descriptor\n");
529 return -1;
530 }
531
532 loop = wl_display_get_event_loop(c->base.wl_display);
533 for (i = 0; i < rcount; i++) {
534 fd = (int)(long)(rfds[i]);
535 c->listener_events[i] = wl_event_loop_add_fd(loop, fd, WL_EVENT_READABLE,
536 rdp_listener_activity, instance);
537 }
538
539 for( ; i < MAX_FREERDP_FDS; i++)
540 c->listener_events[i] = 0;
541 return 0;
542}
543
544
545static void
546rdp_peer_context_new(freerdp_peer* client, RdpPeerContext* context)
547{
548 context->item.peer = client;
549 context->item.flags = 0;
550
551 context->rfx_context = rfx_context_new();
552 context->rfx_context->mode = RLGR3;
553 context->rfx_context->width = client->settings->DesktopWidth;
554 context->rfx_context->height = client->settings->DesktopHeight;
555 rfx_context_set_pixel_format(context->rfx_context, RDP_PIXEL_FORMAT_B8G8R8A8);
556
557 context->nsc_context = nsc_context_new();
Hardening827358e2013-05-22 23:40:18 +0200558 nsc_context_set_pixel_format(context->nsc_context, RDP_PIXEL_FORMAT_B8G8R8A8);
Hardeninga83409c2013-04-01 23:43:58 +0200559
Hardeninga1ce6cd2013-05-22 23:40:16 +0200560 context->encode_stream = Stream_New(NULL, 65536);
Hardeninga83409c2013-04-01 23:43:58 +0200561}
562
563static void
564rdp_peer_context_free(freerdp_peer* client, RdpPeerContext* context)
565{
566 int i;
567 if(!context)
568 return;
569
570 wl_list_remove(&context->item.link);
571 for(i = 0; i < MAX_FREERDP_FDS; i++) {
572 if(context->fds[i] != -1)
573 wl_event_source_remove(context->events[i]);
574 }
575
576 if(context->item.flags & RDP_PEER_ACTIVATED)
577 weston_seat_release(&context->item.seat);
Hardeninga1ce6cd2013-05-22 23:40:16 +0200578 Stream_Free(context->encode_stream, TRUE);
Hardeninga83409c2013-04-01 23:43:58 +0200579 nsc_context_free(context->nsc_context);
580 rfx_context_free(context->rfx_context);
581 free(context->rfx_rects);
582}
583
584
585static int
586rdp_client_activity(int fd, uint32_t mask, void *data) {
587 freerdp_peer* client = (freerdp_peer *)data;
588
589 if (!client->CheckFileDescriptor(client)) {
590 weston_log("unable to checkDescriptor for %p\n", client);
591 goto out_clean;
592 }
593 return 0;
594
595out_clean:
596 freerdp_peer_context_free(client);
597 freerdp_peer_free(client);
598 return 0;
599}
600
601static BOOL
602xf_peer_capabilities(freerdp_peer* client)
603{
604 return TRUE;
605}
606
Hardening4a3c1502013-04-06 23:39:26 +0200607
Hardeninga83409c2013-04-01 23:43:58 +0200608struct rdp_to_xkb_keyboard_layout {
Hardening4a3c1502013-04-06 23:39:26 +0200609 UINT32 rdpLayoutCode;
610 char *xkbLayout;
Hardeninga83409c2013-04-01 23:43:58 +0200611};
612
613/* picked from http://technet.microsoft.com/en-us/library/cc766503(WS.10).aspx */
614static struct rdp_to_xkb_keyboard_layout rdp_keyboards[] = {
615 {0x00000406, "dk"},
616 {0x00000407, "de"},
617 {0x00000409, "us"},
618 {0x0000040c, "fr"},
619 {0x00000410, "it"},
620 {0x00000813, "be"},
621 {0x00000000, 0},
622};
623
624/* taken from 2.2.7.1.6 Input Capability Set (TS_INPUT_CAPABILITYSET) */
625static char *rdp_keyboard_types[] = {
Hardening4a3c1502013-04-06 23:39:26 +0200626 "", /* 0: unused */
627 "", /* 1: IBM PC/XT or compatible (83-key) keyboard */
628 "", /* 2: Olivetti "ICO" (102-key) keyboard */
629 "", /* 3: IBM PC/AT (84-key) or similar keyboard */
Hardeninga83409c2013-04-01 23:43:58 +0200630 "pc102",/* 4: IBM enhanced (101- or 102-key) keyboard */
Hardening4a3c1502013-04-06 23:39:26 +0200631 "", /* 5: Nokia 1050 and similar keyboards */
632 "", /* 6: Nokia 9140 and similar keyboards */
633 "" /* 7: Japanese keyboard */
Hardeninga83409c2013-04-01 23:43:58 +0200634};
635
636static BOOL
637xf_peer_post_connect(freerdp_peer* client)
638{
639 RdpPeerContext *peerCtx;
640 struct rdp_compositor *c;
641 struct rdp_output *output;
642 rdpSettings *settings;
643 struct xkb_context *xkbContext;
644 struct xkb_rule_names xkbRuleNames;
645 struct xkb_keymap *keymap;
646 int i;
647
Hardeninga83409c2013-04-01 23:43:58 +0200648 peerCtx = (RdpPeerContext *)client->context;
649 c = peerCtx->rdpCompositor;
650 output = c->output;
651 settings = client->settings;
652
653 if(!settings->SurfaceCommandsEnabled) {
654 weston_log("client doesn't support required SurfaceCommands\n");
655 return FALSE;
656 }
657
658 if(output->base.width != (int)settings->DesktopWidth ||
659 output->base.height != (int)settings->DesktopHeight)
660 {
661 if(!settings->DesktopResize) {
662 weston_log("client don't support desktopResize()\n");
663 return FALSE;
664 }
665
666 /* force the client size */
667 settings->DesktopWidth = output->base.width;
668 settings->DesktopHeight = output->base.height;
669 client->update->DesktopResize(client->context);
670 }
671
672 weston_log("kbd_layout:%x kbd_type:%x kbd_subType:%x kbd_functionKeys:%x\n",
673 settings->KeyboardLayout, settings->KeyboardType, settings->KeyboardSubType,
674 settings->KeyboardFunctionKey);
675
676 memset(&xkbRuleNames, 0, sizeof(xkbRuleNames));
677 if(settings->KeyboardType <= 7)
678 xkbRuleNames.model = rdp_keyboard_types[settings->KeyboardType];
679 for(i = 0; rdp_keyboards[i].xkbLayout; i++) {
680 if(rdp_keyboards[i].rdpLayoutCode == settings->KeyboardLayout) {
681 xkbRuleNames.layout = rdp_keyboards[i].xkbLayout;
682 break;
683 }
684 }
685
686 keymap = NULL;
687 if(xkbRuleNames.layout) {
688 xkbContext = xkb_context_new(0);
689 if(!xkbContext) {
690 weston_log("unable to create a xkb_context\n");
691 return FALSE;
692 }
693
694 keymap = xkb_keymap_new_from_names(xkbContext, &xkbRuleNames, 0);
695 }
696 weston_seat_init_keyboard(&peerCtx->item.seat, keymap);
697 weston_seat_init_pointer(&peerCtx->item.seat);
698
699 peerCtx->item.flags |= RDP_PEER_ACTIVATED;
700 return TRUE;
701}
702
703static BOOL
704xf_peer_activate(freerdp_peer *client)
705{
Hardening827358e2013-05-22 23:40:18 +0200706 RdpPeerContext *context = (RdpPeerContext *)client->context;
707 rfx_context_reset(context->rfx_context);
Hardeninga83409c2013-04-01 23:43:58 +0200708 return TRUE;
709}
710
711static void
712xf_mouseEvent(rdpInput *input, UINT16 flags, UINT16 x, UINT16 y) {
713 wl_fixed_t wl_x, wl_y;
714 RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
715 struct rdp_output *output;
716 uint32_t button = 0;
717
718 if (flags & PTR_FLAGS_MOVE) {
719 output = peerContext->rdpCompositor->output;
720 if(x < output->base.width && y < output->base.height) {
721 wl_x = wl_fixed_from_int((int)x);
722 wl_y = wl_fixed_from_int((int)y);
723 notify_motion_absolute(&peerContext->item.seat, weston_compositor_get_time(),
724 wl_x, wl_y);
725 }
726 }
727
728 if (flags & PTR_FLAGS_BUTTON1)
729 button = BTN_LEFT;
730 else if (flags & PTR_FLAGS_BUTTON2)
731 button = BTN_RIGHT;
732 else if (flags & PTR_FLAGS_BUTTON3)
733 button = BTN_MIDDLE;
734
735 if(button) {
736 notify_button(&peerContext->item.seat, weston_compositor_get_time(), button,
737 (flags & PTR_FLAGS_DOWN) ? WL_POINTER_BUTTON_STATE_PRESSED : WL_POINTER_BUTTON_STATE_RELEASED
738 );
739 }
740}
741
742static void
743xf_extendedMouseEvent(rdpInput *input, UINT16 flags, UINT16 x, UINT16 y) {
744 wl_fixed_t wl_x, wl_y;
745 RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
746 struct rdp_output *output;
747
748 output = peerContext->rdpCompositor->output;
749 if(x < output->base.width && y < output->base.height) {
750 wl_x = wl_fixed_from_int((int)x);
751 wl_y = wl_fixed_from_int((int)y);
752 notify_motion_absolute(&peerContext->item.seat, weston_compositor_get_time(),
753 wl_x, wl_y);
754 }
755}
756
757
758static void
759xf_input_synchronize_event(rdpInput *input, UINT32 flags)
760{
761 freerdp_peer *client = input->context->peer;
762 rdpPointerUpdate *pointer = client->update->pointer;
763 RdpPeerContext *peerCtx = (RdpPeerContext *)input->context;
764 struct rdp_output *output = peerCtx->rdpCompositor->output;
765 pixman_box32_t box;
766 pixman_region32_t damage;
767
768 /* disable pointer on the client side */
769 pointer->pointer_system.type = SYSPTR_NULL;
770 pointer->PointerSystem(client->context, &pointer->pointer_system);
771
772 /* sends a full refresh */
773 box.x1 = 0;
774 box.y1 = 0;
775 box.x2 = output->base.width;
776 box.y2 = output->base.height;
777 pixman_region32_init_with_extents(&damage, &box);
778
779 rdp_peer_refresh_region(&damage, client);
780
781 pixman_region32_fini(&damage);
782}
783
Hardening4a3c1502013-04-06 23:39:26 +0200784extern DWORD KEYCODE_TO_VKCODE_EVDEV[];
785static uint32_t vk_to_keycode[256];
786static void
787init_vk_translator(void)
788{
789 int i;
790
791 memset(vk_to_keycode, 0, sizeof(vk_to_keycode));
792 for(i = 0; i < 256; i++)
793 vk_to_keycode[KEYCODE_TO_VKCODE_EVDEV[i] & 0xff] = i-8;
794}
795
Hardeninga83409c2013-04-01 23:43:58 +0200796static void
797xf_input_keyboard_event(rdpInput *input, UINT16 flags, UINT16 code)
798{
Hardening4a3c1502013-04-06 23:39:26 +0200799 uint32_t scan_code, vk_code, full_code;
Hardeninga83409c2013-04-01 23:43:58 +0200800 enum wl_keyboard_key_state keyState;
801 RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
802 int notify = 0;
803
804 if (flags & KBD_FLAGS_DOWN) {
805 keyState = WL_KEYBOARD_KEY_STATE_PRESSED;
806 notify = 1;
807 } else if (flags & KBD_FLAGS_RELEASE) {
808 keyState = WL_KEYBOARD_KEY_STATE_RELEASED;
809 notify = 1;
810 }
811
Hardening4a3c1502013-04-06 23:39:26 +0200812 if(notify) {
813 full_code = code;
814 if(flags & KBD_FLAGS_EXTENDED)
815 full_code |= KBD_FLAGS_EXTENDED;
816
817 vk_code = GetVirtualKeyCodeFromVirtualScanCode(full_code, 4);
818 if(vk_code > 0xff) {
819 weston_log("invalid vk_code %x", vk_code);
820 return;
821 }
822 scan_code = vk_to_keycode[vk_code];
823
824
825 /*weston_log("code=%x ext=%d vk_code=%x scan_code=%x\n", code, (flags & KBD_FLAGS_EXTENDED) ? 1 : 0,
826 vk_code, scan_code);*/
Hardeninga83409c2013-04-01 23:43:58 +0200827 notify_key(&peerContext->item.seat, weston_compositor_get_time(),
Hardening4a3c1502013-04-06 23:39:26 +0200828 scan_code, keyState, STATE_UPDATE_AUTOMATIC);
829 }
Hardeninga83409c2013-04-01 23:43:58 +0200830}
831
832static void
833xf_input_unicode_keyboard_event(rdpInput *input, UINT16 flags, UINT16 code)
834{
835 weston_log("Client sent a unicode keyboard event (flags:0x%X code:0x%X)\n", flags, code);
836}
837
838
839static void
840xf_suppress_output(rdpContext *context, BYTE allow, RECTANGLE_16 *area) {
841 RdpPeerContext *peerContext = (RdpPeerContext *)context;
842 if(allow)
843 peerContext->item.flags |= RDP_PEER_OUTPUT_ENABLED;
844 else
845 peerContext->item.flags &= (~RDP_PEER_OUTPUT_ENABLED);
846}
847
848static int
849rdp_peer_init(freerdp_peer *client, struct rdp_compositor *c)
850{
851 int rcount = 0;
852 void *rfds[MAX_FREERDP_FDS];
853 int i, fd;
854 struct wl_event_loop *loop;
855 rdpSettings *settings;
856 rdpInput *input;
857 RdpPeerContext *peerCtx;
858
859 client->context_size = sizeof(RdpPeerContext);
860 client->ContextNew = (psPeerContextNew)rdp_peer_context_new;
861 client->ContextFree = (psPeerContextFree)rdp_peer_context_free;
862 freerdp_peer_context_new(client);
863
864 peerCtx = (RdpPeerContext *) client->context;
865 peerCtx->rdpCompositor = c;
866
867 settings = client->settings;
868 settings->RdpKeyFile = c->rdp_key;
869 if(c->tls_enabled) {
870 settings->CertificateFile = c->server_cert;
871 settings->PrivateKeyFile = c->server_key;
872 } else {
873 settings->TlsSecurity = FALSE;
874 }
875
876 settings->NlaSecurity = FALSE;
877
878 client->Capabilities = xf_peer_capabilities;
879 client->PostConnect = xf_peer_post_connect;
880 client->Activate = xf_peer_activate;
881
882 client->update->SuppressOutput = xf_suppress_output;
883
884 input = client->input;
885 input->SynchronizeEvent = xf_input_synchronize_event;
886 input->MouseEvent = xf_mouseEvent;
887 input->ExtendedMouseEvent = xf_extendedMouseEvent;
888 input->KeyboardEvent = xf_input_keyboard_event;
889 input->UnicodeKeyboardEvent = xf_input_unicode_keyboard_event;
890 weston_seat_init(&peerCtx->item.seat, &c->base);
891
892 client->Initialize(client);
893
894 if (!client->GetFileDescriptor(client, rfds, &rcount)) {
895 weston_log("unable to retrieve client fds\n");
896 return -1;
897 }
898
899 loop = wl_display_get_event_loop(c->base.wl_display);
900 for(i = 0; i < rcount; i++) {
901 fd = (int)(long)(rfds[i]);
902
903 peerCtx->fds[i] = fd;
904 peerCtx->events[i] = wl_event_loop_add_fd(loop, fd, WL_EVENT_READABLE,
905 rdp_client_activity, client);
906 }
907 for( ; i < MAX_FREERDP_FDS; i++) {
908 peerCtx->fds[i] = -1;
909 peerCtx->events[i] = 0;
910 }
911
912 wl_list_insert(&c->output->peers, &peerCtx->item.link);
913 return 0;
914}
915
916
917static void
918rdp_incoming_peer(freerdp_listener *instance, freerdp_peer *client)
919{
920 struct rdp_compositor *c = (struct rdp_compositor *)instance->param4;
921 if (rdp_peer_init(client, c) < 0)
922 return;
923}
924
925static struct weston_compositor *
926rdp_compositor_create(struct wl_display *display,
927 struct rdp_compositor_config *config,
Ossama Othmana50e6e42013-05-14 09:48:26 -0700928 int *argc, char *argv[], int config_fd)
Hardeninga83409c2013-04-01 23:43:58 +0200929{
930 struct rdp_compositor *c;
931 char *fd_str;
932 int fd;
933
934 c = malloc(sizeof *c);
935 if (c == NULL)
936 return NULL;
937
938 memset(c, 0, sizeof *c);
939
940 if (weston_compositor_init(&c->base, display, argc, argv,
Ossama Othmana50e6e42013-05-14 09:48:26 -0700941 config_fd) < 0)
Hardeninga83409c2013-04-01 23:43:58 +0200942 goto err_free;
943
Hardeninga83409c2013-04-01 23:43:58 +0200944 c->base.destroy = rdp_destroy;
945 c->base.restore = rdp_restore;
946 c->rdp_key = config->rdp_key ? strdup(config->rdp_key) : NULL;
947
948 /* activate TLS only if certificate/key are available */
949 if(config->server_cert && config->server_key) {
950 weston_log("TLS support activated\n");
951 c->server_cert = strdup(config->server_cert);
952 c->server_key = strdup(config->server_key);
953 if(!c->server_cert || !c->server_key)
954 goto err_free_strings;
955 c->tls_enabled = 1;
956 }
957
958 if (pixman_renderer_init(&c->base) < 0)
959 goto err_compositor;
960
961 if (rdp_compositor_create_output(c, config->width, config->height, config->extra_modes) < 0)
962 goto err_compositor;
963
964 if(!config->env_socket) {
965 c->listener = freerdp_listener_new();
966 c->listener->PeerAccepted = rdp_incoming_peer;
967 c->listener->param4 = c;
968 if(!c->listener->Open(c->listener, config->bind_address, config->port)) {
969 weston_log("unable to bind rdp socket\n");
970 goto err_listener;
971 }
972
973 if (rdp_implant_listener(c, c->listener) < 0)
974 goto err_compositor;
975 } else {
976 /* get the socket from RDP_FD var */
977 fd_str = getenv("RDP_FD");
978 if(!fd_str) {
979 weston_log("RDP_FD env variable not set");
980 goto err_output;
981 }
982
983 fd = strtoul(fd_str, NULL, 10);
984 if(rdp_peer_init(freerdp_peer_new(fd), c))
985 goto err_output;
986 }
987
988 return &c->base;
989
990err_listener:
991 freerdp_listener_free(c->listener);
992err_output:
993 weston_output_destroy(&c->output->base);
994err_compositor:
995 weston_compositor_shutdown(&c->base);
996err_free_strings:
997 if(c->rdp_key)
998 free(c->rdp_key);
999 if(c->server_cert)
1000 free(c->server_cert);
1001 if(c->server_key)
1002 free(c->server_key);
Hardeninga83409c2013-04-01 23:43:58 +02001003err_free:
1004 free(c);
1005 return NULL;
1006}
1007
1008WL_EXPORT struct weston_compositor *
1009backend_init(struct wl_display *display, int *argc, char *argv[],
Hardeninga1ce6cd2013-05-22 23:40:16 +02001010 int config_fd)
Hardeninga83409c2013-04-01 23:43:58 +02001011{
1012 struct rdp_compositor_config config;
1013 rdp_compositor_config_init(&config);
1014 int major, minor, revision;
1015
1016 freerdp_get_version(&major, &minor, &revision);
1017 weston_log("using FreeRDP version %d.%d.%d\n", major, minor, revision);
Hardening4a3c1502013-04-06 23:39:26 +02001018 init_vk_translator();
Hardeninga83409c2013-04-01 23:43:58 +02001019
1020 const struct weston_option rdp_options[] = {
1021 { WESTON_OPTION_BOOLEAN, "env-socket", 0, &config.env_socket },
1022 { WESTON_OPTION_INTEGER, "width", 0, &config.width },
1023 { WESTON_OPTION_INTEGER, "height", 0, &config.height },
1024 { WESTON_OPTION_STRING, "extra-modes", 0, &config.extra_modes },
1025 { WESTON_OPTION_STRING, "address", 0, &config.bind_address },
1026 { WESTON_OPTION_INTEGER, "port", 0, &config.port },
1027 { WESTON_OPTION_STRING, "rdp4-key", 0, &config.rdp_key },
1028 { WESTON_OPTION_STRING, "rdp-tls-cert", 0, &config.server_cert },
1029 { WESTON_OPTION_STRING, "rdp-tls-key", 0, &config.server_key }
1030 };
1031
1032 parse_options(rdp_options, ARRAY_LENGTH(rdp_options), argc, argv);
Hardeninga1ce6cd2013-05-22 23:40:16 +02001033 return rdp_compositor_create(display, &config, argc, argv, config_fd);
Hardeninga83409c2013-04-01 23:43:58 +02001034}