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 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
static inline void set_wq_data(struct work_struct *work, void *wq)
{
	unsigned long new, old, res;

	/* assume the pending flag is already set and that the task has already
	 * been queued on this workqueue */
	new = (unsigned long) wq | (1UL << WORK_STRUCT_PENDING);
	res = work->management;
	if (res != new) {
		do {
			old = res;
			new = (unsigned long) wq;
			new |= (old & WORK_STRUCT_FLAG_MASK);
			res = cmpxchg(&work->management, old, new);
		} while (res != old);
	}
}

static inline void *get_wq_data(struct work_struct *work)
{
	return (void *) (work->management & WORK_STRUCT_WQ_DATA_MASK);
}

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 136 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 181 182 183
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);

		if (!test_bit(WORK_STRUCT_NOAUTOREL, &work->management))
			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 已提交
184 185 186 187 188 189 190
/* 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);
191
	set_wq_data(work, cwq);
L
Linus Torvalds 已提交
192 193 194 195 196 197
	list_add_tail(&work->entry, &cwq->worklist);
	cwq->insert_sequence++;
	wake_up(&cwq->more_work);
	spin_unlock_irqrestore(&cwq->lock, flags);
}

198 199 200 201 202
/**
 * queue_work - queue work on a workqueue
 * @wq: workqueue to use
 * @work: work to queue
 *
A
Alan Stern 已提交
203
 * Returns 0 if @work was already on a queue, non-zero otherwise.
L
Linus Torvalds 已提交
204 205 206 207 208 209 210 211
 *
 * 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();

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

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

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

233
	__queue_work(per_cpu_ptr(wq->cpu_wq, cpu), &dwork->work);
L
Linus Torvalds 已提交
234 235
}

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

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

254
	if (!test_and_set_bit(WORK_STRUCT_PENDING, &work->management)) {
L
Linus Torvalds 已提交
255 256 257 258
		BUG_ON(timer_pending(timer));
		BUG_ON(!list_empty(&work->entry));

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

270 271 272 273 274 275 276
/**
 * 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 已提交
277
 * Returns 0 if @work was already on a queue, non-zero otherwise.
278
 */
279
int queue_delayed_work_on(int cpu, struct workqueue_struct *wq,
280
			struct delayed_work *dwork, unsigned long delay)
