 17b6dea08b
			
		
	
	
		17b6dea08b
		
	
	
	
	
		
			
			This allows to enforce tls or plaintext usage for certain spice channels. [ v2: code style fixup ]
		
			
				
	
	
		
			350 lines
		
	
	
		
			9.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			350 lines
		
	
	
		
			9.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * Copyright (C) 2010 Red Hat, Inc.
 | |
|  *
 | |
|  * This program is free software; you can redistribute it and/or
 | |
|  * modify it under the terms of the GNU General Public License as
 | |
|  * published by the Free Software Foundation; either version 2 or
 | |
|  * (at your option) version 3 of the License.
 | |
|  *
 | |
|  * This program is distributed in the hope that it will be useful,
 | |
|  * but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
|  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | |
|  * GNU General Public License for more details.
 | |
|  *
 | |
|  * You should have received a copy of the GNU General Public License
 | |
|  * along with this program; if not, see <http://www.gnu.org/licenses/>.
 | |
|  */
 | |
| 
 | |
| #include <spice.h>
 | |
| #include <spice-experimental.h>
 | |
| 
 | |
| #include "qemu-common.h"
 | |
| #include "qemu-spice.h"
 | |
| #include "qemu-timer.h"
 | |
| #include "qemu-queue.h"
 | |
| #include "qemu-x509.h"
 | |
| #include "monitor.h"
 | |
| 
 | |
| /* core bits */
 | |
| 
 | |
| static SpiceServer *spice_server;
 | |
| int using_spice = 0;
 | |
| 
 | |
| struct SpiceTimer {
 | |
|     QEMUTimer *timer;
 | |
|     QTAILQ_ENTRY(SpiceTimer) next;
 | |
| };
 | |
| static QTAILQ_HEAD(, SpiceTimer) timers = QTAILQ_HEAD_INITIALIZER(timers);
 | |
| 
 | |
| static SpiceTimer *timer_add(SpiceTimerFunc func, void *opaque)
 | |
| {
 | |
|     SpiceTimer *timer;
 | |
| 
 | |
|     timer = qemu_mallocz(sizeof(*timer));
 | |
|     timer->timer = qemu_new_timer(rt_clock, func, opaque);
 | |
|     QTAILQ_INSERT_TAIL(&timers, timer, next);
 | |
|     return timer;
 | |
| }
 | |
| 
 | |
| static void timer_start(SpiceTimer *timer, uint32_t ms)
 | |
| {
 | |
|     qemu_mod_timer(timer->timer, qemu_get_clock(rt_clock) + ms);
 | |
| }
 | |
| 
 | |
| static void timer_cancel(SpiceTimer *timer)
 | |
| {
 | |
|     qemu_del_timer(timer->timer);
 | |
| }
 | |
| 
 | |
| static void timer_remove(SpiceTimer *timer)
 | |
| {
 | |
|     qemu_del_timer(timer->timer);
 | |
|     qemu_free_timer(timer->timer);
 | |
|     QTAILQ_REMOVE(&timers, timer, next);
 | |
|     qemu_free(timer);
 | |
| }
 | |
| 
 | |
| struct SpiceWatch {
 | |
|     int fd;
 | |
|     int event_mask;
 | |
|     SpiceWatchFunc func;
 | |
|     void *opaque;
 | |
|     QTAILQ_ENTRY(SpiceWatch) next;
 | |
| };
 | |
| static QTAILQ_HEAD(, SpiceWatch) watches = QTAILQ_HEAD_INITIALIZER(watches);
 | |
| 
 | |
| static void watch_read(void *opaque)
 | |
| {
 | |
|     SpiceWatch *watch = opaque;
 | |
|     watch->func(watch->fd, SPICE_WATCH_EVENT_READ, watch->opaque);
 | |
| }
 | |
| 
 | |
| static void watch_write(void *opaque)
 | |
| {
 | |
|     SpiceWatch *watch = opaque;
 | |
|     watch->func(watch->fd, SPICE_WATCH_EVENT_WRITE, watch->opaque);
 | |
| }
 | |
