workqueue.c 20.8 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 * linux/kernel/workqueue.c
 *
 * Generic mechanism for defining kernel helper threads for running
 * arbitrary tasks in process context.
 *
 * Started by Ingo Molnar, Copyright (C) 2002
 *
 * Derived from the taskqueue/keventd code by:
 *
 *   David Woodhouse <dwmw2@infradead.org>
 *   Andrew Morton <andrewm@uow.edu.au>
 *   Kai Petzke <wpp@marie.physik.tu-berlin.de>
 *   Theodore Ts'o <tytso@mit.edu>
15 16
 *
 * Made to use alloc_percpu by Christoph Lameter <clameter@sgi.com>.
L
Linus Torvalds 已提交
17 18 19 20 21 22 23 24 25 26 27 28 29
 */

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/sched.h>
#include <linux/init.h>
#include <linux/signal.h>
#include <linux/completion.h>
#include <linux/workqueue.h>
#include <linux/slab.h>
#include <linux/cpu.h>
#include <linux/notifier.h>
#include <linux/kthread.h>
30
#include <linux/hardirq.h>
31
#include <linux/mempolicy.h>
32
#include <linux/freezer.h>
33 34
#include <linux/kallsyms.h>
#include <linux/debug_locks.h>
L
Linus Torvalds 已提交
35 36

/*
37 38
 * The per-CPU workqueue (if single thread, we always use the first
 * possible cpu).
L
Linus Torvalds 已提交
39 40
 *
 * The sequence counters are for flush_scheduled_work().  It wants to wait
41
 * until all currently-scheduled works are completed, but it doesn't
L
Linus Torvalds 已提交
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
 * want to be livelocked by new, incoming ones.  So it waits until
 * remove_sequence is >= the insert_sequence which pertained when
 * flush_scheduled_work() was called.
 */
struct cpu_workqueue_struct {

	spinlock_t lock;

	long remove_sequence;	/* Least-recently added (next to run) */
	long insert_sequence;	/* Next to add */

	struct list_head worklist;
	wait_queue_head_t more_work;
	wait_queue_head_t work_done;

	struct workqueue_struct *wq;
58
	struct task_struct *thread;
L
Linus Torvalds 已提交
59 60

	int run_depth;		/* Detect run_workqueue() recursion depth */
61 62

	int freezeable;		/* Freeze the thread during suspend */
L
Linus Torvalds 已提交
63 64 65 66 67 68 69
} ____cacheline_aligned;

/*
 * The externally visible workqueue abstraction is an array of
 * per-CPU workqueues:
 */
struct workqueue_struct {
70
	struct cpu_workqueue_struct *cpu_wq;
L
Linus Torvalds 已提交
71 72 73 74 75 76
	const char *name;
	struct list_head list; 	/* Empty if single thread */
};

/* All the per-cpu workqueues on the system, for hotplug cpu to add/remove
   threads to each one as cpus come/go. */
77
static DEFINE_MUTEX(workqueue_mutex);
L
Linus Torvalds 已提交
78 79
static LIST_HEAD(workqueues);

80 81
static int singlethread_cpu;

L
Linus Torvalds 已提交
82 83 84 85 86 87
/* If it's single threaded, it isn't in the list of workqueues. */
static inline int is_single_threaded(struct workqueue_struct *wq)
{
	return list_empty(&wq->list);
}

88 89 90 91
/*
 * Set the workqueue on which a work item is to be run
 * - Must *only* be called if the pending flag is set
 */
92 93
static inline void set_wq_data(struct work_struct *work, void *wq)
{
94 95 96
	unsigned long new;

	BUG_ON(!work_pending(work));
97 98

	new = (unsigned long) wq | (1UL << WORK_STRUCT_PENDING);
99 100
	new |= WORK_STRUCT_FLAG_MASK & *work_data_bits(work);
	atomic_long_set(&work->data, new);
101 102 103 104
}

static inline void *get_wq_data(struct work_struct *work)
{
105
	return (void *) (atomic_long_read(&work->data) & WORK_STRUCT_WQ_DATA_MASK);
106 107
}

