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             return 8 + read_u32(data, 4);
1200
1201         client_cut_text(vs, read_u32(data, 4), data + 8);
1202         break;
1203     default:
1204         printf("Msg: %d\n", data[0]);
1205         vnc_client_error(vs);
1206         break;
1207     }
1208
1209     vnc_read_when(vs, protocol_client_msg, 1);
1210     return 0;
1211 }
1212
1213 static int protocol_client_init(VncState *vs, char *data, size_t len)
1214 {
1215     char pad[3] = { 0, 0, 0 };
1216     char buf[1024];
1217     int size;
1218
1219     vs->width = vs->ds->width;
1220     vs->height = vs->ds->height;
1221     vnc_write_u16(vs, vs->ds->width);
1222     vnc_write_u16(vs, vs->ds->height);
1223
1224     vnc_write_u8(vs, vs->depth * 8); /* bits-per-pixel */
1225     vnc_write_u8(vs, vs->depth * 8); /* depth */
1226 #ifdef WORDS_BIGENDIAN
1227     vnc_write_u8(vs, 1);             /* big-endian-flag */
1228 #else
1229     vnc_write_u8(vs, 0);             /* big-endian-flag */
1230 #endif
1231     vnc_write_u8(vs, 1);             /* true-color-flag */
1232     if (vs->depth == 4) {
1233         vnc_write_u16(vs, 0xFF);     /* red-max */
1234         vnc_write_u16(vs, 0xFF);     /* green-max */
1235         vnc_write_u16(vs, 0xFF);     /* blue-max */
1236         vnc_write_u8(vs, 16);        /* red-shift */
1237         vnc_write_u8(vs, 8);         /* green-shift */
1238         vnc_write_u8(vs, 0);         /* blue-shift */
1239         vs->send_hextile_tile = send_hextile_tile_32;
1240     } else if (vs->depth == 2) {
1241         vnc_write_u16(vs, 31);       /* red-max */
1242         vnc_write_u16(vs, 63);       /* green-max */
1243         vnc_write_u16(vs, 31);       /* blue-max */
1244         vnc_write_u8(vs, 11);        /* red-shift */
1245         vnc_write_u8(vs, 5);         /* green-shift */
1246         vnc_write_u8(vs, 0);         /* blue-shift */
1247         vs->send_hextile_tile = send_hextile_tile_16;
1248     } else if (vs->depth == 1) {
1249         /* XXX: change QEMU pixel 8 bit pixel format to match the VNC one ? */
1250         vnc_write_u16(vs, 7);        /* red-max */
1251         vnc_write_u16(vs, 7);        /* green-max */
1252         vnc_write_u16(vs, 3);        /* blue-max */
1253         vnc_write_u8(vs, 5);         /* red-shift */
1254         vnc_write_u8(vs, 2);         /* green-shift */
1255         vnc_write_u8(vs, 0);         /* blue-shift */
1256         vs->send_hextile_tile = send_hextile_tile_8;
1257     }
1258     vs->write_pixels = vnc_write_pixels_copy;
1259
1260     vnc_write(vs, pad, 3);           /* padding */
1261
1262     if (qemu_name)
1263         size = snprintf(buf, sizeof(buf), "QEMU (%s)", qemu_name);
1264     else
1265         size = snprintf(buf, sizeof(buf), "QEMU");
1266
1267     vnc_write_u32(vs, size);
1268     vnc_write(vs, buf, size);
1269     vnc_flush(vs);
1270
1271     vnc_read_when(vs, protocol_client_msg, 1);
1272
1273     return 0;
1274 }
1275
1276 static void make_challenge(VncState *vs)
1277 {
1278     int i;
1279
1280     srand(time(NULL)+getpid()+getpid()*987654+rand());
1281
1282     for (i = 0 ; i < sizeof(vs->challenge) ; i++)
1283         vs->challenge[i] = (int) (256.0*rand()/(RAND_MAX+1.0));
1284 }
1285
1286 static int protocol_client_auth_vnc(VncState *vs, char *data, size_t len)
1287 {
1288     char response[VNC_AUTH_CHALLENGE_SIZE];
1289     int i, j, pwlen;
1290     char key[8];
1291
1292     if (!vs->password || !vs->password[0]) {
1293         VNC_DEBUG("No password configured on server");
1294         vnc_write_u32(vs, 1); /* Reject auth */
1295         if (vs->minor >= 8) {
1296             static const char err[] = "Authentication failed";
1297             vnc_write_u32(vs, sizeof(err));
1298             vnc_write(vs, err, sizeof(err));
1299         }
1300         vnc_flush(vs);
1301         vnc_client_error(vs);
1302         return 0;
1303     }
1304
1305     memcpy(response, vs->challenge, VNC_AUTH_CHALLENGE_SIZE);
1306
1307     /* Calculate the expected challenge response */
1308     pwlen = strlen(vs->password);
1309     for (i=0; i<sizeof(key); i++)
1310         key[i] = i<pwlen ? vs->password[i] : 0;
1311     deskey(key, EN0);
1312     for (j = 0; j < VNC_AUTH_CHALLENGE_SIZE; j += 8)
1313         des(response+j, response+j);
1314
1315     /* Compare expected vs actual challenge response */
1316     if (memcmp(response, data, VNC_AUTH_CHALLENGE_SIZE) != 0) {
1317         VNC_DEBUG("Client challenge reponse did not match\n");
1318         vnc_write_u32(vs, 1); /* Reject auth */
1319         if (vs->minor >= 8) {
1320             static const char err[] = "Authentication failed";
1321             vnc_write_u32(vs, sizeof(err));
1322             vnc_write(vs, err, sizeof(err));
1323         }
1324         vnc_flush(vs);
1325         vnc_client_error(vs);
1326     } else {
1327         VNC_DEBUG("Accepting VNC challenge response\n");
1328         vnc_write_u32(vs, 0); /* Accept auth */
1329         vnc_flush(vs);
1330
1331         vnc_read_when(vs, protocol_client_init, 1);
1332     }
1333     return 0;
1334 }
1335
1336 static int start_auth_vnc(VncState *vs)
1337 {
1338     make_challenge(vs);
1339     /* Send client a 'random' challenge */
1340     vnc_write(vs, vs->challenge, sizeof(vs->challenge));
1341     vnc_flush(vs);
1342
1343     vnc_read_when(vs, protocol_client_auth_vnc, sizeof(vs->challenge));
1344     return 0;
1345 }
1346
1347
1348 #if CONFIG_VNC_TLS
1349 #define DH_BITS 1024
1350 static gnutls_dh_params_t dh_params;
1351
1352 static int vnc_tls_initialize(void)
1353 {
1354     static int tlsinitialized = 0;
1355
1356     if (tlsinitialized)
1357         return 1;
1358
1359     if (gnutls_global_init () < 0)
1360         return 0;
1361
1362     /* XXX ought to re-generate diffie-hellmen params periodically */
1363     if (gnutls_dh_params_init (&dh_params) < 0)
1364         return 0;
1365     if (gnutls_dh_params_generate2 (dh_params, DH_BITS) < 0)
1366         return 0;
1367
1368 #if _VNC_DEBUG == 2
1369     gnutls_global_set_log_level(10);
1370     gnutls_global_set_log_function(vnc_debug_gnutls_log);
1371 #endif
1372
1373     tlsinitialized = 1;
1374
1375     return 1;
1376 }
1377
1378 static gnutls_anon_server_credentials vnc_tls_initialize_anon_cred(void)
1379 {
1380     gnutls_anon_server_credentials anon_cred;
1381     int ret;
1382
1383     if ((ret = gnutls_anon_allocate_server_credentials(&anon_cred)) < 0) {
1384         VNC_DEBUG("Cannot allocate credentials %s\n", gnutls_strerror(ret));
1385         return NULL;
1386     }
1387
1388     gnutls_anon_set_server_dh_params(anon_cred, dh_params);
1389
1390     return anon_cred;
1391 }
1392
1393
1394 static gnutls_certificate_credentials_t vnc_tls_initialize_x509_cred(VncState *vs)
1395 {
1396     gnutls_certificate_credentials_t x509_cred;
1397     int ret;
1398
1399     if (!vs->x509cacert) {
1400         VNC_DEBUG("No CA x509 certificate specified\n");
1401         return NULL;
1402     }
1403     if (!vs->x509cert) {
1404         VNC_DEBUG("No server x509 certificate specified\n");
1405         return NULL;
1406     }
1407     if (!vs->x509key) {
1408         VNC_DEBUG("No server private key specified\n");
1409         return NULL;
1410     }
1411
1412     if ((ret = gnutls_certificate_allocate_credentials(&x509_cred)) < 0) {
1413         VNC_DEBUG("Cannot allocate credentials %s\n", gnutls_strerror(ret));
1414         return NULL;
1415     }
1416     if ((ret = gnutls_certificate_set_x509_trust_file(x509_cred,
1417                                                       vs->x509cacert,
1418                                                       GNUTLS_X509_FMT_PEM)) < 0) {
1419         VNC_DEBUG("Cannot load CA certificate %s\n", gnutls_strerror(ret));
1420         gnutls_certificate_free_credentials(x509_cred);
1421         return NULL;
1422     }
1423
1424     if ((ret = gnutls_certificate_set_x509_key_file (x509_cred,
1425                                                      vs->x509cert,
1426                                                      vs->x509key,
1427                                                      GNUTLS_X509_FMT_PEM)) < 0) {
1428         VNC_DEBUG("Cannot load certificate & key %s\n", gnutls_strerror(ret));
1429         gnutls_certificate_free_credentials(x509_cred);
1430         return NULL;
1431     }
1432
1433     if (vs->x509cacrl) {
1434         if ((ret = gnutls_certificate_set_x509_crl_file(x509_cred,
1435                                                         vs->x509cacrl,
1436                                                         GNUTLS_X509_FMT_PEM)) < 0) {
1437             VNC_DEBUG("Cannot load CRL %s\n", gnutls_strerror(ret));
1438             gnutls_certificate_free_credentials(x509_cred);
1439             return NULL;
1440         }
1441     }
1442
1443     gnutls_certificate_set_dh_params (x509_cred, dh_params);
1444
1445     return x509_cred;
1446 }
1447
1448 static int vnc_validate_certificate(struct VncState *vs)
1449 {
1450     int ret;
1451     unsigned int status;
1452     const gnutls_datum_t *certs;
1453     unsigned int nCerts, i;
1454     time_t now;
1455
1456     VNC_DEBUG("Validating client certificate\n");
1457     if ((ret = gnutls_certificate_verify_peers2 (vs->tls_session, &status)) < 0) {
1458         VNC_DEBUG("Verify failed %s\n", gnutls_strerror(ret));
1459         return -1;
1460     }
1461
1462     if ((now = time(NULL)) == ((time_t)-1)) {
1463         return -1;
1464     }
1465
1466     if (status != 0) {
1467         if (status & GNUTLS_CERT_INVALID)
1468             VNC_DEBUG("The certificate is not trusted.\n");
1469
1470         if (status & GNUTLS_CERT_SIGNER_NOT_FOUND)
1471             VNC_DEBUG("The certificate hasn't got a known issuer.\n");
1472
1473         if (status & GNUTLS_CERT_REVOKED)
1474             VNC_DEBUG("The certificate has been revoked.\n");
1475
1476         if (status & GNUTLS_CERT_INSECURE_ALGORITHM)
1477             VNC_DEBUG("The certificate uses an insecure algorithm\n");
1478
1479         return -1;
1480     } else {
1481         VNC_DEBUG("Certificate is valid!\n");
1482     }
1483
1484     /* Only support x509 for now */
1485     if (gnutls_certificate_type_get(vs->tls_session) != GNUTLS_CRT_X509)
1486         return -1;
1487
1488     if (!(certs = gnutls_certificate_get_peers(vs->tls_session, &nCerts)))
1489         return -1;
1490
1491     for (i = 0 ; i < nCerts ; i++) {
1492         gnutls_x509_crt_t cert;
1493         VNC_DEBUG ("Checking certificate chain %d\n", i);
1494         if (gnutls_x509_crt_init (&cert) < 0)
1495             return -1;
1496
1497         if (gnutls_x509_crt_import(cert, &certs[i], GNUTLS_X509_FMT_DER) < 0) {
1498             gnutls_x509_crt_deinit (cert);
1499             return -1;
1500         }
1501
1502         if (gnutls_x509_crt_get_expiration_time (cert) < now) {
1503             VNC_DEBUG("The certificate has expired\n");
1504             gnutls_x509_crt_deinit (cert);
1505             return -1;
1506         }
1507
1508         if (gnutls_x509_crt_get_activation_time (cert) > now) {
1509             VNC_DEBUG("The certificate is not yet activated\n");
1510             gnutls_x509_crt_deinit (cert);
1511             return -1;
1512         }
1513
1514         if (gnutls_x509_crt_get_activation_time (cert) > now) {
1515             VNC_DEBUG("The certificate is not yet activated\n");
1516             gnutls_x509_crt_deinit (cert);
1517             return -1;
1518         }
1519
1520         gnutls_x509_crt_deinit (cert);
1521     }
1522
1523     return 0;
1524 }
1525
1526
1527 static int start_auth_vencrypt_subauth(VncState *vs)
1528 {
1529     switch (vs->subauth) {
1530     case VNC_AUTH_VENCRYPT_TLSNONE:
1531     case VNC_AUTH_VENCRYPT_X509NONE:
1532        VNC_DEBUG("Accept TLS auth none\n");
1533        vnc_write_u32(vs, 0); /* Accept auth completion */
1534        vnc_read_when(vs, protocol_client_init, 1);
1535        break;
1536
1537     case VNC_AUTH_VENCRYPT_TLSVNC:
1538     case VNC_AUTH_VENCRYPT_X509VNC:
1539        VNC_DEBUG("Start TLS auth VNC\n");
1540        return start_auth_vnc(vs);
1541
1542     default: /* Should not be possible, but just in case */
1543        VNC_DEBUG("Reject auth %d\n", vs->auth);
1544        vnc_write_u8(vs, 1);
1545        if (vs->minor >= 8) {
1546            static const char err[] = "Unsupported authentication type";
1547            vnc_write_u32(vs, sizeof(err));
1548            vnc_write(vs, err, sizeof(err));
1549        }
1550        vnc_client_error(vs);
1551     }
1552
1553     return 0;
1554 }
1555
1556 static void vnc_handshake_io(void *opaque);
1557
1558 static int vnc_continue_handshake(struct VncState *vs) {
1559     int ret;
1560
1561     if ((ret = gnutls_handshake(vs->tls_session)) < 0) {
1562        if (!gnutls_error_is_fatal(ret)) {
1563            VNC_DEBUG("Handshake interrupted (blocking)\n");
1564            if (!gnutls_record_get_direction(vs->tls_session))
1565                qemu_set_fd_handler(vs->csock, vnc_handshake_io, NULL, vs);
1566            else
1567                qemu_set_fd_handler(vs->csock, NULL, vnc_handshake_io, vs);
1568            return 0;
1569        }
1570        VNC_DEBUG("Handshake failed %s\n", gnutls_strerror(ret));
1571        vnc_client_error(vs);
1572        return -1;
1573     }
1574
1575     if (vs->x509verify) {
1576         if (vnc_validate_certificate(vs) < 0) {
1577             VNC_DEBUG("Client verification failed\n");
1578             vnc_client_error(vs);
1579             return -1;
1580         } else {
1581             VNC_DEBUG("Client verification passed\n");
1582         }
1583     }
1584
1585     VNC_DEBUG("Handshake done, switching to TLS data mode\n");
1586     vs->wiremode = VNC_WIREMODE_TLS;
1587     qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, vnc_client_write, vs);
1588
1589     return start_auth_vencrypt_subauth(vs);
1590 }
1591
1592 static void vnc_handshake_io(void *opaque) {
1593     struct VncState *vs = (struct VncState *)opaque;
1594
1595     VNC_DEBUG("Handshake IO continue\n");
1596     vnc_continue_handshake(vs);
1597 }
1598
1599 #define NEED_X509_AUTH(vs)                            \
1600     ((vs)->subauth == VNC_AUTH_VENCRYPT_X509NONE ||   \
1601      (vs)->subauth == VNC_AUTH_VENCRYPT_X509VNC ||    \
1602      (vs)->subauth == VNC_AUTH_VENCRYPT_X509PLAIN)
1603
1604
1605 static int vnc_start_tls(struct VncState *vs) {
1606     static const int cert_type_priority[] = { GNUTLS_CRT_X509, 0 };
1607     static const int protocol_priority[]= { GNUTLS_TLS1_1, GNUTLS_TLS1_0, GNUTLS_SSL3, 0 };
1608     static const int kx_anon[] = {GNUTLS_KX_ANON_DH, 0};
1609     static const int kx_x509[] = {GNUTLS_KX_DHE_DSS, GNUTLS_KX_RSA, GNUTLS_KX_DHE_RSA, GNUTLS_KX_SRP, 0};
1610
1611     VNC_DEBUG("Do TLS setup\n");
1612     if (vnc_tls_initialize() < 0) {
1613         VNC_DEBUG("Failed to init TLS\n");
1614         vnc_client_error(vs);
1615         return -1;
1616     }
1617     if (vs->tls_session == NULL) {
1618         if (gnutls_init(&vs->tls_session, GNUTLS_SERVER) < 0) {
1619             vnc_client_error(vs);
1620             return -1;
1621         }
1622
1623         if (gnutls_set_default_priority(vs->tls_session) < 0) {
1624             gnutls_deinit(vs->tls_session);
1625             vs->tls_session = NULL;
1626             vnc_client_error(vs);
1627             return -1;
1628         }
1629
1630         if (gnutls_kx_set_priority(vs->tls_session, NEED_X509_AUTH(vs) ? kx_x509 : kx_anon) < 0) {
1631             gnutls_deinit(vs->tls_session);
1632             vs->tls_session = NULL;
1633             vnc_client_error(vs);
1634             return -1;
1635         }
1636
1637         if (gnutls_certificate_type_set_priority(vs->tls_session, cert_type_priority) < 0) {
1638             gnutls_deinit(vs->tls_session);
1639             vs->tls_session = NULL;
1640             vnc_client_error(vs);
1641             return -1;
1642         }
1643
1644         if (gnutls_protocol_set_priority(vs->tls_session, protocol_priority) < 0) {
1645             gnutls_deinit(vs->tls_session);
1646             vs->tls_session = NULL;
1647             vnc_client_error(vs);
1648             return -1;
1649         }
1650
1651         if (NEED_X509_AUTH(vs)) {
1652             gnutls_certificate_server_credentials x509_cred = vnc_tls_initialize_x509_cred(vs);
1653             if (!x509_cred) {
1654                 gnutls_deinit(vs->tls_session);
1655                 vs->tls_session = NULL;
1656                 vnc_client_error(vs);
1657                 return -1;
1658             }
1659             if (gnutls_credentials_set(vs->tls_session, GNUTLS_CRD_CERTIFICATE, x509_cred) < 0) {
1660                 gnutls_deinit(vs->tls_session);
1661                 vs->tls_session = NULL;
1662                 gnutls_certificate_free_credentials(x509_cred);
1663                 vnc_client_error(vs);
1664                 return -1;
1665             }
1666             if (vs->x509verify) {
1667                 VNC_DEBUG("Requesting a client certificate\n");
1668                 gnutls_certificate_server_set_request (vs->tls_session, GNUTLS_CERT_REQUEST);
1669             }
1670
1671         } else {
1672             gnutls_anon_server_credentials anon_cred = vnc_tls_initialize_anon_cred();
1673             if (!anon_cred) {
1674                 gnutls_deinit(vs->tls_session);
1675                 vs->tls_session = NULL;
1676                 vnc_client_error(vs);
1677                 return -1;
1678             }
1679             if (gnutls_credentials_set(vs->tls_session, GNUTLS_CRD_ANON, anon_cred) < 0) {
1680                 gnutls_deinit(vs->tls_session);
1681                 vs->tls_session = NULL;
1682                 gnutls_anon_free_server_credentials(anon_cred);
1683                 vnc_client_error(vs);
1684                 return -1;
1685             }
1686         }
1687
1688         gnutls_transport_set_ptr(vs->tls_session, (gnutls_transport_ptr_t)vs);
1689         gnutls_transport_set_push_function(vs->tls_session, vnc_tls_push);
1690         gnutls_transport_set_pull_function(vs->tls_session, vnc_tls_pull);
1691     }
1692
1693     VNC_DEBUG("Start TLS handshake process\n");
1694     return vnc_continue_handshake(vs);
1695 }
1696
1697 static int protocol_client_vencrypt_auth(VncState *vs, char *data, size_t len)
1698 {
1699     int auth = read_u32(data, 0);
1700
1701     if (auth != vs->subauth) {
1702         VNC_DEBUG("Rejecting auth %d\n", auth);
1703         vnc_write_u8(vs, 0); /* Reject auth */
1704         vnc_flush(vs);
1705         vnc_client_error(vs);
1706     } else {
1707         VNC_DEBUG("Accepting auth %d, starting handshake\n", auth);
1708         vnc_write_u8(vs, 1); /* Accept auth */
1709         vnc_flush(vs);
1710
1711         if (vnc_start_tls(vs) < 0) {
1712             VNC_DEBUG("Failed to complete TLS\n");
1713             return 0;
1714         }
1715
1716         if (vs->wiremode == VNC_WIREMODE_TLS) {
1717             VNC_DEBUG("Starting VeNCrypt subauth\n");
1718             return start_auth_vencrypt_subauth(vs);
1719         } else {
1720             VNC_DEBUG("TLS handshake blocked\n");
1721             return 0;
1722         }
1723     }
1724     return 0;
1725 }
1726
1727 static int protocol_client_vencrypt_init(VncState *vs, char *data, size_t len)
1728 {
1729     if (data[0] != 0 ||
1730         data[1] != 2) {
1731         VNC_DEBUG("Unsupported VeNCrypt protocol %d.%d\n", (int)data[0], (int)data[1]);
1732         vnc_write_u8(vs, 1); /* Reject version */
1733         vnc_flush(vs);
1734         vnc_client_error(vs);
1735     } else {
1736         VNC_DEBUG("Sending allowed auth %d\n", vs->subauth);
1737         vnc_write_u8(vs, 0); /* Accept version */
1738         vnc_write_u8(vs, 1); /* Number of sub-auths */
1739         vnc_write_u32(vs, vs->subauth); /* The supported auth */
1740         vnc_flush(vs);
1741         vnc_read_when(vs, protocol_client_vencrypt_auth, 4);
1742     }
1743     return 0;
1744 }
1745
1746 static int start_auth_vencrypt(VncState *vs)
1747 {
1748     /* Send VeNCrypt version 0.2 */
1749     vnc_write_u8(vs, 0);
1750     vnc_write_u8(vs, 2);
1751
1752     vnc_read_when(vs, protocol_client_vencrypt_init, 2);
1753     return 0;
1754 }
1755 #endif /* CONFIG_VNC_TLS */
1756
1757 static int protocol_client_auth(VncState *vs, char *data, size_t len)
1758 {
1759     /* We only advertise 1 auth scheme at a time, so client
1760      * must pick the one we sent. Verify this */
1761     if (data[0] != vs->auth) { /* Reject auth */
1762        VNC_DEBUG("Reject auth %d\n", (int)data[0]);
1763        vnc_write_u32(vs, 1);
1764        if (vs->minor >= 8) {
1765            static const char err[] = "Authentication failed";
1766            vnc_write_u32(vs, sizeof(err));
1767            vnc_write(vs, err, sizeof(err));
1768        }
1769        vnc_client_error(vs);
1770     } else { /* Accept requested auth */
1771        VNC_DEBUG("Client requested auth %d\n", (int)data[0]);
1772        switch (vs->auth) {
1773        case VNC_AUTH_NONE:
1774            VNC_DEBUG("Accept auth none\n");
1775            vnc_write_u32(vs, 0); /* Accept auth completion */
1776            vnc_read_when(vs, protocol_client_init, 1);
1777            break;
1778
1779        case VNC_AUTH_VNC:
1780            VNC_DEBUG("Start VNC auth\n");
1781            return start_auth_vnc(vs);
1782
1783 #if CONFIG_VNC_TLS
1784        case VNC_AUTH_VENCRYPT:
1785            VNC_DEBUG("Accept VeNCrypt auth\n");;
1786            return start_auth_vencrypt(vs);
1787 #endif /* CONFIG_VNC_TLS */
1788
1789        default: /* Should not be possible, but just in case */
1790            VNC_DEBUG("Reject auth %d\n", vs->auth);
1791            vnc_write_u8(vs, 1);
1792            if (vs->minor >= 8) {
1793                static const char err[] = "Authentication failed";
1794                vnc_write_u32(vs, sizeof(err));
1795                vnc_write(vs, err, sizeof(err));
1796            }
1797            vnc_client_error(vs);
1798        }
1799     }
1800     return 0;
1801 }
1802
1803 static int protocol_version(VncState *vs, char *version, size_t len)
1804 {
1805     char local[13];
1806
1807     memcpy(local, version, 12);
1808     local[12] = 0;
1809
1810     if (sscanf(local, "RFB %03d.%03d\n", &vs->major, &vs->minor) != 2) {
1811         VNC_DEBUG("Malformed protocol version %s\n", local);
1812         vnc_client_error(vs);
1813         return 0;
1814     }
1815     VNC_DEBUG("Client request protocol version %d.%d\n", vs->major, vs->minor);
1816     if (vs->major != 3 ||
1817         (vs->minor != 3 &&
1818          vs->minor != 5 &&
1819          vs->minor != 7 &&
1820          vs->minor != 8)) {
1821         VNC_DEBUG("Unsupported client version\n");
1822         vnc_write_u32(vs, VNC_AUTH_INVALID);
1823         vnc_flush(vs);
1824         vnc_client_error(vs);
1825         return 0;
1826     }
1827     /* Some broken client report v3.5 which spec requires to be treated
1828      * as equivalent to v3.3 by servers
1829      */
1830     if (vs->minor == 5)
1831         vs->minor = 3;
1832
1833     if (vs->minor == 3) {
1834         if (vs->auth == VNC_AUTH_NONE) {
1835             VNC_DEBUG("Tell client auth none\n");
1836             vnc_write_u32(vs, vs->auth);
1837             vnc_flush(vs);
1838             vnc_read_when(vs, protocol_client_init, 1);
1839        } else if (vs->auth == VNC_AUTH_VNC) {
1840             VNC_DEBUG("Tell client VNC auth\n");
1841             vnc_write_u32(vs, vs->auth);
1842             vnc_flush(vs);
1843             start_auth_vnc(vs);
1844        } else {
1845             VNC_DEBUG("Unsupported auth %d for protocol 3.3\n", vs->auth);
1846             vnc_write_u32(vs, VNC_AUTH_INVALID);
1847             vnc_flush(vs);
1848             vnc_client_error(vs);
1849        }
1850     } else {
1851         VNC_DEBUG("Telling client we support auth %d\n", vs->auth);
1852         vnc_write_u8(vs, 1); /* num auth */
1853         vnc_write_u8(vs, vs->auth);
1854         vnc_read_when(vs, protocol_client_auth, 1);
1855         vnc_flush(vs);
1856     }
1857
1858     return 0;
1859 }
1860
1861 static void vnc_listen_read(void *opaque)
1862 {
1863     VncState *vs = opaque;
1864     struct sockaddr_in addr;
1865     socklen_t addrlen = sizeof(addr);
1866
1867     vs->csock = accept(vs->lsock, (struct sockaddr *)&addr, &addrlen);
1868     if (vs->csock != -1) {
1869         VNC_DEBUG("New client on socket %d\n", vs->csock);
1870         socket_set_nonblock(vs->csock);
1871         qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, opaque);
1872         vnc_write(vs, "RFB 003.008\n", 12);
1873         vnc_flush(vs);
1874         vnc_read_when(vs, protocol_version, 12);
1875         memset(vs->old_data, 0, vs->ds->linesize * vs->ds->height);
1876         memset(vs->dirty_row, 0xFF, sizeof(vs->dirty_row));
1877         vs->has_resize = 0;
1878         vs->has_hextile = 0;
1879         vs->ds->dpy_copy = NULL;
1880     }
1881 }
1882
1883 extern int parse_host_port(struct sockaddr_in *saddr, const char *str);
1884
1885 void vnc_display_init(DisplayState *ds)
1886 {
1887     VncState *vs;
1888
1889     vs = qemu_mallocz(sizeof(VncState));
1890     if (!vs)
1891         exit(1);
1892
1893     ds->opaque = vs;
1894     vnc_state = vs;
1895     vs->display = NULL;
1896     vs->password = NULL;
1897
1898     vs->lsock = -1;
1899     vs->csock = -1;
1900     vs->depth = 4;
1901     vs->last_x = -1;
1902     vs->last_y = -1;
1903
1904     vs->ds = ds;
1905
1906     if (!keyboard_layout)
1907         keyboard_layout = "en-us";
1908
1909     vs->kbd_layout = init_keyboard_layout(keyboard_layout);
1910     if (!vs->kbd_layout)
1911         exit(1);
1912
1913     vs->ds->data = NULL;
1914     vs->ds->dpy_update = vnc_dpy_update;
1915     vs->ds->dpy_resize = vnc_dpy_resize;
1916     vs->ds->dpy_refresh = vnc_dpy_refresh;
1917
1918     memset(vs->dirty_row, 0xFF, sizeof(vs->dirty_row));
1919
1920     vnc_dpy_resize(vs->ds, 640, 400);
1921 }
1922
1923 #if CONFIG_VNC_TLS
1924 static int vnc_set_x509_credential(VncState *vs,
1925                                    const char *certdir,
1926                                    const char *filename,
1927                                    char **cred,
1928                                    int ignoreMissing)
1929 {
1930     struct stat sb;
1931
1932     if (*cred) {
1933         qemu_free(*cred);
1934         *cred = NULL;
1935     }
1936
1937     if (!(*cred = qemu_malloc(strlen(certdir) + strlen(filename) + 2)))
1938         return -1;
1939
1940     strcpy(*cred, certdir);
1941     strcat(*cred, "/");
1942     strcat(*cred, filename);
1943
1944     VNC_DEBUG("Check %s\n", *cred);
1945     if (stat(*cred, &sb) < 0) {
1946         qemu_free(*cred);
1947         *cred = NULL;
1948         if (ignoreMissing && errno == ENOENT)
1949             return 0;
1950         return -1;
1951     }
1952
1953     return 0;
1954 }
1955
1956 static int vnc_set_x509_credential_dir(VncState *vs,
1957                                        const char *certdir)
1958 {
1959     if (vnc_set_x509_credential(vs, certdir, X509_CA_CERT_FILE, &vs->x509cacert, 0) < 0)
1960         goto cleanup;
1961     if (vnc_set_x509_credential(vs, certdir, X509_CA_CRL_FILE, &vs->x509cacrl, 1) < 0)
1962         goto cleanup;
1963     if (vnc_set_x509_credential(vs, certdir, X509_SERVER_CERT_FILE, &vs->x509cert, 0) < 0)
1964         goto cleanup;
1965     if (vnc_set_x509_credential(vs, certdir, X509_SERVER_KEY_FILE, &vs->x509key, 0) < 0)
1966         goto cleanup;
1967
1968     return 0;
1969
1970  cleanup:
1971     qemu_free(vs->x509cacert);
1972     qemu_free(vs->x509cacrl);
1973     qemu_free(vs->x509cert);
1974     qemu_free(vs->x509key);
1975     vs->x509cacert = vs->x509cacrl = vs->x509cert = vs->x509key = NULL;
1976     return -1;
1977 }
1978 #endif /* CONFIG_VNC_TLS */
1979
1980 void vnc_display_close(DisplayState *ds)
1981 {
1982     VncState *vs = ds ? (VncState *)ds->opaque : vnc_state;
1983
1984     if (vs->display) {
1985         qemu_free(vs->display);
1986         vs->display = NULL;
1987     }
1988     if (vs->lsock != -1) {
1989         qemu_set_fd_handler2(vs->lsock, NULL, NULL, NULL, NULL);
1990         close(vs->lsock);
1991         vs->lsock = -1;
1992     }
1993     if (vs->csock != -1) {
1994         qemu_set_fd_handler2(vs->csock, NULL, NULL, NULL, NULL);
1995         closesocket(vs->csock);
1996         vs->csock = -1;
1997         buffer_reset(&vs->input);
1998         buffer_reset(&vs->output);
1999         vs->need_update = 0;
2000 #if CONFIG_VNC_TLS
2001         if (vs->tls_session) {
2002             gnutls_deinit(vs->tls_session);
2003             vs->tls_session = NULL;
2004         }
2005         vs->wiremode = VNC_WIREMODE_CLEAR;
2006 #endif /* CONFIG_VNC_TLS */
2007     }
2008     vs->auth = VNC_AUTH_INVALID;
2009 #if CONFIG_VNC_TLS
2010     vs->subauth = VNC_AUTH_INVALID;
2011     vs->x509verify = 0;
2012 #endif
2013 }
2014
2015 int vnc_display_password(DisplayState *ds, const char *password)
2016 {
2017     VncState *vs = ds ? (VncState *)ds->opaque : vnc_state;
2018
2019     if (vs->password) {
2020         qemu_free(vs->password);
2021         vs->password = NULL;
2022     }
2023     if (password && password[0]) {
2024         if (!(vs->password = qemu_strdup(password)))
2025             return -1;
2026     }
2027
2028     return 0;
2029 }
2030
2031 int vnc_display_open(DisplayState *ds, const char *display)
2032 {
2033     struct sockaddr *addr;
2034     struct sockaddr_in iaddr;
2035 #ifndef _WIN32
2036     struct sockaddr_un uaddr;
2037 #endif
2038     int reuse_addr, ret;
2039     socklen_t addrlen;
2040     const char *p;
2041     VncState *vs = ds ? (VncState *)ds->opaque : vnc_state;
2042     const char *options;
2043     int password = 0;
2044 #if CONFIG_VNC_TLS
2045     int tls = 0, x509 = 0;
2046 #endif
2047
2048     vnc_display_close(ds);
2049     if (strcmp(display, "none") == 0)
2050         return 0;
2051
2052     if (!(vs->display = strdup(display)))
2053         return -1;
2054
2055     options = display;
2056     while ((options = strchr(options, ','))) {
2057         options++;
2058         if (strncmp(options, "password", 8) == 0) {
2059             password = 1; /* Require password auth */
2060 #if CONFIG_VNC_TLS
2061         } else if (strncmp(options, "tls", 3) == 0) {
2062             tls = 1; /* Require TLS */
2063         } else if (strncmp(options, "x509", 4) == 0) {
2064             char *start, *end;
2065             x509 = 1; /* Require x509 certificates */
2066             if (strncmp(options, "x509verify", 10) == 0)
2067                 vs->x509verify = 1; /* ...and verify client certs */
2068
2069             /* Now check for 'x509=/some/path' postfix
2070              * and use that to setup x509 certificate/key paths */
2071             start = strchr(options, '=');
2072             end = strchr(options, ',');
2073             if (start && (!end || (start < end))) {
2074                 int len = end ? end-(start+1) : strlen(start+1);
2075                 char *path = qemu_malloc(len+1);
2076                 strncpy(path, start+1, len);
2077                 path[len] = '\0';
2078                 VNC_DEBUG("Trying certificate path '%s'\n", path);
2079                 if (vnc_set_x509_credential_dir(vs, path) < 0) {
2080                     fprintf(stderr, "Failed to find x509 certificates/keys in %s\n", path);
2081                     qemu_free(path);
2082                     qemu_free(vs->display);
2083                     vs->display = NULL;
2084                     return -1;
2085                 }
2086                 qemu_free(path);
2087             } else {
2088                 fprintf(stderr, "No certificate path provided\n");
2089                 qemu_free(vs->display);
2090                 vs->display = NULL;
2091                 return -1;
2092             }
2093 #endif
2094         }
2095     }
2096
2097     if (password) {
2098 #if CONFIG_VNC_TLS
2099         if (tls) {
2100             vs->auth = VNC_AUTH_VENCRYPT;
2101             if (x509) {
2102                 VNC_DEBUG("Initializing VNC server with x509 password auth\n");
2103                 vs->subauth = VNC_AUTH_VENCRYPT_X509VNC;
2104             } else {
2105                 VNC_DEBUG("Initializing VNC server with TLS password auth\n");
2106                 vs->subauth = VNC_AUTH_VENCRYPT_TLSVNC;
2107             }
2108         } else {
2109 #endif
2110             VNC_DEBUG("Initializing VNC server with password auth\n");
2111             vs->auth = VNC_AUTH_VNC;
2112 #if CONFIG_VNC_TLS
2113             vs->subauth = VNC_AUTH_INVALID;
2114         }
2115 #endif
2116     } else {
2117 #if CONFIG_VNC_TLS
2118         if (tls) {
2119             vs->auth = VNC_AUTH_VENCRYPT;
2120             if (x509) {
2121                 VNC_DEBUG("Initializing VNC server with x509 no auth\n");
2122                 vs->subauth = VNC_AUTH_VENCRYPT_X509NONE;
2123             } else {
2124                 VNC_DEBUG("Initializing VNC server with TLS no auth\n");
2125                 vs->subauth = VNC_AUTH_VENCRYPT_TLSNONE;
2126             }
2127         } else {
2128 #endif
2129             VNC_DEBUG("Initializing VNC server with no auth\n");
2130             vs->auth = VNC_AUTH_NONE;
2131 #if CONFIG_VNC_TLS
2132             vs->subauth = VNC_AUTH_INVALID;
2133         }
2134 #endif
2135     }
2136 #ifndef _WIN32
2137     if (strstart(display, "unix:", &p)) {
2138         addr = (struct sockaddr *)&uaddr;
2139         addrlen = sizeof(uaddr);
2140
2141         vs->lsock = socket(PF_UNIX, SOCK_STREAM, 0);
2142         if (vs->lsock == -1) {
2143             fprintf(stderr, "Could not create socket\n");
2144             free(vs->display);
2145             vs->display = NULL;
2146             return -1;
2147         }
2148
2149         uaddr.sun_family = AF_UNIX;
2150         memset(uaddr.sun_path, 0, 108);
2151         snprintf(uaddr.sun_path, 108, "%s", p);
2152
2153         unlink(uaddr.sun_path);
2154     } else
2155 #endif
2156     {
2157         addr = (struct sockaddr *)&iaddr;
2158         addrlen = sizeof(iaddr);
2159
2160         if (parse_host_port(&iaddr, display) < 0) {
2161             fprintf(stderr, "Could not parse VNC address\n");
2162             free(vs->display);
2163             vs->display = NULL;
2164             return -1;
2165         }
2166
2167         iaddr.sin_port = htons(ntohs(iaddr.sin_port) + 5900);
2168
2169         vs->lsock = socket(PF_INET, SOCK_STREAM, 0);
2170         if (vs->lsock == -1) {
2171             fprintf(stderr, "Could not create socket\n");
2172             free(vs->display);
2173             vs->display = NULL;
2174             return -1;
2175         }
2176
2177         reuse_addr = 1;
2178         ret = setsockopt(vs->lsock, SOL_SOCKET, SO_REUSEADDR,
2179                          (const char *)&reuse_addr, sizeof(reuse_addr));
2180         if (ret == -1) {
2181             fprintf(stderr, "setsockopt() failed\n");
2182             close(vs->lsock);
2183             vs->lsock = -1;
2184             free(vs->display);
2185             vs->display = NULL;
2186             return -1;
2187         }
2188     }
2189
2190     if (bind(vs->lsock, addr, addrlen) == -1) {
2191         fprintf(stderr, "bind() failed\n");
2192         close(vs->lsock);
2193         vs->lsock = -1;
2194         free(vs->display);
2195         vs->display = NULL;
2196         return -1;
2197     }
2198
2199     if (listen(vs->lsock, 1) == -1) {
2200         fprintf(stderr, "listen() failed\n");
2201         close(vs->lsock);
2202         vs->lsock = -1;
2203         free(vs->display);
2204         vs->display = NULL;
2205         return -1;
2206     }
2207
2208     return qemu_set_fd_handler2(vs->lsock, vnc_listen_poll, vnc_listen_read, NULL, vs);
2209 }