1 /* $NetBSD: smp.h,v 1.5 2023/12/29 22:58:23 chs Exp $ */
2
3 /*-
4 * Copyright (c) 2018 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Taylor R. Campbell.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 #ifndef _LINUX_SMP_H_
33 #define _LINUX_SMP_H_
34
35 #include <sys/cpu.h>
36 #include <sys/systm.h>
37 #include <sys/xcall.h>
38
39 #define smp_processor_id() cpu_number()
40
41 static inline int
get_cpu(void)42 get_cpu(void)
43 {
44
45 kpreempt_disable();
46 return cpu_index(curcpu());
47 }
48
49 static inline void
put_cpu(void)50 put_cpu(void)
51 {
52
53 kpreempt_enable();
54 }
55
56 static inline void
on_each_cpu_xc(void * a,void * b)57 on_each_cpu_xc(void *a, void *b)
58 {
59 void (**fp)(void *) = a;
60 void *cookie = b;
61
62 (**fp)(cookie);
63 }
64
65 static inline void
on_each_cpu(void (* f)(void *),void * cookie,int wait)66 on_each_cpu(void (*f)(void *), void *cookie, int wait)
67 {
68 uint64_t ticket;
69
70 ticket = xc_broadcast(0, &on_each_cpu_xc, &f, cookie);
71 if (wait)
72 xc_wait(ticket);
73 }
74
75 #endif /* _LINUX_SMP_H_ */
76