108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135
static int __run_work(struct cpu_workqueue_struct *cwq, struct work_struct *work)
{
	int ret = 0;
	unsigned long flags;

	spin_lock_irqsave(&cwq->lock, flags);
	/*
	 * We need to re-validate the work info after we've gotten
	 * the cpu_workqueue lock. We can run the work now iff:
	 *
	 *  - the wq_data still matches the cpu_workqueue_struct
	 *  - AND the work is still marked pending
	 *  - AND the work is still on a list (which will be this
	 *    workqueue_struct list)
	 *
	 * All these conditions are important, because we
	 * need to protect against the work being run right
	 * now on another CPU (all but the last one might be
	 * true if it's currently running and has not been
	 * released yet, for example).
	 */
	if (get_wq_data(work) == cwq
	    && work_pending(work)
	    && !list_empty(&work->entry)) {
		work_func_t f = work->func;
		list_del_init(&work->entry);
		spin_unlock_irqrestore(&cwq->lock, flags);

136
		if (!test_bit(WORK_STRUCT_NOAUTOREL, work_data_bits(work)))
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180
			work_release(work);
		f(work);

		spin_lock_irqsave(&cwq->lock, flags);
		cwq->remove_sequence++;
		wake_up(&cwq->work_done);
		ret = 1;
	}
	spin_unlock_irqrestore(&cwq->lock, flags);
	return ret;
}

/**
 * run_scheduled_work - run scheduled work synchronously
 * @work: work to run
 *
 * This checks if the work was pending, and runs it
 * synchronously if so. It returns a boolean to indicate
 * whether it had any scheduled work to run or not.
 *
 * NOTE! This _only_ works for normal work_structs. You
 * CANNOT use this for delayed work, because the wq data
 * for delayed work will not point properly to the per-
 * CPU workqueue struct, but will change!
 */
int fastcall run_scheduled_work(struct work_struct *work)
{
	for (;;) {
		struct cpu_workqueue_struct *cwq;

		if (!work_pending(work))
			return 0;
		if (list_empty(&work->entry))
			return 0;
		/* NOTE! This depends intimately on __queue_work! */
		cwq = get_wq_data(work);
		if (!cwq)
			return 0;
		if (__run_work(cwq, work))
			return 1;
	}
}
EXPORT_SYMBOL(run_scheduled_work);

L
Linus Torvalds 已提交
181 182 183 184 185 186 187
/* Preempt must be disabled. */
static void __queue_work(struct cpu_workqueue_struct *cwq,
			 struct work_struct *work)
{
	unsigned long flags;

	spin_lock_irqsave(&cwq->lock, flags);
188
	set_wq_data(work, cwq);
L
Linus Torvalds 已提交
189 190 191 192 193 194
	list_add_tail(&work->entry, &cwq->worklist);
	cwq->insert_sequence++;
	wake_up(&cwq->more_work);
	spin_unlock_irqrestore(&cwq->lock, flags);
}

195 196 197 198 199
/**
 * queue_work - queue work on a workqueue
 * @wq: workqueue to use
 * @work: work to queue
 *
A
Alan Stern 已提交
200
 * Returns 0 if @work was already on a queue, non-zero otherwise.
L
Linus Torvalds 已提交
201 202 203 204 205 206 207 208
 *
 * We queue the work to the CPU it was submitted, but there is no
 * guarantee that it will be processed by that CPU.
 */
int fastcall queue_work(struct workqueue_struct *wq, struct work_struct *work)
{
	int ret = 0, cpu = get_cpu();

209
	if (!test_and_set_bit(WORK_STRUCT_PENDING, work_data_bits(work))) {
L
Linus Torvalds 已提交
210
		if (unlikely(is_single_threaded(wq)))
211
			cpu = singlethread_cpu;
L
Linus Torvalds 已提交
212
		BUG_ON(!list_empty(&work->entry));
213
		__queue_work(per_cpu_ptr(wq->cpu_wq, cpu), work);
L
Linus Torvalds 已提交
214 215 216 217 218
		ret = 1;
	}
	put_cpu();
	return ret;
}
219
EXPORT_SYMBOL_GPL(queue_work);
L
Linus Torvalds 已提交
220 221 222

