1 /* 2 * QEMU live migration via socket 3 * 4 * Copyright Red Hat, Inc. 2009-2016 5 * 6 * Authors: 7 * Chris Lalancette <clalance@redhat.com> 8 * Daniel P. Berrange <berrange@redhat.com> 9 * 10 * This work is licensed under the terms of the GNU GPL, version 2. See 11 * the COPYING file in the top-level directory. 12 * 13 * Contributions after 2012-01-13 are licensed under the terms of the 14 * GNU GPL, version 2 or (at your option) any later version. 15 */ 16 17 #include "qemu/osdep.h" 18 #include "qemu/cutils.h" 19 20 #include "qemu/error-report.h" 21 #include "qapi/error.h" 22 #include "channel.h" 23 #include "socket.h" 24 #include "migration.h" 25 #include "qemu-file.h" 26 #include "io/channel-socket.h" 27 #include "io/net-listener.h" 28 #include "trace.h" 29 30 31 struct SocketOutgoingArgs { 32 SocketAddress *saddr; 33 } outgoing_args; 34 35 void socket_send_channel_create(QIOTaskFunc f, void *data) 36 { 37 QIOChannelSocket *sioc = qio_channel_socket_new(); 38 qio_channel_socket_connect_async(sioc, outgoing_args.saddr, 39 f, data, NULL, NULL); 40 } 41 42 int socket_send_channel_destroy(QIOChannel *send) 43 { 44 /* Remove channel */ 45 object_unref(OBJECT(send)); 46 if (outgoing_args.saddr) { 47 qapi_free_SocketAddress(outgoing_args.saddr); 48 outgoing_args.saddr = NULL; 49 } 50 return 0; 51 } 52 53 struct SocketConnectData { 54 MigrationState *s; 55 char *hostname; 56 }; 57 58 static void socket_connect_data_free(void *opaque) 59 { 60 struct SocketConnectData *data = opaque; 61 if (!data) { 62 return; 63 } 64 g_free(data->hostname); 65 g_free(data); 66 } 67 68 static void socket_outgoing_migration(QIOTask *task, 69 gpointer opaque) 70 { 71 struct SocketConnectData *data = opaque; 72 QIOChannel *sioc = QIO_CHANNEL(qio_task_get_source(task)); 73 Error *err = NULL; 74 75 if (qio_task_propagate_error(task, &err)) { 76 trace_migration_socket_outgoing_error(error_get_pretty(err)); 77 goto out; 78 } 79 80 trace_migration_socket_outgoing_connected(data->hostname); 81 82 if (migrate_use_zero_copy_send()) { 83 error_setg(&err, "Zero copy send not available in migration"); 84 } 85 86 out: 87 migration_channel_connect(data->s, sioc, data->hostname, err); 88 object_unref(OBJECT(sioc)); 89 } 90 91 static void 92 socket_start_outgoing_migration_internal(MigrationState *s, 93 SocketAddress *saddr, 94 Error **errp) 95 { 96 QIOChannelSocket *sioc = qio_channel_socket_new(); 97 struct SocketConnectData *data = g_new0(struct SocketConnectData, 1); 98 99 data->s = s; 100 101 /* in case previous migration leaked it */ 102 qapi_free_SocketAddress(outgoing_args.saddr); 103 outgoing_args.saddr = saddr; 104 105 if (saddr->type == SOCKET_ADDRESS_TYPE_INET) { 106 data->hostname = g_strdup(saddr->u.inet.host); 107 } 108 109 qio_channel_set_name(QIO_CHANNEL(sioc), "migration-socket-outgoing"); 110 qio_channel_socket_connect_async(sioc, 111 saddr, 112 socket_outgoing_migration, 113 data, 114 socket_connect_data_free, 115 NULL); 116 } 117 118 void socket_start_outgoing_migration(MigrationState *s, 119 const char *str, 120 Error **errp) 121 { 122 Error *err = NULL; 123 SocketAddress *saddr = socket_parse(str, &err); 124 if (!err) { 125 socket_start_outgoing_migration_internal(s, saddr, &err); 126 } 127 error_propagate(errp, err); 128 } 129 130 static void socket_accept_incoming_migration(QIONetListener *listener, 131 QIOChannelSocket *cioc, 132 gpointer opaque) 133 { 134 trace_migration_socket_incoming_accepted(); 135 136 if (migration_has_all_channels()) { 137 error_report("%s: Extra incoming migration connection; ignoring", 138 __func__); 139 return; 140 } 141 142 qio_channel_set_name(QIO_CHANNEL(cioc), "migration-socket-incoming"); 143 migration_channel_process_incoming(QIO_CHANNEL(cioc)); 144 } 145 146 static void 147 socket_incoming_migration_end(void *opaque) 148 { 149 QIONetListener *listener = opaque; 150 151 qio_net_listener_disconnect(listener); 152 object_unref(OBJECT(listener)); 153 } 154 155 static void 156 socket_start_incoming_migration_internal(SocketAddress *saddr, 157 Error **errp) 158 { 159 QIONetListener *listener = qio_net_listener_new(); 160 MigrationIncomingState *mis = migration_incoming_get_current(); 161 size_t i; 162 int num = 1; 163 164 qio_net_listener_set_name(listener, "migration-socket-listener"); 165 166 if (migrate_use_multifd()) { 167 num = migrate_multifd_channels(); 168 } 169 170 if (qio_net_listener_open_sync(listener, saddr, num, errp) < 0) { 171 object_unref(OBJECT(listener)); 172 return; 173 } 174 175 mis->transport_data = listener; 176 mis->transport_cleanup = socket_incoming_migration_end; 177 178 qio_net_listener_set_client_func_full(listener, 179 socket_accept_incoming_migration, 180 NULL, NULL, 181 g_main_context_get_thread_default()); 182 183 for (i = 0; i < listener->nsioc; i++) { 184 SocketAddress *address = 185 qio_channel_socket_get_local_address(listener->sioc[i], errp); 186 if (!address) { 187 return; 188 } 189 migrate_add_address(address); 190 qapi_free_SocketAddress(address); 191 } 192 } 193 194 void socket_start_incoming_migration(const char *str, Error **errp) 195 { 196 Error *err = NULL; 197 SocketAddress *saddr = socket_parse(str, &err); 198 if (!err) { 199 socket_start_incoming_migration_internal(saddr, &err); 200 } 201 qapi_free_SocketAddress(saddr); 202 error_propagate(errp, err); 203 } 204