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.
1436 lines
39 KiB
1436 lines
39 KiB
/*
|
|
* Copyright © 2013 Hardening <rdp.effort@gmail.com>
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
* a copy of this software and associated documentation files (the
|
|
* "Software"), to deal in the Software without restriction, including
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
* distribute, sublicense, and/or sell copies of the Software, and to
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
* the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice (including the
|
|
* next paragraph) shall be included in all copies or substantial
|
|
* portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
|
|
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
|
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <assert.h>
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <linux/input.h>
|
|
|
|
#include <freerdp/version.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 <freerdp/locale/keyboard.h>
|
|
#include <winpr/input.h>
|
|
#include <winpr/ssl.h>
|
|
|
|
#include "shared/helpers.h"
|
|
#include "shared/timespec-util.h"
|
|
#include <libweston/libweston.h>
|
|
#include <libweston/backend-rdp.h>
|
|
#include "pixman-renderer.h"
|
|
|
|
#define MAX_FREERDP_FDS 32
|
|
#define DEFAULT_AXIS_STEP_DISTANCE 10
|
|
#define RDP_MODE_FREQ 60 * 1000
|
|
#define DEFAULT_PIXEL_FORMAT PIXEL_FORMAT_BGRA32
|
|
|
|
struct rdp_output;
|
|
|
|
struct rdp_backend {
|
|
struct weston_backend base;
|
|
struct weston_compositor *compositor;
|
|
|
|
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;
|
|
int no_clients_resize;
|
|
int force_no_compression;
|
|
};
|
|
|
|
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_head {
|
|
struct weston_head base;
|
|
};
|
|
|
|
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_backend *rdpBackend;
|
|
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 inline struct rdp_head *
|
|
to_rdp_head(struct weston_head *base)
|
|
{
|
|
return container_of(base, struct rdp_head, base);
|
|
}
|
|
|
|
static inline struct rdp_output *
|
|
to_rdp_output(struct weston_output *base)
|
|
{
|
|
return container_of(base, struct rdp_output, base);
|
|
}
|
|
|
|
static inline struct rdp_backend *
|
|
to_rdp_backend(struct weston_compositor *base)
|
|
{
|
|
return container_of(base->backend, struct rdp_backend, base);
|
|
}
|
|
|
|
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 = { 0 };
|
|
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.skipCompression = TRUE;
|
|
cmd.cmdType = CMDTYPE_STREAM_SURFACE_BITS;
|
|
cmd.destLeft = damage->extents.x1;
|
|
cmd.destTop = damage->extents.y1;
|
|
cmd.destRight = damage->extents.x2;
|
|
cmd.destBottom = damage->extents.y2;
|
|
cmd.bmp.bpp = 32;
|
|
cmd.bmp.codecID = peer->settings->RemoteFxCodecId;
|
|
cmd.bmp.width = width;
|
|
cmd.bmp.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.bmp.bitmapDataLength = Stream_GetPosition(context->encode_stream);
|
|
cmd.bmp.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 = { 0 };
|
|
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.cmdType = CMDTYPE_SET_SURFACE_BITS;
|
|
cmd.skipCompression = TRUE;
|
|
cmd.destLeft = damage->extents.x1;
|
|
cmd.destTop = damage->extents.y1;
|
|
cmd.destRight = damage->extents.x2;
|
|
cmd.destBottom = damage->extents.y2;
|
|
cmd.bmp.bpp = 32;
|
|
cmd.bmp.codecID = peer->settings->NSCodecId;
|
|
cmd.bmp.width = width;
|
|
cmd.bmp.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,
|
|
width, height,
|
|
pixman_image_get_stride(image));
|
|
|
|
cmd.bmp.bitmapDataLength = Stream_GetPosition(context->encode_stream);
|
|
cmd.bmp.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 = { 0 };
|
|
SURFACE_FRAME_MARKER 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.cmdType = CMDTYPE_SET_SURFACE_BITS;
|
|
cmd.bmp.bpp = 32;
|
|
cmd.bmp.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.bmp.width = (rect->x2 - rect->x1);
|
|
|
|
heightIncrement = peer->settings->MultifragMaxRequestSize / (16 + cmd.bmp.width * 4);
|
|
remainingHeight = rect->y2 - rect->y1;
|
|
top = rect->y1;
|
|
|
|
subrect.x1 = rect->x1;
|
|
subrect.x2 = rect->x2;
|
|
|
|
while (remainingHeight) {
|
|
cmd.bmp.height = (remainingHeight > heightIncrement) ? heightIncrement : remainingHeight;
|
|
cmd.destTop = top;
|
|
cmd.destBottom = top + cmd.bmp.height;
|
|
cmd.bmp.bitmapDataLength = cmd.bmp.width * cmd.bmp.height * 4;
|
|
cmd.bmp.bitmapData = (BYTE *)realloc(cmd.bmp.bitmapData, cmd.bmp.bitmapDataLength);
|
|
|
|
subrect.y1 = top;
|
|
subrect.y2 = top + cmd.bmp.height;
|
|
pixman_image_flipped_subrect(&subrect, image, cmd.bmp.bitmapData);
|
|
|
|
/*weston_log("* sending (%d,%d, %d,%d)\n", subrect.x1, subrect.y1, subrect.x2, subrect.y2); */
|
|
update->SurfaceBits(peer->context, &cmd);
|
|
|
|
remainingHeight -= cmd.bmp.height;
|
|
top += cmd.bmp.height;
|
|
}
|
|
}
|
|
|
|
free(cmd.bmp.bitmapData);
|
|
|
|
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->rdpBackend->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 int
|
|
rdp_output_start_repaint_loop(struct weston_output *output)
|
|
{
|
|
struct timespec ts;
|
|
|
|
weston_compositor_read_presentation_clock(output->compositor, &ts);
|
|
weston_output_finish_frame(output, &ts, WP_PRESENTATION_FEEDBACK_INVALID);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
rdp_output_repaint(struct weston_output *output_base, pixman_region32_t *damage,
|
|
void *repaint_data)
|
|
{
|
|
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);
|
|
|
|
if (pixman_region32_not_empty(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);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
finish_frame_handler(void *data)
|
|
{
|
|
struct rdp_output *output = data;
|
|
struct timespec ts;
|
|
|
|
weston_compositor_read_presentation_clock(output->base.compositor, &ts);
|
|
weston_output_finish_frame(&output->base, &ts, 0);
|
|
|
|
return 1;
|
|
}
|
|
|
|
static struct weston_mode *
|
|
rdp_insert_new_mode(struct weston_output *output, int width, int height, int rate)
|
|
{
|
|
struct weston_mode *ret;
|
|
ret = zalloc(sizeof *ret);
|
|
if (!ret)
|
|
return NULL;
|
|
ret->width = width;
|
|
ret->height = height;
|
|
ret->refresh = rate;
|
|
wl_list_insert(&output->mode_list, &ret->link);
|
|
return ret;
|
|
}
|
|
|
|
static struct weston_mode *
|
|
ensure_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 rdp_insert_new_mode(output, target->width, target->height, RDP_MODE_FREQ);
|
|
}
|
|
|
|
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;
|
|
const struct pixman_renderer_output_options options = { .use_shadow = true, };
|
|
|
|
local_mode = ensure_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_mode)
|
|
return 0;
|
|
|
|
output->current_mode->flags &= ~WL_OUTPUT_MODE_CURRENT;
|
|
|
|
output->current_mode = local_mode;
|
|
output->current_mode->flags |= WL_OUTPUT_MODE_CURRENT;
|
|
|
|
pixman_renderer_output_destroy(output);
|
|
pixman_renderer_output_create(output, &options);
|
|
|
|
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->DesktopWidth == (UINT32)target_mode->width &&
|
|
settings->DesktopHeight == (UINT32)target_mode->height)
|
|
continue;
|
|
|
|
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
|
|
rdp_output_set_size(struct weston_output *base,
|
|
int width, int height)
|
|
{
|
|
struct rdp_output *output = to_rdp_output(base);
|
|
struct weston_head *head;
|
|
struct weston_mode *currentMode;
|
|
struct weston_mode initMode;
|
|
|
|
/* We can only be called once. */
|
|
assert(!output->base.current_mode);
|
|
|
|
wl_list_for_each(head, &output->base.head_list, output_link) {
|
|
weston_head_set_monitor_strings(head, "weston", "rdp", NULL);
|
|
|
|
/* This is a virtual output, so report a zero physical size.
|
|
* It's better to let frontends/clients use their defaults. */
|
|
weston_head_set_physical_size(head, 0, 0);
|
|
}
|
|
|
|
wl_list_init(&output->peers);
|
|
|
|
initMode.flags = WL_OUTPUT_MODE_CURRENT | WL_OUTPUT_MODE_PREFERRED;
|
|
initMode.width = width;
|
|
initMode.height = height;
|
|
initMode.refresh = RDP_MODE_FREQ;
|
|
|
|
currentMode = ensure_matching_mode(&output->base, &initMode);
|
|
if (!currentMode)
|
|
return -1;
|
|
|
|
output->base.current_mode = output->base.native_mode = currentMode;
|
|
|
|
output->base.start_repaint_loop = rdp_output_start_repaint_loop;
|
|
output->base.repaint = rdp_output_repaint;
|
|
output->base.assign_planes = NULL;
|
|
output->base.set_backlight = NULL;
|
|
output->base.set_dpms = NULL;
|
|
output->base.switch_mode = rdp_switch_mode;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
rdp_output_enable(struct weston_output *base)
|
|
{
|
|
struct rdp_output *output = to_rdp_output(base);
|
|
struct rdp_backend *b = to_rdp_backend(base->compositor);
|
|
struct wl_event_loop *loop;
|
|
const struct pixman_renderer_output_options options = {
|
|
.use_shadow = true,
|
|
};
|
|
|
|
output->shadow_surface = pixman_image_create_bits(PIXMAN_x8r8g8b8,
|
|
output->base.current_mode->width,
|
|
output->base.current_mode->height,
|
|
NULL,
|
|
output->base.current_mode->width * 4);
|
|
if (output->shadow_surface == NULL) {
|
|
weston_log("Failed to create surface for frame buffer.\n");
|
|
return -1;
|
|
}
|
|
|
|
if (pixman_renderer_output_create(&output->base, &options) < 0) {
|
|
pixman_image_unref(output->shadow_surface);
|
|
return -1;
|
|
}
|
|
|
|
loop = wl_display_get_event_loop(b->compositor->wl_display);
|
|
output->finish_frame_timer = wl_event_loop_add_timer(loop, finish_frame_handler, output);
|
|
|
|
b->output = output;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
rdp_output_disable(struct weston_output *base)
|
|
{
|
|
struct rdp_output *output = to_rdp_output(base);
|
|
struct rdp_backend *b = to_rdp_backend(base->compositor);
|
|
|
|
if (!output->base.enabled)
|
|
return 0;
|
|
|
|
pixman_image_unref(output->shadow_surface);
|
|
pixman_renderer_output_destroy(&output->base);
|
|
|
|
wl_event_source_remove(output->finish_frame_timer);
|
|
b->output = NULL;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
rdp_output_destroy(struct weston_output *base)
|
|
{
|
|
struct rdp_output *output = to_rdp_output(base);
|
|
|
|
rdp_output_disable(&output->base);
|
|
weston_output_release(&output->base);
|
|
|
|
free(output);
|
|
}
|
|
|
|
static struct weston_output *
|
|
rdp_output_create(struct weston_compositor *compositor, const char *name)
|
|
{
|
|
struct rdp_output *output;
|
|
|
|
output = zalloc(sizeof *output);
|
|
if (output == NULL)
|
|
return NULL;
|
|
|
|
weston_output_init(&output->base, compositor, name);
|
|
|
|
output->base.destroy = rdp_output_destroy;
|
|
output->base.disable = rdp_output_disable;
|
|
output->base.enable = rdp_output_enable;
|
|
output->base.attach_head = NULL;
|
|
|
|
weston_compositor_add_pending_output(&output->base, compositor);
|
|
|
|
return &output->base;
|
|
}
|
|
|
|
static int
|
|
rdp_head_create(struct weston_compositor *compositor, const char *name)
|
|
{
|
|
struct rdp_head *head;
|
|
|
|
head = zalloc(sizeof *head);
|
|
if (!head)
|
|
return -1;
|
|
|
|
weston_head_init(&head->base, name);
|
|
weston_head_set_connection_status(&head->base, true);
|
|
weston_compositor_add_head(compositor, &head->base);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
rdp_head_destroy(struct rdp_head *head)
|
|
{
|
|
weston_head_release(&head->base);
|
|
free(head);
|
|
}
|
|
|
|
static void
|
|
rdp_destroy(struct weston_compositor *ec)
|
|
{
|
|
struct rdp_backend *b = to_rdp_backend(ec);
|
|
struct weston_head *base, *next;
|
|
struct rdp_peers_item *rdp_peer, *tmp;
|
|
int i;
|
|
|
|
wl_list_for_each_safe(rdp_peer, tmp, &b->output->peers, link) {
|
|
freerdp_peer* client = rdp_peer->peer;
|
|
|
|
client->Disconnect(client);
|
|
freerdp_peer_context_free(client);
|
|
freerdp_peer_free(client);
|
|
}
|
|
|
|
for (i = 0; i < MAX_FREERDP_FDS; i++)
|
|
if (b->listener_events[i])
|
|
wl_event_source_remove(b->listener_events[i]);
|
|
|
|
weston_compositor_shutdown(ec);
|
|
|
|
wl_list_for_each_safe(base, next, &ec->head_list, compositor_link)
|
|
rdp_head_destroy(to_rdp_head(base));
|
|
|
|
freerdp_listener_free(b->listener);
|
|
|
|
free(b->server_cert);
|
|
free(b->server_key);
|
|
free(b->rdp_key);
|
|
free(b);
|
|
}
|
|
|
|
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_backend *b, 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(b->compositor->wl_display);
|
|
for (i = 0; i < rcount; i++) {
|
|
fd = (int)(long)(rfds[i]);
|
|
b->listener_events[i] = wl_event_loop_add_fd(loop, fd, WL_EVENT_READABLE,
|
|
rdp_listener_activity, instance);
|
|
}
|
|
|
|
for ( ; i < MAX_FREERDP_FDS; i++)
|
|
b->listener_events[i] = 0;
|
|
return 0;
|
|
}
|
|
|
|
|
|
static BOOL
|
|
rdp_peer_context_new(freerdp_peer* client, RdpPeerContext* context)
|
|
{
|
|
context->item.peer = client;
|
|
context->item.flags = RDP_PEER_OUTPUT_ENABLED;
|
|
|
|
context->rfx_context = rfx_context_new(TRUE);
|
|
if (!context->rfx_context)
|
|
return FALSE;
|
|
|
|
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, DEFAULT_PIXEL_FORMAT);
|
|
|
|
context->nsc_context = nsc_context_new();
|
|
if (!context->nsc_context)
|
|
goto out_error_nsc;
|
|
|
|
nsc_context_set_parameters(context->nsc_context, NSC_COLOR_FORMAT, DEFAULT_PIXEL_FORMAT);
|
|
context->encode_stream = Stream_New(NULL, 65536);
|
|
if (!context->encode_stream)
|
|
goto out_error_stream;
|
|
|
|
return TRUE;
|
|
|
|
out_error_nsc:
|
|
rfx_context_free(context->rfx_context);
|
|
out_error_stream:
|
|
nsc_context_free(context->nsc_context);
|
|
return FALSE;
|
|
}
|
|
|
|
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_keyboard(context->item.seat);
|
|
weston_seat_release_pointer(context->item.seat);
|
|
weston_seat_release(context->item.seat);
|
|
free(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;
|
|
const char *xkbLayout;
|
|
const char *xkbVariant;
|
|
};
|
|
|
|
/* table reversed from
|
|
https://github.com/awakecoding/FreeRDP/blob/master/libfreerdp/locale/xkb_layout_ids.c#L811 */
|
|
static const
|
|
struct rdp_to_xkb_keyboard_layout rdp_keyboards[] = {
|
|
{KBD_ARABIC_101, "ara", 0},
|
|
{KBD_BULGARIAN, 0, 0},
|
|
{KBD_CHINESE_TRADITIONAL_US, 0},
|
|
{KBD_CZECH, "cz", 0},
|
|
{KBD_CZECH_PROGRAMMERS, "cz", "bksl"},
|
|
{KBD_CZECH_QWERTY, "cz", "qwerty"},
|
|
{KBD_DANISH, "dk", 0},
|
|
{KBD_GERMAN, "de", 0},
|
|
{KBD_GERMAN_NEO, "de", "neo"},
|
|
{KBD_GERMAN_IBM, "de", "qwerty"},
|
|
{KBD_GREEK, "gr", 0},
|
|
{KBD_GREEK_220, "gr", "simple"},
|
|
{KBD_GREEK_319, "gr", "extended"},
|
|
{KBD_GREEK_POLYTONIC, "gr", "polytonic"},
|
|
{KBD_US, "us", 0},
|
|
{KBD_US_ENGLISH_TABLE_FOR_IBM_ARABIC_238_L, "ara", "buckwalter"},
|
|
{KBD_SPANISH, "es", 0},
|
|
{KBD_SPANISH_VARIATION, "es", "nodeadkeys"},
|
|
{KBD_FINNISH, "fi", 0},
|
|
{KBD_FRENCH, "fr", 0},
|
|
{KBD_HEBREW, "il", 0},
|
|
{KBD_HUNGARIAN, "hu", 0},
|
|
{KBD_HUNGARIAN_101_KEY, "hu", "standard"},
|
|
{KBD_ICELANDIC, "is", 0},
|
|
{KBD_ITALIAN, "it", 0},
|
|
{KBD_ITALIAN_142, "it", "nodeadkeys"},
|
|
{KBD_JAPANESE, "jp", 0},
|
|
{KBD_JAPANESE_INPUT_SYSTEM_MS_IME2002, "jp", "kana"},
|
|
{KBD_KOREAN, "kr", 0},
|
|
{KBD_KOREAN_INPUT_SYSTEM_IME_2000, "kr", "kr104"},
|
|
{KBD_DUTCH, "nl", 0},
|
|
{KBD_NORWEGIAN, "no", 0},
|
|
{KBD_POLISH_PROGRAMMERS, "pl", 0},
|
|
{KBD_POLISH_214, "pl", "qwertz"},
|
|
{KBD_ROMANIAN, "ro", 0},
|
|
{KBD_RUSSIAN, "ru", 0},
|
|
{KBD_RUSSIAN_TYPEWRITER, "ru", "typewriter"},
|
|
{KBD_CROATIAN, "hr", 0},
|
|
{KBD_SLOVAK, "sk", 0},
|
|
{KBD_SLOVAK_QWERTY, "sk", "qwerty"},
|
|
{KBD_ALBANIAN, 0, 0},
|
|
{KBD_SWEDISH, "se", 0},
|
|
{KBD_THAI_KEDMANEE, "th", 0},
|
|
{KBD_THAI_KEDMANEE_NON_SHIFTLOCK, "th", "tis"},
|
|
{KBD_TURKISH_Q, "tr", 0},
|
|
{KBD_TURKISH_F, "tr", "f"},
|
|
{KBD_URDU, "in", "urd-phonetic3"},
|
|
{KBD_UKRAINIAN, "ua", 0},
|
|
{KBD_BELARUSIAN, "by", 0},
|
|
{KBD_SLOVENIAN, "si", 0},
|
|
{KBD_ESTONIAN, "ee", 0},
|
|
{KBD_LATVIAN, "lv", 0},
|
|
{KBD_LITHUANIAN_IBM, "lt", "ibm"},
|
|
{KBD_FARSI, "af", 0},
|
|
{KBD_VIETNAMESE, "vn", 0},
|
|
{KBD_ARMENIAN_EASTERN, "am", 0},
|
|
{KBD_AZERI_LATIN, 0, 0},
|
|
{KBD_FYRO_MACEDONIAN, "mk", 0},
|
|
{KBD_GEORGIAN, "ge", 0},
|
|
{KBD_FAEROESE, 0, 0},
|
|
{KBD_DEVANAGARI_INSCRIPT, 0, 0},
|
|
{KBD_MALTESE_47_KEY, 0, 0},
|
|
{KBD_NORWEGIAN_WITH_SAMI, "no", "smi"},
|
|
{KBD_KAZAKH, "kz", 0},
|
|
{KBD_KYRGYZ_CYRILLIC, "kg", "phonetic"},
|
|
{KBD_TATAR, "ru", "tt"},
|
|
{KBD_BENGALI, "bd", 0},
|
|
{KBD_BENGALI_INSCRIPT, "bd", "probhat"},
|
|
{KBD_PUNJABI, 0, 0},
|
|
{KBD_GUJARATI, "in", "guj"},
|
|
{KBD_TAMIL, "in", "tam"},
|
|
{KBD_TELUGU, "in", "tel"},
|
|
{KBD_KANNADA, "in", "kan"},
|
|
{KBD_MALAYALAM, "in", "mal"},
|
|
{KBD_HINDI_TRADITIONAL, "in", 0},
|
|
{KBD_MARATHI, 0, 0},
|
|
{KBD_MONGOLIAN_CYRILLIC, "mn", 0},
|
|
{KBD_UNITED_KINGDOM_EXTENDED, "gb", "intl"},
|
|
{KBD_SYRIAC, "syc", 0},
|
|
{KBD_SYRIAC_PHONETIC, "syc", "syc_phonetic"},
|
|
{KBD_NEPALI, "np", 0},
|
|
{KBD_PASHTO, "af", "ps"},
|
|
{KBD_DIVEHI_PHONETIC, 0, 0},
|
|
{KBD_LUXEMBOURGISH, 0, 0},
|
|
{KBD_MAORI, "mao", 0},
|
|
{KBD_CHINESE_SIMPLIFIED_US, 0, 0},
|
|
{KBD_SWISS_GERMAN, "ch", "de_nodeadkeys"},
|
|
{KBD_UNITED_KINGDOM, "gb", 0},
|
|
{KBD_LATIN_AMERICAN, "latam", 0},
|
|
{KBD_BELGIAN_FRENCH, "be", 0},
|
|
{KBD_BELGIAN_PERIOD, "be", "oss_sundeadkeys"},
|
|
{KBD_PORTUGUESE, "pt", 0},
|
|
{KBD_SERBIAN_LATIN, "rs", 0},
|
|
{KBD_AZERI_CYRILLIC, "az", "cyrillic"},
|
|
{KBD_SWEDISH_WITH_SAMI, "se", "smi"},
|
|
{KBD_UZBEK_CYRILLIC, "af", "uz"},
|
|
{KBD_INUKTITUT_LATIN, "ca", "ike"},
|
|
{KBD_CANADIAN_FRENCH_LEGACY, "ca", "fr-legacy"},
|
|
{KBD_SERBIAN_CYRILLIC, "rs", 0},
|
|
{KBD_CANADIAN_FRENCH, "ca", "fr-legacy"},
|
|
{KBD_SWISS_FRENCH, "ch", "fr"},
|
|
{KBD_BOSNIAN, "ba", 0},
|
|
{KBD_IRISH, 0, 0},
|
|
{KBD_BOSNIAN_CYRILLIC, "ba", "us"},
|
|
{KBD_UNITED_STATES_DVORAK, "us", "dvorak"},
|
|
{KBD_PORTUGUESE_BRAZILIAN_ABNT2, "br", "nativo"},
|
|
{KBD_CANADIAN_MULTILINGUAL_STANDARD, "ca", "multix"},
|
|
{KBD_GAELIC, "ie", "CloGaelach"},
|
|
|
|
{0x00000000, 0, 0},
|
|
};
|
|
|
|
/* taken from 2.2.7.1.6 Input Capability Set (TS_INPUT_CAPABILITYSET) */
|
|
static const 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_activate(freerdp_peer* client)
|
|
{
|
|
RdpPeerContext *peerCtx;
|
|
struct rdp_backend *b;
|
|
struct rdp_output *output;
|
|
rdpSettings *settings;
|
|
rdpPointerUpdate *pointer;
|
|
struct rdp_peers_item *peersItem;
|
|
struct xkb_rule_names xkbRuleNames;
|
|
struct xkb_keymap *keymap;
|
|
struct weston_output *weston_output;
|
|
int i;
|
|
pixman_box32_t box;
|
|
pixman_region32_t damage;
|
|
char seat_name[50];
|
|
POINTER_SYSTEM_UPDATE pointer_system;
|
|
|
|
peerCtx = (RdpPeerContext *)client->context;
|
|
b = peerCtx->rdpBackend;
|
|
peersItem = &peerCtx->item;
|
|
output = b->output;
|
|
settings = client->settings;
|
|
|
|
if (!settings->SurfaceCommandsEnabled) {
|
|
weston_log("client doesn't support required SurfaceCommands\n");
|
|
return FALSE;
|
|
}
|
|
|
|
if (b->force_no_compression && settings->CompressionEnabled) {
|
|
weston_log("Forcing compression off\n");
|
|
settings->CompressionEnabled = FALSE;
|
|
}
|
|
|
|
if (output->base.width != (int)settings->DesktopWidth ||
|
|
output->base.height != (int)settings->DesktopHeight)
|
|
{
|
|
if (b->no_clients_resize) {
|
|
/* RDP peers don't dictate their resolution to weston */
|
|
if (!settings->DesktopResize) {
|
|
/* peer does not support desktop resize */
|
|
weston_log("%s: client doesn't support resizing, closing connection\n", __FUNCTION__);
|
|
return FALSE;
|
|
} else {
|
|
settings->DesktopWidth = output->base.width;
|
|
settings->DesktopHeight = output->base.height;
|
|
client->update->DesktopResize(client->context);
|
|
}
|
|
} else {
|
|
/* ask weston to adjust size */
|
|
struct weston_mode new_mode;
|
|
struct weston_mode *target_mode;
|
|
new_mode.width = (int)settings->DesktopWidth;
|
|
new_mode.height = (int)settings->DesktopHeight;
|
|
target_mode = ensure_matching_mode(&output->base, &new_mode);
|
|
if (!target_mode) {
|
|
weston_log("client mode not found\n");
|
|
return FALSE;
|
|
}
|
|
weston_output_mode_set_native(&output->base, target_mode, 1);
|
|
output->base.width = new_mode.width;
|
|
output->base.height = new_mode.height;
|
|
}
|
|
}
|
|
|
|
weston_output = &output->base;
|
|
rfx_context_reset(peerCtx->rfx_context, weston_output->width, weston_output->height);
|
|
nsc_context_reset(peerCtx->nsc_context, weston_output->width, weston_output->height);
|
|
|
|
if (peersItem->flags & RDP_PEER_ACTIVATED)
|
|
return TRUE;
|
|
|
|
/* when here it's the first reactivation, we need to setup a little more */
|
|
weston_log("kbd_layout:0x%x kbd_type:0x%x kbd_subType:0x%x kbd_functionKeys:0x%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].rdpLayoutCode; i++) {
|
|
if (rdp_keyboards[i].rdpLayoutCode == settings->KeyboardLayout) {
|
|
xkbRuleNames.layout = rdp_keyboards[i].xkbLayout;
|
|
xkbRuleNames.variant = rdp_keyboards[i].xkbVariant;
|
|
weston_log("%s: matching layout=%s variant=%s\n", __FUNCTION__,
|
|
xkbRuleNames.layout, xkbRuleNames.variant);
|
|
break;
|
|
}
|
|
}
|
|
|
|
keymap = NULL;
|
|
if (xkbRuleNames.layout) {
|
|
keymap = xkb_keymap_new_from_names(b->compositor->xkb_context,
|
|
&xkbRuleNames, 0);
|
|
}
|
|
|
|
if (settings->ClientHostname)
|
|
snprintf(seat_name, sizeof(seat_name), "RDP %s", settings->ClientHostname);
|
|
else
|
|
snprintf(seat_name, sizeof(seat_name), "RDP peer @%s", settings->ClientAddress);
|
|
|
|
peersItem->seat = zalloc(sizeof(*peersItem->seat));
|
|
if (!peersItem->seat) {
|
|
xkb_keymap_unref(keymap);
|
|
weston_log("unable to create a weston_seat\n");
|
|
return FALSE;
|
|
}
|
|
|
|
weston_seat_init(peersItem->seat, b->compositor, seat_name);
|
|
weston_seat_init_keyboard(peersItem->seat, keymap);
|
|
xkb_keymap_unref(keymap);
|
|
weston_seat_init_pointer(peersItem->seat);
|
|
|
|
peersItem->flags |= RDP_PEER_ACTIVATED;
|
|
|
|
/* disable pointer on the client side */
|
|
pointer = client->update->pointer;
|
|
pointer_system.type = SYSPTR_NULL;
|
|
pointer->PointerSystem(client->context, &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_post_connect(freerdp_peer *client)
|
|
{
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL
|
|
xf_mouseEvent(rdpInput *input, UINT16 flags, UINT16 x, UINT16 y)
|
|
{
|
|
RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
|
|
struct rdp_output *output;
|
|
uint32_t button = 0;
|
|
bool need_frame = false;
|
|
struct timespec time;
|
|
|
|
if (flags & PTR_FLAGS_MOVE) {
|
|
output = peerContext->rdpBackend->output;
|
|
if (x < output->base.width && y < output->base.height) {
|
|
weston_compositor_get_time(&time);
|
|
notify_motion_absolute(peerContext->item.seat, &time,
|
|
x, y);
|
|
need_frame = true;
|
|
}
|
|
}
|
|
|
|
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) {
|
|
weston_compositor_get_time(&time);
|
|
notify_button(peerContext->item.seat, &time, button,
|
|
(flags & PTR_FLAGS_DOWN) ? WL_POINTER_BUTTON_STATE_PRESSED : WL_POINTER_BUTTON_STATE_RELEASED
|
|
);
|
|
need_frame = true;
|
|
}
|
|
|
|
if (flags & PTR_FLAGS_WHEEL) {
|
|
struct weston_pointer_axis_event weston_event;
|
|
double value;
|
|
|
|
/* 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".
|
|
*
|
|
* https://blogs.msdn.microsoft.com/oldnewthing/20130123-00/?p=5473 explains the 120 value
|
|
*/
|
|
value = -(flags & 0xff) / 120.0;
|
|
if (flags & PTR_FLAGS_WHEEL_NEGATIVE)
|
|
value = -value;
|
|
|
|
weston_event.axis = WL_POINTER_AXIS_VERTICAL_SCROLL;
|
|
weston_event.value = DEFAULT_AXIS_STEP_DISTANCE * value;
|
|
weston_event.discrete = (int)value;
|
|
weston_event.has_discrete = true;
|
|
|
|
weston_compositor_get_time(&time);
|
|
|
|
notify_axis(peerContext->item.seat, &time, &weston_event);
|
|
need_frame = true;
|
|
}
|
|
|
|
if (need_frame)
|
|
notify_pointer_frame(peerContext->item.seat);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL
|
|
xf_extendedMouseEvent(rdpInput *input, UINT16 flags, UINT16 x, UINT16 y)
|
|
{
|
|
RdpPeerContext *peerContext = (RdpPeerContext *)input->context;
|
|
struct rdp_output *output;
|
|
struct timespec time;
|
|
|
|
output = peerContext->rdpBackend->output;
|
|
if (x < output->base.width && y < output->base.height) {
|
|
weston_compositor_get_time(&time);
|
|
notify_motion_absolute(peerContext->item.seat, &time, x, y);
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
static BOOL
|
|
xf_input_synchronize_event(rdpInput *input, UINT32 flags)
|
|
{
|
|
freerdp_peer *client = input->context->peer;
|
|
RdpPeerContext *peerCtx = (RdpPeerContext *)input->context;
|
|
struct rdp_output *output = peerCtx->rdpBackend->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);
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
static BOOL
|
|
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;
|
|
struct timespec time;
|
|
|
|
if (!(peerContext->item.flags & RDP_PEER_ACTIVATED))
|
|
return TRUE;
|
|
|
|
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 (flags & KBD_FLAGS_EXTENDED)
|
|
vk_code |= KBDEXT;
|
|
|
|
scan_code = GetKeycodeFromVirtualKeyCode(vk_code, KEYCODE_TYPE_EVDEV);
|
|
|
|
/*weston_log("code=%x ext=%d vk_code=%x scan_code=%x\n", code, (flags & KBD_FLAGS_EXTENDED) ? 1 : 0,
|
|
vk_code, scan_code);*/
|
|
weston_compositor_get_time(&time);
|
|
notify_key(peerContext->item.seat, &time,
|
|
scan_code - 8, keyState, STATE_UPDATE_AUTOMATIC);
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL
|
|
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);
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
static BOOL
|
|
xf_suppress_output(rdpContext *context, BYTE allow, const RECTANGLE_16 *area)
|
|
{
|
|
RdpPeerContext *peerContext = (RdpPeerContext *)context;
|
|
|
|
if (allow)
|
|
peerContext->item.flags |= RDP_PEER_OUTPUT_ENABLED;
|
|
else
|
|
peerContext->item.flags &= (~RDP_PEER_OUTPUT_ENABLED);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static int
|
|
rdp_peer_init(freerdp_peer *client, struct rdp_backend *b)
|
|
{
|
|
int rcount = 0;
|
|
void *rfds[MAX_FREERDP_FDS];
|
|
int i, fd;
|
|
struct wl_event_loop *loop;
|
|
rdpSettings *settings;
|
|
rdpInput *input;
|
|
RdpPeerContext *peerCtx;
|
|
|
|
client->ContextSize = 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->rdpBackend = b;
|
|
|
|
settings = client->settings;
|
|
/* configure security settings */
|
|
if (b->rdp_key)
|
|
settings->RdpKeyFile = strdup(b->rdp_key);
|
|
if (b->tls_enabled) {
|
|
settings->CertificateFile = strdup(b->server_cert);
|
|
settings->PrivateKeyFile = strdup(b->server_key);
|
|
} else {
|
|
settings->TlsSecurity = FALSE;
|
|
}
|
|
settings->NlaSecurity = FALSE;
|
|
|
|
if (!client->Initialize(client)) {
|
|
weston_log("peer initialization failed\n");
|
|
goto error_initialize;
|
|
}
|
|
|
|
settings->OsMajorType = OSMAJORTYPE_UNIX;
|
|
settings->OsMinorType = OSMINORTYPE_PSEUDO_XSERVER;
|
|
settings->ColorDepth = 32;
|
|
settings->RefreshRect = TRUE;
|
|
settings->RemoteFxCodec = TRUE;
|
|
settings->NSCodec = TRUE;
|
|
settings->FrameMarkerCommandEnabled = TRUE;
|
|
settings->SurfaceFrameMarkerEnabled = TRUE;
|
|
|
|
client->Capabilities = xf_peer_capabilities;
|
|
client->PostConnect = xf_peer_post_connect;
|
|
client->Activate = xf_peer_activate;
|
|
|
|
client->update->SuppressOutput = (pSuppressOutput)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 (!client->GetFileDescriptor(client, rfds, &rcount)) {
|
|
weston_log("unable to retrieve client fds\n");
|
|
goto error_initialize;
|
|
}
|
|
|
|
loop = wl_display_get_event_loop(b->compositor->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(&b->output->peers, &peerCtx->item.link);
|
|
return 0;
|
|
|
|
error_initialize:
|
|
client->Close(client);
|
|
return -1;
|
|
}
|
|
|
|
|
|
static BOOL
|
|
rdp_incoming_peer(freerdp_listener *instance, freerdp_peer *client)
|
|
{
|
|
struct rdp_backend *b = (struct rdp_backend *)instance->param4;
|
|
if (rdp_peer_init(client, b) < 0) {
|
|
weston_log("error when treating incoming peer\n");
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static const struct weston_rdp_output_api api = {
|
|
rdp_output_set_size,
|
|
};
|
|
|
|
static struct rdp_backend *
|
|
rdp_backend_create(struct weston_compositor *compositor,
|
|
struct weston_rdp_backend_config *config)
|
|
{
|
|
struct rdp_backend *b;
|
|
char *fd_str;
|
|
char *fd_tail;
|
|
int fd, ret;
|
|
|
|
struct weston_head *base, *next;
|
|
|
|
b = zalloc(sizeof *b);
|
|
if (b == NULL)
|
|
return NULL;
|
|
|
|
b->compositor = compositor;
|
|
b->base.destroy = rdp_destroy;
|
|
b->base.create_output = rdp_output_create;
|
|
b->rdp_key = config->rdp_key ? strdup(config->rdp_key) : NULL;
|
|
b->no_clients_resize = config->no_clients_resize;
|
|
b->force_no_compression = config->force_no_compression;
|
|
|
|
compositor->backend = &b->base;
|
|
|
|
/* activate TLS only if certificate/key are available */
|
|
if (config->server_cert && config->server_key) {
|
|
weston_log("TLS support activated\n");
|
|
b->server_cert = strdup(config->server_cert);
|
|
b->server_key = strdup(config->server_key);
|
|
if (!b->server_cert || !b->server_key)
|
|
goto err_free_strings;
|
|
b->tls_enabled = 1;
|
|
}
|
|
|
|
if (weston_compositor_set_presentation_clock_software(compositor) < 0)
|
|
goto err_compositor;
|
|
|
|
if (pixman_renderer_init(compositor) < 0)
|
|
goto err_compositor;
|
|
|
|
if (rdp_head_create(compositor, "rdp") < 0)
|
|
goto err_compositor;
|
|
|
|
compositor->capabilities |= WESTON_CAP_ARBITRARY_MODES;
|
|
|
|
if (!config->env_socket) {
|
|
b->listener = freerdp_listener_new();
|
|
b->listener->PeerAccepted = rdp_incoming_peer;
|
|
b->listener->param4 = b;
|
|
if (!b->listener->Open(b->listener, config->bind_address, config->port)) {
|
|
weston_log("unable to bind rdp socket\n");
|
|
goto err_listener;
|
|
}
|
|
|
|
if (rdp_implant_listener(b, b->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\n");
|
|
goto err_output;
|
|
}
|
|
|
|
fd = strtoul(fd_str, &fd_tail, 10);
|
|
if (errno != 0 || fd_tail == fd_str || *fd_tail != '\0'
|
|
|| rdp_peer_init(freerdp_peer_new(fd), b))
|
|
goto err_output;
|
|
}
|
|
|
|
ret = weston_plugin_api_register(compositor, WESTON_RDP_OUTPUT_API_NAME,
|
|
&api, sizeof(api));
|
|
|
|
if (ret < 0) {
|
|
weston_log("Failed to register output API.\n");
|
|
goto err_output;
|
|
}
|
|
|
|
return b;
|
|
|
|
err_listener:
|
|
freerdp_listener_free(b->listener);
|
|
err_output:
|
|
if (b->output)
|
|
weston_output_release(&b->output->base);
|
|
err_compositor:
|
|
wl_list_for_each_safe(base, next, &compositor->head_list, compositor_link)
|
|
rdp_head_destroy(to_rdp_head(base));
|
|
|
|
weston_compositor_shutdown(compositor);
|
|
err_free_strings:
|
|
free(b->rdp_key);
|
|
free(b->server_cert);
|
|
free(b->server_key);
|
|
free(b);
|
|
return NULL;
|
|
}
|
|
|
|
static void
|
|
config_init_to_defaults(struct weston_rdp_backend_config *config)
|
|
{
|
|
config->bind_address = NULL;
|
|
config->port = 3389;
|
|
config->rdp_key = NULL;
|
|
config->server_cert = NULL;
|
|
config->server_key = NULL;
|
|
config->env_socket = 0;
|
|
config->no_clients_resize = 0;
|
|
config->force_no_compression = 0;
|
|
}
|
|
|
|
WL_EXPORT int
|
|
weston_backend_init(struct weston_compositor *compositor,
|
|
struct weston_backend_config *config_base)
|
|
{
|
|
struct rdp_backend *b;
|
|
struct weston_rdp_backend_config config = {{ 0, }};
|
|
int major, minor, revision;
|
|
|
|
#if FREERDP_VERSION_MAJOR >= 2
|
|
winpr_InitializeSSL(0);
|
|
#endif
|
|
freerdp_get_version(&major, &minor, &revision);
|
|
weston_log("using FreeRDP version %d.%d.%d\n", major, minor, revision);
|
|
|
|
if (config_base == NULL ||
|
|
config_base->struct_version != WESTON_RDP_BACKEND_CONFIG_VERSION ||
|
|
config_base->struct_size > sizeof(struct weston_rdp_backend_config)) {
|
|
weston_log("RDP backend config structure is invalid\n");
|
|
return -1;
|
|
}
|
|
|
|
config_init_to_defaults(&config);
|
|
memcpy(&config, config_base, config_base->struct_size);
|
|
|
|
if (!config.rdp_key && (!config.server_cert || !config.server_key)) {
|
|
weston_log("the RDP compositor requires keys and an optional certificate for RDP or TLS security ("
|
|
"--rdp4-key or --rdp-tls-cert/--rdp-tls-key)\n");
|
|
return -1;
|
|
}
|
|
|
|
b = rdp_backend_create(compositor, &config);
|
|
if (b == NULL)
|
|
return -1;
|
|
return 0;
|
|
}
|
|
|