static void delayed_work_timer_fn(unsigned long __data)
{
223
	struct delayed_work *dwork = (struct delayed_work *)__data;
224
	struct workqueue_struct *wq = get_wq_data(&dwork->work);
L
Linus Torvalds 已提交
225 226 227
	int cpu = smp_processor_id();

	if (unlikely(is_single_threaded(wq)))
228
		cpu = singlethread_cpu;
L
Linus Torvalds 已提交
229

230
	__queue_work(per_cpu_ptr(wq->cpu_wq, cpu), &dwork->work);
L
Linus Torvalds 已提交
231 232
}

233 234 235
/**
 * queue_delayed_work - queue work on a workqueue after delay
 * @wq: workqueue to use
236
 * @work: delayable work to queue
237 238
 * @delay: number of jiffies to wait before queueing
 *
A
Alan Stern 已提交
239
 * Returns 0 if @work was already on a queue, non-zero otherwise.
240
 */
L
Linus Torvalds 已提交
241
int fastcall queue_delayed_work(struct workqueue_struct *wq,
242
			struct delayed_work *dwork, unsigned long delay)
L
Linus Torvalds 已提交
243 244
{
	int ret = 0;
245 246 247 248 249
	struct timer_list *timer = &dwork->timer;
	struct work_struct *work = &dwork->work;

	if (delay == 0)
		return queue_work(wq, work);
L
Linus Torvalds 已提交
250

251
	if (!test_and_set_bit(WORK_STRUCT_PENDING, work_data_bits(work))) {
L
Linus Torvalds 已提交
252 253 254 255
		BUG_ON(timer_pending(timer));
		BUG_ON(!list_empty(&work->entry));

		/* This stores wq for the moment, for the timer_fn */
256
		set_wq_data(work, wq);
L
Linus Torvalds 已提交
257
		timer->expires = jiffies + delay;
258
		timer->data = (unsigned long)dwork;
L
Linus Torvalds 已提交
259 260 261 262 263 264
		timer->function = delayed_work_timer_fn;
		add_timer(timer);
		ret = 1;
	}
	return ret;
}
265
EXPORT_SYMBOL_GPL(queue_delayed_work);
L
Linus Torvalds 已提交
266

267 268 269 270 271 272 273
/**
 * queue_delayed_work_on - queue work on specific CPU after delay
 * @cpu: CPU number to execute work on
 * @wq: workqueue to use
 * @work: work to queue
 * @delay: number of jiffies to wait before queueing
 *
A
Alan Stern 已提交
274
 * Returns 0 if @work was already on a queue, non-zero otherwise.
275
 */
276
int queue_delayed_work_on(int cpu, struct workqueue_struct *wq,
277
			struct delayed_work *dwork, unsigned long delay)
278 279
{
	int ret = 0;
280 281
	struct timer_list *timer = &dwork->timer;
	struct work_struct *work = &dwork->work;
282

283
	if (!test_and_set_bit(WORK_STRUCT_PENDING, work_data_bits(work))) {
284 285 286 287
		BUG_ON(timer_pending(timer));
		BUG_ON(!list_empty(&work->entry));

		/* This stores wq for the moment, for the timer_fn */
288
		set_wq_data(work, wq);
289
		timer->expires = jiffies + delay;
290
		timer->data = (unsigned long)dwork;
291 292 293 294 295 296
		timer->function = delayed_work_timer_fn;
		add_timer_on(timer, cpu);
		ret = 1;
	}
	return ret;
}
297
EXPORT_SYMBOL_GPL(queue_delayed_work_on);
L
Linus Torvalds 已提交
298

