Archipelago » qemu

/*

 * 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, *addr,

        *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, addr_flags;

    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");

    }

    addr = qemu_opt_get(opts, "addr");

    addr_flags = 0;

    if (qemu_opt_get_bool(opts, "ipv4", 0)) {

        addr_flags |= SPICE_ADDR_FLAG_IPV4_ONLY;

    } else if (qemu_opt_get_bool(opts, "ipv6", 0)) {

        addr_flags |= SPICE_ADDR_FLAG_IPV6_ONLY;

    }

    spice_server = spice_server_new();

    spice_server_set_addr(spice_server, addr ? addr : "", addr_flags);

    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);