Blame view

kernel/smp.c 20.8 KB
3d4422332   Jens Axboe   Add generic helpe...
1
2
3
4
  /*
   * Generic helpers for smp ipi calls
   *
   * (C) Jens Axboe <jens.axboe@oracle.com> 2008
3d4422332   Jens Axboe   Add generic helpe...
5
   */
478850160   Frederic Weisbecker   irq_work: Impleme...
6
  #include <linux/irq_work.h>
3d4422332   Jens Axboe   Add generic helpe...
7
  #include <linux/rcupdate.h>
59190f421   Linus Torvalds   Merge branch 'gen...
8
  #include <linux/rculist.h>
641cd4cfc   Ingo Molnar   generic-ipi: elim...
9
  #include <linux/kernel.h>
9984de1a5   Paul Gortmaker   kernel: Map most ...
10
  #include <linux/export.h>
0b13fda1e   Ingo Molnar   generic-ipi: clea...
11
12
  #include <linux/percpu.h>
  #include <linux/init.h>
5a0e3ad6a   Tejun Heo   include cleanup: ...
13
  #include <linux/gfp.h>
3d4422332   Jens Axboe   Add generic helpe...
14
  #include <linux/smp.h>
8969a5ede   Peter Zijlstra   generic-ipi: remo...
15
  #include <linux/cpu.h>
c6f4459fc   Chuansheng Liu   smp: Add new wake...
16
  #include <linux/sched.h>
47ae4b05d   Juergen Gross   virt, sched: Add ...
17
  #include <linux/hypervisor.h>
3d4422332   Jens Axboe   Add generic helpe...
18

3bb5d2ee3   Suresh Siddha   smp, idle: Alloca...
19
  #include "smpboot.h"
3d4422332   Jens Axboe   Add generic helpe...
20
  enum {
6e2756376   Peter Zijlstra   generic-ipi: remo...
21
  	CSD_FLAG_LOCK		= 0x01,
8053871d0   Linus Torvalds   smp: Fix smp_call...
22
  	CSD_FLAG_SYNCHRONOUS	= 0x02,
3d4422332   Jens Axboe   Add generic helpe...
23
24
25
  };
  
  struct call_function_data {
9a46ad6d6   Shaohua Li   smp: make smp_cal...
26
  	struct call_single_data	__percpu *csd;
0b13fda1e   Ingo Molnar   generic-ipi: clea...
27
  	cpumask_var_t		cpumask;
3d4422332   Jens Axboe   Add generic helpe...
28
  };
e03bcb686   Milton Miller   generic-ipi: Opti...
29
  static DEFINE_PER_CPU_SHARED_ALIGNED(struct call_function_data, cfd_data);
6897fc22e   Christoph Hellwig   kernel: use lockl...
30
  static DEFINE_PER_CPU_SHARED_ALIGNED(struct llist_head, call_single_queue);
8969a5ede   Peter Zijlstra   generic-ipi: remo...
31

8d056c48e   Srivatsa S. Bhat   CPU hotplug, smp:...
32
  static void flush_smp_call_function_queue(bool warn_cpu_offline);
31487f832   Richard Weinberger   smp/cfd: Convert ...
33
  int smpcfd_prepare_cpu(unsigned int cpu)
8969a5ede   Peter Zijlstra   generic-ipi: remo...
34
  {
8969a5ede   Peter Zijlstra   generic-ipi: remo...
35
  	struct call_function_data *cfd = &per_cpu(cfd_data, cpu);
31487f832   Richard Weinberger   smp/cfd: Convert ...
36
37
38
39
40
  	if (!zalloc_cpumask_var_node(&cfd->cpumask, GFP_KERNEL,
  				     cpu_to_node(cpu)))
  		return -ENOMEM;
  	cfd->csd = alloc_percpu(struct call_single_data);
  	if (!cfd->csd) {
8969a5ede   Peter Zijlstra   generic-ipi: remo...
41
  		free_cpumask_var(cfd->cpumask);
31487f832   Richard Weinberger   smp/cfd: Convert ...
42
43
44
45
  		return -ENOMEM;
  	}
  
  	return 0;
8969a5ede   Peter Zijlstra   generic-ipi: remo...
46
  }
31487f832   Richard Weinberger   smp/cfd: Convert ...
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
  int smpcfd_dead_cpu(unsigned int cpu)
  {
  	struct call_function_data *cfd = &per_cpu(cfd_data, cpu);
  
  	free_cpumask_var(cfd->cpumask);
  	free_percpu(cfd->csd);
  	return 0;
  }
  
  int smpcfd_dying_cpu(unsigned int cpu)
  {
  	/*
  	 * The IPIs for the smp-call-function callbacks queued by other
  	 * CPUs might arrive late, either due to hardware latencies or
  	 * because this CPU disabled interrupts (inside stop-machine)
  	 * before the IPIs were sent. So flush out any pending callbacks
  	 * explicitly (without waiting for the IPIs to arrive), to
  	 * ensure that the outgoing CPU doesn't go offline with work
  	 * still pending.
  	 */
  	flush_smp_call_function_queue(false);
  	return 0;
  }
8969a5ede   Peter Zijlstra   generic-ipi: remo...
70

d8ad7d112   Takao Indoh   generic-ipi: Fix ...
71
  void __init call_function_init(void)
3d4422332   Jens Axboe   Add generic helpe...
72
73
  {
  	int i;
6897fc22e   Christoph Hellwig   kernel: use lockl...
74
75
  	for_each_possible_cpu(i)
  		init_llist_head(&per_cpu(call_single_queue, i));
8969a5ede   Peter Zijlstra   generic-ipi: remo...
76

31487f832   Richard Weinberger   smp/cfd: Convert ...
77
  	smpcfd_prepare_cpu(smp_processor_id());
3d4422332   Jens Axboe   Add generic helpe...
78
  }
