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