7eb30fe18f
Remove obsolete function vdo_perform_once. Instead, initialize necessary module state when loading the module. Signed-off-by: Matthew Sakai <msakai@redhat.com> Signed-off-by: Mike Snitzer <snitzer@kernel.org>
109 lines
2.9 KiB
C
109 lines
2.9 KiB
C
// SPDX-License-Identifier: GPL-2.0-only
|
|
/*
|
|
* Copyright 2023 Red Hat
|
|
*/
|
|
|
|
#include "thread-utils.h"
|
|
|
|
#include <asm/current.h>
|
|
#include <linux/delay.h>
|
|
#include <linux/kthread.h>
|
|
#include <linux/mutex.h>
|
|
#include <linux/types.h>
|
|
|
|
#include "errors.h"
|
|
#include "logger.h"
|
|
#include "memory-alloc.h"
|
|
|
|
static struct hlist_head thread_list;
|
|
static struct mutex thread_mutex;
|
|
|
|
struct thread {
|
|
void (*thread_function)(void *thread_data);
|
|
void *thread_data;
|
|
struct hlist_node thread_links;
|
|
struct task_struct *thread_task;
|
|
struct completion thread_done;
|
|
};
|
|
|
|
void vdo_initialize_threads_mutex(void)
|
|
{
|
|
mutex_init(&thread_mutex);
|
|
}
|
|
|
|
static int thread_starter(void *arg)
|
|
{
|
|
struct registered_thread allocating_thread;
|
|
struct thread *thread = arg;
|
|
|
|
thread->thread_task = current;
|
|
mutex_lock(&thread_mutex);
|
|
hlist_add_head(&thread->thread_links, &thread_list);
|
|
mutex_unlock(&thread_mutex);
|
|
vdo_register_allocating_thread(&allocating_thread, NULL);
|
|
thread->thread_function(thread->thread_data);
|
|
vdo_unregister_allocating_thread();
|
|
complete(&thread->thread_done);
|
|
return 0;
|
|
}
|
|
|
|
int vdo_create_thread(void (*thread_function)(void *), void *thread_data,
|
|
const char *name, struct thread **new_thread)
|
|
{
|
|
char *name_colon = strchr(name, ':');
|
|
char *my_name_colon = strchr(current->comm, ':');
|
|
struct task_struct *task;
|
|
struct thread *thread;
|
|
int result;
|
|
|
|
result = vdo_allocate(1, struct thread, __func__, &thread);
|
|
if (result != VDO_SUCCESS) {
|
|
vdo_log_warning("Error allocating memory for %s", name);
|
|
return result;
|
|
}
|
|
|
|
thread->thread_function = thread_function;
|
|
thread->thread_data = thread_data;
|
|
init_completion(&thread->thread_done);
|
|
/*
|
|
* Start the thread, with an appropriate thread name.
|
|
*
|
|
* If the name supplied contains a colon character, use that name. This causes uds module
|
|
* threads to have names like "uds:callbackW" and the main test runner thread to be named
|
|
* "zub:runtest".
|
|
*
|
|
* Otherwise if the current thread has a name containing a colon character, prefix the name
|
|
* supplied with the name of the current thread up to (and including) the colon character.
|
|
* Thus when the "kvdo0:dedupeQ" thread opens an index session, all the threads associated
|
|
* with that index will have names like "kvdo0:foo".
|
|
*
|
|
* Otherwise just use the name supplied. This should be a rare occurrence.
|
|
*/
|
|
if ((name_colon == NULL) && (my_name_colon != NULL)) {
|
|
task = kthread_run(thread_starter, thread, "%.*s:%s",
|
|
(int) (my_name_colon - current->comm), current->comm,
|
|
name);
|
|
} else {
|
|
task = kthread_run(thread_starter, thread, "%s", name);
|
|
}
|
|
|
|
if (IS_ERR(task)) {
|
|
vdo_free(thread);
|
|
return PTR_ERR(task);
|
|
}
|
|
|
|
*new_thread = thread;
|
|
return VDO_SUCCESS;
|
|
}
|
|
|
|
void vdo_join_threads(struct thread *thread)
|
|
{
|
|
while (wait_for_completion_interruptible(&thread->thread_done))
|
|
fsleep(1000);
|
|
|
|
mutex_lock(&thread_mutex);
|
|
hlist_del(&thread->thread_links);
|
|
mutex_unlock(&thread_mutex);
|
|
vdo_free(thread);
|
|
}
|