From 253cca4fc3a89c38e79d2e940c5a0b7bb082afcc Mon Sep 17 00:00:00 2001 From: Zhanghongtao2417 <651380626@qq.com> Date: Fri, 26 Apr 2024 22:01:25 +0800 Subject: [PATCH] nvme/cuse: Add ctrlr_lock for cuse register and unregister conflicts: lib/nvme/nvme_io_msg.c nvme_io_msg_ctrlr_update spdk_nvme_cuse_unregister and spdk_nvme_ctrlr_process_admin_completions running at the same time, concurrently operate external_io_msgs. So we add locks to protect. Fixes #3353 Change-Id: Id5176975676c29a475e8e2a0d7c93e44646c00dc Signed-off-by: Zhanghongtao2417 <651380626@qq.com> Reviewed-on: https://review.spdk.io/gerrit/c/spdk/spdk/+/22927 Tested-by: SPDK CI Jenkins Community-CI: Mellanox Build Bot Reviewed-by: Jim Harris Reviewed-by: Konrad Sztyber --- lib/nvme/nvme_io_msg.c | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/lib/nvme/nvme_io_msg.c b/lib/nvme/nvme_io_msg.c index fb5aec3..d74736c 100644 --- a/lib/nvme/nvme_io_msg.c +++ b/lib/nvme/nvme_io_msg.c @@ -130,7 +130,9 @@ nvme_io_msg_ctrlr_register(struct spdk_nvme_ctrlr *ctrlr, return -EINVAL; } + nvme_robust_mutex_lock(&ctrlr->ctrlr_lock); if (nvme_io_msg_is_producer_registered(ctrlr, io_msg_producer)) { + nvme_robust_mutex_unlock(&ctrlr->ctrlr_lock); return -EEXIST; } @@ -148,6 +150,7 @@ nvme_io_msg_ctrlr_register(struct spdk_nvme_ctrlr *ctrlr, ctrlr->external_io_msgs = spdk_ring_create(SPDK_RING_TYPE_MP_SC, 65536, SPDK_ENV_SOCKET_ID_ANY); if (!ctrlr->external_io_msgs) { SPDK_ERRLOG("Unable to allocate memory for message ring\n"); + nvme_robust_mutex_unlock(&ctrlr->ctrlr_lock); return -ENOMEM; } @@ -156,10 +159,12 @@ nvme_io_msg_ctrlr_register(struct spdk_nvme_ctrlr *ctrlr, SPDK_ERRLOG("spdk_nvme_ctrlr_alloc_io_qpair() failed\n"); spdk_ring_free(ctrlr->external_io_msgs); ctrlr->external_io_msgs = NULL; + nvme_robust_mutex_unlock(&ctrlr->ctrlr_lock); return -ENOMEM; } STAILQ_INSERT_TAIL(&ctrlr->io_producers, io_msg_producer, link); + nvme_robust_mutex_unlock(&ctrlr->ctrlr_lock); return 0; } @@ -205,7 +210,9 @@ nvme_io_msg_ctrlr_unregister(struct spdk_nvme_ctrlr *ctrlr, { assert(io_msg_producer != NULL); + nvme_robust_mutex_lock(&ctrlr->ctrlr_lock); if (!nvme_io_msg_is_producer_registered(ctrlr, io_msg_producer)) { + nvme_robust_mutex_unlock(&ctrlr->ctrlr_lock); return; } @@ -213,4 +220,5 @@ nvme_io_msg_ctrlr_unregister(struct spdk_nvme_ctrlr *ctrlr, if (STAILQ_EMPTY(&ctrlr->io_producers)) { nvme_io_msg_ctrlr_detach(ctrlr); } + nvme_robust_mutex_unlock(&ctrlr->ctrlr_lock); } -- 2.27.0