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   <[email protected]>
   9  *  Charles Duffy     <[email protected]>
  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  * Contributions after 2012-01-13 are licensed under the terms of the
  15  * GNU GPL, version 2 or (at your option) any later version.
  16  */
  17
  18 #include "qemu-common.h"
  19 #include "qemu/sockets.h"
  20 #include "migration/migration.h"
  21 #include "buffered_file.h"
  22 #include "block/block.h"
  23 #include <sys/types.h>
  24 #include <sys/wait.h>
  25
  26 //#define DEBUG_MIGRATION_EXEC
  27
  28 #ifdef DEBUG_MIGRATION_EXEC
  29 #define DPRINTF(fmt, ...) \
  30     do { printf("migration-exec: " fmt, ## __VA_ARGS__); } while (0)
  31 #else
  32 #define DPRINTF(fmt, ...) \
  33     do { } while (0)
  34 #endif
  35
  36 static int file_errno(MigrationState *s)
  37 {
  38     return errno;
  39 }
  40
  41 static int file_write(MigrationState *s, const void * buf, size_t size)
  42 {
  43     return write(s->fd, buf, size);
  44 }
  45
  46 static int exec_close(MigrationState *s)
  47 {
  48     int ret = 0;
  49     DPRINTF("exec_close\n");
  50     ret = qemu_fclose(s->opaque);
  51     s->opaque = NULL;
  52     s->fd = -1;
  53     if (ret >= 0 && !(WIFEXITED(ret) && WEXITSTATUS(ret) == 0)) {
  54         /* close succeeded, but non-zero exit code: */
  55         ret = -EIO; /* fake errno value */
  56     }
  57     return ret;
  58 }
  59
  60 void exec_start_outgoing_migration(MigrationState *s, const char *command, Error **errp)
  61 {
  62     FILE *f;
  63
  64     f = popen(command, "w");
  65     if (f == NULL) {
  66         error_setg_errno(errp, errno, "failed to popen the migration target");
  67         return;
  68     }
  69
  70     s->fd = fileno(f);
  71     assert(s->fd != -1);
  72     socket_set_nonblock(s->fd);
  73
  74     s->opaque = qemu_popen(f, "w");
  75
  76     s->close = exec_close;
  77     s->get_error = file_errno;
  78     s->write = file_write;
  79
  80     migrate_fd_connect(s);
  81 }
  82
  83 static void exec_accept_incoming_migration(void *opaque)
  84 {
  85     QEMUFile *f = opaque;
  86
  87     qemu_set_fd_handler2(qemu_get_fd(f), NULL, NULL, NULL, NULL);
  88     process_incoming_migration(f);
  89 }
  90
  91 void exec_start_incoming_migration(const char *command, Error **errp)
  92 {
  93     QEMUFile *f;
  94
  95     DPRINTF("Attempting to start an incoming migration\n");
  96     f = qemu_popen_cmd(command, "r");
  97     if(f == NULL) {
  98         error_setg_errno(errp, errno, "failed to popen the migration source");
  99         return;
 100     }
 101
 102     qemu_set_fd_handler2(qemu_get_fd(f), NULL,
 103                          exec_accept_incoming_migration, NULL, f);
 104 }