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 #include "qemu/bswap.h" 26 #include "trace.h" 27 28 static void vncws_tls_handshake_done(QIOTask *task, 29 gpointer user_data) 30 { 31 VncState *vs = user_data; 32 Error *err = NULL; 33 34 if (qio_task_propagate_error(task, &err)) { 35 VNC_DEBUG("Handshake failed %s\n", error_get_pretty(err)); 36 vnc_client_error(vs); 37 error_free(err); 38 } else { 39 VNC_DEBUG("TLS handshake complete, starting websocket handshake\n"); 40 if (vs->ioc_tag) { 41 g_source_remove(vs->ioc_tag); 42 } 43 vs->ioc_tag = qio_channel_add_watch(vs->ioc, 44 G_IO_IN | G_IO_HUP | G_IO_ERR, 45 vncws_handshake_io, vs, NULL); 46 } 47 } 48 49 50 gboolean vncws_tls_handshake_io(QIOChannel *ioc G_GNUC_UNUSED, 51 GIOCondition condition, 52 void *opaque) 53 { 54 VncState *vs = opaque; 55 QIOChannelTLS *tls; 56 Error *err = NULL; 57 58 if (vs->ioc_tag) { 59 g_source_remove(vs->ioc_tag); 60 vs->ioc_tag = 0; 61 } 62 63 if (condition & (G_IO_HUP | G_IO_ERR)) { 64 vnc_client_error(vs); 65 return TRUE; 66 } 67 68 tls = qio_channel_tls_new_server( 69 vs->ioc, 70 vs->vd->tlscreds, 71 vs->vd->tlsauthzid, 72 &err); 73 if (!tls) { 74 VNC_DEBUG("Failed to setup TLS %s\n", error_get_pretty(err)); 75 error_free(err); 76 vnc_client_error(vs); 77 return TRUE; 78 } 79 80 qio_channel_set_name(QIO_CHANNEL(tls), "vnc-ws-server-tls"); 81 82 object_unref(OBJECT(vs->ioc)); 83 vs->ioc = QIO_CHANNEL(tls); 84 trace_vnc_client_io_wrap(vs, vs->ioc, "tls"); 85 vs->tls = qio_channel_tls_get_session(tls); 86 87 qio_channel_tls_handshake(tls, 88 vncws_tls_handshake_done, 89 vs, 90 NULL, 91 NULL); 92 93 return TRUE; 94 } 95 96 97 static void vncws_handshake_done(QIOTask *task, 98 gpointer user_data) 99 { 100 VncState *vs = user_data; 101 Error *err = NULL; 102 103 if (qio_task_propagate_error(task, &err)) { 104 VNC_DEBUG("Websock handshake failed %s\n", error_get_pretty(err)); 105 vnc_client_error(vs); 106 error_free(err); 107 } else { 108 VNC_DEBUG("Websock handshake complete, starting VNC protocol\n"); 109 vnc_start_protocol(vs); 110 if (vs->ioc_tag) { 111 g_source_remove(vs->ioc_tag); 112 } 113 vs->ioc_tag = qio_channel_add_watch( 114 vs->ioc, G_IO_IN | G_IO_HUP | G_IO_ERR, 115 vnc_client_io, vs, NULL); 116 } 117 } 118 119 120 gboolean vncws_handshake_io(QIOChannel *ioc G_GNUC_UNUSED, 121 GIOCondition condition, 122 void *opaque) 123 { 124 VncState *vs = opaque; 125 QIOChannelWebsock *wioc; 126 127 if (vs->ioc_tag) { 128 g_source_remove(vs->ioc_tag); 129 vs->ioc_tag = 0; 130 } 131 132 if (condition & (G_IO_HUP | G_IO_ERR)) { 133 vnc_client_error(vs); 134 return TRUE; 135 } 136 137 wioc = qio_channel_websock_new_server(vs->ioc); 138 qio_channel_set_name(QIO_CHANNEL(wioc), "vnc-ws-server-websock"); 139 140 object_unref(OBJECT(vs->ioc)); 141 vs->ioc = QIO_CHANNEL(wioc); 142 trace_vnc_client_io_wrap(vs, vs->ioc, "websock"); 143 144 qio_channel_websock_handshake(wioc, 145 vncws_handshake_done, 146 vs, 147 NULL); 148 149 return TRUE; 150 } 151