migration-exec.c

   1 /*
   2  * QEMU live migration
   3  *
   4  * Copyright IBM, Corp. 2008
   5  * Copyright Dell MessageOne 2008
   6  *
   7  * Authors:
   8  *  Anthony Liguori   <aliguori@us.ibm.com>
   9  *  Charles Duffy     <charles_duffy@messageone.com>
  10  *
  11  * This work is licensed under the terms of the GNU GPL, version 2.  See
  12  * the COPYING file in the top-level directory.
  13  *
  14  */
  15
  16 #include "qemu-common.h"
  17 #include "qemu_socket.h"
  18 #include "migration.h"
  19 #include "qemu-char.h"
  20 #include "sysemu.h"
  21 #include "console.h"
  22 #include "buffered_file.h"
  23 #include "block.h"
  24
  25 //#define DEBUG_MIGRATION_EXEC
  26
  27 #ifdef DEBUG_MIGRATION_EXEC
  28 #define dprintf(fmt, ...) \
  29     do { printf("migration-exec: " fmt, ## __VA_ARGS__); } while (0)
  30 #else
  31 #define dprintf(fmt, ...) \
  32     do { } while (0)
  33 #endif
  34
  35 static int file_errno(FdMigrationState *s)
  36 {
  37     return errno;
  38 }
  39
  40 static int file_write(FdMigrationState *s, const void * buf, size_t size)
  41 {
  42     return write(s->fd, buf, size);
  43 }
  44
  45 static int exec_close(FdMigrationState *s)
  46 {
  47     dprintf("exec_close\n");
  48     if (s->opaque) {
  49         qemu_fclose(s->opaque);
  50         s->opaque = NULL;
  51         s->fd = -1;
  52     }
  53     return 0;
  54 }
  55
  56 MigrationState *exec_start_outgoing_migration(const char *command,
  57                                              int64_t bandwidth_limit,
  58                                              int async)
  59 {
  60     FdMigrationState *s;
  61     FILE *f;
  62
  63     s = qemu_mallocz(sizeof(*s));
  64     if (s == NULL) {
  65         dprintf("Unable to allocate FdMigrationState\n");
  66         goto err;
  67     }
  68
  69     f = popen(command, "w");
  70     if (f == NULL) {
  71         dprintf("Unable to popen exec target\n");
  72         goto err_after_alloc;
  73     }
  74
  75     s->fd = fileno(f);
  76     if (s->fd == -1) {
  77         dprintf("Unable to retrieve file descriptor for popen'd handle\n");
  78         goto err_after_open;
  79     }
  80
  81     if (fcntl(s->fd, F_SETFD, O_NONBLOCK) == -1) {
  82         dprintf("Unable to set nonblocking mode on file descriptor\n");
  83         goto err_after_open;
  84     }
  85
  86     s->opaque = qemu_popen(f, "w");
  87
  88     s->close = exec_close;
  89     s->get_error = file_errno;
  90     s->write = file_write;
  91     s->mig_state.cancel = migrate_fd_cancel;
  92     s->mig_state.get_status = migrate_fd_get_status;
  93     s->mig_state.release = migrate_fd_release;
  94
  95     s->state = MIG_STATE_ACTIVE;
  96     s->detach = !async;
  97     s->bandwidth_limit = bandwidth_limit;
  98
  99     if (s->detach == 1) {
 100         dprintf("detaching from monitor\n");
 101         monitor_suspend();
 102         s->detach = 2;
 103     }
 104
 105     migrate_fd_connect(s);
 106     return &s->mig_state;
 107
 108 err_after_open:
 109     pclose(f);
 110 err_after_alloc:
 111     qemu_free(s);
 112 err:
 113     return NULL;
 114 }
 115
 116 int exec_start_incoming_migration(const char *command)
 117 {
 118     int ret;
 119     QEMUFile *f;
 120
 121     dprintf("Attempting to start an incoming migration\n");
 122     f = qemu_popen_cmd(command, "r");
 123     if(f == NULL) {
 124         dprintf("Unable to apply qemu wrapper to popen file\n");
 125         return -errno;
 126     }
 127     vm_stop(0); /* just in case */
 128     ret = qemu_loadvm_state(f);
 129     if (ret < 0) {
 130         fprintf(stderr, "load of migration failed\n");
 131         goto err;
 132     }
 133     qemu_announce_self();
 134     dprintf("successfully loaded vm state\n");
 135     vm_start();
 136     qemu_fclose(f);
 137     return 0;
 138
 139 err:
 140     qemu_fclose(f);
 141     return -errno;
 142 }