workqueue.c 20.4 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 41 42 43 44 45
 */
struct cpu_workqueue_struct {

	spinlock_t lock;

	struct list_head worklist;
	wait_queue_head_t more_work;
46
	struct work_struct *current_work;
L
Linus Torvalds 已提交
47 48

	struct workqueue_struct *wq;
49
	struct task_struct *thread;
50
	int should_stop;
L
Linus Torvalds 已提交
51 52 53 54 55 56 57 58 59

	int run_depth;		/* Detect run_workqueue() recursion depth */
} ____cacheline_aligned;

/*
 * The externally visible workqueue abstraction is an array of
 * per-CPU workqueues:
 */
struct workqueue_struct {
60
	struct cpu_workqueue_struct *cpu_wq;
61
	struct list_head list;
L
Linus Torvalds 已提交
62
	const char *name;
63
	int singlethread;
64
	int freezeable;		/* Freeze threads during suspend */
L
Linus Torvalds 已提交
65 66 67 68
};

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

72
static int singlethread_cpu __read_mostly;
73
static cpumask_t cpu_singlethread_map __read_mostly;
74 75
/* optimization, we could use cpu_possible_map */
static cpumask_t cpu_populated_map __read_mostly;
76

L
Linus Torvalds 已提交
77 78 79
/* If it's single threaded, it isn't in the list of workqueues. */
static inline int is_single_threaded(struct workqueue_struct *wq)
{
80
	return wq->singlethread;
L
Linus Torvalds 已提交
81 82
}

83 84 85 86 87 88
static const cpumask_t *wq_cpu_map(struct workqueue_struct *wq)
{
	return is_single_threaded(wq)
		? &cpu_singlethread_map : &cpu_populated_map;
}

89 90 91 92 93 94 95 96
static
struct cpu_workqueue_struct *wq_per_cpu(struct workqueue_struct *wq, int cpu)
{
	if (unlikely(is_single_threaded(wq)))
		cpu = singlethread_cpu;
	return per_cpu_ptr(wq->cpu_wq, cpu);
}

97 98 99 100
/*
 * Set the workqueue on which a work item is to be run
 * - Must *only* be called if the pending flag is set
 */
101 102
static inline void set_wq_data(struct work_struct *work,
				struct cpu_workqueue_struct *cwq)
103
{
104 105 106
	unsigned long new;

	BUG_ON(!work_pending(work));
107

108
	new = (unsigned long) cwq | (1UL << WORK_STRUCT_PENDING);
109 110
	new |= WORK_STRUCT_FLAG_MASK & *work_data_bits(work);
	atomic_long_set(&work->data, new);
111 112
}

113 114
static inline
struct cpu_workqueue_struct *get_wq_data(struct work_struct *work)
115
{
116
	return (void *) (atomic_long_read(&work->data) & WORK_STRUCT_WQ_DATA_MASK);
117 118
}

O
Oleg Nesterov 已提交
119 120 121 122
static void insert_work(struct cpu_workqueue_struct *cwq,
				struct work_struct *work, int tail)
{
	set_wq_data(work, cwq);
123 124 125 126 127
	/*
	 * Ensure that we get the right work->data if we see the
	 * result of list_add() below, see try_to_grab_pending().
	 */
	smp_wmb();
O
Oleg Nesterov 已提交
128 129 130 131 132 133 134
	if (tail)
		list_add_tail(&work->entry, &cwq->worklist);
	else
		list_add(&work->entry, &cwq->worklist);
	wake_up(&cwq->more_work);
}