299
static void run_workqueue(struct cpu_workqueue_struct *cwq)
L
Linus Torvalds 已提交
300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317
{
	unsigned long flags;

	/*
	 * Keep taking off work from the queue until
	 * done.
	 */
	spin_lock_irqsave(&cwq->lock, flags);
	cwq->run_depth++;
	if (cwq->run_depth > 3) {
		/* morton gets to eat his hat */
		printk("%s: recursion depth exceeded: %d\n",
			__FUNCTION__, cwq->run_depth);
		dump_stack();
	}
	while (!list_empty(&cwq->worklist)) {
		struct work_struct *work = list_entry(cwq->worklist.next,
						struct work_struct, entry);
318
		work_func_t f = work->func;
L
Linus Torvalds 已提交
319 320 321 322

		list_del_init(cwq->worklist.next);
		spin_unlock_irqrestore(&cwq->lock, flags);

323
		BUG_ON(get_wq_data(work) != cwq);
324
		if (!test_bit(WORK_STRUCT_NOAUTOREL, work_data_bits(work)))
325 326
			work_release(work);
		f(work);
L
Linus Torvalds 已提交
327

328 329 330 331 332 333 334 335 336 337 338
		if (unlikely(in_atomic() || lockdep_depth(current) > 0)) {
			printk(KERN_ERR "BUG: workqueue leaked lock or atomic: "
					"%s/0x%08x/%d\n",
					current->comm, preempt_count(),
				       	current->pid);
			printk(KERN_ERR "    last function: ");
			print_symbol("%s\n", (unsigned long)f);
			debug_show_held_locks(current);
			dump_stack();
		}

L
Linus Torvalds 已提交
339 340 341 342 343 344 345 346 347 348 349 350 351 352 353
		spin_lock_irqsave(&cwq->lock, flags);
		cwq->remove_sequence++;
		wake_up(&cwq->work_done);
	}
	cwq->run_depth--;
	spin_unlock_irqrestore(&cwq->lock, flags);
}

static int worker_thread(void *__cwq)
{
	struct cpu_workqueue_struct *cwq = __cwq;
	DECLARE_WAITQUEUE(wait, current);
	struct k_sigaction sa;
	sigset_t blocked;

354 355
	if (!cwq->freezeable)
		current->flags |= PF_NOFREEZE;
L
Linus Torvalds 已提交
356 357 358 359 360 361 362 363

	set_user_nice(current, -5);

	/* Block and flush all signals */
	sigfillset(&blocked);
	sigprocmask(SIG_BLOCK, &blocked, NULL);
	flush_signals(current);

364 365 366 367 368 369
	/*
	 * We inherited MPOL_INTERLEAVE from the booting kernel.
	 * Set MPOL_DEFAULT to insure node local allocations.
	 */
	numa_default_policy();

L
Linus Torvalds 已提交
370 371 372 373 374 375 376 377
	/* SIG_IGN makes children autoreap: see do_notify_parent(). */
	sa.sa.sa_handler = SIG_IGN;
	sa.sa.sa_flags = 0;
	siginitset(&sa.sa.sa_mask, sigmask(SIGCHLD));
	do_sigaction(SIGCHLD, &sa, (struct k_sigaction *)0);

	set_current_state(TASK_INTERRUPTIBLE);
	while (!kthread_should_stop()) {
378 379 380
		if (cwq->freezeable)
			try_to_freeze();

L
Linus Torvalds 已提交
381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422
		add_wait_queue(&cwq->more_work, &wait);
		if (list_empty(&cwq->worklist))
			schedule();
		else
			__set_current_state(TASK_RUNNING);
		remove_wait_queue(&cwq->more_work, &wait);

		if (!list_empty(&cwq->worklist))
			run_workqueue(cwq);
		set_current_state(TASK_INTERRUPTIBLE);
	}
	__set_current_state(TASK_RUNNING);
	return 0;
}

static void flush_cpu_workqueue(struct cpu_workqueue_struct *cwq)
{
	if (cwq->thread == current) {
		/*
		 * Probably keventd trying to flush its own queue. So simply run
		 * it by hand rather than deadlocking.
		 */
		run_workqueue(cwq);
	} else {
		DEFINE_WAIT(wait);
		long sequence_needed;

		spin_lock_irq(&cwq->lock);
		sequence_needed = cwq->insert_sequence;

		while (sequence_needed - cwq->remove_sequence > 0) {
			prepare_to_wait(&cwq->work_done, &wait,
					TASK_UNINTERRUPTIBLE);
			spin_unlock_irq(&cwq->lock);
			schedule();
			spin_lock_irq(&cwq->lock);
		}
		finish_wait(&cwq->work_done, &wait);
		spin_unlock_irq(&cwq->lock);
	}
}

