git.maemo.org Git - qemu/blob - vnc.c

   1 /*
   2  * QEMU VNC display driver
   3  *
   4  * Copyright (C) 2006 Anthony Liguori <anthony@codemonkey.ws>
   5  * Copyright (C) 2006 Fabrice Bellard
   6  *
   7  * Permission is hereby granted, free of charge, to any person obtaining a copy
   8  * of this software and associated documentation files (the "Software"), to deal
   9  * in the Software without restriction, including without limitation the rights
  10  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  11  * copies of the Software, and to permit persons to whom the Software is
  12  * furnished to do so, subject to the following conditions:
  13  *
  14  * The above copyright notice and this permission notice shall be included in
  15  * all copies or substantial portions of the Software.
  16  *
  17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  20  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  21  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  22  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  23  * THE SOFTWARE.
  24  */
  25
  26 #include "vl.h"
  27 #include "qemu_socket.h"
  28
  29 #define VNC_REFRESH_INTERVAL (1000 / 30)
  30
  31 #include "vnc_keysym.h"
  32 #include "keymaps.c"
  33 #include "d3des.h"
  34
  35 #if CONFIG_VNC_TLS
  36 #include <gnutls/gnutls.h>
  37 #include <gnutls/x509.h>
  38 #endif /* CONFIG_VNC_TLS */
  39
  40 // #define _VNC_DEBUG 1
  41
  42 #if _VNC_DEBUG
  43 #define VNC_DEBUG(fmt, ...) do { fprintf(stderr, fmt, ## __VA_ARGS__); } while (0)
  44
  45 #if CONFIG_VNC_TLS && _VNC_DEBUG >= 2
  46 /* Very verbose, so only enabled for _VNC_DEBUG >= 2 */
  47 static void vnc_debug_gnutls_log(int level, const char* str) {
  48     VNC_DEBUG("%d %s", level, str);
  49 }
  50 #endif /* CONFIG_VNC_TLS && _VNC_DEBUG */
  51 #else
  52 #define VNC_DEBUG(fmt, ...) do { } while (0)
  53 #endif
  54
  55
  56 typedef struct Buffer
  57 {
  58     size_t capacity;
  59     size_t offset;
  60     char *buffer;
  61 } Buffer;
  62
  63 typedef struct VncState VncState;
  64
  65 typedef int VncReadEvent(VncState *vs, char *data, size_t len);
  66
  67 typedef void VncWritePixels(VncState *vs, void *data, int size);
  68
  69 typedef void VncSendHextileTile(VncState *vs,
  70                                 int x, int y, int w, int h,
  71                                 uint32_t *last_bg,
  72                                 uint32_t *last_fg,
  73                                 int *has_bg, int *has_fg);
  74
  75 #define VNC_MAX_WIDTH 2048
  76 #define VNC_MAX_HEIGHT 2048
  77 #define VNC_DIRTY_WORDS (VNC_MAX_WIDTH / (16 * 32))
  78
  79 #define VNC_AUTH_CHALLENGE_SIZE 16
  80
  81 enum {
  82     VNC_AUTH_INVALID = 0,
  83     VNC_AUTH_NONE = 1,
  84     VNC_AUTH_VNC = 2,
  85     VNC_AUTH_RA2 = 5,
  86     VNC_AUTH_RA2NE = 6,
  87     VNC_AUTH_TIGHT = 16,
  88     VNC_AUTH_ULTRA = 17,
  89     VNC_AUTH_TLS = 18,
  90     VNC_AUTH_VENCRYPT = 19
  91 };
  92
  93 #if CONFIG_VNC_TLS
  94 enum {
  95     VNC_WIREMODE_CLEAR,
  96     VNC_WIREMODE_TLS,
  97 };
  98
  99 enum {
 100     VNC_AUTH_VENCRYPT_PLAIN = 256,
 101     VNC_AUTH_VENCRYPT_TLSNONE = 257,
 102     VNC_AUTH_VENCRYPT_TLSVNC = 258,
 103     VNC_AUTH_VENCRYPT_TLSPLAIN = 259,
 104     VNC_AUTH_VENCRYPT_X509NONE = 260,
 105     VNC_AUTH_VENCRYPT_X509VNC = 261,
 106     VNC_AUTH_VENCRYPT_X509PLAIN = 262,
 107 };
 108
 109 #if CONFIG_VNC_TLS
 110 #define X509_CA_CERT_FILE "ca-cert.pem"
 111 #define X509_CA_CRL_FILE "ca-crl.pem"
 112 #define X509_SERVER_KEY_FILE "server-key.pem"
 113 #define X509_SERVER_CERT_FILE "server-cert.pem"
 114 #endif
 115
 116 #endif /* CONFIG_VNC_TLS */
 117
 118 struct VncState
 119 {
 120     QEMUTimer *timer;
 121     int lsock;
 122     int csock;
 123     DisplayState *ds;
 124     int need_update;
 125     int width;
 126     int height;
 127     uint32_t dirty_row[VNC_MAX_HEIGHT][VNC_DIRTY_WORDS];
 128     char *old_data;
 129     int depth; /* internal VNC frame buffer byte per pixel */
 130     int has_resize;
 131     int has_hextile;
 132     int has_pointer_type_change;
 133     int absolute;
 134     int last_x;
 135     int last_y;
 136
 137     int major;
 138     int minor;
 139
 140     char *display;
 141     char *password;
 142     int auth;
 143 #if CONFIG_VNC_TLS
 144     int subauth;
 145     int x509verify;
 146
 147     char *x509cacert;
 148     char *x509cacrl;
 149     char *x509cert;
 150     char *x509key;
 151 #endif
 152     char challenge[VNC_AUTH_CHALLENGE_SIZE];
 153
 154 #if CONFIG_VNC_TLS
 155     int wiremode;
 156     gnutls_session_t tls_session;
 157 #endif
 158
 159     Buffer output;
 160     Buffer input;
 161     kbd_layout_t *kbd_layout;
 162     /* current output mode information */
 163     VncWritePixels *write_pixels;
 164     VncSendHextileTile *send_hextile_tile;
 165     int pix_bpp, pix_big_endian;
 166     int red_shift, red_max, red_shift1;
 167     int green_shift, green_max, green_shift1;
 168     int blue_shift, blue_max, blue_shift1;
 169
 170     VncReadEvent *read_handler;
 171     size_t read_handler_expect;
 172     /* input */
 173     uint8_t modifiers_state[256];
 174 };
 175
 176 static VncState *vnc_state; /* needed for info vnc */
 177
 178 void do_info_vnc(void)
 179 {
 180     if (vnc_state == NULL)
 181         term_printf("VNC server disabled\n");
 182     else {
 183         term_printf("VNC server active on: ");
 184         term_print_filename(vnc_state->display);
 185         term_printf("\n");
 186
 187         if (vnc_state->csock == -1)
 188             term_printf("No client connected\n");
 189         else
 190             term_printf("Client connected\n");
 191     }
 192 }
 193
 194 /* TODO
 195    1) Get the queue working for IO.
 196    2) there is some weirdness when using the -S option (the screen is grey
 197       and not totally invalidated
 198    3) resolutions > 1024
 199 */
 200
 201 static void vnc_write(VncState *vs, const void *data, size_t len);
 202 static void vnc_write_u32(VncState *vs, uint32_t value);
 203 static void vnc_write_s32(VncState *vs, int32_t value);
 204 static void vnc_write_u16(VncState *vs, uint16_t value);
 205 static void vnc_write_u8(VncState *vs, uint8_t value);
 206 static void vnc_flush(VncState *vs);
 207 static void vnc_update_client(void *opaque);
 208 static void vnc_client_read(void *opaque);
 209
 210 static inline void vnc_set_bit(uint32_t *d, int k)
 211 {
 212     d[k >> 5] |= 1 << (k & 0x1f);
 213 }
 214
 215 static inline void vnc_clear_bit(uint32_t *d, int k)
 216 {
 217     d[k >> 5] &= ~(1 << (k & 0x1f));
 218 }
 219
 220 static inline void vnc_set_bits(uint32_t *d, int n, int nb_words)
 221 {
 222     int j;
 223
 224     j = 0;
 225     while (n >= 32) {
 226         d[j++] = -1;
 227         n -= 32;
 228     }
 229     if (n > 0)
 230         d[j++] = (1 << n) - 1;
 231     while (j < nb_words)
 232         d[j++] = 0;
 233 }
 234
 235 static inline int vnc_get_bit(const uint32_t *d, int k)
 236 {
 237     return (d[k >> 5] >> (k & 0x1f)) & 1;
 238 }
 239
 240 static inline int vnc_and_bits(const uint32_t *d1, const uint32_t *d2,
 241                                int nb_words)
 242 {
 243     int i;
 244     for(i = 0; i < nb_words; i++) {
 245         if ((d1[i] & d2[i]) != 0)
 246             return 1;
 247     }
 248     return 0;
 249 }
 250
 251 static void vnc_dpy_update(DisplayState *ds, int x, int y, int w, int h)
 252 {
 253     VncState *vs = ds->opaque;
 254     int i;
 255
 256     h += y;
 257
 258     for (; y < h; y++)
 259         for (i = 0; i < w; i += 16)
 260             vnc_set_bit(vs->dirty_row[y], (x + i) / 16);
 261 }
 262
 263 static void vnc_framebuffer_update(VncState *vs, int x, int y, int w, int h,
 264                                    int32_t encoding)
 265 {
 266     vnc_write_u16(vs, x);
 267     vnc_write_u16(vs, y);
 268     vnc_write_u16(vs, w);
 269     vnc_write_u16(vs, h);
 270
 271     vnc_write_s32(vs, encoding);
 272 }
 273
 274 static void vnc_dpy_resize(DisplayState *ds, int w, int h)
 275 {
 276     int size_changed;
 277     VncState *vs = ds->opaque;
 278
 279     ds->data = realloc(ds->data, w * h * vs->depth);
 280     vs->old_data = realloc(vs->old_data, w * h * vs->depth);
 281
 282     if (ds->data == NULL || vs->old_data == NULL) {
 283         fprintf(stderr, "vnc: memory allocation failed\n");
 284         exit(1);
 285     }
 286
 287     ds->depth = vs->depth * 8;
 288     size_changed = ds->width != w || ds->height != h;
 289     ds->width = w;
 290     ds->height = h;
 291     ds->linesize = w * vs->depth;
 292     if (vs->csock != -1 && vs->has_resize && size_changed) {
 293         vnc_write_u8(vs, 0);  /* msg id */
 294         vnc_write_u8(vs, 0);
 295         vnc_write_u16(vs, 1); /* number of rects */
 296         vnc_framebuffer_update(vs, 0, 0, ds->width, ds->height, -223);
 297         vnc_flush(vs);
 298         vs->width = ds->width;
 299         vs->height = ds->height;
 300     }
 301 }
 302
 303 /* fastest code */
 304 static void vnc_write_pixels_copy(VncState *vs, void *pixels, int size)
 305 {
 306     vnc_write(vs, pixels, size);
 307 }
 308
 309 /* slowest but generic code. */
 310 static void vnc_convert_pixel(VncState *vs, uint8_t *buf, uint32_t v)
 311 {
 312     unsigned int r, g, b;
 313
 314     r = (v >> vs->red_shift1) & vs->red_max;
 315     g = (v >> vs->green_shift1) & vs->green_max;
 316     b = (v >> vs->blue_shift1) & vs->blue_max;
 317     v = (r << vs->red_shift) |
 318         (g << vs->green_shift) |
 319         (b << vs->blue_shift);
 320     switch(vs->pix_bpp) {
 321     case 1:
 322         buf[0] = v;
 323         break;
 324     case 2:
 325         if (vs->pix_big_endian) {
 326             buf[0] = v >> 8;
 327             buf[1] = v;
 328         } else {
 329             buf[1] = v >> 8;
 330             buf[0] = v;
 331         }
 332         break;
 333     default:
 334     case 4:
 335         if (vs->pix_big_endian) {
 336             buf[0] = v >> 24;
 337             buf[1] = v >> 16;
 338             buf[2] = v >> 8;
 339             buf[3] = v;
 340         } else {
 341             buf[3] = v >> 24;
 342             buf[2] = v >> 16;
 343             buf[1] = v >> 8;
 344             buf[0] = v;
 345         }
 346         break;
 347     }
 348 }
 349
 350 static void vnc_write_pixels_generic(VncState *vs, void *pixels1, int size)
 351 {
 352     uint32_t *pixels = pixels1;
 353     uint8_t buf[4];
 354     int n, i;
 355
 356     n = size >> 2;
 357     for(i = 0; i < n; i++) {
 358         vnc_convert_pixel(vs, buf, pixels[i]);
 359         vnc_write(vs, buf, vs->pix_bpp);
 360     }
 361 }
 362
 363 static void send_framebuffer_update_raw(VncState *vs, int x, int y, int w, int h)
 364 {
 365     int i;
 366     char *row;
 367
 368     vnc_framebuffer_update(vs, x, y, w, h, 0);
 369
 370     row = vs->ds->data + y * vs->ds->linesize + x * vs->depth;
 371     for (i = 0; i < h; i++) {
 372         vs->write_pixels(vs, row, w * vs->depth);
 373         row += vs->ds->linesize;
 374     }
 375 }
 376
 377 static void hextile_enc_cord(uint8_t *ptr, int x, int y, int w, int h)
 378 {
 379     ptr[0] = ((x & 0x0F) << 4) | (y & 0x0F);
 380     ptr[1] = (((w - 1) & 0x0F) << 4) | ((h - 1) & 0x0F);
 381 }
 382
 383 #define BPP 8
 384 #include "vnchextile.h"
 385 #undef BPP
 386
 387 #define BPP 16
 388 #include "vnchextile.h"
 389 #undef BPP
 390
 391 #define BPP 32
 392 #include "vnchextile.h"
 393 #undef BPP
 394
 395 #define GENERIC
 396 #define BPP 32
 397 #include "vnchextile.h"
 398 #undef BPP
 399 #undef GENERIC
 400
 401 static void send_framebuffer_update_hextile(VncState *vs, int x, int y, int w, int h)
 402 {
 403     int i, j;
 404     int has_fg, has_bg;
 405     uint32_t last_fg32, last_bg32;
 406
 407     vnc_framebuffer_update(vs, x, y, w, h, 5);
 408
 409     has_fg = has_bg = 0;
 410     for (j = y; j < (y + h); j += 16) {
 411         for (i = x; i < (x + w); i += 16) {
 412             vs->send_hextile_tile(vs, i, j,
 413                                   MIN(16, x + w - i), MIN(16, y + h - j),
 414                                   &last_bg32, &last_fg32, &has_bg, &has_fg);
 415         }
 416     }
 417 }
 418
 419 static void send_framebuffer_update(VncState *vs, int x, int y, int w, int h)
 420 {
 421         if (vs->has_hextile)
 422             send_framebuffer_update_hextile(vs, x, y, w, h);
 423         else
 424             send_framebuffer_update_raw(vs, x, y, w, h);
 425 }
 426
 427 static void vnc_copy(DisplayState *ds, int src_x, int src_y, int dst_x, int dst_y, int w, int h)
 428 {
 429     int src, dst;
 430     char *src_row;
 431     char *dst_row;
 432     char *old_row;
 433     int y = 0;
 434     int pitch = ds->linesize;
 435     VncState *vs = ds->opaque;
 436
 437     vnc_update_client(vs);
 438
 439     if (dst_y > src_y) {
 440         y = h - 1;
 441         pitch = -pitch;
 442     }
 443
 444     src = (ds->linesize * (src_y + y) + vs->depth * src_x);
 445     dst = (ds->linesize * (dst_y + y) + vs->depth * dst_x);
 446
 447     src_row = ds->data + src;
 448     dst_row = ds->data + dst;
 449     old_row = vs->old_data + dst;
 450
 451     for (y = 0; y < h; y++) {
 452         memmove(old_row, src_row, w * vs->depth);
 453         memmove(dst_row, src_row, w * vs->depth);
 454         src_row += pitch;
 455         dst_row += pitch;
 456         old_row += pitch;
 457     }
 458
 459     vnc_write_u8(vs, 0);  /* msg id */
 460     vnc_write_u8(vs, 0);
 461     vnc_write_u16(vs, 1); /* number of rects */
 462     vnc_framebuffer_update(vs, dst_x, dst_y, w, h, 1);
 463     vnc_write_u16(vs, src_x);
 464     vnc_write_u16(vs, src_y);
 465     vnc_flush(vs);
 466 }
 467
 468 static int find_dirty_height(VncState *vs, int y, int last_x, int x)
 469 {
 470     int h;
 471
 472     for (h = 1; h < (vs->height - y); h++) {
 473         int tmp_x;
 474         if (!vnc_get_bit(vs->dirty_row[y + h], last_x))
 475             break;
 476         for (tmp_x = last_x; tmp_x < x; tmp_x++)
 477             vnc_clear_bit(vs->dirty_row[y + h], tmp_x);
 478     }
 479
 480     return h;
 481 }
 482
 483 static void vnc_update_client(void *opaque)
 484 {
 485     VncState *vs = opaque;
 486
 487     if (vs->need_update && vs->csock != -1) {
 488         int y;
 489         char *row;
 490         char *old_row;
 491         uint32_t width_mask[VNC_DIRTY_WORDS];
 492         int n_rectangles;
 493         int saved_offset;
 494         int has_dirty = 0;
 495
 496         vnc_set_bits(width_mask, (vs->width / 16), VNC_DIRTY_WORDS);
 497
 498         /* Walk through the dirty map and eliminate tiles that
 499            really aren't dirty */
 500         row = vs->ds->data;
 501         old_row = vs->old_data;
 502
 503         for (y = 0; y < vs->height; y++) {
 504             if (vnc_and_bits(vs->dirty_row[y], width_mask, VNC_DIRTY_WORDS)) {
 505                 int x;
 506                 char *ptr, *old_ptr;
 507
 508                 ptr = row;
 509                 old_ptr = old_row;
 510
 511                 for (x = 0; x < vs->ds->width; x += 16) {
 512                     if (memcmp(old_ptr, ptr, 16 * vs->depth) == 0) {
 513                         vnc_clear_bit(vs->dirty_row[y], (x / 16));
 514                     } else {
 515                         has_dirty = 1;
 516                         memcpy(old_ptr, ptr, 16 * vs->depth);
 517                     }
 518
 519                     ptr += 16 * vs->depth;
 520                     old_ptr += 16 * vs->depth;
 521                 }
 522             }
 523
 524             row += vs->ds->linesize;
 525             old_row += vs->ds->linesize;
 526         }
 527
 528         if (!has_dirty) {
 529             qemu_mod_timer(vs->timer, qemu_get_clock(rt_clock) + VNC_REFRESH_INTERVAL);
 530             return;
 531         }
 532
 533         /* Count rectangles */
 534         n_rectangles = 0;
 535         vnc_write_u8(vs, 0);  /* msg id */
 536         vnc_write_u8(vs, 0);
 537         saved_offset = vs->output.offset;
 538         vnc_write_u16(vs, 0);
 539
 540         for (y = 0; y < vs->height; y++) {
 541             int x;
 542             int last_x = -1;
 543             for (x = 0; x < vs->width / 16; x++) {
 544                 if (vnc_get_bit(vs->dirty_row[y], x)) {
 545                     if (last_x == -1) {
 546                         last_x = x;
 547                     }
 548                     vnc_clear_bit(vs->dirty_row[y], x);
 549                 } else {
 550                     if (last_x != -1) {
 551                         int h = find_dirty_height(vs, y, last_x, x);
 552                         send_framebuffer_update(vs, last_x * 16, y, (x - last_x) * 16, h);
 553                         n_rectangles++;
 554                     }
 555                     last_x = -1;
 556                 }
 557             }
 558             if (last_x != -1) {
 559                 int h = find_dirty_height(vs, y, last_x, x);
 560                 send_framebuffer_update(vs, last_x * 16, y, (x - last_x) * 16, h);
 561                 n_rectangles++;
 562             }
 563         }
 564         vs->output.buffer[saved_offset] = (n_rectangles >> 8) & 0xFF;
 565         vs->output.buffer[saved_offset + 1] = n_rectangles & 0xFF;
 566         vnc_flush(vs);
 567
 568     }
 569     qemu_mod_timer(vs->timer, qemu_get_clock(rt_clock) + VNC_REFRESH_INTERVAL);
 570 }
 571
 572 static void vnc_timer_init(VncState *vs)
 573 {
 574     if (vs->timer == NULL) {
 575         vs->timer = qemu_new_timer(rt_clock, vnc_update_client, vs);
 576         qemu_mod_timer(vs->timer, qemu_get_clock(rt_clock));
 577     }
 578 }
 579
 580 static void vnc_dpy_refresh(DisplayState *ds)
 581 {
 582     VncState *vs = ds->opaque;
 583     vnc_timer_init(vs);
 584     vga_hw_update();
 585 }
 586
 587 static int vnc_listen_poll(void *opaque)
 588 {
 589     VncState *vs = opaque;
 590     if (vs->csock == -1)
 591         return 1;
 592     return 0;
 593 }
 594
 595 static void buffer_reserve(Buffer *buffer, size_t len)
 596 {
 597     if ((buffer->capacity - buffer->offset) < len) {
 598         buffer->capacity += (len + 1024);
 599         buffer->buffer = realloc(buffer->buffer, buffer->capacity);
 600         if (buffer->buffer == NULL) {
 601             fprintf(stderr, "vnc: out of memory\n");
 602             exit(1);
 603         }
 604     }
 605 }
 606
 607 static int buffer_empty(Buffer *buffer)
 608 {
 609     return buffer->offset == 0;
 610 }
 611
 612 static char *buffer_end(Buffer *buffer)
 613 {
 614     return buffer->buffer + buffer->offset;
 615 }
 616
 617 static void buffer_reset(Buffer *buffer)
 618 {
 619         buffer->offset = 0;
 620 }
 621
 622 static void buffer_append(Buffer *buffer, const void *data, size_t len)
 623 {
 624     memcpy(buffer->buffer + buffer->offset, data, len);
 625     buffer->offset += len;
 626 }
 627
 628 static int vnc_client_io_error(VncState *vs, int ret, int last_errno)
 629 {
 630     if (ret == 0 || ret == -1) {
 631         if (ret == -1 && (last_errno == EINTR || last_errno == EAGAIN))
 632             return 0;
 633
 634         VNC_DEBUG("Closing down client sock %d %d\n", ret, ret < 0 ? last_errno : 0);
 635         qemu_set_fd_handler2(vs->csock, NULL, NULL, NULL, NULL);
 636         closesocket(vs->csock);
 637         vs->csock = -1;
 638         buffer_reset(&vs->input);
 639         buffer_reset(&vs->output);
 640         vs->need_update = 0;
 641 #if CONFIG_VNC_TLS
 642         if (vs->tls_session) {
 643             gnutls_deinit(vs->tls_session);
 644             vs->tls_session = NULL;
 645         }
 646         vs->wiremode = VNC_WIREMODE_CLEAR;
 647 #endif /* CONFIG_VNC_TLS */
 648         return 0;
 649     }
 650     return ret;
 651 }
 652
 653 static void vnc_client_error(VncState *vs)
 654 {
 655     vnc_client_io_error(vs, -1, EINVAL);
 656 }
 657
 658 static void vnc_client_write(void *opaque)
 659 {
 660     long ret;
 661     VncState *vs = opaque;
 662
 663 #if CONFIG_VNC_TLS
 664     if (vs->tls_session) {
 665         ret = gnutls_write(vs->tls_session, vs->output.buffer, vs->output.offset);
 666         if (ret < 0) {
 667             if (ret == GNUTLS_E_AGAIN)
 668                 errno = EAGAIN;
 669             else
 670                 errno = EIO;
 671             ret = -1;
 672         }
 673     } else
 674 #endif /* CONFIG_VNC_TLS */
 675         ret = send(vs->csock, vs->output.buffer, vs->output.offset, 0);
 676     ret = vnc_client_io_error(vs, ret, socket_error());
 677     if (!ret)
 678         return;
 679
 680     memmove(vs->output.buffer, vs->output.buffer + ret, (vs->output.offset - ret));
 681     vs->output.offset -= ret;
 682
 683     if (vs->output.offset == 0) {
 684         qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, vs);
 685     }
 686 }
 687
 688 static void vnc_read_when(VncState *vs, VncReadEvent *func, size_t expecting)
 689 {
 690     vs->read_handler = func;
 691     vs->read_handler_expect = expecting;
 692 }
 693
 694 static void vnc_client_read(void *opaque)
 695 {
 696     VncState *vs = opaque;
 697     long ret;
 698
 699     buffer_reserve(&vs->input, 4096);
 700
 701 #if CONFIG_VNC_TLS
 702     if (vs->tls_session) {
 703         ret = gnutls_read(vs->tls_session, buffer_end(&vs->input), 4096);
 704         if (ret < 0) {
 705             if (ret == GNUTLS_E_AGAIN)
 706                 errno = EAGAIN;
 707             else
 708                 errno = EIO;
 709             ret = -1;
 710         }
 711     } else
 712 #endif /* CONFIG_VNC_TLS */
 713         ret = recv(vs->csock, buffer_end(&vs->input), 4096, 0);
 714     ret = vnc_client_io_error(vs, ret, socket_error());
 715     if (!ret)
 716         return;
 717
 718     vs->input.offset += ret;
 719
 720     while (vs->read_handler && vs->input.offset >= vs->read_handler_expect) {
 721         size_t len = vs->read_handler_expect;
 722         int ret;
 723
 724         ret = vs->read_handler(vs, vs->input.buffer, len);
 725         if (vs->csock == -1)
 726             return;
 727
 728         if (!ret) {
 729             memmove(vs->input.buffer, vs->input.buffer + len, (vs->input.offset - len));
 730             vs->input.offset -= len;
 731         } else {
 732             vs->read_handler_expect = ret;
 733         }
 734     }
 735 }
 736
 737 static void vnc_write(VncState *vs, const void *data, size_t len)
 738 {
 739     buffer_reserve(&vs->output, len);
 740
 741     if (buffer_empty(&vs->output)) {
 742         qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, vnc_client_write, vs);
 743     }
 744
 745     buffer_append(&vs->output, data, len);
 746 }
 747
 748 static void vnc_write_s32(VncState *vs, int32_t value)
 749 {
 750     vnc_write_u32(vs, *(uint32_t *)&value);
 751 }
 752
 753 static void vnc_write_u32(VncState *vs, uint32_t value)
 754 {
 755     uint8_t buf[4];
 756
 757     buf[0] = (value >> 24) & 0xFF;
 758     buf[1] = (value >> 16) & 0xFF;
 759     buf[2] = (value >>  8) & 0xFF;
 760     buf[3] = value & 0xFF;
 761
 762     vnc_write(vs, buf, 4);
 763 }
 764
 765 static void vnc_write_u16(VncState *vs, uint16_t value)
 766 {
 767     uint8_t buf[2];
 768
 769     buf[0] = (value >> 8) & 0xFF;
 770     buf[1] = value & 0xFF;
 771
 772     vnc_write(vs, buf, 2);
 773 }
 774
 775 static void vnc_write_u8(VncState *vs, uint8_t value)
 776 {
 777     vnc_write(vs, (char *)&value, 1);
 778 }
 779
 780 static void vnc_flush(VncState *vs)
 781 {
 782     if (vs->output.offset)
 783         vnc_client_write(vs);
 784 }
 785
 786 static uint8_t read_u8(uint8_t *data, size_t offset)
 787 {
 788     return data[offset];
 789 }
 790
 791 static uint16_t read_u16(uint8_t *data, size_t offset)
 792 {
 793     return ((data[offset] & 0xFF) << 8) | (data[offset + 1] & 0xFF);
 794 }
 795
 796 static int32_t read_s32(uint8_t *data, size_t offset)
 797 {
 798     return (int32_t)((data[offset] << 24) | (data[offset + 1] << 16) |
 799                      (data[offset + 2] << 8) | data[offset + 3]);
 800 }
 801
 802 static uint32_t read_u32(uint8_t *data, size_t offset)
 803 {
 804     return ((data[offset] << 24) | (data[offset + 1] << 16) |
 805             (data[offset + 2] << 8) | data[offset + 3]);
 806 }
 807
 808 #if CONFIG_VNC_TLS
 809 ssize_t vnc_tls_push(gnutls_transport_ptr_t transport,
 810                      const void *data,
 811                      size_t len) {
 812     struct VncState *vs = (struct VncState *)transport;
 813     int ret;
 814
 815  retry:
 816     ret = send(vs->csock, data, len, 0);
 817     if (ret < 0) {
 818         if (errno == EINTR)
 819             goto retry;
 820         return -1;
 821     }
 822     return ret;
 823 }
 824
 825
 826 ssize_t vnc_tls_pull(gnutls_transport_ptr_t transport,
 827                      void *data,
 828                      size_t len) {
 829     struct VncState *vs = (struct VncState *)transport;
 830     int ret;
 831
 832  retry:
 833     ret = recv(vs->csock, data, len, 0);
 834     if (ret < 0) {
 835         if (errno == EINTR)
 836             goto retry;
 837         return -1;
 838     }
 839     return ret;
 840 }
 841 #endif /* CONFIG_VNC_TLS */
 842
 843 static void client_cut_text(VncState *vs, size_t len, char *text)
 844 {
 845 }
 846
 847 static void check_pointer_type_change(VncState *vs, int absolute)
 848 {
 849     if (vs->has_pointer_type_change && vs->absolute != absolute) {
 850         vnc_write_u8(vs, 0);
 851         vnc_write_u8(vs, 0);
 852         vnc_write_u16(vs, 1);
 853         vnc_framebuffer_update(vs, absolute, 0,
 854                                vs->ds->width, vs->ds->height, -257);
 855         vnc_flush(vs);
 856     }
 857     vs->absolute = absolute;
 858 }
 859
 860 static void pointer_event(VncState *vs, int button_mask, int x, int y)
 861 {
 862     int buttons = 0;
 863     int dz = 0;
 864
 865     if (button_mask & 0x01)
 866         buttons |= MOUSE_EVENT_LBUTTON;
 867     if (button_mask & 0x02)
 868         buttons |= MOUSE_EVENT_MBUTTON;
 869     if (button_mask & 0x04)
 870         buttons |= MOUSE_EVENT_RBUTTON;
 871     if (button_mask & 0x08)
 872         dz = -1;
 873     if (button_mask & 0x10)
 874         dz = 1;
 875
 876     if (vs->absolute) {
 877         kbd_mouse_event(x * 0x7FFF / vs->ds->width,
 878                         y * 0x7FFF / vs->ds->height,
 879                         dz, buttons);
 880     } else if (vs->has_pointer_type_change) {
 881         x -= 0x7FFF;
 882         y -= 0x7FFF;
 883
 884         kbd_mouse_event(x, y, dz, buttons);
 885     } else {
 886         if (vs->last_x != -1)
 887             kbd_mouse_event(x - vs->last_x,
 888                             y - vs->last_y,
 889                             dz, buttons);
 890         vs->last_x = x;
 891         vs->last_y = y;
 892     }
 893
 894     check_pointer_type_change(vs, kbd_mouse_is_absolute());
 895 }
 896
 897 static void reset_keys(VncState *vs)
 898 {
 899     int i;
 900     for(i = 0; i < 256; i++) {
 901         if (vs->modifiers_state[i]) {
 902             if (i & 0x80)
 903                 kbd_put_keycode(0xe0);
 904             kbd_put_keycode(i | 0x80);
 905             vs->modifiers_state[i] = 0;
 906         }
 907     }
 908 }
 909
 910 static void do_key_event(VncState *vs, int down, uint32_t sym)
 911 {
 912     int keycode;
 913
 914     keycode = keysym2scancode(vs->kbd_layout, sym & 0xFFFF);
 915
 916     /* QEMU console switch */
 917     switch(keycode) {
 918     case 0x2a:                          /* Left Shift */
 919     case 0x36:                          /* Right Shift */
 920     case 0x1d:                          /* Left CTRL */
 921     case 0x9d:                          /* Right CTRL */
 922     case 0x38:                          /* Left ALT */
 923     case 0xb8:                          /* Right ALT */
 924         if (down)
 925             vs->modifiers_state[keycode] = 1;
 926         else
 927             vs->modifiers_state[keycode] = 0;
 928         break;
 929     case 0x02 ... 0x0a: /* '1' to '9' keys */
 930         if (down && vs->modifiers_state[0x1d] && vs->modifiers_state[0x38]) {
 931             /* Reset the modifiers sent to the current console */
 932             reset_keys(vs);
 933             console_select(keycode - 0x02);
 934             return;
 935         }
 936         break;
 937     }
 938
 939     if (is_graphic_console()) {
 940         if (keycode & 0x80)
 941             kbd_put_keycode(0xe0);
 942         if (down)
 943             kbd_put_keycode(keycode & 0x7f);
 944         else
 945             kbd_put_keycode(keycode | 0x80);
 946     } else {
 947         /* QEMU console emulation */
 948         if (down) {
 949             switch (keycode) {
 950             case 0x2a:                          /* Left Shift */
 951             case 0x36:                          /* Right Shift */
 952             case 0x1d:                          /* Left CTRL */
 953             case 0x9d:                          /* Right CTRL */
 954             case 0x38:                          /* Left ALT */
 955             case 0xb8:                          /* Right ALT */
 956                 break;
 957             case 0xc8:
 958                 kbd_put_keysym(QEMU_KEY_UP);
 959                 break;
 960             case 0xd0:
 961                 kbd_put_keysym(QEMU_KEY_DOWN);
 962                 break;
 963             case 0xcb:
 964                 kbd_put_keysym(QEMU_KEY_LEFT);
 965                 break;
 966             case 0xcd:
 967                 kbd_put_keysym(QEMU_KEY_RIGHT);
 968                 break;
 969             case 0xd3:
 970                 kbd_put_keysym(QEMU_KEY_DELETE);
 971                 break;
 972             case 0xc7:
 973                 kbd_put_keysym(QEMU_KEY_HOME);
 974                 break;
 975             case 0xcf:
 976                 kbd_put_keysym(QEMU_KEY_END);
 977                 break;
 978             case 0xc9:
 979                 kbd_put_keysym(QEMU_KEY_PAGEUP);
 980                 break;
 981             case 0xd1:
 982                 kbd_put_keysym(QEMU_KEY_PAGEDOWN);
 983                 break;
 984             default:
 985                 kbd_put_keysym(sym);
 986                 break;
 987             }
 988         }
 989     }
 990 }
 991
 992 static void key_event(VncState *vs, int down, uint32_t sym)
 993 {
 994     if (sym >= 'A' && sym <= 'Z')
 995         sym = sym - 'A' + 'a';
 996     do_key_event(vs, down, sym);
 997 }
 998
 999 static void framebuffer_update_request(VncState *vs, int incremental,
1000                                        int x_position, int y_position,
1001                                        int w, int h)
1002 {
1003     if (x_position > vs->ds->width)
1004         x_position = vs->ds->width;
1005     if (y_position > vs->ds->height)
1006         y_position = vs->ds->height;
1007     if (x_position + w >= vs->ds->width)
1008         w = vs->ds->width  - x_position;
1009     if (y_position + h >= vs->ds->height)
1010         h = vs->ds->height - y_position;
1011
1012     int i;
1013     vs->need_update = 1;
1014     if (!incremental) {
1015         char *old_row = vs->old_data + y_position * vs->ds->linesize;
1016
1017         for (i = 0; i < h; i++) {
1018             vnc_set_bits(vs->dirty_row[y_position + i],
1019                          (vs->ds->width / 16), VNC_DIRTY_WORDS);
1020             memset(old_row, 42, vs->ds->width * vs->depth);
1021             old_row += vs->ds->linesize;
1022         }
1023     }
1024 }
1025
1026 static void set_encodings(VncState *vs, int32_t *encodings, size_t n_encodings)
1027 {
1028     int i;
1029
1030     vs->has_hextile = 0;
1031     vs->has_resize = 0;
1032     vs->has_pointer_type_change = 0;
1033     vs->absolute = -1;
1034     vs->ds->dpy_copy = NULL;
1035
1036     for (i = n_encodings - 1; i >= 0; i--) {
1037         switch (encodings[i]) {
1038         case 0: /* Raw */
1039             vs->has_hextile = 0;
1040             break;
1041         case 1: /* CopyRect */
1042             vs->ds->dpy_copy = vnc_copy;
1043             break;
1044         case 5: /* Hextile */
1045             vs->has_hextile = 1;
1046             break;
1047         case -223: /* DesktopResize */
1048             vs->has_resize = 1;
1049             break;
1050         case -257:
1051             vs->has_pointer_type_change = 1;
1052             break;
1053         default:
1054             break;
1055         }
1056     }
1057
1058     check_pointer_type_change(vs, kbd_mouse_is_absolute());
1059 }
1060
1061 static int compute_nbits(unsigned int val)
1062 {
1063     int n;
1064     n = 0;
1065     while (val != 0) {
1066         n++;
1067         val >>= 1;
1068     }
1069     return n;
1070 }
1071
1072 static void set_pixel_format(VncState *vs,
1073                              int bits_per_pixel, int depth,
1074                              int big_endian_flag, int true_color_flag,
1075                              int red_max, int green_max, int blue_max,
1076                              int red_shift, int green_shift, int blue_shift)
1077 {
1078     int host_big_endian_flag;
1079
1080 #ifdef WORDS_BIGENDIAN
1081     host_big_endian_flag = 1;
1082 #else
1083     host_big_endian_flag = 0;
1084 #endif
1085     if (!true_color_flag) {
1086     fail:
1087         vnc_client_error(vs);
1088         return;
1089     }
1090     if (bits_per_pixel == 32 &&
1091         host_big_endian_flag == big_endian_flag &&
1092         red_max == 0xff && green_max == 0xff && blue_max == 0xff &&
1093         red_shift == 16 && green_shift == 8 && blue_shift == 0) {
1094         vs->depth = 4;
1095         vs->write_pixels = vnc_write_pixels_copy;
1096         vs->send_hextile_tile = send_hextile_tile_32;
1097     } else
1098     if (bits_per_pixel == 16 &&
1099         host_big_endian_flag == big_endian_flag &&
1100         red_max == 31 && green_max == 63 && blue_max == 31 &&
1101         red_shift == 11 && green_shift == 5 && blue_shift == 0) {
1102         vs->depth = 2;
1103         vs->write_pixels = vnc_write_pixels_copy;
1104         vs->send_hextile_tile = send_hextile_tile_16;
1105     } else
1106     if (bits_per_pixel == 8 &&
1107         red_max == 7 && green_max == 7 && blue_max == 3 &&
1108         red_shift == 5 && green_shift == 2 && blue_shift == 0) {
1109         vs->depth = 1;
1110         vs->write_pixels = vnc_write_pixels_copy;
1111         vs->send_hextile_tile = send_hextile_tile_8;
1112     } else
1113     {
1114         /* generic and slower case */
1115         if (bits_per_pixel != 8 &&
1116             bits_per_pixel != 16 &&
1117             bits_per_pixel != 32)
1118             goto fail;
1119         vs->depth = 4;
1120         vs->red_shift = red_shift;
1121         vs->red_max = red_max;
1122         vs->red_shift1 = 24 - compute_nbits(red_max);
1123         vs->green_shift = green_shift;
1124         vs->green_max = green_max;
1125         vs->green_shift1 = 16 - compute_nbits(green_max);
1126         vs->blue_shift = blue_shift;
1127         vs->blue_max = blue_max;
1128         vs->blue_shift1 = 8 - compute_nbits(blue_max);
1129         vs->pix_bpp = bits_per_pixel / 8;
1130         vs->pix_big_endian = big_endian_flag;
1131         vs->write_pixels = vnc_write_pixels_generic;
1132         vs->send_hextile_tile = send_hextile_tile_generic;
1133     }
1134
1135     vnc_dpy_resize(vs->ds, vs->ds->width, vs->ds->height);
1136     memset(vs->dirty_row, 0xFF, sizeof(vs->dirty_row));
1137     memset(vs->old_data, 42, vs->ds->linesize * vs->ds->height);
1138
1139     vga_hw_invalidate();
1140     vga_hw_update();
1141 }
1142
1143 static int protocol_client_msg(VncState *vs, char *data, size_t len)
1144 {
1145     int i;
1146     uint16_t limit;
1147
1148     switch (data[0]) {
1149     case 0:
1150         if (len == 1)
1151             return 20;
1152
1153         set_pixel_format(vs, read_u8(data, 4), read_u8(data, 5),
1154                          read_u8(data, 6), read_u8(data, 7),
1155                          read_u16(data, 8), read_u16(data, 10),
1156                          read_u16(data, 12), read_u8(data, 14),
1157                          read_u8(data, 15), read_u8(data, 16));
1158         break;
1159     case 2:
1160         if (len == 1)
1161             return 4;
1162
1163         if (len == 4)
1164             return 4 + (read_u16(data, 2) * 4);
1165
1166         limit = read_u16(data, 2);
1167         for (i = 0; i < limit; i++) {
1168             int32_t val = read_s32(data, 4 + (i * 4));
1169             memcpy(data + 4 + (i * 4), &val, sizeof(val));
1170         }
1171
1172         set_encodings(vs, (int32_t *)(data + 4), limit);
1173         break;
1174     case 3:
1175         if (len == 1)
1176             return 10;
1177
1178         framebuffer_update_request(vs,
1179                                    read_u8(data, 1), read_u16(data, 2), read_u16(data, 4),
1180                                    read_u16(data, 6), read_u16(data, 8));
1181         break;
1182     case 4:
1183         if (len == 1)
1184             return 8;
1185
1186         key_event(vs, read_u8(data, 1), read_u32(data, 4));
1187         break;
1188     case 5:
1189         if (len == 1)
1190             return 6;
1191
1192         pointer_event(vs, read_u8(data, 1), read_u16(data, 2), read_u16(data, 4));
1193         break;
1194     case 6:
1195         if (len == 1)
1196             return 8;
1197
1198         if (len == 8) {
1199             uint32_t dlen = read_u32(data, 4);
1200             if (dlen > 0)
1201                 return 8 + dlen;
1202         }
1203
1204         client_cut_text(vs, read_u32(data, 4), data + 8);
1205         break;
1206     default:
1207         printf("Msg: %d\n", data[0]);
1208         vnc_client_error(vs);
1209         break;
1210     }
1211
1212     vnc_read_when(vs, protocol_client_msg, 1);
1213     return 0;
1214 }
1215
1216 static int protocol_client_init(VncState *vs, char *data, size_t len)
1217 {
1218     char pad[3] = { 0, 0, 0 };
1219     char buf[1024];
1220     int size;
1221
1222     vs->width = vs->ds->width;
1223     vs->height = vs->ds->height;
1224     vnc_write_u16(vs, vs->ds->width);
1225     vnc_write_u16(vs, vs->ds->height);
1226
1227     vnc_write_u8(vs, vs->depth * 8); /* bits-per-pixel */
1228     vnc_write_u8(vs, vs->depth * 8); /* depth */
1229 #ifdef WORDS_BIGENDIAN
1230     vnc_write_u8(vs, 1);             /* big-endian-flag */
1231 #else
1232     vnc_write_u8(vs, 0);             /* big-endian-flag */
1233 #endif
1234     vnc_write_u8(vs, 1);             /* true-color-flag */
1235     if (vs->depth == 4) {
1236         vnc_write_u16(vs, 0xFF);     /* red-max */
1237         vnc_write_u16(vs, 0xFF);     /* green-max */
1238         vnc_write_u16(vs, 0xFF);     /* blue-max */
1239         vnc_write_u8(vs, 16);        /* red-shift */
1240         vnc_write_u8(vs, 8);         /* green-shift */
1241         vnc_write_u8(vs, 0);         /* blue-shift */
1242         vs->send_hextile_tile = send_hextile_tile_32;
1243     } else if (vs->depth == 2) {
1244         vnc_write_u16(vs, 31);       /* red-max */
1245         vnc_write_u16(vs, 63);       /* green-max */
1246         vnc_write_u16(vs, 31);       /* blue-max */
1247         vnc_write_u8(vs, 11);        /* red-shift */
1248         vnc_write_u8(vs, 5);         /* green-shift */
1249         vnc_write_u8(vs, 0);         /* blue-shift */
1250         vs->send_hextile_tile = send_hextile_tile_16;
1251     } else if (vs->depth == 1) {
1252         /* XXX: change QEMU pixel 8 bit pixel format to match the VNC one ? */
1253         vnc_write_u16(vs, 7);        /* red-max */
1254         vnc_write_u16(vs, 7);        /* green-max */
1255         vnc_write_u16(vs, 3);        /* blue-max */
1256         vnc_write_u8(vs, 5);         /* red-shift */
1257         vnc_write_u8(vs, 2);         /* green-shift */
1258         vnc_write_u8(vs, 0);         /* blue-shift */
1259         vs->send_hextile_tile = send_hextile_tile_8;
1260     }
1261     vs->write_pixels = vnc_write_pixels_copy;
1262
1263     vnc_write(vs, pad, 3);           /* padding */
1264
1265     if (qemu_name)
1266         size = snprintf(buf, sizeof(buf), "QEMU (%s)", qemu_name);
1267     else
1268         size = snprintf(buf, sizeof(buf), "QEMU");
1269
1270     vnc_write_u32(vs, size);
1271     vnc_write(vs, buf, size);
1272     vnc_flush(vs);
1273
1274     vnc_read_when(vs, protocol_client_msg, 1);
1275
1276     return 0;
1277 }
1278
1279 static void make_challenge(VncState *vs)
1280 {
1281     int i;
1282
1283     srand(time(NULL)+getpid()+getpid()*987654+rand());
1284
1285     for (i = 0 ; i < sizeof(vs->challenge) ; i++)
1286         vs->challenge[i] = (int) (256.0*rand()/(RAND_MAX+1.0));
1287 }
1288
1289 static int protocol_client_auth_vnc(VncState *vs, char *data, size_t len)
1290 {
1291     char response[VNC_AUTH_CHALLENGE_SIZE];
1292     int i, j, pwlen;
1293     char key[8];
1294
1295     if (!vs->password || !vs->password[0]) {
1296         VNC_DEBUG("No password configured on server");
1297         vnc_write_u32(vs, 1); /* Reject auth */
1298         if (vs->minor >= 8) {
1299             static const char err[] = "Authentication failed";
1300             vnc_write_u32(vs, sizeof(err));
1301             vnc_write(vs, err, sizeof(err));
1302         }
1303         vnc_flush(vs);
1304         vnc_client_error(vs);
1305         return 0;
1306     }
1307
1308     memcpy(response, vs->challenge, VNC_AUTH_CHALLENGE_SIZE);
1309
1310     /* Calculate the expected challenge response */
1311     pwlen = strlen(vs->password);
1312     for (i=0; i<sizeof(key); i++)
1313         key[i] = i<pwlen ? vs->password[i] : 0;
1314     deskey(key, EN0);
1315     for (j = 0; j < VNC_AUTH_CHALLENGE_SIZE; j += 8)
1316         des(response+j, response+j);
1317
1318     /* Compare expected vs actual challenge response */
1319     if (memcmp(response, data, VNC_AUTH_CHALLENGE_SIZE) != 0) {
1320         VNC_DEBUG("Client challenge reponse did not match\n");
1321         vnc_write_u32(vs, 1); /* Reject auth */
1322         if (vs->minor >= 8) {
1323             static const char err[] = "Authentication failed";
1324             vnc_write_u32(vs, sizeof(err));
1325             vnc_write(vs, err, sizeof(err));
1326         }
1327         vnc_flush(vs);
1328         vnc_client_error(vs);
1329     } else {
1330         VNC_DEBUG("Accepting VNC challenge response\n");
1331         vnc_write_u32(vs, 0); /* Accept auth */
1332         vnc_flush(vs);
1333
1334         vnc_read_when(vs, protocol_client_init, 1);
1335     }
1336     return 0;
1337 }
1338
1339 static int start_auth_vnc(VncState *vs)
1340 {
1341     make_challenge(vs);
1342     /* Send client a 'random' challenge */
1343     vnc_write(vs, vs->challenge, sizeof(vs->challenge));
1344     vnc_flush(vs);
1345
1346     vnc_read_when(vs, protocol_client_auth_vnc, sizeof(vs->challenge));
1347     return 0;
1348 }
1349
1350
1351 #if CONFIG_VNC_TLS
1352 #define DH_BITS 1024
1353 static gnutls_dh_params_t dh_params;
1354
1355 static int vnc_tls_initialize(void)
1356 {
1357     static int tlsinitialized = 0;
1358
1359     if (tlsinitialized)
1360         return 1;
1361
1362     if (gnutls_global_init () < 0)
1363         return 0;
1364
1365     /* XXX ought to re-generate diffie-hellmen params periodically */
1366     if (gnutls_dh_params_init (&dh_params) < 0)
1367         return 0;
1368     if (gnutls_dh_params_generate2 (dh_params, DH_BITS) < 0)
1369         return 0;
1370
1371 #if _VNC_DEBUG == 2
1372     gnutls_global_set_log_level(10);
1373     gnutls_global_set_log_function(vnc_debug_gnutls_log);
1374 #endif
1375
1376     tlsinitialized = 1;
1377
1378     return 1;
1379 }
1380
1381 static gnutls_anon_server_credentials vnc_tls_initialize_anon_cred(void)
1382 {
1383     gnutls_anon_server_credentials anon_cred;
1384     int ret;
1385
1386     if ((ret = gnutls_anon_allocate_server_credentials(&anon_cred)) < 0) {
1387         VNC_DEBUG("Cannot allocate credentials %s\n", gnutls_strerror(ret));
1388         return NULL;
1389     }
1390
1391     gnutls_anon_set_server_dh_params(anon_cred, dh_params);
1392
1393     return anon_cred;
1394 }
1395
1396
1397 static gnutls_certificate_credentials_t vnc_tls_initialize_x509_cred(VncState *vs)
1398 {
1399     gnutls_certificate_credentials_t x509_cred;
1400     int ret;
1401
1402     if (!vs->x509cacert) {
1403         VNC_DEBUG("No CA x509 certificate specified\n");
1404         return NULL;
1405     }
1406     if (!vs->x509cert) {
1407         VNC_DEBUG("No server x509 certificate specified\n");
1408         return NULL;
1409     }
1410     if (!vs->x509key) {
1411         VNC_DEBUG("No server private key specified\n");
1412         return NULL;
1413     }
1414
1415     if ((ret = gnutls_certificate_allocate_credentials(&x509_cred)) < 0) {
1416         VNC_DEBUG("Cannot allocate credentials %s\n", gnutls_strerror(ret));
1417         return NULL;
1418     }
1419     if ((ret = gnutls_certificate_set_x509_trust_file(x509_cred,
1420                                                       vs->x509cacert,
1421                                                       GNUTLS_X509_FMT_PEM)) < 0) {
1422         VNC_DEBUG("Cannot load CA certificate %s\n", gnutls_strerror(ret));
1423         gnutls_certificate_free_credentials(x509_cred);
1424         return NULL;
1425     }
1426
1427     if ((ret = gnutls_certificate_set_x509_key_file (x509_cred,
1428                                                      vs->x509cert,
1429                                                      vs->x509key,
1430                                                      GNUTLS_X509_FMT_PEM)) < 0) {
1431         VNC_DEBUG("Cannot load certificate & key %s\n", gnutls_strerror(ret));
1432         gnutls_certificate_free_credentials(x509_cred);
1433         return NULL;
1434     }
1435
1436     if (vs->x509cacrl) {
1437         if ((ret = gnutls_certificate_set_x509_crl_file(x509_cred,
1438                                                         vs->x509cacrl,
1439                                                         GNUTLS_X509_FMT_PEM)) < 0) {
1440             VNC_DEBUG("Cannot load CRL %s\n", gnutls_strerror(ret));
1441             gnutls_certificate_free_credentials(x509_cred);
1442             return NULL;
1443         }
1444     }
1445
1446     gnutls_certificate_set_dh_params (x509_cred, dh_params);
1447
1448     return x509_cred;
1449 }
1450
1451 static int vnc_validate_certificate(struct VncState *vs)
1452 {
1453     int ret;
1454     unsigned int status;
1455     const gnutls_datum_t *certs;
1456     unsigned int nCerts, i;
1457     time_t now;
1458
1459     VNC_DEBUG("Validating client certificate\n");
1460     if ((ret = gnutls_certificate_verify_peers2 (vs->tls_session, &status)) < 0) {
1461         VNC_DEBUG("Verify failed %s\n", gnutls_strerror(ret));
1462         return -1;
1463     }
1464
1465     if ((now = time(NULL)) == ((time_t)-1)) {
1466         return -1;
1467     }
1468
1469     if (status != 0) {
1470         if (status & GNUTLS_CERT_INVALID)
1471             VNC_DEBUG("The certificate is not trusted.\n");
1472
1473         if (status & GNUTLS_CERT_SIGNER_NOT_FOUND)
1474             VNC_DEBUG("The certificate hasn't got a known issuer.\n");
1475
1476         if (status & GNUTLS_CERT_REVOKED)
1477             VNC_DEBUG("The certificate has been revoked.\n");
1478
1479         if (status & GNUTLS_CERT_INSECURE_ALGORITHM)
1480             VNC_DEBUG("The certificate uses an insecure algorithm\n");
1481
1482         return -1;
1483     } else {
1484         VNC_DEBUG("Certificate is valid!\n");
1485     }
1486
1487     /* Only support x509 for now */
1488     if (gnutls_certificate_type_get(vs->tls_session) != GNUTLS_CRT_X509)
1489         return -1;
1490
1491     if (!(certs = gnutls_certificate_get_peers(vs->tls_session, &nCerts)))
1492         return -1;
1493
1494     for (i = 0 ; i < nCerts ; i++) {
1495         gnutls_x509_crt_t cert;
1496         VNC_DEBUG ("Checking certificate chain %d\n", i);
1497         if (gnutls_x509_crt_init (&cert) < 0)
1498             return -1;
1499
1500         if (gnutls_x509_crt_import(cert, &certs[i], GNUTLS_X509_FMT_DER) < 0) {
1501             gnutls_x509_crt_deinit (cert);
1502             return -1;
1503         }
1504
1505         if (gnutls_x509_crt_get_expiration_time (cert) < now) {
1506             VNC_DEBUG("The certificate has expired\n");
1507             gnutls_x509_crt_deinit (cert);
1508             return -1;
1509         }
1510
1511         if (gnutls_x509_crt_get_activation_time (cert) > now) {
1512             VNC_DEBUG("The certificate is not yet activated\n");
1513             gnutls_x509_crt_deinit (cert);
1514             return -1;
1515         }
1516
1517         if (gnutls_x509_crt_get_activation_time (cert) > now) {
1518             VNC_DEBUG("The certificate is not yet activated\n");
1519             gnutls_x509_crt_deinit (cert);
1520             return -1;
1521         }
1522
1523         gnutls_x509_crt_deinit (cert);
1524     }
1525
1526     return 0;
1527 }
1528
1529
1530 static int start_auth_vencrypt_subauth(VncState *vs)
1531 {
1532     switch (vs->subauth) {
1533     case VNC_AUTH_VENCRYPT_TLSNONE:
1534     case VNC_AUTH_VENCRYPT_X509NONE:
1535        VNC_DEBUG("Accept TLS auth none\n");
1536        vnc_write_u32(vs, 0); /* Accept auth completion */
1537        vnc_read_when(vs, protocol_client_init, 1);
1538        break;
1539
1540     case VNC_AUTH_VENCRYPT_TLSVNC:
1541     case VNC_AUTH_VENCRYPT_X509VNC:
1542        VNC_DEBUG("Start TLS auth VNC\n");
1543        return start_auth_vnc(vs);
1544
1545     default: /* Should not be possible, but just in case */
1546        VNC_DEBUG("Reject auth %d\n", vs->auth);
1547        vnc_write_u8(vs, 1);
1548        if (vs->minor >= 8) {
1549            static const char err[] = "Unsupported authentication type";
1550            vnc_write_u32(vs, sizeof(err));
1551            vnc_write(vs, err, sizeof(err));
1552        }
1553        vnc_client_error(vs);
1554     }
1555
1556     return 0;
1557 }
1558
1559 static void vnc_handshake_io(void *opaque);
1560
1561 static int vnc_continue_handshake(struct VncState *vs) {
1562     int ret;
1563
1564     if ((ret = gnutls_handshake(vs->tls_session)) < 0) {
1565        if (!gnutls_error_is_fatal(ret)) {
1566            VNC_DEBUG("Handshake interrupted (blocking)\n");
1567            if (!gnutls_record_get_direction(vs->tls_session))
1568                qemu_set_fd_handler(vs->csock, vnc_handshake_io, NULL, vs);
1569            else
1570                qemu_set_fd_handler(vs->csock, NULL, vnc_handshake_io, vs);
1571            return 0;
1572        }
1573        VNC_DEBUG("Handshake failed %s\n", gnutls_strerror(ret));
1574        vnc_client_error(vs);
1575        return -1;
1576     }
1577
1578     if (vs->x509verify) {
1579         if (vnc_validate_certificate(vs) < 0) {
1580             VNC_DEBUG("Client verification failed\n");
1581             vnc_client_error(vs);
1582             return -1;
1583         } else {
1584             VNC_DEBUG("Client verification passed\n");
1585         }
1586     }
1587
1588     VNC_DEBUG("Handshake done, switching to TLS data mode\n");
1589     vs->wiremode = VNC_WIREMODE_TLS;
1590     qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, vnc_client_write, vs);
1591
1592     return start_auth_vencrypt_subauth(vs);
1593 }
1594
1595 static void vnc_handshake_io(void *opaque) {
1596     struct VncState *vs = (struct VncState *)opaque;
1597
1598     VNC_DEBUG("Handshake IO continue\n");
1599     vnc_continue_handshake(vs);
1600 }
1601
1602 #define NEED_X509_AUTH(vs)                            \
1603     ((vs)->subauth == VNC_AUTH_VENCRYPT_X509NONE ||   \
1604      (vs)->subauth == VNC_AUTH_VENCRYPT_X509VNC ||    \
1605      (vs)->subauth == VNC_AUTH_VENCRYPT_X509PLAIN)
1606
1607
1608 static int vnc_start_tls(struct VncState *vs) {
1609     static const int cert_type_priority[] = { GNUTLS_CRT_X509, 0 };
1610     static const int protocol_priority[]= { GNUTLS_TLS1_1, GNUTLS_TLS1_0, GNUTLS_SSL3, 0 };
1611     static const int kx_anon[] = {GNUTLS_KX_ANON_DH, 0};
1612     static const int kx_x509[] = {GNUTLS_KX_DHE_DSS, GNUTLS_KX_RSA, GNUTLS_KX_DHE_RSA, GNUTLS_KX_SRP, 0};
1613
1614     VNC_DEBUG("Do TLS setup\n");
1615     if (vnc_tls_initialize() < 0) {
1616         VNC_DEBUG("Failed to init TLS\n");
1617         vnc_client_error(vs);
1618         return -1;
1619     }
1620     if (vs->tls_session == NULL) {
1621         if (gnutls_init(&vs->tls_session, GNUTLS_SERVER) < 0) {
1622             vnc_client_error(vs);
1623             return -1;
1624         }
1625
1626         if (gnutls_set_default_priority(vs->tls_session) < 0) {
1627             gnutls_deinit(vs->tls_session);
1628             vs->tls_session = NULL;
1629             vnc_client_error(vs);
1630             return -1;
1631         }
1632
1633         if (gnutls_kx_set_priority(vs->tls_session, NEED_X509_AUTH(vs) ? kx_x509 : kx_anon) < 0) {
1634             gnutls_deinit(vs->tls_session);
1635             vs->tls_session = NULL;
1636             vnc_client_error(vs);
1637             return -1;
1638         }
1639
1640         if (gnutls_certificate_type_set_priority(vs->tls_session, cert_type_priority) < 0) {
1641             gnutls_deinit(vs->tls_session);
1642             vs->tls_session = NULL;
1643             vnc_client_error(vs);
1644             return -1;
1645         }
1646
1647         if (gnutls_protocol_set_priority(vs->tls_session, protocol_priority) < 0) {
1648             gnutls_deinit(vs->tls_session);
1649             vs->tls_session = NULL;
1650             vnc_client_error(vs);
1651             return -1;
1652         }
1653
1654         if (NEED_X509_AUTH(vs)) {
1655             gnutls_certificate_server_credentials x509_cred = vnc_tls_initialize_x509_cred(vs);
1656             if (!x509_cred) {
1657                 gnutls_deinit(vs->tls_session);
1658                 vs->tls_session = NULL;
1659                 vnc_client_error(vs);
1660                 return -1;
1661             }
1662             if (gnutls_credentials_set(vs->tls_session, GNUTLS_CRD_CERTIFICATE, x509_cred) < 0) {
1663                 gnutls_deinit(vs->tls_session);
1664                 vs->tls_session = NULL;
1665                 gnutls_certificate_free_credentials(x509_cred);
1666                 vnc_client_error(vs);
1667                 return -1;
1668             }
1669             if (vs->x509verify) {
1670                 VNC_DEBUG("Requesting a client certificate\n");
1671                 gnutls_certificate_server_set_request (vs->tls_session, GNUTLS_CERT_REQUEST);
1672             }
1673
1674         } else {
1675             gnutls_anon_server_credentials anon_cred = vnc_tls_initialize_anon_cred();
1676             if (!anon_cred) {
1677                 gnutls_deinit(vs->tls_session);
1678                 vs->tls_session = NULL;
1679                 vnc_client_error(vs);
1680                 return -1;
1681             }
1682             if (gnutls_credentials_set(vs->tls_session, GNUTLS_CRD_ANON, anon_cred) < 0) {
1683                 gnutls_deinit(vs->tls_session);
1684                 vs->tls_session = NULL;
1685                 gnutls_anon_free_server_credentials(anon_cred);
1686                 vnc_client_error(vs);
1687                 return -1;
1688             }
1689         }
1690
1691         gnutls_transport_set_ptr(vs->tls_session, (gnutls_transport_ptr_t)vs);
1692         gnutls_transport_set_push_function(vs->tls_session, vnc_tls_push);
1693         gnutls_transport_set_pull_function(vs->tls_session, vnc_tls_pull);
1694     }
1695
1696     VNC_DEBUG("Start TLS handshake process\n");
1697     return vnc_continue_handshake(vs);
1698 }
1699
1700 static int protocol_client_vencrypt_auth(VncState *vs, char *data, size_t len)
1701 {
1702     int auth = read_u32(data, 0);
1703
1704     if (auth != vs->subauth) {
1705         VNC_DEBUG("Rejecting auth %d\n", auth);
1706         vnc_write_u8(vs, 0); /* Reject auth */
1707         vnc_flush(vs);
1708         vnc_client_error(vs);
1709     } else {
1710         VNC_DEBUG("Accepting auth %d, starting handshake\n", auth);
1711         vnc_write_u8(vs, 1); /* Accept auth */
1712         vnc_flush(vs);
1713
1714         if (vnc_start_tls(vs) < 0) {
1715             VNC_DEBUG("Failed to complete TLS\n");
1716             return 0;
1717         }
1718
1719         if (vs->wiremode == VNC_WIREMODE_TLS) {
1720             VNC_DEBUG("Starting VeNCrypt subauth\n");
1721             return start_auth_vencrypt_subauth(vs);
1722         } else {
1723             VNC_DEBUG("TLS handshake blocked\n");
1724             return 0;
1725         }
1726     }
1727     return 0;
1728 }
1729
1730 static int protocol_client_vencrypt_init(VncState *vs, char *data, size_t len)
1731 {
1732     if (data[0] != 0 ||
1733         data[1] != 2) {
1734         VNC_DEBUG("Unsupported VeNCrypt protocol %d.%d\n", (int)data[0], (int)data[1]);
1735         vnc_write_u8(vs, 1); /* Reject version */
1736         vnc_flush(vs);
1737         vnc_client_error(vs);
1738     } else {
1739         VNC_DEBUG("Sending allowed auth %d\n", vs->subauth);
1740         vnc_write_u8(vs, 0); /* Accept version */
1741         vnc_write_u8(vs, 1); /* Number of sub-auths */
1742         vnc_write_u32(vs, vs->subauth); /* The supported auth */
1743         vnc_flush(vs);
1744         vnc_read_when(vs, protocol_client_vencrypt_auth, 4);
1745     }
1746     return 0;
1747 }
1748
1749 static int start_auth_vencrypt(VncState *vs)
1750 {
1751     /* Send VeNCrypt version 0.2 */
1752     vnc_write_u8(vs, 0);
1753     vnc_write_u8(vs, 2);
1754
1755     vnc_read_when(vs, protocol_client_vencrypt_init, 2);
1756     return 0;
1757 }
1758 #endif /* CONFIG_VNC_TLS */
1759
1760 static int protocol_client_auth(VncState *vs, char *data, size_t len)
1761 {
1762     /* We only advertise 1 auth scheme at a time, so client
1763      * must pick the one we sent. Verify this */
1764     if (data[0] != vs->auth) { /* Reject auth */
1765        VNC_DEBUG("Reject auth %d\n", (int)data[0]);
1766        vnc_write_u32(vs, 1);
1767        if (vs->minor >= 8) {
1768            static const char err[] = "Authentication failed";
1769            vnc_write_u32(vs, sizeof(err));
1770            vnc_write(vs, err, sizeof(err));
1771        }
1772        vnc_client_error(vs);
1773     } else { /* Accept requested auth */
1774        VNC_DEBUG("Client requested auth %d\n", (int)data[0]);
1775        switch (vs->auth) {
1776        case VNC_AUTH_NONE:
1777            VNC_DEBUG("Accept auth none\n");
1778            vnc_write_u32(vs, 0); /* Accept auth completion */
1779            vnc_read_when(vs, protocol_client_init, 1);
1780            break;
1781
1782        case VNC_AUTH_VNC:
1783            VNC_DEBUG("Start VNC auth\n");
1784            return start_auth_vnc(vs);
1785
1786 #if CONFIG_VNC_TLS
1787        case VNC_AUTH_VENCRYPT:
1788            VNC_DEBUG("Accept VeNCrypt auth\n");;
1789            return start_auth_vencrypt(vs);
1790 #endif /* CONFIG_VNC_TLS */
1791
1792        default: /* Should not be possible, but just in case */
1793            VNC_DEBUG("Reject auth %d\n", vs->auth);
1794            vnc_write_u8(vs, 1);
1795            if (vs->minor >= 8) {
1796                static const char err[] = "Authentication failed";
1797                vnc_write_u32(vs, sizeof(err));
1798                vnc_write(vs, err, sizeof(err));
1799            }
1800            vnc_client_error(vs);
1801        }
1802     }
1803     return 0;
1804 }
1805
1806 static int protocol_version(VncState *vs, char *version, size_t len)
1807 {
1808     char local[13];
1809
1810     memcpy(local, version, 12);
1811     local[12] = 0;
1812
1813     if (sscanf(local, "RFB %03d.%03d\n", &vs->major, &vs->minor) != 2) {
1814         VNC_DEBUG("Malformed protocol version %s\n", local);
1815         vnc_client_error(vs);
1816         return 0;
1817     }
1818     VNC_DEBUG("Client request protocol version %d.%d\n", vs->major, vs->minor);
1819     if (vs->major != 3 ||
1820         (vs->minor != 3 &&
1821          vs->minor != 5 &&
1822          vs->minor != 7 &&
1823          vs->minor != 8)) {
1824         VNC_DEBUG("Unsupported client version\n");
1825         vnc_write_u32(vs, VNC_AUTH_INVALID);
1826         vnc_flush(vs);
1827         vnc_client_error(vs);
1828         return 0;
1829     }
1830     /* Some broken client report v3.5 which spec requires to be treated
1831      * as equivalent to v3.3 by servers
1832      */
1833     if (vs->minor == 5)
1834         vs->minor = 3;
1835
1836     if (vs->minor == 3) {
1837         if (vs->auth == VNC_AUTH_NONE) {
1838             VNC_DEBUG("Tell client auth none\n");
1839             vnc_write_u32(vs, vs->auth);
1840             vnc_flush(vs);
1841             vnc_read_when(vs, protocol_client_init, 1);
1842        } else if (vs->auth == VNC_AUTH_VNC) {
1843             VNC_DEBUG("Tell client VNC auth\n");
1844             vnc_write_u32(vs, vs->auth);
1845             vnc_flush(vs);
1846             start_auth_vnc(vs);
1847        } else {
1848             VNC_DEBUG("Unsupported auth %d for protocol 3.3\n", vs->auth);
1849             vnc_write_u32(vs, VNC_AUTH_INVALID);
1850             vnc_flush(vs);
1851             vnc_client_error(vs);
1852        }
1853     } else {
1854         VNC_DEBUG("Telling client we support auth %d\n", vs->auth);
1855         vnc_write_u8(vs, 1); /* num auth */
1856         vnc_write_u8(vs, vs->auth);
1857         vnc_read_when(vs, protocol_client_auth, 1);
1858         vnc_flush(vs);
1859     }
1860
1861     return 0;
1862 }
1863
1864 static void vnc_listen_read(void *opaque)
1865 {
1866     VncState *vs = opaque;
1867     struct sockaddr_in addr;
1868     socklen_t addrlen = sizeof(addr);
1869
1870     vs->csock = accept(vs->lsock, (struct sockaddr *)&addr, &addrlen);
1871     if (vs->csock != -1) {
1872         VNC_DEBUG("New client on socket %d\n", vs->csock);
1873         socket_set_nonblock(vs->csock);
1874         qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, opaque);
1875         vnc_write(vs, "RFB 003.008\n", 12);
1876         vnc_flush(vs);
1877         vnc_read_when(vs, protocol_version, 12);
1878         memset(vs->old_data, 0, vs->ds->linesize * vs->ds->height);
1879         memset(vs->dirty_row, 0xFF, sizeof(vs->dirty_row));
1880         vs->has_resize = 0;
1881         vs->has_hextile = 0;
1882         vs->ds->dpy_copy = NULL;
1883     }
1884 }
1885
1886 extern int parse_host_port(struct sockaddr_in *saddr, const char *str);
1887
1888 void vnc_display_init(DisplayState *ds)
1889 {
1890     VncState *vs;
1891
1892     vs = qemu_mallocz(sizeof(VncState));
1893     if (!vs)
1894         exit(1);
1895
1896     ds->opaque = vs;
1897     vnc_state = vs;
1898     vs->display = NULL;
1899     vs->password = NULL;
1900
1901     vs->lsock = -1;
1902     vs->csock = -1;
1903     vs->depth = 4;
1904     vs->last_x = -1;
1905     vs->last_y = -1;
1906
1907     vs->ds = ds;
1908
1909     if (!keyboard_layout)
1910         keyboard_layout = "en-us";
1911
1912     vs->kbd_layout = init_keyboard_layout(keyboard_layout);
1913     if (!vs->kbd_layout)
1914         exit(1);
1915
1916     vs->ds->data = NULL;
1917     vs->ds->dpy_update = vnc_dpy_update;
1918     vs->ds->dpy_resize = vnc_dpy_resize;
1919     vs->ds->dpy_refresh = vnc_dpy_refresh;
1920
1921     memset(vs->dirty_row, 0xFF, sizeof(vs->dirty_row));
1922
1923     vnc_dpy_resize(vs->ds, 640, 400);
1924 }
1925
1926 #if CONFIG_VNC_TLS
1927 static int vnc_set_x509_credential(VncState *vs,
1928                                    const char *certdir,
1929                                    const char *filename,
1930                                    char **cred,
1931                                    int ignoreMissing)
1932 {
1933     struct stat sb;
1934
1935     if (*cred) {
1936         qemu_free(*cred);
1937         *cred = NULL;
1938     }
1939
1940     if (!(*cred = qemu_malloc(strlen(certdir) + strlen(filename) + 2)))
1941         return -1;
1942
1943     strcpy(*cred, certdir);
1944     strcat(*cred, "/");
1945     strcat(*cred, filename);
1946
1947     VNC_DEBUG("Check %s\n", *cred);
1948     if (stat(*cred, &sb) < 0) {
1949         qemu_free(*cred);
1950         *cred = NULL;
1951         if (ignoreMissing && errno == ENOENT)
1952             return 0;
1953         return -1;
1954     }
1955
1956     return 0;
1957 }
1958
1959 static int vnc_set_x509_credential_dir(VncState *vs,
1960                                        const char *certdir)
1961 {
1962     if (vnc_set_x509_credential(vs, certdir, X509_CA_CERT_FILE, &vs->x509cacert, 0) < 0)
1963         goto cleanup;
1964     if (vnc_set_x509_credential(vs, certdir, X509_CA_CRL_FILE, &vs->x509cacrl, 1) < 0)
1965         goto cleanup;
1966     if (vnc_set_x509_credential(vs, certdir, X509_SERVER_CERT_FILE, &vs->x509cert, 0) < 0)
1967         goto cleanup;
1968     if (vnc_set_x509_credential(vs, certdir, X509_SERVER_KEY_FILE, &vs->x509key, 0) < 0)
1969         goto cleanup;
1970
1971     return 0;
1972
1973  cleanup:
1974     qemu_free(vs->x509cacert);
1975     qemu_free(vs->x509cacrl);
1976     qemu_free(vs->x509cert);
1977     qemu_free(vs->x509key);
1978     vs->x509cacert = vs->x509cacrl = vs->x509cert = vs->x509key = NULL;
1979     return -1;
1980 }
1981 #endif /* CONFIG_VNC_TLS */
1982
1983 void vnc_display_close(DisplayState *ds)
1984 {
1985     VncState *vs = ds ? (VncState *)ds->opaque : vnc_state;
1986
1987     if (vs->display) {
1988         qemu_free(vs->display);
1989         vs->display = NULL;
1990     }
1991     if (vs->lsock != -1) {
1992         qemu_set_fd_handler2(vs->lsock, NULL, NULL, NULL, NULL);
1993         close(vs->lsock);
1994         vs->lsock = -1;
1995     }
1996     if (vs->csock != -1) {
1997         qemu_set_fd_handler2(vs->csock, NULL, NULL, NULL, NULL);
1998         closesocket(vs->csock);
1999         vs->csock = -1;
2000         buffer_reset(&vs->input);
2001         buffer_reset(&vs->output);
2002         vs->need_update = 0;
2003 #if CONFIG_VNC_TLS
2004         if (vs->tls_session) {
2005             gnutls_deinit(vs->tls_session);
2006             vs->tls_session = NULL;
2007         }
2008         vs->wiremode = VNC_WIREMODE_CLEAR;
2009 #endif /* CONFIG_VNC_TLS */
2010     }
2011     vs->auth = VNC_AUTH_INVALID;
2012 #if CONFIG_VNC_TLS
2013     vs->subauth = VNC_AUTH_INVALID;
2014     vs->x509verify = 0;
2015 #endif
2016 }
2017
2018 int vnc_display_password(DisplayState *ds, const char *password)
2019 {
2020     VncState *vs = ds ? (VncState *)ds->opaque : vnc_state;
2021
2022     if (vs->password) {
2023         qemu_free(vs->password);
2024         vs->password = NULL;
2025     }
2026     if (password && password[0]) {
2027         if (!(vs->password = qemu_strdup(password)))
2028             return -1;
2029     }
2030
2031     return 0;
2032 }
2033
2034 int vnc_display_open(DisplayState *ds, const char *display)
2035 {
2036     struct sockaddr *addr;
2037     struct sockaddr_in iaddr;
2038 #ifndef _WIN32
2039     struct sockaddr_un uaddr;
2040 #endif
2041     int reuse_addr, ret;
2042     socklen_t addrlen;
2043     const char *p;
2044     VncState *vs = ds ? (VncState *)ds->opaque : vnc_state;
2045     const char *options;
2046     int password = 0;
2047 #if CONFIG_VNC_TLS
2048     int tls = 0, x509 = 0;
2049 #endif
2050
2051     vnc_display_close(ds);
2052     if (strcmp(display, "none") == 0)
2053         return 0;
2054
2055     if (!(vs->display = strdup(display)))
2056         return -1;
2057
2058     options = display;
2059     while ((options = strchr(options, ','))) {
2060         options++;
2061         if (strncmp(options, "password", 8) == 0) {
2062             password = 1; /* Require password auth */
2063 #if CONFIG_VNC_TLS
2064         } else if (strncmp(options, "tls", 3) == 0) {
2065             tls = 1; /* Require TLS */
2066         } else if (strncmp(options, "x509", 4) == 0) {
2067             char *start, *end;
2068             x509 = 1; /* Require x509 certificates */
2069             if (strncmp(options, "x509verify", 10) == 0)
2070                 vs->x509verify = 1; /* ...and verify client certs */
2071
2072             /* Now check for 'x509=/some/path' postfix
2073              * and use that to setup x509 certificate/key paths */
2074             start = strchr(options, '=');
2075             end = strchr(options, ',');
2076             if (start && (!end || (start < end))) {
2077                 int len = end ? end-(start+1) : strlen(start+1);
2078                 char *path = qemu_malloc(len+1);
2079                 strncpy(path, start+1, len);
2080                 path[len] = '\0';
2081                 VNC_DEBUG("Trying certificate path '%s'\n", path);
2082                 if (vnc_set_x509_credential_dir(vs, path) < 0) {
2083                     fprintf(stderr, "Failed to find x509 certificates/keys in %s\n", path);
2084                     qemu_free(path);
2085                     qemu_free(vs->display);
2086                     vs->display = NULL;
2087                     return -1;
2088                 }
2089                 qemu_free(path);
2090             } else {
2091                 fprintf(stderr, "No certificate path provided\n");
2092                 qemu_free(vs->display);
2093                 vs->display = NULL;
2094                 return -1;
2095             }
2096 #endif
2097         }
2098     }
2099
2100     if (password) {
2101 #if CONFIG_VNC_TLS
2102         if (tls) {
2103             vs->auth = VNC_AUTH_VENCRYPT;
2104             if (x509) {
2105                 VNC_DEBUG("Initializing VNC server with x509 password auth\n");
2106                 vs->subauth = VNC_AUTH_VENCRYPT_X509VNC;
2107             } else {
2108                 VNC_DEBUG("Initializing VNC server with TLS password auth\n");
2109                 vs->subauth = VNC_AUTH_VENCRYPT_TLSVNC;
2110             }
2111         } else {
2112 #endif
2113             VNC_DEBUG("Initializing VNC server with password auth\n");
2114             vs->auth = VNC_AUTH_VNC;
2115 #if CONFIG_VNC_TLS
2116             vs->subauth = VNC_AUTH_INVALID;
2117         }
2118 #endif
2119     } else {
2120 #if CONFIG_VNC_TLS
2121         if (tls) {
2122             vs->auth = VNC_AUTH_VENCRYPT;
2123             if (x509) {
2124                 VNC_DEBUG("Initializing VNC server with x509 no auth\n");
2125                 vs->subauth = VNC_AUTH_VENCRYPT_X509NONE;
2126             } else {
2127                 VNC_DEBUG("Initializing VNC server with TLS no auth\n");
2128                 vs->subauth = VNC_AUTH_VENCRYPT_TLSNONE;
2129             }
2130         } else {
2131 #endif
2132             VNC_DEBUG("Initializing VNC server with no auth\n");
2133             vs->auth = VNC_AUTH_NONE;
2134 #if CONFIG_VNC_TLS
2135             vs->subauth = VNC_AUTH_INVALID;
2136         }
2137 #endif
2138     }
2139 #ifndef _WIN32
2140     if (strstart(display, "unix:", &p)) {
2141         addr = (struct sockaddr *)&uaddr;
2142         addrlen = sizeof(uaddr);
2143
2144         vs->lsock = socket(PF_UNIX, SOCK_STREAM, 0);
2145         if (vs->lsock == -1) {
2146             fprintf(stderr, "Could not create socket\n");
2147             free(vs->display);
2148             vs->display = NULL;
2149             return -1;
2150         }
2151
2152         uaddr.sun_family = AF_UNIX;
2153         memset(uaddr.sun_path, 0, 108);
2154         snprintf(uaddr.sun_path, 108, "%s", p);
2155
2156         unlink(uaddr.sun_path);
2157     } else
2158 #endif
2159     {
2160         addr = (struct sockaddr *)&iaddr;
2161         addrlen = sizeof(iaddr);
2162
2163         if (parse_host_port(&iaddr, display) < 0) {
2164             fprintf(stderr, "Could not parse VNC address\n");
2165             free(vs->display);
2166             vs->display = NULL;
2167             return -1;
2168         }
2169
2170         iaddr.sin_port = htons(ntohs(iaddr.sin_port) + 5900);
2171
2172         vs->lsock = socket(PF_INET, SOCK_STREAM, 0);
2173         if (vs->lsock == -1) {
2174             fprintf(stderr, "Could not create socket\n");
2175             free(vs->display);
2176             vs->display = NULL;
2177             return -1;
2178         }
2179
2180         reuse_addr = 1;
2181         ret = setsockopt(vs->lsock, SOL_SOCKET, SO_REUSEADDR,
2182                          (const char *)&reuse_addr, sizeof(reuse_addr));
2183         if (ret == -1) {
2184             fprintf(stderr, "setsockopt() failed\n");
2185             close(vs->lsock);
2186             vs->lsock = -1;
2187             free(vs->display);
2188             vs->display = NULL;
2189             return -1;
2190         }
2191     }
2192
2193     if (bind(vs->lsock, addr, addrlen) == -1) {
2194         fprintf(stderr, "bind() failed\n");
2195         close(vs->lsock);
2196         vs->lsock = -1;
2197         free(vs->display);
2198         vs->display = NULL;
2199         return -1;
2200     }
2201
2202     if (listen(vs->lsock, 1) == -1) {
2203         fprintf(stderr, "listen() failed\n");
2204         close(vs->lsock);
2205         vs->lsock = -1;
2206         free(vs->display);
2207         vs->display = NULL;
2208         return -1;
2209     }
2210
2211     return qemu_set_fd_handler2(vs->lsock, vnc_listen_poll, vnc_listen_read, NULL, vs);
2212 }