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
26static 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
43gboolean 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
83static 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
101gboolean 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