From: "Dr. David Alan Gilbert" <dgilb...@redhat.com> Postcopy needs a method to send messages from the destination back to the source, this is the 'return path'.
Wire it up for 'socket' QEMUFile's using a dup'd fd. Signed-off-by: Dr. David Alan Gilbert <dgilb...@redhat.com> --- include/migration/qemu-file.h | 7 +++++ qemu-file.c | 73 +++++++++++++++++++++++++++++++++++++------ 2 files changed, 70 insertions(+), 10 deletions(-) diff --git a/include/migration/qemu-file.h b/include/migration/qemu-file.h index 935cf42..210e9c3 100644 --- a/include/migration/qemu-file.h +++ b/include/migration/qemu-file.h @@ -85,6 +85,11 @@ typedef size_t (QEMURamSaveFunc)(QEMUFile *f, void *opaque, int *bytes_sent); /* + * Return a QEMUFile for comms in the opposite direction + */ +typedef QEMUFile *(QEMURetPathFunc)(void *opaque); + +/* * Stop any read or write (depending on flags) on the underlying * transport on the QEMUFile. * Existing blocking reads/writes must be woken @@ -102,6 +107,7 @@ typedef struct QEMUFileOps { QEMURamHookFunc *after_ram_iterate; QEMURamHookFunc *hook_ram_load; QEMURamSaveFunc *save_page; + QEMURetPathFunc *get_return_path; QEMUFileShutdownFunc *shut_down; } QEMUFileOps; @@ -188,6 +194,7 @@ int64_t qemu_file_get_rate_limit(QEMUFile *f); int qemu_file_get_error(QEMUFile *f); void qemu_file_set_error(QEMUFile *f, int ret); void qemu_file_shutdown(QEMUFile *f); +QEMUFile *qemu_file_get_return_path(QEMUFile *f); void qemu_fflush(QEMUFile *f); static inline void qemu_put_be64s(QEMUFile *f, const uint64_t *pv) diff --git a/qemu-file.c b/qemu-file.c index 14dcf34..7393415 100644 --- a/qemu-file.c +++ b/qemu-file.c @@ -26,6 +26,8 @@ struct QEMUFile { unsigned int iovcnt; int last_error; + + struct QEMUFile *return_path; }; typedef struct QEMUFileStdio { @@ -38,6 +40,45 @@ typedef struct QEMUFileSocket { QEMUFile *file; } QEMUFileSocket; +/* + * Give a QEMUFile* off the same socket but data in the opposite + * direction. + */ +static QEMUFile *socket_dup_return_path(void *opaque) +{ + QEMUFileSocket *qfs = opaque; + int revfd; + bool this_is_read; + QEMUFile *result; + + /* We should only be called once to get a RP on a file */ + assert(!qfs->file->return_path); + + if (qemu_file_get_error(qfs->file)) { + /* If the forward file is in error, don't try and open a return */ + return NULL; + } + + /* I don't think there's a better way to tell which direction 'this' is */ + this_is_read = qfs->file->ops->get_buffer != NULL; + + revfd = dup(qfs->fd); + if (revfd == -1) { + error_report("Error duplicating fd for return path: %s", + strerror(errno)); + return NULL; + } + + result = qemu_fopen_socket(revfd, this_is_read ? "wb" : "rb"); + qfs->file->return_path = result; + + if (!result) { + close(revfd); + } + + return result; +} + static ssize_t socket_writev_buffer(void *opaque, struct iovec *iov, int iovcnt, int64_t pos) { @@ -343,19 +384,19 @@ QEMUFile *qemu_fdopen(int fd, const char *mode) } static const QEMUFileOps socket_read_ops = { - .get_fd = socket_get_fd, - .get_buffer = socket_get_buffer, - .close = socket_close, - .shut_down = qemufile_socket_shutdown - + .get_fd = socket_get_fd, + .get_buffer = socket_get_buffer, + .close = socket_close, + .shut_down = qemufile_socket_shutdown, + .get_return_path = socket_dup_return_path }; static const QEMUFileOps socket_write_ops = { - .get_fd = socket_get_fd, - .writev_buffer = socket_writev_buffer, - .close = socket_close, - .shut_down = qemufile_socket_shutdown - + .get_fd = socket_get_fd, + .writev_buffer = socket_writev_buffer, + .close = socket_close, + .shut_down = qemufile_socket_shutdown, + .get_return_path = socket_dup_return_path }; /* @@ -369,6 +410,18 @@ void qemu_file_shutdown(QEMUFile *f) f->ops->shut_down(f, true, true); } +/* + * Result: QEMUFile* for a 'return path' for comms in the opposite direction + * NULL if not available + */ +QEMUFile *qemu_file_get_return_path(QEMUFile *f) +{ + if (!f->ops->get_return_path) { + return NULL; + } + return f->ops->get_return_path(f->opaque); +} + bool qemu_file_mode_is_not_valid(const char *mode) { if (mode == NULL || -- 1.9.3