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