L
Linus Torvalds 已提交
135 136 137 138 139 140 141
/* 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);
O
Oleg Nesterov 已提交
142
	insert_work(cwq, work, 1);
L
Linus Torvalds 已提交
143 144 145
	spin_unlock_irqrestore(&cwq->lock, flags);
}

146 147 148 149 150
/**
 * queue_work - queue work on a workqueue
 * @wq: workqueue to use
 * @work: work to queue
 *
A
Alan Stern 已提交
151
 * Returns 0 if @work was already on a queue, non-zero otherwise.
L
Linus Torvalds 已提交
152 153 154 155 156 157
 *
 * 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)
{
158
	int ret = 0;
L
Linus Torvalds 已提交
159

160
	if (!test_and_set_bit(WORK_STRUCT_PENDING, work_data_bits(work))) {
L
Linus Torvalds 已提交
161
		BUG_ON(!list_empty(&work->entry));
162 163
		__queue_work(wq_per_cpu(wq, get_cpu()), work);
		put_cpu();
L
Linus Torvalds 已提交
164 165 166 167
		ret = 1;
	}
	return ret;
}
168
EXPORT_SYMBOL_GPL(queue_work);
L
Linus Torvalds 已提交
169

170
void delayed_work_timer_fn(unsigned long __data)
L
Linus Torvalds 已提交
171
{
172
	struct delayed_work *dwork = (struct delayed_work *)__data;
173 174
	struct cpu_workqueue_struct *cwq = get_wq_data(&dwork->work);
	struct workqueue_struct *wq = cwq->wq;
L
Linus Torvalds 已提交
175

176
	__queue_work(wq_per_cpu(wq, smp_processor_id()), &dwork->work);
L
Linus Torvalds 已提交
177 178
}

179 180 181
/**
 * queue_delayed_work - queue work on a workqueue after delay
 * @wq: workqueue to use
182
 * @dwork: delayable work to queue
183 184
 * @delay: number of jiffies to wait before queueing
 *
A
Alan Stern 已提交
185
 * Returns 0 if @work was already on a queue, non-zero otherwise.
186
 */
L
Linus Torvalds 已提交
187
int fastcall queue_delayed_work(struct workqueue_struct *wq,
188
			struct delayed_work *dwork, unsigned long delay)
L
Linus Torvalds 已提交
189
{
190
	timer_stats_timer_set_start_info(&dwork->timer);
191
	if (delay == 0)
192
		return queue_work(wq, &dwork->work);
L
Linus Torvalds 已提交
193

194
	return queue_delayed_work_on(-1, wq, dwork, delay);
L
Linus Torvalds 已提交
195
}
196
EXPORT_SYMBOL_GPL(queue_delayed_work);
L
Linus Torvalds 已提交
197

198 199 200 201
/**
 * queue_delayed_work_on - queue work on specific CPU after delay
 * @cpu: CPU number to execute work on
 * @wq: workqueue to use
202
 * @dwork: work to queue
203 204
 * @delay: number of jiffies to wait before queueing
 *
A
Alan Stern 已提交
205
 * Returns 0 if @work was already on a queue, non-zero otherwise.
206
 */
207
int queue_delayed_work_on(int cpu, struct workqueue_struct *wq,
208
			struct delayed_work *dwork, unsigned long delay)
209 210
{
	int ret = 0;
211 212
	struct timer_list *timer = &dwork->timer;
	struct work_struct *work = &dwork->work;
213

214
	if (!test_and_set_bit(WORK_STRUCT_PENDING, work_data_bits(work))) {
215 216 217
		BUG_ON(timer_pending(timer));
		BUG_ON(!list_empty(&work->entry));

218
		/* This stores cwq for the moment, for the timer_fn */
219
		set_wq_data(work, wq_per_cpu(wq, raw_smp_processor_id()));
220
		timer->expires = jiffies + delay;
221
		timer->data = (unsigned long)dwork;
222
		timer->function = delayed_work_timer_fn;
223 224 225 226 227

		if (unlikely(cpu >= 0))
			add_timer_on(timer, cpu);
		else
			add_timer(timer);
228 229 230 231
		ret = 1;
	}
	return ret;
}
232
EXPORT_SYMBOL_GPL(queue_delayed_work_on);
L
Linus Torvalds 已提交
233

234
static void run_workqueue(struct cpu_workqueue_struct *cwq)
L
Linus Torvalds 已提交
235
{
236
	spin_lock_irq(&cwq->lock);
L
Linus Torvalds 已提交
237 238 239 240 241 242 243 244 245 246
	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);
247
		work_func_t f = work->func;
L
Linus Torvalds 已提交
248

O
Oleg Nesterov 已提交
249
		cwq->current_work = work;
L
Linus Torvalds 已提交
250
		list_del_init(cwq->worklist.next);
