migration: use qemu-sockets to establish Unix sockets

This makes migration-unix.c again a cut-and-paste job from migration-tcp.c,
exactly as it was in the beginning. :)

Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>
This commit is contained in:
Paolo Bonzini 2012-10-03 14:05:49 +02:00
parent 342ab8d1e4
commit e08c95ce8d
3 changed files with 21 additions and 81 deletions

View File

@ -53,67 +53,34 @@ static int unix_close(MigrationState *s)
return r; return r;
} }
static void unix_wait_for_connect(void *opaque) static void unix_wait_for_connect(int fd, void *opaque)
{ {
MigrationState *s = opaque; MigrationState *s = opaque;
int val, ret;
socklen_t valsize = sizeof(val);
DPRINTF("connect completed\n"); if (fd < 0) {
do { DPRINTF("migrate connect error\n");
ret = getsockopt(s->fd, SOL_SOCKET, SO_ERROR, (void *) &val, &valsize); s->fd = -1;
} while (ret == -1 && errno == EINTR);
if (ret < 0) {
migrate_fd_error(s); migrate_fd_error(s);
return; } else {
} DPRINTF("migrate connect success\n");
s->fd = fd;
qemu_set_fd_handler2(s->fd, NULL, NULL, NULL, NULL);
if (val == 0)
migrate_fd_connect(s); migrate_fd_connect(s);
else {
DPRINTF("error connecting %d\n", val);
migrate_fd_error(s);
} }
} }
int unix_start_outgoing_migration(MigrationState *s, const char *path) int unix_start_outgoing_migration(MigrationState *s, const char *path, Error **errp)
{ {
struct sockaddr_un addr; Error *local_err = NULL;
int ret;
addr.sun_family = AF_UNIX;
snprintf(addr.sun_path, sizeof(addr.sun_path), "%s", path);
s->get_error = unix_errno; s->get_error = unix_errno;
s->write = unix_write; s->write = unix_write;
s->close = unix_close; s->close = unix_close;
s->fd = qemu_socket(PF_UNIX, SOCK_STREAM, 0); s->fd = unix_nonblocking_connect(path, unix_wait_for_connect, s, &local_err);
if (s->fd == -1) { if (local_err != NULL) {
DPRINTF("Unable to open socket"); error_propagate(errp, local_err);
return -errno; return -1;
} }
socket_set_nonblock(s->fd);
do {
ret = connect(s->fd, (struct sockaddr *)&addr, sizeof(addr));
if (ret == -1) {
ret = -errno;
}
if (ret == -EINPROGRESS || ret == -EWOULDBLOCK) {
qemu_set_fd_handler2(s->fd, NULL, NULL, unix_wait_for_connect, s);
return 0;
}
} while (ret == -EINTR);
if (ret < 0) {
DPRINTF("connect failed\n");
return ret;
}
migrate_fd_connect(s);
return 0; return 0;
} }
@ -151,43 +118,16 @@ out2:
close(s); close(s);
} }
int unix_start_incoming_migration(const char *path) int unix_start_incoming_migration(const char *path, Error **errp)
{ {
struct sockaddr_un addr;
int s; int s;
int ret;
DPRINTF("Attempting to start an incoming migration\n"); s = unix_listen(path, NULL, 0, errp);
if (s < 0) {
s = qemu_socket(PF_UNIX, SOCK_STREAM, 0); return -1;
if (s == -1) {
fprintf(stderr, "Could not open unix socket: %s\n", strerror(errno));
return -errno;
}
memset(&addr, 0, sizeof(addr));
addr.sun_family = AF_UNIX;
snprintf(addr.sun_path, sizeof(addr.sun_path), "%s", path);
unlink(addr.sun_path);
if (bind(s, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
ret = -errno;
fprintf(stderr, "bind(unix:%s): %s\n", addr.sun_path, strerror(errno));
goto err;
}
if (listen(s, 1) == -1) {
fprintf(stderr, "listen(unix:%s): %s\n", addr.sun_path,
strerror(errno));
ret = -errno;
goto err;
} }
qemu_set_fd_handler2(s, NULL, unix_accept_incoming_migration, NULL, qemu_set_fd_handler2(s, NULL, unix_accept_incoming_migration, NULL,
(void *)(intptr_t)s); (void *)(intptr_t)s);
return 0; return 0;
err:
close(s);
return ret;
} }

View File

@ -75,7 +75,7 @@ int qemu_start_incoming_migration(const char *uri, Error **errp)
else if (strstart(uri, "exec:", &p)) else if (strstart(uri, "exec:", &p))
ret = exec_start_incoming_migration(p); ret = exec_start_incoming_migration(p);
else if (strstart(uri, "unix:", &p)) else if (strstart(uri, "unix:", &p))
ret = unix_start_incoming_migration(p); ret = unix_start_incoming_migration(p, errp);
else if (strstart(uri, "fd:", &p)) else if (strstart(uri, "fd:", &p))
ret = fd_start_incoming_migration(p); ret = fd_start_incoming_migration(p);
#endif #endif
@ -514,7 +514,7 @@ void qmp_migrate(const char *uri, bool has_blk, bool blk,
} else if (strstart(uri, "exec:", &p)) { } else if (strstart(uri, "exec:", &p)) {
ret = exec_start_outgoing_migration(s, p); ret = exec_start_outgoing_migration(s, p);
} else if (strstart(uri, "unix:", &p)) { } else if (strstart(uri, "unix:", &p)) {
ret = unix_start_outgoing_migration(s, p); ret = unix_start_outgoing_migration(s, p, &local_err);
} else if (strstart(uri, "fd:", &p)) { } else if (strstart(uri, "fd:", &p)) {
ret = fd_start_outgoing_migration(s, p); ret = fd_start_outgoing_migration(s, p);
#endif #endif

View File

@ -66,9 +66,9 @@ int tcp_start_incoming_migration(const char *host_port, Error **errp);
int tcp_start_outgoing_migration(MigrationState *s, const char *host_port, int tcp_start_outgoing_migration(MigrationState *s, const char *host_port,
Error **errp); Error **errp);
int unix_start_incoming_migration(const char *path); int unix_start_incoming_migration(const char *path, Error **errp);
int unix_start_outgoing_migration(MigrationState *s, const char *path); int unix_start_outgoing_migration(MigrationState *s, const char *path, Error **errp);
int fd_start_incoming_migration(const char *path); int fd_start_incoming_migration(const char *path);