1 /* 2 * Copyright 2009-2016 Samy Al Bahra. 3 * Copyright 2013-2016 Olivier Houchard. 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 */ 27 28 #ifndef CK_PR_AARCH64_LLSC_H 29 #define CK_PR_AARCH64_LLSC_H 30 31 #ifndef CK_PR_H 32 #error Do not include this file directly, use ck_pr.h 33 #endif 34 35 CK_CC_INLINE static bool 36 ck_pr_cas_64_2_value(uint64_t target[2], uint64_t compare[2], uint64_t set[2], uint64_t value[2]) 37 { 38 uint64_t tmp1, tmp2; 39 40 __asm__ __volatile__("1:" 41 "ldxp %0, %1, [%4]\n" 42 "mov %2, %0\n" 43 "mov %3, %1\n" 44 "eor %0, %0, %5\n" 45 "eor %1, %1, %6\n" 46 "orr %1, %0, %1\n" 47 "mov %w0, #0\n" 48 "cbnz %1, 2f\n" 49 "stxp %w0, %7, %8, [%4]\n" 50 "cbnz %w0, 1b\n" 51 "mov %w0, #1\n" 52 "2:" 53 : "=&r" (tmp1), "=&r" (tmp2), "=&r" (value[0]), "=&r" (value[1]) 54 : "r" (target), "r" (compare[0]), "r" (compare[1]), "r" (set[0]), "r" (set[1]) 55 : "cc", "memory"); 56 57 return (tmp1); 58 } 59 60 CK_CC_INLINE static bool 61 ck_pr_cas_ptr_2_value(void *target, void *compare, void *set, void *value) 62 { 63 return (ck_pr_cas_64_2_value(CK_CPP_CAST(uint64_t *, target), 64 CK_CPP_CAST(uint64_t *, compare), 65 CK_CPP_CAST(uint64_t *, set), 66 CK_CPP_CAST(uint64_t *, value))); 67 } 68 69 CK_CC_INLINE static bool 70 ck_pr_cas_64_2(uint64_t target[2], uint64_t compare[2], uint64_t set[2]) 71 { 72 uint64_t tmp1, tmp2; 73 74 __asm__ __volatile__("1:" 75 "ldxp %0, %1, [%2]\n" 76 "eor %0, %0, %3\n" 77 "eor %1, %1, %4\n" 78 "orr %1, %0, %1\n" 79 "mov %w0, #0\n" 80 "cbnz %1, 2f\n" 81 "stxp %w0, %5, %6, [%2]\n" 82 "cbnz %w0, 1b\n" 83 "mov %w0, #1\n" 84 "2:" 85 : "=&r" (tmp1), "=&r" (tmp2) 86 : "r" (target), "r" (compare[0]), "r" (compare[1]), "r" (set[0]), "r" (set[1]) 87 : "cc", "memory"); 88 89 return (tmp1); 90 } 91 CK_CC_INLINE static bool 92 ck_pr_cas_ptr_2(void *target, void *compare, void *set) 93 { 94 return (ck_pr_cas_64_2(CK_CPP_CAST(uint64_t *, target), 95 CK_CPP_CAST(uint64_t *, compare), 96 CK_CPP_CAST(uint64_t *, set))); 97 } 98 99 100 #define CK_PR_CAS(N, M, T, W, R) \ 101 CK_CC_INLINE static bool \ 102 ck_pr_cas_##N##_value(M *target, T compare, T set, M *value) \ 103 { \ 104 T previous; \ 105 T tmp; \ 106 __asm__ __volatile__("1:\n" \ 107 "ldxr" W " %" R "0, [%2]\n" \ 108 "cmp %" R "0, %" R "4\n" \ 109 "b.ne 2f\n" \ 110 "stxr" W " %w1, %" R "3, [%2]\n" \ 111 "cbnz %w1, 1b\n" \ 112 "2:" \ 113 : "=&r" (previous), \ 114 "=&r" (tmp) \ 115 : "r" (target), \ 116 "r" (set), \ 117 "r" (compare) \ 118 : "memory", "cc"); \ 119 *(T *)value = previous; \ 120 return (previous == compare); \ 121 } \ 122 CK_CC_INLINE static bool \ 123 ck_pr_cas_##N(M *target, T compare, T set) \ 124 { \ 125 T previous; \ 126 T tmp; \ 127 __asm__ __volatile__( \ 128 "1:" \ 129 "ldxr" W " %" R "0, [%2]\n" \ 130 "cmp %" R "0, %" R "4\n" \ 131 "b.ne 2f\n" \ 132 "stxr" W " %w1, %" R "3, [%2]\n" \ 133 "cbnz %w1, 1b\n" \ 134 "2:" \ 135 : "=&r" (previous), \ 136 "=&r" (tmp) \ 137 : "r" (target), \ 138 "r" (set), \ 139 "r" (compare) \ 140 : "memory", "cc"); \ 141 return (previous == compare); \ 142 } 143 144 CK_PR_CAS(ptr, void, void *, "", "") 145 146 #define CK_PR_CAS_S(N, M, W, R) CK_PR_CAS(N, M, M, W, R) 147 CK_PR_CAS_S(64, uint64_t, "", "") 148 #ifndef CK_PR_DISABLE_DOUBLE 149 CK_PR_CAS_S(double, double, "", "") 150 #endif 151 CK_PR_CAS_S(32, uint32_t, "", "w") 152 CK_PR_CAS_S(uint, unsigned int, "", "w") 153 CK_PR_CAS_S(int, int, "", "w") 154 CK_PR_CAS_S(16, uint16_t, "h", "w") 155 CK_PR_CAS_S(8, uint8_t, "b", "w") 156 CK_PR_CAS_S(short, short, "h", "w") 157 CK_PR_CAS_S(char, char, "b", "w") 158 159 160 #undef CK_PR_CAS_S 161 #undef CK_PR_CAS 162 163 #define CK_PR_FAS(N, M, T, W, R) \ 164 CK_CC_INLINE static T \ 165 ck_pr_fas_##N(M *target, T v) \ 166 { \ 167 T previous; \ 168 T tmp; \ 169 __asm__ __volatile__("1:" \ 170 "ldxr" W " %" R "0, [%2]\n"\ 171 "stxr" W " %w1, %" R "3, [%2]\n"\ 172 "cbnz %w1, 1b\n" \ 173 : "=&r" (previous), \ 174 "=&r" (tmp) \ 175 : "r" (target), \ 176 "r" (v) \ 177 : "memory", "cc"); \ 178 return (previous); \ 179 } 180 181 CK_PR_FAS(64, uint64_t, uint64_t, "", "") 182 CK_PR_FAS(32, uint32_t, uint32_t, "", "w") 183 CK_PR_FAS(ptr, void, void *, "", "") 184 CK_PR_FAS(int, int, int, "", "w") 185 CK_PR_FAS(uint, unsigned int, unsigned int, "", "w") 186 CK_PR_FAS(16, uint16_t, uint16_t, "h", "w") 187 CK_PR_FAS(8, uint8_t, uint8_t, "b", "w") 188 CK_PR_FAS(short, short, short, "h", "w") 189 CK_PR_FAS(char, char, char, "b", "w") 190 191 192 #undef CK_PR_FAS 193 194 #define CK_PR_UNARY(O, N, M, T, I, W, R) \ 195 CK_CC_INLINE static void \ 196 ck_pr_##O##_##N(M *target) \ 197 { \ 198 T previous = 0; \ 199 T tmp = 0; \ 200 __asm__ __volatile__("1:" \ 201 "ldxr" W " %" R "0, [%2]\n"\ 202 I "\n" \ 203 "stxr" W " %w1, %" R "0, [%2]\n" \ 204 "cbnz %w1, 1b\n" \ 205 : "=&r" (previous), \ 206 "=&r" (tmp) \ 207 : "r" (target) \ 208 : "memory", "cc"); \ 209 return; \ 210 } 211 212 CK_PR_UNARY(inc, ptr, void, void *, "add %0, %0, #1", "", "") 213 CK_PR_UNARY(dec, ptr, void, void *, "sub %0, %0, #1", "", "") 214 CK_PR_UNARY(not, ptr, void, void *, "mvn %0, %0", "", "") 215 CK_PR_UNARY(inc, 64, uint64_t, uint64_t, "add %0, %0, #1", "", "") 216 CK_PR_UNARY(dec, 64, uint64_t, uint64_t, "sub %0, %0, #1", "", "") 217 CK_PR_UNARY(not, 64, uint64_t, uint64_t, "mvn %0, %0", "", "") 218 219 #define CK_PR_UNARY_S(S, T, W) \ 220 CK_PR_UNARY(inc, S, T, T, "add %w0, %w0, #1", W, "w") \ 221 CK_PR_UNARY(dec, S, T, T, "sub %w0, %w0, #1", W, "w") \ 222 CK_PR_UNARY(not, S, T, T, "mvn %w0, %w0", W, "w") \ 223 224 CK_PR_UNARY_S(32, uint32_t, "") 225 CK_PR_UNARY_S(uint, unsigned int, "") 226 CK_PR_UNARY_S(int, int, "") 227 CK_PR_UNARY_S(16, uint16_t, "h") 228 CK_PR_UNARY_S(8, uint8_t, "b") 229 CK_PR_UNARY_S(short, short, "h") 230 CK_PR_UNARY_S(char, char, "b") 231 232 #undef CK_PR_UNARY_S 233 #undef CK_PR_UNARY 234 235 #define CK_PR_BINARY(O, N, M, T, I, W, R) \ 236 CK_CC_INLINE static void \ 237 ck_pr_##O##_##N(M *target, T delta) \ 238 { \ 239 T previous; \ 240 T tmp; \ 241 __asm__ __volatile__("1:" \ 242 "ldxr" W " %" R "0, [%2]\n"\ 243 I " %" R "0, %" R "0, %" R "3\n" \ 244 "stxr" W " %w1, %" R "0, [%2]\n" \ 245 "cbnz %w1, 1b\n" \ 246 : "=&r" (previous), \ 247 "=&r" (tmp) \ 248 : "r" (target), \ 249 "r" (delta) \ 250 : "memory", "cc"); \ 251 return; \ 252 } 253 254 CK_PR_BINARY(and, ptr, void, uintptr_t, "and", "", "") 255 CK_PR_BINARY(add, ptr, void, uintptr_t, "add", "", "") 256 CK_PR_BINARY(or, ptr, void, uintptr_t, "orr", "", "") 257 CK_PR_BINARY(sub, ptr, void, uintptr_t, "sub", "", "") 258 CK_PR_BINARY(xor, ptr, void, uintptr_t, "eor", "", "") 259 CK_PR_BINARY(and, 64, uint64_t, uint64_t, "and", "", "") 260 CK_PR_BINARY(add, 64, uint64_t, uint64_t, "add", "", "") 261 CK_PR_BINARY(or, 64, uint64_t, uint64_t, "orr", "", "") 262 CK_PR_BINARY(sub, 64, uint64_t, uint64_t, "sub", "", "") 263 CK_PR_BINARY(xor, 64, uint64_t, uint64_t, "eor", "", "") 264 265 #define CK_PR_BINARY_S(S, T, W) \ 266 CK_PR_BINARY(and, S, T, T, "and", W, "w") \ 267 CK_PR_BINARY(add, S, T, T, "add", W, "w") \ 268 CK_PR_BINARY(or, S, T, T, "orr", W, "w") \ 269 CK_PR_BINARY(sub, S, T, T, "sub", W, "w") \ 270 CK_PR_BINARY(xor, S, T, T, "eor", W, "w") 271 272 CK_PR_BINARY_S(32, uint32_t, "") 273 CK_PR_BINARY_S(uint, unsigned int, "") 274 CK_PR_BINARY_S(int, int, "") 275 CK_PR_BINARY_S(16, uint16_t, "h") 276 CK_PR_BINARY_S(8, uint8_t, "b") 277 CK_PR_BINARY_S(short, short, "h") 278 CK_PR_BINARY_S(char, char, "b") 279 280 #undef CK_PR_BINARY_S 281 #undef CK_PR_BINARY 282 283 CK_CC_INLINE static void * 284 ck_pr_faa_ptr(void *target, uintptr_t delta) 285 { 286 uintptr_t previous, r, tmp; 287 288 __asm__ __volatile__("1:" 289 "ldxr %0, [%3]\n" 290 "add %1, %4, %0\n" 291 "stxr %w2, %1, [%3]\n" 292 "cbnz %w2, 1b\n" 293 : "=&r" (previous), 294 "=&r" (r), 295 "=&r" (tmp) 296 : "r" (target), 297 "r" (delta) 298 : "memory", "cc"); 299 300 return (void *)(previous); 301 } 302 303 CK_CC_INLINE static uint64_t 304 ck_pr_faa_64(uint64_t *target, uint64_t delta) 305 { 306 uint64_t previous, r, tmp; 307 308 __asm__ __volatile__("1:" 309 "ldxr %0, [%3]\n" 310 "add %1, %4, %0\n" 311 "stxr %w2, %1, [%3]\n" 312 "cbnz %w2, 1b;" 313 : "=&r" (previous), 314 "=&r" (r), 315 "=&r" (tmp) 316 : "r" (target), 317 "r" (delta) 318 : "memory", "cc"); 319 320 return (previous); 321 } 322 323 #define CK_PR_FAA(S, T, W) \ 324 CK_CC_INLINE static T \ 325 ck_pr_faa_##S(T *target, T delta) \ 326 { \ 327 T previous, r, tmp; \ 328 __asm__ __volatile__("1:" \ 329 "ldxr" W " %w0, [%3]\n" \ 330 "add %w1, %w4, %w0\n" \ 331 "stxr" W " %w2, %w1, [%3]\n" \ 332 "cbnz %w2, 1b\n" \ 333 : "=&r" (previous), \ 334 "=&r" (r), \ 335 "=&r" (tmp) \ 336 : "r" (target), \ 337 "r" (delta) \ 338 : "memory", "cc"); \ 339 return (previous); \ 340 } 341 342 CK_PR_FAA(32, uint32_t, "") 343 CK_PR_FAA(uint, unsigned int, "") 344 CK_PR_FAA(int, int, "") 345 CK_PR_FAA(16, uint16_t, "h") 346 CK_PR_FAA(8, uint8_t, "b") 347 CK_PR_FAA(short, short, "h") 348 CK_PR_FAA(char, char, "b") 349 350 #undef CK_PR_FAA 351 352 #endif /* CK_PR_AARCH64_LLSC_H */ 353