251
		spin_unlock_irq(&cwq->lock);
L
Linus Torvalds 已提交
252

253
		BUG_ON(get_wq_data(work) != cwq);
O
Oleg Nesterov 已提交
254
		work_clear_pending(work);
255
		f(work);
L
Linus Torvalds 已提交
256

257 258 259 260 261 262 263 264 265 266 267
		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();
		}

268
		spin_lock_irq(&cwq->lock);
O
Oleg Nesterov 已提交
269
		cwq->current_work = NULL;
L
Linus Torvalds 已提交
270 271
	}
	cwq->run_depth--;
272
	spin_unlock_irq(&cwq->lock);
L
Linus Torvalds 已提交
273 274
}

275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292
/*
 * NOTE: the caller must not touch *cwq if this func returns true
 */
static int cwq_should_stop(struct cpu_workqueue_struct *cwq)
{
	int should_stop = cwq->should_stop;

	if (unlikely(should_stop)) {
		spin_lock_irq(&cwq->lock);
		should_stop = cwq->should_stop && list_empty(&cwq->worklist);
		if (should_stop)
			cwq->thread = NULL;
		spin_unlock_irq(&cwq->lock);
	}

	return should_stop;
}

L
Linus Torvalds 已提交
293 294 295
static int worker_thread(void *__cwq)
{
	struct cpu_workqueue_struct *cwq = __cwq;
296
	DEFINE_WAIT(wait);
L
Linus Torvalds 已提交
297

298
	if (!cwq->wq->freezeable)
299
		current->flags |= PF_NOFREEZE;
L
Linus Torvalds 已提交
300 301 302

	set_user_nice(current, -5);

303 304
	for (;;) {
		prepare_to_wait(&cwq->more_work, &wait, TASK_INTERRUPTIBLE);
305 306
		if (!freezing(current) && !cwq->should_stop
		    && list_empty(&cwq->worklist))
L
Linus Torvalds 已提交
307
			schedule();
308 309
		finish_wait(&cwq->more_work, &wait);

310 311
		try_to_freeze();

312 313
		if (cwq_should_stop(cwq))
			break;
L
Linus Torvalds 已提交
314

315
		run_workqueue(cwq);
L
Linus Torvalds 已提交
316
	}
317

L
Linus Torvalds 已提交
318 319 320
	return 0;
}

O
Oleg Nesterov 已提交
321 322 323 324 325 326 327 328 329 330 331
struct wq_barrier {
	struct work_struct	work;
	struct completion	done;
};

static void wq_barrier_func(struct work_struct *work)
{
	struct wq_barrier *barr = container_of(work, struct wq_barrier, work);
	complete(&barr->done);
}

332 333
static void insert_wq_barrier(struct cpu_workqueue_struct *cwq,
					struct wq_barrier *barr, int tail)
O
Oleg Nesterov 已提交
334 335 336 337 338
{
	INIT_WORK(&barr->work, wq_barrier_func);
	__set_bit(WORK_STRUCT_PENDING, work_data_bits(&barr->work));

	init_completion(&barr->done);
339 340

	insert_work(cwq, &barr->work, tail);
O
Oleg Nesterov 已提交
341 342
}

L
Linus Torvalds 已提交
343 344 345 346 347 348 349 350 351
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 {
O
Oleg Nesterov 已提交
352
		struct wq_barrier barr;
353
		int active = 0;
L
Linus Torvalds 已提交
354

355 356 357 358 359 360
		spin_lock_irq(&cwq->lock);
		if (!list_empty(&cwq->worklist) || cwq->current_work != NULL) {
			insert_wq_barrier(cwq, &barr, 1);
			active = 1;
		}
		spin_unlock_irq(&cwq->lock);
L
Linus Torvalds 已提交
361

362
		if (active)
363
			wait_for_completion(&barr.done);
L
Linus Torvalds 已提交
364 365 366
	}
}