| 
 | |
| static void watch_update_mask(SpiceWatch *watch, int event_mask)
 | |
| {
 | |
|     IOHandler *on_read = NULL;
 | |
|     IOHandler *on_write = NULL;
 | |
| 
 | |
|     watch->event_mask = event_mask;
 | |
|     if (watch->event_mask & SPICE_WATCH_EVENT_READ) {
 | |
|         on_read = watch_read;
 | |
|     }
 | |
|     if (watch->event_mask & SPICE_WATCH_EVENT_WRITE) {
 | |
|         on_read = watch_write;
 | |
|     }
 | |
|     qemu_set_fd_handler(watch->fd, on_read, on_write, watch);
 | |
| }
 | |
| 
 | |
| static SpiceWatch *watch_add(int fd, int event_mask, SpiceWatchFunc func, void *opaque)
 | |
| {
 | |
|     SpiceWatch *watch;
 | |
| 
 | |
|     watch = qemu_mallocz(sizeof(*watch));
 | |
|     watch->fd     = fd;
 | |
|     watch->func   = func;
 | |
|     watch->opaque = opaque;
 | |
|     QTAILQ_INSERT_TAIL(&watches, watch, next);
 | |
| 
 | |
|     watch_update_mask(watch, event_mask);
 | |
|     return watch;
 | |
| }
 | |
| 
 | |
| static void watch_remove(SpiceWatch *watch)
 | |
| {
 | |
|     watch_update_mask(watch, 0);
 | |
|     QTAILQ_REMOVE(&watches, watch, next);
 | |
|     qemu_free(watch);
 | |
| }
 | |
| 
 | |
| static SpiceCoreInterface core_interface = {
 | |
|     .base.type          = SPICE_INTERFACE_CORE,
 | |
|     .base.description   = "qemu core services",
 | |
|     .base.major_version = SPICE_INTERFACE_CORE_MAJOR,
 | |
|     .base.minor_version = SPICE_INTERFACE_CORE_MINOR,
 | |
| 
 | |
|     .timer_add          = timer_add,
 | |
|     .timer_start        = timer_start,
 | |
|     .timer_cancel       = timer_cancel,
 | |
|     .timer_remove       = timer_remove,
 | |
| 
 | |
|     .watch_add          = watch_add,
 | |
|     .watch_update_mask  = watch_update_mask,
 | |
|     .watch_remove       = watch_remove,
 | |
| };
 | |
| 
 | |
| /* config string parsing */
 | |
| 
 | |
| static int name2enum(const char *string, const char *table[], int entries)
 | |
| {
 | |
|     int i;
 | |
| 
 | |
|     if (string) {
 | |
|         for (i = 0; i < entries; i++) {
 | |
|             if (!table[i]) {
 | |
|                 continue;
 | |
|             }
 | |
|             if (strcmp(string, table[i]) != 0) {
 | |
|                 continue;
 | |
|             }
 | |
|             return i;
 | |
|         }
 | |
|     }
 | |
|     return -1;
 | |
| }
 | |
| 
 | |
| static int parse_name(const char *string, const char *optname,
 | |
|                       const char *table[], int entries)
 | |
| {
 | |
|     int value = name2enum(string, table, entries);
 | |
| 
 | |
|     if (value != -1) {
 | |
|         return value;
 | |
|     }
 | |
|     fprintf(stderr, "spice: invalid %s: %s\n", optname, string);
 | |
|     exit(1);
 | |
| }
 | |
| 
 | |
| static const char *compression_names[] = {
 | |
|     [ SPICE_IMAGE_COMPRESS_OFF ]      = "off",
 | |
|     [ SPICE_IMAGE_COMPRESS_AUTO_GLZ ] = "auto_glz",
 | |
|     [ SPICE_IMAGE_COMPRESS_AUTO_LZ ]  = "auto_lz",
 | |
|     [ SPICE_IMAGE_COMPRESS_QUIC ]     = "quic",
 | |
|     [ SPICE_IMAGE_COMPRESS_GLZ ]      = "glz",
 | |
|     [ SPICE_IMAGE_COMPRESS_LZ ]       = "lz",
 | |
| };
 | |