8969a5ede   Peter Zijlstra   generic-ipi: remo...
79
  /*
8969a5ede   Peter Zijlstra   generic-ipi: remo...
80
81
   * csd_lock/csd_unlock used to serialize access to per-cpu csd resources
   *
0b13fda1e   Ingo Molnar   generic-ipi: clea...
82
83
84
   * For non-synchronous ipi calls the csd can still be in use by the
   * previous function call. For multi-cpu calls its even more interesting
   * as we'll have to ensure no other cpu is observing our csd.
8969a5ede   Peter Zijlstra   generic-ipi: remo...
85
   */
90d109847   Davidlohr Bueso   locking/csd_lock:...
86
  static __always_inline void csd_lock_wait(struct call_single_data *csd)
8969a5ede   Peter Zijlstra   generic-ipi: remo...
87
  {
1f03e8d29   Peter Zijlstra   locking/barriers:...
88
  	smp_cond_load_acquire(&csd->flags, !(VAL & CSD_FLAG_LOCK));
6e2756376   Peter Zijlstra   generic-ipi: remo...
89
  }
90d109847   Davidlohr Bueso   locking/csd_lock:...
90
  static __always_inline void csd_lock(struct call_single_data *csd)
6e2756376   Peter Zijlstra   generic-ipi: remo...
91
  {
e1d12f327   Andrew Morton   kernel/smp.c: cle...
92
93
  	csd_lock_wait(csd);
  	csd->flags |= CSD_FLAG_LOCK;
8969a5ede   Peter Zijlstra   generic-ipi: remo...
94
95
  
  	/*
0b13fda1e   Ingo Molnar   generic-ipi: clea...
96
97
98
  	 * prevent CPU from reordering the above assignment
  	 * to ->flags with any subsequent assignments to other
  	 * fields of the specified call_single_data structure:
8969a5ede   Peter Zijlstra   generic-ipi: remo...
99
  	 */
8053871d0   Linus Torvalds   smp: Fix smp_call...
100
  	smp_wmb();
8969a5ede   Peter Zijlstra   generic-ipi: remo...
101
  }
90d109847   Davidlohr Bueso   locking/csd_lock:...
102
  static __always_inline void csd_unlock(struct call_single_data *csd)
8969a5ede   Peter Zijlstra   generic-ipi: remo...
103
  {
8053871d0   Linus Torvalds   smp: Fix smp_call...
104
  	WARN_ON(!(csd->flags & CSD_FLAG_LOCK));
0b13fda1e   Ingo Molnar   generic-ipi: clea...
105

8969a5ede   Peter Zijlstra   generic-ipi: remo...
106
  	/*
0b13fda1e   Ingo Molnar   generic-ipi: clea...
107
  	 * ensure we're all done before releasing data:
8969a5ede   Peter Zijlstra   generic-ipi: remo...
108
  	 */
8053871d0   Linus Torvalds   smp: Fix smp_call...
109
  	smp_store_release(&csd->flags, 0);
3d4422332   Jens Axboe   Add generic helpe...
110
  }
8b28499a7   Frederic Weisbecker   smp: Consolidate ...
111
  static DEFINE_PER_CPU_SHARED_ALIGNED(struct call_single_data, csd_data);
3d4422332   Jens Axboe   Add generic helpe...
112
  /*
0b13fda1e   Ingo Molnar   generic-ipi: clea...
113
114
115
   * Insert a previously allocated call_single_data element
   * for execution on the given CPU. data must already have
   * ->func, ->info, and ->flags set.
3d4422332   Jens Axboe   Add generic helpe...
116
   */
8b28499a7   Frederic Weisbecker   smp: Consolidate ...
117
  static int generic_exec_single(int cpu, struct call_single_data *csd,
8053871d0   Linus Torvalds   smp: Fix smp_call...
118
  			       smp_call_func_t func, void *info)
3d4422332   Jens Axboe   Add generic helpe...
119
  {
8b28499a7   Frederic Weisbecker   smp: Consolidate ...
120
  	if (cpu == smp_processor_id()) {
8053871d0   Linus Torvalds   smp: Fix smp_call...
121
122
123
124
125
126
127
  		unsigned long flags;
  
  		/*
  		 * We can unlock early even for the synchronous on-stack case,
  		 * since we're doing this from the same CPU..
  		 */
  		csd_unlock(csd);
8b28499a7   Frederic Weisbecker   smp: Consolidate ...
128
129
130
131
132
  		local_irq_save(flags);
  		func(info);
  		local_irq_restore(flags);
  		return 0;
  	}
5224b9613   Linus Torvalds   smp: Fix error ca...
133
134
  	if ((unsigned)cpu >= nr_cpu_ids || !cpu_online(cpu)) {
  		csd_unlock(csd);
8b28499a7   Frederic Weisbecker   smp: Consolidate ...
135
  		return -ENXIO;
5224b9613   Linus Torvalds   smp: Fix error ca...
136
  	}
8b28499a7   Frederic Weisbecker   smp: Consolidate ...
137

8b28499a7   Frederic Weisbecker   smp: Consolidate ...
138
139
  	csd->func = func;
  	csd->info = info;
561920a0d   Suresh Siddha   generic-ipi: fix ...
140
  	/*
15d0d3b33   Nick Piggin   generic IPI: simp...
141
142
143
144
145
146
147
  	 * The list addition should be visible before sending the IPI
  	 * handler locks the list to pull the entry off it because of
  	 * normal cache coherency rules implied by spinlocks.
  	 *
  	 * If IPIs can go out of order to the cache coherency protocol
  	 * in an architecture, sufficient synchronisation should be added
  	 * to arch code to make it appear to obey cache coherency WRT
0b13fda1e   Ingo Molnar   generic-ipi: clea...
148
149
  	 * locking and barrier primitives. Generic code isn't really
  	 * equipped to do the right thing...
561920a0d   Suresh Siddha   generic-ipi: fix ...
150
  	 */
6897fc22e   Christoph Hellwig   kernel: use lockl...
151
  	if (llist_add(&csd->llist, &per_cpu(call_single_queue, cpu)))
3d4422332   Jens Axboe   Add generic helpe...
152
  		arch_send_call_function_single_ipi(cpu);
8b28499a7   Frederic Weisbecker   smp: Consolidate ...
153
  	return 0;
3d4422332   Jens Axboe   Add generic helpe...
154
  }
