diff options
author | Drew DeVault <sir@cmpwn.com> | 2019-02-22 13:58:52 -0500 |
---|---|---|
committer | Drew DeVault <sir@cmpwn.com> | 2019-04-08 08:09:07 -0600 |
commit | fd0d7d0907fcb8783c4bd2fbbd8c8812f3ba1494 (patch) | |
tree | 003f3d8f144c92cb4073a348fd4191c96eced18d /backend/rdp | |
parent | 9faea17c738d9bfcdd241de7bae0bbe7590c77bf (diff) |
Add FreeRDP backend for remote desktop support
Diffstat (limited to 'backend/rdp')
-rw-r--r-- | backend/rdp/backend.c | 134 | ||||
-rw-r--r-- | backend/rdp/keyboard.c | 219 | ||||
-rw-r--r-- | backend/rdp/listener.c | 61 | ||||
-rw-r--r-- | backend/rdp/output.c | 288 | ||||
-rw-r--r-- | backend/rdp/peer.c | 360 | ||||
-rw-r--r-- | backend/rdp/pointer.c | 40 |
6 files changed, 1102 insertions, 0 deletions
diff --git a/backend/rdp/backend.c b/backend/rdp/backend.c new file mode 100644 index 00000000..ef0fca85 --- /dev/null +++ b/backend/rdp/backend.c @@ -0,0 +1,134 @@ +#define _POSIX_C_SOURCE 200809L +#include <assert.h> +#include <stdlib.h> +#include <string.h> +#include <wlr/render/egl.h> +#include <wlr/render/gles2.h> +#include <wlr/util/log.h> +#include "backend/rdp.h" +#include "glapi.h" +#include "util/signal.h" + +struct wlr_rdp_backend *rdp_backend_from_backend( + struct wlr_backend *wlr_backend) { + assert(wlr_backend_is_rdp(wlr_backend)); + return (struct wlr_rdp_backend *)wlr_backend; +} + +static bool backend_start(struct wlr_backend *wlr_backend) { + struct wlr_rdp_backend *backend = + rdp_backend_from_backend(wlr_backend); + assert(backend->listener == NULL); + wlr_log(WLR_INFO, "Starting RDP backend"); + if (!rdp_configure_listener(backend)) { + return false; + } + return true; +} + +void wlr_rdp_backend_set_address(struct wlr_backend *wlr_backend, + const char *address) { + struct wlr_rdp_backend *backend = + rdp_backend_from_backend(wlr_backend); + assert(backend->listener == NULL); + backend->address = strdup(address); +} + +void wlr_rdp_backend_set_port(struct wlr_backend *wlr_backend, int port) { + struct wlr_rdp_backend *backend = + rdp_backend_from_backend(wlr_backend); + assert(backend->listener == NULL); + backend->port = port; +} + +static void backend_destroy(struct wlr_backend *wlr_backend) { + struct wlr_rdp_backend *backend = + rdp_backend_from_backend(wlr_backend); + if (!wlr_backend) { + return; + } + + wl_list_remove(&backend->display_destroy.link); + + struct wlr_rdp_peer_context *client; + wl_list_for_each(client, &backend->clients, link) { + freerdp_peer_context_free(client->peer); + freerdp_peer_free(client->peer); + } + + wlr_signal_emit_safe(&wlr_backend->events.destroy, backend); + + wlr_renderer_destroy(backend->renderer); + wlr_egl_finish(&backend->egl); + free(backend->address); + free(backend); +} + +static struct wlr_renderer *backend_get_renderer( + struct wlr_backend *wlr_backend) { + struct wlr_rdp_backend *backend = + rdp_backend_from_backend(wlr_backend); + return backend->renderer; +} + +static const struct wlr_backend_impl backend_impl = { + .start = backend_start, + .destroy = backend_destroy, + .get_renderer = backend_get_renderer, +}; + +static void handle_display_destroy(struct wl_listener *listener, void *data) { + struct wlr_rdp_backend *backend = + wl_container_of(listener, backend, display_destroy); + backend_destroy(&backend->backend); +} + +struct wlr_backend *wlr_rdp_backend_create(struct wl_display *display, + wlr_renderer_create_func_t create_renderer_func, + const char *tls_cert_path, const char *tls_key_path) { + wlr_log(WLR_INFO, "Creating RDP backend"); + + struct wlr_rdp_backend *backend = + calloc(1, sizeof(struct wlr_rdp_backend)); + if (!backend) { + wlr_log(WLR_ERROR, "Failed to allocate wlr_rdp_backend"); + return NULL; + } + wlr_backend_init(&backend->backend, &backend_impl); + backend->display = display; + backend->tls_cert_path = tls_cert_path; + backend->tls_key_path = tls_key_path; + backend->address = strdup("127.0.0.1"); + backend->port = 3389; + wl_list_init(&backend->clients); + + static const EGLint config_attribs[] = { + EGL_SURFACE_TYPE, EGL_PBUFFER_BIT, + EGL_ALPHA_SIZE, 0, + EGL_BLUE_SIZE, 1, + EGL_GREEN_SIZE, 1, + EGL_RED_SIZE, 1, + EGL_NONE, + }; + + if (!create_renderer_func) { + create_renderer_func = wlr_renderer_autocreate; + } + + backend->renderer = create_renderer_func(&backend->egl, + EGL_PLATFORM_SURFACELESS_MESA, NULL, (EGLint*)config_attribs, 0); + if (!backend->renderer) { + wlr_log(WLR_ERROR, "Failed to create renderer"); + free(backend); + return NULL; + } + + backend->display_destroy.notify = handle_display_destroy; + wl_display_add_destroy_listener(display, &backend->display_destroy); + + return &backend->backend; +} + +bool wlr_backend_is_rdp(struct wlr_backend *backend) { + return backend->impl == &backend_impl; +} diff --git a/backend/rdp/keyboard.c b/backend/rdp/keyboard.c new file mode 100644 index 00000000..b3c5869d --- /dev/null +++ b/backend/rdp/keyboard.c @@ -0,0 +1,219 @@ +#include <stdlib.h> +#include <wayland-server.h> +#include <wlr/types/wlr_input_device.h> +#include <wlr/types/wlr_keyboard.h> +#include <wlr/interfaces/wlr_keyboard.h> +#include <wlr/interfaces/wlr_input_device.h> +#include <wlr/util/log.h> +#include <xkbcommon/xkbcommon.h> +#include "backend/rdp.h" +#include "util/signal.h" + +struct rdp_to_xkb_keyboard_layout { + UINT32 rdp_layout_code; + const char *xkb_layout; + const char *xkb_variant; +}; + +/* table reversed from + https://github.com/awakecoding/FreeRDP/blob/master/libfreerdp/locale/xkb_layout_ids.c#L811 */ +static struct rdp_to_xkb_keyboard_layout rdp_keyboards[] = { + {KBD_ARABIC_101, "ara", 0}, + {KBD_BULGARIAN, 0, 0}, + {KBD_CHINESE_TRADITIONAL_US, 0, 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 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 void keyboard_destroy(struct wlr_input_device *wlr_device) { + struct wlr_rdp_keyboard *keyboard = + (struct wlr_rdp_keyboard *)wlr_device->keyboard; + xkb_keymap_unref(keyboard->keymap); + free(keyboard); +} + +static struct wlr_input_device_impl input_device_impl = { + .destroy = keyboard_destroy, +}; + +struct wlr_rdp_input_device *wlr_rdp_keyboard_create( + struct wlr_rdp_backend *backend, rdpSettings *settings) { + struct wlr_rdp_input_device *device = + calloc(1, sizeof(struct wlr_rdp_input_device)); + if (!device) { + wlr_log(WLR_ERROR, "Failed to allcoate RDP input device"); + return NULL; + } + + int vendor = 0; + int product = 0; + const char *name = "rdp"; + struct wlr_input_device *wlr_device = &device->wlr_input_device; + wlr_input_device_init(wlr_device, WLR_INPUT_DEVICE_KEYBOARD, + &input_device_impl, name, vendor, product); + + struct wlr_rdp_keyboard *keyboard = + calloc(1, sizeof(struct wlr_rdp_keyboard)); + if (!keyboard) { + wlr_log(WLR_ERROR, "Failed to allocate RDP pointer device"); + goto error; + } + wlr_device->keyboard = (struct wlr_keyboard *)keyboard; + wlr_keyboard_init(wlr_device->keyboard, NULL); + + wlr_log(WLR_DEBUG, "RDP keyboard layout: 0x%x type: 0x%x subtype: 0x%x " + "function_keys 0x%x", settings->KeyboardLayout, + settings->KeyboardType, settings->KeyboardSubType, + settings->KeyboardFunctionKey); + + // We need to set up an XKB context and jump through some hoops to convert + // RDP input events into scancodes + struct xkb_rule_names xkb_rule_names = { 0 }; + if (settings->KeyboardType <= 7) { + xkb_rule_names.model = rdp_keyboard_types[settings->KeyboardType]; + } + for (int i = 0; rdp_keyboards[i].rdp_layout_code; ++i) { + if (rdp_keyboards[i].rdp_layout_code == settings->KeyboardLayout) { + xkb_rule_names.layout = rdp_keyboards[i].xkb_layout; + xkb_rule_names.variant = rdp_keyboards[i].xkb_variant; + wlr_log(WLR_DEBUG, "Mapped RDP keyboard to xkb layout %s variant " + "%s", xkb_rule_names.layout, xkb_rule_names.variant); + break; + } + } + if (xkb_rule_names.layout) { + struct xkb_context *xkb_context = xkb_context_new(0); + if (!xkb_context) { + wlr_log(WLR_DEBUG, "Failed to allocate xkb context"); + goto error; + } + keyboard->keymap = + xkb_keymap_new_from_names(xkb_context, &xkb_rule_names, 0); + xkb_context_unref(xkb_context); + } + + wlr_keyboard_set_keymap(wlr_device->keyboard, keyboard->keymap); + + wlr_signal_emit_safe(&backend->backend.events.new_input, wlr_device); + return device; +error: + wlr_input_device_destroy(wlr_device); + return NULL; +} diff --git a/backend/rdp/listener.c b/backend/rdp/listener.c new file mode 100644 index 00000000..cfbdc537 --- /dev/null +++ b/backend/rdp/listener.c @@ -0,0 +1,61 @@ +#include <stdlib.h> +#include <wlr/util/log.h> +#include "backend/rdp.h" + +static int rdp_incoming_peer( + freerdp_listener *listener, freerdp_peer *client) { + struct wlr_rdp_backend *backend = + (struct wlr_rdp_backend *)listener->param4; + if (rdp_peer_init(client, backend) < 0) { + wlr_log(WLR_ERROR, "Error initializing incoming peer"); + return false; + } + return true; +} + +static int rdp_listener_activity(int fd, uint32_t mask, void *data) { + freerdp_listener *listener = data; + if (!(mask & WL_EVENT_READABLE)) { + return 0; + } + if (!listener->CheckFileDescriptor(listener)) { + wlr_log(WLR_ERROR, "Failed to check FreeRDP file descriptor"); + return -1; + } + return 0; +} + +bool rdp_configure_listener(struct wlr_rdp_backend *backend) { + backend->listener = freerdp_listener_new(); + if (!backend->listener) { + wlr_log(WLR_ERROR, "Failed to allocate FreeRDP listener"); + return false; + } + backend->listener->PeerAccepted = rdp_incoming_peer; + backend->listener->param4 = backend; + if (!backend->listener->Open(backend->listener, + backend->address, backend->port)) { + wlr_log(WLR_ERROR, "Failed to bind to RDP socket"); + return false; + } + int rcount = 0; + void *rfds[MAX_FREERDP_FDS]; + if (!backend->listener->GetFileDescriptor( + backend->listener, rfds, &rcount)) { + wlr_log(WLR_ERROR, "Failed to get FreeRDP file descriptors"); + return false; + } + struct wl_event_loop *event_loop = + wl_display_get_event_loop(backend->display); + int i; + for (i = 0; i < rcount; ++i) { + int fd = (int)(long)(rfds[i]); + backend->listener_events[i] = wl_event_loop_add_fd( + event_loop, fd, WL_EVENT_READABLE, rdp_listener_activity, + backend->listener); + } + for (; i < MAX_FREERDP_FDS; ++i) { + backend->listener_events[i] = NULL; + } + return true; +} diff --git a/backend/rdp/output.c b/backend/rdp/output.c new file mode 100644 index 00000000..4d5d259d --- /dev/null +++ b/backend/rdp/output.c @@ -0,0 +1,288 @@ +#include <assert.h> +#include <EGL/egl.h> +#include <EGL/eglext.h> +#include <stdlib.h> +#include <wayland-server.h> +#include <wlr/types/wlr_output.h> +#include <wlr/interfaces/wlr_output.h> +#include <wlr/util/log.h> +#include <wlr/render/wlr_renderer.h> +#include "backend/rdp.h" +#include "util/signal.h" + +static struct wlr_rdp_output *rdp_output_from_output( + struct wlr_output *wlr_output) { + assert(wlr_output_is_rdp(wlr_output)); + return (struct wlr_rdp_output *)wlr_output; +} + +static EGLSurface egl_create_surface(struct wlr_egl *egl, unsigned int width, + unsigned int height) { + EGLint attribs[] = { + EGL_WIDTH, width, + EGL_HEIGHT, height, + EGL_NONE, + }; + EGLSurface surf = eglCreatePbufferSurface(egl->display, egl->config, attribs); + if (surf == EGL_NO_SURFACE) { + wlr_log(WLR_ERROR, "Failed to create EGL surface"); + return EGL_NO_SURFACE; + } + return surf; +} + +static bool output_set_custom_mode(struct wlr_output *wlr_output, int32_t width, + int32_t height, int32_t refresh) { + struct wlr_rdp_output *output = + rdp_output_from_output(wlr_output); + struct wlr_rdp_backend *backend = output->backend; + + if (refresh <= 0) { + refresh = 60 * 1000; // 60 Hz + } + + wlr_egl_destroy_surface(&backend->egl, output->egl_surface); + + output->egl_surface = egl_create_surface(&backend->egl, width, height); + if (output->egl_surface == EGL_NO_SURFACE) { + wlr_log(WLR_ERROR, "Failed to recreate EGL surface"); + wlr_output_destroy(wlr_output); + return false; + } + + output->frame_delay = 1000000 / refresh; + + if (output->shadow_surface) { + pixman_image_unref(output->shadow_surface); + } + output->shadow_surface = pixman_image_create_bits(PIXMAN_x8r8g8b8, + width, height, NULL, width * 4); + + wlr_output_update_custom_mode(&output->wlr_output, width, height, refresh); + return true; +} + +static void output_transform(struct wlr_output *wlr_output, + enum wl_output_transform transform) { + struct wlr_rdp_output *output = + rdp_output_from_output(wlr_output); + output->wlr_output.transform = transform; +} + +static bool output_make_current(struct wlr_output *wlr_output, int *buffer_age) { + struct wlr_rdp_output *output = + rdp_output_from_output(wlr_output); + return wlr_egl_make_current(&output->backend->egl, output->egl_surface, + buffer_age); +} + +static bool rfx_swap_buffers( + struct wlr_rdp_output *output, pixman_region32_t *damage) { + struct wlr_rdp_peer_context *context = output->context; + freerdp_peer *peer = context->peer; + rdpUpdate *update = peer->update; + + Stream_Clear(context->encode_stream); + Stream_SetPosition(context->encode_stream, 0); + int width = damage->extents.x2 - damage->extents.x1; + int height = damage->extents.y2 - damage->extents.y1; + + SURFACE_BITS_COMMAND cmd; + 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 = pixman_image_get_depth(output->shadow_surface); + cmd.bmp.codecID = peer->settings->RemoteFxCodecId; + cmd.bmp.width = width; + cmd.bmp.height = height; + + uint32_t *ptr = pixman_image_get_data(output->shadow_surface) + + damage->extents.x1 + damage->extents.y1 * + (pixman_image_get_stride(output->shadow_surface) / sizeof(uint32_t)); + + RFX_RECT *rfx_rect; + int nrects; + pixman_box32_t *rects = + pixman_region32_rectangles(damage, &nrects); + rfx_rect = realloc(context->rfx_rects, nrects * sizeof(*rfx_rect)); + if (rfx_rect == NULL) { + wlr_log(WLR_ERROR, "RDP swap buffers failed: could not realloc rects"); + return false; + } + context->rfx_rects = rfx_rect; + + for (int i = 0; i < nrects; ++i) { + pixman_box32_t *region = &rects[i]; + rfx_rect = &context->rfx_rects[i]; + rfx_rect->x = region->x1 - damage->extents.x1; + rfx_rect->y = region->y1 - damage->extents.y1; + rfx_rect->width = region->x2 - region->x1; + rfx_rect->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(output->shadow_surface)); + cmd.bmp.bitmapDataLength = Stream_GetPosition(context->encode_stream); + cmd.bmp.bitmapData = Stream_Buffer(context->encode_stream); + + update->SurfaceBits(update->context, &cmd); + return true; +} + +static bool nsc_swap_buffers( + struct wlr_rdp_output *output, pixman_region32_t *damage) { + struct wlr_rdp_peer_context *context = output->context; + freerdp_peer *peer = context->peer; + rdpUpdate *update = peer->update; + + Stream_Clear(context->encode_stream); + Stream_SetPosition(context->encode_stream, 0); + int width = damage->extents.x2 - damage->extents.x1; + int height = damage->extents.y2 - damage->extents.y1; + + SURFACE_BITS_COMMAND cmd; + 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 = pixman_image_get_depth(output->shadow_surface); + cmd.bmp.codecID = peer->settings->NSCodecId; + cmd.bmp.width = width; + cmd.bmp.height = height; + + uint32_t *ptr = pixman_image_get_data(output->shadow_surface) + + damage->extents.x1 + damage->extents.y1 * + (pixman_image_get_stride(output->shadow_surface) / sizeof(uint32_t)); + + nsc_compose_message(context->nsc_context, context->encode_stream, + (BYTE *)ptr, width, height, + pixman_image_get_stride(output->shadow_surface)); + + cmd.bmp.bitmapDataLength = Stream_GetPosition(context->encode_stream); + cmd.bmp.bitmapData = Stream_Buffer(context->encode_stream); + + update->SurfaceBits(update->context, &cmd); + return true; +} + +static bool output_swap_buffers( + struct wlr_output *wlr_output, pixman_region32_t *damage) { + if (!pixman_region32_not_empty(damage)) { + return true; + } + + struct wlr_rdp_output *output = + rdp_output_from_output(wlr_output); + + // Update shadow buffer + int width = damage->extents.x2 - damage->extents.x1; + int height = damage->extents.y2 - damage->extents.y1; + struct wlr_renderer *renderer = + wlr_backend_get_renderer(&output->backend->backend); + // TODO performance: add support for flags + if (!wlr_renderer_read_pixels(renderer, WL_SHM_FORMAT_XRGB8888, + NULL, pixman_image_get_stride(output->shadow_surface), + width, height, damage->extents.x1, damage->extents.y1, + damage->extents.x1, damage->extents.y1, + pixman_image_get_data(output->shadow_surface))) { + return false; + } + + // Send along to clients + bool ret = false; + rdpSettings *settings = output->context->peer->settings; + if (settings->RemoteFxCodec) { + ret = rfx_swap_buffers(output, damage); + } else if (settings->NSCodec) { + ret = nsc_swap_buffers(output, damage); + } else { + // This would perform like ass so why bother + wlr_log(WLR_ERROR, "Raw updates are not supported; use rfx or nsc"); + } + wlr_output_send_present(wlr_output, NULL); + return ret; +} + +static void output_destroy(struct wlr_output *wlr_output) { + struct wlr_rdp_output *output = + rdp_output_from_output(wlr_output); + if (output->frame_timer) { + wl_event_source_remove(output->frame_timer); + } + wlr_egl_destroy_surface(&output->backend->egl, output->egl_surface); + if (output->shadow_surface) { + pixman_image_unref(output->shadow_surface); + } + free(output); +} + +static const struct wlr_output_impl output_impl = { + .set_custom_mode = output_set_custom_mode, + .transform = output_transform, + .destroy = output_destroy, + .make_current = output_make_current, + .swap_buffers = output_swap_buffers, +}; + +bool wlr_output_is_rdp(struct wlr_output *wlr_output) { + return wlr_output->impl == &output_impl; +} + +static int signal_frame(void *data) { + struct wlr_rdp_output *output = data; + wlr_output_send_frame(&output->wlr_output); + wl_event_source_timer_update(output->frame_timer, output->frame_delay); + return 0; +} + +struct wlr_rdp_output *wlr_rdp_output_create(struct wlr_rdp_backend *backend, + struct wlr_rdp_peer_context *context, unsigned int width, + unsigned int height) { + struct wlr_rdp_output *output = + calloc(1, sizeof(struct wlr_rdp_output)); + if (output == NULL) { + wlr_log(WLR_ERROR, "Failed to allocate wlr_rdp_output"); + return NULL; + } + output->backend = backend; + output->context = context; + wlr_output_init(&output->wlr_output, &backend->backend, &output_impl, + backend->display); + struct wlr_output *wlr_output = &output->wlr_output; + + output->egl_surface = egl_create_surface(&backend->egl, width, height); + if (output->egl_surface == EGL_NO_SURFACE) { + wlr_log(WLR_ERROR, "Failed to create EGL surface"); + goto error; + } + + output_set_custom_mode(wlr_output, width, height, 0); + strncpy(wlr_output->make, "RDP", sizeof(wlr_output->make)); + strncpy(wlr_output->model, "RDP", sizeof(wlr_output->model)); + snprintf(wlr_output->name, sizeof(wlr_output->name), "RDP-%d", + wl_list_length(&backend->clients)); + + if (!wlr_egl_make_current(&output->backend->egl, output->egl_surface, + NULL)) { + goto error; + } + + wlr_renderer_begin(backend->renderer, wlr_output->width, wlr_output->height); + wlr_renderer_clear(backend->renderer, (float[]){ 1.0, 1.0, 1.0, 1.0 }); + wlr_renderer_end(backend->renderer); + + struct wl_event_loop *ev = wl_display_get_event_loop(backend->display); + output->frame_timer = wl_event_loop_add_timer(ev, signal_frame, output); + wl_event_source_timer_update(output->frame_timer, output->frame_delay); + wlr_output_update_enabled(wlr_output, true); + wlr_signal_emit_safe(&backend->backend.events.new_output, wlr_output); + return output; + +error: + wlr_output_destroy(&output->wlr_output); + return NULL; +} diff --git a/backend/rdp/peer.c b/backend/rdp/peer.c new file mode 100644 index 00000000..350a4721 --- /dev/null +++ b/backend/rdp/peer.c @@ -0,0 +1,360 @@ +#define _POSIX_C_SOURCE 200809L +#include <linux/input.h> +#include <stdlib.h> +#include <string.h> +#include <wlr/types/wlr_output.h> +#include <wlr/interfaces/wlr_input_device.h> +#include <wlr/interfaces/wlr_keyboard.h> +#include <wlr/interfaces/wlr_output.h> +#include <wlr/util/log.h> +#include "backend/rdp.h" +#include "util/signal.h" + +static BOOL xf_peer_capabilities(freerdp_peer *client) { + return TRUE; +} + +static BOOL xf_peer_post_connect(freerdp_peer *client) { + return TRUE; +} + +static BOOL xf_peer_activate(freerdp_peer *client) { + struct wlr_rdp_peer_context *context = + (struct wlr_rdp_peer_context *)client->context; + struct wlr_rdp_backend *backend = context->backend; + rdpSettings *settings = client->settings; + + if (!settings->SurfaceCommandsEnabled) { + wlr_log(WLR_ERROR, "RDP peer does not support SurfaceCommands"); + return FALSE; + } + + context->output = wlr_rdp_output_create(backend, context, + (int)settings->DesktopWidth, (int)settings->DesktopHeight); + if (!context->output) { + wlr_log(WLR_ERROR, "Failed to allcoate output for RDP peer"); + return FALSE; + } + rfx_context_reset(context->rfx_context, + context->output->wlr_output.width, + context->output->wlr_output.height); + nsc_context_reset(context->nsc_context, + context->output->wlr_output.width, + context->output->wlr_output.height); + + if (context->flags & RDP_PEER_ACTIVATED) { + return TRUE; + } + + context->pointer = wlr_rdp_pointer_create(backend, context); + if (!context->pointer) { + wlr_log(WLR_ERROR, "Failed to allocate pointer for RDP peer"); + return FALSE; + } + + // Use wlroots' software cursors instead of remote cursors + POINTER_SYSTEM_UPDATE pointer_system; + rdpPointerUpdate *pointer = client->update->pointer; + pointer_system.type = SYSPTR_NULL; + pointer->PointerSystem(client->context, &pointer_system); + + context->keyboard = wlr_rdp_keyboard_create(backend, settings); + if (!context->keyboard) { + wlr_log(WLR_ERROR, "Failed to allocate keyboard for RDP peer"); + return FALSE; + } + + context->flags |= RDP_PEER_ACTIVATED; + return TRUE; +} + +static int xf_suppress_output(rdpContext *context, + BYTE allow, const RECTANGLE_16 *area) { + struct wlr_rdp_peer_context *peer_context = + (struct wlr_rdp_peer_context *)context; + if (allow) { + peer_context->flags |= RDP_PEER_OUTPUT_ENABLED; + } else { + peer_context->flags &= ~RDP_PEER_OUTPUT_ENABLED; + } + return true; +} + +static int xf_input_synchronize_event(rdpInput *input, UINT32 flags) { + struct wlr_rdp_peer_context *context = + (struct wlr_rdp_peer_context *)input->context; + wlr_output_damage_whole(&context->output->wlr_output); + return true; +} + +static inline int64_t timespec_to_msec(const struct timespec *a) { + return (int64_t)a->tv_sec * 1000 + a->tv_nsec / 1000000; +} + +static int xf_input_mouse_event(rdpInput *input, + UINT16 flags, UINT16 x, UINT16 y) { + struct wlr_rdp_peer_context *context = + (struct wlr_rdp_peer_context *)input->context; + struct wlr_input_device *wlr_device = &context->pointer->wlr_input_device; + struct wlr_pointer *pointer = wlr_device->pointer; + bool frame = false; + struct timespec now; + clock_gettime(CLOCK_MONOTONIC, &now); + + if (flags & PTR_FLAGS_MOVE) { + struct wlr_event_pointer_motion_absolute event = { 0 }; + event.device = wlr_device; + event.time_msec = timespec_to_msec(&now); + event.x = x / (double)context->output->wlr_output.width; + event.y = y / (double)context->output->wlr_output.height; + wlr_signal_emit_safe(&pointer->events.motion_absolute, &event); + frame = true; + } + + uint32_t button = 0; + 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) { + struct wlr_event_pointer_button event = { 0 }; + event.device = wlr_device; + event.time_msec = timespec_to_msec(&now); + event.button = button; + event.state = (flags & PTR_FLAGS_DOWN) ? + WLR_BUTTON_PRESSED : WLR_BUTTON_RELEASED; + wlr_signal_emit_safe(&pointer->events.button, &event); + frame = true; + } + + if (flags & PTR_FLAGS_WHEEL) { + double value = -(flags & 0xFF) / 120.0; + if (flags & PTR_FLAGS_WHEEL_NEGATIVE) { + value = -value; + } + struct wlr_event_pointer_axis event = { 0 }; + event.device = &context->pointer->wlr_input_device; + event.time_msec = timespec_to_msec(&now); + event.source = WLR_AXIS_SOURCE_WHEEL; + event.orientation = WLR_AXIS_ORIENTATION_VERTICAL; + event.delta = value; + event.delta_discrete = (int32_t)value; + wlr_signal_emit_safe(&pointer->events.axis, &event); + frame = true; + } + + if (frame) { + wlr_signal_emit_safe(&pointer->events.frame, pointer); + } + + return true; +} + +static int xf_input_extended_mouse_event( + rdpInput *input, UINT16 flags, UINT16 x, UINT16 y) { + struct wlr_rdp_peer_context *context = + (struct wlr_rdp_peer_context *)input->context; + struct wlr_input_device *wlr_device = &context->pointer->wlr_input_device; + struct wlr_pointer *pointer = wlr_device->pointer; + struct timespec now; + clock_gettime(CLOCK_MONOTONIC, &now); + struct wlr_event_pointer_motion_absolute event = { 0 }; + event.device = wlr_device; + event.time_msec = timespec_to_msec(&now); + event.x = x / (double)context->output->wlr_output.width; + event.y = y / (double)context->output->wlr_output.height; + wlr_signal_emit_safe(&pointer->events.motion_absolute, &event); + wlr_signal_emit_safe(&pointer->events.frame, pointer); + return true; +} + +static int xf_input_keyboard_event(rdpInput *input, UINT16 flags, UINT16 code) { + struct wlr_rdp_peer_context *context = + (struct wlr_rdp_peer_context *)input->context; + struct wlr_input_device *wlr_device = &context->keyboard->wlr_input_device; + struct wlr_keyboard *keyboard = wlr_device->keyboard; + struct timespec now; + clock_gettime(CLOCK_MONOTONIC, &now); + + if (!(context->flags & RDP_PEER_ACTIVATED)) { + return true; + } + + bool notify = false; + enum wlr_key_state state; + if ((flags & KBD_FLAGS_DOWN)) { + state = WLR_KEY_PRESSED; + notify = true; + } else if ((flags & KBD_FLAGS_RELEASE)) { + state = WLR_KEY_RELEASED; + notify = true; + } + + if (notify) { + uint32_t full_code = code; + if (flags & KBD_FLAGS_EXTENDED) { + full_code |= KBD_FLAGS_EXTENDED; + } + uint32_t vk_code = GetVirtualKeyCodeFromVirtualScanCode(full_code, 4); + if (flags & KBD_FLAGS_EXTENDED) { + vk_code |= KBDEXT; + } + uint32_t scan_code = GetKeycodeFromVirtualKeyCode( + vk_code, KEYCODE_TYPE_EVDEV); + struct wlr_event_keyboard_key event = { 0 }; + event.time_msec = timespec_to_msec(&now); + event.keycode = scan_code - 8; + event.state = state; + event.update_state = true; + wlr_keyboard_notify_key(keyboard, &event); + } + + return true; +} + +static int xf_input_unicode_keyboard_event(rdpInput *input, + UINT16 flags, UINT16 code) { + wlr_log(WLR_DEBUG, "Unhandled RDP unicode keyboard event " + "(flags:0x%X code:0x%X)\n", flags, code); + return true; +} + +static int rdp_client_activity(int fd, uint32_t mask, void *data) { + freerdp_peer *client = (freerdp_peer *)data; + if (!client->CheckFileDescriptor(client)) { + wlr_log(WLR_ERROR, + "Unable to check client file descriptor for %p", client); + freerdp_peer_context_free(client); + freerdp_peer_free(client); + } + return 0; +} + +static int rdp_peer_context_new( + freerdp_peer *client, struct wlr_rdp_peer_context *context) { + context->peer = client; + context->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, PIXEL_FORMAT_BGRA32); + + context->nsc_context = nsc_context_new(); + if (!context->nsc_context) { + rfx_context_free(context->rfx_context); + return false; + } + + nsc_context_set_pixel_format(context->nsc_context, PIXEL_FORMAT_BGRA32); + + context->encode_stream = Stream_New(NULL, 65536); + if (!context->encode_stream) { + nsc_context_free(context->nsc_context); + rfx_context_free(context->rfx_context); + return false; + } + return true; +} + +static void rdp_peer_context_free( + freerdp_peer *client, struct wlr_rdp_peer_context *context) { + if (!context) { + return; + } + + for (int i = 0; i < MAX_FREERDP_FDS; ++i) { + if (context->events[i]) { + wl_event_source_remove(context->events[i]); + } + } + + if (context->flags & RDP_PEER_ACTIVATED) { + wlr_output_destroy(&context->output->wlr_output); + wlr_input_device_destroy(&context->pointer->wlr_input_device); + wlr_input_device_destroy(&context->keyboard->wlr_input_device); + } + + wl_list_remove(&context->link); + wlr_output_destroy(&context->output->wlr_output); + + Stream_Free(context->encode_stream, TRUE); + nsc_context_free(context->nsc_context); + rfx_context_free(context->rfx_context); + free(context->rfx_rects); +} + +int rdp_peer_init(freerdp_peer *client, + struct wlr_rdp_backend *backend) { + client->ContextSize = sizeof(struct wlr_rdp_peer_context); + client->ContextNew = (psPeerContextNew)rdp_peer_context_new; + client->ContextFree = (psPeerContextFree)rdp_peer_context_free; + freerdp_peer_context_new(client); + + struct wlr_rdp_peer_context *peer_context = + (struct wlr_rdp_peer_context *)client->context; + peer_context->backend = backend; + + client->settings->CertificateFile = strdup(backend->tls_cert_path); + client->settings->PrivateKeyFile = strdup(backend->tls_key_path); + client->settings->NlaSecurity = FALSE; + + if (!client->Initialize(client)) { + wlr_log(WLR_ERROR, "Failed to initialize FreeRDP peer"); + goto err_init; + } + + client->settings->OsMajorType = OSMAJORTYPE_UNIX; + client->settings->OsMinorType = OSMINORTYPE_PSEUDO_XSERVER; + client->settings->ColorDepth = 32; + client->settings->RefreshRect = TRUE; + client->settings->RemoteFxCodec = TRUE; + client->settings->NSCodec = TRUE; + client->settings->FrameMarkerCommandEnabled = TRUE; + client->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; + + client->input->SynchronizeEvent = xf_input_synchronize_event; + client->input->MouseEvent = xf_input_mouse_event; + client->input->ExtendedMouseEvent = xf_input_extended_mouse_event; + client->input->KeyboardEvent = xf_input_keyboard_event; + client->input->UnicodeKeyboardEvent = xf_input_unicode_keyboard_event; + + int rcount = 0; + void *rfds[MAX_FREERDP_FDS]; + if (!client->GetFileDescriptor(client, rfds, &rcount)) { + wlr_log(WLR_ERROR, "Unable to retrieve client file descriptors"); + goto err_init; + } + struct wl_event_loop *event_loop = + wl_display_get_event_loop(backend->display); + int i; + for (i = 0; i < rcount; ++i) { + int fd = (int)(long)(rfds[i]); + peer_context->events[i] = wl_event_loop_add_fd( + event_loop, fd, WL_EVENT_READABLE, rdp_client_activity, + client); + } + for (; i < MAX_FREERDP_FDS; ++i) { + peer_context->events[i] = NULL; + } + + wl_list_insert(&backend->clients, &peer_context->link); + return 0; + +err_init: + client->Close(client); + return -1; +} diff --git a/backend/rdp/pointer.c b/backend/rdp/pointer.c new file mode 100644 index 00000000..31186669 --- /dev/null +++ b/backend/rdp/pointer.c @@ -0,0 +1,40 @@ +#define _POSIX_C_SOURCE 200809L +#include <stdlib.h> +#include <string.h> +#include <wayland-server.h> +#include <wlr/types/wlr_input_device.h> +#include <wlr/types/wlr_pointer.h> +#include <wlr/interfaces/wlr_pointer.h> +#include <wlr/interfaces/wlr_input_device.h> +#include <wlr/util/log.h> +#include "backend/rdp.h" +#include "util/signal.h" + +static struct wlr_input_device_impl input_device_impl = { 0 }; + +struct wlr_rdp_input_device *wlr_rdp_pointer_create( + struct wlr_rdp_backend *backend, struct wlr_rdp_peer_context *context) { + struct wlr_rdp_input_device *device = + calloc(1, sizeof(struct wlr_rdp_input_device)); + if (!device) { + wlr_log(WLR_ERROR, "Failed to allocate RDP input device"); + return NULL; + } + + int vendor = 0; + int product = 0; + const char *name = "rdp"; + struct wlr_input_device *wlr_device = &device->wlr_input_device; + wlr_input_device_init(wlr_device, WLR_INPUT_DEVICE_POINTER, + &input_device_impl, name, vendor, product); + + if (!(wlr_device->pointer = calloc(1, sizeof(struct wlr_pointer)))) { + wlr_log(WLR_ERROR, "Failed to allocate RDP pointer device"); + return NULL; + } + wlr_device->output_name = strdup(context->output->wlr_output.name); + wlr_pointer_init(wlr_device->pointer, NULL); + + wlr_signal_emit_safe(&backend->backend.events.new_input, wlr_device); + return device; +} |