ui: avoid crash if vnc client disconnects with writes pending
[qemu/ar7.git] / ui / vnc-ws.c
blob3bac46e7741089ac8156a91633962cf93eba784e
1 /*
2 * QEMU VNC display driver: Websockets support
4 * Copyright (C) 2010 Joel Martin
5 * Copyright (C) 2012 Tim Hardeck
7 * This is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation; either version 2 of the License, or
10 * (at your option) any later version.
12 * This software is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with this software; if not, see <http://www.gnu.org/licenses/>.
21 #include "qemu/osdep.h"
22 #include "qapi/error.h"
23 #include "vnc.h"
24 #include "io/channel-websock.h"
25 #include "qemu/bswap.h"
27 static void vncws_tls_handshake_done(Object *source,
28 Error *err,
29 gpointer user_data)
31 VncState *vs = user_data;
33 if (err) {
34 VNC_DEBUG("Handshake failed %s\n", error_get_pretty(err));
35 vnc_client_error(vs);
36 } else {
37 VNC_DEBUG("TLS handshake complete, starting websocket handshake\n");
38 vs->ioc_tag = qio_channel_add_watch(
39 QIO_CHANNEL(vs->ioc), G_IO_IN, vncws_handshake_io, vs, NULL);
44 gboolean vncws_tls_handshake_io(QIOChannel *ioc G_GNUC_UNUSED,
45 GIOCondition condition G_GNUC_UNUSED,
46 void *opaque)
48 VncState *vs = opaque;
49 QIOChannelTLS *tls;
50 Error *err = NULL;
52 VNC_DEBUG("TLS Websocket connection required\n");
53 if (vs->ioc_tag) {
54 g_source_remove(vs->ioc_tag);
55 vs->ioc_tag = 0;
58 tls = qio_channel_tls_new_server(
59 vs->ioc,
60 vs->vd->tlscreds,
61 vs->vd->tlsaclname,
62 &err);
63 if (!tls) {
64 VNC_DEBUG("Failed to setup TLS %s\n", error_get_pretty(err));
65 error_free(err);
66 vnc_client_error(vs);
67 return TRUE;
70 VNC_DEBUG("Start TLS WS handshake process\n");
71 object_unref(OBJECT(vs->ioc));
72 vs->ioc = QIO_CHANNEL(tls);
73 vs->tls = qio_channel_tls_get_session(tls);
75 qio_channel_tls_handshake(tls,
76 vncws_tls_handshake_done,
77 vs,
78 NULL);
80 return TRUE;
84 static void vncws_handshake_done(Object *source,
85 Error *err,
86 gpointer user_data)
88 VncState *vs = user_data;
90 if (err) {
91 VNC_DEBUG("Websock handshake failed %s\n", error_get_pretty(err));
92 vnc_client_error(vs);
93 } else {
94 VNC_DEBUG("Websock handshake complete, starting VNC protocol\n");
95 vnc_init_state(vs);
96 vs->ioc_tag = qio_channel_add_watch(
97 vs->ioc, G_IO_IN, vnc_client_io, vs, NULL);
102 gboolean vncws_handshake_io(QIOChannel *ioc G_GNUC_UNUSED,
103 GIOCondition condition G_GNUC_UNUSED,
104 void *opaque)
106 VncState *vs = opaque;
107 QIOChannelWebsock *wioc;
109 VNC_DEBUG("Websocket negotiate starting\n");
110 if (vs->ioc_tag) {
111 g_source_remove(vs->ioc_tag);
112 vs->ioc_tag = 0;
115 wioc = qio_channel_websock_new_server(vs->ioc);
117 object_unref(OBJECT(vs->ioc));
118 vs->ioc = QIO_CHANNEL(wioc);
120 qio_channel_websock_handshake(wioc,
121 vncws_handshake_done,
123 NULL);
125 return TRUE;