rcutree_plugin.h 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565
  1. /*
  2. * Read-Copy Update mechanism for mutual exclusion (tree-based version)
  3. * Internal non-public definitions that provide either classic
  4. * or preemptable semantics.
  5. *
  6. * This program is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU General Public License as published by
  8. * the Free Software Foundation; either version 2 of the License, or
  9. * (at your option) any later version.
  10. *
  11. * This program is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. * GNU General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU General Public License
  17. * along with this program; if not, write to the Free Software
  18. * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  19. *
  20. * Copyright Red Hat, 2009
  21. * Copyright IBM Corporation, 2009
  22. *
  23. * Author: Ingo Molnar <mingo@elte.hu>
  24. * Paul E. McKenney <paulmck@linux.vnet.ibm.com>
  25. */
  26. #ifdef CONFIG_TREE_PREEMPT_RCU
  27. struct rcu_state rcu_preempt_state = RCU_STATE_INITIALIZER(rcu_preempt_state);
  28. DEFINE_PER_CPU(struct rcu_data, rcu_preempt_data);
  29. /*
  30. * Tell them what RCU they are running.
  31. */
  32. static inline void rcu_bootup_announce(void)
  33. {
  34. printk(KERN_INFO
  35. "Experimental preemptable hierarchical RCU implementation.\n");
  36. }
  37. /*
  38. * Return the number of RCU-preempt batches processed thus far
  39. * for debug and statistics.
  40. */
  41. long rcu_batches_completed_preempt(void)
  42. {
  43. return rcu_preempt_state.completed;
  44. }
  45. EXPORT_SYMBOL_GPL(rcu_batches_completed_preempt);
  46. /*
  47. * Return the number of RCU batches processed thus far for debug & stats.
  48. */
  49. long rcu_batches_completed(void)
  50. {
  51. return rcu_batches_completed_preempt();
  52. }
  53. EXPORT_SYMBOL_GPL(rcu_batches_completed);
  54. /*
  55. * Record a preemptable-RCU quiescent state for the specified CPU. Note
  56. * that this just means that the task currently running on the CPU is
  57. * not in a quiescent state. There might be any number of tasks blocked
  58. * while in an RCU read-side critical section.
  59. */
  60. static void rcu_preempt_qs(int cpu)
  61. {
  62. struct rcu_data *rdp = &per_cpu(rcu_preempt_data, cpu);
  63. rdp->passed_quiesc_completed = rdp->completed;
  64. barrier();
  65. rdp->passed_quiesc = 1;
  66. }
  67. /*
  68. * We have entered the scheduler, and the current task might soon be
  69. * context-switched away from. If this task is in an RCU read-side
  70. * critical section, we will no longer be able to rely on the CPU to
  71. * record that fact, so we enqueue the task on the appropriate entry
  72. * of the blocked_tasks[] array. The task will dequeue itself when
  73. * it exits the outermost enclosing RCU read-side critical section.
  74. * Therefore, the current grace period cannot be permitted to complete
  75. * until the blocked_tasks[] entry indexed by the low-order bit of
  76. * rnp->gpnum empties.
  77. *
  78. * Caller must disable preemption.
  79. */
  80. static void rcu_preempt_note_context_switch(int cpu)
  81. {
  82. struct task_struct *t = current;
  83. unsigned long flags;
  84. int phase;
  85. struct rcu_data *rdp;
  86. struct rcu_node *rnp;
  87. if (t->rcu_read_lock_nesting &&
  88. (t->rcu_read_unlock_special & RCU_READ_UNLOCK_BLOCKED) == 0) {
  89. /* Possibly blocking in an RCU read-side critical section. */
  90. rdp = rcu_preempt_state.rda[cpu];
  91. rnp = rdp->mynode;
  92. spin_lock_irqsave(&rnp->lock, flags);
  93. t->rcu_read_unlock_special |= RCU_READ_UNLOCK_BLOCKED;
  94. t->rcu_blocked_node = rnp;
  95. /*
  96. * If this CPU has already checked in, then this task
  97. * will hold up the next grace period rather than the
  98. * current grace period. Queue the task accordingly.
  99. * If the task is queued for the current grace period
  100. * (i.e., this CPU has not yet passed through a quiescent
  101. * state for the current grace period), then as long
  102. * as that task remains queued, the current grace period
  103. * cannot end.
  104. *
  105. * But first, note that the current CPU must still be
  106. * on line!
  107. */
  108. WARN_ON_ONCE((rdp->grpmask & rnp->qsmaskinit) == 0);
  109. WARN_ON_ONCE(!list_empty(&t->rcu_node_entry));
  110. phase = (rnp->gpnum + !(rnp->qsmask & rdp->grpmask)) & 0x1;
  111. list_add(&t->rcu_node_entry, &rnp->blocked_tasks[phase]);
  112. spin_unlock_irqrestore(&rnp->lock, flags);
  113. }
  114. /*
  115. * Either we were not in an RCU read-side critical section to
  116. * begin with, or we have now recorded that critical section
  117. * globally. Either way, we can now note a quiescent state
  118. * for this CPU. Again, if we were in an RCU read-side critical
  119. * section, and if that critical section was blocking the current
  120. * grace period, then the fact that the task has been enqueued
  121. * means that we continue to block the current grace period.
  122. */
  123. rcu_preempt_qs(cpu);
  124. local_irq_save(flags);
  125. t->rcu_read_unlock_special &= ~RCU_READ_UNLOCK_NEED_QS;
  126. local_irq_restore(flags);
  127. }
  128. /*
  129. * Tree-preemptable RCU implementation for rcu_read_lock().
  130. * Just increment ->rcu_read_lock_nesting, shared state will be updated
  131. * if we block.
  132. */
  133. void __rcu_read_lock(void)
  134. {
  135. ACCESS_ONCE(current->rcu_read_lock_nesting)++;
  136. barrier(); /* needed if we ever invoke rcu_read_lock in rcutree.c */
  137. }
  138. EXPORT_SYMBOL_GPL(__rcu_read_lock);
  139. static void rcu_read_unlock_special(struct task_struct *t)
  140. {
  141. int empty;
  142. unsigned long flags;
  143. unsigned long mask;
  144. struct rcu_node *rnp;
  145. int special;
  146. /* NMI handlers cannot block and cannot safely manipulate state. */
  147. if (in_nmi())
  148. return;
  149. local_irq_save(flags);
  150. /*
  151. * If RCU core is waiting for this CPU to exit critical section,
  152. * let it know that we have done so.
  153. */
  154. special = t->rcu_read_unlock_special;
  155. if (special & RCU_READ_UNLOCK_NEED_QS) {
  156. t->rcu_read_unlock_special &= ~RCU_READ_UNLOCK_NEED_QS;
  157. rcu_preempt_qs(smp_processor_id());
  158. }
  159. /* Hardware IRQ handlers cannot block. */
  160. if (in_irq()) {
  161. local_irq_restore(flags);
  162. return;
  163. }
  164. /* Clean up if blocked during RCU read-side critical section. */
  165. if (special & RCU_READ_UNLOCK_BLOCKED) {
  166. t->rcu_read_unlock_special &= ~RCU_READ_UNLOCK_BLOCKED;
  167. /*
  168. * Remove this task from the list it blocked on. The
  169. * task can migrate while we acquire the lock, but at
  170. * most one time. So at most two passes through loop.
  171. */
  172. for (;;) {
  173. rnp = t->rcu_blocked_node;
  174. spin_lock(&rnp->lock); /* irqs already disabled. */
  175. if (rnp == t->rcu_blocked_node)
  176. break;
  177. spin_unlock(&rnp->lock); /* irqs remain disabled. */
  178. }
  179. empty = list_empty(&rnp->blocked_tasks[rnp->gpnum & 0x1]);
  180. list_del_init(&t->rcu_node_entry);
  181. t->rcu_blocked_node = NULL;
  182. /*
  183. * If this was the last task on the current list, and if
  184. * we aren't waiting on any CPUs, report the quiescent state.
  185. * Note that both cpu_quiet_msk_finish() and cpu_quiet_msk()
  186. * drop rnp->lock and restore irq.
  187. */
  188. if (!empty && rnp->qsmask == 0 &&
  189. list_empty(&rnp->blocked_tasks[rnp->gpnum & 0x1])) {
  190. struct rcu_node *rnp_p;
  191. if (rnp->parent == NULL) {
  192. /* Only one rcu_node in the tree. */
  193. cpu_quiet_msk_finish(&rcu_preempt_state, flags);
  194. return;
  195. }
  196. /* Report up the rest of the hierarchy. */
  197. mask = rnp->grpmask;
  198. spin_unlock_irqrestore(&rnp->lock, flags);
  199. rnp_p = rnp->parent;
  200. spin_lock_irqsave(&rnp_p->lock, flags);
  201. WARN_ON_ONCE(rnp->qsmask);
  202. cpu_quiet_msk(mask, &rcu_preempt_state, rnp_p, flags);
  203. return;
  204. }
  205. spin_unlock(&rnp->lock);
  206. }
  207. local_irq_restore(flags);
  208. }
  209. /*
  210. * Tree-preemptable RCU implementation for rcu_read_unlock().
  211. * Decrement ->rcu_read_lock_nesting. If the result is zero (outermost
  212. * rcu_read_unlock()) and ->rcu_read_unlock_special is non-zero, then
  213. * invoke rcu_read_unlock_special() to clean up after a context switch
  214. * in an RCU read-side critical section and other special cases.
  215. */
  216. void __rcu_read_unlock(void)
  217. {
  218. struct task_struct *t = current;
  219. barrier(); /* needed if we ever invoke rcu_read_unlock in rcutree.c */
  220. if (--ACCESS_ONCE(t->rcu_read_lock_nesting) == 0 &&
  221. unlikely(ACCESS_ONCE(t->rcu_read_unlock_special)))
  222. rcu_read_unlock_special(t);
  223. }
  224. EXPORT_SYMBOL_GPL(__rcu_read_unlock);
  225. #ifdef CONFIG_RCU_CPU_STALL_DETECTOR
  226. /*
  227. * Scan the current list of tasks blocked within RCU read-side critical
  228. * sections, printing out the tid of each.
  229. */
  230. static void rcu_print_task_stall(struct rcu_node *rnp)
  231. {
  232. unsigned long flags;
  233. struct list_head *lp;
  234. int phase = rnp->gpnum & 0x1;
  235. struct task_struct *t;
  236. if (!list_empty(&rnp->blocked_tasks[phase])) {
  237. spin_lock_irqsave(&rnp->lock, flags);
  238. phase = rnp->gpnum & 0x1; /* re-read under lock. */
  239. lp = &rnp->blocked_tasks[phase];
  240. list_for_each_entry(t, lp, rcu_node_entry)
  241. printk(" P%d", t->pid);
  242. spin_unlock_irqrestore(&rnp->lock, flags);
  243. }
  244. }
  245. #endif /* #ifdef CONFIG_RCU_CPU_STALL_DETECTOR */
  246. /*
  247. * Check that the list of blocked tasks for the newly completed grace
  248. * period is in fact empty. It is a serious bug to complete a grace
  249. * period that still has RCU readers blocked! This function must be
  250. * invoked -before- updating this rnp's ->gpnum, and the rnp's ->lock
  251. * must be held by the caller.
  252. */
  253. static void rcu_preempt_check_blocked_tasks(struct rcu_node *rnp)
  254. {
  255. WARN_ON_ONCE(!list_empty(&rnp->blocked_tasks[rnp->gpnum & 0x1]));
  256. WARN_ON_ONCE(rnp->qsmask);
  257. }
  258. /*
  259. * Check for preempted RCU readers for the specified rcu_node structure.
  260. * If the caller needs a reliable answer, it must hold the rcu_node's
  261. * >lock.
  262. */
  263. static int rcu_preempted_readers(struct rcu_node *rnp)
  264. {
  265. return !list_empty(&rnp->blocked_tasks[rnp->gpnum & 0x1]);
  266. }
  267. #ifdef CONFIG_HOTPLUG_CPU
  268. /*
  269. * Handle tasklist migration for case in which all CPUs covered by the
  270. * specified rcu_node have gone offline. Move them up to the root
  271. * rcu_node. The reason for not just moving them to the immediate
  272. * parent is to remove the need for rcu_read_unlock_special() to
  273. * make more than two attempts to acquire the target rcu_node's lock.
  274. *
  275. * The caller must hold rnp->lock with irqs disabled.
  276. */
  277. static void rcu_preempt_offline_tasks(struct rcu_state *rsp,
  278. struct rcu_node *rnp,
  279. struct rcu_data *rdp)
  280. {
  281. int i;
  282. struct list_head *lp;
  283. struct list_head *lp_root;
  284. struct rcu_node *rnp_root = rcu_get_root(rsp);
  285. struct task_struct *tp;
  286. if (rnp == rnp_root) {
  287. WARN_ONCE(1, "Last CPU thought to be offlined?");
  288. return; /* Shouldn't happen: at least one CPU online. */
  289. }
  290. WARN_ON_ONCE(rnp != rdp->mynode &&
  291. (!list_empty(&rnp->blocked_tasks[0]) ||
  292. !list_empty(&rnp->blocked_tasks[1])));
  293. /*
  294. * Move tasks up to root rcu_node. Rely on the fact that the
  295. * root rcu_node can be at most one ahead of the rest of the
  296. * rcu_nodes in terms of gp_num value. This fact allows us to
  297. * move the blocked_tasks[] array directly, element by element.
  298. */
  299. for (i = 0; i < 2; i++) {
  300. lp = &rnp->blocked_tasks[i];
  301. lp_root = &rnp_root->blocked_tasks[i];
  302. while (!list_empty(lp)) {
  303. tp = list_entry(lp->next, typeof(*tp), rcu_node_entry);
  304. spin_lock(&rnp_root->lock); /* irqs already disabled */
  305. list_del(&tp->rcu_node_entry);
  306. tp->rcu_blocked_node = rnp_root;
  307. list_add(&tp->rcu_node_entry, lp_root);
  308. spin_unlock(&rnp_root->lock); /* irqs remain disabled */
  309. }
  310. }
  311. }
  312. /*
  313. * Do CPU-offline processing for preemptable RCU.
  314. */
  315. static void rcu_preempt_offline_cpu(int cpu)
  316. {
  317. __rcu_offline_cpu(cpu, &rcu_preempt_state);
  318. }
  319. #endif /* #ifdef CONFIG_HOTPLUG_CPU */
  320. /*
  321. * Check for a quiescent state from the current CPU. When a task blocks,
  322. * the task is recorded in the corresponding CPU's rcu_node structure,
  323. * which is checked elsewhere.
  324. *
  325. * Caller must disable hard irqs.
  326. */
  327. static void rcu_preempt_check_callbacks(int cpu)
  328. {
  329. struct task_struct *t = current;
  330. if (t->rcu_read_lock_nesting == 0) {
  331. t->rcu_read_unlock_special &= ~RCU_READ_UNLOCK_NEED_QS;
  332. rcu_preempt_qs(cpu);
  333. return;
  334. }
  335. if (per_cpu(rcu_preempt_data, cpu).qs_pending) {
  336. t->rcu_read_unlock_special |= RCU_READ_UNLOCK_NEED_QS;
  337. }
  338. }
  339. /*
  340. * Process callbacks for preemptable RCU.
  341. */
  342. static void rcu_preempt_process_callbacks(void)
  343. {
  344. __rcu_process_callbacks(&rcu_preempt_state,
  345. &__get_cpu_var(rcu_preempt_data));
  346. }
  347. /*
  348. * Queue a preemptable-RCU callback for invocation after a grace period.
  349. */
  350. void call_rcu(struct rcu_head *head, void (*func)(struct rcu_head *rcu))
  351. {
  352. __call_rcu(head, func, &rcu_preempt_state);
  353. }
  354. EXPORT_SYMBOL_GPL(call_rcu);
  355. /*
  356. * Check to see if there is any immediate preemptable-RCU-related work
  357. * to be done.
  358. */
  359. static int rcu_preempt_pending(int cpu)
  360. {
  361. return __rcu_pending(&rcu_preempt_state,
  362. &per_cpu(rcu_preempt_data, cpu));
  363. }
  364. /*
  365. * Does preemptable RCU need the CPU to stay out of dynticks mode?
  366. */
  367. static int rcu_preempt_needs_cpu(int cpu)
  368. {
  369. return !!per_cpu(rcu_preempt_data, cpu).nxtlist;
  370. }
  371. /*
  372. * Initialize preemptable RCU's per-CPU data.
  373. */
  374. static void __cpuinit rcu_preempt_init_percpu_data(int cpu)
  375. {
  376. rcu_init_percpu_data(cpu, &rcu_preempt_state, 1);
  377. }
  378. /*
  379. * Check for a task exiting while in a preemptable-RCU read-side
  380. * critical section, clean up if so. No need to issue warnings,
  381. * as debug_check_no_locks_held() already does this if lockdep
  382. * is enabled.
  383. */
  384. void exit_rcu(void)
  385. {
  386. struct task_struct *t = current;
  387. if (t->rcu_read_lock_nesting == 0)
  388. return;
  389. t->rcu_read_lock_nesting = 1;
  390. rcu_read_unlock();
  391. }
  392. #else /* #ifdef CONFIG_TREE_PREEMPT_RCU */
  393. /*
  394. * Tell them what RCU they are running.
  395. */
  396. static inline void rcu_bootup_announce(void)
  397. {
  398. printk(KERN_INFO "Hierarchical RCU implementation.\n");
  399. }
  400. /*
  401. * Return the number of RCU batches processed thus far for debug & stats.
  402. */
  403. long rcu_batches_completed(void)
  404. {
  405. return rcu_batches_completed_sched();
  406. }
  407. EXPORT_SYMBOL_GPL(rcu_batches_completed);
  408. /*
  409. * Because preemptable RCU does not exist, we never have to check for
  410. * CPUs being in quiescent states.
  411. */
  412. static void rcu_preempt_note_context_switch(int cpu)
  413. {
  414. }
  415. #ifdef CONFIG_RCU_CPU_STALL_DETECTOR
  416. /*
  417. * Because preemptable RCU does not exist, we never have to check for
  418. * tasks blocked within RCU read-side critical sections.
  419. */
  420. static void rcu_print_task_stall(struct rcu_node *rnp)
  421. {
  422. }
  423. #endif /* #ifdef CONFIG_RCU_CPU_STALL_DETECTOR */
  424. /*
  425. * Because there is no preemptable RCU, there can be no readers blocked,
  426. * so there is no need to check for blocked tasks.
  427. */
  428. static void rcu_preempt_check_blocked_tasks(struct rcu_node *rnp)
  429. {
  430. }
  431. /*
  432. * Because preemptable RCU does not exist, there are never any preempted
  433. * RCU readers.
  434. */
  435. static int rcu_preempted_readers(struct rcu_node *rnp)
  436. {
  437. return 0;
  438. }
  439. #ifdef CONFIG_HOTPLUG_CPU
  440. /*
  441. * Because preemptable RCU does not exist, it never needs to migrate
  442. * tasks that were blocked within RCU read-side critical sections.
  443. */
  444. static void rcu_preempt_offline_tasks(struct rcu_state *rsp,
  445. struct rcu_node *rnp,
  446. struct rcu_data *rdp)
  447. {
  448. }
  449. /*
  450. * Because preemptable RCU does not exist, it never needs CPU-offline
  451. * processing.
  452. */
  453. static void rcu_preempt_offline_cpu(int cpu)
  454. {
  455. }
  456. #endif /* #ifdef CONFIG_HOTPLUG_CPU */
  457. /*
  458. * Because preemptable RCU does not exist, it never has any callbacks
  459. * to check.
  460. */
  461. void rcu_preempt_check_callbacks(int cpu)
  462. {
  463. }
  464. /*
  465. * Because preemptable RCU does not exist, it never has any callbacks
  466. * to process.
  467. */
  468. void rcu_preempt_process_callbacks(void)
  469. {
  470. }
  471. /*
  472. * In classic RCU, call_rcu() is just call_rcu_sched().
  473. */
  474. void call_rcu(struct rcu_head *head, void (*func)(struct rcu_head *rcu))
  475. {
  476. call_rcu_sched(head, func);
  477. }
  478. EXPORT_SYMBOL_GPL(call_rcu);
  479. /*
  480. * Because preemptable RCU does not exist, it never has any work to do.
  481. */
  482. static int rcu_preempt_pending(int cpu)
  483. {
  484. return 0;
  485. }
  486. /*
  487. * Because preemptable RCU does not exist, it never needs any CPU.
  488. */
  489. static int rcu_preempt_needs_cpu(int cpu)
  490. {
  491. return 0;
  492. }
  493. /*
  494. * Because preemptable RCU does not exist, there is no per-CPU
  495. * data to initialize.
  496. */
  497. static void __cpuinit rcu_preempt_init_percpu_data(int cpu)
  498. {
  499. }
  500. #endif /* #else #ifdef CONFIG_TREE_PREEMPT_RCU */