xref: /freebsd/sys/sys/pcpu.h (revision 42249ef2)
1 /*-
2  * SPDX-License-Identifier: BSD-3-Clause
3  *
4  * Copyright (c) 2001 Wind River Systems, Inc.
5  * All rights reserved.
6  * Written by: John Baldwin <jhb@FreeBSD.org>
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. Neither the name of the author nor the names of any co-contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  *
32  * $FreeBSD$
33  */
34 
35 #ifndef _SYS_PCPU_H_
36 #define	_SYS_PCPU_H_
37 
38 #ifdef LOCORE
39 #error "no assembler-serviceable parts inside"
40 #endif
41 
42 #include <sys/param.h>
43 #include <sys/_cpuset.h>
44 #include <sys/_lock.h>
45 #include <sys/_mutex.h>
46 #include <sys/_sx.h>
47 #include <sys/queue.h>
48 #include <sys/_rmlock.h>
49 #include <sys/resource.h>
50 #include <machine/pcpu.h>
51 
52 #define	DPCPU_SETNAME		"set_pcpu"
53 #define	DPCPU_SYMPREFIX		"pcpu_entry_"
54 
55 #ifdef _KERNEL
56 
57 /*
58  * Define a set for pcpu data.
59  */
60 extern uintptr_t *__start_set_pcpu;
61 __GLOBL(__start_set_pcpu);
62 extern uintptr_t *__stop_set_pcpu;
63 __GLOBL(__stop_set_pcpu);
64 
65 /*
66  * Array of dynamic pcpu base offsets.  Indexed by id.
67  */
68 extern uintptr_t dpcpu_off[];
69 
70 /*
71  * Convenience defines.
72  */
73 #define	DPCPU_START		((uintptr_t)&__start_set_pcpu)
74 #define	DPCPU_STOP		((uintptr_t)&__stop_set_pcpu)
75 #define	DPCPU_BYTES		(DPCPU_STOP - DPCPU_START)
76 #define	DPCPU_MODMIN		2048
77 #define	DPCPU_SIZE		roundup2(DPCPU_BYTES, PAGE_SIZE)
78 #define	DPCPU_MODSIZE		(DPCPU_SIZE - (DPCPU_BYTES - DPCPU_MODMIN))
79 
80 /*
81  * Declaration and definition.
82  */
83 #define	DPCPU_NAME(n)		pcpu_entry_##n
84 #define	DPCPU_DECLARE(t, n)	extern t DPCPU_NAME(n)
85 /* struct _hack is to stop this from being used with the static keyword. */
86 #define	DPCPU_DEFINE(t, n)	\
87     struct _hack; t DPCPU_NAME(n) __section(DPCPU_SETNAME) __used
88 #if defined(KLD_MODULE) && (defined(__aarch64__) || defined(__riscv) \
89 		|| defined(__powerpc64__))
90 /*
91  * On some architectures the compiler will use PC-relative load to
92  * find the address of DPCPU data with the static keyword. We then
93  * use this to find the offset of the data in a per-CPU region.
94  * This works for in the kernel as we can allocate the space ahead
95  * of time, however modules need to allocate a sepatate space and
96  * then use relocations to fix the address of the data. As
97  * PC-relative data doesn't have a relocation there is nothing for
98  * the kernel module linker to fix so data is accessed from the
99  * wrong location.
100  *
101  * This is a workaround until a better solution can be found.
102  *
103  * VNET_DEFINE_STATIC also has the same workaround.
104  */
105 #define	DPCPU_DEFINE_STATIC(t, n)	\
106     t DPCPU_NAME(n) __section(DPCPU_SETNAME) __used
107 #else
108 #define	DPCPU_DEFINE_STATIC(t, n)	\
109     static t DPCPU_NAME(n) __section(DPCPU_SETNAME) __used
110 #endif
111 
112 /*
113  * Accessors with a given base.
114  */
115 #define	_DPCPU_PTR(b, n)						\
116     (__typeof(DPCPU_NAME(n))*)((b) + (uintptr_t)&DPCPU_NAME(n))
117 #define	_DPCPU_GET(b, n)	(*_DPCPU_PTR(b, n))
118 #define	_DPCPU_SET(b, n, v)	(*_DPCPU_PTR(b, n) = v)
119 
120 /*
121  * Accessors for the current cpu.
122  */
123 #define	DPCPU_PTR(n)		_DPCPU_PTR(PCPU_GET(dynamic), n)
124 #define	DPCPU_GET(n)		(*DPCPU_PTR(n))
125 #define	DPCPU_SET(n, v)		(*DPCPU_PTR(n) = v)
126 
127 /*
128  * Accessors for remote cpus.
129  */
130 #define	DPCPU_ID_PTR(i, n)	_DPCPU_PTR(dpcpu_off[(i)], n)
131 #define	DPCPU_ID_GET(i, n)	(*DPCPU_ID_PTR(i, n))
132 #define	DPCPU_ID_SET(i, n, v)	(*DPCPU_ID_PTR(i, n) = v)
133 
134 /*
135  * Utility macros.
136  */
137 #define	DPCPU_SUM(n) __extension__					\
138 ({									\
139 	u_int _i;							\
140 	__typeof(*DPCPU_PTR(n)) sum;					\
141 									\
142 	sum = 0;							\
143 	CPU_FOREACH(_i) {						\
144 		sum += *DPCPU_ID_PTR(_i, n);				\
145 	}								\
146 	sum;								\
147 })
148 
149 #define	DPCPU_VARSUM(n, var) __extension__				\
150 ({									\
151 	u_int _i;							\
152 	__typeof((DPCPU_PTR(n))->var) sum;				\
153 									\
154 	sum = 0;							\
155 	CPU_FOREACH(_i) {						\
156 		sum += (DPCPU_ID_PTR(_i, n))->var;			\
157 	}								\
158 	sum;								\
159 })
160 
161 #define	DPCPU_ZERO(n) do {						\
162 	u_int _i;							\
163 									\
164 	CPU_FOREACH(_i) {						\
165 		bzero(DPCPU_ID_PTR(_i, n), sizeof(*DPCPU_PTR(n)));	\
166 	}								\
167 } while(0)
168 
169 #endif /* _KERNEL */
170 
171 /*
172  * This structure maps out the global data that needs to be kept on a
173  * per-cpu basis.  The members are accessed via the PCPU_GET/SET/PTR
174  * macros defined in <machine/pcpu.h>.  Machine dependent fields are
175  * defined in the PCPU_MD_FIELDS macro defined in <machine/pcpu.h>.
176  */
177 struct pcpu {
178 	struct thread	*pc_curthread;		/* Current thread */
179 	struct thread	*pc_idlethread;		/* Idle thread */
180 	struct thread	*pc_fpcurthread;	/* Fp state owner */
181 	struct thread	*pc_deadthread;		/* Zombie thread or NULL */
182 	struct pcb	*pc_curpcb;		/* Current pcb */
183 	void		*pc_sched;		/* Scheduler state */
184 	uint64_t	pc_switchtime;		/* cpu_ticks() at last csw */
185 	int		pc_switchticks;		/* `ticks' at last csw */
186 	u_int		pc_cpuid;		/* This cpu number */
187 	STAILQ_ENTRY(pcpu) pc_allcpu;
188 	struct lock_list_entry *pc_spinlocks;
189 	long		pc_cp_time[CPUSTATES];	/* statclock ticks */
190 	struct device	*pc_device;
191 	void		*pc_netisr;		/* netisr SWI cookie */
192 	int		pc_unused1;		/* unused field */
193 	int		pc_domain;		/* Memory domain. */
194 	struct rm_queue	pc_rm_queue;		/* rmlock list of trackers */
195 	uintptr_t	pc_dynamic;		/* Dynamic per-cpu data area */
196 	uint64_t	pc_early_dummy_counter;	/* Startup time counter(9) */
197 
198 	/*
199 	 * Keep MD fields last, so that CPU-specific variations on a
200 	 * single architecture don't result in offset variations of
201 	 * the machine-independent fields of the pcpu.  Even though
202 	 * the pcpu structure is private to the kernel, some ports
203 	 * (e.g., lsof, part of gtop) define _KERNEL and include this
204 	 * header.  While strictly speaking this is wrong, there's no
205 	 * reason not to keep the offsets of the MI fields constant
206 	 * if only to make kernel debugging easier.
207 	 */
208 	PCPU_MD_FIELDS;
209 } __aligned(CACHE_LINE_SIZE);
210 
211 #ifdef _KERNEL
212 
213 STAILQ_HEAD(cpuhead, pcpu);
214 
215 extern struct cpuhead cpuhead;
216 extern struct pcpu *cpuid_to_pcpu[];
217 
218 #define	curcpu		PCPU_GET(cpuid)
219 #define	curvidata	PCPU_GET(vidata)
220 
221 #define UMA_PCPU_ALLOC_SIZE		PAGE_SIZE
222 
223 #include <machine/pcpu_aux.h>
224 
225 #ifndef curthread
226 #define	curthread	PCPU_GET(curthread)
227 #endif
228 #define	curproc		(curthread->td_proc)
229 
230 /* Accessor to elements allocated via UMA_ZONE_PCPU zone. */
231 static inline void *
232 zpcpu_get(void *base)
233 {
234 
235 	return ((char *)(base) + UMA_PCPU_ALLOC_SIZE * curcpu);
236 }
237 
238 static inline void *
239 zpcpu_get_cpu(void *base, int cpu)
240 {
241 
242 	return ((char *)(base) + UMA_PCPU_ALLOC_SIZE * cpu);
243 }
244 
245 /*
246  * Machine dependent callouts.  cpu_pcpu_init() is responsible for
247  * initializing machine dependent fields of struct pcpu, and
248  * db_show_mdpcpu() is responsible for handling machine dependent
249  * fields for the DDB 'show pcpu' command.
250  */
251 void	cpu_pcpu_init(struct pcpu *pcpu, int cpuid, size_t size);
252 void	db_show_mdpcpu(struct pcpu *pcpu);
253 
254 void	*dpcpu_alloc(int size);
255 void	dpcpu_copy(void *s, int size);
256 void	dpcpu_free(void *s, int size);
257 void	dpcpu_init(void *dpcpu, int cpuid);
258 void	pcpu_destroy(struct pcpu *pcpu);
259 struct	pcpu *pcpu_find(u_int cpuid);
260 void	pcpu_init(struct pcpu *pcpu, int cpuid, size_t size);
261 
262 #endif /* _KERNEL */
263 
264 #endif /* !_SYS_PCPU_H_ */
265