1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef _LINUX_TASK_WORK_H
3 #define _LINUX_TASK_WORK_H
4
5 #include <linux/list.h>
6 #include <linux/sched.h>
7
8 typedef void (*task_work_func_t)(struct callback_head *);
9
10 static inline void
init_task_work(struct callback_head * twork,task_work_func_t func)11 init_task_work(struct callback_head *twork, task_work_func_t func)
12 {
13 twork->func = func;
14 }
15
16 enum task_work_notify_mode {
17 TWA_NONE = 0,
18 TWA_RESUME,
19 TWA_SIGNAL,
20 TWA_SIGNAL_NO_IPI,
21 TWA_NMI_CURRENT,
22
23 TWA_FLAGS = 0xff00,
24 TWAF_NO_ALLOC = 0x0100,
25 };
26
task_work_pending(struct task_struct * task)27 static inline bool task_work_pending(struct task_struct *task)
28 {
29 return READ_ONCE(task->task_works);
30 }
31
32 int task_work_add(struct task_struct *task, struct callback_head *twork,
33 enum task_work_notify_mode mode);
34
35 struct callback_head *task_work_cancel_match(struct task_struct *task,
36 bool (*match)(struct callback_head *, void *data), void *data);
37 struct callback_head *task_work_cancel_func(struct task_struct *, task_work_func_t);
38 bool task_work_cancel(struct task_struct *task, struct callback_head *cb);
39 void task_work_run(void);
40
exit_task_work(struct task_struct * task)41 static inline void exit_task_work(struct task_struct *task)
42 {
43 task_work_run();
44 }
45
46 #endif /* _LINUX_TASK_WORK_H */
47