ShuriZma
/
xqemu
mirror of https://github.com/xqemu/xqemu.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

sheepdog: use BDRV_POLL_WHILE 

This is important when the sheepdog driver works on a BlockDriverState
that is attached to an I/O thread other than the main thread.

Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>
Reviewed-by: Fam Zheng <famz@redhat.com>
Message-Id: <1477565348-5458-11-git-send-email-pbonzini@redhat.com>
Signed-off-by: Fam Zheng <famz@redhat.com>
This commit is contained in:
Paolo Bonzini 2016-10-27 12:48:58 +02:00 committed by Fam Zheng
parent d746427aaf
commit f11672dbc0
1 changed files with 35 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
block/sheepdog.c
View File

@ -641,6 +641,7 @@ static void restart_co_req(void *opaque)
typedef struct SheepdogReqCo { typedef struct SheepdogReqCo {
int sockfd; int sockfd;
BlockDriverState *bs;
AioContext *aio_context; AioContext *aio_context;
SheepdogReq *hdr; SheepdogReq *hdr;
void *data; void *data;
@ -708,13 +709,14 @@ out:
* *
* Return 0 on success, -errno in case of error. * Return 0 on success, -errno in case of error.
*/ */
static int do_req(int sockfd, AioContext *aio_context, SheepdogReq *hdr, static int do_req(int sockfd, BlockDriverState *bs, SheepdogReq *hdr,
void *data, unsigned int *wlen, unsigned int *rlen) void *data, unsigned int *wlen, unsigned int *rlen)
{ {
Coroutine *co; Coroutine *co;
SheepdogReqCo srco = { SheepdogReqCo srco = {
.sockfd = sockfd, .sockfd = sockfd,
.aio_context = aio_context, .aio_context = bs ? bdrv_get_aio_context(bs) : qemu_get_aio_context(),
.bs = bs,
.hdr = hdr, .hdr = hdr,
.data = data, .data = data,
.wlen = wlen, .wlen = wlen,
@ -727,9 +729,14 @@ static int do_req(int sockfd, AioContext *aio_context, SheepdogReq *hdr,
do_co_req(&srco); do_co_req(&srco);
} else { } else {
co = qemu_coroutine_create(do_co_req, &srco); co = qemu_coroutine_create(do_co_req, &srco);
if (bs) {
qemu_coroutine_enter(co);
BDRV_POLL_WHILE(bs, !srco.finished);
} else {
qemu_coroutine_enter(co); qemu_coroutine_enter(co);
while (!srco.finished) { while (!srco.finished) {
aio_poll(aio_context, true); aio_poll(qemu_get_aio_context(), true);
}
} }
} }
@ -1125,7 +1132,7 @@ static int find_vdi_name(BDRVSheepdogState *s, const char *filename,
hdr.snapid = snapid; hdr.snapid = snapid;
hdr.flags = SD_FLAG_CMD_WRITE; hdr.flags = SD_FLAG_CMD_WRITE;
ret = do_req(fd, s->aio_context, (SheepdogReq *)&hdr, buf, &wlen, &rlen); ret = do_req(fd, s->bs, (SheepdogReq *)&hdr, buf, &wlen, &rlen);
if (ret) { if (ret) {
error_setg_errno(errp, -ret, "cannot get vdi info"); error_setg_errno(errp, -ret, "cannot get vdi info");
goto out; goto out;
@ -1240,7 +1247,7 @@ out:
qemu_co_mutex_unlock(&s->lock); qemu_co_mutex_unlock(&s->lock);
} }
static int read_write_object(int fd, AioContext *aio_context, char *buf, static int read_write_object(int fd, BlockDriverState *bs, char *buf,
uint64_t oid, uint8_t copies, uint64_t oid, uint8_t copies,
unsigned int datalen, uint64_t offset, unsigned int datalen, uint64_t offset,
bool write, bool create, uint32_t cache_flags) bool write, bool create, uint32_t cache_flags)
@ -1274,7 +1281,7 @@ static int read_write_object(int fd, AioContext *aio_context, char *buf,
hdr.offset = offset; hdr.offset = offset;
hdr.copies = copies; hdr.copies = copies;
ret = do_req(fd, aio_context, (SheepdogReq *)&hdr, buf, &wlen, &rlen); ret = do_req(fd, bs, (SheepdogReq *)&hdr, buf, &wlen, &rlen);
if (ret) { if (ret) {
error_report("failed to send a request to the sheep"); error_report("failed to send a request to the sheep");
return ret; return ret;
@ -1289,22 +1296,22 @@ static int read_write_object(int fd, AioContext *aio_context, char *buf,
} }
} }
static int read_object(int fd, AioContext *aio_context, char *buf, static int read_object(int fd, BlockDriverState *bs, char *buf,
uint64_t oid, uint8_t copies, uint64_t oid, uint8_t copies,
unsigned int datalen, uint64_t offset, unsigned int datalen, uint64_t offset,
uint32_t cache_flags) uint32_t cache_flags)
{ {
return read_write_object(fd, aio_context, buf, oid, copies, return read_write_object(fd, bs, buf, oid, copies,
datalen, offset, false, datalen, offset, false,
false, cache_flags); false, cache_flags);
} }
static int write_object(int fd, AioContext *aio_context, char *buf, static int write_object(int fd, BlockDriverState *bs, char *buf,
uint64_t oid, uint8_t copies, uint64_t oid, uint8_t copies,
unsigned int datalen, uint64_t offset, bool create, unsigned int datalen, uint64_t offset, bool create,
uint32_t cache_flags) uint32_t cache_flags)
{ {
return read_write_object(fd, aio_context, buf, oid, copies, return read_write_object(fd, bs, buf, oid, copies,
datalen, offset, true, datalen, offset, true,
create, cache_flags); create, cache_flags);
} }
@ -1331,7 +1338,7 @@ static int reload_inode(BDRVSheepdogState *s, uint32_t snapid, const char *tag)
goto out; goto out;
} }
ret = read_object(fd, s->aio_context, (char *)inode, vid_to_vdi_oid(vid), ret = read_object(fd, s->bs, (char *)inode, vid_to_vdi_oid(vid),
s->inode.nr_copies, SD_INODE_HEADER_SIZE, 0, s->inode.nr_copies, SD_INODE_HEADER_SIZE, 0,
s->cache_flags); s->cache_flags);
if (ret < 0) { if (ret < 0) {
@ -1489,7 +1496,7 @@ static int sd_open(BlockDriverState *bs, QDict *options, int flags,
} }
buf = g_malloc(SD_INODE_SIZE); buf = g_malloc(SD_INODE_SIZE);
ret = read_object(fd, s->aio_context, buf, vid_to_vdi_oid(vid), ret = read_object(fd, s->bs, buf, vid_to_vdi_oid(vid),
0, SD_INODE_SIZE, 0, s->cache_flags); 0, SD_INODE_SIZE, 0, s->cache_flags);
closesocket(fd); closesocket(fd);
@ -1618,7 +1625,7 @@ static int do_sd_create(BDRVSheepdogState *s, uint32_t *vdi_id, int snapshot,
hdr.copies = s->inode.nr_copies; hdr.copies = s->inode.nr_copies;
hdr.block_size_shift = s->inode.block_size_shift; hdr.block_size_shift = s->inode.block_size_shift;
ret = do_req(fd, s->aio_context, (SheepdogReq *)&hdr, buf, &wlen, &rlen); ret = do_req(fd, NULL, (SheepdogReq *)&hdr, buf, &wlen, &rlen);
closesocket(fd); closesocket(fd);
@ -1886,7 +1893,7 @@ static int sd_create(const char *filename, QemuOpts *opts,
hdr.opcode = SD_OP_GET_CLUSTER_DEFAULT; hdr.opcode = SD_OP_GET_CLUSTER_DEFAULT;
hdr.proto_ver = SD_PROTO_VER; hdr.proto_ver = SD_PROTO_VER;
ret = do_req(fd, s->aio_context, (SheepdogReq *)&hdr, ret = do_req(fd, NULL, (SheepdogReq *)&hdr,
NULL, &wlen, &rlen); NULL, &wlen, &rlen);
closesocket(fd); closesocket(fd);
if (ret) { if (ret) {
@ -1951,7 +1958,7 @@ static void sd_close(BlockDriverState *bs)
hdr.data_length = wlen; hdr.data_length = wlen;
hdr.flags = SD_FLAG_CMD_WRITE; hdr.flags = SD_FLAG_CMD_WRITE;
ret = do_req(fd, s->aio_context, (SheepdogReq *)&hdr, ret = do_req(fd, s->bs, (SheepdogReq *)&hdr,
s->name, &wlen, &rlen); s->name, &wlen, &rlen);
closesocket(fd); closesocket(fd);
@ -2000,7 +2007,7 @@ static int sd_truncate(BlockDriverState *bs, int64_t offset)
/* we don't need to update entire object */ /* we don't need to update entire object */
datalen = SD_INODE_SIZE - sizeof(s->inode.data_vdi_id); datalen = SD_INODE_SIZE - sizeof(s->inode.data_vdi_id);
s->inode.vdi_size = offset; s->inode.vdi_size = offset;
ret = write_object(fd, s->aio_context, (char *)&s->inode, ret = write_object(fd, s->bs, (char *)&s->inode,
vid_to_vdi_oid(s->inode.vdi_id), s->inode.nr_copies, vid_to_vdi_oid(s->inode.vdi_id), s->inode.nr_copies,
datalen, 0, false, s->cache_flags); datalen, 0, false, s->cache_flags);
close(fd); close(fd);
@ -2070,7 +2077,7 @@ static bool sd_delete(BDRVSheepdogState *s)
return false; return false;
} }
ret = do_req(fd, s->aio_context, (SheepdogReq *)&hdr, ret = do_req(fd, s->bs, (SheepdogReq *)&hdr,
s->name, &wlen, &rlen); s->name, &wlen, &rlen);
closesocket(fd); closesocket(fd);
if (ret) { if (ret) {
@ -2126,7 +2133,7 @@ static int sd_create_branch(BDRVSheepdogState *s)
goto out; goto out;
} }
ret = read_object(fd, s->aio_context, buf, vid_to_vdi_oid(vid), ret = read_object(fd, s->bs, buf, vid_to_vdi_oid(vid),
s->inode.nr_copies, SD_INODE_SIZE, 0, s->cache_flags); s->inode.nr_copies, SD_INODE_SIZE, 0, s->cache_flags);
closesocket(fd); closesocket(fd);
@ -2411,7 +2418,7 @@ static int sd_snapshot_create(BlockDriverState *bs, QEMUSnapshotInfo *sn_info)
goto cleanup; goto cleanup;
} }
ret = write_object(fd, s->aio_context, (char *)&s->inode, ret = write_object(fd, s->bs, (char *)&s->inode,
vid_to_vdi_oid(s->inode.vdi_id), s->inode.nr_copies, vid_to_vdi_oid(s->inode.vdi_id), s->inode.nr_copies,
datalen, 0, false, s->cache_flags); datalen, 0, false, s->cache_flags);
if (ret < 0) { if (ret < 0) {
@ -2426,7 +2433,7 @@ static int sd_snapshot_create(BlockDriverState *bs, QEMUSnapshotInfo *sn_info)
goto cleanup; goto cleanup;
} }
ret = read_object(fd, s->aio_context, (char *)inode, ret = read_object(fd, s->bs, (char *)inode,
vid_to_vdi_oid(new_vid), s->inode.nr_copies, datalen, 0, vid_to_vdi_oid(new_vid), s->inode.nr_copies, datalen, 0,
s->cache_flags); s->cache_flags);
@ -2528,7 +2535,7 @@ static bool remove_objects(BDRVSheepdogState *s)
i++; i++;
} }
ret = write_object(fd, s->aio_context, ret = write_object(fd, s->bs,
(char *)&inode->data_vdi_id[start_idx], (char *)&inode->data_vdi_id[start_idx],
vid_to_vdi_oid(s->inode.vdi_id), inode->nr_copies, vid_to_vdi_oid(s->inode.vdi_id), inode->nr_copies,
(i - start_idx) * sizeof(uint32_t), (i - start_idx) * sizeof(uint32_t),
@ -2600,7 +2607,7 @@ static int sd_snapshot_delete(BlockDriverState *bs,
return -1; return -1;
} }
ret = do_req(fd, s->aio_context, (SheepdogReq *)&hdr, ret = do_req(fd, s->bs, (SheepdogReq *)&hdr,
buf, &wlen, &rlen); buf, &wlen, &rlen);
closesocket(fd); closesocket(fd);
if (ret) { if (ret) {
@ -2652,8 +2659,7 @@ static int sd_snapshot_list(BlockDriverState *bs, QEMUSnapshotInfo **psn_tab)
req.opcode = SD_OP_READ_VDIS; req.opcode = SD_OP_READ_VDIS;
req.data_length = max; req.data_length = max;
ret = do_req(fd, s->aio_context, &req, ret = do_req(fd, s->bs, &req, vdi_inuse, &wlen, &rlen);
vdi_inuse, &wlen, &rlen);
closesocket(fd); closesocket(fd);
if (ret) { if (ret) {
@ -2679,7 +2685,7 @@ static int sd_snapshot_list(BlockDriverState *bs, QEMUSnapshotInfo **psn_tab)
} }
/* we don't need to read entire object */ /* we don't need to read entire object */
ret = read_object(fd, s->aio_context, (char *)&inode, ret = read_object(fd, s->bs, (char *)&inode,
vid_to_vdi_oid(vid), vid_to_vdi_oid(vid),
0, SD_INODE_SIZE - sizeof(inode.data_vdi_id), 0, 0, SD_INODE_SIZE - sizeof(inode.data_vdi_id), 0,
s->cache_flags); s->cache_flags);
@ -2745,11 +2751,11 @@ static int do_load_save_vmstate(BDRVSheepdogState *s, uint8_t *data,
create = (offset == 0); create = (offset == 0);
if (load) { if (load) {
ret = read_object(fd, s->aio_context, (char *)data, vmstate_oid, ret = read_object(fd, s->bs, (char *)data, vmstate_oid,
s->inode.nr_copies, data_len, offset, s->inode.nr_copies, data_len, offset,
s->cache_flags); s->cache_flags);
} else { } else {
ret = write_object(fd, s->aio_context, (char *)data, vmstate_oid, ret = write_object(fd, s->bs, (char *)data, vmstate_oid,
s->inode.nr_copies, data_len, offset, create, s->inode.nr_copies, data_len, offset, create,
s->cache_flags); s->cache_flags);
} }