diff options
| -rw-r--r-- | include/migration/qemu-file.h | 1 | ||||
| -rw-r--r-- | migration/qemu-file.c | 15 |
2 files changed, 16 insertions, 0 deletions
diff --git a/include/migration/qemu-file.h b/include/migration/qemu-file.h index 86bb9729d8..66e741f649 100644 --- a/include/migration/qemu-file.h +++ b/include/migration/qemu-file.h @@ -197,6 +197,7 @@ int qemu_file_get_error(QEMUFile *f); void qemu_file_set_error(QEMUFile *f, int ret); int qemu_file_shutdown(QEMUFile *f); void qemu_fflush(QEMUFile *f); +void qemu_file_set_blocking(QEMUFile *f, bool block); static inline void qemu_put_be64s(QEMUFile *f, const uint64_t *pv) { diff --git a/migration/qemu-file.c b/migration/qemu-file.c index e41a677fe3..9ec22678a8 100644 --- a/migration/qemu-file.c +++ b/migration/qemu-file.c @@ -648,3 +648,18 @@ size_t qemu_get_counted_string(QEMUFile *f, char buf[256]) return res == len ? res : 0; } + +/* + * Set the blocking state of the QEMUFile. + * Note: On some transports the OS only keeps a single blocking state for + * both directions, and thus changing the blocking on the main + * QEMUFile can also affect the return path. + */ +void qemu_file_set_blocking(QEMUFile *f, bool block) +{ + if (block) { + qemu_set_block(qemu_get_fd(f)); + } else { + qemu_set_nonblock(qemu_get_fd(f)); + } +} |