18d59ecb2SHans Petter Selasky /*-
28d59ecb2SHans Petter Selasky  * Copyright (c) 2010 Isilon Systems, Inc.
38d59ecb2SHans Petter Selasky  * Copyright (c) 2010 iX Systems, Inc.
48d59ecb2SHans Petter Selasky  * Copyright (c) 2010 Panasas, Inc.
5ca2ad6bdSHans Petter Selasky  * Copyright (c) 2013-2017 Mellanox Technologies, Ltd.
68d59ecb2SHans Petter Selasky  * All rights reserved.
78d59ecb2SHans Petter Selasky  *
88d59ecb2SHans Petter Selasky  * Redistribution and use in source and binary forms, with or without
98d59ecb2SHans Petter Selasky  * modification, are permitted provided that the following conditions
108d59ecb2SHans Petter Selasky  * are met:
118d59ecb2SHans Petter Selasky  * 1. Redistributions of source code must retain the above copyright
128d59ecb2SHans Petter Selasky  *    notice unmodified, this list of conditions, and the following
138d59ecb2SHans Petter Selasky  *    disclaimer.
148d59ecb2SHans Petter Selasky  * 2. Redistributions in binary form must reproduce the above copyright
158d59ecb2SHans Petter Selasky  *    notice, this list of conditions and the following disclaimer in the
168d59ecb2SHans Petter Selasky  *    documentation and/or other materials provided with the distribution.
178d59ecb2SHans Petter Selasky  *
188d59ecb2SHans Petter Selasky  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
198d59ecb2SHans Petter Selasky  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
208d59ecb2SHans Petter Selasky  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
218d59ecb2SHans Petter Selasky  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
228d59ecb2SHans Petter Selasky  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
238d59ecb2SHans Petter Selasky  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
248d59ecb2SHans Petter Selasky  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
258d59ecb2SHans Petter Selasky  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
268d59ecb2SHans Petter Selasky  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
278d59ecb2SHans Petter Selasky  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
288d59ecb2SHans Petter Selasky  *
298d59ecb2SHans Petter Selasky  * $FreeBSD$
308d59ecb2SHans Petter Selasky  */
318d59ecb2SHans Petter Selasky #ifndef	_LINUX_WORKQUEUE_H_
328d59ecb2SHans Petter Selasky #define	_LINUX_WORKQUEUE_H_
338d59ecb2SHans Petter Selasky 
348d59ecb2SHans Petter Selasky #include <linux/types.h>
358d59ecb2SHans Petter Selasky #include <linux/kernel.h>
368d59ecb2SHans Petter Selasky #include <linux/timer.h>
378d59ecb2SHans Petter Selasky #include <linux/slab.h>
388d59ecb2SHans Petter Selasky 
39c4e58b4eSHans Petter Selasky #include <asm/atomic.h>
40c4e58b4eSHans Petter Selasky 
41ca2ad6bdSHans Petter Selasky #include <sys/param.h>
42ca2ad6bdSHans Petter Selasky #include <sys/kernel.h>
438d59ecb2SHans Petter Selasky #include <sys/taskqueue.h>
44ca2ad6bdSHans Petter Selasky #include <sys/mutex.h>
45ca2ad6bdSHans Petter Selasky 
46ca2ad6bdSHans Petter Selasky #define	WORK_CPU_UNBOUND MAXCPU
47ca2ad6bdSHans Petter Selasky #define	WQ_UNBOUND (1 << 0)
48ca2ad6bdSHans Petter Selasky #define	WQ_HIGHPRI (1 << 1)
49ca2ad6bdSHans Petter Selasky 
50ca2ad6bdSHans Petter Selasky struct work_struct;
51ca2ad6bdSHans Petter Selasky typedef void (*work_func_t)(struct work_struct *);
52ca2ad6bdSHans Petter Selasky 
53ca2ad6bdSHans Petter Selasky struct work_exec {
54ca2ad6bdSHans Petter Selasky 	TAILQ_ENTRY(work_exec) entry;
55ca2ad6bdSHans Petter Selasky 	struct work_struct *target;
56ca2ad6bdSHans Petter Selasky };
578d59ecb2SHans Petter Selasky 
588d59ecb2SHans Petter Selasky struct workqueue_struct {
598d59ecb2SHans Petter Selasky 	struct taskqueue *taskqueue;
60ca2ad6bdSHans Petter Selasky 	struct mtx exec_mtx;
61ca2ad6bdSHans Petter Selasky 	TAILQ_HEAD(, work_exec) exec_head;
62c4e58b4eSHans Petter Selasky 	atomic_t draining;
638d59ecb2SHans Petter Selasky };
648d59ecb2SHans Petter Selasky 
65ca2ad6bdSHans Petter Selasky #define	WQ_EXEC_LOCK(wq) mtx_lock(&(wq)->exec_mtx)
66ca2ad6bdSHans Petter Selasky #define	WQ_EXEC_UNLOCK(wq) mtx_unlock(&(wq)->exec_mtx)
67ca2ad6bdSHans Petter Selasky 
688d59ecb2SHans Petter Selasky struct work_struct {
698d59ecb2SHans Petter Selasky 	struct task work_task;
70ca2ad6bdSHans Petter Selasky 	struct workqueue_struct *work_queue;
71ca2ad6bdSHans Petter Selasky 	work_func_t func;
72ca2ad6bdSHans Petter Selasky 	atomic_t state;
738d59ecb2SHans Petter Selasky };
748d59ecb2SHans Petter Selasky 
75ca2ad6bdSHans Petter Selasky #define	DECLARE_WORK(name, fn)						\
76f7328664SHans Petter Selasky 	struct work_struct name;					\
77f7328664SHans Petter Selasky 	static void name##_init(void *arg)				\
78f7328664SHans Petter Selasky 	{								\
79f7328664SHans Petter Selasky 		INIT_WORK(&name, fn);					\
80f7328664SHans Petter Selasky 	}								\
81f7328664SHans Petter Selasky 	SYSINIT(name, SI_SUB_LOCK, SI_ORDER_SECOND, name##_init, NULL)
8252ba0576SHans Petter Selasky 
838d59ecb2SHans Petter Selasky struct delayed_work {
848d59ecb2SHans Petter Selasky 	struct work_struct work;
85ca2ad6bdSHans Petter Selasky 	struct {
86ca2ad6bdSHans Petter Selasky 		struct callout callout;
87ca2ad6bdSHans Petter Selasky 		struct mtx mtx;
88ca2ad6bdSHans Petter Selasky 		int	expires;
89ca2ad6bdSHans Petter Selasky 	} timer;
908d59ecb2SHans Petter Selasky };
918d59ecb2SHans Petter Selasky 
92ca2ad6bdSHans Petter Selasky #define	DECLARE_DELAYED_WORK(name, fn)					\
93ca2ad6bdSHans Petter Selasky 	struct delayed_work name;					\
94ca2ad6bdSHans Petter Selasky 	static void name##_init(void *arg)				\
95ca2ad6bdSHans Petter Selasky 	{								\
96ca2ad6bdSHans Petter Selasky 		linux_init_delayed_work(&name, fn);			\
97ca2ad6bdSHans Petter Selasky 	}								\
98ca2ad6bdSHans Petter Selasky 	SYSINIT(name, SI_SUB_LOCK, SI_ORDER_SECOND, name##_init, NULL)
9955d445d3SHans Petter Selasky 
1008d59ecb2SHans Petter Selasky static inline struct delayed_work *
to_delayed_work(struct work_struct * work)1018d59ecb2SHans Petter Selasky to_delayed_work(struct work_struct *work)
1028d59ecb2SHans Petter Selasky {
103ca2ad6bdSHans Petter Selasky 	return (container_of(work, struct delayed_work, work));
1048d59ecb2SHans Petter Selasky }
1058d59ecb2SHans Petter Selasky 
106ca2ad6bdSHans Petter Selasky #define	INIT_WORK(work, fn)						\
1078d59ecb2SHans Petter Selasky do {									\
108ca2ad6bdSHans Petter Selasky 	(work)->func = (fn);						\
109ca2ad6bdSHans Petter Selasky 	(work)->work_queue = NULL;					\
110ca2ad6bdSHans Petter Selasky 	atomic_set(&(work)->state, 0);					\
11155d445d3SHans Petter Selasky 	TASK_INIT(&(work)->work_task, 0, linux_work_fn, (work));	\
1128d59ecb2SHans Petter Selasky } while (0)
1138d59ecb2SHans Petter Selasky 
114ca2ad6bdSHans Petter Selasky #define	INIT_WORK_ONSTACK(work, fn) \
115ca2ad6bdSHans Petter Selasky 	INIT_WORK(work, fn)
1168d59ecb2SHans Petter Selasky 
117ca2ad6bdSHans Petter Selasky #define	INIT_DELAYED_WORK(dwork, fn) \
118ca2ad6bdSHans Petter Selasky 	linux_init_delayed_work(dwork, fn)
119ca2ad6bdSHans Petter Selasky 
120c6d92030SHans Petter Selasky #define	INIT_DELAYED_WORK_ONSTACK(dwork, fn) \
121c6d92030SHans Petter Selasky 	linux_init_delayed_work(dwork, fn)
122c6d92030SHans Petter Selasky 
123ca2ad6bdSHans Petter Selasky #define	INIT_DEFERRABLE_WORK(dwork, fn) \
124ca2ad6bdSHans Petter Selasky 	INIT_DELAYED_WORK(dwork, fn)
125ca2ad6bdSHans Petter Selasky 
126ca2ad6bdSHans Petter Selasky #define	flush_scheduled_work() \
127ca2ad6bdSHans Petter Selasky 	taskqueue_drain_all(system_wq->taskqueue)
128ca2ad6bdSHans Petter Selasky 
129ca2ad6bdSHans Petter Selasky #define	queue_work(wq, work) \
130ca2ad6bdSHans Petter Selasky 	linux_queue_work_on(WORK_CPU_UNBOUND, wq, work)
1318d59ecb2SHans Petter Selasky 
1328d59ecb2SHans Petter Selasky #define	schedule_work(work) \
133ca2ad6bdSHans Petter Selasky 	linux_queue_work_on(WORK_CPU_UNBOUND, system_wq, work)
1348d59ecb2SHans Petter Selasky 
135ca2ad6bdSHans Petter Selasky #define	queue_delayed_work(wq, dwork, delay) \
136ca2ad6bdSHans Petter Selasky 	linux_queue_delayed_work_on(WORK_CPU_UNBOUND, wq, dwork, delay)
1378d59ecb2SHans Petter Selasky 
138ca2ad6bdSHans Petter Selasky #define	schedule_delayed_work_on(cpu, dwork, delay) \
139ca2ad6bdSHans Petter Selasky 	linux_queue_delayed_work_on(cpu, system_wq, dwork, delay)
1408d59ecb2SHans Petter Selasky 
141ca2ad6bdSHans Petter Selasky #define	queue_work_on(cpu, wq, work) \
142ca2ad6bdSHans Petter Selasky 	linux_queue_work_on(cpu, wq, work)
1438d59ecb2SHans Petter Selasky 
144ca2ad6bdSHans Petter Selasky #define	schedule_delayed_work(dwork, delay) \
145ca2ad6bdSHans Petter Selasky 	linux_queue_delayed_work_on(WORK_CPU_UNBOUND, system_wq, dwork, delay)
1468d59ecb2SHans Petter Selasky 
147ca2ad6bdSHans Petter Selasky #define	queue_delayed_work_on(cpu, wq, dwork, delay) \
148ca2ad6bdSHans Petter Selasky 	linux_queue_delayed_work_on(cpu, wq, dwork, delay)
1498d59ecb2SHans Petter Selasky 
1508d59ecb2SHans Petter Selasky #define	create_singlethread_workqueue(name) \
15155d445d3SHans Petter Selasky 	linux_create_workqueue_common(name, 1)
1528d59ecb2SHans Petter Selasky 
1538d59ecb2SHans Petter Selasky #define	create_workqueue(name) \
154ca2ad6bdSHans Petter Selasky 	linux_create_workqueue_common(name, mp_ncpus)
1558d59ecb2SHans Petter Selasky 
1568d59ecb2SHans Petter Selasky #define	alloc_ordered_workqueue(name, flags) \
15755d445d3SHans Petter Selasky 	linux_create_workqueue_common(name, 1)
1588d59ecb2SHans Petter Selasky 
1598d59ecb2SHans Petter Selasky #define	alloc_workqueue(name, flags, max_active) \
16055d445d3SHans Petter Selasky 	linux_create_workqueue_common(name, max_active)
1618d59ecb2SHans Petter Selasky 
162ca2ad6bdSHans Petter Selasky #define	flush_workqueue(wq) \
163ca2ad6bdSHans Petter Selasky 	taskqueue_drain_all((wq)->taskqueue)
1648d59ecb2SHans Petter Selasky 
165ca2ad6bdSHans Petter Selasky #define	drain_workqueue(wq) do {		\
166ca2ad6bdSHans Petter Selasky 	atomic_inc(&(wq)->draining);		\
167ca2ad6bdSHans Petter Selasky 	taskqueue_drain_all((wq)->taskqueue);	\
168ca2ad6bdSHans Petter Selasky 	atomic_dec(&(wq)->draining);		\
169ca2ad6bdSHans Petter Selasky } while (0)
1708d59ecb2SHans Petter Selasky 
171ca2ad6bdSHans Petter Selasky #define	mod_delayed_work(wq, dwork, delay) ({		\
172ca2ad6bdSHans Petter Selasky 	bool __retval;					\
173ca2ad6bdSHans Petter Selasky 	__retval = linux_cancel_delayed_work(dwork);	\
174ca2ad6bdSHans Petter Selasky 	linux_queue_delayed_work_on(WORK_CPU_UNBOUND,	\
175ca2ad6bdSHans Petter Selasky 	    wq, dwork, delay);				\
176ca2ad6bdSHans Petter Selasky 	__retval;					\
177ca2ad6bdSHans Petter Selasky })
1788d59ecb2SHans Petter Selasky 
179ca2ad6bdSHans Petter Selasky #define	delayed_work_pending(dwork) \
180ca2ad6bdSHans Petter Selasky 	linux_work_pending(&(dwork)->work)
181c4e58b4eSHans Petter Selasky 
182ca2ad6bdSHans Petter Selasky #define	cancel_delayed_work(dwork) \
183ca2ad6bdSHans Petter Selasky 	linux_cancel_delayed_work(dwork)
1848d59ecb2SHans Petter Selasky 
185ca2ad6bdSHans Petter Selasky #define	cancel_work_sync(work) \
186ca2ad6bdSHans Petter Selasky 	linux_cancel_work_sync(work)
1878d59ecb2SHans Petter Selasky 
188ca2ad6bdSHans Petter Selasky #define	cancel_delayed_work_sync(dwork) \
189ca2ad6bdSHans Petter Selasky 	linux_cancel_delayed_work_sync(dwork)
1908d59ecb2SHans Petter Selasky 
191ca2ad6bdSHans Petter Selasky #define	flush_work(work) \
192ca2ad6bdSHans Petter Selasky 	linux_flush_work(work)
1938d59ecb2SHans Petter Selasky 
194ca2ad6bdSHans Petter Selasky #define	flush_delayed_work(dwork) \
195ca2ad6bdSHans Petter Selasky 	linux_flush_delayed_work(dwork)
1968d59ecb2SHans Petter Selasky 
197ca2ad6bdSHans Petter Selasky #define	work_pending(work) \
198ca2ad6bdSHans Petter Selasky 	linux_work_pending(work)
199ca2ad6bdSHans Petter Selasky 
200ca2ad6bdSHans Petter Selasky #define	work_busy(work) \
201ca2ad6bdSHans Petter Selasky 	linux_work_busy(work)
202ca2ad6bdSHans Petter Selasky 
203ca2ad6bdSHans Petter Selasky #define	destroy_work_on_stack(work) \
204ca2ad6bdSHans Petter Selasky 	do { } while (0)
205ca2ad6bdSHans Petter Selasky 
206ca2ad6bdSHans Petter Selasky #define	destroy_delayed_work_on_stack(dwork) \
207ca2ad6bdSHans Petter Selasky 	do { } while (0)
208ca2ad6bdSHans Petter Selasky 
209ca2ad6bdSHans Petter Selasky #define	destroy_workqueue(wq) \
210ca2ad6bdSHans Petter Selasky 	linux_destroy_workqueue(wq)
211ca2ad6bdSHans Petter Selasky 
212*549dcdb3SHans Petter Selasky #define	current_work() \
213*549dcdb3SHans Petter Selasky 	linux_current_work()
214*549dcdb3SHans Petter Selasky 
215ca2ad6bdSHans Petter Selasky /* prototypes */
216ca2ad6bdSHans Petter Selasky 
217ca2ad6bdSHans Petter Selasky extern struct workqueue_struct *system_wq;
218ca2ad6bdSHans Petter Selasky extern struct workqueue_struct *system_long_wq;
219ca2ad6bdSHans Petter Selasky extern struct workqueue_struct *system_unbound_wq;
2207a13eebaSHans Petter Selasky extern struct workqueue_struct *system_highpri_wq;
221ca2ad6bdSHans Petter Selasky extern struct workqueue_struct *system_power_efficient_wq;
222ca2ad6bdSHans Petter Selasky 
223ca2ad6bdSHans Petter Selasky extern void linux_init_delayed_work(struct delayed_work *, work_func_t);
224ca2ad6bdSHans Petter Selasky extern void linux_work_fn(void *, int);
22587a567f1SHans Petter Selasky extern void linux_delayed_work_fn(void *, int);
226ca2ad6bdSHans Petter Selasky extern struct workqueue_struct *linux_create_workqueue_common(const char *, int);
227ca2ad6bdSHans Petter Selasky extern void linux_destroy_workqueue(struct workqueue_struct *);
228ca2ad6bdSHans Petter Selasky extern bool linux_queue_work_on(int cpu, struct workqueue_struct *, struct work_struct *);
229ca2ad6bdSHans Petter Selasky extern bool linux_queue_delayed_work_on(int cpu, struct workqueue_struct *,
230ca2ad6bdSHans Petter Selasky     struct delayed_work *, unsigned delay);
231ca2ad6bdSHans Petter Selasky extern bool linux_cancel_delayed_work(struct delayed_work *);
232ca2ad6bdSHans Petter Selasky extern bool linux_cancel_work_sync(struct work_struct *);
233ca2ad6bdSHans Petter Selasky extern bool linux_cancel_delayed_work_sync(struct delayed_work *);
234ca2ad6bdSHans Petter Selasky extern bool linux_flush_work(struct work_struct *);
235ca2ad6bdSHans Petter Selasky extern bool linux_flush_delayed_work(struct delayed_work *);
236ca2ad6bdSHans Petter Selasky extern bool linux_work_pending(struct work_struct *);
237ca2ad6bdSHans Petter Selasky extern bool linux_work_busy(struct work_struct *);
238*549dcdb3SHans Petter Selasky extern struct work_struct *linux_current_work(void);
2398d59ecb2SHans Petter Selasky 
2408d59ecb2SHans Petter Selasky #endif					/* _LINUX_WORKQUEUE_H_ */
241