Note for the brace police: the style in this commit and the following is consistent with the rest of the file. It is then fixed together with the introduction of coroutines.
Signed-off-by: Paolo Bonzini <pbonz...@redhat.com> --- block/nbd.c | 31 +++++++++++++++++++++++++++++++ nbd.c | 14 +++++++++++++- 2 files changed, 44 insertions(+), 1 deletions(-) diff --git a/block/nbd.c b/block/nbd.c index ffc57a9..4a195dc 100644 --- a/block/nbd.c +++ b/block/nbd.c @@ -237,6 +237,36 @@ static int nbd_write(BlockDriverState *bs, int64_t sector_num, return 0; } +static int nbd_flush(BlockDriverState *bs) +{ + BDRVNBDState *s = bs->opaque; + struct nbd_request request; + struct nbd_reply reply; + + if (!(s->nbdflags & NBD_FLAG_SEND_FLUSH)) { + return 0; + } + + request.type = NBD_CMD_FLUSH; + request.handle = (uint64_t)(intptr_t)bs; + request.from = 0; + request.len = 0; + + if (nbd_send_request(s->sock, &request) == -1) + return -errno; + + if (nbd_receive_reply(s->sock, &reply) == -1) + return -errno; + + if (reply.error !=0) + return -reply.error; + + if (reply.handle != request.handle) + return -EIO; + + return 0; +} + static void nbd_close(BlockDriverState *bs) { BDRVNBDState *s = bs->opaque; @@ -260,6 +290,7 @@ static BlockDriver bdrv_nbd = { .bdrv_read = nbd_read, .bdrv_write = nbd_write, .bdrv_close = nbd_close, + .bdrv_flush = nbd_flush, .bdrv_getlength = nbd_getlength, .protocol_name = "nbd", }; diff --git a/nbd.c b/nbd.c index 30cd78f..4dbbc62 100644 --- a/nbd.c +++ b/nbd.c @@ -193,7 +193,8 @@ int nbd_negotiate(int csock, off_t size, uint32_t flags) memcpy(buf, "NBDMAGIC", 8); cpu_to_be64w((uint64_t*)(buf + 8), 0x00420281861253LL); cpu_to_be64w((uint64_t*)(buf + 16), size); - cpu_to_be32w((uint32_t*)(buf + 24), flags | NBD_FLAG_HAS_FLAGS); + cpu_to_be32w((uint32_t*)(buf + 24), + flags | NBD_FLAG_HAS_FLAGS | NBD_FLAG_SEND_FLUSH); memset(buf + 28, 0, 124); if (write_sync(csock, buf, sizeof(buf)) != sizeof(buf)) { @@ -682,6 +683,18 @@ int nbd_trip(BlockDriverState *bs, int csock, off_t size, uint64_t dev_offset, TRACE("Request type is DISCONNECT"); errno = 0; return 1; + case NBD_CMD_FLUSH: + TRACE("Request type is FLUSH"); + + if (bdrv_flush(bs) == -1) { + LOG("flush failed"); + errno = EINVAL; + return -1; + } + + if (nbd_send_reply(csock, &reply) == -1) + return -1; + break; default: LOG("invalid request type (%u) received", request.type); errno = EINVAL; -- 1.7.6