/* -*- Mode: C; c-basic-offset: 4; indent-tabs-mode: nil -*- */ /* Copyright (C) 2009 Red Hat, Inc. This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. This library 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 Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with this library; if not, see . */ #ifdef HAVE_CONFIG_H #include #endif #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #if HAVE_SASL #include #endif #include #include #include #include #include #include "common/generated_server_marshallers.h" #include "common/ring.h" #include "spice.h" #include "reds.h" #include "agent-msg-filter.h" #include "inputs-channel.h" #include "main-channel.h" #include "red-qxl.h" #include "main-dispatcher.h" #include "sound.h" #include "stat.h" #include "demarshallers.h" #include "char-device.h" #include "migration-protocol.h" #ifdef USE_SMARTCARD #include "smartcard.h" #endif #include "reds-stream.h" #include "utils.h" #include "reds-private.h" static SpiceCoreInterface *core_public = NULL; static SpiceTimer *adapter_timer_add(const SpiceCoreInterfaceInternal *iface, SpiceTimerFunc func, void *opaque) { return core_public->timer_add(func, opaque); } static void adapter_timer_start(SpiceTimer *timer, uint32_t ms) { core_public->timer_start(timer, ms); } static void adapter_timer_cancel(SpiceTimer *timer) { core_public->timer_cancel(timer); } static void adapter_timer_remove(SpiceTimer *timer) { core_public->timer_remove(timer); } static SpiceWatch *adapter_watch_add(const SpiceCoreInterfaceInternal *iface, int fd, int event_mask, SpiceWatchFunc func, void *opaque) { return core_public->watch_add(fd, event_mask, func, opaque); } static void adapter_watch_update_mask(SpiceWatch *watch, int event_mask) { core_public->watch_update_mask(watch, event_mask); } static void adapter_watch_remove(SpiceWatch *watch) { core_public->watch_remove(watch); } static void adapter_channel_event(int event, SpiceChannelEventInfo *info) { if (core_public->base.minor_version >= 3 && core_public->channel_event != NULL) core_public->channel_event(event, info); } static SpiceCoreInterfaceInternal core_interface_adapter = { .timer_add = adapter_timer_add, .timer_start = adapter_timer_start, .timer_cancel = adapter_timer_cancel, .timer_remove = adapter_timer_remove, .watch_add = adapter_watch_add, .watch_update_mask = adapter_watch_update_mask, .watch_remove = adapter_watch_remove, .channel_event = adapter_channel_event, }; /* Debugging only variable: allow multiple client connections to the spice * server */ #define SPICE_DEBUG_ALLOW_MC_ENV "SPICE_DEBUG_ALLOW_MC" #define MIGRATION_NOTIFY_SPICE_KEY "spice_mig_ext" #define REDS_MIG_VERSION 3 #define REDS_MIG_CONTINUE 1 #define REDS_MIG_ABORT 2 #define REDS_MIG_DIFF_VERSION 3 #define REDS_TOKENS_TO_SEND 5 #define REDS_VDI_PORT_NUM_RECEIVE_BUFFS 5 static pthread_mutex_t *lock_cs; static long *lock_count; /* TODO while we can technically create more than one server in a process, * the intended use is to support a single server per process */ GList *servers = NULL; typedef struct RedLinkInfo { RedsState *reds; RedsStream *stream; SpiceLinkHeader link_header; SpiceLinkMess *link_mess; int mess_pos; TicketInfo tiTicketing; SpiceLinkAuthMechanism auth_mechanism; int skip_auth; } RedLinkInfo; struct ChannelSecurityOptions { uint32_t channel_id; uint32_t options; ChannelSecurityOptions *next; }; typedef struct VDIReadBuf { RedCharDeviceVDIPort *dev; RingItem link; uint32_t refs; int len; uint8_t data[SPICE_AGENT_MAX_DATA_SIZE]; } VDIReadBuf; enum { VDI_PORT_READ_STATE_READ_HEADER, VDI_PORT_READ_STATE_GET_BUFF, VDI_PORT_READ_STATE_READ_DATA, }; struct RedCharDeviceVDIPortPrivate { gboolean agent_attached; uint32_t plug_generation; int client_agent_started; /* write to agent */ RedCharDeviceWriteBuffer *recv_from_client_buf; int recv_from_client_buf_pushed; AgentMsgFilter write_filter; /* read from agent */ Ring read_bufs; uint32_t read_state; uint32_t message_receive_len; uint8_t *receive_pos; uint32_t receive_len; VDIReadBuf *current_read_buf; AgentMsgFilter read_filter; VDIChunkHeader vdi_chunk_header; SpiceMigrateDataMain *mig_data; /* storing it when migration data arrives before agent is attached */ }; /* messages that are addressed to the agent and are created in the server */ typedef struct __attribute__ ((__packed__)) VDInternalBuf { VDIChunkHeader chunk_header; VDAgentMessage header; union { VDAgentMouseState mouse_state; } u; } VDInternalBuf; #define RED_TYPE_CHAR_DEVICE_VDIPORT red_char_device_vdi_port_get_type() #define RED_CHAR_DEVICE_VDIPORT(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj), RED_TYPE_CHAR_DEVICE_VDIPORT, RedCharDeviceVDIPort)) #define RED_CHAR_DEVICE_VDIPORT_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass), RED_TYPE_CHAR_DEVICE_VDIPORT, RedCharDeviceVDIPortClass)) #define RED_IS_CHAR_DEVICE_VDIPORT(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj), RED_TYPE_CHAR_DEVICE_VDIPORT)) #define RED_IS_CHAR_DEVICE_VDIPORT_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass), RED_TYPE_CHAR_DEVICE_VDIPORT)) #define RED_CHAR_DEVICE_VDIPORT_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS((obj), RED_TYPE_CHAR_DEVICE_VDIPORT, RedCharDeviceVDIPortClass)) typedef struct RedCharDeviceVDIPortClass RedCharDeviceVDIPortClass; typedef struct RedCharDeviceVDIPortPrivate RedCharDeviceVDIPortPrivate; struct RedCharDeviceVDIPort { RedCharDevice parent; RedCharDeviceVDIPortPrivate *priv; }; struct RedCharDeviceVDIPortClass { RedCharDeviceClass parent_class; }; static GType red_char_device_vdi_port_get_type(void) G_GNUC_CONST; G_DEFINE_TYPE(RedCharDeviceVDIPort, red_char_device_vdi_port, RED_TYPE_CHAR_DEVICE) #define RED_CHAR_DEVICE_VDIPORT_PRIVATE(o) (G_TYPE_INSTANCE_GET_PRIVATE ((o), RED_TYPE_CHAR_DEVICE_VDIPORT, RedCharDeviceVDIPortPrivate)) static RedCharDeviceVDIPort *red_char_device_vdi_port_new(RedsState *reds); static void migrate_timeout(void *opaque); static RedsMigTargetClient* reds_mig_target_client_find(RedsState *reds, RedClient *client); static void reds_mig_target_client_free(RedsMigTargetClient *mig_client); static void reds_mig_cleanup_wait_disconnect(RedsState *reds); static void reds_mig_remove_wait_disconnect_client(RedsState *reds, RedClient *client); static void reds_add_char_device(RedsState *reds, RedCharDevice *dev); static void reds_remove_char_device(RedsState *reds, RedCharDevice *dev); static void reds_send_mm_time(RedsState *reds); static void reds_on_ic_change(RedsState *reds); static void reds_on_sv_change(RedsState *reds); static void reds_on_vm_stop(RedsState *reds); static void reds_on_vm_start(RedsState *reds); static void reds_set_mouse_mode(RedsState *reds, uint32_t mode); static uint32_t reds_qxl_ram_size(RedsState *reds); static int calc_compression_level(RedsState *reds); static VDIReadBuf *vdi_port_get_read_buf(RedCharDeviceVDIPort *dev); static VDIReadBuf *vdi_port_read_buf_ref(VDIReadBuf *buf); static void vdi_port_read_buf_unref(VDIReadBuf *buf); static ChannelSecurityOptions *reds_find_channel_security(RedsState *reds, int id) { ChannelSecurityOptions *now = reds->channels_security; while (now && now->channel_id != id) { now = now->next; } return now; } void reds_handle_channel_event(RedsState *reds, int event, SpiceChannelEventInfo *info) { reds->core->channel_event(event, info); if (event == SPICE_CHANNEL_EVENT_DISCONNECTED) { free(info); } } static void reds_link_free(RedLinkInfo *link) { reds_stream_free(link->stream); link->stream = NULL; free(link->link_mess); link->link_mess = NULL; BN_free(link->tiTicketing.bn); link->tiTicketing.bn = NULL; if (link->tiTicketing.rsa) { RSA_free(link->tiTicketing.rsa); link->tiTicketing.rsa = NULL; } free(link); } #ifdef RED_STATISTICS static void reds_insert_stat_node(RedsState *reds, StatNodeRef parent, StatNodeRef ref) { SpiceStatNode *node = &reds->stat->nodes[ref]; uint32_t pos = INVALID_STAT_REF; uint32_t node_index; uint32_t *head; SpiceStatNode *n; node->first_child_index = INVALID_STAT_REF; head = (parent == INVALID_STAT_REF ? &reds->stat->root_index : &reds->stat->nodes[parent].first_child_index); node_index = *head; while (node_index != INVALID_STAT_REF && (n = &reds->stat->nodes[node_index]) && strcmp(node->name, n->name) > 0) { pos = node_index; node_index = n->next_sibling_index; } if (pos == INVALID_STAT_REF) { node->next_sibling_index = *head; *head = ref; } else { n = &reds->stat->nodes[pos]; node->next_sibling_index = n->next_sibling_index; n->next_sibling_index = ref; } } StatNodeRef stat_add_node(RedsState *reds, StatNodeRef parent, const char *name, int visible) { StatNodeRef ref; SpiceStatNode *node; spice_assert(name && strlen(name) > 0); if (strlen(name) >= sizeof(node->name)) { return INVALID_STAT_REF; } pthread_mutex_lock(&reds->stat_lock); ref = (parent == INVALID_STAT_REF ? reds->stat->root_index : reds->stat->nodes[parent].first_child_index); while (ref != INVALID_STAT_REF) { node = &reds->stat->nodes[ref]; if (strcmp(name, node->name)) { ref = node->next_sibling_index; } else { pthread_mutex_unlock(&reds->stat_lock); return ref; } } if (reds->stat->num_of_nodes >= REDS_MAX_STAT_NODES || reds->stat == NULL) { pthread_mutex_unlock(&reds->stat_lock); return INVALID_STAT_REF; } reds->stat->generation++; reds->stat->num_of_nodes++; for (ref = 0; ref <= REDS_MAX_STAT_NODES; ref++) { node = &reds->stat->nodes[ref]; if (!(node->flags & SPICE_STAT_NODE_FLAG_ENABLED)) { break; } } spice_assert(!(node->flags & SPICE_STAT_NODE_FLAG_ENABLED)); node->value = 0; node->flags = SPICE_STAT_NODE_FLAG_ENABLED | (visible ? SPICE_STAT_NODE_FLAG_VISIBLE : 0); g_strlcpy(node->name, name, sizeof(node->name)); reds_insert_stat_node(reds, parent, ref); pthread_mutex_unlock(&reds->stat_lock); return ref; } static void reds_stat_remove(RedsState *reds, SpiceStatNode *node) { pthread_mutex_lock(&reds->stat_lock); node->flags &= ~SPICE_STAT_NODE_FLAG_ENABLED; reds->stat->generation++; reds->stat->num_of_nodes--; pthread_mutex_unlock(&reds->stat_lock); } void stat_remove_node(RedsState *reds, StatNodeRef ref) { reds_stat_remove(reds, &reds->stat->nodes[ref]); } uint64_t *stat_add_counter(RedsState *reds, StatNodeRef parent, const char *name, int visible) { StatNodeRef ref = stat_add_node(reds, parent, name, visible); SpiceStatNode *node; if (ref == INVALID_STAT_REF) { return NULL; } node = &reds->stat->nodes[ref]; node->flags |= SPICE_STAT_NODE_FLAG_VALUE; return &node->value; } void stat_remove_counter(RedsState *reds, uint64_t *counter) { reds_stat_remove(reds, (SpiceStatNode *)(counter - offsetof(SpiceStatNode, value))); } void stat_update_value(RedsState *reds, uint32_t value) { RedsStatValue *stat_value = &reds->roundtrip_stat; stat_value->value = value; stat_value->min = (stat_value->count ? MIN(stat_value->min, value) : value); stat_value->max = MAX(stat_value->max, value); stat_value->average = (stat_value->average * stat_value->count + value) / (stat_value->count + 1); stat_value->count++; } #endif void reds_register_channel(RedsState *reds, RedChannel *channel) { spice_assert(reds); ring_add(&reds->channels, &channel->link); reds->num_of_channels++; } void reds_unregister_channel(RedsState *reds, RedChannel *channel) { if (ring_item_is_linked(&channel->link)) { ring_remove(&channel->link); reds->num_of_channels--; } else { spice_warning("not found"); } } static RedChannel *reds_find_channel(RedsState *reds, uint32_t type, uint32_t id) { RingItem *now; RING_FOREACH(now, &reds->channels) { RedChannel *channel = SPICE_CONTAINEROF(now, RedChannel, link); if (channel->type == type && channel->id == id) { return channel; } } return NULL; } static void reds_mig_cleanup(RedsState *reds) { if (reds->mig_inprogress) { if (reds->mig_wait_connect || reds->mig_wait_disconnect) { SpiceMigrateInterface *sif; spice_assert(reds->migration_interface); sif = SPICE_CONTAINEROF(reds->migration_interface->base.sif, SpiceMigrateInterface, base); if (reds->mig_wait_connect) { sif->migrate_connect_complete(reds->migration_interface); } else { if (sif->migrate_end_complete) { sif->migrate_end_complete(reds->migration_interface); } } } reds->mig_inprogress = FALSE; reds->mig_wait_connect = FALSE; reds->mig_wait_disconnect = FALSE; reds->core->timer_cancel(reds->mig_timer); reds_mig_cleanup_wait_disconnect(reds); } } static void reds_reset_vdp(RedsState *reds) { RedCharDeviceVDIPort *dev = reds->agent_dev; SpiceCharDeviceInterface *sif; dev->priv->read_state = VDI_PORT_READ_STATE_READ_HEADER; dev->priv->receive_pos = (uint8_t *)&dev->priv->vdi_chunk_header; dev->priv->receive_len = sizeof(dev->priv->vdi_chunk_header); dev->priv->message_receive_len = 0; if (dev->priv->current_read_buf) { vdi_port_read_buf_unref(dev->priv->current_read_buf); dev->priv->current_read_buf = NULL; } /* Reset read filter to start with clean state when the agent reconnects */ agent_msg_filter_init(&dev->priv->read_filter, reds->agent_copypaste, reds->agent_file_xfer, reds_use_client_monitors_config(reds), TRUE); /* Throw away pending chunks from the current (if any) and future * messages written by the client. * TODO: client should clear its agent messages queue when the agent * is disconnected. Currently, when an agent gets disconnected and reconnected, * messages that were directed to the previous instance of the agent continue * to be sent from the client. This TODO will require server, protocol, and client changes */ dev->priv->write_filter.result = AGENT_MSG_FILTER_DISCARD; dev->priv->write_filter.discard_all = TRUE; dev->priv->client_agent_started = FALSE; /* resetting and not destroying the dev as a workaround for a bad * tokens management in the vdagent protocol: * The client tokens' are set only once, when the main channel is initialized. * Instead, it would have been more appropriate to reset them upon AGEN_CONNECT. * The client tokens are tracked as part of the RedCharDeviceClient. Thus, * in order to be backward compatible with the client, we need to track the tokens * even if the agent is detached. We don't destroy the char_device, and * instead we just reset it. * In addition, there used to be a misshandling of AGENT_TOKENS message in spice-gtk: it * overrides the amount of tokens, instead of adding the given amount. */ if (red_channel_test_remote_cap(&reds->main_channel->base, SPICE_MAIN_CAP_AGENT_CONNECTED_TOKENS)) { dev->priv->agent_attached = FALSE; } else { red_char_device_reset(RED_CHAR_DEVICE(dev)); } sif = spice_char_device_get_interface(reds->vdagent); if (sif->state) { sif->state(reds->vdagent, 0); } } static int reds_main_channel_connected(RedsState *reds) { return main_channel_is_connected(reds->main_channel); } void reds_client_disconnect(RedsState *reds, RedClient *client) { RedsMigTargetClient *mig_client; if (reds->exit_on_disconnect) { spice_info("Exiting server because of client disconnect.\n"); exit(0); } if (!client || client->disconnecting) { spice_debug("client %p already during disconnection", client); return; } spice_info(NULL); /* disconnecting is set to prevent recursion because of the following: * main_channel_client_on_disconnect-> * reds_client_disconnect->red_client_destroy->main_channel... */ client->disconnecting = TRUE; // TODO: we need to handle agent properly for all clients!!!! (e.g., cut and paste, how?) // We shouldn't initialize the agent when there are still clients connected mig_client = reds_mig_target_client_find(reds, client); if (mig_client) { reds_mig_target_client_free(mig_client); } if (reds->mig_wait_disconnect) { reds_mig_remove_wait_disconnect_client(reds, client); } if (reds->agent_dev->priv->agent_attached) { /* note that vdagent might be NULL, if the vdagent was once * up and than was removed */ if (red_char_device_client_exists(RED_CHAR_DEVICE(reds->agent_dev), client)) { red_char_device_client_remove(RED_CHAR_DEVICE(reds->agent_dev), client); } } ring_remove(&client->link); reds->num_clients--; red_client_destroy(client); // TODO: we need to handle agent properly for all clients!!!! (e.g., cut and paste, how? Maybe throw away messages // if we are in the middle of one from another client) if (reds->num_clients == 0) { /* Let the agent know the client is disconnected */ if (reds->agent_dev->priv->agent_attached) { RedCharDeviceWriteBuffer *char_dev_buf; VDInternalBuf *internal_buf; uint32_t total_msg_size; total_msg_size = sizeof(VDIChunkHeader) + sizeof(VDAgentMessage); char_dev_buf = red_char_device_write_buffer_get_server_no_token( RED_CHAR_DEVICE(reds->agent_dev), total_msg_size); char_dev_buf->buf_used = total_msg_size; internal_buf = (VDInternalBuf *)char_dev_buf->buf; internal_buf->chunk_header.port = VDP_SERVER_PORT; internal_buf->chunk_header.size = sizeof(VDAgentMessage); internal_buf->header.protocol = VD_AGENT_PROTOCOL; internal_buf->header.type = VD_AGENT_CLIENT_DISCONNECTED; internal_buf->header.opaque = 0; internal_buf->header.size = 0; red_char_device_write_buffer_add(RED_CHAR_DEVICE(reds->agent_dev), char_dev_buf); } /* Reset write filter to start with clean state on client reconnect */ agent_msg_filter_init(&reds->agent_dev->priv->write_filter, reds->agent_copypaste, reds->agent_file_xfer, reds_use_client_monitors_config(reds), TRUE); /* Throw away pending chunks from the current (if any) and future * messages read from the agent */ reds->agent_dev->priv->read_filter.result = AGENT_MSG_FILTER_DISCARD; reds->agent_dev->priv->read_filter.discard_all = TRUE; free(reds->agent_dev->priv->mig_data); reds->agent_dev->priv->mig_data = NULL; reds_mig_cleanup(reds); } } // TODO: go over all usage of reds_disconnect, most/some of it should be converted to // reds_client_disconnect static void reds_disconnect(RedsState *reds) { RingItem *link, *next; spice_info(NULL); RING_FOREACH_SAFE(link, next, &reds->clients) { reds_client_disconnect(reds, SPICE_CONTAINEROF(link, RedClient, link)); } reds_mig_cleanup(reds); } static void reds_mig_disconnect(RedsState *reds) { if (reds_main_channel_connected(reds)) { reds_disconnect(reds); } else { reds_mig_cleanup(reds); } } int reds_get_mouse_mode(RedsState *reds) { return reds->mouse_mode; } static void reds_set_mouse_mode(RedsState *reds, uint32_t mode) { GList *l; if (reds->mouse_mode == mode) { return; } reds->mouse_mode = mode; for (l = reds->qxl_instances; l != NULL; l = l->next) { QXLInstance *qxl = (QXLInstance *)l->data; red_qxl_set_mouse_mode(qxl, mode); } main_channel_push_mouse_mode(reds->main_channel, reds->mouse_mode, reds->is_client_mouse_allowed); } gboolean reds_get_agent_mouse(const RedsState *reds) { return reds->agent_mouse; } static void reds_update_mouse_mode(RedsState *reds) { int allowed = 0; int qxl_count = g_list_length(reds->qxl_instances); if ((reds->agent_mouse && reds->vdagent) || (inputs_channel_has_tablet(reds->inputs_channel) && qxl_count == 1)) { allowed = reds->dispatcher_allows_client_mouse; } if (allowed == reds->is_client_mouse_allowed) { return; } reds->is_client_mouse_allowed = allowed; if (reds->mouse_mode == SPICE_MOUSE_MODE_CLIENT && !allowed) { reds_set_mouse_mode(reds, SPICE_MOUSE_MODE_SERVER); return; } if (reds->main_channel) { main_channel_push_mouse_mode(reds->main_channel, reds->mouse_mode, reds->is_client_mouse_allowed); } } static void reds_agent_remove(RedsState *reds) { // TODO: agent is broken with multiple clients. also need to figure out what to do when // part of the clients are during target migration. reds_reset_vdp(reds); reds->vdagent = NULL; reds_update_mouse_mode(reds); if (reds_main_channel_connected(reds) && !red_channel_is_waiting_for_migrate_data(&reds->main_channel->base)) { main_channel_push_agent_disconnected(reds->main_channel); } } /******************************* * Char device state callbacks * * *****************************/ static void vdi_port_read_buf_release(uint8_t *data, void *opaque) { VDIReadBuf *buf = (VDIReadBuf *)opaque; vdi_port_read_buf_unref(buf); } /* returns TRUE if the buffer can be forwarded */ static gboolean vdi_port_read_buf_process(RedCharDeviceVDIPort *dev, VDIReadBuf *buf, gboolean *error) { int res; *error = FALSE; switch (dev->priv->vdi_chunk_header.port) { case VDP_CLIENT_PORT: { res = agent_msg_filter_process_data(&dev->priv->read_filter, buf->data, buf->len); switch (res) { case AGENT_MSG_FILTER_OK: return TRUE; case AGENT_MSG_FILTER_DISCARD: return FALSE; case AGENT_MSG_FILTER_PROTO_ERROR: *error = TRUE; return FALSE; } } case VDP_SERVER_PORT: return FALSE; default: spice_warning("invalid port"); *error = TRUE; return FALSE; } } static VDIReadBuf *vdi_port_get_read_buf(RedCharDeviceVDIPort *dev) { RingItem *item; VDIReadBuf *buf; if (!(item = ring_get_head(&dev->priv->read_bufs))) { return NULL; } ring_remove(item); buf = SPICE_CONTAINEROF(item, VDIReadBuf, link); buf->refs = 1; return buf; } static VDIReadBuf* vdi_port_read_buf_ref(VDIReadBuf *buf) { buf->refs++; return buf; } /* FIXME: refactor so that unreffing the VDIReadBuf doesn't require accessing * RedsState? */ static void vdi_port_read_buf_unref(VDIReadBuf *buf) { if (!--buf->refs) { ring_add(&buf->dev->priv->read_bufs, &buf->link); /* read_one_msg_from_vdi_port may have never completed because the read_bufs ring was empty. So we call it again so it can complete its work if necessary. Note that since we can be called from red_char_device_wakeup this can cause recursion, but we have protection for that */ if (buf->dev->priv->agent_attached) { red_char_device_wakeup(RED_CHAR_DEVICE(buf->dev)); } } } /* reads from the device till completes reading a message that is addressed to the client, * or otherwise, when reading from the device fails */ static RedCharDeviceMsgToClient *vdi_port_read_one_msg_from_device(SpiceCharDeviceInstance *sin, void *opaque) { RedsState *reds = opaque; RedCharDeviceVDIPort *dev = reds->agent_dev; SpiceCharDeviceInterface *sif; VDIReadBuf *dispatch_buf; int n; if (!reds->vdagent) { return NULL; } spice_assert(reds->vdagent == sin); sif = spice_char_device_get_interface(reds->vdagent); while (reds->vdagent) { switch (dev->priv->read_state) { case VDI_PORT_READ_STATE_READ_HEADER: n = sif->read(reds->vdagent, dev->priv->receive_pos, dev->priv->receive_len); if (!n) { return NULL; } if ((dev->priv->receive_len -= n)) { dev->priv->receive_pos += n; return NULL; } dev->priv->message_receive_len = dev->priv->vdi_chunk_header.size; dev->priv->read_state = VDI_PORT_READ_STATE_GET_BUFF; case VDI_PORT_READ_STATE_GET_BUFF: { if (!(dev->priv->current_read_buf = vdi_port_get_read_buf(reds->agent_dev))) { return NULL; } dev->priv->receive_pos = dev->priv->current_read_buf->data; dev->priv->receive_len = MIN(dev->priv->message_receive_len, sizeof(dev->priv->current_read_buf->data)); dev->priv->current_read_buf->len = dev->priv->receive_len; dev->priv->message_receive_len -= dev->priv->receive_len; dev->priv->read_state = VDI_PORT_READ_STATE_READ_DATA; } case VDI_PORT_READ_STATE_READ_DATA: { gboolean error = FALSE; n = sif->read(reds->vdagent, dev->priv->receive_pos, dev->priv->receive_len); if (!n) { return NULL; } if ((dev->priv->receive_len -= n)) { dev->priv->receive_pos += n; break; } dispatch_buf = dev->priv->current_read_buf; dev->priv->current_read_buf = NULL; dev->priv->receive_pos = NULL; if (dev->priv->message_receive_len == 0) { dev->priv->read_state = VDI_PORT_READ_STATE_READ_HEADER; dev->priv->receive_pos = (uint8_t *)&dev->priv->vdi_chunk_header; dev->priv->receive_len = sizeof(dev->priv->vdi_chunk_header); } else { dev->priv->read_state = VDI_PORT_READ_STATE_GET_BUFF; } if (vdi_port_read_buf_process(reds->agent_dev, dispatch_buf, &error)) { return dispatch_buf; } else { if (error) { reds_agent_remove(reds); } vdi_port_read_buf_unref(dispatch_buf); } } } /* END switch */ } /* END while */ return NULL; } static RedCharDeviceMsgToClient *vdi_port_ref_msg_to_client(RedCharDeviceMsgToClient *msg, void *opaque) { return vdi_port_read_buf_ref(msg); } static void vdi_port_unref_msg_to_client(RedCharDeviceMsgToClient *msg, void *opaque) { vdi_port_read_buf_unref(msg); } /* after calling this, we unref the message, and the ref is in the instance side */ static void vdi_port_send_msg_to_client(RedCharDeviceMsgToClient *msg, RedClient *client, void *opaque) { VDIReadBuf *agent_data_buf = msg; main_channel_client_push_agent_data(red_client_get_main(client), agent_data_buf->data, agent_data_buf->len, vdi_port_read_buf_release, vdi_port_read_buf_ref(agent_data_buf)); } static void vdi_port_send_tokens_to_client(RedClient *client, uint32_t tokens, void *opaque) { main_channel_client_push_agent_tokens(red_client_get_main(client), tokens); } static void vdi_port_on_free_self_token(void *opaque) { RedsState *reds = opaque; if (reds->inputs_channel && reds->pending_mouse_event) { spice_debug("pending mouse event"); reds_handle_agent_mouse_event(reds, inputs_channel_get_mouse_state(reds->inputs_channel)); } } static void vdi_port_remove_client(RedClient *client, void *opaque) { red_channel_client_shutdown(main_channel_client_get_base( red_client_get_main(client))); } /****************************************************************************/ int reds_has_vdagent(RedsState *reds) { return !!reds->vdagent; } void reds_handle_agent_mouse_event(RedsState *reds, const VDAgentMouseState *mouse_state) { RedCharDeviceWriteBuffer *char_dev_buf; VDInternalBuf *internal_buf; uint32_t total_msg_size; if (!reds->inputs_channel || !reds->agent_dev->priv->agent_attached) { return; } total_msg_size = sizeof(VDIChunkHeader) + sizeof(VDAgentMessage) + sizeof(VDAgentMouseState); char_dev_buf = red_char_device_write_buffer_get(RED_CHAR_DEVICE(reds->agent_dev), NULL, total_msg_size); if (!char_dev_buf) { reds->pending_mouse_event = TRUE; return; } reds->pending_mouse_event = FALSE; internal_buf = (VDInternalBuf *)char_dev_buf->buf; internal_buf->chunk_header.port = VDP_SERVER_PORT; internal_buf->chunk_header.size = sizeof(VDAgentMessage) + sizeof(VDAgentMouseState); internal_buf->header.protocol = VD_AGENT_PROTOCOL; internal_buf->header.type = VD_AGENT_MOUSE_STATE; internal_buf->header.opaque = 0; internal_buf->header.size = sizeof(VDAgentMouseState); internal_buf->u.mouse_state = *mouse_state; char_dev_buf->buf_used = total_msg_size; red_char_device_write_buffer_add(RED_CHAR_DEVICE(reds->agent_dev), char_dev_buf); } int reds_get_n_channels(RedsState *reds) { return reds ? reds->num_of_channels : 0; } static int reds_get_n_clients(RedsState *reds) { return reds ? reds->num_clients : 0; } SPICE_GNUC_VISIBLE int spice_server_get_num_clients(SpiceServer *reds) { return reds_get_n_clients(reds); } static int channel_supports_multiple_clients(RedChannel *channel) { switch (channel->type) { case SPICE_CHANNEL_MAIN: case SPICE_CHANNEL_DISPLAY: case SPICE_CHANNEL_CURSOR: case SPICE_CHANNEL_INPUTS: return TRUE; } return FALSE; } void reds_fill_channels(RedsState *reds, SpiceMsgChannels *channels_info) { RingItem *now; int used_channels = 0; channels_info->num_of_channels = reds->num_of_channels; RING_FOREACH(now, &reds->channels) { RedChannel *channel = SPICE_CONTAINEROF(now, RedChannel, link); if (reds->num_clients > 1 && !channel_supports_multiple_clients(channel)) { continue; } channels_info->channels[used_channels].type = channel->type; channels_info->channels[used_channels].id = channel->id; used_channels++; } channels_info->num_of_channels = used_channels; if (used_channels != reds->num_of_channels) { spice_warning("sent %d out of %d", used_channels, reds->num_of_channels); } } void reds_on_main_agent_start(RedsState *reds, MainChannelClient *mcc, uint32_t num_tokens) { RedCharDevice *dev_state = RED_CHAR_DEVICE(reds->agent_dev); RedChannelClient *rcc; if (!reds->vdagent) { return; } spice_assert(reds->vdagent->st && reds->vdagent->st == dev_state); rcc = main_channel_client_get_base(mcc); reds->agent_dev->priv->client_agent_started = TRUE; /* * Note that in older releases, send_tokens were set to ~0 on both client * and server. The server ignored the client given tokens. * Thanks to that, when an old client is connected to a new server, * and vice versa, the sending from the server to the client won't have * flow control, but will have no other problem. */ if (!red_char_device_client_exists(dev_state, rcc->client)) { int client_added; client_added = red_char_device_client_add(dev_state, rcc->client, TRUE, /* flow control */ REDS_VDI_PORT_NUM_RECEIVE_BUFFS, REDS_AGENT_WINDOW_SIZE, num_tokens, red_channel_client_is_waiting_for_migrate_data(rcc)); if (!client_added) { spice_warning("failed to add client to agent"); red_channel_client_shutdown(rcc); return; } } else { red_char_device_send_to_client_tokens_set(dev_state, rcc->client, num_tokens); } agent_msg_filter_config(&reds->agent_dev->priv->write_filter, reds->agent_copypaste, reds->agent_file_xfer, reds_use_client_monitors_config(reds)); reds->agent_dev->priv->write_filter.discard_all = FALSE; } void reds_on_main_agent_tokens(RedsState *reds, MainChannelClient *mcc, uint32_t num_tokens) { if (!reds->vdagent) { return; } spice_assert(reds->vdagent->st); red_char_device_send_to_client_tokens_add(reds->vdagent->st, main_channel_client_get_base(mcc)->client, num_tokens); } uint8_t *reds_get_agent_data_buffer(RedsState *reds, MainChannelClient *mcc, size_t size) { RedCharDeviceVDIPort *dev = reds->agent_dev; RedClient *client; if (!dev->priv->client_agent_started) { /* * agent got disconnected, and possibly got reconnected, but we still can receive * msgs that are addressed to the agent's old instance, in case they were * sent by the client before it received the AGENT_DISCONNECTED msg. * In such case, we will receive and discard the msgs (reds_reset_vdp takes care * of setting dev->write_filter.result = AGENT_MSG_FILTER_DISCARD). */ return spice_malloc(size); } spice_assert(dev->priv->recv_from_client_buf == NULL); client = main_channel_client_get_base(mcc)->client; dev->priv->recv_from_client_buf = red_char_device_write_buffer_get(RED_CHAR_DEVICE(dev), client, size + sizeof(VDIChunkHeader)); dev->priv->recv_from_client_buf_pushed = FALSE; return dev->priv->recv_from_client_buf->buf + sizeof(VDIChunkHeader); } void reds_release_agent_data_buffer(RedsState *reds, uint8_t *buf) { RedCharDeviceVDIPort *dev = reds->agent_dev; if (!dev->priv->recv_from_client_buf) { free(buf); return; } spice_assert(buf == dev->priv->recv_from_client_buf->buf + sizeof(VDIChunkHeader)); if (!dev->priv->recv_from_client_buf_pushed) { red_char_device_write_buffer_release(RED_CHAR_DEVICE(reds->agent_dev), dev->priv->recv_from_client_buf); } dev->priv->recv_from_client_buf = NULL; dev->priv->recv_from_client_buf_pushed = FALSE; } static void reds_client_monitors_config_cleanup(RedsState *reds) { RedsClientMonitorsConfig *cmc = &reds->client_monitors_config; cmc->buffer_size = cmc->buffer_pos = 0; free(cmc->buffer); cmc->buffer = NULL; cmc->mcc = NULL; } static void reds_on_main_agent_monitors_config(RedsState *reds, MainChannelClient *mcc, void *message, size_t size) { VDAgentMessage *msg_header; VDAgentMonitorsConfig *monitors_config; RedsClientMonitorsConfig *cmc = &reds->client_monitors_config; cmc->buffer_size += size; cmc->buffer = realloc(cmc->buffer, cmc->buffer_size); spice_assert(cmc->buffer); cmc->mcc = mcc; memcpy(cmc->buffer + cmc->buffer_pos, message, size); cmc->buffer_pos += size; msg_header = (VDAgentMessage *)cmc->buffer; if (sizeof(VDAgentMessage) > cmc->buffer_size || msg_header->size > cmc->buffer_size - sizeof(VDAgentMessage)) { spice_debug("not enough data yet. %d", cmc->buffer_size); return; } monitors_config = (VDAgentMonitorsConfig *)(cmc->buffer + sizeof(*msg_header)); spice_debug("%s: %d", __func__, monitors_config->num_of_monitors); reds_client_monitors_config(reds, monitors_config); reds_client_monitors_config_cleanup(reds); } void reds_on_main_agent_data(RedsState *reds, MainChannelClient *mcc, void *message, size_t size) { RedCharDeviceVDIPort *dev = reds->agent_dev; VDIChunkHeader *header; int res; res = agent_msg_filter_process_data(&reds->agent_dev->priv->write_filter, message, size); switch (res) { case AGENT_MSG_FILTER_OK: break; case AGENT_MSG_FILTER_DISCARD: return; case AGENT_MSG_FILTER_MONITORS_CONFIG: reds_on_main_agent_monitors_config(reds, mcc, message, size); return; case AGENT_MSG_FILTER_PROTO_ERROR: red_channel_client_shutdown(main_channel_client_get_base(mcc)); return; } spice_assert(reds->agent_dev->priv->recv_from_client_buf); spice_assert(message == reds->agent_dev->priv->recv_from_client_buf->buf + sizeof(VDIChunkHeader)); // TODO - start tracking agent data per channel header = (VDIChunkHeader *)dev->priv->recv_from_client_buf->buf; header->port = VDP_CLIENT_PORT; header->size = size; dev->priv->recv_from_client_buf->buf_used = sizeof(VDIChunkHeader) + size; dev->priv->recv_from_client_buf_pushed = TRUE; red_char_device_write_buffer_add(RED_CHAR_DEVICE(reds->agent_dev), dev->priv->recv_from_client_buf); } void reds_on_main_migrate_connected(RedsState *reds, int seamless) { reds->src_do_seamless_migrate = seamless; if (reds->mig_wait_connect) { reds_mig_cleanup(reds); } } void reds_on_main_mouse_mode_request(RedsState *reds, void *message, size_t size) { switch (((SpiceMsgcMainMouseModeRequest *)message)->mode) { case SPICE_MOUSE_MODE_CLIENT: if (reds->is_client_mouse_allowed) { reds_set_mouse_mode(reds, SPICE_MOUSE_MODE_CLIENT); } else { spice_info("client mouse is disabled"); } break; case SPICE_MOUSE_MODE_SERVER: reds_set_mouse_mode(reds, SPICE_MOUSE_MODE_SERVER); break; default: spice_warning("unsupported mouse mode"); } } /* * Push partial agent data, even if not all the chunk was consumend, * in order to avoid the roundtrip (src-server->client->dest-server) */ void reds_on_main_channel_migrate(RedsState *reds, MainChannelClient *mcc) { RedCharDeviceVDIPort *agent_dev = reds->agent_dev; uint32_t read_data_len; spice_assert(reds->num_clients == 1); if (agent_dev->priv->read_state != VDI_PORT_READ_STATE_READ_DATA) { return; } spice_assert(agent_dev->priv->current_read_buf->data && agent_dev->priv->receive_pos > agent_dev->priv->current_read_buf->data); read_data_len = agent_dev->priv->receive_pos - agent_dev->priv->current_read_buf->data; if (agent_dev->priv->read_filter.msg_data_to_read || read_data_len > sizeof(VDAgentMessage)) { /* msg header has been read */ VDIReadBuf *read_buf = agent_dev->priv->current_read_buf; gboolean error = FALSE; spice_debug("push partial read %u (msg first chunk? %d)", read_data_len, !agent_dev->priv->read_filter.msg_data_to_read); read_buf->len = read_data_len; if (vdi_port_read_buf_process(reds->agent_dev, read_buf, &error)) { main_channel_client_push_agent_data(mcc, read_buf->data, read_buf->len, vdi_port_read_buf_release, read_buf); } else { if (error) { reds_agent_remove(reds); } vdi_port_read_buf_unref(read_buf); } spice_assert(agent_dev->priv->receive_len); agent_dev->priv->message_receive_len += agent_dev->priv->receive_len; agent_dev->priv->read_state = VDI_PORT_READ_STATE_GET_BUFF; agent_dev->priv->current_read_buf = NULL; agent_dev->priv->receive_pos = NULL; } } void reds_marshall_migrate_data(RedsState *reds, SpiceMarshaller *m) { SpiceMigrateDataMain mig_data; RedCharDeviceVDIPort *agent_dev = reds->agent_dev; SpiceMarshaller *m2; memset(&mig_data, 0, sizeof(mig_data)); spice_marshaller_add_uint32(m, SPICE_MIGRATE_DATA_MAIN_MAGIC); spice_marshaller_add_uint32(m, SPICE_MIGRATE_DATA_MAIN_VERSION); if (!reds->vdagent) { uint8_t *null_agent_mig_data; /* MSG_AGENT_CONNECTED_TOKENS is supported by the client (see spice_server_migrate_connect), so agent_attached is set to FALSE when the agent is disconnected and there is no need to track the client tokens (see reds_reset_vdp) */ spice_assert(!agent_dev->priv->agent_attached); red_char_device_migrate_data_marshall_empty(m); null_agent_mig_data = spice_marshaller_reserve_space(m, sizeof(SpiceMigrateDataMain) - sizeof(SpiceMigrateDataCharDevice)); memset(null_agent_mig_data, 0, sizeof(SpiceMigrateDataMain) - sizeof(SpiceMigrateDataCharDevice)); return; } red_char_device_migrate_data_marshall(RED_CHAR_DEVICE(reds->agent_dev), m); spice_marshaller_add_uint8(m, reds->agent_dev->priv->client_agent_started); mig_data.agent2client.chunk_header = agent_dev->priv->vdi_chunk_header; /* agent to client partial msg */ if (agent_dev->priv->read_state == VDI_PORT_READ_STATE_READ_HEADER) { mig_data.agent2client.chunk_header_size = agent_dev->priv->receive_pos - (uint8_t *)&agent_dev->priv->vdi_chunk_header; mig_data.agent2client.msg_header_done = FALSE; mig_data.agent2client.msg_header_partial_len = 0; spice_assert(!agent_dev->priv->read_filter.msg_data_to_read); } else { mig_data.agent2client.chunk_header_size = sizeof(VDIChunkHeader); mig_data.agent2client.chunk_header.size = agent_dev->priv->message_receive_len; if (agent_dev->priv->read_state == VDI_PORT_READ_STATE_READ_DATA) { /* in the middle of reading the message header (see reds_on_main_channel_migrate) */ mig_data.agent2client.msg_header_done = FALSE; mig_data.agent2client.msg_header_partial_len = agent_dev->priv->receive_pos - agent_dev->priv->current_read_buf->data; spice_assert(mig_data.agent2client.msg_header_partial_len < sizeof(VDAgentMessage)); spice_assert(!agent_dev->priv->read_filter.msg_data_to_read); } else { mig_data.agent2client.msg_header_done = TRUE; mig_data.agent2client.msg_remaining = agent_dev->priv->read_filter.msg_data_to_read; mig_data.agent2client.msg_filter_result = agent_dev->priv->read_filter.result; } } spice_marshaller_add_uint32(m, mig_data.agent2client.chunk_header_size); spice_marshaller_add(m, (uint8_t *)&mig_data.agent2client.chunk_header, sizeof(VDIChunkHeader)); spice_marshaller_add_uint8(m, mig_data.agent2client.msg_header_done); spice_marshaller_add_uint32(m, mig_data.agent2client.msg_header_partial_len); m2 = spice_marshaller_get_ptr_submarshaller(m, 0); spice_marshaller_add(m2, agent_dev->priv->current_read_buf->data, mig_data.agent2client.msg_header_partial_len); spice_marshaller_add_uint32(m, mig_data.agent2client.msg_remaining); spice_marshaller_add_uint8(m, mig_data.agent2client.msg_filter_result); mig_data.client2agent.msg_remaining = agent_dev->priv->write_filter.msg_data_to_read; mig_data.client2agent.msg_filter_result = agent_dev->priv->write_filter.result; spice_marshaller_add_uint32(m, mig_data.client2agent.msg_remaining); spice_marshaller_add_uint8(m, mig_data.client2agent.msg_filter_result); spice_debug("from agent filter: discard all %d, wait_msg %u, msg_filter_result %d", agent_dev->priv->read_filter.discard_all, agent_dev->priv->read_filter.msg_data_to_read, agent_dev->priv->read_filter.result); spice_debug("to agent filter: discard all %d, wait_msg %u, msg_filter_result %d", agent_dev->priv->write_filter.discard_all, agent_dev->priv->write_filter.msg_data_to_read, agent_dev->priv->write_filter.result); } static int reds_agent_state_restore(RedsState *reds, SpiceMigrateDataMain *mig_data) { RedCharDeviceVDIPort *agent_dev = reds->agent_dev; uint32_t chunk_header_remaining; agent_dev->priv->vdi_chunk_header = mig_data->agent2client.chunk_header; spice_assert(mig_data->agent2client.chunk_header_size <= sizeof(VDIChunkHeader)); chunk_header_remaining = sizeof(VDIChunkHeader) - mig_data->agent2client.chunk_header_size; if (chunk_header_remaining) { agent_dev->priv->read_state = VDI_PORT_READ_STATE_READ_HEADER; agent_dev->priv->receive_pos = (uint8_t *)&agent_dev->priv->vdi_chunk_header + mig_data->agent2client.chunk_header_size; agent_dev->priv->receive_len = chunk_header_remaining; } else { agent_dev->priv->message_receive_len = agent_dev->priv->vdi_chunk_header.size; } if (!mig_data->agent2client.msg_header_done) { uint8_t *partial_msg_header; if (!chunk_header_remaining) { uint32_t cur_buf_size; agent_dev->priv->read_state = VDI_PORT_READ_STATE_READ_DATA; agent_dev->priv->current_read_buf = vdi_port_get_read_buf(reds->agent_dev); spice_assert(agent_dev->priv->current_read_buf); partial_msg_header = (uint8_t *)mig_data + mig_data->agent2client.msg_header_ptr - sizeof(SpiceMiniDataHeader); memcpy(agent_dev->priv->current_read_buf->data, partial_msg_header, mig_data->agent2client.msg_header_partial_len); agent_dev->priv->receive_pos = agent_dev->priv->current_read_buf->data + mig_data->agent2client.msg_header_partial_len; cur_buf_size = sizeof(agent_dev->priv->current_read_buf->data) - mig_data->agent2client.msg_header_partial_len; agent_dev->priv->receive_len = MIN(agent_dev->priv->message_receive_len, cur_buf_size); agent_dev->priv->current_read_buf->len = agent_dev->priv->receive_len + mig_data->agent2client.msg_header_partial_len; agent_dev->priv->message_receive_len -= agent_dev->priv->receive_len; } else { spice_assert(mig_data->agent2client.msg_header_partial_len == 0); } } else { agent_dev->priv->read_state = VDI_PORT_READ_STATE_GET_BUFF; agent_dev->priv->current_read_buf = NULL; agent_dev->priv->receive_pos = NULL; agent_dev->priv->read_filter.msg_data_to_read = mig_data->agent2client.msg_remaining; agent_dev->priv->read_filter.result = mig_data->agent2client.msg_filter_result; } agent_dev->priv->read_filter.discard_all = FALSE; agent_dev->priv->write_filter.discard_all = !mig_data->client_agent_started; agent_dev->priv->client_agent_started = mig_data->client_agent_started; agent_dev->priv->write_filter.msg_data_to_read = mig_data->client2agent.msg_remaining; agent_dev->priv->write_filter.result = mig_data->client2agent.msg_filter_result; spice_debug("to agent filter: discard all %d, wait_msg %u, msg_filter_result %d", agent_dev->priv->write_filter.discard_all, agent_dev->priv->write_filter.msg_data_to_read, agent_dev->priv->write_filter.result); spice_debug("from agent filter: discard all %d, wait_msg %u, msg_filter_result %d", agent_dev->priv->read_filter.discard_all, agent_dev->priv->read_filter.msg_data_to_read, agent_dev->priv->read_filter.result); return red_char_device_restore(RED_CHAR_DEVICE(agent_dev), &mig_data->agent_base); } /* * The agent device is not attached to the dest before migration is completed. It is * attached only after the vm is started. It might be attached before or after * the migration data has reached the server. */ int reds_handle_migrate_data(RedsState *reds, MainChannelClient *mcc, SpiceMigrateDataMain *mig_data, uint32_t size) { RedCharDeviceVDIPort *agent_dev = reds->agent_dev; spice_debug("main-channel: got migrate data"); /* * Now that the client has switched to the target server, if main_channel * controls the mm-time, we update the client's mm-time. * (MSG_MAIN_INIT is not sent for a migrating connection) */ if (reds->mm_time_enabled) { reds_send_mm_time(reds); } if (mig_data->agent_base.connected) { if (agent_dev->priv->agent_attached) { // agent was attached before migration data has arrived if (!reds->vdagent) { spice_assert(agent_dev->priv->plug_generation > 0); main_channel_push_agent_disconnected(reds->main_channel); spice_debug("agent is no longer connected"); } else { if (agent_dev->priv->plug_generation > 1) { /* red_char_device_state_reset takes care of not making the device wait for migration data */ spice_debug("agent has been detached and reattached before receiving migration data"); main_channel_push_agent_disconnected(reds->main_channel); main_channel_push_agent_connected(reds->main_channel); } else { spice_debug("restoring state from mig_data"); return reds_agent_state_restore(reds, mig_data); } } } else { /* restore agent starte when the agent gets attached */ spice_debug("saving mig_data"); spice_assert(agent_dev->priv->plug_generation == 0); agent_dev->priv->mig_data = spice_memdup(mig_data, size); } } else { spice_debug("agent was not attached on the source host"); if (reds->vdagent) { /* red_char_device_client_remove disables waiting for migration data */ red_char_device_client_remove(RED_CHAR_DEVICE(agent_dev), main_channel_client_get_base(mcc)->client); main_channel_push_agent_connected(reds->main_channel); } } return TRUE; } static void reds_channel_init_auth_caps(RedLinkInfo *link, RedChannel *channel) { RedsState *reds = link->reds; if (reds->sasl_enabled && !link->skip_auth) { red_channel_set_common_cap(channel, SPICE_COMMON_CAP_AUTH_SASL); } else { red_channel_set_common_cap(channel, SPICE_COMMON_CAP_AUTH_SPICE); } red_channel_set_common_cap(channel, SPICE_COMMON_CAP_PROTOCOL_AUTH_SELECTION); } static const uint32_t *red_link_info_get_caps(const RedLinkInfo *link) { const uint8_t *caps_start = (const uint8_t *)link->link_mess; return (const uint32_t *)(caps_start + link->link_mess->caps_offset); } static bool red_link_info_test_capability(const RedLinkInfo *link, uint32_t cap) { const uint32_t *caps = red_link_info_get_caps(link); return test_capability(caps, link->link_mess->num_common_caps, cap); } static int reds_send_link_ack(RedsState *reds, RedLinkInfo *link) { SpiceLinkHeader header; SpiceLinkReply ack; RedChannel *channel; RedChannelCapabilities *channel_caps; BUF_MEM *bmBuf; BIO *bio = NULL; int ret = FALSE; size_t hdr_size; header.magic = SPICE_MAGIC; hdr_size = sizeof(ack); header.major_version = GUINT32_TO_LE(SPICE_VERSION_MAJOR); header.minor_version = GUINT32_TO_LE(SPICE_VERSION_MINOR); ack.error = GUINT32_TO_LE(SPICE_LINK_ERR_OK); channel = reds_find_channel(reds, link->link_mess->channel_type, link->link_mess->channel_id); if (!channel) { if (link->link_mess->channel_type != SPICE_CHANNEL_MAIN) { spice_warning("Received wrong header: channel_type != SPICE_CHANNEL_MAIN"); return FALSE; } spice_assert(reds->main_channel); channel = &reds->main_channel->base; } reds_channel_init_auth_caps(link, channel); /* make sure common caps are set */ channel_caps = &channel->local_caps; ack.num_common_caps = GUINT32_TO_LE(channel_caps->num_common_caps); ack.num_channel_caps = GUINT32_TO_LE(channel_caps->num_caps); hdr_size += channel_caps->num_common_caps * sizeof(uint32_t); hdr_size += channel_caps->num_caps * sizeof(uint32_t); header.size = GUINT32_TO_LE(hdr_size); ack.caps_offset = GUINT32_TO_LE(sizeof(SpiceLinkReply)); if (!reds->sasl_enabled || !red_link_info_test_capability(link, SPICE_COMMON_CAP_AUTH_SASL)) { if (!(link->tiTicketing.rsa = RSA_new())) { spice_warning("RSA new failed"); return FALSE; } if (!(bio = BIO_new(BIO_s_mem()))) { spice_warning("BIO new failed"); return FALSE; } if (RSA_generate_key_ex(link->tiTicketing.rsa, SPICE_TICKET_KEY_PAIR_LENGTH, link->tiTicketing.bn, NULL) != 1) { spice_warning("Failed to generate %d bits RSA key: %s", SPICE_TICKET_KEY_PAIR_LENGTH, ERR_error_string(ERR_get_error(), NULL)); goto end; } link->tiTicketing.rsa_size = RSA_size(link->tiTicketing.rsa); i2d_RSA_PUBKEY_bio(bio, link->tiTicketing.rsa); BIO_get_mem_ptr(bio, &bmBuf); memcpy(ack.pub_key, bmBuf->data, sizeof(ack.pub_key)); } else { /* if the client sets the AUTH_SASL cap, it indicates that it * supports SASL, and will use it if the server supports SASL as * well. Moreover, a client setting the AUTH_SASL cap also * indicates that it will not try using the RSA-related content * in the SpiceLinkReply message, so we don't need to initialize * it. Reason to avoid this is to fix auth in fips mode where * the generation of a 1024 bit RSA key as we are trying to do * will fail. */ spice_warning("not initialising RSA key"); memset(ack.pub_key, '\0', sizeof(ack.pub_key)); } if (!reds_stream_write_all(link->stream, &header, sizeof(header))) goto end; if (!reds_stream_write_all(link->stream, &ack, sizeof(ack))) goto end; for (unsigned int i = 0; i < channel_caps->num_common_caps; i++) { guint32 cap = GUINT32_TO_LE(channel_caps->common_caps[i]); if (!reds_stream_write_all(link->stream, &cap, sizeof(cap))) goto end; } for (unsigned int i = 0; i < channel_caps->num_caps; i++) { guint32 cap = GUINT32_TO_LE(channel_caps->caps[i]); if (!reds_stream_write_all(link->stream, &cap, sizeof(cap))) goto end; } ret = TRUE; end: if (bio != NULL) BIO_free(bio); return ret; } static bool reds_send_link_error(RedLinkInfo *link, uint32_t error) { SpiceLinkHeader header; SpiceLinkReply reply; header.magic = SPICE_MAGIC; header.size = GUINT32_TO_LE(sizeof(reply)); header.major_version = GUINT32_TO_LE(SPICE_VERSION_MAJOR); header.minor_version = GUINT32_TO_LE(SPICE_VERSION_MINOR); memset(&reply, 0, sizeof(reply)); reply.error = GUINT32_TO_LE(error); return reds_stream_write_all(link->stream, &header, sizeof(header)) && reds_stream_write_all(link->stream, &reply, sizeof(reply)); } static void reds_info_new_channel(RedLinkInfo *link, int connection_id) { spice_info("channel %d:%d, connected successfully, over %s link", link->link_mess->channel_type, link->link_mess->channel_id, reds_stream_is_ssl(link->stream) ? "Secure" : "Non Secure"); /* add info + send event */ if (reds_stream_is_ssl(link->stream)) { reds_stream_set_info_flag(link->stream, SPICE_CHANNEL_EVENT_FLAG_TLS); } reds_stream_set_channel(link->stream, connection_id, link->link_mess->channel_type, link->link_mess->channel_id); reds_stream_push_channel_event(link->stream, SPICE_CHANNEL_EVENT_INITIALIZED); } static void reds_send_link_result(RedLinkInfo *link, uint32_t error) { error = GUINT32_TO_LE(error); reds_stream_write_all(link->stream, &error, sizeof(error)); } int reds_expects_link_id(uint32_t connection_id) { spice_info("TODO: keep a list of connection_id's from migration, compare to them"); return 1; } static void reds_mig_target_client_add(RedsState *reds, RedClient *client) { RedsMigTargetClient *mig_client; spice_assert(reds); spice_info(NULL); mig_client = spice_malloc0(sizeof(RedsMigTargetClient)); mig_client->client = client; mig_client->reds = reds; ring_init(&mig_client->pending_links); ring_add(&reds->mig_target_clients, &mig_client->link); reds->num_mig_target_clients++; } static RedsMigTargetClient* reds_mig_target_client_find(RedsState *reds, RedClient *client) { RingItem *item; RING_FOREACH(item, &reds->mig_target_clients) { RedsMigTargetClient *mig_client; mig_client = SPICE_CONTAINEROF(item, RedsMigTargetClient, link); if (mig_client->client == client) { return mig_client; } } return NULL; } static void reds_mig_target_client_add_pending_link(RedsMigTargetClient *client, SpiceLinkMess *link_msg, RedsStream *stream) { RedsMigPendingLink *mig_link; spice_assert(client); mig_link = spice_malloc0(sizeof(RedsMigPendingLink)); mig_link->link_msg = link_msg; mig_link->stream = stream; ring_add(&client->pending_links, &mig_link->ring_link); } static void reds_mig_target_client_free(RedsMigTargetClient *mig_client) { RingItem *now, *next; ring_remove(&mig_client->link); mig_client->reds->num_mig_target_clients--; RING_FOREACH_SAFE(now, next, &mig_client->pending_links) { RedsMigPendingLink *mig_link = SPICE_CONTAINEROF(now, RedsMigPendingLink, ring_link); ring_remove(now); free(mig_link); } free(mig_client); } static void reds_mig_target_client_disconnect_all(RedsState *reds) { RingItem *now, *next; RING_FOREACH_SAFE(now, next, &reds->mig_target_clients) { RedsMigTargetClient *mig_client = SPICE_CONTAINEROF(now, RedsMigTargetClient, link); reds_client_disconnect(reds, mig_client->client); } } static int reds_find_client(RedsState *reds, RedClient *client) { RingItem *item; RING_FOREACH(item, &reds->clients) { RedClient *list_client; list_client = SPICE_CONTAINEROF(item, RedClient, link); if (list_client == client) { return TRUE; } } return FALSE; } /* should be used only when there is one client */ static RedClient *reds_get_client(RedsState *reds) { spice_assert(reds->num_clients <= 1); if (reds->num_clients == 0) { return NULL; } return SPICE_CONTAINEROF(ring_get_head(&reds->clients), RedClient, link); } // TODO: now that main is a separate channel this should // actually be joined with reds_handle_other_links, become reds_handle_link static void reds_handle_main_link(RedsState *reds, RedLinkInfo *link) { RedClient *client; RedsStream *stream; SpiceLinkMess *link_mess; uint32_t *caps; uint32_t connection_id; MainChannelClient *mcc; int mig_target = FALSE; spice_info(NULL); spice_assert(reds->main_channel); link_mess = link->link_mess; if (!reds->allow_multiple_clients) { reds_disconnect(reds); } if (link_mess->connection_id == 0) { reds_send_link_result(link, SPICE_LINK_ERR_OK); while((connection_id = rand()) == 0); mig_target = FALSE; } else { // TODO: make sure link_mess->connection_id is the same // connection id the migration src had (use vmstate to store the connection id) reds_send_link_result(link, SPICE_LINK_ERR_OK); connection_id = link_mess->connection_id; mig_target = TRUE; } reds->mig_inprogress = FALSE; reds->mig_wait_connect = FALSE; reds->mig_wait_disconnect = FALSE; reds_info_new_channel(link, connection_id); stream = link->stream; reds_stream_remove_watch(stream); link->stream = NULL; link->link_mess = NULL; reds_link_free(link); caps = (uint32_t *)((uint8_t *)link_mess + link_mess->caps_offset); client = red_client_new(reds, mig_target); ring_add(&reds->clients, &client->link); reds->num_clients++; mcc = main_channel_link(reds->main_channel, client, stream, connection_id, mig_target, link_mess->num_common_caps, link_mess->num_common_caps ? caps : NULL, link_mess->num_channel_caps, link_mess->num_channel_caps ? caps + link_mess->num_common_caps : NULL); spice_info("NEW Client %p mcc %p connect-id %d", client, mcc, connection_id); free(link_mess); red_client_set_main(client, mcc); if (reds->vdagent) { if (mig_target) { spice_warning("unexpected: vdagent attached to destination during migration"); } agent_msg_filter_config(&reds->agent_dev->priv->read_filter, reds->agent_copypaste, reds->agent_file_xfer, reds_use_client_monitors_config(reds)); reds->agent_dev->priv->read_filter.discard_all = FALSE; reds->agent_dev->priv->plug_generation++; } if (!mig_target) { main_channel_push_init(mcc, g_list_length(reds->qxl_instances), reds->mouse_mode, reds->is_client_mouse_allowed, reds_get_mm_time() - MM_TIME_DELTA, reds_qxl_ram_size(reds)); if (reds->spice_name) main_channel_push_name(mcc, reds->spice_name); if (reds->spice_uuid_is_set) main_channel_push_uuid(mcc, reds->spice_uuid); } else { reds_mig_target_client_add(reds, client); } if (reds_stream_get_family(stream) != AF_UNIX) main_channel_client_start_net_test(mcc, !mig_target); } #define RED_MOUSE_STATE_TO_LOCAL(state) \ ((state & SPICE_MOUSE_BUTTON_MASK_LEFT) | \ ((state & SPICE_MOUSE_BUTTON_MASK_MIDDLE) << 1) | \ ((state & SPICE_MOUSE_BUTTON_MASK_RIGHT) >> 1)) #define RED_MOUSE_BUTTON_STATE_TO_AGENT(state) \ (((state & SPICE_MOUSE_BUTTON_MASK_LEFT) ? VD_AGENT_LBUTTON_MASK : 0) | \ ((state & SPICE_MOUSE_BUTTON_MASK_MIDDLE) ? VD_AGENT_MBUTTON_MASK : 0) | \ ((state & SPICE_MOUSE_BUTTON_MASK_RIGHT) ? VD_AGENT_RBUTTON_MASK : 0)) void reds_set_client_mouse_allowed(RedsState *reds, int is_client_mouse_allowed, int x_res, int y_res) { reds->monitor_mode.x_res = x_res; reds->monitor_mode.y_res = y_res; reds->dispatcher_allows_client_mouse = is_client_mouse_allowed; reds_update_mouse_mode(reds); if (reds->is_client_mouse_allowed && inputs_channel_has_tablet(reds->inputs_channel)) { inputs_channel_set_tablet_logical_size(reds->inputs_channel, reds->monitor_mode.x_res, reds->monitor_mode.y_res); } } static void openssl_init(RedLinkInfo *link) { unsigned long f4 = RSA_F4; link->tiTicketing.bn = BN_new(); if (!link->tiTicketing.bn) { spice_error("OpenSSL BIGNUMS alloc failed"); } BN_set_word(link->tiTicketing.bn, f4); } static void reds_channel_do_link(RedChannel *channel, RedClient *client, SpiceLinkMess *link_msg, RedsStream *stream) { uint32_t *caps; spice_assert(channel); spice_assert(link_msg); spice_assert(stream); caps = (uint32_t *)((uint8_t *)link_msg + link_msg->caps_offset); channel->client_cbs.connect(channel, client, stream, red_client_during_migrate_at_target(client), link_msg->num_common_caps, link_msg->num_common_caps ? caps : NULL, link_msg->num_channel_caps, link_msg->num_channel_caps ? caps + link_msg->num_common_caps : NULL); } /* * migration target side: * In semi-seamless migration, we activate the channels only * after migration is completed. * In seamless migration, in order to keep the continuousness, and * not lose any data, we activate the target channels before * migration completes, as soon as we receive SPICE_MSGC_MAIN_MIGRATE_DST_DO_SEAMLESS */ static int reds_link_mig_target_channels(RedsState *reds, RedClient *client) { RedsMigTargetClient *mig_client; RingItem *item; spice_info("%p", client); mig_client = reds_mig_target_client_find(reds, client); if (!mig_client) { spice_info("Error: mig target client was not found"); return FALSE; } /* Each channel should check if we are during migration, and * act accordingly. */ RING_FOREACH(item, &mig_client->pending_links) { RedsMigPendingLink *mig_link; RedChannel *channel; mig_link = SPICE_CONTAINEROF(item, RedsMigPendingLink, ring_link); channel = reds_find_channel(reds, mig_link->link_msg->channel_type, mig_link->link_msg->channel_id); if (!channel) { spice_warning("client %p channel (%d, %d) (type, id) wasn't found", client, mig_link->link_msg->channel_type, mig_link->link_msg->channel_id); continue; } reds_channel_do_link(channel, client, mig_link->link_msg, mig_link->stream); } reds_mig_target_client_free(mig_client); return TRUE; } int reds_on_migrate_dst_set_seamless(RedsState *reds, MainChannelClient *mcc, uint32_t src_version) { /* seamless migration is not supported with multiple clients*/ if (reds->allow_multiple_clients || src_version > SPICE_MIGRATION_PROTOCOL_VERSION) { reds->dst_do_seamless_migrate = FALSE; } else { RedChannelClient *rcc = main_channel_client_get_base(mcc); red_client_set_migration_seamless(rcc->client); /* linking all the channels that have been connected before migration handshake */ reds->dst_do_seamless_migrate = reds_link_mig_target_channels(reds, rcc->client); } return reds->dst_do_seamless_migrate; } void reds_on_client_seamless_migrate_complete(RedsState *reds, RedClient *client) { spice_debug(NULL); if (!reds_find_client(reds, client)) { spice_info("client no longer exists"); return; } main_channel_migrate_dst_complete(red_client_get_main(client)); } void reds_on_client_semi_seamless_migrate_complete(RedsState *reds, RedClient *client) { MainChannelClient *mcc; spice_info("%p", client); mcc = red_client_get_main(client); // TODO: not doing net test. consider doing it on client_migrate_info main_channel_push_init(mcc, g_list_length(reds->qxl_instances), reds->mouse_mode, reds->is_client_mouse_allowed, reds_get_mm_time() - MM_TIME_DELTA, reds_qxl_ram_size(reds)); reds_link_mig_target_channels(reds, client); main_channel_migrate_dst_complete(mcc); } static void reds_handle_other_links(RedsState *reds, RedLinkInfo *link) { RedChannel *channel; RedClient *client = NULL; SpiceLinkMess *link_mess; RedsMigTargetClient *mig_client; link_mess = link->link_mess; if (reds->main_channel) { client = main_channel_get_client_by_link_id(reds->main_channel, link_mess->connection_id); } // TODO: MC: broke migration (at least for the dont-drop-connection kind). // On migration we should get a connection_id to expect (must be a security measure) // where do we store it? on reds, but should be a list (MC). if (!client) { reds_send_link_result(link, SPICE_LINK_ERR_BAD_CONNECTION_ID); reds_link_free(link); return; } // TODO: MC: be less lenient. Tally connections from same connection_id (by same client). if (!(channel = reds_find_channel(reds, link_mess->channel_type, link_mess->channel_id))) { reds_send_link_result(link, SPICE_LINK_ERR_CHANNEL_NOT_AVAILABLE); reds_link_free(link); return; } reds_send_link_result(link, SPICE_LINK_ERR_OK); reds_info_new_channel(link, link_mess->connection_id); reds_stream_remove_watch(link->stream); mig_client = reds_mig_target_client_find(reds, client); /* * In semi-seamless migration, we activate the channels only * after migration is completed. Since, the session starts almost from * scratch we don't mind if we skip some messages in between the src session end and * dst session start. * In seamless migration, in order to keep the continuousness of the session, and * in order not to lose any data, we activate the target channels before * migration completes, as soon as we receive SPICE_MSGC_MAIN_MIGRATE_DST_DO_SEAMLESS. * If a channel connects before receiving SPICE_MSGC_MAIN_MIGRATE_DST_DO_SEAMLESS, * reds_on_migrate_dst_set_seamless will take care of activating it */ if (red_client_during_migrate_at_target(client) && !reds->dst_do_seamless_migrate) { spice_assert(mig_client); reds_mig_target_client_add_pending_link(mig_client, link_mess, link->stream); } else { spice_assert(!mig_client); reds_channel_do_link(channel, client, link_mess, link->stream); free(link_mess); } link->stream = NULL; link->link_mess = NULL; reds_link_free(link); } static void reds_handle_link(RedsState *reds, RedLinkInfo *link) { if (link->link_mess->channel_type == SPICE_CHANNEL_MAIN) { reds_handle_main_link(reds, link); } else { reds_handle_other_links(reds, link); } } static void reds_handle_ticket(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; RedsState *reds = link->reds; char *password; time_t ltime; int password_size; //todo: use monotonic time time(<ime); if (RSA_size(link->tiTicketing.rsa) < SPICE_MAX_PASSWORD_LENGTH) { spice_warning("RSA modulus size is smaller than SPICE_MAX_PASSWORD_LENGTH (%d < %d), " "SPICE ticket sent from client may be truncated", RSA_size(link->tiTicketing.rsa), SPICE_MAX_PASSWORD_LENGTH); } password = spice_malloc0(RSA_size(link->tiTicketing.rsa) + 1); password_size = RSA_private_decrypt(link->tiTicketing.rsa_size, link->tiTicketing.encrypted_ticket.encrypted_data, (unsigned char *)password, link->tiTicketing.rsa, RSA_PKCS1_OAEP_PADDING); if (password_size == -1) { spice_warning("failed to decrypt RSA encrypted password: %s", ERR_error_string(ERR_get_error(), NULL)); goto error; } password[password_size] = '\0'; if (reds->ticketing_enabled && !link->skip_auth) { int expired = reds->taTicket.expiration_time < ltime; if (strlen(reds->taTicket.password) == 0) { spice_warning("Ticketing is enabled, but no password is set. " "please set a ticket first"); goto error; } if (expired || strcmp(password, reds->taTicket.password) != 0) { if (expired) { spice_warning("Ticket has expired"); } else { spice_warning("Invalid password"); } goto error; } } reds_handle_link(reds, link); goto end; error: reds_send_link_result(link, SPICE_LINK_ERR_PERMISSION_DENIED); reds_link_free(link); end: free(password); } static void reds_get_spice_ticket(RedLinkInfo *link) { reds_stream_async_read(link->stream, (uint8_t *)&link->tiTicketing.encrypted_ticket.encrypted_data, link->tiTicketing.rsa_size, reds_handle_ticket, link); } #if HAVE_SASL /* * Step Msg * * Input from client: * * u32 clientin-length * u8-array clientin-string * * Output to client: * * u32 serverout-length * u8-array serverout-strin * u8 continue */ #define SASL_DATA_MAX_LEN (1024 * 1024) static void reds_handle_auth_sasl_steplen(void *opaque); static void reds_handle_auth_sasl_step(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; RedsState *reds = link->reds; RedsSaslError status; status = reds_sasl_handle_auth_step(link->stream, reds_handle_auth_sasl_steplen, link); if (status == REDS_SASL_ERROR_OK) { reds_handle_link(reds, link); } else if (status != REDS_SASL_ERROR_CONTINUE) { reds_link_free(link); } } static void reds_handle_auth_sasl_steplen(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; RedsSaslError status; status = reds_sasl_handle_auth_steplen(link->stream, reds_handle_auth_sasl_step, link); if (status != REDS_SASL_ERROR_OK) { reds_link_free(link); } } /* * Start Msg * * Input from client: * * u32 clientin-length * u8-array clientin-string * * Output to client: * * u32 serverout-length * u8-array serverout-strin * u8 continue */ static void reds_handle_auth_sasl_start(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; RedsState *reds = link->reds; RedsSaslError status; status = reds_sasl_handle_auth_start(link->stream, reds_handle_auth_sasl_steplen, link); if (status == REDS_SASL_ERROR_OK) { reds_handle_link(reds, link); } else if (status != REDS_SASL_ERROR_CONTINUE) { reds_link_free(link); } } static void reds_handle_auth_startlen(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; RedsSaslError status; status = reds_sasl_handle_auth_startlen(link->stream, reds_handle_auth_sasl_start, link); switch (status) { case REDS_SASL_ERROR_OK: break; case REDS_SASL_ERROR_RETRY: reds_handle_auth_sasl_start(opaque); break; case REDS_SASL_ERROR_GENERIC: case REDS_SASL_ERROR_INVALID_DATA: reds_send_link_error(link, SPICE_LINK_ERR_INVALID_DATA); reds_link_free(link); break; default: g_warn_if_reached(); reds_send_link_error(link, SPICE_LINK_ERR_INVALID_DATA); reds_link_free(link); break; } } static void reds_handle_auth_mechname(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; if (!reds_sasl_handle_auth_mechname(link->stream, reds_handle_auth_startlen, link)) { reds_send_link_error(link, SPICE_LINK_ERR_INVALID_DATA); } } static void reds_handle_auth_mechlen(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; if (!reds_sasl_handle_auth_mechlen(link->stream, reds_handle_auth_mechname, link)) { reds_link_free(link); } } static void reds_start_auth_sasl(RedLinkInfo *link) { if (!reds_sasl_start_auth(link->stream, reds_handle_auth_mechlen, link)) { reds_link_free(link); } } #endif static void reds_handle_auth_mechanism(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; RedsState *reds = link->reds; spice_info("Auth method: %d", link->auth_mechanism.auth_mechanism); link->auth_mechanism.auth_mechanism = GUINT32_FROM_LE(link->auth_mechanism.auth_mechanism); if (link->auth_mechanism.auth_mechanism == SPICE_COMMON_CAP_AUTH_SPICE && !reds->sasl_enabled ) { reds_get_spice_ticket(link); #if HAVE_SASL } else if (link->auth_mechanism.auth_mechanism == SPICE_COMMON_CAP_AUTH_SASL) { spice_info("Starting SASL"); reds_start_auth_sasl(link); #endif } else { spice_warning("Unknown auth method, disconnecting"); if (reds->sasl_enabled) { spice_warning("Your client doesn't handle SASL?"); } reds_send_link_error(link, SPICE_LINK_ERR_INVALID_DATA); reds_link_free(link); } } static int reds_security_check(RedLinkInfo *link) { RedsState *reds = link->reds; ChannelSecurityOptions *security_option = reds_find_channel_security(reds, link->link_mess->channel_type); uint32_t security = security_option ? security_option->options : reds->default_channel_security; return (reds_stream_is_ssl(link->stream) && (security & SPICE_CHANNEL_SECURITY_SSL)) || (!reds_stream_is_ssl(link->stream) && (security & SPICE_CHANNEL_SECURITY_NONE)); } static void reds_handle_read_link_done(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; RedsState *reds = link->reds; SpiceLinkMess *link_mess = link->link_mess; uint32_t num_caps; uint32_t *caps; int auth_selection; unsigned int i; link_mess->caps_offset = GUINT32_FROM_LE(link_mess->caps_offset); link_mess->connection_id = GUINT32_FROM_LE(link_mess->connection_id); link_mess->num_channel_caps = GUINT32_FROM_LE(link_mess->num_channel_caps); link_mess->num_common_caps = GUINT32_FROM_LE(link_mess->num_common_caps); num_caps = link_mess->num_common_caps + link_mess->num_channel_caps; caps = (uint32_t *)((uint8_t *)link_mess + link_mess->caps_offset); if (num_caps && (num_caps * sizeof(uint32_t) + link_mess->caps_offset > link->link_header.size || link_mess->caps_offset < sizeof(*link_mess))) { reds_send_link_error(link, SPICE_LINK_ERR_INVALID_DATA); reds_link_free(link); return; } for(i = 0; i < num_caps;i++) caps[i] = GUINT32_FROM_LE(caps[i]); auth_selection = red_link_info_test_capability(link, SPICE_COMMON_CAP_PROTOCOL_AUTH_SELECTION); if (!reds_security_check(link)) { if (reds_stream_is_ssl(link->stream)) { spice_warning("spice channels %d should not be encrypted", link_mess->channel_type); reds_send_link_error(link, SPICE_LINK_ERR_NEED_UNSECURED); } else { spice_warning("spice channels %d should be encrypted", link_mess->channel_type); reds_send_link_error(link, SPICE_LINK_ERR_NEED_SECURED); } reds_link_free(link); return; } if (!reds_send_link_ack(reds, link)) { reds_link_free(link); return; } if (!auth_selection) { if (reds->sasl_enabled && !link->skip_auth) { spice_warning("SASL enabled, but peer supports only spice authentication"); reds_send_link_error(link, SPICE_LINK_ERR_VERSION_MISMATCH); return; } spice_warning("Peer doesn't support AUTH selection"); reds_get_spice_ticket(link); } else { reds_stream_async_read(link->stream, (uint8_t *)&link->auth_mechanism, sizeof(SpiceLinkAuthMechanism), reds_handle_auth_mechanism, link); } } static void reds_handle_link_error(void *opaque, int err) { RedLinkInfo *link = (RedLinkInfo *)opaque; switch (err) { case 0: case EPIPE: break; default: spice_warning("%s", strerror(errno)); break; } reds_link_free(link); } static void reds_handle_read_header_done(void *opaque) { RedLinkInfo *link = (RedLinkInfo *)opaque; RedsState *reds = link->reds; SpiceLinkHeader *header = &link->link_header; header->major_version = GUINT32_FROM_LE(header->major_version); header->minor_version = GUINT32_FROM_LE(header->minor_version); header->size = GUINT32_FROM_LE(header->size); if (header->magic != SPICE_MAGIC) { reds_send_link_error(link, SPICE_LINK_ERR_INVALID_MAGIC); reds_link_free(link); return; } if (header->major_version != SPICE_VERSION_MAJOR) { if (header->major_version > 0) { reds_send_link_error(link, SPICE_LINK_ERR_VERSION_MISMATCH); } spice_warning("version mismatch"); reds_link_free(link); return; } reds->peer_minor_version = header->minor_version; if (header->size < sizeof(SpiceLinkMess)) { reds_send_link_error(link, SPICE_LINK_ERR_INVALID_DATA); spice_warning("bad size %u", header->size); reds_link_free(link); return; } link->link_mess = spice_malloc(header->size); reds_stream_async_read(link->stream, (uint8_t *)link->link_mess, header->size, reds_handle_read_link_done, link); } static void reds_handle_new_link(RedLinkInfo *link) { reds_stream_set_async_error_handler(link->stream, reds_handle_link_error); reds_stream_async_read(link->stream, (uint8_t *)&link->link_header, sizeof(SpiceLinkHeader), reds_handle_read_header_done, link); } static void reds_handle_ssl_accept(int fd, int event, void *data) { RedLinkInfo *link = (RedLinkInfo *)data; RedsState *reds = link->reds; int return_code = reds_stream_ssl_accept(link->stream); switch (return_code) { case REDS_STREAM_SSL_STATUS_ERROR: reds_link_free(link); return; case REDS_STREAM_SSL_STATUS_WAIT_FOR_READ: reds_core_watch_update_mask(reds, link->stream->watch, SPICE_WATCH_EVENT_READ); return; case REDS_STREAM_SSL_STATUS_WAIT_FOR_WRITE: reds_core_watch_update_mask(reds, link->stream->watch, SPICE_WATCH_EVENT_WRITE); return; case REDS_STREAM_SSL_STATUS_OK: reds_stream_remove_watch(link->stream); reds_handle_new_link(link); } } #define KEEPALIVE_TIMEOUT (10*60) static bool reds_init_keepalive(int socket) { int keepalive = 1; int keepalive_timeout = KEEPALIVE_TIMEOUT; if (setsockopt(socket, SOL_SOCKET, SO_KEEPALIVE, &keepalive, sizeof(keepalive)) == -1) { if (errno != ENOTSUP) { spice_printerr("setsockopt for keepalive failed, %s", strerror(errno)); return false; } } if (setsockopt(socket, SOL_TCP, TCP_KEEPIDLE, &keepalive_timeout, sizeof(keepalive_timeout)) == -1) { if (errno != ENOTSUP) { spice_printerr("setsockopt for keepalive timeout failed, %s", strerror(errno)); return false; } } return true; } static RedLinkInfo *reds_init_client_connection(RedsState *reds, int socket) { RedLinkInfo *link; int delay_val = 1; int flags; if ((flags = fcntl(socket, F_GETFL)) == -1) { spice_warning("accept failed, %s", strerror(errno)); goto error; } if (fcntl(socket, F_SETFL, flags | O_NONBLOCK) == -1) { spice_warning("accept failed, %s", strerror(errno)); goto error; } if (setsockopt(socket, IPPROTO_TCP, TCP_NODELAY, &delay_val, sizeof(delay_val)) == -1) { if (errno != ENOTSUP) { spice_warning("setsockopt failed, %s", strerror(errno)); } } reds_init_keepalive(socket); link = spice_new0(RedLinkInfo, 1); link->reds = reds; link->stream = reds_stream_new(reds, socket); /* gather info + send event */ reds_stream_push_channel_event(link->stream, SPICE_CHANNEL_EVENT_CONNECTED); openssl_init(link); return link; error: return NULL; } static RedLinkInfo *reds_init_client_ssl_connection(RedsState *reds, int socket) { RedLinkInfo *link; int ssl_status; link = reds_init_client_connection(reds, socket); if (link == NULL) goto error; ssl_status = reds_stream_enable_ssl(link->stream, reds->ctx); switch (ssl_status) { case REDS_STREAM_SSL_STATUS_OK: reds_handle_new_link(link); return link; case REDS_STREAM_SSL_STATUS_ERROR: goto error; case REDS_STREAM_SSL_STATUS_WAIT_FOR_READ: link->stream->watch = reds_core_watch_add(reds, link->stream->socket, SPICE_WATCH_EVENT_READ, reds_handle_ssl_accept, link); break; case REDS_STREAM_SSL_STATUS_WAIT_FOR_WRITE: link->stream->watch = reds_core_watch_add(reds, link->stream->socket, SPICE_WATCH_EVENT_WRITE, reds_handle_ssl_accept, link); break; } return link; error: free(link->stream); BN_free(link->tiTicketing.bn); free(link); return NULL; } static void reds_accept_ssl_connection(int fd, int event, void *data) { RedsState *reds = data; RedLinkInfo *link; int socket; if ((socket = accept(reds->secure_listen_socket, NULL, 0)) == -1) { spice_warning("accept failed, %s", strerror(errno)); return; } if (!(link = reds_init_client_ssl_connection(reds, socket))) { close(socket); return; } } static void reds_accept(int fd, int event, void *data) { RedsState *reds = data; int socket; if ((socket = accept(reds->listen_socket, NULL, 0)) == -1) { spice_warning("accept failed, %s", strerror(errno)); return; } if (spice_server_add_client(reds, socket, 0) < 0) close(socket); } SPICE_GNUC_VISIBLE int spice_server_add_client(SpiceServer *reds, int socket, int skip_auth) { RedLinkInfo *link; if (!(link = reds_init_client_connection(reds, socket))) { spice_warning("accept failed"); return -1; } link->skip_auth = skip_auth; reds_handle_new_link(link); return 0; } SPICE_GNUC_VISIBLE int spice_server_add_ssl_client(SpiceServer *reds, int socket, int skip_auth) { RedLinkInfo *link; if (!(link = reds_init_client_ssl_connection(reds, socket))) { return -1; } link->skip_auth = skip_auth; return 0; } static int reds_init_socket(const char *addr, int portnr, int family) { static const int on=1, off=0; struct addrinfo ai,*res,*e; char port[33]; int slisten, rc, len; if (family == AF_UNIX) { struct sockaddr_un local = { 0, }; if ((slisten = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) { perror("socket"); return -1; } local.sun_family = AF_UNIX; strncpy(local.sun_path, addr, sizeof(local.sun_path) -1); unlink(local.sun_path); len = SUN_LEN(&local); if (bind(slisten, (struct sockaddr *)&local, len) == -1) { perror("bind"); return -1; } goto listen; } memset(&ai,0, sizeof(ai)); ai.ai_flags = AI_PASSIVE | AI_ADDRCONFIG; ai.ai_socktype = SOCK_STREAM; ai.ai_family = family; snprintf(port, sizeof(port), "%d", portnr); rc = getaddrinfo(strlen(addr) ? addr : NULL, port, &ai, &res); if (rc != 0) { spice_warning("getaddrinfo(%s,%s): %s", addr, port, gai_strerror(rc)); return -1; } for (e = res; e != NULL; e = e->ai_next) { slisten = socket(e->ai_family, e->ai_socktype, e->ai_protocol); if (slisten < 0) { continue; } setsockopt(slisten,SOL_SOCKET,SO_REUSEADDR,(void*)&on,sizeof(on)); #ifdef IPV6_V6ONLY if (e->ai_family == PF_INET6) { /* listen on both ipv4 and ipv6 */ setsockopt(slisten,IPPROTO_IPV6,IPV6_V6ONLY,(void*)&off, sizeof(off)); } #endif if (bind(slisten, e->ai_addr, e->ai_addrlen) == 0) { char uaddr[INET6_ADDRSTRLEN+1]; char uport[33]; rc = getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen, uaddr,INET6_ADDRSTRLEN, uport,32, NI_NUMERICHOST | NI_NUMERICSERV); if (rc == 0) { spice_info("bound to %s:%s", uaddr, uport); } else { spice_info("cannot resolve address spice-server is bound to"); } freeaddrinfo(res); goto listen; } close(slisten); } spice_warning("%s: binding socket to %s:%d failed", __FUNCTION__, addr, portnr); freeaddrinfo(res); return -1; listen: if (listen(slisten, SOMAXCONN) != 0) { spice_warning("listen: %s", strerror(errno)); close(slisten); return -1; } return slisten; } static void reds_send_mm_time(RedsState *reds) { if (!reds_main_channel_connected(reds)) { return; } spice_debug(NULL); main_channel_push_multi_media_time(reds->main_channel, reds_get_mm_time() - reds->mm_time_latency); } void reds_set_client_mm_time_latency(RedsState *reds, RedClient *client, uint32_t latency) { // TODO: multi-client support for mm_time if (reds->mm_time_enabled) { // TODO: consider network latency if (latency > reds->mm_time_latency) { reds->mm_time_latency = latency; reds_send_mm_time(reds); } else { spice_debug("new latency %u is smaller than existing %u", latency, reds->mm_time_latency); } } else { snd_set_playback_latency(client, latency); } } static int reds_init_net(RedsState *reds) { if (reds->spice_port != -1 || reds->spice_family == AF_UNIX) { reds->listen_socket = reds_init_socket(reds->spice_addr, reds->spice_port, reds->spice_family); if (-1 == reds->listen_socket) { return -1; } reds->listen_watch = reds_core_watch_add(reds, reds->listen_socket, SPICE_WATCH_EVENT_READ, reds_accept, reds); if (reds->listen_watch == NULL) { spice_warning("set fd handle failed"); return -1; } } if (reds->spice_secure_port != -1) { reds->secure_listen_socket = reds_init_socket(reds->spice_addr, reds->spice_secure_port, reds->spice_family); if (-1 == reds->secure_listen_socket) { return -1; } reds->secure_listen_watch = reds_core_watch_add(reds, reds->secure_listen_socket, SPICE_WATCH_EVENT_READ, reds_accept_ssl_connection, reds); if (reds->secure_listen_watch == NULL) { spice_warning("set fd handle failed"); return -1; } } if (reds->spice_listen_socket_fd != -1 ) { reds->listen_socket = reds->spice_listen_socket_fd; reds->listen_watch = reds_core_watch_add(reds, reds->listen_socket, SPICE_WATCH_EVENT_READ, reds_accept, reds); if (reds->listen_watch == NULL) { spice_warning("set fd handle failed"); return -1; } } return 0; } static int load_dh_params(SSL_CTX *ctx, char *file) { DH *ret = 0; BIO *bio; if ((bio = BIO_new_file(file, "r")) == NULL) { spice_warning("Could not open DH file"); return -1; } ret = PEM_read_bio_DHparams(bio, NULL, NULL, NULL); BIO_free(bio); if (ret == 0) { spice_warning("Could not read DH params"); return -1; } if (SSL_CTX_set_tmp_dh(ctx, ret) < 0) { spice_warning("Could not set DH params"); return -1; } return 0; } /*The password code is not thread safe*/ static int ssl_password_cb(char *buf, int size, int flags, void *userdata) { RedsState *reds = userdata; char *pass = reds->ssl_parameters.keyfile_password; if (size < strlen(pass) + 1) { return (0); } strcpy(buf, pass); return (strlen(pass)); } static unsigned long pthreads_thread_id(void) { unsigned long ret; ret = (unsigned long)pthread_self(); return (ret); } static void pthreads_locking_callback(int mode, int type, const char *file, int line) { if (mode & CRYPTO_LOCK) { pthread_mutex_lock(&(lock_cs[type])); lock_count[type]++; } else { pthread_mutex_unlock(&(lock_cs[type])); } } static void openssl_thread_setup(void) { int i; lock_cs = OPENSSL_malloc(CRYPTO_num_locks() * sizeof(pthread_mutex_t)); lock_count = OPENSSL_malloc(CRYPTO_num_locks() * sizeof(long)); for (i = 0; i < CRYPTO_num_locks(); i++) { lock_count[i] = 0; pthread_mutex_init(&(lock_cs[i]), NULL); } CRYPTO_set_id_callback(pthreads_thread_id); CRYPTO_set_locking_callback(pthreads_locking_callback); } static int reds_init_ssl(RedsState *reds) { #if OPENSSL_VERSION_NUMBER >= 0x10000000L const SSL_METHOD *ssl_method; #else SSL_METHOD *ssl_method; #endif int return_code; /* When some other SSL/TLS version becomes obsolete, add it to this * variable. */ long ssl_options = SSL_OP_NO_SSLv2 | SSL_OP_NO_SSLv3; /* Global system initialization*/ SSL_library_init(); SSL_load_error_strings(); /* Create our context*/ /* SSLv23_method() handles TLSv1.x in addition to SSLv2/v3 */ ssl_method = SSLv23_method(); reds->ctx = SSL_CTX_new(ssl_method); if (!reds->ctx) { spice_warning("Could not allocate new SSL context"); return -1; } /* Limit connection to TLSv1 only */ #ifdef SSL_OP_NO_COMPRESSION ssl_options |= SSL_OP_NO_COMPRESSION; #endif SSL_CTX_set_options(reds->ctx, ssl_options); /* Load our keys and certificates*/ return_code = SSL_CTX_use_certificate_chain_file(reds->ctx, reds->ssl_parameters.certs_file); if (return_code == 1) { spice_info("Loaded certificates from %s", reds->ssl_parameters.certs_file); } else { spice_warning("Could not load certificates from %s", reds->ssl_parameters.certs_file); return -1; } SSL_CTX_set_default_passwd_cb(reds->ctx, ssl_password_cb); SSL_CTX_set_default_passwd_cb_userdata(reds->ctx, reds); return_code = SSL_CTX_use_PrivateKey_file(reds->ctx, reds->ssl_parameters.private_key_file, SSL_FILETYPE_PEM); if (return_code == 1) { spice_info("Using private key from %s", reds->ssl_parameters.private_key_file); } else { spice_warning("Could not use private key file"); return -1; } /* Load the CAs we trust*/ return_code = SSL_CTX_load_verify_locations(reds->ctx, reds->ssl_parameters.ca_certificate_file, 0); if (return_code == 1) { spice_info("Loaded CA certificates from %s", reds->ssl_parameters.ca_certificate_file); } else { spice_warning("Could not use CA file %s", reds->ssl_parameters.ca_certificate_file); return -1; } #if (OPENSSL_VERSION_NUMBER < 0x00905100L) SSL_CTX_set_verify_depth(reds->ctx, 1); #endif if (strlen(reds->ssl_parameters.dh_key_file) > 0) { if (load_dh_params(reds->ctx, reds->ssl_parameters.dh_key_file) < 0) { return -1; } } SSL_CTX_set_session_id_context(reds->ctx, (const unsigned char *)"SPICE", 5); if (strlen(reds->ssl_parameters.ciphersuite) > 0) { if (!SSL_CTX_set_cipher_list(reds->ctx, reds->ssl_parameters.ciphersuite)) { return -1; } } openssl_thread_setup(); #ifndef SSL_OP_NO_COMPRESSION STACK *cmp_stack = SSL_COMP_get_compression_methods(); sk_zero(cmp_stack); #endif return 0; } static void reds_cleanup(RedsState *reds) { #ifdef RED_STATISTICS if (reds->stat_shm_name) { shm_unlink(reds->stat_shm_name); free(reds->stat_shm_name); reds->stat_shm_name = NULL; } #endif } SPICE_DESTRUCTOR_FUNC(reds_exit) { GList *l; for (l = servers; l != NULL; l = l->next) { RedsState *reds = l->data; reds_cleanup(reds); } } static inline void on_activating_ticketing(RedsState *reds) { if (!reds->ticketing_enabled && reds_main_channel_connected(reds)) { spice_warning("disconnecting"); reds_disconnect(reds); } } static void reds_set_image_compression(RedsState *reds, SpiceImageCompression val) { if (val == reds->image_compression) { return; } reds->image_compression = val; reds_on_ic_change(reds); } static void reds_set_one_channel_security(RedsState *reds, int id, uint32_t security) { ChannelSecurityOptions *security_options; if ((security_options = reds_find_channel_security(reds, id))) { security_options->options = security; return; } security_options = spice_new(ChannelSecurityOptions, 1); security_options->channel_id = id; security_options->options = security; security_options->next = reds->channels_security; reds->channels_security = security_options; } #define REDS_SAVE_VERSION 1 static void reds_mig_release(RedsState *reds) { if (reds->mig_spice) { free(reds->mig_spice->cert_subject); free(reds->mig_spice->host); free(reds->mig_spice); reds->mig_spice = NULL; } } static void reds_mig_started(RedsState *reds) { spice_info(NULL); spice_assert(reds->mig_spice); reds->mig_inprogress = TRUE; reds->mig_wait_connect = TRUE; reds->core->timer_start(reds->mig_timer, MIGRATE_TIMEOUT); } static void reds_mig_fill_wait_disconnect(RedsState *reds) { RingItem *client_item; spice_assert(reds->num_clients > 0); /* tracking the clients, in order to ignore disconnection * of clients that got connected to the src after migration completion.*/ RING_FOREACH(client_item, &reds->clients) { RedClient *client = SPICE_CONTAINEROF(client_item, RedClient, link); reds->mig_wait_disconnect_clients = g_list_append(reds->mig_wait_disconnect_clients, client); } reds->mig_wait_disconnect = TRUE; reds->core->timer_start(reds->mig_timer, MIGRATE_TIMEOUT); } static void reds_mig_cleanup_wait_disconnect(RedsState *reds) { g_list_free(reds->mig_wait_disconnect_clients); reds->mig_wait_disconnect = FALSE; } static void reds_mig_remove_wait_disconnect_client(RedsState *reds, RedClient *client) { g_warn_if_fail(g_list_find(reds->mig_wait_disconnect_clients, client) != NULL); reds->mig_wait_disconnect_clients = g_list_remove(reds->mig_wait_disconnect_clients, client); if (reds->mig_wait_disconnect_clients == NULL) { reds_mig_cleanup(reds); } } static void reds_migrate_channels_seamless(RedsState *reds) { RedClient *client; /* seamless migration is supported for only one client for now */ client = reds_get_client(reds); red_client_migrate(client); } static void reds_mig_finished(RedsState *reds, int completed) { spice_info(NULL); reds->mig_inprogress = TRUE; if (reds->src_do_seamless_migrate && completed) { reds_migrate_channels_seamless(reds); } else { main_channel_migrate_src_complete(reds->main_channel, completed); } if (completed) { reds_mig_fill_wait_disconnect(reds); } else { reds_mig_cleanup(reds); } reds_mig_release(reds); } static void reds_mig_switch(RedsState *reds) { if (!reds->mig_spice) { spice_warning("reds_mig_switch called without migrate_info set"); return; } main_channel_migrate_switch(reds->main_channel, reds->mig_spice); reds_mig_release(reds); } static void migrate_timeout(void *opaque) { RedsState *reds = opaque; spice_info(NULL); spice_assert(reds->mig_wait_connect || reds->mig_wait_disconnect); if (reds->mig_wait_connect) { /* we will fall back to the switch host scheme when migration completes */ main_channel_migrate_cancel_wait(reds->main_channel); /* in case part of the client haven't yet completed the previous migration, disconnect them */ reds_mig_target_client_disconnect_all(reds); reds_mig_cleanup(reds); } else { reds_mig_disconnect(reds); } } uint32_t reds_get_mm_time(void) { return spice_get_monotonic_time_ms(); } void reds_enable_mm_time(RedsState *reds) { reds->mm_time_enabled = TRUE; reds->mm_time_latency = MM_TIME_DELTA; reds_send_mm_time(reds); } void reds_disable_mm_time(RedsState *reds) { reds->mm_time_enabled = FALSE; } static RedCharDevice *attach_to_red_agent(RedsState *reds, SpiceCharDeviceInstance *sin) { RedCharDeviceVDIPort *dev = reds->agent_dev; SpiceCharDeviceInterface *sif; if (dev->priv->agent_attached) { red_char_device_reset_dev_instance(RED_CHAR_DEVICE(dev), sin); } else { dev->priv->agent_attached = TRUE; g_object_set(G_OBJECT(dev), "sin", sin, NULL); } reds->vdagent = sin; reds_update_mouse_mode(reds); sif = spice_char_device_get_interface(reds->vdagent); if (sif->state) { sif->state(reds->vdagent, 1); } if (!reds_main_channel_connected(reds)) { return RED_CHAR_DEVICE(dev); } dev->priv->read_filter.discard_all = FALSE; dev->priv->plug_generation++; if (dev->priv->mig_data || red_channel_is_waiting_for_migrate_data(&reds->main_channel->base)) { /* Migration in progress (code is running on the destination host): * 1. Add the client to spice char device, if it was not already added. * 2.a If this (qemu-kvm state load side of migration) happens first * then wait for spice migration data to arrive. Otherwise * 2.b If this happens second ==> we already have spice migrate data * then restore state */ if (!red_char_device_client_exists(RED_CHAR_DEVICE(dev), reds_get_client(reds))) { int client_added; client_added = red_char_device_client_add(RED_CHAR_DEVICE(dev), reds_get_client(reds), TRUE, /* flow control */ REDS_VDI_PORT_NUM_RECEIVE_BUFFS, REDS_AGENT_WINDOW_SIZE, ~0, TRUE); if (!client_added) { spice_warning("failed to add client to agent"); reds_disconnect(reds); } } if (dev->priv->mig_data) { spice_debug("restoring dev from stored migration data"); spice_assert(dev->priv->plug_generation == 1); reds_agent_state_restore(reds, dev->priv->mig_data); free(dev->priv->mig_data); dev->priv->mig_data = NULL; } else { spice_debug("waiting for migration data"); } } else { /* we will associate the client with the char device, upon reds_on_main_agent_start, * in response to MSGC_AGENT_START */ main_channel_push_agent_connected(reds->main_channel); } return RED_CHAR_DEVICE(dev); } SPICE_GNUC_VISIBLE void spice_server_char_device_wakeup(SpiceCharDeviceInstance* sin) { if (!sin->st) { spice_warning("no RedCharDevice attached to instance %p", sin); return; } red_char_device_wakeup(sin->st); } #define SUBTYPE_VDAGENT "vdagent" #define SUBTYPE_SMARTCARD "smartcard" #define SUBTYPE_USBREDIR "usbredir" #define SUBTYPE_PORT "port" const char *spice_server_char_device_recognized_subtypes_list[] = { SUBTYPE_VDAGENT, #ifdef USE_SMARTCARD SUBTYPE_SMARTCARD, #endif SUBTYPE_USBREDIR, NULL, }; SPICE_GNUC_VISIBLE const char** spice_server_char_device_recognized_subtypes(void) { return spice_server_char_device_recognized_subtypes_list; } static void reds_add_char_device(RedsState *reds, RedCharDevice *dev) { reds->char_devices = g_list_append(reds->char_devices, dev); } static void reds_remove_char_device(RedsState *reds, RedCharDevice *dev) { g_warn_if_fail(g_list_find(reds->char_devices, dev) != NULL); reds->char_devices = g_list_remove(reds->char_devices, dev); } void reds_on_char_device_state_destroy(RedsState *reds, RedCharDevice *dev) { reds_remove_char_device(reds, dev); } static int spice_server_char_device_add_interface(SpiceServer *reds, SpiceBaseInstance *sin) { SpiceCharDeviceInstance* char_device = SPICE_CONTAINEROF(sin, SpiceCharDeviceInstance, base); RedCharDevice *dev_state = NULL; spice_info("CHAR_DEVICE %s", char_device->subtype); if (strcmp(char_device->subtype, SUBTYPE_VDAGENT) == 0) { if (reds->vdagent) { spice_warning("vdagent already attached"); return -1; } dev_state = attach_to_red_agent(reds, char_device); } #ifdef USE_SMARTCARD else if (strcmp(char_device->subtype, SUBTYPE_SMARTCARD) == 0) { if (!(dev_state = smartcard_device_connect(reds, char_device))) { return -1; } } #endif else if (strcmp(char_device->subtype, SUBTYPE_USBREDIR) == 0) { dev_state = spicevmc_device_connect(reds, char_device, SPICE_CHANNEL_USBREDIR); } else if (strcmp(char_device->subtype, SUBTYPE_PORT) == 0) { if (strcmp(char_device->portname, "org.spice-space.webdav.0") == 0) { dev_state = spicevmc_device_connect(reds, char_device, SPICE_CHANNEL_WEBDAV); } else { dev_state = spicevmc_device_connect(reds, char_device, SPICE_CHANNEL_PORT); } } if (dev_state) { spice_assert(char_device->st); /* setting the char_device state to "started" for backward compatibily with * qemu releases that don't call spice api for start/stop (not implemented yet) */ if (reds->vm_running) { red_char_device_start(char_device->st); } reds_add_char_device(reds, char_device->st); } else { spice_warning("failed to create device state for %s", char_device->subtype); return -1; } return 0; } static void spice_server_char_device_remove_interface(RedsState *reds, SpiceBaseInstance *sin) { SpiceCharDeviceInstance* char_device = SPICE_CONTAINEROF(sin, SpiceCharDeviceInstance, base); spice_info("remove CHAR_DEVICE %s", char_device->subtype); if (strcmp(char_device->subtype, SUBTYPE_VDAGENT) == 0) { if (reds->vdagent) { reds_agent_remove(reds); } } #ifdef USE_SMARTCARD else if (strcmp(char_device->subtype, SUBTYPE_SMARTCARD) == 0) { smartcard_device_disconnect(char_device); } #endif else if (strcmp(char_device->subtype, SUBTYPE_USBREDIR) == 0 || strcmp(char_device->subtype, SUBTYPE_PORT) == 0) { spicevmc_device_disconnect(reds, char_device); } else { spice_warning("failed to remove char device %s", char_device->subtype); } char_device->st = NULL; } SPICE_GNUC_VISIBLE int spice_server_add_interface(SpiceServer *reds, SpiceBaseInstance *sin) { const SpiceBaseInterface *interface = sin->sif; if (strcmp(interface->type, SPICE_INTERFACE_KEYBOARD) == 0) { spice_info("SPICE_INTERFACE_KEYBOARD"); if (interface->major_version != SPICE_INTERFACE_KEYBOARD_MAJOR || interface->minor_version > SPICE_INTERFACE_KEYBOARD_MINOR) { spice_warning("unsupported keyboard interface"); return -1; } if (inputs_channel_set_keyboard(reds->inputs_channel, SPICE_CONTAINEROF(sin, SpiceKbdInstance, base)) != 0) { return -1; } } else if (strcmp(interface->type, SPICE_INTERFACE_MOUSE) == 0) { spice_info("SPICE_INTERFACE_MOUSE"); if (interface->major_version != SPICE_INTERFACE_MOUSE_MAJOR || interface->minor_version > SPICE_INTERFACE_MOUSE_MINOR) { spice_warning("unsupported mouse interface"); return -1; } if (inputs_channel_set_mouse(reds->inputs_channel, SPICE_CONTAINEROF(sin, SpiceMouseInstance, base)) != 0) { return -1; } } else if (strcmp(interface->type, SPICE_INTERFACE_QXL) == 0) { QXLInstance *qxl; spice_info("SPICE_INTERFACE_QXL"); if (interface->major_version != SPICE_INTERFACE_QXL_MAJOR || interface->minor_version > SPICE_INTERFACE_QXL_MINOR) { spice_warning("unsupported qxl interface"); return -1; } qxl = SPICE_CONTAINEROF(sin, QXLInstance, base); red_qxl_init(reds, qxl); reds->qxl_instances = g_list_prepend(reds->qxl_instances, qxl); /* this function has to be called after the qxl is on the list * as QXLInstance clients expect the qxl to be on the list when * this callback is called. This as clients assume they can start the * qxl_instances. Also note that this should be the first callback to * be called. */ red_qxl_attach_worker(qxl); red_qxl_set_compression_level(qxl, calc_compression_level(reds)); } else if (strcmp(interface->type, SPICE_INTERFACE_TABLET) == 0) { SpiceTabletInstance *tablet = SPICE_CONTAINEROF(sin, SpiceTabletInstance, base); spice_info("SPICE_INTERFACE_TABLET"); if (interface->major_version != SPICE_INTERFACE_TABLET_MAJOR || interface->minor_version > SPICE_INTERFACE_TABLET_MINOR) { spice_warning("unsupported tablet interface"); return -1; } if (inputs_channel_set_tablet(reds->inputs_channel, tablet, reds) != 0) { return -1; } reds_update_mouse_mode(reds); if (reds->is_client_mouse_allowed) { inputs_channel_set_tablet_logical_size(reds->inputs_channel, reds->monitor_mode.x_res, reds->monitor_mode.y_res); } } else if (strcmp(interface->type, SPICE_INTERFACE_PLAYBACK) == 0) { spice_info("SPICE_INTERFACE_PLAYBACK"); if (interface->major_version != SPICE_INTERFACE_PLAYBACK_MAJOR || interface->minor_version > SPICE_INTERFACE_PLAYBACK_MINOR) { spice_warning("unsupported playback interface"); return -1; } snd_attach_playback(reds, SPICE_CONTAINEROF(sin, SpicePlaybackInstance, base)); } else if (strcmp(interface->type, SPICE_INTERFACE_RECORD) == 0) { spice_info("SPICE_INTERFACE_RECORD"); if (interface->major_version != SPICE_INTERFACE_RECORD_MAJOR || interface->minor_version > SPICE_INTERFACE_RECORD_MINOR) { spice_warning("unsupported record interface"); return -1; } snd_attach_record(reds, SPICE_CONTAINEROF(sin, SpiceRecordInstance, base)); } else if (strcmp(interface->type, SPICE_INTERFACE_CHAR_DEVICE) == 0) { if (interface->major_version != SPICE_INTERFACE_CHAR_DEVICE_MAJOR || interface->minor_version > SPICE_INTERFACE_CHAR_DEVICE_MINOR) { spice_warning("unsupported char device interface"); return -1; } spice_server_char_device_add_interface(reds, sin); } else if (strcmp(interface->type, SPICE_INTERFACE_MIGRATION) == 0) { spice_info("SPICE_INTERFACE_MIGRATION"); if (reds->migration_interface) { spice_warning("already have migration"); return -1; } if (interface->major_version != SPICE_INTERFACE_MIGRATION_MAJOR || interface->minor_version > SPICE_INTERFACE_MIGRATION_MINOR) { spice_warning("unsupported migration interface"); return -1; } reds->migration_interface = SPICE_CONTAINEROF(sin, SpiceMigrateInstance, base); reds->migration_interface->st = spice_new0(SpiceMigrateState, 1); } return 0; } SPICE_GNUC_VISIBLE int spice_server_remove_interface(SpiceBaseInstance *sin) { RedsState *reds; const SpiceBaseInterface *interface = sin->sif; if (strcmp(interface->type, SPICE_INTERFACE_TABLET) == 0) { SpiceTabletInstance *tablet = SPICE_CONTAINEROF(sin, SpiceTabletInstance, base); reds = spice_tablet_state_get_server(tablet->st); spice_info("remove SPICE_INTERFACE_TABLET"); inputs_channel_detach_tablet(reds->inputs_channel, tablet); reds_update_mouse_mode(reds); } else if (strcmp(interface->type, SPICE_INTERFACE_PLAYBACK) == 0) { spice_info("remove SPICE_INTERFACE_PLAYBACK"); snd_detach_playback(SPICE_CONTAINEROF(sin, SpicePlaybackInstance, base)); } else if (strcmp(interface->type, SPICE_INTERFACE_RECORD) == 0) { spice_info("remove SPICE_INTERFACE_RECORD"); snd_detach_record(SPICE_CONTAINEROF(sin, SpiceRecordInstance, base)); } else if (strcmp(interface->type, SPICE_INTERFACE_CHAR_DEVICE) == 0) { SpiceCharDeviceInstance *char_device = SPICE_CONTAINEROF(sin, SpiceCharDeviceInstance, base); reds = red_char_device_get_server(char_device->st); spice_server_char_device_remove_interface(reds, sin); } else { spice_warning("VD_INTERFACE_REMOVING unsupported"); return -1; } return 0; } static int do_spice_init(RedsState *reds, SpiceCoreInterface *core_interface) { spice_info("starting %s", VERSION); #if !GLIB_CHECK_VERSION(2,36,0) g_type_init(); #endif if (core_interface->base.major_version != SPICE_INTERFACE_CORE_MAJOR) { spice_warning("bad core interface version"); goto err; } core_public = core_interface; reds->core = &core_interface_adapter; reds->listen_socket = -1; reds->secure_listen_socket = -1; reds->agent_dev = red_char_device_vdi_port_new(reds); ring_init(&reds->clients); reds->num_clients = 0; reds->main_dispatcher = main_dispatcher_new(reds, reds->core); ring_init(&reds->channels); ring_init(&reds->mig_target_clients); reds->char_devices = NULL; reds->mig_wait_disconnect_clients = NULL; reds->vm_running = TRUE; /* for backward compatibility */ if (!(reds->mig_timer = reds->core->timer_add(reds->core, migrate_timeout, reds))) { spice_error("migration timer create failed"); } #ifdef RED_STATISTICS int shm_name_len; int fd; shm_name_len = strlen(SPICE_STAT_SHM_NAME) + 20; reds->stat_shm_name = (char *)spice_malloc(shm_name_len); snprintf(reds->stat_shm_name, shm_name_len, SPICE_STAT_SHM_NAME, getpid()); shm_unlink(reds->stat_shm_name); if ((fd = shm_open(reds->stat_shm_name, O_CREAT | O_RDWR, 0444)) == -1) { spice_error("statistics shm_open failed, %s", strerror(errno)); } if (ftruncate(fd, REDS_STAT_SHM_SIZE) == -1) { spice_error("statistics ftruncate failed, %s", strerror(errno)); } reds->stat = (SpiceStat *)mmap(NULL, REDS_STAT_SHM_SIZE, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0); if (reds->stat == (SpiceStat *)MAP_FAILED) { spice_error("statistics mmap failed, %s", strerror(errno)); } memset(reds->stat, 0, REDS_STAT_SHM_SIZE); reds->stat->magic = SPICE_STAT_MAGIC; reds->stat->version = SPICE_STAT_VERSION; reds->stat->root_index = INVALID_STAT_REF; if (pthread_mutex_init(&reds->stat_lock, NULL)) { spice_error("mutex init failed"); } #endif if (reds_init_net(reds) < 0) { goto err; } if (reds->secure_listen_socket != -1) { if (reds_init_ssl(reds) < 0) { goto err; } } #if HAVE_SASL int saslerr; if ((saslerr = sasl_server_init(NULL, reds->sasl_appname ? reds->sasl_appname : "spice")) != SASL_OK) { spice_error("Failed to initialize SASL auth %s", sasl_errstring(saslerr, NULL, NULL)); goto err; } #endif reds->main_channel = main_channel_new(reds); reds->inputs_channel = inputs_channel_new(reds); reds->mouse_mode = SPICE_MOUSE_MODE_SERVER; reds_client_monitors_config_cleanup(reds); reds->allow_multiple_clients = getenv(SPICE_DEBUG_ALLOW_MC_ENV) != NULL; if (reds->allow_multiple_clients) { spice_warning("spice: allowing multiple client connections"); } servers = g_list_prepend(servers, reds); return 0; err: return -1; } static const char default_renderer[] = "sw"; /* new interface */ SPICE_GNUC_VISIBLE SpiceServer *spice_server_new(void) { RedsState *reds = spice_new0(RedsState, 1); reds->default_channel_security = SPICE_CHANNEL_SECURITY_NONE | SPICE_CHANNEL_SECURITY_SSL; reds->renderers = g_array_sized_new(FALSE, TRUE, sizeof(uint32_t), RED_RENDERER_LAST); reds->spice_port = -1; reds->spice_secure_port = -1; reds->spice_listen_socket_fd = -1; reds->spice_family = PF_UNSPEC; reds->sasl_enabled = 0; // sasl disabled by default #if HAVE_SASL reds->sasl_appname = NULL; // default to "spice" if NULL #endif reds->spice_uuid_is_set = FALSE; memset(reds->spice_uuid, 0, sizeof(reds->spice_uuid)); reds->ticketing_enabled = TRUE; /* ticketing enabled by default */ reds->streaming_video = SPICE_STREAM_VIDEO_FILTER; reds->image_compression = SPICE_IMAGE_COMPRESSION_AUTO_GLZ; reds->jpeg_state = SPICE_WAN_COMPRESSION_AUTO; reds->zlib_glz_state = SPICE_WAN_COMPRESSION_AUTO; reds->agent_mouse = TRUE; reds->agent_copypaste = TRUE; reds->agent_file_xfer = TRUE; reds->exit_on_disconnect = FALSE; return reds; } typedef struct RendererInfo { int id; const char *name; } RendererInfo; static const RendererInfo renderers_info[] = { {RED_RENDERER_SW, "sw"}, {RED_RENDERER_INVALID, NULL}, }; static const RendererInfo *find_renderer(const char *name) { const RendererInfo *inf = renderers_info; while (inf->name) { if (strcmp(name, inf->name) == 0) { return inf; } inf++; } return NULL; } static int reds_add_renderer(RedsState *reds, const char *name) { const RendererInfo *inf; if (reds->renderers->len == RED_RENDERER_LAST || !(inf = find_renderer(name))) { return FALSE; } g_array_append_val(reds->renderers, inf->id); return TRUE; } SPICE_GNUC_VISIBLE int spice_server_init(SpiceServer *reds, SpiceCoreInterface *core) { int ret; ret = do_spice_init(reds, core); if (reds->renderers->len == 0) { reds_add_renderer(reds, default_renderer); } return ret; } SPICE_GNUC_VISIBLE void spice_server_destroy(SpiceServer *reds) { g_array_unref(reds->renderers); if (reds->main_channel) { main_channel_close(reds->main_channel); } reds_cleanup(reds); /* remove the server from the list of servers so that we don't attempt to * free it again at exit */ servers = g_list_remove(servers, reds); } SPICE_GNUC_VISIBLE spice_compat_version_t spice_get_current_compat_version(void) { return SPICE_COMPAT_VERSION_CURRENT; } SPICE_GNUC_VISIBLE int spice_server_set_compat_version(SpiceServer *reds, spice_compat_version_t version) { if (version < SPICE_COMPAT_VERSION_0_6) { /* We don't support 0.4 compat mode atm */ return -1; } if (version > SPICE_COMPAT_VERSION_CURRENT) { /* Not compatible with future versions */ return -1; } return 0; } SPICE_GNUC_VISIBLE int spice_server_set_port(SpiceServer *reds, int port) { if (port < 0 || port > 0xffff) { return -1; } reds->spice_port = port; return 0; } SPICE_GNUC_VISIBLE void spice_server_set_addr(SpiceServer *reds, const char *addr, int flags) { g_strlcpy(reds->spice_addr, addr, sizeof(reds->spice_addr)); if (flags == SPICE_ADDR_FLAG_IPV4_ONLY) { reds->spice_family = PF_INET; } else if (flags == SPICE_ADDR_FLAG_IPV6_ONLY) { reds->spice_family = PF_INET6; } else if (flags == SPICE_ADDR_FLAG_UNIX_ONLY) { reds->spice_family = AF_UNIX; } else if (flags != 0) { spice_warning("unknown address flag: 0x%X", flags); } } SPICE_GNUC_VISIBLE int spice_server_set_listen_socket_fd(SpiceServer *s, int listen_fd) { s->spice_listen_socket_fd = listen_fd; return 0; } SPICE_GNUC_VISIBLE int spice_server_set_exit_on_disconnect(SpiceServer *s, int flag) { s->exit_on_disconnect = !!flag; return 0; } SPICE_GNUC_VISIBLE int spice_server_set_noauth(SpiceServer *s) { memset(s->taTicket.password, 0, sizeof(s->taTicket.password)); s->ticketing_enabled = FALSE; return 0; } SPICE_GNUC_VISIBLE int spice_server_set_sasl(SpiceServer *s, int enabled) { #if HAVE_SASL s->sasl_enabled = enabled; return 0; #else return -1; #endif } SPICE_GNUC_VISIBLE int spice_server_set_sasl_appname(SpiceServer *s, const char *appname) { #if HAVE_SASL free(s->sasl_appname); s->sasl_appname = spice_strdup(appname); return 0; #else return -1; #endif } SPICE_GNUC_VISIBLE void spice_server_set_name(SpiceServer *s, const char *name) { free(s->spice_name); s->spice_name = spice_strdup(name); } SPICE_GNUC_VISIBLE void spice_server_set_uuid(SpiceServer *s, const uint8_t uuid[16]) { memcpy(s->spice_uuid, uuid, sizeof(s->spice_uuid)); s->spice_uuid_is_set = TRUE; } SPICE_GNUC_VISIBLE int spice_server_set_ticket(SpiceServer *reds, const char *passwd, int lifetime, int fail_if_connected, int disconnect_if_connected) { if (reds_main_channel_connected(reds)) { if (fail_if_connected) { return -1; } if (disconnect_if_connected) { reds_disconnect(reds); } } on_activating_ticketing(reds); reds->ticketing_enabled = TRUE; if (lifetime == 0) { reds->taTicket.expiration_time = INT_MAX; } else { time_t now = time(NULL); reds->taTicket.expiration_time = now + lifetime; } if (passwd != NULL) { if (strlen(passwd) > SPICE_MAX_PASSWORD_LENGTH) return -1; g_strlcpy(reds->taTicket.password, passwd, sizeof(reds->taTicket.password)); } else { memset(reds->taTicket.password, 0, sizeof(reds->taTicket.password)); reds->taTicket.expiration_time = 0; } return 0; } SPICE_GNUC_VISIBLE int spice_server_set_tls(SpiceServer *s, int port, const char *ca_cert_file, const char *certs_file, const char *private_key_file, const char *key_passwd, const char *dh_key_file, const char *ciphersuite) { if (port == 0 || ca_cert_file == NULL || certs_file == NULL || private_key_file == NULL) { return -1; } if (port < 0 || port > 0xffff) { return -1; } memset(&s->ssl_parameters, 0, sizeof(s->ssl_parameters)); s->spice_secure_port = port; g_strlcpy(s->ssl_parameters.ca_certificate_file, ca_cert_file, sizeof(s->ssl_parameters.ca_certificate_file)); g_strlcpy(s->ssl_parameters.certs_file, certs_file, sizeof(s->ssl_parameters.certs_file)); g_strlcpy(s->ssl_parameters.private_key_file, private_key_file, sizeof(s->ssl_parameters.private_key_file)); if (key_passwd) { g_strlcpy(s->ssl_parameters.keyfile_password, key_passwd, sizeof(s->ssl_parameters.keyfile_password)); } if (ciphersuite) { g_strlcpy(s->ssl_parameters.ciphersuite, ciphersuite, sizeof(s->ssl_parameters.ciphersuite)); } if (dh_key_file) { g_strlcpy(s->ssl_parameters.dh_key_file, dh_key_file, sizeof(s->ssl_parameters.dh_key_file)); } return 0; } SPICE_GNUC_VISIBLE int spice_server_set_image_compression(SpiceServer *s, SpiceImageCompression comp) { #ifndef USE_LZ4 if (comp == SPICE_IMAGE_COMPRESSION_LZ4) { spice_warning("LZ4 compression not supported, falling back to auto GLZ"); comp = SPICE_IMAGE_COMPRESSION_AUTO_GLZ; reds_set_image_compression(s, comp); return -1; } #endif reds_set_image_compression(s, comp); return 0; } SPICE_GNUC_VISIBLE SpiceImageCompression spice_server_get_image_compression(SpiceServer *s) { return s->image_compression; } SPICE_GNUC_VISIBLE int spice_server_set_jpeg_compression(SpiceServer *s, spice_wan_compression_t comp) { if (comp == SPICE_WAN_COMPRESSION_INVALID) { spice_error("invalid jpeg state"); return -1; } // todo: support dynamically changing the state s->jpeg_state = comp; return 0; } SPICE_GNUC_VISIBLE int spice_server_set_zlib_glz_compression(SpiceServer *s, spice_wan_compression_t comp) { if (comp == SPICE_WAN_COMPRESSION_INVALID) { spice_error("invalid zlib_glz state"); return -1; } // todo: support dynamically changing the state s->zlib_glz_state = comp; return 0; } SPICE_GNUC_VISIBLE int spice_server_set_channel_security(SpiceServer *s, const char *channel, int security) { static const char *const names[] = { [ SPICE_CHANNEL_MAIN ] = "main", [ SPICE_CHANNEL_DISPLAY ] = "display", [ SPICE_CHANNEL_INPUTS ] = "inputs", [ SPICE_CHANNEL_CURSOR ] = "cursor", [ SPICE_CHANNEL_PLAYBACK ] = "playback", [ SPICE_CHANNEL_RECORD ] = "record", #ifdef USE_SMARTCARD [ SPICE_CHANNEL_SMARTCARD] = "smartcard", #endif [ SPICE_CHANNEL_USBREDIR ] = "usbredir", [ SPICE_CHANNEL_WEBDAV ] = "webdav", }; int i; if (channel == NULL) { s->default_channel_security = security; return 0; } for (i = 0; i < SPICE_N_ELEMENTS(names); i++) { if (names[i] && strcmp(names[i], channel) == 0) { reds_set_one_channel_security(s, i, security); return 0; } } return -1; } SPICE_GNUC_VISIBLE int spice_server_get_sock_info(SpiceServer *reds, struct sockaddr *sa, socklen_t *salen) { if (main_channel_getsockname(reds->main_channel, sa, salen) < 0) { return -1; } return 0; } SPICE_GNUC_VISIBLE int spice_server_get_peer_info(SpiceServer *reds, struct sockaddr *sa, socklen_t *salen) { if (main_channel_getpeername(reds->main_channel, sa, salen) < 0) { return -1; } return 0; } SPICE_GNUC_VISIBLE int spice_server_is_server_mouse(SpiceServer *reds) { return reds->mouse_mode == SPICE_MOUSE_MODE_SERVER; } SPICE_GNUC_VISIBLE int spice_server_add_renderer(SpiceServer *reds, const char *name) { if (!reds_add_renderer(reds, name)) { return -1; } return 0; } SPICE_GNUC_VISIBLE int spice_server_kbd_leds(SpiceKbdInstance *sin, int leds) { RedsState *reds = spice_kbd_state_get_server(sin->st); inputs_channel_on_keyboard_leds_change(reds->inputs_channel, leds); return 0; } SPICE_GNUC_VISIBLE int spice_server_set_streaming_video(SpiceServer *reds, int value) { if (value != SPICE_STREAM_VIDEO_OFF && value != SPICE_STREAM_VIDEO_ALL && value != SPICE_STREAM_VIDEO_FILTER) return -1; reds->streaming_video = value; reds_on_sv_change(reds); return 0; } uint32_t reds_get_streaming_video(const RedsState *reds) { return reds->streaming_video; } SPICE_GNUC_VISIBLE int spice_server_set_playback_compression(SpiceServer *reds, int enable) { snd_set_playback_compression(enable); return 0; } SPICE_GNUC_VISIBLE int spice_server_set_agent_mouse(SpiceServer *reds, int enable) { reds->agent_mouse = enable; reds_update_mouse_mode(reds); return 0; } SPICE_GNUC_VISIBLE int spice_server_set_agent_copypaste(SpiceServer *reds, int enable) { reds->agent_copypaste = enable; reds->agent_dev->priv->write_filter.copy_paste_enabled = reds->agent_copypaste; reds->agent_dev->priv->read_filter.copy_paste_enabled = reds->agent_copypaste; return 0; } SPICE_GNUC_VISIBLE int spice_server_set_agent_file_xfer(SpiceServer *reds, int enable) { reds->agent_file_xfer = enable; reds->agent_dev->priv->write_filter.file_xfer_enabled = reds->agent_file_xfer; reds->agent_dev->priv->read_filter.file_xfer_enabled = reds->agent_file_xfer; return 0; } /* returns FALSE if info is invalid */ static int reds_set_migration_dest_info(RedsState *reds, const char* dest, int port, int secure_port, const char* cert_subject) { RedsMigSpice *spice_migration = NULL; reds_mig_release(reds); if ((port == -1 && secure_port == -1) || !dest) { return FALSE; } spice_migration = spice_new0(RedsMigSpice, 1); spice_migration->port = port; spice_migration->sport = secure_port; spice_migration->host = spice_strdup(dest); if (cert_subject) { spice_migration->cert_subject = spice_strdup(cert_subject); } reds->mig_spice = spice_migration; return TRUE; } /* semi-seamless client migration */ SPICE_GNUC_VISIBLE int spice_server_migrate_connect(SpiceServer *reds, const char* dest, int port, int secure_port, const char* cert_subject) { SpiceMigrateInterface *sif; int try_seamless; spice_info(NULL); spice_assert(reds->migration_interface); if (reds->expect_migrate) { spice_info("consecutive calls without migration. Canceling previous call"); main_channel_migrate_src_complete(reds->main_channel, FALSE); } sif = SPICE_CONTAINEROF(reds->migration_interface->base.sif, SpiceMigrateInterface, base); if (!reds_set_migration_dest_info(reds, dest, port, secure_port, cert_subject)) { sif->migrate_connect_complete(reds->migration_interface); return -1; } reds->expect_migrate = TRUE; /* * seamless migration support was added to the client after the support in * agent_connect_tokens, so there shouldn't be contradicition - if * the client is capable of seamless migration, it is capbable of agent_connected_tokens. * The demand for agent_connected_tokens support is in order to assure that if migration * occured when the agent was not connected, the tokens state after migration will still * be valid (see reds_reset_vdp for more details). */ try_seamless = reds->seamless_migration_enabled && red_channel_test_remote_cap(&reds->main_channel->base, SPICE_MAIN_CAP_AGENT_CONNECTED_TOKENS); /* main channel will take care of clients that are still during migration (at target)*/ if (main_channel_migrate_connect(reds->main_channel, reds->mig_spice, try_seamless)) { reds_mig_started(reds); } else { if (reds->num_clients == 0) { reds_mig_release(reds); spice_info("no client connected"); } sif->migrate_connect_complete(reds->migration_interface); } return 0; } SPICE_GNUC_VISIBLE int spice_server_migrate_info(SpiceServer *reds, const char* dest, int port, int secure_port, const char* cert_subject) { spice_info(NULL); spice_assert(!reds->migration_interface); if (!reds_set_migration_dest_info(reds, dest, port, secure_port, cert_subject)) { return -1; } return 0; } SPICE_GNUC_VISIBLE int spice_server_migrate_start(SpiceServer *reds) { spice_info(NULL); if (!reds->mig_spice) { return -1; } return 0; } SPICE_GNUC_VISIBLE int spice_server_migrate_end(SpiceServer *reds, int completed) { SpiceMigrateInterface *sif; int ret = 0; spice_info(NULL); spice_assert(reds->migration_interface); sif = SPICE_CONTAINEROF(reds->migration_interface->base.sif, SpiceMigrateInterface, base); if (completed && !reds->expect_migrate && reds->num_clients) { spice_warning("spice_server_migrate_info was not called, disconnecting clients"); reds_disconnect(reds); ret = -1; goto complete; } reds->expect_migrate = FALSE; if (!reds_main_channel_connected(reds)) { spice_info("no peer connected"); goto complete; } reds_mig_finished(reds, completed); return 0; complete: if (sif->migrate_end_complete) { sif->migrate_end_complete(reds->migration_interface); } return ret; } /* interface for switch-host migration */ SPICE_GNUC_VISIBLE int spice_server_migrate_switch(SpiceServer *reds) { spice_info(NULL); if (!reds->num_clients) { return 0; } reds->expect_migrate = FALSE; reds_mig_switch(reds); return 0; } SPICE_GNUC_VISIBLE void spice_server_vm_start(SpiceServer *reds) { GList *it; reds->vm_running = TRUE; for (it = reds->char_devices; it != NULL; it = it->next) { red_char_device_start(it->data); } reds_on_vm_start(reds); } SPICE_GNUC_VISIBLE void spice_server_vm_stop(SpiceServer *reds) { GList *it; reds->vm_running = FALSE; for (it = reds->char_devices; it != NULL; it = it->next) { red_char_device_stop(it->data); } reds_on_vm_stop(reds); } SPICE_GNUC_VISIBLE void spice_server_set_seamless_migration(SpiceServer *reds, int enable) { /* seamless migration is not supported with multiple clients */ reds->seamless_migration_enabled = enable && !reds->allow_multiple_clients; spice_debug("seamless migration enabled=%d", enable); } GArray* reds_get_renderers(RedsState *reds) { return reds->renderers; } spice_wan_compression_t reds_get_jpeg_state(const RedsState *reds) { return reds->jpeg_state; } spice_wan_compression_t reds_get_zlib_glz_state(const RedsState *reds) { return reds->zlib_glz_state; } SpiceCoreInterfaceInternal* reds_get_core_interface(RedsState *reds) { return reds->core; } SpiceWatch *reds_core_watch_add(RedsState *reds, int fd, int event_mask, SpiceWatchFunc func, void *opaque) { g_return_val_if_fail(reds != NULL, NULL); g_return_val_if_fail(reds->core != NULL, NULL); g_return_val_if_fail(reds->core->watch_add != NULL, NULL); return reds->core->watch_add(reds->core, fd, event_mask, func, opaque); } void reds_core_watch_update_mask(RedsState *reds, SpiceWatch *watch, int event_mask) { g_return_if_fail(reds != NULL); g_return_if_fail(reds->core != NULL); g_return_if_fail(reds->core->watch_update_mask != NULL); reds->core->watch_update_mask(watch, event_mask); } void reds_core_watch_remove(RedsState *reds, SpiceWatch *watch) { g_return_if_fail(reds != NULL); g_return_if_fail(reds->core != NULL); g_return_if_fail(reds->core->watch_remove != NULL); reds->core->watch_remove(watch); } SpiceTimer *reds_core_timer_add(RedsState *reds, SpiceTimerFunc func, void *opaque) { g_return_val_if_fail(reds != NULL, NULL); g_return_val_if_fail(reds->core != NULL, NULL); g_return_val_if_fail(reds->core->timer_add != NULL, NULL); return reds->core->timer_add(reds->core, func, opaque); } void reds_core_timer_start(RedsState *reds, SpiceTimer *timer, uint32_t ms) { g_return_if_fail(reds != NULL); g_return_if_fail(reds->core != NULL); g_return_if_fail(reds->core->timer_start != NULL); return reds->core->timer_start(timer, ms); } void reds_core_timer_cancel(RedsState *reds, SpiceTimer *timer) { g_return_if_fail(reds != NULL); g_return_if_fail(reds->core != NULL); g_return_if_fail(reds->core->timer_cancel != NULL); return reds->core->timer_cancel(timer); } void reds_core_timer_remove(RedsState *reds, SpiceTimer *timer) { g_return_if_fail(reds != NULL); g_return_if_fail(reds->core != NULL); g_return_if_fail(reds->core->timer_remove != NULL); return reds->core->timer_remove(timer); } void reds_update_client_mouse_allowed(RedsState *reds) { static int allowed = FALSE; int allow_now = FALSE; int x_res = 0; int y_res = 0; GList *l; int num_active_workers = g_list_length(reds->qxl_instances); if (num_active_workers > 0) { allow_now = TRUE; for (l = reds->qxl_instances; l != NULL && allow_now; l = l->next) { QXLInstance *qxl = l->data; if (red_qxl_get_primary_active(qxl)) { allow_now = red_qxl_get_allow_client_mouse(qxl, &x_res, &y_res); break; } } } if (allow_now || allow_now != allowed) { allowed = allow_now; reds_set_client_mouse_allowed(reds, allowed, x_res, y_res); } } gboolean reds_use_client_monitors_config(RedsState *reds) { GList *l; if (reds->qxl_instances == NULL) { return FALSE; } for (l = reds->qxl_instances; l != NULL ; l = l->next) { QXLInstance *qxl = l->data; if (!red_qxl_use_client_monitors_config(qxl)) return FALSE; } return TRUE; } void reds_client_monitors_config(RedsState *reds, VDAgentMonitorsConfig *monitors_config) { GList *l; for (l = reds->qxl_instances; l != NULL; l = l->next) { QXLInstance *qxl = l->data; if (!red_qxl_client_monitors_config(qxl, monitors_config)) { /* this is a normal condition, some qemu devices might not implement it */ spice_debug("QXLInterface::client_monitors_config failed\n"); } } } static int calc_compression_level(RedsState *reds) { spice_assert(reds_get_streaming_video(reds) != SPICE_STREAM_VIDEO_INVALID); if ((reds_get_streaming_video(reds) != SPICE_STREAM_VIDEO_OFF) || (spice_server_get_image_compression(reds) != SPICE_IMAGE_COMPRESSION_QUIC)) { return 0; } else { return 1; } } void reds_on_ic_change(RedsState *reds) { int compression_level = calc_compression_level(reds); GList *l; for (l = reds->qxl_instances; l != NULL; l = l->next) { QXLInstance *qxl = l->data; red_qxl_set_compression_level(qxl, compression_level); red_qxl_on_ic_change(qxl, spice_server_get_image_compression(reds)); } } void reds_on_sv_change(RedsState *reds) { int compression_level = calc_compression_level(reds); GList *l; for (l = reds->qxl_instances; l != NULL; l = l->next) { QXLInstance *qxl = l->data; red_qxl_set_compression_level(qxl, compression_level); red_qxl_on_sv_change(qxl, reds_get_streaming_video(reds)); } } void reds_on_vm_stop(RedsState *reds) { GList *l; for (l = reds->qxl_instances; l != NULL; l = l->next) { QXLInstance *qxl = l->data; red_qxl_stop(qxl); } } void reds_on_vm_start(RedsState *reds) { GList *l; for (l = reds->qxl_instances; l != NULL; l = l->next) { QXLInstance *qxl = l->data; red_qxl_start(qxl); } } uint32_t reds_qxl_ram_size(RedsState *reds) { QXLInstance *first; if (!reds->qxl_instances) { return 0; } first = reds->qxl_instances->data; return red_qxl_get_ram_size(first); } MainDispatcher* reds_get_main_dispatcher(RedsState *reds) { return reds->main_dispatcher; } static void red_char_device_vdi_port_constructed(GObject *object) { RedCharDeviceVDIPort *dev = RED_CHAR_DEVICE_VDIPORT(object); RedsState *reds; G_OBJECT_CLASS(red_char_device_vdi_port_parent_class)->constructed(object); g_object_get(dev, "spice-server", &reds, NULL); agent_msg_filter_init(&dev->priv->write_filter, reds->agent_copypaste, reds->agent_file_xfer, reds_use_client_monitors_config(reds), TRUE); agent_msg_filter_init(&dev->priv->read_filter, reds->agent_copypaste, reds->agent_file_xfer, reds_use_client_monitors_config(reds), TRUE); } static void red_char_device_vdi_port_init(RedCharDeviceVDIPort *self) { int i; self->priv = RED_CHAR_DEVICE_VDIPORT_PRIVATE(self); ring_init(&self->priv->read_bufs); self->priv->read_state = VDI_PORT_READ_STATE_READ_HEADER; self->priv->receive_pos = (uint8_t *)&self->priv->vdi_chunk_header; self->priv->receive_len = sizeof(self->priv->vdi_chunk_header); for (i = 0; i < REDS_VDI_PORT_NUM_RECEIVE_BUFFS; i++) { VDIReadBuf *buf = spice_new0(VDIReadBuf, 1); buf->dev = self; ring_item_init(&buf->link); ring_add(&self->priv->read_bufs, &buf->link); } } static void red_char_device_vdi_port_finalize(GObject *object) { RedCharDeviceVDIPort *dev = RED_CHAR_DEVICE_VDIPORT(object); free(dev->priv->mig_data); /* FIXME: need to free the VDIReadBuf allocated previously */ } static void red_char_device_vdi_port_class_init(RedCharDeviceVDIPortClass *klass) { GObjectClass *object_class = G_OBJECT_CLASS(klass); RedCharDeviceClass *char_dev_class = RED_CHAR_DEVICE_CLASS(klass); g_type_class_add_private(klass, sizeof (RedCharDeviceVDIPortPrivate)); object_class->finalize = red_char_device_vdi_port_finalize; object_class->constructed = red_char_device_vdi_port_constructed; char_dev_class->read_one_msg_from_device = vdi_port_read_one_msg_from_device; char_dev_class->ref_msg_to_client = vdi_port_ref_msg_to_client; char_dev_class->unref_msg_to_client = vdi_port_unref_msg_to_client; char_dev_class->send_msg_to_client = vdi_port_send_msg_to_client; char_dev_class->send_tokens_to_client = vdi_port_send_tokens_to_client; char_dev_class->remove_client = vdi_port_remove_client; char_dev_class->on_free_self_token = vdi_port_on_free_self_token; } static RedCharDeviceVDIPort *red_char_device_vdi_port_new(RedsState *reds) { return g_object_new(RED_TYPE_CHAR_DEVICE_VDIPORT, "spice-server", reds, "client-tokens-interval", REDS_TOKENS_TO_SEND, "self-tokens", REDS_NUM_INTERNAL_AGENT_MESSAGES, "opaque", reds, NULL); }