io_channel: No longer use thread-local storage
This serves two purposes. First, some older compiler chains don't support thread local storage. Second, we're going to need a way to iterate the list of threads in the future, so keep them in a list. Change-Id: I80e709f4665afb03cf4bcf0db19ef8ea353acdc1 Signed-off-by: Ben Walker <benjamin.walker@intel.com> Reviewed-on: https://review.gerrithub.io/362255 Tested-by: SPDK Automated Test System <sys_sgsw@intel.com> Reviewed-by: Jim Harris <james.r.harris@intel.com>
This commit is contained in:
parent
666bf939f1
commit
a30ffb986e
@ -49,7 +49,7 @@ struct io_device {
|
||||
static TAILQ_HEAD(, io_device) g_io_devices = TAILQ_HEAD_INITIALIZER(g_io_devices);
|
||||
|
||||
struct spdk_io_channel {
|
||||
pthread_t thread_id;
|
||||
struct spdk_thread *thread;
|
||||
void *io_device;
|
||||
uint32_t ref;
|
||||
TAILQ_ENTRY(spdk_io_channel) tailq;
|
||||
@ -64,32 +64,100 @@ struct spdk_io_channel {
|
||||
};
|
||||
|
||||
struct spdk_thread {
|
||||
pthread_t thread_id;
|
||||
spdk_thread_pass_msg fn;
|
||||
void *thread_ctx;
|
||||
TAILQ_HEAD(, spdk_io_channel) io_channels;
|
||||
TAILQ_ENTRY(spdk_thread) tailq;
|
||||
};
|
||||
|
||||
static __thread struct spdk_thread g_thread;
|
||||
static TAILQ_HEAD(, spdk_thread) g_threads = TAILQ_HEAD_INITIALIZER(g_threads);
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
struct spdk_thread *
|
||||
spdk_allocate_thread(spdk_thread_pass_msg fn, void *thread_ctx)
|
||||
{
|
||||
g_thread.fn = fn;
|
||||
g_thread.thread_ctx = thread_ctx;
|
||||
TAILQ_INIT(&g_thread.io_channels);
|
||||
return &g_thread;
|
||||
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->fn = fn;
|
||||
thread->thread_ctx = thread_ctx;
|
||||
TAILQ_INIT(&thread->io_channels);
|
||||
TAILQ_INSERT_TAIL(&g_threads, thread, tailq);
|
||||
|
||||
pthread_mutex_unlock(&g_devlist_mutex);
|
||||
|
||||
return thread;
|
||||
}
|
||||
|
||||
void
|
||||
spdk_free_thread(void)
|
||||
{
|
||||
assert(TAILQ_EMPTY(&g_thread.io_channels));
|
||||
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;
|
||||
}
|
||||
|
||||
TAILQ_REMOVE(&g_threads, thread, tailq);
|
||||
free(thread);
|
||||
|
||||
pthread_mutex_unlock(&g_devlist_mutex);
|
||||
}
|
||||
|
||||
struct spdk_thread *
|
||||
spdk_get_thread(void)
|
||||
{
|
||||
return &g_thread;
|
||||
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;
|
||||
}
|
||||
|
||||
void
|
||||
@ -150,6 +218,7 @@ 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;
|
||||
|
||||
@ -164,9 +233,17 @@ spdk_get_io_channel(void *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;
|
||||
}
|
||||
|
||||
pthread_mutex_unlock(&g_devlist_mutex);
|
||||
|
||||
TAILQ_FOREACH(ch, &g_thread.io_channels, tailq) {
|
||||
TAILQ_FOREACH(ch, &thread->io_channels, tailq) {
|
||||
if (ch->io_device == io_device) {
|
||||
ch->ref++;
|
||||
/*
|
||||
@ -190,9 +267,9 @@ spdk_get_io_channel(void *io_device)
|
||||
|
||||
ch->io_device = io_device;
|
||||
ch->destroy_cb = dev->destroy_cb;
|
||||
ch->thread_id = pthread_self();
|
||||
ch->thread = thread;
|
||||
ch->ref = 1;
|
||||
TAILQ_INSERT_TAIL(&g_thread.io_channels, ch, tailq);
|
||||
TAILQ_INSERT_TAIL(&thread->io_channels, ch, tailq);
|
||||
return ch;
|
||||
}
|
||||
|
||||
@ -207,7 +284,7 @@ spdk_put_io_channel(struct spdk_io_channel *ch)
|
||||
ch->ref--;
|
||||
|
||||
if (ch->ref == 0) {
|
||||
TAILQ_REMOVE(&g_thread.io_channels, ch, tailq);
|
||||
TAILQ_REMOVE(&ch->thread->io_channels, ch, tailq);
|
||||
ch->destroy_cb(ch->io_device, (uint8_t *)ch + sizeof(*ch));
|
||||
free(ch);
|
||||
}
|
||||
|
@ -150,8 +150,8 @@ channel(void)
|
||||
spdk_io_device_unregister(&device2);
|
||||
spdk_io_device_unregister(&device3);
|
||||
CU_ASSERT(TAILQ_EMPTY(&g_io_devices));
|
||||
CU_ASSERT(TAILQ_EMPTY(&g_thread.io_channels));
|
||||
spdk_free_thread();
|
||||
CU_ASSERT(TAILQ_EMPTY(&g_threads));
|
||||
}
|
||||
|
||||
int
|
||||
|
Loading…
Reference in New Issue
Block a user