367
/**
L
Linus Torvalds 已提交
368
 * flush_workqueue - ensure that any scheduled work has run to completion.
369
 * @wq: workqueue to flush
L
Linus Torvalds 已提交
370 371 372 373
 *
 * Forces execution of the workqueue and blocks until its completion.
 * This is typically used in driver shutdown handlers.
 *
O
Oleg Nesterov 已提交
374 375
 * We sleep until all works which were queued on entry have been handled,
 * but we are not livelocked by new incoming ones.
L
Linus Torvalds 已提交
376 377 378 379 380 381
 *
 * 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)
{
382
	const cpumask_t *cpu_map = wq_cpu_map(wq);
383
	int cpu;
L
Linus Torvalds 已提交
384

385 386 387
	might_sleep();
	for_each_cpu_mask(cpu, *cpu_map)
		flush_cpu_workqueue(per_cpu_ptr(wq->cpu_wq, cpu));
L
Linus Torvalds 已提交
388
}
389
EXPORT_SYMBOL_GPL(flush_workqueue);
L
Linus Torvalds 已提交
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 426 427 428 429 430
/*
 * Upon a successful return, the caller "owns" WORK_STRUCT_PENDING bit,
 * so this work can't be re-armed in any way.
 */
static int try_to_grab_pending(struct work_struct *work)
{
	struct cpu_workqueue_struct *cwq;
	int ret = 0;

	if (!test_and_set_bit(WORK_STRUCT_PENDING, work_data_bits(work)))
		return 1;

	/*
	 * The queueing is in progress, or it is already queued. Try to
	 * steal it from ->worklist without clearing WORK_STRUCT_PENDING.
	 */

	cwq = get_wq_data(work);
	if (!cwq)
		return ret;

	spin_lock_irq(&cwq->lock);
	if (!list_empty(&work->entry)) {
		/*
		 * This work is queued, but perhaps we locked the wrong cwq.
		 * In that case we must see the new value after rmb(), see
		 * insert_work()->wmb().
		 */
		smp_rmb();
		if (cwq == get_wq_data(work)) {
			list_del_init(&work->entry);
			ret = 1;
		}
	}
	spin_unlock_irq(&cwq->lock);

	return ret;
}

static void wait_on_cpu_work(struct cpu_workqueue_struct *cwq,
O
Oleg Nesterov 已提交
431 432 433 434 435 436 437
				struct work_struct *work)
{
	struct wq_barrier barr;
	int running = 0;

	spin_lock_irq(&cwq->lock);
	if (unlikely(cwq->current_work == work)) {
438
		insert_wq_barrier(cwq, &barr, 0);
O
Oleg Nesterov 已提交
439 440 441 442
		running = 1;
	}
	spin_unlock_irq(&cwq->lock);

443
	if (unlikely(running))
O
Oleg Nesterov 已提交
444 445 446
		wait_for_completion(&barr.done);
}

447
static void wait_on_work(struct work_struct *work)
O
Oleg Nesterov 已提交
448 449
{
	struct cpu_workqueue_struct *cwq;
450 451
	struct workqueue_struct *wq;
	const cpumask_t *cpu_map;
452
	int cpu;
O
Oleg Nesterov 已提交
453

454 455
	might_sleep();

O
Oleg Nesterov 已提交
456 457
	cwq = get_wq_data(work);
	if (!cwq)
458
		return;
O
Oleg Nesterov 已提交
459

460 461 462
	wq = cwq->wq;
	cpu_map = wq_cpu_map(wq);

463
	for_each_cpu_mask(cpu, *cpu_map)
464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491
		wait_on_cpu_work(per_cpu_ptr(wq->cpu_wq, cpu), work);
}

/**
 * cancel_work_sync - block until a work_struct's callback has terminated
 * @work: the work which is to be flushed
 *
 * cancel_work_sync() will cancel the work if it is queued. If the work's
 * callback appears to be running, cancel_work_sync() will block until it
 * has completed.
 *
 * It is possible to use this function if the work re-queues itself. It can
 * cancel the work even if it migrates to another workqueue, however in that
 * case it only guarantees that work->func() has completed on the last queued
 * workqueue.
 *
 * cancel_work_sync(&delayed_work->work) should be used only if ->timer is not
 * pending, otherwise it goes into a busy-wait loop until the timer expires.
 *
 * The caller must ensure that workqueue_struct on which this work was last
 * queued can't be destroyed before this function returns.
 */