423
/**
L
Linus Torvalds 已提交
424
 * flush_workqueue - ensure that any scheduled work has run to completion.
425
 * @wq: workqueue to flush
L
Linus Torvalds 已提交
426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442
 *
 * Forces execution of the workqueue and blocks until its completion.
 * This is typically used in driver shutdown handlers.
 *
 * This function will sample each workqueue's current insert_sequence number and
 * will sleep until the head sequence is greater than or equal to that.  This
 * means that we sleep until all works which were queued on entry have been
 * handled, but we are not livelocked by new incoming ones.
 *
 * This function used to run the workqueues itself.  Now we just wait for the
 * helper threads to do it.
 */
void fastcall flush_workqueue(struct workqueue_struct *wq)
{
	might_sleep();

	if (is_single_threaded(wq)) {
443
		/* Always use first cpu's area. */
444
		flush_cpu_workqueue(per_cpu_ptr(wq->cpu_wq, singlethread_cpu));
L
Linus Torvalds 已提交
445 446 447
	} else {
		int cpu;

448
		mutex_lock(&workqueue_mutex);
L
Linus Torvalds 已提交
449
		for_each_online_cpu(cpu)
450
			flush_cpu_workqueue(per_cpu_ptr(wq->cpu_wq, cpu));
451
		mutex_unlock(&workqueue_mutex);
L
Linus Torvalds 已提交
452 453
	}
}
454
EXPORT_SYMBOL_GPL(flush_workqueue);
L
Linus Torvalds 已提交
455 456

static struct task_struct *create_workqueue_thread(struct workqueue_struct *wq,
457
						   int cpu, int freezeable)
L
Linus Torvalds 已提交
458
{
459
	struct cpu_workqueue_struct *cwq = per_cpu_ptr(wq->cpu_wq, cpu);
L
Linus Torvalds 已提交
460 461 462 463 464 465 466
	struct task_struct *p;

	spin_lock_init(&cwq->lock);
	cwq->wq = wq;
	cwq->thread = NULL;
	cwq->insert_sequence = 0;
	cwq->remove_sequence = 0;
467
	cwq->freezeable = freezeable;
L
Linus Torvalds 已提交
468 469 470 471 472 473 474 475 476 477 478 479 480 481 482
	INIT_LIST_HEAD(&cwq->worklist);
	init_waitqueue_head(&cwq->more_work);
	init_waitqueue_head(&cwq->work_done);

	if (is_single_threaded(wq))
		p = kthread_create(worker_thread, cwq, "%s", wq->name);
	else
		p = kthread_create(worker_thread, cwq, "%s/%d", wq->name, cpu);
	if (IS_ERR(p))
		return NULL;
	cwq->thread = p;
	return p;
}

struct workqueue_struct *__create_workqueue(const char *name,
483
					    int singlethread, int freezeable)
L
Linus Torvalds 已提交
484 485 486 487 488
{
	int cpu, destroy = 0;
	struct workqueue_struct *wq;
	struct task_struct *p;

489
	wq = kzalloc(sizeof(*wq), GFP_KERNEL);
L
Linus Torvalds 已提交
490 491 492
	if (!wq)
		return NULL;

493
	wq->cpu_wq = alloc_percpu(struct cpu_workqueue_struct);
494 495 496 497 498
	if (!wq->cpu_wq) {
		kfree(wq);
		return NULL;
	}

L
Linus Torvalds 已提交
499
	wq->name = name;
500
	mutex_lock(&workqueue_mutex);
L
Linus Torvalds 已提交
501 502
	if (singlethread) {
		INIT_LIST_HEAD(&wq->list);
503
		p = create_workqueue_thread(wq, singlethread_cpu, freezeable);
L
Linus Torvalds 已提交
504 505 506 507 508 509 510
		if (!p)
			destroy = 1;
		else
			wake_up_process(p);
	} else {
		list_add(&wq->list, &workqueues);
		for_each_online_cpu(cpu) {
511
			p = create_workqueue_thread(wq, cpu, freezeable);
L
Linus Torvalds 已提交
512 513 514 515 516 517 518
			if (p) {
				kthread_bind(p, cpu);
				wake_up_process(p);
			} else
				destroy = 1;
		}
	}
519
	mutex_unlock(&workqueue_mutex);
L
Linus Torvalds 已提交
520 521 522 523 524 525 526 527 528 529

	/*
	 * Was there any error during startup? If yes then clean up:
	 */
	if (destroy) {
		destroy_workqueue(wq);
		wq = NULL;
	}
	return wq;
}
530
EXPORT_SYMBOL_GPL(__create_workqueue);
L
Linus Torvalds 已提交
531 532 533 534 535 536 537

