xref: /linux/arch/s390/include/asm/percpu.h (revision 4b132aacb0768ac1e652cf517097ea6f237214b9)
1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef __ARCH_S390_PERCPU__
3 #define __ARCH_S390_PERCPU__
4 
5 #include <linux/preempt.h>
6 #include <asm/cmpxchg.h>
7 
8 /*
9  * s390 uses its own implementation for per cpu data, the offset of
10  * the cpu local data area is cached in the cpu's lowcore memory.
11  */
12 #define __my_cpu_offset get_lowcore()->percpu_offset
13 
14 /*
15  * For 64 bit module code, the module may be more than 4G above the
16  * per cpu area, use weak definitions to force the compiler to
17  * generate external references.
18  */
19 #if defined(MODULE)
20 #define ARCH_NEEDS_WEAK_PER_CPU
21 #endif
22 
23 /*
24  * We use a compare-and-swap loop since that uses less cpu cycles than
25  * disabling and enabling interrupts like the generic variant would do.
26  */
27 #define arch_this_cpu_to_op_simple(pcp, val, op)			\
28 ({									\
29 	typedef typeof(pcp) pcp_op_T__;					\
30 	pcp_op_T__ old__, new__, prev__;				\
31 	pcp_op_T__ *ptr__;						\
32 	preempt_disable_notrace();					\
33 	ptr__ = raw_cpu_ptr(&(pcp));					\
34 	prev__ = READ_ONCE(*ptr__);					\
35 	do {								\
36 		old__ = prev__;						\
37 		new__ = old__ op (val);					\
38 		prev__ = cmpxchg(ptr__, old__, new__);			\
39 	} while (prev__ != old__);					\
40 	preempt_enable_notrace();					\
41 	new__;								\
42 })
43 
44 #define this_cpu_add_1(pcp, val)	arch_this_cpu_to_op_simple(pcp, val, +)
45 #define this_cpu_add_2(pcp, val)	arch_this_cpu_to_op_simple(pcp, val, +)
46 #define this_cpu_add_return_1(pcp, val) arch_this_cpu_to_op_simple(pcp, val, +)
47 #define this_cpu_add_return_2(pcp, val) arch_this_cpu_to_op_simple(pcp, val, +)
48 #define this_cpu_and_1(pcp, val)	arch_this_cpu_to_op_simple(pcp, val, &)
49 #define this_cpu_and_2(pcp, val)	arch_this_cpu_to_op_simple(pcp, val, &)
50 #define this_cpu_or_1(pcp, val)		arch_this_cpu_to_op_simple(pcp, val, |)
51 #define this_cpu_or_2(pcp, val)		arch_this_cpu_to_op_simple(pcp, val, |)
52 
53 #ifndef CONFIG_HAVE_MARCH_Z196_FEATURES
54 
55 #define this_cpu_add_4(pcp, val)	arch_this_cpu_to_op_simple(pcp, val, +)
56 #define this_cpu_add_8(pcp, val)	arch_this_cpu_to_op_simple(pcp, val, +)
57 #define this_cpu_add_return_4(pcp, val) arch_this_cpu_to_op_simple(pcp, val, +)
58 #define this_cpu_add_return_8(pcp, val) arch_this_cpu_to_op_simple(pcp, val, +)
59 #define this_cpu_and_4(pcp, val)	arch_this_cpu_to_op_simple(pcp, val, &)
60 #define this_cpu_and_8(pcp, val)	arch_this_cpu_to_op_simple(pcp, val, &)
61 #define this_cpu_or_4(pcp, val)		arch_this_cpu_to_op_simple(pcp, val, |)
62 #define this_cpu_or_8(pcp, val)		arch_this_cpu_to_op_simple(pcp, val, |)
63 
64 #else /* CONFIG_HAVE_MARCH_Z196_FEATURES */
65 
66 #define arch_this_cpu_add(pcp, val, op1, op2, szcast)			\
67 {									\
68 	typedef typeof(pcp) pcp_op_T__; 				\
69 	pcp_op_T__ val__ = (val);					\
70 	pcp_op_T__ old__, *ptr__;					\
71 	preempt_disable_notrace();					\
72 	ptr__ = raw_cpu_ptr(&(pcp)); 				\
73 	if (__builtin_constant_p(val__) &&				\
74 	    ((szcast)val__ > -129) && ((szcast)val__ < 128)) {		\
75 		asm volatile(						\
76 			op2 "   %[ptr__],%[val__]\n"			\
77 			: [ptr__] "+Q" (*ptr__) 			\
78 			: [val__] "i" ((szcast)val__)			\
79 			: "cc");					\
80 	} else {							\
81 		asm volatile(						\
82 			op1 "   %[old__],%[val__],%[ptr__]\n"		\
83 			: [old__] "=d" (old__), [ptr__] "+Q" (*ptr__)	\
84 			: [val__] "d" (val__)				\
85 			: "cc");					\
86 	}								\
87 	preempt_enable_notrace();					\
88 }
89 
90 #define this_cpu_add_4(pcp, val) arch_this_cpu_add(pcp, val, "laa", "asi", int)
91 #define this_cpu_add_8(pcp, val) arch_this_cpu_add(pcp, val, "laag", "agsi", long)
92 
93 #define arch_this_cpu_add_return(pcp, val, op)				\
94 ({									\
95 	typedef typeof(pcp) pcp_op_T__; 				\
96 	pcp_op_T__ val__ = (val);					\
97 	pcp_op_T__ old__, *ptr__;					\
98 	preempt_disable_notrace();					\
99 	ptr__ = raw_cpu_ptr(&(pcp));	 				\
100 	asm volatile(							\
101 		op "    %[old__],%[val__],%[ptr__]\n"			\
102 		: [old__] "=d" (old__), [ptr__] "+Q" (*ptr__)		\
103 		: [val__] "d" (val__)					\
104 		: "cc");						\
105 	preempt_enable_notrace();						\
106 	old__ + val__;							\
107 })
108 
109 #define this_cpu_add_return_4(pcp, val) arch_this_cpu_add_return(pcp, val, "laa")
110 #define this_cpu_add_return_8(pcp, val) arch_this_cpu_add_return(pcp, val, "laag")
111 
112 #define arch_this_cpu_to_op(pcp, val, op)				\
113 {									\
114 	typedef typeof(pcp) pcp_op_T__; 				\
115 	pcp_op_T__ val__ = (val);					\
116 	pcp_op_T__ old__, *ptr__;					\
117 	preempt_disable_notrace();					\
118 	ptr__ = raw_cpu_ptr(&(pcp));	 				\
119 	asm volatile(							\
120 		op "    %[old__],%[val__],%[ptr__]\n"			\
121 		: [old__] "=d" (old__), [ptr__] "+Q" (*ptr__)		\
122 		: [val__] "d" (val__)					\
123 		: "cc");						\
124 	preempt_enable_notrace();					\
125 }
126 
127 #define this_cpu_and_4(pcp, val)	arch_this_cpu_to_op(pcp, val, "lan")
128 #define this_cpu_and_8(pcp, val)	arch_this_cpu_to_op(pcp, val, "lang")
129 #define this_cpu_or_4(pcp, val)		arch_this_cpu_to_op(pcp, val, "lao")
130 #define this_cpu_or_8(pcp, val)		arch_this_cpu_to_op(pcp, val, "laog")
131 
132 #endif /* CONFIG_HAVE_MARCH_Z196_FEATURES */
133 
134 #define arch_this_cpu_cmpxchg(pcp, oval, nval)				\
135 ({									\
136 	typedef typeof(pcp) pcp_op_T__;					\
137 	pcp_op_T__ ret__;						\
138 	pcp_op_T__ *ptr__;						\
139 	preempt_disable_notrace();					\
140 	ptr__ = raw_cpu_ptr(&(pcp));					\
141 	ret__ = cmpxchg(ptr__, oval, nval);				\
142 	preempt_enable_notrace();					\
143 	ret__;								\
144 })
145 
146 #define this_cpu_cmpxchg_1(pcp, oval, nval) arch_this_cpu_cmpxchg(pcp, oval, nval)
147 #define this_cpu_cmpxchg_2(pcp, oval, nval) arch_this_cpu_cmpxchg(pcp, oval, nval)
148 #define this_cpu_cmpxchg_4(pcp, oval, nval) arch_this_cpu_cmpxchg(pcp, oval, nval)
149 #define this_cpu_cmpxchg_8(pcp, oval, nval) arch_this_cpu_cmpxchg(pcp, oval, nval)
150 
151 #define this_cpu_cmpxchg64(pcp, o, n)	this_cpu_cmpxchg_8(pcp, o, n)
152 
153 #define this_cpu_cmpxchg128(pcp, oval, nval)				\
154 ({									\
155 	typedef typeof(pcp) pcp_op_T__;					\
156 	u128 old__, new__, ret__;					\
157 	pcp_op_T__ *ptr__;						\
158 	old__ = oval;							\
159 	new__ = nval;							\
160 	preempt_disable_notrace();					\
161 	ptr__ = raw_cpu_ptr(&(pcp));					\
162 	ret__ = cmpxchg128((void *)ptr__, old__, new__);		\
163 	preempt_enable_notrace();					\
164 	ret__;								\
165 })
166 
167 #define arch_this_cpu_xchg(pcp, nval)					\
168 ({									\
169 	typeof(pcp) *ptr__;						\
170 	typeof(pcp) ret__;						\
171 	preempt_disable_notrace();					\
172 	ptr__ = raw_cpu_ptr(&(pcp));					\
173 	ret__ = xchg(ptr__, nval);					\
174 	preempt_enable_notrace();					\
175 	ret__;								\
176 })
177 
178 #define this_cpu_xchg_1(pcp, nval) arch_this_cpu_xchg(pcp, nval)
179 #define this_cpu_xchg_2(pcp, nval) arch_this_cpu_xchg(pcp, nval)
180 #define this_cpu_xchg_4(pcp, nval) arch_this_cpu_xchg(pcp, nval)
181 #define this_cpu_xchg_8(pcp, nval) arch_this_cpu_xchg(pcp, nval)
182 
183 #include <asm-generic/percpu.h>
184 
185 #endif /* __ARCH_S390_PERCPU__ */
186