void cancel_work_sync(struct work_struct *work)
{
	while (!try_to_grab_pending(work))
		cpu_relax();
	wait_on_work(work);
	work_clear_pending(work);
O
Oleg Nesterov 已提交
492
}
493
EXPORT_SYMBOL_GPL(cancel_work_sync);
O
Oleg Nesterov 已提交
494

495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510
/**
 * cancel_rearming_delayed_work - reliably kill off a delayed work.
 * @dwork: the delayed work struct
 *
 * It is possible to use this function if @dwork rearms itself via queue_work()
 * or queue_delayed_work(). See also the comment for cancel_work_sync().
 */
void cancel_rearming_delayed_work(struct delayed_work *dwork)
{
	while (!del_timer(&dwork->timer) &&
	       !try_to_grab_pending(&dwork->work))
		cpu_relax();
	wait_on_work(&dwork->work);
	work_clear_pending(&dwork->work);
}
EXPORT_SYMBOL(cancel_rearming_delayed_work);
L
Linus Torvalds 已提交
511

512
static struct workqueue_struct *keventd_wq __read_mostly;
L
Linus Torvalds 已提交
513

514 515 516 517 518 519
/**
 * 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 已提交
520 521 522 523
int fastcall schedule_work(struct work_struct *work)
{
	return queue_work(keventd_wq, work);
}
524
EXPORT_SYMBOL(schedule_work);
L
Linus Torvalds 已提交
525

526 527
/**
 * schedule_delayed_work - put work task in global workqueue after delay
528 529
 * @dwork: job to be done
 * @delay: number of jiffies to wait or 0 for immediate execution
530 531 532 533
 *
 * After waiting for a given time this puts a job in the kernel-global
 * workqueue.
 */
534 535
int fastcall schedule_delayed_work(struct delayed_work *dwork,
					unsigned long delay)
L
Linus Torvalds 已提交
536
{
537
	timer_stats_timer_set_start_info(&dwork->timer);
538
	return queue_delayed_work(keventd_wq, dwork, delay);
L
Linus Torvalds 已提交
539
}
540
EXPORT_SYMBOL(schedule_delayed_work);
L
Linus Torvalds 已提交
541

542 543 544
/**
 * schedule_delayed_work_on - queue work in global workqueue on CPU after delay
 * @cpu: cpu to use
545
 * @dwork: job to be done
546 547 548 549 550
 * @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 已提交
551
int schedule_delayed_work_on(int cpu,
552
			struct delayed_work *dwork, unsigned long delay)
L
Linus Torvalds 已提交
553
{
554
	return queue_delayed_work_on(cpu, keventd_wq, dwork, delay);
L
Linus Torvalds 已提交
555
}
556
EXPORT_SYMBOL(schedule_delayed_work_on);
L
Linus Torvalds 已提交
557

558 559 560 561 562 563 564 565 566 567 568
/**
 * 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.
 */
569
int schedule_on_each_cpu(work_func_t func)
570 571
{
	int cpu;
572
	struct work_struct *works;
573

574 575
	works = alloc_percpu(struct work_struct);
	if (!works)
576
		return -ENOMEM;
577

578
	preempt_disable();		/* CPU hotplug */
579
	for_each_online_cpu(cpu) {
580 581 582 583 584
		struct work_struct *work = per_cpu_ptr(works, cpu);

		INIT_WORK(work, func);
		set_bit(WORK_STRUCT_PENDING, work_data_bits(work));
		__queue_work(per_cpu_ptr(keventd_wq->cpu_wq, cpu), work);
585
	}
586
	preempt_enable();
587
	flush_workqueue(keventd_wq);
588
	free_percpu(works);
589 590 591
	return 0;
}

L
Linus Torvalds 已提交
592 593 594 595
void flush_scheduled_work(void)
{
	flush_workqueue(keventd_wq);
}
596
EXPORT_SYMBOL(flush_scheduled_work);
L
Linus Torvalds 已提交
597

598 599 600 601 602 603 604 605 606 607 608 609
/**
 * 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
 */
