mirror of
https://github.com/Motorhead1991/qemu.git
synced 2025-08-02 15:23:53 -06:00
block/export: Fix vhost-user-blk shutdown with requests in flight
The vhost-user-blk export runs requests asynchronously in their own coroutine. When the vhost connection goes away and we want to stop the vhost-user server, we need to wait for these coroutines to stop before we can unmap the shared memory. Otherwise, they would still access the unmapped memory and crash. This introduces a refcount to VuServer which is increased when spawning a new request coroutine and decreased before the coroutine exits. The memory is only unmapped when the refcount reaches zero. Signed-off-by: Kevin Wolf <kwolf@redhat.com> Message-Id: <20220125151435.48792-1-kwolf@redhat.com> Signed-off-by: Kevin Wolf <kwolf@redhat.com>
This commit is contained in:
parent
c0829cb1fd
commit
520d8b40e8
3 changed files with 32 additions and 0 deletions
|
@ -74,6 +74,20 @@ static void panic_cb(VuDev *vu_dev, const char *buf)
|
|||
error_report("vu_panic: %s", buf);
|
||||
}
|
||||
|
||||
void vhost_user_server_ref(VuServer *server)
|
||||
{
|
||||
assert(!server->wait_idle);
|
||||
server->refcount++;
|
||||
}
|
||||
|
||||
void vhost_user_server_unref(VuServer *server)
|
||||
{
|
||||
server->refcount--;
|
||||
if (server->wait_idle && !server->refcount) {
|
||||
aio_co_wake(server->co_trip);
|
||||
}
|
||||
}
|
||||
|
||||
static bool coroutine_fn
|
||||
vu_message_read(VuDev *vu_dev, int conn_fd, VhostUserMsg *vmsg)
|
||||
{
|
||||
|
@ -177,6 +191,14 @@ static coroutine_fn void vu_client_trip(void *opaque)
|
|||
/* Keep running */
|
||||
}
|
||||
|
||||
if (server->refcount) {
|
||||
/* Wait for requests to complete before we can unmap the memory */
|
||||
server->wait_idle = true;
|
||||
qemu_coroutine_yield();
|
||||
server->wait_idle = false;
|
||||
}
|
||||
assert(server->refcount == 0);
|
||||
|
||||
vu_deinit(vu_dev);
|
||||
|
||||
/* vu_deinit() should have called remove_watch() */
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue