aboutsummaryrefslogtreecommitdiff
path: root/nbd/nbd-internal.h
diff options
context:
space:
mode:
authorEric Blake <eblake@redhat.com>2017-09-05 14:11:14 -0500
committerEric Blake <eblake@redhat.com>2017-09-06 10:11:54 -0500
commit030fa7f6f950f5c8963f1dee8e6bb7387ed86a99 (patch)
treec604ccc0c3afbfa97fa60839fce99619bf20e457 /nbd/nbd-internal.h
parente8ffaa311080a570a7c86d03c139c160cd11a831 (diff)
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 <eblake@redhat.com> Message-Id: <20170905191114.5959-4-eblake@redhat.com> Reviewed-by: Daniel P. Berrange <berrange@redhat.com>
Diffstat (limited to 'nbd/nbd-internal.h')
-rw-r--r--nbd/nbd-internal.h41
1 files changed, 7 insertions, 34 deletions
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 {