Fix hw/gt64xxx.c compilation with DEBUG defined
[qemu.git] / vnc.c
blob11ae3e5172b2dcdcac8a6d54d952d7316f5d91b7
1 /*
2 * QEMU VNC display driver
4 * Copyright (C) 2006 Anthony Liguori <anthony@codemonkey.ws>
5 * Copyright (C) 2006 Fabrice Bellard
6 * Copyright (C) 2009 Red Hat, Inc
8 * Permission is hereby granted, free of charge, to any person obtaining a copy
9 * of this software and associated documentation files (the "Software"), to deal
10 * in the Software without restriction, including without limitation the rights
11 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
12 * copies of the Software, and to permit persons to whom the Software is
13 * furnished to do so, subject to the following conditions:
15 * The above copyright notice and this permission notice shall be included in
16 * all copies or substantial portions of the Software.
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
19 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
20 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
21 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
22 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
23 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
24 * THE SOFTWARE.
27 #include "vnc.h"
28 #include "sysemu.h"
29 #include "qemu_socket.h"
30 #include "qemu-timer.h"
31 #include "acl.h"
32 #include "qemu-objects.h"
34 #define VNC_REFRESH_INTERVAL_BASE 30
35 #define VNC_REFRESH_INTERVAL_INC 50
36 #define VNC_REFRESH_INTERVAL_MAX 2000
38 #include "vnc_keysym.h"
39 #include "d3des.h"
41 #define count_bits(c, v) { \
42 for (c = 0; v; v >>= 1) \
43 { \
44 c += v & 1; \
45 } \
49 static VncDisplay *vnc_display; /* needed for info vnc */
50 static DisplayChangeListener *dcl;
52 static int vnc_cursor_define(VncState *vs);
54 static char *addr_to_string(const char *format,
55 struct sockaddr_storage *sa,
56 socklen_t salen) {
57 char *addr;
58 char host[NI_MAXHOST];
59 char serv[NI_MAXSERV];
60 int err;
61 size_t addrlen;
63 if ((err = getnameinfo((struct sockaddr *)sa, salen,
64 host, sizeof(host),
65 serv, sizeof(serv),
66 NI_NUMERICHOST | NI_NUMERICSERV)) != 0) {
67 VNC_DEBUG("Cannot resolve address %d: %s\n",
68 err, gai_strerror(err));
69 return NULL;
72 /* Enough for the existing format + the 2 vars we're
73 * substituting in. */
74 addrlen = strlen(format) + strlen(host) + strlen(serv);
75 addr = qemu_malloc(addrlen + 1);
76 snprintf(addr, addrlen, format, host, serv);
77 addr[addrlen] = '\0';
79 return addr;
83 char *vnc_socket_local_addr(const char *format, int fd) {
84 struct sockaddr_storage sa;
85 socklen_t salen;
87 salen = sizeof(sa);
88 if (getsockname(fd, (struct sockaddr*)&sa, &salen) < 0)
89 return NULL;
91 return addr_to_string(format, &sa, salen);
94 char *vnc_socket_remote_addr(const char *format, int fd) {
95 struct sockaddr_storage sa;
96 socklen_t salen;
98 salen = sizeof(sa);
99 if (getpeername(fd, (struct sockaddr*)&sa, &salen) < 0)
100 return NULL;
102 return addr_to_string(format, &sa, salen);
105 static int put_addr_qdict(QDict *qdict, struct sockaddr_storage *sa,
106 socklen_t salen)
108 char host[NI_MAXHOST];
109 char serv[NI_MAXSERV];
110 int err;
112 if ((err = getnameinfo((struct sockaddr *)sa, salen,
113 host, sizeof(host),
114 serv, sizeof(serv),
115 NI_NUMERICHOST | NI_NUMERICSERV)) != 0) {
116 VNC_DEBUG("Cannot resolve address %d: %s\n",
117 err, gai_strerror(err));
118 return -1;
121 qdict_put(qdict, "host", qstring_from_str(host));
122 qdict_put(qdict, "service", qstring_from_str(serv));
123 qdict_put(qdict, "family",qstring_from_str(inet_strfamily(sa->ss_family)));
125 return 0;
128 static int vnc_server_addr_put(QDict *qdict, int fd)
130 struct sockaddr_storage sa;
131 socklen_t salen;
133 salen = sizeof(sa);
134 if (getsockname(fd, (struct sockaddr*)&sa, &salen) < 0) {
135 return -1;
138 return put_addr_qdict(qdict, &sa, salen);
141 static int vnc_qdict_remote_addr(QDict *qdict, int fd)
143 struct sockaddr_storage sa;
144 socklen_t salen;
146 salen = sizeof(sa);
147 if (getpeername(fd, (struct sockaddr*)&sa, &salen) < 0) {
148 return -1;
151 return put_addr_qdict(qdict, &sa, salen);
154 static const char *vnc_auth_name(VncDisplay *vd) {
155 switch (vd->auth) {
156 case VNC_AUTH_INVALID:
157 return "invalid";
158 case VNC_AUTH_NONE:
159 return "none";
160 case VNC_AUTH_VNC:
161 return "vnc";
162 case VNC_AUTH_RA2:
163 return "ra2";
164 case VNC_AUTH_RA2NE:
165 return "ra2ne";
166 case VNC_AUTH_TIGHT:
167 return "tight";
168 case VNC_AUTH_ULTRA:
169 return "ultra";
170 case VNC_AUTH_TLS:
171 return "tls";
172 case VNC_AUTH_VENCRYPT:
173 #ifdef CONFIG_VNC_TLS
174 switch (vd->subauth) {
175 case VNC_AUTH_VENCRYPT_PLAIN:
176 return "vencrypt+plain";
177 case VNC_AUTH_VENCRYPT_TLSNONE:
178 return "vencrypt+tls+none";
179 case VNC_AUTH_VENCRYPT_TLSVNC:
180 return "vencrypt+tls+vnc";
181 case VNC_AUTH_VENCRYPT_TLSPLAIN:
182 return "vencrypt+tls+plain";
183 case VNC_AUTH_VENCRYPT_X509NONE:
184 return "vencrypt+x509+none";
185 case VNC_AUTH_VENCRYPT_X509VNC:
186 return "vencrypt+x509+vnc";
187 case VNC_AUTH_VENCRYPT_X509PLAIN:
188 return "vencrypt+x509+plain";
189 case VNC_AUTH_VENCRYPT_TLSSASL:
190 return "vencrypt+tls+sasl";
191 case VNC_AUTH_VENCRYPT_X509SASL:
192 return "vencrypt+x509+sasl";
193 default:
194 return "vencrypt";
196 #else
197 return "vencrypt";
198 #endif
199 case VNC_AUTH_SASL:
200 return "sasl";
202 return "unknown";
205 static int vnc_server_info_put(QDict *qdict)
207 if (vnc_server_addr_put(qdict, vnc_display->lsock) < 0) {
208 return -1;
211 qdict_put(qdict, "auth", qstring_from_str(vnc_auth_name(vnc_display)));
212 return 0;
215 static void vnc_client_cache_auth(VncState *client)
217 QDict *qdict;
219 if (!client->info) {
220 return;
223 qdict = qobject_to_qdict(client->info);
225 #ifdef CONFIG_VNC_TLS
226 if (client->tls.session &&
227 client->tls.dname) {
228 qdict_put(qdict, "x509_dname", qstring_from_str(client->tls.dname));
230 #endif
231 #ifdef CONFIG_VNC_SASL
232 if (client->sasl.conn &&
233 client->sasl.username) {
234 qdict_put(qdict, "sasl_username",
235 qstring_from_str(client->sasl.username));
237 #endif
240 static void vnc_client_cache_addr(VncState *client)
242 QDict *qdict;
244 qdict = qdict_new();
245 if (vnc_qdict_remote_addr(qdict, client->csock) < 0) {
246 QDECREF(qdict);
247 /* XXX: how to report the error? */
248 return;
251 client->info = QOBJECT(qdict);
254 static void vnc_qmp_event(VncState *vs, MonitorEvent event)
256 QDict *server;
257 QObject *data;
259 if (!vs->info) {
260 return;
263 server = qdict_new();
264 if (vnc_server_info_put(server) < 0) {
265 QDECREF(server);
266 return;
269 data = qobject_from_jsonf("{ 'client': %p, 'server': %p }",
270 vs->info, QOBJECT(server));
272 monitor_protocol_event(event, data);
274 qobject_incref(vs->info);
275 qobject_decref(data);
278 static void info_vnc_iter(QObject *obj, void *opaque)
280 QDict *client;
281 Monitor *mon = opaque;
283 client = qobject_to_qdict(obj);
284 monitor_printf(mon, "Client:\n");
285 monitor_printf(mon, " address: %s:%s\n",
286 qdict_get_str(client, "host"),
287 qdict_get_str(client, "service"));
289 #ifdef CONFIG_VNC_TLS
290 monitor_printf(mon, " x509_dname: %s\n",
291 qdict_haskey(client, "x509_dname") ?
292 qdict_get_str(client, "x509_dname") : "none");
293 #endif
294 #ifdef CONFIG_VNC_SASL
295 monitor_printf(mon, " username: %s\n",
296 qdict_haskey(client, "sasl_username") ?
297 qdict_get_str(client, "sasl_username") : "none");
298 #endif
301 void do_info_vnc_print(Monitor *mon, const QObject *data)
303 QDict *server;
304 QList *clients;
306 server = qobject_to_qdict(data);
307 if (qdict_get_bool(server, "enabled") == 0) {
308 monitor_printf(mon, "Server: disabled\n");
309 return;
312 monitor_printf(mon, "Server:\n");
313 monitor_printf(mon, " address: %s:%s\n",
314 qdict_get_str(server, "host"),
315 qdict_get_str(server, "service"));
316 monitor_printf(mon, " auth: %s\n", qdict_get_str(server, "auth"));
318 clients = qdict_get_qlist(server, "clients");
319 if (qlist_empty(clients)) {
320 monitor_printf(mon, "Client: none\n");
321 } else {
322 qlist_iter(clients, info_vnc_iter, mon);
327 * do_info_vnc(): Show VNC server information
329 * Return a QDict with server information. Connected clients are returned
330 * as a QList of QDicts.
332 * The main QDict contains the following:
334 * - "enabled": true or false
335 * - "host": server's IP address
336 * - "family": address family ("ipv4" or "ipv6")
337 * - "service": server's port number
338 * - "auth": authentication method
339 * - "clients": a QList of all connected clients
341 * Clients are described by a QDict, with the following information:
343 * - "host": client's IP address
344 * - "family": address family ("ipv4" or "ipv6")
345 * - "service": client's port number
346 * - "x509_dname": TLS dname (optional)
347 * - "sasl_username": SASL username (optional)
349 * Example:
351 * { "enabled": true, "host": "0.0.0.0", "service": "50402", "auth": "vnc",
352 * "family": "ipv4",
353 * "clients": [{ "host": "127.0.0.1", "service": "50401", "family": "ipv4" }]}
355 void do_info_vnc(Monitor *mon, QObject **ret_data)
357 if (vnc_display == NULL || vnc_display->display == NULL) {
358 *ret_data = qobject_from_jsonf("{ 'enabled': false }");
359 } else {
360 QList *clist;
361 VncState *client;
363 clist = qlist_new();
364 QTAILQ_FOREACH(client, &vnc_display->clients, next) {
365 if (client->info) {
366 /* incref so that it's not freed by upper layers */
367 qobject_incref(client->info);
368 qlist_append_obj(clist, client->info);
372 *ret_data = qobject_from_jsonf("{ 'enabled': true, 'clients': %p }",
373 QOBJECT(clist));
374 assert(*ret_data != NULL);
376 if (vnc_server_info_put(qobject_to_qdict(*ret_data)) < 0) {
377 qobject_decref(*ret_data);
378 *ret_data = NULL;
383 static inline uint32_t vnc_has_feature(VncState *vs, int feature) {
384 return (vs->features & (1 << feature));
387 /* TODO
388 1) Get the queue working for IO.
389 2) there is some weirdness when using the -S option (the screen is grey
390 and not totally invalidated
391 3) resolutions > 1024
394 static int vnc_update_client(VncState *vs, int has_dirty);
395 static void vnc_disconnect_start(VncState *vs);
396 static void vnc_disconnect_finish(VncState *vs);
397 static void vnc_init_timer(VncDisplay *vd);
398 static void vnc_remove_timer(VncDisplay *vd);
400 static void vnc_colordepth(VncState *vs);
401 static void framebuffer_update_request(VncState *vs, int incremental,
402 int x_position, int y_position,
403 int w, int h);
404 static void vnc_refresh(void *opaque);
405 static int vnc_refresh_server_surface(VncDisplay *vd);
407 static inline void vnc_set_bit(uint32_t *d, int k)
409 d[k >> 5] |= 1 << (k & 0x1f);
412 static inline void vnc_clear_bit(uint32_t *d, int k)
414 d[k >> 5] &= ~(1 << (k & 0x1f));
417 static inline void vnc_set_bits(uint32_t *d, int n, int nb_words)
419 int j;
421 j = 0;
422 while (n >= 32) {
423 d[j++] = -1;
424 n -= 32;
426 if (n > 0)
427 d[j++] = (1 << n) - 1;
428 while (j < nb_words)
429 d[j++] = 0;
432 static inline int vnc_get_bit(const uint32_t *d, int k)
434 return (d[k >> 5] >> (k & 0x1f)) & 1;
437 static inline int vnc_and_bits(const uint32_t *d1, const uint32_t *d2,
438 int nb_words)
440 int i;
441 for(i = 0; i < nb_words; i++) {
442 if ((d1[i] & d2[i]) != 0)
443 return 1;
445 return 0;
448 static void vnc_dpy_update(DisplayState *ds, int x, int y, int w, int h)
450 int i;
451 VncDisplay *vd = ds->opaque;
452 struct VncSurface *s = &vd->guest;
454 h += y;
456 /* round x down to ensure the loop only spans one 16-pixel block per,
457 iteration. otherwise, if (x % 16) != 0, the last iteration may span
458 two 16-pixel blocks but we only mark the first as dirty
460 w += (x % 16);
461 x -= (x % 16);
463 x = MIN(x, s->ds->width);
464 y = MIN(y, s->ds->height);
465 w = MIN(x + w, s->ds->width) - x;
466 h = MIN(h, s->ds->height);
468 for (; y < h; y++)
469 for (i = 0; i < w; i += 16)
470 vnc_set_bit(s->dirty[y], (x + i) / 16);
473 void vnc_framebuffer_update(VncState *vs, int x, int y, int w, int h,
474 int32_t encoding)
476 vnc_write_u16(vs, x);
477 vnc_write_u16(vs, y);
478 vnc_write_u16(vs, w);
479 vnc_write_u16(vs, h);
481 vnc_write_s32(vs, encoding);
484 void buffer_reserve(Buffer *buffer, size_t len)
486 if ((buffer->capacity - buffer->offset) < len) {
487 buffer->capacity += (len + 1024);
488 buffer->buffer = qemu_realloc(buffer->buffer, buffer->capacity);
489 if (buffer->buffer == NULL) {
490 fprintf(stderr, "vnc: out of memory\n");
491 exit(1);
496 int buffer_empty(Buffer *buffer)
498 return buffer->offset == 0;
501 uint8_t *buffer_end(Buffer *buffer)
503 return buffer->buffer + buffer->offset;
506 void buffer_reset(Buffer *buffer)
508 buffer->offset = 0;
511 void buffer_append(Buffer *buffer, const void *data, size_t len)
513 memcpy(buffer->buffer + buffer->offset, data, len);
514 buffer->offset += len;
517 static void vnc_dpy_resize(DisplayState *ds)
519 int size_changed;
520 VncDisplay *vd = ds->opaque;
521 VncState *vs;
523 /* server surface */
524 if (!vd->server)
525 vd->server = qemu_mallocz(sizeof(*vd->server));
526 if (vd->server->data)
527 qemu_free(vd->server->data);
528 *(vd->server) = *(ds->surface);
529 vd->server->data = qemu_mallocz(vd->server->linesize *
530 vd->server->height);
532 /* guest surface */
533 if (!vd->guest.ds)
534 vd->guest.ds = qemu_mallocz(sizeof(*vd->guest.ds));
535 if (ds_get_bytes_per_pixel(ds) != vd->guest.ds->pf.bytes_per_pixel)
536 console_color_init(ds);
537 size_changed = ds_get_width(ds) != vd->guest.ds->width ||
538 ds_get_height(ds) != vd->guest.ds->height;
539 *(vd->guest.ds) = *(ds->surface);
540 memset(vd->guest.dirty, 0xFF, sizeof(vd->guest.dirty));
542 QTAILQ_FOREACH(vs, &vd->clients, next) {
543 vnc_colordepth(vs);
544 if (size_changed) {
545 if (vs->csock != -1 && vnc_has_feature(vs, VNC_FEATURE_RESIZE)) {
546 vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
547 vnc_write_u8(vs, 0);
548 vnc_write_u16(vs, 1); /* number of rects */
549 vnc_framebuffer_update(vs, 0, 0, ds_get_width(ds), ds_get_height(ds),
550 VNC_ENCODING_DESKTOPRESIZE);
551 vnc_flush(vs);
554 if (vs->vd->cursor) {
555 vnc_cursor_define(vs);
557 memset(vs->dirty, 0xFF, sizeof(vs->dirty));
561 /* fastest code */
562 static void vnc_write_pixels_copy(VncState *vs, struct PixelFormat *pf,
563 void *pixels, int size)
565 vnc_write(vs, pixels, size);
568 /* slowest but generic code. */
569 void vnc_convert_pixel(VncState *vs, uint8_t *buf, uint32_t v)
571 uint8_t r, g, b;
572 VncDisplay *vd = vs->vd;
574 r = ((((v & vd->server->pf.rmask) >> vd->server->pf.rshift) << vs->clientds.pf.rbits) >>
575 vd->server->pf.rbits);
576 g = ((((v & vd->server->pf.gmask) >> vd->server->pf.gshift) << vs->clientds.pf.gbits) >>
577 vd->server->pf.gbits);
578 b = ((((v & vd->server->pf.bmask) >> vd->server->pf.bshift) << vs->clientds.pf.bbits) >>
579 vd->server->pf.bbits);
580 v = (r << vs->clientds.pf.rshift) |
581 (g << vs->clientds.pf.gshift) |
582 (b << vs->clientds.pf.bshift);
583 switch(vs->clientds.pf.bytes_per_pixel) {
584 case 1:
585 buf[0] = v;
586 break;
587 case 2:
588 if (vs->clientds.flags & QEMU_BIG_ENDIAN_FLAG) {
589 buf[0] = v >> 8;
590 buf[1] = v;
591 } else {
592 buf[1] = v >> 8;
593 buf[0] = v;
595 break;
596 default:
597 case 4:
598 if (vs->clientds.flags & QEMU_BIG_ENDIAN_FLAG) {
599 buf[0] = v >> 24;
600 buf[1] = v >> 16;
601 buf[2] = v >> 8;
602 buf[3] = v;
603 } else {
604 buf[3] = v >> 24;
605 buf[2] = v >> 16;
606 buf[1] = v >> 8;
607 buf[0] = v;
609 break;
613 static void vnc_write_pixels_generic(VncState *vs, struct PixelFormat *pf,
614 void *pixels1, int size)
616 uint8_t buf[4];
618 if (pf->bytes_per_pixel == 4) {
619 uint32_t *pixels = pixels1;
620 int n, i;
621 n = size >> 2;
622 for(i = 0; i < n; i++) {
623 vnc_convert_pixel(vs, buf, pixels[i]);
624 vnc_write(vs, buf, vs->clientds.pf.bytes_per_pixel);
626 } else if (pf->bytes_per_pixel == 2) {
627 uint16_t *pixels = pixels1;
628 int n, i;
629 n = size >> 1;
630 for(i = 0; i < n; i++) {
631 vnc_convert_pixel(vs, buf, pixels[i]);
632 vnc_write(vs, buf, vs->clientds.pf.bytes_per_pixel);
634 } else if (pf->bytes_per_pixel == 1) {
635 uint8_t *pixels = pixels1;
636 int n, i;
637 n = size;
638 for(i = 0; i < n; i++) {
639 vnc_convert_pixel(vs, buf, pixels[i]);
640 vnc_write(vs, buf, vs->clientds.pf.bytes_per_pixel);
642 } else {
643 fprintf(stderr, "vnc_write_pixels_generic: VncState color depth not supported\n");
647 void vnc_raw_send_framebuffer_update(VncState *vs, int x, int y, int w, int h)
649 int i;
650 uint8_t *row;
651 VncDisplay *vd = vs->vd;
653 row = vd->server->data + y * ds_get_linesize(vs->ds) + x * ds_get_bytes_per_pixel(vs->ds);
654 for (i = 0; i < h; i++) {
655 vs->write_pixels(vs, &vd->server->pf, row, w * ds_get_bytes_per_pixel(vs->ds));
656 row += ds_get_linesize(vs->ds);
660 static void send_framebuffer_update(VncState *vs, int x, int y, int w, int h)
662 switch(vs->vnc_encoding) {
663 case VNC_ENCODING_ZLIB:
664 vnc_hextile_send_framebuffer_update(vs, x, y, w, h);
665 break;
666 case VNC_ENCODING_HEXTILE:
667 vnc_framebuffer_update(vs, x, y, w, h, VNC_ENCODING_HEXTILE);
668 vnc_hextile_send_framebuffer_update(vs, x, y, w, h);
669 break;
670 default:
671 vnc_framebuffer_update(vs, x, y, w, h, VNC_ENCODING_RAW);
672 vnc_raw_send_framebuffer_update(vs, x, y, w, h);
673 break;
677 static void vnc_copy(VncState *vs, int src_x, int src_y, int dst_x, int dst_y, int w, int h)
679 /* send bitblit op to the vnc client */
680 vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
681 vnc_write_u8(vs, 0);
682 vnc_write_u16(vs, 1); /* number of rects */
683 vnc_framebuffer_update(vs, dst_x, dst_y, w, h, VNC_ENCODING_COPYRECT);
684 vnc_write_u16(vs, src_x);
685 vnc_write_u16(vs, src_y);
686 vnc_flush(vs);
689 static void vnc_dpy_copy(DisplayState *ds, int src_x, int src_y, int dst_x, int dst_y, int w, int h)
691 VncDisplay *vd = ds->opaque;
692 VncState *vs, *vn;
693 uint8_t *src_row;
694 uint8_t *dst_row;
695 int i,x,y,pitch,depth,inc,w_lim,s;
696 int cmp_bytes;
698 vnc_refresh_server_surface(vd);
699 QTAILQ_FOREACH_SAFE(vs, &vd->clients, next, vn) {
700 if (vnc_has_feature(vs, VNC_FEATURE_COPYRECT)) {
701 vs->force_update = 1;
702 vnc_update_client(vs, 1);
703 /* vs might be free()ed here */
707 /* do bitblit op on the local surface too */
708 pitch = ds_get_linesize(vd->ds);
709 depth = ds_get_bytes_per_pixel(vd->ds);
710 src_row = vd->server->data + pitch * src_y + depth * src_x;
711 dst_row = vd->server->data + pitch * dst_y + depth * dst_x;
712 y = dst_y;
713 inc = 1;
714 if (dst_y > src_y) {
715 /* copy backwards */
716 src_row += pitch * (h-1);
717 dst_row += pitch * (h-1);
718 pitch = -pitch;
719 y = dst_y + h - 1;
720 inc = -1;
722 w_lim = w - (16 - (dst_x % 16));
723 if (w_lim < 0)
724 w_lim = w;
725 else
726 w_lim = w - (w_lim % 16);
727 for (i = 0; i < h; i++) {
728 for (x = 0; x <= w_lim;
729 x += s, src_row += cmp_bytes, dst_row += cmp_bytes) {
730 if (x == w_lim) {
731 if ((s = w - w_lim) == 0)
732 break;
733 } else if (!x) {
734 s = (16 - (dst_x % 16));
735 s = MIN(s, w_lim);
736 } else {
737 s = 16;
739 cmp_bytes = s * depth;
740 if (memcmp(src_row, dst_row, cmp_bytes) == 0)
741 continue;
742 memmove(dst_row, src_row, cmp_bytes);
743 QTAILQ_FOREACH(vs, &vd->clients, next) {
744 if (!vnc_has_feature(vs, VNC_FEATURE_COPYRECT)) {
745 vnc_set_bit(vs->dirty[y], ((x + dst_x) / 16));
749 src_row += pitch - w * depth;
750 dst_row += pitch - w * depth;
751 y += inc;
754 QTAILQ_FOREACH(vs, &vd->clients, next) {
755 if (vnc_has_feature(vs, VNC_FEATURE_COPYRECT)) {
756 vnc_copy(vs, src_x, src_y, dst_x, dst_y, w, h);
761 static void vnc_mouse_set(int x, int y, int visible)
763 /* can we ask the client(s) to move the pointer ??? */
766 static int vnc_cursor_define(VncState *vs)
768 QEMUCursor *c = vs->vd->cursor;
769 PixelFormat pf = qemu_default_pixelformat(32);
770 int isize;
772 if (vnc_has_feature(vs, VNC_FEATURE_RICH_CURSOR)) {
773 vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
774 vnc_write_u8(vs, 0); /* padding */
775 vnc_write_u16(vs, 1); /* # of rects */
776 vnc_framebuffer_update(vs, c->hot_x, c->hot_y, c->width, c->height,
777 VNC_ENCODING_RICH_CURSOR);
778 isize = c->width * c->height * vs->clientds.pf.bytes_per_pixel;
779 vnc_write_pixels_generic(vs, &pf, c->data, isize);
780 vnc_write(vs, vs->vd->cursor_mask, vs->vd->cursor_msize);
781 return 0;
783 return -1;
786 static void vnc_dpy_cursor_define(QEMUCursor *c)
788 VncDisplay *vd = vnc_display;
789 VncState *vs;
791 cursor_put(vd->cursor);
792 qemu_free(vd->cursor_mask);
794 vd->cursor = c;
795 cursor_get(vd->cursor);
796 vd->cursor_msize = cursor_get_mono_bpl(c) * c->height;
797 vd->cursor_mask = qemu_mallocz(vd->cursor_msize);
798 cursor_get_mono_mask(c, 0, vd->cursor_mask);
800 QTAILQ_FOREACH(vs, &vd->clients, next) {
801 vnc_cursor_define(vs);
805 static int find_and_clear_dirty_height(struct VncState *vs,
806 int y, int last_x, int x)
808 int h;
809 VncDisplay *vd = vs->vd;
811 for (h = 1; h < (vd->server->height - y); h++) {
812 int tmp_x;
813 if (!vnc_get_bit(vs->dirty[y + h], last_x))
814 break;
815 for (tmp_x = last_x; tmp_x < x; tmp_x++)
816 vnc_clear_bit(vs->dirty[y + h], tmp_x);
819 return h;
822 static int vnc_update_client(VncState *vs, int has_dirty)
824 if (vs->need_update && vs->csock != -1) {
825 VncDisplay *vd = vs->vd;
826 int y;
827 int n_rectangles;
828 int saved_offset;
830 if (vs->output.offset && !vs->audio_cap && !vs->force_update)
831 /* kernel send buffers are full -> drop frames to throttle */
832 return 0;
834 if (!has_dirty && !vs->audio_cap && !vs->force_update)
835 return 0;
838 * Send screen updates to the vnc client using the server
839 * surface and server dirty map. guest surface updates
840 * happening in parallel don't disturb us, the next pass will
841 * send them to the client.
843 n_rectangles = 0;
844 vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
845 vnc_write_u8(vs, 0);
846 saved_offset = vs->output.offset;
847 vnc_write_u16(vs, 0);
849 for (y = 0; y < vd->server->height; y++) {
850 int x;
851 int last_x = -1;
852 for (x = 0; x < vd->server->width / 16; x++) {
853 if (vnc_get_bit(vs->dirty[y], x)) {
854 if (last_x == -1) {
855 last_x = x;
857 vnc_clear_bit(vs->dirty[y], x);
858 } else {
859 if (last_x != -1) {
860 int h = find_and_clear_dirty_height(vs, y, last_x, x);
861 send_framebuffer_update(vs, last_x * 16, y, (x - last_x) * 16, h);
862 n_rectangles++;
864 last_x = -1;
867 if (last_x != -1) {
868 int h = find_and_clear_dirty_height(vs, y, last_x, x);
869 send_framebuffer_update(vs, last_x * 16, y, (x - last_x) * 16, h);
870 n_rectangles++;
873 vs->output.buffer[saved_offset] = (n_rectangles >> 8) & 0xFF;
874 vs->output.buffer[saved_offset + 1] = n_rectangles & 0xFF;
875 vnc_flush(vs);
876 vs->force_update = 0;
877 return n_rectangles;
880 if (vs->csock == -1)
881 vnc_disconnect_finish(vs);
883 return 0;
886 /* audio */
887 static void audio_capture_notify(void *opaque, audcnotification_e cmd)
889 VncState *vs = opaque;
891 switch (cmd) {
892 case AUD_CNOTIFY_DISABLE:
893 vnc_write_u8(vs, VNC_MSG_SERVER_QEMU);
894 vnc_write_u8(vs, VNC_MSG_SERVER_QEMU_AUDIO);
895 vnc_write_u16(vs, VNC_MSG_SERVER_QEMU_AUDIO_END);
896 vnc_flush(vs);
897 break;
899 case AUD_CNOTIFY_ENABLE:
900 vnc_write_u8(vs, VNC_MSG_SERVER_QEMU);
901 vnc_write_u8(vs, VNC_MSG_SERVER_QEMU_AUDIO);
902 vnc_write_u16(vs, VNC_MSG_SERVER_QEMU_AUDIO_BEGIN);
903 vnc_flush(vs);
904 break;
908 static void audio_capture_destroy(void *opaque)
912 static void audio_capture(void *opaque, void *buf, int size)
914 VncState *vs = opaque;
916 vnc_write_u8(vs, VNC_MSG_SERVER_QEMU);
917 vnc_write_u8(vs, VNC_MSG_SERVER_QEMU_AUDIO);
918 vnc_write_u16(vs, VNC_MSG_SERVER_QEMU_AUDIO_DATA);
919 vnc_write_u32(vs, size);
920 vnc_write(vs, buf, size);
921 vnc_flush(vs);
924 static void audio_add(VncState *vs)
926 struct audio_capture_ops ops;
928 if (vs->audio_cap) {
929 monitor_printf(default_mon, "audio already running\n");
930 return;
933 ops.notify = audio_capture_notify;
934 ops.destroy = audio_capture_destroy;
935 ops.capture = audio_capture;
937 vs->audio_cap = AUD_add_capture(&vs->as, &ops, vs);
938 if (!vs->audio_cap) {
939 monitor_printf(default_mon, "Failed to add audio capture\n");
943 static void audio_del(VncState *vs)
945 if (vs->audio_cap) {
946 AUD_del_capture(vs->audio_cap, vs);
947 vs->audio_cap = NULL;
951 static void vnc_disconnect_start(VncState *vs)
953 if (vs->csock == -1)
954 return;
955 qemu_set_fd_handler2(vs->csock, NULL, NULL, NULL, NULL);
956 closesocket(vs->csock);
957 vs->csock = -1;
960 static void vnc_disconnect_finish(VncState *vs)
962 vnc_qmp_event(vs, QEVENT_VNC_DISCONNECTED);
964 if (vs->input.buffer) {
965 qemu_free(vs->input.buffer);
966 vs->input.buffer = NULL;
968 if (vs->output.buffer) {
969 qemu_free(vs->output.buffer);
970 vs->output.buffer = NULL;
973 qobject_decref(vs->info);
975 #ifdef CONFIG_VNC_TLS
976 vnc_tls_client_cleanup(vs);
977 #endif /* CONFIG_VNC_TLS */
978 #ifdef CONFIG_VNC_SASL
979 vnc_sasl_client_cleanup(vs);
980 #endif /* CONFIG_VNC_SASL */
981 audio_del(vs);
983 QTAILQ_REMOVE(&vs->vd->clients, vs, next);
985 if (QTAILQ_EMPTY(&vs->vd->clients)) {
986 dcl->idle = 1;
989 qemu_remove_mouse_mode_change_notifier(&vs->mouse_mode_notifier);
990 vnc_remove_timer(vs->vd);
991 if (vs->vd->lock_key_sync)
992 qemu_remove_led_event_handler(vs->led);
993 qemu_free(vs);
996 int vnc_client_io_error(VncState *vs, int ret, int last_errno)
998 if (ret == 0 || ret == -1) {
999 if (ret == -1) {
1000 switch (last_errno) {
1001 case EINTR:
1002 case EAGAIN:
1003 #ifdef _WIN32
1004 case WSAEWOULDBLOCK:
1005 #endif
1006 return 0;
1007 default:
1008 break;
1012 VNC_DEBUG("Closing down client sock: ret %d, errno %d\n",
1013 ret, ret < 0 ? last_errno : 0);
1014 vnc_disconnect_start(vs);
1016 return 0;
1018 return ret;
1022 void vnc_client_error(VncState *vs)
1024 VNC_DEBUG("Closing down client sock: protocol error\n");
1025 vnc_disconnect_start(vs);
1030 * Called to write a chunk of data to the client socket. The data may
1031 * be the raw data, or may have already been encoded by SASL.
1032 * The data will be written either straight onto the socket, or
1033 * written via the GNUTLS wrappers, if TLS/SSL encryption is enabled
1035 * NB, it is theoretically possible to have 2 layers of encryption,
1036 * both SASL, and this TLS layer. It is highly unlikely in practice
1037 * though, since SASL encryption will typically be a no-op if TLS
1038 * is active
1040 * Returns the number of bytes written, which may be less than
1041 * the requested 'datalen' if the socket would block. Returns
1042 * -1 on error, and disconnects the client socket.
1044 long vnc_client_write_buf(VncState *vs, const uint8_t *data, size_t datalen)
1046 long ret;
1047 #ifdef CONFIG_VNC_TLS
1048 if (vs->tls.session) {
1049 ret = gnutls_write(vs->tls.session, data, datalen);
1050 if (ret < 0) {
1051 if (ret == GNUTLS_E_AGAIN)
1052 errno = EAGAIN;
1053 else
1054 errno = EIO;
1055 ret = -1;
1057 } else
1058 #endif /* CONFIG_VNC_TLS */
1059 ret = send(vs->csock, (const void *)data, datalen, 0);
1060 VNC_DEBUG("Wrote wire %p %zd -> %ld\n", data, datalen, ret);
1061 return vnc_client_io_error(vs, ret, socket_error());
1066 * Called to write buffered data to the client socket, when not
1067 * using any SASL SSF encryption layers. Will write as much data
1068 * as possible without blocking. If all buffered data is written,
1069 * will switch the FD poll() handler back to read monitoring.
1071 * Returns the number of bytes written, which may be less than
1072 * the buffered output data if the socket would block. Returns
1073 * -1 on error, and disconnects the client socket.
1075 static long vnc_client_write_plain(VncState *vs)
1077 long ret;
1079 #ifdef CONFIG_VNC_SASL
1080 VNC_DEBUG("Write Plain: Pending output %p size %zd offset %zd. Wait SSF %d\n",
1081 vs->output.buffer, vs->output.capacity, vs->output.offset,
1082 vs->sasl.waitWriteSSF);
1084 if (vs->sasl.conn &&
1085 vs->sasl.runSSF &&
1086 vs->sasl.waitWriteSSF) {
1087 ret = vnc_client_write_buf(vs, vs->output.buffer, vs->sasl.waitWriteSSF);
1088 if (ret)
1089 vs->sasl.waitWriteSSF -= ret;
1090 } else
1091 #endif /* CONFIG_VNC_SASL */
1092 ret = vnc_client_write_buf(vs, vs->output.buffer, vs->output.offset);
1093 if (!ret)
1094 return 0;
1096 memmove(vs->output.buffer, vs->output.buffer + ret, (vs->output.offset - ret));
1097 vs->output.offset -= ret;
1099 if (vs->output.offset == 0) {
1100 qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, vs);
1103 return ret;
1108 * First function called whenever there is data to be written to
1109 * the client socket. Will delegate actual work according to whether
1110 * SASL SSF layers are enabled (thus requiring encryption calls)
1112 void vnc_client_write(void *opaque)
1114 VncState *vs = opaque;
1116 #ifdef CONFIG_VNC_SASL
1117 if (vs->sasl.conn &&
1118 vs->sasl.runSSF &&
1119 !vs->sasl.waitWriteSSF) {
1120 vnc_client_write_sasl(vs);
1121 } else
1122 #endif /* CONFIG_VNC_SASL */
1123 vnc_client_write_plain(vs);
1126 void vnc_read_when(VncState *vs, VncReadEvent *func, size_t expecting)
1128 vs->read_handler = func;
1129 vs->read_handler_expect = expecting;
1134 * Called to read a chunk of data from the client socket. The data may
1135 * be the raw data, or may need to be further decoded by SASL.
1136 * The data will be read either straight from to the socket, or
1137 * read via the GNUTLS wrappers, if TLS/SSL encryption is enabled
1139 * NB, it is theoretically possible to have 2 layers of encryption,
1140 * both SASL, and this TLS layer. It is highly unlikely in practice
1141 * though, since SASL encryption will typically be a no-op if TLS
1142 * is active
1144 * Returns the number of bytes read, which may be less than
1145 * the requested 'datalen' if the socket would block. Returns
1146 * -1 on error, and disconnects the client socket.
1148 long vnc_client_read_buf(VncState *vs, uint8_t *data, size_t datalen)
1150 long ret;
1151 #ifdef CONFIG_VNC_TLS
1152 if (vs->tls.session) {
1153 ret = gnutls_read(vs->tls.session, data, datalen);
1154 if (ret < 0) {
1155 if (ret == GNUTLS_E_AGAIN)
1156 errno = EAGAIN;
1157 else
1158 errno = EIO;
1159 ret = -1;
1161 } else
1162 #endif /* CONFIG_VNC_TLS */
1163 ret = recv(vs->csock, (void *)data, datalen, 0);
1164 VNC_DEBUG("Read wire %p %zd -> %ld\n", data, datalen, ret);
1165 return vnc_client_io_error(vs, ret, socket_error());
1170 * Called to read data from the client socket to the input buffer,
1171 * when not using any SASL SSF encryption layers. Will read as much
1172 * data as possible without blocking.
1174 * Returns the number of bytes read. Returns -1 on error, and
1175 * disconnects the client socket.
1177 static long vnc_client_read_plain(VncState *vs)
1179 int ret;
1180 VNC_DEBUG("Read plain %p size %zd offset %zd\n",
1181 vs->input.buffer, vs->input.capacity, vs->input.offset);
1182 buffer_reserve(&vs->input, 4096);
1183 ret = vnc_client_read_buf(vs, buffer_end(&vs->input), 4096);
1184 if (!ret)
1185 return 0;
1186 vs->input.offset += ret;
1187 return ret;
1192 * First function called whenever there is more data to be read from
1193 * the client socket. Will delegate actual work according to whether
1194 * SASL SSF layers are enabled (thus requiring decryption calls)
1196 void vnc_client_read(void *opaque)
1198 VncState *vs = opaque;
1199 long ret;
1201 #ifdef CONFIG_VNC_SASL
1202 if (vs->sasl.conn && vs->sasl.runSSF)
1203 ret = vnc_client_read_sasl(vs);
1204 else
1205 #endif /* CONFIG_VNC_SASL */
1206 ret = vnc_client_read_plain(vs);
1207 if (!ret) {
1208 if (vs->csock == -1)
1209 vnc_disconnect_finish(vs);
1210 return;
1213 while (vs->read_handler && vs->input.offset >= vs->read_handler_expect) {
1214 size_t len = vs->read_handler_expect;
1215 int ret;
1217 ret = vs->read_handler(vs, vs->input.buffer, len);
1218 if (vs->csock == -1) {
1219 vnc_disconnect_finish(vs);
1220 return;
1223 if (!ret) {
1224 memmove(vs->input.buffer, vs->input.buffer + len, (vs->input.offset - len));
1225 vs->input.offset -= len;
1226 } else {
1227 vs->read_handler_expect = ret;
1232 void vnc_write(VncState *vs, const void *data, size_t len)
1234 buffer_reserve(&vs->output, len);
1236 if (vs->csock != -1 && buffer_empty(&vs->output)) {
1237 qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, vnc_client_write, vs);
1240 buffer_append(&vs->output, data, len);
1243 void vnc_write_s32(VncState *vs, int32_t value)
1245 vnc_write_u32(vs, *(uint32_t *)&value);
1248 void vnc_write_u32(VncState *vs, uint32_t value)
1250 uint8_t buf[4];
1252 buf[0] = (value >> 24) & 0xFF;
1253 buf[1] = (value >> 16) & 0xFF;
1254 buf[2] = (value >> 8) & 0xFF;
1255 buf[3] = value & 0xFF;
1257 vnc_write(vs, buf, 4);
1260 void vnc_write_u16(VncState *vs, uint16_t value)
1262 uint8_t buf[2];
1264 buf[0] = (value >> 8) & 0xFF;
1265 buf[1] = value & 0xFF;
1267 vnc_write(vs, buf, 2);
1270 void vnc_write_u8(VncState *vs, uint8_t value)
1272 vnc_write(vs, (char *)&value, 1);
1275 void vnc_flush(VncState *vs)
1277 if (vs->csock != -1 && vs->output.offset)
1278 vnc_client_write(vs);
1281 uint8_t read_u8(uint8_t *data, size_t offset)
1283 return data[offset];
1286 uint16_t read_u16(uint8_t *data, size_t offset)
1288 return ((data[offset] & 0xFF) << 8) | (data[offset + 1] & 0xFF);
1291 int32_t read_s32(uint8_t *data, size_t offset)
1293 return (int32_t)((data[offset] << 24) | (data[offset + 1] << 16) |
1294 (data[offset + 2] << 8) | data[offset + 3]);
1297 uint32_t read_u32(uint8_t *data, size_t offset)
1299 return ((data[offset] << 24) | (data[offset + 1] << 16) |
1300 (data[offset + 2] << 8) | data[offset + 3]);
1303 static void client_cut_text(VncState *vs, size_t len, uint8_t *text)
1307 static void check_pointer_type_change(Notifier *notifier)
1309 VncState *vs = container_of(notifier, VncState, mouse_mode_notifier);
1310 int absolute = kbd_mouse_is_absolute();
1312 if (vnc_has_feature(vs, VNC_FEATURE_POINTER_TYPE_CHANGE) && vs->absolute != absolute) {
1313 vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
1314 vnc_write_u8(vs, 0);
1315 vnc_write_u16(vs, 1);
1316 vnc_framebuffer_update(vs, absolute, 0,
1317 ds_get_width(vs->ds), ds_get_height(vs->ds),
1318 VNC_ENCODING_POINTER_TYPE_CHANGE);
1319 vnc_flush(vs);
1321 vs->absolute = absolute;
1324 static void pointer_event(VncState *vs, int button_mask, int x, int y)
1326 int buttons = 0;
1327 int dz = 0;
1329 if (button_mask & 0x01)
1330 buttons |= MOUSE_EVENT_LBUTTON;
1331 if (button_mask & 0x02)
1332 buttons |= MOUSE_EVENT_MBUTTON;
1333 if (button_mask & 0x04)
1334 buttons |= MOUSE_EVENT_RBUTTON;
1335 if (button_mask & 0x08)
1336 dz = -1;
1337 if (button_mask & 0x10)
1338 dz = 1;
1340 if (vs->absolute) {
1341 kbd_mouse_event(ds_get_width(vs->ds) > 1 ?
1342 x * 0x7FFF / (ds_get_width(vs->ds) - 1) : 0x4000,
1343 ds_get_height(vs->ds) > 1 ?
1344 y * 0x7FFF / (ds_get_height(vs->ds) - 1) : 0x4000,
1345 dz, buttons);
1346 } else if (vnc_has_feature(vs, VNC_FEATURE_POINTER_TYPE_CHANGE)) {
1347 x -= 0x7FFF;
1348 y -= 0x7FFF;
1350 kbd_mouse_event(x, y, dz, buttons);
1351 } else {
1352 if (vs->last_x != -1)
1353 kbd_mouse_event(x - vs->last_x,
1354 y - vs->last_y,
1355 dz, buttons);
1356 vs->last_x = x;
1357 vs->last_y = y;
1361 static void reset_keys(VncState *vs)
1363 int i;
1364 for(i = 0; i < 256; i++) {
1365 if (vs->modifiers_state[i]) {
1366 if (i & SCANCODE_GREY)
1367 kbd_put_keycode(SCANCODE_EMUL0);
1368 kbd_put_keycode(i | SCANCODE_UP);
1369 vs->modifiers_state[i] = 0;
1374 static void press_key(VncState *vs, int keysym)
1376 int keycode = keysym2scancode(vs->vd->kbd_layout, keysym) & SCANCODE_KEYMASK;
1377 if (keycode & SCANCODE_GREY)
1378 kbd_put_keycode(SCANCODE_EMUL0);
1379 kbd_put_keycode(keycode & SCANCODE_KEYCODEMASK);
1380 if (keycode & SCANCODE_GREY)
1381 kbd_put_keycode(SCANCODE_EMUL0);
1382 kbd_put_keycode(keycode | SCANCODE_UP);
1385 static void kbd_leds(void *opaque, int ledstate)
1387 VncState *vs = opaque;
1388 int caps, num;
1390 caps = ledstate & QEMU_CAPS_LOCK_LED ? 1 : 0;
1391 num = ledstate & QEMU_NUM_LOCK_LED ? 1 : 0;
1393 if (vs->modifiers_state[0x3a] != caps) {
1394 vs->modifiers_state[0x3a] = caps;
1396 if (vs->modifiers_state[0x45] != num) {
1397 vs->modifiers_state[0x45] = num;
1401 static void do_key_event(VncState *vs, int down, int keycode, int sym)
1403 /* QEMU console switch */
1404 switch(keycode) {
1405 case 0x2a: /* Left Shift */
1406 case 0x36: /* Right Shift */
1407 case 0x1d: /* Left CTRL */
1408 case 0x9d: /* Right CTRL */
1409 case 0x38: /* Left ALT */
1410 case 0xb8: /* Right ALT */
1411 if (down)
1412 vs->modifiers_state[keycode] = 1;
1413 else
1414 vs->modifiers_state[keycode] = 0;
1415 break;
1416 case 0x02 ... 0x0a: /* '1' to '9' keys */
1417 if (down && vs->modifiers_state[0x1d] && vs->modifiers_state[0x38]) {
1418 /* Reset the modifiers sent to the current console */
1419 reset_keys(vs);
1420 console_select(keycode - 0x02);
1421 return;
1423 break;
1424 case 0x3a: /* CapsLock */
1425 case 0x45: /* NumLock */
1426 if (down)
1427 vs->modifiers_state[keycode] ^= 1;
1428 break;
1431 if (vs->vd->lock_key_sync &&
1432 keycode_is_keypad(vs->vd->kbd_layout, keycode)) {
1433 /* If the numlock state needs to change then simulate an additional
1434 keypress before sending this one. This will happen if the user
1435 toggles numlock away from the VNC window.
1437 if (keysym_is_numlock(vs->vd->kbd_layout, sym & 0xFFFF)) {
1438 if (!vs->modifiers_state[0x45]) {
1439 vs->modifiers_state[0x45] = 1;
1440 press_key(vs, 0xff7f);
1442 } else {
1443 if (vs->modifiers_state[0x45]) {
1444 vs->modifiers_state[0x45] = 0;
1445 press_key(vs, 0xff7f);
1450 if (vs->vd->lock_key_sync &&
1451 ((sym >= 'A' && sym <= 'Z') || (sym >= 'a' && sym <= 'z'))) {
1452 /* If the capslock state needs to change then simulate an additional
1453 keypress before sending this one. This will happen if the user
1454 toggles capslock away from the VNC window.
1456 int uppercase = !!(sym >= 'A' && sym <= 'Z');
1457 int shift = !!(vs->modifiers_state[0x2a] | vs->modifiers_state[0x36]);
1458 int capslock = !!(vs->modifiers_state[0x3a]);
1459 if (capslock) {
1460 if (uppercase == shift) {
1461 vs->modifiers_state[0x3a] = 0;
1462 press_key(vs, 0xffe5);
1464 } else {
1465 if (uppercase != shift) {
1466 vs->modifiers_state[0x3a] = 1;
1467 press_key(vs, 0xffe5);
1472 if (is_graphic_console()) {
1473 if (keycode & SCANCODE_GREY)
1474 kbd_put_keycode(SCANCODE_EMUL0);
1475 if (down)
1476 kbd_put_keycode(keycode & SCANCODE_KEYCODEMASK);
1477 else
1478 kbd_put_keycode(keycode | SCANCODE_UP);
1479 } else {
1480 /* QEMU console emulation */
1481 if (down) {
1482 int numlock = vs->modifiers_state[0x45];
1483 switch (keycode) {
1484 case 0x2a: /* Left Shift */
1485 case 0x36: /* Right Shift */
1486 case 0x1d: /* Left CTRL */
1487 case 0x9d: /* Right CTRL */
1488 case 0x38: /* Left ALT */
1489 case 0xb8: /* Right ALT */
1490 break;
1491 case 0xc8:
1492 kbd_put_keysym(QEMU_KEY_UP);
1493 break;
1494 case 0xd0:
1495 kbd_put_keysym(QEMU_KEY_DOWN);
1496 break;
1497 case 0xcb:
1498 kbd_put_keysym(QEMU_KEY_LEFT);
1499 break;
1500 case 0xcd:
1501 kbd_put_keysym(QEMU_KEY_RIGHT);
1502 break;
1503 case 0xd3:
1504 kbd_put_keysym(QEMU_KEY_DELETE);
1505 break;
1506 case 0xc7:
1507 kbd_put_keysym(QEMU_KEY_HOME);
1508 break;
1509 case 0xcf:
1510 kbd_put_keysym(QEMU_KEY_END);
1511 break;
1512 case 0xc9:
1513 kbd_put_keysym(QEMU_KEY_PAGEUP);
1514 break;
1515 case 0xd1:
1516 kbd_put_keysym(QEMU_KEY_PAGEDOWN);
1517 break;
1519 case 0x47:
1520 kbd_put_keysym(numlock ? '7' : QEMU_KEY_HOME);
1521 break;
1522 case 0x48:
1523 kbd_put_keysym(numlock ? '8' : QEMU_KEY_UP);
1524 break;
1525 case 0x49:
1526 kbd_put_keysym(numlock ? '9' : QEMU_KEY_PAGEUP);
1527 break;
1528 case 0x4b:
1529 kbd_put_keysym(numlock ? '4' : QEMU_KEY_LEFT);
1530 break;
1531 case 0x4c:
1532 kbd_put_keysym('5');
1533 break;
1534 case 0x4d:
1535 kbd_put_keysym(numlock ? '6' : QEMU_KEY_RIGHT);
1536 break;
1537 case 0x4f:
1538 kbd_put_keysym(numlock ? '1' : QEMU_KEY_END);
1539 break;
1540 case 0x50:
1541 kbd_put_keysym(numlock ? '2' : QEMU_KEY_DOWN);
1542 break;
1543 case 0x51:
1544 kbd_put_keysym(numlock ? '3' : QEMU_KEY_PAGEDOWN);
1545 break;
1546 case 0x52:
1547 kbd_put_keysym('0');
1548 break;
1549 case 0x53:
1550 kbd_put_keysym(numlock ? '.' : QEMU_KEY_DELETE);
1551 break;
1553 case 0xb5:
1554 kbd_put_keysym('/');
1555 break;
1556 case 0x37:
1557 kbd_put_keysym('*');
1558 break;
1559 case 0x4a:
1560 kbd_put_keysym('-');
1561 break;
1562 case 0x4e:
1563 kbd_put_keysym('+');
1564 break;
1565 case 0x9c:
1566 kbd_put_keysym('\n');
1567 break;
1569 default:
1570 kbd_put_keysym(sym);
1571 break;
1577 static void key_event(VncState *vs, int down, uint32_t sym)
1579 int keycode;
1580 int lsym = sym;
1582 if (lsym >= 'A' && lsym <= 'Z' && is_graphic_console()) {
1583 lsym = lsym - 'A' + 'a';
1586 keycode = keysym2scancode(vs->vd->kbd_layout, lsym & 0xFFFF) & SCANCODE_KEYMASK;
1587 do_key_event(vs, down, keycode, sym);
1590 static void ext_key_event(VncState *vs, int down,
1591 uint32_t sym, uint16_t keycode)
1593 /* if the user specifies a keyboard layout, always use it */
1594 if (keyboard_layout)
1595 key_event(vs, down, sym);
1596 else
1597 do_key_event(vs, down, keycode, sym);
1600 static void framebuffer_update_request(VncState *vs, int incremental,
1601 int x_position, int y_position,
1602 int w, int h)
1604 if (y_position > ds_get_height(vs->ds))
1605 y_position = ds_get_height(vs->ds);
1606 if (y_position + h >= ds_get_height(vs->ds))
1607 h = ds_get_height(vs->ds) - y_position;
1609 int i;
1610 vs->need_update = 1;
1611 if (!incremental) {
1612 vs->force_update = 1;
1613 for (i = 0; i < h; i++) {
1614 vnc_set_bits(vs->dirty[y_position + i],
1615 (ds_get_width(vs->ds) / 16), VNC_DIRTY_WORDS);
1620 static void send_ext_key_event_ack(VncState *vs)
1622 vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
1623 vnc_write_u8(vs, 0);
1624 vnc_write_u16(vs, 1);
1625 vnc_framebuffer_update(vs, 0, 0, ds_get_width(vs->ds), ds_get_height(vs->ds),
1626 VNC_ENCODING_EXT_KEY_EVENT);
1627 vnc_flush(vs);
1630 static void send_ext_audio_ack(VncState *vs)
1632 vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
1633 vnc_write_u8(vs, 0);
1634 vnc_write_u16(vs, 1);
1635 vnc_framebuffer_update(vs, 0, 0, ds_get_width(vs->ds), ds_get_height(vs->ds),
1636 VNC_ENCODING_AUDIO);
1637 vnc_flush(vs);
1640 static void set_encodings(VncState *vs, int32_t *encodings, size_t n_encodings)
1642 int i;
1643 unsigned int enc = 0;
1645 vnc_zlib_init(vs);
1646 vs->features = 0;
1647 vs->vnc_encoding = -1;
1648 vs->tight_compression = 9;
1649 vs->tight_quality = 9;
1650 vs->absolute = -1;
1652 for (i = n_encodings - 1; i >= 0; i--) {
1653 enc = encodings[i];
1654 switch (enc) {
1655 case VNC_ENCODING_RAW:
1656 if (vs->vnc_encoding != -1) {
1657 vs->vnc_encoding = enc;
1659 break;
1660 case VNC_ENCODING_COPYRECT:
1661 vs->features |= VNC_FEATURE_COPYRECT_MASK;
1662 break;
1663 case VNC_ENCODING_HEXTILE:
1664 vs->features |= VNC_FEATURE_HEXTILE_MASK;
1665 if (vs->vnc_encoding != -1) {
1666 vs->vnc_encoding = enc;
1668 break;
1669 case VNC_ENCODING_ZLIB:
1670 vs->features |= VNC_FEATURE_ZLIB_MASK;
1671 if (vs->vnc_encoding != -1) {
1672 vs->vnc_encoding = enc;
1674 break;
1675 case VNC_ENCODING_DESKTOPRESIZE:
1676 vs->features |= VNC_FEATURE_RESIZE_MASK;
1677 break;
1678 case VNC_ENCODING_POINTER_TYPE_CHANGE:
1679 vs->features |= VNC_FEATURE_POINTER_TYPE_CHANGE_MASK;
1680 break;
1681 case VNC_ENCODING_RICH_CURSOR:
1682 vs->features |= VNC_FEATURE_RICH_CURSOR_MASK;
1683 break;
1684 case VNC_ENCODING_EXT_KEY_EVENT:
1685 send_ext_key_event_ack(vs);
1686 break;
1687 case VNC_ENCODING_AUDIO:
1688 send_ext_audio_ack(vs);
1689 break;
1690 case VNC_ENCODING_WMVi:
1691 vs->features |= VNC_FEATURE_WMVI_MASK;
1692 break;
1693 case VNC_ENCODING_COMPRESSLEVEL0 ... VNC_ENCODING_COMPRESSLEVEL0 + 9:
1694 vs->tight_compression = (enc & 0x0F);
1695 break;
1696 case VNC_ENCODING_QUALITYLEVEL0 ... VNC_ENCODING_QUALITYLEVEL0 + 9:
1697 vs->tight_quality = (enc & 0x0F);
1698 break;
1699 default:
1700 VNC_DEBUG("Unknown encoding: %d (0x%.8x): %d\n", i, enc, enc);
1701 break;
1704 check_pointer_type_change(&vs->mouse_mode_notifier);
1707 static void set_pixel_conversion(VncState *vs)
1709 if ((vs->clientds.flags & QEMU_BIG_ENDIAN_FLAG) ==
1710 (vs->ds->surface->flags & QEMU_BIG_ENDIAN_FLAG) &&
1711 !memcmp(&(vs->clientds.pf), &(vs->ds->surface->pf), sizeof(PixelFormat))) {
1712 vs->write_pixels = vnc_write_pixels_copy;
1713 vnc_hextile_set_pixel_conversion(vs, 0);
1714 } else {
1715 vs->write_pixels = vnc_write_pixels_generic;
1716 vnc_hextile_set_pixel_conversion(vs, 1);
1720 static void set_pixel_format(VncState *vs,
1721 int bits_per_pixel, int depth,
1722 int big_endian_flag, int true_color_flag,
1723 int red_max, int green_max, int blue_max,
1724 int red_shift, int green_shift, int blue_shift)
1726 if (!true_color_flag) {
1727 vnc_client_error(vs);
1728 return;
1731 vs->clientds = *(vs->vd->guest.ds);
1732 vs->clientds.pf.rmax = red_max;
1733 count_bits(vs->clientds.pf.rbits, red_max);
1734 vs->clientds.pf.rshift = red_shift;
1735 vs->clientds.pf.rmask = red_max << red_shift;
1736 vs->clientds.pf.gmax = green_max;
1737 count_bits(vs->clientds.pf.gbits, green_max);
1738 vs->clientds.pf.gshift = green_shift;
1739 vs->clientds.pf.gmask = green_max << green_shift;
1740 vs->clientds.pf.bmax = blue_max;
1741 count_bits(vs->clientds.pf.bbits, blue_max);
1742 vs->clientds.pf.bshift = blue_shift;
1743 vs->clientds.pf.bmask = blue_max << blue_shift;
1744 vs->clientds.pf.bits_per_pixel = bits_per_pixel;
1745 vs->clientds.pf.bytes_per_pixel = bits_per_pixel / 8;
1746 vs->clientds.pf.depth = bits_per_pixel == 32 ? 24 : bits_per_pixel;
1747 vs->clientds.flags = big_endian_flag ? QEMU_BIG_ENDIAN_FLAG : 0x00;
1749 set_pixel_conversion(vs);
1751 vga_hw_invalidate();
1752 vga_hw_update();
1755 static void pixel_format_message (VncState *vs) {
1756 char pad[3] = { 0, 0, 0 };
1758 vnc_write_u8(vs, vs->ds->surface->pf.bits_per_pixel); /* bits-per-pixel */
1759 vnc_write_u8(vs, vs->ds->surface->pf.depth); /* depth */
1761 #ifdef HOST_WORDS_BIGENDIAN
1762 vnc_write_u8(vs, 1); /* big-endian-flag */
1763 #else
1764 vnc_write_u8(vs, 0); /* big-endian-flag */
1765 #endif
1766 vnc_write_u8(vs, 1); /* true-color-flag */
1767 vnc_write_u16(vs, vs->ds->surface->pf.rmax); /* red-max */
1768 vnc_write_u16(vs, vs->ds->surface->pf.gmax); /* green-max */
1769 vnc_write_u16(vs, vs->ds->surface->pf.bmax); /* blue-max */
1770 vnc_write_u8(vs, vs->ds->surface->pf.rshift); /* red-shift */
1771 vnc_write_u8(vs, vs->ds->surface->pf.gshift); /* green-shift */
1772 vnc_write_u8(vs, vs->ds->surface->pf.bshift); /* blue-shift */
1774 vnc_hextile_set_pixel_conversion(vs, 0);
1776 vs->clientds = *(vs->ds->surface);
1777 vs->clientds.flags &= ~QEMU_ALLOCATED_FLAG;
1778 vs->write_pixels = vnc_write_pixels_copy;
1780 vnc_write(vs, pad, 3); /* padding */
1783 static void vnc_dpy_setdata(DisplayState *ds)
1785 /* We don't have to do anything */
1788 static void vnc_colordepth(VncState *vs)
1790 if (vnc_has_feature(vs, VNC_FEATURE_WMVI)) {
1791 /* Sending a WMVi message to notify the client*/
1792 vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
1793 vnc_write_u8(vs, 0);
1794 vnc_write_u16(vs, 1); /* number of rects */
1795 vnc_framebuffer_update(vs, 0, 0, ds_get_width(vs->ds),
1796 ds_get_height(vs->ds), VNC_ENCODING_WMVi);
1797 pixel_format_message(vs);
1798 vnc_flush(vs);
1799 } else {
1800 set_pixel_conversion(vs);
1804 static int protocol_client_msg(VncState *vs, uint8_t *data, size_t len)
1806 int i;
1807 uint16_t limit;
1808 VncDisplay *vd = vs->vd;
1810 if (data[0] > 3) {
1811 vd->timer_interval = VNC_REFRESH_INTERVAL_BASE;
1812 if (!qemu_timer_expired(vd->timer, qemu_get_clock(rt_clock) + vd->timer_interval))
1813 qemu_mod_timer(vd->timer, qemu_get_clock(rt_clock) + vd->timer_interval);
1816 switch (data[0]) {
1817 case VNC_MSG_CLIENT_SET_PIXEL_FORMAT:
1818 if (len == 1)
1819 return 20;
1821 set_pixel_format(vs, read_u8(data, 4), read_u8(data, 5),
1822 read_u8(data, 6), read_u8(data, 7),
1823 read_u16(data, 8), read_u16(data, 10),
1824 read_u16(data, 12), read_u8(data, 14),
1825 read_u8(data, 15), read_u8(data, 16));
1826 break;
1827 case VNC_MSG_CLIENT_SET_ENCODINGS:
1828 if (len == 1)
1829 return 4;
1831 if (len == 4) {
1832 limit = read_u16(data, 2);
1833 if (limit > 0)
1834 return 4 + (limit * 4);
1835 } else
1836 limit = read_u16(data, 2);
1838 for (i = 0; i < limit; i++) {
1839 int32_t val = read_s32(data, 4 + (i * 4));
1840 memcpy(data + 4 + (i * 4), &val, sizeof(val));
1843 set_encodings(vs, (int32_t *)(data + 4), limit);
1844 break;
1845 case VNC_MSG_CLIENT_FRAMEBUFFER_UPDATE_REQUEST:
1846 if (len == 1)
1847 return 10;
1849 framebuffer_update_request(vs,
1850 read_u8(data, 1), read_u16(data, 2), read_u16(data, 4),
1851 read_u16(data, 6), read_u16(data, 8));
1852 break;
1853 case VNC_MSG_CLIENT_KEY_EVENT:
1854 if (len == 1)
1855 return 8;
1857 key_event(vs, read_u8(data, 1), read_u32(data, 4));
1858 break;
1859 case VNC_MSG_CLIENT_POINTER_EVENT:
1860 if (len == 1)
1861 return 6;
1863 pointer_event(vs, read_u8(data, 1), read_u16(data, 2), read_u16(data, 4));
1864 break;
1865 case VNC_MSG_CLIENT_CUT_TEXT:
1866 if (len == 1)
1867 return 8;
1869 if (len == 8) {
1870 uint32_t dlen = read_u32(data, 4);
1871 if (dlen > 0)
1872 return 8 + dlen;
1875 client_cut_text(vs, read_u32(data, 4), data + 8);
1876 break;
1877 case VNC_MSG_CLIENT_QEMU:
1878 if (len == 1)
1879 return 2;
1881 switch (read_u8(data, 1)) {
1882 case VNC_MSG_CLIENT_QEMU_EXT_KEY_EVENT:
1883 if (len == 2)
1884 return 12;
1886 ext_key_event(vs, read_u16(data, 2),
1887 read_u32(data, 4), read_u32(data, 8));
1888 break;
1889 case VNC_MSG_CLIENT_QEMU_AUDIO:
1890 if (len == 2)
1891 return 4;
1893 switch (read_u16 (data, 2)) {
1894 case VNC_MSG_CLIENT_QEMU_AUDIO_ENABLE:
1895 audio_add(vs);
1896 break;
1897 case VNC_MSG_CLIENT_QEMU_AUDIO_DISABLE:
1898 audio_del(vs);
1899 break;
1900 case VNC_MSG_CLIENT_QEMU_AUDIO_SET_FORMAT:
1901 if (len == 4)
1902 return 10;
1903 switch (read_u8(data, 4)) {
1904 case 0: vs->as.fmt = AUD_FMT_U8; break;
1905 case 1: vs->as.fmt = AUD_FMT_S8; break;
1906 case 2: vs->as.fmt = AUD_FMT_U16; break;
1907 case 3: vs->as.fmt = AUD_FMT_S16; break;
1908 case 4: vs->as.fmt = AUD_FMT_U32; break;
1909 case 5: vs->as.fmt = AUD_FMT_S32; break;
1910 default:
1911 printf("Invalid audio format %d\n", read_u8(data, 4));
1912 vnc_client_error(vs);
1913 break;
1915 vs->as.nchannels = read_u8(data, 5);
1916 if (vs->as.nchannels != 1 && vs->as.nchannels != 2) {
1917 printf("Invalid audio channel coount %d\n",
1918 read_u8(data, 5));
1919 vnc_client_error(vs);
1920 break;
1922 vs->as.freq = read_u32(data, 6);
1923 break;
1924 default:
1925 printf ("Invalid audio message %d\n", read_u8(data, 4));
1926 vnc_client_error(vs);
1927 break;
1929 break;
1931 default:
1932 printf("Msg: %d\n", read_u16(data, 0));
1933 vnc_client_error(vs);
1934 break;
1936 break;
1937 default:
1938 printf("Msg: %d\n", data[0]);
1939 vnc_client_error(vs);
1940 break;
1943 vnc_read_when(vs, protocol_client_msg, 1);
1944 return 0;
1947 static int protocol_client_init(VncState *vs, uint8_t *data, size_t len)
1949 char buf[1024];
1950 int size;
1952 vnc_write_u16(vs, ds_get_width(vs->ds));
1953 vnc_write_u16(vs, ds_get_height(vs->ds));
1955 pixel_format_message(vs);
1957 if (qemu_name)
1958 size = snprintf(buf, sizeof(buf), "QEMU (%s)", qemu_name);
1959 else
1960 size = snprintf(buf, sizeof(buf), "QEMU");
1962 vnc_write_u32(vs, size);
1963 vnc_write(vs, buf, size);
1964 vnc_flush(vs);
1966 vnc_client_cache_auth(vs);
1967 vnc_qmp_event(vs, QEVENT_VNC_INITIALIZED);
1969 vnc_read_when(vs, protocol_client_msg, 1);
1971 return 0;
1974 void start_client_init(VncState *vs)
1976 vnc_read_when(vs, protocol_client_init, 1);
1979 static void make_challenge(VncState *vs)
1981 int i;
1983 srand(time(NULL)+getpid()+getpid()*987654+rand());
1985 for (i = 0 ; i < sizeof(vs->challenge) ; i++)
1986 vs->challenge[i] = (int) (256.0*rand()/(RAND_MAX+1.0));
1989 static int protocol_client_auth_vnc(VncState *vs, uint8_t *data, size_t len)
1991 unsigned char response[VNC_AUTH_CHALLENGE_SIZE];
1992 int i, j, pwlen;
1993 unsigned char key[8];
1995 if (!vs->vd->password || !vs->vd->password[0]) {
1996 VNC_DEBUG("No password configured on server");
1997 vnc_write_u32(vs, 1); /* Reject auth */
1998 if (vs->minor >= 8) {
1999 static const char err[] = "Authentication failed";
2000 vnc_write_u32(vs, sizeof(err));
2001 vnc_write(vs, err, sizeof(err));
2003 vnc_flush(vs);
2004 vnc_client_error(vs);
2005 return 0;
2008 memcpy(response, vs->challenge, VNC_AUTH_CHALLENGE_SIZE);
2010 /* Calculate the expected challenge response */
2011 pwlen = strlen(vs->vd->password);
2012 for (i=0; i<sizeof(key); i++)
2013 key[i] = i<pwlen ? vs->vd->password[i] : 0;
2014 deskey(key, EN0);
2015 for (j = 0; j < VNC_AUTH_CHALLENGE_SIZE; j += 8)
2016 des(response+j, response+j);
2018 /* Compare expected vs actual challenge response */
2019 if (memcmp(response, data, VNC_AUTH_CHALLENGE_SIZE) != 0) {
2020 VNC_DEBUG("Client challenge reponse did not match\n");
2021 vnc_write_u32(vs, 1); /* Reject auth */
2022 if (vs->minor >= 8) {
2023 static const char err[] = "Authentication failed";
2024 vnc_write_u32(vs, sizeof(err));
2025 vnc_write(vs, err, sizeof(err));
2027 vnc_flush(vs);
2028 vnc_client_error(vs);
2029 } else {
2030 VNC_DEBUG("Accepting VNC challenge response\n");
2031 vnc_write_u32(vs, 0); /* Accept auth */
2032 vnc_flush(vs);
2034 start_client_init(vs);
2036 return 0;
2039 void start_auth_vnc(VncState *vs)
2041 make_challenge(vs);
2042 /* Send client a 'random' challenge */
2043 vnc_write(vs, vs->challenge, sizeof(vs->challenge));
2044 vnc_flush(vs);
2046 vnc_read_when(vs, protocol_client_auth_vnc, sizeof(vs->challenge));
2050 static int protocol_client_auth(VncState *vs, uint8_t *data, size_t len)
2052 /* We only advertise 1 auth scheme at a time, so client
2053 * must pick the one we sent. Verify this */
2054 if (data[0] != vs->vd->auth) { /* Reject auth */
2055 VNC_DEBUG("Reject auth %d because it didn't match advertized\n", (int)data[0]);
2056 vnc_write_u32(vs, 1);
2057 if (vs->minor >= 8) {
2058 static const char err[] = "Authentication failed";
2059 vnc_write_u32(vs, sizeof(err));
2060 vnc_write(vs, err, sizeof(err));
2062 vnc_client_error(vs);
2063 } else { /* Accept requested auth */
2064 VNC_DEBUG("Client requested auth %d\n", (int)data[0]);
2065 switch (vs->vd->auth) {
2066 case VNC_AUTH_NONE:
2067 VNC_DEBUG("Accept auth none\n");
2068 if (vs->minor >= 8) {
2069 vnc_write_u32(vs, 0); /* Accept auth completion */
2070 vnc_flush(vs);
2072 start_client_init(vs);
2073 break;
2075 case VNC_AUTH_VNC:
2076 VNC_DEBUG("Start VNC auth\n");
2077 start_auth_vnc(vs);
2078 break;
2080 #ifdef CONFIG_VNC_TLS
2081 case VNC_AUTH_VENCRYPT:
2082 VNC_DEBUG("Accept VeNCrypt auth\n");;
2083 start_auth_vencrypt(vs);
2084 break;
2085 #endif /* CONFIG_VNC_TLS */
2087 #ifdef CONFIG_VNC_SASL
2088 case VNC_AUTH_SASL:
2089 VNC_DEBUG("Accept SASL auth\n");
2090 start_auth_sasl(vs);
2091 break;
2092 #endif /* CONFIG_VNC_SASL */
2094 default: /* Should not be possible, but just in case */
2095 VNC_DEBUG("Reject auth %d server code bug\n", vs->vd->auth);
2096 vnc_write_u8(vs, 1);
2097 if (vs->minor >= 8) {
2098 static const char err[] = "Authentication failed";
2099 vnc_write_u32(vs, sizeof(err));
2100 vnc_write(vs, err, sizeof(err));
2102 vnc_client_error(vs);
2105 return 0;
2108 static int protocol_version(VncState *vs, uint8_t *version, size_t len)
2110 char local[13];
2112 memcpy(local, version, 12);
2113 local[12] = 0;
2115 if (sscanf(local, "RFB %03d.%03d\n", &vs->major, &vs->minor) != 2) {
2116 VNC_DEBUG("Malformed protocol version %s\n", local);
2117 vnc_client_error(vs);
2118 return 0;
2120 VNC_DEBUG("Client request protocol version %d.%d\n", vs->major, vs->minor);
2121 if (vs->major != 3 ||
2122 (vs->minor != 3 &&
2123 vs->minor != 4 &&
2124 vs->minor != 5 &&
2125 vs->minor != 7 &&
2126 vs->minor != 8)) {
2127 VNC_DEBUG("Unsupported client version\n");
2128 vnc_write_u32(vs, VNC_AUTH_INVALID);
2129 vnc_flush(vs);
2130 vnc_client_error(vs);
2131 return 0;
2133 /* Some broken clients report v3.4 or v3.5, which spec requires to be treated
2134 * as equivalent to v3.3 by servers
2136 if (vs->minor == 4 || vs->minor == 5)
2137 vs->minor = 3;
2139 if (vs->minor == 3) {
2140 if (vs->vd->auth == VNC_AUTH_NONE) {
2141 VNC_DEBUG("Tell client auth none\n");
2142 vnc_write_u32(vs, vs->vd->auth);
2143 vnc_flush(vs);
2144 start_client_init(vs);
2145 } else if (vs->vd->auth == VNC_AUTH_VNC) {
2146 VNC_DEBUG("Tell client VNC auth\n");
2147 vnc_write_u32(vs, vs->vd->auth);
2148 vnc_flush(vs);
2149 start_auth_vnc(vs);
2150 } else {
2151 VNC_DEBUG("Unsupported auth %d for protocol 3.3\n", vs->vd->auth);
2152 vnc_write_u32(vs, VNC_AUTH_INVALID);
2153 vnc_flush(vs);
2154 vnc_client_error(vs);
2156 } else {
2157 VNC_DEBUG("Telling client we support auth %d\n", vs->vd->auth);
2158 vnc_write_u8(vs, 1); /* num auth */
2159 vnc_write_u8(vs, vs->vd->auth);
2160 vnc_read_when(vs, protocol_client_auth, 1);
2161 vnc_flush(vs);
2164 return 0;
2167 static int vnc_refresh_server_surface(VncDisplay *vd)
2169 int y;
2170 uint8_t *guest_row;
2171 uint8_t *server_row;
2172 int cmp_bytes;
2173 uint32_t width_mask[VNC_DIRTY_WORDS];
2174 VncState *vs;
2175 int has_dirty = 0;
2178 * Walk through the guest dirty map.
2179 * Check and copy modified bits from guest to server surface.
2180 * Update server dirty map.
2182 vnc_set_bits(width_mask, (ds_get_width(vd->ds) / 16), VNC_DIRTY_WORDS);
2183 cmp_bytes = 16 * ds_get_bytes_per_pixel(vd->ds);
2184 guest_row = vd->guest.ds->data;
2185 server_row = vd->server->data;
2186 for (y = 0; y < vd->guest.ds->height; y++) {
2187 if (vnc_and_bits(vd->guest.dirty[y], width_mask, VNC_DIRTY_WORDS)) {
2188 int x;
2189 uint8_t *guest_ptr;
2190 uint8_t *server_ptr;
2192 guest_ptr = guest_row;
2193 server_ptr = server_row;
2195 for (x = 0; x < vd->guest.ds->width;
2196 x += 16, guest_ptr += cmp_bytes, server_ptr += cmp_bytes) {
2197 if (!vnc_get_bit(vd->guest.dirty[y], (x / 16)))
2198 continue;
2199 vnc_clear_bit(vd->guest.dirty[y], (x / 16));
2200 if (memcmp(server_ptr, guest_ptr, cmp_bytes) == 0)
2201 continue;
2202 memcpy(server_ptr, guest_ptr, cmp_bytes);
2203 QTAILQ_FOREACH(vs, &vd->clients, next) {
2204 vnc_set_bit(vs->dirty[y], (x / 16));
2206 has_dirty++;
2209 guest_row += ds_get_linesize(vd->ds);
2210 server_row += ds_get_linesize(vd->ds);
2212 return has_dirty;
2215 static void vnc_refresh(void *opaque)
2217 VncDisplay *vd = opaque;
2218 VncState *vs, *vn;
2219 int has_dirty, rects = 0;
2221 vga_hw_update();
2223 has_dirty = vnc_refresh_server_surface(vd);
2225 QTAILQ_FOREACH_SAFE(vs, &vd->clients, next, vn) {
2226 rects += vnc_update_client(vs, has_dirty);
2227 /* vs might be free()ed here */
2229 /* vd->timer could be NULL now if the last client disconnected,
2230 * in this case don't update the timer */
2231 if (vd->timer == NULL)
2232 return;
2234 if (has_dirty && rects) {
2235 vd->timer_interval /= 2;
2236 if (vd->timer_interval < VNC_REFRESH_INTERVAL_BASE)
2237 vd->timer_interval = VNC_REFRESH_INTERVAL_BASE;
2238 } else {
2239 vd->timer_interval += VNC_REFRESH_INTERVAL_INC;
2240 if (vd->timer_interval > VNC_REFRESH_INTERVAL_MAX)
2241 vd->timer_interval = VNC_REFRESH_INTERVAL_MAX;
2243 qemu_mod_timer(vd->timer, qemu_get_clock(rt_clock) + vd->timer_interval);
2246 static void vnc_init_timer(VncDisplay *vd)
2248 vd->timer_interval = VNC_REFRESH_INTERVAL_BASE;
2249 if (vd->timer == NULL && !QTAILQ_EMPTY(&vd->clients)) {
2250 vd->timer = qemu_new_timer(rt_clock, vnc_refresh, vd);
2251 vnc_refresh(vd);
2255 static void vnc_remove_timer(VncDisplay *vd)
2257 if (vd->timer != NULL && QTAILQ_EMPTY(&vd->clients)) {
2258 qemu_del_timer(vd->timer);
2259 qemu_free_timer(vd->timer);
2260 vd->timer = NULL;
2264 static void vnc_connect(VncDisplay *vd, int csock)
2266 VncState *vs = qemu_mallocz(sizeof(VncState));
2267 vs->csock = csock;
2269 VNC_DEBUG("New client on socket %d\n", csock);
2270 dcl->idle = 0;
2271 socket_set_nonblock(vs->csock);
2272 qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, vs);
2274 vnc_client_cache_addr(vs);
2275 vnc_qmp_event(vs, QEVENT_VNC_CONNECTED);
2277 vs->vd = vd;
2278 vs->ds = vd->ds;
2279 vs->last_x = -1;
2280 vs->last_y = -1;
2282 vs->as.freq = 44100;
2283 vs->as.nchannels = 2;
2284 vs->as.fmt = AUD_FMT_S16;
2285 vs->as.endianness = 0;
2287 QTAILQ_INSERT_HEAD(&vd->clients, vs, next);
2289 vga_hw_update();
2291 vnc_write(vs, "RFB 003.008\n", 12);
2292 vnc_flush(vs);
2293 vnc_read_when(vs, protocol_version, 12);
2294 reset_keys(vs);
2295 if (vs->vd->lock_key_sync)
2296 vs->led = qemu_add_led_event_handler(kbd_leds, vs);
2298 vs->mouse_mode_notifier.notify = check_pointer_type_change;
2299 qemu_add_mouse_mode_change_notifier(&vs->mouse_mode_notifier);
2301 vnc_init_timer(vd);
2303 /* vs might be free()ed here */
2306 static void vnc_listen_read(void *opaque)
2308 VncDisplay *vs = opaque;
2309 struct sockaddr_in addr;
2310 socklen_t addrlen = sizeof(addr);
2312 /* Catch-up */
2313 vga_hw_update();
2315 int csock = qemu_accept(vs->lsock, (struct sockaddr *)&addr, &addrlen);
2316 if (csock != -1) {
2317 vnc_connect(vs, csock);
2321 void vnc_display_init(DisplayState *ds)
2323 VncDisplay *vs = qemu_mallocz(sizeof(*vs));
2325 dcl = qemu_mallocz(sizeof(DisplayChangeListener));
2327 ds->opaque = vs;
2328 dcl->idle = 1;
2329 vnc_display = vs;
2331 vs->lsock = -1;
2333 vs->ds = ds;
2334 QTAILQ_INIT(&vs->clients);
2336 if (keyboard_layout)
2337 vs->kbd_layout = init_keyboard_layout(name2keysym, keyboard_layout);
2338 else
2339 vs->kbd_layout = init_keyboard_layout(name2keysym, "en-us");
2341 if (!vs->kbd_layout)
2342 exit(1);
2344 dcl->dpy_copy = vnc_dpy_copy;
2345 dcl->dpy_update = vnc_dpy_update;
2346 dcl->dpy_resize = vnc_dpy_resize;
2347 dcl->dpy_setdata = vnc_dpy_setdata;
2348 register_displaychangelistener(ds, dcl);
2349 ds->mouse_set = vnc_mouse_set;
2350 ds->cursor_define = vnc_dpy_cursor_define;
2354 void vnc_display_close(DisplayState *ds)
2356 VncDisplay *vs = ds ? (VncDisplay *)ds->opaque : vnc_display;
2358 if (!vs)
2359 return;
2360 if (vs->display) {
2361 qemu_free(vs->display);
2362 vs->display = NULL;
2364 if (vs->lsock != -1) {
2365 qemu_set_fd_handler2(vs->lsock, NULL, NULL, NULL, NULL);
2366 close(vs->lsock);
2367 vs->lsock = -1;
2369 vs->auth = VNC_AUTH_INVALID;
2370 #ifdef CONFIG_VNC_TLS
2371 vs->subauth = VNC_AUTH_INVALID;
2372 vs->tls.x509verify = 0;
2373 #endif
2376 int vnc_display_password(DisplayState *ds, const char *password)
2378 VncDisplay *vs = ds ? (VncDisplay *)ds->opaque : vnc_display;
2380 if (!vs) {
2381 return -1;
2384 if (vs->password) {
2385 qemu_free(vs->password);
2386 vs->password = NULL;
2388 if (password && password[0]) {
2389 if (!(vs->password = qemu_strdup(password)))
2390 return -1;
2391 if (vs->auth == VNC_AUTH_NONE) {
2392 vs->auth = VNC_AUTH_VNC;
2394 } else {
2395 vs->auth = VNC_AUTH_NONE;
2398 return 0;
2401 char *vnc_display_local_addr(DisplayState *ds)
2403 VncDisplay *vs = ds ? (VncDisplay *)ds->opaque : vnc_display;
2405 return vnc_socket_local_addr("%s:%s", vs->lsock);
2408 int vnc_display_open(DisplayState *ds, const char *display)
2410 VncDisplay *vs = ds ? (VncDisplay *)ds->opaque : vnc_display;
2411 const char *options;
2412 int password = 0;
2413 int reverse = 0;
2414 #ifdef CONFIG_VNC_TLS
2415 int tls = 0, x509 = 0;
2416 #endif
2417 #ifdef CONFIG_VNC_SASL
2418 int sasl = 0;
2419 int saslErr;
2420 #endif
2421 int acl = 0;
2422 int lock_key_sync = 1;
2424 if (!vnc_display)
2425 return -1;
2426 vnc_display_close(ds);
2427 if (strcmp(display, "none") == 0)
2428 return 0;
2430 if (!(vs->display = strdup(display)))
2431 return -1;
2433 options = display;
2434 while ((options = strchr(options, ','))) {
2435 options++;
2436 if (strncmp(options, "password", 8) == 0) {
2437 password = 1; /* Require password auth */
2438 } else if (strncmp(options, "reverse", 7) == 0) {
2439 reverse = 1;
2440 } else if (strncmp(options, "no-lock-key-sync", 9) == 0) {
2441 lock_key_sync = 0;
2442 #ifdef CONFIG_VNC_SASL
2443 } else if (strncmp(options, "sasl", 4) == 0) {
2444 sasl = 1; /* Require SASL auth */
2445 #endif
2446 #ifdef CONFIG_VNC_TLS
2447 } else if (strncmp(options, "tls", 3) == 0) {
2448 tls = 1; /* Require TLS */
2449 } else if (strncmp(options, "x509", 4) == 0) {
2450 char *start, *end;
2451 x509 = 1; /* Require x509 certificates */
2452 if (strncmp(options, "x509verify", 10) == 0)
2453 vs->tls.x509verify = 1; /* ...and verify client certs */
2455 /* Now check for 'x509=/some/path' postfix
2456 * and use that to setup x509 certificate/key paths */
2457 start = strchr(options, '=');
2458 end = strchr(options, ',');
2459 if (start && (!end || (start < end))) {
2460 int len = end ? end-(start+1) : strlen(start+1);
2461 char *path = qemu_strndup(start + 1, len);
2463 VNC_DEBUG("Trying certificate path '%s'\n", path);
2464 if (vnc_tls_set_x509_creds_dir(vs, path) < 0) {
2465 fprintf(stderr, "Failed to find x509 certificates/keys in %s\n", path);
2466 qemu_free(path);
2467 qemu_free(vs->display);
2468 vs->display = NULL;
2469 return -1;
2471 qemu_free(path);
2472 } else {
2473 fprintf(stderr, "No certificate path provided\n");
2474 qemu_free(vs->display);
2475 vs->display = NULL;
2476 return -1;
2478 #endif
2479 } else if (strncmp(options, "acl", 3) == 0) {
2480 acl = 1;
2484 #ifdef CONFIG_VNC_TLS
2485 if (acl && x509 && vs->tls.x509verify) {
2486 if (!(vs->tls.acl = qemu_acl_init("vnc.x509dname"))) {
2487 fprintf(stderr, "Failed to create x509 dname ACL\n");
2488 exit(1);
2491 #endif
2492 #ifdef CONFIG_VNC_SASL
2493 if (acl && sasl) {
2494 if (!(vs->sasl.acl = qemu_acl_init("vnc.username"))) {
2495 fprintf(stderr, "Failed to create username ACL\n");
2496 exit(1);
2499 #endif
2502 * Combinations we support here:
2504 * - no-auth (clear text, no auth)
2505 * - password (clear text, weak auth)
2506 * - sasl (encrypt, good auth *IF* using Kerberos via GSSAPI)
2507 * - tls (encrypt, weak anonymous creds, no auth)
2508 * - tls + password (encrypt, weak anonymous creds, weak auth)
2509 * - tls + sasl (encrypt, weak anonymous creds, good auth)
2510 * - tls + x509 (encrypt, good x509 creds, no auth)
2511 * - tls + x509 + password (encrypt, good x509 creds, weak auth)
2512 * - tls + x509 + sasl (encrypt, good x509 creds, good auth)
2514 * NB1. TLS is a stackable auth scheme.
2515 * NB2. the x509 schemes have option to validate a client cert dname
2517 if (password) {
2518 #ifdef CONFIG_VNC_TLS
2519 if (tls) {
2520 vs->auth = VNC_AUTH_VENCRYPT;
2521 if (x509) {
2522 VNC_DEBUG("Initializing VNC server with x509 password auth\n");
2523 vs->subauth = VNC_AUTH_VENCRYPT_X509VNC;
2524 } else {
2525 VNC_DEBUG("Initializing VNC server with TLS password auth\n");
2526 vs->subauth = VNC_AUTH_VENCRYPT_TLSVNC;
2528 } else {
2529 #endif /* CONFIG_VNC_TLS */
2530 VNC_DEBUG("Initializing VNC server with password auth\n");
2531 vs->auth = VNC_AUTH_VNC;
2532 #ifdef CONFIG_VNC_TLS
2533 vs->subauth = VNC_AUTH_INVALID;
2535 #endif /* CONFIG_VNC_TLS */
2536 #ifdef CONFIG_VNC_SASL
2537 } else if (sasl) {
2538 #ifdef CONFIG_VNC_TLS
2539 if (tls) {
2540 vs->auth = VNC_AUTH_VENCRYPT;
2541 if (x509) {
2542 VNC_DEBUG("Initializing VNC server with x509 SASL auth\n");
2543 vs->subauth = VNC_AUTH_VENCRYPT_X509SASL;
2544 } else {
2545 VNC_DEBUG("Initializing VNC server with TLS SASL auth\n");
2546 vs->subauth = VNC_AUTH_VENCRYPT_TLSSASL;
2548 } else {
2549 #endif /* CONFIG_VNC_TLS */
2550 VNC_DEBUG("Initializing VNC server with SASL auth\n");
2551 vs->auth = VNC_AUTH_SASL;
2552 #ifdef CONFIG_VNC_TLS
2553 vs->subauth = VNC_AUTH_INVALID;
2555 #endif /* CONFIG_VNC_TLS */
2556 #endif /* CONFIG_VNC_SASL */
2557 } else {
2558 #ifdef CONFIG_VNC_TLS
2559 if (tls) {
2560 vs->auth = VNC_AUTH_VENCRYPT;
2561 if (x509) {
2562 VNC_DEBUG("Initializing VNC server with x509 no auth\n");
2563 vs->subauth = VNC_AUTH_VENCRYPT_X509NONE;
2564 } else {
2565 VNC_DEBUG("Initializing VNC server with TLS no auth\n");
2566 vs->subauth = VNC_AUTH_VENCRYPT_TLSNONE;
2568 } else {
2569 #endif
2570 VNC_DEBUG("Initializing VNC server with no auth\n");
2571 vs->auth = VNC_AUTH_NONE;
2572 #ifdef CONFIG_VNC_TLS
2573 vs->subauth = VNC_AUTH_INVALID;
2575 #endif
2578 #ifdef CONFIG_VNC_SASL
2579 if ((saslErr = sasl_server_init(NULL, "qemu")) != SASL_OK) {
2580 fprintf(stderr, "Failed to initialize SASL auth %s",
2581 sasl_errstring(saslErr, NULL, NULL));
2582 free(vs->display);
2583 vs->display = NULL;
2584 return -1;
2586 #endif
2587 vs->lock_key_sync = lock_key_sync;
2589 if (reverse) {
2590 /* connect to viewer */
2591 if (strncmp(display, "unix:", 5) == 0)
2592 vs->lsock = unix_connect(display+5);
2593 else
2594 vs->lsock = inet_connect(display, SOCK_STREAM);
2595 if (-1 == vs->lsock) {
2596 free(vs->display);
2597 vs->display = NULL;
2598 return -1;
2599 } else {
2600 int csock = vs->lsock;
2601 vs->lsock = -1;
2602 vnc_connect(vs, csock);
2604 return 0;
2606 } else {
2607 /* listen for connects */
2608 char *dpy;
2609 dpy = qemu_malloc(256);
2610 if (strncmp(display, "unix:", 5) == 0) {
2611 pstrcpy(dpy, 256, "unix:");
2612 vs->lsock = unix_listen(display+5, dpy+5, 256-5);
2613 } else {
2614 vs->lsock = inet_listen(display, dpy, 256, SOCK_STREAM, 5900);
2616 if (-1 == vs->lsock) {
2617 free(dpy);
2618 return -1;
2619 } else {
2620 free(vs->display);
2621 vs->display = dpy;
2624 return qemu_set_fd_handler2(vs->lsock, NULL, vnc_listen_read, NULL, vs);