xref: /openbmc/qemu/migration/migration.c (revision 0f550c5c)
1 /*
2  * QEMU live migration
3  *
4  * Copyright IBM, Corp. 2008
5  *
6  * Authors:
7  *  Anthony Liguori   <aliguori@us.ibm.com>
8  *
9  * This work is licensed under the terms of the GNU GPL, version 2.  See
10  * the COPYING file in the top-level directory.
11  *
12  * Contributions after 2012-01-13 are licensed under the terms of the
13  * GNU GPL, version 2 or (at your option) any later version.
14  */
15 
16 #include "qemu/osdep.h"
17 #include "qemu/cutils.h"
18 #include "qemu/error-report.h"
19 #include "migration/blocker.h"
20 #include "exec.h"
21 #include "fd.h"
22 #include "socket.h"
23 #include "rdma.h"
24 #include "ram.h"
25 #include "migration/global_state.h"
26 #include "migration/misc.h"
27 #include "migration.h"
28 #include "savevm.h"
29 #include "qemu-file-channel.h"
30 #include "qemu-file.h"
31 #include "migration/vmstate.h"
32 #include "block/block.h"
33 #include "qapi/qmp/qerror.h"
34 #include "qemu/rcu.h"
35 #include "block.h"
36 #include "postcopy-ram.h"
37 #include "qemu/thread.h"
38 #include "qmp-commands.h"
39 #include "trace.h"
40 #include "qapi-event.h"
41 #include "exec/target_page.h"
42 #include "io/channel-buffer.h"
43 #include "migration/colo.h"
44 #include "hw/boards.h"
45 #include "monitor/monitor.h"
46 
47 #define MAX_THROTTLE  (32 << 20)      /* Migration transfer speed throttling */
48 
49 /* Amount of time to allocate to each "chunk" of bandwidth-throttled
50  * data. */
51 #define BUFFER_DELAY     100
52 #define XFER_LIMIT_RATIO (1000 / BUFFER_DELAY)
53 
54 /* Time in milliseconds we are allowed to stop the source,
55  * for sending the last part */
56 #define DEFAULT_MIGRATE_SET_DOWNTIME 300
57 
58 /* Maximum migrate downtime set to 2000 seconds */
59 #define MAX_MIGRATE_DOWNTIME_SECONDS 2000
60 #define MAX_MIGRATE_DOWNTIME (MAX_MIGRATE_DOWNTIME_SECONDS * 1000)
61 
62 /* Default compression thread count */
63 #define DEFAULT_MIGRATE_COMPRESS_THREAD_COUNT 8
64 /* Default decompression thread count, usually decompression is at
65  * least 4 times as fast as compression.*/
66 #define DEFAULT_MIGRATE_DECOMPRESS_THREAD_COUNT 2
67 /*0: means nocompress, 1: best speed, ... 9: best compress ratio */
68 #define DEFAULT_MIGRATE_COMPRESS_LEVEL 1
69 /* Define default autoconverge cpu throttle migration parameters */
70 #define DEFAULT_MIGRATE_CPU_THROTTLE_INITIAL 20
71 #define DEFAULT_MIGRATE_CPU_THROTTLE_INCREMENT 10
72 
73 /* Migration XBZRLE default cache size */
74 #define DEFAULT_MIGRATE_CACHE_SIZE (64 * 1024 * 1024)
75 
76 /* The delay time (in ms) between two COLO checkpoints
77  * Note: Please change this default value to 10000 when we support hybrid mode.
78  */
79 #define DEFAULT_MIGRATE_X_CHECKPOINT_DELAY 200
80 #define DEFAULT_MIGRATE_MULTIFD_CHANNELS 2
81 #define DEFAULT_MIGRATE_MULTIFD_PAGE_COUNT 16
82 
83 static NotifierList migration_state_notifiers =
84     NOTIFIER_LIST_INITIALIZER(migration_state_notifiers);
85 
86 static bool deferred_incoming;
87 
88 /* Messages sent on the return path from destination to source */
89 enum mig_rp_message_type {
90     MIG_RP_MSG_INVALID = 0,  /* Must be 0 */
91     MIG_RP_MSG_SHUT,         /* sibling will not send any more RP messages */
92     MIG_RP_MSG_PONG,         /* Response to a PING; data (seq: be32 ) */
93 
94     MIG_RP_MSG_REQ_PAGES_ID, /* data (start: be64, len: be32, id: string) */
95     MIG_RP_MSG_REQ_PAGES,    /* data (start: be64, len: be32) */
96 
97     MIG_RP_MSG_MAX
98 };
99 
100 /* When we add fault tolerance, we could have several
101    migrations at once.  For now we don't need to add
102    dynamic creation of migration */
103 
104 static MigrationState *current_migration;
105 
106 static bool migration_object_check(MigrationState *ms, Error **errp);
107 static int migration_maybe_pause(MigrationState *s,
108                                  int *current_active_state,
109                                  int new_state);
110 
111 void migration_object_init(void)
112 {
113     MachineState *ms = MACHINE(qdev_get_machine());
114     Error *err = NULL;
115 
116     /* This can only be called once. */
117     assert(!current_migration);
118     current_migration = MIGRATION_OBJ(object_new(TYPE_MIGRATION));
119 
120     if (!migration_object_check(current_migration, &err)) {
121         error_report_err(err);
122         exit(1);
123     }
124 
125     /*
126      * We cannot really do this in migration_instance_init() since at
127      * that time global properties are not yet applied, then this
128      * value will be definitely replaced by something else.
129      */
130     if (ms->enforce_config_section) {
131         current_migration->send_configuration = true;
132     }
133 }
134 
135 /* For outgoing */
136 MigrationState *migrate_get_current(void)
137 {
138     /* This can only be called after the object created. */
139     assert(current_migration);
140     return current_migration;
141 }
142 
143 MigrationIncomingState *migration_incoming_get_current(void)
144 {
145     static bool once;
146     static MigrationIncomingState mis_current;
147 
148     if (!once) {
149         mis_current.state = MIGRATION_STATUS_NONE;
150         memset(&mis_current, 0, sizeof(MigrationIncomingState));
151         qemu_mutex_init(&mis_current.rp_mutex);
152         qemu_event_init(&mis_current.main_thread_load_event, false);
153         once = true;
154     }
155     return &mis_current;
156 }
157 
158 void migration_incoming_state_destroy(void)
159 {
160     struct MigrationIncomingState *mis = migration_incoming_get_current();
161 
162     if (mis->to_src_file) {
163         /* Tell source that we are done */
164         migrate_send_rp_shut(mis, qemu_file_get_error(mis->from_src_file) != 0);
165         qemu_fclose(mis->to_src_file);
166         mis->to_src_file = NULL;
167     }
168 
169     if (mis->from_src_file) {
170         qemu_fclose(mis->from_src_file);
171         mis->from_src_file = NULL;
172     }
173 
174     qemu_event_reset(&mis->main_thread_load_event);
175 }
176 
177 static void migrate_generate_event(int new_state)
178 {
179     if (migrate_use_events()) {
180         qapi_event_send_migration(new_state, &error_abort);
181     }
182 }
183 
184 /*
185  * Called on -incoming with a defer: uri.
186  * The migration can be started later after any parameters have been
187  * changed.
188  */
189 static void deferred_incoming_migration(Error **errp)
190 {
191     if (deferred_incoming) {
192         error_setg(errp, "Incoming migration already deferred");
193     }
194     deferred_incoming = true;
195 }
196 
197 /*
198  * Send a message on the return channel back to the source
199  * of the migration.
200  */
201 static void migrate_send_rp_message(MigrationIncomingState *mis,
202                                     enum mig_rp_message_type message_type,
203                                     uint16_t len, void *data)
204 {
205     trace_migrate_send_rp_message((int)message_type, len);
206     qemu_mutex_lock(&mis->rp_mutex);
207     qemu_put_be16(mis->to_src_file, (unsigned int)message_type);
208     qemu_put_be16(mis->to_src_file, len);
209     qemu_put_buffer(mis->to_src_file, data, len);
210     qemu_fflush(mis->to_src_file);
211     qemu_mutex_unlock(&mis->rp_mutex);
212 }
213 
214 /* Request a range of pages from the source VM at the given
215  * start address.
216  *   rbname: Name of the RAMBlock to request the page in, if NULL it's the same
217  *           as the last request (a name must have been given previously)
218  *   Start: Address offset within the RB
219  *   Len: Length in bytes required - must be a multiple of pagesize
220  */
221 void migrate_send_rp_req_pages(MigrationIncomingState *mis, const char *rbname,
222                                ram_addr_t start, size_t len)
223 {
224     uint8_t bufc[12 + 1 + 255]; /* start (8), len (4), rbname up to 256 */
225     size_t msglen = 12; /* start + len */
226 
227     *(uint64_t *)bufc = cpu_to_be64((uint64_t)start);
228     *(uint32_t *)(bufc + 8) = cpu_to_be32((uint32_t)len);
229 
230     if (rbname) {
231         int rbname_len = strlen(rbname);
232         assert(rbname_len < 256);
233 
234         bufc[msglen++] = rbname_len;
235         memcpy(bufc + msglen, rbname, rbname_len);
236         msglen += rbname_len;
237         migrate_send_rp_message(mis, MIG_RP_MSG_REQ_PAGES_ID, msglen, bufc);
238     } else {
239         migrate_send_rp_message(mis, MIG_RP_MSG_REQ_PAGES, msglen, bufc);
240     }
241 }
242 
243 void qemu_start_incoming_migration(const char *uri, Error **errp)
244 {
245     const char *p;
246 
247     qapi_event_send_migration(MIGRATION_STATUS_SETUP, &error_abort);
248     if (!strcmp(uri, "defer")) {
249         deferred_incoming_migration(errp);
250     } else if (strstart(uri, "tcp:", &p)) {
251         tcp_start_incoming_migration(p, errp);
252 #ifdef CONFIG_RDMA
253     } else if (strstart(uri, "rdma:", &p)) {
254         rdma_start_incoming_migration(p, errp);
255 #endif
256     } else if (strstart(uri, "exec:", &p)) {
257         exec_start_incoming_migration(p, errp);
258     } else if (strstart(uri, "unix:", &p)) {
259         unix_start_incoming_migration(p, errp);
260     } else if (strstart(uri, "fd:", &p)) {
261         fd_start_incoming_migration(p, errp);
262     } else {
263         error_setg(errp, "unknown migration protocol: %s", uri);
264     }
265 }
266 
267 static void process_incoming_migration_bh(void *opaque)
268 {
269     Error *local_err = NULL;
270     MigrationIncomingState *mis = opaque;
271 
272     /* Make sure all file formats flush their mutable metadata.
273      * If we get an error here, just don't restart the VM yet. */
274     bdrv_invalidate_cache_all(&local_err);
275     if (local_err) {
276         error_report_err(local_err);
277         local_err = NULL;
278         autostart = false;
279     }
280 
281     /*
282      * This must happen after all error conditions are dealt with and
283      * we're sure the VM is going to be running on this host.
284      */
285     qemu_announce_self();
286 
287     if (multifd_load_cleanup(&local_err) != 0) {
288         error_report_err(local_err);
289         autostart = false;
290     }
291     /* If global state section was not received or we are in running
292        state, we need to obey autostart. Any other state is set with
293        runstate_set. */
294 
295     if (!global_state_received() ||
296         global_state_get_runstate() == RUN_STATE_RUNNING) {
297         if (autostart) {
298             vm_start();
299         } else {
300             runstate_set(RUN_STATE_PAUSED);
301         }
302     } else {
303         runstate_set(global_state_get_runstate());
304     }
305     /*
306      * This must happen after any state changes since as soon as an external
307      * observer sees this event they might start to prod at the VM assuming
308      * it's ready to use.
309      */
310     migrate_set_state(&mis->state, MIGRATION_STATUS_ACTIVE,
311                       MIGRATION_STATUS_COMPLETED);
312     qemu_bh_delete(mis->bh);
313     migration_incoming_state_destroy();
314 }
315 
316 static void process_incoming_migration_co(void *opaque)
317 {
318     MigrationIncomingState *mis = migration_incoming_get_current();
319     PostcopyState ps;
320     int ret;
321 
322     assert(mis->from_src_file);
323     mis->largest_page_size = qemu_ram_pagesize_largest();
324     postcopy_state_set(POSTCOPY_INCOMING_NONE);
325     migrate_set_state(&mis->state, MIGRATION_STATUS_NONE,
326                       MIGRATION_STATUS_ACTIVE);
327     ret = qemu_loadvm_state(mis->from_src_file);
328 
329     ps = postcopy_state_get();
330     trace_process_incoming_migration_co_end(ret, ps);
331     if (ps != POSTCOPY_INCOMING_NONE) {
332         if (ps == POSTCOPY_INCOMING_ADVISE) {
333             /*
334              * Where a migration had postcopy enabled (and thus went to advise)
335              * but managed to complete within the precopy period, we can use
336              * the normal exit.
337              */
338             postcopy_ram_incoming_cleanup(mis);
339         } else if (ret >= 0) {
340             /*
341              * Postcopy was started, cleanup should happen at the end of the
342              * postcopy thread.
343              */
344             trace_process_incoming_migration_co_postcopy_end_main();
345             return;
346         }
347         /* Else if something went wrong then just fall out of the normal exit */
348     }
349 
350     /* we get COLO info, and know if we are in COLO mode */
351     if (!ret && migration_incoming_enable_colo()) {
352         mis->migration_incoming_co = qemu_coroutine_self();
353         qemu_thread_create(&mis->colo_incoming_thread, "COLO incoming",
354              colo_process_incoming_thread, mis, QEMU_THREAD_JOINABLE);
355         mis->have_colo_incoming_thread = true;
356         qemu_coroutine_yield();
357 
358         /* Wait checkpoint incoming thread exit before free resource */
359         qemu_thread_join(&mis->colo_incoming_thread);
360     }
361 
362     if (ret < 0) {
363         Error *local_err = NULL;
364 
365         migrate_set_state(&mis->state, MIGRATION_STATUS_ACTIVE,
366                           MIGRATION_STATUS_FAILED);
367         error_report("load of migration failed: %s", strerror(-ret));
368         qemu_fclose(mis->from_src_file);
369         if (multifd_load_cleanup(&local_err) != 0) {
370             error_report_err(local_err);
371         }
372         exit(EXIT_FAILURE);
373     }
374     mis->bh = qemu_bh_new(process_incoming_migration_bh, mis);
375     qemu_bh_schedule(mis->bh);
376 }
377 
378 static void migration_incoming_setup(QEMUFile *f)
379 {
380     MigrationIncomingState *mis = migration_incoming_get_current();
381 
382     if (multifd_load_setup() != 0) {
383         /* We haven't been able to create multifd threads
384            nothing better to do */
385         exit(EXIT_FAILURE);
386     }
387 
388     if (!mis->from_src_file) {
389         mis->from_src_file = f;
390     }
391     qemu_file_set_blocking(f, false);
392 }
393 
394 static void migration_incoming_process(void)
395 {
396     Coroutine *co = qemu_coroutine_create(process_incoming_migration_co, NULL);
397     qemu_coroutine_enter(co);
398 }
399 
400 void migration_fd_process_incoming(QEMUFile *f)
401 {
402     migration_incoming_setup(f);
403     migration_incoming_process();
404 }
405 
406 void migration_ioc_process_incoming(QIOChannel *ioc)
407 {
408     MigrationIncomingState *mis = migration_incoming_get_current();
409 
410     if (!mis->from_src_file) {
411         QEMUFile *f = qemu_fopen_channel_input(ioc);
412         migration_fd_process_incoming(f);
413     }
414     /* We still only have a single channel.  Nothing to do here yet */
415 }
416 
417 /**
418  * @migration_has_all_channels: We have received all channels that we need
419  *
420  * Returns true when we have got connections to all the channels that
421  * we need for migration.
422  */
423 bool migration_has_all_channels(void)
424 {
425     return true;
426 }
427 
428 /*
429  * Send a 'SHUT' message on the return channel with the given value
430  * to indicate that we've finished with the RP.  Non-0 value indicates
431  * error.
432  */
433 void migrate_send_rp_shut(MigrationIncomingState *mis,
434                           uint32_t value)
435 {
436     uint32_t buf;
437 
438     buf = cpu_to_be32(value);
439     migrate_send_rp_message(mis, MIG_RP_MSG_SHUT, sizeof(buf), &buf);
440 }
441 
442 /*
443  * Send a 'PONG' message on the return channel with the given value
444  * (normally in response to a 'PING')
445  */
446 void migrate_send_rp_pong(MigrationIncomingState *mis,
447                           uint32_t value)
448 {
449     uint32_t buf;
450 
451     buf = cpu_to_be32(value);
452     migrate_send_rp_message(mis, MIG_RP_MSG_PONG, sizeof(buf), &buf);
453 }
454 
455 MigrationCapabilityStatusList *qmp_query_migrate_capabilities(Error **errp)
456 {
457     MigrationCapabilityStatusList *head = NULL;
458     MigrationCapabilityStatusList *caps;
459     MigrationState *s = migrate_get_current();
460     int i;
461 
462     caps = NULL; /* silence compiler warning */
463     for (i = 0; i < MIGRATION_CAPABILITY__MAX; i++) {
464 #ifndef CONFIG_LIVE_BLOCK_MIGRATION
465         if (i == MIGRATION_CAPABILITY_BLOCK) {
466             continue;
467         }
468 #endif
469         if (head == NULL) {
470             head = g_malloc0(sizeof(*caps));
471             caps = head;
472         } else {
473             caps->next = g_malloc0(sizeof(*caps));
474             caps = caps->next;
475         }
476         caps->value =
477             g_malloc(sizeof(*caps->value));
478         caps->value->capability = i;
479         caps->value->state = s->enabled_capabilities[i];
480     }
481 
482     return head;
483 }
484 
485 MigrationParameters *qmp_query_migrate_parameters(Error **errp)
486 {
487     MigrationParameters *params;
488     MigrationState *s = migrate_get_current();
489 
490     /* TODO use QAPI_CLONE() instead of duplicating it inline */
491     params = g_malloc0(sizeof(*params));
492     params->has_compress_level = true;
493     params->compress_level = s->parameters.compress_level;
494     params->has_compress_threads = true;
495     params->compress_threads = s->parameters.compress_threads;
496     params->has_decompress_threads = true;
497     params->decompress_threads = s->parameters.decompress_threads;
498     params->has_cpu_throttle_initial = true;
499     params->cpu_throttle_initial = s->parameters.cpu_throttle_initial;
500     params->has_cpu_throttle_increment = true;
501     params->cpu_throttle_increment = s->parameters.cpu_throttle_increment;
502     params->has_tls_creds = true;
503     params->tls_creds = g_strdup(s->parameters.tls_creds);
504     params->has_tls_hostname = true;
505     params->tls_hostname = g_strdup(s->parameters.tls_hostname);
506     params->has_max_bandwidth = true;
507     params->max_bandwidth = s->parameters.max_bandwidth;
508     params->has_downtime_limit = true;
509     params->downtime_limit = s->parameters.downtime_limit;
510     params->has_x_checkpoint_delay = true;
511     params->x_checkpoint_delay = s->parameters.x_checkpoint_delay;
512     params->has_block_incremental = true;
513     params->block_incremental = s->parameters.block_incremental;
514     params->has_x_multifd_channels = true;
515     params->x_multifd_channels = s->parameters.x_multifd_channels;
516     params->has_x_multifd_page_count = true;
517     params->x_multifd_page_count = s->parameters.x_multifd_page_count;
518 
519     return params;
520 }
521 
522 /*
523  * Return true if we're already in the middle of a migration
524  * (i.e. any of the active or setup states)
525  */
526 static bool migration_is_setup_or_active(int state)
527 {
528     switch (state) {
529     case MIGRATION_STATUS_ACTIVE:
530     case MIGRATION_STATUS_POSTCOPY_ACTIVE:
531     case MIGRATION_STATUS_SETUP:
532     case MIGRATION_STATUS_PRE_SWITCHOVER:
533     case MIGRATION_STATUS_DEVICE:
534         return true;
535 
536     default:
537         return false;
538 
539     }
540 }
541 
542 static void populate_ram_info(MigrationInfo *info, MigrationState *s)
543 {
544     info->has_ram = true;
545     info->ram = g_malloc0(sizeof(*info->ram));
546     info->ram->transferred = ram_counters.transferred;
547     info->ram->total = ram_bytes_total();
548     info->ram->duplicate = ram_counters.duplicate;
549     /* legacy value.  It is not used anymore */
550     info->ram->skipped = 0;
551     info->ram->normal = ram_counters.normal;
552     info->ram->normal_bytes = ram_counters.normal *
553         qemu_target_page_size();
554     info->ram->mbps = s->mbps;
555     info->ram->dirty_sync_count = ram_counters.dirty_sync_count;
556     info->ram->postcopy_requests = ram_counters.postcopy_requests;
557     info->ram->page_size = qemu_target_page_size();
558 
559     if (migrate_use_xbzrle()) {
560         info->has_xbzrle_cache = true;
561         info->xbzrle_cache = g_malloc0(sizeof(*info->xbzrle_cache));
562         info->xbzrle_cache->cache_size = migrate_xbzrle_cache_size();
563         info->xbzrle_cache->bytes = xbzrle_counters.bytes;
564         info->xbzrle_cache->pages = xbzrle_counters.pages;
565         info->xbzrle_cache->cache_miss = xbzrle_counters.cache_miss;
566         info->xbzrle_cache->cache_miss_rate = xbzrle_counters.cache_miss_rate;
567         info->xbzrle_cache->overflow = xbzrle_counters.overflow;
568     }
569 
570     if (cpu_throttle_active()) {
571         info->has_cpu_throttle_percentage = true;
572         info->cpu_throttle_percentage = cpu_throttle_get_percentage();
573     }
574 
575     if (s->state != MIGRATION_STATUS_COMPLETED) {
576         info->ram->remaining = ram_bytes_remaining();
577         info->ram->dirty_pages_rate = ram_counters.dirty_pages_rate;
578     }
579 }
580 
581 static void populate_disk_info(MigrationInfo *info)
582 {
583     if (blk_mig_active()) {
584         info->has_disk = true;
585         info->disk = g_malloc0(sizeof(*info->disk));
586         info->disk->transferred = blk_mig_bytes_transferred();
587         info->disk->remaining = blk_mig_bytes_remaining();
588         info->disk->total = blk_mig_bytes_total();
589     }
590 }
591 
592 MigrationInfo *qmp_query_migrate(Error **errp)
593 {
594     MigrationInfo *info = g_malloc0(sizeof(*info));
595     MigrationState *s = migrate_get_current();
596 
597     switch (s->state) {
598     case MIGRATION_STATUS_NONE:
599         /* no migration has happened ever */
600         break;
601     case MIGRATION_STATUS_SETUP:
602         info->has_status = true;
603         info->has_total_time = false;
604         break;
605     case MIGRATION_STATUS_ACTIVE:
606     case MIGRATION_STATUS_CANCELLING:
607     case MIGRATION_STATUS_POSTCOPY_ACTIVE:
608     case MIGRATION_STATUS_PRE_SWITCHOVER:
609     case MIGRATION_STATUS_DEVICE:
610          /* TODO add some postcopy stats */
611         info->has_status = true;
612         info->has_total_time = true;
613         info->total_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME)
614             - s->total_time;
615         info->has_expected_downtime = true;
616         info->expected_downtime = s->expected_downtime;
617         info->has_setup_time = true;
618         info->setup_time = s->setup_time;
619 
620         populate_ram_info(info, s);
621         populate_disk_info(info);
622         break;
623     case MIGRATION_STATUS_COLO:
624         info->has_status = true;
625         /* TODO: display COLO specific information (checkpoint info etc.) */
626         break;
627     case MIGRATION_STATUS_COMPLETED:
628         info->has_status = true;
629         info->has_total_time = true;
630         info->total_time = s->total_time;
631         info->has_downtime = true;
632         info->downtime = s->downtime;
633         info->has_setup_time = true;
634         info->setup_time = s->setup_time;
635 
636         populate_ram_info(info, s);
637         break;
638     case MIGRATION_STATUS_FAILED:
639         info->has_status = true;
640         if (s->error) {
641             info->has_error_desc = true;
642             info->error_desc = g_strdup(error_get_pretty(s->error));
643         }
644         break;
645     case MIGRATION_STATUS_CANCELLED:
646         info->has_status = true;
647         break;
648     }
649     info->status = s->state;
650 
651     return info;
652 }
653 
654 /**
655  * @migration_caps_check - check capability validity
656  *
657  * @cap_list: old capability list, array of bool
658  * @params: new capabilities to be applied soon
659  * @errp: set *errp if the check failed, with reason
660  *
661  * Returns true if check passed, otherwise false.
662  */
663 static bool migrate_caps_check(bool *cap_list,
664                                MigrationCapabilityStatusList *params,
665                                Error **errp)
666 {
667     MigrationCapabilityStatusList *cap;
668     bool old_postcopy_cap;
669     MigrationIncomingState *mis = migration_incoming_get_current();
670 
671     old_postcopy_cap = cap_list[MIGRATION_CAPABILITY_POSTCOPY_RAM];
672 
673     for (cap = params; cap; cap = cap->next) {
674         cap_list[cap->value->capability] = cap->value->state;
675     }
676 
677 #ifndef CONFIG_LIVE_BLOCK_MIGRATION
678     if (cap_list[MIGRATION_CAPABILITY_BLOCK]) {
679         error_setg(errp, "QEMU compiled without old-style (blk/-b, inc/-i) "
680                    "block migration");
681         error_append_hint(errp, "Use drive_mirror+NBD instead.\n");
682         return false;
683     }
684 #endif
685 
686     if (cap_list[MIGRATION_CAPABILITY_POSTCOPY_RAM]) {
687         if (cap_list[MIGRATION_CAPABILITY_COMPRESS]) {
688             /* The decompression threads asynchronously write into RAM
689              * rather than use the atomic copies needed to avoid
690              * userfaulting.  It should be possible to fix the decompression
691              * threads for compatibility in future.
692              */
693             error_setg(errp, "Postcopy is not currently compatible "
694                        "with compression");
695             return false;
696         }
697 
698         /* This check is reasonably expensive, so only when it's being
699          * set the first time, also it's only the destination that needs
700          * special support.
701          */
702         if (!old_postcopy_cap && runstate_check(RUN_STATE_INMIGRATE) &&
703             !postcopy_ram_supported_by_host(mis)) {
704             /* postcopy_ram_supported_by_host will have emitted a more
705              * detailed message
706              */
707             error_setg(errp, "Postcopy is not supported");
708             return false;
709         }
710     }
711 
712     return true;
713 }
714 
715 void qmp_migrate_set_capabilities(MigrationCapabilityStatusList *params,
716                                   Error **errp)
717 {
718     MigrationState *s = migrate_get_current();
719     MigrationCapabilityStatusList *cap;
720 
721     if (migration_is_setup_or_active(s->state)) {
722         error_setg(errp, QERR_MIGRATION_ACTIVE);
723         return;
724     }
725 
726     if (!migrate_caps_check(s->enabled_capabilities, params, errp)) {
727         return;
728     }
729 
730     for (cap = params; cap; cap = cap->next) {
731         s->enabled_capabilities[cap->value->capability] = cap->value->state;
732     }
733 }
734 
735 /*
736  * Check whether the parameters are valid. Error will be put into errp
737  * (if provided). Return true if valid, otherwise false.
738  */
739 static bool migrate_params_check(MigrationParameters *params, Error **errp)
740 {
741     if (params->has_compress_level &&
742         (params->compress_level < 0 || params->compress_level > 9)) {
743         error_setg(errp, QERR_INVALID_PARAMETER_VALUE, "compress_level",
744                    "is invalid, it should be in the range of 0 to 9");
745         return false;
746     }
747 
748     if (params->has_compress_threads &&
749         (params->compress_threads < 1 || params->compress_threads > 255)) {
750         error_setg(errp, QERR_INVALID_PARAMETER_VALUE,
751                    "compress_threads",
752                    "is invalid, it should be in the range of 1 to 255");
753         return false;
754     }
755 
756     if (params->has_decompress_threads &&
757         (params->decompress_threads < 1 || params->decompress_threads > 255)) {
758         error_setg(errp, QERR_INVALID_PARAMETER_VALUE,
759                    "decompress_threads",
760                    "is invalid, it should be in the range of 1 to 255");
761         return false;
762     }
763 
764     if (params->has_cpu_throttle_initial &&
765         (params->cpu_throttle_initial < 1 ||
766          params->cpu_throttle_initial > 99)) {
767         error_setg(errp, QERR_INVALID_PARAMETER_VALUE,
768                    "cpu_throttle_initial",
769                    "an integer in the range of 1 to 99");
770         return false;
771     }
772 
773     if (params->has_cpu_throttle_increment &&
774         (params->cpu_throttle_increment < 1 ||
775          params->cpu_throttle_increment > 99)) {
776         error_setg(errp, QERR_INVALID_PARAMETER_VALUE,
777                    "cpu_throttle_increment",
778                    "an integer in the range of 1 to 99");
779         return false;
780     }
781 
782     if (params->has_max_bandwidth &&
783         (params->max_bandwidth < 0 || params->max_bandwidth > SIZE_MAX)) {
784         error_setg(errp, "Parameter 'max_bandwidth' expects an integer in the"
785                          " range of 0 to %zu bytes/second", SIZE_MAX);
786         return false;
787     }
788 
789     if (params->has_downtime_limit &&
790         (params->downtime_limit < 0 ||
791          params->downtime_limit > MAX_MIGRATE_DOWNTIME)) {
792         error_setg(errp, "Parameter 'downtime_limit' expects an integer in "
793                          "the range of 0 to %d milliseconds",
794                          MAX_MIGRATE_DOWNTIME);
795         return false;
796     }
797 
798     if (params->has_x_checkpoint_delay && (params->x_checkpoint_delay < 0)) {
799         error_setg(errp, QERR_INVALID_PARAMETER_VALUE,
800                     "x_checkpoint_delay",
801                     "is invalid, it should be positive");
802         return false;
803     }
804     if (params->has_x_multifd_channels &&
805         (params->x_multifd_channels < 1 || params->x_multifd_channels > 255)) {
806         error_setg(errp, QERR_INVALID_PARAMETER_VALUE,
807                    "multifd_channels",
808                    "is invalid, it should be in the range of 1 to 255");
809         return false;
810     }
811     if (params->has_x_multifd_page_count &&
812             (params->x_multifd_page_count < 1 ||
813              params->x_multifd_page_count > 10000)) {
814         error_setg(errp, QERR_INVALID_PARAMETER_VALUE,
815                    "multifd_page_count",
816                    "is invalid, it should be in the range of 1 to 10000");
817         return false;
818     }
819 
820     return true;
821 }
822 
823 static void migrate_params_test_apply(MigrateSetParameters *params,
824                                       MigrationParameters *dest)
825 {
826     *dest = migrate_get_current()->parameters;
827 
828     /* TODO use QAPI_CLONE() instead of duplicating it inline */
829 
830     if (params->has_compress_level) {
831         dest->compress_level = params->compress_level;
832     }
833 
834     if (params->has_compress_threads) {
835         dest->compress_threads = params->compress_threads;
836     }
837 
838     if (params->has_decompress_threads) {
839         dest->decompress_threads = params->decompress_threads;
840     }
841 
842     if (params->has_cpu_throttle_initial) {
843         dest->cpu_throttle_initial = params->cpu_throttle_initial;
844     }
845 
846     if (params->has_cpu_throttle_increment) {
847         dest->cpu_throttle_increment = params->cpu_throttle_increment;
848     }
849 
850     if (params->has_tls_creds) {
851         assert(params->tls_creds->type == QTYPE_QSTRING);
852         dest->tls_creds = g_strdup(params->tls_creds->u.s);
853     }
854 
855     if (params->has_tls_hostname) {
856         assert(params->tls_hostname->type == QTYPE_QSTRING);
857         dest->tls_hostname = g_strdup(params->tls_hostname->u.s);
858     }
859 
860     if (params->has_max_bandwidth) {
861         dest->max_bandwidth = params->max_bandwidth;
862     }
863 
864     if (params->has_downtime_limit) {
865         dest->downtime_limit = params->downtime_limit;
866     }
867 
868     if (params->has_x_checkpoint_delay) {
869         dest->x_checkpoint_delay = params->x_checkpoint_delay;
870     }
871 
872     if (params->has_block_incremental) {
873         dest->block_incremental = params->block_incremental;
874     }
875     if (params->has_x_multifd_channels) {
876         dest->x_multifd_channels = params->x_multifd_channels;
877     }
878     if (params->has_x_multifd_page_count) {
879         dest->x_multifd_page_count = params->x_multifd_page_count;
880     }
881 }
882 
883 static void migrate_params_apply(MigrateSetParameters *params)
884 {
885     MigrationState *s = migrate_get_current();
886 
887     /* TODO use QAPI_CLONE() instead of duplicating it inline */
888 
889     if (params->has_compress_level) {
890         s->parameters.compress_level = params->compress_level;
891     }
892 
893     if (params->has_compress_threads) {
894         s->parameters.compress_threads = params->compress_threads;
895     }
896 
897     if (params->has_decompress_threads) {
898         s->parameters.decompress_threads = params->decompress_threads;
899     }
900 
901     if (params->has_cpu_throttle_initial) {
902         s->parameters.cpu_throttle_initial = params->cpu_throttle_initial;
903     }
904 
905     if (params->has_cpu_throttle_increment) {
906         s->parameters.cpu_throttle_increment = params->cpu_throttle_increment;
907     }
908 
909     if (params->has_tls_creds) {
910         g_free(s->parameters.tls_creds);
911         assert(params->tls_creds->type == QTYPE_QSTRING);
912         s->parameters.tls_creds = g_strdup(params->tls_creds->u.s);
913     }
914 
915     if (params->has_tls_hostname) {
916         g_free(s->parameters.tls_hostname);
917         assert(params->tls_hostname->type == QTYPE_QSTRING);
918         s->parameters.tls_hostname = g_strdup(params->tls_hostname->u.s);
919     }
920 
921     if (params->has_max_bandwidth) {
922         s->parameters.max_bandwidth = params->max_bandwidth;
923         if (s->to_dst_file) {
924             qemu_file_set_rate_limit(s->to_dst_file,
925                                 s->parameters.max_bandwidth / XFER_LIMIT_RATIO);
926         }
927     }
928 
929     if (params->has_downtime_limit) {
930         s->parameters.downtime_limit = params->downtime_limit;
931     }
932 
933     if (params->has_x_checkpoint_delay) {
934         s->parameters.x_checkpoint_delay = params->x_checkpoint_delay;
935         if (migration_in_colo_state()) {
936             colo_checkpoint_notify(s);
937         }
938     }
939 
940     if (params->has_block_incremental) {
941         s->parameters.block_incremental = params->block_incremental;
942     }
943     if (params->has_x_multifd_channels) {
944         s->parameters.x_multifd_channels = params->x_multifd_channels;
945     }
946     if (params->has_x_multifd_page_count) {
947         s->parameters.x_multifd_page_count = params->x_multifd_page_count;
948     }
949 }
950 
951 void qmp_migrate_set_parameters(MigrateSetParameters *params, Error **errp)
952 {
953     MigrationParameters tmp;
954 
955     /* TODO Rewrite "" to null instead */
956     if (params->has_tls_creds
957         && params->tls_creds->type == QTYPE_QNULL) {
958         QDECREF(params->tls_creds->u.n);
959         params->tls_creds->type = QTYPE_QSTRING;
960         params->tls_creds->u.s = strdup("");
961     }
962     /* TODO Rewrite "" to null instead */
963     if (params->has_tls_hostname
964         && params->tls_hostname->type == QTYPE_QNULL) {
965         QDECREF(params->tls_hostname->u.n);
966         params->tls_hostname->type = QTYPE_QSTRING;
967         params->tls_hostname->u.s = strdup("");
968     }
969 
970     migrate_params_test_apply(params, &tmp);
971 
972     if (!migrate_params_check(&tmp, errp)) {
973         /* Invalid parameter */
974         return;
975     }
976 
977     migrate_params_apply(params);
978 }
979 
980 
981 void qmp_migrate_start_postcopy(Error **errp)
982 {
983     MigrationState *s = migrate_get_current();
984 
985     if (!migrate_postcopy_ram()) {
986         error_setg(errp, "Enable postcopy with migrate_set_capability before"
987                          " the start of migration");
988         return;
989     }
990 
991     if (s->state == MIGRATION_STATUS_NONE) {
992         error_setg(errp, "Postcopy must be started after migration has been"
993                          " started");
994         return;
995     }
996     /*
997      * we don't error if migration has finished since that would be racy
998      * with issuing this command.
999      */
1000     atomic_set(&s->start_postcopy, true);
1001 }
1002 
1003 /* shared migration helpers */
1004 
1005 void migrate_set_state(int *state, int old_state, int new_state)
1006 {
1007     assert(new_state < MIGRATION_STATUS__MAX);
1008     if (atomic_cmpxchg(state, old_state, new_state) == old_state) {
1009         trace_migrate_set_state(MigrationStatus_str(new_state));
1010         migrate_generate_event(new_state);
1011     }
1012 }
1013 
1014 static MigrationCapabilityStatusList *migrate_cap_add(
1015     MigrationCapabilityStatusList *list,
1016     MigrationCapability index,
1017     bool state)
1018 {
1019     MigrationCapabilityStatusList *cap;
1020 
1021     cap = g_new0(MigrationCapabilityStatusList, 1);
1022     cap->value = g_new0(MigrationCapabilityStatus, 1);
1023     cap->value->capability = index;
1024     cap->value->state = state;
1025     cap->next = list;
1026 
1027     return cap;
1028 }
1029 
1030 void migrate_set_block_enabled(bool value, Error **errp)
1031 {
1032     MigrationCapabilityStatusList *cap;
1033 
1034     cap = migrate_cap_add(NULL, MIGRATION_CAPABILITY_BLOCK, value);
1035     qmp_migrate_set_capabilities(cap, errp);
1036     qapi_free_MigrationCapabilityStatusList(cap);
1037 }
1038 
1039 static void migrate_set_block_incremental(MigrationState *s, bool value)
1040 {
1041     s->parameters.block_incremental = value;
1042 }
1043 
1044 static void block_cleanup_parameters(MigrationState *s)
1045 {
1046     if (s->must_remove_block_options) {
1047         /* setting to false can never fail */
1048         migrate_set_block_enabled(false, &error_abort);
1049         migrate_set_block_incremental(s, false);
1050         s->must_remove_block_options = false;
1051     }
1052 }
1053 
1054 static void migrate_fd_cleanup(void *opaque)
1055 {
1056     MigrationState *s = opaque;
1057 
1058     qemu_bh_delete(s->cleanup_bh);
1059     s->cleanup_bh = NULL;
1060 
1061     if (s->to_dst_file) {
1062         Error *local_err = NULL;
1063 
1064         trace_migrate_fd_cleanup();
1065         qemu_mutex_unlock_iothread();
1066         if (s->migration_thread_running) {
1067             qemu_thread_join(&s->thread);
1068             s->migration_thread_running = false;
1069         }
1070         qemu_mutex_lock_iothread();
1071 
1072         if (multifd_save_cleanup(&local_err) != 0) {
1073             error_report_err(local_err);
1074         }
1075         qemu_fclose(s->to_dst_file);
1076         s->to_dst_file = NULL;
1077     }
1078 
1079     assert((s->state != MIGRATION_STATUS_ACTIVE) &&
1080            (s->state != MIGRATION_STATUS_POSTCOPY_ACTIVE));
1081 
1082     if (s->state == MIGRATION_STATUS_CANCELLING) {
1083         migrate_set_state(&s->state, MIGRATION_STATUS_CANCELLING,
1084                           MIGRATION_STATUS_CANCELLED);
1085     }
1086 
1087     if (s->error) {
1088         /* It is used on info migrate.  We can't free it */
1089         error_report_err(error_copy(s->error));
1090     }
1091     notifier_list_notify(&migration_state_notifiers, s);
1092     block_cleanup_parameters(s);
1093 }
1094 
1095 void migrate_set_error(MigrationState *s, const Error *error)
1096 {
1097     qemu_mutex_lock(&s->error_mutex);
1098     if (!s->error) {
1099         s->error = error_copy(error);
1100     }
1101     qemu_mutex_unlock(&s->error_mutex);
1102 }
1103 
1104 void migrate_fd_error(MigrationState *s, const Error *error)
1105 {
1106     trace_migrate_fd_error(error_get_pretty(error));
1107     assert(s->to_dst_file == NULL);
1108     migrate_set_state(&s->state, MIGRATION_STATUS_SETUP,
1109                       MIGRATION_STATUS_FAILED);
1110     migrate_set_error(s, error);
1111     notifier_list_notify(&migration_state_notifiers, s);
1112     block_cleanup_parameters(s);
1113 }
1114 
1115 static void migrate_fd_cancel(MigrationState *s)
1116 {
1117     int old_state ;
1118     QEMUFile *f = migrate_get_current()->to_dst_file;
1119     trace_migrate_fd_cancel();
1120 
1121     if (s->rp_state.from_dst_file) {
1122         /* shutdown the rp socket, so causing the rp thread to shutdown */
1123         qemu_file_shutdown(s->rp_state.from_dst_file);
1124     }
1125 
1126     do {
1127         old_state = s->state;
1128         if (!migration_is_setup_or_active(old_state)) {
1129             break;
1130         }
1131         /* If the migration is paused, kick it out of the pause */
1132         if (old_state == MIGRATION_STATUS_PRE_SWITCHOVER) {
1133             qemu_sem_post(&s->pause_sem);
1134         }
1135         migrate_set_state(&s->state, old_state, MIGRATION_STATUS_CANCELLING);
1136     } while (s->state != MIGRATION_STATUS_CANCELLING);
1137 
1138     /*
1139      * If we're unlucky the migration code might be stuck somewhere in a
1140      * send/write while the network has failed and is waiting to timeout;
1141      * if we've got shutdown(2) available then we can force it to quit.
1142      * The outgoing qemu file gets closed in migrate_fd_cleanup that is
1143      * called in a bh, so there is no race against this cancel.
1144      */
1145     if (s->state == MIGRATION_STATUS_CANCELLING && f) {
1146         qemu_file_shutdown(f);
1147     }
1148     if (s->state == MIGRATION_STATUS_CANCELLING && s->block_inactive) {
1149         Error *local_err = NULL;
1150 
1151         bdrv_invalidate_cache_all(&local_err);
1152         if (local_err) {
1153             error_report_err(local_err);
1154         } else {
1155             s->block_inactive = false;
1156         }
1157     }
1158     block_cleanup_parameters(s);
1159 }
1160 
1161 void add_migration_state_change_notifier(Notifier *notify)
1162 {
1163     notifier_list_add(&migration_state_notifiers, notify);
1164 }
1165 
1166 void remove_migration_state_change_notifier(Notifier *notify)
1167 {
1168     notifier_remove(notify);
1169 }
1170 
1171 bool migration_in_setup(MigrationState *s)
1172 {
1173     return s->state == MIGRATION_STATUS_SETUP;
1174 }
1175 
1176 bool migration_has_finished(MigrationState *s)
1177 {
1178     return s->state == MIGRATION_STATUS_COMPLETED;
1179 }
1180 
1181 bool migration_has_failed(MigrationState *s)
1182 {
1183     return (s->state == MIGRATION_STATUS_CANCELLED ||
1184             s->state == MIGRATION_STATUS_FAILED);
1185 }
1186 
1187 bool migration_in_postcopy(void)
1188 {
1189     MigrationState *s = migrate_get_current();
1190 
1191     return (s->state == MIGRATION_STATUS_POSTCOPY_ACTIVE);
1192 }
1193 
1194 bool migration_in_postcopy_after_devices(MigrationState *s)
1195 {
1196     return migration_in_postcopy() && s->postcopy_after_devices;
1197 }
1198 
1199 bool migration_is_idle(void)
1200 {
1201     MigrationState *s = migrate_get_current();
1202 
1203     switch (s->state) {
1204     case MIGRATION_STATUS_NONE:
1205     case MIGRATION_STATUS_CANCELLED:
1206     case MIGRATION_STATUS_COMPLETED:
1207     case MIGRATION_STATUS_FAILED:
1208         return true;
1209     case MIGRATION_STATUS_SETUP:
1210     case MIGRATION_STATUS_CANCELLING:
1211     case MIGRATION_STATUS_ACTIVE:
1212     case MIGRATION_STATUS_POSTCOPY_ACTIVE:
1213     case MIGRATION_STATUS_COLO:
1214     case MIGRATION_STATUS_PRE_SWITCHOVER:
1215     case MIGRATION_STATUS_DEVICE:
1216         return false;
1217     case MIGRATION_STATUS__MAX:
1218         g_assert_not_reached();
1219     }
1220 
1221     return false;
1222 }
1223 
1224 MigrationState *migrate_init(void)
1225 {
1226     MigrationState *s = migrate_get_current();
1227 
1228     /*
1229      * Reinitialise all migration state, except
1230      * parameters/capabilities that the user set, and
1231      * locks.
1232      */
1233     s->bytes_xfer = 0;
1234     s->xfer_limit = 0;
1235     s->cleanup_bh = 0;
1236     s->to_dst_file = NULL;
1237     s->state = MIGRATION_STATUS_NONE;
1238     s->rp_state.from_dst_file = NULL;
1239     s->rp_state.error = false;
1240     s->mbps = 0.0;
1241     s->downtime = 0;
1242     s->expected_downtime = 0;
1243     s->setup_time = 0;
1244     s->start_postcopy = false;
1245     s->postcopy_after_devices = false;
1246     s->migration_thread_running = false;
1247     error_free(s->error);
1248     s->error = NULL;
1249 
1250     migrate_set_state(&s->state, MIGRATION_STATUS_NONE, MIGRATION_STATUS_SETUP);
1251 
1252     s->total_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
1253     return s;
1254 }
1255 
1256 static GSList *migration_blockers;
1257 
1258 int migrate_add_blocker(Error *reason, Error **errp)
1259 {
1260     if (migrate_get_current()->only_migratable) {
1261         error_propagate(errp, error_copy(reason));
1262         error_prepend(errp, "disallowing migration blocker "
1263                           "(--only_migratable) for: ");
1264         return -EACCES;
1265     }
1266 
1267     if (migration_is_idle()) {
1268         migration_blockers = g_slist_prepend(migration_blockers, reason);
1269         return 0;
1270     }
1271 
1272     error_propagate(errp, error_copy(reason));
1273     error_prepend(errp, "disallowing migration blocker (migration in "
1274                       "progress) for: ");
1275     return -EBUSY;
1276 }
1277 
1278 void migrate_del_blocker(Error *reason)
1279 {
1280     migration_blockers = g_slist_remove(migration_blockers, reason);
1281 }
1282 
1283 void qmp_migrate_incoming(const char *uri, Error **errp)
1284 {
1285     Error *local_err = NULL;
1286     static bool once = true;
1287 
1288     if (!deferred_incoming) {
1289         error_setg(errp, "For use with '-incoming defer'");
1290         return;
1291     }
1292     if (!once) {
1293         error_setg(errp, "The incoming migration has already been started");
1294     }
1295 
1296     qemu_start_incoming_migration(uri, &local_err);
1297 
1298     if (local_err) {
1299         error_propagate(errp, local_err);
1300         return;
1301     }
1302 
1303     once = false;
1304 }
1305 
1306 bool migration_is_blocked(Error **errp)
1307 {
1308     if (qemu_savevm_state_blocked(errp)) {
1309         return true;
1310     }
1311 
1312     if (migration_blockers) {
1313         error_propagate(errp, error_copy(migration_blockers->data));
1314         return true;
1315     }
1316 
1317     return false;
1318 }
1319 
1320 void qmp_migrate(const char *uri, bool has_blk, bool blk,
1321                  bool has_inc, bool inc, bool has_detach, bool detach,
1322                  Error **errp)
1323 {
1324     Error *local_err = NULL;
1325     MigrationState *s = migrate_get_current();
1326     const char *p;
1327 
1328     if (migration_is_setup_or_active(s->state) ||
1329         s->state == MIGRATION_STATUS_CANCELLING ||
1330         s->state == MIGRATION_STATUS_COLO) {
1331         error_setg(errp, QERR_MIGRATION_ACTIVE);
1332         return;
1333     }
1334     if (runstate_check(RUN_STATE_INMIGRATE)) {
1335         error_setg(errp, "Guest is waiting for an incoming migration");
1336         return;
1337     }
1338 
1339     if (migration_is_blocked(errp)) {
1340         return;
1341     }
1342 
1343     if ((has_blk && blk) || (has_inc && inc)) {
1344         if (migrate_use_block() || migrate_use_block_incremental()) {
1345             error_setg(errp, "Command options are incompatible with "
1346                        "current migration capabilities");
1347             return;
1348         }
1349         migrate_set_block_enabled(true, &local_err);
1350         if (local_err) {
1351             error_propagate(errp, local_err);
1352             return;
1353         }
1354         s->must_remove_block_options = true;
1355     }
1356 
1357     if (has_inc && inc) {
1358         migrate_set_block_incremental(s, true);
1359     }
1360 
1361     s = migrate_init();
1362 
1363     if (strstart(uri, "tcp:", &p)) {
1364         tcp_start_outgoing_migration(s, p, &local_err);
1365 #ifdef CONFIG_RDMA
1366     } else if (strstart(uri, "rdma:", &p)) {
1367         rdma_start_outgoing_migration(s, p, &local_err);
1368 #endif
1369     } else if (strstart(uri, "exec:", &p)) {
1370         exec_start_outgoing_migration(s, p, &local_err);
1371     } else if (strstart(uri, "unix:", &p)) {
1372         unix_start_outgoing_migration(s, p, &local_err);
1373     } else if (strstart(uri, "fd:", &p)) {
1374         fd_start_outgoing_migration(s, p, &local_err);
1375     } else {
1376         error_setg(errp, QERR_INVALID_PARAMETER_VALUE, "uri",
1377                    "a valid migration protocol");
1378         migrate_set_state(&s->state, MIGRATION_STATUS_SETUP,
1379                           MIGRATION_STATUS_FAILED);
1380         return;
1381     }
1382 
1383     if (local_err) {
1384         migrate_fd_error(s, local_err);
1385         error_propagate(errp, local_err);
1386         return;
1387     }
1388 }
1389 
1390 void qmp_migrate_cancel(Error **errp)
1391 {
1392     migrate_fd_cancel(migrate_get_current());
1393 }
1394 
1395 void qmp_migrate_continue(MigrationStatus state, Error **errp)
1396 {
1397     MigrationState *s = migrate_get_current();
1398     if (s->state != state) {
1399         error_setg(errp,  "Migration not in expected state: %s",
1400                    MigrationStatus_str(s->state));
1401         return;
1402     }
1403     qemu_sem_post(&s->pause_sem);
1404 }
1405 
1406 void qmp_migrate_set_cache_size(int64_t value, Error **errp)
1407 {
1408     MigrationState *s = migrate_get_current();
1409     int64_t new_size;
1410 
1411     new_size = xbzrle_cache_resize(value, errp);
1412     if (new_size < 0) {
1413         return;
1414     }
1415 
1416     s->xbzrle_cache_size = new_size;
1417 }
1418 
1419 int64_t qmp_query_migrate_cache_size(Error **errp)
1420 {
1421     return migrate_xbzrle_cache_size();
1422 }
1423 
1424 void qmp_migrate_set_speed(int64_t value, Error **errp)
1425 {
1426     MigrateSetParameters p = {
1427         .has_max_bandwidth = true,
1428         .max_bandwidth = value,
1429     };
1430 
1431     qmp_migrate_set_parameters(&p, errp);
1432 }
1433 
1434 void qmp_migrate_set_downtime(double value, Error **errp)
1435 {
1436     if (value < 0 || value > MAX_MIGRATE_DOWNTIME_SECONDS) {
1437         error_setg(errp, "Parameter 'downtime_limit' expects an integer in "
1438                          "the range of 0 to %d seconds",
1439                          MAX_MIGRATE_DOWNTIME_SECONDS);
1440         return;
1441     }
1442 
1443     value *= 1000; /* Convert to milliseconds */
1444     value = MAX(0, MIN(INT64_MAX, value));
1445 
1446     MigrateSetParameters p = {
1447         .has_downtime_limit = true,
1448         .downtime_limit = value,
1449     };
1450 
1451     qmp_migrate_set_parameters(&p, errp);
1452 }
1453 
1454 bool migrate_release_ram(void)
1455 {
1456     MigrationState *s;
1457 
1458     s = migrate_get_current();
1459 
1460     return s->enabled_capabilities[MIGRATION_CAPABILITY_RELEASE_RAM];
1461 }
1462 
1463 bool migrate_postcopy_ram(void)
1464 {
1465     MigrationState *s;
1466 
1467     s = migrate_get_current();
1468 
1469     return s->enabled_capabilities[MIGRATION_CAPABILITY_POSTCOPY_RAM];
1470 }
1471 
1472 bool migrate_postcopy(void)
1473 {
1474     return migrate_postcopy_ram();
1475 }
1476 
1477 bool migrate_auto_converge(void)
1478 {
1479     MigrationState *s;
1480 
1481     s = migrate_get_current();
1482 
1483     return s->enabled_capabilities[MIGRATION_CAPABILITY_AUTO_CONVERGE];
1484 }
1485 
1486 bool migrate_zero_blocks(void)
1487 {
1488     MigrationState *s;
1489 
1490     s = migrate_get_current();
1491 
1492     return s->enabled_capabilities[MIGRATION_CAPABILITY_ZERO_BLOCKS];
1493 }
1494 
1495 bool migrate_use_compression(void)
1496 {
1497     MigrationState *s;
1498 
1499     s = migrate_get_current();
1500 
1501     return s->enabled_capabilities[MIGRATION_CAPABILITY_COMPRESS];
1502 }
1503 
1504 int migrate_compress_level(void)
1505 {
1506     MigrationState *s;
1507 
1508     s = migrate_get_current();
1509 
1510     return s->parameters.compress_level;
1511 }
1512 
1513 int migrate_compress_threads(void)
1514 {
1515     MigrationState *s;
1516 
1517     s = migrate_get_current();
1518 
1519     return s->parameters.compress_threads;
1520 }
1521 
1522 int migrate_decompress_threads(void)
1523 {
1524     MigrationState *s;
1525 
1526     s = migrate_get_current();
1527 
1528     return s->parameters.decompress_threads;
1529 }
1530 
1531 bool migrate_use_events(void)
1532 {
1533     MigrationState *s;
1534 
1535     s = migrate_get_current();
1536 
1537     return s->enabled_capabilities[MIGRATION_CAPABILITY_EVENTS];
1538 }
1539 
1540 bool migrate_use_multifd(void)
1541 {
1542     MigrationState *s;
1543 
1544     s = migrate_get_current();
1545 
1546     return s->enabled_capabilities[MIGRATION_CAPABILITY_X_MULTIFD];
1547 }
1548 
1549 bool migrate_pause_before_switchover(void)
1550 {
1551     MigrationState *s;
1552 
1553     s = migrate_get_current();
1554 
1555     return s->enabled_capabilities[
1556         MIGRATION_CAPABILITY_PAUSE_BEFORE_SWITCHOVER];
1557 }
1558 
1559 int migrate_multifd_channels(void)
1560 {
1561     MigrationState *s;
1562 
1563     s = migrate_get_current();
1564 
1565     return s->parameters.x_multifd_channels;
1566 }
1567 
1568 int migrate_multifd_page_count(void)
1569 {
1570     MigrationState *s;
1571 
1572     s = migrate_get_current();
1573 
1574     return s->parameters.x_multifd_page_count;
1575 }
1576 
1577 int migrate_use_xbzrle(void)
1578 {
1579     MigrationState *s;
1580 
1581     s = migrate_get_current();
1582 
1583     return s->enabled_capabilities[MIGRATION_CAPABILITY_XBZRLE];
1584 }
1585 
1586 int64_t migrate_xbzrle_cache_size(void)
1587 {
1588     MigrationState *s;
1589 
1590     s = migrate_get_current();
1591 
1592     return s->xbzrle_cache_size;
1593 }
1594 
1595 bool migrate_use_block(void)
1596 {
1597     MigrationState *s;
1598 
1599     s = migrate_get_current();
1600 
1601     return s->enabled_capabilities[MIGRATION_CAPABILITY_BLOCK];
1602 }
1603 
1604 bool migrate_use_return_path(void)
1605 {
1606     MigrationState *s;
1607 
1608     s = migrate_get_current();
1609 
1610     return s->enabled_capabilities[MIGRATION_CAPABILITY_RETURN_PATH];
1611 }
1612 
1613 bool migrate_use_block_incremental(void)
1614 {
1615     MigrationState *s;
1616 
1617     s = migrate_get_current();
1618 
1619     return s->parameters.block_incremental;
1620 }
1621 
1622 /* migration thread support */
1623 /*
1624  * Something bad happened to the RP stream, mark an error
1625  * The caller shall print or trace something to indicate why
1626  */
1627 static void mark_source_rp_bad(MigrationState *s)
1628 {
1629     s->rp_state.error = true;
1630 }
1631 
1632 static struct rp_cmd_args {
1633     ssize_t     len; /* -1 = variable */
1634     const char *name;
1635 } rp_cmd_args[] = {
1636     [MIG_RP_MSG_INVALID]        = { .len = -1, .name = "INVALID" },
1637     [MIG_RP_MSG_SHUT]           = { .len =  4, .name = "SHUT" },
1638     [MIG_RP_MSG_PONG]           = { .len =  4, .name = "PONG" },
1639     [MIG_RP_MSG_REQ_PAGES]      = { .len = 12, .name = "REQ_PAGES" },
1640     [MIG_RP_MSG_REQ_PAGES_ID]   = { .len = -1, .name = "REQ_PAGES_ID" },
1641     [MIG_RP_MSG_MAX]            = { .len = -1, .name = "MAX" },
1642 };
1643 
1644 /*
1645  * Process a request for pages received on the return path,
1646  * We're allowed to send more than requested (e.g. to round to our page size)
1647  * and we don't need to send pages that have already been sent.
1648  */
1649 static void migrate_handle_rp_req_pages(MigrationState *ms, const char* rbname,
1650                                        ram_addr_t start, size_t len)
1651 {
1652     long our_host_ps = getpagesize();
1653 
1654     trace_migrate_handle_rp_req_pages(rbname, start, len);
1655 
1656     /*
1657      * Since we currently insist on matching page sizes, just sanity check
1658      * we're being asked for whole host pages.
1659      */
1660     if (start & (our_host_ps-1) ||
1661        (len & (our_host_ps-1))) {
1662         error_report("%s: Misaligned page request, start: " RAM_ADDR_FMT
1663                      " len: %zd", __func__, start, len);
1664         mark_source_rp_bad(ms);
1665         return;
1666     }
1667 
1668     if (ram_save_queue_pages(rbname, start, len)) {
1669         mark_source_rp_bad(ms);
1670     }
1671 }
1672 
1673 /*
1674  * Handles messages sent on the return path towards the source VM
1675  *
1676  */
1677 static void *source_return_path_thread(void *opaque)
1678 {
1679     MigrationState *ms = opaque;
1680     QEMUFile *rp = ms->rp_state.from_dst_file;
1681     uint16_t header_len, header_type;
1682     uint8_t buf[512];
1683     uint32_t tmp32, sibling_error;
1684     ram_addr_t start = 0; /* =0 to silence warning */
1685     size_t  len = 0, expected_len;
1686     int res;
1687 
1688     trace_source_return_path_thread_entry();
1689     while (!ms->rp_state.error && !qemu_file_get_error(rp) &&
1690            migration_is_setup_or_active(ms->state)) {
1691         trace_source_return_path_thread_loop_top();
1692         header_type = qemu_get_be16(rp);
1693         header_len = qemu_get_be16(rp);
1694 
1695         if (header_type >= MIG_RP_MSG_MAX ||
1696             header_type == MIG_RP_MSG_INVALID) {
1697             error_report("RP: Received invalid message 0x%04x length 0x%04x",
1698                     header_type, header_len);
1699             mark_source_rp_bad(ms);
1700             goto out;
1701         }
1702 
1703         if ((rp_cmd_args[header_type].len != -1 &&
1704             header_len != rp_cmd_args[header_type].len) ||
1705             header_len > sizeof(buf)) {
1706             error_report("RP: Received '%s' message (0x%04x) with"
1707                     "incorrect length %d expecting %zu",
1708                     rp_cmd_args[header_type].name, header_type, header_len,
1709                     (size_t)rp_cmd_args[header_type].len);
1710             mark_source_rp_bad(ms);
1711             goto out;
1712         }
1713 
1714         /* We know we've got a valid header by this point */
1715         res = qemu_get_buffer(rp, buf, header_len);
1716         if (res != header_len) {
1717             error_report("RP: Failed reading data for message 0x%04x"
1718                          " read %d expected %d",
1719                          header_type, res, header_len);
1720             mark_source_rp_bad(ms);
1721             goto out;
1722         }
1723 
1724         /* OK, we have the message and the data */
1725         switch (header_type) {
1726         case MIG_RP_MSG_SHUT:
1727             sibling_error = ldl_be_p(buf);
1728             trace_source_return_path_thread_shut(sibling_error);
1729             if (sibling_error) {
1730                 error_report("RP: Sibling indicated error %d", sibling_error);
1731                 mark_source_rp_bad(ms);
1732             }
1733             /*
1734              * We'll let the main thread deal with closing the RP
1735              * we could do a shutdown(2) on it, but we're the only user
1736              * anyway, so there's nothing gained.
1737              */
1738             goto out;
1739 
1740         case MIG_RP_MSG_PONG:
1741             tmp32 = ldl_be_p(buf);
1742             trace_source_return_path_thread_pong(tmp32);
1743             break;
1744 
1745         case MIG_RP_MSG_REQ_PAGES:
1746             start = ldq_be_p(buf);
1747             len = ldl_be_p(buf + 8);
1748             migrate_handle_rp_req_pages(ms, NULL, start, len);
1749             break;
1750 
1751         case MIG_RP_MSG_REQ_PAGES_ID:
1752             expected_len = 12 + 1; /* header + termination */
1753 
1754             if (header_len >= expected_len) {
1755                 start = ldq_be_p(buf);
1756                 len = ldl_be_p(buf + 8);
1757                 /* Now we expect an idstr */
1758                 tmp32 = buf[12]; /* Length of the following idstr */
1759                 buf[13 + tmp32] = '\0';
1760                 expected_len += tmp32;
1761             }
1762             if (header_len != expected_len) {
1763                 error_report("RP: Req_Page_id with length %d expecting %zd",
1764                         header_len, expected_len);
1765                 mark_source_rp_bad(ms);
1766                 goto out;
1767             }
1768             migrate_handle_rp_req_pages(ms, (char *)&buf[13], start, len);
1769             break;
1770 
1771         default:
1772             break;
1773         }
1774     }
1775     if (qemu_file_get_error(rp)) {
1776         trace_source_return_path_thread_bad_end();
1777         mark_source_rp_bad(ms);
1778     }
1779 
1780     trace_source_return_path_thread_end();
1781 out:
1782     ms->rp_state.from_dst_file = NULL;
1783     qemu_fclose(rp);
1784     return NULL;
1785 }
1786 
1787 static int open_return_path_on_source(MigrationState *ms)
1788 {
1789 
1790     ms->rp_state.from_dst_file = qemu_file_get_return_path(ms->to_dst_file);
1791     if (!ms->rp_state.from_dst_file) {
1792         return -1;
1793     }
1794 
1795     trace_open_return_path_on_source();
1796     qemu_thread_create(&ms->rp_state.rp_thread, "return path",
1797                        source_return_path_thread, ms, QEMU_THREAD_JOINABLE);
1798 
1799     trace_open_return_path_on_source_continue();
1800 
1801     return 0;
1802 }
1803 
1804 /* Returns 0 if the RP was ok, otherwise there was an error on the RP */
1805 static int await_return_path_close_on_source(MigrationState *ms)
1806 {
1807     /*
1808      * If this is a normal exit then the destination will send a SHUT and the
1809      * rp_thread will exit, however if there's an error we need to cause
1810      * it to exit.
1811      */
1812     if (qemu_file_get_error(ms->to_dst_file) && ms->rp_state.from_dst_file) {
1813         /*
1814          * shutdown(2), if we have it, will cause it to unblock if it's stuck
1815          * waiting for the destination.
1816          */
1817         qemu_file_shutdown(ms->rp_state.from_dst_file);
1818         mark_source_rp_bad(ms);
1819     }
1820     trace_await_return_path_close_on_source_joining();
1821     qemu_thread_join(&ms->rp_state.rp_thread);
1822     trace_await_return_path_close_on_source_close();
1823     return ms->rp_state.error;
1824 }
1825 
1826 /*
1827  * Switch from normal iteration to postcopy
1828  * Returns non-0 on error
1829  */
1830 static int postcopy_start(MigrationState *ms, bool *old_vm_running)
1831 {
1832     int ret;
1833     QIOChannelBuffer *bioc;
1834     QEMUFile *fb;
1835     int64_t time_at_stop = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
1836     bool restart_block = false;
1837     int cur_state = MIGRATION_STATUS_ACTIVE;
1838     if (!migrate_pause_before_switchover()) {
1839         migrate_set_state(&ms->state, MIGRATION_STATUS_ACTIVE,
1840                           MIGRATION_STATUS_POSTCOPY_ACTIVE);
1841     }
1842 
1843     trace_postcopy_start();
1844     qemu_mutex_lock_iothread();
1845     trace_postcopy_start_set_run();
1846 
1847     qemu_system_wakeup_request(QEMU_WAKEUP_REASON_OTHER);
1848     *old_vm_running = runstate_is_running();
1849     global_state_store();
1850     ret = vm_stop_force_state(RUN_STATE_FINISH_MIGRATE);
1851     if (ret < 0) {
1852         goto fail;
1853     }
1854 
1855     ret = migration_maybe_pause(ms, &cur_state,
1856                                 MIGRATION_STATUS_POSTCOPY_ACTIVE);
1857     if (ret < 0) {
1858         goto fail;
1859     }
1860 
1861     ret = bdrv_inactivate_all();
1862     if (ret < 0) {
1863         goto fail;
1864     }
1865     restart_block = true;
1866 
1867     /*
1868      * Cause any non-postcopiable, but iterative devices to
1869      * send out their final data.
1870      */
1871     qemu_savevm_state_complete_precopy(ms->to_dst_file, true, false);
1872 
1873     /*
1874      * in Finish migrate and with the io-lock held everything should
1875      * be quiet, but we've potentially still got dirty pages and we
1876      * need to tell the destination to throw any pages it's already received
1877      * that are dirty
1878      */
1879     if (migrate_postcopy_ram()) {
1880         if (ram_postcopy_send_discard_bitmap(ms)) {
1881             error_report("postcopy send discard bitmap failed");
1882             goto fail;
1883         }
1884     }
1885 
1886     /*
1887      * send rest of state - note things that are doing postcopy
1888      * will notice we're in POSTCOPY_ACTIVE and not actually
1889      * wrap their state up here
1890      */
1891     qemu_file_set_rate_limit(ms->to_dst_file, INT64_MAX);
1892     if (migrate_postcopy_ram()) {
1893         /* Ping just for debugging, helps line traces up */
1894         qemu_savevm_send_ping(ms->to_dst_file, 2);
1895     }
1896 
1897     /*
1898      * While loading the device state we may trigger page transfer
1899      * requests and the fd must be free to process those, and thus
1900      * the destination must read the whole device state off the fd before
1901      * it starts processing it.  Unfortunately the ad-hoc migration format
1902      * doesn't allow the destination to know the size to read without fully
1903      * parsing it through each devices load-state code (especially the open
1904      * coded devices that use get/put).
1905      * So we wrap the device state up in a package with a length at the start;
1906      * to do this we use a qemu_buf to hold the whole of the device state.
1907      */
1908     bioc = qio_channel_buffer_new(4096);
1909     qio_channel_set_name(QIO_CHANNEL(bioc), "migration-postcopy-buffer");
1910     fb = qemu_fopen_channel_output(QIO_CHANNEL(bioc));
1911     object_unref(OBJECT(bioc));
1912 
1913     /*
1914      * Make sure the receiver can get incoming pages before we send the rest
1915      * of the state
1916      */
1917     qemu_savevm_send_postcopy_listen(fb);
1918 
1919     qemu_savevm_state_complete_precopy(fb, false, false);
1920     if (migrate_postcopy_ram()) {
1921         qemu_savevm_send_ping(fb, 3);
1922     }
1923 
1924     qemu_savevm_send_postcopy_run(fb);
1925 
1926     /* <><> end of stuff going into the package */
1927 
1928     /* Last point of recovery; as soon as we send the package the destination
1929      * can open devices and potentially start running.
1930      * Lets just check again we've not got any errors.
1931      */
1932     ret = qemu_file_get_error(ms->to_dst_file);
1933     if (ret) {
1934         error_report("postcopy_start: Migration stream errored (pre package)");
1935         goto fail_closefb;
1936     }
1937 
1938     restart_block = false;
1939 
1940     /* Now send that blob */
1941     if (qemu_savevm_send_packaged(ms->to_dst_file, bioc->data, bioc->usage)) {
1942         goto fail_closefb;
1943     }
1944     qemu_fclose(fb);
1945 
1946     /* Send a notify to give a chance for anything that needs to happen
1947      * at the transition to postcopy and after the device state; in particular
1948      * spice needs to trigger a transition now
1949      */
1950     ms->postcopy_after_devices = true;
1951     notifier_list_notify(&migration_state_notifiers, ms);
1952 
1953     ms->downtime =  qemu_clock_get_ms(QEMU_CLOCK_REALTIME) - time_at_stop;
1954 
1955     qemu_mutex_unlock_iothread();
1956 
1957     if (migrate_postcopy_ram()) {
1958         /*
1959          * Although this ping is just for debug, it could potentially be
1960          * used for getting a better measurement of downtime at the source.
1961          */
1962         qemu_savevm_send_ping(ms->to_dst_file, 4);
1963     }
1964 
1965     if (migrate_release_ram()) {
1966         ram_postcopy_migrated_memory_release(ms);
1967     }
1968 
1969     ret = qemu_file_get_error(ms->to_dst_file);
1970     if (ret) {
1971         error_report("postcopy_start: Migration stream errored");
1972         migrate_set_state(&ms->state, MIGRATION_STATUS_POSTCOPY_ACTIVE,
1973                               MIGRATION_STATUS_FAILED);
1974     }
1975 
1976     return ret;
1977 
1978 fail_closefb:
1979     qemu_fclose(fb);
1980 fail:
1981     migrate_set_state(&ms->state, MIGRATION_STATUS_POSTCOPY_ACTIVE,
1982                           MIGRATION_STATUS_FAILED);
1983     if (restart_block) {
1984         /* A failure happened early enough that we know the destination hasn't
1985          * accessed block devices, so we're safe to recover.
1986          */
1987         Error *local_err = NULL;
1988 
1989         bdrv_invalidate_cache_all(&local_err);
1990         if (local_err) {
1991             error_report_err(local_err);
1992         }
1993     }
1994     qemu_mutex_unlock_iothread();
1995     return -1;
1996 }
1997 
1998 /**
1999  * migration_maybe_pause: Pause if required to by
2000  * migrate_pause_before_switchover called with the iothread locked
2001  * Returns: 0 on success
2002  */
2003 static int migration_maybe_pause(MigrationState *s,
2004                                  int *current_active_state,
2005                                  int new_state)
2006 {
2007     if (!migrate_pause_before_switchover()) {
2008         return 0;
2009     }
2010 
2011     /* Since leaving this state is not atomic with posting the semaphore
2012      * it's possible that someone could have issued multiple migrate_continue
2013      * and the semaphore is incorrectly positive at this point;
2014      * the docs say it's undefined to reinit a semaphore that's already
2015      * init'd, so use timedwait to eat up any existing posts.
2016      */
2017     while (qemu_sem_timedwait(&s->pause_sem, 1) == 0) {
2018         /* This block intentionally left blank */
2019     }
2020 
2021     qemu_mutex_unlock_iothread();
2022     migrate_set_state(&s->state, *current_active_state,
2023                       MIGRATION_STATUS_PRE_SWITCHOVER);
2024     qemu_sem_wait(&s->pause_sem);
2025     migrate_set_state(&s->state, MIGRATION_STATUS_PRE_SWITCHOVER,
2026                       new_state);
2027     *current_active_state = new_state;
2028     qemu_mutex_lock_iothread();
2029 
2030     return s->state == new_state ? 0 : -EINVAL;
2031 }
2032 
2033 /**
2034  * migration_completion: Used by migration_thread when there's not much left.
2035  *   The caller 'breaks' the loop when this returns.
2036  *
2037  * @s: Current migration state
2038  * @current_active_state: The migration state we expect to be in
2039  * @*old_vm_running: Pointer to old_vm_running flag
2040  * @*start_time: Pointer to time to update
2041  */
2042 static void migration_completion(MigrationState *s, int current_active_state,
2043                                  bool *old_vm_running,
2044                                  int64_t *start_time)
2045 {
2046     int ret;
2047 
2048     if (s->state == MIGRATION_STATUS_ACTIVE) {
2049         qemu_mutex_lock_iothread();
2050         *start_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
2051         qemu_system_wakeup_request(QEMU_WAKEUP_REASON_OTHER);
2052         *old_vm_running = runstate_is_running();
2053         ret = global_state_store();
2054 
2055         if (!ret) {
2056             bool inactivate = !migrate_colo_enabled();
2057             ret = vm_stop_force_state(RUN_STATE_FINISH_MIGRATE);
2058             if (ret >= 0) {
2059                 ret = migration_maybe_pause(s, &current_active_state,
2060                                             MIGRATION_STATUS_DEVICE);
2061             }
2062             if (ret >= 0) {
2063                 qemu_file_set_rate_limit(s->to_dst_file, INT64_MAX);
2064                 ret = qemu_savevm_state_complete_precopy(s->to_dst_file, false,
2065                                                          inactivate);
2066             }
2067             if (inactivate && ret >= 0) {
2068                 s->block_inactive = true;
2069             }
2070         }
2071         qemu_mutex_unlock_iothread();
2072 
2073         if (ret < 0) {
2074             goto fail;
2075         }
2076     } else if (s->state == MIGRATION_STATUS_POSTCOPY_ACTIVE) {
2077         trace_migration_completion_postcopy_end();
2078 
2079         qemu_savevm_state_complete_postcopy(s->to_dst_file);
2080         trace_migration_completion_postcopy_end_after_complete();
2081     }
2082 
2083     /*
2084      * If rp was opened we must clean up the thread before
2085      * cleaning everything else up (since if there are no failures
2086      * it will wait for the destination to send it's status in
2087      * a SHUT command).
2088      */
2089     if (s->rp_state.from_dst_file) {
2090         int rp_error;
2091         trace_migration_return_path_end_before();
2092         rp_error = await_return_path_close_on_source(s);
2093         trace_migration_return_path_end_after(rp_error);
2094         if (rp_error) {
2095             goto fail_invalidate;
2096         }
2097     }
2098 
2099     if (qemu_file_get_error(s->to_dst_file)) {
2100         trace_migration_completion_file_err();
2101         goto fail_invalidate;
2102     }
2103 
2104     if (!migrate_colo_enabled()) {
2105         migrate_set_state(&s->state, current_active_state,
2106                           MIGRATION_STATUS_COMPLETED);
2107     }
2108 
2109     return;
2110 
2111 fail_invalidate:
2112     /* If not doing postcopy, vm_start() will be called: let's regain
2113      * control on images.
2114      */
2115     if (s->state == MIGRATION_STATUS_ACTIVE) {
2116         Error *local_err = NULL;
2117 
2118         qemu_mutex_lock_iothread();
2119         bdrv_invalidate_cache_all(&local_err);
2120         if (local_err) {
2121             error_report_err(local_err);
2122         } else {
2123             s->block_inactive = false;
2124         }
2125         qemu_mutex_unlock_iothread();
2126     }
2127 
2128 fail:
2129     migrate_set_state(&s->state, current_active_state,
2130                       MIGRATION_STATUS_FAILED);
2131 }
2132 
2133 bool migrate_colo_enabled(void)
2134 {
2135     MigrationState *s = migrate_get_current();
2136     return s->enabled_capabilities[MIGRATION_CAPABILITY_X_COLO];
2137 }
2138 
2139 /*
2140  * Master migration thread on the source VM.
2141  * It drives the migration and pumps the data down the outgoing channel.
2142  */
2143 static void *migration_thread(void *opaque)
2144 {
2145     MigrationState *s = opaque;
2146     /* Used by the bandwidth calcs, updated later */
2147     int64_t initial_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
2148     int64_t setup_start = qemu_clock_get_ms(QEMU_CLOCK_HOST);
2149     int64_t initial_bytes = 0;
2150     /*
2151      * The final stage happens when the remaining data is smaller than
2152      * this threshold; it's calculated from the requested downtime and
2153      * measured bandwidth
2154      */
2155     int64_t threshold_size = 0;
2156     int64_t start_time = initial_time;
2157     int64_t end_time;
2158     bool old_vm_running = false;
2159     bool entered_postcopy = false;
2160     /* The active state we expect to be in; ACTIVE or POSTCOPY_ACTIVE */
2161     enum MigrationStatus current_active_state = MIGRATION_STATUS_ACTIVE;
2162     bool enable_colo = migrate_colo_enabled();
2163 
2164     rcu_register_thread();
2165 
2166     qemu_savevm_state_header(s->to_dst_file);
2167 
2168     /*
2169      * If we opened the return path, we need to make sure dst has it
2170      * opened as well.
2171      */
2172     if (s->rp_state.from_dst_file) {
2173         /* Now tell the dest that it should open its end so it can reply */
2174         qemu_savevm_send_open_return_path(s->to_dst_file);
2175 
2176         /* And do a ping that will make stuff easier to debug */
2177         qemu_savevm_send_ping(s->to_dst_file, 1);
2178     }
2179 
2180     if (migrate_postcopy()) {
2181         /*
2182          * Tell the destination that we *might* want to do postcopy later;
2183          * if the other end can't do postcopy it should fail now, nice and
2184          * early.
2185          */
2186         qemu_savevm_send_postcopy_advise(s->to_dst_file);
2187     }
2188 
2189     qemu_savevm_state_setup(s->to_dst_file);
2190 
2191     s->setup_time = qemu_clock_get_ms(QEMU_CLOCK_HOST) - setup_start;
2192     migrate_set_state(&s->state, MIGRATION_STATUS_SETUP,
2193                       MIGRATION_STATUS_ACTIVE);
2194 
2195     trace_migration_thread_setup_complete();
2196 
2197     while (s->state == MIGRATION_STATUS_ACTIVE ||
2198            s->state == MIGRATION_STATUS_POSTCOPY_ACTIVE) {
2199         int64_t current_time;
2200         uint64_t pending_size;
2201 
2202         if (!qemu_file_rate_limit(s->to_dst_file)) {
2203             uint64_t pend_post, pend_nonpost;
2204 
2205             qemu_savevm_state_pending(s->to_dst_file, threshold_size,
2206                                       &pend_nonpost, &pend_post);
2207             pending_size = pend_nonpost + pend_post;
2208             trace_migrate_pending(pending_size, threshold_size,
2209                                   pend_post, pend_nonpost);
2210             if (pending_size && pending_size >= threshold_size) {
2211                 /* Still a significant amount to transfer */
2212 
2213                 if (migrate_postcopy() &&
2214                     s->state != MIGRATION_STATUS_POSTCOPY_ACTIVE &&
2215                     pend_nonpost <= threshold_size &&
2216                     atomic_read(&s->start_postcopy)) {
2217 
2218                     if (!postcopy_start(s, &old_vm_running)) {
2219                         current_active_state = MIGRATION_STATUS_POSTCOPY_ACTIVE;
2220                         entered_postcopy = true;
2221                     }
2222 
2223                     continue;
2224                 }
2225                 /* Just another iteration step */
2226                 qemu_savevm_state_iterate(s->to_dst_file, entered_postcopy);
2227             } else {
2228                 trace_migration_thread_low_pending(pending_size);
2229                 migration_completion(s, current_active_state,
2230                                      &old_vm_running, &start_time);
2231                 break;
2232             }
2233         }
2234 
2235         if (qemu_file_get_error(s->to_dst_file)) {
2236             migrate_set_state(&s->state, current_active_state,
2237                               MIGRATION_STATUS_FAILED);
2238             trace_migration_thread_file_err();
2239             break;
2240         }
2241         current_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
2242         if (current_time >= initial_time + BUFFER_DELAY) {
2243             uint64_t transferred_bytes = qemu_ftell(s->to_dst_file) -
2244                                          initial_bytes;
2245             uint64_t time_spent = current_time - initial_time;
2246             double bandwidth = (double)transferred_bytes / time_spent;
2247             threshold_size = bandwidth * s->parameters.downtime_limit;
2248 
2249             s->mbps = (((double) transferred_bytes * 8.0) /
2250                     ((double) time_spent / 1000.0)) / 1000.0 / 1000.0;
2251 
2252             trace_migrate_transferred(transferred_bytes, time_spent,
2253                                       bandwidth, threshold_size);
2254             /* if we haven't sent anything, we don't want to recalculate
2255                10000 is a small enough number for our purposes */
2256             if (ram_counters.dirty_pages_rate && transferred_bytes > 10000) {
2257                 s->expected_downtime = ram_counters.dirty_pages_rate *
2258                     qemu_target_page_size() / bandwidth;
2259             }
2260 
2261             qemu_file_reset_rate_limit(s->to_dst_file);
2262             initial_time = current_time;
2263             initial_bytes = qemu_ftell(s->to_dst_file);
2264         }
2265         if (qemu_file_rate_limit(s->to_dst_file)) {
2266             /* usleep expects microseconds */
2267             g_usleep((initial_time + BUFFER_DELAY - current_time)*1000);
2268         }
2269     }
2270 
2271     trace_migration_thread_after_loop();
2272     /* If we enabled cpu throttling for auto-converge, turn it off. */
2273     cpu_throttle_stop();
2274     end_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
2275 
2276     qemu_mutex_lock_iothread();
2277     /*
2278      * The resource has been allocated by migration will be reused in COLO
2279      * process, so don't release them.
2280      */
2281     if (!enable_colo) {
2282         qemu_savevm_state_cleanup();
2283     }
2284     if (s->state == MIGRATION_STATUS_COMPLETED) {
2285         uint64_t transferred_bytes = qemu_ftell(s->to_dst_file);
2286         s->total_time = end_time - s->total_time;
2287         if (!entered_postcopy) {
2288             s->downtime = end_time - start_time;
2289         }
2290         if (s->total_time) {
2291             s->mbps = (((double) transferred_bytes * 8.0) /
2292                        ((double) s->total_time)) / 1000;
2293         }
2294         runstate_set(RUN_STATE_POSTMIGRATE);
2295     } else {
2296         if (s->state == MIGRATION_STATUS_ACTIVE && enable_colo) {
2297             migrate_start_colo_process(s);
2298             qemu_savevm_state_cleanup();
2299             /*
2300             * Fixme: we will run VM in COLO no matter its old running state.
2301             * After exited COLO, we will keep running.
2302             */
2303             old_vm_running = true;
2304         }
2305         if (old_vm_running && !entered_postcopy) {
2306             vm_start();
2307         } else {
2308             if (runstate_check(RUN_STATE_FINISH_MIGRATE)) {
2309                 runstate_set(RUN_STATE_POSTMIGRATE);
2310             }
2311         }
2312     }
2313     qemu_bh_schedule(s->cleanup_bh);
2314     qemu_mutex_unlock_iothread();
2315 
2316     rcu_unregister_thread();
2317     return NULL;
2318 }
2319 
2320 void migrate_fd_connect(MigrationState *s)
2321 {
2322     s->expected_downtime = s->parameters.downtime_limit;
2323     s->cleanup_bh = qemu_bh_new(migrate_fd_cleanup, s);
2324 
2325     qemu_file_set_blocking(s->to_dst_file, true);
2326     qemu_file_set_rate_limit(s->to_dst_file,
2327                              s->parameters.max_bandwidth / XFER_LIMIT_RATIO);
2328 
2329     /* Notify before starting migration thread */
2330     notifier_list_notify(&migration_state_notifiers, s);
2331 
2332     /*
2333      * Open the return path. For postcopy, it is used exclusively. For
2334      * precopy, only if user specified "return-path" capability would
2335      * QEMU uses the return path.
2336      */
2337     if (migrate_postcopy_ram() || migrate_use_return_path()) {
2338         if (open_return_path_on_source(s)) {
2339             error_report("Unable to open return-path for postcopy");
2340             migrate_set_state(&s->state, MIGRATION_STATUS_SETUP,
2341                               MIGRATION_STATUS_FAILED);
2342             migrate_fd_cleanup(s);
2343             return;
2344         }
2345     }
2346 
2347     if (multifd_save_setup() != 0) {
2348         migrate_set_state(&s->state, MIGRATION_STATUS_SETUP,
2349                           MIGRATION_STATUS_FAILED);
2350         migrate_fd_cleanup(s);
2351         return;
2352     }
2353     qemu_thread_create(&s->thread, "live_migration", migration_thread, s,
2354                        QEMU_THREAD_JOINABLE);
2355     s->migration_thread_running = true;
2356 }
2357 
2358 void migration_global_dump(Monitor *mon)
2359 {
2360     MigrationState *ms = migrate_get_current();
2361 
2362     monitor_printf(mon, "globals: store-global-state=%d, only_migratable=%d, "
2363                    "send-configuration=%d, send-section-footer=%d\n",
2364                    ms->store_global_state, ms->only_migratable,
2365                    ms->send_configuration, ms->send_section_footer);
2366 }
2367 
2368 #define DEFINE_PROP_MIG_CAP(name, x)             \
2369     DEFINE_PROP_BOOL(name, MigrationState, enabled_capabilities[x], false)
2370 
2371 static Property migration_properties[] = {
2372     DEFINE_PROP_BOOL("store-global-state", MigrationState,
2373                      store_global_state, true),
2374     DEFINE_PROP_BOOL("only-migratable", MigrationState, only_migratable, false),
2375     DEFINE_PROP_BOOL("send-configuration", MigrationState,
2376                      send_configuration, true),
2377     DEFINE_PROP_BOOL("send-section-footer", MigrationState,
2378                      send_section_footer, true),
2379 
2380     /* Migration parameters */
2381     DEFINE_PROP_INT64("x-compress-level", MigrationState,
2382                       parameters.compress_level,
2383                       DEFAULT_MIGRATE_COMPRESS_LEVEL),
2384     DEFINE_PROP_INT64("x-compress-threads", MigrationState,
2385                       parameters.compress_threads,
2386                       DEFAULT_MIGRATE_COMPRESS_THREAD_COUNT),
2387     DEFINE_PROP_INT64("x-decompress-threads", MigrationState,
2388                       parameters.decompress_threads,
2389                       DEFAULT_MIGRATE_DECOMPRESS_THREAD_COUNT),
2390     DEFINE_PROP_INT64("x-cpu-throttle-initial", MigrationState,
2391                       parameters.cpu_throttle_initial,
2392                       DEFAULT_MIGRATE_CPU_THROTTLE_INITIAL),
2393     DEFINE_PROP_INT64("x-cpu-throttle-increment", MigrationState,
2394                       parameters.cpu_throttle_increment,
2395                       DEFAULT_MIGRATE_CPU_THROTTLE_INCREMENT),
2396     DEFINE_PROP_INT64("x-max-bandwidth", MigrationState,
2397                       parameters.max_bandwidth, MAX_THROTTLE),
2398     DEFINE_PROP_INT64("x-downtime-limit", MigrationState,
2399                       parameters.downtime_limit,
2400                       DEFAULT_MIGRATE_SET_DOWNTIME),
2401     DEFINE_PROP_INT64("x-checkpoint-delay", MigrationState,
2402                       parameters.x_checkpoint_delay,
2403                       DEFAULT_MIGRATE_X_CHECKPOINT_DELAY),
2404     DEFINE_PROP_INT64("x-multifd-channels", MigrationState,
2405                       parameters.x_multifd_channels,
2406                       DEFAULT_MIGRATE_MULTIFD_CHANNELS),
2407     DEFINE_PROP_INT64("x-multifd-page-count", MigrationState,
2408                       parameters.x_multifd_page_count,
2409                       DEFAULT_MIGRATE_MULTIFD_PAGE_COUNT),
2410 
2411     /* Migration capabilities */
2412     DEFINE_PROP_MIG_CAP("x-xbzrle", MIGRATION_CAPABILITY_XBZRLE),
2413     DEFINE_PROP_MIG_CAP("x-rdma-pin-all", MIGRATION_CAPABILITY_RDMA_PIN_ALL),
2414     DEFINE_PROP_MIG_CAP("x-auto-converge", MIGRATION_CAPABILITY_AUTO_CONVERGE),
2415     DEFINE_PROP_MIG_CAP("x-zero-blocks", MIGRATION_CAPABILITY_ZERO_BLOCKS),
2416     DEFINE_PROP_MIG_CAP("x-compress", MIGRATION_CAPABILITY_COMPRESS),
2417     DEFINE_PROP_MIG_CAP("x-events", MIGRATION_CAPABILITY_EVENTS),
2418     DEFINE_PROP_MIG_CAP("x-postcopy-ram", MIGRATION_CAPABILITY_POSTCOPY_RAM),
2419     DEFINE_PROP_MIG_CAP("x-colo", MIGRATION_CAPABILITY_X_COLO),
2420     DEFINE_PROP_MIG_CAP("x-release-ram", MIGRATION_CAPABILITY_RELEASE_RAM),
2421     DEFINE_PROP_MIG_CAP("x-block", MIGRATION_CAPABILITY_BLOCK),
2422     DEFINE_PROP_MIG_CAP("x-return-path", MIGRATION_CAPABILITY_RETURN_PATH),
2423     DEFINE_PROP_MIG_CAP("x-multifd", MIGRATION_CAPABILITY_X_MULTIFD),
2424 
2425     DEFINE_PROP_END_OF_LIST(),
2426 };
2427 
2428 static void migration_class_init(ObjectClass *klass, void *data)
2429 {
2430     DeviceClass *dc = DEVICE_CLASS(klass);
2431 
2432     dc->user_creatable = false;
2433     dc->props = migration_properties;
2434 }
2435 
2436 static void migration_instance_finalize(Object *obj)
2437 {
2438     MigrationState *ms = MIGRATION_OBJ(obj);
2439     MigrationParameters *params = &ms->parameters;
2440 
2441     qemu_mutex_destroy(&ms->error_mutex);
2442     g_free(params->tls_hostname);
2443     g_free(params->tls_creds);
2444     qemu_sem_destroy(&ms->pause_sem);
2445 }
2446 
2447 static void migration_instance_init(Object *obj)
2448 {
2449     MigrationState *ms = MIGRATION_OBJ(obj);
2450     MigrationParameters *params = &ms->parameters;
2451 
2452     ms->state = MIGRATION_STATUS_NONE;
2453     ms->xbzrle_cache_size = DEFAULT_MIGRATE_CACHE_SIZE;
2454     ms->mbps = -1;
2455     qemu_sem_init(&ms->pause_sem, 0);
2456     qemu_mutex_init(&ms->error_mutex);
2457 
2458     params->tls_hostname = g_strdup("");
2459     params->tls_creds = g_strdup("");
2460 
2461     /* Set has_* up only for parameter checks */
2462     params->has_compress_level = true;
2463     params->has_compress_threads = true;
2464     params->has_decompress_threads = true;
2465     params->has_cpu_throttle_initial = true;
2466     params->has_cpu_throttle_increment = true;
2467     params->has_max_bandwidth = true;
2468     params->has_downtime_limit = true;
2469     params->has_x_checkpoint_delay = true;
2470     params->has_block_incremental = true;
2471     params->has_x_multifd_channels = true;
2472     params->has_x_multifd_page_count = true;
2473 }
2474 
2475 /*
2476  * Return true if check pass, false otherwise. Error will be put
2477  * inside errp if provided.
2478  */
2479 static bool migration_object_check(MigrationState *ms, Error **errp)
2480 {
2481     MigrationCapabilityStatusList *head = NULL;
2482     /* Assuming all off */
2483     bool cap_list[MIGRATION_CAPABILITY__MAX] = { 0 }, ret;
2484     int i;
2485 
2486     if (!migrate_params_check(&ms->parameters, errp)) {
2487         return false;
2488     }
2489 
2490     for (i = 0; i < MIGRATION_CAPABILITY__MAX; i++) {
2491         if (ms->enabled_capabilities[i]) {
2492             head = migrate_cap_add(head, i, true);
2493         }
2494     }
2495 
2496     ret = migrate_caps_check(cap_list, head, errp);
2497 
2498     /* It works with head == NULL */
2499     qapi_free_MigrationCapabilityStatusList(head);
2500 
2501     return ret;
2502 }
2503 
2504 static const TypeInfo migration_type = {
2505     .name = TYPE_MIGRATION,
2506     /*
2507      * NOTE: TYPE_MIGRATION is not really a device, as the object is
2508      * not created using qdev_create(), it is not attached to the qdev
2509      * device tree, and it is never realized.
2510      *
2511      * TODO: Make this TYPE_OBJECT once QOM provides something like
2512      * TYPE_DEVICE's "-global" properties.
2513      */
2514     .parent = TYPE_DEVICE,
2515     .class_init = migration_class_init,
2516     .class_size = sizeof(MigrationClass),
2517     .instance_size = sizeof(MigrationState),
2518     .instance_init = migration_instance_init,
2519     .instance_finalize = migration_instance_finalize,
2520 };
2521 
2522 static void register_migration_types(void)
2523 {
2524     type_register_static(&migration_type);
2525 }
2526 
2527 type_init(register_migration_types);
2528