xref: /linux-6.15/kernel/workqueue.c (revision db7bccf4)
11da177e4SLinus Torvalds /*
21da177e4SLinus Torvalds  * linux/kernel/workqueue.c
31da177e4SLinus Torvalds  *
41da177e4SLinus Torvalds  * Generic mechanism for defining kernel helper threads for running
51da177e4SLinus Torvalds  * arbitrary tasks in process context.
61da177e4SLinus Torvalds  *
71da177e4SLinus Torvalds  * Started by Ingo Molnar, Copyright (C) 2002
81da177e4SLinus Torvalds  *
91da177e4SLinus Torvalds  * Derived from the taskqueue/keventd code by:
101da177e4SLinus Torvalds  *
111da177e4SLinus Torvalds  *   David Woodhouse <[email protected]>
12e1f8e874SFrancois Cami  *   Andrew Morton
131da177e4SLinus Torvalds  *   Kai Petzke <[email protected]>
141da177e4SLinus Torvalds  *   Theodore Ts'o <[email protected]>
1589ada679SChristoph Lameter  *
16cde53535SChristoph Lameter  * Made to use alloc_percpu by Christoph Lameter.
171da177e4SLinus Torvalds  */
181da177e4SLinus Torvalds 
191da177e4SLinus Torvalds #include <linux/module.h>
201da177e4SLinus Torvalds #include <linux/kernel.h>
211da177e4SLinus Torvalds #include <linux/sched.h>
221da177e4SLinus Torvalds #include <linux/init.h>
231da177e4SLinus Torvalds #include <linux/signal.h>
241da177e4SLinus Torvalds #include <linux/completion.h>
251da177e4SLinus Torvalds #include <linux/workqueue.h>
261da177e4SLinus Torvalds #include <linux/slab.h>
271da177e4SLinus Torvalds #include <linux/cpu.h>
281da177e4SLinus Torvalds #include <linux/notifier.h>
291da177e4SLinus Torvalds #include <linux/kthread.h>
301fa44ecaSJames Bottomley #include <linux/hardirq.h>
3146934023SChristoph Lameter #include <linux/mempolicy.h>
32341a5958SRafael J. Wysocki #include <linux/freezer.h>
33d5abe669SPeter Zijlstra #include <linux/kallsyms.h>
34d5abe669SPeter Zijlstra #include <linux/debug_locks.h>
354e6045f1SJohannes Berg #include <linux/lockdep.h>
36c34056a3STejun Heo #include <linux/idr.h>
371da177e4SLinus Torvalds 
38c8e55f36STejun Heo enum {
39*db7bccf4STejun Heo 	/* global_cwq flags */
40*db7bccf4STejun Heo 	GCWQ_FREEZING		= 1 << 3,	/* freeze in progress */
41*db7bccf4STejun Heo 
42c8e55f36STejun Heo 	/* worker flags */
43c8e55f36STejun Heo 	WORKER_STARTED		= 1 << 0,	/* started */
44c8e55f36STejun Heo 	WORKER_DIE		= 1 << 1,	/* die die die */
45c8e55f36STejun Heo 	WORKER_IDLE		= 1 << 2,	/* is idle */
46*db7bccf4STejun Heo 	WORKER_ROGUE		= 1 << 4,	/* not bound to any cpu */
47*db7bccf4STejun Heo 
48*db7bccf4STejun Heo 	/* gcwq->trustee_state */
49*db7bccf4STejun Heo 	TRUSTEE_START		= 0,		/* start */
50*db7bccf4STejun Heo 	TRUSTEE_IN_CHARGE	= 1,		/* trustee in charge of gcwq */
51*db7bccf4STejun Heo 	TRUSTEE_BUTCHER		= 2,		/* butcher workers */
52*db7bccf4STejun Heo 	TRUSTEE_RELEASE		= 3,		/* release workers */
53*db7bccf4STejun Heo 	TRUSTEE_DONE		= 4,		/* trustee is done */
54c8e55f36STejun Heo 
55c8e55f36STejun Heo 	BUSY_WORKER_HASH_ORDER	= 6,		/* 64 pointers */
56c8e55f36STejun Heo 	BUSY_WORKER_HASH_SIZE	= 1 << BUSY_WORKER_HASH_ORDER,
57c8e55f36STejun Heo 	BUSY_WORKER_HASH_MASK	= BUSY_WORKER_HASH_SIZE - 1,
58*db7bccf4STejun Heo 
59*db7bccf4STejun Heo 	TRUSTEE_COOLDOWN	= HZ / 10,	/* for trustee draining */
60c8e55f36STejun Heo };
61c8e55f36STejun Heo 
621da177e4SLinus Torvalds /*
634690c4abSTejun Heo  * Structure fields follow one of the following exclusion rules.
644690c4abSTejun Heo  *
654690c4abSTejun Heo  * I: Set during initialization and read-only afterwards.
664690c4abSTejun Heo  *
678b03ae3cSTejun Heo  * L: gcwq->lock protected.  Access with gcwq->lock held.
684690c4abSTejun Heo  *
6973f53c4aSTejun Heo  * F: wq->flush_mutex protected.
7073f53c4aSTejun Heo  *
714690c4abSTejun Heo  * W: workqueue_lock protected.
724690c4abSTejun Heo  */
734690c4abSTejun Heo 
748b03ae3cSTejun Heo struct global_cwq;
75c34056a3STejun Heo struct cpu_workqueue_struct;
76c34056a3STejun Heo 
77c34056a3STejun Heo struct worker {
78c8e55f36STejun Heo 	/* on idle list while idle, on busy hash table while busy */
79c8e55f36STejun Heo 	union {
80c8e55f36STejun Heo 		struct list_head	entry;	/* L: while idle */
81c8e55f36STejun Heo 		struct hlist_node	hentry;	/* L: while busy */
82c8e55f36STejun Heo 	};
83c8e55f36STejun Heo 
84c34056a3STejun Heo 	struct work_struct	*current_work;	/* L: work being processed */
85affee4b2STejun Heo 	struct list_head	scheduled;	/* L: scheduled works */
86c34056a3STejun Heo 	struct task_struct	*task;		/* I: worker task */
878b03ae3cSTejun Heo 	struct global_cwq	*gcwq;		/* I: the associated gcwq */
88c34056a3STejun Heo 	struct cpu_workqueue_struct *cwq;	/* I: the associated cwq */
89c8e55f36STejun Heo 	unsigned int		flags;		/* L: flags */
90c34056a3STejun Heo 	int			id;		/* I: worker id */
91c34056a3STejun Heo };
92c34056a3STejun Heo 
934690c4abSTejun Heo /*
948b03ae3cSTejun Heo  * Global per-cpu workqueue.
958b03ae3cSTejun Heo  */
968b03ae3cSTejun Heo struct global_cwq {
978b03ae3cSTejun Heo 	spinlock_t		lock;		/* the gcwq lock */
988b03ae3cSTejun Heo 	unsigned int		cpu;		/* I: the associated cpu */
99*db7bccf4STejun Heo 	unsigned int		flags;		/* L: GCWQ_* flags */
100c8e55f36STejun Heo 
101c8e55f36STejun Heo 	int			nr_workers;	/* L: total number of workers */
102c8e55f36STejun Heo 	int			nr_idle;	/* L: currently idle ones */
103c8e55f36STejun Heo 
104c8e55f36STejun Heo 	/* workers are chained either in the idle_list or busy_hash */
105c8e55f36STejun Heo 	struct list_head	idle_list;	/* L: list of idle workers */
106c8e55f36STejun Heo 	struct hlist_head	busy_hash[BUSY_WORKER_HASH_SIZE];
107c8e55f36STejun Heo 						/* L: hash of busy workers */
108c8e55f36STejun Heo 
1098b03ae3cSTejun Heo 	struct ida		worker_ida;	/* L: for worker IDs */
110*db7bccf4STejun Heo 
111*db7bccf4STejun Heo 	struct task_struct	*trustee;	/* L: for gcwq shutdown */
112*db7bccf4STejun Heo 	unsigned int		trustee_state;	/* L: trustee state */
113*db7bccf4STejun Heo 	wait_queue_head_t	trustee_wait;	/* trustee wait */
1148b03ae3cSTejun Heo } ____cacheline_aligned_in_smp;
1158b03ae3cSTejun Heo 
1168b03ae3cSTejun Heo /*
117f756d5e2SNathan Lynch  * The per-CPU workqueue (if single thread, we always use the first
1180f900049STejun Heo  * possible cpu).  The lower WORK_STRUCT_FLAG_BITS of
1190f900049STejun Heo  * work_struct->data are used for flags and thus cwqs need to be
1200f900049STejun Heo  * aligned at two's power of the number of flag bits.
1211da177e4SLinus Torvalds  */
1221da177e4SLinus Torvalds struct cpu_workqueue_struct {
1238b03ae3cSTejun Heo 	struct global_cwq	*gcwq;		/* I: the associated gcwq */
1241da177e4SLinus Torvalds 	struct list_head worklist;
125c34056a3STejun Heo 	struct worker		*worker;
1264690c4abSTejun Heo 	struct workqueue_struct *wq;		/* I: the owning workqueue */
12773f53c4aSTejun Heo 	int			work_color;	/* L: current color */
12873f53c4aSTejun Heo 	int			flush_color;	/* L: flushing color */
12973f53c4aSTejun Heo 	int			nr_in_flight[WORK_NR_COLORS];
13073f53c4aSTejun Heo 						/* L: nr of in_flight works */
1311e19ffc6STejun Heo 	int			nr_active;	/* L: nr of active works */
132a0a1a5fdSTejun Heo 	int			max_active;	/* L: max active works */
1331e19ffc6STejun Heo 	struct list_head	delayed_works;	/* L: delayed works */
1340f900049STejun Heo };
1351da177e4SLinus Torvalds 
1361da177e4SLinus Torvalds /*
13773f53c4aSTejun Heo  * Structure used to wait for workqueue flush.
13873f53c4aSTejun Heo  */
13973f53c4aSTejun Heo struct wq_flusher {
14073f53c4aSTejun Heo 	struct list_head	list;		/* F: list of flushers */
14173f53c4aSTejun Heo 	int			flush_color;	/* F: flush color waiting for */
14273f53c4aSTejun Heo 	struct completion	done;		/* flush completion */
14373f53c4aSTejun Heo };
14473f53c4aSTejun Heo 
14573f53c4aSTejun Heo /*
1461da177e4SLinus Torvalds  * The externally visible workqueue abstraction is an array of
1471da177e4SLinus Torvalds  * per-CPU workqueues:
1481da177e4SLinus Torvalds  */
1491da177e4SLinus Torvalds struct workqueue_struct {
15097e37d7bSTejun Heo 	unsigned int		flags;		/* I: WQ_* flags */
1514690c4abSTejun Heo 	struct cpu_workqueue_struct *cpu_wq;	/* I: cwq's */
1524690c4abSTejun Heo 	struct list_head	list;		/* W: list of all workqueues */
15373f53c4aSTejun Heo 
15473f53c4aSTejun Heo 	struct mutex		flush_mutex;	/* protects wq flushing */
15573f53c4aSTejun Heo 	int			work_color;	/* F: current work color */
15673f53c4aSTejun Heo 	int			flush_color;	/* F: current flush color */
15773f53c4aSTejun Heo 	atomic_t		nr_cwqs_to_flush; /* flush in progress */
15873f53c4aSTejun Heo 	struct wq_flusher	*first_flusher;	/* F: first flusher */
15973f53c4aSTejun Heo 	struct list_head	flusher_queue;	/* F: flush waiters */
16073f53c4aSTejun Heo 	struct list_head	flusher_overflow; /* F: flush overflow list */
16173f53c4aSTejun Heo 
162a0a1a5fdSTejun Heo 	int			saved_max_active; /* I: saved cwq max_active */
1634690c4abSTejun Heo 	const char		*name;		/* I: workqueue name */
1644e6045f1SJohannes Berg #ifdef CONFIG_LOCKDEP
1654e6045f1SJohannes Berg 	struct lockdep_map	lockdep_map;
1664e6045f1SJohannes Berg #endif
1671da177e4SLinus Torvalds };
1681da177e4SLinus Torvalds 
169*db7bccf4STejun Heo #define for_each_busy_worker(worker, i, pos, gcwq)			\
170*db7bccf4STejun Heo 	for (i = 0; i < BUSY_WORKER_HASH_SIZE; i++)			\
171*db7bccf4STejun Heo 		hlist_for_each_entry(worker, pos, &gcwq->busy_hash[i], hentry)
172*db7bccf4STejun Heo 
173dc186ad7SThomas Gleixner #ifdef CONFIG_DEBUG_OBJECTS_WORK
174dc186ad7SThomas Gleixner 
175dc186ad7SThomas Gleixner static struct debug_obj_descr work_debug_descr;
176dc186ad7SThomas Gleixner 
177dc186ad7SThomas Gleixner /*
178dc186ad7SThomas Gleixner  * fixup_init is called when:
179dc186ad7SThomas Gleixner  * - an active object is initialized
180dc186ad7SThomas Gleixner  */
181dc186ad7SThomas Gleixner static int work_fixup_init(void *addr, enum debug_obj_state state)
182dc186ad7SThomas Gleixner {
183dc186ad7SThomas Gleixner 	struct work_struct *work = addr;
184dc186ad7SThomas Gleixner 
185dc186ad7SThomas Gleixner 	switch (state) {
186dc186ad7SThomas Gleixner 	case ODEBUG_STATE_ACTIVE:
187dc186ad7SThomas Gleixner 		cancel_work_sync(work);
188dc186ad7SThomas Gleixner 		debug_object_init(work, &work_debug_descr);
189dc186ad7SThomas Gleixner 		return 1;
190dc186ad7SThomas Gleixner 	default:
191dc186ad7SThomas Gleixner 		return 0;
192dc186ad7SThomas Gleixner 	}
193dc186ad7SThomas Gleixner }
194dc186ad7SThomas Gleixner 
195dc186ad7SThomas Gleixner /*
196dc186ad7SThomas Gleixner  * fixup_activate is called when:
197dc186ad7SThomas Gleixner  * - an active object is activated
198dc186ad7SThomas Gleixner  * - an unknown object is activated (might be a statically initialized object)
199dc186ad7SThomas Gleixner  */
200dc186ad7SThomas Gleixner static int work_fixup_activate(void *addr, enum debug_obj_state state)
201dc186ad7SThomas Gleixner {
202dc186ad7SThomas Gleixner 	struct work_struct *work = addr;
203dc186ad7SThomas Gleixner 
204dc186ad7SThomas Gleixner 	switch (state) {
205dc186ad7SThomas Gleixner 
206dc186ad7SThomas Gleixner 	case ODEBUG_STATE_NOTAVAILABLE:
207dc186ad7SThomas Gleixner 		/*
208dc186ad7SThomas Gleixner 		 * This is not really a fixup. The work struct was
209dc186ad7SThomas Gleixner 		 * statically initialized. We just make sure that it
210dc186ad7SThomas Gleixner 		 * is tracked in the object tracker.
211dc186ad7SThomas Gleixner 		 */
21222df02bbSTejun Heo 		if (test_bit(WORK_STRUCT_STATIC_BIT, work_data_bits(work))) {
213dc186ad7SThomas Gleixner 			debug_object_init(work, &work_debug_descr);
214dc186ad7SThomas Gleixner 			debug_object_activate(work, &work_debug_descr);
215dc186ad7SThomas Gleixner 			return 0;
216dc186ad7SThomas Gleixner 		}
217dc186ad7SThomas Gleixner 		WARN_ON_ONCE(1);
218dc186ad7SThomas Gleixner 		return 0;
219dc186ad7SThomas Gleixner 
220dc186ad7SThomas Gleixner 	case ODEBUG_STATE_ACTIVE:
221dc186ad7SThomas Gleixner 		WARN_ON(1);
222dc186ad7SThomas Gleixner 
223dc186ad7SThomas Gleixner 	default:
224dc186ad7SThomas Gleixner 		return 0;
225dc186ad7SThomas Gleixner 	}
226dc186ad7SThomas Gleixner }
227dc186ad7SThomas Gleixner 
228dc186ad7SThomas Gleixner /*
229dc186ad7SThomas Gleixner  * fixup_free is called when:
230dc186ad7SThomas Gleixner  * - an active object is freed
231dc186ad7SThomas Gleixner  */
232dc186ad7SThomas Gleixner static int work_fixup_free(void *addr, enum debug_obj_state state)
233dc186ad7SThomas Gleixner {
234dc186ad7SThomas Gleixner 	struct work_struct *work = addr;
235dc186ad7SThomas Gleixner 
236dc186ad7SThomas Gleixner 	switch (state) {
237dc186ad7SThomas Gleixner 	case ODEBUG_STATE_ACTIVE:
238dc186ad7SThomas Gleixner 		cancel_work_sync(work);
239dc186ad7SThomas Gleixner 		debug_object_free(work, &work_debug_descr);
240dc186ad7SThomas Gleixner 		return 1;
241dc186ad7SThomas Gleixner 	default:
242dc186ad7SThomas Gleixner 		return 0;
243dc186ad7SThomas Gleixner 	}
244dc186ad7SThomas Gleixner }
245dc186ad7SThomas Gleixner 
246dc186ad7SThomas Gleixner static struct debug_obj_descr work_debug_descr = {
247dc186ad7SThomas Gleixner 	.name		= "work_struct",
248dc186ad7SThomas Gleixner 	.fixup_init	= work_fixup_init,
249dc186ad7SThomas Gleixner 	.fixup_activate	= work_fixup_activate,
250dc186ad7SThomas Gleixner 	.fixup_free	= work_fixup_free,
251dc186ad7SThomas Gleixner };
252dc186ad7SThomas Gleixner 
253dc186ad7SThomas Gleixner static inline void debug_work_activate(struct work_struct *work)
254dc186ad7SThomas Gleixner {
255dc186ad7SThomas Gleixner 	debug_object_activate(work, &work_debug_descr);
256dc186ad7SThomas Gleixner }
257dc186ad7SThomas Gleixner 
258dc186ad7SThomas Gleixner static inline void debug_work_deactivate(struct work_struct *work)
259dc186ad7SThomas Gleixner {
260dc186ad7SThomas Gleixner 	debug_object_deactivate(work, &work_debug_descr);
261dc186ad7SThomas Gleixner }
262dc186ad7SThomas Gleixner 
263dc186ad7SThomas Gleixner void __init_work(struct work_struct *work, int onstack)
264dc186ad7SThomas Gleixner {
265dc186ad7SThomas Gleixner 	if (onstack)
266dc186ad7SThomas Gleixner 		debug_object_init_on_stack(work, &work_debug_descr);
267dc186ad7SThomas Gleixner 	else
268dc186ad7SThomas Gleixner 		debug_object_init(work, &work_debug_descr);
269dc186ad7SThomas Gleixner }
270dc186ad7SThomas Gleixner EXPORT_SYMBOL_GPL(__init_work);
271dc186ad7SThomas Gleixner 
272dc186ad7SThomas Gleixner void destroy_work_on_stack(struct work_struct *work)
273dc186ad7SThomas Gleixner {
274dc186ad7SThomas Gleixner 	debug_object_free(work, &work_debug_descr);
275dc186ad7SThomas Gleixner }
276dc186ad7SThomas Gleixner EXPORT_SYMBOL_GPL(destroy_work_on_stack);
277dc186ad7SThomas Gleixner 
278dc186ad7SThomas Gleixner #else
279dc186ad7SThomas Gleixner static inline void debug_work_activate(struct work_struct *work) { }
280dc186ad7SThomas Gleixner static inline void debug_work_deactivate(struct work_struct *work) { }
281dc186ad7SThomas Gleixner #endif
282dc186ad7SThomas Gleixner 
28395402b38SGautham R Shenoy /* Serializes the accesses to the list of workqueues. */
28495402b38SGautham R Shenoy static DEFINE_SPINLOCK(workqueue_lock);
2851da177e4SLinus Torvalds static LIST_HEAD(workqueues);
286a0a1a5fdSTejun Heo static bool workqueue_freezing;		/* W: have wqs started freezing? */
287c34056a3STejun Heo 
2888b03ae3cSTejun Heo static DEFINE_PER_CPU(struct global_cwq, global_cwq);
2898b03ae3cSTejun Heo 
290c34056a3STejun Heo static int worker_thread(void *__worker);
2911da177e4SLinus Torvalds 
2923af24433SOleg Nesterov static int singlethread_cpu __read_mostly;
293b1f4ec17SOleg Nesterov 
2948b03ae3cSTejun Heo static struct global_cwq *get_gcwq(unsigned int cpu)
2958b03ae3cSTejun Heo {
2968b03ae3cSTejun Heo 	return &per_cpu(global_cwq, cpu);
2978b03ae3cSTejun Heo }
2988b03ae3cSTejun Heo 
2994690c4abSTejun Heo static struct cpu_workqueue_struct *get_cwq(unsigned int cpu,
3004690c4abSTejun Heo 					    struct workqueue_struct *wq)
301a848e3b6SOleg Nesterov {
302a848e3b6SOleg Nesterov 	return per_cpu_ptr(wq->cpu_wq, cpu);
303a848e3b6SOleg Nesterov }
304a848e3b6SOleg Nesterov 
3051537663fSTejun Heo static struct cpu_workqueue_struct *target_cwq(unsigned int cpu,
3061537663fSTejun Heo 					       struct workqueue_struct *wq)
3071537663fSTejun Heo {
3081537663fSTejun Heo 	if (unlikely(wq->flags & WQ_SINGLE_THREAD))
3091537663fSTejun Heo 		cpu = singlethread_cpu;
3101537663fSTejun Heo 	return get_cwq(cpu, wq);
3111537663fSTejun Heo }
3121537663fSTejun Heo 
31373f53c4aSTejun Heo static unsigned int work_color_to_flags(int color)
31473f53c4aSTejun Heo {
31573f53c4aSTejun Heo 	return color << WORK_STRUCT_COLOR_SHIFT;
31673f53c4aSTejun Heo }
31773f53c4aSTejun Heo 
31873f53c4aSTejun Heo static int get_work_color(struct work_struct *work)
31973f53c4aSTejun Heo {
32073f53c4aSTejun Heo 	return (*work_data_bits(work) >> WORK_STRUCT_COLOR_SHIFT) &
32173f53c4aSTejun Heo 		((1 << WORK_STRUCT_COLOR_BITS) - 1);
32273f53c4aSTejun Heo }
32373f53c4aSTejun Heo 
32473f53c4aSTejun Heo static int work_next_color(int color)
32573f53c4aSTejun Heo {
32673f53c4aSTejun Heo 	return (color + 1) % WORK_NR_COLORS;
32773f53c4aSTejun Heo }
32873f53c4aSTejun Heo 
3294594bf15SDavid Howells /*
3304594bf15SDavid Howells  * Set the workqueue on which a work item is to be run
3314594bf15SDavid Howells  * - Must *only* be called if the pending flag is set
3324594bf15SDavid Howells  */
333ed7c0feeSOleg Nesterov static inline void set_wq_data(struct work_struct *work,
3344690c4abSTejun Heo 			       struct cpu_workqueue_struct *cwq,
3354690c4abSTejun Heo 			       unsigned long extra_flags)
336365970a1SDavid Howells {
3374594bf15SDavid Howells 	BUG_ON(!work_pending(work));
3384594bf15SDavid Howells 
3394690c4abSTejun Heo 	atomic_long_set(&work->data, (unsigned long)cwq | work_static(work) |
34022df02bbSTejun Heo 			WORK_STRUCT_PENDING | extra_flags);
341365970a1SDavid Howells }
342365970a1SDavid Howells 
3434d707b9fSOleg Nesterov /*
3444d707b9fSOleg Nesterov  * Clear WORK_STRUCT_PENDING and the workqueue on which it was queued.
3454d707b9fSOleg Nesterov  */
3464d707b9fSOleg Nesterov static inline void clear_wq_data(struct work_struct *work)
3474d707b9fSOleg Nesterov {
3484690c4abSTejun Heo 	atomic_long_set(&work->data, work_static(work));
3494d707b9fSOleg Nesterov }
3504d707b9fSOleg Nesterov 
35164166699STejun Heo static inline struct cpu_workqueue_struct *get_wq_data(struct work_struct *work)
352365970a1SDavid Howells {
35364166699STejun Heo 	return (void *)(atomic_long_read(&work->data) &
35464166699STejun Heo 			WORK_STRUCT_WQ_DATA_MASK);
355365970a1SDavid Howells }
356365970a1SDavid Howells 
3574690c4abSTejun Heo /**
358c8e55f36STejun Heo  * busy_worker_head - return the busy hash head for a work
359c8e55f36STejun Heo  * @gcwq: gcwq of interest
360c8e55f36STejun Heo  * @work: work to be hashed
361c8e55f36STejun Heo  *
362c8e55f36STejun Heo  * Return hash head of @gcwq for @work.
363c8e55f36STejun Heo  *
364c8e55f36STejun Heo  * CONTEXT:
365c8e55f36STejun Heo  * spin_lock_irq(gcwq->lock).
366c8e55f36STejun Heo  *
367c8e55f36STejun Heo  * RETURNS:
368c8e55f36STejun Heo  * Pointer to the hash head.
369c8e55f36STejun Heo  */
370c8e55f36STejun Heo static struct hlist_head *busy_worker_head(struct global_cwq *gcwq,
371c8e55f36STejun Heo 					   struct work_struct *work)
372c8e55f36STejun Heo {
373c8e55f36STejun Heo 	const int base_shift = ilog2(sizeof(struct work_struct));
374c8e55f36STejun Heo 	unsigned long v = (unsigned long)work;
375c8e55f36STejun Heo 
376c8e55f36STejun Heo 	/* simple shift and fold hash, do we need something better? */
377c8e55f36STejun Heo 	v >>= base_shift;
378c8e55f36STejun Heo 	v += v >> BUSY_WORKER_HASH_ORDER;
379c8e55f36STejun Heo 	v &= BUSY_WORKER_HASH_MASK;
380c8e55f36STejun Heo 
381c8e55f36STejun Heo 	return &gcwq->busy_hash[v];
382c8e55f36STejun Heo }
383c8e55f36STejun Heo 
384c8e55f36STejun Heo /**
3854690c4abSTejun Heo  * insert_work - insert a work into cwq
3864690c4abSTejun Heo  * @cwq: cwq @work belongs to
3874690c4abSTejun Heo  * @work: work to insert
3884690c4abSTejun Heo  * @head: insertion point
3894690c4abSTejun Heo  * @extra_flags: extra WORK_STRUCT_* flags to set
3904690c4abSTejun Heo  *
3914690c4abSTejun Heo  * Insert @work into @cwq after @head.
3924690c4abSTejun Heo  *
3934690c4abSTejun Heo  * CONTEXT:
3948b03ae3cSTejun Heo  * spin_lock_irq(gcwq->lock).
3954690c4abSTejun Heo  */
396b89deed3SOleg Nesterov static void insert_work(struct cpu_workqueue_struct *cwq,
3974690c4abSTejun Heo 			struct work_struct *work, struct list_head *head,
3984690c4abSTejun Heo 			unsigned int extra_flags)
399b89deed3SOleg Nesterov {
4004690c4abSTejun Heo 	/* we own @work, set data and link */
4014690c4abSTejun Heo 	set_wq_data(work, cwq, extra_flags);
4024690c4abSTejun Heo 
4036e84d644SOleg Nesterov 	/*
4046e84d644SOleg Nesterov 	 * Ensure that we get the right work->data if we see the
4056e84d644SOleg Nesterov 	 * result of list_add() below, see try_to_grab_pending().
4066e84d644SOleg Nesterov 	 */
4076e84d644SOleg Nesterov 	smp_wmb();
4084690c4abSTejun Heo 
4091a4d9b0aSOleg Nesterov 	list_add_tail(&work->entry, head);
410c8e55f36STejun Heo 	wake_up_process(cwq->worker->task);
411b89deed3SOleg Nesterov }
412b89deed3SOleg Nesterov 
4134690c4abSTejun Heo static void __queue_work(unsigned int cpu, struct workqueue_struct *wq,
4141da177e4SLinus Torvalds 			 struct work_struct *work)
4151da177e4SLinus Torvalds {
4161537663fSTejun Heo 	struct cpu_workqueue_struct *cwq = target_cwq(cpu, wq);
4178b03ae3cSTejun Heo 	struct global_cwq *gcwq = cwq->gcwq;
4181e19ffc6STejun Heo 	struct list_head *worklist;
4191da177e4SLinus Torvalds 	unsigned long flags;
4201da177e4SLinus Torvalds 
421dc186ad7SThomas Gleixner 	debug_work_activate(work);
4221e19ffc6STejun Heo 
4238b03ae3cSTejun Heo 	spin_lock_irqsave(&gcwq->lock, flags);
4244690c4abSTejun Heo 	BUG_ON(!list_empty(&work->entry));
4251e19ffc6STejun Heo 
42673f53c4aSTejun Heo 	cwq->nr_in_flight[cwq->work_color]++;
4271e19ffc6STejun Heo 
4281e19ffc6STejun Heo 	if (likely(cwq->nr_active < cwq->max_active)) {
4291e19ffc6STejun Heo 		cwq->nr_active++;
4301e19ffc6STejun Heo 		worklist = &cwq->worklist;
4311e19ffc6STejun Heo 	} else
4321e19ffc6STejun Heo 		worklist = &cwq->delayed_works;
4331e19ffc6STejun Heo 
4341e19ffc6STejun Heo 	insert_work(cwq, work, worklist, work_color_to_flags(cwq->work_color));
4351e19ffc6STejun Heo 
4368b03ae3cSTejun Heo 	spin_unlock_irqrestore(&gcwq->lock, flags);
4371da177e4SLinus Torvalds }
4381da177e4SLinus Torvalds 
4390fcb78c2SRolf Eike Beer /**
4400fcb78c2SRolf Eike Beer  * queue_work - queue work on a workqueue
4410fcb78c2SRolf Eike Beer  * @wq: workqueue to use
4420fcb78c2SRolf Eike Beer  * @work: work to queue
4430fcb78c2SRolf Eike Beer  *
444057647fcSAlan Stern  * Returns 0 if @work was already on a queue, non-zero otherwise.
4451da177e4SLinus Torvalds  *
44600dfcaf7SOleg Nesterov  * We queue the work to the CPU on which it was submitted, but if the CPU dies
44700dfcaf7SOleg Nesterov  * it can be processed by another CPU.
4481da177e4SLinus Torvalds  */
4497ad5b3a5SHarvey Harrison int queue_work(struct workqueue_struct *wq, struct work_struct *work)
4501da177e4SLinus Torvalds {
451ef1ca236SOleg Nesterov 	int ret;
4521da177e4SLinus Torvalds 
453ef1ca236SOleg Nesterov 	ret = queue_work_on(get_cpu(), wq, work);
454a848e3b6SOleg Nesterov 	put_cpu();
455ef1ca236SOleg Nesterov 
4561da177e4SLinus Torvalds 	return ret;
4571da177e4SLinus Torvalds }
458ae90dd5dSDave Jones EXPORT_SYMBOL_GPL(queue_work);
4591da177e4SLinus Torvalds 
460c1a220e7SZhang Rui /**
461c1a220e7SZhang Rui  * queue_work_on - queue work on specific cpu
462c1a220e7SZhang Rui  * @cpu: CPU number to execute work on
463c1a220e7SZhang Rui  * @wq: workqueue to use
464c1a220e7SZhang Rui  * @work: work to queue
465c1a220e7SZhang Rui  *
466c1a220e7SZhang Rui  * Returns 0 if @work was already on a queue, non-zero otherwise.
467c1a220e7SZhang Rui  *
468c1a220e7SZhang Rui  * We queue the work to a specific CPU, the caller must ensure it
469c1a220e7SZhang Rui  * can't go away.
470c1a220e7SZhang Rui  */
471c1a220e7SZhang Rui int
472c1a220e7SZhang Rui queue_work_on(int cpu, struct workqueue_struct *wq, struct work_struct *work)
473c1a220e7SZhang Rui {
474c1a220e7SZhang Rui 	int ret = 0;
475c1a220e7SZhang Rui 
47622df02bbSTejun Heo 	if (!test_and_set_bit(WORK_STRUCT_PENDING_BIT, work_data_bits(work))) {
4774690c4abSTejun Heo 		__queue_work(cpu, wq, work);
478c1a220e7SZhang Rui 		ret = 1;
479c1a220e7SZhang Rui 	}
480c1a220e7SZhang Rui 	return ret;
481c1a220e7SZhang Rui }
482c1a220e7SZhang Rui EXPORT_SYMBOL_GPL(queue_work_on);
483c1a220e7SZhang Rui 
4846d141c3fSLi Zefan static void delayed_work_timer_fn(unsigned long __data)
4851da177e4SLinus Torvalds {
48652bad64dSDavid Howells 	struct delayed_work *dwork = (struct delayed_work *)__data;
487ed7c0feeSOleg Nesterov 	struct cpu_workqueue_struct *cwq = get_wq_data(&dwork->work);
4881da177e4SLinus Torvalds 
4894690c4abSTejun Heo 	__queue_work(smp_processor_id(), cwq->wq, &dwork->work);
4901da177e4SLinus Torvalds }
4911da177e4SLinus Torvalds 
4920fcb78c2SRolf Eike Beer /**
4930fcb78c2SRolf Eike Beer  * queue_delayed_work - queue work on a workqueue after delay
4940fcb78c2SRolf Eike Beer  * @wq: workqueue to use
495af9997e4SRandy Dunlap  * @dwork: delayable work to queue
4960fcb78c2SRolf Eike Beer  * @delay: number of jiffies to wait before queueing
4970fcb78c2SRolf Eike Beer  *
498057647fcSAlan Stern  * Returns 0 if @work was already on a queue, non-zero otherwise.
4990fcb78c2SRolf Eike Beer  */
5007ad5b3a5SHarvey Harrison int queue_delayed_work(struct workqueue_struct *wq,
50152bad64dSDavid Howells 			struct delayed_work *dwork, unsigned long delay)
5021da177e4SLinus Torvalds {
50352bad64dSDavid Howells 	if (delay == 0)
50463bc0362SOleg Nesterov 		return queue_work(wq, &dwork->work);
5051da177e4SLinus Torvalds 
50663bc0362SOleg Nesterov 	return queue_delayed_work_on(-1, wq, dwork, delay);
5071da177e4SLinus Torvalds }
508ae90dd5dSDave Jones EXPORT_SYMBOL_GPL(queue_delayed_work);
5091da177e4SLinus Torvalds 
5100fcb78c2SRolf Eike Beer /**
5110fcb78c2SRolf Eike Beer  * queue_delayed_work_on - queue work on specific CPU after delay
5120fcb78c2SRolf Eike Beer  * @cpu: CPU number to execute work on
5130fcb78c2SRolf Eike Beer  * @wq: workqueue to use
514af9997e4SRandy Dunlap  * @dwork: work to queue
5150fcb78c2SRolf Eike Beer  * @delay: number of jiffies to wait before queueing
5160fcb78c2SRolf Eike Beer  *
517057647fcSAlan Stern  * Returns 0 if @work was already on a queue, non-zero otherwise.
5180fcb78c2SRolf Eike Beer  */
5197a6bc1cdSVenkatesh Pallipadi int queue_delayed_work_on(int cpu, struct workqueue_struct *wq,
52052bad64dSDavid Howells 			struct delayed_work *dwork, unsigned long delay)
5217a6bc1cdSVenkatesh Pallipadi {
5227a6bc1cdSVenkatesh Pallipadi 	int ret = 0;
52352bad64dSDavid Howells 	struct timer_list *timer = &dwork->timer;
52452bad64dSDavid Howells 	struct work_struct *work = &dwork->work;
5257a6bc1cdSVenkatesh Pallipadi 
52622df02bbSTejun Heo 	if (!test_and_set_bit(WORK_STRUCT_PENDING_BIT, work_data_bits(work))) {
5277a6bc1cdSVenkatesh Pallipadi 		BUG_ON(timer_pending(timer));
5287a6bc1cdSVenkatesh Pallipadi 		BUG_ON(!list_empty(&work->entry));
5297a6bc1cdSVenkatesh Pallipadi 
5308a3e77ccSAndrew Liu 		timer_stats_timer_set_start_info(&dwork->timer);
5318a3e77ccSAndrew Liu 
532ed7c0feeSOleg Nesterov 		/* This stores cwq for the moment, for the timer_fn */
5331537663fSTejun Heo 		set_wq_data(work, target_cwq(raw_smp_processor_id(), wq), 0);
5347a6bc1cdSVenkatesh Pallipadi 		timer->expires = jiffies + delay;
53552bad64dSDavid Howells 		timer->data = (unsigned long)dwork;
5367a6bc1cdSVenkatesh Pallipadi 		timer->function = delayed_work_timer_fn;
53763bc0362SOleg Nesterov 
53863bc0362SOleg Nesterov 		if (unlikely(cpu >= 0))
5397a6bc1cdSVenkatesh Pallipadi 			add_timer_on(timer, cpu);
54063bc0362SOleg Nesterov 		else
54163bc0362SOleg Nesterov 			add_timer(timer);
5427a6bc1cdSVenkatesh Pallipadi 		ret = 1;
5437a6bc1cdSVenkatesh Pallipadi 	}
5447a6bc1cdSVenkatesh Pallipadi 	return ret;
5457a6bc1cdSVenkatesh Pallipadi }
546ae90dd5dSDave Jones EXPORT_SYMBOL_GPL(queue_delayed_work_on);
5471da177e4SLinus Torvalds 
548c8e55f36STejun Heo /**
549c8e55f36STejun Heo  * worker_enter_idle - enter idle state
550c8e55f36STejun Heo  * @worker: worker which is entering idle state
551c8e55f36STejun Heo  *
552c8e55f36STejun Heo  * @worker is entering idle state.  Update stats and idle timer if
553c8e55f36STejun Heo  * necessary.
554c8e55f36STejun Heo  *
555c8e55f36STejun Heo  * LOCKING:
556c8e55f36STejun Heo  * spin_lock_irq(gcwq->lock).
557c8e55f36STejun Heo  */
558c8e55f36STejun Heo static void worker_enter_idle(struct worker *worker)
559c8e55f36STejun Heo {
560c8e55f36STejun Heo 	struct global_cwq *gcwq = worker->gcwq;
561c8e55f36STejun Heo 
562c8e55f36STejun Heo 	BUG_ON(worker->flags & WORKER_IDLE);
563c8e55f36STejun Heo 	BUG_ON(!list_empty(&worker->entry) &&
564c8e55f36STejun Heo 	       (worker->hentry.next || worker->hentry.pprev));
565c8e55f36STejun Heo 
566c8e55f36STejun Heo 	worker->flags |= WORKER_IDLE;
567c8e55f36STejun Heo 	gcwq->nr_idle++;
568c8e55f36STejun Heo 
569c8e55f36STejun Heo 	/* idle_list is LIFO */
570c8e55f36STejun Heo 	list_add(&worker->entry, &gcwq->idle_list);
571*db7bccf4STejun Heo 
572*db7bccf4STejun Heo 	if (unlikely(worker->flags & WORKER_ROGUE))
573*db7bccf4STejun Heo 		wake_up_all(&gcwq->trustee_wait);
574c8e55f36STejun Heo }
575c8e55f36STejun Heo 
576c8e55f36STejun Heo /**
577c8e55f36STejun Heo  * worker_leave_idle - leave idle state
578c8e55f36STejun Heo  * @worker: worker which is leaving idle state
579c8e55f36STejun Heo  *
580c8e55f36STejun Heo  * @worker is leaving idle state.  Update stats.
581c8e55f36STejun Heo  *
582c8e55f36STejun Heo  * LOCKING:
583c8e55f36STejun Heo  * spin_lock_irq(gcwq->lock).
584c8e55f36STejun Heo  */
585c8e55f36STejun Heo static void worker_leave_idle(struct worker *worker)
586c8e55f36STejun Heo {
587c8e55f36STejun Heo 	struct global_cwq *gcwq = worker->gcwq;
588c8e55f36STejun Heo 
589c8e55f36STejun Heo 	BUG_ON(!(worker->flags & WORKER_IDLE));
590c8e55f36STejun Heo 	worker->flags &= ~WORKER_IDLE;
591c8e55f36STejun Heo 	gcwq->nr_idle--;
592c8e55f36STejun Heo 	list_del_init(&worker->entry);
593c8e55f36STejun Heo }
594c8e55f36STejun Heo 
595c34056a3STejun Heo static struct worker *alloc_worker(void)
596c34056a3STejun Heo {
597c34056a3STejun Heo 	struct worker *worker;
598c34056a3STejun Heo 
599c34056a3STejun Heo 	worker = kzalloc(sizeof(*worker), GFP_KERNEL);
600c8e55f36STejun Heo 	if (worker) {
601c8e55f36STejun Heo 		INIT_LIST_HEAD(&worker->entry);
602affee4b2STejun Heo 		INIT_LIST_HEAD(&worker->scheduled);
603c8e55f36STejun Heo 	}
604c34056a3STejun Heo 	return worker;
605c34056a3STejun Heo }
606c34056a3STejun Heo 
607c34056a3STejun Heo /**
608c34056a3STejun Heo  * create_worker - create a new workqueue worker
609c34056a3STejun Heo  * @cwq: cwq the new worker will belong to
610c34056a3STejun Heo  * @bind: whether to set affinity to @cpu or not
611c34056a3STejun Heo  *
612c34056a3STejun Heo  * Create a new worker which is bound to @cwq.  The returned worker
613c34056a3STejun Heo  * can be started by calling start_worker() or destroyed using
614c34056a3STejun Heo  * destroy_worker().
615c34056a3STejun Heo  *
616c34056a3STejun Heo  * CONTEXT:
617c34056a3STejun Heo  * Might sleep.  Does GFP_KERNEL allocations.
618c34056a3STejun Heo  *
619c34056a3STejun Heo  * RETURNS:
620c34056a3STejun Heo  * Pointer to the newly created worker.
621c34056a3STejun Heo  */
622c34056a3STejun Heo static struct worker *create_worker(struct cpu_workqueue_struct *cwq, bool bind)
623c34056a3STejun Heo {
6248b03ae3cSTejun Heo 	struct global_cwq *gcwq = cwq->gcwq;
625c34056a3STejun Heo 	int id = -1;
626c34056a3STejun Heo 	struct worker *worker = NULL;
627c34056a3STejun Heo 
6288b03ae3cSTejun Heo 	spin_lock_irq(&gcwq->lock);
6298b03ae3cSTejun Heo 	while (ida_get_new(&gcwq->worker_ida, &id)) {
6308b03ae3cSTejun Heo 		spin_unlock_irq(&gcwq->lock);
6318b03ae3cSTejun Heo 		if (!ida_pre_get(&gcwq->worker_ida, GFP_KERNEL))
632c34056a3STejun Heo 			goto fail;
6338b03ae3cSTejun Heo 		spin_lock_irq(&gcwq->lock);
634c34056a3STejun Heo 	}
6358b03ae3cSTejun Heo 	spin_unlock_irq(&gcwq->lock);
636c34056a3STejun Heo 
637c34056a3STejun Heo 	worker = alloc_worker();
638c34056a3STejun Heo 	if (!worker)
639c34056a3STejun Heo 		goto fail;
640c34056a3STejun Heo 
6418b03ae3cSTejun Heo 	worker->gcwq = gcwq;
642c34056a3STejun Heo 	worker->cwq = cwq;
643c34056a3STejun Heo 	worker->id = id;
644c34056a3STejun Heo 
645c34056a3STejun Heo 	worker->task = kthread_create(worker_thread, worker, "kworker/%u:%d",
6468b03ae3cSTejun Heo 				      gcwq->cpu, id);
647c34056a3STejun Heo 	if (IS_ERR(worker->task))
648c34056a3STejun Heo 		goto fail;
649c34056a3STejun Heo 
650*db7bccf4STejun Heo 	/*
651*db7bccf4STejun Heo 	 * A rogue worker will become a regular one if CPU comes
652*db7bccf4STejun Heo 	 * online later on.  Make sure every worker has
653*db7bccf4STejun Heo 	 * PF_THREAD_BOUND set.
654*db7bccf4STejun Heo 	 */
655c34056a3STejun Heo 	if (bind)
6568b03ae3cSTejun Heo 		kthread_bind(worker->task, gcwq->cpu);
657*db7bccf4STejun Heo 	else
658*db7bccf4STejun Heo 		worker->task->flags |= PF_THREAD_BOUND;
659c34056a3STejun Heo 
660c34056a3STejun Heo 	return worker;
661c34056a3STejun Heo fail:
662c34056a3STejun Heo 	if (id >= 0) {
6638b03ae3cSTejun Heo 		spin_lock_irq(&gcwq->lock);
6648b03ae3cSTejun Heo 		ida_remove(&gcwq->worker_ida, id);
6658b03ae3cSTejun Heo 		spin_unlock_irq(&gcwq->lock);
666c34056a3STejun Heo 	}
667c34056a3STejun Heo 	kfree(worker);
668c34056a3STejun Heo 	return NULL;
669c34056a3STejun Heo }
670c34056a3STejun Heo 
671c34056a3STejun Heo /**
672c34056a3STejun Heo  * start_worker - start a newly created worker
673c34056a3STejun Heo  * @worker: worker to start
674c34056a3STejun Heo  *
675c8e55f36STejun Heo  * Make the gcwq aware of @worker and start it.
676c34056a3STejun Heo  *
677c34056a3STejun Heo  * CONTEXT:
6788b03ae3cSTejun Heo  * spin_lock_irq(gcwq->lock).
679c34056a3STejun Heo  */
680c34056a3STejun Heo static void start_worker(struct worker *worker)
681c34056a3STejun Heo {
682c8e55f36STejun Heo 	worker->flags |= WORKER_STARTED;
683c8e55f36STejun Heo 	worker->gcwq->nr_workers++;
684c8e55f36STejun Heo 	worker_enter_idle(worker);
685c34056a3STejun Heo 	wake_up_process(worker->task);
686c34056a3STejun Heo }
687c34056a3STejun Heo 
688c34056a3STejun Heo /**
689c34056a3STejun Heo  * destroy_worker - destroy a workqueue worker
690c34056a3STejun Heo  * @worker: worker to be destroyed
691c34056a3STejun Heo  *
692c8e55f36STejun Heo  * Destroy @worker and adjust @gcwq stats accordingly.
693c8e55f36STejun Heo  *
694c8e55f36STejun Heo  * CONTEXT:
695c8e55f36STejun Heo  * spin_lock_irq(gcwq->lock) which is released and regrabbed.
696c34056a3STejun Heo  */
697c34056a3STejun Heo static void destroy_worker(struct worker *worker)
698c34056a3STejun Heo {
6998b03ae3cSTejun Heo 	struct global_cwq *gcwq = worker->gcwq;
700c34056a3STejun Heo 	int id = worker->id;
701c34056a3STejun Heo 
702c34056a3STejun Heo 	/* sanity check frenzy */
703c34056a3STejun Heo 	BUG_ON(worker->current_work);
704affee4b2STejun Heo 	BUG_ON(!list_empty(&worker->scheduled));
705c34056a3STejun Heo 
706c8e55f36STejun Heo 	if (worker->flags & WORKER_STARTED)
707c8e55f36STejun Heo 		gcwq->nr_workers--;
708c8e55f36STejun Heo 	if (worker->flags & WORKER_IDLE)
709c8e55f36STejun Heo 		gcwq->nr_idle--;
710c8e55f36STejun Heo 
711c8e55f36STejun Heo 	list_del_init(&worker->entry);
712c8e55f36STejun Heo 	worker->flags |= WORKER_DIE;
713c8e55f36STejun Heo 
714c8e55f36STejun Heo 	spin_unlock_irq(&gcwq->lock);
715c8e55f36STejun Heo 
716c34056a3STejun Heo 	kthread_stop(worker->task);
717c34056a3STejun Heo 	kfree(worker);
718c34056a3STejun Heo 
7198b03ae3cSTejun Heo 	spin_lock_irq(&gcwq->lock);
7208b03ae3cSTejun Heo 	ida_remove(&gcwq->worker_ida, id);
721c34056a3STejun Heo }
722c34056a3STejun Heo 
723a62428c0STejun Heo /**
724affee4b2STejun Heo  * move_linked_works - move linked works to a list
725affee4b2STejun Heo  * @work: start of series of works to be scheduled
726affee4b2STejun Heo  * @head: target list to append @work to
727affee4b2STejun Heo  * @nextp: out paramter for nested worklist walking
728affee4b2STejun Heo  *
729affee4b2STejun Heo  * Schedule linked works starting from @work to @head.  Work series to
730affee4b2STejun Heo  * be scheduled starts at @work and includes any consecutive work with
731affee4b2STejun Heo  * WORK_STRUCT_LINKED set in its predecessor.
732affee4b2STejun Heo  *
733affee4b2STejun Heo  * If @nextp is not NULL, it's updated to point to the next work of
734affee4b2STejun Heo  * the last scheduled work.  This allows move_linked_works() to be
735affee4b2STejun Heo  * nested inside outer list_for_each_entry_safe().
736affee4b2STejun Heo  *
737affee4b2STejun Heo  * CONTEXT:
7388b03ae3cSTejun Heo  * spin_lock_irq(gcwq->lock).
739affee4b2STejun Heo  */
740affee4b2STejun Heo static void move_linked_works(struct work_struct *work, struct list_head *head,
741affee4b2STejun Heo 			      struct work_struct **nextp)
742affee4b2STejun Heo {
743affee4b2STejun Heo 	struct work_struct *n;
744affee4b2STejun Heo 
745affee4b2STejun Heo 	/*
746affee4b2STejun Heo 	 * Linked worklist will always end before the end of the list,
747affee4b2STejun Heo 	 * use NULL for list head.
748affee4b2STejun Heo 	 */
749affee4b2STejun Heo 	list_for_each_entry_safe_from(work, n, NULL, entry) {
750affee4b2STejun Heo 		list_move_tail(&work->entry, head);
751affee4b2STejun Heo 		if (!(*work_data_bits(work) & WORK_STRUCT_LINKED))
752affee4b2STejun Heo 			break;
753affee4b2STejun Heo 	}
754affee4b2STejun Heo 
755affee4b2STejun Heo 	/*
756affee4b2STejun Heo 	 * If we're already inside safe list traversal and have moved
757affee4b2STejun Heo 	 * multiple works to the scheduled queue, the next position
758affee4b2STejun Heo 	 * needs to be updated.
759affee4b2STejun Heo 	 */
760affee4b2STejun Heo 	if (nextp)
761affee4b2STejun Heo 		*nextp = n;
762affee4b2STejun Heo }
763affee4b2STejun Heo 
7641e19ffc6STejun Heo static void cwq_activate_first_delayed(struct cpu_workqueue_struct *cwq)
7651e19ffc6STejun Heo {
7661e19ffc6STejun Heo 	struct work_struct *work = list_first_entry(&cwq->delayed_works,
7671e19ffc6STejun Heo 						    struct work_struct, entry);
7681e19ffc6STejun Heo 
7691e19ffc6STejun Heo 	move_linked_works(work, &cwq->worklist, NULL);
7701e19ffc6STejun Heo 	cwq->nr_active++;
7711e19ffc6STejun Heo }
7721e19ffc6STejun Heo 
773affee4b2STejun Heo /**
77473f53c4aSTejun Heo  * cwq_dec_nr_in_flight - decrement cwq's nr_in_flight
77573f53c4aSTejun Heo  * @cwq: cwq of interest
77673f53c4aSTejun Heo  * @color: color of work which left the queue
77773f53c4aSTejun Heo  *
77873f53c4aSTejun Heo  * A work either has completed or is removed from pending queue,
77973f53c4aSTejun Heo  * decrement nr_in_flight of its cwq and handle workqueue flushing.
78073f53c4aSTejun Heo  *
78173f53c4aSTejun Heo  * CONTEXT:
7828b03ae3cSTejun Heo  * spin_lock_irq(gcwq->lock).
78373f53c4aSTejun Heo  */
78473f53c4aSTejun Heo static void cwq_dec_nr_in_flight(struct cpu_workqueue_struct *cwq, int color)
78573f53c4aSTejun Heo {
78673f53c4aSTejun Heo 	/* ignore uncolored works */
78773f53c4aSTejun Heo 	if (color == WORK_NO_COLOR)
78873f53c4aSTejun Heo 		return;
78973f53c4aSTejun Heo 
79073f53c4aSTejun Heo 	cwq->nr_in_flight[color]--;
7911e19ffc6STejun Heo 	cwq->nr_active--;
7921e19ffc6STejun Heo 
7931e19ffc6STejun Heo 	/* one down, submit a delayed one */
7941e19ffc6STejun Heo 	if (!list_empty(&cwq->delayed_works) &&
7951e19ffc6STejun Heo 	    cwq->nr_active < cwq->max_active)
7961e19ffc6STejun Heo 		cwq_activate_first_delayed(cwq);
79773f53c4aSTejun Heo 
79873f53c4aSTejun Heo 	/* is flush in progress and are we at the flushing tip? */
79973f53c4aSTejun Heo 	if (likely(cwq->flush_color != color))
80073f53c4aSTejun Heo 		return;
80173f53c4aSTejun Heo 
80273f53c4aSTejun Heo 	/* are there still in-flight works? */
80373f53c4aSTejun Heo 	if (cwq->nr_in_flight[color])
80473f53c4aSTejun Heo 		return;
80573f53c4aSTejun Heo 
80673f53c4aSTejun Heo 	/* this cwq is done, clear flush_color */
80773f53c4aSTejun Heo 	cwq->flush_color = -1;
80873f53c4aSTejun Heo 
80973f53c4aSTejun Heo 	/*
81073f53c4aSTejun Heo 	 * If this was the last cwq, wake up the first flusher.  It
81173f53c4aSTejun Heo 	 * will handle the rest.
81273f53c4aSTejun Heo 	 */
81373f53c4aSTejun Heo 	if (atomic_dec_and_test(&cwq->wq->nr_cwqs_to_flush))
81473f53c4aSTejun Heo 		complete(&cwq->wq->first_flusher->done);
81573f53c4aSTejun Heo }
81673f53c4aSTejun Heo 
81773f53c4aSTejun Heo /**
818a62428c0STejun Heo  * process_one_work - process single work
819c34056a3STejun Heo  * @worker: self
820a62428c0STejun Heo  * @work: work to process
821a62428c0STejun Heo  *
822a62428c0STejun Heo  * Process @work.  This function contains all the logics necessary to
823a62428c0STejun Heo  * process a single work including synchronization against and
824a62428c0STejun Heo  * interaction with other workers on the same cpu, queueing and
825a62428c0STejun Heo  * flushing.  As long as context requirement is met, any worker can
826a62428c0STejun Heo  * call this function to process a work.
827a62428c0STejun Heo  *
828a62428c0STejun Heo  * CONTEXT:
8298b03ae3cSTejun Heo  * spin_lock_irq(gcwq->lock) which is released and regrabbed.
830a62428c0STejun Heo  */
831c34056a3STejun Heo static void process_one_work(struct worker *worker, struct work_struct *work)
8321da177e4SLinus Torvalds {
833c34056a3STejun Heo 	struct cpu_workqueue_struct *cwq = worker->cwq;
8348b03ae3cSTejun Heo 	struct global_cwq *gcwq = cwq->gcwq;
835c8e55f36STejun Heo 	struct hlist_head *bwh = busy_worker_head(gcwq, work);
8366bb49e59SDavid Howells 	work_func_t f = work->func;
83773f53c4aSTejun Heo 	int work_color;
8384e6045f1SJohannes Berg #ifdef CONFIG_LOCKDEP
8394e6045f1SJohannes Berg 	/*
840a62428c0STejun Heo 	 * It is permissible to free the struct work_struct from
841a62428c0STejun Heo 	 * inside the function that is called from it, this we need to
842a62428c0STejun Heo 	 * take into account for lockdep too.  To avoid bogus "held
843a62428c0STejun Heo 	 * lock freed" warnings as well as problems when looking into
844a62428c0STejun Heo 	 * work->lockdep_map, make a copy and use that here.
8454e6045f1SJohannes Berg 	 */
8464e6045f1SJohannes Berg 	struct lockdep_map lockdep_map = work->lockdep_map;
8474e6045f1SJohannes Berg #endif
848a62428c0STejun Heo 	/* claim and process */
849dc186ad7SThomas Gleixner 	debug_work_deactivate(work);
850c8e55f36STejun Heo 	hlist_add_head(&worker->hentry, bwh);
851c34056a3STejun Heo 	worker->current_work = work;
85273f53c4aSTejun Heo 	work_color = get_work_color(work);
853a62428c0STejun Heo 	list_del_init(&work->entry);
854a62428c0STejun Heo 
8558b03ae3cSTejun Heo 	spin_unlock_irq(&gcwq->lock);
8561da177e4SLinus Torvalds 
857365970a1SDavid Howells 	BUG_ON(get_wq_data(work) != cwq);
85823b2e599SOleg Nesterov 	work_clear_pending(work);
8593295f0efSIngo Molnar 	lock_map_acquire(&cwq->wq->lockdep_map);
8603295f0efSIngo Molnar 	lock_map_acquire(&lockdep_map);
86165f27f38SDavid Howells 	f(work);
8623295f0efSIngo Molnar 	lock_map_release(&lockdep_map);
8633295f0efSIngo Molnar 	lock_map_release(&cwq->wq->lockdep_map);
8641da177e4SLinus Torvalds 
865d5abe669SPeter Zijlstra 	if (unlikely(in_atomic() || lockdep_depth(current) > 0)) {
866d5abe669SPeter Zijlstra 		printk(KERN_ERR "BUG: workqueue leaked lock or atomic: "
867d5abe669SPeter Zijlstra 		       "%s/0x%08x/%d\n",
868a62428c0STejun Heo 		       current->comm, preempt_count(), task_pid_nr(current));
869d5abe669SPeter Zijlstra 		printk(KERN_ERR "    last function: ");
870d5abe669SPeter Zijlstra 		print_symbol("%s\n", (unsigned long)f);
871d5abe669SPeter Zijlstra 		debug_show_held_locks(current);
872d5abe669SPeter Zijlstra 		dump_stack();
873d5abe669SPeter Zijlstra 	}
874d5abe669SPeter Zijlstra 
8758b03ae3cSTejun Heo 	spin_lock_irq(&gcwq->lock);
876a62428c0STejun Heo 
877a62428c0STejun Heo 	/* we're done with it, release */
878c8e55f36STejun Heo 	hlist_del_init(&worker->hentry);
879c34056a3STejun Heo 	worker->current_work = NULL;
88073f53c4aSTejun Heo 	cwq_dec_nr_in_flight(cwq, work_color);
8811da177e4SLinus Torvalds }
882a62428c0STejun Heo 
883affee4b2STejun Heo /**
884affee4b2STejun Heo  * process_scheduled_works - process scheduled works
885affee4b2STejun Heo  * @worker: self
886affee4b2STejun Heo  *
887affee4b2STejun Heo  * Process all scheduled works.  Please note that the scheduled list
888affee4b2STejun Heo  * may change while processing a work, so this function repeatedly
889affee4b2STejun Heo  * fetches a work from the top and executes it.
890affee4b2STejun Heo  *
891affee4b2STejun Heo  * CONTEXT:
8928b03ae3cSTejun Heo  * spin_lock_irq(gcwq->lock) which may be released and regrabbed
893affee4b2STejun Heo  * multiple times.
894affee4b2STejun Heo  */
895affee4b2STejun Heo static void process_scheduled_works(struct worker *worker)
896a62428c0STejun Heo {
897affee4b2STejun Heo 	while (!list_empty(&worker->scheduled)) {
898affee4b2STejun Heo 		struct work_struct *work = list_first_entry(&worker->scheduled,
899a62428c0STejun Heo 						struct work_struct, entry);
900c34056a3STejun Heo 		process_one_work(worker, work);
901a62428c0STejun Heo 	}
9021da177e4SLinus Torvalds }
9031da177e4SLinus Torvalds 
9044690c4abSTejun Heo /**
9054690c4abSTejun Heo  * worker_thread - the worker thread function
906c34056a3STejun Heo  * @__worker: self
9074690c4abSTejun Heo  *
9084690c4abSTejun Heo  * The cwq worker thread function.
9094690c4abSTejun Heo  */
910c34056a3STejun Heo static int worker_thread(void *__worker)
9111da177e4SLinus Torvalds {
912c34056a3STejun Heo 	struct worker *worker = __worker;
9138b03ae3cSTejun Heo 	struct global_cwq *gcwq = worker->gcwq;
914c34056a3STejun Heo 	struct cpu_workqueue_struct *cwq = worker->cwq;
9151da177e4SLinus Torvalds 
916c8e55f36STejun Heo woke_up:
9178b03ae3cSTejun Heo 	spin_lock_irq(&gcwq->lock);
918affee4b2STejun Heo 
919c8e55f36STejun Heo 	/* DIE can be set only while we're idle, checking here is enough */
920c8e55f36STejun Heo 	if (worker->flags & WORKER_DIE) {
921c8e55f36STejun Heo 		spin_unlock_irq(&gcwq->lock);
922c8e55f36STejun Heo 		return 0;
923c8e55f36STejun Heo 	}
924c8e55f36STejun Heo 
925c8e55f36STejun Heo 	worker_leave_idle(worker);
926*db7bccf4STejun Heo recheck:
927c8e55f36STejun Heo 	/*
928c8e55f36STejun Heo 	 * ->scheduled list can only be filled while a worker is
929c8e55f36STejun Heo 	 * preparing to process a work or actually processing it.
930c8e55f36STejun Heo 	 * Make sure nobody diddled with it while I was sleeping.
931c8e55f36STejun Heo 	 */
932c8e55f36STejun Heo 	BUG_ON(!list_empty(&worker->scheduled));
933c8e55f36STejun Heo 
934affee4b2STejun Heo 	while (!list_empty(&cwq->worklist)) {
935affee4b2STejun Heo 		struct work_struct *work =
936affee4b2STejun Heo 			list_first_entry(&cwq->worklist,
937affee4b2STejun Heo 					 struct work_struct, entry);
938affee4b2STejun Heo 
939*db7bccf4STejun Heo 		/*
940*db7bccf4STejun Heo 		 * The following is a rather inefficient way to close
941*db7bccf4STejun Heo 		 * race window against cpu hotplug operations.  Will
942*db7bccf4STejun Heo 		 * be replaced soon.
943*db7bccf4STejun Heo 		 */
944*db7bccf4STejun Heo 		if (unlikely(!(worker->flags & WORKER_ROGUE) &&
945*db7bccf4STejun Heo 			     !cpumask_equal(&worker->task->cpus_allowed,
946*db7bccf4STejun Heo 					    get_cpu_mask(gcwq->cpu)))) {
947*db7bccf4STejun Heo 			spin_unlock_irq(&gcwq->lock);
948*db7bccf4STejun Heo 			set_cpus_allowed_ptr(worker->task,
949*db7bccf4STejun Heo 					     get_cpu_mask(gcwq->cpu));
950*db7bccf4STejun Heo 			cpu_relax();
951*db7bccf4STejun Heo 			spin_lock_irq(&gcwq->lock);
952*db7bccf4STejun Heo 			goto recheck;
953*db7bccf4STejun Heo 		}
954*db7bccf4STejun Heo 
955c8e55f36STejun Heo 		if (likely(!(*work_data_bits(work) & WORK_STRUCT_LINKED))) {
956affee4b2STejun Heo 			/* optimization path, not strictly necessary */
957affee4b2STejun Heo 			process_one_work(worker, work);
958affee4b2STejun Heo 			if (unlikely(!list_empty(&worker->scheduled)))
959affee4b2STejun Heo 				process_scheduled_works(worker);
960affee4b2STejun Heo 		} else {
961c8e55f36STejun Heo 			move_linked_works(work, &worker->scheduled, NULL);
962affee4b2STejun Heo 			process_scheduled_works(worker);
963affee4b2STejun Heo 		}
964affee4b2STejun Heo 	}
965affee4b2STejun Heo 
966c8e55f36STejun Heo 	/*
967c8e55f36STejun Heo 	 * gcwq->lock is held and there's no work to process, sleep.
968c8e55f36STejun Heo 	 * Workers are woken up only while holding gcwq->lock, so
969c8e55f36STejun Heo 	 * setting the current state before releasing gcwq->lock is
970c8e55f36STejun Heo 	 * enough to prevent losing any event.
971c8e55f36STejun Heo 	 */
972c8e55f36STejun Heo 	worker_enter_idle(worker);
973c8e55f36STejun Heo 	__set_current_state(TASK_INTERRUPTIBLE);
9748b03ae3cSTejun Heo 	spin_unlock_irq(&gcwq->lock);
975c8e55f36STejun Heo 	schedule();
976c8e55f36STejun Heo 	goto woke_up;
9771da177e4SLinus Torvalds }
9781da177e4SLinus Torvalds 
979fc2e4d70SOleg Nesterov struct wq_barrier {
980fc2e4d70SOleg Nesterov 	struct work_struct	work;
981fc2e4d70SOleg Nesterov 	struct completion	done;
982fc2e4d70SOleg Nesterov };
983fc2e4d70SOleg Nesterov 
984fc2e4d70SOleg Nesterov static void wq_barrier_func(struct work_struct *work)
985fc2e4d70SOleg Nesterov {
986fc2e4d70SOleg Nesterov 	struct wq_barrier *barr = container_of(work, struct wq_barrier, work);
987fc2e4d70SOleg Nesterov 	complete(&barr->done);
988fc2e4d70SOleg Nesterov }
989fc2e4d70SOleg Nesterov 
9904690c4abSTejun Heo /**
9914690c4abSTejun Heo  * insert_wq_barrier - insert a barrier work
9924690c4abSTejun Heo  * @cwq: cwq to insert barrier into
9934690c4abSTejun Heo  * @barr: wq_barrier to insert
994affee4b2STejun Heo  * @target: target work to attach @barr to
995affee4b2STejun Heo  * @worker: worker currently executing @target, NULL if @target is not executing
9964690c4abSTejun Heo  *
997affee4b2STejun Heo  * @barr is linked to @target such that @barr is completed only after
998affee4b2STejun Heo  * @target finishes execution.  Please note that the ordering
999affee4b2STejun Heo  * guarantee is observed only with respect to @target and on the local
1000affee4b2STejun Heo  * cpu.
1001affee4b2STejun Heo  *
1002affee4b2STejun Heo  * Currently, a queued barrier can't be canceled.  This is because
1003affee4b2STejun Heo  * try_to_grab_pending() can't determine whether the work to be
1004affee4b2STejun Heo  * grabbed is at the head of the queue and thus can't clear LINKED
1005affee4b2STejun Heo  * flag of the previous work while there must be a valid next work
1006affee4b2STejun Heo  * after a work with LINKED flag set.
1007affee4b2STejun Heo  *
1008affee4b2STejun Heo  * Note that when @worker is non-NULL, @target may be modified
1009affee4b2STejun Heo  * underneath us, so we can't reliably determine cwq from @target.
10104690c4abSTejun Heo  *
10114690c4abSTejun Heo  * CONTEXT:
10128b03ae3cSTejun Heo  * spin_lock_irq(gcwq->lock).
10134690c4abSTejun Heo  */
101483c22520SOleg Nesterov static void insert_wq_barrier(struct cpu_workqueue_struct *cwq,
1015affee4b2STejun Heo 			      struct wq_barrier *barr,
1016affee4b2STejun Heo 			      struct work_struct *target, struct worker *worker)
1017fc2e4d70SOleg Nesterov {
1018affee4b2STejun Heo 	struct list_head *head;
1019affee4b2STejun Heo 	unsigned int linked = 0;
1020affee4b2STejun Heo 
1021dc186ad7SThomas Gleixner 	/*
10228b03ae3cSTejun Heo 	 * debugobject calls are safe here even with gcwq->lock locked
1023dc186ad7SThomas Gleixner 	 * as we know for sure that this will not trigger any of the
1024dc186ad7SThomas Gleixner 	 * checks and call back into the fixup functions where we
1025dc186ad7SThomas Gleixner 	 * might deadlock.
1026dc186ad7SThomas Gleixner 	 */
1027dc186ad7SThomas Gleixner 	INIT_WORK_ON_STACK(&barr->work, wq_barrier_func);
102822df02bbSTejun Heo 	__set_bit(WORK_STRUCT_PENDING_BIT, work_data_bits(&barr->work));
1029fc2e4d70SOleg Nesterov 	init_completion(&barr->done);
103083c22520SOleg Nesterov 
1031affee4b2STejun Heo 	/*
1032affee4b2STejun Heo 	 * If @target is currently being executed, schedule the
1033affee4b2STejun Heo 	 * barrier to the worker; otherwise, put it after @target.
1034affee4b2STejun Heo 	 */
1035affee4b2STejun Heo 	if (worker)
1036affee4b2STejun Heo 		head = worker->scheduled.next;
1037affee4b2STejun Heo 	else {
1038affee4b2STejun Heo 		unsigned long *bits = work_data_bits(target);
1039affee4b2STejun Heo 
1040affee4b2STejun Heo 		head = target->entry.next;
1041affee4b2STejun Heo 		/* there can already be other linked works, inherit and set */
1042affee4b2STejun Heo 		linked = *bits & WORK_STRUCT_LINKED;
1043affee4b2STejun Heo 		__set_bit(WORK_STRUCT_LINKED_BIT, bits);
1044affee4b2STejun Heo 	}
1045affee4b2STejun Heo 
1046dc186ad7SThomas Gleixner 	debug_work_activate(&barr->work);
1047affee4b2STejun Heo 	insert_work(cwq, &barr->work, head,
1048affee4b2STejun Heo 		    work_color_to_flags(WORK_NO_COLOR) | linked);
1049fc2e4d70SOleg Nesterov }
1050fc2e4d70SOleg Nesterov 
105173f53c4aSTejun Heo /**
105273f53c4aSTejun Heo  * flush_workqueue_prep_cwqs - prepare cwqs for workqueue flushing
105373f53c4aSTejun Heo  * @wq: workqueue being flushed
105473f53c4aSTejun Heo  * @flush_color: new flush color, < 0 for no-op
105573f53c4aSTejun Heo  * @work_color: new work color, < 0 for no-op
105673f53c4aSTejun Heo  *
105773f53c4aSTejun Heo  * Prepare cwqs for workqueue flushing.
105873f53c4aSTejun Heo  *
105973f53c4aSTejun Heo  * If @flush_color is non-negative, flush_color on all cwqs should be
106073f53c4aSTejun Heo  * -1.  If no cwq has in-flight commands at the specified color, all
106173f53c4aSTejun Heo  * cwq->flush_color's stay at -1 and %false is returned.  If any cwq
106273f53c4aSTejun Heo  * has in flight commands, its cwq->flush_color is set to
106373f53c4aSTejun Heo  * @flush_color, @wq->nr_cwqs_to_flush is updated accordingly, cwq
106473f53c4aSTejun Heo  * wakeup logic is armed and %true is returned.
106573f53c4aSTejun Heo  *
106673f53c4aSTejun Heo  * The caller should have initialized @wq->first_flusher prior to
106773f53c4aSTejun Heo  * calling this function with non-negative @flush_color.  If
106873f53c4aSTejun Heo  * @flush_color is negative, no flush color update is done and %false
106973f53c4aSTejun Heo  * is returned.
107073f53c4aSTejun Heo  *
107173f53c4aSTejun Heo  * If @work_color is non-negative, all cwqs should have the same
107273f53c4aSTejun Heo  * work_color which is previous to @work_color and all will be
107373f53c4aSTejun Heo  * advanced to @work_color.
107473f53c4aSTejun Heo  *
107573f53c4aSTejun Heo  * CONTEXT:
107673f53c4aSTejun Heo  * mutex_lock(wq->flush_mutex).
107773f53c4aSTejun Heo  *
107873f53c4aSTejun Heo  * RETURNS:
107973f53c4aSTejun Heo  * %true if @flush_color >= 0 and there's something to flush.  %false
108073f53c4aSTejun Heo  * otherwise.
108173f53c4aSTejun Heo  */
108273f53c4aSTejun Heo static bool flush_workqueue_prep_cwqs(struct workqueue_struct *wq,
108373f53c4aSTejun Heo 				      int flush_color, int work_color)
10841da177e4SLinus Torvalds {
108573f53c4aSTejun Heo 	bool wait = false;
108673f53c4aSTejun Heo 	unsigned int cpu;
10871da177e4SLinus Torvalds 
108873f53c4aSTejun Heo 	if (flush_color >= 0) {
108973f53c4aSTejun Heo 		BUG_ON(atomic_read(&wq->nr_cwqs_to_flush));
109073f53c4aSTejun Heo 		atomic_set(&wq->nr_cwqs_to_flush, 1);
109173f53c4aSTejun Heo 	}
109273f53c4aSTejun Heo 
109373f53c4aSTejun Heo 	for_each_possible_cpu(cpu) {
109473f53c4aSTejun Heo 		struct cpu_workqueue_struct *cwq = get_cwq(cpu, wq);
10958b03ae3cSTejun Heo 		struct global_cwq *gcwq = cwq->gcwq;
10962355b70fSLai Jiangshan 
10978b03ae3cSTejun Heo 		spin_lock_irq(&gcwq->lock);
109873f53c4aSTejun Heo 
109973f53c4aSTejun Heo 		if (flush_color >= 0) {
110073f53c4aSTejun Heo 			BUG_ON(cwq->flush_color != -1);
110173f53c4aSTejun Heo 
110273f53c4aSTejun Heo 			if (cwq->nr_in_flight[flush_color]) {
110373f53c4aSTejun Heo 				cwq->flush_color = flush_color;
110473f53c4aSTejun Heo 				atomic_inc(&wq->nr_cwqs_to_flush);
110573f53c4aSTejun Heo 				wait = true;
110683c22520SOleg Nesterov 			}
110773f53c4aSTejun Heo 		}
110873f53c4aSTejun Heo 
110973f53c4aSTejun Heo 		if (work_color >= 0) {
111073f53c4aSTejun Heo 			BUG_ON(work_color != work_next_color(cwq->work_color));
111173f53c4aSTejun Heo 			cwq->work_color = work_color;
111273f53c4aSTejun Heo 		}
111373f53c4aSTejun Heo 
11148b03ae3cSTejun Heo 		spin_unlock_irq(&gcwq->lock);
1115dc186ad7SThomas Gleixner 	}
111614441960SOleg Nesterov 
111773f53c4aSTejun Heo 	if (flush_color >= 0 && atomic_dec_and_test(&wq->nr_cwqs_to_flush))
111873f53c4aSTejun Heo 		complete(&wq->first_flusher->done);
111973f53c4aSTejun Heo 
112073f53c4aSTejun Heo 	return wait;
112183c22520SOleg Nesterov }
11221da177e4SLinus Torvalds 
11230fcb78c2SRolf Eike Beer /**
11241da177e4SLinus Torvalds  * flush_workqueue - ensure that any scheduled work has run to completion.
11250fcb78c2SRolf Eike Beer  * @wq: workqueue to flush
11261da177e4SLinus Torvalds  *
11271da177e4SLinus Torvalds  * Forces execution of the workqueue and blocks until its completion.
11281da177e4SLinus Torvalds  * This is typically used in driver shutdown handlers.
11291da177e4SLinus Torvalds  *
1130fc2e4d70SOleg Nesterov  * We sleep until all works which were queued on entry have been handled,
1131fc2e4d70SOleg Nesterov  * but we are not livelocked by new incoming ones.
11321da177e4SLinus Torvalds  */
11337ad5b3a5SHarvey Harrison void flush_workqueue(struct workqueue_struct *wq)
11341da177e4SLinus Torvalds {
113573f53c4aSTejun Heo 	struct wq_flusher this_flusher = {
113673f53c4aSTejun Heo 		.list = LIST_HEAD_INIT(this_flusher.list),
113773f53c4aSTejun Heo 		.flush_color = -1,
113873f53c4aSTejun Heo 		.done = COMPLETION_INITIALIZER_ONSTACK(this_flusher.done),
113973f53c4aSTejun Heo 	};
114073f53c4aSTejun Heo 	int next_color;
1141b1f4ec17SOleg Nesterov 
11423295f0efSIngo Molnar 	lock_map_acquire(&wq->lockdep_map);
11433295f0efSIngo Molnar 	lock_map_release(&wq->lockdep_map);
114473f53c4aSTejun Heo 
114573f53c4aSTejun Heo 	mutex_lock(&wq->flush_mutex);
114673f53c4aSTejun Heo 
114773f53c4aSTejun Heo 	/*
114873f53c4aSTejun Heo 	 * Start-to-wait phase
114973f53c4aSTejun Heo 	 */
115073f53c4aSTejun Heo 	next_color = work_next_color(wq->work_color);
115173f53c4aSTejun Heo 
115273f53c4aSTejun Heo 	if (next_color != wq->flush_color) {
115373f53c4aSTejun Heo 		/*
115473f53c4aSTejun Heo 		 * Color space is not full.  The current work_color
115573f53c4aSTejun Heo 		 * becomes our flush_color and work_color is advanced
115673f53c4aSTejun Heo 		 * by one.
115773f53c4aSTejun Heo 		 */
115873f53c4aSTejun Heo 		BUG_ON(!list_empty(&wq->flusher_overflow));
115973f53c4aSTejun Heo 		this_flusher.flush_color = wq->work_color;
116073f53c4aSTejun Heo 		wq->work_color = next_color;
116173f53c4aSTejun Heo 
116273f53c4aSTejun Heo 		if (!wq->first_flusher) {
116373f53c4aSTejun Heo 			/* no flush in progress, become the first flusher */
116473f53c4aSTejun Heo 			BUG_ON(wq->flush_color != this_flusher.flush_color);
116573f53c4aSTejun Heo 
116673f53c4aSTejun Heo 			wq->first_flusher = &this_flusher;
116773f53c4aSTejun Heo 
116873f53c4aSTejun Heo 			if (!flush_workqueue_prep_cwqs(wq, wq->flush_color,
116973f53c4aSTejun Heo 						       wq->work_color)) {
117073f53c4aSTejun Heo 				/* nothing to flush, done */
117173f53c4aSTejun Heo 				wq->flush_color = next_color;
117273f53c4aSTejun Heo 				wq->first_flusher = NULL;
117373f53c4aSTejun Heo 				goto out_unlock;
117473f53c4aSTejun Heo 			}
117573f53c4aSTejun Heo 		} else {
117673f53c4aSTejun Heo 			/* wait in queue */
117773f53c4aSTejun Heo 			BUG_ON(wq->flush_color == this_flusher.flush_color);
117873f53c4aSTejun Heo 			list_add_tail(&this_flusher.list, &wq->flusher_queue);
117973f53c4aSTejun Heo 			flush_workqueue_prep_cwqs(wq, -1, wq->work_color);
118073f53c4aSTejun Heo 		}
118173f53c4aSTejun Heo 	} else {
118273f53c4aSTejun Heo 		/*
118373f53c4aSTejun Heo 		 * Oops, color space is full, wait on overflow queue.
118473f53c4aSTejun Heo 		 * The next flush completion will assign us
118573f53c4aSTejun Heo 		 * flush_color and transfer to flusher_queue.
118673f53c4aSTejun Heo 		 */
118773f53c4aSTejun Heo 		list_add_tail(&this_flusher.list, &wq->flusher_overflow);
118873f53c4aSTejun Heo 	}
118973f53c4aSTejun Heo 
119073f53c4aSTejun Heo 	mutex_unlock(&wq->flush_mutex);
119173f53c4aSTejun Heo 
119273f53c4aSTejun Heo 	wait_for_completion(&this_flusher.done);
119373f53c4aSTejun Heo 
119473f53c4aSTejun Heo 	/*
119573f53c4aSTejun Heo 	 * Wake-up-and-cascade phase
119673f53c4aSTejun Heo 	 *
119773f53c4aSTejun Heo 	 * First flushers are responsible for cascading flushes and
119873f53c4aSTejun Heo 	 * handling overflow.  Non-first flushers can simply return.
119973f53c4aSTejun Heo 	 */
120073f53c4aSTejun Heo 	if (wq->first_flusher != &this_flusher)
120173f53c4aSTejun Heo 		return;
120273f53c4aSTejun Heo 
120373f53c4aSTejun Heo 	mutex_lock(&wq->flush_mutex);
120473f53c4aSTejun Heo 
120573f53c4aSTejun Heo 	wq->first_flusher = NULL;
120673f53c4aSTejun Heo 
120773f53c4aSTejun Heo 	BUG_ON(!list_empty(&this_flusher.list));
120873f53c4aSTejun Heo 	BUG_ON(wq->flush_color != this_flusher.flush_color);
120973f53c4aSTejun Heo 
121073f53c4aSTejun Heo 	while (true) {
121173f53c4aSTejun Heo 		struct wq_flusher *next, *tmp;
121273f53c4aSTejun Heo 
121373f53c4aSTejun Heo 		/* complete all the flushers sharing the current flush color */
121473f53c4aSTejun Heo 		list_for_each_entry_safe(next, tmp, &wq->flusher_queue, list) {
121573f53c4aSTejun Heo 			if (next->flush_color != wq->flush_color)
121673f53c4aSTejun Heo 				break;
121773f53c4aSTejun Heo 			list_del_init(&next->list);
121873f53c4aSTejun Heo 			complete(&next->done);
121973f53c4aSTejun Heo 		}
122073f53c4aSTejun Heo 
122173f53c4aSTejun Heo 		BUG_ON(!list_empty(&wq->flusher_overflow) &&
122273f53c4aSTejun Heo 		       wq->flush_color != work_next_color(wq->work_color));
122373f53c4aSTejun Heo 
122473f53c4aSTejun Heo 		/* this flush_color is finished, advance by one */
122573f53c4aSTejun Heo 		wq->flush_color = work_next_color(wq->flush_color);
122673f53c4aSTejun Heo 
122773f53c4aSTejun Heo 		/* one color has been freed, handle overflow queue */
122873f53c4aSTejun Heo 		if (!list_empty(&wq->flusher_overflow)) {
122973f53c4aSTejun Heo 			/*
123073f53c4aSTejun Heo 			 * Assign the same color to all overflowed
123173f53c4aSTejun Heo 			 * flushers, advance work_color and append to
123273f53c4aSTejun Heo 			 * flusher_queue.  This is the start-to-wait
123373f53c4aSTejun Heo 			 * phase for these overflowed flushers.
123473f53c4aSTejun Heo 			 */
123573f53c4aSTejun Heo 			list_for_each_entry(tmp, &wq->flusher_overflow, list)
123673f53c4aSTejun Heo 				tmp->flush_color = wq->work_color;
123773f53c4aSTejun Heo 
123873f53c4aSTejun Heo 			wq->work_color = work_next_color(wq->work_color);
123973f53c4aSTejun Heo 
124073f53c4aSTejun Heo 			list_splice_tail_init(&wq->flusher_overflow,
124173f53c4aSTejun Heo 					      &wq->flusher_queue);
124273f53c4aSTejun Heo 			flush_workqueue_prep_cwqs(wq, -1, wq->work_color);
124373f53c4aSTejun Heo 		}
124473f53c4aSTejun Heo 
124573f53c4aSTejun Heo 		if (list_empty(&wq->flusher_queue)) {
124673f53c4aSTejun Heo 			BUG_ON(wq->flush_color != wq->work_color);
124773f53c4aSTejun Heo 			break;
124873f53c4aSTejun Heo 		}
124973f53c4aSTejun Heo 
125073f53c4aSTejun Heo 		/*
125173f53c4aSTejun Heo 		 * Need to flush more colors.  Make the next flusher
125273f53c4aSTejun Heo 		 * the new first flusher and arm cwqs.
125373f53c4aSTejun Heo 		 */
125473f53c4aSTejun Heo 		BUG_ON(wq->flush_color == wq->work_color);
125573f53c4aSTejun Heo 		BUG_ON(wq->flush_color != next->flush_color);
125673f53c4aSTejun Heo 
125773f53c4aSTejun Heo 		list_del_init(&next->list);
125873f53c4aSTejun Heo 		wq->first_flusher = next;
125973f53c4aSTejun Heo 
126073f53c4aSTejun Heo 		if (flush_workqueue_prep_cwqs(wq, wq->flush_color, -1))
126173f53c4aSTejun Heo 			break;
126273f53c4aSTejun Heo 
126373f53c4aSTejun Heo 		/*
126473f53c4aSTejun Heo 		 * Meh... this color is already done, clear first
126573f53c4aSTejun Heo 		 * flusher and repeat cascading.
126673f53c4aSTejun Heo 		 */
126773f53c4aSTejun Heo 		wq->first_flusher = NULL;
126873f53c4aSTejun Heo 	}
126973f53c4aSTejun Heo 
127073f53c4aSTejun Heo out_unlock:
127173f53c4aSTejun Heo 	mutex_unlock(&wq->flush_mutex);
12721da177e4SLinus Torvalds }
1273ae90dd5dSDave Jones EXPORT_SYMBOL_GPL(flush_workqueue);
12741da177e4SLinus Torvalds 
1275db700897SOleg Nesterov /**
1276db700897SOleg Nesterov  * flush_work - block until a work_struct's callback has terminated
1277db700897SOleg Nesterov  * @work: the work which is to be flushed
1278db700897SOleg Nesterov  *
1279a67da70dSOleg Nesterov  * Returns false if @work has already terminated.
1280a67da70dSOleg Nesterov  *
1281db700897SOleg Nesterov  * It is expected that, prior to calling flush_work(), the caller has
1282db700897SOleg Nesterov  * arranged for the work to not be requeued, otherwise it doesn't make
1283db700897SOleg Nesterov  * sense to use this function.
1284db700897SOleg Nesterov  */
1285db700897SOleg Nesterov int flush_work(struct work_struct *work)
1286db700897SOleg Nesterov {
1287affee4b2STejun Heo 	struct worker *worker = NULL;
1288db700897SOleg Nesterov 	struct cpu_workqueue_struct *cwq;
12898b03ae3cSTejun Heo 	struct global_cwq *gcwq;
1290db700897SOleg Nesterov 	struct wq_barrier barr;
1291db700897SOleg Nesterov 
1292db700897SOleg Nesterov 	might_sleep();
1293db700897SOleg Nesterov 	cwq = get_wq_data(work);
1294db700897SOleg Nesterov 	if (!cwq)
1295db700897SOleg Nesterov 		return 0;
12968b03ae3cSTejun Heo 	gcwq = cwq->gcwq;
1297db700897SOleg Nesterov 
12983295f0efSIngo Molnar 	lock_map_acquire(&cwq->wq->lockdep_map);
12993295f0efSIngo Molnar 	lock_map_release(&cwq->wq->lockdep_map);
1300a67da70dSOleg Nesterov 
13018b03ae3cSTejun Heo 	spin_lock_irq(&gcwq->lock);
1302db700897SOleg Nesterov 	if (!list_empty(&work->entry)) {
1303db700897SOleg Nesterov 		/*
1304db700897SOleg Nesterov 		 * See the comment near try_to_grab_pending()->smp_rmb().
1305db700897SOleg Nesterov 		 * If it was re-queued under us we are not going to wait.
1306db700897SOleg Nesterov 		 */
1307db700897SOleg Nesterov 		smp_rmb();
1308db700897SOleg Nesterov 		if (unlikely(cwq != get_wq_data(work)))
13094690c4abSTejun Heo 			goto already_gone;
1310db700897SOleg Nesterov 	} else {
1311affee4b2STejun Heo 		if (cwq->worker && cwq->worker->current_work == work)
1312affee4b2STejun Heo 			worker = cwq->worker;
1313affee4b2STejun Heo 		if (!worker)
13144690c4abSTejun Heo 			goto already_gone;
1315db700897SOleg Nesterov 	}
1316db700897SOleg Nesterov 
1317affee4b2STejun Heo 	insert_wq_barrier(cwq, &barr, work, worker);
13188b03ae3cSTejun Heo 	spin_unlock_irq(&gcwq->lock);
1319db700897SOleg Nesterov 	wait_for_completion(&barr.done);
1320dc186ad7SThomas Gleixner 	destroy_work_on_stack(&barr.work);
1321db700897SOleg Nesterov 	return 1;
13224690c4abSTejun Heo already_gone:
13238b03ae3cSTejun Heo 	spin_unlock_irq(&gcwq->lock);
13244690c4abSTejun Heo 	return 0;
1325db700897SOleg Nesterov }
1326db700897SOleg Nesterov EXPORT_SYMBOL_GPL(flush_work);
1327db700897SOleg Nesterov 
13286e84d644SOleg Nesterov /*
13291f1f642eSOleg Nesterov  * Upon a successful return (>= 0), the caller "owns" WORK_STRUCT_PENDING bit,
13306e84d644SOleg Nesterov  * so this work can't be re-armed in any way.
13316e84d644SOleg Nesterov  */
13326e84d644SOleg Nesterov static int try_to_grab_pending(struct work_struct *work)
13336e84d644SOleg Nesterov {
13348b03ae3cSTejun Heo 	struct global_cwq *gcwq;
13356e84d644SOleg Nesterov 	struct cpu_workqueue_struct *cwq;
13361f1f642eSOleg Nesterov 	int ret = -1;
13376e84d644SOleg Nesterov 
133822df02bbSTejun Heo 	if (!test_and_set_bit(WORK_STRUCT_PENDING_BIT, work_data_bits(work)))
13391f1f642eSOleg Nesterov 		return 0;
13406e84d644SOleg Nesterov 
13416e84d644SOleg Nesterov 	/*
13426e84d644SOleg Nesterov 	 * The queueing is in progress, or it is already queued. Try to
13436e84d644SOleg Nesterov 	 * steal it from ->worklist without clearing WORK_STRUCT_PENDING.
13446e84d644SOleg Nesterov 	 */
13456e84d644SOleg Nesterov 
13466e84d644SOleg Nesterov 	cwq = get_wq_data(work);
13476e84d644SOleg Nesterov 	if (!cwq)
13486e84d644SOleg Nesterov 		return ret;
13498b03ae3cSTejun Heo 	gcwq = cwq->gcwq;
13506e84d644SOleg Nesterov 
13518b03ae3cSTejun Heo 	spin_lock_irq(&gcwq->lock);
13526e84d644SOleg Nesterov 	if (!list_empty(&work->entry)) {
13536e84d644SOleg Nesterov 		/*
13546e84d644SOleg Nesterov 		 * This work is queued, but perhaps we locked the wrong cwq.
13556e84d644SOleg Nesterov 		 * In that case we must see the new value after rmb(), see
13566e84d644SOleg Nesterov 		 * insert_work()->wmb().
13576e84d644SOleg Nesterov 		 */
13586e84d644SOleg Nesterov 		smp_rmb();
13596e84d644SOleg Nesterov 		if (cwq == get_wq_data(work)) {
1360dc186ad7SThomas Gleixner 			debug_work_deactivate(work);
13616e84d644SOleg Nesterov 			list_del_init(&work->entry);
136273f53c4aSTejun Heo 			cwq_dec_nr_in_flight(cwq, get_work_color(work));
13636e84d644SOleg Nesterov 			ret = 1;
13646e84d644SOleg Nesterov 		}
13656e84d644SOleg Nesterov 	}
13668b03ae3cSTejun Heo 	spin_unlock_irq(&gcwq->lock);
13676e84d644SOleg Nesterov 
13686e84d644SOleg Nesterov 	return ret;
13696e84d644SOleg Nesterov }
13706e84d644SOleg Nesterov 
13716e84d644SOleg Nesterov static void wait_on_cpu_work(struct cpu_workqueue_struct *cwq,
1372b89deed3SOleg Nesterov 				struct work_struct *work)
1373b89deed3SOleg Nesterov {
13748b03ae3cSTejun Heo 	struct global_cwq *gcwq = cwq->gcwq;
1375b89deed3SOleg Nesterov 	struct wq_barrier barr;
1376affee4b2STejun Heo 	struct worker *worker;
1377b89deed3SOleg Nesterov 
13788b03ae3cSTejun Heo 	spin_lock_irq(&gcwq->lock);
1379affee4b2STejun Heo 
1380affee4b2STejun Heo 	worker = NULL;
1381c34056a3STejun Heo 	if (unlikely(cwq->worker && cwq->worker->current_work == work)) {
1382affee4b2STejun Heo 		worker = cwq->worker;
1383affee4b2STejun Heo 		insert_wq_barrier(cwq, &barr, work, worker);
1384b89deed3SOleg Nesterov 	}
1385affee4b2STejun Heo 
13868b03ae3cSTejun Heo 	spin_unlock_irq(&gcwq->lock);
1387b89deed3SOleg Nesterov 
1388affee4b2STejun Heo 	if (unlikely(worker)) {
1389b89deed3SOleg Nesterov 		wait_for_completion(&barr.done);
1390dc186ad7SThomas Gleixner 		destroy_work_on_stack(&barr.work);
1391dc186ad7SThomas Gleixner 	}
1392b89deed3SOleg Nesterov }
1393b89deed3SOleg Nesterov 
13946e84d644SOleg Nesterov static void wait_on_work(struct work_struct *work)
1395b89deed3SOleg Nesterov {
1396b89deed3SOleg Nesterov 	struct cpu_workqueue_struct *cwq;
139728e53bddSOleg Nesterov 	struct workqueue_struct *wq;
1398b1f4ec17SOleg Nesterov 	int cpu;
1399b89deed3SOleg Nesterov 
1400f293ea92SOleg Nesterov 	might_sleep();
1401f293ea92SOleg Nesterov 
14023295f0efSIngo Molnar 	lock_map_acquire(&work->lockdep_map);
14033295f0efSIngo Molnar 	lock_map_release(&work->lockdep_map);
14044e6045f1SJohannes Berg 
1405b89deed3SOleg Nesterov 	cwq = get_wq_data(work);
1406b89deed3SOleg Nesterov 	if (!cwq)
14073af24433SOleg Nesterov 		return;
1408b89deed3SOleg Nesterov 
140928e53bddSOleg Nesterov 	wq = cwq->wq;
141028e53bddSOleg Nesterov 
14111537663fSTejun Heo 	for_each_possible_cpu(cpu)
14124690c4abSTejun Heo 		wait_on_cpu_work(get_cwq(cpu, wq), work);
14136e84d644SOleg Nesterov }
14146e84d644SOleg Nesterov 
14151f1f642eSOleg Nesterov static int __cancel_work_timer(struct work_struct *work,
14161f1f642eSOleg Nesterov 				struct timer_list* timer)
14171f1f642eSOleg Nesterov {
14181f1f642eSOleg Nesterov 	int ret;
14191f1f642eSOleg Nesterov 
14201f1f642eSOleg Nesterov 	do {
14211f1f642eSOleg Nesterov 		ret = (timer && likely(del_timer(timer)));
14221f1f642eSOleg Nesterov 		if (!ret)
14231f1f642eSOleg Nesterov 			ret = try_to_grab_pending(work);
14241f1f642eSOleg Nesterov 		wait_on_work(work);
14251f1f642eSOleg Nesterov 	} while (unlikely(ret < 0));
14261f1f642eSOleg Nesterov 
14274d707b9fSOleg Nesterov 	clear_wq_data(work);
14281f1f642eSOleg Nesterov 	return ret;
14291f1f642eSOleg Nesterov }
14301f1f642eSOleg Nesterov 
14316e84d644SOleg Nesterov /**
14326e84d644SOleg Nesterov  * cancel_work_sync - block until a work_struct's callback has terminated
14336e84d644SOleg Nesterov  * @work: the work which is to be flushed
14346e84d644SOleg Nesterov  *
14351f1f642eSOleg Nesterov  * Returns true if @work was pending.
14361f1f642eSOleg Nesterov  *
14376e84d644SOleg Nesterov  * cancel_work_sync() will cancel the work if it is queued. If the work's
14386e84d644SOleg Nesterov  * callback appears to be running, cancel_work_sync() will block until it
14396e84d644SOleg Nesterov  * has completed.
14406e84d644SOleg Nesterov  *
14416e84d644SOleg Nesterov  * It is possible to use this function if the work re-queues itself. It can
14426e84d644SOleg Nesterov  * cancel the work even if it migrates to another workqueue, however in that
14436e84d644SOleg Nesterov  * case it only guarantees that work->func() has completed on the last queued
14446e84d644SOleg Nesterov  * workqueue.
14456e84d644SOleg Nesterov  *
14466e84d644SOleg Nesterov  * cancel_work_sync(&delayed_work->work) should be used only if ->timer is not
14476e84d644SOleg Nesterov  * pending, otherwise it goes into a busy-wait loop until the timer expires.
14486e84d644SOleg Nesterov  *
14496e84d644SOleg Nesterov  * The caller must ensure that workqueue_struct on which this work was last
14506e84d644SOleg Nesterov  * queued can't be destroyed before this function returns.
14516e84d644SOleg Nesterov  */
14521f1f642eSOleg Nesterov int cancel_work_sync(struct work_struct *work)
14536e84d644SOleg Nesterov {
14541f1f642eSOleg Nesterov 	return __cancel_work_timer(work, NULL);
1455b89deed3SOleg Nesterov }
145628e53bddSOleg Nesterov EXPORT_SYMBOL_GPL(cancel_work_sync);
1457b89deed3SOleg Nesterov 
14586e84d644SOleg Nesterov /**
1459f5a421a4SOleg Nesterov  * cancel_delayed_work_sync - reliably kill off a delayed work.
14606e84d644SOleg Nesterov  * @dwork: the delayed work struct
14616e84d644SOleg Nesterov  *
14621f1f642eSOleg Nesterov  * Returns true if @dwork was pending.
14631f1f642eSOleg Nesterov  *
14646e84d644SOleg Nesterov  * It is possible to use this function if @dwork rearms itself via queue_work()
14656e84d644SOleg Nesterov  * or queue_delayed_work(). See also the comment for cancel_work_sync().
14666e84d644SOleg Nesterov  */
14671f1f642eSOleg Nesterov int cancel_delayed_work_sync(struct delayed_work *dwork)
14686e84d644SOleg Nesterov {
14691f1f642eSOleg Nesterov 	return __cancel_work_timer(&dwork->work, &dwork->timer);
14706e84d644SOleg Nesterov }
1471f5a421a4SOleg Nesterov EXPORT_SYMBOL(cancel_delayed_work_sync);
14721da177e4SLinus Torvalds 
14736e84d644SOleg Nesterov static struct workqueue_struct *keventd_wq __read_mostly;
14741da177e4SLinus Torvalds 
14750fcb78c2SRolf Eike Beer /**
14760fcb78c2SRolf Eike Beer  * schedule_work - put work task in global workqueue
14770fcb78c2SRolf Eike Beer  * @work: job to be done
14780fcb78c2SRolf Eike Beer  *
14795b0f437dSBart Van Assche  * Returns zero if @work was already on the kernel-global workqueue and
14805b0f437dSBart Van Assche  * non-zero otherwise.
14815b0f437dSBart Van Assche  *
14825b0f437dSBart Van Assche  * This puts a job in the kernel-global workqueue if it was not already
14835b0f437dSBart Van Assche  * queued and leaves it in the same position on the kernel-global
14845b0f437dSBart Van Assche  * workqueue otherwise.
14850fcb78c2SRolf Eike Beer  */
14867ad5b3a5SHarvey Harrison int schedule_work(struct work_struct *work)
14871da177e4SLinus Torvalds {
14881da177e4SLinus Torvalds 	return queue_work(keventd_wq, work);
14891da177e4SLinus Torvalds }
1490ae90dd5dSDave Jones EXPORT_SYMBOL(schedule_work);
14911da177e4SLinus Torvalds 
1492c1a220e7SZhang Rui /*
1493c1a220e7SZhang Rui  * schedule_work_on - put work task on a specific cpu
1494c1a220e7SZhang Rui  * @cpu: cpu to put the work task on
1495c1a220e7SZhang Rui  * @work: job to be done
1496c1a220e7SZhang Rui  *
1497c1a220e7SZhang Rui  * This puts a job on a specific cpu
1498c1a220e7SZhang Rui  */
1499c1a220e7SZhang Rui int schedule_work_on(int cpu, struct work_struct *work)
1500c1a220e7SZhang Rui {
1501c1a220e7SZhang Rui 	return queue_work_on(cpu, keventd_wq, work);
1502c1a220e7SZhang Rui }
1503c1a220e7SZhang Rui EXPORT_SYMBOL(schedule_work_on);
1504c1a220e7SZhang Rui 
15050fcb78c2SRolf Eike Beer /**
15060fcb78c2SRolf Eike Beer  * schedule_delayed_work - put work task in global workqueue after delay
150752bad64dSDavid Howells  * @dwork: job to be done
150852bad64dSDavid Howells  * @delay: number of jiffies to wait or 0 for immediate execution
15090fcb78c2SRolf Eike Beer  *
15100fcb78c2SRolf Eike Beer  * After waiting for a given time this puts a job in the kernel-global
15110fcb78c2SRolf Eike Beer  * workqueue.
15120fcb78c2SRolf Eike Beer  */
15137ad5b3a5SHarvey Harrison int schedule_delayed_work(struct delayed_work *dwork,
151482f67cd9SIngo Molnar 					unsigned long delay)
15151da177e4SLinus Torvalds {
151652bad64dSDavid Howells 	return queue_delayed_work(keventd_wq, dwork, delay);
15171da177e4SLinus Torvalds }
1518ae90dd5dSDave Jones EXPORT_SYMBOL(schedule_delayed_work);
15191da177e4SLinus Torvalds 
15200fcb78c2SRolf Eike Beer /**
15218c53e463SLinus Torvalds  * flush_delayed_work - block until a dwork_struct's callback has terminated
15228c53e463SLinus Torvalds  * @dwork: the delayed work which is to be flushed
15238c53e463SLinus Torvalds  *
15248c53e463SLinus Torvalds  * Any timeout is cancelled, and any pending work is run immediately.
15258c53e463SLinus Torvalds  */
15268c53e463SLinus Torvalds void flush_delayed_work(struct delayed_work *dwork)
15278c53e463SLinus Torvalds {
15288c53e463SLinus Torvalds 	if (del_timer_sync(&dwork->timer)) {
15294690c4abSTejun Heo 		__queue_work(get_cpu(), get_wq_data(&dwork->work)->wq,
15304690c4abSTejun Heo 			     &dwork->work);
15318c53e463SLinus Torvalds 		put_cpu();
15328c53e463SLinus Torvalds 	}
15338c53e463SLinus Torvalds 	flush_work(&dwork->work);
15348c53e463SLinus Torvalds }
15358c53e463SLinus Torvalds EXPORT_SYMBOL(flush_delayed_work);
15368c53e463SLinus Torvalds 
15378c53e463SLinus Torvalds /**
15380fcb78c2SRolf Eike Beer  * schedule_delayed_work_on - queue work in global workqueue on CPU after delay
15390fcb78c2SRolf Eike Beer  * @cpu: cpu to use
154052bad64dSDavid Howells  * @dwork: job to be done
15410fcb78c2SRolf Eike Beer  * @delay: number of jiffies to wait
15420fcb78c2SRolf Eike Beer  *
15430fcb78c2SRolf Eike Beer  * After waiting for a given time this puts a job in the kernel-global
15440fcb78c2SRolf Eike Beer  * workqueue on the specified CPU.
15450fcb78c2SRolf Eike Beer  */
15461da177e4SLinus Torvalds int schedule_delayed_work_on(int cpu,
154752bad64dSDavid Howells 			struct delayed_work *dwork, unsigned long delay)
15481da177e4SLinus Torvalds {
154952bad64dSDavid Howells 	return queue_delayed_work_on(cpu, keventd_wq, dwork, delay);
15501da177e4SLinus Torvalds }
1551ae90dd5dSDave Jones EXPORT_SYMBOL(schedule_delayed_work_on);
15521da177e4SLinus Torvalds 
1553b6136773SAndrew Morton /**
1554b6136773SAndrew Morton  * schedule_on_each_cpu - call a function on each online CPU from keventd
1555b6136773SAndrew Morton  * @func: the function to call
1556b6136773SAndrew Morton  *
1557b6136773SAndrew Morton  * Returns zero on success.
1558b6136773SAndrew Morton  * Returns -ve errno on failure.
1559b6136773SAndrew Morton  *
1560b6136773SAndrew Morton  * schedule_on_each_cpu() is very slow.
1561b6136773SAndrew Morton  */
156265f27f38SDavid Howells int schedule_on_each_cpu(work_func_t func)
156315316ba8SChristoph Lameter {
156415316ba8SChristoph Lameter 	int cpu;
156565a64464SAndi Kleen 	int orig = -1;
1566b6136773SAndrew Morton 	struct work_struct *works;
156715316ba8SChristoph Lameter 
1568b6136773SAndrew Morton 	works = alloc_percpu(struct work_struct);
1569b6136773SAndrew Morton 	if (!works)
157015316ba8SChristoph Lameter 		return -ENOMEM;
1571b6136773SAndrew Morton 
157295402b38SGautham R Shenoy 	get_online_cpus();
157393981800STejun Heo 
157493981800STejun Heo 	/*
157593981800STejun Heo 	 * When running in keventd don't schedule a work item on
157693981800STejun Heo 	 * itself.  Can just call directly because the work queue is
157793981800STejun Heo 	 * already bound.  This also is faster.
157893981800STejun Heo 	 */
157993981800STejun Heo 	if (current_is_keventd())
158093981800STejun Heo 		orig = raw_smp_processor_id();
158193981800STejun Heo 
158215316ba8SChristoph Lameter 	for_each_online_cpu(cpu) {
15839bfb1839SIngo Molnar 		struct work_struct *work = per_cpu_ptr(works, cpu);
15849bfb1839SIngo Molnar 
15859bfb1839SIngo Molnar 		INIT_WORK(work, func);
158693981800STejun Heo 		if (cpu != orig)
15878de6d308SOleg Nesterov 			schedule_work_on(cpu, work);
158815316ba8SChristoph Lameter 	}
158993981800STejun Heo 	if (orig >= 0)
159093981800STejun Heo 		func(per_cpu_ptr(works, orig));
159193981800STejun Heo 
159293981800STejun Heo 	for_each_online_cpu(cpu)
15938616a89aSOleg Nesterov 		flush_work(per_cpu_ptr(works, cpu));
159493981800STejun Heo 
159595402b38SGautham R Shenoy 	put_online_cpus();
1596b6136773SAndrew Morton 	free_percpu(works);
159715316ba8SChristoph Lameter 	return 0;
159815316ba8SChristoph Lameter }
159915316ba8SChristoph Lameter 
1600eef6a7d5SAlan Stern /**
1601eef6a7d5SAlan Stern  * flush_scheduled_work - ensure that any scheduled work has run to completion.
1602eef6a7d5SAlan Stern  *
1603eef6a7d5SAlan Stern  * Forces execution of the kernel-global workqueue and blocks until its
1604eef6a7d5SAlan Stern  * completion.
1605eef6a7d5SAlan Stern  *
1606eef6a7d5SAlan Stern  * Think twice before calling this function!  It's very easy to get into
1607eef6a7d5SAlan Stern  * trouble if you don't take great care.  Either of the following situations
1608eef6a7d5SAlan Stern  * will lead to deadlock:
1609eef6a7d5SAlan Stern  *
1610eef6a7d5SAlan Stern  *	One of the work items currently on the workqueue needs to acquire
1611eef6a7d5SAlan Stern  *	a lock held by your code or its caller.
1612eef6a7d5SAlan Stern  *
1613eef6a7d5SAlan Stern  *	Your code is running in the context of a work routine.
1614eef6a7d5SAlan Stern  *
1615eef6a7d5SAlan Stern  * They will be detected by lockdep when they occur, but the first might not
1616eef6a7d5SAlan Stern  * occur very often.  It depends on what work items are on the workqueue and
1617eef6a7d5SAlan Stern  * what locks they need, which you have no control over.
1618eef6a7d5SAlan Stern  *
1619eef6a7d5SAlan Stern  * In most situations flushing the entire workqueue is overkill; you merely
1620eef6a7d5SAlan Stern  * need to know that a particular work item isn't queued and isn't running.
1621eef6a7d5SAlan Stern  * In such cases you should use cancel_delayed_work_sync() or
1622eef6a7d5SAlan Stern  * cancel_work_sync() instead.
1623eef6a7d5SAlan Stern  */
16241da177e4SLinus Torvalds void flush_scheduled_work(void)
16251da177e4SLinus Torvalds {
16261da177e4SLinus Torvalds 	flush_workqueue(keventd_wq);
16271da177e4SLinus Torvalds }
1628ae90dd5dSDave Jones EXPORT_SYMBOL(flush_scheduled_work);
16291da177e4SLinus Torvalds 
16301da177e4SLinus Torvalds /**
16311fa44ecaSJames Bottomley  * execute_in_process_context - reliably execute the routine with user context
16321fa44ecaSJames Bottomley  * @fn:		the function to execute
16331fa44ecaSJames Bottomley  * @ew:		guaranteed storage for the execute work structure (must
16341fa44ecaSJames Bottomley  *		be available when the work executes)
16351fa44ecaSJames Bottomley  *
16361fa44ecaSJames Bottomley  * Executes the function immediately if process context is available,
16371fa44ecaSJames Bottomley  * otherwise schedules the function for delayed execution.
16381fa44ecaSJames Bottomley  *
16391fa44ecaSJames Bottomley  * Returns:	0 - function was executed
16401fa44ecaSJames Bottomley  *		1 - function was scheduled for execution
16411fa44ecaSJames Bottomley  */
164265f27f38SDavid Howells int execute_in_process_context(work_func_t fn, struct execute_work *ew)
16431fa44ecaSJames Bottomley {
16441fa44ecaSJames Bottomley 	if (!in_interrupt()) {
164565f27f38SDavid Howells 		fn(&ew->work);
16461fa44ecaSJames Bottomley 		return 0;
16471fa44ecaSJames Bottomley 	}
16481fa44ecaSJames Bottomley 
164965f27f38SDavid Howells 	INIT_WORK(&ew->work, fn);
16501fa44ecaSJames Bottomley 	schedule_work(&ew->work);
16511fa44ecaSJames Bottomley 
16521fa44ecaSJames Bottomley 	return 1;
16531fa44ecaSJames Bottomley }
16541fa44ecaSJames Bottomley EXPORT_SYMBOL_GPL(execute_in_process_context);
16551fa44ecaSJames Bottomley 
16561da177e4SLinus Torvalds int keventd_up(void)
16571da177e4SLinus Torvalds {
16581da177e4SLinus Torvalds 	return keventd_wq != NULL;
16591da177e4SLinus Torvalds }
16601da177e4SLinus Torvalds 
16611da177e4SLinus Torvalds int current_is_keventd(void)
16621da177e4SLinus Torvalds {
16631da177e4SLinus Torvalds 	struct cpu_workqueue_struct *cwq;
1664d243769dSHugh Dickins 	int cpu = raw_smp_processor_id(); /* preempt-safe: keventd is per-cpu */
16651da177e4SLinus Torvalds 	int ret = 0;
16661da177e4SLinus Torvalds 
16671da177e4SLinus Torvalds 	BUG_ON(!keventd_wq);
16681da177e4SLinus Torvalds 
16691537663fSTejun Heo 	cwq = get_cwq(cpu, keventd_wq);
1670c34056a3STejun Heo 	if (current == cwq->worker->task)
16711da177e4SLinus Torvalds 		ret = 1;
16721da177e4SLinus Torvalds 
16731da177e4SLinus Torvalds 	return ret;
16741da177e4SLinus Torvalds 
16751da177e4SLinus Torvalds }
16761da177e4SLinus Torvalds 
16770f900049STejun Heo static struct cpu_workqueue_struct *alloc_cwqs(void)
16780f900049STejun Heo {
16790f900049STejun Heo 	/*
16800f900049STejun Heo 	 * cwqs are forced aligned according to WORK_STRUCT_FLAG_BITS.
16810f900049STejun Heo 	 * Make sure that the alignment isn't lower than that of
16820f900049STejun Heo 	 * unsigned long long.
16830f900049STejun Heo 	 */
16840f900049STejun Heo 	const size_t size = sizeof(struct cpu_workqueue_struct);
16850f900049STejun Heo 	const size_t align = max_t(size_t, 1 << WORK_STRUCT_FLAG_BITS,
16860f900049STejun Heo 				   __alignof__(unsigned long long));
16870f900049STejun Heo 	struct cpu_workqueue_struct *cwqs;
16880f900049STejun Heo #ifndef CONFIG_SMP
16890f900049STejun Heo 	void *ptr;
16900f900049STejun Heo 
16910f900049STejun Heo 	/*
16920f900049STejun Heo 	 * On UP, percpu allocator doesn't honor alignment parameter
16930f900049STejun Heo 	 * and simply uses arch-dependent default.  Allocate enough
16940f900049STejun Heo 	 * room to align cwq and put an extra pointer at the end
16950f900049STejun Heo 	 * pointing back to the originally allocated pointer which
16960f900049STejun Heo 	 * will be used for free.
16970f900049STejun Heo 	 *
16980f900049STejun Heo 	 * FIXME: This really belongs to UP percpu code.  Update UP
16990f900049STejun Heo 	 * percpu code to honor alignment and remove this ugliness.
17000f900049STejun Heo 	 */
17010f900049STejun Heo 	ptr = __alloc_percpu(size + align + sizeof(void *), 1);
17020f900049STejun Heo 	cwqs = PTR_ALIGN(ptr, align);
17030f900049STejun Heo 	*(void **)per_cpu_ptr(cwqs + 1, 0) = ptr;
17040f900049STejun Heo #else
17050f900049STejun Heo 	/* On SMP, percpu allocator can do it itself */
17060f900049STejun Heo 	cwqs = __alloc_percpu(size, align);
17070f900049STejun Heo #endif
17080f900049STejun Heo 	/* just in case, make sure it's actually aligned */
17090f900049STejun Heo 	BUG_ON(!IS_ALIGNED((unsigned long)cwqs, align));
17100f900049STejun Heo 	return cwqs;
17110f900049STejun Heo }
17120f900049STejun Heo 
17130f900049STejun Heo static void free_cwqs(struct cpu_workqueue_struct *cwqs)
17140f900049STejun Heo {
17150f900049STejun Heo #ifndef CONFIG_SMP
17160f900049STejun Heo 	/* on UP, the pointer to free is stored right after the cwq */
17170f900049STejun Heo 	if (cwqs)
17180f900049STejun Heo 		free_percpu(*(void **)per_cpu_ptr(cwqs + 1, 0));
17190f900049STejun Heo #else
17200f900049STejun Heo 	free_percpu(cwqs);
17210f900049STejun Heo #endif
17220f900049STejun Heo }
17230f900049STejun Heo 
17244e6045f1SJohannes Berg struct workqueue_struct *__create_workqueue_key(const char *name,
172597e37d7bSTejun Heo 						unsigned int flags,
17261e19ffc6STejun Heo 						int max_active,
1727eb13ba87SJohannes Berg 						struct lock_class_key *key,
1728eb13ba87SJohannes Berg 						const char *lock_name)
17293af24433SOleg Nesterov {
17301537663fSTejun Heo 	bool singlethread = flags & WQ_SINGLE_THREAD;
17313af24433SOleg Nesterov 	struct workqueue_struct *wq;
1732c34056a3STejun Heo 	bool failed = false;
1733c34056a3STejun Heo 	unsigned int cpu;
17343af24433SOleg Nesterov 
17351e19ffc6STejun Heo 	max_active = clamp_val(max_active, 1, INT_MAX);
17361e19ffc6STejun Heo 
17373af24433SOleg Nesterov 	wq = kzalloc(sizeof(*wq), GFP_KERNEL);
17383af24433SOleg Nesterov 	if (!wq)
17394690c4abSTejun Heo 		goto err;
17403af24433SOleg Nesterov 
17410f900049STejun Heo 	wq->cpu_wq = alloc_cwqs();
17424690c4abSTejun Heo 	if (!wq->cpu_wq)
17434690c4abSTejun Heo 		goto err;
17443af24433SOleg Nesterov 
174597e37d7bSTejun Heo 	wq->flags = flags;
1746a0a1a5fdSTejun Heo 	wq->saved_max_active = max_active;
174773f53c4aSTejun Heo 	mutex_init(&wq->flush_mutex);
174873f53c4aSTejun Heo 	atomic_set(&wq->nr_cwqs_to_flush, 0);
174973f53c4aSTejun Heo 	INIT_LIST_HEAD(&wq->flusher_queue);
175073f53c4aSTejun Heo 	INIT_LIST_HEAD(&wq->flusher_overflow);
17513af24433SOleg Nesterov 	wq->name = name;
1752eb13ba87SJohannes Berg 	lockdep_init_map(&wq->lockdep_map, lock_name, key, 0);
1753cce1a165SOleg Nesterov 	INIT_LIST_HEAD(&wq->list);
17543af24433SOleg Nesterov 
17553da1c84cSOleg Nesterov 	cpu_maps_update_begin();
17566af8bf3dSOleg Nesterov 	/*
17576af8bf3dSOleg Nesterov 	 * We must initialize cwqs for each possible cpu even if we
17586af8bf3dSOleg Nesterov 	 * are going to call destroy_workqueue() finally. Otherwise
17596af8bf3dSOleg Nesterov 	 * cpu_up() can hit the uninitialized cwq once we drop the
17606af8bf3dSOleg Nesterov 	 * lock.
17616af8bf3dSOleg Nesterov 	 */
17623af24433SOleg Nesterov 	for_each_possible_cpu(cpu) {
17631537663fSTejun Heo 		struct cpu_workqueue_struct *cwq = get_cwq(cpu, wq);
17648b03ae3cSTejun Heo 		struct global_cwq *gcwq = get_gcwq(cpu);
17651537663fSTejun Heo 
17660f900049STejun Heo 		BUG_ON((unsigned long)cwq & WORK_STRUCT_FLAG_MASK);
17678b03ae3cSTejun Heo 		cwq->gcwq = gcwq;
1768c34056a3STejun Heo 		cwq->wq = wq;
176973f53c4aSTejun Heo 		cwq->flush_color = -1;
17701e19ffc6STejun Heo 		cwq->max_active = max_active;
17711537663fSTejun Heo 		INIT_LIST_HEAD(&cwq->worklist);
17721e19ffc6STejun Heo 		INIT_LIST_HEAD(&cwq->delayed_works);
17731537663fSTejun Heo 
1774c34056a3STejun Heo 		if (failed)
17753af24433SOleg Nesterov 			continue;
1776c34056a3STejun Heo 		cwq->worker = create_worker(cwq,
1777c34056a3STejun Heo 					    cpu_online(cpu) && !singlethread);
1778c34056a3STejun Heo 		if (cwq->worker)
1779c34056a3STejun Heo 			start_worker(cwq->worker);
17801537663fSTejun Heo 		else
1781c34056a3STejun Heo 			failed = true;
17823af24433SOleg Nesterov 	}
17831537663fSTejun Heo 
1784a0a1a5fdSTejun Heo 	/*
1785a0a1a5fdSTejun Heo 	 * workqueue_lock protects global freeze state and workqueues
1786a0a1a5fdSTejun Heo 	 * list.  Grab it, set max_active accordingly and add the new
1787a0a1a5fdSTejun Heo 	 * workqueue to workqueues list.
1788a0a1a5fdSTejun Heo 	 */
17891537663fSTejun Heo 	spin_lock(&workqueue_lock);
1790a0a1a5fdSTejun Heo 
1791a0a1a5fdSTejun Heo 	if (workqueue_freezing && wq->flags & WQ_FREEZEABLE)
1792a0a1a5fdSTejun Heo 		for_each_possible_cpu(cpu)
1793a0a1a5fdSTejun Heo 			get_cwq(cpu, wq)->max_active = 0;
1794a0a1a5fdSTejun Heo 
17951537663fSTejun Heo 	list_add(&wq->list, &workqueues);
1796a0a1a5fdSTejun Heo 
17971537663fSTejun Heo 	spin_unlock(&workqueue_lock);
17981537663fSTejun Heo 
17993da1c84cSOleg Nesterov 	cpu_maps_update_done();
18003af24433SOleg Nesterov 
1801c34056a3STejun Heo 	if (failed) {
18023af24433SOleg Nesterov 		destroy_workqueue(wq);
18033af24433SOleg Nesterov 		wq = NULL;
18043af24433SOleg Nesterov 	}
18053af24433SOleg Nesterov 	return wq;
18064690c4abSTejun Heo err:
18074690c4abSTejun Heo 	if (wq) {
18080f900049STejun Heo 		free_cwqs(wq->cpu_wq);
18094690c4abSTejun Heo 		kfree(wq);
18104690c4abSTejun Heo 	}
18114690c4abSTejun Heo 	return NULL;
18123af24433SOleg Nesterov }
18134e6045f1SJohannes Berg EXPORT_SYMBOL_GPL(__create_workqueue_key);
18143af24433SOleg Nesterov 
18153af24433SOleg Nesterov /**
18163af24433SOleg Nesterov  * destroy_workqueue - safely terminate a workqueue
18173af24433SOleg Nesterov  * @wq: target workqueue
18183af24433SOleg Nesterov  *
18193af24433SOleg Nesterov  * Safely destroy a workqueue. All work currently pending will be done first.
18203af24433SOleg Nesterov  */
18213af24433SOleg Nesterov void destroy_workqueue(struct workqueue_struct *wq)
18223af24433SOleg Nesterov {
1823c8e55f36STejun Heo 	unsigned int cpu;
18243af24433SOleg Nesterov 
1825a0a1a5fdSTejun Heo 	flush_workqueue(wq);
1826a0a1a5fdSTejun Heo 
1827a0a1a5fdSTejun Heo 	/*
1828a0a1a5fdSTejun Heo 	 * wq list is used to freeze wq, remove from list after
1829a0a1a5fdSTejun Heo 	 * flushing is complete in case freeze races us.
1830a0a1a5fdSTejun Heo 	 */
18313da1c84cSOleg Nesterov 	cpu_maps_update_begin();
183295402b38SGautham R Shenoy 	spin_lock(&workqueue_lock);
18333af24433SOleg Nesterov 	list_del(&wq->list);
183495402b38SGautham R Shenoy 	spin_unlock(&workqueue_lock);
18353da1c84cSOleg Nesterov 	cpu_maps_update_done();
18363af24433SOleg Nesterov 
183773f53c4aSTejun Heo 	for_each_possible_cpu(cpu) {
183873f53c4aSTejun Heo 		struct cpu_workqueue_struct *cwq = get_cwq(cpu, wq);
183973f53c4aSTejun Heo 		int i;
184073f53c4aSTejun Heo 
1841c34056a3STejun Heo 		if (cwq->worker) {
1842c8e55f36STejun Heo 			spin_lock_irq(&cwq->gcwq->lock);
1843c34056a3STejun Heo 			destroy_worker(cwq->worker);
1844c34056a3STejun Heo 			cwq->worker = NULL;
1845c8e55f36STejun Heo 			spin_unlock_irq(&cwq->gcwq->lock);
184673f53c4aSTejun Heo 		}
184773f53c4aSTejun Heo 
184873f53c4aSTejun Heo 		for (i = 0; i < WORK_NR_COLORS; i++)
184973f53c4aSTejun Heo 			BUG_ON(cwq->nr_in_flight[i]);
18501e19ffc6STejun Heo 		BUG_ON(cwq->nr_active);
18511e19ffc6STejun Heo 		BUG_ON(!list_empty(&cwq->delayed_works));
185273f53c4aSTejun Heo 	}
18531537663fSTejun Heo 
18540f900049STejun Heo 	free_cwqs(wq->cpu_wq);
18553af24433SOleg Nesterov 	kfree(wq);
18563af24433SOleg Nesterov }
18573af24433SOleg Nesterov EXPORT_SYMBOL_GPL(destroy_workqueue);
18583af24433SOleg Nesterov 
1859*db7bccf4STejun Heo /*
1860*db7bccf4STejun Heo  * CPU hotplug.
1861*db7bccf4STejun Heo  *
1862*db7bccf4STejun Heo  * CPU hotplug is implemented by allowing cwqs to be detached from
1863*db7bccf4STejun Heo  * CPU, running with unbound workers and allowing them to be
1864*db7bccf4STejun Heo  * reattached later if the cpu comes back online.  A separate thread
1865*db7bccf4STejun Heo  * is created to govern cwqs in such state and is called the trustee.
1866*db7bccf4STejun Heo  *
1867*db7bccf4STejun Heo  * Trustee states and their descriptions.
1868*db7bccf4STejun Heo  *
1869*db7bccf4STejun Heo  * START	Command state used on startup.  On CPU_DOWN_PREPARE, a
1870*db7bccf4STejun Heo  *		new trustee is started with this state.
1871*db7bccf4STejun Heo  *
1872*db7bccf4STejun Heo  * IN_CHARGE	Once started, trustee will enter this state after
1873*db7bccf4STejun Heo  *		making all existing workers rogue.  DOWN_PREPARE waits
1874*db7bccf4STejun Heo  *		for trustee to enter this state.  After reaching
1875*db7bccf4STejun Heo  *		IN_CHARGE, trustee tries to execute the pending
1876*db7bccf4STejun Heo  *		worklist until it's empty and the state is set to
1877*db7bccf4STejun Heo  *		BUTCHER, or the state is set to RELEASE.
1878*db7bccf4STejun Heo  *
1879*db7bccf4STejun Heo  * BUTCHER	Command state which is set by the cpu callback after
1880*db7bccf4STejun Heo  *		the cpu has went down.  Once this state is set trustee
1881*db7bccf4STejun Heo  *		knows that there will be no new works on the worklist
1882*db7bccf4STejun Heo  *		and once the worklist is empty it can proceed to
1883*db7bccf4STejun Heo  *		killing idle workers.
1884*db7bccf4STejun Heo  *
1885*db7bccf4STejun Heo  * RELEASE	Command state which is set by the cpu callback if the
1886*db7bccf4STejun Heo  *		cpu down has been canceled or it has come online
1887*db7bccf4STejun Heo  *		again.  After recognizing this state, trustee stops
1888*db7bccf4STejun Heo  *		trying to drain or butcher and transits to DONE.
1889*db7bccf4STejun Heo  *
1890*db7bccf4STejun Heo  * DONE		Trustee will enter this state after BUTCHER or RELEASE
1891*db7bccf4STejun Heo  *		is complete.
1892*db7bccf4STejun Heo  *
1893*db7bccf4STejun Heo  *          trustee                 CPU                draining
1894*db7bccf4STejun Heo  *         took over                down               complete
1895*db7bccf4STejun Heo  * START -----------> IN_CHARGE -----------> BUTCHER -----------> DONE
1896*db7bccf4STejun Heo  *                        |                     |                  ^
1897*db7bccf4STejun Heo  *                        | CPU is back online  v   return workers |
1898*db7bccf4STejun Heo  *                         ----------------> RELEASE --------------
1899*db7bccf4STejun Heo  */
1900*db7bccf4STejun Heo 
1901*db7bccf4STejun Heo /**
1902*db7bccf4STejun Heo  * trustee_wait_event_timeout - timed event wait for trustee
1903*db7bccf4STejun Heo  * @cond: condition to wait for
1904*db7bccf4STejun Heo  * @timeout: timeout in jiffies
1905*db7bccf4STejun Heo  *
1906*db7bccf4STejun Heo  * wait_event_timeout() for trustee to use.  Handles locking and
1907*db7bccf4STejun Heo  * checks for RELEASE request.
1908*db7bccf4STejun Heo  *
1909*db7bccf4STejun Heo  * CONTEXT:
1910*db7bccf4STejun Heo  * spin_lock_irq(gcwq->lock) which may be released and regrabbed
1911*db7bccf4STejun Heo  * multiple times.  To be used by trustee.
1912*db7bccf4STejun Heo  *
1913*db7bccf4STejun Heo  * RETURNS:
1914*db7bccf4STejun Heo  * Positive indicating left time if @cond is satisfied, 0 if timed
1915*db7bccf4STejun Heo  * out, -1 if canceled.
1916*db7bccf4STejun Heo  */
1917*db7bccf4STejun Heo #define trustee_wait_event_timeout(cond, timeout) ({			\
1918*db7bccf4STejun Heo 	long __ret = (timeout);						\
1919*db7bccf4STejun Heo 	while (!((cond) || (gcwq->trustee_state == TRUSTEE_RELEASE)) &&	\
1920*db7bccf4STejun Heo 	       __ret) {							\
1921*db7bccf4STejun Heo 		spin_unlock_irq(&gcwq->lock);				\
1922*db7bccf4STejun Heo 		__wait_event_timeout(gcwq->trustee_wait, (cond) ||	\
1923*db7bccf4STejun Heo 			(gcwq->trustee_state == TRUSTEE_RELEASE),	\
1924*db7bccf4STejun Heo 			__ret);						\
1925*db7bccf4STejun Heo 		spin_lock_irq(&gcwq->lock);				\
1926*db7bccf4STejun Heo 	}								\
1927*db7bccf4STejun Heo 	gcwq->trustee_state == TRUSTEE_RELEASE ? -1 : (__ret);		\
1928*db7bccf4STejun Heo })
1929*db7bccf4STejun Heo 
1930*db7bccf4STejun Heo /**
1931*db7bccf4STejun Heo  * trustee_wait_event - event wait for trustee
1932*db7bccf4STejun Heo  * @cond: condition to wait for
1933*db7bccf4STejun Heo  *
1934*db7bccf4STejun Heo  * wait_event() for trustee to use.  Automatically handles locking and
1935*db7bccf4STejun Heo  * checks for CANCEL request.
1936*db7bccf4STejun Heo  *
1937*db7bccf4STejun Heo  * CONTEXT:
1938*db7bccf4STejun Heo  * spin_lock_irq(gcwq->lock) which may be released and regrabbed
1939*db7bccf4STejun Heo  * multiple times.  To be used by trustee.
1940*db7bccf4STejun Heo  *
1941*db7bccf4STejun Heo  * RETURNS:
1942*db7bccf4STejun Heo  * 0 if @cond is satisfied, -1 if canceled.
1943*db7bccf4STejun Heo  */
1944*db7bccf4STejun Heo #define trustee_wait_event(cond) ({					\
1945*db7bccf4STejun Heo 	long __ret1;							\
1946*db7bccf4STejun Heo 	__ret1 = trustee_wait_event_timeout(cond, MAX_SCHEDULE_TIMEOUT);\
1947*db7bccf4STejun Heo 	__ret1 < 0 ? -1 : 0;						\
1948*db7bccf4STejun Heo })
1949*db7bccf4STejun Heo 
1950*db7bccf4STejun Heo static int __cpuinit trustee_thread(void *__gcwq)
1951*db7bccf4STejun Heo {
1952*db7bccf4STejun Heo 	struct global_cwq *gcwq = __gcwq;
1953*db7bccf4STejun Heo 	struct worker *worker;
1954*db7bccf4STejun Heo 	struct hlist_node *pos;
1955*db7bccf4STejun Heo 	int i;
1956*db7bccf4STejun Heo 
1957*db7bccf4STejun Heo 	BUG_ON(gcwq->cpu != smp_processor_id());
1958*db7bccf4STejun Heo 
1959*db7bccf4STejun Heo 	spin_lock_irq(&gcwq->lock);
1960*db7bccf4STejun Heo 	/*
1961*db7bccf4STejun Heo 	 * Make all multithread workers rogue.  Trustee must be bound
1962*db7bccf4STejun Heo 	 * to the target cpu and can't be cancelled.
1963*db7bccf4STejun Heo 	 */
1964*db7bccf4STejun Heo 	BUG_ON(gcwq->cpu != smp_processor_id());
1965*db7bccf4STejun Heo 
1966*db7bccf4STejun Heo 	list_for_each_entry(worker, &gcwq->idle_list, entry)
1967*db7bccf4STejun Heo 		if (!(worker->cwq->wq->flags & WQ_SINGLE_THREAD))
1968*db7bccf4STejun Heo 			worker->flags |= WORKER_ROGUE;
1969*db7bccf4STejun Heo 
1970*db7bccf4STejun Heo 	for_each_busy_worker(worker, i, pos, gcwq)
1971*db7bccf4STejun Heo 		if (!(worker->cwq->wq->flags & WQ_SINGLE_THREAD))
1972*db7bccf4STejun Heo 			worker->flags |= WORKER_ROGUE;
1973*db7bccf4STejun Heo 
1974*db7bccf4STejun Heo 	/*
1975*db7bccf4STejun Heo 	 * We're now in charge.  Notify and proceed to drain.  We need
1976*db7bccf4STejun Heo 	 * to keep the gcwq running during the whole CPU down
1977*db7bccf4STejun Heo 	 * procedure as other cpu hotunplug callbacks may need to
1978*db7bccf4STejun Heo 	 * flush currently running tasks.
1979*db7bccf4STejun Heo 	 */
1980*db7bccf4STejun Heo 	gcwq->trustee_state = TRUSTEE_IN_CHARGE;
1981*db7bccf4STejun Heo 	wake_up_all(&gcwq->trustee_wait);
1982*db7bccf4STejun Heo 
1983*db7bccf4STejun Heo 	/*
1984*db7bccf4STejun Heo 	 * The original cpu is in the process of dying and may go away
1985*db7bccf4STejun Heo 	 * anytime now.  When that happens, we and all workers would
1986*db7bccf4STejun Heo 	 * be migrated to other cpus.  Try draining any left work.
1987*db7bccf4STejun Heo 	 * Note that if the gcwq is frozen, there may be frozen works
1988*db7bccf4STejun Heo 	 * in freezeable cwqs.  Don't declare completion while frozen.
1989*db7bccf4STejun Heo 	 */
1990*db7bccf4STejun Heo 	while (gcwq->nr_workers != gcwq->nr_idle ||
1991*db7bccf4STejun Heo 	       gcwq->flags & GCWQ_FREEZING ||
1992*db7bccf4STejun Heo 	       gcwq->trustee_state == TRUSTEE_IN_CHARGE) {
1993*db7bccf4STejun Heo 		/* give a breather */
1994*db7bccf4STejun Heo 		if (trustee_wait_event_timeout(false, TRUSTEE_COOLDOWN) < 0)
1995*db7bccf4STejun Heo 			break;
1996*db7bccf4STejun Heo 	}
1997*db7bccf4STejun Heo 
1998*db7bccf4STejun Heo 	/* notify completion */
1999*db7bccf4STejun Heo 	gcwq->trustee = NULL;
2000*db7bccf4STejun Heo 	gcwq->trustee_state = TRUSTEE_DONE;
2001*db7bccf4STejun Heo 	wake_up_all(&gcwq->trustee_wait);
2002*db7bccf4STejun Heo 	spin_unlock_irq(&gcwq->lock);
2003*db7bccf4STejun Heo 	return 0;
2004*db7bccf4STejun Heo }
2005*db7bccf4STejun Heo 
2006*db7bccf4STejun Heo /**
2007*db7bccf4STejun Heo  * wait_trustee_state - wait for trustee to enter the specified state
2008*db7bccf4STejun Heo  * @gcwq: gcwq the trustee of interest belongs to
2009*db7bccf4STejun Heo  * @state: target state to wait for
2010*db7bccf4STejun Heo  *
2011*db7bccf4STejun Heo  * Wait for the trustee to reach @state.  DONE is already matched.
2012*db7bccf4STejun Heo  *
2013*db7bccf4STejun Heo  * CONTEXT:
2014*db7bccf4STejun Heo  * spin_lock_irq(gcwq->lock) which may be released and regrabbed
2015*db7bccf4STejun Heo  * multiple times.  To be used by cpu_callback.
2016*db7bccf4STejun Heo  */
2017*db7bccf4STejun Heo static void __cpuinit wait_trustee_state(struct global_cwq *gcwq, int state)
2018*db7bccf4STejun Heo {
2019*db7bccf4STejun Heo 	if (!(gcwq->trustee_state == state ||
2020*db7bccf4STejun Heo 	      gcwq->trustee_state == TRUSTEE_DONE)) {
2021*db7bccf4STejun Heo 		spin_unlock_irq(&gcwq->lock);
2022*db7bccf4STejun Heo 		__wait_event(gcwq->trustee_wait,
2023*db7bccf4STejun Heo 			     gcwq->trustee_state == state ||
2024*db7bccf4STejun Heo 			     gcwq->trustee_state == TRUSTEE_DONE);
2025*db7bccf4STejun Heo 		spin_lock_irq(&gcwq->lock);
2026*db7bccf4STejun Heo 	}
2027*db7bccf4STejun Heo }
2028*db7bccf4STejun Heo 
20299c7b216dSChandra Seetharaman static int __devinit workqueue_cpu_callback(struct notifier_block *nfb,
20301da177e4SLinus Torvalds 						unsigned long action,
20311da177e4SLinus Torvalds 						void *hcpu)
20321da177e4SLinus Torvalds {
20333af24433SOleg Nesterov 	unsigned int cpu = (unsigned long)hcpu;
2034*db7bccf4STejun Heo 	struct global_cwq *gcwq = get_gcwq(cpu);
2035*db7bccf4STejun Heo 	struct task_struct *new_trustee = NULL;
2036*db7bccf4STejun Heo 	struct worker *worker;
2037*db7bccf4STejun Heo 	struct hlist_node *pos;
2038*db7bccf4STejun Heo 	unsigned long flags;
2039*db7bccf4STejun Heo 	int i;
20401da177e4SLinus Torvalds 
20418bb78442SRafael J. Wysocki 	action &= ~CPU_TASKS_FROZEN;
20428bb78442SRafael J. Wysocki 
2043*db7bccf4STejun Heo 	switch (action) {
2044*db7bccf4STejun Heo 	case CPU_DOWN_PREPARE:
2045*db7bccf4STejun Heo 		new_trustee = kthread_create(trustee_thread, gcwq,
2046*db7bccf4STejun Heo 					     "workqueue_trustee/%d\n", cpu);
2047*db7bccf4STejun Heo 		if (IS_ERR(new_trustee))
2048*db7bccf4STejun Heo 			return notifier_from_errno(PTR_ERR(new_trustee));
2049*db7bccf4STejun Heo 		kthread_bind(new_trustee, cpu);
2050*db7bccf4STejun Heo 	}
20511537663fSTejun Heo 
2052*db7bccf4STejun Heo 	/* some are called w/ irq disabled, don't disturb irq status */
2053*db7bccf4STejun Heo 	spin_lock_irqsave(&gcwq->lock, flags);
20543af24433SOleg Nesterov 
20553af24433SOleg Nesterov 	switch (action) {
2056*db7bccf4STejun Heo 	case CPU_DOWN_PREPARE:
2057*db7bccf4STejun Heo 		/* initialize trustee and tell it to acquire the gcwq */
2058*db7bccf4STejun Heo 		BUG_ON(gcwq->trustee || gcwq->trustee_state != TRUSTEE_DONE);
2059*db7bccf4STejun Heo 		gcwq->trustee = new_trustee;
2060*db7bccf4STejun Heo 		gcwq->trustee_state = TRUSTEE_START;
2061*db7bccf4STejun Heo 		wake_up_process(gcwq->trustee);
2062*db7bccf4STejun Heo 		wait_trustee_state(gcwq, TRUSTEE_IN_CHARGE);
2063*db7bccf4STejun Heo 		break;
2064*db7bccf4STejun Heo 
20653da1c84cSOleg Nesterov 	case CPU_POST_DEAD:
2066*db7bccf4STejun Heo 		gcwq->trustee_state = TRUSTEE_BUTCHER;
2067*db7bccf4STejun Heo 		break;
2068*db7bccf4STejun Heo 
2069*db7bccf4STejun Heo 	case CPU_DOWN_FAILED:
2070*db7bccf4STejun Heo 	case CPU_ONLINE:
2071*db7bccf4STejun Heo 		if (gcwq->trustee_state != TRUSTEE_DONE) {
2072*db7bccf4STejun Heo 			gcwq->trustee_state = TRUSTEE_RELEASE;
2073*db7bccf4STejun Heo 			wake_up_process(gcwq->trustee);
2074*db7bccf4STejun Heo 			wait_trustee_state(gcwq, TRUSTEE_DONE);
2075*db7bccf4STejun Heo 		}
2076*db7bccf4STejun Heo 
2077*db7bccf4STejun Heo 		/* clear ROGUE from all multithread workers */
2078*db7bccf4STejun Heo 		list_for_each_entry(worker, &gcwq->idle_list, entry)
2079*db7bccf4STejun Heo 			if (!(worker->cwq->wq->flags & WQ_SINGLE_THREAD))
2080*db7bccf4STejun Heo 				worker->flags &= ~WORKER_ROGUE;
2081*db7bccf4STejun Heo 
2082*db7bccf4STejun Heo 		for_each_busy_worker(worker, i, pos, gcwq)
2083*db7bccf4STejun Heo 			if (!(worker->cwq->wq->flags & WQ_SINGLE_THREAD))
2084*db7bccf4STejun Heo 				worker->flags &= ~WORKER_ROGUE;
20851da177e4SLinus Torvalds 		break;
20861da177e4SLinus Torvalds 	}
2087*db7bccf4STejun Heo 
2088*db7bccf4STejun Heo 	spin_unlock_irqrestore(&gcwq->lock, flags);
20891da177e4SLinus Torvalds 
20901537663fSTejun Heo 	return notifier_from_errno(0);
20911da177e4SLinus Torvalds }
20921da177e4SLinus Torvalds 
20932d3854a3SRusty Russell #ifdef CONFIG_SMP
20948ccad40dSRusty Russell 
20952d3854a3SRusty Russell struct work_for_cpu {
20966b44003eSAndrew Morton 	struct completion completion;
20972d3854a3SRusty Russell 	long (*fn)(void *);
20982d3854a3SRusty Russell 	void *arg;
20992d3854a3SRusty Russell 	long ret;
21002d3854a3SRusty Russell };
21012d3854a3SRusty Russell 
21026b44003eSAndrew Morton static int do_work_for_cpu(void *_wfc)
21032d3854a3SRusty Russell {
21046b44003eSAndrew Morton 	struct work_for_cpu *wfc = _wfc;
21052d3854a3SRusty Russell 	wfc->ret = wfc->fn(wfc->arg);
21066b44003eSAndrew Morton 	complete(&wfc->completion);
21076b44003eSAndrew Morton 	return 0;
21082d3854a3SRusty Russell }
21092d3854a3SRusty Russell 
21102d3854a3SRusty Russell /**
21112d3854a3SRusty Russell  * work_on_cpu - run a function in user context on a particular cpu
21122d3854a3SRusty Russell  * @cpu: the cpu to run on
21132d3854a3SRusty Russell  * @fn: the function to run
21142d3854a3SRusty Russell  * @arg: the function arg
21152d3854a3SRusty Russell  *
211631ad9081SRusty Russell  * This will return the value @fn returns.
211731ad9081SRusty Russell  * It is up to the caller to ensure that the cpu doesn't go offline.
21186b44003eSAndrew Morton  * The caller must not hold any locks which would prevent @fn from completing.
21192d3854a3SRusty Russell  */
21202d3854a3SRusty Russell long work_on_cpu(unsigned int cpu, long (*fn)(void *), void *arg)
21212d3854a3SRusty Russell {
21226b44003eSAndrew Morton 	struct task_struct *sub_thread;
21236b44003eSAndrew Morton 	struct work_for_cpu wfc = {
21246b44003eSAndrew Morton 		.completion = COMPLETION_INITIALIZER_ONSTACK(wfc.completion),
21256b44003eSAndrew Morton 		.fn = fn,
21266b44003eSAndrew Morton 		.arg = arg,
21276b44003eSAndrew Morton 	};
21282d3854a3SRusty Russell 
21296b44003eSAndrew Morton 	sub_thread = kthread_create(do_work_for_cpu, &wfc, "work_for_cpu");
21306b44003eSAndrew Morton 	if (IS_ERR(sub_thread))
21316b44003eSAndrew Morton 		return PTR_ERR(sub_thread);
21326b44003eSAndrew Morton 	kthread_bind(sub_thread, cpu);
21336b44003eSAndrew Morton 	wake_up_process(sub_thread);
21346b44003eSAndrew Morton 	wait_for_completion(&wfc.completion);
21352d3854a3SRusty Russell 	return wfc.ret;
21362d3854a3SRusty Russell }
21372d3854a3SRusty Russell EXPORT_SYMBOL_GPL(work_on_cpu);
21382d3854a3SRusty Russell #endif /* CONFIG_SMP */
21392d3854a3SRusty Russell 
2140a0a1a5fdSTejun Heo #ifdef CONFIG_FREEZER
2141a0a1a5fdSTejun Heo 
2142a0a1a5fdSTejun Heo /**
2143a0a1a5fdSTejun Heo  * freeze_workqueues_begin - begin freezing workqueues
2144a0a1a5fdSTejun Heo  *
2145a0a1a5fdSTejun Heo  * Start freezing workqueues.  After this function returns, all
2146a0a1a5fdSTejun Heo  * freezeable workqueues will queue new works to their frozen_works
2147a0a1a5fdSTejun Heo  * list instead of the cwq ones.
2148a0a1a5fdSTejun Heo  *
2149a0a1a5fdSTejun Heo  * CONTEXT:
21508b03ae3cSTejun Heo  * Grabs and releases workqueue_lock and gcwq->lock's.
2151a0a1a5fdSTejun Heo  */
2152a0a1a5fdSTejun Heo void freeze_workqueues_begin(void)
2153a0a1a5fdSTejun Heo {
2154a0a1a5fdSTejun Heo 	struct workqueue_struct *wq;
2155a0a1a5fdSTejun Heo 	unsigned int cpu;
2156a0a1a5fdSTejun Heo 
2157a0a1a5fdSTejun Heo 	spin_lock(&workqueue_lock);
2158a0a1a5fdSTejun Heo 
2159a0a1a5fdSTejun Heo 	BUG_ON(workqueue_freezing);
2160a0a1a5fdSTejun Heo 	workqueue_freezing = true;
2161a0a1a5fdSTejun Heo 
2162a0a1a5fdSTejun Heo 	for_each_possible_cpu(cpu) {
21638b03ae3cSTejun Heo 		struct global_cwq *gcwq = get_gcwq(cpu);
21648b03ae3cSTejun Heo 
21658b03ae3cSTejun Heo 		spin_lock_irq(&gcwq->lock);
21668b03ae3cSTejun Heo 
2167*db7bccf4STejun Heo 		BUG_ON(gcwq->flags & GCWQ_FREEZING);
2168*db7bccf4STejun Heo 		gcwq->flags |= GCWQ_FREEZING;
2169*db7bccf4STejun Heo 
2170a0a1a5fdSTejun Heo 		list_for_each_entry(wq, &workqueues, list) {
2171a0a1a5fdSTejun Heo 			struct cpu_workqueue_struct *cwq = get_cwq(cpu, wq);
2172a0a1a5fdSTejun Heo 
2173a0a1a5fdSTejun Heo 			if (wq->flags & WQ_FREEZEABLE)
2174a0a1a5fdSTejun Heo 				cwq->max_active = 0;
2175a0a1a5fdSTejun Heo 		}
21768b03ae3cSTejun Heo 
21778b03ae3cSTejun Heo 		spin_unlock_irq(&gcwq->lock);
2178a0a1a5fdSTejun Heo 	}
2179a0a1a5fdSTejun Heo 
2180a0a1a5fdSTejun Heo 	spin_unlock(&workqueue_lock);
2181a0a1a5fdSTejun Heo }
2182a0a1a5fdSTejun Heo 
2183a0a1a5fdSTejun Heo /**
2184a0a1a5fdSTejun Heo  * freeze_workqueues_busy - are freezeable workqueues still busy?
2185a0a1a5fdSTejun Heo  *
2186a0a1a5fdSTejun Heo  * Check whether freezing is complete.  This function must be called
2187a0a1a5fdSTejun Heo  * between freeze_workqueues_begin() and thaw_workqueues().
2188a0a1a5fdSTejun Heo  *
2189a0a1a5fdSTejun Heo  * CONTEXT:
2190a0a1a5fdSTejun Heo  * Grabs and releases workqueue_lock.
2191a0a1a5fdSTejun Heo  *
2192a0a1a5fdSTejun Heo  * RETURNS:
2193a0a1a5fdSTejun Heo  * %true if some freezeable workqueues are still busy.  %false if
2194a0a1a5fdSTejun Heo  * freezing is complete.
2195a0a1a5fdSTejun Heo  */
2196a0a1a5fdSTejun Heo bool freeze_workqueues_busy(void)
2197a0a1a5fdSTejun Heo {
2198a0a1a5fdSTejun Heo 	struct workqueue_struct *wq;
2199a0a1a5fdSTejun Heo 	unsigned int cpu;
2200a0a1a5fdSTejun Heo 	bool busy = false;
2201a0a1a5fdSTejun Heo 
2202a0a1a5fdSTejun Heo 	spin_lock(&workqueue_lock);
2203a0a1a5fdSTejun Heo 
2204a0a1a5fdSTejun Heo 	BUG_ON(!workqueue_freezing);
2205a0a1a5fdSTejun Heo 
2206a0a1a5fdSTejun Heo 	for_each_possible_cpu(cpu) {
2207a0a1a5fdSTejun Heo 		/*
2208a0a1a5fdSTejun Heo 		 * nr_active is monotonically decreasing.  It's safe
2209a0a1a5fdSTejun Heo 		 * to peek without lock.
2210a0a1a5fdSTejun Heo 		 */
2211a0a1a5fdSTejun Heo 		list_for_each_entry(wq, &workqueues, list) {
2212a0a1a5fdSTejun Heo 			struct cpu_workqueue_struct *cwq = get_cwq(cpu, wq);
2213a0a1a5fdSTejun Heo 
2214a0a1a5fdSTejun Heo 			if (!(wq->flags & WQ_FREEZEABLE))
2215a0a1a5fdSTejun Heo 				continue;
2216a0a1a5fdSTejun Heo 
2217a0a1a5fdSTejun Heo 			BUG_ON(cwq->nr_active < 0);
2218a0a1a5fdSTejun Heo 			if (cwq->nr_active) {
2219a0a1a5fdSTejun Heo 				busy = true;
2220a0a1a5fdSTejun Heo 				goto out_unlock;
2221a0a1a5fdSTejun Heo 			}
2222a0a1a5fdSTejun Heo 		}
2223a0a1a5fdSTejun Heo 	}
2224a0a1a5fdSTejun Heo out_unlock:
2225a0a1a5fdSTejun Heo 	spin_unlock(&workqueue_lock);
2226a0a1a5fdSTejun Heo 	return busy;
2227a0a1a5fdSTejun Heo }
2228a0a1a5fdSTejun Heo 
2229a0a1a5fdSTejun Heo /**
2230a0a1a5fdSTejun Heo  * thaw_workqueues - thaw workqueues
2231a0a1a5fdSTejun Heo  *
2232a0a1a5fdSTejun Heo  * Thaw workqueues.  Normal queueing is restored and all collected
2233a0a1a5fdSTejun Heo  * frozen works are transferred to their respective cwq worklists.
2234a0a1a5fdSTejun Heo  *
2235a0a1a5fdSTejun Heo  * CONTEXT:
22368b03ae3cSTejun Heo  * Grabs and releases workqueue_lock and gcwq->lock's.
2237a0a1a5fdSTejun Heo  */
2238a0a1a5fdSTejun Heo void thaw_workqueues(void)
2239a0a1a5fdSTejun Heo {
2240a0a1a5fdSTejun Heo 	struct workqueue_struct *wq;
2241a0a1a5fdSTejun Heo 	unsigned int cpu;
2242a0a1a5fdSTejun Heo 
2243a0a1a5fdSTejun Heo 	spin_lock(&workqueue_lock);
2244a0a1a5fdSTejun Heo 
2245a0a1a5fdSTejun Heo 	if (!workqueue_freezing)
2246a0a1a5fdSTejun Heo 		goto out_unlock;
2247a0a1a5fdSTejun Heo 
2248a0a1a5fdSTejun Heo 	for_each_possible_cpu(cpu) {
22498b03ae3cSTejun Heo 		struct global_cwq *gcwq = get_gcwq(cpu);
22508b03ae3cSTejun Heo 
22518b03ae3cSTejun Heo 		spin_lock_irq(&gcwq->lock);
22528b03ae3cSTejun Heo 
2253*db7bccf4STejun Heo 		BUG_ON(!(gcwq->flags & GCWQ_FREEZING));
2254*db7bccf4STejun Heo 		gcwq->flags &= ~GCWQ_FREEZING;
2255*db7bccf4STejun Heo 
2256a0a1a5fdSTejun Heo 		list_for_each_entry(wq, &workqueues, list) {
2257a0a1a5fdSTejun Heo 			struct cpu_workqueue_struct *cwq = get_cwq(cpu, wq);
2258a0a1a5fdSTejun Heo 
2259a0a1a5fdSTejun Heo 			if (!(wq->flags & WQ_FREEZEABLE))
2260a0a1a5fdSTejun Heo 				continue;
2261a0a1a5fdSTejun Heo 
2262a0a1a5fdSTejun Heo 			/* restore max_active and repopulate worklist */
2263a0a1a5fdSTejun Heo 			cwq->max_active = wq->saved_max_active;
2264a0a1a5fdSTejun Heo 
2265a0a1a5fdSTejun Heo 			while (!list_empty(&cwq->delayed_works) &&
2266a0a1a5fdSTejun Heo 			       cwq->nr_active < cwq->max_active)
2267a0a1a5fdSTejun Heo 				cwq_activate_first_delayed(cwq);
2268a0a1a5fdSTejun Heo 
2269c8e55f36STejun Heo 			wake_up_process(cwq->worker->task);
2270a0a1a5fdSTejun Heo 		}
22718b03ae3cSTejun Heo 
22728b03ae3cSTejun Heo 		spin_unlock_irq(&gcwq->lock);
2273a0a1a5fdSTejun Heo 	}
2274a0a1a5fdSTejun Heo 
2275a0a1a5fdSTejun Heo 	workqueue_freezing = false;
2276a0a1a5fdSTejun Heo out_unlock:
2277a0a1a5fdSTejun Heo 	spin_unlock(&workqueue_lock);
2278a0a1a5fdSTejun Heo }
2279a0a1a5fdSTejun Heo #endif /* CONFIG_FREEZER */
2280a0a1a5fdSTejun Heo 
2281c12920d1SOleg Nesterov void __init init_workqueues(void)
22821da177e4SLinus Torvalds {
2283c34056a3STejun Heo 	unsigned int cpu;
2284c8e55f36STejun Heo 	int i;
2285c34056a3STejun Heo 
2286e7577c50SRusty Russell 	singlethread_cpu = cpumask_first(cpu_possible_mask);
2287*db7bccf4STejun Heo 	hotcpu_notifier(workqueue_cpu_callback, CPU_PRI_WORKQUEUE);
22888b03ae3cSTejun Heo 
22898b03ae3cSTejun Heo 	/* initialize gcwqs */
22908b03ae3cSTejun Heo 	for_each_possible_cpu(cpu) {
22918b03ae3cSTejun Heo 		struct global_cwq *gcwq = get_gcwq(cpu);
22928b03ae3cSTejun Heo 
22938b03ae3cSTejun Heo 		spin_lock_init(&gcwq->lock);
22948b03ae3cSTejun Heo 		gcwq->cpu = cpu;
22958b03ae3cSTejun Heo 
2296c8e55f36STejun Heo 		INIT_LIST_HEAD(&gcwq->idle_list);
2297c8e55f36STejun Heo 		for (i = 0; i < BUSY_WORKER_HASH_SIZE; i++)
2298c8e55f36STejun Heo 			INIT_HLIST_HEAD(&gcwq->busy_hash[i]);
2299c8e55f36STejun Heo 
23008b03ae3cSTejun Heo 		ida_init(&gcwq->worker_ida);
2301*db7bccf4STejun Heo 
2302*db7bccf4STejun Heo 		gcwq->trustee_state = TRUSTEE_DONE;
2303*db7bccf4STejun Heo 		init_waitqueue_head(&gcwq->trustee_wait);
23048b03ae3cSTejun Heo 	}
23058b03ae3cSTejun Heo 
23061da177e4SLinus Torvalds 	keventd_wq = create_workqueue("events");
23071da177e4SLinus Torvalds 	BUG_ON(!keventd_wq);
23081da177e4SLinus Torvalds }
2309