summaryrefslogtreecommitdiffstats
path: root/src/spdk/lib/thread
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 18:24:20 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 18:24:20 +0000
commit483eb2f56657e8e7f419ab1a4fab8dce9ade8609 (patch)
treee5d88d25d870d5dedacb6bbdbe2a966086a0a5cf /src/spdk/lib/thread
parentInitial commit. (diff)
downloadceph-483eb2f56657e8e7f419ab1a4fab8dce9ade8609.tar.xz
ceph-483eb2f56657e8e7f419ab1a4fab8dce9ade8609.zip
Adding upstream version 14.2.21.upstream/14.2.21upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/spdk/lib/thread')
-rw-r--r--src/spdk/lib/thread/Makefile40
-rw-r--r--src/spdk/lib/thread/thread.c768
2 files changed, 808 insertions, 0 deletions
diff --git a/src/spdk/lib/thread/Makefile b/src/spdk/lib/thread/Makefile
new file mode 100644
index 00000000..467e32ff
--- /dev/null
+++ b/src/spdk/lib/thread/Makefile
@@ -0,0 +1,40 @@
+#
+# BSD LICENSE
+#
+# Copyright (c) Intel Corporation.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+#
+# * Redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer.
+# * Redistributions in binary form must reproduce the above copyright
+# notice, this list of conditions and the following disclaimer in
+# the documentation and/or other materials provided with the
+# distribution.
+# * Neither the name of Intel Corporation nor the names of its
+# contributors may be used to endorse or promote products derived
+# from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+
+SPDK_ROOT_DIR := $(abspath $(CURDIR)/../..)
+include $(SPDK_ROOT_DIR)/mk/spdk.common.mk
+
+C_SRCS = thread.c
+LIBNAME = thread
+
+include $(SPDK_ROOT_DIR)/mk/spdk.lib.mk
diff --git a/src/spdk/lib/thread/thread.c b/src/spdk/lib/thread/thread.c
new file mode 100644
index 00000000..c014f4ed
--- /dev/null
+++ b/src/spdk/lib/thread/thread.c
@@ -0,0 +1,768 @@
+/*-
+ * BSD LICENSE
+ *
+ * Copyright (c) Intel Corporation.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * * Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in
+ * the documentation and/or other materials provided with the
+ * distribution.
+ * * Neither the name of Intel Corporation nor the names of its
+ * contributors may be used to endorse or promote products derived
+ * from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "spdk/stdinc.h"
+
+#include "spdk/string.h"
+#include "spdk/thread.h"
+
+#include "spdk_internal/log.h"
+
+#ifdef __linux__
+#include <sys/prctl.h>
+#endif
+
+#ifdef __FreeBSD__
+#include <pthread_np.h>
+#endif
+
+static pthread_mutex_t g_devlist_mutex = PTHREAD_MUTEX_INITIALIZER;
+
+struct io_device {
+ void *io_device;
+ char *name;
+ spdk_io_channel_create_cb create_cb;
+ spdk_io_channel_destroy_cb destroy_cb;
+ spdk_io_device_unregister_cb unregister_cb;
+ struct spdk_thread *unregister_thread;
+ uint32_t ctx_size;
+ uint32_t for_each_count;
+ TAILQ_ENTRY(io_device) tailq;
+
+ uint32_t refcnt;
+
+ bool unregistered;
+};
+
+static TAILQ_HEAD(, io_device) g_io_devices = TAILQ_HEAD_INITIALIZER(g_io_devices);
+
+struct spdk_thread {
+ pthread_t thread_id;
+ spdk_thread_pass_msg msg_fn;
+ spdk_start_poller start_poller_fn;
+ spdk_stop_poller stop_poller_fn;
+ void *thread_ctx;
+ TAILQ_HEAD(, spdk_io_channel) io_channels;
+ TAILQ_ENTRY(spdk_thread) tailq;
+ char *name;
+};
+
+static TAILQ_HEAD(, spdk_thread) g_threads = TAILQ_HEAD_INITIALIZER(g_threads);
+static uint32_t g_thread_count = 0;
+
+static struct spdk_thread *
+_get_thread(void)
+{
+ pthread_t thread_id;
+ struct spdk_thread *thread;
+
+ thread_id = pthread_self();
+
+ thread = NULL;
+ TAILQ_FOREACH(thread, &g_threads, tailq) {
+ if (thread->thread_id == thread_id) {
+ return thread;
+ }
+ }
+
+ return NULL;
+}
+
+static void
+_set_thread_name(const char *thread_name)
+{
+#if defined(__linux__)
+ prctl(PR_SET_NAME, thread_name, 0, 0, 0);
+#elif defined(__FreeBSD__)
+ pthread_set_name_np(pthread_self(), thread_name);
+#else
+#error missing platform support for thread name
+#endif
+}
+
+int
+spdk_thread_lib_init(void)
+{
+ return 0;
+}
+
+void
+spdk_thread_lib_fini(void)
+{
+}
+
+struct spdk_thread *
+spdk_allocate_thread(spdk_thread_pass_msg msg_fn,
+ spdk_start_poller start_poller_fn,
+ spdk_stop_poller stop_poller_fn,
+ void *thread_ctx, const char *name)
+{
+ struct spdk_thread *thread;
+
+ pthread_mutex_lock(&g_devlist_mutex);
+
+ thread = _get_thread();
+ if (thread) {
+ SPDK_ERRLOG("Double allocated SPDK thread\n");
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return NULL;
+ }
+
+ thread = calloc(1, sizeof(*thread));
+ if (!thread) {
+ SPDK_ERRLOG("Unable to allocate memory for thread\n");
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return NULL;
+ }
+
+ thread->thread_id = pthread_self();
+ thread->msg_fn = msg_fn;
+ thread->start_poller_fn = start_poller_fn;
+ thread->stop_poller_fn = stop_poller_fn;
+ thread->thread_ctx = thread_ctx;
+ TAILQ_INIT(&thread->io_channels);
+ TAILQ_INSERT_TAIL(&g_threads, thread, tailq);
+ g_thread_count++;
+ if (name) {
+ _set_thread_name(name);
+ thread->name = strdup(name);
+ } else {
+ thread->name = spdk_sprintf_alloc("%p", thread);
+ }
+
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Allocating new thread %s\n", thread->name);
+
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ return thread;
+}
+
+void
+spdk_free_thread(void)
+{
+ struct spdk_thread *thread;
+
+ pthread_mutex_lock(&g_devlist_mutex);
+
+ thread = _get_thread();
+ if (!thread) {
+ SPDK_ERRLOG("No thread allocated\n");
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return;
+ }
+
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Freeing thread %s\n", thread->name);
+
+ assert(g_thread_count > 0);
+ g_thread_count--;
+ TAILQ_REMOVE(&g_threads, thread, tailq);
+ free(thread->name);
+ free(thread);
+
+ pthread_mutex_unlock(&g_devlist_mutex);
+}
+
+uint32_t
+spdk_thread_get_count(void)
+{
+ /*
+ * Return cached value of the current thread count. We could acquire the
+ * lock and iterate through the TAILQ of threads to count them, but that
+ * count could still be invalidated after we release the lock.
+ */
+ return g_thread_count;
+}
+
+struct spdk_thread *
+spdk_get_thread(void)
+{
+ struct spdk_thread *thread;
+
+ pthread_mutex_lock(&g_devlist_mutex);
+
+ thread = _get_thread();
+ if (!thread) {
+ SPDK_ERRLOG("No thread allocated\n");
+ }
+
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ return thread;
+}
+
+const char *
+spdk_thread_get_name(const struct spdk_thread *thread)
+{
+ return thread->name;
+}
+
+void
+spdk_thread_send_msg(const struct spdk_thread *thread, spdk_thread_fn fn, void *ctx)
+{
+ thread->msg_fn(fn, ctx, thread->thread_ctx);
+}
+
+
+struct spdk_poller *
+spdk_poller_register(spdk_poller_fn fn,
+ void *arg,
+ uint64_t period_microseconds)
+{
+ struct spdk_thread *thread;
+ struct spdk_poller *poller;
+
+ thread = spdk_get_thread();
+ if (!thread) {
+ assert(false);
+ return NULL;
+ }
+
+ if (!thread->start_poller_fn || !thread->stop_poller_fn) {
+ SPDK_ERRLOG("No related functions to start requested poller\n");
+ assert(false);
+ return NULL;
+ }
+
+ poller = thread->start_poller_fn(thread->thread_ctx, fn, arg, period_microseconds);
+ if (!poller) {
+ SPDK_ERRLOG("Unable to start requested poller\n");
+ assert(false);
+ return NULL;
+ }
+
+ return poller;
+}
+
+void
+spdk_poller_unregister(struct spdk_poller **ppoller)
+{
+ struct spdk_thread *thread;
+ struct spdk_poller *poller;
+
+ poller = *ppoller;
+ if (poller == NULL) {
+ return;
+ }
+
+ *ppoller = NULL;
+
+ thread = spdk_get_thread();
+
+ if (thread) {
+ thread->stop_poller_fn(poller, thread->thread_ctx);
+ }
+}
+
+struct call_thread {
+ struct spdk_thread *cur_thread;
+ spdk_thread_fn fn;
+ void *ctx;
+
+ struct spdk_thread *orig_thread;
+ spdk_thread_fn cpl;
+};
+
+static void
+spdk_on_thread(void *ctx)
+{
+ struct call_thread *ct = ctx;
+
+ ct->fn(ct->ctx);
+
+ pthread_mutex_lock(&g_devlist_mutex);
+ ct->cur_thread = TAILQ_NEXT(ct->cur_thread, tailq);
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ if (!ct->cur_thread) {
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Completed thread iteration\n");
+
+ spdk_thread_send_msg(ct->orig_thread, ct->cpl, ct->ctx);
+ free(ctx);
+ } else {
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Continuing thread iteration to %s\n",
+ ct->cur_thread->name);
+
+ spdk_thread_send_msg(ct->cur_thread, spdk_on_thread, ctx);
+ }
+}
+
+void
+spdk_for_each_thread(spdk_thread_fn fn, void *ctx, spdk_thread_fn cpl)
+{
+ struct call_thread *ct;
+
+ ct = calloc(1, sizeof(*ct));
+ if (!ct) {
+ SPDK_ERRLOG("Unable to perform thread iteration\n");
+ cpl(ctx);
+ return;
+ }
+
+ ct->fn = fn;
+ ct->ctx = ctx;
+ ct->cpl = cpl;
+
+ pthread_mutex_lock(&g_devlist_mutex);
+ ct->orig_thread = _get_thread();
+ ct->cur_thread = TAILQ_FIRST(&g_threads);
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Starting thread iteration from %s\n",
+ ct->orig_thread->name);
+
+ spdk_thread_send_msg(ct->cur_thread, spdk_on_thread, ct);
+}
+
+void
+spdk_io_device_register(void *io_device, spdk_io_channel_create_cb create_cb,
+ spdk_io_channel_destroy_cb destroy_cb, uint32_t ctx_size,
+ const char *name)
+{
+ struct io_device *dev, *tmp;
+
+ assert(io_device != NULL);
+ assert(create_cb != NULL);
+ assert(destroy_cb != NULL);
+
+ dev = calloc(1, sizeof(struct io_device));
+ if (dev == NULL) {
+ SPDK_ERRLOG("could not allocate io_device\n");
+ return;
+ }
+
+ dev->io_device = io_device;
+ if (name) {
+ dev->name = strdup(name);
+ } else {
+ dev->name = spdk_sprintf_alloc("%p", dev);
+ }
+ dev->create_cb = create_cb;
+ dev->destroy_cb = destroy_cb;
+ dev->unregister_cb = NULL;
+ dev->ctx_size = ctx_size;
+ dev->for_each_count = 0;
+ dev->unregistered = false;
+ dev->refcnt = 0;
+
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Registering io_device %s (%p) on thread %s\n",
+ dev->name, dev->io_device, spdk_get_thread()->name);
+
+ pthread_mutex_lock(&g_devlist_mutex);
+ TAILQ_FOREACH(tmp, &g_io_devices, tailq) {
+ if (tmp->io_device == io_device) {
+ SPDK_ERRLOG("io_device %p already registered\n", io_device);
+ free(dev->name);
+ free(dev);
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return;
+ }
+ }
+ TAILQ_INSERT_TAIL(&g_io_devices, dev, tailq);
+ pthread_mutex_unlock(&g_devlist_mutex);
+}
+
+static void
+_finish_unregister(void *arg)
+{
+ struct io_device *dev = arg;
+
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Finishing unregistration of io_device %s (%p) on thread %s\n",
+ dev->name, dev->io_device, dev->unregister_thread->name);
+
+ dev->unregister_cb(dev->io_device);
+ free(dev->name);
+ free(dev);
+}
+
+static void
+_spdk_io_device_free(struct io_device *dev)
+{
+ if (dev->unregister_cb == NULL) {
+ free(dev->name);
+ free(dev);
+ } else {
+ assert(dev->unregister_thread != NULL);
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "io_device %s (%p) needs to unregister from thread %s\n",
+ dev->name, dev->io_device, dev->unregister_thread->name);
+ spdk_thread_send_msg(dev->unregister_thread, _finish_unregister, dev);
+ }
+}
+
+void
+spdk_io_device_unregister(void *io_device, spdk_io_device_unregister_cb unregister_cb)
+{
+ struct io_device *dev;
+ uint32_t refcnt;
+ struct spdk_thread *thread;
+
+ thread = spdk_get_thread();
+
+ pthread_mutex_lock(&g_devlist_mutex);
+ TAILQ_FOREACH(dev, &g_io_devices, tailq) {
+ if (dev->io_device == io_device) {
+ break;
+ }
+ }
+
+ if (!dev) {
+ SPDK_ERRLOG("io_device %p not found\n", io_device);
+ assert(false);
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return;
+ }
+
+ if (dev->for_each_count > 0) {
+ SPDK_ERRLOG("io_device %p has %u for_each calls outstanding\n", io_device, dev->for_each_count);
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return;
+ }
+
+ dev->unregister_cb = unregister_cb;
+ dev->unregistered = true;
+ TAILQ_REMOVE(&g_io_devices, dev, tailq);
+ refcnt = dev->refcnt;
+ dev->unregister_thread = thread;
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Unregistering io_device %s (%p) from thread %s\n",
+ dev->name, dev->io_device, thread->name);
+
+ if (refcnt > 0) {
+ /* defer deletion */
+ return;
+ }
+
+ _spdk_io_device_free(dev);
+}
+
+struct spdk_io_channel *
+spdk_get_io_channel(void *io_device)
+{
+ struct spdk_io_channel *ch;
+ struct spdk_thread *thread;
+ struct io_device *dev;
+ int rc;
+
+ pthread_mutex_lock(&g_devlist_mutex);
+ TAILQ_FOREACH(dev, &g_io_devices, tailq) {
+ if (dev->io_device == io_device) {
+ break;
+ }
+ }
+ if (dev == NULL) {
+ SPDK_ERRLOG("could not find io_device %p\n", io_device);
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return NULL;
+ }
+
+ thread = _get_thread();
+ if (!thread) {
+ SPDK_ERRLOG("No thread allocated\n");
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return NULL;
+ }
+
+ TAILQ_FOREACH(ch, &thread->io_channels, tailq) {
+ if (ch->dev == dev) {
+ ch->ref++;
+
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Get io_channel %p for io_device %s (%p) on thread %s refcnt %u\n",
+ ch, dev->name, dev->io_device, thread->name, ch->ref);
+
+ /*
+ * An I/O channel already exists for this device on this
+ * thread, so return it.
+ */
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return ch;
+ }
+ }
+
+ ch = calloc(1, sizeof(*ch) + dev->ctx_size);
+ if (ch == NULL) {
+ SPDK_ERRLOG("could not calloc spdk_io_channel\n");
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return NULL;
+ }
+
+ ch->dev = dev;
+ ch->destroy_cb = dev->destroy_cb;
+ ch->thread = thread;
+ ch->ref = 1;
+ ch->destroy_ref = 0;
+ TAILQ_INSERT_TAIL(&thread->io_channels, ch, tailq);
+
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD, "Get io_channel %p for io_device %s (%p) on thread %s refcnt %u\n",
+ ch, dev->name, dev->io_device, thread->name, ch->ref);
+
+ dev->refcnt++;
+
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ rc = dev->create_cb(io_device, (uint8_t *)ch + sizeof(*ch));
+ if (rc == -1) {
+ pthread_mutex_lock(&g_devlist_mutex);
+ TAILQ_REMOVE(&ch->thread->io_channels, ch, tailq);
+ dev->refcnt--;
+ free(ch);
+ pthread_mutex_unlock(&g_devlist_mutex);
+ return NULL;
+ }
+
+ return ch;
+}
+
+static void
+_spdk_put_io_channel(void *arg)
+{
+ struct spdk_io_channel *ch = arg;
+ bool do_remove_dev = true;
+
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD,
+ "Releasing io_channel %p for io_device %s (%p). Channel thread %p. Current thread %s\n",
+ ch, ch->dev->name, ch->dev->io_device, ch->thread, spdk_get_thread()->name);
+
+ assert(ch->thread == spdk_get_thread());
+
+ ch->destroy_ref--;
+
+ if (ch->ref > 0 || ch->destroy_ref > 0) {
+ /*
+ * Another reference to the associated io_device was requested
+ * after this message was sent but before it had a chance to
+ * execute.
+ */
+ return;
+ }
+
+ pthread_mutex_lock(&g_devlist_mutex);
+ TAILQ_REMOVE(&ch->thread->io_channels, ch, tailq);
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ /* Don't hold the devlist mutex while the destroy_cb is called. */
+ ch->destroy_cb(ch->dev->io_device, spdk_io_channel_get_ctx(ch));
+
+ pthread_mutex_lock(&g_devlist_mutex);
+ ch->dev->refcnt--;
+
+ if (!ch->dev->unregistered) {
+ do_remove_dev = false;
+ }
+
+ if (ch->dev->refcnt > 0) {
+ do_remove_dev = false;
+ }
+
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ if (do_remove_dev) {
+ _spdk_io_device_free(ch->dev);
+ }
+ free(ch);
+}
+
+void
+spdk_put_io_channel(struct spdk_io_channel *ch)
+{
+ SPDK_DEBUGLOG(SPDK_LOG_THREAD,
+ "Putting io_channel %p for io_device %s (%p) on thread %s refcnt %u\n",
+ ch, ch->dev->name, ch->dev->io_device, ch->thread->name, ch->ref);
+
+ ch->ref--;
+
+ if (ch->ref == 0) {
+ ch->destroy_ref++;
+ spdk_thread_send_msg(ch->thread, _spdk_put_io_channel, ch);
+ }
+}
+
+struct spdk_io_channel *
+spdk_io_channel_from_ctx(void *ctx)
+{
+ return (struct spdk_io_channel *)((uint8_t *)ctx - sizeof(struct spdk_io_channel));
+}
+
+struct spdk_thread *
+spdk_io_channel_get_thread(struct spdk_io_channel *ch)
+{
+ return ch->thread;
+}
+
+struct spdk_io_channel_iter {
+ void *io_device;
+ struct io_device *dev;
+ spdk_channel_msg fn;
+ int status;
+ void *ctx;
+ struct spdk_io_channel *ch;
+
+ struct spdk_thread *cur_thread;
+
+ struct spdk_thread *orig_thread;
+ spdk_channel_for_each_cpl cpl;
+};
+
+void *
+spdk_io_channel_iter_get_io_device(struct spdk_io_channel_iter *i)
+{
+ return i->io_device;
+}
+
+struct spdk_io_channel *
+spdk_io_channel_iter_get_channel(struct spdk_io_channel_iter *i)
+{
+ return i->ch;
+}
+
+void *
+spdk_io_channel_iter_get_ctx(struct spdk_io_channel_iter *i)
+{
+ return i->ctx;
+}
+
+static void
+_call_completion(void *ctx)
+{
+ struct spdk_io_channel_iter *i = ctx;
+
+ if (i->cpl != NULL) {
+ i->cpl(i, i->status);
+ }
+ free(i);
+}
+
+static void
+_call_channel(void *ctx)
+{
+ struct spdk_io_channel_iter *i = ctx;
+ struct spdk_io_channel *ch;
+
+ /*
+ * It is possible that the channel was deleted before this
+ * message had a chance to execute. If so, skip calling
+ * the fn() on this thread.
+ */
+ pthread_mutex_lock(&g_devlist_mutex);
+ TAILQ_FOREACH(ch, &i->cur_thread->io_channels, tailq) {
+ if (ch->dev->io_device == i->io_device) {
+ break;
+ }
+ }
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ if (ch) {
+ i->fn(i);
+ } else {
+ spdk_for_each_channel_continue(i, 0);
+ }
+}
+
+void
+spdk_for_each_channel(void *io_device, spdk_channel_msg fn, void *ctx,
+ spdk_channel_for_each_cpl cpl)
+{
+ struct spdk_thread *thread;
+ struct spdk_io_channel *ch;
+ struct spdk_io_channel_iter *i;
+
+ i = calloc(1, sizeof(*i));
+ if (!i) {
+ SPDK_ERRLOG("Unable to allocate iterator\n");
+ return;
+ }
+
+ i->io_device = io_device;
+ i->fn = fn;
+ i->ctx = ctx;
+ i->cpl = cpl;
+
+ pthread_mutex_lock(&g_devlist_mutex);
+ i->orig_thread = _get_thread();
+
+ TAILQ_FOREACH(thread, &g_threads, tailq) {
+ TAILQ_FOREACH(ch, &thread->io_channels, tailq) {
+ if (ch->dev->io_device == io_device) {
+ ch->dev->for_each_count++;
+ i->dev = ch->dev;
+ i->cur_thread = thread;
+ i->ch = ch;
+ pthread_mutex_unlock(&g_devlist_mutex);
+ spdk_thread_send_msg(thread, _call_channel, i);
+ return;
+ }
+ }
+ }
+
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ spdk_thread_send_msg(i->orig_thread, _call_completion, i);
+}
+
+void
+spdk_for_each_channel_continue(struct spdk_io_channel_iter *i, int status)
+{
+ struct spdk_thread *thread;
+ struct spdk_io_channel *ch;
+
+ assert(i->cur_thread == spdk_get_thread());
+
+ i->status = status;
+
+ pthread_mutex_lock(&g_devlist_mutex);
+ if (status) {
+ goto end;
+ }
+ thread = TAILQ_NEXT(i->cur_thread, tailq);
+ while (thread) {
+ TAILQ_FOREACH(ch, &thread->io_channels, tailq) {
+ if (ch->dev->io_device == i->io_device) {
+ i->cur_thread = thread;
+ i->ch = ch;
+ pthread_mutex_unlock(&g_devlist_mutex);
+ spdk_thread_send_msg(thread, _call_channel, i);
+ return;
+ }
+ }
+ thread = TAILQ_NEXT(thread, tailq);
+ }
+
+end:
+ i->dev->for_each_count--;
+ i->ch = NULL;
+ pthread_mutex_unlock(&g_devlist_mutex);
+
+ spdk_thread_send_msg(i->orig_thread, _call_completion, i);
+}
+
+
+SPDK_LOG_REGISTER_COMPONENT("thread", SPDK_LOG_THREAD)