rcu: Renamings to increase RCU clarity
[linux-2.6.git] / kernel / rcupreempt.c
1 /*
2  * Read-Copy Update mechanism for mutual exclusion, realtime implementation
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17  *
18  * Copyright IBM Corporation, 2006
19  *
20  * Authors: Paul E. McKenney <paulmck@us.ibm.com>
21  *              With thanks to Esben Nielsen, Bill Huey, and Ingo Molnar
22  *              for pushing me away from locks and towards counters, and
23  *              to Suparna Bhattacharya for pushing me completely away
24  *              from atomic instructions on the read side.
25  *
26  *  - Added handling of Dynamic Ticks
27  *      Copyright 2007 - Paul E. Mckenney <paulmck@us.ibm.com>
28  *                     - Steven Rostedt <srostedt@redhat.com>
29  *
30  * Papers:  http://www.rdrop.com/users/paulmck/RCU
31  *
32  * Design Document: http://lwn.net/Articles/253651/
33  *
34  * For detailed explanation of Read-Copy Update mechanism see -
35  *              Documentation/RCU/ *.txt
36  *
37  */
38 #include <linux/types.h>
39 #include <linux/kernel.h>
40 #include <linux/init.h>
41 #include <linux/spinlock.h>
42 #include <linux/smp.h>
43 #include <linux/rcupdate.h>
44 #include <linux/interrupt.h>
45 #include <linux/sched.h>
46 #include <asm/atomic.h>
47 #include <linux/bitops.h>
48 #include <linux/module.h>
49 #include <linux/kthread.h>
50 #include <linux/completion.h>
51 #include <linux/moduleparam.h>
52 #include <linux/percpu.h>
53 #include <linux/notifier.h>
54 #include <linux/cpu.h>
55 #include <linux/random.h>
56 #include <linux/delay.h>
57 #include <linux/cpumask.h>
58 #include <linux/rcupreempt_trace.h>
59 #include <asm/byteorder.h>
60
61 /*
62  * PREEMPT_RCU data structures.
63  */
64
65 /*
66  * GP_STAGES specifies the number of times the state machine has
67  * to go through the all the rcu_try_flip_states (see below)
68  * in a single Grace Period.
69  *
70  * GP in GP_STAGES stands for Grace Period ;)
71  */
72 #define GP_STAGES    2
73 struct rcu_data {
74         spinlock_t      lock;           /* Protect rcu_data fields. */
75         long            completed;      /* Number of last completed batch. */
76         int             waitlistcount;
77         struct rcu_head *nextlist;
78         struct rcu_head **nexttail;
79         struct rcu_head *waitlist[GP_STAGES];
80         struct rcu_head **waittail[GP_STAGES];
81         struct rcu_head *donelist;      /* from waitlist & waitschedlist */
82         struct rcu_head **donetail;
83         long rcu_flipctr[2];
84         struct rcu_head *nextschedlist;
85         struct rcu_head **nextschedtail;
86         struct rcu_head *waitschedlist;
87         struct rcu_head **waitschedtail;
88         int rcu_sched_sleeping;
89 #ifdef CONFIG_RCU_TRACE
90         struct rcupreempt_trace trace;
91 #endif /* #ifdef CONFIG_RCU_TRACE */
92 };
93
94 /*
95  * States for rcu_try_flip() and friends.
96  */
97
98 enum rcu_try_flip_states {
99
100         /*
101          * Stay here if nothing is happening. Flip the counter if somthing
102          * starts happening. Denoted by "I"
103          */
104         rcu_try_flip_idle_state,
105
106         /*
107          * Wait here for all CPUs to notice that the counter has flipped. This
108          * prevents the old set of counters from ever being incremented once
109          * we leave this state, which in turn is necessary because we cannot
110          * test any individual counter for zero -- we can only check the sum.
111          * Denoted by "A".
112          */
113         rcu_try_flip_waitack_state,
114
115         /*
116          * Wait here for the sum of the old per-CPU counters to reach zero.
117          * Denoted by "Z".
118          */
119         rcu_try_flip_waitzero_state,
120
121         /*
122          * Wait here for each of the other CPUs to execute a memory barrier.
123          * This is necessary to ensure that these other CPUs really have
124          * completed executing their RCU read-side critical sections, despite
125          * their CPUs wildly reordering memory. Denoted by "M".
126          */
127         rcu_try_flip_waitmb_state,
128 };
129
130 /*
131  * States for rcu_ctrlblk.rcu_sched_sleep.
132  */
133
134 enum rcu_sched_sleep_states {
135         rcu_sched_not_sleeping, /* Not sleeping, callbacks need GP.  */
136         rcu_sched_sleep_prep,   /* Thinking of sleeping, rechecking. */
137         rcu_sched_sleeping,     /* Sleeping, awaken if GP needed. */
138 };
139
140 struct rcu_ctrlblk {
141         spinlock_t      fliplock;       /* Protect state-machine transitions. */
142         long            completed;      /* Number of last completed batch. */
143         enum rcu_try_flip_states rcu_try_flip_state; /* The current state of
144                                                         the rcu state machine */
145         spinlock_t      schedlock;      /* Protect rcu_sched sleep state. */
146         enum rcu_sched_sleep_states sched_sleep; /* rcu_sched state. */
147         wait_queue_head_t sched_wq;     /* Place for rcu_sched to sleep. */
148 };
149
150 struct rcu_dyntick_sched {
151         int dynticks;
152         int dynticks_snap;
153         int sched_qs;
154         int sched_qs_snap;
155         int sched_dynticks_snap;
156 };
157
158 static DEFINE_PER_CPU_SHARED_ALIGNED(struct rcu_dyntick_sched, rcu_dyntick_sched) = {
159         .dynticks = 1,
160 };
161
162 void rcu_sched_qs(int cpu)
163 {
164         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
165
166         rdssp->sched_qs++;
167 }
168
169 #ifdef CONFIG_NO_HZ
170
171 void rcu_enter_nohz(void)
172 {
173         static DEFINE_RATELIMIT_STATE(rs, 10 * HZ, 1);
174
175         smp_mb(); /* CPUs seeing ++ must see prior RCU read-side crit sects */
176         __get_cpu_var(rcu_dyntick_sched).dynticks++;
177         WARN_ON_RATELIMIT(__get_cpu_var(rcu_dyntick_sched).dynticks & 0x1, &rs);
178 }
179
180 void rcu_exit_nohz(void)
181 {
182         static DEFINE_RATELIMIT_STATE(rs, 10 * HZ, 1);
183
184         __get_cpu_var(rcu_dyntick_sched).dynticks++;
185         smp_mb(); /* CPUs seeing ++ must see later RCU read-side crit sects */
186         WARN_ON_RATELIMIT(!(__get_cpu_var(rcu_dyntick_sched).dynticks & 0x1),
187                                 &rs);
188 }
189
190 #endif /* CONFIG_NO_HZ */
191
192
193 static DEFINE_PER_CPU(struct rcu_data, rcu_data);
194
195 static struct rcu_ctrlblk rcu_ctrlblk = {
196         .fliplock = __SPIN_LOCK_UNLOCKED(rcu_ctrlblk.fliplock),
197         .completed = 0,
198         .rcu_try_flip_state = rcu_try_flip_idle_state,
199         .schedlock = __SPIN_LOCK_UNLOCKED(rcu_ctrlblk.schedlock),
200         .sched_sleep = rcu_sched_not_sleeping,
201         .sched_wq = __WAIT_QUEUE_HEAD_INITIALIZER(rcu_ctrlblk.sched_wq),
202 };
203
204 static struct task_struct *rcu_sched_grace_period_task;
205
206 #ifdef CONFIG_RCU_TRACE
207 static char *rcu_try_flip_state_names[] =
208         { "idle", "waitack", "waitzero", "waitmb" };
209 #endif /* #ifdef CONFIG_RCU_TRACE */
210
211 static DECLARE_BITMAP(rcu_cpu_online_map, NR_CPUS) __read_mostly
212         = CPU_BITS_NONE;
213
214 /*
215  * Enum and per-CPU flag to determine when each CPU has seen
216  * the most recent counter flip.
217  */
218
219 enum rcu_flip_flag_values {
220         rcu_flip_seen,          /* Steady/initial state, last flip seen. */
221                                 /* Only GP detector can update. */
222         rcu_flipped             /* Flip just completed, need confirmation. */
223                                 /* Only corresponding CPU can update. */
224 };
225 static DEFINE_PER_CPU_SHARED_ALIGNED(enum rcu_flip_flag_values, rcu_flip_flag)
226                                                                 = rcu_flip_seen;
227
228 /*
229  * Enum and per-CPU flag to determine when each CPU has executed the
230  * needed memory barrier to fence in memory references from its last RCU
231  * read-side critical section in the just-completed grace period.
232  */
233
234 enum rcu_mb_flag_values {
235         rcu_mb_done,            /* Steady/initial state, no mb()s required. */
236                                 /* Only GP detector can update. */
237         rcu_mb_needed           /* Flip just completed, need an mb(). */
238                                 /* Only corresponding CPU can update. */
239 };
240 static DEFINE_PER_CPU_SHARED_ALIGNED(enum rcu_mb_flag_values, rcu_mb_flag)
241                                                                 = rcu_mb_done;
242
243 /*
244  * RCU_DATA_ME: find the current CPU's rcu_data structure.
245  * RCU_DATA_CPU: find the specified CPU's rcu_data structure.
246  */
247 #define RCU_DATA_ME()           (&__get_cpu_var(rcu_data))
248 #define RCU_DATA_CPU(cpu)       (&per_cpu(rcu_data, cpu))
249
250 /*
251  * Helper macro for tracing when the appropriate rcu_data is not
252  * cached in a local variable, but where the CPU number is so cached.
253  */
254 #define RCU_TRACE_CPU(f, cpu) RCU_TRACE(f, &(RCU_DATA_CPU(cpu)->trace));
255
256 /*
257  * Helper macro for tracing when the appropriate rcu_data is not
258  * cached in a local variable.
259  */
260 #define RCU_TRACE_ME(f) RCU_TRACE(f, &(RCU_DATA_ME()->trace));
261
262 /*
263  * Helper macro for tracing when the appropriate rcu_data is pointed
264  * to by a local variable.
265  */
266 #define RCU_TRACE_RDP(f, rdp) RCU_TRACE(f, &((rdp)->trace));
267
268 #define RCU_SCHED_BATCH_TIME (HZ / 50)
269
270 /*
271  * Return the number of RCU batches processed thus far.  Useful
272  * for debug and statistics.
273  */
274 long rcu_batches_completed(void)
275 {
276         return rcu_ctrlblk.completed;
277 }
278 EXPORT_SYMBOL_GPL(rcu_batches_completed);
279
280 void __rcu_read_lock(void)
281 {
282         int idx;
283         struct task_struct *t = current;
284         int nesting;
285
286         nesting = ACCESS_ONCE(t->rcu_read_lock_nesting);
287         if (nesting != 0) {
288
289                 /* An earlier rcu_read_lock() covers us, just count it. */
290
291                 t->rcu_read_lock_nesting = nesting + 1;
292
293         } else {
294                 unsigned long flags;
295
296                 /*
297                  * We disable interrupts for the following reasons:
298                  * - If we get scheduling clock interrupt here, and we
299                  *   end up acking the counter flip, it's like a promise
300                  *   that we will never increment the old counter again.
301                  *   Thus we will break that promise if that
302                  *   scheduling clock interrupt happens between the time
303                  *   we pick the .completed field and the time that we
304                  *   increment our counter.
305                  *
306                  * - We don't want to be preempted out here.
307                  *
308                  * NMIs can still occur, of course, and might themselves
309                  * contain rcu_read_lock().
310                  */
311
312                 local_irq_save(flags);
313
314                 /*
315                  * Outermost nesting of rcu_read_lock(), so increment
316                  * the current counter for the current CPU.  Use volatile
317                  * casts to prevent the compiler from reordering.
318                  */
319
320                 idx = ACCESS_ONCE(rcu_ctrlblk.completed) & 0x1;
321                 ACCESS_ONCE(RCU_DATA_ME()->rcu_flipctr[idx])++;
322
323                 /*
324                  * Now that the per-CPU counter has been incremented, we
325                  * are protected from races with rcu_read_lock() invoked
326                  * from NMI handlers on this CPU.  We can therefore safely
327                  * increment the nesting counter, relieving further NMIs
328                  * of the need to increment the per-CPU counter.
329                  */
330
331                 ACCESS_ONCE(t->rcu_read_lock_nesting) = nesting + 1;
332
333                 /*
334                  * Now that we have preventing any NMIs from storing
335                  * to the ->rcu_flipctr_idx, we can safely use it to
336                  * remember which counter to decrement in the matching
337                  * rcu_read_unlock().
338                  */
339
340                 ACCESS_ONCE(t->rcu_flipctr_idx) = idx;
341                 local_irq_restore(flags);
342         }
343 }
344 EXPORT_SYMBOL_GPL(__rcu_read_lock);
345
346 void __rcu_read_unlock(void)
347 {
348         int idx;
349         struct task_struct *t = current;
350         int nesting;
351
352         nesting = ACCESS_ONCE(t->rcu_read_lock_nesting);
353         if (nesting > 1) {
354
355                 /*
356                  * We are still protected by the enclosing rcu_read_lock(),
357                  * so simply decrement the counter.
358                  */
359
360                 t->rcu_read_lock_nesting = nesting - 1;
361
362         } else {
363                 unsigned long flags;
364
365                 /*
366                  * Disable local interrupts to prevent the grace-period
367                  * detection state machine from seeing us half-done.
368                  * NMIs can still occur, of course, and might themselves
369                  * contain rcu_read_lock() and rcu_read_unlock().
370                  */
371
372                 local_irq_save(flags);
373
374                 /*
375                  * Outermost nesting of rcu_read_unlock(), so we must
376                  * decrement the current counter for the current CPU.
377                  * This must be done carefully, because NMIs can
378                  * occur at any point in this code, and any rcu_read_lock()
379                  * and rcu_read_unlock() pairs in the NMI handlers
380                  * must interact non-destructively with this code.
381                  * Lots of volatile casts, and -very- careful ordering.
382                  *
383                  * Changes to this code, including this one, must be
384                  * inspected, validated, and tested extremely carefully!!!
385                  */
386
387                 /*
388                  * First, pick up the index.
389                  */
390
391                 idx = ACCESS_ONCE(t->rcu_flipctr_idx);
392
393                 /*
394                  * Now that we have fetched the counter index, it is
395                  * safe to decrement the per-task RCU nesting counter.
396                  * After this, any interrupts or NMIs will increment and
397                  * decrement the per-CPU counters.
398                  */
399                 ACCESS_ONCE(t->rcu_read_lock_nesting) = nesting - 1;
400
401                 /*
402                  * It is now safe to decrement this task's nesting count.
403                  * NMIs that occur after this statement will route their
404                  * rcu_read_lock() calls through this "else" clause, and
405                  * will thus start incrementing the per-CPU counter on
406                  * their own.  They will also clobber ->rcu_flipctr_idx,
407                  * but that is OK, since we have already fetched it.
408                  */
409
410                 ACCESS_ONCE(RCU_DATA_ME()->rcu_flipctr[idx])--;
411                 local_irq_restore(flags);
412         }
413 }
414 EXPORT_SYMBOL_GPL(__rcu_read_unlock);
415
416 /*
417  * If a global counter flip has occurred since the last time that we
418  * advanced callbacks, advance them.  Hardware interrupts must be
419  * disabled when calling this function.
420  */
421 static void __rcu_advance_callbacks(struct rcu_data *rdp)
422 {
423         int cpu;
424         int i;
425         int wlc = 0;
426
427         if (rdp->completed != rcu_ctrlblk.completed) {
428                 if (rdp->waitlist[GP_STAGES - 1] != NULL) {
429                         *rdp->donetail = rdp->waitlist[GP_STAGES - 1];
430                         rdp->donetail = rdp->waittail[GP_STAGES - 1];
431                         RCU_TRACE_RDP(rcupreempt_trace_move2done, rdp);
432                 }
433                 for (i = GP_STAGES - 2; i >= 0; i--) {
434                         if (rdp->waitlist[i] != NULL) {
435                                 rdp->waitlist[i + 1] = rdp->waitlist[i];
436                                 rdp->waittail[i + 1] = rdp->waittail[i];
437                                 wlc++;
438                         } else {
439                                 rdp->waitlist[i + 1] = NULL;
440                                 rdp->waittail[i + 1] =
441                                         &rdp->waitlist[i + 1];
442                         }
443                 }
444                 if (rdp->nextlist != NULL) {
445                         rdp->waitlist[0] = rdp->nextlist;
446                         rdp->waittail[0] = rdp->nexttail;
447                         wlc++;
448                         rdp->nextlist = NULL;
449                         rdp->nexttail = &rdp->nextlist;
450                         RCU_TRACE_RDP(rcupreempt_trace_move2wait, rdp);
451                 } else {
452                         rdp->waitlist[0] = NULL;
453                         rdp->waittail[0] = &rdp->waitlist[0];
454                 }
455                 rdp->waitlistcount = wlc;
456                 rdp->completed = rcu_ctrlblk.completed;
457         }
458
459         /*
460          * Check to see if this CPU needs to report that it has seen
461          * the most recent counter flip, thereby declaring that all
462          * subsequent rcu_read_lock() invocations will respect this flip.
463          */
464
465         cpu = raw_smp_processor_id();
466         if (per_cpu(rcu_flip_flag, cpu) == rcu_flipped) {
467                 smp_mb();  /* Subsequent counter accesses must see new value */
468                 per_cpu(rcu_flip_flag, cpu) = rcu_flip_seen;
469                 smp_mb();  /* Subsequent RCU read-side critical sections */
470                            /*  seen -after- acknowledgement. */
471         }
472 }
473
474 #ifdef CONFIG_NO_HZ
475 static DEFINE_PER_CPU(int, rcu_update_flag);
476
477 /**
478  * rcu_irq_enter - Called from Hard irq handlers and NMI/SMI.
479  *
480  * If the CPU was idle with dynamic ticks active, this updates the
481  * rcu_dyntick_sched.dynticks to let the RCU handling know that the
482  * CPU is active.
483  */
484 void rcu_irq_enter(void)
485 {
486         int cpu = smp_processor_id();
487         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
488
489         if (per_cpu(rcu_update_flag, cpu))
490                 per_cpu(rcu_update_flag, cpu)++;
491
492         /*
493          * Only update if we are coming from a stopped ticks mode
494          * (rcu_dyntick_sched.dynticks is even).
495          */
496         if (!in_interrupt() &&
497             (rdssp->dynticks & 0x1) == 0) {
498                 /*
499                  * The following might seem like we could have a race
500                  * with NMI/SMIs. But this really isn't a problem.
501                  * Here we do a read/modify/write, and the race happens
502                  * when an NMI/SMI comes in after the read and before
503                  * the write. But NMI/SMIs will increment this counter
504                  * twice before returning, so the zero bit will not
505                  * be corrupted by the NMI/SMI which is the most important
506                  * part.
507                  *
508                  * The only thing is that we would bring back the counter
509                  * to a postion that it was in during the NMI/SMI.
510                  * But the zero bit would be set, so the rest of the
511                  * counter would again be ignored.
512                  *
513                  * On return from the IRQ, the counter may have the zero
514                  * bit be 0 and the counter the same as the return from
515                  * the NMI/SMI. If the state machine was so unlucky to
516                  * see that, it still doesn't matter, since all
517                  * RCU read-side critical sections on this CPU would
518                  * have already completed.
519                  */
520                 rdssp->dynticks++;
521                 /*
522                  * The following memory barrier ensures that any
523                  * rcu_read_lock() primitives in the irq handler
524                  * are seen by other CPUs to follow the above
525                  * increment to rcu_dyntick_sched.dynticks. This is
526                  * required in order for other CPUs to correctly
527                  * determine when it is safe to advance the RCU
528                  * grace-period state machine.
529                  */
530                 smp_mb(); /* see above block comment. */
531                 /*
532                  * Since we can't determine the dynamic tick mode from
533                  * the rcu_dyntick_sched.dynticks after this routine,
534                  * we use a second flag to acknowledge that we came
535                  * from an idle state with ticks stopped.
536                  */
537                 per_cpu(rcu_update_flag, cpu)++;
538                 /*
539                  * If we take an NMI/SMI now, they will also increment
540                  * the rcu_update_flag, and will not update the
541                  * rcu_dyntick_sched.dynticks on exit. That is for
542                  * this IRQ to do.
543                  */
544         }
545 }
546
547 /**
548  * rcu_irq_exit - Called from exiting Hard irq context.
549  *
550  * If the CPU was idle with dynamic ticks active, update the
551  * rcu_dyntick_sched.dynticks to let the RCU handling be
552  * aware that the CPU is going back to idle with no ticks.
553  */
554 void rcu_irq_exit(void)
555 {
556         int cpu = smp_processor_id();
557         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
558
559         /*
560          * rcu_update_flag is set if we interrupted the CPU
561          * when it was idle with ticks stopped.
562          * Once this occurs, we keep track of interrupt nesting
563          * because a NMI/SMI could also come in, and we still
564          * only want the IRQ that started the increment of the
565          * rcu_dyntick_sched.dynticks to be the one that modifies
566          * it on exit.
567          */
568         if (per_cpu(rcu_update_flag, cpu)) {
569                 if (--per_cpu(rcu_update_flag, cpu))
570                         return;
571
572                 /* This must match the interrupt nesting */
573                 WARN_ON(in_interrupt());
574
575                 /*
576                  * If an NMI/SMI happens now we are still
577                  * protected by the rcu_dyntick_sched.dynticks being odd.
578                  */
579
580                 /*
581                  * The following memory barrier ensures that any
582                  * rcu_read_unlock() primitives in the irq handler
583                  * are seen by other CPUs to preceed the following
584                  * increment to rcu_dyntick_sched.dynticks. This
585                  * is required in order for other CPUs to determine
586                  * when it is safe to advance the RCU grace-period
587                  * state machine.
588                  */
589                 smp_mb(); /* see above block comment. */
590                 rdssp->dynticks++;
591                 WARN_ON(rdssp->dynticks & 0x1);
592         }
593 }
594
595 void rcu_nmi_enter(void)
596 {
597         rcu_irq_enter();
598 }
599
600 void rcu_nmi_exit(void)
601 {
602         rcu_irq_exit();
603 }
604
605 static void dyntick_save_progress_counter(int cpu)
606 {
607         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
608
609         rdssp->dynticks_snap = rdssp->dynticks;
610 }
611
612 static inline int
613 rcu_try_flip_waitack_needed(int cpu)
614 {
615         long curr;
616         long snap;
617         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
618
619         curr = rdssp->dynticks;
620         snap = rdssp->dynticks_snap;
621         smp_mb(); /* force ordering with cpu entering/leaving dynticks. */
622
623         /*
624          * If the CPU remained in dynticks mode for the entire time
625          * and didn't take any interrupts, NMIs, SMIs, or whatever,
626          * then it cannot be in the middle of an rcu_read_lock(), so
627          * the next rcu_read_lock() it executes must use the new value
628          * of the counter.  So we can safely pretend that this CPU
629          * already acknowledged the counter.
630          */
631
632         if ((curr == snap) && ((curr & 0x1) == 0))
633                 return 0;
634
635         /*
636          * If the CPU passed through or entered a dynticks idle phase with
637          * no active irq handlers, then, as above, we can safely pretend
638          * that this CPU already acknowledged the counter.
639          */
640
641         if ((curr - snap) > 2 || (curr & 0x1) == 0)
642                 return 0;
643
644         /* We need this CPU to explicitly acknowledge the counter flip. */
645
646         return 1;
647 }
648
649 static inline int
650 rcu_try_flip_waitmb_needed(int cpu)
651 {
652         long curr;
653         long snap;
654         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
655
656         curr = rdssp->dynticks;
657         snap = rdssp->dynticks_snap;
658         smp_mb(); /* force ordering with cpu entering/leaving dynticks. */
659
660         /*
661          * If the CPU remained in dynticks mode for the entire time
662          * and didn't take any interrupts, NMIs, SMIs, or whatever,
663          * then it cannot have executed an RCU read-side critical section
664          * during that time, so there is no need for it to execute a
665          * memory barrier.
666          */
667
668         if ((curr == snap) && ((curr & 0x1) == 0))
669                 return 0;
670
671         /*
672          * If the CPU either entered or exited an outermost interrupt,
673          * SMI, NMI, or whatever handler, then we know that it executed
674          * a memory barrier when doing so.  So we don't need another one.
675          */
676         if (curr != snap)
677                 return 0;
678
679         /* We need the CPU to execute a memory barrier. */
680
681         return 1;
682 }
683
684 static void dyntick_save_progress_counter_sched(int cpu)
685 {
686         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
687
688         rdssp->sched_dynticks_snap = rdssp->dynticks;
689 }
690
691 static int rcu_qsctr_inc_needed_dyntick(int cpu)
692 {
693         long curr;
694         long snap;
695         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
696
697         curr = rdssp->dynticks;
698         snap = rdssp->sched_dynticks_snap;
699         smp_mb(); /* force ordering with cpu entering/leaving dynticks. */
700
701         /*
702          * If the CPU remained in dynticks mode for the entire time
703          * and didn't take any interrupts, NMIs, SMIs, or whatever,
704          * then it cannot be in the middle of an rcu_read_lock(), so
705          * the next rcu_read_lock() it executes must use the new value
706          * of the counter.  Therefore, this CPU has been in a quiescent
707          * state the entire time, and we don't need to wait for it.
708          */
709
710         if ((curr == snap) && ((curr & 0x1) == 0))
711                 return 0;
712
713         /*
714          * If the CPU passed through or entered a dynticks idle phase with
715          * no active irq handlers, then, as above, this CPU has already
716          * passed through a quiescent state.
717          */
718
719         if ((curr - snap) > 2 || (snap & 0x1) == 0)
720                 return 0;
721
722         /* We need this CPU to go through a quiescent state. */
723
724         return 1;
725 }
726
727 #else /* !CONFIG_NO_HZ */
728
729 # define dyntick_save_progress_counter(cpu)             do { } while (0)
730 # define rcu_try_flip_waitack_needed(cpu)               (1)
731 # define rcu_try_flip_waitmb_needed(cpu)                (1)
732
733 # define dyntick_save_progress_counter_sched(cpu)       do { } while (0)
734 # define rcu_qsctr_inc_needed_dyntick(cpu)              (1)
735
736 #endif /* CONFIG_NO_HZ */
737
738 static void save_qsctr_sched(int cpu)
739 {
740         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
741
742         rdssp->sched_qs_snap = rdssp->sched_qs;
743 }
744
745 static inline int rcu_qsctr_inc_needed(int cpu)
746 {
747         struct rcu_dyntick_sched *rdssp = &per_cpu(rcu_dyntick_sched, cpu);
748
749         /*
750          * If there has been a quiescent state, no more need to wait
751          * on this CPU.
752          */
753
754         if (rdssp->sched_qs != rdssp->sched_qs_snap) {
755                 smp_mb(); /* force ordering with cpu entering schedule(). */
756                 return 0;
757         }
758
759         /* We need this CPU to go through a quiescent state. */
760
761         return 1;
762 }
763
764 /*
765  * Get here when RCU is idle.  Decide whether we need to
766  * move out of idle state, and return non-zero if so.
767  * "Straightforward" approach for the moment, might later
768  * use callback-list lengths, grace-period duration, or
769  * some such to determine when to exit idle state.
770  * Might also need a pre-idle test that does not acquire
771  * the lock, but let's get the simple case working first...
772  */
773
774 static int
775 rcu_try_flip_idle(void)
776 {
777         int cpu;
778
779         RCU_TRACE_ME(rcupreempt_trace_try_flip_i1);
780         if (!rcu_pending(smp_processor_id())) {
781                 RCU_TRACE_ME(rcupreempt_trace_try_flip_ie1);
782                 return 0;
783         }
784
785         /*
786          * Do the flip.
787          */
788
789         RCU_TRACE_ME(rcupreempt_trace_try_flip_g1);
790         rcu_ctrlblk.completed++;  /* stands in for rcu_try_flip_g2 */
791
792         /*
793          * Need a memory barrier so that other CPUs see the new
794          * counter value before they see the subsequent change of all
795          * the rcu_flip_flag instances to rcu_flipped.
796          */
797
798         smp_mb();       /* see above block comment. */
799
800         /* Now ask each CPU for acknowledgement of the flip. */
801
802         for_each_cpu(cpu, to_cpumask(rcu_cpu_online_map)) {
803                 per_cpu(rcu_flip_flag, cpu) = rcu_flipped;
804                 dyntick_save_progress_counter(cpu);
805         }
806
807         return 1;
808 }
809
810 /*
811  * Wait for CPUs to acknowledge the flip.
812  */
813
814 static int
815 rcu_try_flip_waitack(void)
816 {
817         int cpu;
818
819         RCU_TRACE_ME(rcupreempt_trace_try_flip_a1);
820         for_each_cpu(cpu, to_cpumask(rcu_cpu_online_map))
821                 if (rcu_try_flip_waitack_needed(cpu) &&
822                     per_cpu(rcu_flip_flag, cpu) != rcu_flip_seen) {
823                         RCU_TRACE_ME(rcupreempt_trace_try_flip_ae1);
824                         return 0;
825                 }
826
827         /*
828          * Make sure our checks above don't bleed into subsequent
829          * waiting for the sum of the counters to reach zero.
830          */
831
832         smp_mb();       /* see above block comment. */
833         RCU_TRACE_ME(rcupreempt_trace_try_flip_a2);
834         return 1;
835 }
836
837 /*
838  * Wait for collective ``last'' counter to reach zero,
839  * then tell all CPUs to do an end-of-grace-period memory barrier.
840  */
841
842 static int
843 rcu_try_flip_waitzero(void)
844 {
845         int cpu;
846         int lastidx = !(rcu_ctrlblk.completed & 0x1);
847         int sum = 0;
848
849         /* Check to see if the sum of the "last" counters is zero. */
850
851         RCU_TRACE_ME(rcupreempt_trace_try_flip_z1);
852         for_each_possible_cpu(cpu)
853                 sum += RCU_DATA_CPU(cpu)->rcu_flipctr[lastidx];
854         if (sum != 0) {
855                 RCU_TRACE_ME(rcupreempt_trace_try_flip_ze1);
856                 return 0;
857         }
858
859         /*
860          * This ensures that the other CPUs see the call for
861          * memory barriers -after- the sum to zero has been
862          * detected here
863          */
864         smp_mb();  /*  ^^^^^^^^^^^^ */
865
866         /* Call for a memory barrier from each CPU. */
867         for_each_cpu(cpu, to_cpumask(rcu_cpu_online_map)) {
868                 per_cpu(rcu_mb_flag, cpu) = rcu_mb_needed;
869                 dyntick_save_progress_counter(cpu);
870         }
871
872         RCU_TRACE_ME(rcupreempt_trace_try_flip_z2);
873         return 1;
874 }
875
876 /*
877  * Wait for all CPUs to do their end-of-grace-period memory barrier.
878  * Return 0 once all CPUs have done so.
879  */
880
881 static int
882 rcu_try_flip_waitmb(void)
883 {
884         int cpu;
885
886         RCU_TRACE_ME(rcupreempt_trace_try_flip_m1);
887         for_each_cpu(cpu, to_cpumask(rcu_cpu_online_map))
888                 if (rcu_try_flip_waitmb_needed(cpu) &&
889                     per_cpu(rcu_mb_flag, cpu) != rcu_mb_done) {
890                         RCU_TRACE_ME(rcupreempt_trace_try_flip_me1);
891                         return 0;
892                 }
893
894         smp_mb(); /* Ensure that the above checks precede any following flip. */
895         RCU_TRACE_ME(rcupreempt_trace_try_flip_m2);
896         return 1;
897 }
898
899 /*
900  * Attempt a single flip of the counters.  Remember, a single flip does
901  * -not- constitute a grace period.  Instead, the interval between
902  * at least GP_STAGES consecutive flips is a grace period.
903  *
904  * If anyone is nuts enough to run this CONFIG_PREEMPT_RCU implementation
905  * on a large SMP, they might want to use a hierarchical organization of
906  * the per-CPU-counter pairs.
907  */
908 static void rcu_try_flip(void)
909 {
910         unsigned long flags;
911
912         RCU_TRACE_ME(rcupreempt_trace_try_flip_1);
913         if (unlikely(!spin_trylock_irqsave(&rcu_ctrlblk.fliplock, flags))) {
914                 RCU_TRACE_ME(rcupreempt_trace_try_flip_e1);
915                 return;
916         }
917
918         /*
919          * Take the next transition(s) through the RCU grace-period
920          * flip-counter state machine.
921          */
922
923         switch (rcu_ctrlblk.rcu_try_flip_state) {
924         case rcu_try_flip_idle_state:
925                 if (rcu_try_flip_idle())
926                         rcu_ctrlblk.rcu_try_flip_state =
927                                 rcu_try_flip_waitack_state;
928                 break;
929         case rcu_try_flip_waitack_state:
930                 if (rcu_try_flip_waitack())
931                         rcu_ctrlblk.rcu_try_flip_state =
932                                 rcu_try_flip_waitzero_state;
933                 break;
934         case rcu_try_flip_waitzero_state:
935                 if (rcu_try_flip_waitzero())
936                         rcu_ctrlblk.rcu_try_flip_state =
937                                 rcu_try_flip_waitmb_state;
938                 break;
939         case rcu_try_flip_waitmb_state:
940                 if (rcu_try_flip_waitmb())
941                         rcu_ctrlblk.rcu_try_flip_state =
942                                 rcu_try_flip_idle_state;
943         }
944         spin_unlock_irqrestore(&rcu_ctrlblk.fliplock, flags);
945 }
946
947 /*
948  * Check to see if this CPU needs to do a memory barrier in order to
949  * ensure that any prior RCU read-side critical sections have committed
950  * their counter manipulations and critical-section memory references
951  * before declaring the grace period to be completed.
952  */
953 static void rcu_check_mb(int cpu)
954 {
955         if (per_cpu(rcu_mb_flag, cpu) == rcu_mb_needed) {
956                 smp_mb();  /* Ensure RCU read-side accesses are visible. */
957                 per_cpu(rcu_mb_flag, cpu) = rcu_mb_done;
958         }
959 }
960
961 void rcu_check_callbacks(int cpu, int user)
962 {
963         unsigned long flags;
964         struct rcu_data *rdp = RCU_DATA_CPU(cpu);
965
966         /*
967          * If this CPU took its interrupt from user mode or from the
968          * idle loop, and this is not a nested interrupt, then
969          * this CPU has to have exited all prior preept-disable
970          * sections of code.  So invoke rcu_sched_qs() to note this.
971          *
972          * The memory barrier is needed to handle the case where
973          * writes from a preempt-disable section of code get reordered
974          * into schedule() by this CPU's write buffer.  So the memory
975          * barrier makes sure that the rcu_sched_qs() is seen by other
976          * CPUs to happen after any such write.
977          */
978
979         if (user ||
980             (idle_cpu(cpu) && !in_softirq() &&
981              hardirq_count() <= (1 << HARDIRQ_SHIFT))) {
982                 smp_mb();       /* Guard against aggressive schedule(). */
983                 rcu_sched_qs(cpu);
984         }
985
986         rcu_check_mb(cpu);
987         if (rcu_ctrlblk.completed == rdp->completed)
988                 rcu_try_flip();
989         spin_lock_irqsave(&rdp->lock, flags);
990         RCU_TRACE_RDP(rcupreempt_trace_check_callbacks, rdp);
991         __rcu_advance_callbacks(rdp);
992         if (rdp->donelist == NULL) {
993                 spin_unlock_irqrestore(&rdp->lock, flags);
994         } else {
995                 spin_unlock_irqrestore(&rdp->lock, flags);
996                 raise_softirq(RCU_SOFTIRQ);
997         }
998 }
999
1000 /*
1001  * Needed by dynticks, to make sure all RCU processing has finished
1002  * when we go idle:
1003  */
1004 void rcu_advance_callbacks(int cpu, int user)
1005 {
1006         unsigned long flags;
1007         struct rcu_data *rdp = RCU_DATA_CPU(cpu);
1008
1009         if (rcu_ctrlblk.completed == rdp->completed) {
1010                 rcu_try_flip();
1011                 if (rcu_ctrlblk.completed == rdp->completed)
1012                         return;
1013         }
1014         spin_lock_irqsave(&rdp->lock, flags);
1015         RCU_TRACE_RDP(rcupreempt_trace_check_callbacks, rdp);
1016         __rcu_advance_callbacks(rdp);
1017         spin_unlock_irqrestore(&rdp->lock, flags);
1018 }
1019
1020 #ifdef CONFIG_HOTPLUG_CPU
1021 #define rcu_offline_cpu_enqueue(srclist, srctail, dstlist, dsttail) do { \
1022                 *dsttail = srclist; \
1023                 if (srclist != NULL) { \
1024                         dsttail = srctail; \
1025                         srclist = NULL; \
1026                         srctail = &srclist;\
1027                 } \
1028         } while (0)
1029
1030 void rcu_offline_cpu(int cpu)
1031 {
1032         int i;
1033         struct rcu_head *list = NULL;
1034         unsigned long flags;
1035         struct rcu_data *rdp = RCU_DATA_CPU(cpu);
1036         struct rcu_head *schedlist = NULL;
1037         struct rcu_head **schedtail = &schedlist;
1038         struct rcu_head **tail = &list;
1039
1040         /*
1041          * Remove all callbacks from the newly dead CPU, retaining order.
1042          * Otherwise rcu_barrier() will fail
1043          */
1044
1045         spin_lock_irqsave(&rdp->lock, flags);
1046         rcu_offline_cpu_enqueue(rdp->donelist, rdp->donetail, list, tail);
1047         for (i = GP_STAGES - 1; i >= 0; i--)
1048                 rcu_offline_cpu_enqueue(rdp->waitlist[i], rdp->waittail[i],
1049                                                 list, tail);
1050         rcu_offline_cpu_enqueue(rdp->nextlist, rdp->nexttail, list, tail);
1051         rcu_offline_cpu_enqueue(rdp->waitschedlist, rdp->waitschedtail,
1052                                 schedlist, schedtail);
1053         rcu_offline_cpu_enqueue(rdp->nextschedlist, rdp->nextschedtail,
1054                                 schedlist, schedtail);
1055         rdp->rcu_sched_sleeping = 0;
1056         spin_unlock_irqrestore(&rdp->lock, flags);
1057         rdp->waitlistcount = 0;
1058
1059         /* Disengage the newly dead CPU from the grace-period computation. */
1060
1061         spin_lock_irqsave(&rcu_ctrlblk.fliplock, flags);
1062         rcu_check_mb(cpu);
1063         if (per_cpu(rcu_flip_flag, cpu) == rcu_flipped) {
1064                 smp_mb();  /* Subsequent counter accesses must see new value */
1065                 per_cpu(rcu_flip_flag, cpu) = rcu_flip_seen;
1066                 smp_mb();  /* Subsequent RCU read-side critical sections */
1067                            /*  seen -after- acknowledgement. */
1068         }
1069
1070         cpumask_clear_cpu(cpu, to_cpumask(rcu_cpu_online_map));
1071
1072         spin_unlock_irqrestore(&rcu_ctrlblk.fliplock, flags);
1073
1074         /*
1075          * Place the removed callbacks on the current CPU's queue.
1076          * Make them all start a new grace period: simple approach,
1077          * in theory could starve a given set of callbacks, but
1078          * you would need to be doing some serious CPU hotplugging
1079          * to make this happen.  If this becomes a problem, adding
1080          * a synchronize_rcu() to the hotplug path would be a simple
1081          * fix.
1082          */
1083
1084         local_irq_save(flags);  /* disable preempt till we know what lock. */
1085         rdp = RCU_DATA_ME();
1086         spin_lock(&rdp->lock);
1087         *rdp->nexttail = list;
1088         if (list)
1089                 rdp->nexttail = tail;
1090         *rdp->nextschedtail = schedlist;
1091         if (schedlist)
1092                 rdp->nextschedtail = schedtail;
1093         spin_unlock_irqrestore(&rdp->lock, flags);
1094 }
1095
1096 #else /* #ifdef CONFIG_HOTPLUG_CPU */
1097
1098 void rcu_offline_cpu(int cpu)
1099 {
1100 }
1101
1102 #endif /* #else #ifdef CONFIG_HOTPLUG_CPU */
1103
1104 void __cpuinit rcu_online_cpu(int cpu)
1105 {
1106         unsigned long flags;
1107         struct rcu_data *rdp;
1108
1109         spin_lock_irqsave(&rcu_ctrlblk.fliplock, flags);
1110         cpumask_set_cpu(cpu, to_cpumask(rcu_cpu_online_map));
1111         spin_unlock_irqrestore(&rcu_ctrlblk.fliplock, flags);
1112
1113         /*
1114          * The rcu_sched grace-period processing might have bypassed
1115          * this CPU, given that it was not in the rcu_cpu_online_map
1116          * when the grace-period scan started.  This means that the
1117          * grace-period task might sleep.  So make sure that if this
1118          * should happen, the first callback posted to this CPU will
1119          * wake up the grace-period task if need be.
1120          */
1121
1122         rdp = RCU_DATA_CPU(cpu);
1123         spin_lock_irqsave(&rdp->lock, flags);
1124         rdp->rcu_sched_sleeping = 1;
1125         spin_unlock_irqrestore(&rdp->lock, flags);
1126 }
1127
1128 static void rcu_process_callbacks(struct softirq_action *unused)
1129 {
1130         unsigned long flags;
1131         struct rcu_head *next, *list;
1132         struct rcu_data *rdp;
1133
1134         local_irq_save(flags);
1135         rdp = RCU_DATA_ME();
1136         spin_lock(&rdp->lock);
1137         list = rdp->donelist;
1138         if (list == NULL) {
1139                 spin_unlock_irqrestore(&rdp->lock, flags);
1140                 return;
1141         }
1142         rdp->donelist = NULL;
1143         rdp->donetail = &rdp->donelist;
1144         RCU_TRACE_RDP(rcupreempt_trace_done_remove, rdp);
1145         spin_unlock_irqrestore(&rdp->lock, flags);
1146         while (list) {
1147                 next = list->next;
1148                 list->func(list);
1149                 list = next;
1150                 RCU_TRACE_ME(rcupreempt_trace_invoke);
1151         }
1152 }
1153
1154 void call_rcu(struct rcu_head *head, void (*func)(struct rcu_head *rcu))
1155 {
1156         unsigned long flags;
1157         struct rcu_data *rdp;
1158
1159         head->func = func;
1160         head->next = NULL;
1161         local_irq_save(flags);
1162         rdp = RCU_DATA_ME();
1163         spin_lock(&rdp->lock);
1164         __rcu_advance_callbacks(rdp);
1165         *rdp->nexttail = head;
1166         rdp->nexttail = &head->next;
1167         RCU_TRACE_RDP(rcupreempt_trace_next_add, rdp);
1168         spin_unlock_irqrestore(&rdp->lock, flags);
1169 }
1170 EXPORT_SYMBOL_GPL(call_rcu);
1171
1172 void call_rcu_sched(struct rcu_head *head, void (*func)(struct rcu_head *rcu))
1173 {
1174         unsigned long flags;
1175         struct rcu_data *rdp;
1176         int wake_gp = 0;
1177
1178         head->func = func;
1179         head->next = NULL;
1180         local_irq_save(flags);
1181         rdp = RCU_DATA_ME();
1182         spin_lock(&rdp->lock);
1183         *rdp->nextschedtail = head;
1184         rdp->nextschedtail = &head->next;
1185         if (rdp->rcu_sched_sleeping) {
1186
1187                 /* Grace-period processing might be sleeping... */
1188
1189                 rdp->rcu_sched_sleeping = 0;
1190                 wake_gp = 1;
1191         }
1192         spin_unlock_irqrestore(&rdp->lock, flags);
1193         if (wake_gp) {
1194
1195                 /* Wake up grace-period processing, unless someone beat us. */
1196
1197                 spin_lock_irqsave(&rcu_ctrlblk.schedlock, flags);
1198                 if (rcu_ctrlblk.sched_sleep != rcu_sched_sleeping)
1199                         wake_gp = 0;
1200                 rcu_ctrlblk.sched_sleep = rcu_sched_not_sleeping;
1201                 spin_unlock_irqrestore(&rcu_ctrlblk.schedlock, flags);
1202                 if (wake_gp)
1203                         wake_up_interruptible(&rcu_ctrlblk.sched_wq);
1204         }
1205 }
1206 EXPORT_SYMBOL_GPL(call_rcu_sched);
1207
1208 /*
1209  * Wait until all currently running preempt_disable() code segments
1210  * (including hardware-irq-disable segments) complete.  Note that
1211  * in -rt this does -not- necessarily result in all currently executing
1212  * interrupt -handlers- having completed.
1213  */
1214 void __synchronize_sched(void)
1215 {
1216         struct rcu_synchronize rcu;
1217
1218         if (num_online_cpus() == 1)
1219                 return;  /* blocking is gp if only one CPU! */
1220
1221         init_completion(&rcu.completion);
1222         /* Will wake me after RCU finished. */
1223         call_rcu_sched(&rcu.head, wakeme_after_rcu);
1224         /* Wait for it. */
1225         wait_for_completion(&rcu.completion);
1226 }
1227 EXPORT_SYMBOL_GPL(__synchronize_sched);
1228
1229 /*
1230  * kthread function that manages call_rcu_sched grace periods.
1231  */
1232 static int rcu_sched_grace_period(void *arg)
1233 {
1234         int couldsleep;         /* might sleep after current pass. */
1235         int couldsleepnext = 0; /* might sleep after next pass. */
1236         int cpu;
1237         unsigned long flags;
1238         struct rcu_data *rdp;
1239         int ret;
1240
1241         /*
1242          * Each pass through the following loop handles one
1243          * rcu_sched grace period cycle.
1244          */
1245         do {
1246                 /* Save each CPU's current state. */
1247
1248                 for_each_online_cpu(cpu) {
1249                         dyntick_save_progress_counter_sched(cpu);
1250                         save_qsctr_sched(cpu);
1251                 }
1252
1253                 /*
1254                  * Sleep for about an RCU grace-period's worth to
1255                  * allow better batching and to consume less CPU.
1256                  */
1257                 schedule_timeout_interruptible(RCU_SCHED_BATCH_TIME);
1258
1259                 /*
1260                  * If there was nothing to do last time, prepare to
1261                  * sleep at the end of the current grace period cycle.
1262                  */
1263                 couldsleep = couldsleepnext;
1264                 couldsleepnext = 1;
1265                 if (couldsleep) {
1266                         spin_lock_irqsave(&rcu_ctrlblk.schedlock, flags);
1267                         rcu_ctrlblk.sched_sleep = rcu_sched_sleep_prep;
1268                         spin_unlock_irqrestore(&rcu_ctrlblk.schedlock, flags);
1269                 }
1270
1271                 /*
1272                  * Wait on each CPU in turn to have either visited
1273                  * a quiescent state or been in dynticks-idle mode.
1274                  */
1275                 for_each_online_cpu(cpu) {
1276                         while (rcu_qsctr_inc_needed(cpu) &&
1277                                rcu_qsctr_inc_needed_dyntick(cpu)) {
1278                                 /* resched_cpu(cpu); @@@ */
1279                                 schedule_timeout_interruptible(1);
1280                         }
1281                 }
1282
1283                 /* Advance callbacks for each CPU.  */
1284
1285                 for_each_online_cpu(cpu) {
1286
1287                         rdp = RCU_DATA_CPU(cpu);
1288                         spin_lock_irqsave(&rdp->lock, flags);
1289
1290                         /*
1291                          * We are running on this CPU irq-disabled, so no
1292                          * CPU can go offline until we re-enable irqs.
1293                          * The current CPU might have already gone
1294                          * offline (between the for_each_offline_cpu and
1295                          * the spin_lock_irqsave), but in that case all its
1296                          * callback lists will be empty, so no harm done.
1297                          *
1298                          * Advance the callbacks!  We share normal RCU's
1299                          * donelist, since callbacks are invoked the
1300                          * same way in either case.
1301                          */
1302                         if (rdp->waitschedlist != NULL) {
1303                                 *rdp->donetail = rdp->waitschedlist;
1304                                 rdp->donetail = rdp->waitschedtail;
1305
1306                                 /*
1307                                  * Next rcu_check_callbacks() will
1308                                  * do the required raise_softirq().
1309                                  */
1310                         }
1311                         if (rdp->nextschedlist != NULL) {
1312                                 rdp->waitschedlist = rdp->nextschedlist;
1313                                 rdp->waitschedtail = rdp->nextschedtail;
1314                                 couldsleep = 0;
1315                                 couldsleepnext = 0;
1316                         } else {
1317                                 rdp->waitschedlist = NULL;
1318                                 rdp->waitschedtail = &rdp->waitschedlist;
1319                         }
1320                         rdp->nextschedlist = NULL;
1321                         rdp->nextschedtail = &rdp->nextschedlist;
1322
1323                         /* Mark sleep intention. */
1324
1325                         rdp->rcu_sched_sleeping = couldsleep;
1326
1327                         spin_unlock_irqrestore(&rdp->lock, flags);
1328                 }
1329
1330                 /* If we saw callbacks on the last scan, go deal with them. */
1331
1332                 if (!couldsleep)
1333                         continue;
1334
1335                 /* Attempt to block... */
1336
1337                 spin_lock_irqsave(&rcu_ctrlblk.schedlock, flags);
1338                 if (rcu_ctrlblk.sched_sleep != rcu_sched_sleep_prep) {
1339
1340                         /*
1341                          * Someone posted a callback after we scanned.
1342                          * Go take care of it.
1343                          */
1344                         spin_unlock_irqrestore(&rcu_ctrlblk.schedlock, flags);
1345                         couldsleepnext = 0;
1346                         continue;
1347                 }
1348
1349                 /* Block until the next person posts a callback. */
1350
1351                 rcu_ctrlblk.sched_sleep = rcu_sched_sleeping;
1352                 spin_unlock_irqrestore(&rcu_ctrlblk.schedlock, flags);
1353                 ret = 0; /* unused */
1354                 __wait_event_interruptible(rcu_ctrlblk.sched_wq,
1355                         rcu_ctrlblk.sched_sleep != rcu_sched_sleeping,
1356                         ret);
1357
1358                 couldsleepnext = 0;
1359
1360         } while (!kthread_should_stop());
1361
1362         return (0);
1363 }
1364
1365 /*
1366  * Check to see if any future RCU-related work will need to be done
1367  * by the current CPU, even if none need be done immediately, returning
1368  * 1 if so.  Assumes that notifiers would take care of handling any
1369  * outstanding requests from the RCU core.
1370  *
1371  * This function is part of the RCU implementation; it is -not-
1372  * an exported member of the RCU API.
1373  */
1374 int rcu_needs_cpu(int cpu)
1375 {
1376         struct rcu_data *rdp = RCU_DATA_CPU(cpu);
1377
1378         return (rdp->donelist != NULL ||
1379                 !!rdp->waitlistcount ||
1380                 rdp->nextlist != NULL ||
1381                 rdp->nextschedlist != NULL ||
1382                 rdp->waitschedlist != NULL);
1383 }
1384
1385 int rcu_pending(int cpu)
1386 {
1387         struct rcu_data *rdp = RCU_DATA_CPU(cpu);
1388
1389         /* The CPU has at least one callback queued somewhere. */
1390
1391         if (rdp->donelist != NULL ||
1392             !!rdp->waitlistcount ||
1393             rdp->nextlist != NULL ||
1394             rdp->nextschedlist != NULL ||
1395             rdp->waitschedlist != NULL)
1396                 return 1;
1397
1398         /* The RCU core needs an acknowledgement from this CPU. */
1399
1400         if ((per_cpu(rcu_flip_flag, cpu) == rcu_flipped) ||
1401             (per_cpu(rcu_mb_flag, cpu) == rcu_mb_needed))
1402                 return 1;
1403
1404         /* This CPU has fallen behind the global grace-period number. */
1405
1406         if (rdp->completed != rcu_ctrlblk.completed)
1407                 return 1;
1408
1409         /* Nothing needed from this CPU. */
1410
1411         return 0;
1412 }
1413
1414 int __cpuinit rcu_cpu_notify(struct notifier_block *self,
1415                              unsigned long action, void *hcpu)
1416 {
1417         long cpu = (long)hcpu;
1418
1419         switch (action) {
1420         case CPU_UP_PREPARE:
1421         case CPU_UP_PREPARE_FROZEN:
1422                 rcu_online_cpu(cpu);
1423                 break;
1424         case CPU_UP_CANCELED:
1425         case CPU_UP_CANCELED_FROZEN:
1426         case CPU_DEAD:
1427         case CPU_DEAD_FROZEN:
1428                 rcu_offline_cpu(cpu);
1429                 break;
1430         default:
1431                 break;
1432         }
1433         return NOTIFY_OK;
1434 }
1435
1436 void __init __rcu_init(void)
1437 {
1438         int cpu;
1439         int i;
1440         struct rcu_data *rdp;
1441
1442         printk(KERN_NOTICE "Preemptible RCU implementation.\n");
1443         for_each_possible_cpu(cpu) {
1444                 rdp = RCU_DATA_CPU(cpu);
1445                 spin_lock_init(&rdp->lock);
1446                 rdp->completed = 0;
1447                 rdp->waitlistcount = 0;
1448                 rdp->nextlist = NULL;
1449                 rdp->nexttail = &rdp->nextlist;
1450                 for (i = 0; i < GP_STAGES; i++) {
1451                         rdp->waitlist[i] = NULL;
1452                         rdp->waittail[i] = &rdp->waitlist[i];
1453                 }
1454                 rdp->donelist = NULL;
1455                 rdp->donetail = &rdp->donelist;
1456                 rdp->rcu_flipctr[0] = 0;
1457                 rdp->rcu_flipctr[1] = 0;
1458                 rdp->nextschedlist = NULL;
1459                 rdp->nextschedtail = &rdp->nextschedlist;
1460                 rdp->waitschedlist = NULL;
1461                 rdp->waitschedtail = &rdp->waitschedlist;
1462                 rdp->rcu_sched_sleeping = 0;
1463         }
1464         open_softirq(RCU_SOFTIRQ, rcu_process_callbacks);
1465 }
1466
1467 /*
1468  * Late-boot-time RCU initialization that must wait until after scheduler
1469  * has been initialized.
1470  */
1471 void __init rcu_init_sched(void)
1472 {
1473         rcu_sched_grace_period_task = kthread_run(rcu_sched_grace_period,
1474                                                   NULL,
1475                                                   "rcu_sched_grace_period");
1476         WARN_ON(IS_ERR(rcu_sched_grace_period_task));
1477 }
1478
1479 #ifdef CONFIG_RCU_TRACE
1480 long *rcupreempt_flipctr(int cpu)
1481 {
1482         return &RCU_DATA_CPU(cpu)->rcu_flipctr[0];
1483 }
1484 EXPORT_SYMBOL_GPL(rcupreempt_flipctr);
1485
1486 int rcupreempt_flip_flag(int cpu)
1487 {
1488         return per_cpu(rcu_flip_flag, cpu);
1489 }
1490 EXPORT_SYMBOL_GPL(rcupreempt_flip_flag);
1491
1492 int rcupreempt_mb_flag(int cpu)
1493 {
1494         return per_cpu(rcu_mb_flag, cpu);
1495 }
1496 EXPORT_SYMBOL_GPL(rcupreempt_mb_flag);
1497
1498 char *rcupreempt_try_flip_state_name(void)
1499 {
1500         return rcu_try_flip_state_names[rcu_ctrlblk.rcu_try_flip_state];
1501 }
1502 EXPORT_SYMBOL_GPL(rcupreempt_try_flip_state_name);
1503
1504 struct rcupreempt_trace *rcupreempt_trace_cpu(int cpu)
1505 {
1506         struct rcu_data *rdp = RCU_DATA_CPU(cpu);
1507
1508         return &rdp->trace;
1509 }
1510 EXPORT_SYMBOL_GPL(rcupreempt_trace_cpu);
1511
1512 #endif /* #ifdef RCU_TRACE */