xref: /openbmc/qemu/ui/vnc.c (revision e556032960eaa403da6069aa9abf4dcaddefd710)
13e230dd2SCorentin Chary /*
23e230dd2SCorentin Chary  * QEMU VNC display driver
33e230dd2SCorentin Chary  *
43e230dd2SCorentin Chary  * Copyright (C) 2006 Anthony Liguori <anthony@codemonkey.ws>
53e230dd2SCorentin Chary  * Copyright (C) 2006 Fabrice Bellard
63e230dd2SCorentin Chary  * Copyright (C) 2009 Red Hat, Inc
73e230dd2SCorentin Chary  *
83e230dd2SCorentin Chary  * Permission is hereby granted, free of charge, to any person obtaining a copy
93e230dd2SCorentin Chary  * of this software and associated documentation files (the "Software"), to deal
103e230dd2SCorentin Chary  * in the Software without restriction, including without limitation the rights
113e230dd2SCorentin Chary  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
123e230dd2SCorentin Chary  * copies of the Software, and to permit persons to whom the Software is
133e230dd2SCorentin Chary  * furnished to do so, subject to the following conditions:
143e230dd2SCorentin Chary  *
153e230dd2SCorentin Chary  * The above copyright notice and this permission notice shall be included in
163e230dd2SCorentin Chary  * all copies or substantial portions of the Software.
173e230dd2SCorentin Chary  *
183e230dd2SCorentin Chary  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
193e230dd2SCorentin Chary  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
203e230dd2SCorentin Chary  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
213e230dd2SCorentin Chary  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
223e230dd2SCorentin Chary  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
233e230dd2SCorentin Chary  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
243e230dd2SCorentin Chary  * THE SOFTWARE.
253e230dd2SCorentin Chary  */
263e230dd2SCorentin Chary 
273e230dd2SCorentin Chary #include "vnc.h"
28bd023f95SCorentin Chary #include "vnc-jobs.h"
2940066175SGerd Hoffmann #include "trace.h"
301d0d59feSGerd Hoffmann #include "hw/qdev.h"
319c17d615SPaolo Bonzini #include "sysemu/sysemu.h"
321de7afc9SPaolo Bonzini #include "qemu/sockets.h"
331de7afc9SPaolo Bonzini #include "qemu/timer.h"
341de7afc9SPaolo Bonzini #include "qemu/acl.h"
354db14629SGerd Hoffmann #include "qemu/config-file.h"
367b1b5d19SPaolo Bonzini #include "qapi/qmp/types.h"
372b54aa87SLuiz Capitulino #include "qmp-commands.h"
381de7afc9SPaolo Bonzini #include "qemu/osdep.h"
398d447d10SGerd Hoffmann #include "ui/input.h"
40fb6ba0d5SWenchao Xia #include "qapi-event.h"
413e230dd2SCorentin Chary 
420f7b2864SGerd Hoffmann #define VNC_REFRESH_INTERVAL_BASE GUI_REFRESH_INTERVAL_DEFAULT
433e230dd2SCorentin Chary #define VNC_REFRESH_INTERVAL_INC  50
440f7b2864SGerd Hoffmann #define VNC_REFRESH_INTERVAL_MAX  GUI_REFRESH_INTERVAL_IDLE
45999342a0SCorentin Chary static const struct timeval VNC_REFRESH_STATS = { 0, 500000 };
46999342a0SCorentin Chary static const struct timeval VNC_REFRESH_LOSSY = { 2, 0 };
473e230dd2SCorentin Chary 
483e230dd2SCorentin Chary #include "vnc_keysym.h"
493e230dd2SCorentin Chary #include "d3des.h"
503e230dd2SCorentin Chary 
51d616ccc5SGerd Hoffmann static QTAILQ_HEAD(, VncDisplay) vnc_displays =
52d616ccc5SGerd Hoffmann     QTAILQ_HEAD_INITIALIZER(vnc_displays);
533e230dd2SCorentin Chary 
543e230dd2SCorentin Chary static int vnc_cursor_define(VncState *vs);
557bc9318bSGerd Hoffmann static void vnc_release_modifiers(VncState *vs);
563e230dd2SCorentin Chary 
578cf36489SGerd Hoffmann static void vnc_set_share_mode(VncState *vs, VncShareMode mode)
588cf36489SGerd Hoffmann {
598cf36489SGerd Hoffmann #ifdef _VNC_DEBUG
608cf36489SGerd Hoffmann     static const char *mn[] = {
618cf36489SGerd Hoffmann         [0]                           = "undefined",
628cf36489SGerd Hoffmann         [VNC_SHARE_MODE_CONNECTING]   = "connecting",
638cf36489SGerd Hoffmann         [VNC_SHARE_MODE_SHARED]       = "shared",
648cf36489SGerd Hoffmann         [VNC_SHARE_MODE_EXCLUSIVE]    = "exclusive",
658cf36489SGerd Hoffmann         [VNC_SHARE_MODE_DISCONNECTED] = "disconnected",
668cf36489SGerd Hoffmann     };
678cf36489SGerd Hoffmann     fprintf(stderr, "%s/%d: %s -> %s\n", __func__,
688cf36489SGerd Hoffmann             vs->csock, mn[vs->share_mode], mn[mode]);
698cf36489SGerd Hoffmann #endif
708cf36489SGerd Hoffmann 
71e5f34cddSGerd Hoffmann     switch (vs->share_mode) {
72e5f34cddSGerd Hoffmann     case VNC_SHARE_MODE_CONNECTING:
73e5f34cddSGerd Hoffmann         vs->vd->num_connecting--;
74e5f34cddSGerd Hoffmann         break;
75e5f34cddSGerd Hoffmann     case VNC_SHARE_MODE_SHARED:
76e5f34cddSGerd Hoffmann         vs->vd->num_shared--;
77e5f34cddSGerd Hoffmann         break;
78e5f34cddSGerd Hoffmann     case VNC_SHARE_MODE_EXCLUSIVE:
798cf36489SGerd Hoffmann         vs->vd->num_exclusive--;
80e5f34cddSGerd Hoffmann         break;
81e5f34cddSGerd Hoffmann     default:
82e5f34cddSGerd Hoffmann         break;
838cf36489SGerd Hoffmann     }
84e5f34cddSGerd Hoffmann 
858cf36489SGerd Hoffmann     vs->share_mode = mode;
86e5f34cddSGerd Hoffmann 
87e5f34cddSGerd Hoffmann     switch (vs->share_mode) {
88e5f34cddSGerd Hoffmann     case VNC_SHARE_MODE_CONNECTING:
89e5f34cddSGerd Hoffmann         vs->vd->num_connecting++;
90e5f34cddSGerd Hoffmann         break;
91e5f34cddSGerd Hoffmann     case VNC_SHARE_MODE_SHARED:
92e5f34cddSGerd Hoffmann         vs->vd->num_shared++;
93e5f34cddSGerd Hoffmann         break;
94e5f34cddSGerd Hoffmann     case VNC_SHARE_MODE_EXCLUSIVE:
958cf36489SGerd Hoffmann         vs->vd->num_exclusive++;
96e5f34cddSGerd Hoffmann         break;
97e5f34cddSGerd Hoffmann     default:
98e5f34cddSGerd Hoffmann         break;
998cf36489SGerd Hoffmann     }
1008cf36489SGerd Hoffmann }
1018cf36489SGerd Hoffmann 
1023e230dd2SCorentin Chary static char *addr_to_string(const char *format,
1033e230dd2SCorentin Chary                             struct sockaddr_storage *sa,
1043e230dd2SCorentin Chary                             socklen_t salen) {
1053e230dd2SCorentin Chary     char *addr;
1063e230dd2SCorentin Chary     char host[NI_MAXHOST];
1073e230dd2SCorentin Chary     char serv[NI_MAXSERV];
1083e230dd2SCorentin Chary     int err;
1093e230dd2SCorentin Chary     size_t addrlen;
1103e230dd2SCorentin Chary 
1113e230dd2SCorentin Chary     if ((err = getnameinfo((struct sockaddr *)sa, salen,
1123e230dd2SCorentin Chary                            host, sizeof(host),
1133e230dd2SCorentin Chary                            serv, sizeof(serv),
1143e230dd2SCorentin Chary                            NI_NUMERICHOST | NI_NUMERICSERV)) != 0) {
1153e230dd2SCorentin Chary         VNC_DEBUG("Cannot resolve address %d: %s\n",
1163e230dd2SCorentin Chary                   err, gai_strerror(err));
1173e230dd2SCorentin Chary         return NULL;
1183e230dd2SCorentin Chary     }
1193e230dd2SCorentin Chary 
1203e230dd2SCorentin Chary     /* Enough for the existing format + the 2 vars we're
1213e230dd2SCorentin Chary      * substituting in. */
1223e230dd2SCorentin Chary     addrlen = strlen(format) + strlen(host) + strlen(serv);
1237267c094SAnthony Liguori     addr = g_malloc(addrlen + 1);
1243e230dd2SCorentin Chary     snprintf(addr, addrlen, format, host, serv);
1253e230dd2SCorentin Chary     addr[addrlen] = '\0';
1263e230dd2SCorentin Chary 
1273e230dd2SCorentin Chary     return addr;
1283e230dd2SCorentin Chary }
1293e230dd2SCorentin Chary 
1303e230dd2SCorentin Chary 
1313e230dd2SCorentin Chary char *vnc_socket_local_addr(const char *format, int fd) {
1323e230dd2SCorentin Chary     struct sockaddr_storage sa;
1333e230dd2SCorentin Chary     socklen_t salen;
1343e230dd2SCorentin Chary 
1353e230dd2SCorentin Chary     salen = sizeof(sa);
1363e230dd2SCorentin Chary     if (getsockname(fd, (struct sockaddr*)&sa, &salen) < 0)
1373e230dd2SCorentin Chary         return NULL;
1383e230dd2SCorentin Chary 
1393e230dd2SCorentin Chary     return addr_to_string(format, &sa, salen);
1403e230dd2SCorentin Chary }
1413e230dd2SCorentin Chary 
1423e230dd2SCorentin Chary char *vnc_socket_remote_addr(const char *format, int fd) {
1433e230dd2SCorentin Chary     struct sockaddr_storage sa;
1443e230dd2SCorentin Chary     socklen_t salen;
1453e230dd2SCorentin Chary 
1463e230dd2SCorentin Chary     salen = sizeof(sa);
1473e230dd2SCorentin Chary     if (getpeername(fd, (struct sockaddr*)&sa, &salen) < 0)
1483e230dd2SCorentin Chary         return NULL;
1493e230dd2SCorentin Chary 
1503e230dd2SCorentin Chary     return addr_to_string(format, &sa, salen);
1513e230dd2SCorentin Chary }
1523e230dd2SCorentin Chary 
153fb6ba0d5SWenchao Xia static VncBasicInfo *vnc_basic_info_get(struct sockaddr_storage *sa,
1543e230dd2SCorentin Chary                                         socklen_t salen)
1553e230dd2SCorentin Chary {
156fb6ba0d5SWenchao Xia     VncBasicInfo *info;
1573e230dd2SCorentin Chary     char host[NI_MAXHOST];
1583e230dd2SCorentin Chary     char serv[NI_MAXSERV];
1593e230dd2SCorentin Chary     int err;
1603e230dd2SCorentin Chary 
1613e230dd2SCorentin Chary     if ((err = getnameinfo((struct sockaddr *)sa, salen,
1623e230dd2SCorentin Chary                            host, sizeof(host),
1633e230dd2SCorentin Chary                            serv, sizeof(serv),
1643e230dd2SCorentin Chary                            NI_NUMERICHOST | NI_NUMERICSERV)) != 0) {
1653e230dd2SCorentin Chary         VNC_DEBUG("Cannot resolve address %d: %s\n",
1663e230dd2SCorentin Chary                   err, gai_strerror(err));
167fb6ba0d5SWenchao Xia         return NULL;
1683e230dd2SCorentin Chary     }
1693e230dd2SCorentin Chary 
170fb6ba0d5SWenchao Xia     info = g_malloc0(sizeof(VncBasicInfo));
171fb6ba0d5SWenchao Xia     info->host = g_strdup(host);
172fb6ba0d5SWenchao Xia     info->service = g_strdup(serv);
173fb6ba0d5SWenchao Xia     info->family = inet_netfamily(sa->ss_family);
174fb6ba0d5SWenchao Xia     return info;
1753e230dd2SCorentin Chary }
1763e230dd2SCorentin Chary 
177fb6ba0d5SWenchao Xia static VncBasicInfo *vnc_basic_info_get_from_server_addr(int fd)
1783e230dd2SCorentin Chary {
1793e230dd2SCorentin Chary     struct sockaddr_storage sa;
1803e230dd2SCorentin Chary     socklen_t salen;
1813e230dd2SCorentin Chary 
1823e230dd2SCorentin Chary     salen = sizeof(sa);
1833e230dd2SCorentin Chary     if (getsockname(fd, (struct sockaddr*)&sa, &salen) < 0) {
184fb6ba0d5SWenchao Xia         return NULL;
1853e230dd2SCorentin Chary     }
1863e230dd2SCorentin Chary 
187fb6ba0d5SWenchao Xia     return vnc_basic_info_get(&sa, salen);
1883e230dd2SCorentin Chary }
1893e230dd2SCorentin Chary 
190fb6ba0d5SWenchao Xia static VncBasicInfo *vnc_basic_info_get_from_remote_addr(int fd)
1913e230dd2SCorentin Chary {
1923e230dd2SCorentin Chary     struct sockaddr_storage sa;
1933e230dd2SCorentin Chary     socklen_t salen;
1943e230dd2SCorentin Chary 
1953e230dd2SCorentin Chary     salen = sizeof(sa);
1963e230dd2SCorentin Chary     if (getpeername(fd, (struct sockaddr*)&sa, &salen) < 0) {
197fb6ba0d5SWenchao Xia         return NULL;
1983e230dd2SCorentin Chary     }
1993e230dd2SCorentin Chary 
200fb6ba0d5SWenchao Xia     return vnc_basic_info_get(&sa, salen);
2013e230dd2SCorentin Chary }
2023e230dd2SCorentin Chary 
2033e230dd2SCorentin Chary static const char *vnc_auth_name(VncDisplay *vd) {
2043e230dd2SCorentin Chary     switch (vd->auth) {
2053e230dd2SCorentin Chary     case VNC_AUTH_INVALID:
2063e230dd2SCorentin Chary         return "invalid";
2073e230dd2SCorentin Chary     case VNC_AUTH_NONE:
2083e230dd2SCorentin Chary         return "none";
2093e230dd2SCorentin Chary     case VNC_AUTH_VNC:
2103e230dd2SCorentin Chary         return "vnc";
2113e230dd2SCorentin Chary     case VNC_AUTH_RA2:
2123e230dd2SCorentin Chary         return "ra2";
2133e230dd2SCorentin Chary     case VNC_AUTH_RA2NE:
2143e230dd2SCorentin Chary         return "ra2ne";
2153e230dd2SCorentin Chary     case VNC_AUTH_TIGHT:
2163e230dd2SCorentin Chary         return "tight";
2173e230dd2SCorentin Chary     case VNC_AUTH_ULTRA:
2183e230dd2SCorentin Chary         return "ultra";
2193e230dd2SCorentin Chary     case VNC_AUTH_TLS:
2203e230dd2SCorentin Chary         return "tls";
2213e230dd2SCorentin Chary     case VNC_AUTH_VENCRYPT:
2223e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
2233e230dd2SCorentin Chary         switch (vd->subauth) {
2243e230dd2SCorentin Chary         case VNC_AUTH_VENCRYPT_PLAIN:
2253e230dd2SCorentin Chary             return "vencrypt+plain";
2263e230dd2SCorentin Chary         case VNC_AUTH_VENCRYPT_TLSNONE:
2273e230dd2SCorentin Chary             return "vencrypt+tls+none";
2283e230dd2SCorentin Chary         case VNC_AUTH_VENCRYPT_TLSVNC:
2293e230dd2SCorentin Chary             return "vencrypt+tls+vnc";
2303e230dd2SCorentin Chary         case VNC_AUTH_VENCRYPT_TLSPLAIN:
2313e230dd2SCorentin Chary             return "vencrypt+tls+plain";
2323e230dd2SCorentin Chary         case VNC_AUTH_VENCRYPT_X509NONE:
2333e230dd2SCorentin Chary             return "vencrypt+x509+none";
2343e230dd2SCorentin Chary         case VNC_AUTH_VENCRYPT_X509VNC:
2353e230dd2SCorentin Chary             return "vencrypt+x509+vnc";
2363e230dd2SCorentin Chary         case VNC_AUTH_VENCRYPT_X509PLAIN:
2373e230dd2SCorentin Chary             return "vencrypt+x509+plain";
2383e230dd2SCorentin Chary         case VNC_AUTH_VENCRYPT_TLSSASL:
2393e230dd2SCorentin Chary             return "vencrypt+tls+sasl";
2403e230dd2SCorentin Chary         case VNC_AUTH_VENCRYPT_X509SASL:
2413e230dd2SCorentin Chary             return "vencrypt+x509+sasl";
2423e230dd2SCorentin Chary         default:
2433e230dd2SCorentin Chary             return "vencrypt";
2443e230dd2SCorentin Chary         }
2453e230dd2SCorentin Chary #else
2463e230dd2SCorentin Chary         return "vencrypt";
2473e230dd2SCorentin Chary #endif
2483e230dd2SCorentin Chary     case VNC_AUTH_SASL:
2493e230dd2SCorentin Chary         return "sasl";
2503e230dd2SCorentin Chary     }
2513e230dd2SCorentin Chary     return "unknown";
2523e230dd2SCorentin Chary }
2533e230dd2SCorentin Chary 
254d616ccc5SGerd Hoffmann static VncServerInfo *vnc_server_info_get(VncDisplay *vd)
2553e230dd2SCorentin Chary {
256fb6ba0d5SWenchao Xia     VncServerInfo *info;
257d616ccc5SGerd Hoffmann     VncBasicInfo *bi = vnc_basic_info_get_from_server_addr(vd->lsock);
258fb6ba0d5SWenchao Xia     if (!bi) {
259fb6ba0d5SWenchao Xia         return NULL;
2603e230dd2SCorentin Chary     }
2613e230dd2SCorentin Chary 
262fb6ba0d5SWenchao Xia     info = g_malloc(sizeof(*info));
263fb6ba0d5SWenchao Xia     info->base = bi;
264fb6ba0d5SWenchao Xia     info->has_auth = true;
265d616ccc5SGerd Hoffmann     info->auth = g_strdup(vnc_auth_name(vd));
266fb6ba0d5SWenchao Xia     return info;
2673e230dd2SCorentin Chary }
2683e230dd2SCorentin Chary 
2693e230dd2SCorentin Chary static void vnc_client_cache_auth(VncState *client)
2703e230dd2SCorentin Chary {
2713e230dd2SCorentin Chary     if (!client->info) {
2723e230dd2SCorentin Chary         return;
2733e230dd2SCorentin Chary     }
2743e230dd2SCorentin Chary 
2753e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
2763e230dd2SCorentin Chary     if (client->tls.session &&
2773e230dd2SCorentin Chary         client->tls.dname) {
278fb6ba0d5SWenchao Xia         client->info->has_x509_dname = true;
279fb6ba0d5SWenchao Xia         client->info->x509_dname = g_strdup(client->tls.dname);
2803e230dd2SCorentin Chary     }
2813e230dd2SCorentin Chary #endif
2823e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
2833e230dd2SCorentin Chary     if (client->sasl.conn &&
2843e230dd2SCorentin Chary         client->sasl.username) {
285fb6ba0d5SWenchao Xia         client->info->has_sasl_username = true;
286fb6ba0d5SWenchao Xia         client->info->sasl_username = g_strdup(client->sasl.username);
2873e230dd2SCorentin Chary     }
2883e230dd2SCorentin Chary #endif
2893e230dd2SCorentin Chary }
2903e230dd2SCorentin Chary 
2913e230dd2SCorentin Chary static void vnc_client_cache_addr(VncState *client)
2923e230dd2SCorentin Chary {
293fb6ba0d5SWenchao Xia     VncBasicInfo *bi = vnc_basic_info_get_from_remote_addr(client->csock);
2943e230dd2SCorentin Chary 
295fb6ba0d5SWenchao Xia     if (bi) {
296fb6ba0d5SWenchao Xia         client->info = g_malloc0(sizeof(*client->info));
297fb6ba0d5SWenchao Xia         client->info->base = bi;
298fb6ba0d5SWenchao Xia     }
2993e230dd2SCorentin Chary }
3003e230dd2SCorentin Chary 
301fb6ba0d5SWenchao Xia static void vnc_qmp_event(VncState *vs, QAPIEvent event)
3023e230dd2SCorentin Chary {
303fb6ba0d5SWenchao Xia     VncServerInfo *si;
3043e230dd2SCorentin Chary 
3053e230dd2SCorentin Chary     if (!vs->info) {
3063e230dd2SCorentin Chary         return;
3073e230dd2SCorentin Chary     }
308fb6ba0d5SWenchao Xia     g_assert(vs->info->base);
3093e230dd2SCorentin Chary 
310d616ccc5SGerd Hoffmann     si = vnc_server_info_get(vs->vd);
311fb6ba0d5SWenchao Xia     if (!si) {
3123e230dd2SCorentin Chary         return;
3133e230dd2SCorentin Chary     }
3143e230dd2SCorentin Chary 
315fb6ba0d5SWenchao Xia     switch (event) {
316fb6ba0d5SWenchao Xia     case QAPI_EVENT_VNC_CONNECTED:
317fb6ba0d5SWenchao Xia         qapi_event_send_vnc_connected(si, vs->info->base, &error_abort);
318fb6ba0d5SWenchao Xia         break;
319fb6ba0d5SWenchao Xia     case QAPI_EVENT_VNC_INITIALIZED:
320fb6ba0d5SWenchao Xia         qapi_event_send_vnc_initialized(si, vs->info, &error_abort);
321fb6ba0d5SWenchao Xia         break;
322fb6ba0d5SWenchao Xia     case QAPI_EVENT_VNC_DISCONNECTED:
323fb6ba0d5SWenchao Xia         qapi_event_send_vnc_disconnected(si, vs->info, &error_abort);
324fb6ba0d5SWenchao Xia         break;
325fb6ba0d5SWenchao Xia     default:
326fb6ba0d5SWenchao Xia         break;
327fb6ba0d5SWenchao Xia     }
3283e230dd2SCorentin Chary 
329fb6ba0d5SWenchao Xia     qapi_free_VncServerInfo(si);
3303e230dd2SCorentin Chary }
3313e230dd2SCorentin Chary 
3322b54aa87SLuiz Capitulino static VncClientInfo *qmp_query_vnc_client(const VncState *client)
3333e230dd2SCorentin Chary {
3342b54aa87SLuiz Capitulino     struct sockaddr_storage sa;
3352b54aa87SLuiz Capitulino     socklen_t salen = sizeof(sa);
3362b54aa87SLuiz Capitulino     char host[NI_MAXHOST];
3372b54aa87SLuiz Capitulino     char serv[NI_MAXSERV];
3382b54aa87SLuiz Capitulino     VncClientInfo *info;
3393e230dd2SCorentin Chary 
3402b54aa87SLuiz Capitulino     if (getpeername(client->csock, (struct sockaddr *)&sa, &salen) < 0) {
3412b54aa87SLuiz Capitulino         return NULL;
3422b54aa87SLuiz Capitulino     }
3432b54aa87SLuiz Capitulino 
3442b54aa87SLuiz Capitulino     if (getnameinfo((struct sockaddr *)&sa, salen,
3452b54aa87SLuiz Capitulino                     host, sizeof(host),
3462b54aa87SLuiz Capitulino                     serv, sizeof(serv),
3472b54aa87SLuiz Capitulino                     NI_NUMERICHOST | NI_NUMERICSERV) < 0) {
3482b54aa87SLuiz Capitulino         return NULL;
3492b54aa87SLuiz Capitulino     }
3502b54aa87SLuiz Capitulino 
3512b54aa87SLuiz Capitulino     info = g_malloc0(sizeof(*info));
352a589569fSWenchao Xia     info->base = g_malloc0(sizeof(*info->base));
353a589569fSWenchao Xia     info->base->host = g_strdup(host);
354a589569fSWenchao Xia     info->base->service = g_strdup(serv);
355a589569fSWenchao Xia     info->base->family = inet_netfamily(sa.ss_family);
3564478aa76SGerd Hoffmann #ifdef CONFIG_VNC_WS
3574478aa76SGerd Hoffmann     info->base->websocket = client->websocket;
3584478aa76SGerd Hoffmann #endif
3593e230dd2SCorentin Chary 
3603e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
3612b54aa87SLuiz Capitulino     if (client->tls.session && client->tls.dname) {
3622b54aa87SLuiz Capitulino         info->has_x509_dname = true;
3632b54aa87SLuiz Capitulino         info->x509_dname = g_strdup(client->tls.dname);
3642b54aa87SLuiz Capitulino     }
3653e230dd2SCorentin Chary #endif
3663e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
3672b54aa87SLuiz Capitulino     if (client->sasl.conn && client->sasl.username) {
3682b54aa87SLuiz Capitulino         info->has_sasl_username = true;
3692b54aa87SLuiz Capitulino         info->sasl_username = g_strdup(client->sasl.username);
3702b54aa87SLuiz Capitulino     }
3713e230dd2SCorentin Chary #endif
3722b54aa87SLuiz Capitulino 
3732b54aa87SLuiz Capitulino     return info;
3743e230dd2SCorentin Chary }
3753e230dd2SCorentin Chary 
376d616ccc5SGerd Hoffmann static VncDisplay *vnc_display_find(const char *id)
377d616ccc5SGerd Hoffmann {
378d616ccc5SGerd Hoffmann     VncDisplay *vd;
379d616ccc5SGerd Hoffmann 
380d616ccc5SGerd Hoffmann     if (id == NULL) {
381d616ccc5SGerd Hoffmann         return QTAILQ_FIRST(&vnc_displays);
382d616ccc5SGerd Hoffmann     }
383d616ccc5SGerd Hoffmann     QTAILQ_FOREACH(vd, &vnc_displays, next) {
384d616ccc5SGerd Hoffmann         if (strcmp(id, vd->id) == 0) {
385d616ccc5SGerd Hoffmann             return vd;
386d616ccc5SGerd Hoffmann         }
387d616ccc5SGerd Hoffmann     }
388d616ccc5SGerd Hoffmann     return NULL;
389d616ccc5SGerd Hoffmann }
390d616ccc5SGerd Hoffmann 
3912d29a436SGerd Hoffmann static VncClientInfoList *qmp_query_client_list(VncDisplay *vd)
3922d29a436SGerd Hoffmann {
3932d29a436SGerd Hoffmann     VncClientInfoList *cinfo, *prev = NULL;
3942d29a436SGerd Hoffmann     VncState *client;
3952d29a436SGerd Hoffmann 
3962d29a436SGerd Hoffmann     QTAILQ_FOREACH(client, &vd->clients, next) {
3972d29a436SGerd Hoffmann         cinfo = g_new0(VncClientInfoList, 1);
3982d29a436SGerd Hoffmann         cinfo->value = qmp_query_vnc_client(client);
3992d29a436SGerd Hoffmann         cinfo->next = prev;
4002d29a436SGerd Hoffmann         prev = cinfo;
4012d29a436SGerd Hoffmann     }
4022d29a436SGerd Hoffmann     return prev;
4032d29a436SGerd Hoffmann }
4042d29a436SGerd Hoffmann 
4052b54aa87SLuiz Capitulino VncInfo *qmp_query_vnc(Error **errp)
4063e230dd2SCorentin Chary {
4072b54aa87SLuiz Capitulino     VncInfo *info = g_malloc0(sizeof(*info));
408d616ccc5SGerd Hoffmann     VncDisplay *vd = vnc_display_find(NULL);
4093e230dd2SCorentin Chary 
410bf7aa45eSGerd Hoffmann     if (vd == NULL || !vd->enabled) {
4112b54aa87SLuiz Capitulino         info->enabled = false;
4123e230dd2SCorentin Chary     } else {
4132b54aa87SLuiz Capitulino         struct sockaddr_storage sa;
4142b54aa87SLuiz Capitulino         socklen_t salen = sizeof(sa);
4152b54aa87SLuiz Capitulino         char host[NI_MAXHOST];
4162b54aa87SLuiz Capitulino         char serv[NI_MAXSERV];
4173e230dd2SCorentin Chary 
4182b54aa87SLuiz Capitulino         info->enabled = true;
4192b54aa87SLuiz Capitulino 
4202b54aa87SLuiz Capitulino         /* for compatibility with the original command */
4212b54aa87SLuiz Capitulino         info->has_clients = true;
4222d29a436SGerd Hoffmann         info->clients = qmp_query_client_list(vd);
4233e230dd2SCorentin Chary 
424d616ccc5SGerd Hoffmann         if (vd->lsock == -1) {
425417b0b88SPaolo Bonzini             return info;
426417b0b88SPaolo Bonzini         }
427417b0b88SPaolo Bonzini 
428d616ccc5SGerd Hoffmann         if (getsockname(vd->lsock, (struct sockaddr *)&sa,
4292b54aa87SLuiz Capitulino                         &salen) == -1) {
4302b54aa87SLuiz Capitulino             error_set(errp, QERR_UNDEFINED_ERROR);
4312b54aa87SLuiz Capitulino             goto out_error;
4322b54aa87SLuiz Capitulino         }
4333e230dd2SCorentin Chary 
4342b54aa87SLuiz Capitulino         if (getnameinfo((struct sockaddr *)&sa, salen,
4352b54aa87SLuiz Capitulino                         host, sizeof(host),
4362b54aa87SLuiz Capitulino                         serv, sizeof(serv),
4372b54aa87SLuiz Capitulino                         NI_NUMERICHOST | NI_NUMERICSERV) < 0) {
4382b54aa87SLuiz Capitulino             error_set(errp, QERR_UNDEFINED_ERROR);
4392b54aa87SLuiz Capitulino             goto out_error;
4403e230dd2SCorentin Chary         }
4412b54aa87SLuiz Capitulino 
4422b54aa87SLuiz Capitulino         info->has_host = true;
4432b54aa87SLuiz Capitulino         info->host = g_strdup(host);
4442b54aa87SLuiz Capitulino 
4452b54aa87SLuiz Capitulino         info->has_service = true;
4462b54aa87SLuiz Capitulino         info->service = g_strdup(serv);
4472b54aa87SLuiz Capitulino 
4482b54aa87SLuiz Capitulino         info->has_family = true;
449a589569fSWenchao Xia         info->family = inet_netfamily(sa.ss_family);
4502b54aa87SLuiz Capitulino 
4512b54aa87SLuiz Capitulino         info->has_auth = true;
452d616ccc5SGerd Hoffmann         info->auth = g_strdup(vnc_auth_name(vd));
4533e230dd2SCorentin Chary     }
4542b54aa87SLuiz Capitulino 
4552b54aa87SLuiz Capitulino     return info;
4562b54aa87SLuiz Capitulino 
4572b54aa87SLuiz Capitulino out_error:
4582b54aa87SLuiz Capitulino     qapi_free_VncInfo(info);
4592b54aa87SLuiz Capitulino     return NULL;
4603e230dd2SCorentin Chary }
4613e230dd2SCorentin Chary 
462df887684SGerd Hoffmann static VncBasicInfoList *qmp_query_server_entry(int socket,
4634478aa76SGerd Hoffmann                                                 bool websocket,
464df887684SGerd Hoffmann                                                 VncBasicInfoList *prev)
465df887684SGerd Hoffmann {
466df887684SGerd Hoffmann     VncBasicInfoList *list;
467df887684SGerd Hoffmann     VncBasicInfo *info;
468df887684SGerd Hoffmann     struct sockaddr_storage sa;
469df887684SGerd Hoffmann     socklen_t salen = sizeof(sa);
470df887684SGerd Hoffmann     char host[NI_MAXHOST];
471df887684SGerd Hoffmann     char serv[NI_MAXSERV];
472df887684SGerd Hoffmann 
473df887684SGerd Hoffmann     if (getsockname(socket, (struct sockaddr *)&sa, &salen) < 0 ||
474df887684SGerd Hoffmann         getnameinfo((struct sockaddr *)&sa, salen,
475df887684SGerd Hoffmann                     host, sizeof(host), serv, sizeof(serv),
476df887684SGerd Hoffmann                     NI_NUMERICHOST | NI_NUMERICSERV) < 0) {
477df887684SGerd Hoffmann         return prev;
478df887684SGerd Hoffmann     }
479df887684SGerd Hoffmann 
480df887684SGerd Hoffmann     info = g_new0(VncBasicInfo, 1);
481df887684SGerd Hoffmann     info->host = g_strdup(host);
482df887684SGerd Hoffmann     info->service = g_strdup(serv);
483df887684SGerd Hoffmann     info->family = inet_netfamily(sa.ss_family);
4844478aa76SGerd Hoffmann     info->websocket = websocket;
485df887684SGerd Hoffmann 
486df887684SGerd Hoffmann     list = g_new0(VncBasicInfoList, 1);
487df887684SGerd Hoffmann     list->value = info;
488df887684SGerd Hoffmann     list->next = prev;
489df887684SGerd Hoffmann     return list;
490df887684SGerd Hoffmann }
491df887684SGerd Hoffmann 
492df887684SGerd Hoffmann static void qmp_query_auth(VncDisplay *vd, VncInfo2 *info)
493df887684SGerd Hoffmann {
494df887684SGerd Hoffmann     switch (vd->auth) {
495df887684SGerd Hoffmann     case VNC_AUTH_VNC:
496df887684SGerd Hoffmann         info->auth = VNC_PRIMARY_AUTH_VNC;
497df887684SGerd Hoffmann         break;
498df887684SGerd Hoffmann     case VNC_AUTH_RA2:
499df887684SGerd Hoffmann         info->auth = VNC_PRIMARY_AUTH_RA2;
500df887684SGerd Hoffmann         break;
501df887684SGerd Hoffmann     case VNC_AUTH_RA2NE:
502df887684SGerd Hoffmann         info->auth = VNC_PRIMARY_AUTH_RA2NE;
503df887684SGerd Hoffmann         break;
504df887684SGerd Hoffmann     case VNC_AUTH_TIGHT:
505df887684SGerd Hoffmann         info->auth = VNC_PRIMARY_AUTH_TIGHT;
506df887684SGerd Hoffmann         break;
507df887684SGerd Hoffmann     case VNC_AUTH_ULTRA:
508df887684SGerd Hoffmann         info->auth = VNC_PRIMARY_AUTH_ULTRA;
509df887684SGerd Hoffmann         break;
510df887684SGerd Hoffmann     case VNC_AUTH_TLS:
511df887684SGerd Hoffmann         info->auth = VNC_PRIMARY_AUTH_TLS;
512df887684SGerd Hoffmann         break;
513df887684SGerd Hoffmann     case VNC_AUTH_VENCRYPT:
514df887684SGerd Hoffmann         info->auth = VNC_PRIMARY_AUTH_VENCRYPT;
515df887684SGerd Hoffmann #ifdef CONFIG_VNC_TLS
516df887684SGerd Hoffmann         info->has_vencrypt = true;
517df887684SGerd Hoffmann         switch (vd->subauth) {
518df887684SGerd Hoffmann         case VNC_AUTH_VENCRYPT_PLAIN:
519df887684SGerd Hoffmann             info->vencrypt = VNC_VENCRYPT_SUB_AUTH_PLAIN;
520df887684SGerd Hoffmann             break;
521df887684SGerd Hoffmann         case VNC_AUTH_VENCRYPT_TLSNONE:
522df887684SGerd Hoffmann             info->vencrypt = VNC_VENCRYPT_SUB_AUTH_TLS_NONE;
523df887684SGerd Hoffmann             break;
524df887684SGerd Hoffmann         case VNC_AUTH_VENCRYPT_TLSVNC:
525df887684SGerd Hoffmann             info->vencrypt = VNC_VENCRYPT_SUB_AUTH_TLS_VNC;
526df887684SGerd Hoffmann             break;
527df887684SGerd Hoffmann         case VNC_AUTH_VENCRYPT_TLSPLAIN:
528df887684SGerd Hoffmann             info->vencrypt = VNC_VENCRYPT_SUB_AUTH_TLS_PLAIN;
529df887684SGerd Hoffmann             break;
530df887684SGerd Hoffmann         case VNC_AUTH_VENCRYPT_X509NONE:
531df887684SGerd Hoffmann             info->vencrypt = VNC_VENCRYPT_SUB_AUTH_X509_NONE;
532df887684SGerd Hoffmann             break;
533df887684SGerd Hoffmann         case VNC_AUTH_VENCRYPT_X509VNC:
534df887684SGerd Hoffmann             info->vencrypt = VNC_VENCRYPT_SUB_AUTH_X509_VNC;
535df887684SGerd Hoffmann             break;
536df887684SGerd Hoffmann         case VNC_AUTH_VENCRYPT_X509PLAIN:
537df887684SGerd Hoffmann             info->vencrypt = VNC_VENCRYPT_SUB_AUTH_X509_PLAIN;
538df887684SGerd Hoffmann             break;
539df887684SGerd Hoffmann         case VNC_AUTH_VENCRYPT_TLSSASL:
540df887684SGerd Hoffmann             info->vencrypt = VNC_VENCRYPT_SUB_AUTH_TLS_SASL;
541df887684SGerd Hoffmann             break;
542df887684SGerd Hoffmann         case VNC_AUTH_VENCRYPT_X509SASL:
543df887684SGerd Hoffmann             info->vencrypt = VNC_VENCRYPT_SUB_AUTH_X509_SASL;
544df887684SGerd Hoffmann             break;
545df887684SGerd Hoffmann         default:
546df887684SGerd Hoffmann             info->has_vencrypt = false;
547df887684SGerd Hoffmann             break;
548df887684SGerd Hoffmann         }
549df887684SGerd Hoffmann #endif
550df887684SGerd Hoffmann         break;
551df887684SGerd Hoffmann     case VNC_AUTH_SASL:
552df887684SGerd Hoffmann         info->auth = VNC_PRIMARY_AUTH_SASL;
553df887684SGerd Hoffmann         break;
554df887684SGerd Hoffmann     case VNC_AUTH_NONE:
555df887684SGerd Hoffmann     default:
556df887684SGerd Hoffmann         info->auth = VNC_PRIMARY_AUTH_NONE;
557df887684SGerd Hoffmann         break;
558df887684SGerd Hoffmann     }
559df887684SGerd Hoffmann }
560df887684SGerd Hoffmann 
561df887684SGerd Hoffmann VncInfo2List *qmp_query_vnc_servers(Error **errp)
562df887684SGerd Hoffmann {
563df887684SGerd Hoffmann     VncInfo2List *item, *prev = NULL;
564df887684SGerd Hoffmann     VncInfo2 *info;
565df887684SGerd Hoffmann     VncDisplay *vd;
566df887684SGerd Hoffmann     DeviceState *dev;
567df887684SGerd Hoffmann 
568df887684SGerd Hoffmann     QTAILQ_FOREACH(vd, &vnc_displays, next) {
569df887684SGerd Hoffmann         info = g_new0(VncInfo2, 1);
570df887684SGerd Hoffmann         info->id = g_strdup(vd->id);
571df887684SGerd Hoffmann         info->clients = qmp_query_client_list(vd);
572df887684SGerd Hoffmann         qmp_query_auth(vd, info);
573df887684SGerd Hoffmann         if (vd->dcl.con) {
574df887684SGerd Hoffmann             dev = DEVICE(object_property_get_link(OBJECT(vd->dcl.con),
575df887684SGerd Hoffmann                                                   "device", NULL));
576df887684SGerd Hoffmann             info->has_display = true;
577df887684SGerd Hoffmann             info->display = g_strdup(dev->id);
578df887684SGerd Hoffmann         }
579df887684SGerd Hoffmann         if (vd->lsock != -1) {
5804478aa76SGerd Hoffmann             info->server = qmp_query_server_entry(vd->lsock, false,
581df887684SGerd Hoffmann                                                   info->server);
582df887684SGerd Hoffmann         }
583df887684SGerd Hoffmann #ifdef CONFIG_VNC_WS
584df887684SGerd Hoffmann         if (vd->lwebsock != -1) {
5854478aa76SGerd Hoffmann             info->server = qmp_query_server_entry(vd->lwebsock, true,
5864478aa76SGerd Hoffmann                                                   info->server);
587df887684SGerd Hoffmann         }
588df887684SGerd Hoffmann #endif
589df887684SGerd Hoffmann 
590df887684SGerd Hoffmann         item = g_new0(VncInfo2List, 1);
591df887684SGerd Hoffmann         item->value = info;
592df887684SGerd Hoffmann         item->next = prev;
593df887684SGerd Hoffmann         prev = item;
594df887684SGerd Hoffmann     }
595df887684SGerd Hoffmann     return prev;
596df887684SGerd Hoffmann }
597df887684SGerd Hoffmann 
5983e230dd2SCorentin Chary /* TODO
5993e230dd2SCorentin Chary    1) Get the queue working for IO.
6003e230dd2SCorentin Chary    2) there is some weirdness when using the -S option (the screen is grey
6013e230dd2SCorentin Chary       and not totally invalidated
6023e230dd2SCorentin Chary    3) resolutions > 1024
6033e230dd2SCorentin Chary */
6043e230dd2SCorentin Chary 
60538ee14f4SGerd Hoffmann static int vnc_update_client(VncState *vs, int has_dirty, bool sync);
6063e230dd2SCorentin Chary static void vnc_disconnect_start(VncState *vs);
6073e230dd2SCorentin Chary 
6083e230dd2SCorentin Chary static void vnc_colordepth(VncState *vs);
6093e230dd2SCorentin Chary static void framebuffer_update_request(VncState *vs, int incremental,
6103e230dd2SCorentin Chary                                        int x_position, int y_position,
6113e230dd2SCorentin Chary                                        int w, int h);
6120f7b2864SGerd Hoffmann static void vnc_refresh(DisplayChangeListener *dcl);
6133e230dd2SCorentin Chary static int vnc_refresh_server_surface(VncDisplay *vd);
6143e230dd2SCorentin Chary 
615bea60dd7SPeter Lieven static void vnc_set_area_dirty(DECLARE_BITMAP(dirty[VNC_MAX_HEIGHT],
616bea60dd7SPeter Lieven                                VNC_MAX_WIDTH / VNC_DIRTY_PIXELS_PER_BIT),
617bea60dd7SPeter Lieven                                int width, int height,
618bea60dd7SPeter Lieven                                int x, int y, int w, int h) {
61991937225SPeter Lieven     /* this is needed this to ensure we updated all affected
62091937225SPeter Lieven      * blocks if x % VNC_DIRTY_PIXELS_PER_BIT != 0 */
621b4c85ddcSPeter Lieven     w += (x % VNC_DIRTY_PIXELS_PER_BIT);
622b4c85ddcSPeter Lieven     x -= (x % VNC_DIRTY_PIXELS_PER_BIT);
6233e230dd2SCorentin Chary 
6249f64916dSGerd Hoffmann     x = MIN(x, width);
6259f64916dSGerd Hoffmann     y = MIN(y, height);
6269f64916dSGerd Hoffmann     w = MIN(x + w, width) - x;
62791937225SPeter Lieven     h = MIN(y + h, height);
6283e230dd2SCorentin Chary 
629b4c85ddcSPeter Lieven     for (; y < h; y++) {
630bea60dd7SPeter Lieven         bitmap_set(dirty[y], x / VNC_DIRTY_PIXELS_PER_BIT,
63191937225SPeter Lieven                    DIV_ROUND_UP(w, VNC_DIRTY_PIXELS_PER_BIT));
632b4c85ddcSPeter Lieven     }
6333e230dd2SCorentin Chary }
6343e230dd2SCorentin Chary 
635bea60dd7SPeter Lieven static void vnc_dpy_update(DisplayChangeListener *dcl,
636bea60dd7SPeter Lieven                            int x, int y, int w, int h)
637bea60dd7SPeter Lieven {
638bea60dd7SPeter Lieven     VncDisplay *vd = container_of(dcl, VncDisplay, dcl);
639bea60dd7SPeter Lieven     struct VncSurface *s = &vd->guest;
640bea60dd7SPeter Lieven     int width = pixman_image_get_width(vd->server);
641bea60dd7SPeter Lieven     int height = pixman_image_get_height(vd->server);
642bea60dd7SPeter Lieven 
643bea60dd7SPeter Lieven     vnc_set_area_dirty(s->dirty, width, height, x, y, w, h);
644bea60dd7SPeter Lieven }
645bea60dd7SPeter Lieven 
6463e230dd2SCorentin Chary void vnc_framebuffer_update(VncState *vs, int x, int y, int w, int h,
6473e230dd2SCorentin Chary                             int32_t encoding)
6483e230dd2SCorentin Chary {
6493e230dd2SCorentin Chary     vnc_write_u16(vs, x);
6503e230dd2SCorentin Chary     vnc_write_u16(vs, y);
6513e230dd2SCorentin Chary     vnc_write_u16(vs, w);
6523e230dd2SCorentin Chary     vnc_write_u16(vs, h);
6533e230dd2SCorentin Chary 
6543e230dd2SCorentin Chary     vnc_write_s32(vs, encoding);
6553e230dd2SCorentin Chary }
6563e230dd2SCorentin Chary 
6573e230dd2SCorentin Chary void buffer_reserve(Buffer *buffer, size_t len)
6583e230dd2SCorentin Chary {
6593e230dd2SCorentin Chary     if ((buffer->capacity - buffer->offset) < len) {
6603e230dd2SCorentin Chary         buffer->capacity += (len + 1024);
6617267c094SAnthony Liguori         buffer->buffer = g_realloc(buffer->buffer, buffer->capacity);
6623e230dd2SCorentin Chary     }
6633e230dd2SCorentin Chary }
6643e230dd2SCorentin Chary 
66571a8cdecSBlue Swirl static int buffer_empty(Buffer *buffer)
6663e230dd2SCorentin Chary {
6673e230dd2SCorentin Chary     return buffer->offset == 0;
6683e230dd2SCorentin Chary }
6693e230dd2SCorentin Chary 
6707536ee4bSTim Hardeck uint8_t *buffer_end(Buffer *buffer)
6713e230dd2SCorentin Chary {
6723e230dd2SCorentin Chary     return buffer->buffer + buffer->offset;
6733e230dd2SCorentin Chary }
6743e230dd2SCorentin Chary 
6753e230dd2SCorentin Chary void buffer_reset(Buffer *buffer)
6763e230dd2SCorentin Chary {
6773e230dd2SCorentin Chary         buffer->offset = 0;
6783e230dd2SCorentin Chary }
6793e230dd2SCorentin Chary 
6803e230dd2SCorentin Chary void buffer_free(Buffer *buffer)
6813e230dd2SCorentin Chary {
6827267c094SAnthony Liguori     g_free(buffer->buffer);
6833e230dd2SCorentin Chary     buffer->offset = 0;
6843e230dd2SCorentin Chary     buffer->capacity = 0;
6853e230dd2SCorentin Chary     buffer->buffer = NULL;
6863e230dd2SCorentin Chary }
6873e230dd2SCorentin Chary 
6883e230dd2SCorentin Chary void buffer_append(Buffer *buffer, const void *data, size_t len)
6893e230dd2SCorentin Chary {
6903e230dd2SCorentin Chary     memcpy(buffer->buffer + buffer->offset, data, len);
6913e230dd2SCorentin Chary     buffer->offset += len;
6923e230dd2SCorentin Chary }
6933e230dd2SCorentin Chary 
69432ed2680STim Hardeck void buffer_advance(Buffer *buf, size_t len)
69532ed2680STim Hardeck {
69632ed2680STim Hardeck     memmove(buf->buffer, buf->buffer + len,
69732ed2680STim Hardeck             (buf->offset - len));
69832ed2680STim Hardeck     buf->offset -= len;
69932ed2680STim Hardeck }
70032ed2680STim Hardeck 
7013e230dd2SCorentin Chary static void vnc_desktop_resize(VncState *vs)
7023e230dd2SCorentin Chary {
7033e230dd2SCorentin Chary     if (vs->csock == -1 || !vnc_has_feature(vs, VNC_FEATURE_RESIZE)) {
7043e230dd2SCorentin Chary         return;
7053e230dd2SCorentin Chary     }
706bea60dd7SPeter Lieven     if (vs->client_width == pixman_image_get_width(vs->vd->server) &&
707bea60dd7SPeter Lieven         vs->client_height == pixman_image_get_height(vs->vd->server)) {
7083e230dd2SCorentin Chary         return;
7093e230dd2SCorentin Chary     }
710bea60dd7SPeter Lieven     vs->client_width = pixman_image_get_width(vs->vd->server);
711bea60dd7SPeter Lieven     vs->client_height = pixman_image_get_height(vs->vd->server);
712bd023f95SCorentin Chary     vnc_lock_output(vs);
7133e230dd2SCorentin Chary     vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
7143e230dd2SCorentin Chary     vnc_write_u8(vs, 0);
7153e230dd2SCorentin Chary     vnc_write_u16(vs, 1); /* number of rects */
7163e230dd2SCorentin Chary     vnc_framebuffer_update(vs, 0, 0, vs->client_width, vs->client_height,
7173e230dd2SCorentin Chary                            VNC_ENCODING_DESKTOPRESIZE);
718bd023f95SCorentin Chary     vnc_unlock_output(vs);
7193e230dd2SCorentin Chary     vnc_flush(vs);
7203e230dd2SCorentin Chary }
7213e230dd2SCorentin Chary 
722bd023f95SCorentin Chary static void vnc_abort_display_jobs(VncDisplay *vd)
723bd023f95SCorentin Chary {
724bd023f95SCorentin Chary     VncState *vs;
725bd023f95SCorentin Chary 
726bd023f95SCorentin Chary     QTAILQ_FOREACH(vs, &vd->clients, next) {
727bd023f95SCorentin Chary         vnc_lock_output(vs);
728bd023f95SCorentin Chary         vs->abort = true;
729bd023f95SCorentin Chary         vnc_unlock_output(vs);
730bd023f95SCorentin Chary     }
731bd023f95SCorentin Chary     QTAILQ_FOREACH(vs, &vd->clients, next) {
732bd023f95SCorentin Chary         vnc_jobs_join(vs);
733bd023f95SCorentin Chary     }
734bd023f95SCorentin Chary     QTAILQ_FOREACH(vs, &vd->clients, next) {
735bd023f95SCorentin Chary         vnc_lock_output(vs);
736bd023f95SCorentin Chary         vs->abort = false;
737bd023f95SCorentin Chary         vnc_unlock_output(vs);
738bd023f95SCorentin Chary     }
739bd023f95SCorentin Chary }
740bd023f95SCorentin Chary 
7419f64916dSGerd Hoffmann int vnc_server_fb_stride(VncDisplay *vd)
7429f64916dSGerd Hoffmann {
7439f64916dSGerd Hoffmann     return pixman_image_get_stride(vd->server);
7449f64916dSGerd Hoffmann }
7459f64916dSGerd Hoffmann 
7469f64916dSGerd Hoffmann void *vnc_server_fb_ptr(VncDisplay *vd, int x, int y)
7479f64916dSGerd Hoffmann {
7489f64916dSGerd Hoffmann     uint8_t *ptr;
7499f64916dSGerd Hoffmann 
7509f64916dSGerd Hoffmann     ptr  = (uint8_t *)pixman_image_get_data(vd->server);
7519f64916dSGerd Hoffmann     ptr += y * vnc_server_fb_stride(vd);
7529f64916dSGerd Hoffmann     ptr += x * VNC_SERVER_FB_BYTES;
7539f64916dSGerd Hoffmann     return ptr;
7549f64916dSGerd Hoffmann }
7559f64916dSGerd Hoffmann 
756c12aeb86SGerd Hoffmann static void vnc_dpy_switch(DisplayChangeListener *dcl,
757c12aeb86SGerd Hoffmann                            DisplaySurface *surface)
7583e230dd2SCorentin Chary {
75921ef45d7SGerd Hoffmann     VncDisplay *vd = container_of(dcl, VncDisplay, dcl);
7603e230dd2SCorentin Chary     VncState *vs;
761bea60dd7SPeter Lieven     int width, height;
7623e230dd2SCorentin Chary 
763bd023f95SCorentin Chary     vnc_abort_display_jobs(vd);
764bd023f95SCorentin Chary 
7653e230dd2SCorentin Chary     /* server surface */
7669f64916dSGerd Hoffmann     qemu_pixman_image_unref(vd->server);
767d39fa6d8SGerd Hoffmann     vd->ds = surface;
768bea60dd7SPeter Lieven     width = MIN(VNC_MAX_WIDTH, ROUND_UP(surface_width(vd->ds),
769bea60dd7SPeter Lieven                                         VNC_DIRTY_PIXELS_PER_BIT));
770bea60dd7SPeter Lieven     height = MIN(VNC_MAX_HEIGHT, surface_height(vd->ds));
7719f64916dSGerd Hoffmann     vd->server = pixman_image_create_bits(VNC_SERVER_FB_FORMAT,
772bea60dd7SPeter Lieven                                           width, height, NULL, 0);
7733e230dd2SCorentin Chary 
7743e230dd2SCorentin Chary     /* guest surface */
7759f64916dSGerd Hoffmann #if 0 /* FIXME */
7763e230dd2SCorentin Chary     if (ds_get_bytes_per_pixel(ds) != vd->guest.ds->pf.bytes_per_pixel)
7773e230dd2SCorentin Chary         console_color_init(ds);
7789f64916dSGerd Hoffmann #endif
7799f64916dSGerd Hoffmann     qemu_pixman_image_unref(vd->guest.fb);
780d39fa6d8SGerd Hoffmann     vd->guest.fb = pixman_image_ref(surface->image);
781d39fa6d8SGerd Hoffmann     vd->guest.format = surface->format;
782bea60dd7SPeter Lieven     memset(vd->guest.dirty, 0x00, sizeof(vd->guest.dirty));
783bea60dd7SPeter Lieven     vnc_set_area_dirty(vd->guest.dirty, width, height, 0, 0,
784bea60dd7SPeter Lieven                        width, height);
7853e230dd2SCorentin Chary 
7863e230dd2SCorentin Chary     QTAILQ_FOREACH(vs, &vd->clients, next) {
7873e230dd2SCorentin Chary         vnc_colordepth(vs);
7883e230dd2SCorentin Chary         vnc_desktop_resize(vs);
7893e230dd2SCorentin Chary         if (vs->vd->cursor) {
7903e230dd2SCorentin Chary             vnc_cursor_define(vs);
7913e230dd2SCorentin Chary         }
792bea60dd7SPeter Lieven         memset(vs->dirty, 0x00, sizeof(vs->dirty));
793bea60dd7SPeter Lieven         vnc_set_area_dirty(vs->dirty, width, height, 0, 0,
794bea60dd7SPeter Lieven                            width, height);
7953e230dd2SCorentin Chary     }
7963e230dd2SCorentin Chary }
7973e230dd2SCorentin Chary 
7983e230dd2SCorentin Chary /* fastest code */
7999f64916dSGerd Hoffmann static void vnc_write_pixels_copy(VncState *vs,
8003e230dd2SCorentin Chary                                   void *pixels, int size)
8013e230dd2SCorentin Chary {
8023e230dd2SCorentin Chary     vnc_write(vs, pixels, size);
8033e230dd2SCorentin Chary }
8043e230dd2SCorentin Chary 
8053e230dd2SCorentin Chary /* slowest but generic code. */
8063e230dd2SCorentin Chary void vnc_convert_pixel(VncState *vs, uint8_t *buf, uint32_t v)
8073e230dd2SCorentin Chary {
8083e230dd2SCorentin Chary     uint8_t r, g, b;
8093e230dd2SCorentin Chary 
8109f64916dSGerd Hoffmann #if VNC_SERVER_FB_FORMAT == PIXMAN_FORMAT(32, PIXMAN_TYPE_ARGB, 0, 8, 8, 8)
8119f64916dSGerd Hoffmann     r = (((v & 0x00ff0000) >> 16) << vs->client_pf.rbits) >> 8;
8129f64916dSGerd Hoffmann     g = (((v & 0x0000ff00) >>  8) << vs->client_pf.gbits) >> 8;
8139f64916dSGerd Hoffmann     b = (((v & 0x000000ff) >>  0) << vs->client_pf.bbits) >> 8;
8149f64916dSGerd Hoffmann #else
8159f64916dSGerd Hoffmann # error need some bits here if you change VNC_SERVER_FB_FORMAT
8169f64916dSGerd Hoffmann #endif
8179f64916dSGerd Hoffmann     v = (r << vs->client_pf.rshift) |
8189f64916dSGerd Hoffmann         (g << vs->client_pf.gshift) |
8199f64916dSGerd Hoffmann         (b << vs->client_pf.bshift);
8209f64916dSGerd Hoffmann     switch (vs->client_pf.bytes_per_pixel) {
8213e230dd2SCorentin Chary     case 1:
8223e230dd2SCorentin Chary         buf[0] = v;
8233e230dd2SCorentin Chary         break;
8243e230dd2SCorentin Chary     case 2:
8259f64916dSGerd Hoffmann         if (vs->client_be) {
8263e230dd2SCorentin Chary             buf[0] = v >> 8;
8273e230dd2SCorentin Chary             buf[1] = v;
8283e230dd2SCorentin Chary         } else {
8293e230dd2SCorentin Chary             buf[1] = v >> 8;
8303e230dd2SCorentin Chary             buf[0] = v;
8313e230dd2SCorentin Chary         }
8323e230dd2SCorentin Chary         break;
8333e230dd2SCorentin Chary     default:
8343e230dd2SCorentin Chary     case 4:
8359f64916dSGerd Hoffmann         if (vs->client_be) {
8363e230dd2SCorentin Chary             buf[0] = v >> 24;
8373e230dd2SCorentin Chary             buf[1] = v >> 16;
8383e230dd2SCorentin Chary             buf[2] = v >> 8;
8393e230dd2SCorentin Chary             buf[3] = v;
8403e230dd2SCorentin Chary         } else {
8413e230dd2SCorentin Chary             buf[3] = v >> 24;
8423e230dd2SCorentin Chary             buf[2] = v >> 16;
8433e230dd2SCorentin Chary             buf[1] = v >> 8;
8443e230dd2SCorentin Chary             buf[0] = v;
8453e230dd2SCorentin Chary         }
8463e230dd2SCorentin Chary         break;
8473e230dd2SCorentin Chary     }
8483e230dd2SCorentin Chary }
8493e230dd2SCorentin Chary 
8509f64916dSGerd Hoffmann static void vnc_write_pixels_generic(VncState *vs,
8513e230dd2SCorentin Chary                                      void *pixels1, int size)
8523e230dd2SCorentin Chary {
8533e230dd2SCorentin Chary     uint8_t buf[4];
8543e230dd2SCorentin Chary 
8559f64916dSGerd Hoffmann     if (VNC_SERVER_FB_BYTES == 4) {
8563e230dd2SCorentin Chary         uint32_t *pixels = pixels1;
8573e230dd2SCorentin Chary         int n, i;
8583e230dd2SCorentin Chary         n = size >> 2;
8593e230dd2SCorentin Chary         for (i = 0; i < n; i++) {
8603e230dd2SCorentin Chary             vnc_convert_pixel(vs, buf, pixels[i]);
8619f64916dSGerd Hoffmann             vnc_write(vs, buf, vs->client_pf.bytes_per_pixel);
8623e230dd2SCorentin Chary         }
8633e230dd2SCorentin Chary     }
8643e230dd2SCorentin Chary }
8653e230dd2SCorentin Chary 
8663e230dd2SCorentin Chary int vnc_raw_send_framebuffer_update(VncState *vs, int x, int y, int w, int h)
8673e230dd2SCorentin Chary {
8683e230dd2SCorentin Chary     int i;
8693e230dd2SCorentin Chary     uint8_t *row;
8703e230dd2SCorentin Chary     VncDisplay *vd = vs->vd;
8713e230dd2SCorentin Chary 
8729f64916dSGerd Hoffmann     row = vnc_server_fb_ptr(vd, x, y);
8733e230dd2SCorentin Chary     for (i = 0; i < h; i++) {
8749f64916dSGerd Hoffmann         vs->write_pixels(vs, row, w * VNC_SERVER_FB_BYTES);
8759f64916dSGerd Hoffmann         row += vnc_server_fb_stride(vd);
8763e230dd2SCorentin Chary     }
8773e230dd2SCorentin Chary     return 1;
8783e230dd2SCorentin Chary }
8793e230dd2SCorentin Chary 
880bd023f95SCorentin Chary int vnc_send_framebuffer_update(VncState *vs, int x, int y, int w, int h)
8813e230dd2SCorentin Chary {
8823e230dd2SCorentin Chary     int n = 0;
8833e230dd2SCorentin Chary 
8843e230dd2SCorentin Chary     switch(vs->vnc_encoding) {
8853e230dd2SCorentin Chary         case VNC_ENCODING_ZLIB:
8863e230dd2SCorentin Chary             n = vnc_zlib_send_framebuffer_update(vs, x, y, w, h);
8873e230dd2SCorentin Chary             break;
8883e230dd2SCorentin Chary         case VNC_ENCODING_HEXTILE:
8893e230dd2SCorentin Chary             vnc_framebuffer_update(vs, x, y, w, h, VNC_ENCODING_HEXTILE);
8903e230dd2SCorentin Chary             n = vnc_hextile_send_framebuffer_update(vs, x, y, w, h);
8913e230dd2SCorentin Chary             break;
8923e230dd2SCorentin Chary         case VNC_ENCODING_TIGHT:
8933e230dd2SCorentin Chary             n = vnc_tight_send_framebuffer_update(vs, x, y, w, h);
8943e230dd2SCorentin Chary             break;
895efe556adSCorentin Chary         case VNC_ENCODING_TIGHT_PNG:
896efe556adSCorentin Chary             n = vnc_tight_png_send_framebuffer_update(vs, x, y, w, h);
897efe556adSCorentin Chary             break;
898148954faSCorentin Chary         case VNC_ENCODING_ZRLE:
899148954faSCorentin Chary             n = vnc_zrle_send_framebuffer_update(vs, x, y, w, h);
900148954faSCorentin Chary             break;
901148954faSCorentin Chary         case VNC_ENCODING_ZYWRLE:
902148954faSCorentin Chary             n = vnc_zywrle_send_framebuffer_update(vs, x, y, w, h);
903148954faSCorentin Chary             break;
9043e230dd2SCorentin Chary         default:
9053e230dd2SCorentin Chary             vnc_framebuffer_update(vs, x, y, w, h, VNC_ENCODING_RAW);
9063e230dd2SCorentin Chary             n = vnc_raw_send_framebuffer_update(vs, x, y, w, h);
9073e230dd2SCorentin Chary             break;
9083e230dd2SCorentin Chary     }
9093e230dd2SCorentin Chary     return n;
9103e230dd2SCorentin Chary }
9113e230dd2SCorentin Chary 
9123e230dd2SCorentin Chary static void vnc_copy(VncState *vs, int src_x, int src_y, int dst_x, int dst_y, int w, int h)
9133e230dd2SCorentin Chary {
9143e230dd2SCorentin Chary     /* send bitblit op to the vnc client */
915bd023f95SCorentin Chary     vnc_lock_output(vs);
9163e230dd2SCorentin Chary     vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
9173e230dd2SCorentin Chary     vnc_write_u8(vs, 0);
9183e230dd2SCorentin Chary     vnc_write_u16(vs, 1); /* number of rects */
9193e230dd2SCorentin Chary     vnc_framebuffer_update(vs, dst_x, dst_y, w, h, VNC_ENCODING_COPYRECT);
9203e230dd2SCorentin Chary     vnc_write_u16(vs, src_x);
9213e230dd2SCorentin Chary     vnc_write_u16(vs, src_y);
922bd023f95SCorentin Chary     vnc_unlock_output(vs);
9233e230dd2SCorentin Chary     vnc_flush(vs);
9243e230dd2SCorentin Chary }
9253e230dd2SCorentin Chary 
9267c20b4a3SGerd Hoffmann static void vnc_dpy_copy(DisplayChangeListener *dcl,
9277c20b4a3SGerd Hoffmann                          int src_x, int src_y,
9287c20b4a3SGerd Hoffmann                          int dst_x, int dst_y, int w, int h)
9293e230dd2SCorentin Chary {
93021ef45d7SGerd Hoffmann     VncDisplay *vd = container_of(dcl, VncDisplay, dcl);
9313e230dd2SCorentin Chary     VncState *vs, *vn;
9323e230dd2SCorentin Chary     uint8_t *src_row;
9333e230dd2SCorentin Chary     uint8_t *dst_row;
9349f64916dSGerd Hoffmann     int i, x, y, pitch, inc, w_lim, s;
9353e230dd2SCorentin Chary     int cmp_bytes;
9363e230dd2SCorentin Chary 
9373e230dd2SCorentin Chary     vnc_refresh_server_surface(vd);
9383e230dd2SCorentin Chary     QTAILQ_FOREACH_SAFE(vs, &vd->clients, next, vn) {
9393e230dd2SCorentin Chary         if (vnc_has_feature(vs, VNC_FEATURE_COPYRECT)) {
9403e230dd2SCorentin Chary             vs->force_update = 1;
94138ee14f4SGerd Hoffmann             vnc_update_client(vs, 1, true);
9423e230dd2SCorentin Chary             /* vs might be free()ed here */
9433e230dd2SCorentin Chary         }
9443e230dd2SCorentin Chary     }
9453e230dd2SCorentin Chary 
9463e230dd2SCorentin Chary     /* do bitblit op on the local surface too */
9479f64916dSGerd Hoffmann     pitch = vnc_server_fb_stride(vd);
9489f64916dSGerd Hoffmann     src_row = vnc_server_fb_ptr(vd, src_x, src_y);
9499f64916dSGerd Hoffmann     dst_row = vnc_server_fb_ptr(vd, dst_x, dst_y);
9503e230dd2SCorentin Chary     y = dst_y;
9513e230dd2SCorentin Chary     inc = 1;
9523e230dd2SCorentin Chary     if (dst_y > src_y) {
9533e230dd2SCorentin Chary         /* copy backwards */
9543e230dd2SCorentin Chary         src_row += pitch * (h-1);
9553e230dd2SCorentin Chary         dst_row += pitch * (h-1);
9563e230dd2SCorentin Chary         pitch = -pitch;
9573e230dd2SCorentin Chary         y = dst_y + h - 1;
9583e230dd2SCorentin Chary         inc = -1;
9593e230dd2SCorentin Chary     }
960b4c85ddcSPeter Lieven     w_lim = w - (VNC_DIRTY_PIXELS_PER_BIT - (dst_x % VNC_DIRTY_PIXELS_PER_BIT));
961b4c85ddcSPeter Lieven     if (w_lim < 0) {
9623e230dd2SCorentin Chary         w_lim = w;
963b4c85ddcSPeter Lieven     } else {
964b4c85ddcSPeter Lieven         w_lim = w - (w_lim % VNC_DIRTY_PIXELS_PER_BIT);
965b4c85ddcSPeter Lieven     }
9663e230dd2SCorentin Chary     for (i = 0; i < h; i++) {
9673e230dd2SCorentin Chary         for (x = 0; x <= w_lim;
9683e230dd2SCorentin Chary                 x += s, src_row += cmp_bytes, dst_row += cmp_bytes) {
9693e230dd2SCorentin Chary             if (x == w_lim) {
9703e230dd2SCorentin Chary                 if ((s = w - w_lim) == 0)
9713e230dd2SCorentin Chary                     break;
9723e230dd2SCorentin Chary             } else if (!x) {
973b4c85ddcSPeter Lieven                 s = (VNC_DIRTY_PIXELS_PER_BIT -
974b4c85ddcSPeter Lieven                     (dst_x % VNC_DIRTY_PIXELS_PER_BIT));
9753e230dd2SCorentin Chary                 s = MIN(s, w_lim);
9763e230dd2SCorentin Chary             } else {
977b4c85ddcSPeter Lieven                 s = VNC_DIRTY_PIXELS_PER_BIT;
9783e230dd2SCorentin Chary             }
9799f64916dSGerd Hoffmann             cmp_bytes = s * VNC_SERVER_FB_BYTES;
9803e230dd2SCorentin Chary             if (memcmp(src_row, dst_row, cmp_bytes) == 0)
9813e230dd2SCorentin Chary                 continue;
9823e230dd2SCorentin Chary             memmove(dst_row, src_row, cmp_bytes);
9833e230dd2SCorentin Chary             QTAILQ_FOREACH(vs, &vd->clients, next) {
9843e230dd2SCorentin Chary                 if (!vnc_has_feature(vs, VNC_FEATURE_COPYRECT)) {
985b4c85ddcSPeter Lieven                     set_bit(((x + dst_x) / VNC_DIRTY_PIXELS_PER_BIT),
986b4c85ddcSPeter Lieven                             vs->dirty[y]);
9873e230dd2SCorentin Chary                 }
9883e230dd2SCorentin Chary             }
9893e230dd2SCorentin Chary         }
9909f64916dSGerd Hoffmann         src_row += pitch - w * VNC_SERVER_FB_BYTES;
9919f64916dSGerd Hoffmann         dst_row += pitch - w * VNC_SERVER_FB_BYTES;
9923e230dd2SCorentin Chary         y += inc;
9933e230dd2SCorentin Chary     }
9943e230dd2SCorentin Chary 
9953e230dd2SCorentin Chary     QTAILQ_FOREACH(vs, &vd->clients, next) {
9963e230dd2SCorentin Chary         if (vnc_has_feature(vs, VNC_FEATURE_COPYRECT)) {
9973e230dd2SCorentin Chary             vnc_copy(vs, src_x, src_y, dst_x, dst_y, w, h);
9983e230dd2SCorentin Chary         }
9993e230dd2SCorentin Chary     }
10003e230dd2SCorentin Chary }
10013e230dd2SCorentin Chary 
10027c20b4a3SGerd Hoffmann static void vnc_mouse_set(DisplayChangeListener *dcl,
10037c20b4a3SGerd Hoffmann                           int x, int y, int visible)
10043e230dd2SCorentin Chary {
10053e230dd2SCorentin Chary     /* can we ask the client(s) to move the pointer ??? */
10063e230dd2SCorentin Chary }
10073e230dd2SCorentin Chary 
10083e230dd2SCorentin Chary static int vnc_cursor_define(VncState *vs)
10093e230dd2SCorentin Chary {
10103e230dd2SCorentin Chary     QEMUCursor *c = vs->vd->cursor;
10113e230dd2SCorentin Chary     int isize;
10123e230dd2SCorentin Chary 
10133e230dd2SCorentin Chary     if (vnc_has_feature(vs, VNC_FEATURE_RICH_CURSOR)) {
1014d01f9595SCorentin Chary         vnc_lock_output(vs);
10153e230dd2SCorentin Chary         vnc_write_u8(vs,  VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
10163e230dd2SCorentin Chary         vnc_write_u8(vs,  0);  /*  padding     */
10173e230dd2SCorentin Chary         vnc_write_u16(vs, 1);  /*  # of rects  */
10183e230dd2SCorentin Chary         vnc_framebuffer_update(vs, c->hot_x, c->hot_y, c->width, c->height,
10193e230dd2SCorentin Chary                                VNC_ENCODING_RICH_CURSOR);
10209f64916dSGerd Hoffmann         isize = c->width * c->height * vs->client_pf.bytes_per_pixel;
10219f64916dSGerd Hoffmann         vnc_write_pixels_generic(vs, c->data, isize);
10223e230dd2SCorentin Chary         vnc_write(vs, vs->vd->cursor_mask, vs->vd->cursor_msize);
1023d01f9595SCorentin Chary         vnc_unlock_output(vs);
10243e230dd2SCorentin Chary         return 0;
10253e230dd2SCorentin Chary     }
10263e230dd2SCorentin Chary     return -1;
10273e230dd2SCorentin Chary }
10283e230dd2SCorentin Chary 
10297c20b4a3SGerd Hoffmann static void vnc_dpy_cursor_define(DisplayChangeListener *dcl,
10307c20b4a3SGerd Hoffmann                                   QEMUCursor *c)
10313e230dd2SCorentin Chary {
1032d616ccc5SGerd Hoffmann     VncDisplay *vd = container_of(dcl, VncDisplay, dcl);
10333e230dd2SCorentin Chary     VncState *vs;
10343e230dd2SCorentin Chary 
10353e230dd2SCorentin Chary     cursor_put(vd->cursor);
10367267c094SAnthony Liguori     g_free(vd->cursor_mask);
10373e230dd2SCorentin Chary 
10383e230dd2SCorentin Chary     vd->cursor = c;
10393e230dd2SCorentin Chary     cursor_get(vd->cursor);
10403e230dd2SCorentin Chary     vd->cursor_msize = cursor_get_mono_bpl(c) * c->height;
10417267c094SAnthony Liguori     vd->cursor_mask = g_malloc0(vd->cursor_msize);
10423e230dd2SCorentin Chary     cursor_get_mono_mask(c, 0, vd->cursor_mask);
10433e230dd2SCorentin Chary 
10443e230dd2SCorentin Chary     QTAILQ_FOREACH(vs, &vd->clients, next) {
10453e230dd2SCorentin Chary         vnc_cursor_define(vs);
10463e230dd2SCorentin Chary     }
10473e230dd2SCorentin Chary }
10483e230dd2SCorentin Chary 
10493e230dd2SCorentin Chary static int find_and_clear_dirty_height(struct VncState *vs,
10506c71a539SCorentin Chary                                        int y, int last_x, int x, int height)
10513e230dd2SCorentin Chary {
10523e230dd2SCorentin Chary     int h;
10533e230dd2SCorentin Chary 
10546c71a539SCorentin Chary     for (h = 1; h < (height - y); h++) {
1055bc2429b9SCorentin Chary         if (!test_bit(last_x, vs->dirty[y + h])) {
10563e230dd2SCorentin Chary             break;
1057bc2429b9SCorentin Chary         }
1058863d7c91SPeter Lieven         bitmap_clear(vs->dirty[y + h], last_x, x - last_x);
10593e230dd2SCorentin Chary     }
10603e230dd2SCorentin Chary 
10613e230dd2SCorentin Chary     return h;
10623e230dd2SCorentin Chary }
10633e230dd2SCorentin Chary 
106438ee14f4SGerd Hoffmann static int vnc_update_client(VncState *vs, int has_dirty, bool sync)
10653e230dd2SCorentin Chary {
106663658280SGerd Hoffmann     vs->has_dirty += has_dirty;
10673e230dd2SCorentin Chary     if (vs->need_update && vs->csock != -1) {
10683e230dd2SCorentin Chary         VncDisplay *vd = vs->vd;
1069bd023f95SCorentin Chary         VncJob *job;
10703e230dd2SCorentin Chary         int y;
10712f487a3dSPeter Lieven         int height, width;
1072bd023f95SCorentin Chary         int n = 0;
1073bd023f95SCorentin Chary 
10743e230dd2SCorentin Chary         if (vs->output.offset && !vs->audio_cap && !vs->force_update)
10753e230dd2SCorentin Chary             /* kernel send buffers are full -> drop frames to throttle */
10763e230dd2SCorentin Chary             return 0;
10773e230dd2SCorentin Chary 
107863658280SGerd Hoffmann         if (!vs->has_dirty && !vs->audio_cap && !vs->force_update)
10793e230dd2SCorentin Chary             return 0;
10803e230dd2SCorentin Chary 
10813e230dd2SCorentin Chary         /*
10823e230dd2SCorentin Chary          * Send screen updates to the vnc client using the server
10833e230dd2SCorentin Chary          * surface and server dirty map.  guest surface updates
10843e230dd2SCorentin Chary          * happening in parallel don't disturb us, the next pass will
10853e230dd2SCorentin Chary          * send them to the client.
10863e230dd2SCorentin Chary          */
1087bd023f95SCorentin Chary         job = vnc_job_new(vs);
10883e230dd2SCorentin Chary 
1089bea60dd7SPeter Lieven         height = pixman_image_get_height(vd->server);
1090bea60dd7SPeter Lieven         width = pixman_image_get_width(vd->server);
10913e230dd2SCorentin Chary 
109212b316d4SPeter Lieven         y = 0;
109312b316d4SPeter Lieven         for (;;) {
109412b316d4SPeter Lieven             int x, h;
109512b316d4SPeter Lieven             unsigned long x2;
109612b316d4SPeter Lieven             unsigned long offset = find_next_bit((unsigned long *) &vs->dirty,
109712b316d4SPeter Lieven                                                  height * VNC_DIRTY_BPL(vs),
109812b316d4SPeter Lieven                                                  y * VNC_DIRTY_BPL(vs));
109912b316d4SPeter Lieven             if (offset == height * VNC_DIRTY_BPL(vs)) {
110012b316d4SPeter Lieven                 /* no more dirty bits */
110112b316d4SPeter Lieven                 break;
11023e230dd2SCorentin Chary             }
110312b316d4SPeter Lieven             y = offset / VNC_DIRTY_BPL(vs);
110412b316d4SPeter Lieven             x = offset % VNC_DIRTY_BPL(vs);
110512b316d4SPeter Lieven             x2 = find_next_zero_bit((unsigned long *) &vs->dirty[y],
110612b316d4SPeter Lieven                                     VNC_DIRTY_BPL(vs), x);
110712b316d4SPeter Lieven             bitmap_clear(vs->dirty[y], x, x2 - x);
110812b316d4SPeter Lieven             h = find_and_clear_dirty_height(vs, y, x, x2, height);
11092f487a3dSPeter Lieven             x2 = MIN(x2, width / VNC_DIRTY_PIXELS_PER_BIT);
11102f487a3dSPeter Lieven             if (x2 > x) {
111112b316d4SPeter Lieven                 n += vnc_job_add_rect(job, x * VNC_DIRTY_PIXELS_PER_BIT, y,
111212b316d4SPeter Lieven                                       (x2 - x) * VNC_DIRTY_PIXELS_PER_BIT, h);
11133e230dd2SCorentin Chary             }
11140e7d6f60SPeter Lieven             if (!x && x2 == width / VNC_DIRTY_PIXELS_PER_BIT) {
11150e7d6f60SPeter Lieven                 y += h;
11160e7d6f60SPeter Lieven                 if (y == height) {
11170e7d6f60SPeter Lieven                     break;
11180e7d6f60SPeter Lieven                 }
11190e7d6f60SPeter Lieven             }
11202f487a3dSPeter Lieven         }
1121bd023f95SCorentin Chary 
1122bd023f95SCorentin Chary         vnc_job_push(job);
1123eb214ff8SGerd Hoffmann         if (sync) {
1124eb214ff8SGerd Hoffmann             vnc_jobs_join(vs);
1125eb214ff8SGerd Hoffmann         }
11263e230dd2SCorentin Chary         vs->force_update = 0;
112763658280SGerd Hoffmann         vs->has_dirty = 0;
1128bd023f95SCorentin Chary         return n;
11293e230dd2SCorentin Chary     }
11303e230dd2SCorentin Chary 
113138ee14f4SGerd Hoffmann     if (vs->csock == -1) {
11323e230dd2SCorentin Chary         vnc_disconnect_finish(vs);
113338ee14f4SGerd Hoffmann     } else if (sync) {
113438ee14f4SGerd Hoffmann         vnc_jobs_join(vs);
113538ee14f4SGerd Hoffmann     }
11363e230dd2SCorentin Chary 
11373e230dd2SCorentin Chary     return 0;
11383e230dd2SCorentin Chary }
11393e230dd2SCorentin Chary 
11403e230dd2SCorentin Chary /* audio */
11413e230dd2SCorentin Chary static void audio_capture_notify(void *opaque, audcnotification_e cmd)
11423e230dd2SCorentin Chary {
11433e230dd2SCorentin Chary     VncState *vs = opaque;
11443e230dd2SCorentin Chary 
11453e230dd2SCorentin Chary     switch (cmd) {
11463e230dd2SCorentin Chary     case AUD_CNOTIFY_DISABLE:
1147bd023f95SCorentin Chary         vnc_lock_output(vs);
11483e230dd2SCorentin Chary         vnc_write_u8(vs, VNC_MSG_SERVER_QEMU);
11493e230dd2SCorentin Chary         vnc_write_u8(vs, VNC_MSG_SERVER_QEMU_AUDIO);
11503e230dd2SCorentin Chary         vnc_write_u16(vs, VNC_MSG_SERVER_QEMU_AUDIO_END);
1151bd023f95SCorentin Chary         vnc_unlock_output(vs);
11523e230dd2SCorentin Chary         vnc_flush(vs);
11533e230dd2SCorentin Chary         break;
11543e230dd2SCorentin Chary 
11553e230dd2SCorentin Chary     case AUD_CNOTIFY_ENABLE:
1156bd023f95SCorentin Chary         vnc_lock_output(vs);
11573e230dd2SCorentin Chary         vnc_write_u8(vs, VNC_MSG_SERVER_QEMU);
11583e230dd2SCorentin Chary         vnc_write_u8(vs, VNC_MSG_SERVER_QEMU_AUDIO);
11593e230dd2SCorentin Chary         vnc_write_u16(vs, VNC_MSG_SERVER_QEMU_AUDIO_BEGIN);
1160bd023f95SCorentin Chary         vnc_unlock_output(vs);
11613e230dd2SCorentin Chary         vnc_flush(vs);
11623e230dd2SCorentin Chary         break;
11633e230dd2SCorentin Chary     }
11643e230dd2SCorentin Chary }
11653e230dd2SCorentin Chary 
11663e230dd2SCorentin Chary static void audio_capture_destroy(void *opaque)
11673e230dd2SCorentin Chary {
11683e230dd2SCorentin Chary }
11693e230dd2SCorentin Chary 
11703e230dd2SCorentin Chary static void audio_capture(void *opaque, void *buf, int size)
11713e230dd2SCorentin Chary {
11723e230dd2SCorentin Chary     VncState *vs = opaque;
11733e230dd2SCorentin Chary 
1174bd023f95SCorentin Chary     vnc_lock_output(vs);
11753e230dd2SCorentin Chary     vnc_write_u8(vs, VNC_MSG_SERVER_QEMU);
11763e230dd2SCorentin Chary     vnc_write_u8(vs, VNC_MSG_SERVER_QEMU_AUDIO);
11773e230dd2SCorentin Chary     vnc_write_u16(vs, VNC_MSG_SERVER_QEMU_AUDIO_DATA);
11783e230dd2SCorentin Chary     vnc_write_u32(vs, size);
11793e230dd2SCorentin Chary     vnc_write(vs, buf, size);
1180bd023f95SCorentin Chary     vnc_unlock_output(vs);
11813e230dd2SCorentin Chary     vnc_flush(vs);
11823e230dd2SCorentin Chary }
11833e230dd2SCorentin Chary 
11843e230dd2SCorentin Chary static void audio_add(VncState *vs)
11853e230dd2SCorentin Chary {
11863e230dd2SCorentin Chary     struct audio_capture_ops ops;
11873e230dd2SCorentin Chary 
11883e230dd2SCorentin Chary     if (vs->audio_cap) {
1189027a79c3SCole Robinson         error_report("audio already running");
11903e230dd2SCorentin Chary         return;
11913e230dd2SCorentin Chary     }
11923e230dd2SCorentin Chary 
11933e230dd2SCorentin Chary     ops.notify = audio_capture_notify;
11943e230dd2SCorentin Chary     ops.destroy = audio_capture_destroy;
11953e230dd2SCorentin Chary     ops.capture = audio_capture;
11963e230dd2SCorentin Chary 
11973e230dd2SCorentin Chary     vs->audio_cap = AUD_add_capture(&vs->as, &ops, vs);
11983e230dd2SCorentin Chary     if (!vs->audio_cap) {
1199027a79c3SCole Robinson         error_report("Failed to add audio capture");
12003e230dd2SCorentin Chary     }
12013e230dd2SCorentin Chary }
12023e230dd2SCorentin Chary 
12033e230dd2SCorentin Chary static void audio_del(VncState *vs)
12043e230dd2SCorentin Chary {
12053e230dd2SCorentin Chary     if (vs->audio_cap) {
12063e230dd2SCorentin Chary         AUD_del_capture(vs->audio_cap, vs);
12073e230dd2SCorentin Chary         vs->audio_cap = NULL;
12083e230dd2SCorentin Chary     }
12093e230dd2SCorentin Chary }
12103e230dd2SCorentin Chary 
12113e230dd2SCorentin Chary static void vnc_disconnect_start(VncState *vs)
12123e230dd2SCorentin Chary {
12133e230dd2SCorentin Chary     if (vs->csock == -1)
12143e230dd2SCorentin Chary         return;
12158cf36489SGerd Hoffmann     vnc_set_share_mode(vs, VNC_SHARE_MODE_DISCONNECTED);
12163e230dd2SCorentin Chary     qemu_set_fd_handler2(vs->csock, NULL, NULL, NULL, NULL);
12173e230dd2SCorentin Chary     closesocket(vs->csock);
12183e230dd2SCorentin Chary     vs->csock = -1;
12193e230dd2SCorentin Chary }
12203e230dd2SCorentin Chary 
12217536ee4bSTim Hardeck void vnc_disconnect_finish(VncState *vs)
12223e230dd2SCorentin Chary {
12237d964c9dSCorentin Chary     int i;
12247d964c9dSCorentin Chary 
1225bd023f95SCorentin Chary     vnc_jobs_join(vs); /* Wait encoding jobs */
1226bd023f95SCorentin Chary 
1227bd023f95SCorentin Chary     vnc_lock_output(vs);
1228fb6ba0d5SWenchao Xia     vnc_qmp_event(vs, QAPI_EVENT_VNC_DISCONNECTED);
12293e230dd2SCorentin Chary 
12303e230dd2SCorentin Chary     buffer_free(&vs->input);
12313e230dd2SCorentin Chary     buffer_free(&vs->output);
12327536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
12337536ee4bSTim Hardeck     buffer_free(&vs->ws_input);
12347536ee4bSTim Hardeck     buffer_free(&vs->ws_output);
12357536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
12363e230dd2SCorentin Chary 
1237fb6ba0d5SWenchao Xia     qapi_free_VncClientInfo(vs->info);
12383e230dd2SCorentin Chary 
12393e230dd2SCorentin Chary     vnc_zlib_clear(vs);
12403e230dd2SCorentin Chary     vnc_tight_clear(vs);
1241148954faSCorentin Chary     vnc_zrle_clear(vs);
12423e230dd2SCorentin Chary 
12433e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
12443e230dd2SCorentin Chary     vnc_tls_client_cleanup(vs);
12453e230dd2SCorentin Chary #endif /* CONFIG_VNC_TLS */
12463e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
12473e230dd2SCorentin Chary     vnc_sasl_client_cleanup(vs);
12483e230dd2SCorentin Chary #endif /* CONFIG_VNC_SASL */
12493e230dd2SCorentin Chary     audio_del(vs);
12507bc9318bSGerd Hoffmann     vnc_release_modifiers(vs);
12513e230dd2SCorentin Chary 
12526fd8e79aSTim Hardeck     if (vs->initialized) {
12533e230dd2SCorentin Chary         QTAILQ_REMOVE(&vs->vd->clients, vs, next);
12546fd8e79aSTim Hardeck         qemu_remove_mouse_mode_change_notifier(&vs->mouse_mode_notifier);
12556fd8e79aSTim Hardeck     }
12563e230dd2SCorentin Chary 
12573e230dd2SCorentin Chary     if (vs->vd->lock_key_sync)
12583e230dd2SCorentin Chary         qemu_remove_led_event_handler(vs->led);
1259bd023f95SCorentin Chary     vnc_unlock_output(vs);
1260bd023f95SCorentin Chary 
1261bd023f95SCorentin Chary     qemu_mutex_destroy(&vs->output_mutex);
12626fd8e79aSTim Hardeck     if (vs->bh != NULL) {
1263175b2a6eSCorentin Chary         qemu_bh_delete(vs->bh);
12646fd8e79aSTim Hardeck     }
1265175b2a6eSCorentin Chary     buffer_free(&vs->jobs_buffer);
1266175b2a6eSCorentin Chary 
12677d964c9dSCorentin Chary     for (i = 0; i < VNC_STAT_ROWS; ++i) {
12687267c094SAnthony Liguori         g_free(vs->lossy_rect[i]);
12697d964c9dSCorentin Chary     }
12707267c094SAnthony Liguori     g_free(vs->lossy_rect);
12717267c094SAnthony Liguori     g_free(vs);
12723e230dd2SCorentin Chary }
12733e230dd2SCorentin Chary 
12743e230dd2SCorentin Chary int vnc_client_io_error(VncState *vs, int ret, int last_errno)
12753e230dd2SCorentin Chary {
12763e230dd2SCorentin Chary     if (ret == 0 || ret == -1) {
12773e230dd2SCorentin Chary         if (ret == -1) {
12783e230dd2SCorentin Chary             switch (last_errno) {
12793e230dd2SCorentin Chary                 case EINTR:
12803e230dd2SCorentin Chary                 case EAGAIN:
12813e230dd2SCorentin Chary #ifdef _WIN32
12823e230dd2SCorentin Chary                 case WSAEWOULDBLOCK:
12833e230dd2SCorentin Chary #endif
12843e230dd2SCorentin Chary                     return 0;
12853e230dd2SCorentin Chary                 default:
12863e230dd2SCorentin Chary                     break;
12873e230dd2SCorentin Chary             }
12883e230dd2SCorentin Chary         }
12893e230dd2SCorentin Chary 
12903e230dd2SCorentin Chary         VNC_DEBUG("Closing down client sock: ret %d, errno %d\n",
12913e230dd2SCorentin Chary                   ret, ret < 0 ? last_errno : 0);
12923e230dd2SCorentin Chary         vnc_disconnect_start(vs);
12933e230dd2SCorentin Chary 
12943e230dd2SCorentin Chary         return 0;
12953e230dd2SCorentin Chary     }
12963e230dd2SCorentin Chary     return ret;
12973e230dd2SCorentin Chary }
12983e230dd2SCorentin Chary 
12993e230dd2SCorentin Chary 
13003e230dd2SCorentin Chary void vnc_client_error(VncState *vs)
13013e230dd2SCorentin Chary {
13023e230dd2SCorentin Chary     VNC_DEBUG("Closing down client sock: protocol error\n");
13033e230dd2SCorentin Chary     vnc_disconnect_start(vs);
13043e230dd2SCorentin Chary }
13053e230dd2SCorentin Chary 
13060057a0d5STim Hardeck #ifdef CONFIG_VNC_TLS
13070057a0d5STim Hardeck static long vnc_client_write_tls(gnutls_session_t *session,
13080057a0d5STim Hardeck                                  const uint8_t *data,
13090057a0d5STim Hardeck                                  size_t datalen)
13100057a0d5STim Hardeck {
13110057a0d5STim Hardeck     long ret = gnutls_write(*session, data, datalen);
13120057a0d5STim Hardeck     if (ret < 0) {
13130057a0d5STim Hardeck         if (ret == GNUTLS_E_AGAIN) {
13140057a0d5STim Hardeck             errno = EAGAIN;
13150057a0d5STim Hardeck         } else {
13160057a0d5STim Hardeck             errno = EIO;
13170057a0d5STim Hardeck         }
13180057a0d5STim Hardeck         ret = -1;
13190057a0d5STim Hardeck     }
13200057a0d5STim Hardeck     return ret;
13210057a0d5STim Hardeck }
13220057a0d5STim Hardeck #endif /* CONFIG_VNC_TLS */
13233e230dd2SCorentin Chary 
13243e230dd2SCorentin Chary /*
13253e230dd2SCorentin Chary  * Called to write a chunk of data to the client socket. The data may
13263e230dd2SCorentin Chary  * be the raw data, or may have already been encoded by SASL.
13273e230dd2SCorentin Chary  * The data will be written either straight onto the socket, or
13283e230dd2SCorentin Chary  * written via the GNUTLS wrappers, if TLS/SSL encryption is enabled
13293e230dd2SCorentin Chary  *
13303e230dd2SCorentin Chary  * NB, it is theoretically possible to have 2 layers of encryption,
13313e230dd2SCorentin Chary  * both SASL, and this TLS layer. It is highly unlikely in practice
13323e230dd2SCorentin Chary  * though, since SASL encryption will typically be a no-op if TLS
13333e230dd2SCorentin Chary  * is active
13343e230dd2SCorentin Chary  *
13353e230dd2SCorentin Chary  * Returns the number of bytes written, which may be less than
13363e230dd2SCorentin Chary  * the requested 'datalen' if the socket would block. Returns
13373e230dd2SCorentin Chary  * -1 on error, and disconnects the client socket.
13383e230dd2SCorentin Chary  */
13393e230dd2SCorentin Chary long vnc_client_write_buf(VncState *vs, const uint8_t *data, size_t datalen)
13403e230dd2SCorentin Chary {
13413e230dd2SCorentin Chary     long ret;
13423e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
13433e230dd2SCorentin Chary     if (vs->tls.session) {
13440057a0d5STim Hardeck         ret = vnc_client_write_tls(&vs->tls.session, data, datalen);
13450057a0d5STim Hardeck     } else {
13460057a0d5STim Hardeck #ifdef CONFIG_VNC_WS
13470057a0d5STim Hardeck         if (vs->ws_tls.session) {
13480057a0d5STim Hardeck             ret = vnc_client_write_tls(&vs->ws_tls.session, data, datalen);
13493e230dd2SCorentin Chary         } else
13500057a0d5STim Hardeck #endif /* CONFIG_VNC_WS */
13513e230dd2SCorentin Chary #endif /* CONFIG_VNC_TLS */
13520057a0d5STim Hardeck         {
13533e230dd2SCorentin Chary             ret = send(vs->csock, (const void *)data, datalen, 0);
13540057a0d5STim Hardeck         }
13550057a0d5STim Hardeck #ifdef CONFIG_VNC_TLS
13560057a0d5STim Hardeck     }
13570057a0d5STim Hardeck #endif /* CONFIG_VNC_TLS */
13583e230dd2SCorentin Chary     VNC_DEBUG("Wrote wire %p %zd -> %ld\n", data, datalen, ret);
13593e230dd2SCorentin Chary     return vnc_client_io_error(vs, ret, socket_error());
13603e230dd2SCorentin Chary }
13613e230dd2SCorentin Chary 
13623e230dd2SCorentin Chary 
13633e230dd2SCorentin Chary /*
13643e230dd2SCorentin Chary  * Called to write buffered data to the client socket, when not
13653e230dd2SCorentin Chary  * using any SASL SSF encryption layers. Will write as much data
13663e230dd2SCorentin Chary  * as possible without blocking. If all buffered data is written,
13673e230dd2SCorentin Chary  * will switch the FD poll() handler back to read monitoring.
13683e230dd2SCorentin Chary  *
13693e230dd2SCorentin Chary  * Returns the number of bytes written, which may be less than
13703e230dd2SCorentin Chary  * the buffered output data if the socket would block. Returns
13713e230dd2SCorentin Chary  * -1 on error, and disconnects the client socket.
13723e230dd2SCorentin Chary  */
13733e230dd2SCorentin Chary static long vnc_client_write_plain(VncState *vs)
13743e230dd2SCorentin Chary {
13753e230dd2SCorentin Chary     long ret;
13763e230dd2SCorentin Chary 
13773e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
13783e230dd2SCorentin Chary     VNC_DEBUG("Write Plain: Pending output %p size %zd offset %zd. Wait SSF %d\n",
13793e230dd2SCorentin Chary               vs->output.buffer, vs->output.capacity, vs->output.offset,
13803e230dd2SCorentin Chary               vs->sasl.waitWriteSSF);
13813e230dd2SCorentin Chary 
13823e230dd2SCorentin Chary     if (vs->sasl.conn &&
13833e230dd2SCorentin Chary         vs->sasl.runSSF &&
13843e230dd2SCorentin Chary         vs->sasl.waitWriteSSF) {
13853e230dd2SCorentin Chary         ret = vnc_client_write_buf(vs, vs->output.buffer, vs->sasl.waitWriteSSF);
13863e230dd2SCorentin Chary         if (ret)
13873e230dd2SCorentin Chary             vs->sasl.waitWriteSSF -= ret;
13883e230dd2SCorentin Chary     } else
13893e230dd2SCorentin Chary #endif /* CONFIG_VNC_SASL */
13903e230dd2SCorentin Chary         ret = vnc_client_write_buf(vs, vs->output.buffer, vs->output.offset);
13913e230dd2SCorentin Chary     if (!ret)
13923e230dd2SCorentin Chary         return 0;
13933e230dd2SCorentin Chary 
139432ed2680STim Hardeck     buffer_advance(&vs->output, ret);
13953e230dd2SCorentin Chary 
13963e230dd2SCorentin Chary     if (vs->output.offset == 0) {
13973e230dd2SCorentin Chary         qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, vs);
13983e230dd2SCorentin Chary     }
13993e230dd2SCorentin Chary 
14003e230dd2SCorentin Chary     return ret;
14013e230dd2SCorentin Chary }
14023e230dd2SCorentin Chary 
14033e230dd2SCorentin Chary 
14043e230dd2SCorentin Chary /*
14053e230dd2SCorentin Chary  * First function called whenever there is data to be written to
14063e230dd2SCorentin Chary  * the client socket. Will delegate actual work according to whether
14073e230dd2SCorentin Chary  * SASL SSF layers are enabled (thus requiring encryption calls)
14083e230dd2SCorentin Chary  */
1409bd023f95SCorentin Chary static void vnc_client_write_locked(void *opaque)
14103e230dd2SCorentin Chary {
14113e230dd2SCorentin Chary     VncState *vs = opaque;
14123e230dd2SCorentin Chary 
14133e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
14143e230dd2SCorentin Chary     if (vs->sasl.conn &&
14153e230dd2SCorentin Chary         vs->sasl.runSSF &&
14163e230dd2SCorentin Chary         !vs->sasl.waitWriteSSF) {
14173e230dd2SCorentin Chary         vnc_client_write_sasl(vs);
14183e230dd2SCorentin Chary     } else
14193e230dd2SCorentin Chary #endif /* CONFIG_VNC_SASL */
14207536ee4bSTim Hardeck     {
14217536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
14227536ee4bSTim Hardeck         if (vs->encode_ws) {
14237536ee4bSTim Hardeck             vnc_client_write_ws(vs);
14247536ee4bSTim Hardeck         } else
14257536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
14267536ee4bSTim Hardeck         {
14273e230dd2SCorentin Chary             vnc_client_write_plain(vs);
14283e230dd2SCorentin Chary         }
14297536ee4bSTim Hardeck     }
14307536ee4bSTim Hardeck }
14313e230dd2SCorentin Chary 
1432bd023f95SCorentin Chary void vnc_client_write(void *opaque)
1433bd023f95SCorentin Chary {
1434bd023f95SCorentin Chary     VncState *vs = opaque;
1435bd023f95SCorentin Chary 
1436bd023f95SCorentin Chary     vnc_lock_output(vs);
14377536ee4bSTim Hardeck     if (vs->output.offset
14387536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
14397536ee4bSTim Hardeck             || vs->ws_output.offset
14407536ee4bSTim Hardeck #endif
14417536ee4bSTim Hardeck             ) {
1442bd023f95SCorentin Chary         vnc_client_write_locked(opaque);
1443ac71103dSYoshiaki Tamura     } else if (vs->csock != -1) {
1444bd023f95SCorentin Chary         qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, vs);
1445bd023f95SCorentin Chary     }
1446bd023f95SCorentin Chary     vnc_unlock_output(vs);
1447bd023f95SCorentin Chary }
1448bd023f95SCorentin Chary 
14493e230dd2SCorentin Chary void vnc_read_when(VncState *vs, VncReadEvent *func, size_t expecting)
14503e230dd2SCorentin Chary {
14513e230dd2SCorentin Chary     vs->read_handler = func;
14523e230dd2SCorentin Chary     vs->read_handler_expect = expecting;
14533e230dd2SCorentin Chary }
14543e230dd2SCorentin Chary 
14550057a0d5STim Hardeck #ifdef CONFIG_VNC_TLS
14560057a0d5STim Hardeck static long vnc_client_read_tls(gnutls_session_t *session, uint8_t *data,
14570057a0d5STim Hardeck                                 size_t datalen)
14580057a0d5STim Hardeck {
14590057a0d5STim Hardeck     long ret = gnutls_read(*session, data, datalen);
14600057a0d5STim Hardeck     if (ret < 0) {
14610057a0d5STim Hardeck         if (ret == GNUTLS_E_AGAIN) {
14620057a0d5STim Hardeck             errno = EAGAIN;
14630057a0d5STim Hardeck         } else {
14640057a0d5STim Hardeck             errno = EIO;
14650057a0d5STim Hardeck         }
14660057a0d5STim Hardeck         ret = -1;
14670057a0d5STim Hardeck     }
14680057a0d5STim Hardeck     return ret;
14690057a0d5STim Hardeck }
14700057a0d5STim Hardeck #endif /* CONFIG_VNC_TLS */
14713e230dd2SCorentin Chary 
14723e230dd2SCorentin Chary /*
14733e230dd2SCorentin Chary  * Called to read a chunk of data from the client socket. The data may
14743e230dd2SCorentin Chary  * be the raw data, or may need to be further decoded by SASL.
14753e230dd2SCorentin Chary  * The data will be read either straight from to the socket, or
14763e230dd2SCorentin Chary  * read via the GNUTLS wrappers, if TLS/SSL encryption is enabled
14773e230dd2SCorentin Chary  *
14783e230dd2SCorentin Chary  * NB, it is theoretically possible to have 2 layers of encryption,
14793e230dd2SCorentin Chary  * both SASL, and this TLS layer. It is highly unlikely in practice
14803e230dd2SCorentin Chary  * though, since SASL encryption will typically be a no-op if TLS
14813e230dd2SCorentin Chary  * is active
14823e230dd2SCorentin Chary  *
14833e230dd2SCorentin Chary  * Returns the number of bytes read, which may be less than
14843e230dd2SCorentin Chary  * the requested 'datalen' if the socket would block. Returns
14853e230dd2SCorentin Chary  * -1 on error, and disconnects the client socket.
14863e230dd2SCorentin Chary  */
14873e230dd2SCorentin Chary long vnc_client_read_buf(VncState *vs, uint8_t *data, size_t datalen)
14883e230dd2SCorentin Chary {
14893e230dd2SCorentin Chary     long ret;
14903e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
14913e230dd2SCorentin Chary     if (vs->tls.session) {
14920057a0d5STim Hardeck         ret = vnc_client_read_tls(&vs->tls.session, data, datalen);
14930057a0d5STim Hardeck     } else {
14940057a0d5STim Hardeck #ifdef CONFIG_VNC_WS
14950057a0d5STim Hardeck         if (vs->ws_tls.session) {
14960057a0d5STim Hardeck             ret = vnc_client_read_tls(&vs->ws_tls.session, data, datalen);
14973e230dd2SCorentin Chary         } else
14980057a0d5STim Hardeck #endif /* CONFIG_VNC_WS */
14993e230dd2SCorentin Chary #endif /* CONFIG_VNC_TLS */
15000057a0d5STim Hardeck         {
150100aa0040SBlue Swirl             ret = qemu_recv(vs->csock, data, datalen, 0);
15020057a0d5STim Hardeck         }
15030057a0d5STim Hardeck #ifdef CONFIG_VNC_TLS
15040057a0d5STim Hardeck     }
15050057a0d5STim Hardeck #endif /* CONFIG_VNC_TLS */
15063e230dd2SCorentin Chary     VNC_DEBUG("Read wire %p %zd -> %ld\n", data, datalen, ret);
15073e230dd2SCorentin Chary     return vnc_client_io_error(vs, ret, socket_error());
15083e230dd2SCorentin Chary }
15093e230dd2SCorentin Chary 
15103e230dd2SCorentin Chary 
15113e230dd2SCorentin Chary /*
15123e230dd2SCorentin Chary  * Called to read data from the client socket to the input buffer,
15133e230dd2SCorentin Chary  * when not using any SASL SSF encryption layers. Will read as much
15143e230dd2SCorentin Chary  * data as possible without blocking.
15153e230dd2SCorentin Chary  *
15163e230dd2SCorentin Chary  * Returns the number of bytes read. Returns -1 on error, and
15173e230dd2SCorentin Chary  * disconnects the client socket.
15183e230dd2SCorentin Chary  */
15193e230dd2SCorentin Chary static long vnc_client_read_plain(VncState *vs)
15203e230dd2SCorentin Chary {
15213e230dd2SCorentin Chary     int ret;
15223e230dd2SCorentin Chary     VNC_DEBUG("Read plain %p size %zd offset %zd\n",
15233e230dd2SCorentin Chary               vs->input.buffer, vs->input.capacity, vs->input.offset);
15243e230dd2SCorentin Chary     buffer_reserve(&vs->input, 4096);
15253e230dd2SCorentin Chary     ret = vnc_client_read_buf(vs, buffer_end(&vs->input), 4096);
15263e230dd2SCorentin Chary     if (!ret)
15273e230dd2SCorentin Chary         return 0;
15283e230dd2SCorentin Chary     vs->input.offset += ret;
15293e230dd2SCorentin Chary     return ret;
15303e230dd2SCorentin Chary }
15313e230dd2SCorentin Chary 
1532175b2a6eSCorentin Chary static void vnc_jobs_bh(void *opaque)
1533175b2a6eSCorentin Chary {
1534175b2a6eSCorentin Chary     VncState *vs = opaque;
1535175b2a6eSCorentin Chary 
1536175b2a6eSCorentin Chary     vnc_jobs_consume_buffer(vs);
1537175b2a6eSCorentin Chary }
15383e230dd2SCorentin Chary 
15393e230dd2SCorentin Chary /*
15403e230dd2SCorentin Chary  * First function called whenever there is more data to be read from
15413e230dd2SCorentin Chary  * the client socket. Will delegate actual work according to whether
15423e230dd2SCorentin Chary  * SASL SSF layers are enabled (thus requiring decryption calls)
15433e230dd2SCorentin Chary  */
15443e230dd2SCorentin Chary void vnc_client_read(void *opaque)
15453e230dd2SCorentin Chary {
15463e230dd2SCorentin Chary     VncState *vs = opaque;
15473e230dd2SCorentin Chary     long ret;
15483e230dd2SCorentin Chary 
15493e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
15503e230dd2SCorentin Chary     if (vs->sasl.conn && vs->sasl.runSSF)
15513e230dd2SCorentin Chary         ret = vnc_client_read_sasl(vs);
15523e230dd2SCorentin Chary     else
15533e230dd2SCorentin Chary #endif /* CONFIG_VNC_SASL */
15547536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
15557536ee4bSTim Hardeck         if (vs->encode_ws) {
15567536ee4bSTim Hardeck             ret = vnc_client_read_ws(vs);
15577536ee4bSTim Hardeck             if (ret == -1) {
15587536ee4bSTim Hardeck                 vnc_disconnect_start(vs);
15597536ee4bSTim Hardeck                 return;
15607536ee4bSTim Hardeck             } else if (ret == -2) {
15617536ee4bSTim Hardeck                 vnc_client_error(vs);
15627536ee4bSTim Hardeck                 return;
15637536ee4bSTim Hardeck             }
15647536ee4bSTim Hardeck         } else
15657536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
15667536ee4bSTim Hardeck         {
15673e230dd2SCorentin Chary         ret = vnc_client_read_plain(vs);
15687536ee4bSTim Hardeck         }
15693e230dd2SCorentin Chary     if (!ret) {
15703e230dd2SCorentin Chary         if (vs->csock == -1)
15713e230dd2SCorentin Chary             vnc_disconnect_finish(vs);
15723e230dd2SCorentin Chary         return;
15733e230dd2SCorentin Chary     }
15743e230dd2SCorentin Chary 
15753e230dd2SCorentin Chary     while (vs->read_handler && vs->input.offset >= vs->read_handler_expect) {
15763e230dd2SCorentin Chary         size_t len = vs->read_handler_expect;
15773e230dd2SCorentin Chary         int ret;
15783e230dd2SCorentin Chary 
15793e230dd2SCorentin Chary         ret = vs->read_handler(vs, vs->input.buffer, len);
15803e230dd2SCorentin Chary         if (vs->csock == -1) {
15813e230dd2SCorentin Chary             vnc_disconnect_finish(vs);
15823e230dd2SCorentin Chary             return;
15833e230dd2SCorentin Chary         }
15843e230dd2SCorentin Chary 
15853e230dd2SCorentin Chary         if (!ret) {
158632ed2680STim Hardeck             buffer_advance(&vs->input, len);
15873e230dd2SCorentin Chary         } else {
15883e230dd2SCorentin Chary             vs->read_handler_expect = ret;
15893e230dd2SCorentin Chary         }
15903e230dd2SCorentin Chary     }
15913e230dd2SCorentin Chary }
15923e230dd2SCorentin Chary 
15933e230dd2SCorentin Chary void vnc_write(VncState *vs, const void *data, size_t len)
15943e230dd2SCorentin Chary {
15953e230dd2SCorentin Chary     buffer_reserve(&vs->output, len);
15963e230dd2SCorentin Chary 
15973e230dd2SCorentin Chary     if (vs->csock != -1 && buffer_empty(&vs->output)) {
15983e230dd2SCorentin Chary         qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, vnc_client_write, vs);
15993e230dd2SCorentin Chary     }
16003e230dd2SCorentin Chary 
16013e230dd2SCorentin Chary     buffer_append(&vs->output, data, len);
16023e230dd2SCorentin Chary }
16033e230dd2SCorentin Chary 
16043e230dd2SCorentin Chary void vnc_write_s32(VncState *vs, int32_t value)
16053e230dd2SCorentin Chary {
16063e230dd2SCorentin Chary     vnc_write_u32(vs, *(uint32_t *)&value);
16073e230dd2SCorentin Chary }
16083e230dd2SCorentin Chary 
16093e230dd2SCorentin Chary void vnc_write_u32(VncState *vs, uint32_t value)
16103e230dd2SCorentin Chary {
16113e230dd2SCorentin Chary     uint8_t buf[4];
16123e230dd2SCorentin Chary 
16133e230dd2SCorentin Chary     buf[0] = (value >> 24) & 0xFF;
16143e230dd2SCorentin Chary     buf[1] = (value >> 16) & 0xFF;
16153e230dd2SCorentin Chary     buf[2] = (value >>  8) & 0xFF;
16163e230dd2SCorentin Chary     buf[3] = value & 0xFF;
16173e230dd2SCorentin Chary 
16183e230dd2SCorentin Chary     vnc_write(vs, buf, 4);
16193e230dd2SCorentin Chary }
16203e230dd2SCorentin Chary 
16213e230dd2SCorentin Chary void vnc_write_u16(VncState *vs, uint16_t value)
16223e230dd2SCorentin Chary {
16233e230dd2SCorentin Chary     uint8_t buf[2];
16243e230dd2SCorentin Chary 
16253e230dd2SCorentin Chary     buf[0] = (value >> 8) & 0xFF;
16263e230dd2SCorentin Chary     buf[1] = value & 0xFF;
16273e230dd2SCorentin Chary 
16283e230dd2SCorentin Chary     vnc_write(vs, buf, 2);
16293e230dd2SCorentin Chary }
16303e230dd2SCorentin Chary 
16313e230dd2SCorentin Chary void vnc_write_u8(VncState *vs, uint8_t value)
16323e230dd2SCorentin Chary {
16333e230dd2SCorentin Chary     vnc_write(vs, (char *)&value, 1);
16343e230dd2SCorentin Chary }
16353e230dd2SCorentin Chary 
16363e230dd2SCorentin Chary void vnc_flush(VncState *vs)
16373e230dd2SCorentin Chary {
1638bd023f95SCorentin Chary     vnc_lock_output(vs);
16397536ee4bSTim Hardeck     if (vs->csock != -1 && (vs->output.offset
16407536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
16417536ee4bSTim Hardeck                 || vs->ws_output.offset
16427536ee4bSTim Hardeck #endif
16437536ee4bSTim Hardeck                 )) {
1644bd023f95SCorentin Chary         vnc_client_write_locked(vs);
1645bd023f95SCorentin Chary     }
1646bd023f95SCorentin Chary     vnc_unlock_output(vs);
16473e230dd2SCorentin Chary }
16483e230dd2SCorentin Chary 
164971a8cdecSBlue Swirl static uint8_t read_u8(uint8_t *data, size_t offset)
16503e230dd2SCorentin Chary {
16513e230dd2SCorentin Chary     return data[offset];
16523e230dd2SCorentin Chary }
16533e230dd2SCorentin Chary 
165471a8cdecSBlue Swirl static uint16_t read_u16(uint8_t *data, size_t offset)
16553e230dd2SCorentin Chary {
16563e230dd2SCorentin Chary     return ((data[offset] & 0xFF) << 8) | (data[offset + 1] & 0xFF);
16573e230dd2SCorentin Chary }
16583e230dd2SCorentin Chary 
165971a8cdecSBlue Swirl static int32_t read_s32(uint8_t *data, size_t offset)
16603e230dd2SCorentin Chary {
16613e230dd2SCorentin Chary     return (int32_t)((data[offset] << 24) | (data[offset + 1] << 16) |
16623e230dd2SCorentin Chary                      (data[offset + 2] << 8) | data[offset + 3]);
16633e230dd2SCorentin Chary }
16643e230dd2SCorentin Chary 
16653e230dd2SCorentin Chary uint32_t read_u32(uint8_t *data, size_t offset)
16663e230dd2SCorentin Chary {
16673e230dd2SCorentin Chary     return ((data[offset] << 24) | (data[offset + 1] << 16) |
16683e230dd2SCorentin Chary             (data[offset + 2] << 8) | data[offset + 3]);
16693e230dd2SCorentin Chary }
16703e230dd2SCorentin Chary 
16713e230dd2SCorentin Chary static void client_cut_text(VncState *vs, size_t len, uint8_t *text)
16723e230dd2SCorentin Chary {
16733e230dd2SCorentin Chary }
16743e230dd2SCorentin Chary 
16759e8dd451SJan Kiszka static void check_pointer_type_change(Notifier *notifier, void *data)
16763e230dd2SCorentin Chary {
16773e230dd2SCorentin Chary     VncState *vs = container_of(notifier, VncState, mouse_mode_notifier);
167814768ebaSGerd Hoffmann     int absolute = qemu_input_is_absolute();
16793e230dd2SCorentin Chary 
16803e230dd2SCorentin Chary     if (vnc_has_feature(vs, VNC_FEATURE_POINTER_TYPE_CHANGE) && vs->absolute != absolute) {
1681bd023f95SCorentin Chary         vnc_lock_output(vs);
16823e230dd2SCorentin Chary         vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
16833e230dd2SCorentin Chary         vnc_write_u8(vs, 0);
16843e230dd2SCorentin Chary         vnc_write_u16(vs, 1);
16853e230dd2SCorentin Chary         vnc_framebuffer_update(vs, absolute, 0,
1686bea60dd7SPeter Lieven                                pixman_image_get_width(vs->vd->server),
1687bea60dd7SPeter Lieven                                pixman_image_get_height(vs->vd->server),
16883e230dd2SCorentin Chary                                VNC_ENCODING_POINTER_TYPE_CHANGE);
1689bd023f95SCorentin Chary         vnc_unlock_output(vs);
16903e230dd2SCorentin Chary         vnc_flush(vs);
16913e230dd2SCorentin Chary     }
16923e230dd2SCorentin Chary     vs->absolute = absolute;
16933e230dd2SCorentin Chary }
16943e230dd2SCorentin Chary 
16953e230dd2SCorentin Chary static void pointer_event(VncState *vs, int button_mask, int x, int y)
16963e230dd2SCorentin Chary {
169714768ebaSGerd Hoffmann     static uint32_t bmap[INPUT_BUTTON_MAX] = {
169814768ebaSGerd Hoffmann         [INPUT_BUTTON_LEFT]       = 0x01,
169914768ebaSGerd Hoffmann         [INPUT_BUTTON_MIDDLE]     = 0x02,
170014768ebaSGerd Hoffmann         [INPUT_BUTTON_RIGHT]      = 0x04,
170114768ebaSGerd Hoffmann         [INPUT_BUTTON_WHEEL_UP]   = 0x08,
170214768ebaSGerd Hoffmann         [INPUT_BUTTON_WHEEL_DOWN] = 0x10,
170314768ebaSGerd Hoffmann     };
170414768ebaSGerd Hoffmann     QemuConsole *con = vs->vd->dcl.con;
1705bea60dd7SPeter Lieven     int width = pixman_image_get_width(vs->vd->server);
1706bea60dd7SPeter Lieven     int height = pixman_image_get_height(vs->vd->server);
17073e230dd2SCorentin Chary 
170814768ebaSGerd Hoffmann     if (vs->last_bmask != button_mask) {
170914768ebaSGerd Hoffmann         qemu_input_update_buttons(con, bmap, vs->last_bmask, button_mask);
171014768ebaSGerd Hoffmann         vs->last_bmask = button_mask;
171114768ebaSGerd Hoffmann     }
17123e230dd2SCorentin Chary 
17133e230dd2SCorentin Chary     if (vs->absolute) {
171414768ebaSGerd Hoffmann         qemu_input_queue_abs(con, INPUT_AXIS_X, x, width);
171514768ebaSGerd Hoffmann         qemu_input_queue_abs(con, INPUT_AXIS_Y, y, height);
17163e230dd2SCorentin Chary     } else if (vnc_has_feature(vs, VNC_FEATURE_POINTER_TYPE_CHANGE)) {
171714768ebaSGerd Hoffmann         qemu_input_queue_rel(con, INPUT_AXIS_X, x - 0x7FFF);
171814768ebaSGerd Hoffmann         qemu_input_queue_rel(con, INPUT_AXIS_Y, y - 0x7FFF);
17193e230dd2SCorentin Chary     } else {
172014768ebaSGerd Hoffmann         if (vs->last_x != -1) {
172114768ebaSGerd Hoffmann             qemu_input_queue_rel(con, INPUT_AXIS_X, x - vs->last_x);
172214768ebaSGerd Hoffmann             qemu_input_queue_rel(con, INPUT_AXIS_Y, y - vs->last_y);
172314768ebaSGerd Hoffmann         }
17243e230dd2SCorentin Chary         vs->last_x = x;
17253e230dd2SCorentin Chary         vs->last_y = y;
17263e230dd2SCorentin Chary     }
172714768ebaSGerd Hoffmann     qemu_input_event_sync();
17283e230dd2SCorentin Chary }
17293e230dd2SCorentin Chary 
17303e230dd2SCorentin Chary static void reset_keys(VncState *vs)
17313e230dd2SCorentin Chary {
17323e230dd2SCorentin Chary     int i;
17333e230dd2SCorentin Chary     for(i = 0; i < 256; i++) {
17343e230dd2SCorentin Chary         if (vs->modifiers_state[i]) {
17358d447d10SGerd Hoffmann             qemu_input_event_send_key_number(vs->vd->dcl.con, i, false);
17363e230dd2SCorentin Chary             vs->modifiers_state[i] = 0;
17373e230dd2SCorentin Chary         }
17383e230dd2SCorentin Chary     }
17393e230dd2SCorentin Chary }
17403e230dd2SCorentin Chary 
17413e230dd2SCorentin Chary static void press_key(VncState *vs, int keysym)
17423e230dd2SCorentin Chary {
17433e230dd2SCorentin Chary     int keycode = keysym2scancode(vs->vd->kbd_layout, keysym) & SCANCODE_KEYMASK;
17448d447d10SGerd Hoffmann     qemu_input_event_send_key_number(vs->vd->dcl.con, keycode, true);
17452deb4accSGerd Hoffmann     qemu_input_event_send_key_delay(0);
17468d447d10SGerd Hoffmann     qemu_input_event_send_key_number(vs->vd->dcl.con, keycode, false);
17472deb4accSGerd Hoffmann     qemu_input_event_send_key_delay(0);
17483e230dd2SCorentin Chary }
17493e230dd2SCorentin Chary 
1750ab99e5c1SLei Li static int current_led_state(VncState *vs)
1751ab99e5c1SLei Li {
1752ab99e5c1SLei Li     int ledstate = 0;
1753ab99e5c1SLei Li 
1754ab99e5c1SLei Li     if (vs->modifiers_state[0x46]) {
1755ab99e5c1SLei Li         ledstate |= QEMU_SCROLL_LOCK_LED;
1756ab99e5c1SLei Li     }
1757ab99e5c1SLei Li     if (vs->modifiers_state[0x45]) {
1758ab99e5c1SLei Li         ledstate |= QEMU_NUM_LOCK_LED;
1759ab99e5c1SLei Li     }
1760ab99e5c1SLei Li     if (vs->modifiers_state[0x3a]) {
1761ab99e5c1SLei Li         ledstate |= QEMU_CAPS_LOCK_LED;
1762ab99e5c1SLei Li     }
1763ab99e5c1SLei Li 
1764ab99e5c1SLei Li     return ledstate;
1765ab99e5c1SLei Li }
1766ab99e5c1SLei Li 
1767ab99e5c1SLei Li static void vnc_led_state_change(VncState *vs)
1768ab99e5c1SLei Li {
1769ab99e5c1SLei Li     int ledstate = 0;
1770ab99e5c1SLei Li 
1771ab99e5c1SLei Li     if (!vnc_has_feature(vs, VNC_FEATURE_LED_STATE)) {
1772ab99e5c1SLei Li         return;
1773ab99e5c1SLei Li     }
1774ab99e5c1SLei Li 
1775ab99e5c1SLei Li     ledstate = current_led_state(vs);
1776ab99e5c1SLei Li     vnc_lock_output(vs);
1777ab99e5c1SLei Li     vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
1778ab99e5c1SLei Li     vnc_write_u8(vs, 0);
1779ab99e5c1SLei Li     vnc_write_u16(vs, 1);
1780ab99e5c1SLei Li     vnc_framebuffer_update(vs, 0, 0, 1, 1, VNC_ENCODING_LED_STATE);
1781ab99e5c1SLei Li     vnc_write_u8(vs, ledstate);
1782ab99e5c1SLei Li     vnc_unlock_output(vs);
1783ab99e5c1SLei Li     vnc_flush(vs);
1784ab99e5c1SLei Li }
1785ab99e5c1SLei Li 
17863e230dd2SCorentin Chary static void kbd_leds(void *opaque, int ledstate)
17873e230dd2SCorentin Chary {
17883e230dd2SCorentin Chary     VncState *vs = opaque;
178996f3d174SLei Li     int caps, num, scr;
17901483adcfSLei Li     bool has_changed = (ledstate != current_led_state(vs));
17913e230dd2SCorentin Chary 
179240066175SGerd Hoffmann     trace_vnc_key_guest_leds((ledstate & QEMU_CAPS_LOCK_LED),
179340066175SGerd Hoffmann                              (ledstate & QEMU_NUM_LOCK_LED),
179440066175SGerd Hoffmann                              (ledstate & QEMU_SCROLL_LOCK_LED));
179540066175SGerd Hoffmann 
17963e230dd2SCorentin Chary     caps = ledstate & QEMU_CAPS_LOCK_LED ? 1 : 0;
17973e230dd2SCorentin Chary     num  = ledstate & QEMU_NUM_LOCK_LED  ? 1 : 0;
179896f3d174SLei Li     scr  = ledstate & QEMU_SCROLL_LOCK_LED ? 1 : 0;
17993e230dd2SCorentin Chary 
18003e230dd2SCorentin Chary     if (vs->modifiers_state[0x3a] != caps) {
18013e230dd2SCorentin Chary         vs->modifiers_state[0x3a] = caps;
18023e230dd2SCorentin Chary     }
18033e230dd2SCorentin Chary     if (vs->modifiers_state[0x45] != num) {
18043e230dd2SCorentin Chary         vs->modifiers_state[0x45] = num;
18053e230dd2SCorentin Chary     }
180696f3d174SLei Li     if (vs->modifiers_state[0x46] != scr) {
180796f3d174SLei Li         vs->modifiers_state[0x46] = scr;
180896f3d174SLei Li     }
1809ab99e5c1SLei Li 
1810ab99e5c1SLei Li     /* Sending the current led state message to the client */
18111483adcfSLei Li     if (has_changed) {
1812ab99e5c1SLei Li         vnc_led_state_change(vs);
1813ab99e5c1SLei Li     }
18143e230dd2SCorentin Chary }
18153e230dd2SCorentin Chary 
18163e230dd2SCorentin Chary static void do_key_event(VncState *vs, int down, int keycode, int sym)
18173e230dd2SCorentin Chary {
18183e230dd2SCorentin Chary     /* QEMU console switch */
18193e230dd2SCorentin Chary     switch(keycode) {
18203e230dd2SCorentin Chary     case 0x2a:                          /* Left Shift */
18213e230dd2SCorentin Chary     case 0x36:                          /* Right Shift */
18223e230dd2SCorentin Chary     case 0x1d:                          /* Left CTRL */
18233e230dd2SCorentin Chary     case 0x9d:                          /* Right CTRL */
18243e230dd2SCorentin Chary     case 0x38:                          /* Left ALT */
18253e230dd2SCorentin Chary     case 0xb8:                          /* Right ALT */
18263e230dd2SCorentin Chary         if (down)
18273e230dd2SCorentin Chary             vs->modifiers_state[keycode] = 1;
18283e230dd2SCorentin Chary         else
18293e230dd2SCorentin Chary             vs->modifiers_state[keycode] = 0;
18303e230dd2SCorentin Chary         break;
18313e230dd2SCorentin Chary     case 0x02 ... 0x0a: /* '1' to '9' keys */
18321d0d59feSGerd Hoffmann         if (vs->vd->dcl.con == NULL &&
18331d0d59feSGerd Hoffmann             down && vs->modifiers_state[0x1d] && vs->modifiers_state[0x38]) {
18343e230dd2SCorentin Chary             /* Reset the modifiers sent to the current console */
18353e230dd2SCorentin Chary             reset_keys(vs);
18363e230dd2SCorentin Chary             console_select(keycode - 0x02);
18373e230dd2SCorentin Chary             return;
18383e230dd2SCorentin Chary         }
18393e230dd2SCorentin Chary         break;
18403e230dd2SCorentin Chary     case 0x3a:                        /* CapsLock */
18413e230dd2SCorentin Chary     case 0x45:                        /* NumLock */
18423e230dd2SCorentin Chary         if (down)
18433e230dd2SCorentin Chary             vs->modifiers_state[keycode] ^= 1;
18443e230dd2SCorentin Chary         break;
18453e230dd2SCorentin Chary     }
18463e230dd2SCorentin Chary 
1847e7b2aaccSLei Li     /* Turn off the lock state sync logic if the client support the led
1848e7b2aaccSLei Li        state extension.
1849e7b2aaccSLei Li     */
18509892088bSGerd Hoffmann     if (down && vs->vd->lock_key_sync &&
1851e7b2aaccSLei Li         !vnc_has_feature(vs, VNC_FEATURE_LED_STATE) &&
18523e230dd2SCorentin Chary         keycode_is_keypad(vs->vd->kbd_layout, keycode)) {
18533e230dd2SCorentin Chary         /* If the numlock state needs to change then simulate an additional
18543e230dd2SCorentin Chary            keypress before sending this one.  This will happen if the user
18553e230dd2SCorentin Chary            toggles numlock away from the VNC window.
18563e230dd2SCorentin Chary         */
18573e230dd2SCorentin Chary         if (keysym_is_numlock(vs->vd->kbd_layout, sym & 0xFFFF)) {
18583e230dd2SCorentin Chary             if (!vs->modifiers_state[0x45]) {
185940066175SGerd Hoffmann                 trace_vnc_key_sync_numlock(true);
18603e230dd2SCorentin Chary                 vs->modifiers_state[0x45] = 1;
18613e230dd2SCorentin Chary                 press_key(vs, 0xff7f);
18623e230dd2SCorentin Chary             }
18633e230dd2SCorentin Chary         } else {
18643e230dd2SCorentin Chary             if (vs->modifiers_state[0x45]) {
186540066175SGerd Hoffmann                 trace_vnc_key_sync_numlock(false);
18663e230dd2SCorentin Chary                 vs->modifiers_state[0x45] = 0;
18673e230dd2SCorentin Chary                 press_key(vs, 0xff7f);
18683e230dd2SCorentin Chary             }
18693e230dd2SCorentin Chary         }
18703e230dd2SCorentin Chary     }
18713e230dd2SCorentin Chary 
18729892088bSGerd Hoffmann     if (down && vs->vd->lock_key_sync &&
1873e7b2aaccSLei Li         !vnc_has_feature(vs, VNC_FEATURE_LED_STATE) &&
18743e230dd2SCorentin Chary         ((sym >= 'A' && sym <= 'Z') || (sym >= 'a' && sym <= 'z'))) {
18753e230dd2SCorentin Chary         /* If the capslock state needs to change then simulate an additional
18763e230dd2SCorentin Chary            keypress before sending this one.  This will happen if the user
18773e230dd2SCorentin Chary            toggles capslock away from the VNC window.
18783e230dd2SCorentin Chary         */
18793e230dd2SCorentin Chary         int uppercase = !!(sym >= 'A' && sym <= 'Z');
18803e230dd2SCorentin Chary         int shift = !!(vs->modifiers_state[0x2a] | vs->modifiers_state[0x36]);
18813e230dd2SCorentin Chary         int capslock = !!(vs->modifiers_state[0x3a]);
18823e230dd2SCorentin Chary         if (capslock) {
18833e230dd2SCorentin Chary             if (uppercase == shift) {
188440066175SGerd Hoffmann                 trace_vnc_key_sync_capslock(false);
18853e230dd2SCorentin Chary                 vs->modifiers_state[0x3a] = 0;
18863e230dd2SCorentin Chary                 press_key(vs, 0xffe5);
18873e230dd2SCorentin Chary             }
18883e230dd2SCorentin Chary         } else {
18893e230dd2SCorentin Chary             if (uppercase != shift) {
189040066175SGerd Hoffmann                 trace_vnc_key_sync_capslock(true);
18913e230dd2SCorentin Chary                 vs->modifiers_state[0x3a] = 1;
18923e230dd2SCorentin Chary                 press_key(vs, 0xffe5);
18933e230dd2SCorentin Chary             }
18943e230dd2SCorentin Chary         }
18953e230dd2SCorentin Chary     }
18963e230dd2SCorentin Chary 
189781c0d5a6SGerd Hoffmann     if (qemu_console_is_graphic(NULL)) {
18988d447d10SGerd Hoffmann         qemu_input_event_send_key_number(vs->vd->dcl.con, keycode, down);
18993e230dd2SCorentin Chary     } else {
1900e26437c2SGerd Hoffmann         bool numlock = vs->modifiers_state[0x45];
1901e26437c2SGerd Hoffmann         bool control = (vs->modifiers_state[0x1d] ||
1902e26437c2SGerd Hoffmann                         vs->modifiers_state[0x9d]);
19033e230dd2SCorentin Chary         /* QEMU console emulation */
19043e230dd2SCorentin Chary         if (down) {
19053e230dd2SCorentin Chary             switch (keycode) {
19063e230dd2SCorentin Chary             case 0x2a:                          /* Left Shift */
19073e230dd2SCorentin Chary             case 0x36:                          /* Right Shift */
19083e230dd2SCorentin Chary             case 0x1d:                          /* Left CTRL */
19093e230dd2SCorentin Chary             case 0x9d:                          /* Right CTRL */
19103e230dd2SCorentin Chary             case 0x38:                          /* Left ALT */
19113e230dd2SCorentin Chary             case 0xb8:                          /* Right ALT */
19123e230dd2SCorentin Chary                 break;
19133e230dd2SCorentin Chary             case 0xc8:
19143e230dd2SCorentin Chary                 kbd_put_keysym(QEMU_KEY_UP);
19153e230dd2SCorentin Chary                 break;
19163e230dd2SCorentin Chary             case 0xd0:
19173e230dd2SCorentin Chary                 kbd_put_keysym(QEMU_KEY_DOWN);
19183e230dd2SCorentin Chary                 break;
19193e230dd2SCorentin Chary             case 0xcb:
19203e230dd2SCorentin Chary                 kbd_put_keysym(QEMU_KEY_LEFT);
19213e230dd2SCorentin Chary                 break;
19223e230dd2SCorentin Chary             case 0xcd:
19233e230dd2SCorentin Chary                 kbd_put_keysym(QEMU_KEY_RIGHT);
19243e230dd2SCorentin Chary                 break;
19253e230dd2SCorentin Chary             case 0xd3:
19263e230dd2SCorentin Chary                 kbd_put_keysym(QEMU_KEY_DELETE);
19273e230dd2SCorentin Chary                 break;
19283e230dd2SCorentin Chary             case 0xc7:
19293e230dd2SCorentin Chary                 kbd_put_keysym(QEMU_KEY_HOME);
19303e230dd2SCorentin Chary                 break;
19313e230dd2SCorentin Chary             case 0xcf:
19323e230dd2SCorentin Chary                 kbd_put_keysym(QEMU_KEY_END);
19333e230dd2SCorentin Chary                 break;
19343e230dd2SCorentin Chary             case 0xc9:
19353e230dd2SCorentin Chary                 kbd_put_keysym(QEMU_KEY_PAGEUP);
19363e230dd2SCorentin Chary                 break;
19373e230dd2SCorentin Chary             case 0xd1:
19383e230dd2SCorentin Chary                 kbd_put_keysym(QEMU_KEY_PAGEDOWN);
19393e230dd2SCorentin Chary                 break;
19403e230dd2SCorentin Chary 
19413e230dd2SCorentin Chary             case 0x47:
19423e230dd2SCorentin Chary                 kbd_put_keysym(numlock ? '7' : QEMU_KEY_HOME);
19433e230dd2SCorentin Chary                 break;
19443e230dd2SCorentin Chary             case 0x48:
19453e230dd2SCorentin Chary                 kbd_put_keysym(numlock ? '8' : QEMU_KEY_UP);
19463e230dd2SCorentin Chary                 break;
19473e230dd2SCorentin Chary             case 0x49:
19483e230dd2SCorentin Chary                 kbd_put_keysym(numlock ? '9' : QEMU_KEY_PAGEUP);
19493e230dd2SCorentin Chary                 break;
19503e230dd2SCorentin Chary             case 0x4b:
19513e230dd2SCorentin Chary                 kbd_put_keysym(numlock ? '4' : QEMU_KEY_LEFT);
19523e230dd2SCorentin Chary                 break;
19533e230dd2SCorentin Chary             case 0x4c:
19543e230dd2SCorentin Chary                 kbd_put_keysym('5');
19553e230dd2SCorentin Chary                 break;
19563e230dd2SCorentin Chary             case 0x4d:
19573e230dd2SCorentin Chary                 kbd_put_keysym(numlock ? '6' : QEMU_KEY_RIGHT);
19583e230dd2SCorentin Chary                 break;
19593e230dd2SCorentin Chary             case 0x4f:
19603e230dd2SCorentin Chary                 kbd_put_keysym(numlock ? '1' : QEMU_KEY_END);
19613e230dd2SCorentin Chary                 break;
19623e230dd2SCorentin Chary             case 0x50:
19633e230dd2SCorentin Chary                 kbd_put_keysym(numlock ? '2' : QEMU_KEY_DOWN);
19643e230dd2SCorentin Chary                 break;
19653e230dd2SCorentin Chary             case 0x51:
19663e230dd2SCorentin Chary                 kbd_put_keysym(numlock ? '3' : QEMU_KEY_PAGEDOWN);
19673e230dd2SCorentin Chary                 break;
19683e230dd2SCorentin Chary             case 0x52:
19693e230dd2SCorentin Chary                 kbd_put_keysym('0');
19703e230dd2SCorentin Chary                 break;
19713e230dd2SCorentin Chary             case 0x53:
19723e230dd2SCorentin Chary                 kbd_put_keysym(numlock ? '.' : QEMU_KEY_DELETE);
19733e230dd2SCorentin Chary                 break;
19743e230dd2SCorentin Chary 
19753e230dd2SCorentin Chary             case 0xb5:
19763e230dd2SCorentin Chary                 kbd_put_keysym('/');
19773e230dd2SCorentin Chary                 break;
19783e230dd2SCorentin Chary             case 0x37:
19793e230dd2SCorentin Chary                 kbd_put_keysym('*');
19803e230dd2SCorentin Chary                 break;
19813e230dd2SCorentin Chary             case 0x4a:
19823e230dd2SCorentin Chary                 kbd_put_keysym('-');
19833e230dd2SCorentin Chary                 break;
19843e230dd2SCorentin Chary             case 0x4e:
19853e230dd2SCorentin Chary                 kbd_put_keysym('+');
19863e230dd2SCorentin Chary                 break;
19873e230dd2SCorentin Chary             case 0x9c:
19883e230dd2SCorentin Chary                 kbd_put_keysym('\n');
19893e230dd2SCorentin Chary                 break;
19903e230dd2SCorentin Chary 
19913e230dd2SCorentin Chary             default:
1992e26437c2SGerd Hoffmann                 if (control) {
1993e26437c2SGerd Hoffmann                     kbd_put_keysym(sym & 0x1f);
1994e26437c2SGerd Hoffmann                 } else {
19953e230dd2SCorentin Chary                     kbd_put_keysym(sym);
1996e26437c2SGerd Hoffmann                 }
19973e230dd2SCorentin Chary                 break;
19983e230dd2SCorentin Chary             }
19993e230dd2SCorentin Chary         }
20003e230dd2SCorentin Chary     }
20013e230dd2SCorentin Chary }
20023e230dd2SCorentin Chary 
20037bc9318bSGerd Hoffmann static void vnc_release_modifiers(VncState *vs)
20047bc9318bSGerd Hoffmann {
20057bc9318bSGerd Hoffmann     static const int keycodes[] = {
20067bc9318bSGerd Hoffmann         /* shift, control, alt keys, both left & right */
20077bc9318bSGerd Hoffmann         0x2a, 0x36, 0x1d, 0x9d, 0x38, 0xb8,
20087bc9318bSGerd Hoffmann     };
20097bc9318bSGerd Hoffmann     int i, keycode;
20107bc9318bSGerd Hoffmann 
201181c0d5a6SGerd Hoffmann     if (!qemu_console_is_graphic(NULL)) {
20127bc9318bSGerd Hoffmann         return;
20137bc9318bSGerd Hoffmann     }
20147bc9318bSGerd Hoffmann     for (i = 0; i < ARRAY_SIZE(keycodes); i++) {
20157bc9318bSGerd Hoffmann         keycode = keycodes[i];
20167bc9318bSGerd Hoffmann         if (!vs->modifiers_state[keycode]) {
20177bc9318bSGerd Hoffmann             continue;
20187bc9318bSGerd Hoffmann         }
20198d447d10SGerd Hoffmann         qemu_input_event_send_key_number(vs->vd->dcl.con, keycode, false);
20207bc9318bSGerd Hoffmann     }
20217bc9318bSGerd Hoffmann }
20227bc9318bSGerd Hoffmann 
202340066175SGerd Hoffmann static const char *code2name(int keycode)
202440066175SGerd Hoffmann {
202540066175SGerd Hoffmann     return QKeyCode_lookup[qemu_input_key_number_to_qcode(keycode)];
202640066175SGerd Hoffmann }
202740066175SGerd Hoffmann 
20283e230dd2SCorentin Chary static void key_event(VncState *vs, int down, uint32_t sym)
20293e230dd2SCorentin Chary {
20303e230dd2SCorentin Chary     int keycode;
20313e230dd2SCorentin Chary     int lsym = sym;
20323e230dd2SCorentin Chary 
203381c0d5a6SGerd Hoffmann     if (lsym >= 'A' && lsym <= 'Z' && qemu_console_is_graphic(NULL)) {
20343e230dd2SCorentin Chary         lsym = lsym - 'A' + 'a';
20353e230dd2SCorentin Chary     }
20363e230dd2SCorentin Chary 
20373e230dd2SCorentin Chary     keycode = keysym2scancode(vs->vd->kbd_layout, lsym & 0xFFFF) & SCANCODE_KEYMASK;
203840066175SGerd Hoffmann     trace_vnc_key_event_map(down, sym, keycode, code2name(keycode));
20393e230dd2SCorentin Chary     do_key_event(vs, down, keycode, sym);
20403e230dd2SCorentin Chary }
20413e230dd2SCorentin Chary 
20423e230dd2SCorentin Chary static void ext_key_event(VncState *vs, int down,
20433e230dd2SCorentin Chary                           uint32_t sym, uint16_t keycode)
20443e230dd2SCorentin Chary {
20453e230dd2SCorentin Chary     /* if the user specifies a keyboard layout, always use it */
204640066175SGerd Hoffmann     if (keyboard_layout) {
20473e230dd2SCorentin Chary         key_event(vs, down, sym);
204840066175SGerd Hoffmann     } else {
204940066175SGerd Hoffmann         trace_vnc_key_event_ext(down, sym, keycode, code2name(keycode));
20503e230dd2SCorentin Chary         do_key_event(vs, down, keycode, sym);
20513e230dd2SCorentin Chary     }
205240066175SGerd Hoffmann }
20533e230dd2SCorentin Chary 
20543e230dd2SCorentin Chary static void framebuffer_update_request(VncState *vs, int incremental,
2055bea60dd7SPeter Lieven                                        int x, int y, int w, int h)
20563e230dd2SCorentin Chary {
2057bea60dd7SPeter Lieven     int width = pixman_image_get_width(vs->vd->server);
2058bea60dd7SPeter Lieven     int height = pixman_image_get_height(vs->vd->server);
20593e230dd2SCorentin Chary 
20603e230dd2SCorentin Chary     vs->need_update = 1;
2061bea60dd7SPeter Lieven 
2062bea60dd7SPeter Lieven     if (incremental) {
2063bea60dd7SPeter Lieven         return;
20643e230dd2SCorentin Chary     }
2065bea60dd7SPeter Lieven 
206607535a89SStephan Kulow     vs->force_update = 1;
2067bea60dd7SPeter Lieven     vnc_set_area_dirty(vs->dirty, width, height, x, y, w, h);
20683e230dd2SCorentin Chary }
20693e230dd2SCorentin Chary 
20703e230dd2SCorentin Chary static void send_ext_key_event_ack(VncState *vs)
20713e230dd2SCorentin Chary {
2072bd023f95SCorentin Chary     vnc_lock_output(vs);
20733e230dd2SCorentin Chary     vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
20743e230dd2SCorentin Chary     vnc_write_u8(vs, 0);
20753e230dd2SCorentin Chary     vnc_write_u16(vs, 1);
2076d39fa6d8SGerd Hoffmann     vnc_framebuffer_update(vs, 0, 0,
2077bea60dd7SPeter Lieven                            pixman_image_get_width(vs->vd->server),
2078bea60dd7SPeter Lieven                            pixman_image_get_height(vs->vd->server),
20793e230dd2SCorentin Chary                            VNC_ENCODING_EXT_KEY_EVENT);
2080bd023f95SCorentin Chary     vnc_unlock_output(vs);
20813e230dd2SCorentin Chary     vnc_flush(vs);
20823e230dd2SCorentin Chary }
20833e230dd2SCorentin Chary 
20843e230dd2SCorentin Chary static void send_ext_audio_ack(VncState *vs)
20853e230dd2SCorentin Chary {
2086bd023f95SCorentin Chary     vnc_lock_output(vs);
20873e230dd2SCorentin Chary     vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
20883e230dd2SCorentin Chary     vnc_write_u8(vs, 0);
20893e230dd2SCorentin Chary     vnc_write_u16(vs, 1);
2090d39fa6d8SGerd Hoffmann     vnc_framebuffer_update(vs, 0, 0,
2091bea60dd7SPeter Lieven                            pixman_image_get_width(vs->vd->server),
2092bea60dd7SPeter Lieven                            pixman_image_get_height(vs->vd->server),
20933e230dd2SCorentin Chary                            VNC_ENCODING_AUDIO);
2094bd023f95SCorentin Chary     vnc_unlock_output(vs);
20953e230dd2SCorentin Chary     vnc_flush(vs);
20963e230dd2SCorentin Chary }
20973e230dd2SCorentin Chary 
20983e230dd2SCorentin Chary static void set_encodings(VncState *vs, int32_t *encodings, size_t n_encodings)
20993e230dd2SCorentin Chary {
21003e230dd2SCorentin Chary     int i;
21013e230dd2SCorentin Chary     unsigned int enc = 0;
21023e230dd2SCorentin Chary 
21033e230dd2SCorentin Chary     vs->features = 0;
21043e230dd2SCorentin Chary     vs->vnc_encoding = 0;
2105d1af0e05SCorentin Chary     vs->tight.compression = 9;
2106d1af0e05SCorentin Chary     vs->tight.quality = -1; /* Lossless by default */
21073e230dd2SCorentin Chary     vs->absolute = -1;
21083e230dd2SCorentin Chary 
21093e230dd2SCorentin Chary     /*
21103e230dd2SCorentin Chary      * Start from the end because the encodings are sent in order of preference.
2111e5bed759SDong Xu Wang      * This way the preferred encoding (first encoding defined in the array)
21123e230dd2SCorentin Chary      * will be set at the end of the loop.
21133e230dd2SCorentin Chary      */
21143e230dd2SCorentin Chary     for (i = n_encodings - 1; i >= 0; i--) {
21153e230dd2SCorentin Chary         enc = encodings[i];
21163e230dd2SCorentin Chary         switch (enc) {
21173e230dd2SCorentin Chary         case VNC_ENCODING_RAW:
21183e230dd2SCorentin Chary             vs->vnc_encoding = enc;
21193e230dd2SCorentin Chary             break;
21203e230dd2SCorentin Chary         case VNC_ENCODING_COPYRECT:
21213e230dd2SCorentin Chary             vs->features |= VNC_FEATURE_COPYRECT_MASK;
21223e230dd2SCorentin Chary             break;
21233e230dd2SCorentin Chary         case VNC_ENCODING_HEXTILE:
21243e230dd2SCorentin Chary             vs->features |= VNC_FEATURE_HEXTILE_MASK;
21253e230dd2SCorentin Chary             vs->vnc_encoding = enc;
21263e230dd2SCorentin Chary             break;
21273e230dd2SCorentin Chary         case VNC_ENCODING_TIGHT:
21283e230dd2SCorentin Chary             vs->features |= VNC_FEATURE_TIGHT_MASK;
21293e230dd2SCorentin Chary             vs->vnc_encoding = enc;
21303e230dd2SCorentin Chary             break;
2131fe3e7f2dSJoel Martin #ifdef CONFIG_VNC_PNG
2132efe556adSCorentin Chary         case VNC_ENCODING_TIGHT_PNG:
2133efe556adSCorentin Chary             vs->features |= VNC_FEATURE_TIGHT_PNG_MASK;
2134efe556adSCorentin Chary             vs->vnc_encoding = enc;
2135efe556adSCorentin Chary             break;
2136fe3e7f2dSJoel Martin #endif
21373e230dd2SCorentin Chary         case VNC_ENCODING_ZLIB:
21383e230dd2SCorentin Chary             vs->features |= VNC_FEATURE_ZLIB_MASK;
21393e230dd2SCorentin Chary             vs->vnc_encoding = enc;
21403e230dd2SCorentin Chary             break;
2141148954faSCorentin Chary         case VNC_ENCODING_ZRLE:
2142148954faSCorentin Chary             vs->features |= VNC_FEATURE_ZRLE_MASK;
2143148954faSCorentin Chary             vs->vnc_encoding = enc;
2144148954faSCorentin Chary             break;
2145148954faSCorentin Chary         case VNC_ENCODING_ZYWRLE:
2146148954faSCorentin Chary             vs->features |= VNC_FEATURE_ZYWRLE_MASK;
2147148954faSCorentin Chary             vs->vnc_encoding = enc;
2148148954faSCorentin Chary             break;
21493e230dd2SCorentin Chary         case VNC_ENCODING_DESKTOPRESIZE:
21503e230dd2SCorentin Chary             vs->features |= VNC_FEATURE_RESIZE_MASK;
21513e230dd2SCorentin Chary             break;
21523e230dd2SCorentin Chary         case VNC_ENCODING_POINTER_TYPE_CHANGE:
21533e230dd2SCorentin Chary             vs->features |= VNC_FEATURE_POINTER_TYPE_CHANGE_MASK;
21543e230dd2SCorentin Chary             break;
21553e230dd2SCorentin Chary         case VNC_ENCODING_RICH_CURSOR:
21563e230dd2SCorentin Chary             vs->features |= VNC_FEATURE_RICH_CURSOR_MASK;
21573e230dd2SCorentin Chary             break;
21583e230dd2SCorentin Chary         case VNC_ENCODING_EXT_KEY_EVENT:
21593e230dd2SCorentin Chary             send_ext_key_event_ack(vs);
21603e230dd2SCorentin Chary             break;
21613e230dd2SCorentin Chary         case VNC_ENCODING_AUDIO:
21623e230dd2SCorentin Chary             send_ext_audio_ack(vs);
21633e230dd2SCorentin Chary             break;
21643e230dd2SCorentin Chary         case VNC_ENCODING_WMVi:
21653e230dd2SCorentin Chary             vs->features |= VNC_FEATURE_WMVI_MASK;
21663e230dd2SCorentin Chary             break;
2167ab99e5c1SLei Li         case VNC_ENCODING_LED_STATE:
2168ab99e5c1SLei Li             vs->features |= VNC_FEATURE_LED_STATE_MASK;
2169ab99e5c1SLei Li             break;
21703e230dd2SCorentin Chary         case VNC_ENCODING_COMPRESSLEVEL0 ... VNC_ENCODING_COMPRESSLEVEL0 + 9:
2171d1af0e05SCorentin Chary             vs->tight.compression = (enc & 0x0F);
21723e230dd2SCorentin Chary             break;
21733e230dd2SCorentin Chary         case VNC_ENCODING_QUALITYLEVEL0 ... VNC_ENCODING_QUALITYLEVEL0 + 9:
2174b31f519eSCorentin Chary             if (vs->vd->lossy) {
2175d1af0e05SCorentin Chary                 vs->tight.quality = (enc & 0x0F);
2176b31f519eSCorentin Chary             }
21773e230dd2SCorentin Chary             break;
21783e230dd2SCorentin Chary         default:
21793e230dd2SCorentin Chary             VNC_DEBUG("Unknown encoding: %d (0x%.8x): %d\n", i, enc, enc);
21803e230dd2SCorentin Chary             break;
21813e230dd2SCorentin Chary         }
21823e230dd2SCorentin Chary     }
21833e230dd2SCorentin Chary     vnc_desktop_resize(vs);
21849e8dd451SJan Kiszka     check_pointer_type_change(&vs->mouse_mode_notifier, NULL);
2185ab99e5c1SLei Li     vnc_led_state_change(vs);
21863e230dd2SCorentin Chary }
21873e230dd2SCorentin Chary 
21883e230dd2SCorentin Chary static void set_pixel_conversion(VncState *vs)
21893e230dd2SCorentin Chary {
21909f64916dSGerd Hoffmann     pixman_format_code_t fmt = qemu_pixman_get_format(&vs->client_pf);
21919f64916dSGerd Hoffmann 
21929f64916dSGerd Hoffmann     if (fmt == VNC_SERVER_FB_FORMAT) {
21933e230dd2SCorentin Chary         vs->write_pixels = vnc_write_pixels_copy;
21943e230dd2SCorentin Chary         vnc_hextile_set_pixel_conversion(vs, 0);
21953e230dd2SCorentin Chary     } else {
21963e230dd2SCorentin Chary         vs->write_pixels = vnc_write_pixels_generic;
21973e230dd2SCorentin Chary         vnc_hextile_set_pixel_conversion(vs, 1);
21983e230dd2SCorentin Chary     }
21993e230dd2SCorentin Chary }
22003e230dd2SCorentin Chary 
22013e230dd2SCorentin Chary static void set_pixel_format(VncState *vs,
22023e230dd2SCorentin Chary                              int bits_per_pixel, int depth,
22033e230dd2SCorentin Chary                              int big_endian_flag, int true_color_flag,
22043e230dd2SCorentin Chary                              int red_max, int green_max, int blue_max,
22053e230dd2SCorentin Chary                              int red_shift, int green_shift, int blue_shift)
22063e230dd2SCorentin Chary {
22073e230dd2SCorentin Chary     if (!true_color_flag) {
22083e230dd2SCorentin Chary         vnc_client_error(vs);
22093e230dd2SCorentin Chary         return;
22103e230dd2SCorentin Chary     }
22113e230dd2SCorentin Chary 
2212e6908bfeSPetr Matousek     switch (bits_per_pixel) {
2213e6908bfeSPetr Matousek     case 8:
2214e6908bfeSPetr Matousek     case 16:
2215e6908bfeSPetr Matousek     case 32:
2216e6908bfeSPetr Matousek         break;
2217e6908bfeSPetr Matousek     default:
2218e6908bfeSPetr Matousek         vnc_client_error(vs);
2219e6908bfeSPetr Matousek         return;
2220e6908bfeSPetr Matousek     }
2221e6908bfeSPetr Matousek 
22229f64916dSGerd Hoffmann     vs->client_pf.rmax = red_max;
22239f64916dSGerd Hoffmann     vs->client_pf.rbits = hweight_long(red_max);
22249f64916dSGerd Hoffmann     vs->client_pf.rshift = red_shift;
22259f64916dSGerd Hoffmann     vs->client_pf.rmask = red_max << red_shift;
22269f64916dSGerd Hoffmann     vs->client_pf.gmax = green_max;
22279f64916dSGerd Hoffmann     vs->client_pf.gbits = hweight_long(green_max);
22289f64916dSGerd Hoffmann     vs->client_pf.gshift = green_shift;
22299f64916dSGerd Hoffmann     vs->client_pf.gmask = green_max << green_shift;
22309f64916dSGerd Hoffmann     vs->client_pf.bmax = blue_max;
22319f64916dSGerd Hoffmann     vs->client_pf.bbits = hweight_long(blue_max);
22329f64916dSGerd Hoffmann     vs->client_pf.bshift = blue_shift;
22339f64916dSGerd Hoffmann     vs->client_pf.bmask = blue_max << blue_shift;
22349f64916dSGerd Hoffmann     vs->client_pf.bits_per_pixel = bits_per_pixel;
22359f64916dSGerd Hoffmann     vs->client_pf.bytes_per_pixel = bits_per_pixel / 8;
22369f64916dSGerd Hoffmann     vs->client_pf.depth = bits_per_pixel == 32 ? 24 : bits_per_pixel;
22379f64916dSGerd Hoffmann     vs->client_be = big_endian_flag;
22383e230dd2SCorentin Chary 
22393e230dd2SCorentin Chary     set_pixel_conversion(vs);
22403e230dd2SCorentin Chary 
22411d0d59feSGerd Hoffmann     graphic_hw_invalidate(vs->vd->dcl.con);
22421d0d59feSGerd Hoffmann     graphic_hw_update(vs->vd->dcl.con);
22433e230dd2SCorentin Chary }
22443e230dd2SCorentin Chary 
22453e230dd2SCorentin Chary static void pixel_format_message (VncState *vs) {
22463e230dd2SCorentin Chary     char pad[3] = { 0, 0, 0 };
22473e230dd2SCorentin Chary 
22489f64916dSGerd Hoffmann     vs->client_pf = qemu_default_pixelformat(32);
22499f64916dSGerd Hoffmann 
22509f64916dSGerd Hoffmann     vnc_write_u8(vs, vs->client_pf.bits_per_pixel); /* bits-per-pixel */
22519f64916dSGerd Hoffmann     vnc_write_u8(vs, vs->client_pf.depth); /* depth */
22523e230dd2SCorentin Chary 
22533e230dd2SCorentin Chary #ifdef HOST_WORDS_BIGENDIAN
22543e230dd2SCorentin Chary     vnc_write_u8(vs, 1);             /* big-endian-flag */
22553e230dd2SCorentin Chary #else
22563e230dd2SCorentin Chary     vnc_write_u8(vs, 0);             /* big-endian-flag */
22573e230dd2SCorentin Chary #endif
22583e230dd2SCorentin Chary     vnc_write_u8(vs, 1);             /* true-color-flag */
22599f64916dSGerd Hoffmann     vnc_write_u16(vs, vs->client_pf.rmax);     /* red-max */
22609f64916dSGerd Hoffmann     vnc_write_u16(vs, vs->client_pf.gmax);     /* green-max */
22619f64916dSGerd Hoffmann     vnc_write_u16(vs, vs->client_pf.bmax);     /* blue-max */
22629f64916dSGerd Hoffmann     vnc_write_u8(vs, vs->client_pf.rshift);    /* red-shift */
22639f64916dSGerd Hoffmann     vnc_write_u8(vs, vs->client_pf.gshift);    /* green-shift */
22649f64916dSGerd Hoffmann     vnc_write_u8(vs, vs->client_pf.bshift);    /* blue-shift */
22659f64916dSGerd Hoffmann     vnc_write(vs, pad, 3);           /* padding */
22663e230dd2SCorentin Chary 
22673e230dd2SCorentin Chary     vnc_hextile_set_pixel_conversion(vs, 0);
22683e230dd2SCorentin Chary     vs->write_pixels = vnc_write_pixels_copy;
22693e230dd2SCorentin Chary }
22703e230dd2SCorentin Chary 
22713e230dd2SCorentin Chary static void vnc_colordepth(VncState *vs)
22723e230dd2SCorentin Chary {
22733e230dd2SCorentin Chary     if (vnc_has_feature(vs, VNC_FEATURE_WMVI)) {
22743e230dd2SCorentin Chary         /* Sending a WMVi message to notify the client*/
2275bd023f95SCorentin Chary         vnc_lock_output(vs);
22763e230dd2SCorentin Chary         vnc_write_u8(vs, VNC_MSG_SERVER_FRAMEBUFFER_UPDATE);
22773e230dd2SCorentin Chary         vnc_write_u8(vs, 0);
22783e230dd2SCorentin Chary         vnc_write_u16(vs, 1); /* number of rects */
2279d39fa6d8SGerd Hoffmann         vnc_framebuffer_update(vs, 0, 0,
2280bea60dd7SPeter Lieven                                pixman_image_get_width(vs->vd->server),
2281bea60dd7SPeter Lieven                                pixman_image_get_height(vs->vd->server),
2282d39fa6d8SGerd Hoffmann                                VNC_ENCODING_WMVi);
22833e230dd2SCorentin Chary         pixel_format_message(vs);
2284bd023f95SCorentin Chary         vnc_unlock_output(vs);
22853e230dd2SCorentin Chary         vnc_flush(vs);
22863e230dd2SCorentin Chary     } else {
22873e230dd2SCorentin Chary         set_pixel_conversion(vs);
22883e230dd2SCorentin Chary     }
22893e230dd2SCorentin Chary }
22903e230dd2SCorentin Chary 
22913e230dd2SCorentin Chary static int protocol_client_msg(VncState *vs, uint8_t *data, size_t len)
22923e230dd2SCorentin Chary {
22933e230dd2SCorentin Chary     int i;
22943e230dd2SCorentin Chary     uint16_t limit;
22953e230dd2SCorentin Chary     VncDisplay *vd = vs->vd;
22963e230dd2SCorentin Chary 
22973e230dd2SCorentin Chary     if (data[0] > 3) {
22980f7b2864SGerd Hoffmann         update_displaychangelistener(&vd->dcl, VNC_REFRESH_INTERVAL_BASE);
22993e230dd2SCorentin Chary     }
23003e230dd2SCorentin Chary 
23013e230dd2SCorentin Chary     switch (data[0]) {
23023e230dd2SCorentin Chary     case VNC_MSG_CLIENT_SET_PIXEL_FORMAT:
23033e230dd2SCorentin Chary         if (len == 1)
23043e230dd2SCorentin Chary             return 20;
23053e230dd2SCorentin Chary 
23063e230dd2SCorentin Chary         set_pixel_format(vs, read_u8(data, 4), read_u8(data, 5),
23073e230dd2SCorentin Chary                          read_u8(data, 6), read_u8(data, 7),
23083e230dd2SCorentin Chary                          read_u16(data, 8), read_u16(data, 10),
23093e230dd2SCorentin Chary                          read_u16(data, 12), read_u8(data, 14),
23103e230dd2SCorentin Chary                          read_u8(data, 15), read_u8(data, 16));
23113e230dd2SCorentin Chary         break;
23123e230dd2SCorentin Chary     case VNC_MSG_CLIENT_SET_ENCODINGS:
23133e230dd2SCorentin Chary         if (len == 1)
23143e230dd2SCorentin Chary             return 4;
23153e230dd2SCorentin Chary 
23163e230dd2SCorentin Chary         if (len == 4) {
23173e230dd2SCorentin Chary             limit = read_u16(data, 2);
23183e230dd2SCorentin Chary             if (limit > 0)
23193e230dd2SCorentin Chary                 return 4 + (limit * 4);
23203e230dd2SCorentin Chary         } else
23213e230dd2SCorentin Chary             limit = read_u16(data, 2);
23223e230dd2SCorentin Chary 
23233e230dd2SCorentin Chary         for (i = 0; i < limit; i++) {
23243e230dd2SCorentin Chary             int32_t val = read_s32(data, 4 + (i * 4));
23253e230dd2SCorentin Chary             memcpy(data + 4 + (i * 4), &val, sizeof(val));
23263e230dd2SCorentin Chary         }
23273e230dd2SCorentin Chary 
23283e230dd2SCorentin Chary         set_encodings(vs, (int32_t *)(data + 4), limit);
23293e230dd2SCorentin Chary         break;
23303e230dd2SCorentin Chary     case VNC_MSG_CLIENT_FRAMEBUFFER_UPDATE_REQUEST:
23313e230dd2SCorentin Chary         if (len == 1)
23323e230dd2SCorentin Chary             return 10;
23333e230dd2SCorentin Chary 
23343e230dd2SCorentin Chary         framebuffer_update_request(vs,
23353e230dd2SCorentin Chary                                    read_u8(data, 1), read_u16(data, 2), read_u16(data, 4),
23363e230dd2SCorentin Chary                                    read_u16(data, 6), read_u16(data, 8));
23373e230dd2SCorentin Chary         break;
23383e230dd2SCorentin Chary     case VNC_MSG_CLIENT_KEY_EVENT:
23393e230dd2SCorentin Chary         if (len == 1)
23403e230dd2SCorentin Chary             return 8;
23413e230dd2SCorentin Chary 
23423e230dd2SCorentin Chary         key_event(vs, read_u8(data, 1), read_u32(data, 4));
23433e230dd2SCorentin Chary         break;
23443e230dd2SCorentin Chary     case VNC_MSG_CLIENT_POINTER_EVENT:
23453e230dd2SCorentin Chary         if (len == 1)
23463e230dd2SCorentin Chary             return 6;
23473e230dd2SCorentin Chary 
23483e230dd2SCorentin Chary         pointer_event(vs, read_u8(data, 1), read_u16(data, 2), read_u16(data, 4));
23493e230dd2SCorentin Chary         break;
23503e230dd2SCorentin Chary     case VNC_MSG_CLIENT_CUT_TEXT:
2351f9a70e79SPeter Lieven         if (len == 1) {
23523e230dd2SCorentin Chary             return 8;
2353f9a70e79SPeter Lieven         }
23543e230dd2SCorentin Chary         if (len == 8) {
23553e230dd2SCorentin Chary             uint32_t dlen = read_u32(data, 4);
2356f9a70e79SPeter Lieven             if (dlen > (1 << 20)) {
2357f9a70e79SPeter Lieven                 error_report("vnc: client_cut_text msg payload has %u bytes"
2358f9a70e79SPeter Lieven                              " which exceeds our limit of 1MB.", dlen);
2359f9a70e79SPeter Lieven                 vnc_client_error(vs);
2360f9a70e79SPeter Lieven                 break;
2361f9a70e79SPeter Lieven             }
2362f9a70e79SPeter Lieven             if (dlen > 0) {
23633e230dd2SCorentin Chary                 return 8 + dlen;
23643e230dd2SCorentin Chary             }
2365f9a70e79SPeter Lieven         }
23663e230dd2SCorentin Chary 
23673e230dd2SCorentin Chary         client_cut_text(vs, read_u32(data, 4), data + 8);
23683e230dd2SCorentin Chary         break;
23693e230dd2SCorentin Chary     case VNC_MSG_CLIENT_QEMU:
23703e230dd2SCorentin Chary         if (len == 1)
23713e230dd2SCorentin Chary             return 2;
23723e230dd2SCorentin Chary 
23733e230dd2SCorentin Chary         switch (read_u8(data, 1)) {
23743e230dd2SCorentin Chary         case VNC_MSG_CLIENT_QEMU_EXT_KEY_EVENT:
23753e230dd2SCorentin Chary             if (len == 2)
23763e230dd2SCorentin Chary                 return 12;
23773e230dd2SCorentin Chary 
23783e230dd2SCorentin Chary             ext_key_event(vs, read_u16(data, 2),
23793e230dd2SCorentin Chary                           read_u32(data, 4), read_u32(data, 8));
23803e230dd2SCorentin Chary             break;
23813e230dd2SCorentin Chary         case VNC_MSG_CLIENT_QEMU_AUDIO:
23823e230dd2SCorentin Chary             if (len == 2)
23833e230dd2SCorentin Chary                 return 4;
23843e230dd2SCorentin Chary 
23853e230dd2SCorentin Chary             switch (read_u16 (data, 2)) {
23863e230dd2SCorentin Chary             case VNC_MSG_CLIENT_QEMU_AUDIO_ENABLE:
23873e230dd2SCorentin Chary                 audio_add(vs);
23883e230dd2SCorentin Chary                 break;
23893e230dd2SCorentin Chary             case VNC_MSG_CLIENT_QEMU_AUDIO_DISABLE:
23903e230dd2SCorentin Chary                 audio_del(vs);
23913e230dd2SCorentin Chary                 break;
23923e230dd2SCorentin Chary             case VNC_MSG_CLIENT_QEMU_AUDIO_SET_FORMAT:
23933e230dd2SCorentin Chary                 if (len == 4)
23943e230dd2SCorentin Chary                     return 10;
23953e230dd2SCorentin Chary                 switch (read_u8(data, 4)) {
23963e230dd2SCorentin Chary                 case 0: vs->as.fmt = AUD_FMT_U8; break;
23973e230dd2SCorentin Chary                 case 1: vs->as.fmt = AUD_FMT_S8; break;
23983e230dd2SCorentin Chary                 case 2: vs->as.fmt = AUD_FMT_U16; break;
23993e230dd2SCorentin Chary                 case 3: vs->as.fmt = AUD_FMT_S16; break;
24003e230dd2SCorentin Chary                 case 4: vs->as.fmt = AUD_FMT_U32; break;
24013e230dd2SCorentin Chary                 case 5: vs->as.fmt = AUD_FMT_S32; break;
24023e230dd2SCorentin Chary                 default:
24033e230dd2SCorentin Chary                     printf("Invalid audio format %d\n", read_u8(data, 4));
24043e230dd2SCorentin Chary                     vnc_client_error(vs);
24053e230dd2SCorentin Chary                     break;
24063e230dd2SCorentin Chary                 }
24073e230dd2SCorentin Chary                 vs->as.nchannels = read_u8(data, 5);
24083e230dd2SCorentin Chary                 if (vs->as.nchannels != 1 && vs->as.nchannels != 2) {
24093e230dd2SCorentin Chary                     printf("Invalid audio channel coount %d\n",
24103e230dd2SCorentin Chary                            read_u8(data, 5));
24113e230dd2SCorentin Chary                     vnc_client_error(vs);
24123e230dd2SCorentin Chary                     break;
24133e230dd2SCorentin Chary                 }
24143e230dd2SCorentin Chary                 vs->as.freq = read_u32(data, 6);
24153e230dd2SCorentin Chary                 break;
24163e230dd2SCorentin Chary             default:
24173e230dd2SCorentin Chary                 printf ("Invalid audio message %d\n", read_u8(data, 4));
24183e230dd2SCorentin Chary                 vnc_client_error(vs);
24193e230dd2SCorentin Chary                 break;
24203e230dd2SCorentin Chary             }
24213e230dd2SCorentin Chary             break;
24223e230dd2SCorentin Chary 
24233e230dd2SCorentin Chary         default:
24243e230dd2SCorentin Chary             printf("Msg: %d\n", read_u16(data, 0));
24253e230dd2SCorentin Chary             vnc_client_error(vs);
24263e230dd2SCorentin Chary             break;
24273e230dd2SCorentin Chary         }
24283e230dd2SCorentin Chary         break;
24293e230dd2SCorentin Chary     default:
24303e230dd2SCorentin Chary         printf("Msg: %d\n", data[0]);
24313e230dd2SCorentin Chary         vnc_client_error(vs);
24323e230dd2SCorentin Chary         break;
24333e230dd2SCorentin Chary     }
24343e230dd2SCorentin Chary 
24353e230dd2SCorentin Chary     vnc_read_when(vs, protocol_client_msg, 1);
24363e230dd2SCorentin Chary     return 0;
24373e230dd2SCorentin Chary }
24383e230dd2SCorentin Chary 
24393e230dd2SCorentin Chary static int protocol_client_init(VncState *vs, uint8_t *data, size_t len)
24403e230dd2SCorentin Chary {
24413e230dd2SCorentin Chary     char buf[1024];
24428cf36489SGerd Hoffmann     VncShareMode mode;
24433e230dd2SCorentin Chary     int size;
24443e230dd2SCorentin Chary 
24458cf36489SGerd Hoffmann     mode = data[0] ? VNC_SHARE_MODE_SHARED : VNC_SHARE_MODE_EXCLUSIVE;
24468cf36489SGerd Hoffmann     switch (vs->vd->share_policy) {
24478cf36489SGerd Hoffmann     case VNC_SHARE_POLICY_IGNORE:
24488cf36489SGerd Hoffmann         /*
24498cf36489SGerd Hoffmann          * Ignore the shared flag.  Nothing to do here.
24508cf36489SGerd Hoffmann          *
24518cf36489SGerd Hoffmann          * Doesn't conform to the rfb spec but is traditional qemu
24528cf36489SGerd Hoffmann          * behavior, thus left here as option for compatibility
24538cf36489SGerd Hoffmann          * reasons.
24548cf36489SGerd Hoffmann          */
24558cf36489SGerd Hoffmann         break;
24568cf36489SGerd Hoffmann     case VNC_SHARE_POLICY_ALLOW_EXCLUSIVE:
24578cf36489SGerd Hoffmann         /*
24588cf36489SGerd Hoffmann          * Policy: Allow clients ask for exclusive access.
24598cf36489SGerd Hoffmann          *
24608cf36489SGerd Hoffmann          * Implementation: When a client asks for exclusive access,
24618cf36489SGerd Hoffmann          * disconnect all others. Shared connects are allowed as long
24628cf36489SGerd Hoffmann          * as no exclusive connection exists.
24638cf36489SGerd Hoffmann          *
24648cf36489SGerd Hoffmann          * This is how the rfb spec suggests to handle the shared flag.
24658cf36489SGerd Hoffmann          */
24668cf36489SGerd Hoffmann         if (mode == VNC_SHARE_MODE_EXCLUSIVE) {
24678cf36489SGerd Hoffmann             VncState *client;
24688cf36489SGerd Hoffmann             QTAILQ_FOREACH(client, &vs->vd->clients, next) {
24698cf36489SGerd Hoffmann                 if (vs == client) {
24708cf36489SGerd Hoffmann                     continue;
24718cf36489SGerd Hoffmann                 }
24728cf36489SGerd Hoffmann                 if (client->share_mode != VNC_SHARE_MODE_EXCLUSIVE &&
24738cf36489SGerd Hoffmann                     client->share_mode != VNC_SHARE_MODE_SHARED) {
24748cf36489SGerd Hoffmann                     continue;
24758cf36489SGerd Hoffmann                 }
24768cf36489SGerd Hoffmann                 vnc_disconnect_start(client);
24778cf36489SGerd Hoffmann             }
24788cf36489SGerd Hoffmann         }
24798cf36489SGerd Hoffmann         if (mode == VNC_SHARE_MODE_SHARED) {
24808cf36489SGerd Hoffmann             if (vs->vd->num_exclusive > 0) {
24818cf36489SGerd Hoffmann                 vnc_disconnect_start(vs);
24828cf36489SGerd Hoffmann                 return 0;
24838cf36489SGerd Hoffmann             }
24848cf36489SGerd Hoffmann         }
24858cf36489SGerd Hoffmann         break;
24868cf36489SGerd Hoffmann     case VNC_SHARE_POLICY_FORCE_SHARED:
24878cf36489SGerd Hoffmann         /*
24888cf36489SGerd Hoffmann          * Policy: Shared connects only.
24898cf36489SGerd Hoffmann          * Implementation: Disallow clients asking for exclusive access.
24908cf36489SGerd Hoffmann          *
24918cf36489SGerd Hoffmann          * Useful for shared desktop sessions where you don't want
24928cf36489SGerd Hoffmann          * someone forgetting to say -shared when running the vnc
24938cf36489SGerd Hoffmann          * client disconnect everybody else.
24948cf36489SGerd Hoffmann          */
24958cf36489SGerd Hoffmann         if (mode == VNC_SHARE_MODE_EXCLUSIVE) {
24968cf36489SGerd Hoffmann             vnc_disconnect_start(vs);
24978cf36489SGerd Hoffmann             return 0;
24988cf36489SGerd Hoffmann         }
24998cf36489SGerd Hoffmann         break;
25008cf36489SGerd Hoffmann     }
25018cf36489SGerd Hoffmann     vnc_set_share_mode(vs, mode);
25028cf36489SGerd Hoffmann 
2503e5f34cddSGerd Hoffmann     if (vs->vd->num_shared > vs->vd->connections_limit) {
2504e5f34cddSGerd Hoffmann         vnc_disconnect_start(vs);
2505e5f34cddSGerd Hoffmann         return 0;
2506e5f34cddSGerd Hoffmann     }
2507e5f34cddSGerd Hoffmann 
2508bea60dd7SPeter Lieven     vs->client_width = pixman_image_get_width(vs->vd->server);
2509bea60dd7SPeter Lieven     vs->client_height = pixman_image_get_height(vs->vd->server);
25103e230dd2SCorentin Chary     vnc_write_u16(vs, vs->client_width);
25113e230dd2SCorentin Chary     vnc_write_u16(vs, vs->client_height);
25123e230dd2SCorentin Chary 
25133e230dd2SCorentin Chary     pixel_format_message(vs);
25143e230dd2SCorentin Chary 
25153e230dd2SCorentin Chary     if (qemu_name)
25163e230dd2SCorentin Chary         size = snprintf(buf, sizeof(buf), "QEMU (%s)", qemu_name);
25173e230dd2SCorentin Chary     else
25183e230dd2SCorentin Chary         size = snprintf(buf, sizeof(buf), "QEMU");
25193e230dd2SCorentin Chary 
25203e230dd2SCorentin Chary     vnc_write_u32(vs, size);
25213e230dd2SCorentin Chary     vnc_write(vs, buf, size);
25223e230dd2SCorentin Chary     vnc_flush(vs);
25233e230dd2SCorentin Chary 
25243e230dd2SCorentin Chary     vnc_client_cache_auth(vs);
2525fb6ba0d5SWenchao Xia     vnc_qmp_event(vs, QAPI_EVENT_VNC_INITIALIZED);
25263e230dd2SCorentin Chary 
25273e230dd2SCorentin Chary     vnc_read_when(vs, protocol_client_msg, 1);
25283e230dd2SCorentin Chary 
25293e230dd2SCorentin Chary     return 0;
25303e230dd2SCorentin Chary }
25313e230dd2SCorentin Chary 
25323e230dd2SCorentin Chary void start_client_init(VncState *vs)
25333e230dd2SCorentin Chary {
25343e230dd2SCorentin Chary     vnc_read_when(vs, protocol_client_init, 1);
25353e230dd2SCorentin Chary }
25363e230dd2SCorentin Chary 
25373e230dd2SCorentin Chary static void make_challenge(VncState *vs)
25383e230dd2SCorentin Chary {
25393e230dd2SCorentin Chary     int i;
25403e230dd2SCorentin Chary 
25413e230dd2SCorentin Chary     srand(time(NULL)+getpid()+getpid()*987654+rand());
25423e230dd2SCorentin Chary 
25433e230dd2SCorentin Chary     for (i = 0 ; i < sizeof(vs->challenge) ; i++)
25443e230dd2SCorentin Chary         vs->challenge[i] = (int) (256.0*rand()/(RAND_MAX+1.0));
25453e230dd2SCorentin Chary }
25463e230dd2SCorentin Chary 
25473e230dd2SCorentin Chary static int protocol_client_auth_vnc(VncState *vs, uint8_t *data, size_t len)
25483e230dd2SCorentin Chary {
25493e230dd2SCorentin Chary     unsigned char response[VNC_AUTH_CHALLENGE_SIZE];
25503e230dd2SCorentin Chary     int i, j, pwlen;
25513e230dd2SCorentin Chary     unsigned char key[8];
25523c9405a0SGerd Hoffmann     time_t now = time(NULL);
25533e230dd2SCorentin Chary 
25541cd20f8bSAnthony Liguori     if (!vs->vd->password) {
25553e230dd2SCorentin Chary         VNC_DEBUG("No password configured on server");
25566bffdf0fSGerd Hoffmann         goto reject;
25573e230dd2SCorentin Chary     }
25583c9405a0SGerd Hoffmann     if (vs->vd->expires < now) {
25593c9405a0SGerd Hoffmann         VNC_DEBUG("Password is expired");
25603c9405a0SGerd Hoffmann         goto reject;
25613c9405a0SGerd Hoffmann     }
25623e230dd2SCorentin Chary 
25633e230dd2SCorentin Chary     memcpy(response, vs->challenge, VNC_AUTH_CHALLENGE_SIZE);
25643e230dd2SCorentin Chary 
25653e230dd2SCorentin Chary     /* Calculate the expected challenge response */
25663e230dd2SCorentin Chary     pwlen = strlen(vs->vd->password);
25673e230dd2SCorentin Chary     for (i=0; i<sizeof(key); i++)
25683e230dd2SCorentin Chary         key[i] = i<pwlen ? vs->vd->password[i] : 0;
25693e230dd2SCorentin Chary     deskey(key, EN0);
25703e230dd2SCorentin Chary     for (j = 0; j < VNC_AUTH_CHALLENGE_SIZE; j += 8)
25713e230dd2SCorentin Chary         des(response+j, response+j);
25723e230dd2SCorentin Chary 
25733e230dd2SCorentin Chary     /* Compare expected vs actual challenge response */
25743e230dd2SCorentin Chary     if (memcmp(response, data, VNC_AUTH_CHALLENGE_SIZE) != 0) {
2575e5bed759SDong Xu Wang         VNC_DEBUG("Client challenge response did not match\n");
25766bffdf0fSGerd Hoffmann         goto reject;
25776bffdf0fSGerd Hoffmann     } else {
25786bffdf0fSGerd Hoffmann         VNC_DEBUG("Accepting VNC challenge response\n");
25796bffdf0fSGerd Hoffmann         vnc_write_u32(vs, 0); /* Accept auth */
25806bffdf0fSGerd Hoffmann         vnc_flush(vs);
25816bffdf0fSGerd Hoffmann 
25826bffdf0fSGerd Hoffmann         start_client_init(vs);
25836bffdf0fSGerd Hoffmann     }
25846bffdf0fSGerd Hoffmann     return 0;
25856bffdf0fSGerd Hoffmann 
25866bffdf0fSGerd Hoffmann reject:
25873e230dd2SCorentin Chary     vnc_write_u32(vs, 1); /* Reject auth */
25883e230dd2SCorentin Chary     if (vs->minor >= 8) {
25893e230dd2SCorentin Chary         static const char err[] = "Authentication failed";
25903e230dd2SCorentin Chary         vnc_write_u32(vs, sizeof(err));
25913e230dd2SCorentin Chary         vnc_write(vs, err, sizeof(err));
25923e230dd2SCorentin Chary     }
25933e230dd2SCorentin Chary     vnc_flush(vs);
25943e230dd2SCorentin Chary     vnc_client_error(vs);
25953e230dd2SCorentin Chary     return 0;
25963e230dd2SCorentin Chary }
25973e230dd2SCorentin Chary 
25983e230dd2SCorentin Chary void start_auth_vnc(VncState *vs)
25993e230dd2SCorentin Chary {
26003e230dd2SCorentin Chary     make_challenge(vs);
26013e230dd2SCorentin Chary     /* Send client a 'random' challenge */
26023e230dd2SCorentin Chary     vnc_write(vs, vs->challenge, sizeof(vs->challenge));
26033e230dd2SCorentin Chary     vnc_flush(vs);
26043e230dd2SCorentin Chary 
26053e230dd2SCorentin Chary     vnc_read_when(vs, protocol_client_auth_vnc, sizeof(vs->challenge));
26063e230dd2SCorentin Chary }
26073e230dd2SCorentin Chary 
26083e230dd2SCorentin Chary 
26093e230dd2SCorentin Chary static int protocol_client_auth(VncState *vs, uint8_t *data, size_t len)
26103e230dd2SCorentin Chary {
26113e230dd2SCorentin Chary     /* We only advertise 1 auth scheme at a time, so client
26123e230dd2SCorentin Chary      * must pick the one we sent. Verify this */
26137e7e2ebcSDaniel P. Berrange     if (data[0] != vs->auth) { /* Reject auth */
26143e230dd2SCorentin Chary        VNC_DEBUG("Reject auth %d because it didn't match advertized\n", (int)data[0]);
26153e230dd2SCorentin Chary        vnc_write_u32(vs, 1);
26163e230dd2SCorentin Chary        if (vs->minor >= 8) {
26173e230dd2SCorentin Chary            static const char err[] = "Authentication failed";
26183e230dd2SCorentin Chary            vnc_write_u32(vs, sizeof(err));
26193e230dd2SCorentin Chary            vnc_write(vs, err, sizeof(err));
26203e230dd2SCorentin Chary        }
26213e230dd2SCorentin Chary        vnc_client_error(vs);
26223e230dd2SCorentin Chary     } else { /* Accept requested auth */
26233e230dd2SCorentin Chary        VNC_DEBUG("Client requested auth %d\n", (int)data[0]);
26247e7e2ebcSDaniel P. Berrange        switch (vs->auth) {
26253e230dd2SCorentin Chary        case VNC_AUTH_NONE:
26263e230dd2SCorentin Chary            VNC_DEBUG("Accept auth none\n");
26273e230dd2SCorentin Chary            if (vs->minor >= 8) {
26283e230dd2SCorentin Chary                vnc_write_u32(vs, 0); /* Accept auth completion */
26293e230dd2SCorentin Chary                vnc_flush(vs);
26303e230dd2SCorentin Chary            }
26313e230dd2SCorentin Chary            start_client_init(vs);
26323e230dd2SCorentin Chary            break;
26333e230dd2SCorentin Chary 
26343e230dd2SCorentin Chary        case VNC_AUTH_VNC:
26353e230dd2SCorentin Chary            VNC_DEBUG("Start VNC auth\n");
26363e230dd2SCorentin Chary            start_auth_vnc(vs);
26373e230dd2SCorentin Chary            break;
26383e230dd2SCorentin Chary 
26393e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
26403e230dd2SCorentin Chary        case VNC_AUTH_VENCRYPT:
26413a93113aSDong Xu Wang            VNC_DEBUG("Accept VeNCrypt auth\n");
26423e230dd2SCorentin Chary            start_auth_vencrypt(vs);
26433e230dd2SCorentin Chary            break;
26443e230dd2SCorentin Chary #endif /* CONFIG_VNC_TLS */
26453e230dd2SCorentin Chary 
26463e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
26473e230dd2SCorentin Chary        case VNC_AUTH_SASL:
26483e230dd2SCorentin Chary            VNC_DEBUG("Accept SASL auth\n");
26493e230dd2SCorentin Chary            start_auth_sasl(vs);
26503e230dd2SCorentin Chary            break;
26513e230dd2SCorentin Chary #endif /* CONFIG_VNC_SASL */
26523e230dd2SCorentin Chary 
26533e230dd2SCorentin Chary        default: /* Should not be possible, but just in case */
26547e7e2ebcSDaniel P. Berrange            VNC_DEBUG("Reject auth %d server code bug\n", vs->auth);
26553e230dd2SCorentin Chary            vnc_write_u8(vs, 1);
26563e230dd2SCorentin Chary            if (vs->minor >= 8) {
26573e230dd2SCorentin Chary                static const char err[] = "Authentication failed";
26583e230dd2SCorentin Chary                vnc_write_u32(vs, sizeof(err));
26593e230dd2SCorentin Chary                vnc_write(vs, err, sizeof(err));
26603e230dd2SCorentin Chary            }
26613e230dd2SCorentin Chary            vnc_client_error(vs);
26623e230dd2SCorentin Chary        }
26633e230dd2SCorentin Chary     }
26643e230dd2SCorentin Chary     return 0;
26653e230dd2SCorentin Chary }
26663e230dd2SCorentin Chary 
26673e230dd2SCorentin Chary static int protocol_version(VncState *vs, uint8_t *version, size_t len)
26683e230dd2SCorentin Chary {
26693e230dd2SCorentin Chary     char local[13];
26703e230dd2SCorentin Chary 
26713e230dd2SCorentin Chary     memcpy(local, version, 12);
26723e230dd2SCorentin Chary     local[12] = 0;
26733e230dd2SCorentin Chary 
26743e230dd2SCorentin Chary     if (sscanf(local, "RFB %03d.%03d\n", &vs->major, &vs->minor) != 2) {
26753e230dd2SCorentin Chary         VNC_DEBUG("Malformed protocol version %s\n", local);
26763e230dd2SCorentin Chary         vnc_client_error(vs);
26773e230dd2SCorentin Chary         return 0;
26783e230dd2SCorentin Chary     }
26793e230dd2SCorentin Chary     VNC_DEBUG("Client request protocol version %d.%d\n", vs->major, vs->minor);
26803e230dd2SCorentin Chary     if (vs->major != 3 ||
26813e230dd2SCorentin Chary         (vs->minor != 3 &&
26823e230dd2SCorentin Chary          vs->minor != 4 &&
26833e230dd2SCorentin Chary          vs->minor != 5 &&
26843e230dd2SCorentin Chary          vs->minor != 7 &&
26853e230dd2SCorentin Chary          vs->minor != 8)) {
26863e230dd2SCorentin Chary         VNC_DEBUG("Unsupported client version\n");
26873e230dd2SCorentin Chary         vnc_write_u32(vs, VNC_AUTH_INVALID);
26883e230dd2SCorentin Chary         vnc_flush(vs);
26893e230dd2SCorentin Chary         vnc_client_error(vs);
26903e230dd2SCorentin Chary         return 0;
26913e230dd2SCorentin Chary     }
26923e230dd2SCorentin Chary     /* Some broken clients report v3.4 or v3.5, which spec requires to be treated
26933e230dd2SCorentin Chary      * as equivalent to v3.3 by servers
26943e230dd2SCorentin Chary      */
26953e230dd2SCorentin Chary     if (vs->minor == 4 || vs->minor == 5)
26963e230dd2SCorentin Chary         vs->minor = 3;
26973e230dd2SCorentin Chary 
26983e230dd2SCorentin Chary     if (vs->minor == 3) {
26997e7e2ebcSDaniel P. Berrange         if (vs->auth == VNC_AUTH_NONE) {
27003e230dd2SCorentin Chary             VNC_DEBUG("Tell client auth none\n");
27017e7e2ebcSDaniel P. Berrange             vnc_write_u32(vs, vs->auth);
27023e230dd2SCorentin Chary             vnc_flush(vs);
27033e230dd2SCorentin Chary             start_client_init(vs);
27047e7e2ebcSDaniel P. Berrange        } else if (vs->auth == VNC_AUTH_VNC) {
27053e230dd2SCorentin Chary             VNC_DEBUG("Tell client VNC auth\n");
27067e7e2ebcSDaniel P. Berrange             vnc_write_u32(vs, vs->auth);
27073e230dd2SCorentin Chary             vnc_flush(vs);
27083e230dd2SCorentin Chary             start_auth_vnc(vs);
27093e230dd2SCorentin Chary        } else {
27107e7e2ebcSDaniel P. Berrange             VNC_DEBUG("Unsupported auth %d for protocol 3.3\n", vs->auth);
27113e230dd2SCorentin Chary             vnc_write_u32(vs, VNC_AUTH_INVALID);
27123e230dd2SCorentin Chary             vnc_flush(vs);
27133e230dd2SCorentin Chary             vnc_client_error(vs);
27143e230dd2SCorentin Chary        }
27153e230dd2SCorentin Chary     } else {
27167e7e2ebcSDaniel P. Berrange         VNC_DEBUG("Telling client we support auth %d\n", vs->auth);
27173e230dd2SCorentin Chary         vnc_write_u8(vs, 1); /* num auth */
27187e7e2ebcSDaniel P. Berrange         vnc_write_u8(vs, vs->auth);
27193e230dd2SCorentin Chary         vnc_read_when(vs, protocol_client_auth, 1);
27203e230dd2SCorentin Chary         vnc_flush(vs);
27213e230dd2SCorentin Chary     }
27223e230dd2SCorentin Chary 
27233e230dd2SCorentin Chary     return 0;
27243e230dd2SCorentin Chary }
27253e230dd2SCorentin Chary 
2726999342a0SCorentin Chary static VncRectStat *vnc_stat_rect(VncDisplay *vd, int x, int y)
2727999342a0SCorentin Chary {
2728999342a0SCorentin Chary     struct VncSurface *vs = &vd->guest;
2729999342a0SCorentin Chary 
2730999342a0SCorentin Chary     return &vs->stats[y / VNC_STAT_RECT][x / VNC_STAT_RECT];
2731999342a0SCorentin Chary }
2732999342a0SCorentin Chary 
27337d964c9dSCorentin Chary void vnc_sent_lossy_rect(VncState *vs, int x, int y, int w, int h)
27347d964c9dSCorentin Chary {
27357d964c9dSCorentin Chary     int i, j;
27367d964c9dSCorentin Chary 
27377d964c9dSCorentin Chary     w = (x + w) / VNC_STAT_RECT;
27387d964c9dSCorentin Chary     h = (y + h) / VNC_STAT_RECT;
27397d964c9dSCorentin Chary     x /= VNC_STAT_RECT;
27407d964c9dSCorentin Chary     y /= VNC_STAT_RECT;
27417d964c9dSCorentin Chary 
2742207f328aSCorentin Chary     for (j = y; j <= h; j++) {
2743207f328aSCorentin Chary         for (i = x; i <= w; i++) {
27447d964c9dSCorentin Chary             vs->lossy_rect[j][i] = 1;
27457d964c9dSCorentin Chary         }
27467d964c9dSCorentin Chary     }
27477d964c9dSCorentin Chary }
27487d964c9dSCorentin Chary 
27497d964c9dSCorentin Chary static int vnc_refresh_lossy_rect(VncDisplay *vd, int x, int y)
27507d964c9dSCorentin Chary {
27517d964c9dSCorentin Chary     VncState *vs;
27527d964c9dSCorentin Chary     int sty = y / VNC_STAT_RECT;
27537d964c9dSCorentin Chary     int stx = x / VNC_STAT_RECT;
27547d964c9dSCorentin Chary     int has_dirty = 0;
27557d964c9dSCorentin Chary 
27567d964c9dSCorentin Chary     y = y / VNC_STAT_RECT * VNC_STAT_RECT;
27577d964c9dSCorentin Chary     x = x / VNC_STAT_RECT * VNC_STAT_RECT;
27587d964c9dSCorentin Chary 
27597d964c9dSCorentin Chary     QTAILQ_FOREACH(vs, &vd->clients, next) {
2760bc2429b9SCorentin Chary         int j;
27617d964c9dSCorentin Chary 
27627d964c9dSCorentin Chary         /* kernel send buffers are full -> refresh later */
27637d964c9dSCorentin Chary         if (vs->output.offset) {
27647d964c9dSCorentin Chary             continue;
27657d964c9dSCorentin Chary         }
27667d964c9dSCorentin Chary 
27677d964c9dSCorentin Chary         if (!vs->lossy_rect[sty][stx]) {
27687d964c9dSCorentin Chary             continue;
27697d964c9dSCorentin Chary         }
2770207f328aSCorentin Chary 
27717d964c9dSCorentin Chary         vs->lossy_rect[sty][stx] = 0;
27727d964c9dSCorentin Chary         for (j = 0; j < VNC_STAT_RECT; ++j) {
2773b4c85ddcSPeter Lieven             bitmap_set(vs->dirty[y + j],
2774b4c85ddcSPeter Lieven                        x / VNC_DIRTY_PIXELS_PER_BIT,
2775b4c85ddcSPeter Lieven                        VNC_STAT_RECT / VNC_DIRTY_PIXELS_PER_BIT);
27767d964c9dSCorentin Chary         }
27777d964c9dSCorentin Chary         has_dirty++;
27787d964c9dSCorentin Chary     }
2779207f328aSCorentin Chary 
27807d964c9dSCorentin Chary     return has_dirty;
27817d964c9dSCorentin Chary }
27827d964c9dSCorentin Chary 
27837d964c9dSCorentin Chary static int vnc_update_stats(VncDisplay *vd,  struct timeval * tv)
2784999342a0SCorentin Chary {
27859f64916dSGerd Hoffmann     int width = pixman_image_get_width(vd->guest.fb);
27869f64916dSGerd Hoffmann     int height = pixman_image_get_height(vd->guest.fb);
2787999342a0SCorentin Chary     int x, y;
2788999342a0SCorentin Chary     struct timeval res;
27897d964c9dSCorentin Chary     int has_dirty = 0;
2790999342a0SCorentin Chary 
27919f64916dSGerd Hoffmann     for (y = 0; y < height; y += VNC_STAT_RECT) {
27929f64916dSGerd Hoffmann         for (x = 0; x < width; x += VNC_STAT_RECT) {
2793999342a0SCorentin Chary             VncRectStat *rect = vnc_stat_rect(vd, x, y);
2794999342a0SCorentin Chary 
2795999342a0SCorentin Chary             rect->updated = false;
2796999342a0SCorentin Chary         }
2797999342a0SCorentin Chary     }
2798999342a0SCorentin Chary 
2799ad620c29SBlue Swirl     qemu_timersub(tv, &VNC_REFRESH_STATS, &res);
2800999342a0SCorentin Chary 
2801999342a0SCorentin Chary     if (timercmp(&vd->guest.last_freq_check, &res, >)) {
28027d964c9dSCorentin Chary         return has_dirty;
2803999342a0SCorentin Chary     }
2804999342a0SCorentin Chary     vd->guest.last_freq_check = *tv;
2805999342a0SCorentin Chary 
28069f64916dSGerd Hoffmann     for (y = 0; y < height; y += VNC_STAT_RECT) {
28079f64916dSGerd Hoffmann         for (x = 0; x < width; x += VNC_STAT_RECT) {
2808999342a0SCorentin Chary             VncRectStat *rect= vnc_stat_rect(vd, x, y);
2809999342a0SCorentin Chary             int count = ARRAY_SIZE(rect->times);
2810999342a0SCorentin Chary             struct timeval min, max;
2811999342a0SCorentin Chary 
2812999342a0SCorentin Chary             if (!timerisset(&rect->times[count - 1])) {
2813999342a0SCorentin Chary                 continue ;
2814999342a0SCorentin Chary             }
2815999342a0SCorentin Chary 
2816999342a0SCorentin Chary             max = rect->times[(rect->idx + count - 1) % count];
2817ad620c29SBlue Swirl             qemu_timersub(tv, &max, &res);
2818999342a0SCorentin Chary 
2819999342a0SCorentin Chary             if (timercmp(&res, &VNC_REFRESH_LOSSY, >)) {
2820999342a0SCorentin Chary                 rect->freq = 0;
28217d964c9dSCorentin Chary                 has_dirty += vnc_refresh_lossy_rect(vd, x, y);
2822999342a0SCorentin Chary                 memset(rect->times, 0, sizeof (rect->times));
2823999342a0SCorentin Chary                 continue ;
2824999342a0SCorentin Chary             }
2825999342a0SCorentin Chary 
2826999342a0SCorentin Chary             min = rect->times[rect->idx];
2827999342a0SCorentin Chary             max = rect->times[(rect->idx + count - 1) % count];
2828ad620c29SBlue Swirl             qemu_timersub(&max, &min, &res);
2829999342a0SCorentin Chary 
2830999342a0SCorentin Chary             rect->freq = res.tv_sec + res.tv_usec / 1000000.;
2831999342a0SCorentin Chary             rect->freq /= count;
2832999342a0SCorentin Chary             rect->freq = 1. / rect->freq;
2833999342a0SCorentin Chary         }
2834999342a0SCorentin Chary     }
28357d964c9dSCorentin Chary     return has_dirty;
2836999342a0SCorentin Chary }
2837999342a0SCorentin Chary 
2838999342a0SCorentin Chary double vnc_update_freq(VncState *vs, int x, int y, int w, int h)
2839999342a0SCorentin Chary {
2840999342a0SCorentin Chary     int i, j;
2841999342a0SCorentin Chary     double total = 0;
2842999342a0SCorentin Chary     int num = 0;
2843999342a0SCorentin Chary 
2844999342a0SCorentin Chary     x =  (x / VNC_STAT_RECT) * VNC_STAT_RECT;
2845999342a0SCorentin Chary     y =  (y / VNC_STAT_RECT) * VNC_STAT_RECT;
2846999342a0SCorentin Chary 
2847999342a0SCorentin Chary     for (j = y; j <= y + h; j += VNC_STAT_RECT) {
2848999342a0SCorentin Chary         for (i = x; i <= x + w; i += VNC_STAT_RECT) {
2849999342a0SCorentin Chary             total += vnc_stat_rect(vs->vd, i, j)->freq;
2850999342a0SCorentin Chary             num++;
2851999342a0SCorentin Chary         }
2852999342a0SCorentin Chary     }
2853999342a0SCorentin Chary 
2854999342a0SCorentin Chary     if (num) {
2855999342a0SCorentin Chary         return total / num;
2856999342a0SCorentin Chary     } else {
2857999342a0SCorentin Chary         return 0;
2858999342a0SCorentin Chary     }
2859999342a0SCorentin Chary }
2860999342a0SCorentin Chary 
2861999342a0SCorentin Chary static void vnc_rect_updated(VncDisplay *vd, int x, int y, struct timeval * tv)
2862999342a0SCorentin Chary {
2863999342a0SCorentin Chary     VncRectStat *rect;
2864999342a0SCorentin Chary 
2865999342a0SCorentin Chary     rect = vnc_stat_rect(vd, x, y);
2866999342a0SCorentin Chary     if (rect->updated) {
2867999342a0SCorentin Chary         return ;
2868999342a0SCorentin Chary     }
2869999342a0SCorentin Chary     rect->times[rect->idx] = *tv;
2870999342a0SCorentin Chary     rect->idx = (rect->idx + 1) % ARRAY_SIZE(rect->times);
2871999342a0SCorentin Chary     rect->updated = true;
2872999342a0SCorentin Chary }
2873999342a0SCorentin Chary 
28743e230dd2SCorentin Chary static int vnc_refresh_server_surface(VncDisplay *vd)
28753e230dd2SCorentin Chary {
2876bea60dd7SPeter Lieven     int width = MIN(pixman_image_get_width(vd->guest.fb),
2877bea60dd7SPeter Lieven                     pixman_image_get_width(vd->server));
2878bea60dd7SPeter Lieven     int height = MIN(pixman_image_get_height(vd->guest.fb),
2879bea60dd7SPeter Lieven                      pixman_image_get_height(vd->server));
2880bea60dd7SPeter Lieven     int cmp_bytes, server_stride, min_stride, guest_stride, y = 0;
288112b316d4SPeter Lieven     uint8_t *guest_row0 = NULL, *server_row0;
28823e230dd2SCorentin Chary     VncState *vs;
28833e230dd2SCorentin Chary     int has_dirty = 0;
28849f64916dSGerd Hoffmann     pixman_image_t *tmpbuf = NULL;
28853e230dd2SCorentin Chary 
288680e0c8c3SCorentin Chary     struct timeval tv = { 0, 0 };
2887999342a0SCorentin Chary 
288880e0c8c3SCorentin Chary     if (!vd->non_adaptive) {
2889999342a0SCorentin Chary         gettimeofday(&tv, NULL);
28907d964c9dSCorentin Chary         has_dirty = vnc_update_stats(vd, &tv);
289180e0c8c3SCorentin Chary     }
2892999342a0SCorentin Chary 
28933e230dd2SCorentin Chary     /*
28943e230dd2SCorentin Chary      * Walk through the guest dirty map.
28953e230dd2SCorentin Chary      * Check and copy modified bits from guest to server surface.
28963e230dd2SCorentin Chary      * Update server dirty map.
28973e230dd2SCorentin Chary      */
2898bea60dd7SPeter Lieven     server_row0 = (uint8_t *)pixman_image_get_data(vd->server);
2899bea60dd7SPeter Lieven     server_stride = guest_stride = pixman_image_get_stride(vd->server);
2900bea60dd7SPeter Lieven     cmp_bytes = MIN(VNC_DIRTY_PIXELS_PER_BIT * VNC_SERVER_FB_BYTES,
2901bea60dd7SPeter Lieven                     server_stride);
29029f64916dSGerd Hoffmann     if (vd->guest.format != VNC_SERVER_FB_FORMAT) {
29039f64916dSGerd Hoffmann         int width = pixman_image_get_width(vd->server);
29049f64916dSGerd Hoffmann         tmpbuf = qemu_pixman_linebuf_create(VNC_SERVER_FB_FORMAT, width);
290512b316d4SPeter Lieven     } else {
290612b316d4SPeter Lieven         guest_row0 = (uint8_t *)pixman_image_get_data(vd->guest.fb);
290712b316d4SPeter Lieven         guest_stride = pixman_image_get_stride(vd->guest.fb);
29089f64916dSGerd Hoffmann     }
2909bea60dd7SPeter Lieven     min_stride = MIN(server_stride, guest_stride);
291012b316d4SPeter Lieven 
291112b316d4SPeter Lieven     for (;;) {
29123e230dd2SCorentin Chary         int x;
291312b316d4SPeter Lieven         uint8_t *guest_ptr, *server_ptr;
291412b316d4SPeter Lieven         unsigned long offset = find_next_bit((unsigned long *) &vd->guest.dirty,
291512b316d4SPeter Lieven                                              height * VNC_DIRTY_BPL(&vd->guest),
291612b316d4SPeter Lieven                                              y * VNC_DIRTY_BPL(&vd->guest));
291712b316d4SPeter Lieven         if (offset == height * VNC_DIRTY_BPL(&vd->guest)) {
291812b316d4SPeter Lieven             /* no more dirty bits */
291912b316d4SPeter Lieven             break;
292012b316d4SPeter Lieven         }
292112b316d4SPeter Lieven         y = offset / VNC_DIRTY_BPL(&vd->guest);
292212b316d4SPeter Lieven         x = offset % VNC_DIRTY_BPL(&vd->guest);
292312b316d4SPeter Lieven 
292412b316d4SPeter Lieven         server_ptr = server_row0 + y * server_stride + x * cmp_bytes;
29253e230dd2SCorentin Chary 
29269f64916dSGerd Hoffmann         if (vd->guest.format != VNC_SERVER_FB_FORMAT) {
2927bc210eb1SGerd Hoffmann             qemu_pixman_linebuf_fill(tmpbuf, vd->guest.fb, width, 0, y);
29289f64916dSGerd Hoffmann             guest_ptr = (uint8_t *)pixman_image_get_data(tmpbuf);
29299f64916dSGerd Hoffmann         } else {
293012b316d4SPeter Lieven             guest_ptr = guest_row0 + y * guest_stride;
29319f64916dSGerd Hoffmann         }
293212b316d4SPeter Lieven         guest_ptr += x * cmp_bytes;
29333e230dd2SCorentin Chary 
293412b316d4SPeter Lieven         for (; x < DIV_ROUND_UP(width, VNC_DIRTY_PIXELS_PER_BIT);
293512b316d4SPeter Lieven              x++, guest_ptr += cmp_bytes, server_ptr += cmp_bytes) {
2936bea60dd7SPeter Lieven             int _cmp_bytes = cmp_bytes;
293712b316d4SPeter Lieven             if (!test_and_clear_bit(x, vd->guest.dirty[y])) {
29383e230dd2SCorentin Chary                 continue;
2939b4c85ddcSPeter Lieven             }
2940bea60dd7SPeter Lieven             if ((x + 1) * cmp_bytes > min_stride) {
2941bea60dd7SPeter Lieven                 _cmp_bytes = min_stride - x * cmp_bytes;
2942bea60dd7SPeter Lieven             }
2943bea60dd7SPeter Lieven             if (memcmp(server_ptr, guest_ptr, _cmp_bytes) == 0) {
29443e230dd2SCorentin Chary                 continue;
2945b4c85ddcSPeter Lieven             }
2946bea60dd7SPeter Lieven             memcpy(server_ptr, guest_ptr, _cmp_bytes);
294712b316d4SPeter Lieven             if (!vd->non_adaptive) {
294812b316d4SPeter Lieven                 vnc_rect_updated(vd, x * VNC_DIRTY_PIXELS_PER_BIT,
294912b316d4SPeter Lieven                                  y, &tv);
295012b316d4SPeter Lieven             }
29513e230dd2SCorentin Chary             QTAILQ_FOREACH(vs, &vd->clients, next) {
295212b316d4SPeter Lieven                 set_bit(x, vs->dirty[y]);
29533e230dd2SCorentin Chary             }
29543e230dd2SCorentin Chary             has_dirty++;
29553e230dd2SCorentin Chary         }
295612b316d4SPeter Lieven 
295712b316d4SPeter Lieven         y++;
29583e230dd2SCorentin Chary     }
29599f64916dSGerd Hoffmann     qemu_pixman_image_unref(tmpbuf);
29603e230dd2SCorentin Chary     return has_dirty;
29613e230dd2SCorentin Chary }
29623e230dd2SCorentin Chary 
29630f7b2864SGerd Hoffmann static void vnc_refresh(DisplayChangeListener *dcl)
29643e230dd2SCorentin Chary {
29650f7b2864SGerd Hoffmann     VncDisplay *vd = container_of(dcl, VncDisplay, dcl);
29663e230dd2SCorentin Chary     VncState *vs, *vn;
29673e230dd2SCorentin Chary     int has_dirty, rects = 0;
29683e230dd2SCorentin Chary 
29699d6b2070SChenLiang     if (QTAILQ_EMPTY(&vd->clients)) {
29709d6b2070SChenLiang         update_displaychangelistener(&vd->dcl, VNC_REFRESH_INTERVAL_MAX);
29719d6b2070SChenLiang         return;
29729d6b2070SChenLiang     }
29739d6b2070SChenLiang 
29741d0d59feSGerd Hoffmann     graphic_hw_update(vd->dcl.con);
29753e230dd2SCorentin Chary 
2976bd023f95SCorentin Chary     if (vnc_trylock_display(vd)) {
29770f7b2864SGerd Hoffmann         update_displaychangelistener(&vd->dcl, VNC_REFRESH_INTERVAL_BASE);
2978bd023f95SCorentin Chary         return;
2979bd023f95SCorentin Chary     }
2980bd023f95SCorentin Chary 
29813e230dd2SCorentin Chary     has_dirty = vnc_refresh_server_surface(vd);
2982bd023f95SCorentin Chary     vnc_unlock_display(vd);
29833e230dd2SCorentin Chary 
29843e230dd2SCorentin Chary     QTAILQ_FOREACH_SAFE(vs, &vd->clients, next, vn) {
298538ee14f4SGerd Hoffmann         rects += vnc_update_client(vs, has_dirty, false);
29863e230dd2SCorentin Chary         /* vs might be free()ed here */
29873e230dd2SCorentin Chary     }
2988bd023f95SCorentin Chary 
29893e230dd2SCorentin Chary     if (has_dirty && rects) {
29900f7b2864SGerd Hoffmann         vd->dcl.update_interval /= 2;
29910f7b2864SGerd Hoffmann         if (vd->dcl.update_interval < VNC_REFRESH_INTERVAL_BASE) {
29920f7b2864SGerd Hoffmann             vd->dcl.update_interval = VNC_REFRESH_INTERVAL_BASE;
29930f7b2864SGerd Hoffmann         }
29943e230dd2SCorentin Chary     } else {
29950f7b2864SGerd Hoffmann         vd->dcl.update_interval += VNC_REFRESH_INTERVAL_INC;
29960f7b2864SGerd Hoffmann         if (vd->dcl.update_interval > VNC_REFRESH_INTERVAL_MAX) {
29970f7b2864SGerd Hoffmann             vd->dcl.update_interval = VNC_REFRESH_INTERVAL_MAX;
29983e230dd2SCorentin Chary         }
29993e230dd2SCorentin Chary     }
30003e230dd2SCorentin Chary }
30013e230dd2SCorentin Chary 
30022c8cf549SMichael Tokarev static void vnc_connect(VncDisplay *vd, int csock,
30032c8cf549SMichael Tokarev                         bool skipauth, bool websocket)
30043e230dd2SCorentin Chary {
30057267c094SAnthony Liguori     VncState *vs = g_malloc0(sizeof(VncState));
30067d964c9dSCorentin Chary     int i;
30077d964c9dSCorentin Chary 
30083e230dd2SCorentin Chary     vs->csock = csock;
3009d616ccc5SGerd Hoffmann     vs->vd = vd;
30107e7e2ebcSDaniel P. Berrange 
30117e7e2ebcSDaniel P. Berrange     if (skipauth) {
30127e7e2ebcSDaniel P. Berrange 	vs->auth = VNC_AUTH_NONE;
30137e7e2ebcSDaniel P. Berrange #ifdef CONFIG_VNC_TLS
30147e7e2ebcSDaniel P. Berrange 	vs->subauth = VNC_AUTH_INVALID;
30157e7e2ebcSDaniel P. Berrange #endif
30167e7e2ebcSDaniel P. Berrange     } else {
30177e7e2ebcSDaniel P. Berrange 	vs->auth = vd->auth;
30187e7e2ebcSDaniel P. Berrange #ifdef CONFIG_VNC_TLS
30197e7e2ebcSDaniel P. Berrange 	vs->subauth = vd->subauth;
30207e7e2ebcSDaniel P. Berrange #endif
30217e7e2ebcSDaniel P. Berrange     }
30227e7e2ebcSDaniel P. Berrange 
30237267c094SAnthony Liguori     vs->lossy_rect = g_malloc0(VNC_STAT_ROWS * sizeof (*vs->lossy_rect));
30247d964c9dSCorentin Chary     for (i = 0; i < VNC_STAT_ROWS; ++i) {
30257267c094SAnthony Liguori         vs->lossy_rect[i] = g_malloc0(VNC_STAT_COLS * sizeof (uint8_t));
30267d964c9dSCorentin Chary     }
30273e230dd2SCorentin Chary 
30283e230dd2SCorentin Chary     VNC_DEBUG("New client on socket %d\n", csock);
30290f7b2864SGerd Hoffmann     update_displaychangelistener(&vd->dcl, VNC_REFRESH_INTERVAL_BASE);
3030f9e8caccSStefan Hajnoczi     qemu_set_nonblock(vs->csock);
30317536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
30327536ee4bSTim Hardeck     if (websocket) {
30337536ee4bSTim Hardeck         vs->websocket = 1;
30340057a0d5STim Hardeck #ifdef CONFIG_VNC_TLS
30350057a0d5STim Hardeck         if (vd->tls.x509cert) {
30360057a0d5STim Hardeck             qemu_set_fd_handler2(vs->csock, NULL, vncws_tls_handshake_peek,
30370057a0d5STim Hardeck                                  NULL, vs);
30380057a0d5STim Hardeck         } else
30390057a0d5STim Hardeck #endif /* CONFIG_VNC_TLS */
30400057a0d5STim Hardeck         {
30410057a0d5STim Hardeck             qemu_set_fd_handler2(vs->csock, NULL, vncws_handshake_read,
30420057a0d5STim Hardeck                                  NULL, vs);
30430057a0d5STim Hardeck         }
30447536ee4bSTim Hardeck     } else
30457536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
30467536ee4bSTim Hardeck     {
30473e230dd2SCorentin Chary         qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, vs);
30487536ee4bSTim Hardeck     }
30493e230dd2SCorentin Chary 
30503e230dd2SCorentin Chary     vnc_client_cache_addr(vs);
3051fb6ba0d5SWenchao Xia     vnc_qmp_event(vs, QAPI_EVENT_VNC_CONNECTED);
30528cf36489SGerd Hoffmann     vnc_set_share_mode(vs, VNC_SHARE_MODE_CONNECTING);
30533e230dd2SCorentin Chary 
30547536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
30557536ee4bSTim Hardeck     if (!vs->websocket)
30567536ee4bSTim Hardeck #endif
30577536ee4bSTim Hardeck     {
30587536ee4bSTim Hardeck         vnc_init_state(vs);
30597536ee4bSTim Hardeck     }
3060e5f34cddSGerd Hoffmann 
3061e5f34cddSGerd Hoffmann     if (vd->num_connecting > vd->connections_limit) {
3062e5f34cddSGerd Hoffmann         QTAILQ_FOREACH(vs, &vd->clients, next) {
3063e5f34cddSGerd Hoffmann             if (vs->share_mode == VNC_SHARE_MODE_CONNECTING) {
3064e5f34cddSGerd Hoffmann                 vnc_disconnect_start(vs);
3065e5f34cddSGerd Hoffmann                 return;
3066e5f34cddSGerd Hoffmann             }
3067e5f34cddSGerd Hoffmann         }
3068e5f34cddSGerd Hoffmann     }
30697536ee4bSTim Hardeck }
30707536ee4bSTim Hardeck 
30717536ee4bSTim Hardeck void vnc_init_state(VncState *vs)
30727536ee4bSTim Hardeck {
30736fd8e79aSTim Hardeck     vs->initialized = true;
30747536ee4bSTim Hardeck     VncDisplay *vd = vs->vd;
30757536ee4bSTim Hardeck 
30763e230dd2SCorentin Chary     vs->last_x = -1;
30773e230dd2SCorentin Chary     vs->last_y = -1;
30783e230dd2SCorentin Chary 
30793e230dd2SCorentin Chary     vs->as.freq = 44100;
30803e230dd2SCorentin Chary     vs->as.nchannels = 2;
30813e230dd2SCorentin Chary     vs->as.fmt = AUD_FMT_S16;
30823e230dd2SCorentin Chary     vs->as.endianness = 0;
30833e230dd2SCorentin Chary 
3084bd023f95SCorentin Chary     qemu_mutex_init(&vs->output_mutex);
3085175b2a6eSCorentin Chary     vs->bh = qemu_bh_new(vnc_jobs_bh, vs);
3086bd023f95SCorentin Chary 
3087e5f34cddSGerd Hoffmann     QTAILQ_INSERT_TAIL(&vd->clients, vs, next);
30883e230dd2SCorentin Chary 
30891d0d59feSGerd Hoffmann     graphic_hw_update(vd->dcl.con);
30903e230dd2SCorentin Chary 
30913e230dd2SCorentin Chary     vnc_write(vs, "RFB 003.008\n", 12);
30923e230dd2SCorentin Chary     vnc_flush(vs);
30933e230dd2SCorentin Chary     vnc_read_when(vs, protocol_version, 12);
30943e230dd2SCorentin Chary     reset_keys(vs);
30953e230dd2SCorentin Chary     if (vs->vd->lock_key_sync)
30963e230dd2SCorentin Chary         vs->led = qemu_add_led_event_handler(kbd_leds, vs);
30973e230dd2SCorentin Chary 
30983e230dd2SCorentin Chary     vs->mouse_mode_notifier.notify = check_pointer_type_change;
30993e230dd2SCorentin Chary     qemu_add_mouse_mode_change_notifier(&vs->mouse_mode_notifier);
31003e230dd2SCorentin Chary 
31013e230dd2SCorentin Chary     /* vs might be free()ed here */
31023e230dd2SCorentin Chary }
31033e230dd2SCorentin Chary 
31047536ee4bSTim Hardeck static void vnc_listen_read(void *opaque, bool websocket)
31053e230dd2SCorentin Chary {
31063e230dd2SCorentin Chary     VncDisplay *vs = opaque;
31073e230dd2SCorentin Chary     struct sockaddr_in addr;
31083e230dd2SCorentin Chary     socklen_t addrlen = sizeof(addr);
31097536ee4bSTim Hardeck     int csock;
31103e230dd2SCorentin Chary 
31113e230dd2SCorentin Chary     /* Catch-up */
31121d0d59feSGerd Hoffmann     graphic_hw_update(vs->dcl.con);
31137536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
31147536ee4bSTim Hardeck     if (websocket) {
31157536ee4bSTim Hardeck         csock = qemu_accept(vs->lwebsock, (struct sockaddr *)&addr, &addrlen);
31167536ee4bSTim Hardeck     } else
31177536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
31187536ee4bSTim Hardeck     {
31197536ee4bSTim Hardeck         csock = qemu_accept(vs->lsock, (struct sockaddr *)&addr, &addrlen);
31207536ee4bSTim Hardeck     }
31213e230dd2SCorentin Chary 
31223e230dd2SCorentin Chary     if (csock != -1) {
312386152436SPeter Lieven         socket_set_nodelay(csock);
31242c8cf549SMichael Tokarev         vnc_connect(vs, csock, false, websocket);
31253e230dd2SCorentin Chary     }
31263e230dd2SCorentin Chary }
31273e230dd2SCorentin Chary 
31287536ee4bSTim Hardeck static void vnc_listen_regular_read(void *opaque)
31297536ee4bSTim Hardeck {
31302c8cf549SMichael Tokarev     vnc_listen_read(opaque, false);
31317536ee4bSTim Hardeck }
31327536ee4bSTim Hardeck 
31337536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
31347536ee4bSTim Hardeck static void vnc_listen_websocket_read(void *opaque)
31357536ee4bSTim Hardeck {
31362c8cf549SMichael Tokarev     vnc_listen_read(opaque, true);
31377536ee4bSTim Hardeck }
31387536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
31397536ee4bSTim Hardeck 
31407c20b4a3SGerd Hoffmann static const DisplayChangeListenerOps dcl_ops = {
31417c20b4a3SGerd Hoffmann     .dpy_name             = "vnc",
31420f7b2864SGerd Hoffmann     .dpy_refresh          = vnc_refresh,
31437c20b4a3SGerd Hoffmann     .dpy_gfx_copy         = vnc_dpy_copy,
31447c20b4a3SGerd Hoffmann     .dpy_gfx_update       = vnc_dpy_update,
3145c12aeb86SGerd Hoffmann     .dpy_gfx_switch       = vnc_dpy_switch,
314634da30afSBenjamin Herrenschmidt     .dpy_gfx_check_format = qemu_pixman_check_format,
31477c20b4a3SGerd Hoffmann     .dpy_mouse_set        = vnc_mouse_set,
31487c20b4a3SGerd Hoffmann     .dpy_cursor_define    = vnc_dpy_cursor_define,
31497c20b4a3SGerd Hoffmann };
31507c20b4a3SGerd Hoffmann 
315114f7143eSGerd Hoffmann void vnc_display_init(const char *id)
31523e230dd2SCorentin Chary {
31534db14629SGerd Hoffmann     VncDisplay *vs;
31544db14629SGerd Hoffmann 
31554db14629SGerd Hoffmann     if (vnc_display_find(id) != NULL) {
31564db14629SGerd Hoffmann         return;
31574db14629SGerd Hoffmann     }
31584db14629SGerd Hoffmann     vs = g_malloc0(sizeof(*vs));
31593e230dd2SCorentin Chary 
316014f7143eSGerd Hoffmann     vs->id = strdup(id);
3161d616ccc5SGerd Hoffmann     QTAILQ_INSERT_TAIL(&vnc_displays, vs, next);
31623e230dd2SCorentin Chary 
31633e230dd2SCorentin Chary     vs->lsock = -1;
31647536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
31657536ee4bSTim Hardeck     vs->lwebsock = -1;
31667536ee4bSTim Hardeck #endif
31673e230dd2SCorentin Chary 
31683e230dd2SCorentin Chary     QTAILQ_INIT(&vs->clients);
31693c9405a0SGerd Hoffmann     vs->expires = TIME_MAX;
31703e230dd2SCorentin Chary 
317140066175SGerd Hoffmann     if (keyboard_layout) {
317240066175SGerd Hoffmann         trace_vnc_key_map_init(keyboard_layout);
31733e230dd2SCorentin Chary         vs->kbd_layout = init_keyboard_layout(name2keysym, keyboard_layout);
317440066175SGerd Hoffmann     } else {
31753e230dd2SCorentin Chary         vs->kbd_layout = init_keyboard_layout(name2keysym, "en-us");
317640066175SGerd Hoffmann     }
31773e230dd2SCorentin Chary 
31783e230dd2SCorentin Chary     if (!vs->kbd_layout)
31793e230dd2SCorentin Chary         exit(1);
31803e230dd2SCorentin Chary 
3181bd023f95SCorentin Chary     qemu_mutex_init(&vs->mutex);
3182bd023f95SCorentin Chary     vnc_start_worker_thread();
3183bd023f95SCorentin Chary 
318421ef45d7SGerd Hoffmann     vs->dcl.ops = &dcl_ops;
31855209089fSGerd Hoffmann     register_displaychangelistener(&vs->dcl);
31863e230dd2SCorentin Chary }
31873e230dd2SCorentin Chary 
31883e230dd2SCorentin Chary 
318914f7143eSGerd Hoffmann static void vnc_display_close(VncDisplay *vs)
31903e230dd2SCorentin Chary {
31913e230dd2SCorentin Chary     if (!vs)
31923e230dd2SCorentin Chary         return;
3193bf7aa45eSGerd Hoffmann     vs->enabled = false;
3194bf7aa45eSGerd Hoffmann     vs->is_unix = false;
31953e230dd2SCorentin Chary     if (vs->lsock != -1) {
31963e230dd2SCorentin Chary         qemu_set_fd_handler2(vs->lsock, NULL, NULL, NULL, NULL);
31973e230dd2SCorentin Chary         close(vs->lsock);
31983e230dd2SCorentin Chary         vs->lsock = -1;
31993e230dd2SCorentin Chary     }
32007536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
3201bf7aa45eSGerd Hoffmann     vs->ws_enabled = false;
32027536ee4bSTim Hardeck     if (vs->lwebsock != -1) {
32037536ee4bSTim Hardeck         qemu_set_fd_handler2(vs->lwebsock, NULL, NULL, NULL, NULL);
32047536ee4bSTim Hardeck         close(vs->lwebsock);
32057536ee4bSTim Hardeck         vs->lwebsock = -1;
32067536ee4bSTim Hardeck     }
32077536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
32083e230dd2SCorentin Chary     vs->auth = VNC_AUTH_INVALID;
32093e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
32103e230dd2SCorentin Chary     vs->subauth = VNC_AUTH_INVALID;
32113e230dd2SCorentin Chary     vs->tls.x509verify = 0;
32123e230dd2SCorentin Chary #endif
32133e230dd2SCorentin Chary }
32143e230dd2SCorentin Chary 
321514f7143eSGerd Hoffmann int vnc_display_password(const char *id, const char *password)
32161cd20f8bSAnthony Liguori {
321714f7143eSGerd Hoffmann     VncDisplay *vs = vnc_display_find(id);
32181cd20f8bSAnthony Liguori 
32191cd20f8bSAnthony Liguori     if (!vs) {
3220a6aa9d3eSLuiz Capitulino         return -EINVAL;
32211cd20f8bSAnthony Liguori     }
3222cf864569SGerd Hoffmann     if (vs->auth == VNC_AUTH_NONE) {
3223cf864569SGerd Hoffmann         error_printf_unless_qmp("If you want use passwords please enable "
3224cf864569SGerd Hoffmann                                 "password auth using '-vnc ${dpy},password'.");
3225cf864569SGerd Hoffmann         return -EINVAL;
32261cd20f8bSAnthony Liguori     }
32271cd20f8bSAnthony Liguori 
32287267c094SAnthony Liguori     g_free(vs->password);
32297267c094SAnthony Liguori     vs->password = g_strdup(password);
3230a6aa9d3eSLuiz Capitulino 
3231a6aa9d3eSLuiz Capitulino     return 0;
32323e230dd2SCorentin Chary }
32333e230dd2SCorentin Chary 
323414f7143eSGerd Hoffmann int vnc_display_pw_expire(const char *id, time_t expires)
32353c9405a0SGerd Hoffmann {
323614f7143eSGerd Hoffmann     VncDisplay *vs = vnc_display_find(id);
32373c9405a0SGerd Hoffmann 
32381643f2b2SGerd Hoffmann     if (!vs) {
32391643f2b2SGerd Hoffmann         return -EINVAL;
32401643f2b2SGerd Hoffmann     }
32411643f2b2SGerd Hoffmann 
32423c9405a0SGerd Hoffmann     vs->expires = expires;
32433c9405a0SGerd Hoffmann     return 0;
32443c9405a0SGerd Hoffmann }
32453c9405a0SGerd Hoffmann 
324614f7143eSGerd Hoffmann char *vnc_display_local_addr(const char *id)
32473e230dd2SCorentin Chary {
324814f7143eSGerd Hoffmann     VncDisplay *vs = vnc_display_find(id);
32493e230dd2SCorentin Chary 
32509e0ff75eSGerd Hoffmann     assert(vs);
32513e230dd2SCorentin Chary     return vnc_socket_local_addr("%s:%s", vs->lsock);
32523e230dd2SCorentin Chary }
32533e230dd2SCorentin Chary 
32544db14629SGerd Hoffmann static QemuOptsList qemu_vnc_opts = {
32554db14629SGerd Hoffmann     .name = "vnc",
32564db14629SGerd Hoffmann     .head = QTAILQ_HEAD_INITIALIZER(qemu_vnc_opts.head),
32574db14629SGerd Hoffmann     .implied_opt_name = "vnc",
32584db14629SGerd Hoffmann     .desc = {
32594db14629SGerd Hoffmann         {
32604db14629SGerd Hoffmann             .name = "vnc",
32614db14629SGerd Hoffmann             .type = QEMU_OPT_STRING,
32624db14629SGerd Hoffmann         },{
32634db14629SGerd Hoffmann             .name = "websocket",
32644db14629SGerd Hoffmann             .type = QEMU_OPT_STRING,
32654db14629SGerd Hoffmann         },{
32664db14629SGerd Hoffmann             .name = "x509",
32674db14629SGerd Hoffmann             .type = QEMU_OPT_STRING,
32684db14629SGerd Hoffmann         },{
32694db14629SGerd Hoffmann             .name = "share",
32704db14629SGerd Hoffmann             .type = QEMU_OPT_STRING,
32714db14629SGerd Hoffmann         },{
32721d0d59feSGerd Hoffmann             .name = "display",
32731d0d59feSGerd Hoffmann             .type = QEMU_OPT_STRING,
32741d0d59feSGerd Hoffmann         },{
32751d0d59feSGerd Hoffmann             .name = "head",
32761d0d59feSGerd Hoffmann             .type = QEMU_OPT_NUMBER,
32771d0d59feSGerd Hoffmann         },{
3278e5f34cddSGerd Hoffmann             .name = "connections",
3279e5f34cddSGerd Hoffmann             .type = QEMU_OPT_NUMBER,
3280e5f34cddSGerd Hoffmann         },{
328188428b7aSGonglei             .name = "to",
328288428b7aSGonglei             .type = QEMU_OPT_NUMBER,
328388428b7aSGonglei         },{
328488428b7aSGonglei             .name = "ipv4",
328588428b7aSGonglei             .type = QEMU_OPT_BOOL,
328688428b7aSGonglei         },{
328788428b7aSGonglei             .name = "ipv6",
328888428b7aSGonglei             .type = QEMU_OPT_BOOL,
328988428b7aSGonglei         },{
32904db14629SGerd Hoffmann             .name = "password",
32914db14629SGerd Hoffmann             .type = QEMU_OPT_BOOL,
32924db14629SGerd Hoffmann         },{
32934db14629SGerd Hoffmann             .name = "reverse",
32944db14629SGerd Hoffmann             .type = QEMU_OPT_BOOL,
32954db14629SGerd Hoffmann         },{
32964db14629SGerd Hoffmann             .name = "lock-key-sync",
32974db14629SGerd Hoffmann             .type = QEMU_OPT_BOOL,
32984db14629SGerd Hoffmann         },{
32994db14629SGerd Hoffmann             .name = "sasl",
33004db14629SGerd Hoffmann             .type = QEMU_OPT_BOOL,
33014db14629SGerd Hoffmann         },{
33024db14629SGerd Hoffmann             .name = "tls",
33034db14629SGerd Hoffmann             .type = QEMU_OPT_BOOL,
33044db14629SGerd Hoffmann         },{
33054db14629SGerd Hoffmann             .name = "x509verify",
33064db14629SGerd Hoffmann             .type = QEMU_OPT_BOOL,
33074db14629SGerd Hoffmann         },{
33084db14629SGerd Hoffmann             .name = "acl",
33094db14629SGerd Hoffmann             .type = QEMU_OPT_BOOL,
33104db14629SGerd Hoffmann         },{
33114db14629SGerd Hoffmann             .name = "lossy",
33124db14629SGerd Hoffmann             .type = QEMU_OPT_BOOL,
33134db14629SGerd Hoffmann         },{
33144db14629SGerd Hoffmann             .name = "non-adaptive",
33154db14629SGerd Hoffmann             .type = QEMU_OPT_BOOL,
33164db14629SGerd Hoffmann         },
33174db14629SGerd Hoffmann         { /* end of list */ }
33184db14629SGerd Hoffmann     },
33194db14629SGerd Hoffmann };
33204db14629SGerd Hoffmann 
33214db14629SGerd Hoffmann void vnc_display_open(const char *id, Error **errp)
33223e230dd2SCorentin Chary {
332314f7143eSGerd Hoffmann     VncDisplay *vs = vnc_display_find(id);
33244db14629SGerd Hoffmann     QemuOpts *opts = qemu_opts_find(&qemu_vnc_opts, id);
3325*e5560329SGerd Hoffmann     QemuOpts *sopts, *wsopts;
3326e2a11d9dSGonglei     const char *share, *device_id;
33271d0d59feSGerd Hoffmann     QemuConsole *con;
3328a2c72de0SGonglei     bool password = false;
3329a2c72de0SGonglei     bool reverse = false;
3330e2a11d9dSGonglei     const char *vnc;
3331e2a11d9dSGonglei     const char *has_to;
3332*e5560329SGerd Hoffmann     char *h;
3333e2a11d9dSGonglei     bool has_ipv4 = false;
3334e2a11d9dSGonglei     bool has_ipv6 = false;
33354478aa76SGerd Hoffmann #ifdef CONFIG_VNC_WS
33364478aa76SGerd Hoffmann     const char *websocket;
33374478aa76SGerd Hoffmann #endif
33383e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
3339a2c72de0SGonglei     bool tls = false, x509 = false;
33404db14629SGerd Hoffmann     const char *path;
33413e230dd2SCorentin Chary #endif
33423e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
3343a2c72de0SGonglei     bool sasl = false;
33443e230dd2SCorentin Chary     int saslErr;
33453e230dd2SCorentin Chary #endif
33462ded6ad7SBlue Swirl #if defined(CONFIG_VNC_TLS) || defined(CONFIG_VNC_SASL)
33473e230dd2SCorentin Chary     int acl = 0;
33482ded6ad7SBlue Swirl #endif
33493e230dd2SCorentin Chary     int lock_key_sync = 1;
33503e230dd2SCorentin Chary 
3351d616ccc5SGerd Hoffmann     if (!vs) {
33522d55f0e8SPaolo Bonzini         error_setg(errp, "VNC display not active");
33532d55f0e8SPaolo Bonzini         return;
33542d55f0e8SPaolo Bonzini     }
335514f7143eSGerd Hoffmann     vnc_display_close(vs);
33564db14629SGerd Hoffmann 
33574db14629SGerd Hoffmann     if (!opts) {
33582d55f0e8SPaolo Bonzini         return;
33594db14629SGerd Hoffmann     }
3360e2a11d9dSGonglei     vnc = qemu_opt_get(opts, "vnc");
3361e2a11d9dSGonglei     if (!vnc || strcmp(vnc, "none") == 0) {
33624db14629SGerd Hoffmann         return;
33634db14629SGerd Hoffmann     }
3364e2a11d9dSGonglei 
3365*e5560329SGerd Hoffmann     sopts = qemu_opts_create(&socket_optslist, NULL, 0, &error_abort);
3366*e5560329SGerd Hoffmann     wsopts = qemu_opts_create(&socket_optslist, NULL, 0, &error_abort);
3367*e5560329SGerd Hoffmann 
3368*e5560329SGerd Hoffmann     h = strrchr(vnc, ':');
3369*e5560329SGerd Hoffmann     if (h) {
3370*e5560329SGerd Hoffmann         char *host = g_strndup(vnc, h - vnc);
3371*e5560329SGerd Hoffmann         qemu_opt_set(sopts, "host", host, &error_abort);
3372*e5560329SGerd Hoffmann         qemu_opt_set(wsopts, "host", host, &error_abort);
3373*e5560329SGerd Hoffmann         qemu_opt_set(sopts, "port", h+1, &error_abort);
3374*e5560329SGerd Hoffmann         g_free(host);
3375*e5560329SGerd Hoffmann     } else {
3376*e5560329SGerd Hoffmann         error_setg(errp, "no vnc port specified");
3377*e5560329SGerd Hoffmann         goto fail;
3378e2a11d9dSGonglei     }
3379*e5560329SGerd Hoffmann 
3380*e5560329SGerd Hoffmann     has_to = qemu_opt_get(opts, "to");
3381e2a11d9dSGonglei     has_ipv4 = qemu_opt_get_bool(opts, "ipv4", false);
3382e2a11d9dSGonglei     has_ipv6 = qemu_opt_get_bool(opts, "ipv6", false);
3383*e5560329SGerd Hoffmann     if (has_to) {
3384*e5560329SGerd Hoffmann         qemu_opt_set(sopts, "to", has_to, &error_abort);
3385*e5560329SGerd Hoffmann         qemu_opt_set(wsopts, "to", has_to, &error_abort);
3386*e5560329SGerd Hoffmann     }
3387*e5560329SGerd Hoffmann     if (has_ipv4) {
3388*e5560329SGerd Hoffmann         qemu_opt_set(sopts, "ipv4", "on", &error_abort);
3389*e5560329SGerd Hoffmann         qemu_opt_set(wsopts, "ipv4", "on", &error_abort);
3390*e5560329SGerd Hoffmann     }
3391*e5560329SGerd Hoffmann     if (has_ipv6) {
3392*e5560329SGerd Hoffmann         qemu_opt_set(sopts, "ipv6", "on", &error_abort);
3393*e5560329SGerd Hoffmann         qemu_opt_set(wsopts, "ipv6", "on", &error_abort);
3394*e5560329SGerd Hoffmann     }
33953e230dd2SCorentin Chary 
33964db14629SGerd Hoffmann     password = qemu_opt_get_bool(opts, "password", false);
33974db14629SGerd Hoffmann     if (password && fips_get_state()) {
33982d55f0e8SPaolo Bonzini         error_setg(errp,
33990f66998fSPaul Moore                    "VNC password auth disabled due to FIPS mode, "
34000f66998fSPaul Moore                    "consider using the VeNCrypt or SASL authentication "
34012d55f0e8SPaolo Bonzini                    "methods as an alternative");
34021ce52c78SPaolo Bonzini         goto fail;
34030f66998fSPaul Moore     }
34044db14629SGerd Hoffmann 
34054db14629SGerd Hoffmann     reverse = qemu_opt_get_bool(opts, "reverse", false);
34064db14629SGerd Hoffmann     lock_key_sync = qemu_opt_get_bool(opts, "lock-key-sync", true);
34073e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
34084db14629SGerd Hoffmann     sasl = qemu_opt_get_bool(opts, "sasl", false);
34093e230dd2SCorentin Chary #endif
34103e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
34114db14629SGerd Hoffmann     tls  = qemu_opt_get_bool(opts, "tls", false);
34124db14629SGerd Hoffmann     path = qemu_opt_get(opts, "x509");
34134db14629SGerd Hoffmann     if (path) {
3414a2c72de0SGonglei         x509 = true;
34154db14629SGerd Hoffmann         vs->tls.x509verify = qemu_opt_get_bool(opts, "x509verify", false);
34163e230dd2SCorentin Chary         if (vnc_tls_set_x509_creds_dir(vs, path) < 0) {
34174db14629SGerd Hoffmann             error_setg(errp, "Failed to find x509 certificates/keys in %s",
34184db14629SGerd Hoffmann                        path);
34191ce52c78SPaolo Bonzini             goto fail;
34203e230dd2SCorentin Chary         }
34213e230dd2SCorentin Chary     }
34223e230dd2SCorentin Chary #endif
34232ded6ad7SBlue Swirl #if defined(CONFIG_VNC_TLS) || defined(CONFIG_VNC_SASL)
34244db14629SGerd Hoffmann     acl = qemu_opt_get_bool(opts, "acl", false);
34252ded6ad7SBlue Swirl #endif
34264db14629SGerd Hoffmann 
34274db14629SGerd Hoffmann     share = qemu_opt_get(opts, "share");
34284db14629SGerd Hoffmann     if (share) {
34294db14629SGerd Hoffmann         if (strcmp(share, "ignore") == 0) {
34308cf36489SGerd Hoffmann             vs->share_policy = VNC_SHARE_POLICY_IGNORE;
34314db14629SGerd Hoffmann         } else if (strcmp(share, "allow-exclusive") == 0) {
34328cf36489SGerd Hoffmann             vs->share_policy = VNC_SHARE_POLICY_ALLOW_EXCLUSIVE;
34334db14629SGerd Hoffmann         } else if (strcmp(share, "force-shared") == 0) {
34348cf36489SGerd Hoffmann             vs->share_policy = VNC_SHARE_POLICY_FORCE_SHARED;
34358cf36489SGerd Hoffmann         } else {
34362d55f0e8SPaolo Bonzini             error_setg(errp, "unknown vnc share= option");
34371ce52c78SPaolo Bonzini             goto fail;
34388cf36489SGerd Hoffmann         }
34394db14629SGerd Hoffmann     } else {
34404db14629SGerd Hoffmann         vs->share_policy = VNC_SHARE_POLICY_ALLOW_EXCLUSIVE;
34413e230dd2SCorentin Chary     }
3442e5f34cddSGerd Hoffmann     vs->connections_limit = qemu_opt_get_number(opts, "connections", 32);
34433e230dd2SCorentin Chary 
34444db14629SGerd Hoffmann  #ifdef CONFIG_VNC_WS
34454db14629SGerd Hoffmann     websocket = qemu_opt_get(opts, "websocket");
34464db14629SGerd Hoffmann     if (websocket) {
3447bf7aa45eSGerd Hoffmann         vs->ws_enabled = true;
3448*e5560329SGerd Hoffmann         qemu_opt_set(wsopts, "port", websocket, &error_abort);
34494db14629SGerd Hoffmann 
34504db14629SGerd Hoffmann     }
34514db14629SGerd Hoffmann #endif /* CONFIG_VNC_WS */
34524db14629SGerd Hoffmann 
34534db14629SGerd Hoffmann #ifdef CONFIG_VNC_JPEG
34544db14629SGerd Hoffmann     vs->lossy = qemu_opt_get_bool(opts, "lossy", false);
34554db14629SGerd Hoffmann #endif
34564db14629SGerd Hoffmann     vs->non_adaptive = qemu_opt_get_bool(opts, "non-adaptive", false);
3457e22492d3SPeter Lieven     /* adaptive updates are only used with tight encoding and
3458e22492d3SPeter Lieven      * if lossy updates are enabled so we can disable all the
3459e22492d3SPeter Lieven      * calculations otherwise */
3460e22492d3SPeter Lieven     if (!vs->lossy) {
3461e22492d3SPeter Lieven         vs->non_adaptive = true;
3462e22492d3SPeter Lieven     }
3463e22492d3SPeter Lieven 
34643e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
34653e230dd2SCorentin Chary     if (acl && x509 && vs->tls.x509verify) {
3466c8496408SGerd Hoffmann         char *aclname;
3467c8496408SGerd Hoffmann 
3468c8496408SGerd Hoffmann         if (strcmp(vs->id, "default") == 0) {
3469c8496408SGerd Hoffmann             aclname = g_strdup("vnc.x509dname");
3470c8496408SGerd Hoffmann         } else {
3471c8496408SGerd Hoffmann             aclname = g_strdup_printf("vnc.%s.x509dname", vs->id);
3472c8496408SGerd Hoffmann         }
3473c8496408SGerd Hoffmann         vs->tls.acl = qemu_acl_init(aclname);
3474c8496408SGerd Hoffmann         if (!vs->tls.acl) {
34753e230dd2SCorentin Chary             fprintf(stderr, "Failed to create x509 dname ACL\n");
34763e230dd2SCorentin Chary             exit(1);
34773e230dd2SCorentin Chary         }
3478c8496408SGerd Hoffmann         g_free(aclname);
34793e230dd2SCorentin Chary     }
34803e230dd2SCorentin Chary #endif
34813e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
34823e230dd2SCorentin Chary     if (acl && sasl) {
3483c8496408SGerd Hoffmann         char *aclname;
3484c8496408SGerd Hoffmann 
3485c8496408SGerd Hoffmann         if (strcmp(vs->id, "default") == 0) {
3486c8496408SGerd Hoffmann             aclname = g_strdup("vnc.username");
3487c8496408SGerd Hoffmann         } else {
3488c8496408SGerd Hoffmann             aclname = g_strdup_printf("vnc.%s.username", vs->id);
3489c8496408SGerd Hoffmann         }
3490c8496408SGerd Hoffmann         vs->sasl.acl = qemu_acl_init(aclname);
3491c8496408SGerd Hoffmann         if (!vs->sasl.acl) {
34923e230dd2SCorentin Chary             fprintf(stderr, "Failed to create username ACL\n");
34933e230dd2SCorentin Chary             exit(1);
34943e230dd2SCorentin Chary         }
3495c8496408SGerd Hoffmann         g_free(aclname);
34963e230dd2SCorentin Chary     }
34973e230dd2SCorentin Chary #endif
34983e230dd2SCorentin Chary 
34993e230dd2SCorentin Chary     /*
35003e230dd2SCorentin Chary      * Combinations we support here:
35013e230dd2SCorentin Chary      *
35023e230dd2SCorentin Chary      *  - no-auth                (clear text, no auth)
35033e230dd2SCorentin Chary      *  - password               (clear text, weak auth)
35043e230dd2SCorentin Chary      *  - sasl                   (encrypt, good auth *IF* using Kerberos via GSSAPI)
35053e230dd2SCorentin Chary      *  - tls                    (encrypt, weak anonymous creds, no auth)
35063e230dd2SCorentin Chary      *  - tls + password         (encrypt, weak anonymous creds, weak auth)
35073e230dd2SCorentin Chary      *  - tls + sasl             (encrypt, weak anonymous creds, good auth)
35083e230dd2SCorentin Chary      *  - tls + x509             (encrypt, good x509 creds, no auth)
35093e230dd2SCorentin Chary      *  - tls + x509 + password  (encrypt, good x509 creds, weak auth)
35103e230dd2SCorentin Chary      *  - tls + x509 + sasl      (encrypt, good x509 creds, good auth)
35113e230dd2SCorentin Chary      *
35123e230dd2SCorentin Chary      * NB1. TLS is a stackable auth scheme.
35133e230dd2SCorentin Chary      * NB2. the x509 schemes have option to validate a client cert dname
35143e230dd2SCorentin Chary      */
35153e230dd2SCorentin Chary     if (password) {
35163e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
35173e230dd2SCorentin Chary         if (tls) {
35183e230dd2SCorentin Chary             vs->auth = VNC_AUTH_VENCRYPT;
35193e230dd2SCorentin Chary             if (x509) {
35203e230dd2SCorentin Chary                 VNC_DEBUG("Initializing VNC server with x509 password auth\n");
35213e230dd2SCorentin Chary                 vs->subauth = VNC_AUTH_VENCRYPT_X509VNC;
35223e230dd2SCorentin Chary             } else {
35233e230dd2SCorentin Chary                 VNC_DEBUG("Initializing VNC server with TLS password auth\n");
35243e230dd2SCorentin Chary                 vs->subauth = VNC_AUTH_VENCRYPT_TLSVNC;
35253e230dd2SCorentin Chary             }
35263e230dd2SCorentin Chary         } else {
35273e230dd2SCorentin Chary #endif /* CONFIG_VNC_TLS */
35283e230dd2SCorentin Chary             VNC_DEBUG("Initializing VNC server with password auth\n");
35293e230dd2SCorentin Chary             vs->auth = VNC_AUTH_VNC;
35303e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
35313e230dd2SCorentin Chary             vs->subauth = VNC_AUTH_INVALID;
35323e230dd2SCorentin Chary         }
35333e230dd2SCorentin Chary #endif /* CONFIG_VNC_TLS */
35343e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
35353e230dd2SCorentin Chary     } else if (sasl) {
35363e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
35373e230dd2SCorentin Chary         if (tls) {
35383e230dd2SCorentin Chary             vs->auth = VNC_AUTH_VENCRYPT;
35393e230dd2SCorentin Chary             if (x509) {
35403e230dd2SCorentin Chary                 VNC_DEBUG("Initializing VNC server with x509 SASL auth\n");
35413e230dd2SCorentin Chary                 vs->subauth = VNC_AUTH_VENCRYPT_X509SASL;
35423e230dd2SCorentin Chary             } else {
35433e230dd2SCorentin Chary                 VNC_DEBUG("Initializing VNC server with TLS SASL auth\n");
35443e230dd2SCorentin Chary                 vs->subauth = VNC_AUTH_VENCRYPT_TLSSASL;
35453e230dd2SCorentin Chary             }
35463e230dd2SCorentin Chary         } else {
35473e230dd2SCorentin Chary #endif /* CONFIG_VNC_TLS */
35483e230dd2SCorentin Chary             VNC_DEBUG("Initializing VNC server with SASL auth\n");
35493e230dd2SCorentin Chary             vs->auth = VNC_AUTH_SASL;
35503e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
35513e230dd2SCorentin Chary             vs->subauth = VNC_AUTH_INVALID;
35523e230dd2SCorentin Chary         }
35533e230dd2SCorentin Chary #endif /* CONFIG_VNC_TLS */
35543e230dd2SCorentin Chary #endif /* CONFIG_VNC_SASL */
35553e230dd2SCorentin Chary     } else {
35563e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
35573e230dd2SCorentin Chary         if (tls) {
35583e230dd2SCorentin Chary             vs->auth = VNC_AUTH_VENCRYPT;
35593e230dd2SCorentin Chary             if (x509) {
35603e230dd2SCorentin Chary                 VNC_DEBUG("Initializing VNC server with x509 no auth\n");
35613e230dd2SCorentin Chary                 vs->subauth = VNC_AUTH_VENCRYPT_X509NONE;
35623e230dd2SCorentin Chary             } else {
35633e230dd2SCorentin Chary                 VNC_DEBUG("Initializing VNC server with TLS no auth\n");
35643e230dd2SCorentin Chary                 vs->subauth = VNC_AUTH_VENCRYPT_TLSNONE;
35653e230dd2SCorentin Chary             }
35663e230dd2SCorentin Chary         } else {
35673e230dd2SCorentin Chary #endif
35683e230dd2SCorentin Chary             VNC_DEBUG("Initializing VNC server with no auth\n");
35693e230dd2SCorentin Chary             vs->auth = VNC_AUTH_NONE;
35703e230dd2SCorentin Chary #ifdef CONFIG_VNC_TLS
35713e230dd2SCorentin Chary             vs->subauth = VNC_AUTH_INVALID;
35723e230dd2SCorentin Chary         }
35733e230dd2SCorentin Chary #endif
35743e230dd2SCorentin Chary     }
35753e230dd2SCorentin Chary 
35763e230dd2SCorentin Chary #ifdef CONFIG_VNC_SASL
35773e230dd2SCorentin Chary     if ((saslErr = sasl_server_init(NULL, "qemu")) != SASL_OK) {
35782d55f0e8SPaolo Bonzini         error_setg(errp, "Failed to initialize SASL auth: %s",
35793e230dd2SCorentin Chary                    sasl_errstring(saslErr, NULL, NULL));
35801ce52c78SPaolo Bonzini         goto fail;
35813e230dd2SCorentin Chary     }
35823e230dd2SCorentin Chary #endif
35833e230dd2SCorentin Chary     vs->lock_key_sync = lock_key_sync;
35843e230dd2SCorentin Chary 
35851d0d59feSGerd Hoffmann     device_id = qemu_opt_get(opts, "display");
35861d0d59feSGerd Hoffmann     if (device_id) {
35871d0d59feSGerd Hoffmann         DeviceState *dev;
35881d0d59feSGerd Hoffmann         int head = qemu_opt_get_number(opts, "head", 0);
35891d0d59feSGerd Hoffmann 
35901d0d59feSGerd Hoffmann         dev = qdev_find_recursive(sysbus_get_default(), device_id);
35911d0d59feSGerd Hoffmann         if (dev == NULL) {
35921d0d59feSGerd Hoffmann             error_set(errp, QERR_DEVICE_NOT_FOUND, device_id);
35931d0d59feSGerd Hoffmann             goto fail;
35941d0d59feSGerd Hoffmann         }
35951d0d59feSGerd Hoffmann 
35961d0d59feSGerd Hoffmann         con = qemu_console_lookup_by_device(dev, head);
35971d0d59feSGerd Hoffmann         if (con == NULL) {
35981d0d59feSGerd Hoffmann             error_setg(errp, "Device %s is not bound to a QemuConsole",
35991d0d59feSGerd Hoffmann                        device_id);
36001d0d59feSGerd Hoffmann             goto fail;
36011d0d59feSGerd Hoffmann         }
36021d0d59feSGerd Hoffmann     } else {
36031d0d59feSGerd Hoffmann         con = NULL;
36041d0d59feSGerd Hoffmann     }
36051d0d59feSGerd Hoffmann 
36061d0d59feSGerd Hoffmann     if (con != vs->dcl.con) {
36071d0d59feSGerd Hoffmann         unregister_displaychangelistener(&vs->dcl);
36081d0d59feSGerd Hoffmann         vs->dcl.con = con;
36091d0d59feSGerd Hoffmann         register_displaychangelistener(&vs->dcl);
36101d0d59feSGerd Hoffmann     }
36111d0d59feSGerd Hoffmann 
36123e230dd2SCorentin Chary     if (reverse) {
36133e230dd2SCorentin Chary         /* connect to viewer */
3614007fcd3eSPaolo Bonzini         int csock;
36153e230dd2SCorentin Chary         vs->lsock = -1;
36167536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
36177536ee4bSTim Hardeck         vs->lwebsock = -1;
36187536ee4bSTim Hardeck #endif
3619*e5560329SGerd Hoffmann         if (strncmp(vnc, "unix:", 5) == 0) {
3620*e5560329SGerd Hoffmann             csock = unix_connect(vnc+5, errp);
3621007fcd3eSPaolo Bonzini         } else {
3622*e5560329SGerd Hoffmann             csock = inet_connect(vnc, errp);
36233e230dd2SCorentin Chary         }
3624007fcd3eSPaolo Bonzini         if (csock < 0) {
3625007fcd3eSPaolo Bonzini             goto fail;
3626007fcd3eSPaolo Bonzini         }
36272c8cf549SMichael Tokarev         vnc_connect(vs, csock, false, false);
36283e230dd2SCorentin Chary     } else {
36293e230dd2SCorentin Chary         /* listen for connects */
3630*e5560329SGerd Hoffmann         if (strncmp(vnc, "unix:", 5) == 0) {
3631*e5560329SGerd Hoffmann             vs->lsock = unix_listen(vnc+5, NULL, 0, errp);
3632bf7aa45eSGerd Hoffmann             vs->is_unix = true;
36333e230dd2SCorentin Chary         } else {
3634*e5560329SGerd Hoffmann             vs->lsock = inet_listen_opts(sopts, 5900, errp);
3635c1c1619cSPaolo Bonzini             if (vs->lsock < 0) {
36361ce52c78SPaolo Bonzini                 goto fail;
36371ce52c78SPaolo Bonzini             }
36387536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
3639bf7aa45eSGerd Hoffmann             if (vs->ws_enabled) {
3640*e5560329SGerd Hoffmann                 vs->lwebsock = inet_listen_opts(wsopts, 0, errp);
36417536ee4bSTim Hardeck                 if (vs->lwebsock < 0) {
36427536ee4bSTim Hardeck                     if (vs->lsock) {
36437536ee4bSTim Hardeck                         close(vs->lsock);
36447536ee4bSTim Hardeck                         vs->lsock = -1;
36457536ee4bSTim Hardeck                     }
36467536ee4bSTim Hardeck                     goto fail;
36477536ee4bSTim Hardeck                 }
36487536ee4bSTim Hardeck             }
36497536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
36507536ee4bSTim Hardeck         }
3651bf7aa45eSGerd Hoffmann         vs->enabled = true;
36527536ee4bSTim Hardeck         qemu_set_fd_handler2(vs->lsock, NULL,
36537536ee4bSTim Hardeck                 vnc_listen_regular_read, NULL, vs);
36547536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
3655bf7aa45eSGerd Hoffmann         if (vs->ws_enabled) {
36567536ee4bSTim Hardeck             qemu_set_fd_handler2(vs->lwebsock, NULL,
36577536ee4bSTim Hardeck                     vnc_listen_websocket_read, NULL, vs);
36587536ee4bSTim Hardeck         }
36597536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
36603e230dd2SCorentin Chary     }
3661*e5560329SGerd Hoffmann     qemu_opts_del(sopts);
3662*e5560329SGerd Hoffmann     qemu_opts_del(wsopts);
36632d55f0e8SPaolo Bonzini     return;
36641ce52c78SPaolo Bonzini 
36651ce52c78SPaolo Bonzini fail:
3666*e5560329SGerd Hoffmann     qemu_opts_del(sopts);
3667*e5560329SGerd Hoffmann     qemu_opts_del(wsopts);
3668bf7aa45eSGerd Hoffmann     vs->enabled = false;
36697536ee4bSTim Hardeck #ifdef CONFIG_VNC_WS
3670bf7aa45eSGerd Hoffmann     vs->ws_enabled = false;
36717536ee4bSTim Hardeck #endif /* CONFIG_VNC_WS */
36723e230dd2SCorentin Chary }
367313661089SDaniel P. Berrange 
367414f7143eSGerd Hoffmann void vnc_display_add_client(const char *id, int csock, bool skipauth)
367513661089SDaniel P. Berrange {
367614f7143eSGerd Hoffmann     VncDisplay *vs = vnc_display_find(id);
367713661089SDaniel P. Berrange 
3678d616ccc5SGerd Hoffmann     if (!vs) {
3679d616ccc5SGerd Hoffmann         return;
3680d616ccc5SGerd Hoffmann     }
36812c8cf549SMichael Tokarev     vnc_connect(vs, csock, skipauth, false);
368213661089SDaniel P. Berrange }
36834db14629SGerd Hoffmann 
36849634f4e3SGerd Hoffmann static void vnc_auto_assign_id(QemuOptsList *olist, QemuOpts *opts)
36852779672fSGonglei {
36862779672fSGonglei     int i = 2;
36872779672fSGonglei     char *id;
36882779672fSGonglei 
36892779672fSGonglei     id = g_strdup("default");
36902779672fSGonglei     while (qemu_opts_find(olist, id)) {
36912779672fSGonglei         g_free(id);
36922779672fSGonglei         id = g_strdup_printf("vnc%d", i++);
36932779672fSGonglei     }
36942779672fSGonglei     qemu_opts_set_id(opts, id);
36952779672fSGonglei }
36962779672fSGonglei 
36979634f4e3SGerd Hoffmann QemuOpts *vnc_parse_func(const char *str)
36984db14629SGerd Hoffmann {
36994db14629SGerd Hoffmann     QemuOptsList *olist = qemu_find_opts("vnc");
37009634f4e3SGerd Hoffmann     QemuOpts *opts = qemu_opts_parse(olist, str, 1);
37019634f4e3SGerd Hoffmann     const char *id = qemu_opts_id(opts);
37024db14629SGerd Hoffmann 
37034db14629SGerd Hoffmann     if (!id) {
37044db14629SGerd Hoffmann         /* auto-assign id if not present */
37052779672fSGonglei         vnc_auto_assign_id(olist, opts);
37069634f4e3SGerd Hoffmann     }
37079634f4e3SGerd Hoffmann     return opts;
37084db14629SGerd Hoffmann }
37094db14629SGerd Hoffmann 
37109634f4e3SGerd Hoffmann int vnc_init_func(QemuOpts *opts, void *opaque)
37119634f4e3SGerd Hoffmann {
37129634f4e3SGerd Hoffmann     Error *local_err = NULL;
37139634f4e3SGerd Hoffmann     char *id = (char *)qemu_opts_id(opts);
37149634f4e3SGerd Hoffmann 
37159634f4e3SGerd Hoffmann     assert(id);
37164db14629SGerd Hoffmann     vnc_display_init(id);
37174db14629SGerd Hoffmann     vnc_display_open(id, &local_err);
37184db14629SGerd Hoffmann     if (local_err != NULL) {
37194db14629SGerd Hoffmann         error_report("Failed to start VNC server on `%s': %s",
37204db14629SGerd Hoffmann                      qemu_opt_get(opts, "display"),
37214db14629SGerd Hoffmann                      error_get_pretty(local_err));
37224db14629SGerd Hoffmann         error_free(local_err);
37234db14629SGerd Hoffmann         exit(1);
37244db14629SGerd Hoffmann     }
37254db14629SGerd Hoffmann     return 0;
37264db14629SGerd Hoffmann }
37274db14629SGerd Hoffmann 
37284db14629SGerd Hoffmann static void vnc_register_config(void)
37294db14629SGerd Hoffmann {
37304db14629SGerd Hoffmann     qemu_add_opts(&qemu_vnc_opts);
37314db14629SGerd Hoffmann }
37324db14629SGerd Hoffmann machine_init(vnc_register_config);
3733