xref: /linux/arch/x86/include/asm/atomic64_32.h (revision 44f57d78)
1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef _ASM_X86_ATOMIC64_32_H
3 #define _ASM_X86_ATOMIC64_32_H
4 
5 #include <linux/compiler.h>
6 #include <linux/types.h>
7 //#include <asm/cmpxchg.h>
8 
9 /* An 64bit atomic type */
10 
11 typedef struct {
12 	u64 __aligned(8) counter;
13 } atomic64_t;
14 
15 #define ATOMIC64_INIT(val)	{ (val) }
16 
17 #define __ATOMIC64_DECL(sym) void atomic64_##sym(atomic64_t *, ...)
18 #ifndef ATOMIC64_EXPORT
19 #define ATOMIC64_DECL_ONE __ATOMIC64_DECL
20 #else
21 #define ATOMIC64_DECL_ONE(sym) __ATOMIC64_DECL(sym); \
22 	ATOMIC64_EXPORT(atomic64_##sym)
23 #endif
24 
25 #ifdef CONFIG_X86_CMPXCHG64
26 #define __alternative_atomic64(f, g, out, in...) \
27 	asm volatile("call %P[func]" \
28 		     : out : [func] "i" (atomic64_##g##_cx8), ## in)
29 
30 #define ATOMIC64_DECL(sym) ATOMIC64_DECL_ONE(sym##_cx8)
31 #else
32 #define __alternative_atomic64(f, g, out, in...) \
33 	alternative_call(atomic64_##f##_386, atomic64_##g##_cx8, \
34 			 X86_FEATURE_CX8, ASM_OUTPUT2(out), ## in)
35 
36 #define ATOMIC64_DECL(sym) ATOMIC64_DECL_ONE(sym##_cx8); \
37 	ATOMIC64_DECL_ONE(sym##_386)
38 
39 ATOMIC64_DECL_ONE(add_386);
40 ATOMIC64_DECL_ONE(sub_386);
41 ATOMIC64_DECL_ONE(inc_386);
42 ATOMIC64_DECL_ONE(dec_386);
43 #endif
44 
45 #define alternative_atomic64(f, out, in...) \
46 	__alternative_atomic64(f, f, ASM_OUTPUT2(out), ## in)
47 
48 ATOMIC64_DECL(read);
49 ATOMIC64_DECL(set);
50 ATOMIC64_DECL(xchg);
51 ATOMIC64_DECL(add_return);
52 ATOMIC64_DECL(sub_return);
53 ATOMIC64_DECL(inc_return);
54 ATOMIC64_DECL(dec_return);
55 ATOMIC64_DECL(dec_if_positive);
56 ATOMIC64_DECL(inc_not_zero);
57 ATOMIC64_DECL(add_unless);
58 
59 #undef ATOMIC64_DECL
60 #undef ATOMIC64_DECL_ONE
61 #undef __ATOMIC64_DECL
62 #undef ATOMIC64_EXPORT
63 
64 /**
65  * arch_atomic64_cmpxchg - cmpxchg atomic64 variable
66  * @v: pointer to type atomic64_t
67  * @o: expected value
68  * @n: new value
69  *
70  * Atomically sets @v to @n if it was equal to @o and returns
71  * the old value.
72  */
73 
74 static inline long long arch_atomic64_cmpxchg(atomic64_t *v, long long o,
75 					      long long n)
76 {
77 	return arch_cmpxchg64(&v->counter, o, n);
78 }
79 
80 /**
81  * arch_atomic64_xchg - xchg atomic64 variable
82  * @v: pointer to type atomic64_t
83  * @n: value to assign
84  *
85  * Atomically xchgs the value of @v to @n and returns
86  * the old value.
87  */
88 static inline long long arch_atomic64_xchg(atomic64_t *v, long long n)
89 {
90 	long long o;
91 	unsigned high = (unsigned)(n >> 32);
92 	unsigned low = (unsigned)n;
93 	alternative_atomic64(xchg, "=&A" (o),
94 			     "S" (v), "b" (low), "c" (high)
95 			     : "memory");
96 	return o;
97 }
98 
99 /**
100  * arch_atomic64_set - set atomic64 variable
101  * @v: pointer to type atomic64_t
102  * @i: value to assign
103  *
104  * Atomically sets the value of @v to @n.
105  */
106 static inline void arch_atomic64_set(atomic64_t *v, long long i)
107 {
108 	unsigned high = (unsigned)(i >> 32);
109 	unsigned low = (unsigned)i;
110 	alternative_atomic64(set, /* no output */,
111 			     "S" (v), "b" (low), "c" (high)
112 			     : "eax", "edx", "memory");
113 }
114 
115 /**
116  * arch_atomic64_read - read atomic64 variable
117  * @v: pointer to type atomic64_t
118  *
119  * Atomically reads the value of @v and returns it.
120  */
121 static inline long long arch_atomic64_read(const atomic64_t *v)
122 {
123 	long long r;
124 	alternative_atomic64(read, "=&A" (r), "c" (v) : "memory");
125 	return r;
126 }
127 
128 /**
129  * arch_atomic64_add_return - add and return
130  * @i: integer value to add
131  * @v: pointer to type atomic64_t
132  *
133  * Atomically adds @i to @v and returns @i + *@v
134  */
135 static inline long long arch_atomic64_add_return(long long i, atomic64_t *v)
136 {
137 	alternative_atomic64(add_return,
138 			     ASM_OUTPUT2("+A" (i), "+c" (v)),
139 			     ASM_NO_INPUT_CLOBBER("memory"));
140 	return i;
141 }
142 
143 /*
144  * Other variants with different arithmetic operators:
145  */
146 static inline long long arch_atomic64_sub_return(long long i, atomic64_t *v)
147 {
148 	alternative_atomic64(sub_return,
149 			     ASM_OUTPUT2("+A" (i), "+c" (v)),
150 			     ASM_NO_INPUT_CLOBBER("memory"));
151 	return i;
152 }
153 
154 static inline long long arch_atomic64_inc_return(atomic64_t *v)
155 {
156 	long long a;
157 	alternative_atomic64(inc_return, "=&A" (a),
158 			     "S" (v) : "memory", "ecx");
159 	return a;
160 }
161 #define arch_atomic64_inc_return arch_atomic64_inc_return
162 
163 static inline long long arch_atomic64_dec_return(atomic64_t *v)
164 {
165 	long long a;
166 	alternative_atomic64(dec_return, "=&A" (a),
167 			     "S" (v) : "memory", "ecx");
168 	return a;
169 }
170 #define arch_atomic64_dec_return arch_atomic64_dec_return
171 
172 /**
173  * arch_atomic64_add - add integer to atomic64 variable
174  * @i: integer value to add
175  * @v: pointer to type atomic64_t
176  *
177  * Atomically adds @i to @v.
178  */
179 static inline long long arch_atomic64_add(long long i, atomic64_t *v)
180 {
181 	__alternative_atomic64(add, add_return,
182 			       ASM_OUTPUT2("+A" (i), "+c" (v)),
183 			       ASM_NO_INPUT_CLOBBER("memory"));
184 	return i;
185 }
186 
187 /**
188  * arch_atomic64_sub - subtract the atomic64 variable
189  * @i: integer value to subtract
190  * @v: pointer to type atomic64_t
191  *
192  * Atomically subtracts @i from @v.
193  */
194 static inline long long arch_atomic64_sub(long long i, atomic64_t *v)
195 {
196 	__alternative_atomic64(sub, sub_return,
197 			       ASM_OUTPUT2("+A" (i), "+c" (v)),
198 			       ASM_NO_INPUT_CLOBBER("memory"));
199 	return i;
200 }
201 
202 /**
203  * arch_atomic64_inc - increment atomic64 variable
204  * @v: pointer to type atomic64_t
205  *
206  * Atomically increments @v by 1.
207  */
208 static inline void arch_atomic64_inc(atomic64_t *v)
209 {
210 	__alternative_atomic64(inc, inc_return, /* no output */,
211 			       "S" (v) : "memory", "eax", "ecx", "edx");
212 }
213 #define arch_atomic64_inc arch_atomic64_inc
214 
215 /**
216  * arch_atomic64_dec - decrement atomic64 variable
217  * @v: pointer to type atomic64_t
218  *
219  * Atomically decrements @v by 1.
220  */
221 static inline void arch_atomic64_dec(atomic64_t *v)
222 {
223 	__alternative_atomic64(dec, dec_return, /* no output */,
224 			       "S" (v) : "memory", "eax", "ecx", "edx");
225 }
226 #define arch_atomic64_dec arch_atomic64_dec
227 
228 /**
229  * arch_atomic64_add_unless - add unless the number is a given value
230  * @v: pointer of type atomic64_t
231  * @a: the amount to add to v...
232  * @u: ...unless v is equal to u.
233  *
234  * Atomically adds @a to @v, so long as it was not @u.
235  * Returns non-zero if the add was done, zero otherwise.
236  */
237 static inline int arch_atomic64_add_unless(atomic64_t *v, long long a,
238 					   long long u)
239 {
240 	unsigned low = (unsigned)u;
241 	unsigned high = (unsigned)(u >> 32);
242 	alternative_atomic64(add_unless,
243 			     ASM_OUTPUT2("+A" (a), "+c" (low), "+D" (high)),
244 			     "S" (v) : "memory");
245 	return (int)a;
246 }
247 
248 static inline int arch_atomic64_inc_not_zero(atomic64_t *v)
249 {
250 	int r;
251 	alternative_atomic64(inc_not_zero, "=&a" (r),
252 			     "S" (v) : "ecx", "edx", "memory");
253 	return r;
254 }
255 #define arch_atomic64_inc_not_zero arch_atomic64_inc_not_zero
256 
257 static inline long long arch_atomic64_dec_if_positive(atomic64_t *v)
258 {
259 	long long r;
260 	alternative_atomic64(dec_if_positive, "=&A" (r),
261 			     "S" (v) : "ecx", "memory");
262 	return r;
263 }
264 #define arch_atomic64_dec_if_positive arch_atomic64_dec_if_positive
265 
266 #undef alternative_atomic64
267 #undef __alternative_atomic64
268 
269 static inline void arch_atomic64_and(long long i, atomic64_t *v)
270 {
271 	long long old, c = 0;
272 
273 	while ((old = arch_atomic64_cmpxchg(v, c, c & i)) != c)
274 		c = old;
275 }
276 
277 static inline long long arch_atomic64_fetch_and(long long i, atomic64_t *v)
278 {
279 	long long old, c = 0;
280 
281 	while ((old = arch_atomic64_cmpxchg(v, c, c & i)) != c)
282 		c = old;
283 
284 	return old;
285 }
286 
287 static inline void arch_atomic64_or(long long i, atomic64_t *v)
288 {
289 	long long old, c = 0;
290 
291 	while ((old = arch_atomic64_cmpxchg(v, c, c | i)) != c)
292 		c = old;
293 }
294 
295 static inline long long arch_atomic64_fetch_or(long long i, atomic64_t *v)
296 {
297 	long long old, c = 0;
298 
299 	while ((old = arch_atomic64_cmpxchg(v, c, c | i)) != c)
300 		c = old;
301 
302 	return old;
303 }
304 
305 static inline void arch_atomic64_xor(long long i, atomic64_t *v)
306 {
307 	long long old, c = 0;
308 
309 	while ((old = arch_atomic64_cmpxchg(v, c, c ^ i)) != c)
310 		c = old;
311 }
312 
313 static inline long long arch_atomic64_fetch_xor(long long i, atomic64_t *v)
314 {
315 	long long old, c = 0;
316 
317 	while ((old = arch_atomic64_cmpxchg(v, c, c ^ i)) != c)
318 		c = old;
319 
320 	return old;
321 }
322 
323 static inline long long arch_atomic64_fetch_add(long long i, atomic64_t *v)
324 {
325 	long long old, c = 0;
326 
327 	while ((old = arch_atomic64_cmpxchg(v, c, c + i)) != c)
328 		c = old;
329 
330 	return old;
331 }
332 
333 #define arch_atomic64_fetch_sub(i, v)	arch_atomic64_fetch_add(-(i), (v))
334 
335 #endif /* _ASM_X86_ATOMIC64_32_H */
336