cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

atomic_ll_sc.h (10754B)


      1/* SPDX-License-Identifier: GPL-2.0-only */
      2/*
      3 * Based on arch/arm/include/asm/atomic.h
      4 *
      5 * Copyright (C) 1996 Russell King.
      6 * Copyright (C) 2002 Deep Blue Solutions Ltd.
      7 * Copyright (C) 2012 ARM Ltd.
      8 */
      9
     10#ifndef __ASM_ATOMIC_LL_SC_H
     11#define __ASM_ATOMIC_LL_SC_H
     12
     13#include <linux/stringify.h>
     14
     15#ifdef CONFIG_ARM64_LSE_ATOMICS
     16#define __LL_SC_FALLBACK(asm_ops)					\
     17"	b	3f\n"							\
     18"	.subsection	1\n"						\
     19"3:\n"									\
     20asm_ops "\n"								\
     21"	b	4f\n"							\
     22"	.previous\n"							\
     23"4:\n"
     24#else
     25#define __LL_SC_FALLBACK(asm_ops) asm_ops
     26#endif
     27
     28#ifndef CONFIG_CC_HAS_K_CONSTRAINT
     29#define K
     30#endif
     31
     32/*
     33 * AArch64 UP and SMP safe atomic ops.  We use load exclusive and
     34 * store exclusive to ensure that these are atomic.  We may loop
     35 * to ensure that the update happens.
     36 */
     37
     38#define ATOMIC_OP(op, asm_op, constraint)				\
     39static inline void							\
     40__ll_sc_atomic_##op(int i, atomic_t *v)					\
     41{									\
     42	unsigned long tmp;						\
     43	int result;							\
     44									\
     45	asm volatile("// atomic_" #op "\n"				\
     46	__LL_SC_FALLBACK(						\
     47	"	prfm	pstl1strm, %2\n"				\
     48	"1:	ldxr	%w0, %2\n"					\
     49	"	" #asm_op "	%w0, %w0, %w3\n"			\
     50	"	stxr	%w1, %w0, %2\n"					\
     51	"	cbnz	%w1, 1b\n")					\
     52	: "=&r" (result), "=&r" (tmp), "+Q" (v->counter)		\
     53	: __stringify(constraint) "r" (i));				\
     54}
     55
     56#define ATOMIC_OP_RETURN(name, mb, acq, rel, cl, op, asm_op, constraint)\
     57static inline int							\
     58__ll_sc_atomic_##op##_return##name(int i, atomic_t *v)			\
     59{									\
     60	unsigned long tmp;						\
     61	int result;							\
     62									\
     63	asm volatile("// atomic_" #op "_return" #name "\n"		\
     64	__LL_SC_FALLBACK(						\
     65	"	prfm	pstl1strm, %2\n"				\
     66	"1:	ld" #acq "xr	%w0, %2\n"				\
     67	"	" #asm_op "	%w0, %w0, %w3\n"			\
     68	"	st" #rel "xr	%w1, %w0, %2\n"				\
     69	"	cbnz	%w1, 1b\n"					\
     70	"	" #mb )							\
     71	: "=&r" (result), "=&r" (tmp), "+Q" (v->counter)		\
     72	: __stringify(constraint) "r" (i)				\
     73	: cl);								\
     74									\
     75	return result;							\
     76}
     77
     78#define ATOMIC_FETCH_OP(name, mb, acq, rel, cl, op, asm_op, constraint) \
     79static inline int							\
     80__ll_sc_atomic_fetch_##op##name(int i, atomic_t *v)			\
     81{									\
     82	unsigned long tmp;						\
     83	int val, result;						\
     84									\
     85	asm volatile("// atomic_fetch_" #op #name "\n"			\
     86	__LL_SC_FALLBACK(						\
     87	"	prfm	pstl1strm, %3\n"				\
     88	"1:	ld" #acq "xr	%w0, %3\n"				\
     89	"	" #asm_op "	%w1, %w0, %w4\n"			\
     90	"	st" #rel "xr	%w2, %w1, %3\n"				\
     91	"	cbnz	%w2, 1b\n"					\
     92	"	" #mb )							\
     93	: "=&r" (result), "=&r" (val), "=&r" (tmp), "+Q" (v->counter)	\
     94	: __stringify(constraint) "r" (i)				\
     95	: cl);								\
     96									\
     97	return result;							\
     98}
     99
    100#define ATOMIC_OPS(...)							\
    101	ATOMIC_OP(__VA_ARGS__)						\
    102	ATOMIC_OP_RETURN(        , dmb ish,  , l, "memory", __VA_ARGS__)\
    103	ATOMIC_OP_RETURN(_relaxed,        ,  ,  ,         , __VA_ARGS__)\
    104	ATOMIC_OP_RETURN(_acquire,        , a,  , "memory", __VA_ARGS__)\
    105	ATOMIC_OP_RETURN(_release,        ,  , l, "memory", __VA_ARGS__)\
    106	ATOMIC_FETCH_OP (        , dmb ish,  , l, "memory", __VA_ARGS__)\
    107	ATOMIC_FETCH_OP (_relaxed,        ,  ,  ,         , __VA_ARGS__)\
    108	ATOMIC_FETCH_OP (_acquire,        , a,  , "memory", __VA_ARGS__)\
    109	ATOMIC_FETCH_OP (_release,        ,  , l, "memory", __VA_ARGS__)
    110
    111ATOMIC_OPS(add, add, I)
    112ATOMIC_OPS(sub, sub, J)
    113
    114#undef ATOMIC_OPS
    115#define ATOMIC_OPS(...)							\
    116	ATOMIC_OP(__VA_ARGS__)						\
    117	ATOMIC_FETCH_OP (        , dmb ish,  , l, "memory", __VA_ARGS__)\
    118	ATOMIC_FETCH_OP (_relaxed,        ,  ,  ,         , __VA_ARGS__)\
    119	ATOMIC_FETCH_OP (_acquire,        , a,  , "memory", __VA_ARGS__)\
    120	ATOMIC_FETCH_OP (_release,        ,  , l, "memory", __VA_ARGS__)
    121
    122ATOMIC_OPS(and, and, K)
    123ATOMIC_OPS(or, orr, K)
    124ATOMIC_OPS(xor, eor, K)
    125/*
    126 * GAS converts the mysterious and undocumented BIC (immediate) alias to
    127 * an AND (immediate) instruction with the immediate inverted. We don't
    128 * have a constraint for this, so fall back to register.
    129 */
    130ATOMIC_OPS(andnot, bic, )
    131
    132#undef ATOMIC_OPS
    133#undef ATOMIC_FETCH_OP
    134#undef ATOMIC_OP_RETURN
    135#undef ATOMIC_OP
    136
    137#define ATOMIC64_OP(op, asm_op, constraint)				\
    138static inline void							\
    139__ll_sc_atomic64_##op(s64 i, atomic64_t *v)				\
    140{									\
    141	s64 result;							\
    142	unsigned long tmp;						\
    143									\
    144	asm volatile("// atomic64_" #op "\n"				\
    145	__LL_SC_FALLBACK(						\
    146	"	prfm	pstl1strm, %2\n"				\
    147	"1:	ldxr	%0, %2\n"					\
    148	"	" #asm_op "	%0, %0, %3\n"				\
    149	"	stxr	%w1, %0, %2\n"					\
    150	"	cbnz	%w1, 1b")					\
    151	: "=&r" (result), "=&r" (tmp), "+Q" (v->counter)		\
    152	: __stringify(constraint) "r" (i));				\
    153}
    154
    155#define ATOMIC64_OP_RETURN(name, mb, acq, rel, cl, op, asm_op, constraint)\
    156static inline long							\
    157__ll_sc_atomic64_##op##_return##name(s64 i, atomic64_t *v)		\
    158{									\
    159	s64 result;							\
    160	unsigned long tmp;						\
    161									\
    162	asm volatile("// atomic64_" #op "_return" #name "\n"		\
    163	__LL_SC_FALLBACK(						\
    164	"	prfm	pstl1strm, %2\n"				\
    165	"1:	ld" #acq "xr	%0, %2\n"				\
    166	"	" #asm_op "	%0, %0, %3\n"				\
    167	"	st" #rel "xr	%w1, %0, %2\n"				\
    168	"	cbnz	%w1, 1b\n"					\
    169	"	" #mb )							\
    170	: "=&r" (result), "=&r" (tmp), "+Q" (v->counter)		\
    171	: __stringify(constraint) "r" (i)				\
    172	: cl);								\
    173									\
    174	return result;							\
    175}
    176
    177#define ATOMIC64_FETCH_OP(name, mb, acq, rel, cl, op, asm_op, constraint)\
    178static inline long							\
    179__ll_sc_atomic64_fetch_##op##name(s64 i, atomic64_t *v)			\
    180{									\
    181	s64 result, val;						\
    182	unsigned long tmp;						\
    183									\
    184	asm volatile("// atomic64_fetch_" #op #name "\n"		\
    185	__LL_SC_FALLBACK(						\
    186	"	prfm	pstl1strm, %3\n"				\
    187	"1:	ld" #acq "xr	%0, %3\n"				\
    188	"	" #asm_op "	%1, %0, %4\n"				\
    189	"	st" #rel "xr	%w2, %1, %3\n"				\
    190	"	cbnz	%w2, 1b\n"					\
    191	"	" #mb )							\
    192	: "=&r" (result), "=&r" (val), "=&r" (tmp), "+Q" (v->counter)	\
    193	: __stringify(constraint) "r" (i)				\
    194	: cl);								\
    195									\
    196	return result;							\
    197}
    198
    199#define ATOMIC64_OPS(...)						\
    200	ATOMIC64_OP(__VA_ARGS__)					\
    201	ATOMIC64_OP_RETURN(, dmb ish,  , l, "memory", __VA_ARGS__)	\
    202	ATOMIC64_OP_RETURN(_relaxed,,  ,  ,         , __VA_ARGS__)	\
    203	ATOMIC64_OP_RETURN(_acquire,, a,  , "memory", __VA_ARGS__)	\
    204	ATOMIC64_OP_RETURN(_release,,  , l, "memory", __VA_ARGS__)	\
    205	ATOMIC64_FETCH_OP (, dmb ish,  , l, "memory", __VA_ARGS__)	\
    206	ATOMIC64_FETCH_OP (_relaxed,,  ,  ,         , __VA_ARGS__)	\
    207	ATOMIC64_FETCH_OP (_acquire,, a,  , "memory", __VA_ARGS__)	\
    208	ATOMIC64_FETCH_OP (_release,,  , l, "memory", __VA_ARGS__)
    209
    210ATOMIC64_OPS(add, add, I)
    211ATOMIC64_OPS(sub, sub, J)
    212
    213#undef ATOMIC64_OPS
    214#define ATOMIC64_OPS(...)						\
    215	ATOMIC64_OP(__VA_ARGS__)					\
    216	ATOMIC64_FETCH_OP (, dmb ish,  , l, "memory", __VA_ARGS__)	\
    217	ATOMIC64_FETCH_OP (_relaxed,,  ,  ,         , __VA_ARGS__)	\
    218	ATOMIC64_FETCH_OP (_acquire,, a,  , "memory", __VA_ARGS__)	\
    219	ATOMIC64_FETCH_OP (_release,,  , l, "memory", __VA_ARGS__)
    220
    221ATOMIC64_OPS(and, and, L)
    222ATOMIC64_OPS(or, orr, L)
    223ATOMIC64_OPS(xor, eor, L)
    224/*
    225 * GAS converts the mysterious and undocumented BIC (immediate) alias to
    226 * an AND (immediate) instruction with the immediate inverted. We don't
    227 * have a constraint for this, so fall back to register.
    228 */
    229ATOMIC64_OPS(andnot, bic, )
    230
    231#undef ATOMIC64_OPS
    232#undef ATOMIC64_FETCH_OP
    233#undef ATOMIC64_OP_RETURN
    234#undef ATOMIC64_OP
    235
    236static inline s64
    237__ll_sc_atomic64_dec_if_positive(atomic64_t *v)
    238{
    239	s64 result;
    240	unsigned long tmp;
    241
    242	asm volatile("// atomic64_dec_if_positive\n"
    243	__LL_SC_FALLBACK(
    244	"	prfm	pstl1strm, %2\n"
    245	"1:	ldxr	%0, %2\n"
    246	"	subs	%0, %0, #1\n"
    247	"	b.lt	2f\n"
    248	"	stlxr	%w1, %0, %2\n"
    249	"	cbnz	%w1, 1b\n"
    250	"	dmb	ish\n"
    251	"2:")
    252	: "=&r" (result), "=&r" (tmp), "+Q" (v->counter)
    253	:
    254	: "cc", "memory");
    255
    256	return result;
    257}
    258
    259#define __CMPXCHG_CASE(w, sfx, name, sz, mb, acq, rel, cl, constraint)	\
    260static inline u##sz							\
    261__ll_sc__cmpxchg_case_##name##sz(volatile void *ptr,			\
    262					 unsigned long old,		\
    263					 u##sz new)			\
    264{									\
    265	unsigned long tmp;						\
    266	u##sz oldval;							\
    267									\
    268	/*								\
    269	 * Sub-word sizes require explicit casting so that the compare  \
    270	 * part of the cmpxchg doesn't end up interpreting non-zero	\
    271	 * upper bits of the register containing "old".			\
    272	 */								\
    273	if (sz < 32)							\
    274		old = (u##sz)old;					\
    275									\
    276	asm volatile(							\
    277	__LL_SC_FALLBACK(						\
    278	"	prfm	pstl1strm, %[v]\n"				\
    279	"1:	ld" #acq "xr" #sfx "\t%" #w "[oldval], %[v]\n"		\
    280	"	eor	%" #w "[tmp], %" #w "[oldval], %" #w "[old]\n"	\
    281	"	cbnz	%" #w "[tmp], 2f\n"				\
    282	"	st" #rel "xr" #sfx "\t%w[tmp], %" #w "[new], %[v]\n"	\
    283	"	cbnz	%w[tmp], 1b\n"					\
    284	"	" #mb "\n"						\
    285	"2:")								\
    286	: [tmp] "=&r" (tmp), [oldval] "=&r" (oldval),			\
    287	  [v] "+Q" (*(u##sz *)ptr)					\
    288	: [old] __stringify(constraint) "r" (old), [new] "r" (new)	\
    289	: cl);								\
    290									\
    291	return oldval;							\
    292}
    293
    294/*
    295 * Earlier versions of GCC (no later than 8.1.0) appear to incorrectly
    296 * handle the 'K' constraint for the value 4294967295 - thus we use no
    297 * constraint for 32 bit operations.
    298 */
    299__CMPXCHG_CASE(w, b,     ,  8,        ,  ,  ,         , K)
    300__CMPXCHG_CASE(w, h,     , 16,        ,  ,  ,         , K)
    301__CMPXCHG_CASE(w,  ,     , 32,        ,  ,  ,         , K)
    302__CMPXCHG_CASE( ,  ,     , 64,        ,  ,  ,         , L)
    303__CMPXCHG_CASE(w, b, acq_,  8,        , a,  , "memory", K)
    304__CMPXCHG_CASE(w, h, acq_, 16,        , a,  , "memory", K)
    305__CMPXCHG_CASE(w,  , acq_, 32,        , a,  , "memory", K)
    306__CMPXCHG_CASE( ,  , acq_, 64,        , a,  , "memory", L)
    307__CMPXCHG_CASE(w, b, rel_,  8,        ,  , l, "memory", K)
    308__CMPXCHG_CASE(w, h, rel_, 16,        ,  , l, "memory", K)
    309__CMPXCHG_CASE(w,  , rel_, 32,        ,  , l, "memory", K)
    310__CMPXCHG_CASE( ,  , rel_, 64,        ,  , l, "memory", L)
    311__CMPXCHG_CASE(w, b,  mb_,  8, dmb ish,  , l, "memory", K)
    312__CMPXCHG_CASE(w, h,  mb_, 16, dmb ish,  , l, "memory", K)
    313__CMPXCHG_CASE(w,  ,  mb_, 32, dmb ish,  , l, "memory", K)
    314__CMPXCHG_CASE( ,  ,  mb_, 64, dmb ish,  , l, "memory", L)
    315
    316#undef __CMPXCHG_CASE
    317
    318#define __CMPXCHG_DBL(name, mb, rel, cl)				\
    319static inline long							\
    320__ll_sc__cmpxchg_double##name(unsigned long old1,			\
    321				      unsigned long old2,		\
    322				      unsigned long new1,		\
    323				      unsigned long new2,		\
    324				      volatile void *ptr)		\
    325{									\
    326	unsigned long tmp, ret;						\
    327									\
    328	asm volatile("// __cmpxchg_double" #name "\n"			\
    329	__LL_SC_FALLBACK(						\
    330	"	prfm	pstl1strm, %2\n"				\
    331	"1:	ldxp	%0, %1, %2\n"					\
    332	"	eor	%0, %0, %3\n"					\
    333	"	eor	%1, %1, %4\n"					\
    334	"	orr	%1, %0, %1\n"					\
    335	"	cbnz	%1, 2f\n"					\
    336	"	st" #rel "xp	%w0, %5, %6, %2\n"			\
    337	"	cbnz	%w0, 1b\n"					\
    338	"	" #mb "\n"						\
    339	"2:")								\
    340	: "=&r" (tmp), "=&r" (ret), "+Q" (*(unsigned long *)ptr)	\
    341	: "r" (old1), "r" (old2), "r" (new1), "r" (new2)		\
    342	: cl);								\
    343									\
    344	return ret;							\
    345}
    346
    347__CMPXCHG_DBL(   ,        ,  ,         )
    348__CMPXCHG_DBL(_mb, dmb ish, l, "memory")
    349
    350#undef __CMPXCHG_DBL
    351#undef K
    352
    353#endif	/* __ASM_ATOMIC_LL_SC_H */