rcupdate.c 3.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * Read-Copy Update mechanism for mutual exclusion
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
18
 * Copyright IBM Corporation, 2001
L
Linus Torvalds 已提交
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
 *
 * Authors: Dipankar Sarma <dipankar@in.ibm.com>
 *	    Manfred Spraul <manfred@colorfullife.com>
 * 
 * Based on the original work by Paul McKenney <paulmck@us.ibm.com>
 * and inputs from Rusty Russell, Andrea Arcangeli and Andi Kleen.
 * Papers:
 * http://www.rdrop.com/users/paulmck/paper/rclockpdcsproof.pdf
 * http://lse.sourceforge.net/locking/rclock_OLS.2001.05.01c.sc.pdf (OLS2001)
 *
 * For detailed explanation of Read-Copy Update mechanism see -
 * 		http://lse.sourceforge.net/locking/rcupdate.html
 *
 */
#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/spinlock.h>
#include <linux/smp.h>
#include <linux/interrupt.h>
#include <linux/sched.h>
#include <asm/atomic.h>
#include <linux/bitops.h>
#include <linux/percpu.h>
#include <linux/notifier.h>
#include <linux/cpu.h>
45
#include <linux/mutex.h>
46
#include <linux/module.h>
L
Linus Torvalds 已提交
47

48
static DEFINE_PER_CPU(struct rcu_head, rcu_barrier_head) = {NULL};
D
Dipankar Sarma 已提交
49
static atomic_t rcu_barrier_cpu_count;
50
static DEFINE_MUTEX(rcu_barrier_mutex);
D
Dipankar Sarma 已提交
51 52
static struct completion rcu_barrier_completion;

P
Paul E. McKenney 已提交
53 54 55 56
/*
 * Awaken the corresponding synchronize_rcu() instance now that a
 * grace period has elapsed.
 */
P
Paul E. McKenney 已提交
57
void wakeme_after_rcu(struct rcu_head  *head)
D
Dipankar Sarma 已提交
58
{
59 60 61 62
	struct rcu_synchronize *rcu;

	rcu = container_of(head, struct rcu_synchronize, head);
	complete(&rcu->completion);
D
Dipankar Sarma 已提交
63
}
L
Linus Torvalds 已提交
64 65

/**
66
 * synchronize_rcu - wait until a grace period has elapsed.
L
Linus Torvalds 已提交
67
 *
68 69
 * Control will return to the caller some time after a full grace
 * period has elapsed, in other words after all currently executing RCU
L
Linus Torvalds 已提交
70 71 72 73
 * read-side critical sections have completed.  RCU read-side critical
 * sections are delimited by rcu_read_lock() and rcu_read_unlock(),
 * and may be nested.
 */
P
Paul E. McKenney 已提交
74
synchronize_rcu_xxx(synchronize_rcu, call_rcu)
75
EXPORT_SYMBOL_GPL(synchronize_rcu);
76

77 78 79 80 81 82 83 84 85 86 87 88
static void rcu_barrier_callback(struct rcu_head *notused)
{
	if (atomic_dec_and_test(&rcu_barrier_cpu_count))
		complete(&rcu_barrier_completion);
}

/*
 * Called with preemption disabled, and from cross-cpu IRQ context.
 */
static void rcu_barrier_func(void *notused)
{
	int cpu = smp_processor_id();
89
	struct rcu_head *head = &per_cpu(rcu_barrier_head, cpu);
90 91 92 93 94 95 96 97 98 99 100

	atomic_inc(&rcu_barrier_cpu_count);
	call_rcu(head, rcu_barrier_callback);
}

/**
 * rcu_barrier - Wait until all the in-flight RCUs are complete.
 */
void rcu_barrier(void)
{
	BUG_ON(in_interrupt());
101 102
	/* Take cpucontrol mutex to protect against CPU hotplug */
	mutex_lock(&rcu_barrier_mutex);
103 104
	init_completion(&rcu_barrier_completion);
	atomic_set(&rcu_barrier_cpu_count, 0);
105 106 107 108 109 110 111 112 113
	/*
	 * The queueing of callbacks in all CPUs must be atomic with
	 * respect to RCU, otherwise one CPU may queue a callback,
	 * wait for a grace period, decrement barrier count and call
	 * complete(), while other CPUs have not yet queued anything.
	 * So, we need to make sure that grace periods cannot complete
	 * until all the callbacks are queued.
	 */
	rcu_read_lock();
114
	on_each_cpu(rcu_barrier_func, NULL, 0, 1);
115
	rcu_read_unlock();
116
	wait_for_completion(&rcu_barrier_completion);
117
	mutex_unlock(&rcu_barrier_mutex);
118 119 120
}
EXPORT_SYMBOL_GPL(rcu_barrier);

L
Linus Torvalds 已提交
121 122
void __init rcu_init(void)
{
123
	__rcu_init();
L
Linus Torvalds 已提交
124 125
}