root / vnc.c @ d78f3995
History | View | Annotate | Download (65.6 kB)
1 | 7d510b8c | bellard | /*
|
---|---|---|---|
2 | 7d510b8c | bellard | * QEMU VNC display driver
|
3 | 5fafdf24 | ths | *
|
4 | 7d510b8c | bellard | * Copyright (C) 2006 Anthony Liguori <anthony@codemonkey.ws>
|
5 | 7d510b8c | bellard | * Copyright (C) 2006 Fabrice Bellard
|
6 | 19a490bf | aliguori | * Copyright (C) 2009 Red Hat, Inc
|
7 | 5fafdf24 | ths | *
|
8 | 7d510b8c | bellard | * Permission is hereby granted, free of charge, to any person obtaining a copy
|
9 | 7d510b8c | bellard | * of this software and associated documentation files (the "Software"), to deal
|
10 | 7d510b8c | bellard | * in the Software without restriction, including without limitation the rights
|
11 | 7d510b8c | bellard | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
12 | 7d510b8c | bellard | * copies of the Software, and to permit persons to whom the Software is
|
13 | 7d510b8c | bellard | * furnished to do so, subject to the following conditions:
|
14 | 7d510b8c | bellard | *
|
15 | 7d510b8c | bellard | * The above copyright notice and this permission notice shall be included in
|
16 | 7d510b8c | bellard | * all copies or substantial portions of the Software.
|
17 | 7d510b8c | bellard | *
|
18 | 7d510b8c | bellard | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
19 | 7d510b8c | bellard | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
20 | 7d510b8c | bellard | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
21 | 7d510b8c | bellard | * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
22 | 7d510b8c | bellard | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
23 | 7d510b8c | bellard | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
24 | 7d510b8c | bellard | * THE SOFTWARE.
|
25 | 7d510b8c | bellard | */
|
26 | 7d510b8c | bellard | |
27 | 19a490bf | aliguori | #include "vnc.h" |
28 | 87ecb68b | pbrook | #include "sysemu.h" |
29 | 6ca957f0 | bellard | #include "qemu_socket.h" |
30 | 87ecb68b | pbrook | #include "qemu-timer.h" |
31 | 76655d6d | aliguori | #include "acl.h" |
32 | 24236869 | bellard | |
33 | 24236869 | bellard | #define VNC_REFRESH_INTERVAL (1000 / 30) |
34 | 24236869 | bellard | |
35 | 24236869 | bellard | #include "vnc_keysym.h" |
36 | 70848515 | ths | #include "d3des.h" |
37 | 70848515 | ths | |
38 | 90a1e3c0 | aliguori | #define count_bits(c, v) { \
|
39 | 90a1e3c0 | aliguori | for (c = 0; v; v >>= 1) \ |
40 | 90a1e3c0 | aliguori | { \ |
41 | 90a1e3c0 | aliguori | c += v & 1; \
|
42 | 90a1e3c0 | aliguori | } \ |
43 | 90a1e3c0 | aliguori | } |
44 | 8d5d2d4c | ths | |
45 | 24236869 | bellard | |
46 | 753b4053 | aliguori | static VncDisplay *vnc_display; /* needed for info vnc */ |
47 | 7d957bd8 | aliguori | static DisplayChangeListener *dcl;
|
48 | a9ce8590 | bellard | |
49 | 1ff7df1a | aliguori | static char *addr_to_string(const char *format, |
50 | 1ff7df1a | aliguori | struct sockaddr_storage *sa,
|
51 | 1ff7df1a | aliguori | socklen_t salen) { |
52 | 1ff7df1a | aliguori | char *addr;
|
53 | 1ff7df1a | aliguori | char host[NI_MAXHOST];
|
54 | 1ff7df1a | aliguori | char serv[NI_MAXSERV];
|
55 | 1ff7df1a | aliguori | int err;
|
56 | 457772e6 | aliguori | size_t addrlen; |
57 | 1ff7df1a | aliguori | |
58 | 1ff7df1a | aliguori | if ((err = getnameinfo((struct sockaddr *)sa, salen, |
59 | 1ff7df1a | aliguori | host, sizeof(host),
|
60 | 1ff7df1a | aliguori | serv, sizeof(serv),
|
61 | 1ff7df1a | aliguori | NI_NUMERICHOST | NI_NUMERICSERV)) != 0) {
|
62 | 1ff7df1a | aliguori | VNC_DEBUG("Cannot resolve address %d: %s\n",
|
63 | 1ff7df1a | aliguori | err, gai_strerror(err)); |
64 | 1ff7df1a | aliguori | return NULL; |
65 | 1ff7df1a | aliguori | } |
66 | 1ff7df1a | aliguori | |
67 | 457772e6 | aliguori | /* Enough for the existing format + the 2 vars we're
|
68 | 457772e6 | aliguori | * subsituting in. */
|
69 | 457772e6 | aliguori | addrlen = strlen(format) + strlen(host) + strlen(serv); |
70 | 457772e6 | aliguori | addr = qemu_malloc(addrlen + 1);
|
71 | 457772e6 | aliguori | snprintf(addr, addrlen, format, host, serv); |
72 | 457772e6 | aliguori | addr[addrlen] = '\0';
|
73 | 1ff7df1a | aliguori | |
74 | 1ff7df1a | aliguori | return addr;
|
75 | 1ff7df1a | aliguori | } |
76 | 1ff7df1a | aliguori | |
77 | 2f9606b3 | aliguori | |
78 | 2f9606b3 | aliguori | char *vnc_socket_local_addr(const char *format, int fd) { |
79 | 1ff7df1a | aliguori | struct sockaddr_storage sa;
|
80 | 1ff7df1a | aliguori | socklen_t salen; |
81 | 1ff7df1a | aliguori | |
82 | 1ff7df1a | aliguori | salen = sizeof(sa);
|
83 | 1ff7df1a | aliguori | if (getsockname(fd, (struct sockaddr*)&sa, &salen) < 0) |
84 | 1ff7df1a | aliguori | return NULL; |
85 | 1ff7df1a | aliguori | |
86 | 1ff7df1a | aliguori | return addr_to_string(format, &sa, salen);
|
87 | 1ff7df1a | aliguori | } |
88 | 1ff7df1a | aliguori | |
89 | 2f9606b3 | aliguori | |
90 | 2f9606b3 | aliguori | char *vnc_socket_remote_addr(const char *format, int fd) { |
91 | 1ff7df1a | aliguori | struct sockaddr_storage sa;
|
92 | 1ff7df1a | aliguori | socklen_t salen; |
93 | 1ff7df1a | aliguori | |
94 | 1ff7df1a | aliguori | salen = sizeof(sa);
|
95 | 1ff7df1a | aliguori | if (getpeername(fd, (struct sockaddr*)&sa, &salen) < 0) |
96 | 1ff7df1a | aliguori | return NULL; |
97 | 1ff7df1a | aliguori | |
98 | 1ff7df1a | aliguori | return addr_to_string(format, &sa, salen);
|
99 | 1ff7df1a | aliguori | } |
100 | 1ff7df1a | aliguori | |
101 | 1ff7df1a | aliguori | static const char *vnc_auth_name(VncDisplay *vd) { |
102 | 1ff7df1a | aliguori | switch (vd->auth) {
|
103 | 1ff7df1a | aliguori | case VNC_AUTH_INVALID:
|
104 | 1ff7df1a | aliguori | return "invalid"; |
105 | 1ff7df1a | aliguori | case VNC_AUTH_NONE:
|
106 | 1ff7df1a | aliguori | return "none"; |
107 | 1ff7df1a | aliguori | case VNC_AUTH_VNC:
|
108 | 1ff7df1a | aliguori | return "vnc"; |
109 | 1ff7df1a | aliguori | case VNC_AUTH_RA2:
|
110 | 1ff7df1a | aliguori | return "ra2"; |
111 | 1ff7df1a | aliguori | case VNC_AUTH_RA2NE:
|
112 | 1ff7df1a | aliguori | return "ra2ne"; |
113 | 1ff7df1a | aliguori | case VNC_AUTH_TIGHT:
|
114 | 1ff7df1a | aliguori | return "tight"; |
115 | 1ff7df1a | aliguori | case VNC_AUTH_ULTRA:
|
116 | 1ff7df1a | aliguori | return "ultra"; |
117 | 1ff7df1a | aliguori | case VNC_AUTH_TLS:
|
118 | 1ff7df1a | aliguori | return "tls"; |
119 | 1ff7df1a | aliguori | case VNC_AUTH_VENCRYPT:
|
120 | 1ff7df1a | aliguori | #ifdef CONFIG_VNC_TLS
|
121 | 1ff7df1a | aliguori | switch (vd->subauth) {
|
122 | 1ff7df1a | aliguori | case VNC_AUTH_VENCRYPT_PLAIN:
|
123 | 1ff7df1a | aliguori | return "vencrypt+plain"; |
124 | 1ff7df1a | aliguori | case VNC_AUTH_VENCRYPT_TLSNONE:
|
125 | 1ff7df1a | aliguori | return "vencrypt+tls+none"; |
126 | 1ff7df1a | aliguori | case VNC_AUTH_VENCRYPT_TLSVNC:
|
127 | 1ff7df1a | aliguori | return "vencrypt+tls+vnc"; |
128 | 1ff7df1a | aliguori | case VNC_AUTH_VENCRYPT_TLSPLAIN:
|
129 | 1ff7df1a | aliguori | return "vencrypt+tls+plain"; |
130 | 1ff7df1a | aliguori | case VNC_AUTH_VENCRYPT_X509NONE:
|
131 | 1ff7df1a | aliguori | return "vencrypt+x509+none"; |
132 | 1ff7df1a | aliguori | case VNC_AUTH_VENCRYPT_X509VNC:
|
133 | 1ff7df1a | aliguori | return "vencrypt+x509+vnc"; |
134 | 1ff7df1a | aliguori | case VNC_AUTH_VENCRYPT_X509PLAIN:
|
135 | 1ff7df1a | aliguori | return "vencrypt+x509+plain"; |
136 | 28a76be8 | aliguori | case VNC_AUTH_VENCRYPT_TLSSASL:
|
137 | 28a76be8 | aliguori | return "vencrypt+tls+sasl"; |
138 | 28a76be8 | aliguori | case VNC_AUTH_VENCRYPT_X509SASL:
|
139 | 28a76be8 | aliguori | return "vencrypt+x509+sasl"; |
140 | 1ff7df1a | aliguori | default:
|
141 | 1ff7df1a | aliguori | return "vencrypt"; |
142 | 1ff7df1a | aliguori | } |
143 | 1ff7df1a | aliguori | #else
|
144 | 1ff7df1a | aliguori | return "vencrypt"; |
145 | 1ff7df1a | aliguori | #endif
|
146 | 2f9606b3 | aliguori | case VNC_AUTH_SASL:
|
147 | 28a76be8 | aliguori | return "sasl"; |
148 | 1ff7df1a | aliguori | } |
149 | 1ff7df1a | aliguori | return "unknown"; |
150 | 1ff7df1a | aliguori | } |
151 | 1ff7df1a | aliguori | |
152 | 1ff7df1a | aliguori | static void do_info_vnc_client(Monitor *mon, VncState *client) |
153 | 1ff7df1a | aliguori | { |
154 | 1ff7df1a | aliguori | char *clientAddr =
|
155 | 1ff7df1a | aliguori | vnc_socket_remote_addr(" address: %s:%s\n",
|
156 | 1ff7df1a | aliguori | client->csock); |
157 | 1ff7df1a | aliguori | if (!clientAddr)
|
158 | 1ff7df1a | aliguori | return;
|
159 | 1ff7df1a | aliguori | |
160 | 1ff7df1a | aliguori | monitor_printf(mon, "Client:\n");
|
161 | 1ff7df1a | aliguori | monitor_printf(mon, "%s", clientAddr);
|
162 | 1ff7df1a | aliguori | free(clientAddr); |
163 | 1263b7d6 | aliguori | |
164 | 1263b7d6 | aliguori | #ifdef CONFIG_VNC_TLS
|
165 | 1263b7d6 | aliguori | if (client->tls.session &&
|
166 | 28a76be8 | aliguori | client->tls.dname) |
167 | 28a76be8 | aliguori | monitor_printf(mon, " x509 dname: %s\n", client->tls.dname);
|
168 | 1263b7d6 | aliguori | else
|
169 | 28a76be8 | aliguori | monitor_printf(mon, " x509 dname: none\n");
|
170 | 1263b7d6 | aliguori | #endif
|
171 | 1263b7d6 | aliguori | #ifdef CONFIG_VNC_SASL
|
172 | 1263b7d6 | aliguori | if (client->sasl.conn &&
|
173 | 28a76be8 | aliguori | client->sasl.username) |
174 | 28a76be8 | aliguori | monitor_printf(mon, " username: %s\n", client->sasl.username);
|
175 | 1263b7d6 | aliguori | else
|
176 | 28a76be8 | aliguori | monitor_printf(mon, " username: none\n");
|
177 | 1263b7d6 | aliguori | #endif
|
178 | 1ff7df1a | aliguori | } |
179 | 1ff7df1a | aliguori | |
180 | 376253ec | aliguori | void do_info_vnc(Monitor *mon)
|
181 | a9ce8590 | bellard | { |
182 | 1ff7df1a | aliguori | if (vnc_display == NULL || vnc_display->display == NULL) { |
183 | 1ff7df1a | aliguori | monitor_printf(mon, "Server: disabled\n");
|
184 | 1ff7df1a | aliguori | } else {
|
185 | 1ff7df1a | aliguori | char *serverAddr = vnc_socket_local_addr(" address: %s:%s\n", |
186 | 1ff7df1a | aliguori | vnc_display->lsock); |
187 | 1ff7df1a | aliguori | |
188 | 1ff7df1a | aliguori | if (!serverAddr)
|
189 | 1ff7df1a | aliguori | return;
|
190 | 1ff7df1a | aliguori | |
191 | 1ff7df1a | aliguori | monitor_printf(mon, "Server:\n");
|
192 | 1ff7df1a | aliguori | monitor_printf(mon, "%s", serverAddr);
|
193 | 1ff7df1a | aliguori | free(serverAddr); |
194 | 1ff7df1a | aliguori | monitor_printf(mon, " auth: %s\n", vnc_auth_name(vnc_display));
|
195 | 1ff7df1a | aliguori | |
196 | 1ff7df1a | aliguori | if (vnc_display->clients) {
|
197 | 1ff7df1a | aliguori | VncState *client = vnc_display->clients; |
198 | 1ff7df1a | aliguori | while (client) {
|
199 | 1ff7df1a | aliguori | do_info_vnc_client(mon, client); |
200 | 1ff7df1a | aliguori | client = client->next; |
201 | 1ff7df1a | aliguori | } |
202 | 1ff7df1a | aliguori | } else {
|
203 | 1ff7df1a | aliguori | monitor_printf(mon, "Client: none\n");
|
204 | 1ff7df1a | aliguori | } |
205 | a9ce8590 | bellard | } |
206 | a9ce8590 | bellard | } |
207 | a9ce8590 | bellard | |
208 | 29fa4ed9 | aliguori | static inline uint32_t vnc_has_feature(VncState *vs, int feature) { |
209 | 29fa4ed9 | aliguori | return (vs->features & (1 << feature)); |
210 | 29fa4ed9 | aliguori | } |
211 | 29fa4ed9 | aliguori | |
212 | 24236869 | bellard | /* TODO
|
213 | 24236869 | bellard | 1) Get the queue working for IO.
|
214 | 24236869 | bellard | 2) there is some weirdness when using the -S option (the screen is grey
|
215 | 24236869 | bellard | and not totally invalidated
|
216 | 24236869 | bellard | 3) resolutions > 1024
|
217 | 24236869 | bellard | */
|
218 | 24236869 | bellard | |
219 | 24236869 | bellard | static void vnc_update_client(void *opaque); |
220 | 24236869 | bellard | |
221 | 753b4053 | aliguori | static void vnc_colordepth(VncState *vs); |
222 | 7eac3a87 | aliguori | |
223 | 99589bdc | bellard | static inline void vnc_set_bit(uint32_t *d, int k) |
224 | 99589bdc | bellard | { |
225 | 99589bdc | bellard | d[k >> 5] |= 1 << (k & 0x1f); |
226 | 99589bdc | bellard | } |
227 | 99589bdc | bellard | |
228 | 99589bdc | bellard | static inline void vnc_clear_bit(uint32_t *d, int k) |
229 | 99589bdc | bellard | { |
230 | 99589bdc | bellard | d[k >> 5] &= ~(1 << (k & 0x1f)); |
231 | 99589bdc | bellard | } |
232 | 99589bdc | bellard | |
233 | 99589bdc | bellard | static inline void vnc_set_bits(uint32_t *d, int n, int nb_words) |
234 | 99589bdc | bellard | { |
235 | 99589bdc | bellard | int j;
|
236 | 99589bdc | bellard | |
237 | 99589bdc | bellard | j = 0;
|
238 | 99589bdc | bellard | while (n >= 32) { |
239 | 99589bdc | bellard | d[j++] = -1;
|
240 | 99589bdc | bellard | n -= 32;
|
241 | 99589bdc | bellard | } |
242 | 5fafdf24 | ths | if (n > 0) |
243 | 99589bdc | bellard | d[j++] = (1 << n) - 1; |
244 | 99589bdc | bellard | while (j < nb_words)
|
245 | 99589bdc | bellard | d[j++] = 0;
|
246 | 99589bdc | bellard | } |
247 | 99589bdc | bellard | |
248 | 99589bdc | bellard | static inline int vnc_get_bit(const uint32_t *d, int k) |
249 | 99589bdc | bellard | { |
250 | 99589bdc | bellard | return (d[k >> 5] >> (k & 0x1f)) & 1; |
251 | 99589bdc | bellard | } |
252 | 99589bdc | bellard | |
253 | 5fafdf24 | ths | static inline int vnc_and_bits(const uint32_t *d1, const uint32_t *d2, |
254 | 99589bdc | bellard | int nb_words)
|
255 | 99589bdc | bellard | { |
256 | 99589bdc | bellard | int i;
|
257 | 99589bdc | bellard | for(i = 0; i < nb_words; i++) { |
258 | 99589bdc | bellard | if ((d1[i] & d2[i]) != 0) |
259 | 99589bdc | bellard | return 1; |
260 | 99589bdc | bellard | } |
261 | 99589bdc | bellard | return 0; |
262 | 99589bdc | bellard | } |
263 | 99589bdc | bellard | |
264 | 753b4053 | aliguori | static void vnc_update(VncState *vs, int x, int y, int w, int h) |
265 | 24236869 | bellard | { |
266 | 24236869 | bellard | int i;
|
267 | 24236869 | bellard | |
268 | 24236869 | bellard | h += y; |
269 | 24236869 | bellard | |
270 | 0486e8a7 | balrog | /* round x down to ensure the loop only spans one 16-pixel block per,
|
271 | 0486e8a7 | balrog | iteration. otherwise, if (x % 16) != 0, the last iteration may span
|
272 | 0486e8a7 | balrog | two 16-pixel blocks but we only mark the first as dirty
|
273 | 0486e8a7 | balrog | */
|
274 | 0486e8a7 | balrog | w += (x % 16);
|
275 | 0486e8a7 | balrog | x -= (x % 16);
|
276 | 0486e8a7 | balrog | |
277 | 6cec5487 | aliguori | x = MIN(x, vs->serverds.width); |
278 | 6cec5487 | aliguori | y = MIN(y, vs->serverds.height); |
279 | 6cec5487 | aliguori | w = MIN(x + w, vs->serverds.width) - x; |
280 | 6cec5487 | aliguori | h = MIN(h, vs->serverds.height); |
281 | 788abf8e | balrog | |
282 | 24236869 | bellard | for (; y < h; y++)
|
283 | 28a76be8 | aliguori | for (i = 0; i < w; i += 16) |
284 | 28a76be8 | aliguori | vnc_set_bit(vs->dirty_row[y], (x + i) / 16);
|
285 | 24236869 | bellard | } |
286 | 24236869 | bellard | |
287 | 753b4053 | aliguori | static void vnc_dpy_update(DisplayState *ds, int x, int y, int w, int h) |
288 | 753b4053 | aliguori | { |
289 | 753b4053 | aliguori | VncDisplay *vd = ds->opaque; |
290 | 753b4053 | aliguori | VncState *vs = vd->clients; |
291 | 753b4053 | aliguori | while (vs != NULL) { |
292 | 753b4053 | aliguori | vnc_update(vs, x, y, w, h); |
293 | 753b4053 | aliguori | vs = vs->next; |
294 | 753b4053 | aliguori | } |
295 | 753b4053 | aliguori | } |
296 | 753b4053 | aliguori | |
297 | 24236869 | bellard | static void vnc_framebuffer_update(VncState *vs, int x, int y, int w, int h, |
298 | 28a76be8 | aliguori | int32_t encoding) |
299 | 24236869 | bellard | { |
300 | 24236869 | bellard | vnc_write_u16(vs, x); |
301 | 24236869 | bellard | vnc_write_u16(vs, y); |
302 | 24236869 | bellard | vnc_write_u16(vs, w); |
303 | 24236869 | bellard | vnc_write_u16(vs, h); |
304 | 24236869 | bellard | |
305 | 24236869 | bellard | vnc_write_s32(vs, encoding); |
306 | 24236869 | bellard | } |
307 | 24236869 | bellard | |
308 | 2f9606b3 | aliguori | void buffer_reserve(Buffer *buffer, size_t len)
|
309 | 89064286 | aliguori | { |
310 | 89064286 | aliguori | if ((buffer->capacity - buffer->offset) < len) {
|
311 | 28a76be8 | aliguori | buffer->capacity += (len + 1024);
|
312 | 28a76be8 | aliguori | buffer->buffer = qemu_realloc(buffer->buffer, buffer->capacity); |
313 | 28a76be8 | aliguori | if (buffer->buffer == NULL) { |
314 | 28a76be8 | aliguori | fprintf(stderr, "vnc: out of memory\n");
|
315 | 28a76be8 | aliguori | exit(1);
|
316 | 28a76be8 | aliguori | } |
317 | 89064286 | aliguori | } |
318 | 89064286 | aliguori | } |
319 | 89064286 | aliguori | |
320 | 2f9606b3 | aliguori | int buffer_empty(Buffer *buffer)
|
321 | 89064286 | aliguori | { |
322 | 89064286 | aliguori | return buffer->offset == 0; |
323 | 89064286 | aliguori | } |
324 | 89064286 | aliguori | |
325 | 2f9606b3 | aliguori | uint8_t *buffer_end(Buffer *buffer) |
326 | 89064286 | aliguori | { |
327 | 89064286 | aliguori | return buffer->buffer + buffer->offset;
|
328 | 89064286 | aliguori | } |
329 | 89064286 | aliguori | |
330 | 2f9606b3 | aliguori | void buffer_reset(Buffer *buffer)
|
331 | 89064286 | aliguori | { |
332 | 28a76be8 | aliguori | buffer->offset = 0;
|
333 | 89064286 | aliguori | } |
334 | 89064286 | aliguori | |
335 | 2f9606b3 | aliguori | void buffer_append(Buffer *buffer, const void *data, size_t len) |
336 | 89064286 | aliguori | { |
337 | 89064286 | aliguori | memcpy(buffer->buffer + buffer->offset, data, len); |
338 | 89064286 | aliguori | buffer->offset += len; |
339 | 89064286 | aliguori | } |
340 | 89064286 | aliguori | |
341 | 753b4053 | aliguori | static void vnc_resize(VncState *vs) |
342 | 24236869 | bellard | { |
343 | 753b4053 | aliguori | DisplayState *ds = vs->ds; |
344 | 753b4053 | aliguori | |
345 | 73e14b62 | ths | int size_changed;
|
346 | 24236869 | bellard | |
347 | 7d957bd8 | aliguori | vs->old_data = qemu_realloc(vs->old_data, ds_get_linesize(ds) * ds_get_height(ds)); |
348 | 24236869 | bellard | |
349 | 7d957bd8 | aliguori | if (vs->old_data == NULL) { |
350 | 28a76be8 | aliguori | fprintf(stderr, "vnc: memory allocation failed\n");
|
351 | 28a76be8 | aliguori | exit(1);
|
352 | 24236869 | bellard | } |
353 | 24236869 | bellard | |
354 | 6cec5487 | aliguori | if (ds_get_bytes_per_pixel(ds) != vs->serverds.pf.bytes_per_pixel)
|
355 | a528b80c | balrog | console_color_init(ds); |
356 | 753b4053 | aliguori | vnc_colordepth(vs); |
357 | 6cec5487 | aliguori | size_changed = ds_get_width(ds) != vs->serverds.width || |
358 | 6cec5487 | aliguori | ds_get_height(ds) != vs->serverds.height; |
359 | 6cec5487 | aliguori | vs->serverds = *(ds->surface); |
360 | b94eb43f | balrog | if (size_changed) {
|
361 | 29fa4ed9 | aliguori | if (vs->csock != -1 && vnc_has_feature(vs, VNC_FEATURE_RESIZE)) { |
362 | b94eb43f | balrog | vnc_write_u8(vs, 0); /* msg id */ |
363 | b94eb43f | balrog | vnc_write_u8(vs, 0);
|
364 | b94eb43f | balrog | vnc_write_u16(vs, 1); /* number of rects */ |
365 | 29fa4ed9 | aliguori | vnc_framebuffer_update(vs, 0, 0, ds_get_width(ds), ds_get_height(ds), |
366 | 29fa4ed9 | aliguori | VNC_ENCODING_DESKTOPRESIZE); |
367 | b94eb43f | balrog | vnc_flush(vs); |
368 | b94eb43f | balrog | } |
369 | 24236869 | bellard | } |
370 | 8bba5c81 | balrog | |
371 | 8bba5c81 | balrog | memset(vs->dirty_row, 0xFF, sizeof(vs->dirty_row)); |
372 | 0e1f5a0c | aliguori | memset(vs->old_data, 42, ds_get_linesize(vs->ds) * ds_get_height(vs->ds));
|
373 | 24236869 | bellard | } |
374 | 24236869 | bellard | |
375 | 753b4053 | aliguori | static void vnc_dpy_resize(DisplayState *ds) |
376 | 753b4053 | aliguori | { |
377 | 753b4053 | aliguori | VncDisplay *vd = ds->opaque; |
378 | 753b4053 | aliguori | VncState *vs = vd->clients; |
379 | 753b4053 | aliguori | while (vs != NULL) { |
380 | 753b4053 | aliguori | vnc_resize(vs); |
381 | 753b4053 | aliguori | vs = vs->next; |
382 | 753b4053 | aliguori | } |
383 | 753b4053 | aliguori | } |
384 | 753b4053 | aliguori | |
385 | 3512779a | bellard | /* fastest code */
|
386 | 3512779a | bellard | static void vnc_write_pixels_copy(VncState *vs, void *pixels, int size) |
387 | 3512779a | bellard | { |
388 | 3512779a | bellard | vnc_write(vs, pixels, size); |
389 | 3512779a | bellard | } |
390 | 3512779a | bellard | |
391 | 3512779a | bellard | /* slowest but generic code. */
|
392 | 3512779a | bellard | static void vnc_convert_pixel(VncState *vs, uint8_t *buf, uint32_t v) |
393 | 3512779a | bellard | { |
394 | 7eac3a87 | aliguori | uint8_t r, g, b; |
395 | 7eac3a87 | aliguori | |
396 | 90a1e3c0 | aliguori | r = ((((v & vs->serverds.pf.rmask) >> vs->serverds.pf.rshift) << vs->clientds.pf.rbits) >> |
397 | 90a1e3c0 | aliguori | vs->serverds.pf.rbits); |
398 | 90a1e3c0 | aliguori | g = ((((v & vs->serverds.pf.gmask) >> vs->serverds.pf.gshift) << vs->clientds.pf.gbits) >> |
399 | 90a1e3c0 | aliguori | vs->serverds.pf.gbits); |
400 | 90a1e3c0 | aliguori | b = ((((v & vs->serverds.pf.bmask) >> vs->serverds.pf.bshift) << vs->clientds.pf.bbits) >> |
401 | 90a1e3c0 | aliguori | vs->serverds.pf.bbits); |
402 | 6cec5487 | aliguori | v = (r << vs->clientds.pf.rshift) | |
403 | 6cec5487 | aliguori | (g << vs->clientds.pf.gshift) | |
404 | 6cec5487 | aliguori | (b << vs->clientds.pf.bshift); |
405 | 6cec5487 | aliguori | switch(vs->clientds.pf.bytes_per_pixel) {
|
406 | 3512779a | bellard | case 1: |
407 | 3512779a | bellard | buf[0] = v;
|
408 | 3512779a | bellard | break;
|
409 | 3512779a | bellard | case 2: |
410 | 6cec5487 | aliguori | if (vs->clientds.flags & QEMU_BIG_ENDIAN_FLAG) {
|
411 | 3512779a | bellard | buf[0] = v >> 8; |
412 | 3512779a | bellard | buf[1] = v;
|
413 | 3512779a | bellard | } else {
|
414 | 3512779a | bellard | buf[1] = v >> 8; |
415 | 3512779a | bellard | buf[0] = v;
|
416 | 3512779a | bellard | } |
417 | 3512779a | bellard | break;
|
418 | 3512779a | bellard | default:
|
419 | 3512779a | bellard | case 4: |
420 | 6cec5487 | aliguori | if (vs->clientds.flags & QEMU_BIG_ENDIAN_FLAG) {
|
421 | 3512779a | bellard | buf[0] = v >> 24; |
422 | 3512779a | bellard | buf[1] = v >> 16; |
423 | 3512779a | bellard | buf[2] = v >> 8; |
424 | 3512779a | bellard | buf[3] = v;
|
425 | 3512779a | bellard | } else {
|
426 | 3512779a | bellard | buf[3] = v >> 24; |
427 | 3512779a | bellard | buf[2] = v >> 16; |
428 | 3512779a | bellard | buf[1] = v >> 8; |
429 | 3512779a | bellard | buf[0] = v;
|
430 | 3512779a | bellard | } |
431 | 3512779a | bellard | break;
|
432 | 3512779a | bellard | } |
433 | 3512779a | bellard | } |
434 | 3512779a | bellard | |
435 | 3512779a | bellard | static void vnc_write_pixels_generic(VncState *vs, void *pixels1, int size) |
436 | 3512779a | bellard | { |
437 | 3512779a | bellard | uint8_t buf[4];
|
438 | 3512779a | bellard | |
439 | 6cec5487 | aliguori | if (vs->serverds.pf.bytes_per_pixel == 4) { |
440 | 7eac3a87 | aliguori | uint32_t *pixels = pixels1; |
441 | 7eac3a87 | aliguori | int n, i;
|
442 | 7eac3a87 | aliguori | n = size >> 2;
|
443 | 7eac3a87 | aliguori | for(i = 0; i < n; i++) { |
444 | 7eac3a87 | aliguori | vnc_convert_pixel(vs, buf, pixels[i]); |
445 | 6cec5487 | aliguori | vnc_write(vs, buf, vs->clientds.pf.bytes_per_pixel); |
446 | 7eac3a87 | aliguori | } |
447 | 6cec5487 | aliguori | } else if (vs->serverds.pf.bytes_per_pixel == 2) { |
448 | 7eac3a87 | aliguori | uint16_t *pixels = pixels1; |
449 | 7eac3a87 | aliguori | int n, i;
|
450 | 7eac3a87 | aliguori | n = size >> 1;
|
451 | 7eac3a87 | aliguori | for(i = 0; i < n; i++) { |
452 | 7eac3a87 | aliguori | vnc_convert_pixel(vs, buf, pixels[i]); |
453 | 6cec5487 | aliguori | vnc_write(vs, buf, vs->clientds.pf.bytes_per_pixel); |
454 | 7eac3a87 | aliguori | } |
455 | 6cec5487 | aliguori | } else if (vs->serverds.pf.bytes_per_pixel == 1) { |
456 | 7eac3a87 | aliguori | uint8_t *pixels = pixels1; |
457 | 7eac3a87 | aliguori | int n, i;
|
458 | 7eac3a87 | aliguori | n = size; |
459 | 7eac3a87 | aliguori | for(i = 0; i < n; i++) { |
460 | 7eac3a87 | aliguori | vnc_convert_pixel(vs, buf, pixels[i]); |
461 | 6cec5487 | aliguori | vnc_write(vs, buf, vs->clientds.pf.bytes_per_pixel); |
462 | 7eac3a87 | aliguori | } |
463 | 7eac3a87 | aliguori | } else {
|
464 | 7eac3a87 | aliguori | fprintf(stderr, "vnc_write_pixels_generic: VncState color depth not supported\n");
|
465 | 3512779a | bellard | } |
466 | 3512779a | bellard | } |
467 | 3512779a | bellard | |
468 | 24236869 | bellard | static void send_framebuffer_update_raw(VncState *vs, int x, int y, int w, int h) |
469 | 24236869 | bellard | { |
470 | 24236869 | bellard | int i;
|
471 | 60fe76f3 | ths | uint8_t *row; |
472 | 24236869 | bellard | |
473 | 6cec5487 | aliguori | row = ds_get_data(vs->ds) + y * ds_get_linesize(vs->ds) + x * ds_get_bytes_per_pixel(vs->ds); |
474 | 24236869 | bellard | for (i = 0; i < h; i++) { |
475 | 28a76be8 | aliguori | vs->write_pixels(vs, row, w * ds_get_bytes_per_pixel(vs->ds)); |
476 | 28a76be8 | aliguori | row += ds_get_linesize(vs->ds); |
477 | 24236869 | bellard | } |
478 | 24236869 | bellard | } |
479 | 24236869 | bellard | |
480 | 24236869 | bellard | static void hextile_enc_cord(uint8_t *ptr, int x, int y, int w, int h) |
481 | 24236869 | bellard | { |
482 | 24236869 | bellard | ptr[0] = ((x & 0x0F) << 4) | (y & 0x0F); |
483 | 24236869 | bellard | ptr[1] = (((w - 1) & 0x0F) << 4) | ((h - 1) & 0x0F); |
484 | 24236869 | bellard | } |
485 | 24236869 | bellard | |
486 | 24236869 | bellard | #define BPP 8 |
487 | 24236869 | bellard | #include "vnchextile.h" |
488 | 24236869 | bellard | #undef BPP
|
489 | 24236869 | bellard | |
490 | 24236869 | bellard | #define BPP 16 |
491 | 24236869 | bellard | #include "vnchextile.h" |
492 | 24236869 | bellard | #undef BPP
|
493 | 24236869 | bellard | |
494 | 24236869 | bellard | #define BPP 32 |
495 | 24236869 | bellard | #include "vnchextile.h" |
496 | 24236869 | bellard | #undef BPP
|
497 | 24236869 | bellard | |
498 | 3512779a | bellard | #define GENERIC
|
499 | 7eac3a87 | aliguori | #define BPP 8 |
500 | 7eac3a87 | aliguori | #include "vnchextile.h" |
501 | 7eac3a87 | aliguori | #undef BPP
|
502 | 7eac3a87 | aliguori | #undef GENERIC
|
503 | 7eac3a87 | aliguori | |
504 | 7eac3a87 | aliguori | #define GENERIC
|
505 | 7eac3a87 | aliguori | #define BPP 16 |
506 | 7eac3a87 | aliguori | #include "vnchextile.h" |
507 | 7eac3a87 | aliguori | #undef BPP
|
508 | 7eac3a87 | aliguori | #undef GENERIC
|
509 | 7eac3a87 | aliguori | |
510 | 7eac3a87 | aliguori | #define GENERIC
|
511 | 3512779a | bellard | #define BPP 32 |
512 | 3512779a | bellard | #include "vnchextile.h" |
513 | 3512779a | bellard | #undef BPP
|
514 | 3512779a | bellard | #undef GENERIC
|
515 | 3512779a | bellard | |
516 | 24236869 | bellard | static void send_framebuffer_update_hextile(VncState *vs, int x, int y, int w, int h) |
517 | 24236869 | bellard | { |
518 | 24236869 | bellard | int i, j;
|
519 | 24236869 | bellard | int has_fg, has_bg;
|
520 | 7eac3a87 | aliguori | uint8_t *last_fg, *last_bg; |
521 | 24236869 | bellard | |
522 | 1eec614b | aliguori | last_fg = (uint8_t *) qemu_malloc(vs->serverds.pf.bytes_per_pixel); |
523 | 1eec614b | aliguori | last_bg = (uint8_t *) qemu_malloc(vs->serverds.pf.bytes_per_pixel); |
524 | 24236869 | bellard | has_fg = has_bg = 0;
|
525 | 24236869 | bellard | for (j = y; j < (y + h); j += 16) { |
526 | 28a76be8 | aliguori | for (i = x; i < (x + w); i += 16) { |
527 | 5fafdf24 | ths | vs->send_hextile_tile(vs, i, j, |
528 | 3512779a | bellard | MIN(16, x + w - i), MIN(16, y + h - j), |
529 | 7eac3a87 | aliguori | last_bg, last_fg, &has_bg, &has_fg); |
530 | 28a76be8 | aliguori | } |
531 | 24236869 | bellard | } |
532 | 7eac3a87 | aliguori | free(last_fg); |
533 | 7eac3a87 | aliguori | free(last_bg); |
534 | 7eac3a87 | aliguori | |
535 | 24236869 | bellard | } |
536 | 24236869 | bellard | |
537 | 059cef40 | aliguori | static void vnc_zlib_init(VncState *vs) |
538 | 059cef40 | aliguori | { |
539 | 059cef40 | aliguori | int i;
|
540 | 059cef40 | aliguori | for (i=0; i<(sizeof(vs->zlib_stream) / sizeof(z_stream)); i++) |
541 | 059cef40 | aliguori | vs->zlib_stream[i].opaque = NULL;
|
542 | 059cef40 | aliguori | } |
543 | 059cef40 | aliguori | |
544 | 059cef40 | aliguori | static void vnc_zlib_start(VncState *vs) |
545 | 059cef40 | aliguori | { |
546 | 059cef40 | aliguori | buffer_reset(&vs->zlib); |
547 | 059cef40 | aliguori | |
548 | 059cef40 | aliguori | // make the output buffer be the zlib buffer, so we can compress it later
|
549 | 059cef40 | aliguori | vs->zlib_tmp = vs->output; |
550 | 059cef40 | aliguori | vs->output = vs->zlib; |
551 | 059cef40 | aliguori | } |
552 | 059cef40 | aliguori | |
553 | 059cef40 | aliguori | static int vnc_zlib_stop(VncState *vs, int stream_id) |
554 | 059cef40 | aliguori | { |
555 | 059cef40 | aliguori | z_streamp zstream = &vs->zlib_stream[stream_id]; |
556 | 059cef40 | aliguori | int previous_out;
|
557 | 059cef40 | aliguori | |
558 | 059cef40 | aliguori | // switch back to normal output/zlib buffers
|
559 | 059cef40 | aliguori | vs->zlib = vs->output; |
560 | 059cef40 | aliguori | vs->output = vs->zlib_tmp; |
561 | 059cef40 | aliguori | |
562 | 059cef40 | aliguori | // compress the zlib buffer
|
563 | 059cef40 | aliguori | |
564 | 059cef40 | aliguori | // initialize the stream
|
565 | 059cef40 | aliguori | // XXX need one stream per session
|
566 | 059cef40 | aliguori | if (zstream->opaque != vs) {
|
567 | 059cef40 | aliguori | int err;
|
568 | 059cef40 | aliguori | |
569 | 059cef40 | aliguori | VNC_DEBUG("VNC: initializing zlib stream %d\n", stream_id);
|
570 | 059cef40 | aliguori | VNC_DEBUG("VNC: opaque = %p | vs = %p\n", zstream->opaque, vs);
|
571 | 059cef40 | aliguori | zstream->zalloc = Z_NULL; |
572 | 059cef40 | aliguori | zstream->zfree = Z_NULL; |
573 | 059cef40 | aliguori | |
574 | 059cef40 | aliguori | err = deflateInit2(zstream, vs->tight_compression, Z_DEFLATED, MAX_WBITS, |
575 | 059cef40 | aliguori | MAX_MEM_LEVEL, Z_DEFAULT_STRATEGY); |
576 | 059cef40 | aliguori | |
577 | 059cef40 | aliguori | if (err != Z_OK) {
|
578 | 059cef40 | aliguori | fprintf(stderr, "VNC: error initializing zlib\n");
|
579 | 059cef40 | aliguori | return -1; |
580 | 059cef40 | aliguori | } |
581 | 059cef40 | aliguori | |
582 | 059cef40 | aliguori | zstream->opaque = vs; |
583 | 059cef40 | aliguori | } |
584 | 059cef40 | aliguori | |
585 | 059cef40 | aliguori | // XXX what to do if tight_compression changed in between?
|
586 | 059cef40 | aliguori | |
587 | 059cef40 | aliguori | // reserve memory in output buffer
|
588 | 059cef40 | aliguori | buffer_reserve(&vs->output, vs->zlib.offset + 64);
|
589 | 059cef40 | aliguori | |
590 | 059cef40 | aliguori | // set pointers
|
591 | 059cef40 | aliguori | zstream->next_in = vs->zlib.buffer; |
592 | 059cef40 | aliguori | zstream->avail_in = vs->zlib.offset; |
593 | 059cef40 | aliguori | zstream->next_out = vs->output.buffer + vs->output.offset; |
594 | 059cef40 | aliguori | zstream->avail_out = vs->output.capacity - vs->output.offset; |
595 | 059cef40 | aliguori | zstream->data_type = Z_BINARY; |
596 | 059cef40 | aliguori | previous_out = zstream->total_out; |
597 | 059cef40 | aliguori | |
598 | 059cef40 | aliguori | // start encoding
|
599 | 059cef40 | aliguori | if (deflate(zstream, Z_SYNC_FLUSH) != Z_OK) {
|
600 | 059cef40 | aliguori | fprintf(stderr, "VNC: error during zlib compression\n");
|
601 | 059cef40 | aliguori | return -1; |
602 | 059cef40 | aliguori | } |
603 | 059cef40 | aliguori | |
604 | 059cef40 | aliguori | vs->output.offset = vs->output.capacity - zstream->avail_out; |
605 | 059cef40 | aliguori | return zstream->total_out - previous_out;
|
606 | 059cef40 | aliguori | } |
607 | 059cef40 | aliguori | |
608 | 059cef40 | aliguori | static void send_framebuffer_update_zlib(VncState *vs, int x, int y, int w, int h) |
609 | 059cef40 | aliguori | { |
610 | 059cef40 | aliguori | int old_offset, new_offset, bytes_written;
|
611 | 059cef40 | aliguori | |
612 | 059cef40 | aliguori | vnc_framebuffer_update(vs, x, y, w, h, VNC_ENCODING_ZLIB); |
613 | 059cef40 | aliguori | |
614 | 059cef40 | aliguori | // remember where we put in the follow-up size
|
615 | 059cef40 | aliguori | old_offset = vs->output.offset; |
616 | 059cef40 | aliguori | vnc_write_s32(vs, 0);
|
617 | 059cef40 | aliguori | |
618 | 059cef40 | aliguori | // compress the stream
|
619 | 059cef40 | aliguori | vnc_zlib_start(vs); |
620 | 059cef40 | aliguori | send_framebuffer_update_raw(vs, x, y, w, h); |
621 | 059cef40 | aliguori | bytes_written = vnc_zlib_stop(vs, 0);
|
622 | 059cef40 | aliguori | |
623 | 059cef40 | aliguori | if (bytes_written == -1) |
624 | 059cef40 | aliguori | return;
|
625 | 059cef40 | aliguori | |
626 | 059cef40 | aliguori | // hack in the size
|
627 | 059cef40 | aliguori | new_offset = vs->output.offset; |
628 | 059cef40 | aliguori | vs->output.offset = old_offset; |
629 | 059cef40 | aliguori | vnc_write_u32(vs, bytes_written); |
630 | 059cef40 | aliguori | vs->output.offset = new_offset; |
631 | 059cef40 | aliguori | } |
632 | 059cef40 | aliguori | |
633 | 24236869 | bellard | static void send_framebuffer_update(VncState *vs, int x, int y, int w, int h) |
634 | 24236869 | bellard | { |
635 | fb437313 | aliguori | switch(vs->vnc_encoding) {
|
636 | 28a76be8 | aliguori | case VNC_ENCODING_ZLIB:
|
637 | 28a76be8 | aliguori | send_framebuffer_update_zlib(vs, x, y, w, h); |
638 | 28a76be8 | aliguori | break;
|
639 | 28a76be8 | aliguori | case VNC_ENCODING_HEXTILE:
|
640 | 28a76be8 | aliguori | vnc_framebuffer_update(vs, x, y, w, h, VNC_ENCODING_HEXTILE); |
641 | 28a76be8 | aliguori | send_framebuffer_update_hextile(vs, x, y, w, h); |
642 | 28a76be8 | aliguori | break;
|
643 | 28a76be8 | aliguori | default:
|
644 | 28a76be8 | aliguori | vnc_framebuffer_update(vs, x, y, w, h, VNC_ENCODING_RAW); |
645 | 28a76be8 | aliguori | send_framebuffer_update_raw(vs, x, y, w, h); |
646 | 28a76be8 | aliguori | break;
|
647 | fb437313 | aliguori | } |
648 | 24236869 | bellard | } |
649 | 24236869 | bellard | |
650 | 753b4053 | aliguori | static void vnc_copy(VncState *vs, int src_x, int src_y, int dst_x, int dst_y, int w, int h) |
651 | 24236869 | bellard | { |
652 | 24236869 | bellard | vnc_update_client(vs); |
653 | 24236869 | bellard | |
654 | 24236869 | bellard | vnc_write_u8(vs, 0); /* msg id */ |
655 | 24236869 | bellard | vnc_write_u8(vs, 0);
|
656 | 24236869 | bellard | vnc_write_u16(vs, 1); /* number of rects */ |
657 | 29fa4ed9 | aliguori | vnc_framebuffer_update(vs, dst_x, dst_y, w, h, VNC_ENCODING_COPYRECT); |
658 | 24236869 | bellard | vnc_write_u16(vs, src_x); |
659 | 24236869 | bellard | vnc_write_u16(vs, src_y); |
660 | 24236869 | bellard | vnc_flush(vs); |
661 | 24236869 | bellard | } |
662 | 24236869 | bellard | |
663 | 753b4053 | aliguori | static void vnc_dpy_copy(DisplayState *ds, int src_x, int src_y, int dst_x, int dst_y, int w, int h) |
664 | 753b4053 | aliguori | { |
665 | 753b4053 | aliguori | VncDisplay *vd = ds->opaque; |
666 | 753b4053 | aliguori | VncState *vs = vd->clients; |
667 | 753b4053 | aliguori | while (vs != NULL) { |
668 | 753b4053 | aliguori | if (vnc_has_feature(vs, VNC_FEATURE_COPYRECT))
|
669 | 753b4053 | aliguori | vnc_copy(vs, src_x, src_y, dst_x, dst_y, w, h); |
670 | 753b4053 | aliguori | else /* TODO */ |
671 | 753b4053 | aliguori | vnc_update(vs, dst_x, dst_y, w, h); |
672 | 753b4053 | aliguori | vs = vs->next; |
673 | 753b4053 | aliguori | } |
674 | 753b4053 | aliguori | } |
675 | 753b4053 | aliguori | |
676 | 24236869 | bellard | static int find_dirty_height(VncState *vs, int y, int last_x, int x) |
677 | 24236869 | bellard | { |
678 | 24236869 | bellard | int h;
|
679 | 24236869 | bellard | |
680 | 6cec5487 | aliguori | for (h = 1; h < (vs->serverds.height - y); h++) { |
681 | 28a76be8 | aliguori | int tmp_x;
|
682 | 28a76be8 | aliguori | if (!vnc_get_bit(vs->dirty_row[y + h], last_x))
|
683 | 28a76be8 | aliguori | break;
|
684 | 28a76be8 | aliguori | for (tmp_x = last_x; tmp_x < x; tmp_x++)
|
685 | 28a76be8 | aliguori | vnc_clear_bit(vs->dirty_row[y + h], tmp_x); |
686 | 24236869 | bellard | } |
687 | 24236869 | bellard | |
688 | 24236869 | bellard | return h;
|
689 | 24236869 | bellard | } |
690 | 24236869 | bellard | |
691 | 24236869 | bellard | static void vnc_update_client(void *opaque) |
692 | 24236869 | bellard | { |
693 | 24236869 | bellard | VncState *vs = opaque; |
694 | 24236869 | bellard | if (vs->need_update && vs->csock != -1) { |
695 | 28a76be8 | aliguori | int y;
|
696 | 28a76be8 | aliguori | uint8_t *row; |
697 | 28a76be8 | aliguori | char *old_row;
|
698 | 28a76be8 | aliguori | uint32_t width_mask[VNC_DIRTY_WORDS]; |
699 | 28a76be8 | aliguori | int n_rectangles;
|
700 | 28a76be8 | aliguori | int saved_offset;
|
701 | 28a76be8 | aliguori | int has_dirty = 0; |
702 | 24236869 | bellard | |
703 | a0ecfb73 | balrog | vga_hw_update(); |
704 | a0ecfb73 | balrog | |
705 | 6cec5487 | aliguori | vnc_set_bits(width_mask, (ds_get_width(vs->ds) / 16), VNC_DIRTY_WORDS);
|
706 | 24236869 | bellard | |
707 | 28a76be8 | aliguori | /* Walk through the dirty map and eliminate tiles that
|
708 | 28a76be8 | aliguori | really aren't dirty */
|
709 | 28a76be8 | aliguori | row = ds_get_data(vs->ds); |
710 | 28a76be8 | aliguori | old_row = vs->old_data; |
711 | 28a76be8 | aliguori | |
712 | 28a76be8 | aliguori | for (y = 0; y < ds_get_height(vs->ds); y++) { |
713 | 28a76be8 | aliguori | if (vnc_and_bits(vs->dirty_row[y], width_mask, VNC_DIRTY_WORDS)) {
|
714 | 28a76be8 | aliguori | int x;
|
715 | 28a76be8 | aliguori | uint8_t *ptr; |
716 | 28a76be8 | aliguori | char *old_ptr;
|
717 | 28a76be8 | aliguori | |
718 | 28a76be8 | aliguori | ptr = row; |
719 | 28a76be8 | aliguori | old_ptr = (char*)old_row;
|
720 | 28a76be8 | aliguori | |
721 | 28a76be8 | aliguori | for (x = 0; x < ds_get_width(vs->ds); x += 16) { |
722 | 28a76be8 | aliguori | if (memcmp(old_ptr, ptr, 16 * ds_get_bytes_per_pixel(vs->ds)) == 0) { |
723 | 28a76be8 | aliguori | vnc_clear_bit(vs->dirty_row[y], (x / 16));
|
724 | 28a76be8 | aliguori | } else {
|
725 | 28a76be8 | aliguori | has_dirty = 1;
|
726 | 28a76be8 | aliguori | memcpy(old_ptr, ptr, 16 * ds_get_bytes_per_pixel(vs->ds));
|
727 | 28a76be8 | aliguori | } |
728 | 28a76be8 | aliguori | |
729 | 28a76be8 | aliguori | ptr += 16 * ds_get_bytes_per_pixel(vs->ds);
|
730 | 28a76be8 | aliguori | old_ptr += 16 * ds_get_bytes_per_pixel(vs->ds);
|
731 | 28a76be8 | aliguori | } |
732 | 28a76be8 | aliguori | } |
733 | 28a76be8 | aliguori | |
734 | 28a76be8 | aliguori | row += ds_get_linesize(vs->ds); |
735 | 28a76be8 | aliguori | old_row += ds_get_linesize(vs->ds); |
736 | 28a76be8 | aliguori | } |
737 | 28a76be8 | aliguori | |
738 | 28a76be8 | aliguori | if (!has_dirty && !vs->audio_cap) {
|
739 | 28a76be8 | aliguori | qemu_mod_timer(vs->timer, qemu_get_clock(rt_clock) + VNC_REFRESH_INTERVAL); |
740 | 28a76be8 | aliguori | return;
|
741 | 28a76be8 | aliguori | } |
742 | 28a76be8 | aliguori | |
743 | 28a76be8 | aliguori | /* Count rectangles */
|
744 | 28a76be8 | aliguori | n_rectangles = 0;
|
745 | 28a76be8 | aliguori | vnc_write_u8(vs, 0); /* msg id */ |
746 | 28a76be8 | aliguori | vnc_write_u8(vs, 0);
|
747 | 28a76be8 | aliguori | saved_offset = vs->output.offset; |
748 | 28a76be8 | aliguori | vnc_write_u16(vs, 0);
|
749 | 28a76be8 | aliguori | |
750 | 28a76be8 | aliguori | for (y = 0; y < vs->serverds.height; y++) { |
751 | 28a76be8 | aliguori | int x;
|
752 | 28a76be8 | aliguori | int last_x = -1; |
753 | 28a76be8 | aliguori | for (x = 0; x < vs->serverds.width / 16; x++) { |
754 | 28a76be8 | aliguori | if (vnc_get_bit(vs->dirty_row[y], x)) {
|
755 | 28a76be8 | aliguori | if (last_x == -1) { |
756 | 28a76be8 | aliguori | last_x = x; |
757 | 28a76be8 | aliguori | } |
758 | 28a76be8 | aliguori | vnc_clear_bit(vs->dirty_row[y], x); |
759 | 28a76be8 | aliguori | } else {
|
760 | 28a76be8 | aliguori | if (last_x != -1) { |
761 | 28a76be8 | aliguori | int h = find_dirty_height(vs, y, last_x, x);
|
762 | 28a76be8 | aliguori | send_framebuffer_update(vs, last_x * 16, y, (x - last_x) * 16, h); |
763 | 28a76be8 | aliguori | n_rectangles++; |
764 | 28a76be8 | aliguori | } |
765 | 28a76be8 | aliguori | last_x = -1;
|
766 | 28a76be8 | aliguori | } |
767 | 28a76be8 | aliguori | } |
768 | 28a76be8 | aliguori | if (last_x != -1) { |
769 | 28a76be8 | aliguori | int h = find_dirty_height(vs, y, last_x, x);
|
770 | 28a76be8 | aliguori | send_framebuffer_update(vs, last_x * 16, y, (x - last_x) * 16, h); |
771 | 28a76be8 | aliguori | n_rectangles++; |
772 | 28a76be8 | aliguori | } |
773 | 28a76be8 | aliguori | } |
774 | 28a76be8 | aliguori | vs->output.buffer[saved_offset] = (n_rectangles >> 8) & 0xFF; |
775 | 28a76be8 | aliguori | vs->output.buffer[saved_offset + 1] = n_rectangles & 0xFF; |
776 | 28a76be8 | aliguori | vnc_flush(vs); |
777 | 24236869 | bellard | |
778 | 24236869 | bellard | } |
779 | 24236869 | bellard | |
780 | a0ecfb73 | balrog | if (vs->csock != -1) { |
781 | a0ecfb73 | balrog | qemu_mod_timer(vs->timer, qemu_get_clock(rt_clock) + VNC_REFRESH_INTERVAL); |
782 | 24236869 | bellard | } |
783 | 24236869 | bellard | |
784 | 24236869 | bellard | } |
785 | 24236869 | bellard | |
786 | 429a8ed3 | malc | /* audio */
|
787 | 429a8ed3 | malc | static void audio_capture_notify(void *opaque, audcnotification_e cmd) |
788 | 429a8ed3 | malc | { |
789 | 429a8ed3 | malc | VncState *vs = opaque; |
790 | 429a8ed3 | malc | |
791 | 429a8ed3 | malc | switch (cmd) {
|
792 | 429a8ed3 | malc | case AUD_CNOTIFY_DISABLE:
|
793 | 429a8ed3 | malc | vnc_write_u8(vs, 255);
|
794 | 429a8ed3 | malc | vnc_write_u8(vs, 1);
|
795 | 429a8ed3 | malc | vnc_write_u16(vs, 0);
|
796 | 429a8ed3 | malc | vnc_flush(vs); |
797 | 429a8ed3 | malc | break;
|
798 | 429a8ed3 | malc | |
799 | 429a8ed3 | malc | case AUD_CNOTIFY_ENABLE:
|
800 | 429a8ed3 | malc | vnc_write_u8(vs, 255);
|
801 | 429a8ed3 | malc | vnc_write_u8(vs, 1);
|
802 | 429a8ed3 | malc | vnc_write_u16(vs, 1);
|
803 | 429a8ed3 | malc | vnc_flush(vs); |
804 | 429a8ed3 | malc | break;
|
805 | 429a8ed3 | malc | } |
806 | 429a8ed3 | malc | } |
807 | 429a8ed3 | malc | |
808 | 429a8ed3 | malc | static void audio_capture_destroy(void *opaque) |
809 | 429a8ed3 | malc | { |
810 | 429a8ed3 | malc | } |
811 | 429a8ed3 | malc | |
812 | 429a8ed3 | malc | static void audio_capture(void *opaque, void *buf, int size) |
813 | 429a8ed3 | malc | { |
814 | 429a8ed3 | malc | VncState *vs = opaque; |
815 | 429a8ed3 | malc | |
816 | 429a8ed3 | malc | vnc_write_u8(vs, 255);
|
817 | 429a8ed3 | malc | vnc_write_u8(vs, 1);
|
818 | 429a8ed3 | malc | vnc_write_u16(vs, 2);
|
819 | 429a8ed3 | malc | vnc_write_u32(vs, size); |
820 | 429a8ed3 | malc | vnc_write(vs, buf, size); |
821 | 429a8ed3 | malc | vnc_flush(vs); |
822 | 429a8ed3 | malc | } |
823 | 429a8ed3 | malc | |
824 | 429a8ed3 | malc | static void audio_add(VncState *vs) |
825 | 429a8ed3 | malc | { |
826 | 376253ec | aliguori | Monitor *mon = cur_mon; |
827 | 429a8ed3 | malc | struct audio_capture_ops ops;
|
828 | 429a8ed3 | malc | |
829 | 429a8ed3 | malc | if (vs->audio_cap) {
|
830 | 376253ec | aliguori | monitor_printf(mon, "audio already running\n");
|
831 | 429a8ed3 | malc | return;
|
832 | 429a8ed3 | malc | } |
833 | 429a8ed3 | malc | |
834 | 429a8ed3 | malc | ops.notify = audio_capture_notify; |
835 | 429a8ed3 | malc | ops.destroy = audio_capture_destroy; |
836 | 429a8ed3 | malc | ops.capture = audio_capture; |
837 | 429a8ed3 | malc | |
838 | 429a8ed3 | malc | vs->audio_cap = AUD_add_capture(NULL, &vs->as, &ops, vs);
|
839 | 429a8ed3 | malc | if (!vs->audio_cap) {
|
840 | 376253ec | aliguori | monitor_printf(mon, "Failed to add audio capture\n");
|
841 | 429a8ed3 | malc | } |
842 | 429a8ed3 | malc | } |
843 | 429a8ed3 | malc | |
844 | 429a8ed3 | malc | static void audio_del(VncState *vs) |
845 | 429a8ed3 | malc | { |
846 | 429a8ed3 | malc | if (vs->audio_cap) {
|
847 | 429a8ed3 | malc | AUD_del_capture(vs->audio_cap, vs); |
848 | 429a8ed3 | malc | vs->audio_cap = NULL;
|
849 | 429a8ed3 | malc | } |
850 | 429a8ed3 | malc | } |
851 | 429a8ed3 | malc | |
852 | 2f9606b3 | aliguori | |
853 | 2f9606b3 | aliguori | int vnc_client_io_error(VncState *vs, int ret, int last_errno) |
854 | 24236869 | bellard | { |
855 | 24236869 | bellard | if (ret == 0 || ret == -1) { |
856 | ea01e5fd | balrog | if (ret == -1) { |
857 | ea01e5fd | balrog | switch (last_errno) {
|
858 | ea01e5fd | balrog | case EINTR:
|
859 | ea01e5fd | balrog | case EAGAIN:
|
860 | ea01e5fd | balrog | #ifdef _WIN32
|
861 | ea01e5fd | balrog | case WSAEWOULDBLOCK:
|
862 | ea01e5fd | balrog | #endif
|
863 | ea01e5fd | balrog | return 0; |
864 | ea01e5fd | balrog | default:
|
865 | ea01e5fd | balrog | break;
|
866 | ea01e5fd | balrog | } |
867 | ea01e5fd | balrog | } |
868 | 24236869 | bellard | |
869 | 28a76be8 | aliguori | VNC_DEBUG("Closing down client sock %d %d\n", ret, ret < 0 ? last_errno : 0); |
870 | 28a76be8 | aliguori | qemu_set_fd_handler2(vs->csock, NULL, NULL, NULL, NULL); |
871 | 28a76be8 | aliguori | closesocket(vs->csock); |
872 | 753b4053 | aliguori | qemu_del_timer(vs->timer); |
873 | 753b4053 | aliguori | qemu_free_timer(vs->timer); |
874 | 753b4053 | aliguori | if (vs->input.buffer) qemu_free(vs->input.buffer);
|
875 | 753b4053 | aliguori | if (vs->output.buffer) qemu_free(vs->output.buffer);
|
876 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
877 | 28a76be8 | aliguori | vnc_tls_client_cleanup(vs); |
878 | 8d5d2d4c | ths | #endif /* CONFIG_VNC_TLS */ |
879 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_SASL
|
880 | 2f9606b3 | aliguori | vnc_sasl_client_cleanup(vs); |
881 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_SASL */ |
882 | 429a8ed3 | malc | audio_del(vs); |
883 | 753b4053 | aliguori | |
884 | 753b4053 | aliguori | VncState *p, *parent = NULL;
|
885 | 753b4053 | aliguori | for (p = vs->vd->clients; p != NULL; p = p->next) { |
886 | 753b4053 | aliguori | if (p == vs) {
|
887 | 753b4053 | aliguori | if (parent)
|
888 | 753b4053 | aliguori | parent->next = p->next; |
889 | 753b4053 | aliguori | else
|
890 | 753b4053 | aliguori | vs->vd->clients = p->next; |
891 | 753b4053 | aliguori | break;
|
892 | 753b4053 | aliguori | } |
893 | 753b4053 | aliguori | parent = p; |
894 | 753b4053 | aliguori | } |
895 | 753b4053 | aliguori | if (!vs->vd->clients)
|
896 | 753b4053 | aliguori | dcl->idle = 1;
|
897 | 753b4053 | aliguori | |
898 | 753b4053 | aliguori | qemu_free(vs->old_data); |
899 | 753b4053 | aliguori | qemu_free(vs); |
900 | 753b4053 | aliguori | |
901 | 28a76be8 | aliguori | return 0; |
902 | 24236869 | bellard | } |
903 | 24236869 | bellard | return ret;
|
904 | 24236869 | bellard | } |
905 | 24236869 | bellard | |
906 | 5fb6c7a8 | aliguori | |
907 | 5fb6c7a8 | aliguori | void vnc_client_error(VncState *vs)
|
908 | 24236869 | bellard | { |
909 | 6ca957f0 | bellard | vnc_client_io_error(vs, -1, EINVAL);
|
910 | 24236869 | bellard | } |
911 | 24236869 | bellard | |
912 | 2f9606b3 | aliguori | |
913 | 2f9606b3 | aliguori | /*
|
914 | 2f9606b3 | aliguori | * Called to write a chunk of data to the client socket. The data may
|
915 | 2f9606b3 | aliguori | * be the raw data, or may have already been encoded by SASL.
|
916 | 2f9606b3 | aliguori | * The data will be written either straight onto the socket, or
|
917 | 2f9606b3 | aliguori | * written via the GNUTLS wrappers, if TLS/SSL encryption is enabled
|
918 | 2f9606b3 | aliguori | *
|
919 | 2f9606b3 | aliguori | * NB, it is theoretically possible to have 2 layers of encryption,
|
920 | 2f9606b3 | aliguori | * both SASL, and this TLS layer. It is highly unlikely in practice
|
921 | 2f9606b3 | aliguori | * though, since SASL encryption will typically be a no-op if TLS
|
922 | 2f9606b3 | aliguori | * is active
|
923 | 2f9606b3 | aliguori | *
|
924 | 2f9606b3 | aliguori | * Returns the number of bytes written, which may be less than
|
925 | 2f9606b3 | aliguori | * the requested 'datalen' if the socket would block. Returns
|
926 | 2f9606b3 | aliguori | * -1 on error, and disconnects the client socket.
|
927 | 2f9606b3 | aliguori | */
|
928 | 2f9606b3 | aliguori | long vnc_client_write_buf(VncState *vs, const uint8_t *data, size_t datalen) |
929 | 24236869 | bellard | { |
930 | ceb5caaf | bellard | long ret;
|
931 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
932 | 5fb6c7a8 | aliguori | if (vs->tls.session) {
|
933 | 28a76be8 | aliguori | ret = gnutls_write(vs->tls.session, data, datalen); |
934 | 28a76be8 | aliguori | if (ret < 0) { |
935 | 28a76be8 | aliguori | if (ret == GNUTLS_E_AGAIN)
|
936 | 28a76be8 | aliguori | errno = EAGAIN; |
937 | 28a76be8 | aliguori | else
|
938 | 28a76be8 | aliguori | errno = EIO; |
939 | 28a76be8 | aliguori | ret = -1;
|
940 | 28a76be8 | aliguori | } |
941 | 8d5d2d4c | ths | } else
|
942 | 8d5d2d4c | ths | #endif /* CONFIG_VNC_TLS */ |
943 | 28a76be8 | aliguori | ret = send(vs->csock, data, datalen, 0);
|
944 | 2f9606b3 | aliguori | VNC_DEBUG("Wrote wire %p %d -> %ld\n", data, datalen, ret);
|
945 | 2f9606b3 | aliguori | return vnc_client_io_error(vs, ret, socket_error());
|
946 | 2f9606b3 | aliguori | } |
947 | 2f9606b3 | aliguori | |
948 | 2f9606b3 | aliguori | |
949 | 2f9606b3 | aliguori | /*
|
950 | 2f9606b3 | aliguori | * Called to write buffered data to the client socket, when not
|
951 | 2f9606b3 | aliguori | * using any SASL SSF encryption layers. Will write as much data
|
952 | 2f9606b3 | aliguori | * as possible without blocking. If all buffered data is written,
|
953 | 2f9606b3 | aliguori | * will switch the FD poll() handler back to read monitoring.
|
954 | 2f9606b3 | aliguori | *
|
955 | 2f9606b3 | aliguori | * Returns the number of bytes written, which may be less than
|
956 | 2f9606b3 | aliguori | * the buffered output data if the socket would block. Returns
|
957 | 2f9606b3 | aliguori | * -1 on error, and disconnects the client socket.
|
958 | 2f9606b3 | aliguori | */
|
959 | 2f9606b3 | aliguori | static long vnc_client_write_plain(VncState *vs) |
960 | 2f9606b3 | aliguori | { |
961 | 2f9606b3 | aliguori | long ret;
|
962 | 2f9606b3 | aliguori | |
963 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_SASL
|
964 | 2f9606b3 | aliguori | VNC_DEBUG("Write Plain: Pending output %p size %d offset %d. Wait SSF %d\n",
|
965 | 2f9606b3 | aliguori | vs->output.buffer, vs->output.capacity, vs->output.offset, |
966 | 2f9606b3 | aliguori | vs->sasl.waitWriteSSF); |
967 | 2f9606b3 | aliguori | |
968 | 2f9606b3 | aliguori | if (vs->sasl.conn &&
|
969 | 2f9606b3 | aliguori | vs->sasl.runSSF && |
970 | 2f9606b3 | aliguori | vs->sasl.waitWriteSSF) { |
971 | 2f9606b3 | aliguori | ret = vnc_client_write_buf(vs, vs->output.buffer, vs->sasl.waitWriteSSF); |
972 | 2f9606b3 | aliguori | if (ret)
|
973 | 2f9606b3 | aliguori | vs->sasl.waitWriteSSF -= ret; |
974 | 2f9606b3 | aliguori | } else
|
975 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_SASL */ |
976 | 2f9606b3 | aliguori | ret = vnc_client_write_buf(vs, vs->output.buffer, vs->output.offset); |
977 | 24236869 | bellard | if (!ret)
|
978 | 2f9606b3 | aliguori | return 0; |
979 | 24236869 | bellard | |
980 | 24236869 | bellard | memmove(vs->output.buffer, vs->output.buffer + ret, (vs->output.offset - ret)); |
981 | 24236869 | bellard | vs->output.offset -= ret; |
982 | 24236869 | bellard | |
983 | 24236869 | bellard | if (vs->output.offset == 0) { |
984 | 28a76be8 | aliguori | qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, vs); |
985 | 24236869 | bellard | } |
986 | 2f9606b3 | aliguori | |
987 | 2f9606b3 | aliguori | return ret;
|
988 | 2f9606b3 | aliguori | } |
989 | 2f9606b3 | aliguori | |
990 | 2f9606b3 | aliguori | |
991 | 2f9606b3 | aliguori | /*
|
992 | 2f9606b3 | aliguori | * First function called whenever there is data to be written to
|
993 | 2f9606b3 | aliguori | * the client socket. Will delegate actual work according to whether
|
994 | 2f9606b3 | aliguori | * SASL SSF layers are enabled (thus requiring encryption calls)
|
995 | 2f9606b3 | aliguori | */
|
996 | 2f9606b3 | aliguori | void vnc_client_write(void *opaque) |
997 | 2f9606b3 | aliguori | { |
998 | 2f9606b3 | aliguori | long ret;
|
999 | 2f9606b3 | aliguori | VncState *vs = opaque; |
1000 | 2f9606b3 | aliguori | |
1001 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_SASL
|
1002 | 2f9606b3 | aliguori | if (vs->sasl.conn &&
|
1003 | 2f9606b3 | aliguori | vs->sasl.runSSF && |
1004 | 2f9606b3 | aliguori | !vs->sasl.waitWriteSSF) |
1005 | 2f9606b3 | aliguori | ret = vnc_client_write_sasl(vs); |
1006 | 2f9606b3 | aliguori | else
|
1007 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_SASL */ |
1008 | 2f9606b3 | aliguori | ret = vnc_client_write_plain(vs); |
1009 | 24236869 | bellard | } |
1010 | 24236869 | bellard | |
1011 | 5fb6c7a8 | aliguori | void vnc_read_when(VncState *vs, VncReadEvent *func, size_t expecting)
|
1012 | 24236869 | bellard | { |
1013 | 24236869 | bellard | vs->read_handler = func; |
1014 | 24236869 | bellard | vs->read_handler_expect = expecting; |
1015 | 24236869 | bellard | } |
1016 | 24236869 | bellard | |
1017 | 2f9606b3 | aliguori | |
1018 | 2f9606b3 | aliguori | /*
|
1019 | 2f9606b3 | aliguori | * Called to read a chunk of data from the client socket. The data may
|
1020 | 2f9606b3 | aliguori | * be the raw data, or may need to be further decoded by SASL.
|
1021 | 2f9606b3 | aliguori | * The data will be read either straight from to the socket, or
|
1022 | 2f9606b3 | aliguori | * read via the GNUTLS wrappers, if TLS/SSL encryption is enabled
|
1023 | 2f9606b3 | aliguori | *
|
1024 | 2f9606b3 | aliguori | * NB, it is theoretically possible to have 2 layers of encryption,
|
1025 | 2f9606b3 | aliguori | * both SASL, and this TLS layer. It is highly unlikely in practice
|
1026 | 2f9606b3 | aliguori | * though, since SASL encryption will typically be a no-op if TLS
|
1027 | 2f9606b3 | aliguori | * is active
|
1028 | 2f9606b3 | aliguori | *
|
1029 | 2f9606b3 | aliguori | * Returns the number of bytes read, which may be less than
|
1030 | 2f9606b3 | aliguori | * the requested 'datalen' if the socket would block. Returns
|
1031 | 2f9606b3 | aliguori | * -1 on error, and disconnects the client socket.
|
1032 | 2f9606b3 | aliguori | */
|
1033 | 2f9606b3 | aliguori | long vnc_client_read_buf(VncState *vs, uint8_t *data, size_t datalen)
|
1034 | 24236869 | bellard | { |
1035 | ceb5caaf | bellard | long ret;
|
1036 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
1037 | 5fb6c7a8 | aliguori | if (vs->tls.session) {
|
1038 | 28a76be8 | aliguori | ret = gnutls_read(vs->tls.session, data, datalen); |
1039 | 28a76be8 | aliguori | if (ret < 0) { |
1040 | 28a76be8 | aliguori | if (ret == GNUTLS_E_AGAIN)
|
1041 | 28a76be8 | aliguori | errno = EAGAIN; |
1042 | 28a76be8 | aliguori | else
|
1043 | 28a76be8 | aliguori | errno = EIO; |
1044 | 28a76be8 | aliguori | ret = -1;
|
1045 | 28a76be8 | aliguori | } |
1046 | 8d5d2d4c | ths | } else
|
1047 | 8d5d2d4c | ths | #endif /* CONFIG_VNC_TLS */ |
1048 | 28a76be8 | aliguori | ret = recv(vs->csock, data, datalen, 0);
|
1049 | 2f9606b3 | aliguori | VNC_DEBUG("Read wire %p %d -> %ld\n", data, datalen, ret);
|
1050 | 2f9606b3 | aliguori | return vnc_client_io_error(vs, ret, socket_error());
|
1051 | 2f9606b3 | aliguori | } |
1052 | 24236869 | bellard | |
1053 | 2f9606b3 | aliguori | |
1054 | 2f9606b3 | aliguori | /*
|
1055 | 2f9606b3 | aliguori | * Called to read data from the client socket to the input buffer,
|
1056 | 2f9606b3 | aliguori | * when not using any SASL SSF encryption layers. Will read as much
|
1057 | 2f9606b3 | aliguori | * data as possible without blocking.
|
1058 | 2f9606b3 | aliguori | *
|
1059 | 2f9606b3 | aliguori | * Returns the number of bytes read. Returns -1 on error, and
|
1060 | 2f9606b3 | aliguori | * disconnects the client socket.
|
1061 | 2f9606b3 | aliguori | */
|
1062 | 2f9606b3 | aliguori | static long vnc_client_read_plain(VncState *vs) |
1063 | 2f9606b3 | aliguori | { |
1064 | 2f9606b3 | aliguori | int ret;
|
1065 | 2f9606b3 | aliguori | VNC_DEBUG("Read plain %p size %d offset %d\n",
|
1066 | 2f9606b3 | aliguori | vs->input.buffer, vs->input.capacity, vs->input.offset); |
1067 | 2f9606b3 | aliguori | buffer_reserve(&vs->input, 4096);
|
1068 | 2f9606b3 | aliguori | ret = vnc_client_read_buf(vs, buffer_end(&vs->input), 4096);
|
1069 | 2f9606b3 | aliguori | if (!ret)
|
1070 | 2f9606b3 | aliguori | return 0; |
1071 | 24236869 | bellard | vs->input.offset += ret; |
1072 | 2f9606b3 | aliguori | return ret;
|
1073 | 2f9606b3 | aliguori | } |
1074 | 2f9606b3 | aliguori | |
1075 | 2f9606b3 | aliguori | |
1076 | 2f9606b3 | aliguori | /*
|
1077 | 2f9606b3 | aliguori | * First function called whenever there is more data to be read from
|
1078 | 2f9606b3 | aliguori | * the client socket. Will delegate actual work according to whether
|
1079 | 2f9606b3 | aliguori | * SASL SSF layers are enabled (thus requiring decryption calls)
|
1080 | 2f9606b3 | aliguori | */
|
1081 | 2f9606b3 | aliguori | void vnc_client_read(void *opaque) |
1082 | 2f9606b3 | aliguori | { |
1083 | 2f9606b3 | aliguori | VncState *vs = opaque; |
1084 | 2f9606b3 | aliguori | long ret;
|
1085 | 2f9606b3 | aliguori | |
1086 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_SASL
|
1087 | 2f9606b3 | aliguori | if (vs->sasl.conn && vs->sasl.runSSF)
|
1088 | 2f9606b3 | aliguori | ret = vnc_client_read_sasl(vs); |
1089 | 2f9606b3 | aliguori | else
|
1090 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_SASL */ |
1091 | 2f9606b3 | aliguori | ret = vnc_client_read_plain(vs); |
1092 | 2f9606b3 | aliguori | if (!ret)
|
1093 | 28a76be8 | aliguori | return;
|
1094 | 24236869 | bellard | |
1095 | 24236869 | bellard | while (vs->read_handler && vs->input.offset >= vs->read_handler_expect) {
|
1096 | 28a76be8 | aliguori | size_t len = vs->read_handler_expect; |
1097 | 28a76be8 | aliguori | int ret;
|
1098 | 28a76be8 | aliguori | |
1099 | 28a76be8 | aliguori | ret = vs->read_handler(vs, vs->input.buffer, len); |
1100 | 28a76be8 | aliguori | if (vs->csock == -1) |
1101 | 28a76be8 | aliguori | return;
|
1102 | 28a76be8 | aliguori | |
1103 | 28a76be8 | aliguori | if (!ret) {
|
1104 | 28a76be8 | aliguori | memmove(vs->input.buffer, vs->input.buffer + len, (vs->input.offset - len)); |
1105 | 28a76be8 | aliguori | vs->input.offset -= len; |
1106 | 28a76be8 | aliguori | } else {
|
1107 | 28a76be8 | aliguori | vs->read_handler_expect = ret; |
1108 | 28a76be8 | aliguori | } |
1109 | 24236869 | bellard | } |
1110 | 24236869 | bellard | } |
1111 | 24236869 | bellard | |
1112 | 5fb6c7a8 | aliguori | void vnc_write(VncState *vs, const void *data, size_t len) |
1113 | 24236869 | bellard | { |
1114 | 24236869 | bellard | buffer_reserve(&vs->output, len); |
1115 | 24236869 | bellard | |
1116 | 24236869 | bellard | if (buffer_empty(&vs->output)) {
|
1117 | 28a76be8 | aliguori | qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, vnc_client_write, vs);
|
1118 | 24236869 | bellard | } |
1119 | 24236869 | bellard | |
1120 | 24236869 | bellard | buffer_append(&vs->output, data, len); |
1121 | 24236869 | bellard | } |
1122 | 24236869 | bellard | |
1123 | 5fb6c7a8 | aliguori | void vnc_write_s32(VncState *vs, int32_t value)
|
1124 | 24236869 | bellard | { |
1125 | 24236869 | bellard | vnc_write_u32(vs, *(uint32_t *)&value); |
1126 | 24236869 | bellard | } |
1127 | 24236869 | bellard | |
1128 | 5fb6c7a8 | aliguori | void vnc_write_u32(VncState *vs, uint32_t value)
|
1129 | 24236869 | bellard | { |
1130 | 24236869 | bellard | uint8_t buf[4];
|
1131 | 24236869 | bellard | |
1132 | 24236869 | bellard | buf[0] = (value >> 24) & 0xFF; |
1133 | 24236869 | bellard | buf[1] = (value >> 16) & 0xFF; |
1134 | 24236869 | bellard | buf[2] = (value >> 8) & 0xFF; |
1135 | 24236869 | bellard | buf[3] = value & 0xFF; |
1136 | 24236869 | bellard | |
1137 | 24236869 | bellard | vnc_write(vs, buf, 4);
|
1138 | 24236869 | bellard | } |
1139 | 24236869 | bellard | |
1140 | 5fb6c7a8 | aliguori | void vnc_write_u16(VncState *vs, uint16_t value)
|
1141 | 24236869 | bellard | { |
1142 | 64f5a135 | bellard | uint8_t buf[2];
|
1143 | 24236869 | bellard | |
1144 | 24236869 | bellard | buf[0] = (value >> 8) & 0xFF; |
1145 | 24236869 | bellard | buf[1] = value & 0xFF; |
1146 | 24236869 | bellard | |
1147 | 24236869 | bellard | vnc_write(vs, buf, 2);
|
1148 | 24236869 | bellard | } |
1149 | 24236869 | bellard | |
1150 | 5fb6c7a8 | aliguori | void vnc_write_u8(VncState *vs, uint8_t value)
|
1151 | 24236869 | bellard | { |
1152 | 24236869 | bellard | vnc_write(vs, (char *)&value, 1); |
1153 | 24236869 | bellard | } |
1154 | 24236869 | bellard | |
1155 | 5fb6c7a8 | aliguori | void vnc_flush(VncState *vs)
|
1156 | 24236869 | bellard | { |
1157 | 24236869 | bellard | if (vs->output.offset)
|
1158 | 28a76be8 | aliguori | vnc_client_write(vs); |
1159 | 24236869 | bellard | } |
1160 | 24236869 | bellard | |
1161 | 5fb6c7a8 | aliguori | uint8_t read_u8(uint8_t *data, size_t offset) |
1162 | 24236869 | bellard | { |
1163 | 24236869 | bellard | return data[offset];
|
1164 | 24236869 | bellard | } |
1165 | 24236869 | bellard | |
1166 | 5fb6c7a8 | aliguori | uint16_t read_u16(uint8_t *data, size_t offset) |
1167 | 24236869 | bellard | { |
1168 | 24236869 | bellard | return ((data[offset] & 0xFF) << 8) | (data[offset + 1] & 0xFF); |
1169 | 24236869 | bellard | } |
1170 | 24236869 | bellard | |
1171 | 5fb6c7a8 | aliguori | int32_t read_s32(uint8_t *data, size_t offset) |
1172 | 24236869 | bellard | { |
1173 | 24236869 | bellard | return (int32_t)((data[offset] << 24) | (data[offset + 1] << 16) | |
1174 | 28a76be8 | aliguori | (data[offset + 2] << 8) | data[offset + 3]); |
1175 | 24236869 | bellard | } |
1176 | 24236869 | bellard | |
1177 | 5fb6c7a8 | aliguori | uint32_t read_u32(uint8_t *data, size_t offset) |
1178 | 24236869 | bellard | { |
1179 | 24236869 | bellard | return ((data[offset] << 24) | (data[offset + 1] << 16) | |
1180 | 28a76be8 | aliguori | (data[offset + 2] << 8) | data[offset + 3]); |
1181 | 24236869 | bellard | } |
1182 | 24236869 | bellard | |
1183 | 60fe76f3 | ths | static void client_cut_text(VncState *vs, size_t len, uint8_t *text) |
1184 | 24236869 | bellard | { |
1185 | 24236869 | bellard | } |
1186 | 24236869 | bellard | |
1187 | 564c337e | bellard | static void check_pointer_type_change(VncState *vs, int absolute) |
1188 | 564c337e | bellard | { |
1189 | 29fa4ed9 | aliguori | if (vnc_has_feature(vs, VNC_FEATURE_POINTER_TYPE_CHANGE) && vs->absolute != absolute) {
|
1190 | 28a76be8 | aliguori | vnc_write_u8(vs, 0);
|
1191 | 28a76be8 | aliguori | vnc_write_u8(vs, 0);
|
1192 | 28a76be8 | aliguori | vnc_write_u16(vs, 1);
|
1193 | 28a76be8 | aliguori | vnc_framebuffer_update(vs, absolute, 0,
|
1194 | 28a76be8 | aliguori | ds_get_width(vs->ds), ds_get_height(vs->ds), |
1195 | 29fa4ed9 | aliguori | VNC_ENCODING_POINTER_TYPE_CHANGE); |
1196 | 28a76be8 | aliguori | vnc_flush(vs); |
1197 | 564c337e | bellard | } |
1198 | 564c337e | bellard | vs->absolute = absolute; |
1199 | 564c337e | bellard | } |
1200 | 564c337e | bellard | |
1201 | 24236869 | bellard | static void pointer_event(VncState *vs, int button_mask, int x, int y) |
1202 | 24236869 | bellard | { |
1203 | 24236869 | bellard | int buttons = 0; |
1204 | 24236869 | bellard | int dz = 0; |
1205 | 24236869 | bellard | |
1206 | 24236869 | bellard | if (button_mask & 0x01) |
1207 | 28a76be8 | aliguori | buttons |= MOUSE_EVENT_LBUTTON; |
1208 | 24236869 | bellard | if (button_mask & 0x02) |
1209 | 28a76be8 | aliguori | buttons |= MOUSE_EVENT_MBUTTON; |
1210 | 24236869 | bellard | if (button_mask & 0x04) |
1211 | 28a76be8 | aliguori | buttons |= MOUSE_EVENT_RBUTTON; |
1212 | 24236869 | bellard | if (button_mask & 0x08) |
1213 | 28a76be8 | aliguori | dz = -1;
|
1214 | 24236869 | bellard | if (button_mask & 0x10) |
1215 | 28a76be8 | aliguori | dz = 1;
|
1216 | 564c337e | bellard | |
1217 | 564c337e | bellard | if (vs->absolute) {
|
1218 | 28a76be8 | aliguori | kbd_mouse_event(x * 0x7FFF / (ds_get_width(vs->ds) - 1), |
1219 | 28a76be8 | aliguori | y * 0x7FFF / (ds_get_height(vs->ds) - 1), |
1220 | 28a76be8 | aliguori | dz, buttons); |
1221 | 29fa4ed9 | aliguori | } else if (vnc_has_feature(vs, VNC_FEATURE_POINTER_TYPE_CHANGE)) { |
1222 | 28a76be8 | aliguori | x -= 0x7FFF;
|
1223 | 28a76be8 | aliguori | y -= 0x7FFF;
|
1224 | 24236869 | bellard | |
1225 | 28a76be8 | aliguori | kbd_mouse_event(x, y, dz, buttons); |
1226 | 564c337e | bellard | } else {
|
1227 | 28a76be8 | aliguori | if (vs->last_x != -1) |
1228 | 28a76be8 | aliguori | kbd_mouse_event(x - vs->last_x, |
1229 | 28a76be8 | aliguori | y - vs->last_y, |
1230 | 28a76be8 | aliguori | dz, buttons); |
1231 | 28a76be8 | aliguori | vs->last_x = x; |
1232 | 28a76be8 | aliguori | vs->last_y = y; |
1233 | 24236869 | bellard | } |
1234 | 564c337e | bellard | |
1235 | 564c337e | bellard | check_pointer_type_change(vs, kbd_mouse_is_absolute()); |
1236 | 24236869 | bellard | } |
1237 | 24236869 | bellard | |
1238 | 64f5a135 | bellard | static void reset_keys(VncState *vs) |
1239 | 64f5a135 | bellard | { |
1240 | 64f5a135 | bellard | int i;
|
1241 | 64f5a135 | bellard | for(i = 0; i < 256; i++) { |
1242 | 64f5a135 | bellard | if (vs->modifiers_state[i]) {
|
1243 | 64f5a135 | bellard | if (i & 0x80) |
1244 | 64f5a135 | bellard | kbd_put_keycode(0xe0);
|
1245 | 64f5a135 | bellard | kbd_put_keycode(i | 0x80);
|
1246 | 64f5a135 | bellard | vs->modifiers_state[i] = 0;
|
1247 | 64f5a135 | bellard | } |
1248 | 64f5a135 | bellard | } |
1249 | 64f5a135 | bellard | } |
1250 | 64f5a135 | bellard | |
1251 | a528b80c | balrog | static void press_key(VncState *vs, int keysym) |
1252 | a528b80c | balrog | { |
1253 | 753b4053 | aliguori | kbd_put_keycode(keysym2scancode(vs->vd->kbd_layout, keysym) & 0x7f);
|
1254 | 753b4053 | aliguori | kbd_put_keycode(keysym2scancode(vs->vd->kbd_layout, keysym) | 0x80);
|
1255 | a528b80c | balrog | } |
1256 | a528b80c | balrog | |
1257 | 9ca313aa | aliguori | static void do_key_event(VncState *vs, int down, int keycode, int sym) |
1258 | 24236869 | bellard | { |
1259 | 64f5a135 | bellard | /* QEMU console switch */
|
1260 | 64f5a135 | bellard | switch(keycode) {
|
1261 | 64f5a135 | bellard | case 0x2a: /* Left Shift */ |
1262 | 64f5a135 | bellard | case 0x36: /* Right Shift */ |
1263 | 64f5a135 | bellard | case 0x1d: /* Left CTRL */ |
1264 | 64f5a135 | bellard | case 0x9d: /* Right CTRL */ |
1265 | 64f5a135 | bellard | case 0x38: /* Left ALT */ |
1266 | 64f5a135 | bellard | case 0xb8: /* Right ALT */ |
1267 | 64f5a135 | bellard | if (down)
|
1268 | 64f5a135 | bellard | vs->modifiers_state[keycode] = 1;
|
1269 | 64f5a135 | bellard | else
|
1270 | 64f5a135 | bellard | vs->modifiers_state[keycode] = 0;
|
1271 | 64f5a135 | bellard | break;
|
1272 | 5fafdf24 | ths | case 0x02 ... 0x0a: /* '1' to '9' keys */ |
1273 | 64f5a135 | bellard | if (down && vs->modifiers_state[0x1d] && vs->modifiers_state[0x38]) { |
1274 | 64f5a135 | bellard | /* Reset the modifiers sent to the current console */
|
1275 | 64f5a135 | bellard | reset_keys(vs); |
1276 | 64f5a135 | bellard | console_select(keycode - 0x02);
|
1277 | 64f5a135 | bellard | return;
|
1278 | 64f5a135 | bellard | } |
1279 | 64f5a135 | bellard | break;
|
1280 | 28a76be8 | aliguori | case 0x3a: /* CapsLock */ |
1281 | 28a76be8 | aliguori | case 0x45: /* NumLock */ |
1282 | a528b80c | balrog | if (!down)
|
1283 | a528b80c | balrog | vs->modifiers_state[keycode] ^= 1;
|
1284 | a528b80c | balrog | break;
|
1285 | a528b80c | balrog | } |
1286 | a528b80c | balrog | |
1287 | 753b4053 | aliguori | if (keycode_is_keypad(vs->vd->kbd_layout, keycode)) {
|
1288 | a528b80c | balrog | /* If the numlock state needs to change then simulate an additional
|
1289 | a528b80c | balrog | keypress before sending this one. This will happen if the user
|
1290 | a528b80c | balrog | toggles numlock away from the VNC window.
|
1291 | a528b80c | balrog | */
|
1292 | 753b4053 | aliguori | if (keysym_is_numlock(vs->vd->kbd_layout, sym & 0xFFFF)) { |
1293 | a528b80c | balrog | if (!vs->modifiers_state[0x45]) { |
1294 | a528b80c | balrog | vs->modifiers_state[0x45] = 1; |
1295 | a528b80c | balrog | press_key(vs, 0xff7f);
|
1296 | a528b80c | balrog | } |
1297 | a528b80c | balrog | } else {
|
1298 | a528b80c | balrog | if (vs->modifiers_state[0x45]) { |
1299 | a528b80c | balrog | vs->modifiers_state[0x45] = 0; |
1300 | a528b80c | balrog | press_key(vs, 0xff7f);
|
1301 | a528b80c | balrog | } |
1302 | a528b80c | balrog | } |
1303 | 64f5a135 | bellard | } |
1304 | 24236869 | bellard | |
1305 | 64f5a135 | bellard | if (is_graphic_console()) {
|
1306 | 64f5a135 | bellard | if (keycode & 0x80) |
1307 | 64f5a135 | bellard | kbd_put_keycode(0xe0);
|
1308 | 64f5a135 | bellard | if (down)
|
1309 | 64f5a135 | bellard | kbd_put_keycode(keycode & 0x7f);
|
1310 | 64f5a135 | bellard | else
|
1311 | 64f5a135 | bellard | kbd_put_keycode(keycode | 0x80);
|
1312 | 64f5a135 | bellard | } else {
|
1313 | 64f5a135 | bellard | /* QEMU console emulation */
|
1314 | 64f5a135 | bellard | if (down) {
|
1315 | 64f5a135 | bellard | switch (keycode) {
|
1316 | 64f5a135 | bellard | case 0x2a: /* Left Shift */ |
1317 | 64f5a135 | bellard | case 0x36: /* Right Shift */ |
1318 | 64f5a135 | bellard | case 0x1d: /* Left CTRL */ |
1319 | 64f5a135 | bellard | case 0x9d: /* Right CTRL */ |
1320 | 64f5a135 | bellard | case 0x38: /* Left ALT */ |
1321 | 64f5a135 | bellard | case 0xb8: /* Right ALT */ |
1322 | 64f5a135 | bellard | break;
|
1323 | 64f5a135 | bellard | case 0xc8: |
1324 | 64f5a135 | bellard | kbd_put_keysym(QEMU_KEY_UP); |
1325 | 64f5a135 | bellard | break;
|
1326 | 64f5a135 | bellard | case 0xd0: |
1327 | 64f5a135 | bellard | kbd_put_keysym(QEMU_KEY_DOWN); |
1328 | 64f5a135 | bellard | break;
|
1329 | 64f5a135 | bellard | case 0xcb: |
1330 | 64f5a135 | bellard | kbd_put_keysym(QEMU_KEY_LEFT); |
1331 | 64f5a135 | bellard | break;
|
1332 | 64f5a135 | bellard | case 0xcd: |
1333 | 64f5a135 | bellard | kbd_put_keysym(QEMU_KEY_RIGHT); |
1334 | 64f5a135 | bellard | break;
|
1335 | 64f5a135 | bellard | case 0xd3: |
1336 | 64f5a135 | bellard | kbd_put_keysym(QEMU_KEY_DELETE); |
1337 | 64f5a135 | bellard | break;
|
1338 | 64f5a135 | bellard | case 0xc7: |
1339 | 64f5a135 | bellard | kbd_put_keysym(QEMU_KEY_HOME); |
1340 | 64f5a135 | bellard | break;
|
1341 | 64f5a135 | bellard | case 0xcf: |
1342 | 64f5a135 | bellard | kbd_put_keysym(QEMU_KEY_END); |
1343 | 64f5a135 | bellard | break;
|
1344 | 64f5a135 | bellard | case 0xc9: |
1345 | 64f5a135 | bellard | kbd_put_keysym(QEMU_KEY_PAGEUP); |
1346 | 64f5a135 | bellard | break;
|
1347 | 64f5a135 | bellard | case 0xd1: |
1348 | 64f5a135 | bellard | kbd_put_keysym(QEMU_KEY_PAGEDOWN); |
1349 | 64f5a135 | bellard | break;
|
1350 | 64f5a135 | bellard | default:
|
1351 | 64f5a135 | bellard | kbd_put_keysym(sym); |
1352 | 64f5a135 | bellard | break;
|
1353 | 64f5a135 | bellard | } |
1354 | 64f5a135 | bellard | } |
1355 | 64f5a135 | bellard | } |
1356 | 24236869 | bellard | } |
1357 | 24236869 | bellard | |
1358 | bdbd7676 | bellard | static void key_event(VncState *vs, int down, uint32_t sym) |
1359 | bdbd7676 | bellard | { |
1360 | 9ca313aa | aliguori | int keycode;
|
1361 | 9ca313aa | aliguori | |
1362 | a528b80c | balrog | if (sym >= 'A' && sym <= 'Z' && is_graphic_console()) |
1363 | 28a76be8 | aliguori | sym = sym - 'A' + 'a'; |
1364 | 9ca313aa | aliguori | |
1365 | 753b4053 | aliguori | keycode = keysym2scancode(vs->vd->kbd_layout, sym & 0xFFFF);
|
1366 | 9ca313aa | aliguori | do_key_event(vs, down, keycode, sym); |
1367 | 9ca313aa | aliguori | } |
1368 | 9ca313aa | aliguori | |
1369 | 9ca313aa | aliguori | static void ext_key_event(VncState *vs, int down, |
1370 | 9ca313aa | aliguori | uint32_t sym, uint16_t keycode) |
1371 | 9ca313aa | aliguori | { |
1372 | 9ca313aa | aliguori | /* if the user specifies a keyboard layout, always use it */
|
1373 | 9ca313aa | aliguori | if (keyboard_layout)
|
1374 | 9ca313aa | aliguori | key_event(vs, down, sym); |
1375 | 9ca313aa | aliguori | else
|
1376 | 9ca313aa | aliguori | do_key_event(vs, down, keycode, sym); |
1377 | bdbd7676 | bellard | } |
1378 | bdbd7676 | bellard | |
1379 | 24236869 | bellard | static void framebuffer_update_request(VncState *vs, int incremental, |
1380 | 28a76be8 | aliguori | int x_position, int y_position, |
1381 | 28a76be8 | aliguori | int w, int h) |
1382 | 24236869 | bellard | { |
1383 | 0e1f5a0c | aliguori | if (x_position > ds_get_width(vs->ds))
|
1384 | 0e1f5a0c | aliguori | x_position = ds_get_width(vs->ds); |
1385 | 0e1f5a0c | aliguori | if (y_position > ds_get_height(vs->ds))
|
1386 | 0e1f5a0c | aliguori | y_position = ds_get_height(vs->ds); |
1387 | 0e1f5a0c | aliguori | if (x_position + w >= ds_get_width(vs->ds))
|
1388 | 0e1f5a0c | aliguori | w = ds_get_width(vs->ds) - x_position; |
1389 | 0e1f5a0c | aliguori | if (y_position + h >= ds_get_height(vs->ds))
|
1390 | 0e1f5a0c | aliguori | h = ds_get_height(vs->ds) - y_position; |
1391 | cf2d385c | ths | |
1392 | 24236869 | bellard | int i;
|
1393 | 24236869 | bellard | vs->need_update = 1;
|
1394 | 24236869 | bellard | if (!incremental) {
|
1395 | 28a76be8 | aliguori | char *old_row = vs->old_data + y_position * ds_get_linesize(vs->ds);
|
1396 | 24236869 | bellard | |
1397 | 28a76be8 | aliguori | for (i = 0; i < h; i++) { |
1398 | 5fafdf24 | ths | vnc_set_bits(vs->dirty_row[y_position + i], |
1399 | 0e1f5a0c | aliguori | (ds_get_width(vs->ds) / 16), VNC_DIRTY_WORDS);
|
1400 | 28a76be8 | aliguori | memset(old_row, 42, ds_get_width(vs->ds) * ds_get_bytes_per_pixel(vs->ds));
|
1401 | 28a76be8 | aliguori | old_row += ds_get_linesize(vs->ds); |
1402 | 28a76be8 | aliguori | } |
1403 | 24236869 | bellard | } |
1404 | 24236869 | bellard | } |
1405 | 24236869 | bellard | |
1406 | 9ca313aa | aliguori | static void send_ext_key_event_ack(VncState *vs) |
1407 | 9ca313aa | aliguori | { |
1408 | 9ca313aa | aliguori | vnc_write_u8(vs, 0);
|
1409 | 9ca313aa | aliguori | vnc_write_u8(vs, 0);
|
1410 | 9ca313aa | aliguori | vnc_write_u16(vs, 1);
|
1411 | 29fa4ed9 | aliguori | vnc_framebuffer_update(vs, 0, 0, ds_get_width(vs->ds), ds_get_height(vs->ds), |
1412 | 29fa4ed9 | aliguori | VNC_ENCODING_EXT_KEY_EVENT); |
1413 | 9ca313aa | aliguori | vnc_flush(vs); |
1414 | 9ca313aa | aliguori | } |
1415 | 9ca313aa | aliguori | |
1416 | 429a8ed3 | malc | static void send_ext_audio_ack(VncState *vs) |
1417 | 429a8ed3 | malc | { |
1418 | 429a8ed3 | malc | vnc_write_u8(vs, 0);
|
1419 | 429a8ed3 | malc | vnc_write_u8(vs, 0);
|
1420 | 429a8ed3 | malc | vnc_write_u16(vs, 1);
|
1421 | 29fa4ed9 | aliguori | vnc_framebuffer_update(vs, 0, 0, ds_get_width(vs->ds), ds_get_height(vs->ds), |
1422 | 29fa4ed9 | aliguori | VNC_ENCODING_AUDIO); |
1423 | 429a8ed3 | malc | vnc_flush(vs); |
1424 | 429a8ed3 | malc | } |
1425 | 429a8ed3 | malc | |
1426 | 24236869 | bellard | static void set_encodings(VncState *vs, int32_t *encodings, size_t n_encodings) |
1427 | 24236869 | bellard | { |
1428 | 24236869 | bellard | int i;
|
1429 | 29fa4ed9 | aliguori | unsigned int enc = 0; |
1430 | 24236869 | bellard | |
1431 | 059cef40 | aliguori | vnc_zlib_init(vs); |
1432 | 29fa4ed9 | aliguori | vs->features = 0;
|
1433 | fb437313 | aliguori | vs->vnc_encoding = 0;
|
1434 | fb437313 | aliguori | vs->tight_compression = 9;
|
1435 | fb437313 | aliguori | vs->tight_quality = 9;
|
1436 | 564c337e | bellard | vs->absolute = -1;
|
1437 | 24236869 | bellard | |
1438 | 24236869 | bellard | for (i = n_encodings - 1; i >= 0; i--) { |
1439 | 29fa4ed9 | aliguori | enc = encodings[i]; |
1440 | 29fa4ed9 | aliguori | switch (enc) {
|
1441 | 29fa4ed9 | aliguori | case VNC_ENCODING_RAW:
|
1442 | fb437313 | aliguori | vs->vnc_encoding = enc; |
1443 | 29fa4ed9 | aliguori | break;
|
1444 | 29fa4ed9 | aliguori | case VNC_ENCODING_COPYRECT:
|
1445 | 753b4053 | aliguori | vs->features |= VNC_FEATURE_COPYRECT_MASK; |
1446 | 29fa4ed9 | aliguori | break;
|
1447 | 29fa4ed9 | aliguori | case VNC_ENCODING_HEXTILE:
|
1448 | 29fa4ed9 | aliguori | vs->features |= VNC_FEATURE_HEXTILE_MASK; |
1449 | fb437313 | aliguori | vs->vnc_encoding = enc; |
1450 | 29fa4ed9 | aliguori | break;
|
1451 | 059cef40 | aliguori | case VNC_ENCODING_ZLIB:
|
1452 | 059cef40 | aliguori | vs->features |= VNC_FEATURE_ZLIB_MASK; |
1453 | 059cef40 | aliguori | vs->vnc_encoding = enc; |
1454 | 059cef40 | aliguori | break;
|
1455 | 29fa4ed9 | aliguori | case VNC_ENCODING_DESKTOPRESIZE:
|
1456 | 29fa4ed9 | aliguori | vs->features |= VNC_FEATURE_RESIZE_MASK; |
1457 | 29fa4ed9 | aliguori | break;
|
1458 | 29fa4ed9 | aliguori | case VNC_ENCODING_POINTER_TYPE_CHANGE:
|
1459 | 29fa4ed9 | aliguori | vs->features |= VNC_FEATURE_POINTER_TYPE_CHANGE_MASK; |
1460 | 29fa4ed9 | aliguori | break;
|
1461 | 29fa4ed9 | aliguori | case VNC_ENCODING_EXT_KEY_EVENT:
|
1462 | 9ca313aa | aliguori | send_ext_key_event_ack(vs); |
1463 | 9ca313aa | aliguori | break;
|
1464 | 29fa4ed9 | aliguori | case VNC_ENCODING_AUDIO:
|
1465 | 429a8ed3 | malc | send_ext_audio_ack(vs); |
1466 | 429a8ed3 | malc | break;
|
1467 | 29fa4ed9 | aliguori | case VNC_ENCODING_WMVi:
|
1468 | 29fa4ed9 | aliguori | vs->features |= VNC_FEATURE_WMVI_MASK; |
1469 | ca4cca4d | aliguori | break;
|
1470 | fb437313 | aliguori | case VNC_ENCODING_COMPRESSLEVEL0 ... VNC_ENCODING_COMPRESSLEVEL0 + 9: |
1471 | fb437313 | aliguori | vs->tight_compression = (enc & 0x0F);
|
1472 | fb437313 | aliguori | break;
|
1473 | fb437313 | aliguori | case VNC_ENCODING_QUALITYLEVEL0 ... VNC_ENCODING_QUALITYLEVEL0 + 9: |
1474 | fb437313 | aliguori | vs->tight_quality = (enc & 0x0F);
|
1475 | fb437313 | aliguori | break;
|
1476 | 29fa4ed9 | aliguori | default:
|
1477 | 29fa4ed9 | aliguori | VNC_DEBUG("Unknown encoding: %d (0x%.8x): %d\n", i, enc, enc);
|
1478 | 29fa4ed9 | aliguori | break;
|
1479 | 29fa4ed9 | aliguori | } |
1480 | 24236869 | bellard | } |
1481 | 564c337e | bellard | |
1482 | 564c337e | bellard | check_pointer_type_change(vs, kbd_mouse_is_absolute()); |
1483 | 24236869 | bellard | } |
1484 | 24236869 | bellard | |
1485 | 6cec5487 | aliguori | static void set_pixel_conversion(VncState *vs) |
1486 | 6cec5487 | aliguori | { |
1487 | 6cec5487 | aliguori | if ((vs->clientds.flags & QEMU_BIG_ENDIAN_FLAG) ==
|
1488 | 6cec5487 | aliguori | (vs->ds->surface->flags & QEMU_BIG_ENDIAN_FLAG) && |
1489 | 6cec5487 | aliguori | !memcmp(&(vs->clientds.pf), &(vs->ds->surface->pf), sizeof(PixelFormat))) {
|
1490 | 6cec5487 | aliguori | vs->write_pixels = vnc_write_pixels_copy; |
1491 | 6cec5487 | aliguori | switch (vs->ds->surface->pf.bits_per_pixel) {
|
1492 | 6cec5487 | aliguori | case 8: |
1493 | 6cec5487 | aliguori | vs->send_hextile_tile = send_hextile_tile_8; |
1494 | 6cec5487 | aliguori | break;
|
1495 | 6cec5487 | aliguori | case 16: |
1496 | 6cec5487 | aliguori | vs->send_hextile_tile = send_hextile_tile_16; |
1497 | 6cec5487 | aliguori | break;
|
1498 | 6cec5487 | aliguori | case 32: |
1499 | 6cec5487 | aliguori | vs->send_hextile_tile = send_hextile_tile_32; |
1500 | 6cec5487 | aliguori | break;
|
1501 | 6cec5487 | aliguori | } |
1502 | 6cec5487 | aliguori | } else {
|
1503 | 6cec5487 | aliguori | vs->write_pixels = vnc_write_pixels_generic; |
1504 | 6cec5487 | aliguori | switch (vs->ds->surface->pf.bits_per_pixel) {
|
1505 | 6cec5487 | aliguori | case 8: |
1506 | 6cec5487 | aliguori | vs->send_hextile_tile = send_hextile_tile_generic_8; |
1507 | 6cec5487 | aliguori | break;
|
1508 | 6cec5487 | aliguori | case 16: |
1509 | 6cec5487 | aliguori | vs->send_hextile_tile = send_hextile_tile_generic_16; |
1510 | 6cec5487 | aliguori | break;
|
1511 | 6cec5487 | aliguori | case 32: |
1512 | 6cec5487 | aliguori | vs->send_hextile_tile = send_hextile_tile_generic_32; |
1513 | 6cec5487 | aliguori | break;
|
1514 | 6cec5487 | aliguori | } |
1515 | 6cec5487 | aliguori | } |
1516 | 6cec5487 | aliguori | } |
1517 | 6cec5487 | aliguori | |
1518 | 24236869 | bellard | static void set_pixel_format(VncState *vs, |
1519 | 28a76be8 | aliguori | int bits_per_pixel, int depth, |
1520 | 28a76be8 | aliguori | int big_endian_flag, int true_color_flag, |
1521 | 28a76be8 | aliguori | int red_max, int green_max, int blue_max, |
1522 | 28a76be8 | aliguori | int red_shift, int green_shift, int blue_shift) |
1523 | 24236869 | bellard | { |
1524 | 3512779a | bellard | if (!true_color_flag) {
|
1525 | 28a76be8 | aliguori | vnc_client_error(vs); |
1526 | 3512779a | bellard | return;
|
1527 | 3512779a | bellard | } |
1528 | 24236869 | bellard | |
1529 | 6cec5487 | aliguori | vs->clientds = vs->serverds; |
1530 | 6cec5487 | aliguori | vs->clientds.pf.rmax = red_max; |
1531 | 90a1e3c0 | aliguori | count_bits(vs->clientds.pf.rbits, red_max); |
1532 | 6cec5487 | aliguori | vs->clientds.pf.rshift = red_shift; |
1533 | 6cec5487 | aliguori | vs->clientds.pf.rmask = red_max << red_shift; |
1534 | 6cec5487 | aliguori | vs->clientds.pf.gmax = green_max; |
1535 | 90a1e3c0 | aliguori | count_bits(vs->clientds.pf.gbits, green_max); |
1536 | 6cec5487 | aliguori | vs->clientds.pf.gshift = green_shift; |
1537 | 6cec5487 | aliguori | vs->clientds.pf.gmask = green_max << green_shift; |
1538 | 6cec5487 | aliguori | vs->clientds.pf.bmax = blue_max; |
1539 | 90a1e3c0 | aliguori | count_bits(vs->clientds.pf.bbits, blue_max); |
1540 | 6cec5487 | aliguori | vs->clientds.pf.bshift = blue_shift; |
1541 | 6cec5487 | aliguori | vs->clientds.pf.bmask = blue_max << blue_shift; |
1542 | 6cec5487 | aliguori | vs->clientds.pf.bits_per_pixel = bits_per_pixel; |
1543 | 6cec5487 | aliguori | vs->clientds.pf.bytes_per_pixel = bits_per_pixel / 8;
|
1544 | 6cec5487 | aliguori | vs->clientds.pf.depth = bits_per_pixel == 32 ? 24 : bits_per_pixel; |
1545 | 6cec5487 | aliguori | vs->clientds.flags = big_endian_flag ? QEMU_BIG_ENDIAN_FLAG : 0x00;
|
1546 | 6cec5487 | aliguori | |
1547 | 6cec5487 | aliguori | set_pixel_conversion(vs); |
1548 | 24236869 | bellard | |
1549 | 24236869 | bellard | vga_hw_invalidate(); |
1550 | 24236869 | bellard | vga_hw_update(); |
1551 | 24236869 | bellard | } |
1552 | 24236869 | bellard | |
1553 | ca4cca4d | aliguori | static void pixel_format_message (VncState *vs) { |
1554 | ca4cca4d | aliguori | char pad[3] = { 0, 0, 0 }; |
1555 | ca4cca4d | aliguori | |
1556 | 6cec5487 | aliguori | vnc_write_u8(vs, vs->ds->surface->pf.bits_per_pixel); /* bits-per-pixel */
|
1557 | 6cec5487 | aliguori | vnc_write_u8(vs, vs->ds->surface->pf.depth); /* depth */
|
1558 | ca4cca4d | aliguori | |
1559 | ca4cca4d | aliguori | #ifdef WORDS_BIGENDIAN
|
1560 | ca4cca4d | aliguori | vnc_write_u8(vs, 1); /* big-endian-flag */ |
1561 | ca4cca4d | aliguori | #else
|
1562 | ca4cca4d | aliguori | vnc_write_u8(vs, 0); /* big-endian-flag */ |
1563 | ca4cca4d | aliguori | #endif
|
1564 | ca4cca4d | aliguori | vnc_write_u8(vs, 1); /* true-color-flag */ |
1565 | 6cec5487 | aliguori | vnc_write_u16(vs, vs->ds->surface->pf.rmax); /* red-max */
|
1566 | 6cec5487 | aliguori | vnc_write_u16(vs, vs->ds->surface->pf.gmax); /* green-max */
|
1567 | 6cec5487 | aliguori | vnc_write_u16(vs, vs->ds->surface->pf.bmax); /* blue-max */
|
1568 | 6cec5487 | aliguori | vnc_write_u8(vs, vs->ds->surface->pf.rshift); /* red-shift */
|
1569 | 6cec5487 | aliguori | vnc_write_u8(vs, vs->ds->surface->pf.gshift); /* green-shift */
|
1570 | 6cec5487 | aliguori | vnc_write_u8(vs, vs->ds->surface->pf.bshift); /* blue-shift */
|
1571 | 6cec5487 | aliguori | if (vs->ds->surface->pf.bits_per_pixel == 32) |
1572 | ca4cca4d | aliguori | vs->send_hextile_tile = send_hextile_tile_32; |
1573 | 6cec5487 | aliguori | else if (vs->ds->surface->pf.bits_per_pixel == 16) |
1574 | ca4cca4d | aliguori | vs->send_hextile_tile = send_hextile_tile_16; |
1575 | 6cec5487 | aliguori | else if (vs->ds->surface->pf.bits_per_pixel == 8) |
1576 | ca4cca4d | aliguori | vs->send_hextile_tile = send_hextile_tile_8; |
1577 | 6cec5487 | aliguori | vs->clientds = *(vs->ds->surface); |
1578 | 6cec5487 | aliguori | vs->clientds.flags |= ~QEMU_ALLOCATED_FLAG; |
1579 | ca4cca4d | aliguori | vs->write_pixels = vnc_write_pixels_copy; |
1580 | ca4cca4d | aliguori | |
1581 | ca4cca4d | aliguori | vnc_write(vs, pad, 3); /* padding */ |
1582 | ca4cca4d | aliguori | } |
1583 | ca4cca4d | aliguori | |
1584 | 7d957bd8 | aliguori | static void vnc_dpy_setdata(DisplayState *ds) |
1585 | 7d957bd8 | aliguori | { |
1586 | 7d957bd8 | aliguori | /* We don't have to do anything */
|
1587 | 7d957bd8 | aliguori | } |
1588 | 7d957bd8 | aliguori | |
1589 | 753b4053 | aliguori | static void vnc_colordepth(VncState *vs) |
1590 | 7eac3a87 | aliguori | { |
1591 | 753b4053 | aliguori | if (vnc_has_feature(vs, VNC_FEATURE_WMVI)) {
|
1592 | ca4cca4d | aliguori | /* Sending a WMVi message to notify the client*/
|
1593 | ca4cca4d | aliguori | vnc_write_u8(vs, 0); /* msg id */ |
1594 | ca4cca4d | aliguori | vnc_write_u8(vs, 0);
|
1595 | ca4cca4d | aliguori | vnc_write_u16(vs, 1); /* number of rects */ |
1596 | 753b4053 | aliguori | vnc_framebuffer_update(vs, 0, 0, ds_get_width(vs->ds), |
1597 | 753b4053 | aliguori | ds_get_height(vs->ds), VNC_ENCODING_WMVi); |
1598 | ca4cca4d | aliguori | pixel_format_message(vs); |
1599 | ca4cca4d | aliguori | vnc_flush(vs); |
1600 | 7eac3a87 | aliguori | } else {
|
1601 | 6cec5487 | aliguori | set_pixel_conversion(vs); |
1602 | 7eac3a87 | aliguori | } |
1603 | 7eac3a87 | aliguori | } |
1604 | 7eac3a87 | aliguori | |
1605 | 60fe76f3 | ths | static int protocol_client_msg(VncState *vs, uint8_t *data, size_t len) |
1606 | 24236869 | bellard | { |
1607 | 24236869 | bellard | int i;
|
1608 | 24236869 | bellard | uint16_t limit; |
1609 | 24236869 | bellard | |
1610 | 24236869 | bellard | switch (data[0]) { |
1611 | 24236869 | bellard | case 0: |
1612 | 28a76be8 | aliguori | if (len == 1) |
1613 | 28a76be8 | aliguori | return 20; |
1614 | 28a76be8 | aliguori | |
1615 | 28a76be8 | aliguori | set_pixel_format(vs, read_u8(data, 4), read_u8(data, 5), |
1616 | 28a76be8 | aliguori | read_u8(data, 6), read_u8(data, 7), |
1617 | 28a76be8 | aliguori | read_u16(data, 8), read_u16(data, 10), |
1618 | 28a76be8 | aliguori | read_u16(data, 12), read_u8(data, 14), |
1619 | 28a76be8 | aliguori | read_u8(data, 15), read_u8(data, 16)); |
1620 | 28a76be8 | aliguori | break;
|
1621 | 24236869 | bellard | case 2: |
1622 | 28a76be8 | aliguori | if (len == 1) |
1623 | 28a76be8 | aliguori | return 4; |
1624 | 24236869 | bellard | |
1625 | 28a76be8 | aliguori | if (len == 4) { |
1626 | 69dd5c9f | aliguori | limit = read_u16(data, 2);
|
1627 | 69dd5c9f | aliguori | if (limit > 0) |
1628 | 69dd5c9f | aliguori | return 4 + (limit * 4); |
1629 | 69dd5c9f | aliguori | } else
|
1630 | 69dd5c9f | aliguori | limit = read_u16(data, 2);
|
1631 | 24236869 | bellard | |
1632 | 28a76be8 | aliguori | for (i = 0; i < limit; i++) { |
1633 | 28a76be8 | aliguori | int32_t val = read_s32(data, 4 + (i * 4)); |
1634 | 28a76be8 | aliguori | memcpy(data + 4 + (i * 4), &val, sizeof(val)); |
1635 | 28a76be8 | aliguori | } |
1636 | 24236869 | bellard | |
1637 | 28a76be8 | aliguori | set_encodings(vs, (int32_t *)(data + 4), limit);
|
1638 | 28a76be8 | aliguori | break;
|
1639 | 24236869 | bellard | case 3: |
1640 | 28a76be8 | aliguori | if (len == 1) |
1641 | 28a76be8 | aliguori | return 10; |
1642 | 24236869 | bellard | |
1643 | 28a76be8 | aliguori | framebuffer_update_request(vs, |
1644 | 28a76be8 | aliguori | read_u8(data, 1), read_u16(data, 2), read_u16(data, 4), |
1645 | 28a76be8 | aliguori | read_u16(data, 6), read_u16(data, 8)); |
1646 | 28a76be8 | aliguori | break;
|
1647 | 24236869 | bellard | case 4: |
1648 | 28a76be8 | aliguori | if (len == 1) |
1649 | 28a76be8 | aliguori | return 8; |
1650 | 24236869 | bellard | |
1651 | 28a76be8 | aliguori | key_event(vs, read_u8(data, 1), read_u32(data, 4)); |
1652 | 28a76be8 | aliguori | break;
|
1653 | 24236869 | bellard | case 5: |
1654 | 28a76be8 | aliguori | if (len == 1) |
1655 | 28a76be8 | aliguori | return 6; |
1656 | 24236869 | bellard | |
1657 | 28a76be8 | aliguori | pointer_event(vs, read_u8(data, 1), read_u16(data, 2), read_u16(data, 4)); |
1658 | 28a76be8 | aliguori | break;
|
1659 | 24236869 | bellard | case 6: |
1660 | 28a76be8 | aliguori | if (len == 1) |
1661 | 28a76be8 | aliguori | return 8; |
1662 | 24236869 | bellard | |
1663 | 28a76be8 | aliguori | if (len == 8) { |
1664 | baa7666c | ths | uint32_t dlen = read_u32(data, 4);
|
1665 | baa7666c | ths | if (dlen > 0) |
1666 | baa7666c | ths | return 8 + dlen; |
1667 | baa7666c | ths | } |
1668 | 24236869 | bellard | |
1669 | 28a76be8 | aliguori | client_cut_text(vs, read_u32(data, 4), data + 8); |
1670 | 28a76be8 | aliguori | break;
|
1671 | 9ca313aa | aliguori | case 255: |
1672 | 9ca313aa | aliguori | if (len == 1) |
1673 | 9ca313aa | aliguori | return 2; |
1674 | 9ca313aa | aliguori | |
1675 | 9ca313aa | aliguori | switch (read_u8(data, 1)) { |
1676 | 9ca313aa | aliguori | case 0: |
1677 | 9ca313aa | aliguori | if (len == 2) |
1678 | 9ca313aa | aliguori | return 12; |
1679 | 9ca313aa | aliguori | |
1680 | 9ca313aa | aliguori | ext_key_event(vs, read_u16(data, 2),
|
1681 | 9ca313aa | aliguori | read_u32(data, 4), read_u32(data, 8)); |
1682 | 9ca313aa | aliguori | break;
|
1683 | 429a8ed3 | malc | case 1: |
1684 | 429a8ed3 | malc | if (len == 2) |
1685 | 429a8ed3 | malc | return 4; |
1686 | 429a8ed3 | malc | |
1687 | 429a8ed3 | malc | switch (read_u16 (data, 2)) { |
1688 | 429a8ed3 | malc | case 0: |
1689 | 429a8ed3 | malc | audio_add(vs); |
1690 | 429a8ed3 | malc | break;
|
1691 | 429a8ed3 | malc | case 1: |
1692 | 429a8ed3 | malc | audio_del(vs); |
1693 | 429a8ed3 | malc | break;
|
1694 | 429a8ed3 | malc | case 2: |
1695 | 429a8ed3 | malc | if (len == 4) |
1696 | 429a8ed3 | malc | return 10; |
1697 | 429a8ed3 | malc | switch (read_u8(data, 4)) { |
1698 | 429a8ed3 | malc | case 0: vs->as.fmt = AUD_FMT_U8; break; |
1699 | 429a8ed3 | malc | case 1: vs->as.fmt = AUD_FMT_S8; break; |
1700 | 429a8ed3 | malc | case 2: vs->as.fmt = AUD_FMT_U16; break; |
1701 | 429a8ed3 | malc | case 3: vs->as.fmt = AUD_FMT_S16; break; |
1702 | 429a8ed3 | malc | case 4: vs->as.fmt = AUD_FMT_U32; break; |
1703 | 429a8ed3 | malc | case 5: vs->as.fmt = AUD_FMT_S32; break; |
1704 | 429a8ed3 | malc | default:
|
1705 | 429a8ed3 | malc | printf("Invalid audio format %d\n", read_u8(data, 4)); |
1706 | 429a8ed3 | malc | vnc_client_error(vs); |
1707 | 429a8ed3 | malc | break;
|
1708 | 429a8ed3 | malc | } |
1709 | 429a8ed3 | malc | vs->as.nchannels = read_u8(data, 5);
|
1710 | 429a8ed3 | malc | if (vs->as.nchannels != 1 && vs->as.nchannels != 2) { |
1711 | 429a8ed3 | malc | printf("Invalid audio channel coount %d\n",
|
1712 | 429a8ed3 | malc | read_u8(data, 5));
|
1713 | 429a8ed3 | malc | vnc_client_error(vs); |
1714 | 429a8ed3 | malc | break;
|
1715 | 429a8ed3 | malc | } |
1716 | 429a8ed3 | malc | vs->as.freq = read_u32(data, 6);
|
1717 | 429a8ed3 | malc | break;
|
1718 | 429a8ed3 | malc | default:
|
1719 | 429a8ed3 | malc | printf ("Invalid audio message %d\n", read_u8(data, 4)); |
1720 | 429a8ed3 | malc | vnc_client_error(vs); |
1721 | 429a8ed3 | malc | break;
|
1722 | 429a8ed3 | malc | } |
1723 | 429a8ed3 | malc | break;
|
1724 | 429a8ed3 | malc | |
1725 | 9ca313aa | aliguori | default:
|
1726 | 9ca313aa | aliguori | printf("Msg: %d\n", read_u16(data, 0)); |
1727 | 9ca313aa | aliguori | vnc_client_error(vs); |
1728 | 9ca313aa | aliguori | break;
|
1729 | 9ca313aa | aliguori | } |
1730 | 9ca313aa | aliguori | break;
|
1731 | 24236869 | bellard | default:
|
1732 | 28a76be8 | aliguori | printf("Msg: %d\n", data[0]); |
1733 | 28a76be8 | aliguori | vnc_client_error(vs); |
1734 | 28a76be8 | aliguori | break;
|
1735 | 24236869 | bellard | } |
1736 | 5fafdf24 | ths | |
1737 | 24236869 | bellard | vnc_read_when(vs, protocol_client_msg, 1);
|
1738 | 24236869 | bellard | return 0; |
1739 | 24236869 | bellard | } |
1740 | 24236869 | bellard | |
1741 | 60fe76f3 | ths | static int protocol_client_init(VncState *vs, uint8_t *data, size_t len) |
1742 | 24236869 | bellard | { |
1743 | c35734b2 | ths | char buf[1024]; |
1744 | c35734b2 | ths | int size;
|
1745 | 24236869 | bellard | |
1746 | 0e1f5a0c | aliguori | vnc_write_u16(vs, ds_get_width(vs->ds)); |
1747 | 0e1f5a0c | aliguori | vnc_write_u16(vs, ds_get_height(vs->ds)); |
1748 | 24236869 | bellard | |
1749 | ca4cca4d | aliguori | pixel_format_message(vs); |
1750 | 24236869 | bellard | |
1751 | c35734b2 | ths | if (qemu_name)
|
1752 | c35734b2 | ths | size = snprintf(buf, sizeof(buf), "QEMU (%s)", qemu_name); |
1753 | c35734b2 | ths | else
|
1754 | c35734b2 | ths | size = snprintf(buf, sizeof(buf), "QEMU"); |
1755 | c35734b2 | ths | |
1756 | c35734b2 | ths | vnc_write_u32(vs, size); |
1757 | c35734b2 | ths | vnc_write(vs, buf, size); |
1758 | 24236869 | bellard | vnc_flush(vs); |
1759 | 24236869 | bellard | |
1760 | 24236869 | bellard | vnc_read_when(vs, protocol_client_msg, 1);
|
1761 | 24236869 | bellard | |
1762 | 24236869 | bellard | return 0; |
1763 | 24236869 | bellard | } |
1764 | 24236869 | bellard | |
1765 | 5fb6c7a8 | aliguori | void start_client_init(VncState *vs)
|
1766 | 5fb6c7a8 | aliguori | { |
1767 | 5fb6c7a8 | aliguori | vnc_read_when(vs, protocol_client_init, 1);
|
1768 | 5fb6c7a8 | aliguori | } |
1769 | 5fb6c7a8 | aliguori | |
1770 | 70848515 | ths | static void make_challenge(VncState *vs) |
1771 | 70848515 | ths | { |
1772 | 70848515 | ths | int i;
|
1773 | 70848515 | ths | |
1774 | 70848515 | ths | srand(time(NULL)+getpid()+getpid()*987654+rand()); |
1775 | 70848515 | ths | |
1776 | 70848515 | ths | for (i = 0 ; i < sizeof(vs->challenge) ; i++) |
1777 | 70848515 | ths | vs->challenge[i] = (int) (256.0*rand()/(RAND_MAX+1.0)); |
1778 | 70848515 | ths | } |
1779 | 70848515 | ths | |
1780 | 60fe76f3 | ths | static int protocol_client_auth_vnc(VncState *vs, uint8_t *data, size_t len) |
1781 | 70848515 | ths | { |
1782 | 60fe76f3 | ths | unsigned char response[VNC_AUTH_CHALLENGE_SIZE]; |
1783 | 70848515 | ths | int i, j, pwlen;
|
1784 | 60fe76f3 | ths | unsigned char key[8]; |
1785 | 70848515 | ths | |
1786 | 753b4053 | aliguori | if (!vs->vd->password || !vs->vd->password[0]) { |
1787 | 28a76be8 | aliguori | VNC_DEBUG("No password configured on server");
|
1788 | 28a76be8 | aliguori | vnc_write_u32(vs, 1); /* Reject auth */ |
1789 | 28a76be8 | aliguori | if (vs->minor >= 8) { |
1790 | 28a76be8 | aliguori | static const char err[] = "Authentication failed"; |
1791 | 28a76be8 | aliguori | vnc_write_u32(vs, sizeof(err));
|
1792 | 28a76be8 | aliguori | vnc_write(vs, err, sizeof(err));
|
1793 | 28a76be8 | aliguori | } |
1794 | 28a76be8 | aliguori | vnc_flush(vs); |
1795 | 28a76be8 | aliguori | vnc_client_error(vs); |
1796 | 28a76be8 | aliguori | return 0; |
1797 | 70848515 | ths | } |
1798 | 70848515 | ths | |
1799 | 70848515 | ths | memcpy(response, vs->challenge, VNC_AUTH_CHALLENGE_SIZE); |
1800 | 70848515 | ths | |
1801 | 70848515 | ths | /* Calculate the expected challenge response */
|
1802 | 753b4053 | aliguori | pwlen = strlen(vs->vd->password); |
1803 | 70848515 | ths | for (i=0; i<sizeof(key); i++) |
1804 | 753b4053 | aliguori | key[i] = i<pwlen ? vs->vd->password[i] : 0;
|
1805 | 70848515 | ths | deskey(key, EN0); |
1806 | 70848515 | ths | for (j = 0; j < VNC_AUTH_CHALLENGE_SIZE; j += 8) |
1807 | 70848515 | ths | des(response+j, response+j); |
1808 | 70848515 | ths | |
1809 | 70848515 | ths | /* Compare expected vs actual challenge response */
|
1810 | 70848515 | ths | if (memcmp(response, data, VNC_AUTH_CHALLENGE_SIZE) != 0) { |
1811 | 28a76be8 | aliguori | VNC_DEBUG("Client challenge reponse did not match\n");
|
1812 | 28a76be8 | aliguori | vnc_write_u32(vs, 1); /* Reject auth */ |
1813 | 28a76be8 | aliguori | if (vs->minor >= 8) { |
1814 | 28a76be8 | aliguori | static const char err[] = "Authentication failed"; |
1815 | 28a76be8 | aliguori | vnc_write_u32(vs, sizeof(err));
|
1816 | 28a76be8 | aliguori | vnc_write(vs, err, sizeof(err));
|
1817 | 28a76be8 | aliguori | } |
1818 | 28a76be8 | aliguori | vnc_flush(vs); |
1819 | 28a76be8 | aliguori | vnc_client_error(vs); |
1820 | 70848515 | ths | } else {
|
1821 | 28a76be8 | aliguori | VNC_DEBUG("Accepting VNC challenge response\n");
|
1822 | 28a76be8 | aliguori | vnc_write_u32(vs, 0); /* Accept auth */ |
1823 | 28a76be8 | aliguori | vnc_flush(vs); |
1824 | 70848515 | ths | |
1825 | 5fb6c7a8 | aliguori | start_client_init(vs); |
1826 | 70848515 | ths | } |
1827 | 70848515 | ths | return 0; |
1828 | 70848515 | ths | } |
1829 | 70848515 | ths | |
1830 | 5fb6c7a8 | aliguori | void start_auth_vnc(VncState *vs)
|
1831 | 70848515 | ths | { |
1832 | 70848515 | ths | make_challenge(vs); |
1833 | 70848515 | ths | /* Send client a 'random' challenge */
|
1834 | 70848515 | ths | vnc_write(vs, vs->challenge, sizeof(vs->challenge));
|
1835 | 70848515 | ths | vnc_flush(vs); |
1836 | 70848515 | ths | |
1837 | 70848515 | ths | vnc_read_when(vs, protocol_client_auth_vnc, sizeof(vs->challenge));
|
1838 | 469b15c6 | ths | } |
1839 | 469b15c6 | ths | |
1840 | 469b15c6 | ths | |
1841 | 60fe76f3 | ths | static int protocol_client_auth(VncState *vs, uint8_t *data, size_t len) |
1842 | 70848515 | ths | { |
1843 | 70848515 | ths | /* We only advertise 1 auth scheme at a time, so client
|
1844 | 70848515 | ths | * must pick the one we sent. Verify this */
|
1845 | 753b4053 | aliguori | if (data[0] != vs->vd->auth) { /* Reject auth */ |
1846 | 1263b7d6 | aliguori | VNC_DEBUG("Reject auth %d because it didn't match advertized\n", (int)data[0]); |
1847 | 70848515 | ths | vnc_write_u32(vs, 1);
|
1848 | 70848515 | ths | if (vs->minor >= 8) { |
1849 | 70848515 | ths | static const char err[] = "Authentication failed"; |
1850 | 70848515 | ths | vnc_write_u32(vs, sizeof(err));
|
1851 | 70848515 | ths | vnc_write(vs, err, sizeof(err));
|
1852 | 70848515 | ths | } |
1853 | 70848515 | ths | vnc_client_error(vs); |
1854 | 70848515 | ths | } else { /* Accept requested auth */ |
1855 | 70848515 | ths | VNC_DEBUG("Client requested auth %d\n", (int)data[0]); |
1856 | 753b4053 | aliguori | switch (vs->vd->auth) {
|
1857 | 70848515 | ths | case VNC_AUTH_NONE:
|
1858 | 70848515 | ths | VNC_DEBUG("Accept auth none\n");
|
1859 | a26c97ad | balrog | if (vs->minor >= 8) { |
1860 | a26c97ad | balrog | vnc_write_u32(vs, 0); /* Accept auth completion */ |
1861 | a26c97ad | balrog | vnc_flush(vs); |
1862 | a26c97ad | balrog | } |
1863 | 5fb6c7a8 | aliguori | start_client_init(vs); |
1864 | 70848515 | ths | break;
|
1865 | 70848515 | ths | |
1866 | 70848515 | ths | case VNC_AUTH_VNC:
|
1867 | 70848515 | ths | VNC_DEBUG("Start VNC auth\n");
|
1868 | 5fb6c7a8 | aliguori | start_auth_vnc(vs); |
1869 | 5fb6c7a8 | aliguori | break;
|
1870 | 70848515 | ths | |
1871 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
1872 | 8d5d2d4c | ths | case VNC_AUTH_VENCRYPT:
|
1873 | 8d5d2d4c | ths | VNC_DEBUG("Accept VeNCrypt auth\n");;
|
1874 | 5fb6c7a8 | aliguori | start_auth_vencrypt(vs); |
1875 | 5fb6c7a8 | aliguori | break;
|
1876 | 8d5d2d4c | ths | #endif /* CONFIG_VNC_TLS */ |
1877 | 8d5d2d4c | ths | |
1878 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_SASL
|
1879 | 2f9606b3 | aliguori | case VNC_AUTH_SASL:
|
1880 | 2f9606b3 | aliguori | VNC_DEBUG("Accept SASL auth\n");
|
1881 | 2f9606b3 | aliguori | start_auth_sasl(vs); |
1882 | 2f9606b3 | aliguori | break;
|
1883 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_SASL */ |
1884 | 2f9606b3 | aliguori | |
1885 | 70848515 | ths | default: /* Should not be possible, but just in case */ |
1886 | 1263b7d6 | aliguori | VNC_DEBUG("Reject auth %d server code bug\n", vs->vd->auth);
|
1887 | 70848515 | ths | vnc_write_u8(vs, 1);
|
1888 | 70848515 | ths | if (vs->minor >= 8) { |
1889 | 70848515 | ths | static const char err[] = "Authentication failed"; |
1890 | 70848515 | ths | vnc_write_u32(vs, sizeof(err));
|
1891 | 70848515 | ths | vnc_write(vs, err, sizeof(err));
|
1892 | 70848515 | ths | } |
1893 | 70848515 | ths | vnc_client_error(vs); |
1894 | 70848515 | ths | } |
1895 | 70848515 | ths | } |
1896 | 70848515 | ths | return 0; |
1897 | 70848515 | ths | } |
1898 | 70848515 | ths | |
1899 | 60fe76f3 | ths | static int protocol_version(VncState *vs, uint8_t *version, size_t len) |
1900 | 24236869 | bellard | { |
1901 | 24236869 | bellard | char local[13]; |
1902 | 24236869 | bellard | |
1903 | 24236869 | bellard | memcpy(local, version, 12);
|
1904 | 24236869 | bellard | local[12] = 0; |
1905 | 24236869 | bellard | |
1906 | 70848515 | ths | if (sscanf(local, "RFB %03d.%03d\n", &vs->major, &vs->minor) != 2) { |
1907 | 28a76be8 | aliguori | VNC_DEBUG("Malformed protocol version %s\n", local);
|
1908 | 28a76be8 | aliguori | vnc_client_error(vs); |
1909 | 28a76be8 | aliguori | return 0; |
1910 | 24236869 | bellard | } |
1911 | 70848515 | ths | VNC_DEBUG("Client request protocol version %d.%d\n", vs->major, vs->minor);
|
1912 | 70848515 | ths | if (vs->major != 3 || |
1913 | 28a76be8 | aliguori | (vs->minor != 3 &&
|
1914 | 28a76be8 | aliguori | vs->minor != 4 &&
|
1915 | 28a76be8 | aliguori | vs->minor != 5 &&
|
1916 | 28a76be8 | aliguori | vs->minor != 7 &&
|
1917 | 28a76be8 | aliguori | vs->minor != 8)) {
|
1918 | 28a76be8 | aliguori | VNC_DEBUG("Unsupported client version\n");
|
1919 | 28a76be8 | aliguori | vnc_write_u32(vs, VNC_AUTH_INVALID); |
1920 | 28a76be8 | aliguori | vnc_flush(vs); |
1921 | 28a76be8 | aliguori | vnc_client_error(vs); |
1922 | 28a76be8 | aliguori | return 0; |
1923 | 70848515 | ths | } |
1924 | b0566f4f | ths | /* Some broken clients report v3.4 or v3.5, which spec requires to be treated
|
1925 | 70848515 | ths | * as equivalent to v3.3 by servers
|
1926 | 70848515 | ths | */
|
1927 | b0566f4f | ths | if (vs->minor == 4 || vs->minor == 5) |
1928 | 28a76be8 | aliguori | vs->minor = 3;
|
1929 | 70848515 | ths | |
1930 | 70848515 | ths | if (vs->minor == 3) { |
1931 | 28a76be8 | aliguori | if (vs->vd->auth == VNC_AUTH_NONE) {
|
1932 | 70848515 | ths | VNC_DEBUG("Tell client auth none\n");
|
1933 | 753b4053 | aliguori | vnc_write_u32(vs, vs->vd->auth); |
1934 | 70848515 | ths | vnc_flush(vs); |
1935 | 28a76be8 | aliguori | start_client_init(vs); |
1936 | 753b4053 | aliguori | } else if (vs->vd->auth == VNC_AUTH_VNC) { |
1937 | 70848515 | ths | VNC_DEBUG("Tell client VNC auth\n");
|
1938 | 753b4053 | aliguori | vnc_write_u32(vs, vs->vd->auth); |
1939 | 70848515 | ths | vnc_flush(vs); |
1940 | 70848515 | ths | start_auth_vnc(vs); |
1941 | 70848515 | ths | } else {
|
1942 | 753b4053 | aliguori | VNC_DEBUG("Unsupported auth %d for protocol 3.3\n", vs->vd->auth);
|
1943 | 70848515 | ths | vnc_write_u32(vs, VNC_AUTH_INVALID); |
1944 | 70848515 | ths | vnc_flush(vs); |
1945 | 70848515 | ths | vnc_client_error(vs); |
1946 | 70848515 | ths | } |
1947 | 70848515 | ths | } else {
|
1948 | 28a76be8 | aliguori | VNC_DEBUG("Telling client we support auth %d\n", vs->vd->auth);
|
1949 | 28a76be8 | aliguori | vnc_write_u8(vs, 1); /* num auth */ |
1950 | 28a76be8 | aliguori | vnc_write_u8(vs, vs->vd->auth); |
1951 | 28a76be8 | aliguori | vnc_read_when(vs, protocol_client_auth, 1);
|
1952 | 28a76be8 | aliguori | vnc_flush(vs); |
1953 | 70848515 | ths | } |
1954 | 24236869 | bellard | |
1955 | 24236869 | bellard | return 0; |
1956 | 24236869 | bellard | } |
1957 | 24236869 | bellard | |
1958 | 753b4053 | aliguori | static void vnc_connect(VncDisplay *vd, int csock) |
1959 | 3aa3eea3 | balrog | { |
1960 | 753b4053 | aliguori | VncState *vs = qemu_mallocz(sizeof(VncState));
|
1961 | 753b4053 | aliguori | vs->csock = csock; |
1962 | 753b4053 | aliguori | |
1963 | 753b4053 | aliguori | VNC_DEBUG("New client on socket %d\n", csock);
|
1964 | 7d957bd8 | aliguori | dcl->idle = 0;
|
1965 | 3aa3eea3 | balrog | socket_set_nonblock(vs->csock); |
1966 | 3aa3eea3 | balrog | qemu_set_fd_handler2(vs->csock, NULL, vnc_client_read, NULL, vs); |
1967 | 753b4053 | aliguori | |
1968 | 753b4053 | aliguori | vs->vd = vd; |
1969 | 753b4053 | aliguori | vs->ds = vd->ds; |
1970 | 753b4053 | aliguori | vs->timer = qemu_new_timer(rt_clock, vnc_update_client, vs); |
1971 | 753b4053 | aliguori | vs->last_x = -1;
|
1972 | 753b4053 | aliguori | vs->last_y = -1;
|
1973 | 753b4053 | aliguori | |
1974 | 753b4053 | aliguori | vs->as.freq = 44100;
|
1975 | 753b4053 | aliguori | vs->as.nchannels = 2;
|
1976 | 753b4053 | aliguori | vs->as.fmt = AUD_FMT_S16; |
1977 | 753b4053 | aliguori | vs->as.endianness = 0;
|
1978 | 753b4053 | aliguori | |
1979 | 753b4053 | aliguori | vnc_resize(vs); |
1980 | 3aa3eea3 | balrog | vnc_write(vs, "RFB 003.008\n", 12); |
1981 | 3aa3eea3 | balrog | vnc_flush(vs); |
1982 | 3aa3eea3 | balrog | vnc_read_when(vs, protocol_version, 12);
|
1983 | 0e1f5a0c | aliguori | memset(vs->old_data, 0, ds_get_linesize(vs->ds) * ds_get_height(vs->ds));
|
1984 | 3aa3eea3 | balrog | memset(vs->dirty_row, 0xFF, sizeof(vs->dirty_row)); |
1985 | 3aa3eea3 | balrog | vnc_update_client(vs); |
1986 | 53762ddb | malc | reset_keys(vs); |
1987 | 753b4053 | aliguori | |
1988 | 753b4053 | aliguori | vs->next = vd->clients; |
1989 | 753b4053 | aliguori | vd->clients = vs; |
1990 | 3aa3eea3 | balrog | } |
1991 | 3aa3eea3 | balrog | |
1992 | 24236869 | bellard | static void vnc_listen_read(void *opaque) |
1993 | 24236869 | bellard | { |
1994 | 753b4053 | aliguori | VncDisplay *vs = opaque; |
1995 | 24236869 | bellard | struct sockaddr_in addr;
|
1996 | 24236869 | bellard | socklen_t addrlen = sizeof(addr);
|
1997 | 24236869 | bellard | |
1998 | 9f60ad50 | balrog | /* Catch-up */
|
1999 | 9f60ad50 | balrog | vga_hw_update(); |
2000 | 9f60ad50 | balrog | |
2001 | 753b4053 | aliguori | int csock = accept(vs->lsock, (struct sockaddr *)&addr, &addrlen); |
2002 | 753b4053 | aliguori | if (csock != -1) { |
2003 | 753b4053 | aliguori | vnc_connect(vs, csock); |
2004 | 24236869 | bellard | } |
2005 | 24236869 | bellard | } |
2006 | 24236869 | bellard | |
2007 | 71cab5ca | ths | void vnc_display_init(DisplayState *ds)
|
2008 | 24236869 | bellard | { |
2009 | 753b4053 | aliguori | VncDisplay *vs; |
2010 | 24236869 | bellard | |
2011 | 24236869 | bellard | vs = qemu_mallocz(sizeof(VncState));
|
2012 | 7d957bd8 | aliguori | dcl = qemu_mallocz(sizeof(DisplayChangeListener));
|
2013 | 24236869 | bellard | |
2014 | 24236869 | bellard | ds->opaque = vs; |
2015 | 7d957bd8 | aliguori | dcl->idle = 1;
|
2016 | 753b4053 | aliguori | vnc_display = vs; |
2017 | 24236869 | bellard | |
2018 | 24236869 | bellard | vs->lsock = -1;
|
2019 | 24236869 | bellard | |
2020 | 24236869 | bellard | vs->ds = ds; |
2021 | 24236869 | bellard | |
2022 | 9ca313aa | aliguori | if (keyboard_layout)
|
2023 | 0483755a | aliguori | vs->kbd_layout = init_keyboard_layout(name2keysym, keyboard_layout); |
2024 | 9ca313aa | aliguori | else
|
2025 | 0483755a | aliguori | vs->kbd_layout = init_keyboard_layout(name2keysym, "en-us");
|
2026 | 24236869 | bellard | |
2027 | 24236869 | bellard | if (!vs->kbd_layout)
|
2028 | 28a76be8 | aliguori | exit(1);
|
2029 | 24236869 | bellard | |
2030 | 753b4053 | aliguori | dcl->dpy_copy = vnc_dpy_copy; |
2031 | 7d957bd8 | aliguori | dcl->dpy_update = vnc_dpy_update; |
2032 | 7d957bd8 | aliguori | dcl->dpy_resize = vnc_dpy_resize; |
2033 | 7d957bd8 | aliguori | dcl->dpy_setdata = vnc_dpy_setdata; |
2034 | 7d957bd8 | aliguori | register_displaychangelistener(ds, dcl); |
2035 | 71cab5ca | ths | } |
2036 | 71cab5ca | ths | |
2037 | 6f43024c | ths | |
2038 | 71cab5ca | ths | void vnc_display_close(DisplayState *ds)
|
2039 | 71cab5ca | ths | { |
2040 | 753b4053 | aliguori | VncDisplay *vs = ds ? (VncDisplay *)ds->opaque : vnc_display; |
2041 | 71cab5ca | ths | |
2042 | 452b4d88 | aliguori | if (!vs)
|
2043 | 452b4d88 | aliguori | return;
|
2044 | 71cab5ca | ths | if (vs->display) {
|
2045 | 28a76be8 | aliguori | qemu_free(vs->display); |
2046 | 28a76be8 | aliguori | vs->display = NULL;
|
2047 | 71cab5ca | ths | } |
2048 | 71cab5ca | ths | if (vs->lsock != -1) { |
2049 | 28a76be8 | aliguori | qemu_set_fd_handler2(vs->lsock, NULL, NULL, NULL, NULL); |
2050 | 28a76be8 | aliguori | close(vs->lsock); |
2051 | 28a76be8 | aliguori | vs->lsock = -1;
|
2052 | 71cab5ca | ths | } |
2053 | 70848515 | ths | vs->auth = VNC_AUTH_INVALID; |
2054 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
2055 | 8d5d2d4c | ths | vs->subauth = VNC_AUTH_INVALID; |
2056 | 5fb6c7a8 | aliguori | vs->tls.x509verify = 0;
|
2057 | 8d5d2d4c | ths | #endif
|
2058 | 70848515 | ths | } |
2059 | 70848515 | ths | |
2060 | 70848515 | ths | int vnc_display_password(DisplayState *ds, const char *password) |
2061 | 70848515 | ths | { |
2062 | 753b4053 | aliguori | VncDisplay *vs = ds ? (VncDisplay *)ds->opaque : vnc_display; |
2063 | 70848515 | ths | |
2064 | 70848515 | ths | if (vs->password) {
|
2065 | 28a76be8 | aliguori | qemu_free(vs->password); |
2066 | 28a76be8 | aliguori | vs->password = NULL;
|
2067 | 70848515 | ths | } |
2068 | 70848515 | ths | if (password && password[0]) { |
2069 | 28a76be8 | aliguori | if (!(vs->password = qemu_strdup(password)))
|
2070 | 28a76be8 | aliguori | return -1; |
2071 | 70848515 | ths | } |
2072 | 70848515 | ths | |
2073 | 70848515 | ths | return 0; |
2074 | 71cab5ca | ths | } |
2075 | 71cab5ca | ths | |
2076 | 70848515 | ths | int vnc_display_open(DisplayState *ds, const char *display) |
2077 | 71cab5ca | ths | { |
2078 | 753b4053 | aliguori | VncDisplay *vs = ds ? (VncDisplay *)ds->opaque : vnc_display; |
2079 | 70848515 | ths | const char *options; |
2080 | 70848515 | ths | int password = 0; |
2081 | 3aa3eea3 | balrog | int reverse = 0; |
2082 | 9712ecaf | aliguori | int to_port = 0; |
2083 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
2084 | 3a702699 | ths | int tls = 0, x509 = 0; |
2085 | 8d5d2d4c | ths | #endif
|
2086 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_SASL
|
2087 | 2f9606b3 | aliguori | int sasl = 0; |
2088 | 2f9606b3 | aliguori | int saslErr;
|
2089 | 2f9606b3 | aliguori | #endif
|
2090 | 76655d6d | aliguori | int acl = 0; |
2091 | 71cab5ca | ths | |
2092 | 753b4053 | aliguori | if (!vnc_display)
|
2093 | 452b4d88 | aliguori | return -1; |
2094 | 71cab5ca | ths | vnc_display_close(ds); |
2095 | 70848515 | ths | if (strcmp(display, "none") == 0) |
2096 | 28a76be8 | aliguori | return 0; |
2097 | 24236869 | bellard | |
2098 | 70848515 | ths | if (!(vs->display = strdup(display)))
|
2099 | 28a76be8 | aliguori | return -1; |
2100 | 70848515 | ths | |
2101 | 70848515 | ths | options = display; |
2102 | 70848515 | ths | while ((options = strchr(options, ','))) { |
2103 | 28a76be8 | aliguori | options++; |
2104 | 28a76be8 | aliguori | if (strncmp(options, "password", 8) == 0) { |
2105 | 28a76be8 | aliguori | password = 1; /* Require password auth */ |
2106 | 28a76be8 | aliguori | } else if (strncmp(options, "reverse", 7) == 0) { |
2107 | 28a76be8 | aliguori | reverse = 1;
|
2108 | 28a76be8 | aliguori | } else if (strncmp(options, "to=", 3) == 0) { |
2109 | 9712ecaf | aliguori | to_port = atoi(options+3) + 5900; |
2110 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_SASL
|
2111 | 28a76be8 | aliguori | } else if (strncmp(options, "sasl", 4) == 0) { |
2112 | 28a76be8 | aliguori | sasl = 1; /* Require SASL auth */ |
2113 | 2f9606b3 | aliguori | #endif
|
2114 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
2115 | 28a76be8 | aliguori | } else if (strncmp(options, "tls", 3) == 0) { |
2116 | 28a76be8 | aliguori | tls = 1; /* Require TLS */ |
2117 | 28a76be8 | aliguori | } else if (strncmp(options, "x509", 4) == 0) { |
2118 | 28a76be8 | aliguori | char *start, *end;
|
2119 | 28a76be8 | aliguori | x509 = 1; /* Require x509 certificates */ |
2120 | 28a76be8 | aliguori | if (strncmp(options, "x509verify", 10) == 0) |
2121 | 28a76be8 | aliguori | vs->tls.x509verify = 1; /* ...and verify client certs */ |
2122 | 28a76be8 | aliguori | |
2123 | 28a76be8 | aliguori | /* Now check for 'x509=/some/path' postfix
|
2124 | 28a76be8 | aliguori | * and use that to setup x509 certificate/key paths */
|
2125 | 28a76be8 | aliguori | start = strchr(options, '=');
|
2126 | 28a76be8 | aliguori | end = strchr(options, ',');
|
2127 | 28a76be8 | aliguori | if (start && (!end || (start < end))) {
|
2128 | 28a76be8 | aliguori | int len = end ? end-(start+1) : strlen(start+1); |
2129 | 28a76be8 | aliguori | char *path = qemu_strndup(start + 1, len); |
2130 | 28a76be8 | aliguori | |
2131 | 28a76be8 | aliguori | VNC_DEBUG("Trying certificate path '%s'\n", path);
|
2132 | 28a76be8 | aliguori | if (vnc_tls_set_x509_creds_dir(vs, path) < 0) { |
2133 | 28a76be8 | aliguori | fprintf(stderr, "Failed to find x509 certificates/keys in %s\n", path);
|
2134 | 28a76be8 | aliguori | qemu_free(path); |
2135 | 28a76be8 | aliguori | qemu_free(vs->display); |
2136 | 28a76be8 | aliguori | vs->display = NULL;
|
2137 | 28a76be8 | aliguori | return -1; |
2138 | 28a76be8 | aliguori | } |
2139 | 28a76be8 | aliguori | qemu_free(path); |
2140 | 28a76be8 | aliguori | } else {
|
2141 | 28a76be8 | aliguori | fprintf(stderr, "No certificate path provided\n");
|
2142 | 28a76be8 | aliguori | qemu_free(vs->display); |
2143 | 28a76be8 | aliguori | vs->display = NULL;
|
2144 | 28a76be8 | aliguori | return -1; |
2145 | 28a76be8 | aliguori | } |
2146 | 8d5d2d4c | ths | #endif
|
2147 | 28a76be8 | aliguori | } else if (strncmp(options, "acl", 3) == 0) { |
2148 | 28a76be8 | aliguori | acl = 1;
|
2149 | 28a76be8 | aliguori | } |
2150 | 70848515 | ths | } |
2151 | 70848515 | ths | |
2152 | 76655d6d | aliguori | #ifdef CONFIG_VNC_TLS
|
2153 | 76655d6d | aliguori | if (acl && x509 && vs->tls.x509verify) {
|
2154 | 28a76be8 | aliguori | if (!(vs->tls.acl = qemu_acl_init("vnc.x509dname"))) { |
2155 | 28a76be8 | aliguori | fprintf(stderr, "Failed to create x509 dname ACL\n");
|
2156 | 28a76be8 | aliguori | exit(1);
|
2157 | 28a76be8 | aliguori | } |
2158 | 76655d6d | aliguori | } |
2159 | 76655d6d | aliguori | #endif
|
2160 | 76655d6d | aliguori | #ifdef CONFIG_VNC_SASL
|
2161 | 76655d6d | aliguori | if (acl && sasl) {
|
2162 | 28a76be8 | aliguori | if (!(vs->sasl.acl = qemu_acl_init("vnc.username"))) { |
2163 | 28a76be8 | aliguori | fprintf(stderr, "Failed to create username ACL\n");
|
2164 | 28a76be8 | aliguori | exit(1);
|
2165 | 28a76be8 | aliguori | } |
2166 | 76655d6d | aliguori | } |
2167 | 76655d6d | aliguori | #endif
|
2168 | 76655d6d | aliguori | |
2169 | 2f9606b3 | aliguori | /*
|
2170 | 2f9606b3 | aliguori | * Combinations we support here:
|
2171 | 2f9606b3 | aliguori | *
|
2172 | 2f9606b3 | aliguori | * - no-auth (clear text, no auth)
|
2173 | 2f9606b3 | aliguori | * - password (clear text, weak auth)
|
2174 | 2f9606b3 | aliguori | * - sasl (encrypt, good auth *IF* using Kerberos via GSSAPI)
|
2175 | 2f9606b3 | aliguori | * - tls (encrypt, weak anonymous creds, no auth)
|
2176 | 2f9606b3 | aliguori | * - tls + password (encrypt, weak anonymous creds, weak auth)
|
2177 | 2f9606b3 | aliguori | * - tls + sasl (encrypt, weak anonymous creds, good auth)
|
2178 | 2f9606b3 | aliguori | * - tls + x509 (encrypt, good x509 creds, no auth)
|
2179 | 2f9606b3 | aliguori | * - tls + x509 + password (encrypt, good x509 creds, weak auth)
|
2180 | 2f9606b3 | aliguori | * - tls + x509 + sasl (encrypt, good x509 creds, good auth)
|
2181 | 2f9606b3 | aliguori | *
|
2182 | 2f9606b3 | aliguori | * NB1. TLS is a stackable auth scheme.
|
2183 | 2f9606b3 | aliguori | * NB2. the x509 schemes have option to validate a client cert dname
|
2184 | 2f9606b3 | aliguori | */
|
2185 | 70848515 | ths | if (password) {
|
2186 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
2187 | 28a76be8 | aliguori | if (tls) {
|
2188 | 28a76be8 | aliguori | vs->auth = VNC_AUTH_VENCRYPT; |
2189 | 28a76be8 | aliguori | if (x509) {
|
2190 | 28a76be8 | aliguori | VNC_DEBUG("Initializing VNC server with x509 password auth\n");
|
2191 | 28a76be8 | aliguori | vs->subauth = VNC_AUTH_VENCRYPT_X509VNC; |
2192 | 28a76be8 | aliguori | } else {
|
2193 | 28a76be8 | aliguori | VNC_DEBUG("Initializing VNC server with TLS password auth\n");
|
2194 | 28a76be8 | aliguori | vs->subauth = VNC_AUTH_VENCRYPT_TLSVNC; |
2195 | 28a76be8 | aliguori | } |
2196 | 28a76be8 | aliguori | } else {
|
2197 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_TLS */ |
2198 | 28a76be8 | aliguori | VNC_DEBUG("Initializing VNC server with password auth\n");
|
2199 | 28a76be8 | aliguori | vs->auth = VNC_AUTH_VNC; |
2200 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
2201 | 28a76be8 | aliguori | vs->subauth = VNC_AUTH_INVALID; |
2202 | 28a76be8 | aliguori | } |
2203 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_TLS */ |
2204 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_SASL
|
2205 | 2f9606b3 | aliguori | } else if (sasl) { |
2206 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_TLS
|
2207 | 2f9606b3 | aliguori | if (tls) {
|
2208 | 2f9606b3 | aliguori | vs->auth = VNC_AUTH_VENCRYPT; |
2209 | 2f9606b3 | aliguori | if (x509) {
|
2210 | 28a76be8 | aliguori | VNC_DEBUG("Initializing VNC server with x509 SASL auth\n");
|
2211 | 2f9606b3 | aliguori | vs->subauth = VNC_AUTH_VENCRYPT_X509SASL; |
2212 | 2f9606b3 | aliguori | } else {
|
2213 | 28a76be8 | aliguori | VNC_DEBUG("Initializing VNC server with TLS SASL auth\n");
|
2214 | 2f9606b3 | aliguori | vs->subauth = VNC_AUTH_VENCRYPT_TLSSASL; |
2215 | 2f9606b3 | aliguori | } |
2216 | 2f9606b3 | aliguori | } else {
|
2217 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_TLS */ |
2218 | 28a76be8 | aliguori | VNC_DEBUG("Initializing VNC server with SASL auth\n");
|
2219 | 2f9606b3 | aliguori | vs->auth = VNC_AUTH_SASL; |
2220 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_TLS
|
2221 | 2f9606b3 | aliguori | vs->subauth = VNC_AUTH_INVALID; |
2222 | 2f9606b3 | aliguori | } |
2223 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_TLS */ |
2224 | 2f9606b3 | aliguori | #endif /* CONFIG_VNC_SASL */ |
2225 | 70848515 | ths | } else {
|
2226 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
2227 | 28a76be8 | aliguori | if (tls) {
|
2228 | 28a76be8 | aliguori | vs->auth = VNC_AUTH_VENCRYPT; |
2229 | 28a76be8 | aliguori | if (x509) {
|
2230 | 28a76be8 | aliguori | VNC_DEBUG("Initializing VNC server with x509 no auth\n");
|
2231 | 28a76be8 | aliguori | vs->subauth = VNC_AUTH_VENCRYPT_X509NONE; |
2232 | 28a76be8 | aliguori | } else {
|
2233 | 28a76be8 | aliguori | VNC_DEBUG("Initializing VNC server with TLS no auth\n");
|
2234 | 28a76be8 | aliguori | vs->subauth = VNC_AUTH_VENCRYPT_TLSNONE; |
2235 | 28a76be8 | aliguori | } |
2236 | 28a76be8 | aliguori | } else {
|
2237 | 8d5d2d4c | ths | #endif
|
2238 | 28a76be8 | aliguori | VNC_DEBUG("Initializing VNC server with no auth\n");
|
2239 | 28a76be8 | aliguori | vs->auth = VNC_AUTH_NONE; |
2240 | eb38c52c | blueswir1 | #ifdef CONFIG_VNC_TLS
|
2241 | 28a76be8 | aliguori | vs->subauth = VNC_AUTH_INVALID; |
2242 | 28a76be8 | aliguori | } |
2243 | 8d5d2d4c | ths | #endif
|
2244 | 70848515 | ths | } |
2245 | 24236869 | bellard | |
2246 | 2f9606b3 | aliguori | #ifdef CONFIG_VNC_SASL
|
2247 | 2f9606b3 | aliguori | if ((saslErr = sasl_server_init(NULL, "qemu")) != SASL_OK) { |
2248 | 2f9606b3 | aliguori | fprintf(stderr, "Failed to initialize SASL auth %s",
|
2249 | 2f9606b3 | aliguori | sasl_errstring(saslErr, NULL, NULL)); |
2250 | 2f9606b3 | aliguori | free(vs->display); |
2251 | 2f9606b3 | aliguori | vs->display = NULL;
|
2252 | 2f9606b3 | aliguori | return -1; |
2253 | 2f9606b3 | aliguori | } |
2254 | 2f9606b3 | aliguori | #endif
|
2255 | 2f9606b3 | aliguori | |
2256 | 3aa3eea3 | balrog | if (reverse) {
|
2257 | 9712ecaf | aliguori | /* connect to viewer */
|
2258 | 9712ecaf | aliguori | if (strncmp(display, "unix:", 5) == 0) |
2259 | 9712ecaf | aliguori | vs->lsock = unix_connect(display+5);
|
2260 | 9712ecaf | aliguori | else
|
2261 | 9712ecaf | aliguori | vs->lsock = inet_connect(display, SOCK_STREAM); |
2262 | 9712ecaf | aliguori | if (-1 == vs->lsock) { |
2263 | 3aa3eea3 | balrog | free(vs->display); |
2264 | 3aa3eea3 | balrog | vs->display = NULL;
|
2265 | 3aa3eea3 | balrog | return -1; |
2266 | 3aa3eea3 | balrog | } else {
|
2267 | 753b4053 | aliguori | int csock = vs->lsock;
|
2268 | 3aa3eea3 | balrog | vs->lsock = -1;
|
2269 | 753b4053 | aliguori | vnc_connect(vs, csock); |
2270 | 3aa3eea3 | balrog | } |
2271 | 9712ecaf | aliguori | return 0; |
2272 | 24236869 | bellard | |
2273 | 9712ecaf | aliguori | } else {
|
2274 | 9712ecaf | aliguori | /* listen for connects */
|
2275 | 9712ecaf | aliguori | char *dpy;
|
2276 | 9712ecaf | aliguori | dpy = qemu_malloc(256);
|
2277 | 9712ecaf | aliguori | if (strncmp(display, "unix:", 5) == 0) { |
2278 | bc575e95 | blueswir1 | pstrcpy(dpy, 256, "unix:"); |
2279 | 4a55bfdf | aliguori | vs->lsock = unix_listen(display+5, dpy+5, 256-5); |
2280 | 9712ecaf | aliguori | } else {
|
2281 | 9712ecaf | aliguori | vs->lsock = inet_listen(display, dpy, 256, SOCK_STREAM, 5900); |
2282 | 9712ecaf | aliguori | } |
2283 | 9712ecaf | aliguori | if (-1 == vs->lsock) { |
2284 | 9712ecaf | aliguori | free(dpy); |
2285 | d0513623 | balrog | return -1; |
2286 | 9712ecaf | aliguori | } else {
|
2287 | 9712ecaf | aliguori | free(vs->display); |
2288 | 9712ecaf | aliguori | vs->display = dpy; |
2289 | 9712ecaf | aliguori | } |
2290 | 24236869 | bellard | } |
2291 | 753b4053 | aliguori | return qemu_set_fd_handler2(vs->lsock, NULL, vnc_listen_read, NULL, vs); |
2292 | 24236869 | bellard | } |