1 /* 2 * QEMU VNC display driver: Websockets support 3 * 4 * Copyright (C) 2010 Joel Martin 5 * Copyright (C) 2012 Tim Hardeck 6 * 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. 11 * 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. 16 * 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/>. 19 */ 20 21 #include "qemu/osdep.h" 22 #include "qapi/error.h" 23 #include "vnc.h" 24 #include "io/channel-websock.h" 25 26 static void vncws_tls_handshake_done(Object *source, 27 Error *err, 28 gpointer user_data) 29 { 30 VncState *vs = user_data; 31 32 if (err) { 33 VNC_DEBUG("Handshake failed %s\n", error_get_pretty(err)); 34 vnc_client_error(vs); 35 } else { 36 VNC_DEBUG("TLS handshake complete, starting websocket handshake\n"); 37 vs->ioc_tag = qio_channel_add_watch( 38 QIO_CHANNEL(vs->ioc), G_IO_IN, vncws_handshake_io, vs, NULL); 39 } 40 } 41 42 43 gboolean vncws_tls_handshake_io(QIOChannel *ioc G_GNUC_UNUSED, 44 GIOCondition condition G_GNUC_UNUSED, 45 void *opaque) 46 { 47 VncState *vs = opaque; 48 QIOChannelTLS *tls; 49 Error *err = NULL; 50 51 VNC_DEBUG("TLS Websocket connection required\n"); 52 if (vs->ioc_tag) { 53 g_source_remove(vs->ioc_tag); 54 vs->ioc_tag = 0; 55 } 56 57 tls = qio_channel_tls_new_server( 58 vs->ioc, 59 vs->vd->tlscreds, 60 vs->vd->tlsaclname, 61 &err); 62 if (!tls) { 63 VNC_DEBUG("Failed to setup TLS %s\n", error_get_pretty(err)); 64 error_free(err); 65 vnc_client_error(vs); 66 return TRUE; 67 } 68 69 VNC_DEBUG("Start TLS WS handshake process\n"); 70 object_unref(OBJECT(vs->ioc)); 71 vs->ioc = QIO_CHANNEL(tls); 72 vs->tls = qio_channel_tls_get_session(tls); 73 74 qio_channel_tls_handshake(tls, 75 vncws_tls_handshake_done, 76 vs, 77 NULL); 78 79 return TRUE; 80 } 81 82 83 static void vncws_handshake_done(Object *source, 84 Error *err, 85 gpointer user_data) 86 { 87 VncState *vs = user_data; 88 89 if (err) { 90 VNC_DEBUG("Websock handshake failed %s\n", error_get_pretty(err)); 91 vnc_client_error(vs); 92 } else { 93 VNC_DEBUG("Websock handshake complete, starting VNC protocol\n"); 94 vnc_init_state(vs); 95 vs->ioc_tag = qio_channel_add_watch( 96 vs->ioc, G_IO_IN, vnc_client_io, vs, NULL); 97 } 98 } 99 100 101 gboolean vncws_handshake_io(QIOChannel *ioc G_GNUC_UNUSED, 102 GIOCondition condition G_GNUC_UNUSED, 103 void *opaque) 104 { 105 VncState *vs = opaque; 106 QIOChannelWebsock *wioc; 107 108 VNC_DEBUG("Websocket negotiate starting\n"); 109 if (vs->ioc_tag) { 110 g_source_remove(vs->ioc_tag); 111 vs->ioc_tag = 0; 112 } 113 114 wioc = qio_channel_websock_new_server(vs->ioc); 115 116 object_unref(OBJECT(vs->ioc)); 117 vs->ioc = QIO_CHANNEL(wioc); 118 119 qio_channel_websock_handshake(wioc, 120 vncws_handshake_done, 121 vs, 122 NULL); 123 124 return TRUE; 125 } 126