You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
weston/src/compositor-rdp.c

1093 lines
30 KiB

/*
* Copyright © 2013 Hardening <rdp.effort@gmail.com>
*
* Permission to use, copy, modify, distribute, and sell this software and
* its documentation for any purpose is hereby granted without fee, provided
* that the above copyright notice appear in all copies and that both that
* copyright notice and this permission notice appear in supporting
* documentation, and that the name of the copyright holders not be used in
* advertising or publicity pertaining to distribution of the software
* without specific, written prior permission. The copyright holders make
* no representations about the suitability of this software for any
* purpose. It is provided "as is" without express or implied warranty.
*
* THE COPYRIGHT HOLDERS DISCLAIM ALL WARRANTIES WITH REGARD TO THIS
* SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
* FITNESS, IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
* SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
* RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
* CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
* CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include "config.h"
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <linux/input.h>
#include <freerdp/freerdp.h>
#include <freerdp/listener.h>
#include <freerdp/update.h>
#include <freerdp/input.h>
#include <freerdp/codec/color.h>
#include <freerdp/codec/rfx.h>
#include <freerdp/codec/nsc.h>
#include <winpr/input.h>
#include "compositor.h"
#include "pixman-renderer.h"
#define MAX_FREERDP_FDS 32
#define DEFAULT_AXIS_STEP_DISTANCE wl_fixed_from_int(10)
struct rdp_compositor_config {
int width;
int height;
char *bind_address;
int port;
char *rdp_key;
char *server_cert;
char *server_key;
char *extra_modes;
int env_socket;
};
struct rdp_output;
struct rdp_compositor {
struct weston_compositor base;
freerdp_listener *listener;
struct wl_event_source *listener_events[MAX_FREERDP_FDS];
struct rdp_output *output;
char *server_cert;
char *server_key;
char *rdp_key;
int tls_enabled;
};
enum peer_item_flags {
RDP_PEER_ACTIVATED = (1 << 0),
RDP_PEER_OUTPUT_ENABLED = (1 << 1),
};
struct rdp_peers_item {
int flags;
freerdp_peer *peer;
struct weston_seat seat;
struct wl_list link;
};
struct rdp_output {
struct weston_output base;
struct wl_event_source *finish_frame_timer;
pixman_image_t *shadow_surface;
struct wl_list peers;
};
struct rdp_peer_context {
rdpContext _p;
struct rdp_compositor *rdpCompositor;
struct wl_event_source *events[MAX_FREERDP_FDS];
RFX_CONTEXT *rfx_context;
wStream *encode_stream;
RFX_RECT *rfx_rects;
NSC_CONTEXT *nsc_context;
struct rdp_peers_item item;
};
typedef struct rdp_peer_context RdpPeerContext;
static void
rdp_compositor_config_init(struct rdp_compositor_config *config) {
config->width = 640;
config->height = 480;
config->bind_address = NULL;
config->port = 3389;
config->rdp_key = NULL;
config->server_cert = NULL;
config->server_key = NULL;
config->extra_modes = NULL;
config->env_socket = 0;
}
static void
rdp_peer_refresh_rfx(pixman_region32_t *damage, pixman_image_t *image, freerdp_peer *peer)
{
int width, height, nrects, i;
pixman_box32_t *region, *rects;
uint32_t *ptr;
RFX_RECT *rfxRect;
rdpUpdate *update = peer->update;
SURFACE_BITS_COMMAND *cmd = &update->surface_bits_command;
RdpPeerContext *context = (RdpPeerContext *)peer->context;
Stream_Clear(context->encode_stream);
Stream_SetPosition(context->encode_stream, 0);
width = (damage->extents.x2 - damage->extents.x1);
height = (damage->extents.y2 - damage->extents.y1);
cmd->destLeft = damage->extents.x1;
cmd->destTop = damage->extents.y1;
cmd->destRight = damage->extents.x2;
cmd->destBottom = damage->extents.y2;
cmd->bpp = 32;
cmd->codecID = peer->settings->RemoteFxCodecId;
cmd->width = width;
cmd->height = height;
ptr = pixman_image_get_data(image) + damage->extents.x1 +
damage->extents.y1 * (pixman_image_get_stride(image) / sizeof(uint32_t));
rects = pixman_region32_rectangles(damage, &nrects);
context->rfx_rects = realloc(context->rfx_rects, nrects * sizeof *rfxRect);
for (i = 0; i < nrects; i++) {
region = &rects[i];
rfxRect = &context->rfx_rects[i];
rfxRect->x = (region->x1 - damage->extents.x1);
rfxRect->y = (region->y1 - damage->extents.y1);
rfxRect->width = (region->x2 - region->x1);
rfxRect->height = (region->y2 - region->y1);
}
rfx_compose_message(context->rfx_context, context->encode_stream, context->rfx_rects, nrects,
(BYTE *)ptr, width, height,
pixman_image_get_stride(image)
);
cmd->bitmapDataLength = Stream_GetPosition(context->encode_stream);
cmd->bitmapData = Stream_Buffer(context->encode_stream);
update->SurfaceBits(update->context, cmd);
}
static void
rdp_peer_refresh_nsc(pixman_region32_t *damage, pixman_image_t *image, freerdp_peer *peer)
{
int width, height;
uint32_t *ptr;
rdpUpdate *update = peer->update;
SURFACE_BITS_COMMAND *cmd = &update->surface_bits_command;
RdpPeerContext *context = (RdpPeerContext *)peer->context;
Stream_Clear(context->encode_stream);
Stream_SetPosition(context->encode_stream, 0);
width = (damage->extents.x2 - damage->extents.x1);
height = (damage->extents.y2 - damage->extents.y1);
cmd->destLeft = damage->extents.x1;
cmd->destTop = damage->extents.y1;
cmd->destRight = damage->extents.x2;
cmd->destBottom = damage->extents.y2;
cmd->bpp = 32;
cmd->codecID = peer->settings->NSCodecId;
cmd->width = width;
cmd->height = height;
ptr = pixman_image_get_data(image) + damage->extents.x1 +
damage->extents.y1 * (pixman_image_get_stride(image) / sizeof(uint32_t));
nsc_compose_message(context->nsc_context, context->encode_stream, (BYTE *)ptr,
cmd->width, cmd->height,
pixman_image_get_stride(image));
cmd->bitmapDataLength = Stream_GetPosition(context->encode_stream);
cmd->bitmapData = Stream_Buffer(context->encode_stream);
update->SurfaceBits(update->context, cmd);
}
static void
pixman_image_flipped_subrect(const pixman_box32_t *rect, pixman_image_t *img, BYTE *dest) {
int stride = pixman_image_get_stride(img);
int h;
int toCopy = (rect->x2 - rect->x1) * 4;
int height = (rect->y2 - rect->y1);
const BYTE *src = (const BYTE *)pixman_image_get_data(img);
src += ((rect->y2-1) * stride) + (rect->x1 * 4);
for (h = 0; h < height; h++, src -= stride, dest += toCopy)
memcpy(dest, src, toCopy);
}
static void
rdp_peer_refresh_raw(pixman_region32_t *region, pixman_image_t *image, freerdp_peer *peer)
{
rdpUpdate *update = peer->update;
SURFACE_BITS_COMMAND *cmd = &update->surface_bits_command;
SURFACE_FRAME_MARKER *marker = &update->surface_frame_marker;
pixman_box32_t *rect, subrect;
int nrects, i;
int heightIncrement, remainingHeight, top;
rect = pixman_region32_rectangles(region, &nrects);
if (!nrects)
return;
marker->frameId++;
marker->frameAction = SURFACECMD_FRAMEACTION_BEGIN;
update->SurfaceFrameMarker(peer->context, marker);
cmd->bpp = 32;
cmd->codecID = 0;
for (i = 0; i < nrects; i++, rect++) {
/*weston_log("rect(%d,%d, %d,%d)\n", rect->x1, rect->y1, rect->x2, rect->y2);*/
cmd->destLeft = rect->x1;
cmd->destRight = rect->x2;
cmd->width = rect->x2 - rect->x1;
heightIncrement = peer->settings->MultifragMaxRequestSize / (16 + cmd->width * 4);
remainingHeight = rect->y2 - rect->y1;
top = rect->y1;
subrect.x1 = rect->x1;
subrect.x2 = rect->x2;
while (remainingHeight) {
cmd->height = (remainingHeight > heightIncrement) ? heightIncrement : remainingHeight;
cmd->destTop = top;
cmd->destBottom = top + cmd->height;
cmd->bitmapDataLength = cmd->width * cmd->height * 4;
cmd->bitmapData = (BYTE *)realloc(cmd->bitmapData, cmd->bitmapDataLength);
subrect.y1 = top;
subrect.y2 = top + cmd->height;
pixman_image_flipped_subrect(&subrect, image, cmd->bitmapData);
/*weston_log("* sending (%d,%d, %d,%d)\n", subrect.x1, subrect.y1, subrect.x2, subrect.y2); */
update->SurfaceBits(peer->context, cmd);
remainingHeight -= cmd->height;
top += cmd->height;
}
}
marker->frameAction = SURFACECMD_FRAMEACTION_END;
update->SurfaceFrameMarker(peer->context, marker);
}
static void
rdp_peer_refresh_region(pixman_region32_t *region, freerdp_peer *peer)
{
RdpPeerContext *context = (RdpPeerContext *)peer->context;
struct rdp_output *output = context->rdpCompositor->output;
rdpSettings *settings = peer->settings;
if (settings->RemoteFxCodec)
rdp_peer_refresh_rfx(region, output->shadow_surface, peer);
else if (settings->NSCodec)
rdp_peer_refresh_nsc(region, output->shadow_surface, peer);
else
rdp_peer_refresh_raw(region, output->shadow_surface, peer);
}
static void
rdp_output_start_repaint_loop(struct weston_output *output)
{
uint32_t msec;
struct timeval tv;
gettimeofday(&tv, NULL);
msec = tv.tv_sec * 1000 + tv.tv_usec / 1000;
weston_output_finish_frame(output, msec);
}
static void
rdp_output_repaint(struct weston_output *output_base, pixman_region32_t *damage)
{
struct rdp_output *output = container_of(output_base, struct rdp_output, base);
struct weston_compositor *ec = output->base.compositor;
struct rdp_peers_item *outputPeer;
pixman_renderer_output_set_buffer(output_base, output->shadow_surface);
ec->renderer->repaint_output(&output->base, damage);
wl_list_for_each(outputPeer, &output->peers, link) {
if ((outputPeer->flags & RDP_PEER_ACTIVATED) &&
(outputPeer->flags & RDP_PEER_OUTPUT_ENABLED))
{
rdp_peer_refresh_region(damage, outputPeer->peer);
}
}
pixman_region32_subtract(&ec->primary_plane.damage,
&ec->primary_plane.damage, damage);
wl_event_source_timer_update(output->finish_frame_timer, 16);
}
static void
rdp_output_destroy(struct weston_output *output_base)
{
struct rdp_output *output = (struct rdp_output *)output_base;
wl_event_source_remove(output->finish_frame_timer);
free(output);
}
static int
finish_frame_handler(void *data)
{
rdp_output_start_repaint_loop(data);
return 1;
}
static struct weston_mode *
find_matching_mode(struct weston_output *output, struct weston_mode *target) {
struct weston_mode *local;
wl_list_for_each(local, &output->mode_list, link) {
if((local->width == target->width) && (local->height == target->height))
return local;
}
return 0;
}
static int
rdp_switch_mode(struct weston_output *output, struct weston_mode *target_mode) {
struct rdp_output *rdpOutput = container_of(output, struct rdp_output, base);
struct rdp_peers_item *rdpPeer;
rdpSettings *settings;
pixman_image_t *new_shadow_buffer;
struct weston_mode *local_mode;
local_mode = find_matching_mode(output, target_mode);
if(!local_mode) {
weston_log("mode %dx%d not available\n", target_mode->width, target_mode->height);
return -ENOENT;
}
if(local_mode == output->current)
return 0;
output->current->flags = 0;
output->current = local_mode;
output->current->flags = WL_OUTPUT_MODE_CURRENT | WL_OUTPUT_MODE_PREFERRED;
pixman_renderer_output_destroy(output);
pixman_renderer_output_create(output);
new_shadow_buffer = pixman_image_create_bits(PIXMAN_x8r8g8b8, target_mode->width,
target_mode->height, 0, target_mode->width * 4);
pixman_image_composite32(PIXMAN_OP_SRC, rdpOutput->shadow_surface, 0, new_shadow_buffer,
0, 0, 0, 0, 0, 0, target_mode->width, target_mode->height);
pixman_image_unref(rdpOutput->shadow_surface);
rdpOutput->shadow_surface = new_shadow_buffer;
wl_list_for_each(rdpPeer, &rdpOutput->peers, link) {
settings = rdpPeer->peer->settings;
if(!settings->DesktopResize) {
/* too bad this peer does not support desktop resize */
rdpPeer->peer->Close(rdpPeer->peer);
} else {
settings->DesktopWidth = target_mode->width;
settings->DesktopHeight = target_mode->height;
rdpPeer->peer->update->DesktopResize(rdpPeer->peer->context);
}
}
return 0;
}
static int
parse_extra_modes(const char *modes_str, struct rdp_output *output) {
const char *startAt = modes_str;
const char *nextPos;
int w, h;
struct weston_mode *mode;
while(startAt && *startAt) {
nextPos = strchr(startAt, 'x');
if(!nextPos)
return -1;
w = strtoul(startAt, NULL, 0);
startAt = nextPos + 1;
if(!*startAt)
return -1;
h = strtoul(startAt, NULL, 0);
if(!w || (w > 3000) || !h || (h > 3000))
return -1;
mode = malloc(sizeof *mode);
if(!mode)
return -1;
mode->width = w;
mode->height = h;
mode->refresh = 5;
mode->flags = 0;
wl_list_insert(&output->base.mode_list, &mode->link);
startAt = strchr(startAt, ',');
if(startAt && *startAt == ',')
startAt++;
}
return 0;
}
static int
rdp_compositor_create_output(struct rdp_compositor *c, int width, int height,
const char *extraModes)
{
struct rdp_output *output;
struct wl_event_loop *loop;
struct weston_mode *currentMode, *next;
output = malloc(sizeof *output);
if (output == NULL)
return -1;
memset(output, 0, sizeof *output);
wl_list_init(&output->peers);
wl_list_init(&output->base.mode_list);
currentMode = malloc(sizeof *currentMode);
if(!currentMode)
goto out_free_output;
currentMode->flags = WL_OUTPUT_MODE_CURRENT | WL_OUTPUT_MODE_PREFERRED;
currentMode->width = width;
currentMode->height = height;
currentMode->refresh = 5;
wl_list_insert(&output->base.mode_list, &currentMode->link);
if(parse_extra_modes(extraModes, output) < 0) {
weston_log("invalid extra modes\n");
goto out_free_output_and_modes;
}
output->base.current = currentMode;
weston_output_init(&output->base, &c->base, 0, 0, width, height,
WL_OUTPUT_TRANSFORM_NORMAL, 1);
output->base.make = "weston";
output->base.model = "rdp";
output->shadow_surface = pixman_image_create_bits(PIXMAN_x8r8g8b8,
width, height,
NULL,
width * 4);
if (output->shadow_surface == NULL) {
weston_log("Failed to create surface for frame buffer.\n");
goto out_output;
}
if (pixman_renderer_output_create(&output->base) < 0)
goto out_shadow_surface;
weston_output_move(&output->base, 0, 0);
loop = wl_display_get_event_loop(c->base.wl_display);
output->finish_frame_timer = wl_event_loop_add_timer(loop, finish_frame_handler, output);
output->base.origin = output->base.current;
output->base.start_repaint_loop = rdp_output_start_repaint_loop;
output->base.repaint = rdp_output_repaint;
output->base.destroy = rdp_output_destroy;
output->base.assign_planes = NULL;
output->base.set_backlight = NULL;
output->base.set_dpms = NULL;
output->base.switch_mode = rdp_switch_mode;
c->output = output;
wl_list_insert(c->base.output_list.prev, &output->base.link);
return 0;
out_shadow_surface:
pixman_image_unref(output->shadow_surface);
out_output:
weston_output_destroy(&output->base);
out_free_output_and_modes:
wl_list_for_each_safe(currentMode, next, &output->base.mode_list, link)
free(currentMode);
out_free_output:
free(output);
return -1;
}
static void
rdp_restore(struct weston_compositor *ec)
{
}
static void
rdp_destroy(struct weston_compositor *ec)
{
ec->renderer->destroy(ec);
weston_compositor_shutdown(ec);
free(ec);
}
static
int rdp_listener_activity(int fd, uint32_t mask, void *data) {
freerdp_listener* instance = (freerdp_listener *)data;
if (!(mask & WL_EVENT_READABLE))
return 0;
if (!instance->CheckFileDescriptor(instance))
{
weston_log("failed to check FreeRDP file descriptor\n");
return -1;
}
return 0;
}
static
int rdp_implant_listener(struct rdp_compositor *c, freerdp_listener* instance) {
int i, fd;
int rcount = 0;
void* rfds[MAX_FREERDP_FDS];
struct wl_event_loop *loop;
if (!instance->GetFileDescriptor(instance, rfds, &rcount)) {
weston_log("Failed to get FreeRDP file descriptor\n");
return -1;
}
loop = wl_display_get_event_loop(c->base.wl_display);
for (i = 0; i < rcount; i++) {
fd = (int)(long)(rfds[i]);
c->listener_events[i] = wl_event_loop_add_fd(loop, fd, WL_EVENT_READABLE,
rdp_listener_activity, instance);
}
for( ; i < MAX_FREERDP_FDS; i++)
c->listener_events[i] = 0;
return 0;
}
static void
rdp_peer_context_new(freerdp_peer* client, RdpPeerContext* context)
{
context->item.peer = client;
context->item.flags = 0;
context->rfx_context = rfx_context_new();
context->rfx_context->mode = RLGR3;
context->rfx_context->width = client->settings->DesktopWidth;
context->rfx_context->height = client->settings->DesktopHeight;
rfx_context_set_pixel_format(context->rfx_context, RDP_PIXEL_FORMAT_B8G8R8A8);
context->nsc_context = nsc_context_new();
nsc_context_set_pixel_format(context->nsc_context, RDP_PIXEL_FORMAT_B8G8R8A8);
context->encode_stream = Stream_New(NULL, 65536);
}
static void
rdp_peer_context_free(freerdp_peer* client, RdpPeerContext* context)
{
int i;
if(!context)
return;
wl_list_remove(&context->item.link);
for(i = 0; i < MAX_FREERDP_FDS; i++) {
if (context->events[i])
wl_event_source_remove(context->events[i]);
}
if(context->item.flags & RDP_PEER_ACTIVATED)
weston_seat_release(&context->item.seat);
Stream_Free(context->encode_stream, TRUE);
nsc_context_free(context->nsc_context);
rfx_context_free(context->rfx_context);
free(context->rfx_rects);
}
static int
rdp_client_activity(int fd, uint32_t mask, void *data) {
freerdp_peer* client = (freerdp_peer *)data;
if (!client->CheckFileDescriptor(client)) {
weston_log("unable to checkDescriptor for %p\n", client);
goto out_clean;
}
return 0;
out_clean:
freerdp_peer_context_free(client);
freerdp_peer_free(client);
return 0;
}
static BOOL
xf_peer_capabilities(freerdp_peer* client)
{
return TRUE;
}
struct rdp_to_xkb_keyboard_layout {
UINT32 rdpLayoutCode;
char *xkbLayout;
};
/* picked from http://technet.microsoft.com/en-us/library/cc766503(WS.10).aspx */
static struct rdp_to_xkb_keyboard_layout rdp_keyboards[] = {
{0x00000406, "dk"},
{0x00000407, "de"},
{0x00000409, "us"},
{0x0000040c, "fr"},
{0x00000410, "it"},
{0x00000813, "be"},
{0x00000000, 0},
};
/* taken from 2.2.7.1.6 Input Capability Set (TS_INPUT_CAPABILITYSET) */
static char *rdp_keyboard_types[] = {
"", /* 0: unused */
"", /* 1: IBM PC/XT or compatible (83-key) keyboard */
"", /* 2: Olivetti "ICO" (102-key) keyboard */
"", /* 3: IBM PC/AT (84-key) or similar keyboard */
"pc102",/* 4: IBM enhanced (101- or 102-key) keyboard */
"", /* 5: Nokia 1050 and similar keyboards */
"", /* 6: Nokia 9140 and similar keyboards */
"" /* 7: Japanese keyboard */
};
static BOOL
xf_peer_post_connect(freerdp_peer* client)
{
RdpPeerContext *peerCtx;
struct rdp_compositor *c;
struct rdp_output *output;
rdpSettings *settings;
rdpPointerUpdate *pointer;
struct xkb_context *xkbContext;
struct xkb_rule_names xkbRuleNames;
struct xkb_keymap *keymap;
int i;
pixman_box32_t box;
pixman_region32_t damage;
peerCtx = (RdpPeerContext *)client->context;
c = peerCtx->rdpCompositor;
output = c->output;
settings = client->settings;
if(!settings->SurfaceCommandsEnabled) {
weston_log("client doesn't support required SurfaceCommands\n");
return FALSE;
}
if(output->base.width != (int)settings->DesktopWidth ||
output->base.height != (int)settings->DesktopHeight)
{
if(!settings->DesktopResize) {
weston_log("client don't support desktopResize()\n");
return FALSE;
}
/* force the client size */
settings->DesktopWidth = output->base.width;
settings->DesktopHeight = output->base.height;
client->update->DesktopResize(client->context);
}
weston_log("kbd_layout:%x kbd_type:%x kbd_subType:%x kbd_functionKeys:%x\n",
settings->KeyboardLayout, settings->KeyboardType, settings->KeyboardSubType,
settings->KeyboardFunctionKey);
memset(&xkbRuleNames, 0, sizeof(xkbRuleNames));
if(settings->KeyboardType <= 7)
xkbRuleNames.model = rdp_keyboard_types[settings->KeyboardType];
for(i = 0; rdp_keyboards[i].xkbLayout; i++) {
if(rdp_keyboards[i].rdpLayoutCode == settings->KeyboardLayout) {
xkbRuleNames.layout = rdp_keyboards[i].xkbLayout;
break;
}
}
keymap = NULL;
if(xkbRuleNames.layout) {
xkbContext = xkb_context_new(0);
if(!xkbContext) {
weston_log("unable to create a xkb_context\n");
return FALSE;
}
keymap = xkb_keymap_new_from_names(xkbContext, &xkbRuleNames, 0);
}
weston_seat_init_keyboard(&peerCtx->item.seat, keymap);
weston_seat_init_pointer(&peerCtx->item.seat);
peerCtx->item.flags |= RDP_PEER_ACTIVATED;
/* disable pointer on the client side */
pointer = client->update->pointer;
pointer->pointer_system.type = SYSPTR_NULL;
pointer->PointerSystem(client->context, &pointer->pointer_system);
/* sends a full refresh */
box.x1 = 0;
box.y1 = 0;
box.x2 = output->base.width;
box.y2 = output->base.height;
pixman_region32_init_with_extents(&damage, &box);
rdp_peer_refresh_region(&damage, client);
pixman_region32_fini(&damage);
return TRUE;
}
static BOOL
xf_peer_activate(freerdp_peer *client)
{
RdpPeerContext *context = (RdpPeerContext *)client->context;
rfx_context_reset(context->rfx_context);
return TRUE;
}
static void
xf_mouseEvent(rdpInput *input, UINT16 flags, UINT16 x, UINT16 y) {
wl_fixed_t wl_x, wl_y, axis;
RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
struct rdp_output *output;
uint32_t button = 0;
if (flags & PTR_FLAGS_MOVE) {
output = peerContext->rdpCompositor->output;
if(x < output->base.width && y < output->base.height) {
wl_x = wl_fixed_from_int((int)x);
wl_y = wl_fixed_from_int((int)y);
notify_motion_absolute(&peerContext->item.seat, weston_compositor_get_time(),
wl_x, wl_y);
}
}
if (flags & PTR_FLAGS_BUTTON1)
button = BTN_LEFT;
else if (flags & PTR_FLAGS_BUTTON2)
button = BTN_RIGHT;
else if (flags & PTR_FLAGS_BUTTON3)
button = BTN_MIDDLE;
if(button) {
notify_button(&peerContext->item.seat, weston_compositor_get_time(), button,
(flags & PTR_FLAGS_DOWN) ? WL_POINTER_BUTTON_STATE_PRESSED : WL_POINTER_BUTTON_STATE_RELEASED
);
}
if (flags & PTR_FLAGS_WHEEL) {
/* DEFAULT_AXIS_STEP_DISTANCE is stolen from compositor-x11.c
* The RDP specs says the lower bits of flags contains the "the number of rotation
* units the mouse wheel was rotated".
*
* http://blogs.msdn.com/b/oldnewthing/archive/2013/01/23/10387366.aspx explains the 120 value
*/
axis = (DEFAULT_AXIS_STEP_DISTANCE * (flags & 0xff)) / 120;
if (flags & PTR_FLAGS_WHEEL_NEGATIVE)
axis = -axis;
notify_axis(&peerContext->item.seat, weston_compositor_get_time(),
WL_POINTER_AXIS_VERTICAL_SCROLL,
axis);
}
}
static void
xf_extendedMouseEvent(rdpInput *input, UINT16 flags, UINT16 x, UINT16 y) {
wl_fixed_t wl_x, wl_y;
RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
struct rdp_output *output;
output = peerContext->rdpCompositor->output;
if(x < output->base.width && y < output->base.height) {
wl_x = wl_fixed_from_int((int)x);
wl_y = wl_fixed_from_int((int)y);
notify_motion_absolute(&peerContext->item.seat, weston_compositor_get_time(),
wl_x, wl_y);
}
}
static void
xf_input_synchronize_event(rdpInput *input, UINT32 flags)
{
freerdp_peer *client = input->context->peer;
RdpPeerContext *peerCtx = (RdpPeerContext *)input->context;
struct rdp_output *output = peerCtx->rdpCompositor->output;
pixman_box32_t box;
pixman_region32_t damage;
/* sends a full refresh */
box.x1 = 0;
box.y1 = 0;
box.x2 = output->base.width;
box.y2 = output->base.height;
pixman_region32_init_with_extents(&damage, &box);
rdp_peer_refresh_region(&damage, client);
pixman_region32_fini(&damage);
}
extern DWORD KEYCODE_TO_VKCODE_EVDEV[];
static uint32_t vk_to_keycode[256];
static void
init_vk_translator(void)
{
int i;
memset(vk_to_keycode, 0, sizeof(vk_to_keycode));
for(i = 0; i < 256; i++)
vk_to_keycode[KEYCODE_TO_VKCODE_EVDEV[i] & 0xff] = i-8;
}
static void
xf_input_keyboard_event(rdpInput *input, UINT16 flags, UINT16 code)
{
uint32_t scan_code, vk_code, full_code;
enum wl_keyboard_key_state keyState;
RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
int notify = 0;
if (flags & KBD_FLAGS_DOWN) {
keyState = WL_KEYBOARD_KEY_STATE_PRESSED;
notify = 1;
} else if (flags & KBD_FLAGS_RELEASE) {
keyState = WL_KEYBOARD_KEY_STATE_RELEASED;
notify = 1;
}
if(notify) {
full_code = code;
if(flags & KBD_FLAGS_EXTENDED)
full_code |= KBD_FLAGS_EXTENDED;
vk_code = GetVirtualKeyCodeFromVirtualScanCode(full_code, 4);
if(vk_code > 0xff) {
weston_log("invalid vk_code %x", vk_code);
return;
}
scan_code = vk_to_keycode[vk_code];
/*weston_log("code=%x ext=%d vk_code=%x scan_code=%x\n", code, (flags & KBD_FLAGS_EXTENDED) ? 1 : 0,
vk_code, scan_code);*/
notify_key(&peerContext->item.seat, weston_compositor_get_time(),
scan_code, keyState, STATE_UPDATE_AUTOMATIC);
}
}
static void
xf_input_unicode_keyboard_event(rdpInput *input, UINT16 flags, UINT16 code)
{
weston_log("Client sent a unicode keyboard event (flags:0x%X code:0x%X)\n", flags, code);
}
static void
xf_suppress_output(rdpContext *context, BYTE allow, RECTANGLE_16 *area) {
RdpPeerContext *peerContext = (RdpPeerContext *)context;
if(allow)
peerContext->item.flags |= RDP_PEER_OUTPUT_ENABLED;
else
peerContext->item.flags &= (~RDP_PEER_OUTPUT_ENABLED);
}
static int
rdp_peer_init(freerdp_peer *client, struct rdp_compositor *c)
{
int rcount = 0;
void *rfds[MAX_FREERDP_FDS];
int i, fd;
struct wl_event_loop *loop;
rdpSettings *settings;
rdpInput *input;
RdpPeerContext *peerCtx;
char seat_name[32];
client->context_size = sizeof(RdpPeerContext);
client->ContextNew = (psPeerContextNew)rdp_peer_context_new;
client->ContextFree = (psPeerContextFree)rdp_peer_context_free;
freerdp_peer_context_new(client);
peerCtx = (RdpPeerContext *) client->context;
peerCtx->rdpCompositor = c;
settings = client->settings;
settings->RdpKeyFile = c->rdp_key;
if(c->tls_enabled) {
settings->CertificateFile = c->server_cert;
settings->PrivateKeyFile = c->server_key;
} else {
settings->TlsSecurity = FALSE;
}
settings->NlaSecurity = FALSE;
client->Capabilities = xf_peer_capabilities;
client->PostConnect = xf_peer_post_connect;
client->Activate = xf_peer_activate;
client->update->SuppressOutput = xf_suppress_output;
input = client->input;
input->SynchronizeEvent = xf_input_synchronize_event;
input->MouseEvent = xf_mouseEvent;
input->ExtendedMouseEvent = xf_extendedMouseEvent;
input->KeyboardEvent = xf_input_keyboard_event;
input->UnicodeKeyboardEvent = xf_input_unicode_keyboard_event;
if (snprintf(seat_name, 32, "rdp:%d:%s", client->sockfd, client->hostname) >= 32)
seat_name[31] = '\0';
weston_seat_init(&peerCtx->item.seat, &c->base, seat_name);
client->Initialize(client);
if (!client->GetFileDescriptor(client, rfds, &rcount)) {
weston_log("unable to retrieve client fds\n");
return -1;
}
loop = wl_display_get_event_loop(c->base.wl_display);
for(i = 0; i < rcount; i++) {
fd = (int)(long)(rfds[i]);
peerCtx->events[i] = wl_event_loop_add_fd(loop, fd, WL_EVENT_READABLE,
rdp_client_activity, client);
}
for ( ; i < MAX_FREERDP_FDS; i++)
peerCtx->events[i] = 0;
wl_list_insert(&c->output->peers, &peerCtx->item.link);
return 0;
}
static void
rdp_incoming_peer(freerdp_listener *instance, freerdp_peer *client)
{
struct rdp_compositor *c = (struct rdp_compositor *)instance->param4;
if (rdp_peer_init(client, c) < 0)
return;
}
static struct weston_compositor *
rdp_compositor_create(struct wl_display *display,
struct rdp_compositor_config *config,
int *argc, char *argv[], struct weston_config *wconfig)
{
struct rdp_compositor *c;
char *fd_str;
int fd;
c = malloc(sizeof *c);
if (c == NULL)
return NULL;
memset(c, 0, sizeof *c);
if (weston_compositor_init(&c->base, display, argc, argv, wconfig) < 0)
goto err_free;
c->base.destroy = rdp_destroy;
c->base.restore = rdp_restore;
c->rdp_key = config->rdp_key ? strdup(config->rdp_key) : NULL;
/* activate TLS only if certificate/key are available */
if(config->server_cert && config->server_key) {
weston_log("TLS support activated\n");
c->server_cert = strdup(config->server_cert);
c->server_key = strdup(config->server_key);
if(!c->server_cert || !c->server_key)
goto err_free_strings;
c->tls_enabled = 1;
}
if (pixman_renderer_init(&c->base) < 0)
goto err_compositor;
if (rdp_compositor_create_output(c, config->width, config->height, config->extra_modes) < 0)
goto err_compositor;
if(!config->env_socket) {
c->listener = freerdp_listener_new();
c->listener->PeerAccepted = rdp_incoming_peer;
c->listener->param4 = c;
if(!c->listener->Open(c->listener, config->bind_address, config->port)) {
weston_log("unable to bind rdp socket\n");
goto err_listener;
}
if (rdp_implant_listener(c, c->listener) < 0)
goto err_compositor;
} else {
/* get the socket from RDP_FD var */
fd_str = getenv("RDP_FD");
if(!fd_str) {
weston_log("RDP_FD env variable not set");
goto err_output;
}
fd = strtoul(fd_str, NULL, 10);
if(rdp_peer_init(freerdp_peer_new(fd), c))
goto err_output;
}
return &c->base;
err_listener:
freerdp_listener_free(c->listener);
err_output:
weston_output_destroy(&c->output->base);
err_compositor:
weston_compositor_shutdown(&c->base);
err_free_strings:
if(c->rdp_key)
free(c->rdp_key);
if(c->server_cert)
free(c->server_cert);
if(c->server_key)
free(c->server_key);
err_free:
free(c);
return NULL;
}
WL_EXPORT struct weston_compositor *
backend_init(struct wl_display *display, int *argc, char *argv[],
struct weston_config *wconfig)
{
struct rdp_compositor_config config;
rdp_compositor_config_init(&config);
int major, minor, revision;
freerdp_get_version(&major, &minor, &revision);
weston_log("using FreeRDP version %d.%d.%d\n", major, minor, revision);
init_vk_translator();
const struct weston_option rdp_options[] = {
{ WESTON_OPTION_BOOLEAN, "env-socket", 0, &config.env_socket },
{ WESTON_OPTION_INTEGER, "width", 0, &config.width },
{ WESTON_OPTION_INTEGER, "height", 0, &config.height },
{ WESTON_OPTION_STRING, "extra-modes", 0, &config.extra_modes },
{ WESTON_OPTION_STRING, "address", 0, &config.bind_address },
{ WESTON_OPTION_INTEGER, "port", 0, &config.port },
{ WESTON_OPTION_STRING, "rdp4-key", 0, &config.rdp_key },
{ WESTON_OPTION_STRING, "rdp-tls-cert", 0, &config.server_cert },
{ WESTON_OPTION_STRING, "rdp-tls-key", 0, &config.server_key }
};
parse_options(rdp_options, ARRAY_LENGTH(rdp_options), argc, argv);
return rdp_compositor_create(display, &config, argc, argv, wconfig);
}