static void cleanup_workqueue_thread(struct workqueue_struct *wq, int cpu)
{
	struct cpu_workqueue_struct *cwq;
	unsigned long flags;
	struct task_struct *p;

538
	cwq = per_cpu_ptr(wq->cpu_wq, cpu);
L
Linus Torvalds 已提交
539 540 541 542 543 544 545 546
	spin_lock_irqsave(&cwq->lock, flags);
	p = cwq->thread;
	cwq->thread = NULL;
	spin_unlock_irqrestore(&cwq->lock, flags);
	if (p)
		kthread_stop(p);
}

547 548 549 550 551 552
/**
 * destroy_workqueue - safely terminate a workqueue
 * @wq: target workqueue
 *
 * Safely destroy a workqueue. All work currently pending will be done first.
 */
L
Linus Torvalds 已提交
553 554 555 556 557 558 559
void destroy_workqueue(struct workqueue_struct *wq)
{
	int cpu;

	flush_workqueue(wq);

	/* We don't need the distraction of CPUs appearing and vanishing. */
560
	mutex_lock(&workqueue_mutex);
L
Linus Torvalds 已提交
561
	if (is_single_threaded(wq))
562
		cleanup_workqueue_thread(wq, singlethread_cpu);
L
Linus Torvalds 已提交
563 564 565 566 567
	else {
		for_each_online_cpu(cpu)
			cleanup_workqueue_thread(wq, cpu);
		list_del(&wq->list);
	}
568
	mutex_unlock(&workqueue_mutex);
569
	free_percpu(wq->cpu_wq);
L
Linus Torvalds 已提交
570 571
	kfree(wq);
}
572
EXPORT_SYMBOL_GPL(destroy_workqueue);
L
Linus Torvalds 已提交
573 574 575

static struct workqueue_struct *keventd_wq;

576 577 578 579 580 581
/**
 * schedule_work - put work task in global workqueue
 * @work: job to be done
 *
 * This puts a job in the kernel-global workqueue.
 */
L
Linus Torvalds 已提交
582 583 584 585
int fastcall schedule_work(struct work_struct *work)
{
	return queue_work(keventd_wq, work);
}
586
EXPORT_SYMBOL(schedule_work);
L
Linus Torvalds 已提交
587

588 589
/**
 * schedule_delayed_work - put work task in global workqueue after delay
590 591
 * @dwork: job to be done
 * @delay: number of jiffies to wait or 0 for immediate execution
592 593 594 595
 *
 * After waiting for a given time this puts a job in the kernel-global
 * workqueue.
 */
596
int fastcall schedule_delayed_work(struct delayed_work *dwork, unsigned long delay)
L
Linus Torvalds 已提交
597
{
598
	return queue_delayed_work(keventd_wq, dwork, delay);
L
Linus Torvalds 已提交
599
}
600
EXPORT_SYMBOL(schedule_delayed_work);
L
Linus Torvalds 已提交
601

602 603 604
/**
 * schedule_delayed_work_on - queue work in global workqueue on CPU after delay
 * @cpu: cpu to use
605
 * @dwork: job to be done
606 607 608 609 610
 * @delay: number of jiffies to wait
 *
 * After waiting for a given time this puts a job in the kernel-global
 * workqueue on the specified CPU.
 */
L
Linus Torvalds 已提交
611
int schedule_delayed_work_on(int cpu,
612
			struct delayed_work *dwork, unsigned long delay)
L
Linus Torvalds 已提交
613
{
614
	return queue_delayed_work_on(cpu, keventd_wq, dwork, delay);
L
Linus Torvalds 已提交
615
}
616
EXPORT_SYMBOL(schedule_delayed_work_on);
L
Linus Torvalds 已提交
617

