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