[Top][All Lists]
[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]
[Qemu-devel] [PATCH 15/25] qemu-nbd: introduce nbd_do_send_reply
From: |
Paolo Bonzini |
Subject: |
[Qemu-devel] [PATCH 15/25] qemu-nbd: introduce nbd_do_send_reply |
Date: |
Tue, 6 Dec 2011 16:27:42 +0100 |
Group the sending of a reply and the associated data into a new function.
Without corking, the caller would be forced to leave 12 free bytes at the
beginning of the data pointer. Not too ugly, but still ugly. :)
Using nbd_do_send_reply everywhere will help when the routine will set up
the write handler that re-enters the send coroutine.
Signed-off-by: Paolo Bonzini <address@hidden>
---
nbd.c | 46 ++++++++++++++++++++++++++++++++--------------
1 files changed, 32 insertions(+), 14 deletions(-)
diff --git a/nbd.c b/nbd.c
index 71447cd..fb074ef 100644
--- a/nbd.c
+++ b/nbd.c
@@ -583,6 +583,34 @@ static int nbd_send_reply(int csock, struct nbd_reply
*reply)
return 0;
}
+static int nbd_do_send_reply(int csock, struct nbd_reply *reply,
+ uint8_t *data, int len)
+{
+ int rc, ret;
+
+ if (!len) {
+ rc = nbd_send_reply(csock, reply);
+ if (rc == -1) {
+ rc = -errno;
+ }
+ } else {
+ socket_set_cork(csock, 1);
+ rc = nbd_send_reply(csock, reply);
+ if (rc != -1) {
+ ret = write_sync(csock, data, len);
+ if (ret != len) {
+ errno = EIO;
+ rc = -1;
+ }
+ }
+ if (rc == -1) {
+ rc = -errno;
+ }
+ socket_set_cork(csock, 0);
+ }
+ return rc;
+}
+
int nbd_trip(BlockDriverState *bs, int csock, off_t size,
uint64_t dev_offset, uint32_t nbdflags,
uint8_t *data)
@@ -637,18 +665,8 @@ int nbd_trip(BlockDriverState *bs, int csock, off_t size,
}
TRACE("Read %u byte(s)", request.len);
- socket_set_cork(csock, 1);
- if (nbd_send_reply(csock, &reply) == -1)
+ if (nbd_do_send_reply(csock, &reply, data, request.len) < 0)
return -1;
-
- TRACE("Sending data to client");
-
- if (write_sync(csock, data, request.len) != request.len) {
- LOG("writing to socket failed");
- errno = EINVAL;
- return -1;
- }
- socket_set_cork(csock, 0);
break;
case NBD_CMD_WRITE:
TRACE("Request type is WRITE");
@@ -684,7 +702,7 @@ int nbd_trip(BlockDriverState *bs, int csock, off_t size,
}
}
- if (nbd_send_reply(csock, &reply) == -1)
+ if (nbd_do_send_reply(csock, &reply, NULL, 0) < 0)
return -1;
break;
case NBD_CMD_DISC:
@@ -700,7 +718,7 @@ int nbd_trip(BlockDriverState *bs, int csock, off_t size,
reply.error = -ret;
}
- if (nbd_send_reply(csock, &reply) == -1)
+ if (nbd_do_send_reply(csock, &reply, NULL, 0) < 0)
return -1;
break;
case NBD_CMD_TRIM:
@@ -711,7 +729,7 @@ int nbd_trip(BlockDriverState *bs, int csock, off_t size,
LOG("discard failed");
reply.error = -ret;
}
- if (nbd_send_reply(csock, &reply) == -1)
+ if (nbd_do_send_reply(csock, &reply, NULL, 0) < 0)
return -1;
break;
default:
--
1.7.7.1
- [Qemu-devel] [PATCH 02/25] sheepdog: move coroutine send/recv function to generic code, (continued)
- [Qemu-devel] [PATCH 02/25] sheepdog: move coroutine send/recv function to generic code, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 06/25] nbd: fix error handling in the server, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 08/25] nbd: add support for NBD_CMD_FLUSH, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 07/25] nbd: add support for NBD_CMD_FLAG_FUA, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 09/25] nbd: add support for NBD_CMD_TRIM, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 11/25] qemu-nbd: remove offset argument to nbd_trip, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 10/25] Update ioctl order in nbd_init() to detect EBUSY, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 16/25] qemu-nbd: more robust handling of invalid requests, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 14/25] qemu-nbd: simplify nbd_trip, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 12/25] qemu-nbd: remove data_size argument to nbd_trip, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 15/25] qemu-nbd: introduce nbd_do_send_reply,
Paolo Bonzini <=
- [Qemu-devel] [PATCH 25/25] qemu-nbd: throttle requests, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 23/25] qemu-nbd: add client pointer to NBDRequest, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 24/25] qemu-nbd: asynchronous operation, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 20/25] link the main loop and its dependencies into the tools, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 19/25] qemu-nbd: introduce NBDRequest, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 18/25] qemu-nbd: introduce NBDExport, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 13/25] move corking functions to osdep.c, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 17/25] qemu-nbd: introduce nbd_do_receive_request, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 21/25] qemu-nbd: use common main loop, Paolo Bonzini, 2011/12/06
- [Qemu-devel] [PATCH 22/25] qemu-nbd: move client handling to nbd.c, Paolo Bonzini, 2011/12/06