xref: /openbmc/linux/kernel/rcu/tree_exp.h (revision fc28ab18)
1 /*
2  * RCU expedited grace periods
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, you can access it online at
16  * http://www.gnu.org/licenses/gpl-2.0.html.
17  *
18  * Copyright IBM Corporation, 2016
19  *
20  * Authors: Paul E. McKenney <paulmck@linux.vnet.ibm.com>
21  */
22 
23 /* Wrapper functions for expedited grace periods.  */
24 static void rcu_exp_gp_seq_start(struct rcu_state *rsp)
25 {
26 	rcu_seq_start(&rsp->expedited_sequence);
27 }
28 static void rcu_exp_gp_seq_end(struct rcu_state *rsp)
29 {
30 	rcu_seq_end(&rsp->expedited_sequence);
31 	smp_mb(); /* Ensure that consecutive grace periods serialize. */
32 }
33 static unsigned long rcu_exp_gp_seq_snap(struct rcu_state *rsp)
34 {
35 	unsigned long s;
36 
37 	smp_mb(); /* Caller's modifications seen first by other CPUs. */
38 	s = rcu_seq_snap(&rsp->expedited_sequence);
39 	trace_rcu_exp_grace_period(rsp->name, s, TPS("snap"));
40 	return s;
41 }
42 static bool rcu_exp_gp_seq_done(struct rcu_state *rsp, unsigned long s)
43 {
44 	return rcu_seq_done(&rsp->expedited_sequence, s);
45 }
46 
47 /*
48  * Reset the ->expmaskinit values in the rcu_node tree to reflect any
49  * recent CPU-online activity.  Note that these masks are not cleared
50  * when CPUs go offline, so they reflect the union of all CPUs that have
51  * ever been online.  This means that this function normally takes its
52  * no-work-to-do fastpath.
53  */
54 static void sync_exp_reset_tree_hotplug(struct rcu_state *rsp)
55 {
56 	bool done;
57 	unsigned long flags;
58 	unsigned long mask;
59 	unsigned long oldmask;
60 	int ncpus = READ_ONCE(rsp->ncpus);
61 	struct rcu_node *rnp;
62 	struct rcu_node *rnp_up;
63 
64 	/* If no new CPUs onlined since last time, nothing to do. */
65 	if (likely(ncpus == rsp->ncpus_snap))
66 		return;
67 	rsp->ncpus_snap = ncpus;
68 
69 	/*
70 	 * Each pass through the following loop propagates newly onlined
71 	 * CPUs for the current rcu_node structure up the rcu_node tree.
72 	 */
73 	rcu_for_each_leaf_node(rsp, rnp) {
74 		raw_spin_lock_irqsave_rcu_node(rnp, flags);
75 		if (rnp->expmaskinit == rnp->expmaskinitnext) {
76 			raw_spin_unlock_irqrestore_rcu_node(rnp, flags);
77 			continue;  /* No new CPUs, nothing to do. */
78 		}
79 
80 		/* Update this node's mask, track old value for propagation. */
81 		oldmask = rnp->expmaskinit;
82 		rnp->expmaskinit = rnp->expmaskinitnext;
83 		raw_spin_unlock_irqrestore_rcu_node(rnp, flags);
84 
85 		/* If was already nonzero, nothing to propagate. */
86 		if (oldmask)
87 			continue;
88 
89 		/* Propagate the new CPU up the tree. */
90 		mask = rnp->grpmask;
91 		rnp_up = rnp->parent;
92 		done = false;
93 		while (rnp_up) {
94 			raw_spin_lock_irqsave_rcu_node(rnp_up, flags);
95 			if (rnp_up->expmaskinit)
96 				done = true;
97 			rnp_up->expmaskinit |= mask;
98 			raw_spin_unlock_irqrestore_rcu_node(rnp_up, flags);
99 			if (done)
100 				break;
101 			mask = rnp_up->grpmask;
102 			rnp_up = rnp_up->parent;
103 		}
104 	}
105 }
106 
107 /*
108  * Reset the ->expmask values in the rcu_node tree in preparation for
109  * a new expedited grace period.
110  */
111 static void __maybe_unused sync_exp_reset_tree(struct rcu_state *rsp)
112 {
113 	unsigned long flags;
114 	struct rcu_node *rnp;
115 
116 	sync_exp_reset_tree_hotplug(rsp);
117 	rcu_for_each_node_breadth_first(rsp, rnp) {
118 		raw_spin_lock_irqsave_rcu_node(rnp, flags);
119 		WARN_ON_ONCE(rnp->expmask);
120 		rnp->expmask = rnp->expmaskinit;
121 		raw_spin_unlock_irqrestore_rcu_node(rnp, flags);
122 	}
123 }
124 
125 /*
126  * Return non-zero if there is no RCU expedited grace period in progress
127  * for the specified rcu_node structure, in other words, if all CPUs and
128  * tasks covered by the specified rcu_node structure have done their bit
129  * for the current expedited grace period.  Works only for preemptible
130  * RCU -- other RCU implementation use other means.
131  *
132  * Caller must hold the rcu_state's exp_mutex.
133  */
134 static int sync_rcu_preempt_exp_done(struct rcu_node *rnp)
135 {
136 	return rnp->exp_tasks == NULL &&
137 	       READ_ONCE(rnp->expmask) == 0;
138 }
139 
140 /*
141  * Report the exit from RCU read-side critical section for the last task
142  * that queued itself during or before the current expedited preemptible-RCU
143  * grace period.  This event is reported either to the rcu_node structure on
144  * which the task was queued or to one of that rcu_node structure's ancestors,
145  * recursively up the tree.  (Calm down, calm down, we do the recursion
146  * iteratively!)
147  *
148  * Caller must hold the rcu_state's exp_mutex and the specified rcu_node
149  * structure's ->lock.
150  */
151 static void __rcu_report_exp_rnp(struct rcu_state *rsp, struct rcu_node *rnp,
152 				 bool wake, unsigned long flags)
153 	__releases(rnp->lock)
154 {
155 	unsigned long mask;
156 
157 	for (;;) {
158 		if (!sync_rcu_preempt_exp_done(rnp)) {
159 			if (!rnp->expmask)
160 				rcu_initiate_boost(rnp, flags);
161 			else
162 				raw_spin_unlock_irqrestore_rcu_node(rnp, flags);
163 			break;
164 		}
165 		if (rnp->parent == NULL) {
166 			raw_spin_unlock_irqrestore_rcu_node(rnp, flags);
167 			if (wake) {
168 				smp_mb(); /* EGP done before wake_up(). */
169 				swake_up(&rsp->expedited_wq);
170 			}
171 			break;
172 		}
173 		mask = rnp->grpmask;
174 		raw_spin_unlock_rcu_node(rnp); /* irqs remain disabled */
175 		rnp = rnp->parent;
176 		raw_spin_lock_rcu_node(rnp); /* irqs already disabled */
177 		WARN_ON_ONCE(!(rnp->expmask & mask));
178 		rnp->expmask &= ~mask;
179 	}
180 }
181 
182 /*
183  * Report expedited quiescent state for specified node.  This is a
184  * lock-acquisition wrapper function for __rcu_report_exp_rnp().
185  *
186  * Caller must hold the rcu_state's exp_mutex.
187  */
188 static void __maybe_unused rcu_report_exp_rnp(struct rcu_state *rsp,
189 					      struct rcu_node *rnp, bool wake)
190 {
191 	unsigned long flags;
192 
193 	raw_spin_lock_irqsave_rcu_node(rnp, flags);
194 	__rcu_report_exp_rnp(rsp, rnp, wake, flags);
195 }
196 
197 /*
198  * Report expedited quiescent state for multiple CPUs, all covered by the
199  * specified leaf rcu_node structure.  Caller must hold the rcu_state's
200  * exp_mutex.
201  */
202 static void rcu_report_exp_cpu_mult(struct rcu_state *rsp, struct rcu_node *rnp,
203 				    unsigned long mask, bool wake)
204 {
205 	unsigned long flags;
206 
207 	raw_spin_lock_irqsave_rcu_node(rnp, flags);
208 	if (!(rnp->expmask & mask)) {
209 		raw_spin_unlock_irqrestore_rcu_node(rnp, flags);
210 		return;
211 	}
212 	rnp->expmask &= ~mask;
213 	__rcu_report_exp_rnp(rsp, rnp, wake, flags); /* Releases rnp->lock. */
214 }
215 
216 /*
217  * Report expedited quiescent state for specified rcu_data (CPU).
218  */
219 static void rcu_report_exp_rdp(struct rcu_state *rsp, struct rcu_data *rdp,
220 			       bool wake)
221 {
222 	rcu_report_exp_cpu_mult(rsp, rdp->mynode, rdp->grpmask, wake);
223 }
224 
225 /* Common code for synchronize_{rcu,sched}_expedited() work-done checking. */
226 static bool sync_exp_work_done(struct rcu_state *rsp, atomic_long_t *stat,
227 			       unsigned long s)
228 {
229 	if (rcu_exp_gp_seq_done(rsp, s)) {
230 		trace_rcu_exp_grace_period(rsp->name, s, TPS("done"));
231 		/* Ensure test happens before caller kfree(). */
232 		smp_mb__before_atomic(); /* ^^^ */
233 		atomic_long_inc(stat);
234 		return true;
235 	}
236 	return false;
237 }
238 
239 /*
240  * Funnel-lock acquisition for expedited grace periods.  Returns true
241  * if some other task completed an expedited grace period that this task
242  * can piggy-back on, and with no mutex held.  Otherwise, returns false
243  * with the mutex held, indicating that the caller must actually do the
244  * expedited grace period.
245  */
246 static bool exp_funnel_lock(struct rcu_state *rsp, unsigned long s)
247 {
248 	struct rcu_data *rdp = per_cpu_ptr(rsp->rda, raw_smp_processor_id());
249 	struct rcu_node *rnp = rdp->mynode;
250 	struct rcu_node *rnp_root = rcu_get_root(rsp);
251 
252 	/* Low-contention fastpath. */
253 	if (ULONG_CMP_LT(READ_ONCE(rnp->exp_seq_rq), s) &&
254 	    (rnp == rnp_root ||
255 	     ULONG_CMP_LT(READ_ONCE(rnp_root->exp_seq_rq), s)) &&
256 	    mutex_trylock(&rsp->exp_mutex))
257 		goto fastpath;
258 
259 	/*
260 	 * Each pass through the following loop works its way up
261 	 * the rcu_node tree, returning if others have done the work or
262 	 * otherwise falls through to acquire rsp->exp_mutex.  The mapping
263 	 * from CPU to rcu_node structure can be inexact, as it is just
264 	 * promoting locality and is not strictly needed for correctness.
265 	 */
266 	for (; rnp != NULL; rnp = rnp->parent) {
267 		if (sync_exp_work_done(rsp, &rdp->exp_workdone1, s))
268 			return true;
269 
270 		/* Work not done, either wait here or go up. */
271 		spin_lock(&rnp->exp_lock);
272 		if (ULONG_CMP_GE(rnp->exp_seq_rq, s)) {
273 
274 			/* Someone else doing GP, so wait for them. */
275 			spin_unlock(&rnp->exp_lock);
276 			trace_rcu_exp_funnel_lock(rsp->name, rnp->level,
277 						  rnp->grplo, rnp->grphi,
278 						  TPS("wait"));
279 			wait_event(rnp->exp_wq[(s >> 1) & 0x3],
280 				   sync_exp_work_done(rsp,
281 						      &rdp->exp_workdone2, s));
282 			return true;
283 		}
284 		rnp->exp_seq_rq = s; /* Followers can wait on us. */
285 		spin_unlock(&rnp->exp_lock);
286 		trace_rcu_exp_funnel_lock(rsp->name, rnp->level, rnp->grplo,
287 					  rnp->grphi, TPS("nxtlvl"));
288 	}
289 	mutex_lock(&rsp->exp_mutex);
290 fastpath:
291 	if (sync_exp_work_done(rsp, &rdp->exp_workdone3, s)) {
292 		mutex_unlock(&rsp->exp_mutex);
293 		return true;
294 	}
295 	rcu_exp_gp_seq_start(rsp);
296 	trace_rcu_exp_grace_period(rsp->name, s, TPS("start"));
297 	return false;
298 }
299 
300 /* Invoked on each online non-idle CPU for expedited quiescent state. */
301 static void sync_sched_exp_handler(void *data)
302 {
303 	struct rcu_data *rdp;
304 	struct rcu_node *rnp;
305 	struct rcu_state *rsp = data;
306 
307 	rdp = this_cpu_ptr(rsp->rda);
308 	rnp = rdp->mynode;
309 	if (!(READ_ONCE(rnp->expmask) & rdp->grpmask) ||
310 	    __this_cpu_read(rcu_sched_data.cpu_no_qs.b.exp))
311 		return;
312 	if (rcu_is_cpu_rrupt_from_idle()) {
313 		rcu_report_exp_rdp(&rcu_sched_state,
314 				   this_cpu_ptr(&rcu_sched_data), true);
315 		return;
316 	}
317 	__this_cpu_write(rcu_sched_data.cpu_no_qs.b.exp, true);
318 	resched_cpu(smp_processor_id());
319 }
320 
321 /* Send IPI for expedited cleanup if needed at end of CPU-hotplug operation. */
322 static void sync_sched_exp_online_cleanup(int cpu)
323 {
324 	struct rcu_data *rdp;
325 	int ret;
326 	struct rcu_node *rnp;
327 	struct rcu_state *rsp = &rcu_sched_state;
328 
329 	rdp = per_cpu_ptr(rsp->rda, cpu);
330 	rnp = rdp->mynode;
331 	if (!(READ_ONCE(rnp->expmask) & rdp->grpmask))
332 		return;
333 	ret = smp_call_function_single(cpu, sync_sched_exp_handler, rsp, 0);
334 	WARN_ON_ONCE(ret);
335 }
336 
337 /*
338  * Select the nodes that the upcoming expedited grace period needs
339  * to wait for.
340  */
341 static void sync_rcu_exp_select_cpus(struct rcu_state *rsp,
342 				     smp_call_func_t func)
343 {
344 	int cpu;
345 	unsigned long flags;
346 	unsigned long mask_ofl_test;
347 	unsigned long mask_ofl_ipi;
348 	int ret;
349 	struct rcu_node *rnp;
350 
351 	sync_exp_reset_tree(rsp);
352 	rcu_for_each_leaf_node(rsp, rnp) {
353 		raw_spin_lock_irqsave_rcu_node(rnp, flags);
354 
355 		/* Each pass checks a CPU for identity, offline, and idle. */
356 		mask_ofl_test = 0;
357 		for_each_leaf_node_possible_cpu(rnp, cpu) {
358 			struct rcu_data *rdp = per_cpu_ptr(rsp->rda, cpu);
359 			struct rcu_dynticks *rdtp = &per_cpu(rcu_dynticks, cpu);
360 
361 			rdp->exp_dynticks_snap =
362 				atomic_add_return(0, &rdtp->dynticks);
363 			if (raw_smp_processor_id() == cpu ||
364 			    !(rdp->exp_dynticks_snap & 0x1) ||
365 			    !(rnp->qsmaskinitnext & rdp->grpmask))
366 				mask_ofl_test |= rdp->grpmask;
367 		}
368 		mask_ofl_ipi = rnp->expmask & ~mask_ofl_test;
369 
370 		/*
371 		 * Need to wait for any blocked tasks as well.  Note that
372 		 * additional blocking tasks will also block the expedited
373 		 * GP until such time as the ->expmask bits are cleared.
374 		 */
375 		if (rcu_preempt_has_tasks(rnp))
376 			rnp->exp_tasks = rnp->blkd_tasks.next;
377 		raw_spin_unlock_irqrestore_rcu_node(rnp, flags);
378 
379 		/* IPI the remaining CPUs for expedited quiescent state. */
380 		for_each_leaf_node_possible_cpu(rnp, cpu) {
381 			unsigned long mask = leaf_node_cpu_bit(rnp, cpu);
382 			struct rcu_data *rdp = per_cpu_ptr(rsp->rda, cpu);
383 			struct rcu_dynticks *rdtp = &per_cpu(rcu_dynticks, cpu);
384 
385 			if (!(mask_ofl_ipi & mask))
386 				continue;
387 retry_ipi:
388 			if (atomic_add_return(0, &rdtp->dynticks) !=
389 			    rdp->exp_dynticks_snap) {
390 				mask_ofl_test |= mask;
391 				continue;
392 			}
393 			ret = smp_call_function_single(cpu, func, rsp, 0);
394 			if (!ret) {
395 				mask_ofl_ipi &= ~mask;
396 				continue;
397 			}
398 			/* Failed, raced with CPU hotplug operation. */
399 			raw_spin_lock_irqsave_rcu_node(rnp, flags);
400 			if ((rnp->qsmaskinitnext & mask) &&
401 			    (rnp->expmask & mask)) {
402 				/* Online, so delay for a bit and try again. */
403 				raw_spin_unlock_irqrestore_rcu_node(rnp, flags);
404 				schedule_timeout_uninterruptible(1);
405 				goto retry_ipi;
406 			}
407 			/* CPU really is offline, so we can ignore it. */
408 			if (!(rnp->expmask & mask))
409 				mask_ofl_ipi &= ~mask;
410 			raw_spin_unlock_irqrestore_rcu_node(rnp, flags);
411 		}
412 		/* Report quiescent states for those that went offline. */
413 		mask_ofl_test |= mask_ofl_ipi;
414 		if (mask_ofl_test)
415 			rcu_report_exp_cpu_mult(rsp, rnp, mask_ofl_test, false);
416 	}
417 }
418 
419 static void synchronize_sched_expedited_wait(struct rcu_state *rsp)
420 {
421 	int cpu;
422 	unsigned long jiffies_stall;
423 	unsigned long jiffies_start;
424 	unsigned long mask;
425 	int ndetected;
426 	struct rcu_node *rnp;
427 	struct rcu_node *rnp_root = rcu_get_root(rsp);
428 	int ret;
429 
430 	jiffies_stall = rcu_jiffies_till_stall_check();
431 	jiffies_start = jiffies;
432 
433 	for (;;) {
434 		ret = swait_event_timeout(
435 				rsp->expedited_wq,
436 				sync_rcu_preempt_exp_done(rnp_root),
437 				jiffies_stall);
438 		if (ret > 0 || sync_rcu_preempt_exp_done(rnp_root))
439 			return;
440 		WARN_ON(ret < 0);  /* workqueues should not be signaled. */
441 		if (rcu_cpu_stall_suppress)
442 			continue;
443 		panic_on_rcu_stall();
444 		pr_err("INFO: %s detected expedited stalls on CPUs/tasks: {",
445 		       rsp->name);
446 		ndetected = 0;
447 		rcu_for_each_leaf_node(rsp, rnp) {
448 			ndetected += rcu_print_task_exp_stall(rnp);
449 			for_each_leaf_node_possible_cpu(rnp, cpu) {
450 				struct rcu_data *rdp;
451 
452 				mask = leaf_node_cpu_bit(rnp, cpu);
453 				if (!(rnp->expmask & mask))
454 					continue;
455 				ndetected++;
456 				rdp = per_cpu_ptr(rsp->rda, cpu);
457 				pr_cont(" %d-%c%c%c", cpu,
458 					"O."[!!cpu_online(cpu)],
459 					"o."[!!(rdp->grpmask & rnp->expmaskinit)],
460 					"N."[!!(rdp->grpmask & rnp->expmaskinitnext)]);
461 			}
462 		}
463 		pr_cont(" } %lu jiffies s: %lu root: %#lx/%c\n",
464 			jiffies - jiffies_start, rsp->expedited_sequence,
465 			rnp_root->expmask, ".T"[!!rnp_root->exp_tasks]);
466 		if (ndetected) {
467 			pr_err("blocking rcu_node structures:");
468 			rcu_for_each_node_breadth_first(rsp, rnp) {
469 				if (rnp == rnp_root)
470 					continue; /* printed unconditionally */
471 				if (sync_rcu_preempt_exp_done(rnp))
472 					continue;
473 				pr_cont(" l=%u:%d-%d:%#lx/%c",
474 					rnp->level, rnp->grplo, rnp->grphi,
475 					rnp->expmask,
476 					".T"[!!rnp->exp_tasks]);
477 			}
478 			pr_cont("\n");
479 		}
480 		rcu_for_each_leaf_node(rsp, rnp) {
481 			for_each_leaf_node_possible_cpu(rnp, cpu) {
482 				mask = leaf_node_cpu_bit(rnp, cpu);
483 				if (!(rnp->expmask & mask))
484 					continue;
485 				dump_cpu_task(cpu);
486 			}
487 		}
488 		jiffies_stall = 3 * rcu_jiffies_till_stall_check() + 3;
489 	}
490 }
491 
492 /*
493  * Wait for the current expedited grace period to complete, and then
494  * wake up everyone who piggybacked on the just-completed expedited
495  * grace period.  Also update all the ->exp_seq_rq counters as needed
496  * in order to avoid counter-wrap problems.
497  */
498 static void rcu_exp_wait_wake(struct rcu_state *rsp, unsigned long s)
499 {
500 	struct rcu_node *rnp;
501 
502 	synchronize_sched_expedited_wait(rsp);
503 	rcu_exp_gp_seq_end(rsp);
504 	trace_rcu_exp_grace_period(rsp->name, s, TPS("end"));
505 
506 	/*
507 	 * Switch over to wakeup mode, allowing the next GP, but -only- the
508 	 * next GP, to proceed.
509 	 */
510 	mutex_lock(&rsp->exp_wake_mutex);
511 
512 	rcu_for_each_node_breadth_first(rsp, rnp) {
513 		if (ULONG_CMP_LT(READ_ONCE(rnp->exp_seq_rq), s)) {
514 			spin_lock(&rnp->exp_lock);
515 			/* Recheck, avoid hang in case someone just arrived. */
516 			if (ULONG_CMP_LT(rnp->exp_seq_rq, s))
517 				rnp->exp_seq_rq = s;
518 			spin_unlock(&rnp->exp_lock);
519 		}
520 		wake_up_all(&rnp->exp_wq[(rsp->expedited_sequence >> 1) & 0x3]);
521 	}
522 	trace_rcu_exp_grace_period(rsp->name, s, TPS("endwake"));
523 	mutex_unlock(&rsp->exp_wake_mutex);
524 }
525 
526 /* Let the workqueue handler know what it is supposed to do. */
527 struct rcu_exp_work {
528 	smp_call_func_t rew_func;
529 	struct rcu_state *rew_rsp;
530 	unsigned long rew_s;
531 	struct work_struct rew_work;
532 };
533 
534 /*
535  * Common code to drive an expedited grace period forward, used by
536  * workqueues and mid-boot-time tasks.
537  */
538 static void rcu_exp_sel_wait_wake(struct rcu_state *rsp,
539 				  smp_call_func_t func, unsigned long s)
540 {
541 	/* Initialize the rcu_node tree in preparation for the wait. */
542 	sync_rcu_exp_select_cpus(rsp, func);
543 
544 	/* Wait and clean up, including waking everyone. */
545 	rcu_exp_wait_wake(rsp, s);
546 }
547 
548 /*
549  * Work-queue handler to drive an expedited grace period forward.
550  */
551 static void wait_rcu_exp_gp(struct work_struct *wp)
552 {
553 	struct rcu_exp_work *rewp;
554 
555 	rewp = container_of(wp, struct rcu_exp_work, rew_work);
556 	rcu_exp_sel_wait_wake(rewp->rew_rsp, rewp->rew_func, rewp->rew_s);
557 }
558 
559 /*
560  * Given an rcu_state pointer and a smp_call_function() handler, kick
561  * off the specified flavor of expedited grace period.
562  */
563 static void _synchronize_rcu_expedited(struct rcu_state *rsp,
564 				       smp_call_func_t func)
565 {
566 	struct rcu_data *rdp;
567 	struct rcu_exp_work rew;
568 	struct rcu_node *rnp;
569 	unsigned long s;
570 
571 	/* If expedited grace periods are prohibited, fall back to normal. */
572 	if (rcu_gp_is_normal()) {
573 		wait_rcu_gp(rsp->call);
574 		return;
575 	}
576 
577 	/* Take a snapshot of the sequence number.  */
578 	s = rcu_exp_gp_seq_snap(rsp);
579 	if (exp_funnel_lock(rsp, s))
580 		return;  /* Someone else did our work for us. */
581 
582 	/* Ensure that load happens before action based on it. */
583 	if (unlikely(rcu_scheduler_active == RCU_SCHEDULER_INIT)) {
584 		/* Direct call during scheduler init and early_initcalls(). */
585 		rcu_exp_sel_wait_wake(rsp, func, s);
586 	} else {
587 		/* Marshall arguments & schedule the expedited grace period. */
588 		rew.rew_func = func;
589 		rew.rew_rsp = rsp;
590 		rew.rew_s = s;
591 		INIT_WORK_ONSTACK(&rew.rew_work, wait_rcu_exp_gp);
592 		schedule_work(&rew.rew_work);
593 	}
594 
595 	/* Wait for expedited grace period to complete. */
596 	rdp = per_cpu_ptr(rsp->rda, raw_smp_processor_id());
597 	rnp = rcu_get_root(rsp);
598 	wait_event(rnp->exp_wq[(s >> 1) & 0x3],
599 		   sync_exp_work_done(rsp,
600 				      &rdp->exp_workdone0, s));
601 
602 	/* Let the next expedited grace period start. */
603 	mutex_unlock(&rsp->exp_mutex);
604 }
605 
606 /**
607  * synchronize_sched_expedited - Brute-force RCU-sched grace period
608  *
609  * Wait for an RCU-sched grace period to elapse, but use a "big hammer"
610  * approach to force the grace period to end quickly.  This consumes
611  * significant time on all CPUs and is unfriendly to real-time workloads,
612  * so is thus not recommended for any sort of common-case code.  In fact,
613  * if you are using synchronize_sched_expedited() in a loop, please
614  * restructure your code to batch your updates, and then use a single
615  * synchronize_sched() instead.
616  *
617  * This implementation can be thought of as an application of sequence
618  * locking to expedited grace periods, but using the sequence counter to
619  * determine when someone else has already done the work instead of for
620  * retrying readers.
621  */
622 void synchronize_sched_expedited(void)
623 {
624 	struct rcu_state *rsp = &rcu_sched_state;
625 
626 	/* If only one CPU, this is automatically a grace period. */
627 	if (rcu_blocking_is_gp())
628 		return;
629 
630 	_synchronize_rcu_expedited(rsp, sync_sched_exp_handler);
631 }
632 EXPORT_SYMBOL_GPL(synchronize_sched_expedited);
633 
634 #ifdef CONFIG_PREEMPT_RCU
635 
636 /*
637  * Remote handler for smp_call_function_single().  If there is an
638  * RCU read-side critical section in effect, request that the
639  * next rcu_read_unlock() record the quiescent state up the
640  * ->expmask fields in the rcu_node tree.  Otherwise, immediately
641  * report the quiescent state.
642  */
643 static void sync_rcu_exp_handler(void *info)
644 {
645 	struct rcu_data *rdp;
646 	struct rcu_state *rsp = info;
647 	struct task_struct *t = current;
648 
649 	/*
650 	 * Within an RCU read-side critical section, request that the next
651 	 * rcu_read_unlock() report.  Unless this RCU read-side critical
652 	 * section has already blocked, in which case it is already set
653 	 * up for the expedited grace period to wait on it.
654 	 */
655 	if (t->rcu_read_lock_nesting > 0 &&
656 	    !t->rcu_read_unlock_special.b.blocked) {
657 		t->rcu_read_unlock_special.b.exp_need_qs = true;
658 		return;
659 	}
660 
661 	/*
662 	 * We are either exiting an RCU read-side critical section (negative
663 	 * values of t->rcu_read_lock_nesting) or are not in one at all
664 	 * (zero value of t->rcu_read_lock_nesting).  Or we are in an RCU
665 	 * read-side critical section that blocked before this expedited
666 	 * grace period started.  Either way, we can immediately report
667 	 * the quiescent state.
668 	 */
669 	rdp = this_cpu_ptr(rsp->rda);
670 	rcu_report_exp_rdp(rsp, rdp, true);
671 }
672 
673 /**
674  * synchronize_rcu_expedited - Brute-force RCU grace period
675  *
676  * Wait for an RCU-preempt grace period, but expedite it.  The basic
677  * idea is to IPI all non-idle non-nohz online CPUs.  The IPI handler
678  * checks whether the CPU is in an RCU-preempt critical section, and
679  * if so, it sets a flag that causes the outermost rcu_read_unlock()
680  * to report the quiescent state.  On the other hand, if the CPU is
681  * not in an RCU read-side critical section, the IPI handler reports
682  * the quiescent state immediately.
683  *
684  * Although this is a greate improvement over previous expedited
685  * implementations, it is still unfriendly to real-time workloads, so is
686  * thus not recommended for any sort of common-case code.  In fact, if
687  * you are using synchronize_rcu_expedited() in a loop, please restructure
688  * your code to batch your updates, and then Use a single synchronize_rcu()
689  * instead.
690  */
691 void synchronize_rcu_expedited(void)
692 {
693 	struct rcu_state *rsp = rcu_state_p;
694 
695 	if (rcu_scheduler_active == RCU_SCHEDULER_INACTIVE)
696 		return;
697 	_synchronize_rcu_expedited(rsp, sync_rcu_exp_handler);
698 }
699 EXPORT_SYMBOL_GPL(synchronize_rcu_expedited);
700 
701 #else /* #ifdef CONFIG_PREEMPT_RCU */
702 
703 /*
704  * Wait for an rcu-preempt grace period, but make it happen quickly.
705  * But because preemptible RCU does not exist, map to rcu-sched.
706  */
707 void synchronize_rcu_expedited(void)
708 {
709 	synchronize_sched_expedited();
710 }
711 EXPORT_SYMBOL_GPL(synchronize_rcu_expedited);
712 
713 #endif /* #else #ifdef CONFIG_PREEMPT_RCU */
714 
715 /*
716  * Switch to run-time mode once Tree RCU has fully initialized.
717  */
718 static int __init rcu_exp_runtime_mode(void)
719 {
720 	rcu_test_sync_prims();
721 	rcu_scheduler_active = RCU_SCHEDULER_RUNNING;
722 	rcu_test_sync_prims();
723 	return 0;
724 }
725 core_initcall(rcu_exp_runtime_mode);
726