| #define parse_compression(_name)                                        \
 | |
|     parse_name(_name, "image compression",                              \
 | |
|                compression_names, ARRAY_SIZE(compression_names))
 | |
| 
 | |
| static const char *wan_compression_names[] = {
 | |
|     [ SPICE_WAN_COMPRESSION_AUTO   ] = "auto",
 | |
|     [ SPICE_WAN_COMPRESSION_NEVER  ] = "never",
 | |
|     [ SPICE_WAN_COMPRESSION_ALWAYS ] = "always",
 | |
| };
 | |
| #define parse_wan_compression(_name)                                    \
 | |
|     parse_name(_name, "wan compression",                                \
 | |
|                wan_compression_names, ARRAY_SIZE(wan_compression_names))
 | |
| 
 | |
| /* functions for the rest of qemu */
 | |
| 
 | |
| static int add_channel(const char *name, const char *value, void *opaque)
 | |
| {
 | |
|     int security = 0;
 | |
|     int rc;
 | |
| 
 | |
|     if (strcmp(name, "tls-channel") == 0) {
 | |
|         security = SPICE_CHANNEL_SECURITY_SSL;
 | |
|     }
 | |
|     if (strcmp(name, "plaintext-channel") == 0) {
 | |
|         security = SPICE_CHANNEL_SECURITY_NONE;
 | |
|     }
 | |
|     if (security == 0) {
 | |
|         return 0;
 | |
|     }
 | |
|     if (strcmp(value, "default") == 0) {
 | |
|         rc = spice_server_set_channel_security(spice_server, NULL, security);
 | |
|     } else {
 | |
|         rc = spice_server_set_channel_security(spice_server, value, security);
 | |
|     }
 | |
|     if (rc != 0) {
 | |
|         fprintf(stderr, "spice: failed to set channel security for %s\n", value);
 | |
|         exit(1);
 | |
|     }
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| void qemu_spice_init(void)
 | |
| {
 | |
|     QemuOpts *opts = QTAILQ_FIRST(&qemu_spice_opts.head);
 | |
|     const char *password, *str, *x509_dir,
 | |
|         *x509_key_password = NULL,
 | |
|         *x509_dh_file = NULL,
 | |
|         *tls_ciphers = NULL;
 | |
|     char *x509_key_file = NULL,
 | |
|         *x509_cert_file = NULL,
 | |
|         *x509_cacert_file = NULL;
 | |
|     int port, tls_port, len;
 | |
|     spice_image_compression_t compression;
 | |
|     spice_wan_compression_t wan_compr;
 | |
| 
 | |
|     if (!opts) {
 | |
|         return;
 | |
|     }
 | |
|     port = qemu_opt_get_number(opts, "port", 0);
 | |
|     tls_port = qemu_opt_get_number(opts, "tls-port", 0);
 | |
|     if (!port && !tls_port) {
 | |
|         return;
 | |
|     }
 | |
|     password = qemu_opt_get(opts, "password");
 | |
| 
 | |
|     if (tls_port) {
 | |
|         x509_dir = qemu_opt_get(opts, "x509-dir");
 | |
|         if (NULL == x509_dir) {
 | |
|             x509_dir = ".";
 | |
|         }
 | |
|         len = strlen(x509_dir) + 32;
 | |
| 
 | |
|         str = qemu_opt_get(opts, "x509-key-file");
 | |
|         if (str) {
 | |
|             x509_key_file = qemu_strdup(str);
 | |
|         } else {
 | |
|             x509_key_file = qemu_malloc(len);
 | |
|             snprintf(x509_key_file, len, "%s/%s", x509_dir, X509_SERVER_KEY_FILE);
 | |
|         }
 | |
| 
 | |
|         str = qemu_opt_get(opts, "x509-cert-file");
 | |
|         if (str) {
 | |
|             x509_cert_file = qemu_strdup(str);
 | |
|         } else {
 | |
|             x509_cert_file = qemu_malloc(len);
 | |
|             snprintf(x509_cert_file, len, "%s/%s", x509_dir, X509_SERVER_CERT_FILE);
 | |
|         }
 | |
| 
 | |
|         str = qemu_opt_get(opts, "x509-cacert-file");
 | |
|         if (str) {
 | |
|             x509_cacert_file = qemu_strdup(str);
 | |
|         } else {
 | |
|             x509_cacert_file = qemu_malloc(len);
 | |
|             snprintf(x509_cacert_file, len, "%s/%s", x509_dir, X509_CA_CERT_FILE);
 | |
|         }
 | |
| 
 | |
|         x509_key_password = qemu_opt_get(opts, "x509-key-password");
 | |
|         x509_dh_file = qemu_opt_get(opts, "x509-dh-file");
 | |
|         tls_ciphers = qemu_opt_get(opts, "tls-ciphers");
 | |
|     }
 | |
| 
 | |
|     spice_server = spice_server_new();
 | |
|     if (port) {
 | |
|         spice_server_set_port(spice_server, port);
 | |
|     }
 | |
|     if (tls_port) {
 | |
|         spice_server_set_tls(spice_server, tls_port,
 | |
|                              x509_cacert_file,
 | |
|                              x509_cert_file,
 | |
|                              x509_key_file,
 | |
|                              x509_key_password,
 | |
|                              x509_dh_file,
 | |
|                              tls_ciphers);
 | |
|     }
 | |
|     if (password) {
 | |
|         spice_server_set_ticket(spice_server, password, 0, 0, 0);
 | |
|     }
 | |
|     if (qemu_opt_get_bool(opts, "disable-ticketing", 0)) {
 | |
|         spice_server_set_noauth(spice_server);
 | |
|     }
 | |
| 
 | |
|     compression = SPICE_IMAGE_COMPRESS_AUTO_GLZ;
 | |
|     str = qemu_opt_get(opts, "image-compression");
 | |
|     if (str) {
 | |
|         compression = parse_compression(str);
 | |
|     }
 | |
|     spice_server_set_image_compression(spice_server, compression);
 | |
| 
 | |
|     wan_compr = SPICE_WAN_COMPRESSION_AUTO;
 | |
|     str = qemu_opt_get(opts, "jpeg-wan-compression");
 | |
|     if (str) {
 | |
|         wan_compr = parse_wan_compression(str);
 | |
|     }
 | |
|     spice_server_set_jpeg_compression(spice_server, wan_compr);
 | |
| 
 | |
|     wan_compr = SPICE_WAN_COMPRESSION_AUTO;
 | |
|     str = qemu_opt_get(opts, "zlib-glz-wan-compression");
 | |
|     if (str) {
 | |
|         wan_compr = parse_wan_compression(str);
 | |
|     }
 | |
|     spice_server_set_zlib_glz_compression(spice_server, wan_compr);
 | |
| 
 | |
|     qemu_opt_foreach(opts, add_channel, NULL, 0);
 | |
| 
 | |
|     spice_server_init(spice_server, &core_interface);
 | |
|     using_spice = 1;
 | |
| 
 | |
|     qemu_spice_input_init();
 | |
| 
 | |
|     qemu_free(x509_key_file);
 | |
|     qemu_free(x509_cert_file);
 | |
|     qemu_free(x509_cacert_file);
 | |
| }
 | |
| 
 | |
| int qemu_spice_add_interface(SpiceBaseInstance *sin)
 | |
| {
 | |
|     return spice_server_add_interface(spice_server, sin);
 | |
| }
 | |
| 
 | |
| static void spice_register_config(void)
 | |
| {
 | |
|     qemu_add_opts(&qemu_spice_opts);
 | |
| }
 | |
| machine_init(spice_register_config);
 | |
| 
 | |
| static void spice_initialize(void)
 | |
| {
 | |
|     qemu_spice_init();
 | |
| }
 | |
| device_init(spice_initialize);
 |