8d056c48e   Srivatsa S. Bhat   CPU hotplug, smp:...
155
156
157
158
159
  /**
   * generic_smp_call_function_single_interrupt - Execute SMP IPI callbacks
   *
   * Invoked by arch to handle an IPI for call function single.
   * Must be called with interrupts disabled.
3d4422332   Jens Axboe   Add generic helpe...
160
161
162
   */
  void generic_smp_call_function_single_interrupt(void)
  {
8d056c48e   Srivatsa S. Bhat   CPU hotplug, smp:...
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
  	flush_smp_call_function_queue(true);
  }
  
  /**
   * flush_smp_call_function_queue - Flush pending smp-call-function callbacks
   *
   * @warn_cpu_offline: If set to 'true', warn if callbacks were queued on an
   *		      offline CPU. Skip this check if set to 'false'.
   *
   * Flush any pending smp-call-function callbacks queued on this CPU. This is
   * invoked by the generic IPI handler, as well as by a CPU about to go offline,
   * to ensure that all pending IPI callbacks are run before it goes completely
   * offline.
   *
   * Loop through the call_single_queue and run all the queued callbacks.
   * Must be called with interrupts disabled.
   */
  static void flush_smp_call_function_queue(bool warn_cpu_offline)
  {
  	struct llist_head *head;
5fd77595e   Jan Kara   smp: Iterate func...
183
184
  	struct llist_node *entry;
  	struct call_single_data *csd, *csd_next;
a219ccf46   Srivatsa S. Bhat   smp: print more u...
185
  	static bool warned;
8d056c48e   Srivatsa S. Bhat   CPU hotplug, smp:...
186
  	WARN_ON(!irqs_disabled());
bb964a92c   Christoph Lameter   kernel misc: Repl...
187
  	head = this_cpu_ptr(&call_single_queue);
8d056c48e   Srivatsa S. Bhat   CPU hotplug, smp:...
188
  	entry = llist_del_all(head);
a219ccf46   Srivatsa S. Bhat   smp: print more u...
189
  	entry = llist_reverse_order(entry);
3d4422332   Jens Axboe   Add generic helpe...
190

8d056c48e   Srivatsa S. Bhat   CPU hotplug, smp:...
191
192
193
  	/* There shouldn't be any pending callbacks on an offline CPU. */
  	if (unlikely(warn_cpu_offline && !cpu_online(smp_processor_id()) &&
  		     !warned && !llist_empty(head))) {
a219ccf46   Srivatsa S. Bhat   smp: print more u...
194
195
196
197
198
199
200
201
202
203
204
205
206
  		warned = true;
  		WARN(1, "IPI on offline CPU %d
  ", smp_processor_id());
  
  		/*
  		 * We don't have to use the _safe() variant here
  		 * because we are not invoking the IPI handlers yet.
  		 */
  		llist_for_each_entry(csd, entry, llist)
  			pr_warn("IPI callback %pS sent to offline CPU
  ",
  				csd->func);
  	}
3d4422332   Jens Axboe   Add generic helpe...
207

5fd77595e   Jan Kara   smp: Iterate func...
208
  	llist_for_each_entry_safe(csd, csd_next, entry, llist) {
8053871d0   Linus Torvalds   smp: Fix smp_call...
209
210
211
212
213
214
215
216
217
218
219
  		smp_call_func_t func = csd->func;
  		void *info = csd->info;
  
  		/* Do we wait until *after* callback? */
  		if (csd->flags & CSD_FLAG_SYNCHRONOUS) {
  			func(info);
  			csd_unlock(csd);
  		} else {
  			csd_unlock(csd);
  			func(info);
  		}
3d4422332   Jens Axboe   Add generic helpe...
220
  	}
478850160   Frederic Weisbecker   irq_work: Impleme...
221
222
223
224
225
226
227
228
  
  	/*
  	 * Handle irq works queued remotely by irq_work_queue_on().
  	 * Smp functions above are typically synchronous so they
  	 * better run first since some other CPUs may be busy waiting
  	 * for them.
  	 */
  	irq_work_run();
3d4422332   Jens Axboe   Add generic helpe...
229
230
231
232
233
234
  }
  
  /*
   * smp_call_function_single - Run a function on a specific CPU
   * @func: The function to run. This must be fast and non-blocking.
   * @info: An arbitrary pointer to pass to the function.
3d4422332   Jens Axboe   Add generic helpe...
235
236
   * @wait: If true, wait until function has completed on other CPUs.
   *
72f279b25   Sheng Yang   generic-ipi: Fix ...
237
   * Returns 0 on success, else a negative status code.
3d4422332   Jens Axboe   Add generic helpe...
238
   */
3a5f65df5   David Howells   Typedef SMP call ...
239
  int smp_call_function_single(int cpu, smp_call_func_t func, void *info,
8691e5a8f   Jens Axboe   smp_call_function...
240
  			     int wait)
3d4422332   Jens Axboe   Add generic helpe...
241
  {
8053871d0   Linus Torvalds   smp: Fix smp_call...
242
243
  	struct call_single_data *csd;
  	struct call_single_data csd_stack = { .flags = CSD_FLAG_LOCK | CSD_FLAG_SYNCHRONOUS };
0b13fda1e   Ingo Molnar   generic-ipi: clea...
244
  	int this_cpu;
8b28499a7   Frederic Weisbecker   smp: Consolidate ...
245
  	int err;
3d4422332   Jens Axboe   Add generic helpe...
246

0b13fda1e   Ingo Molnar   generic-ipi: clea...
247
248
249
250
251
  	/*
  	 * prevent preemption and reschedule on another processor,
  	 * as well as CPU removal
  	 */
  	this_cpu = get_cpu();
269c861ba   Suresh Siddha   generic-ipi: Allo...
252
253
254
255
256
257
258
259
  	/*
  	 * Can deadlock when called with interrupts disabled.
  	 * We allow cpu's that are not yet online though, as no one else can
  	 * send smp call function interrupt to this cpu and as such deadlocks
  	 * can't happen.
  	 */
  	WARN_ON_ONCE(cpu_online(this_cpu) && irqs_disabled()
  		     && !oops_in_progress);
3d4422332   Jens Axboe   Add generic helpe...
260

8053871d0   Linus Torvalds   smp: Fix smp_call...
261
262
263
264
265
266
267
268
269
270
  	csd = &csd_stack;
  	if (!wait) {
  		csd = this_cpu_ptr(&csd_data);
  		csd_lock(csd);
  	}
  
  	err = generic_exec_single(cpu, csd, func, info);
  
  	if (wait)
  		csd_lock_wait(csd);
3d4422332   Jens Axboe   Add generic helpe...
271
272
  
  	put_cpu();
0b13fda1e   Ingo Molnar   generic-ipi: clea...
273

f73be6ded   H. Peter Anvin   smp: have smp_cal...
274
  	return err;
3d4422332   Jens Axboe   Add generic helpe...
275
276
  }
  EXPORT_SYMBOL(smp_call_function_single);
d7877c03f   Frederic Weisbecker   smp: Move __smp_c...
277
  /**
c46fff2a3   Frederic Weisbecker   smp: Rename __smp...
278
279
   * smp_call_function_single_async(): Run an asynchronous function on a
   * 			         specific CPU.
d7877c03f   Frederic Weisbecker   smp: Move __smp_c...
280
281
   * @cpu: The CPU to run on.
   * @csd: Pre-allocated and setup data structure
d7877c03f   Frederic Weisbecker   smp: Move __smp_c...
282
   *
c46fff2a3   Frederic Weisbecker   smp: Rename __smp...
283
284
285
286
287
288
289
290
291
   * Like smp_call_function_single(), but the call is asynchonous and
   * can thus be done from contexts with disabled interrupts.
   *
   * The caller passes his own pre-allocated data structure
   * (ie: embedded in an object) and is responsible for synchronizing it
   * such that the IPIs performed on the @csd are strictly serialized.
   *
   * NOTE: Be careful, there is unfortunately no current debugging facility to
   * validate the correctness of this serialization.
d7877c03f   Frederic Weisbecker   smp: Move __smp_c...
292
   */
c46fff2a3   Frederic Weisbecker   smp: Rename __smp...
293
  int smp_call_function_single_async(int cpu, struct call_single_data *csd)
d7877c03f   Frederic Weisbecker   smp: Move __smp_c...
294
295
  {
  	int err = 0;
d7877c03f   Frederic Weisbecker   smp: Move __smp_c...
296

fce8ad156   Frederic Weisbecker   smp: Remove wait ...
297
  	preempt_disable();
8053871d0   Linus Torvalds   smp: Fix smp_call...
298
299
300
301
302
303
304
305
306
  
  	/* We could deadlock if we have to wait here with interrupts disabled! */
  	if (WARN_ON_ONCE(csd->flags & CSD_FLAG_LOCK))
  		csd_lock_wait(csd);
  
  	csd->flags = CSD_FLAG_LOCK;
  	smp_wmb();
  
  	err = generic_exec_single(cpu, csd, csd->func, csd->info);
fce8ad156   Frederic Weisbecker   smp: Remove wait ...
307
  	preempt_enable();
d7877c03f   Frederic Weisbecker   smp: Move __smp_c...
308
309
310
  
  	return err;
  }
c46fff2a3   Frederic Weisbecker   smp: Rename __smp...
311
  EXPORT_SYMBOL_GPL(smp_call_function_single_async);
d7877c03f   Frederic Weisbecker   smp: Move __smp_c...
312

2ea6dec4a   Rusty Russell   generic-ipi: Add ...
313
314
315
316
317
318
319
320
  /*
   * smp_call_function_any - Run a function on any of the given cpus
   * @mask: The mask of cpus it can run on.
   * @func: The function to run. This must be fast and non-blocking.
   * @info: An arbitrary pointer to pass to the function.
   * @wait: If true, wait until function has completed.
   *
   * Returns 0 on success, else a negative status code (if no cpus were online).
2ea6dec4a   Rusty Russell   generic-ipi: Add ...
321
322
323
324
325
326
327
   *
   * Selection preference:
   *	1) current cpu if in @mask
   *	2) any cpu of current node if in @mask
   *	3) any other online cpu in @mask
   */
  int smp_call_function_any(const struct cpumask *mask,
3a5f65df5   David Howells   Typedef SMP call ...
328
  			  smp_call_func_t func, void *info, int wait)
2ea6dec4a   Rusty Russell   generic-ipi: Add ...
329
330
331
332
333
334
335
336
337
338
339
  {
  	unsigned int cpu;
  	const struct cpumask *nodemask;
  	int ret;
  
  	/* Try for same CPU (cheapest) */
  	cpu = get_cpu();
  	if (cpumask_test_cpu(cpu, mask))
  		goto call;
  
  	/* Try for same node. */
af2422c42   David John   smp_call_function...
340
  	nodemask = cpumask_of_node(cpu_to_node(cpu));
2ea6dec4a   Rusty Russell   generic-ipi: Add ...
341
342
343
344
345
346
347
348
349
350
351
352
353
354
  	for (cpu = cpumask_first_and(nodemask, mask); cpu < nr_cpu_ids;
  	     cpu = cpumask_next_and(cpu, nodemask, mask)) {
  		if (cpu_online(cpu))
  			goto call;
  	}
  
  	/* Any online will do: smp_call_function_single handles nr_cpu_ids. */
  	cpu = cpumask_any_and(mask, cpu_online_mask);
  call:
  	ret = smp_call_function_single(cpu, func, info, wait);
  	put_cpu();
  	return ret;
  }
  EXPORT_SYMBOL_GPL(smp_call_function_any);
3d4422332   Jens Axboe   Add generic helpe...
355
  /**
54b11e6d5   Rusty Russell   cpumask: smp_call...
356
357
   * smp_call_function_many(): Run a function on a set of other CPUs.
   * @mask: The set of cpus to run on (only runs on online subset).
3d4422332   Jens Axboe   Add generic helpe...
358
359
   * @func: The function to run. This must be fast and non-blocking.
   * @info: An arbitrary pointer to pass to the function.
0b13fda1e   Ingo Molnar   generic-ipi: clea...
360
361
   * @wait: If true, wait (atomically) until function has completed
   *        on other CPUs.
3d4422332   Jens Axboe   Add generic helpe...
362
   *
72f279b25   Sheng Yang   generic-ipi: Fix ...
363
   * If @wait is true, then returns once @func has returned.
3d4422332   Jens Axboe   Add generic helpe...
364
365
366
367
368
   *
   * You must not call this function with disabled interrupts or from a
   * hardware interrupt handler or from a bottom half handler. Preemption
   * must be disabled when calling this function.
   */
54b11e6d5   Rusty Russell   cpumask: smp_call...
369
  void smp_call_function_many(const struct cpumask *mask,
3a5f65df5   David Howells   Typedef SMP call ...
370
  			    smp_call_func_t func, void *info, bool wait)
3d4422332   Jens Axboe   Add generic helpe...
371
  {
e1d12f327   Andrew Morton   kernel/smp.c: cle...
372
  	struct call_function_data *cfd;
9a46ad6d6   Shaohua Li   smp: make smp_cal...
373
  	int cpu, next_cpu, this_cpu = smp_processor_id();
3d4422332   Jens Axboe   Add generic helpe...
374

269c861ba   Suresh Siddha   generic-ipi: Allo...
375
376
377
378
379
380
381
  	/*
  	 * Can deadlock when called with interrupts disabled.
  	 * We allow cpu's that are not yet online though, as no one else can
  	 * send smp call function interrupt to this cpu and as such deadlocks
  	 * can't happen.
  	 */
  	WARN_ON_ONCE(cpu_online(this_cpu) && irqs_disabled()
bd924e8cb   Tejun Heo   smp: Allow on_eac...
382
  		     && !oops_in_progress && !early_boot_irqs_disabled);
3d4422332   Jens Axboe   Add generic helpe...
383

723aae25d   Milton Miller   smp_call_function...
384
  	/* Try to fastpath.  So, what's a CPU they want? Ignoring this one. */
54b11e6d5   Rusty Russell   cpumask: smp_call...
385
  	cpu = cpumask_first_and(mask, cpu_online_mask);
0b13fda1e   Ingo Molnar   generic-ipi: clea...
386
  	if (cpu == this_cpu)
54b11e6d5   Rusty Russell   cpumask: smp_call...
387
  		cpu = cpumask_next_and(cpu, mask, cpu_online_mask);
0b13fda1e   Ingo Molnar   generic-ipi: clea...
388

54b11e6d5   Rusty Russell   cpumask: smp_call...
389
390
391
392
393
394
  	/* No online cpus?  We're done. */
  	if (cpu >= nr_cpu_ids)
  		return;
  
  	/* Do we have another CPU which isn't us? */
  	next_cpu = cpumask_next_and(cpu, mask, cpu_online_mask);
0b13fda1e   Ingo Molnar   generic-ipi: clea...
395
  	if (next_cpu == this_cpu)
54b11e6d5   Rusty Russell   cpumask: smp_call...
396
397
398
399
400
401
  		next_cpu = cpumask_next_and(next_cpu, mask, cpu_online_mask);
  
  	/* Fastpath: do that cpu by itself. */
  	if (next_cpu >= nr_cpu_ids) {
  		smp_call_function_single(cpu, func, info, wait);
  		return;
3d4422332   Jens Axboe   Add generic helpe...
402
  	}
bb964a92c   Christoph Lameter   kernel misc: Repl...
403
  	cfd = this_cpu_ptr(&cfd_data);
45a579192   Milton Miller   call_function_man...
404

e1d12f327   Andrew Morton   kernel/smp.c: cle...
405
406
  	cpumask_and(cfd->cpumask, mask, cpu_online_mask);
  	cpumask_clear_cpu(this_cpu, cfd->cpumask);
723aae25d   Milton Miller   smp_call_function...
407
408
  
  	/* Some callers race with other cpus changing the passed mask */
e1d12f327   Andrew Morton   kernel/smp.c: cle...
409
  	if (unlikely(!cpumask_weight(cfd->cpumask)))
723aae25d   Milton Miller   smp_call_function...
410
  		return;
3d4422332   Jens Axboe   Add generic helpe...
411

e1d12f327   Andrew Morton   kernel/smp.c: cle...
412
413
  	for_each_cpu(cpu, cfd->cpumask) {
  		struct call_single_data *csd = per_cpu_ptr(cfd->csd, cpu);
9a46ad6d6   Shaohua Li   smp: make smp_cal...
414
415
  
  		csd_lock(csd);
8053871d0   Linus Torvalds   smp: Fix smp_call...
416
417
  		if (wait)
  			csd->flags |= CSD_FLAG_SYNCHRONOUS;
9a46ad6d6   Shaohua Li   smp: make smp_cal...
418
419
  		csd->func = func;
  		csd->info = info;
6897fc22e   Christoph Hellwig   kernel: use lockl...
420
  		llist_add(&csd->llist, &per_cpu(call_single_queue, cpu));
9a46ad6d6   Shaohua Li   smp: make smp_cal...
421
  	}
561920a0d   Suresh Siddha   generic-ipi: fix ...
422

3d4422332   Jens Axboe   Add generic helpe...
423
  	/* Send a message to all CPUs in the map */
73f945505   Roman Gushchin   kernel/smp.c: rem...
424
  	arch_send_call_function_ipi_mask(cfd->cpumask);
3d4422332   Jens Axboe   Add generic helpe...
425

9a46ad6d6   Shaohua Li   smp: make smp_cal...
426
  	if (wait) {
e1d12f327   Andrew Morton   kernel/smp.c: cle...
427
428
429
430
  		for_each_cpu(cpu, cfd->cpumask) {
  			struct call_single_data *csd;
  
  			csd = per_cpu_ptr(cfd->csd, cpu);
9a46ad6d6   Shaohua Li   smp: make smp_cal...
431
432
433
  			csd_lock_wait(csd);
  		}
  	}
3d4422332   Jens Axboe   Add generic helpe...
434
  }
54b11e6d5   Rusty Russell   cpumask: smp_call...
435
  EXPORT_SYMBOL(smp_call_function_many);
3d4422332   Jens Axboe   Add generic helpe...
436
437
438
439
440
  
  /**
   * smp_call_function(): Run a function on all other CPUs.
   * @func: The function to run. This must be fast and non-blocking.
   * @info: An arbitrary pointer to pass to the function.
0b13fda1e   Ingo Molnar   generic-ipi: clea...
441
442
   * @wait: If true, wait (atomically) until function has completed
   *        on other CPUs.
3d4422332   Jens Axboe   Add generic helpe...
443
   *
54b11e6d5   Rusty Russell   cpumask: smp_call...
444
   * Returns 0.
3d4422332   Jens Axboe   Add generic helpe...
445
446
   *
   * If @wait is true, then returns once @func has returned; otherwise
72f279b25   Sheng Yang   generic-ipi: Fix ...
447
   * it returns just before the target cpu calls @func.
3d4422332   Jens Axboe   Add generic helpe...
448
449
450
451
   *
   * You must not call this function with disabled interrupts or from a
   * hardware interrupt handler or from a bottom half handler.
   */
3a5f65df5   David Howells   Typedef SMP call ...
452
  int smp_call_function(smp_call_func_t func, void *info, int wait)
3d4422332   Jens Axboe   Add generic helpe...
453
  {
3d4422332   Jens Axboe   Add generic helpe...
454
  	preempt_disable();
54b11e6d5   Rusty Russell   cpumask: smp_call...
455
  	smp_call_function_many(cpu_online_mask, func, info, wait);
3d4422332   Jens Axboe   Add generic helpe...
456
  	preempt_enable();
0b13fda1e   Ingo Molnar   generic-ipi: clea...
457

54b11e6d5   Rusty Russell   cpumask: smp_call...
458
  	return 0;
3d4422332   Jens Axboe   Add generic helpe...
459
460
  }
  EXPORT_SYMBOL(smp_call_function);
351f8f8e6   Amerigo Wang   kernel: clean up ...
461

34db18a05   Amerigo Wang   smp: move smp set...
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
  /* Setup configured maximum number of CPUs to activate */
  unsigned int setup_max_cpus = NR_CPUS;
  EXPORT_SYMBOL(setup_max_cpus);
  
  
  /*
   * Setup routine for controlling SMP activation
   *
   * Command-line option of "nosmp" or "maxcpus=0" will disable SMP
   * activation entirely (the MPS table probe still happens, though).
   *
   * Command-line option of "maxcpus=<NUM>", where <NUM> is an integer
   * greater than 0, limits the maximum number of CPUs activated in
   * SMP mode to <NUM>.
   */
  
  void __weak arch_disable_smp_support(void) { }
  
  static int __init nosmp(char *str)
  {
  	setup_max_cpus = 0;
  	arch_disable_smp_support();
  
  	return 0;
  }
  
  early_param("nosmp", nosmp);
  
  /* this is hard limit */
  static int __init nrcpus(char *str)
  {
  	int nr_cpus;
  
  	get_option(&str, &nr_cpus);
  	if (nr_cpus > 0 && nr_cpus < nr_cpu_ids)
  		nr_cpu_ids = nr_cpus;
  
  	return 0;
  }
  
  early_param("nr_cpus", nrcpus);
  
  static int __init maxcpus(char *str)
  {
  	get_option(&str, &setup_max_cpus);
  	if (setup_max_cpus == 0)
  		arch_disable_smp_support();
  
  	return 0;
  }
  
  early_param("maxcpus", maxcpus);
  
  /* Setup number of possible processor ids */
  int nr_cpu_ids __read_mostly = NR_CPUS;
  EXPORT_SYMBOL(nr_cpu_ids);
  
  /* An arch may set nr_cpu_ids earlier if needed, so this would be redundant */
  void __init setup_nr_cpu_ids(void)
  {
  	nr_cpu_ids = find_last_bit(cpumask_bits(cpu_possible_mask),NR_CPUS) + 1;
  }
a17bce4d1   Borislav Petkov   x86/boot: Further...
524
525
526
527
528
  void __weak smp_announce(void)
  {
  	printk(KERN_INFO "Brought up %d CPUs
  ", num_online_cpus());
  }
34db18a05   Amerigo Wang   smp: move smp set...
529
530
531
532
  /* Called by boot processor to activate the rest. */
  void __init smp_init(void)
  {
  	unsigned int cpu;
3bb5d2ee3   Suresh Siddha   smp, idle: Alloca...
533
  	idle_threads_init();
4cb28ced2   Thomas Gleixner   cpu/hotplug: Crea...
534
  	cpuhp_threads_init();
3bb5d2ee3   Suresh Siddha   smp, idle: Alloca...
535

34db18a05   Amerigo Wang   smp: move smp set...
536
537
538
539
540
541
542
543
544
  	/* FIXME: This should be done in userspace --RR */
  	for_each_present_cpu(cpu) {
  		if (num_online_cpus() >= setup_max_cpus)
  			break;
  		if (!cpu_online(cpu))
  			cpu_up(cpu);
  	}
  
  	/* Any cleanup work */
a17bce4d1   Borislav Petkov   x86/boot: Further...
545
  	smp_announce();
34db18a05   Amerigo Wang   smp: move smp set...
546
547
  	smp_cpus_done(setup_max_cpus);
  }
351f8f8e6   Amerigo Wang   kernel: clean up ...
548
  /*
bd924e8cb   Tejun Heo   smp: Allow on_eac...
549
550
551
   * Call a function on all processors.  May be used during early boot while
   * early_boot_irqs_disabled is set.  Use local_irq_save/restore() instead
   * of local_irq_disable/enable().
351f8f8e6   Amerigo Wang   kernel: clean up ...
552
553
554
   */
  int on_each_cpu(void (*func) (void *info), void *info, int wait)
  {
bd924e8cb   Tejun Heo   smp: Allow on_eac...
555
  	unsigned long flags;
351f8f8e6   Amerigo Wang   kernel: clean up ...
556
557
558
559
  	int ret = 0;
  
  	preempt_disable();
  	ret = smp_call_function(func, info, wait);
bd924e8cb   Tejun Heo   smp: Allow on_eac...
560
  	local_irq_save(flags);
351f8f8e6   Amerigo Wang   kernel: clean up ...
561
  	func(info);
bd924e8cb   Tejun Heo   smp: Allow on_eac...
562
  	local_irq_restore(flags);
351f8f8e6   Amerigo Wang   kernel: clean up ...
563
564
565
566
  	preempt_enable();
  	return ret;
  }
  EXPORT_SYMBOL(on_each_cpu);
3fc498f16   Gilad Ben-Yossef   smp: introduce a ...
567
568
569
570
571
572
573
574
575
576
577
578
  
  /**
   * on_each_cpu_mask(): Run a function on processors specified by
   * cpumask, which may include the local processor.
   * @mask: The set of cpus to run on (only runs on online subset).
   * @func: The function to run. This must be fast and non-blocking.
   * @info: An arbitrary pointer to pass to the function.
   * @wait: If true, wait (atomically) until function has completed
   *        on other CPUs.
   *
   * If @wait is true, then returns once @func has returned.
   *
202da4005   David Daney   kernel/smp.c: qui...
579
580
581
582
   * You must not call this function with disabled interrupts or from a
   * hardware interrupt handler or from a bottom half handler.  The
   * exception is that it may be used during early boot while
   * early_boot_irqs_disabled is set.
3fc498f16   Gilad Ben-Yossef   smp: introduce a ...
583
584
585
586
587
588
589
590
   */
  void on_each_cpu_mask(const struct cpumask *mask, smp_call_func_t func,
  			void *info, bool wait)
  {
  	int cpu = get_cpu();
  
  	smp_call_function_many(mask, func, info, wait);
  	if (cpumask_test_cpu(cpu, mask)) {
202da4005   David Daney   kernel/smp.c: qui...
591
592
  		unsigned long flags;
  		local_irq_save(flags);
3fc498f16   Gilad Ben-Yossef   smp: introduce a ...
593
  		func(info);
202da4005   David Daney   kernel/smp.c: qui...
594
  		local_irq_restore(flags);
3fc498f16   Gilad Ben-Yossef   smp: introduce a ...
595
596
597
598
  	}
  	put_cpu();
  }
  EXPORT_SYMBOL(on_each_cpu_mask);
b3a7e98e0   Gilad Ben-Yossef   smp: add func to ...
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
  
  /*
   * on_each_cpu_cond(): Call a function on each processor for which
   * the supplied function cond_func returns true, optionally waiting
   * for all the required CPUs to finish. This may include the local
   * processor.
   * @cond_func:	A callback function that is passed a cpu id and
   *		the the info parameter. The function is called
   *		with preemption disabled. The function should
   *		return a blooean value indicating whether to IPI
   *		the specified CPU.
   * @func:	The function to run on all applicable CPUs.
   *		This must be fast and non-blocking.
   * @info:	An arbitrary pointer to pass to both functions.
   * @wait:	If true, wait (atomically) until function has
   *		completed on other CPUs.
   * @gfp_flags:	GFP flags to use when allocating the cpumask
   *		used internally by the function.
   *
   * The function might sleep if the GFP flags indicates a non
   * atomic allocation is allowed.
   *
   * Preemption is disabled to protect against CPUs going offline but not online.
   * CPUs going online during the call will not be seen or sent an IPI.
   *
   * You must not call this function with disabled interrupts or
   * from a hardware interrupt handler or from a bottom half handler.
   */
  void on_each_cpu_cond(bool (*cond_func)(int cpu, void *info),
  			smp_call_func_t func, void *info, bool wait,
  			gfp_t gfp_flags)
  {
  	cpumask_var_t cpus;
  	int cpu, ret;
d0164adc8   Mel Gorman   mm, page_alloc: d...
633
  	might_sleep_if(gfpflags_allow_blocking(gfp_flags));
b3a7e98e0   Gilad Ben-Yossef   smp: add func to ...
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
  
  	if (likely(zalloc_cpumask_var(&cpus, (gfp_flags|__GFP_NOWARN)))) {
  		preempt_disable();
  		for_each_online_cpu(cpu)
  			if (cond_func(cpu, info))
  				cpumask_set_cpu(cpu, cpus);
  		on_each_cpu_mask(cpus, func, info, wait);
  		preempt_enable();
  		free_cpumask_var(cpus);
  	} else {
  		/*
  		 * No free cpumask, bother. No matter, we'll
  		 * just have to IPI them one by one.
  		 */
  		preempt_disable();
  		for_each_online_cpu(cpu)
  			if (cond_func(cpu, info)) {
  				ret = smp_call_function_single(cpu, func,
  								info, wait);
618fde872   Sasha Levin   kernel/smp.c:on_e...
653
  				WARN_ON_ONCE(ret);
b3a7e98e0   Gilad Ben-Yossef   smp: add func to ...
654
655
656
657
658
  			}
  		preempt_enable();
  	}
  }
  EXPORT_SYMBOL(on_each_cpu_cond);
f37f435f3   Thomas Gleixner   smp: Implement ki...
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
  
  static void do_nothing(void *unused)
  {
  }
  
  /**
   * kick_all_cpus_sync - Force all cpus out of idle
   *
   * Used to synchronize the update of pm_idle function pointer. It's
   * called after the pointer is updated and returns after the dummy
   * callback function has been executed on all cpus. The execution of
   * the function can only happen on the remote cpus after they have
   * left the idle function which had been called via pm_idle function
   * pointer. So it's guaranteed that nothing uses the previous pointer
   * anymore.
   */
  void kick_all_cpus_sync(void)
  {
  	/* Make sure the change is visible before we kick the cpus */
  	smp_mb();
  	smp_call_function(do_nothing, NULL, 1);
  }
  EXPORT_SYMBOL_GPL(kick_all_cpus_sync);
c6f4459fc   Chuansheng Liu   smp: Add new wake...
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
  
  /**
   * wake_up_all_idle_cpus - break all cpus out of idle
   * wake_up_all_idle_cpus try to break all cpus which is in idle state even
   * including idle polling cpus, for non-idle cpus, we will do nothing
   * for them.
   */
  void wake_up_all_idle_cpus(void)
  {
  	int cpu;
  
  	preempt_disable();
  	for_each_online_cpu(cpu) {
  		if (cpu == smp_processor_id())
  			continue;
  
  		wake_up_if_idle(cpu);
  	}
  	preempt_enable();
  }
  EXPORT_SYMBOL_GPL(wake_up_all_idle_cpus);
df8ce9d78   Juergen Gross   smp: Add function...
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
  
  /**
   * smp_call_on_cpu - Call a function on a specific cpu
   *
   * Used to call a function on a specific cpu and wait for it to return.
   * Optionally make sure the call is done on a specified physical cpu via vcpu
   * pinning in order to support virtualized environments.
   */
  struct smp_call_on_cpu_struct {
  	struct work_struct	work;
  	struct completion	done;
  	int			(*func)(void *);
  	void			*data;
  	int			ret;
  	int			cpu;
  };
  
  static void smp_call_on_cpu_callback(struct work_struct *work)
  {
  	struct smp_call_on_cpu_struct *sscs;
  
  	sscs = container_of(work, struct smp_call_on_cpu_struct, work);
  	if (sscs->cpu >= 0)
  		hypervisor_pin_vcpu(sscs->cpu);
  	sscs->ret = sscs->func(sscs->data);
  	if (sscs->cpu >= 0)
  		hypervisor_pin_vcpu(-1);
  
  	complete(&sscs->done);
  }
  
  int smp_call_on_cpu(unsigned int cpu, int (*func)(void *), void *par, bool phys)
  {
  	struct smp_call_on_cpu_struct sscs = {
df8ce9d78   Juergen Gross   smp: Add function...
737
738
739
740
741
  		.done = COMPLETION_INITIALIZER_ONSTACK(sscs.done),
  		.func = func,
  		.data = par,
  		.cpu  = phys ? cpu : -1,
  	};
8db549491   Peter Zijlstra   smp: Allocate smp...
742
  	INIT_WORK_ONSTACK(&sscs.work, smp_call_on_cpu_callback);
df8ce9d78   Juergen Gross   smp: Add function...
743
744
745
746
747
748
749
750
751
  	if (cpu >= nr_cpu_ids || !cpu_online(cpu))
  		return -ENXIO;
  
  	queue_work_on(cpu, system_wq, &sscs.work);
  	wait_for_completion(&sscs.done);
  
  	return sscs.ret;
  }
  EXPORT_SYMBOL_GPL(smp_call_on_cpu);