618 619 620 621 622 623 624 625 626 627 628
/**
 * schedule_on_each_cpu - call a function on each online CPU from keventd
 * @func: the function to call
 *
 * Returns zero on success.
 * Returns -ve errno on failure.
 *
 * Appears to be racy against CPU hotplug.
 *
 * schedule_on_each_cpu() is very slow.
 */
629
int schedule_on_each_cpu(work_func_t func)
630 631
{
	int cpu;
632
	struct work_struct *works;
633

634 635
	works = alloc_percpu(struct work_struct);
	if (!works)
636
		return -ENOMEM;
637

638
	mutex_lock(&workqueue_mutex);
639
	for_each_online_cpu(cpu) {
640
		INIT_WORK(per_cpu_ptr(works, cpu), func);
641
		__queue_work(per_cpu_ptr(keventd_wq->cpu_wq, cpu),
642
				per_cpu_ptr(works, cpu));
643
	}
644
	mutex_unlock(&workqueue_mutex);
645
	flush_workqueue(keventd_wq);
646
	free_percpu(works);
647 648 649
	return 0;
}

L
Linus Torvalds 已提交
650 651 652 653
void flush_scheduled_work(void)
{
	flush_workqueue(keventd_wq);
}
654
EXPORT_SYMBOL(flush_scheduled_work);
L
Linus Torvalds 已提交
655 656 657 658 659

/**
 * cancel_rearming_delayed_workqueue - reliably kill off a delayed
 *			work whose handler rearms the delayed work.
 * @wq:   the controlling workqueue structure
660
 * @dwork: the delayed work struct
L
Linus Torvalds 已提交
661
 */
662
void cancel_rearming_delayed_workqueue(struct workqueue_struct *wq,
663
				       struct delayed_work *dwork)
L
Linus Torvalds 已提交
664
{
665
	while (!cancel_delayed_work(dwork))
L
Linus Torvalds 已提交
666 667
		flush_workqueue(wq);
}
668
EXPORT_SYMBOL(cancel_rearming_delayed_workqueue);
L
Linus Torvalds 已提交
669 670 671 672

/**
 * cancel_rearming_delayed_work - reliably kill off a delayed keventd
 *			work whose handler rearms the delayed work.
673
 * @dwork: the delayed work struct
L
Linus Torvalds 已提交
674
 */
675
void cancel_rearming_delayed_work(struct delayed_work *dwork)
L
Linus Torvalds 已提交
676
{
677
	cancel_rearming_delayed_workqueue(keventd_wq, dwork);
L
Linus Torvalds 已提交
678 679 680
}
EXPORT_SYMBOL(cancel_rearming_delayed_work);

681 682 683 684 685 686 687 688 689 690 691 692
/**
 * execute_in_process_context - reliably execute the routine with user context
 * @fn:		the function to execute
 * @ew:		guaranteed storage for the execute work structure (must
 *		be available when the work executes)
 *
 * Executes the function immediately if process context is available,
 * otherwise schedules the function for delayed execution.
 *
 * Returns:	0 - function was executed
 *		1 - function was scheduled for execution
 */
693
int execute_in_process_context(work_func_t fn, struct execute_work *ew)
694 695
{
	if (!in_interrupt()) {
696
		fn(&ew->work);
697 698 699
		return 0;
	}

700
	INIT_WORK(&ew->work, fn);
701 702 703 704 705 706
	schedule_work(&ew->work);

	return 1;
}
EXPORT_SYMBOL_GPL(execute_in_process_context);

L
Linus Torvalds 已提交
707 708 709 710 711 712 713 714 715 716 717 718 719
int keventd_up(void)
{
	return keventd_wq != NULL;
}

int current_is_keventd(void)
{
	struct cpu_workqueue_struct *cwq;
	int cpu = smp_processor_id();	/* preempt-safe: keventd is per-cpu */
	int ret = 0;

	BUG_ON(!keventd_wq);

720
	cwq = per_cpu_ptr(keventd_wq->cpu_wq, cpu);
L
Linus Torvalds 已提交
721 722 723 724 725 726 727 728 729 730
	if (current == cwq->thread)
		ret = 1;

	return ret;

}

