Signed-off-by: Changzhi Xie <s...@qq.com> --- block/export/fuse.c | 167 ++++++++++++++++++++++++++++++-------------- 1 file changed, 114 insertions(+), 53 deletions(-)
diff --git a/block/export/fuse.c b/block/export/fuse.c index 465cc9891d..f47117a00d 100644 --- a/block/export/fuse.c +++ b/block/export/fuse.c @@ -64,6 +64,16 @@ typedef struct FuseExport { gid_t st_gid; } FuseExport; +typedef struct FuseIORequest { + fuse_req_t req; + fuse_ino_t inode; + size_t size; + off_t offset; + struct fuse_file_info *fi; + FuseExport *exp; + char *write_buf; +} FuseIORequest; + static GHashTable *exports; static const struct fuse_lowlevel_ops fuse_ops; @@ -570,102 +580,153 @@ static void fuse_open(fuse_req_t req, fuse_ino_t inode, fuse_reply_open(req, fi); } -/** - * Handle client reads from the exported image. - */ -static void fuse_read(fuse_req_t req, fuse_ino_t inode, - size_t size, off_t offset, struct fuse_file_info *fi) +static void coroutine_fn fuse_read_coroutine(void *opaque) { - FuseExport *exp = fuse_req_userdata(req); + FuseIORequest *io_req = opaque; + FuseExport *exp = io_req->exp; int64_t length; - void *buf; + void *buffer; int ret; - /* Limited by max_read, should not happen */ - if (size > FUSE_MAX_BOUNCE_BYTES) { - fuse_reply_err(req, EINVAL); - return; + if (io_req->size > FUSE_MAX_BOUNCE_BYTES) { + fuse_reply_err(io_req->req, EINVAL); + goto cleanup; } - /** - * Clients will expect short reads at EOF, so we have to limit - * offset+size to the image length. - */ length = blk_getlength(exp->common.blk); if (length < 0) { - fuse_reply_err(req, -length); - return; + fuse_reply_err(io_req->req, -length); + goto cleanup; } - if (offset + size > length) { - size = length - offset; + if (io_req->offset + io_req->size > length) { + io_req->size = length - io_req->offset; } - buf = qemu_try_blockalign(blk_bs(exp->common.blk), size); - if (!buf) { - fuse_reply_err(req, ENOMEM); - return; + if (io_req->size == 0) { + fuse_reply_buf(io_req->req, NULL, 0); + goto cleanup; + } + + buffer = qemu_try_blockalign(blk_bs(exp->common.blk), io_req->size); + if (!buffer) { + fuse_reply_err(io_req->req, ENOMEM); + goto cleanup; } - ret = blk_pread(exp->common.blk, offset, size, buf, 0); + ret = blk_co_pread(exp->common.blk, io_req->offset, + io_req->size, buffer, 0); if (ret >= 0) { - fuse_reply_buf(req, buf, size); + fuse_reply_buf(io_req->req, buffer, io_req->size); } else { - fuse_reply_err(req, -ret); + fuse_reply_err(io_req->req, -ret); } - qemu_vfree(buf); + qemu_vfree(buffer); + +cleanup: + g_free(io_req); } -/** - * Handle client writes to the exported image. - */ -static void fuse_write(fuse_req_t req, fuse_ino_t inode, const char *buf, - size_t size, off_t offset, struct fuse_file_info *fi) +static void coroutine_fn fuse_write_coroutine(void *opaque) { - FuseExport *exp = fuse_req_userdata(req); + FuseIORequest *io_req = opaque; + FuseExport *exp = io_req->exp; int64_t length; int ret; - /* Limited by max_write, should not happen */ - if (size > BDRV_REQUEST_MAX_BYTES) { - fuse_reply_err(req, EINVAL); - return; + if (io_req->size > BDRV_REQUEST_MAX_BYTES) { + fuse_reply_err(io_req->req, EINVAL); + goto cleanup; } if (!exp->writable) { - fuse_reply_err(req, EACCES); - return; + fuse_reply_err(io_req->req, EACCES); + goto cleanup; } - /** - * Clients will expect short writes at EOF, so we have to limit - * offset+size to the image length. - */ length = blk_getlength(exp->common.blk); if (length < 0) { - fuse_reply_err(req, -length); - return; + fuse_reply_err(io_req->req, -length); + goto cleanup; } - if (offset + size > length) { + if (io_req->offset + io_req->size > length) { if (exp->growable) { - ret = fuse_do_truncate(exp, offset + size, true, PREALLOC_MODE_OFF); + ret = fuse_do_truncate(exp, io_req->offset + io_req->size, + true, PREALLOC_MODE_OFF); if (ret < 0) { - fuse_reply_err(req, -ret); - return; + fuse_reply_err(io_req->req, -ret); + goto cleanup; } } else { - size = length - offset; + io_req->size = MAX(0, length - io_req->offset); + if (io_req->size == 0) { + fuse_reply_write(io_req->req, 0); + goto cleanup; + } } } - ret = blk_pwrite(exp->common.blk, offset, size, buf, 0); + ret = blk_co_pwrite(exp->common.blk, io_req->offset, io_req->size, + io_req->write_buf, 0); if (ret >= 0) { - fuse_reply_write(req, size); + fuse_reply_write(io_req->req, io_req->size); } else { - fuse_reply_err(req, -ret); + fuse_reply_err(io_req->req, -ret); } + +cleanup: + g_free(io_req->write_buf); + g_free(io_req); +} + +/** + * Handle client reads from the exported image. + */ +static void fuse_read(fuse_req_t req, fuse_ino_t inode, + size_t size, off_t offset, struct fuse_file_info *fi) +{ + FuseExport *exp = fuse_req_userdata(req); + FuseIORequest *io_req = g_new(FuseIORequest, 1); + io_req->req = req; + io_req->inode = inode; + io_req->size = size; + io_req->offset = offset; + io_req->fi = fi; + io_req->exp = exp; + + Coroutine *co = qemu_coroutine_create(fuse_read_coroutine, io_req); + qemu_coroutine_enter(co); +} + + +/** + * Handle client writes to the exported image. + */ +static void fuse_write(fuse_req_t req, fuse_ino_t inode, const char *buf, + size_t size, off_t offset, struct fuse_file_info *fi) +{ + FuseExport *exp = fuse_req_userdata(req); + FuseIORequest *io_req = g_new(FuseIORequest, 1); + + io_req->write_buf = g_try_malloc(size); + if (!io_req->write_buf) { + fuse_reply_err(req, ENOMEM); + g_free(io_req); + return; + } + memcpy(io_req->write_buf, buf, size); + + io_req->req = req; + io_req->inode = inode; + io_req->size = size; + io_req->offset = offset; + io_req->fi = fi; + io_req->exp = exp; + + Coroutine *co = qemu_coroutine_create(fuse_write_coroutine, io_req); + qemu_coroutine_enter(co); } /** -- 2.34.1 >From 3d2d317a49eb4cbf401294c6e19c72533eeeefad Mon Sep 17 00:00:00 2001 From: saz97 <s...@qq.com> Date: Wed, 12 Mar 2025 13:47:01 +0800 Subject: [PATCH 2/2] allocate independent fuse_buffor each coroutine and store fuse_file_info copy instead of pointerin FuseIORequest Signed-off-by: Changzhi Xie <s...@qq.com> --- block/export/fuse.c | 45 ++++++++++++++++++++++++--------------------- 1 file changed, 24 insertions(+), 21 deletions(-) diff --git a/block/export/fuse.c b/block/export/fuse.c index f47117a00d..69ffe4f0ca 100644 --- a/block/export/fuse.c +++ b/block/export/fuse.c @@ -69,7 +69,7 @@ typedef struct FuseIORequest { fuse_ino_t inode; size_t size; off_t offset; - struct fuse_file_info *fi; + struct fuse_file_info fi; FuseExport *exp; char *write_buf; } FuseIORequest; @@ -298,6 +298,10 @@ fail: static void read_from_fuse_export(void *opaque) { FuseExport *exp = opaque; + struct fuse_buf buf = { + .mem = g_malloc(FUSE_MAX_BOUNCE_BYTES), + .size = FUSE_MAX_BOUNCE_BYTES, + }; int ret; blk_exp_ref(&exp->common); @@ -314,6 +318,7 @@ static void read_from_fuse_export(void *opaque) fuse_session_process_buf(exp->fuse_session, &exp->fuse_buf); out: + g_free(buf.mem); if (qatomic_fetch_dec(&exp->in_flight) == 1) { aio_wait_kick(); /* wake AIO_WAIT_WHILE() */ } @@ -689,12 +694,15 @@ static void fuse_read(fuse_req_t req, fuse_ino_t inode, { FuseExport *exp = fuse_req_userdata(req); FuseIORequest *io_req = g_new(FuseIORequest, 1); - io_req->req = req; - io_req->inode = inode; - io_req->size = size; - io_req->offset = offset; - io_req->fi = fi; - io_req->exp = exp; + + *io_req = (FuseIORequest) { + .req = req, + .inode = inode, + .size = size, + .offset = offset, + .exp = exp, + .fi = *fi, + }; Coroutine *co = qemu_coroutine_create(fuse_read_coroutine, io_req); qemu_coroutine_enter(co); @@ -710,20 +718,15 @@ static void fuse_write(fuse_req_t req, fuse_ino_t inode, const char *buf, FuseExport *exp = fuse_req_userdata(req); FuseIORequest *io_req = g_new(FuseIORequest, 1); - io_req->write_buf = g_try_malloc(size); - if (!io_req->write_buf) { - fuse_reply_err(req, ENOMEM); - g_free(io_req); - return; - } - memcpy(io_req->write_buf, buf, size); - - io_req->req = req; - io_req->inode = inode; - io_req->size = size; - io_req->offset = offset; - io_req->fi = fi; - io_req->exp = exp; + *io_req = (FuseIORequest) { + .req = req, + .inode = inode, + .size = size, + .offset = offset, + .exp = exp, + .fi = *fi, + .write_buf = g_memdup2_qemu(buf, size), + }; Coroutine *co = qemu_coroutine_create(fuse_write_coroutine, io_req); qemu_coroutine_enter(co); -- 2.34.1