From 030fa7f6f950f5c8963f1dee8e6bb7387ed86a99 Mon Sep 17 00:00:00 2001 From: Eric Blake Date: Tue, 5 Sep 2017 14:11:14 -0500 Subject: nbd: Use new qio_channel_*_all() functions Rather than open-coding our own read/write-all functions, we can make use of the recently-added qio code. It slightly changes the error message in one of the iotests. Signed-off-by: Eric Blake Message-Id: <20170905191114.5959-4-eblake@redhat.com> Reviewed-by: Daniel P. Berrange --- nbd/nbd-internal.h | 41 +++++++---------------------------------- 1 file changed, 7 insertions(+), 34 deletions(-) (limited to 'nbd/nbd-internal.h') diff --git a/nbd/nbd-internal.h b/nbd/nbd-internal.h index 03549e3f39..8a609a227f 100644 --- a/nbd/nbd-internal.h +++ b/nbd/nbd-internal.h @@ -85,28 +85,14 @@ static inline int nbd_read_eof(QIOChannel *ioc, void *buffer, size_t size, Error **errp) { - struct iovec iov = { .iov_base = buffer, .iov_len = size }; - ssize_t ret; - - /* Sockets are kept in blocking mode in the negotiation phase. After - * that, a non-readable socket simply means that another thread stole - * our request/reply. Synchronization is done with recv_coroutine, so - * that this is coroutine-safe. - */ + int ret; assert(size); - - ret = nbd_rwv(ioc, &iov, 1, size, true, errp); - if (ret <= 0) { - return ret; - } - - if (ret != size) { - error_setg(errp, "End of file"); - return -EINVAL; + ret = qio_channel_read_all_eof(ioc, buffer, size, errp); + if (ret < 0) { + ret = -EIO; } - - return 1; + return ret; } /* nbd_read @@ -115,14 +101,7 @@ static inline int nbd_read_eof(QIOChannel *ioc, void *buffer, size_t size, static inline int nbd_read(QIOChannel *ioc, void *buffer, size_t size, Error **errp) { - int ret = nbd_read_eof(ioc, buffer, size, errp); - - if (ret == 0) { - ret = -EINVAL; - error_setg(errp, "End of file"); - } - - return ret < 0 ? ret : 0; + return qio_channel_read_all(ioc, buffer, size, errp) < 0 ? -EIO : 0; } /* nbd_write @@ -131,13 +110,7 @@ static inline int nbd_read(QIOChannel *ioc, void *buffer, size_t size, static inline int nbd_write(QIOChannel *ioc, const void *buffer, size_t size, Error **errp) { - struct iovec iov = { .iov_base = (void *) buffer, .iov_len = size }; - - ssize_t ret = nbd_rwv(ioc, &iov, 1, size, false, errp); - - assert(ret < 0 || ret == size); - - return ret < 0 ? ret : 0; + return qio_channel_write_all(ioc, buffer, size, errp) < 0 ? -EIO : 0; } struct NBDTLSHandshakeData { -- cgit v1.2.1