281 282
{
	int ret = 0;
283 284
	struct timer_list *timer = &dwork->timer;
	struct work_struct *work = &dwork->work;
285

286
	if (!test_and_set_bit(WORK_STRUCT_PENDING, &work->management)) {
287 288 289 290
		BUG_ON(timer_pending(timer));
		BUG_ON(!list_empty(&work->entry));

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

302
static void run_workqueue(struct cpu_workqueue_struct *cwq)
L
Linus Torvalds 已提交
303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320
{
	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);
321
		work_func_t f = work->func;
L
Linus Torvalds 已提交
322 323 324 325

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

326
		BUG_ON(get_wq_data(work) != cwq);
327 328 329
		if (!test_bit(WORK_STRUCT_NOAUTOREL, &work->management))
			work_release(work);
		f(work);
L
Linus Torvalds 已提交
330

331 332 333 334 335 336 337 338 339 340 341
		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 已提交
342 343 344 345 346 347 348 349 350 351 352 353 354 355 356
		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;

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

	set_user_nice(current, -5);

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

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

L
Linus Torvalds 已提交
373 374 375 376 377 378 379 380
	/* 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()) {
381 382 383
		if (cwq->freezeable)
			try_to_freeze();

L
Linus Torvalds 已提交
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 423 424 425
		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);
	}
}

426
/**
L
Linus Torvalds 已提交
427
 * flush_workqueue - ensure that any scheduled work has run to completion.
428
 * @wq: workqueue to flush
L
Linus Torvalds 已提交
429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445
 *
 * 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)) {
446
		/* Always use first cpu's area. */
447
		flush_cpu_workqueue(per_cpu_ptr(wq->cpu_wq, singlethread_cpu));
L
Linus Torvalds 已提交
448 449 450
	} else {
		int cpu;

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

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

	spin_lock_init(&cwq->lock);
	cwq->wq = wq;
	cwq->thread = NULL;
	cwq->insert_sequence = 0;
	cwq->remove_sequence = 0;
470
	cwq->freezeable = freezeable;
L
Linus Torvalds 已提交
471 472 473 474 475 476 477 478 479 480 481 482 483 484 485
	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,
486
					    int singlethread, int freezeable)
L
Linus Torvalds 已提交
487 488 489 490 491
{
	int cpu, destroy = 0;
	struct workqueue_struct *wq;
	struct task_struct *p;

492
	wq = kzalloc(sizeof(*wq), GFP_KERNEL);
L
Linus Torvalds 已提交
493 494 495
	if (!wq)
		return NULL;

496
	wq->cpu_wq = alloc_percpu(struct cpu_workqueue_struct);
497 498 499 500 501
	if (!wq->cpu_wq) {
		kfree(wq);
		return NULL;
	}

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

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

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

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

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

	flush_workqueue(wq);

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

static struct workqueue_struct *keventd_wq;

579 580 581 582 583 584
/**
 * 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 已提交
585 586 587 588
int fastcall schedule_work(struct work_struct *work)
{
	return queue_work(keventd_wq, work);
}
589
EXPORT_SYMBOL(schedule_work);
L
Linus Torvalds 已提交
590

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

605 606 607
/**
 * schedule_delayed_work_on - queue work in global workqueue on CPU after delay
 * @cpu: cpu to use
608
 * @dwork: job to be done
609 610 611 612 613
 * @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 已提交
614
int schedule_delayed_work_on(int cpu,
615
			struct delayed_work *dwork, unsigned long delay)
L
Linus Torvalds 已提交
616
{
617
	return queue_delayed_work_on(cpu, keventd_wq, dwork, delay);
L
Linus Torvalds 已提交
618
}
619
EXPORT_SYMBOL(schedule_delayed_work_on);
L
Linus Torvalds 已提交
620

621 622 623 624 625 626 627 628 629 630 631
/**
 * 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.
 */
632
int schedule_on_each_cpu(work_func_t func)
633 634
{
	int cpu;
635
	struct work_struct *works;
636

637 638
	works = alloc_percpu(struct work_struct);
	if (!works)
639
		return -ENOMEM;
640

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

L
Linus Torvalds 已提交
653 654 655 656
void flush_scheduled_work(void)
{
	flush_workqueue(keventd_wq);
}
657
EXPORT_SYMBOL(flush_scheduled_work);
L
Linus Torvalds 已提交
658 659 660 661 662

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

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

684 685 686 687 688 689 690 691 692 693 694 695
/**
 * 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
 */
696
int execute_in_process_context(work_func_t fn, struct execute_work *ew)
697 698
{
	if (!in_interrupt()) {
699
		fn(&ew->work);
700 701 702
		return 0;
	}

703
	INIT_WORK(&ew->work, fn);
704 705 706 707 708 709
	schedule_work(&ew->work);

	return 1;
}
EXPORT_SYMBOL_GPL(execute_in_process_context);

L
Linus Torvalds 已提交
710 711 712 713 714 715 716 717 718 719 720 721 722
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);

723
	cwq = per_cpu_ptr(keventd_wq->cpu_wq, cpu);
L
Linus Torvalds 已提交
724 725 726 727 728 729 730 731 732 733
	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)
{
734
	struct cpu_workqueue_struct *cwq = per_cpu_ptr(wq->cpu_wq, cpu);
735
	struct list_head list;
L
Linus Torvalds 已提交
736 737 738
	struct work_struct *work;

	spin_lock_irq(&cwq->lock);
739
	list_replace_init(&cwq->worklist, &list);
L
Linus Torvalds 已提交
740 741 742 743 744

	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);
745
		__queue_work(per_cpu_ptr(wq->cpu_wq, smp_processor_id()), work);
L
Linus Torvalds 已提交
746 747 748 749 750
	}
	spin_unlock_irq(&cwq->lock);
}

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

	switch (action) {
	case CPU_UP_PREPARE:
760
		mutex_lock(&workqueue_mutex);
L
Linus Torvalds 已提交
761 762
		/* Create a new workqueue thread for it. */
		list_for_each_entry(wq, &workqueues, list) {
763
			if (!create_workqueue_thread(wq, hotcpu, 0)) {
L
Linus Torvalds 已提交
764 765 766 767 768 769 770 771 772
				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) {
773 774 775 776 777
			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 已提交
778
		}
779
		mutex_unlock(&workqueue_mutex);
L
Linus Torvalds 已提交
780 781 782 783
		break;

	case CPU_UP_CANCELED:
		list_for_each_entry(wq, &workqueues, list) {
784 785
			if (!per_cpu_ptr(wq->cpu_wq, hotcpu)->thread)
				continue;
L
Linus Torvalds 已提交
786
			/* Unbind so it can run. */
787
			kthread_bind(per_cpu_ptr(wq->cpu_wq, hotcpu)->thread,
788
				     any_online_cpu(cpu_online_map));
L
Linus Torvalds 已提交
789 790
			cleanup_workqueue_thread(wq, hotcpu);
		}
791 792 793 794 795 796 797 798 799
		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 已提交
800 801 802 803 804 805 806
		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);
807
		mutex_unlock(&workqueue_mutex);
L
Linus Torvalds 已提交
808 809 810 811 812 813 814 815
		break;
	}

	return NOTIFY_OK;
}

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