mirror of
https://github.com/sysprog21/lkmpg.git
synced 2024-11-22 15:44:11 +08:00
e62dff0df4
The rule of thumb is to include the headers we are the direct user of. In particular, if we need an atomic API, we include <linux/atomic.h>. On the other hand we should not use headers for no reason. In particular, if we are not doing any IRQ job, why is the <linux/irq.h> included? Signed-off-by: Andy Shevchenko <andriy.shevchenko@linux.intel.com>
78 lines
1.7 KiB
C
78 lines
1.7 KiB
C
/*
|
|
* completions.c
|
|
*/
|
|
#include <linux/completion.h>
|
|
#include <linux/err.h> /* for IS_ERR() */
|
|
#include <linux/init.h>
|
|
#include <linux/kernel.h>
|
|
#include <linux/kthread.h>
|
|
#include <linux/module.h>
|
|
|
|
static struct {
|
|
struct completion crank_comp;
|
|
struct completion flywheel_comp;
|
|
} machine;
|
|
|
|
static int machine_crank_thread(void *arg)
|
|
{
|
|
pr_info("Turn the crank\n");
|
|
|
|
complete_all(&machine.crank_comp);
|
|
complete_and_exit(&machine.crank_comp, 0);
|
|
}
|
|
|
|
static int machine_flywheel_spinup_thread(void *arg)
|
|
{
|
|
wait_for_completion(&machine.crank_comp);
|
|
|
|
pr_info("Flywheel spins up\n");
|
|
|
|
complete_all(&machine.flywheel_comp);
|
|
complete_and_exit(&machine.flywheel_comp, 0);
|
|
}
|
|
|
|
static int completions_init(void)
|
|
{
|
|
struct task_struct *crank_thread;
|
|
struct task_struct *flywheel_thread;
|
|
|
|
pr_info("completions example\n");
|
|
|
|
init_completion(&machine.crank_comp);
|
|
init_completion(&machine.flywheel_comp);
|
|
|
|
crank_thread = kthread_create(machine_crank_thread, NULL, "KThread Crank");
|
|
if (IS_ERR(crank_thread))
|
|
goto ERROR_THREAD_1;
|
|
|
|
flywheel_thread = kthread_create(machine_flywheel_spinup_thread, NULL,
|
|
"KThread Flywheel");
|
|
if (IS_ERR(flywheel_thread))
|
|
goto ERROR_THREAD_2;
|
|
|
|
wake_up_process(flywheel_thread);
|
|
wake_up_process(crank_thread);
|
|
|
|
return 0;
|
|
|
|
ERROR_THREAD_2:
|
|
kthread_stop(crank_thread);
|
|
ERROR_THREAD_1:
|
|
|
|
return -1;
|
|
}
|
|
|
|
static void completions_exit(void)
|
|
{
|
|
wait_for_completion(&machine.crank_comp);
|
|
wait_for_completion(&machine.flywheel_comp);
|
|
|
|
pr_info("completions exit\n");
|
|
}
|
|
|
|
module_init(completions_init);
|
|
module_exit(completions_exit);
|
|
|
|
MODULE_DESCRIPTION("Completions example");
|
|
MODULE_LICENSE("GPL");
|