610
int execute_in_process_context(work_func_t fn, struct execute_work *ew)
611 612
{
	if (!in_interrupt()) {
613
		fn(&ew->work);
614 615 616
		return 0;
	}

617
	INIT_WORK(&ew->work, fn);
618 619 620 621 622 623
	schedule_work(&ew->work);

	return 1;
}
EXPORT_SYMBOL_GPL(execute_in_process_context);

L
Linus Torvalds 已提交
624 625 626 627 628 629 630 631 632 633 634 635 636
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);

637
	cwq = per_cpu_ptr(keventd_wq->cpu_wq, cpu);
L
Linus Torvalds 已提交
638 639 640 641 642 643 644
	if (current == cwq->thread)
		ret = 1;

	return ret;

}

645 646
static struct cpu_workqueue_struct *
init_cpu_workqueue(struct workqueue_struct *wq, int cpu)
L
Linus Torvalds 已提交
647
{
648
	struct cpu_workqueue_struct *cwq = per_cpu_ptr(wq->cpu_wq, cpu);
L
Linus Torvalds 已提交
649

650 651 652 653 654 655
	cwq->wq = wq;
	spin_lock_init(&cwq->lock);
	INIT_LIST_HEAD(&cwq->worklist);
	init_waitqueue_head(&cwq->more_work);

	return cwq;
L
Linus Torvalds 已提交
656 657
}

658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681
static int create_workqueue_thread(struct cpu_workqueue_struct *cwq, int cpu)
{
	struct workqueue_struct *wq = cwq->wq;
	const char *fmt = is_single_threaded(wq) ? "%s" : "%s/%d";
	struct task_struct *p;

	p = kthread_create(worker_thread, cwq, fmt, wq->name, cpu);
	/*
	 * Nobody can add the work_struct to this cwq,
	 *	if (caller is __create_workqueue)
	 *		nobody should see this wq
	 *	else // caller is CPU_UP_PREPARE
	 *		cpu is not on cpu_online_map
	 * so we can abort safely.
	 */
	if (IS_ERR(p))
		return PTR_ERR(p);

	cwq->thread = p;
	cwq->should_stop = 0;

	return 0;
}

682 683 684 685 686 687 688 689 690 691 692
static void start_workqueue_thread(struct cpu_workqueue_struct *cwq, int cpu)
{
	struct task_struct *p = cwq->thread;

	if (p != NULL) {
		if (cpu >= 0)
			kthread_bind(p, cpu);
		wake_up_process(p);
	}
}

693 694
struct workqueue_struct *__create_workqueue(const char *name,
					    int singlethread, int freezeable)
L
Linus Torvalds 已提交
695 696
{
	struct workqueue_struct *wq;
697 698
	struct cpu_workqueue_struct *cwq;
	int err = 0, cpu;
L
Linus Torvalds 已提交
699

700 701 702 703 704 705 706 707 708 709 710
	wq = kzalloc(sizeof(*wq), GFP_KERNEL);
	if (!wq)
		return NULL;

	wq->cpu_wq = alloc_percpu(struct cpu_workqueue_struct);
	if (!wq->cpu_wq) {
		kfree(wq);
		return NULL;
	}

	wq->name = name;
711
	wq->singlethread = singlethread;
712
	wq->freezeable = freezeable;
713
	INIT_LIST_HEAD(&wq->list);
714 715 716 717

	if (singlethread) {
		cwq = init_cpu_workqueue(wq, singlethread_cpu);
		err = create_workqueue_thread(cwq, singlethread_cpu);
718
		start_workqueue_thread(cwq, -1);
719
	} else {
720
		mutex_lock(&workqueue_mutex);
721 722 723 724 725 726 727
		list_add(&wq->list, &workqueues);

		for_each_possible_cpu(cpu) {
			cwq = init_cpu_workqueue(wq, cpu);
			if (err || !cpu_online(cpu))
				continue;
			err = create_workqueue_thread(cwq, cpu);
728
			start_workqueue_thread(cwq, cpu);
L
Linus Torvalds 已提交
729
		}
730 731 732 733 734 735 736 737 738 739
		mutex_unlock(&workqueue_mutex);
	}

	if (err) {
		destroy_workqueue(wq);
		wq = NULL;
	}
	return wq;
}
EXPORT_SYMBOL_GPL(__create_workqueue);
L
Linus Torvalds 已提交
740

