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