1 /* 2 * Copyright (c) 2019 François Tigeot <ftigeot@wolfpond.org> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice unmodified, this list of conditions, and the following 10 * disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27 #include <linux/kthread.h> 28 #include <linux/sched.h> 29 #include <linux/slab.h> 30 31 #include <sys/kthread.h> 32 33 /* 34 All Linux threads/processes have an associated task_struct 35 a kthread is a pure kernel thread without userland context 36 */ 37 38 static void 39 linux_ktfn_wrapper(void *arg) 40 { 41 struct task_struct *task = arg; 42 43 task->kt_exitvalue = task->kt_fn(task->kt_fndata); 44 } 45 46 struct task_struct * 47 kthread_run(int (*lfn)(void *), void *data, const char *namefmt, ...) 48 { 49 struct task_struct *task; 50 struct thread *td; 51 __va_list args; 52 int ret; 53 54 task = kzalloc(sizeof(*task), GFP_KERNEL); 55 56 __va_start(args, namefmt); 57 ret = kthread_alloc(linux_ktfn_wrapper, task, &td, namefmt, args); 58 __va_end(args); 59 if (ret) { 60 kfree(task); 61 return ERR_PTR(-ENOMEM); 62 } 63 64 task->dfly_td = td; 65 td->td_linux_task = task; 66 67 task->mm = NULL; /* kthreads have no userland address space */ 68 69 task->kt_fn = lfn; 70 task->kt_fndata = data; 71 72 /* Start the thread here */ 73 lwkt_schedule(td); 74 75 return task; 76 } 77 78 #define KTHREAD_SHOULD_STOP 1 79 80 bool 81 kthread_should_stop(void) 82 { 83 return test_bit(KTHREAD_SHOULD_STOP, ¤t->kt_flags); 84 } 85 86 int 87 kthread_stop(struct task_struct *ts) 88 { 89 set_bit(KTHREAD_SHOULD_STOP, &ts->kt_flags); 90 91 wake_up_process(ts); 92 wait_for_completion(&ts->kt_exited); 93 94 return ts->kt_exitvalue; 95 } 96