741 742 743 744
static void cleanup_workqueue_thread(struct cpu_workqueue_struct *cwq, int cpu)
{
	struct wq_barrier barr;
	int alive = 0;
745

746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775
	spin_lock_irq(&cwq->lock);
	if (cwq->thread != NULL) {
		insert_wq_barrier(cwq, &barr, 1);
		cwq->should_stop = 1;
		alive = 1;
	}
	spin_unlock_irq(&cwq->lock);

	if (alive) {
		wait_for_completion(&barr.done);

		while (unlikely(cwq->thread != NULL))
			cpu_relax();
		/*
		 * Wait until cwq->thread unlocks cwq->lock,
		 * it won't touch *cwq after that.
		 */
		smp_rmb();
		spin_unlock_wait(&cwq->lock);
	}
}

/**
 * destroy_workqueue - safely terminate a workqueue
 * @wq: target workqueue
 *
 * Safely destroy a workqueue. All work currently pending will be done first.
 */
void destroy_workqueue(struct workqueue_struct *wq)
{
776
	const cpumask_t *cpu_map = wq_cpu_map(wq);
777
	struct cpu_workqueue_struct *cwq;
778
	int cpu;
779

780 781 782
	mutex_lock(&workqueue_mutex);
	list_del(&wq->list);
	mutex_unlock(&workqueue_mutex);
783

784 785 786
	for_each_cpu_mask(cpu, *cpu_map) {
		cwq = per_cpu_ptr(wq->cpu_wq, cpu);
		cleanup_workqueue_thread(cwq, cpu);
787
	}
788

789 790 791 792 793 794 795 796 797 798 799 800 801
	free_percpu(wq->cpu_wq);
	kfree(wq);
}
EXPORT_SYMBOL_GPL(destroy_workqueue);

static int __devinit workqueue_cpu_callback(struct notifier_block *nfb,
						unsigned long action,
						void *hcpu)
{
	unsigned int cpu = (unsigned long)hcpu;
	struct cpu_workqueue_struct *cwq;
	struct workqueue_struct *wq;

802 803
	action &= ~CPU_TASKS_FROZEN;

804 805
	switch (action) {
	case CPU_LOCK_ACQUIRE:
806
		mutex_lock(&workqueue_mutex);
807
		return NOTIFY_OK;
808

809
	case CPU_LOCK_RELEASE:
810
		mutex_unlock(&workqueue_mutex);
811
		return NOTIFY_OK;
L
Linus Torvalds 已提交
812

813 814 815 816 817 818 819 820 821 822 823 824 825 826 827
	case CPU_UP_PREPARE:
		cpu_set(cpu, cpu_populated_map);
	}

	list_for_each_entry(wq, &workqueues, list) {
		cwq = per_cpu_ptr(wq->cpu_wq, cpu);

		switch (action) {
		case CPU_UP_PREPARE:
			if (!create_workqueue_thread(cwq, cpu))
				break;
			printk(KERN_ERR "workqueue for %i failed\n", cpu);
			return NOTIFY_BAD;

		case CPU_ONLINE:
828
			start_workqueue_thread(cwq, cpu);
829 830 831
			break;

		case CPU_UP_CANCELED:
832
			start_workqueue_thread(cwq, -1);
833 834 835 836
		case CPU_DEAD:
			cleanup_workqueue_thread(cwq, cpu);
			break;
		}
L
Linus Torvalds 已提交
837 838 839 840 841
	}

	return NOTIFY_OK;
}

842
void __init init_workqueues(void)
L
Linus Torvalds 已提交
843
{
844
	cpu_populated_map = cpu_online_map;
845
	singlethread_cpu = first_cpu(cpu_possible_map);
846
	cpu_singlethread_map = cpumask_of_cpu(singlethread_cpu);
L
Linus Torvalds 已提交
847 848 849 850
	hotcpu_notifier(workqueue_cpu_callback, 0);
	keventd_wq = create_workqueue("events");
	BUG_ON(!keventd_wq);
}