4 * Copyright IBM, Corp. 2008
7 * Anthony Liguori <aliguori@us.ibm.com>
9 * This work is licensed under the terms of the GNU GPL, version 2. See
10 * the COPYING file in the top-level directory.
14 #include "qemu-common.h"
15 #include "migration.h"
17 #include "buffered_file.h"
20 #include "qemu_socket.h"
22 //#define DEBUG_MIGRATION
24 #ifdef DEBUG_MIGRATION
25 #define dprintf(fmt, ...) \
26 do { printf("migration: " fmt, ## __VA_ARGS__); } while (0)
28 #define dprintf(fmt, ...) \
32 /* Migration speed throttling */
33 static uint32_t max_throttle = (32 << 20);
35 static MigrationState *current_migration;
37 void qemu_start_incoming_migration(const char *uri)
41 if (strstart(uri, "tcp:", &p))
42 tcp_start_incoming_migration(p);
44 else if (strstart(uri, "exec:", &p))
45 exec_start_incoming_migration(p);
48 fprintf(stderr, "unknown migration protocol: %s\n", uri);
51 void do_migrate(Monitor *mon, int detach, const char *uri)
53 MigrationState *s = NULL;
56 if (strstart(uri, "tcp:", &p))
57 s = tcp_start_outgoing_migration(p, max_throttle, detach);
59 else if (strstart(uri, "exec:", &p))
60 s = exec_start_outgoing_migration(p, max_throttle, detach);
63 monitor_printf(mon, "unknown migration protocol: %s\n", uri);
66 monitor_printf(mon, "migration failed\n");
68 if (current_migration)
69 current_migration->release(current_migration);
71 current_migration = s;
75 void do_migrate_cancel(Monitor *mon)
77 MigrationState *s = current_migration;
83 void do_migrate_set_speed(Monitor *mon, const char *value)
88 d = strtod(value, &ptr);
100 max_throttle = (uint32_t)d;
103 void do_info_migrate(Monitor *mon)
105 MigrationState *s = current_migration;
108 monitor_printf(mon, "Migration status: ");
109 switch (s->get_status(s)) {
110 case MIG_STATE_ACTIVE:
111 monitor_printf(mon, "active\n");
113 case MIG_STATE_COMPLETED:
114 monitor_printf(mon, "completed\n");
116 case MIG_STATE_ERROR:
117 monitor_printf(mon, "failed\n");
119 case MIG_STATE_CANCELLED:
120 monitor_printf(mon, "cancelled\n");
126 /* shared migration helpers */
128 void migrate_fd_monitor_suspend(FdMigrationState *s)
130 s->mon_resume = cur_mon;
131 monitor_suspend(cur_mon);
132 dprintf("suspending monitor\n");
135 void migrate_fd_error(FdMigrationState *s)
137 dprintf("setting error state\n");
138 s->state = MIG_STATE_ERROR;
139 migrate_fd_cleanup(s);
142 void migrate_fd_cleanup(FdMigrationState *s)
144 qemu_set_fd_handler2(s->fd, NULL, NULL, NULL, NULL);
147 dprintf("closing file\n");
148 qemu_fclose(s->file);
154 /* Don't resume monitor until we've flushed all of the buffers */
156 monitor_resume(s->mon_resume);
161 void migrate_fd_put_notify(void *opaque)
163 FdMigrationState *s = opaque;
165 qemu_set_fd_handler2(s->fd, NULL, NULL, NULL, NULL);
166 qemu_file_put_notify(s->file);
169 ssize_t migrate_fd_put_buffer(void *opaque, const void *data, size_t size)
171 FdMigrationState *s = opaque;
175 ret = s->write(s, data, size);
176 } while (ret == -1 && ((s->get_error(s)) == EINTR || (s->get_error(s)) == EWOULDBLOCK));
179 ret = -(s->get_error(s));
182 qemu_set_fd_handler2(s->fd, NULL, NULL, migrate_fd_put_notify, s);
187 void migrate_fd_connect(FdMigrationState *s)
191 s->file = qemu_fopen_ops_buffered(s,
193 migrate_fd_put_buffer,
194 migrate_fd_put_ready,
195 migrate_fd_wait_for_unfreeze,
198 dprintf("beginning savevm\n");
199 ret = qemu_savevm_state_begin(s->file);
201 dprintf("failed, %d\n", ret);
206 migrate_fd_put_ready(s);
209 void migrate_fd_put_ready(void *opaque)
211 FdMigrationState *s = opaque;
213 if (s->state != MIG_STATE_ACTIVE) {
214 dprintf("put_ready returning because of non-active state\n");
218 dprintf("iterate\n");
219 if (qemu_savevm_state_iterate(s->file) == 1) {
220 dprintf("done iterating\n");
224 qemu_savevm_state_complete(s->file);
225 s->state = MIG_STATE_COMPLETED;
226 migrate_fd_cleanup(s);
230 int migrate_fd_get_status(MigrationState *mig_state)
232 FdMigrationState *s = migrate_to_fms(mig_state);
236 void migrate_fd_cancel(MigrationState *mig_state)
238 FdMigrationState *s = migrate_to_fms(mig_state);
240 if (s->state != MIG_STATE_ACTIVE)
243 dprintf("cancelling migration\n");
245 s->state = MIG_STATE_CANCELLED;
247 migrate_fd_cleanup(s);
250 void migrate_fd_release(MigrationState *mig_state)
252 FdMigrationState *s = migrate_to_fms(mig_state);
254 dprintf("releasing state\n");
256 if (s->state == MIG_STATE_ACTIVE) {
257 s->state = MIG_STATE_CANCELLED;
258 migrate_fd_cleanup(s);
263 void migrate_fd_wait_for_unfreeze(void *opaque)
265 FdMigrationState *s = opaque;
268 dprintf("wait for unfreeze\n");
269 if (s->state != MIG_STATE_ACTIVE)
276 FD_SET(s->fd, &wfds);
278 ret = select(s->fd + 1, NULL, &wfds, NULL, NULL);
279 } while (ret == -1 && (s->get_error(s)) == EINTR);
282 int migrate_fd_close(void *opaque)
284 FdMigrationState *s = opaque;