/* Take the work from this (downed) CPU. */
static void take_over_work(struct workqueue_struct *wq, unsigned int cpu)
{
731
	struct cpu_workqueue_struct *cwq = per_cpu_ptr(wq->cpu_wq, cpu);
732
	struct list_head list;
L
Linus Torvalds 已提交
733 734 735
	struct work_struct *work;

	spin_lock_irq(&cwq->lock);
736
	list_replace_init(&cwq->worklist, &list);
L
Linus Torvalds 已提交
737 738 739 740 741

	while (!list_empty(&list)) {
		printk("Taking work for %s\n", wq->name);
		work = list_entry(list.next,struct work_struct,entry);
		list_del(&work->entry);
742
		__queue_work(per_cpu_ptr(wq->cpu_wq, smp_processor_id()), work);
L
Linus Torvalds 已提交
743 744 745 746 747
	}
	spin_unlock_irq(&cwq->lock);
}

/* We're holding the cpucontrol mutex here */
748
static int __devinit workqueue_cpu_callback(struct notifier_block *nfb,
L
Linus Torvalds 已提交
749 750 751 752 753 754 755 756
				  unsigned long action,
				  void *hcpu)
{
	unsigned int hotcpu = (unsigned long)hcpu;
	struct workqueue_struct *wq;

	switch (action) {
	case CPU_UP_PREPARE:
757
		mutex_lock(&workqueue_mutex);
L
Linus Torvalds 已提交
758 759
		/* Create a new workqueue thread for it. */
		list_for_each_entry(wq, &workqueues, list) {
760
			if (!create_workqueue_thread(wq, hotcpu, 0)) {
L
Linus Torvalds 已提交
761 762 763 764 765 766 767 768 769
				printk("workqueue for %i failed\n", hotcpu);
				return NOTIFY_BAD;
			}
		}
		break;

	case CPU_ONLINE:
		/* Kick off worker threads. */
		list_for_each_entry(wq, &workqueues, list) {
770 771 772 773 774
			struct cpu_workqueue_struct *cwq;

			cwq = per_cpu_ptr(wq->cpu_wq, hotcpu);
			kthread_bind(cwq->thread, hotcpu);
			wake_up_process(cwq->thread);
L
Linus Torvalds 已提交
775
		}
776
		mutex_unlock(&workqueue_mutex);
L
Linus Torvalds 已提交
777 778 779 780
		break;

	case CPU_UP_CANCELED:
		list_for_each_entry(wq, &workqueues, list) {
781 782
			if (!per_cpu_ptr(wq->cpu_wq, hotcpu)->thread)
				continue;
L
Linus Torvalds 已提交
783
			/* Unbind so it can run. */
784
			kthread_bind(per_cpu_ptr(wq->cpu_wq, hotcpu)->thread,
785
				     any_online_cpu(cpu_online_map));
L
Linus Torvalds 已提交
786 787
			cleanup_workqueue_thread(wq, hotcpu);
		}
788 789 790 791 792 793 794 795 796
		mutex_unlock(&workqueue_mutex);
		break;

	case CPU_DOWN_PREPARE:
		mutex_lock(&workqueue_mutex);
		break;

	case CPU_DOWN_FAILED:
		mutex_unlock(&workqueue_mutex);
L
Linus Torvalds 已提交
797 798 799 800 801 802 803
		break;

	case CPU_DEAD:
		list_for_each_entry(wq, &workqueues, list)
			cleanup_workqueue_thread(wq, hotcpu);
		list_for_each_entry(wq, &workqueues, list)
			take_over_work(wq, hotcpu);
804
		mutex_unlock(&workqueue_mutex);
L
Linus Torvalds 已提交
805 806 807 808 809 810 811 812
		break;
	}

	return NOTIFY_OK;
}

void init_workqueues(void)
{
813
	singlethread_cpu = first_cpu(cpu_possible_map);
L
Linus Torvalds 已提交
814 815 816 817 818
	hotcpu_notifier(workqueue_cpu_callback, 0);
	keventd_wq = create_workqueue("events");
	BUG_ON(!keventd_wq);
}