1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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
28static 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(
44 QIO_CHANNEL(vs->ioc), G_IO_IN, vncws_handshake_io, vs, NULL);
45 }
46}
47
48
49gboolean vncws_tls_handshake_io(QIOChannel *ioc G_GNUC_UNUSED,
50 GIOCondition condition G_GNUC_UNUSED,
51 void *opaque)
52{
53 VncState *vs = opaque;
54 QIOChannelTLS *tls;
55 Error *err = NULL;
56
57 if (vs->ioc_tag) {
58 g_source_remove(vs->ioc_tag);
59 vs->ioc_tag = 0;
60 }
61
62 tls = qio_channel_tls_new_server(
63 vs->ioc,
64 vs->vd->tlscreds,
65 vs->vd->tlsauthzid,
66 &err);
67 if (!tls) {
68 VNC_DEBUG("Failed to setup TLS %s\n", error_get_pretty(err));
69 error_free(err);
70 vnc_client_error(vs);
71 return TRUE;
72 }
73
74 qio_channel_set_name(QIO_CHANNEL(tls), "vnc-ws-server-tls");
75
76 object_unref(OBJECT(vs->ioc));
77 vs->ioc = QIO_CHANNEL(tls);
78 trace_vnc_client_io_wrap(vs, vs->ioc, "tls");
79 vs->tls = qio_channel_tls_get_session(tls);
80
81 qio_channel_tls_handshake(tls,
82 vncws_tls_handshake_done,
83 vs,
84 NULL,
85 NULL);
86
87 return TRUE;
88}
89
90
91static void vncws_handshake_done(QIOTask *task,
92 gpointer user_data)
93{
94 VncState *vs = user_data;
95 Error *err = NULL;
96
97 if (qio_task_propagate_error(task, &err)) {
98 VNC_DEBUG("Websock handshake failed %s\n", error_get_pretty(err));
99 vnc_client_error(vs);
100 error_free(err);
101 } else {
102 VNC_DEBUG("Websock handshake complete, starting VNC protocol\n");
103 vnc_start_protocol(vs);
104 if (vs->ioc_tag) {
105 g_source_remove(vs->ioc_tag);
106 }
107 vs->ioc_tag = qio_channel_add_watch(
108 vs->ioc, G_IO_IN, vnc_client_io, vs, NULL);
109 }
110}
111
112
113gboolean vncws_handshake_io(QIOChannel *ioc G_GNUC_UNUSED,
114 GIOCondition condition G_GNUC_UNUSED,
115 void *opaque)
116{
117 VncState *vs = opaque;
118 QIOChannelWebsock *wioc;
119
120 if (vs->ioc_tag) {
121 g_source_remove(vs->ioc_tag);
122 vs->ioc_tag = 0;
123 }
124
125 wioc = qio_channel_websock_new_server(vs->ioc);
126 qio_channel_set_name(QIO_CHANNEL(wioc), "vnc-ws-server-websock");
127
128 object_unref(OBJECT(vs->ioc));
129 vs->ioc = QIO_CHANNEL(wioc);
130 trace_vnc_client_io_wrap(vs, vs->ioc, "websock");
131
132 qio_channel_websock_handshake(wioc,
133 vncws_handshake_done,
134 vs,
135 NULL);
136
137 return TRUE;
138}
139