xref: /linux/arch/x86/include/asm/atomic64_32.h (revision 649e441f49d4bfb717e9467950891dc452f4e063)
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  * 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 atomic64_cmpxchg(atomic64_t *v, long long o, long long n)
75 {
76 	return cmpxchg64(&v->counter, o, n);
77 }
78 
79 /**
80  * atomic64_xchg - xchg atomic64 variable
81  * @v: pointer to type atomic64_t
82  * @n: value to assign
83  *
84  * Atomically xchgs the value of @v to @n and returns
85  * the old value.
86  */
87 static inline long long atomic64_xchg(atomic64_t *v, long long n)
88 {
89 	long long o;
90 	unsigned high = (unsigned)(n >> 32);
91 	unsigned low = (unsigned)n;
92 	alternative_atomic64(xchg, "=&A" (o),
93 			     "S" (v), "b" (low), "c" (high)
94 			     : "memory");
95 	return o;
96 }
97 
98 /**
99  * atomic64_set - set atomic64 variable
100  * @v: pointer to type atomic64_t
101  * @i: value to assign
102  *
103  * Atomically sets the value of @v to @n.
104  */
105 static inline void atomic64_set(atomic64_t *v, long long i)
106 {
107 	unsigned high = (unsigned)(i >> 32);
108 	unsigned low = (unsigned)i;
109 	alternative_atomic64(set, /* no output */,
110 			     "S" (v), "b" (low), "c" (high)
111 			     : "eax", "edx", "memory");
112 }
113 
114 /**
115  * atomic64_read - read atomic64 variable
116  * @v: pointer to type atomic64_t
117  *
118  * Atomically reads the value of @v and returns it.
119  */
120 static inline long long atomic64_read(const atomic64_t *v)
121 {
122 	long long r;
123 	alternative_atomic64(read, "=&A" (r), "c" (v) : "memory");
124 	return r;
125  }
126 
127 /**
128  * atomic64_add_return - add and return
129  * @i: integer value to add
130  * @v: pointer to type atomic64_t
131  *
132  * Atomically adds @i to @v and returns @i + *@v
133  */
134 static inline long long atomic64_add_return(long long i, atomic64_t *v)
135 {
136 	alternative_atomic64(add_return,
137 			     ASM_OUTPUT2("+A" (i), "+c" (v)),
138 			     ASM_NO_INPUT_CLOBBER("memory"));
139 	return i;
140 }
141 
142 /*
143  * Other variants with different arithmetic operators:
144  */
145 static inline long long atomic64_sub_return(long long i, atomic64_t *v)
146 {
147 	alternative_atomic64(sub_return,
148 			     ASM_OUTPUT2("+A" (i), "+c" (v)),
149 			     ASM_NO_INPUT_CLOBBER("memory"));
150 	return i;
151 }
152 
153 static inline long long atomic64_inc_return(atomic64_t *v)
154 {
155 	long long a;
156 	alternative_atomic64(inc_return, "=&A" (a),
157 			     "S" (v) : "memory", "ecx");
158 	return a;
159 }
160 
161 static inline long long atomic64_dec_return(atomic64_t *v)
162 {
163 	long long a;
164 	alternative_atomic64(dec_return, "=&A" (a),
165 			     "S" (v) : "memory", "ecx");
166 	return a;
167 }
168 
169 /**
170  * atomic64_add - add integer to atomic64 variable
171  * @i: integer value to add
172  * @v: pointer to type atomic64_t
173  *
174  * Atomically adds @i to @v.
175  */
176 static inline long long atomic64_add(long long i, atomic64_t *v)
177 {
178 	__alternative_atomic64(add, add_return,
179 			       ASM_OUTPUT2("+A" (i), "+c" (v)),
180 			       ASM_NO_INPUT_CLOBBER("memory"));
181 	return i;
182 }
183 
184 /**
185  * atomic64_sub - subtract the atomic64 variable
186  * @i: integer value to subtract
187  * @v: pointer to type atomic64_t
188  *
189  * Atomically subtracts @i from @v.
190  */
191 static inline long long atomic64_sub(long long i, atomic64_t *v)
192 {
193 	__alternative_atomic64(sub, sub_return,
194 			       ASM_OUTPUT2("+A" (i), "+c" (v)),
195 			       ASM_NO_INPUT_CLOBBER("memory"));
196 	return i;
197 }
198 
199 /**
200  * atomic64_sub_and_test - subtract value from variable and test result
201  * @i: integer value to subtract
202  * @v: pointer to type atomic64_t
203  *
204  * Atomically subtracts @i from @v and returns
205  * true if the result is zero, or false for all
206  * other cases.
207  */
208 static inline int atomic64_sub_and_test(long long i, atomic64_t *v)
209 {
210 	return atomic64_sub_return(i, v) == 0;
211 }
212 
213 /**
214  * atomic64_inc - increment atomic64 variable
215  * @v: pointer to type atomic64_t
216  *
217  * Atomically increments @v by 1.
218  */
219 static inline void atomic64_inc(atomic64_t *v)
220 {
221 	__alternative_atomic64(inc, inc_return, /* no output */,
222 			       "S" (v) : "memory", "eax", "ecx", "edx");
223 }
224 
225 /**
226  * atomic64_dec - decrement atomic64 variable
227  * @v: pointer to type atomic64_t
228  *
229  * Atomically decrements @v by 1.
230  */
231 static inline void atomic64_dec(atomic64_t *v)
232 {
233 	__alternative_atomic64(dec, dec_return, /* no output */,
234 			       "S" (v) : "memory", "eax", "ecx", "edx");
235 }
236 
237 /**
238  * atomic64_dec_and_test - decrement and test
239  * @v: pointer to type atomic64_t
240  *
241  * Atomically decrements @v by 1 and
242  * returns true if the result is 0, or false for all other
243  * cases.
244  */
245 static inline int atomic64_dec_and_test(atomic64_t *v)
246 {
247 	return atomic64_dec_return(v) == 0;
248 }
249 
250 /**
251  * atomic64_inc_and_test - increment and test
252  * @v: pointer to type atomic64_t
253  *
254  * Atomically increments @v by 1
255  * and returns true if the result is zero, or false for all
256  * other cases.
257  */
258 static inline int atomic64_inc_and_test(atomic64_t *v)
259 {
260 	return atomic64_inc_return(v) == 0;
261 }
262 
263 /**
264  * atomic64_add_negative - add and test if negative
265  * @i: integer value to add
266  * @v: pointer to type atomic64_t
267  *
268  * Atomically adds @i to @v and returns true
269  * if the result is negative, or false when
270  * result is greater than or equal to zero.
271  */
272 static inline int atomic64_add_negative(long long i, atomic64_t *v)
273 {
274 	return atomic64_add_return(i, v) < 0;
275 }
276 
277 /**
278  * atomic64_add_unless - add unless the number is a given value
279  * @v: pointer of type atomic64_t
280  * @a: the amount to add to v...
281  * @u: ...unless v is equal to u.
282  *
283  * Atomically adds @a to @v, so long as it was not @u.
284  * Returns non-zero if the add was done, zero otherwise.
285  */
286 static inline int atomic64_add_unless(atomic64_t *v, long long a, long long u)
287 {
288 	unsigned low = (unsigned)u;
289 	unsigned high = (unsigned)(u >> 32);
290 	alternative_atomic64(add_unless,
291 			     ASM_OUTPUT2("+A" (a), "+c" (low), "+D" (high)),
292 			     "S" (v) : "memory");
293 	return (int)a;
294 }
295 
296 
297 static inline int atomic64_inc_not_zero(atomic64_t *v)
298 {
299 	int r;
300 	alternative_atomic64(inc_not_zero, "=&a" (r),
301 			     "S" (v) : "ecx", "edx", "memory");
302 	return r;
303 }
304 
305 static inline long long atomic64_dec_if_positive(atomic64_t *v)
306 {
307 	long long r;
308 	alternative_atomic64(dec_if_positive, "=&A" (r),
309 			     "S" (v) : "ecx", "memory");
310 	return r;
311 }
312 
313 #undef alternative_atomic64
314 #undef __alternative_atomic64
315 
316 static inline void atomic64_and(long long i, atomic64_t *v)
317 {
318 	long long old, c = 0;
319 
320 	while ((old = atomic64_cmpxchg(v, c, c & i)) != c)
321 		c = old;
322 }
323 
324 static inline long long atomic64_fetch_and(long long i, atomic64_t *v)
325 {
326 	long long old, c = 0;
327 
328 	while ((old = atomic64_cmpxchg(v, c, c & i)) != c)
329 		c = old;
330 
331 	return old;
332 }
333 
334 static inline void atomic64_or(long long i, atomic64_t *v)
335 {
336 	long long old, c = 0;
337 
338 	while ((old = atomic64_cmpxchg(v, c, c | i)) != c)
339 		c = old;
340 }
341 
342 static inline long long atomic64_fetch_or(long long i, atomic64_t *v)
343 {
344 	long long old, c = 0;
345 
346 	while ((old = atomic64_cmpxchg(v, c, c | i)) != c)
347 		c = old;
348 
349 	return old;
350 }
351 
352 static inline void atomic64_xor(long long i, atomic64_t *v)
353 {
354 	long long old, c = 0;
355 
356 	while ((old = atomic64_cmpxchg(v, c, c ^ i)) != c)
357 		c = old;
358 }
359 
360 static inline long long atomic64_fetch_xor(long long i, atomic64_t *v)
361 {
362 	long long old, c = 0;
363 
364 	while ((old = atomic64_cmpxchg(v, c, c ^ i)) != c)
365 		c = old;
366 
367 	return old;
368 }
369 
370 static inline long long atomic64_fetch_add(long long i, atomic64_t *v)
371 {
372 	long long old, c = 0;
373 
374 	while ((old = atomic64_cmpxchg(v, c, c + i)) != c)
375 		c = old;
376 
377 	return old;
378 }
379 
380 #define atomic64_fetch_sub(i, v)	atomic64_fetch_add(-(i), (v))
381 
382 #endif /* _ASM_X86_ATOMIC64_32_H */
383