2 * Copyright © 2013 Hardening <rdp.effort@gmail.com>
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.
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.
30 #include <linux/input.h>
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>
39 #include <winpr/input.h>
41 #include "compositor.h"
42 #include "pixman-renderer.h"
44 #define MAX_FREERDP_FDS 32
46 struct rdp_compositor_config {
60 struct rdp_compositor {
61 struct weston_compositor base;
63 freerdp_listener *listener;
64 struct wl_event_source *listener_events[MAX_FREERDP_FDS];
65 struct rdp_output *output;
73 enum peer_item_flags {
74 RDP_PEER_ACTIVATED = (1 << 0),
75 RDP_PEER_OUTPUT_ENABLED = (1 << 1),
78 struct rdp_peers_item {
81 struct weston_seat seat;
87 struct weston_output base;
88 struct wl_event_source *finish_frame_timer;
89 pixman_image_t *shadow_surface;
94 struct rdp_peer_context {
97 struct rdp_compositor *rdpCompositor;
98 struct wl_event_source *events[MAX_FREERDP_FDS];
99 RFX_CONTEXT *rfx_context;
100 wStream *encode_stream;
102 NSC_CONTEXT *nsc_context;
104 struct rdp_peers_item item;
106 typedef struct rdp_peer_context RdpPeerContext;
109 rdp_compositor_config_init(struct rdp_compositor_config *config) {
111 config->height = 480;
112 config->bind_address = NULL;
114 config->rdp_key = NULL;
115 config->server_cert = NULL;
116 config->server_key = NULL;
117 config->extra_modes = NULL;
118 config->env_socket = 0;
122 rdp_peer_refresh_rfx(pixman_region32_t *damage, pixman_image_t *image, freerdp_peer *peer)
124 int width, height, nrects, i;
125 pixman_box32_t *region, *rects;
128 rdpUpdate *update = peer->update;
129 SURFACE_BITS_COMMAND *cmd = &update->surface_bits_command;
130 RdpPeerContext *context = (RdpPeerContext *)peer->context;
132 Stream_Clear(context->encode_stream);
133 Stream_SetPosition(context->encode_stream, 0);
135 width = (damage->extents.x2 - damage->extents.x1);
136 height = (damage->extents.y2 - damage->extents.y1);
138 cmd->destLeft = damage->extents.x1;
139 cmd->destTop = damage->extents.y1;
140 cmd->destRight = damage->extents.x2;
141 cmd->destBottom = damage->extents.y2;
143 cmd->codecID = peer->settings->RemoteFxCodecId;
145 cmd->height = height;
147 ptr = pixman_image_get_data(image) + damage->extents.x1 +
148 damage->extents.y1 * (pixman_image_get_stride(image) / sizeof(uint32_t));
150 rects = pixman_region32_rectangles(damage, &nrects);
151 context->rfx_rects = realloc(context->rfx_rects, nrects * sizeof *rfxRect);
153 for (i = 0; i < nrects; i++) {
155 rfxRect = &context->rfx_rects[i];
157 rfxRect->x = (region->x1 - damage->extents.x1);
158 rfxRect->y = (region->y1 - damage->extents.y1);
159 rfxRect->width = (region->x2 - region->x1);
160 rfxRect->height = (region->y2 - region->y1);
163 rfx_compose_message(context->rfx_context, context->encode_stream, context->rfx_rects, nrects,
164 (BYTE *)ptr, width, height,
165 pixman_image_get_stride(image)
168 cmd->bitmapDataLength = Stream_GetPosition(context->encode_stream);
169 cmd->bitmapData = Stream_Buffer(context->encode_stream);
171 update->SurfaceBits(update->context, cmd);
176 rdp_peer_refresh_nsc(pixman_region32_t *damage, pixman_image_t *image, freerdp_peer *peer)
180 rdpUpdate *update = peer->update;
181 SURFACE_BITS_COMMAND *cmd = &update->surface_bits_command;
182 RdpPeerContext *context = (RdpPeerContext *)peer->context;
184 Stream_Clear(context->encode_stream);
185 Stream_SetPosition(context->encode_stream, 0);
187 width = (damage->extents.x2 - damage->extents.x1);
188 height = (damage->extents.y2 - damage->extents.y1);
190 cmd->destLeft = damage->extents.x1;
191 cmd->destTop = damage->extents.y1;
192 cmd->destRight = damage->extents.x2;
193 cmd->destBottom = damage->extents.y2;
195 cmd->codecID = peer->settings->NSCodecId;
197 cmd->height = height;
199 ptr = pixman_image_get_data(image) + damage->extents.x1 +
200 damage->extents.y1 * (pixman_image_get_stride(image) / sizeof(uint32_t));
202 nsc_compose_message(context->nsc_context, context->encode_stream, (BYTE *)ptr,
203 cmd->width, cmd->height,
204 pixman_image_get_stride(image));
205 cmd->bitmapDataLength = Stream_GetPosition(context->encode_stream);
206 cmd->bitmapData = Stream_Buffer(context->encode_stream);
207 update->SurfaceBits(update->context, cmd);
211 pixman_image_flipped_subrect(const pixman_box32_t *rect, pixman_image_t *img, BYTE *dest) {
212 int stride = pixman_image_get_stride(img);
214 int toCopy = (rect->x2 - rect->x1) * 4;
215 int height = (rect->y2 - rect->y1);
216 const BYTE *src = (const BYTE *)pixman_image_get_data(img);
217 src += ((rect->y2-1) * stride) + (rect->x1 * 4);
219 for (h = 0; h < height; h++, src -= stride, dest += toCopy)
220 memcpy(dest, src, toCopy);
224 rdp_peer_refresh_raw(pixman_region32_t *region, pixman_image_t *image, freerdp_peer *peer)
226 rdpUpdate *update = peer->update;
227 SURFACE_BITS_COMMAND *cmd = &update->surface_bits_command;
228 SURFACE_FRAME_MARKER *marker = &update->surface_frame_marker;
229 pixman_box32_t *rect, subrect;
231 int heightIncrement, remainingHeight, top;
233 rect = pixman_region32_rectangles(region, &nrects);
238 marker->frameAction = SURFACECMD_FRAMEACTION_BEGIN;
239 update->SurfaceFrameMarker(peer->context, marker);
244 for (i = 0; i < nrects; i++, rect++) {
245 /*weston_log("rect(%d,%d, %d,%d)\n", rect->x1, rect->y1, rect->x2, rect->y2);*/
246 cmd->destLeft = rect->x1;
247 cmd->destRight = rect->x2;
248 cmd->width = rect->x2 - rect->x1;
250 heightIncrement = peer->settings->MultifragMaxRequestSize / (16 + cmd->width * 4);
251 remainingHeight = rect->y2 - rect->y1;
254 subrect.x1 = rect->x1;
255 subrect.x2 = rect->x2;
257 while (remainingHeight) {
258 cmd->height = (remainingHeight > heightIncrement) ? heightIncrement : remainingHeight;
260 cmd->destBottom = top + cmd->height;
261 cmd->bitmapDataLength = cmd->width * cmd->height * 4;
262 cmd->bitmapData = (BYTE *)realloc(cmd->bitmapData, cmd->bitmapDataLength);
265 subrect.y2 = top + cmd->height;
266 pixman_image_flipped_subrect(&subrect, image, cmd->bitmapData);
268 /*weston_log("* sending (%d,%d, %d,%d)\n", subrect.x1, subrect.y1, subrect.x2, subrect.y2); */
269 update->SurfaceBits(peer->context, cmd);
271 remainingHeight -= cmd->height;
276 marker->frameAction = SURFACECMD_FRAMEACTION_END;
277 update->SurfaceFrameMarker(peer->context, marker);
281 rdp_peer_refresh_region(pixman_region32_t *region, freerdp_peer *peer)
283 RdpPeerContext *context = (RdpPeerContext *)peer->context;
284 struct rdp_output *output = context->rdpCompositor->output;
285 rdpSettings *settings = peer->settings;
287 if (settings->RemoteFxCodec)
288 rdp_peer_refresh_rfx(region, output->shadow_surface, peer);
289 else if (settings->NSCodec)
290 rdp_peer_refresh_nsc(region, output->shadow_surface, peer);
292 rdp_peer_refresh_raw(region, output->shadow_surface, peer);
296 rdp_output_start_repaint_loop(struct weston_output *output)
301 gettimeofday(&tv, NULL);
302 msec = tv.tv_sec * 1000 + tv.tv_usec / 1000;
303 weston_output_finish_frame(output, msec);
307 rdp_output_repaint(struct weston_output *output_base, pixman_region32_t *damage)
309 struct rdp_output *output = container_of(output_base, struct rdp_output, base);
310 struct weston_compositor *ec = output->base.compositor;
311 struct rdp_peers_item *outputPeer;
313 pixman_renderer_output_set_buffer(output_base, output->shadow_surface);
314 ec->renderer->repaint_output(&output->base, damage);
316 wl_list_for_each(outputPeer, &output->peers, link) {
317 if ((outputPeer->flags & RDP_PEER_ACTIVATED) &&
318 (outputPeer->flags & RDP_PEER_OUTPUT_ENABLED))
320 rdp_peer_refresh_region(damage, outputPeer->peer);
324 pixman_region32_subtract(&ec->primary_plane.damage,
325 &ec->primary_plane.damage, damage);
327 wl_event_source_timer_update(output->finish_frame_timer, 16);
331 rdp_output_destroy(struct weston_output *output_base)
333 struct rdp_output *output = (struct rdp_output *)output_base;
335 wl_event_source_remove(output->finish_frame_timer);
340 finish_frame_handler(void *data)
342 rdp_output_start_repaint_loop(data);
348 static struct weston_mode *
349 find_matching_mode(struct weston_output *output, struct weston_mode *target) {
350 struct weston_mode *local;
352 wl_list_for_each(local, &output->mode_list, link) {
353 if((local->width == target->width) && (local->height == target->height))
360 rdp_switch_mode(struct weston_output *output, struct weston_mode *target_mode) {
361 struct rdp_output *rdpOutput = container_of(output, struct rdp_output, base);
362 struct rdp_peers_item *rdpPeer;
363 rdpSettings *settings;
364 pixman_image_t *new_shadow_buffer;
365 struct weston_mode *local_mode;
367 local_mode = find_matching_mode(output, target_mode);
369 weston_log("mode %dx%d not available\n", target_mode->width, target_mode->height);
373 if(local_mode == output->current)
376 output->current->flags = 0;
377 output->current = local_mode;
378 output->current->flags = WL_OUTPUT_MODE_CURRENT | WL_OUTPUT_MODE_PREFERRED;
380 pixman_renderer_output_destroy(output);
381 pixman_renderer_output_create(output);
383 new_shadow_buffer = pixman_image_create_bits(PIXMAN_x8r8g8b8, target_mode->width,
384 target_mode->height, 0, target_mode->width * 4);
385 pixman_image_composite32(PIXMAN_OP_SRC, rdpOutput->shadow_surface, 0, new_shadow_buffer,
386 0, 0, 0, 0, 0, 0, target_mode->width, target_mode->height);
387 pixman_image_unref(rdpOutput->shadow_surface);
388 rdpOutput->shadow_surface = new_shadow_buffer;
390 wl_list_for_each(rdpPeer, &rdpOutput->peers, link) {
391 settings = rdpPeer->peer->settings;
392 if(!settings->DesktopResize) {
393 /* too bad this peer does not support desktop resize */
394 rdpPeer->peer->Close(rdpPeer->peer);
396 settings->DesktopWidth = target_mode->width;
397 settings->DesktopHeight = target_mode->height;
398 rdpPeer->peer->update->DesktopResize(rdpPeer->peer->context);
405 parse_extra_modes(const char *modes_str, struct rdp_output *output) {
406 const char *startAt = modes_str;
409 struct weston_mode *mode;
411 while(startAt && *startAt) {
412 nextPos = strchr(startAt, 'x');
416 w = strtoul(startAt, NULL, 0);
417 startAt = nextPos + 1;
421 h = strtoul(startAt, NULL, 0);
423 if(!w || (w > 3000) || !h || (h > 3000))
425 mode = malloc(sizeof *mode);
433 wl_list_insert(&output->base.mode_list, &mode->link);
435 startAt = strchr(startAt, ',');
436 if(startAt && *startAt == ',')
442 rdp_compositor_create_output(struct rdp_compositor *c, int width, int height,
443 const char *extraModes)
445 struct rdp_output *output;
446 struct wl_event_loop *loop;
447 struct weston_mode *currentMode, *next;
449 output = malloc(sizeof *output);
452 memset(output, 0, sizeof *output);
454 wl_list_init(&output->peers);
455 wl_list_init(&output->base.mode_list);
457 currentMode = malloc(sizeof *currentMode);
459 goto out_free_output;
460 currentMode->flags = WL_OUTPUT_MODE_CURRENT | WL_OUTPUT_MODE_PREFERRED;
461 currentMode->width = width;
462 currentMode->height = height;
463 currentMode->refresh = 5;
464 wl_list_insert(&output->base.mode_list, ¤tMode->link);
466 if(parse_extra_modes(extraModes, output) < 0) {
467 weston_log("invalid extra modes\n");
468 goto out_free_output_and_modes;
471 output->base.current = currentMode;
472 weston_output_init(&output->base, &c->base, 0, 0, width, height,
473 WL_OUTPUT_TRANSFORM_NORMAL, 1);
475 output->base.make = "weston";
476 output->base.model = "rdp";
477 output->shadow_surface = pixman_image_create_bits(PIXMAN_x8r8g8b8,
481 if (output->shadow_surface == NULL) {
482 weston_log("Failed to create surface for frame buffer.\n");
486 if (pixman_renderer_output_create(&output->base) < 0)
487 goto out_shadow_surface;
489 weston_output_move(&output->base, 0, 0);
491 loop = wl_display_get_event_loop(c->base.wl_display);
492 output->finish_frame_timer = wl_event_loop_add_timer(loop, finish_frame_handler, output);
494 output->base.origin = output->base.current;
495 output->base.start_repaint_loop = rdp_output_start_repaint_loop;
496 output->base.repaint = rdp_output_repaint;
497 output->base.destroy = rdp_output_destroy;
498 output->base.assign_planes = NULL;
499 output->base.set_backlight = NULL;
500 output->base.set_dpms = NULL;
501 output->base.switch_mode = rdp_switch_mode;
504 wl_list_insert(c->base.output_list.prev, &output->base.link);
508 pixman_image_unref(output->shadow_surface);
510 weston_output_destroy(&output->base);
511 out_free_output_and_modes:
512 wl_list_for_each_safe(currentMode, next, &output->base.mode_list, link)
520 rdp_restore(struct weston_compositor *ec)
525 rdp_destroy(struct weston_compositor *ec)
527 ec->renderer->destroy(ec);
528 weston_compositor_shutdown(ec);
534 int rdp_listener_activity(int fd, uint32_t mask, void *data) {
535 freerdp_listener* instance = (freerdp_listener *)data;
537 if (!(mask & WL_EVENT_READABLE))
539 if (!instance->CheckFileDescriptor(instance))
541 weston_log("failed to check FreeRDP file descriptor\n");
548 int rdp_implant_listener(struct rdp_compositor *c, freerdp_listener* instance) {
551 void* rfds[MAX_FREERDP_FDS];
552 struct wl_event_loop *loop;
554 if (!instance->GetFileDescriptor(instance, rfds, &rcount)) {
555 weston_log("Failed to get FreeRDP file descriptor\n");
559 loop = wl_display_get_event_loop(c->base.wl_display);
560 for (i = 0; i < rcount; i++) {
561 fd = (int)(long)(rfds[i]);
562 c->listener_events[i] = wl_event_loop_add_fd(loop, fd, WL_EVENT_READABLE,
563 rdp_listener_activity, instance);
566 for( ; i < MAX_FREERDP_FDS; i++)
567 c->listener_events[i] = 0;
573 rdp_peer_context_new(freerdp_peer* client, RdpPeerContext* context)
575 context->item.peer = client;
576 context->item.flags = 0;
578 context->rfx_context = rfx_context_new();
579 context->rfx_context->mode = RLGR3;
580 context->rfx_context->width = client->settings->DesktopWidth;
581 context->rfx_context->height = client->settings->DesktopHeight;
582 rfx_context_set_pixel_format(context->rfx_context, RDP_PIXEL_FORMAT_B8G8R8A8);
584 context->nsc_context = nsc_context_new();
585 nsc_context_set_pixel_format(context->nsc_context, RDP_PIXEL_FORMAT_B8G8R8A8);
587 context->encode_stream = Stream_New(NULL, 65536);
591 rdp_peer_context_free(freerdp_peer* client, RdpPeerContext* context)
597 wl_list_remove(&context->item.link);
598 for(i = 0; i < MAX_FREERDP_FDS; i++) {
599 if (context->events[i])
600 wl_event_source_remove(context->events[i]);
603 if(context->item.flags & RDP_PEER_ACTIVATED)
604 weston_seat_release(&context->item.seat);
605 Stream_Free(context->encode_stream, TRUE);
606 nsc_context_free(context->nsc_context);
607 rfx_context_free(context->rfx_context);
608 free(context->rfx_rects);
613 rdp_client_activity(int fd, uint32_t mask, void *data) {
614 freerdp_peer* client = (freerdp_peer *)data;
616 if (!client->CheckFileDescriptor(client)) {
617 weston_log("unable to checkDescriptor for %p\n", client);
623 freerdp_peer_context_free(client);
624 freerdp_peer_free(client);
629 xf_peer_capabilities(freerdp_peer* client)
635 struct rdp_to_xkb_keyboard_layout {
636 UINT32 rdpLayoutCode;
640 /* picked from http://technet.microsoft.com/en-us/library/cc766503(WS.10).aspx */
641 static struct rdp_to_xkb_keyboard_layout rdp_keyboards[] = {
651 /* taken from 2.2.7.1.6 Input Capability Set (TS_INPUT_CAPABILITYSET) */
652 static char *rdp_keyboard_types[] = {
654 "", /* 1: IBM PC/XT or compatible (83-key) keyboard */
655 "", /* 2: Olivetti "ICO" (102-key) keyboard */
656 "", /* 3: IBM PC/AT (84-key) or similar keyboard */
657 "pc102",/* 4: IBM enhanced (101- or 102-key) keyboard */
658 "", /* 5: Nokia 1050 and similar keyboards */
659 "", /* 6: Nokia 9140 and similar keyboards */
660 "" /* 7: Japanese keyboard */
664 xf_peer_post_connect(freerdp_peer* client)
666 RdpPeerContext *peerCtx;
667 struct rdp_compositor *c;
668 struct rdp_output *output;
669 rdpSettings *settings;
670 rdpPointerUpdate *pointer;
671 struct xkb_context *xkbContext;
672 struct xkb_rule_names xkbRuleNames;
673 struct xkb_keymap *keymap;
676 pixman_region32_t damage;
679 peerCtx = (RdpPeerContext *)client->context;
680 c = peerCtx->rdpCompositor;
682 settings = client->settings;
684 if(!settings->SurfaceCommandsEnabled) {
685 weston_log("client doesn't support required SurfaceCommands\n");
689 if(output->base.width != (int)settings->DesktopWidth ||
690 output->base.height != (int)settings->DesktopHeight)
692 if(!settings->DesktopResize) {
693 weston_log("client don't support desktopResize()\n");
697 /* force the client size */
698 settings->DesktopWidth = output->base.width;
699 settings->DesktopHeight = output->base.height;
700 client->update->DesktopResize(client->context);
703 weston_log("kbd_layout:%x kbd_type:%x kbd_subType:%x kbd_functionKeys:%x\n",
704 settings->KeyboardLayout, settings->KeyboardType, settings->KeyboardSubType,
705 settings->KeyboardFunctionKey);
707 memset(&xkbRuleNames, 0, sizeof(xkbRuleNames));
708 if(settings->KeyboardType <= 7)
709 xkbRuleNames.model = rdp_keyboard_types[settings->KeyboardType];
710 for(i = 0; rdp_keyboards[i].xkbLayout; i++) {
711 if(rdp_keyboards[i].rdpLayoutCode == settings->KeyboardLayout) {
712 xkbRuleNames.layout = rdp_keyboards[i].xkbLayout;
718 if(xkbRuleNames.layout) {
719 xkbContext = xkb_context_new(0);
721 weston_log("unable to create a xkb_context\n");
725 keymap = xkb_keymap_new_from_names(xkbContext, &xkbRuleNames, 0);
727 weston_seat_init_keyboard(&peerCtx->item.seat, keymap);
728 weston_seat_init_pointer(&peerCtx->item.seat);
730 peerCtx->item.flags |= RDP_PEER_ACTIVATED;
732 /* disable pointer on the client side */
733 pointer = client->update->pointer;
734 pointer->pointer_system.type = SYSPTR_NULL;
735 pointer->PointerSystem(client->context, &pointer->pointer_system);
737 /* sends a full refresh */
740 box.x2 = output->base.width;
741 box.y2 = output->base.height;
742 pixman_region32_init_with_extents(&damage, &box);
744 rdp_peer_refresh_region(&damage, client);
746 pixman_region32_fini(&damage);
752 xf_peer_activate(freerdp_peer *client)
754 RdpPeerContext *context = (RdpPeerContext *)client->context;
755 rfx_context_reset(context->rfx_context);
760 xf_mouseEvent(rdpInput *input, UINT16 flags, UINT16 x, UINT16 y) {
761 wl_fixed_t wl_x, wl_y;
762 RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
763 struct rdp_output *output;
766 if (flags & PTR_FLAGS_MOVE) {
767 output = peerContext->rdpCompositor->output;
768 if(x < output->base.width && y < output->base.height) {
769 wl_x = wl_fixed_from_int((int)x);
770 wl_y = wl_fixed_from_int((int)y);
771 notify_motion_absolute(&peerContext->item.seat, weston_compositor_get_time(),
776 if (flags & PTR_FLAGS_BUTTON1)
778 else if (flags & PTR_FLAGS_BUTTON2)
780 else if (flags & PTR_FLAGS_BUTTON3)
784 notify_button(&peerContext->item.seat, weston_compositor_get_time(), button,
785 (flags & PTR_FLAGS_DOWN) ? WL_POINTER_BUTTON_STATE_PRESSED : WL_POINTER_BUTTON_STATE_RELEASED
791 xf_extendedMouseEvent(rdpInput *input, UINT16 flags, UINT16 x, UINT16 y) {
792 wl_fixed_t wl_x, wl_y;
793 RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
794 struct rdp_output *output;
796 output = peerContext->rdpCompositor->output;
797 if(x < output->base.width && y < output->base.height) {
798 wl_x = wl_fixed_from_int((int)x);
799 wl_y = wl_fixed_from_int((int)y);
800 notify_motion_absolute(&peerContext->item.seat, weston_compositor_get_time(),
807 xf_input_synchronize_event(rdpInput *input, UINT32 flags)
809 freerdp_peer *client = input->context->peer;
810 RdpPeerContext *peerCtx = (RdpPeerContext *)input->context;
811 struct rdp_output *output = peerCtx->rdpCompositor->output;
813 pixman_region32_t damage;
815 /* sends a full refresh */
818 box.x2 = output->base.width;
819 box.y2 = output->base.height;
820 pixman_region32_init_with_extents(&damage, &box);
822 rdp_peer_refresh_region(&damage, client);
824 pixman_region32_fini(&damage);
827 extern DWORD KEYCODE_TO_VKCODE_EVDEV[];
828 static uint32_t vk_to_keycode[256];
830 init_vk_translator(void)
834 memset(vk_to_keycode, 0, sizeof(vk_to_keycode));
835 for(i = 0; i < 256; i++)
836 vk_to_keycode[KEYCODE_TO_VKCODE_EVDEV[i] & 0xff] = i-8;
840 xf_input_keyboard_event(rdpInput *input, UINT16 flags, UINT16 code)
842 uint32_t scan_code, vk_code, full_code;
843 enum wl_keyboard_key_state keyState;
844 RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
847 if (flags & KBD_FLAGS_DOWN) {
848 keyState = WL_KEYBOARD_KEY_STATE_PRESSED;
850 } else if (flags & KBD_FLAGS_RELEASE) {
851 keyState = WL_KEYBOARD_KEY_STATE_RELEASED;
857 if(flags & KBD_FLAGS_EXTENDED)
858 full_code |= KBD_FLAGS_EXTENDED;
860 vk_code = GetVirtualKeyCodeFromVirtualScanCode(full_code, 4);
862 weston_log("invalid vk_code %x", vk_code);
865 scan_code = vk_to_keycode[vk_code];
868 /*weston_log("code=%x ext=%d vk_code=%x scan_code=%x\n", code, (flags & KBD_FLAGS_EXTENDED) ? 1 : 0,
869 vk_code, scan_code);*/
870 notify_key(&peerContext->item.seat, weston_compositor_get_time(),
871 scan_code, keyState, STATE_UPDATE_AUTOMATIC);
876 xf_input_unicode_keyboard_event(rdpInput *input, UINT16 flags, UINT16 code)
878 weston_log("Client sent a unicode keyboard event (flags:0x%X code:0x%X)\n", flags, code);
883 xf_suppress_output(rdpContext *context, BYTE allow, RECTANGLE_16 *area) {
884 RdpPeerContext *peerContext = (RdpPeerContext *)context;
886 peerContext->item.flags |= RDP_PEER_OUTPUT_ENABLED;
888 peerContext->item.flags &= (~RDP_PEER_OUTPUT_ENABLED);
892 rdp_peer_init(freerdp_peer *client, struct rdp_compositor *c)
895 void *rfds[MAX_FREERDP_FDS];
897 struct wl_event_loop *loop;
898 rdpSettings *settings;
900 RdpPeerContext *peerCtx;
902 client->context_size = sizeof(RdpPeerContext);
903 client->ContextNew = (psPeerContextNew)rdp_peer_context_new;
904 client->ContextFree = (psPeerContextFree)rdp_peer_context_free;
905 freerdp_peer_context_new(client);
907 peerCtx = (RdpPeerContext *) client->context;
908 peerCtx->rdpCompositor = c;
910 settings = client->settings;
911 settings->RdpKeyFile = c->rdp_key;
913 settings->CertificateFile = c->server_cert;
914 settings->PrivateKeyFile = c->server_key;
916 settings->TlsSecurity = FALSE;
919 settings->NlaSecurity = FALSE;
921 client->Capabilities = xf_peer_capabilities;
922 client->PostConnect = xf_peer_post_connect;
923 client->Activate = xf_peer_activate;
925 client->update->SuppressOutput = xf_suppress_output;
927 input = client->input;
928 input->SynchronizeEvent = xf_input_synchronize_event;
929 input->MouseEvent = xf_mouseEvent;
930 input->ExtendedMouseEvent = xf_extendedMouseEvent;
931 input->KeyboardEvent = xf_input_keyboard_event;
932 input->UnicodeKeyboardEvent = xf_input_unicode_keyboard_event;
933 weston_seat_init(&peerCtx->item.seat, &c->base);
935 client->Initialize(client);
937 if (!client->GetFileDescriptor(client, rfds, &rcount)) {
938 weston_log("unable to retrieve client fds\n");
942 loop = wl_display_get_event_loop(c->base.wl_display);
943 for(i = 0; i < rcount; i++) {
944 fd = (int)(long)(rfds[i]);
946 peerCtx->events[i] = wl_event_loop_add_fd(loop, fd, WL_EVENT_READABLE,
947 rdp_client_activity, client);
949 for ( ; i < MAX_FREERDP_FDS; i++)
950 peerCtx->events[i] = 0;
952 wl_list_insert(&c->output->peers, &peerCtx->item.link);
958 rdp_incoming_peer(freerdp_listener *instance, freerdp_peer *client)
960 struct rdp_compositor *c = (struct rdp_compositor *)instance->param4;
961 if (rdp_peer_init(client, c) < 0)
965 static struct weston_compositor *
966 rdp_compositor_create(struct wl_display *display,
967 struct rdp_compositor_config *config,
968 int *argc, char *argv[], struct weston_config *config)
970 struct rdp_compositor *c;
974 c = malloc(sizeof *c);
978 memset(c, 0, sizeof *c);
980 if (weston_compositor_init(&c->base, display, argc, argv, config) < 0)
983 c->base.destroy = rdp_destroy;
984 c->base.restore = rdp_restore;
985 c->rdp_key = config->rdp_key ? strdup(config->rdp_key) : NULL;
987 /* activate TLS only if certificate/key are available */
988 if(config->server_cert && config->server_key) {
989 weston_log("TLS support activated\n");
990 c->server_cert = strdup(config->server_cert);
991 c->server_key = strdup(config->server_key);
992 if(!c->server_cert || !c->server_key)
993 goto err_free_strings;
997 if (pixman_renderer_init(&c->base) < 0)
1000 if (rdp_compositor_create_output(c, config->width, config->height, config->extra_modes) < 0)
1001 goto err_compositor;
1003 if(!config->env_socket) {
1004 c->listener = freerdp_listener_new();
1005 c->listener->PeerAccepted = rdp_incoming_peer;
1006 c->listener->param4 = c;
1007 if(!c->listener->Open(c->listener, config->bind_address, config->port)) {
1008 weston_log("unable to bind rdp socket\n");
1012 if (rdp_implant_listener(c, c->listener) < 0)
1013 goto err_compositor;
1015 /* get the socket from RDP_FD var */
1016 fd_str = getenv("RDP_FD");
1018 weston_log("RDP_FD env variable not set");
1022 fd = strtoul(fd_str, NULL, 10);
1023 if(rdp_peer_init(freerdp_peer_new(fd), c))
1030 freerdp_listener_free(c->listener);
1032 weston_output_destroy(&c->output->base);
1034 weston_compositor_shutdown(&c->base);
1039 free(c->server_cert);
1041 free(c->server_key);
1047 WL_EXPORT struct weston_compositor *
1048 backend_init(struct wl_display *display, int *argc, char *argv[],
1049 struct weston_config *config)
1051 struct rdp_compositor_config config;
1052 rdp_compositor_config_init(&config);
1053 int major, minor, revision;
1055 freerdp_get_version(&major, &minor, &revision);
1056 weston_log("using FreeRDP version %d.%d.%d\n", major, minor, revision);
1057 init_vk_translator();
1059 const struct weston_option rdp_options[] = {
1060 { WESTON_OPTION_BOOLEAN, "env-socket", 0, &config.env_socket },
1061 { WESTON_OPTION_INTEGER, "width", 0, &config.width },
1062 { WESTON_OPTION_INTEGER, "height", 0, &config.height },
1063 { WESTON_OPTION_STRING, "extra-modes", 0, &config.extra_modes },
1064 { WESTON_OPTION_STRING, "address", 0, &config.bind_address },
1065 { WESTON_OPTION_INTEGER, "port", 0, &config.port },
1066 { WESTON_OPTION_STRING, "rdp4-key", 0, &config.rdp_key },
1067 { WESTON_OPTION_STRING, "rdp-tls-cert", 0, &config.server_cert },
1068 { WESTON_OPTION_STRING, "rdp-tls-key", 0, &config.server_key }
1071 parse_options(rdp_options, ARRAY_LENGTH(rdp_options), argc, argv);
1072 return rdp_compositor_create(display, &config, argc, argv, config);