1 /*- 2 * Copyright (c) 1998 Doug Rabson 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 * 26 * $FreeBSD: src/sys/i386/include/atomic.h,v 1.9.2.1 2000/07/07 00:38:47 obrien Exp $ 27 */ 28 #ifndef _CPU_ATOMIC_H_ 29 #define _CPU_ATOMIC_H_ 30 31 #ifndef _SYS_TYPES_H_ 32 #include <sys/types.h> 33 #endif 34 35 /* 36 * Various simple arithmetic on memory which is atomic in the presence 37 * of interrupts and multiple processors. 38 * 39 * atomic_set_char(P, V) (*(u_char*)(P) |= (V)) 40 * atomic_clear_char(P, V) (*(u_char*)(P) &= ~(V)) 41 * atomic_add_char(P, V) (*(u_char*)(P) += (V)) 42 * atomic_subtract_char(P, V) (*(u_char*)(P) -= (V)) 43 * 44 * atomic_set_short(P, V) (*(u_short*)(P) |= (V)) 45 * atomic_clear_short(P, V) (*(u_short*)(P) &= ~(V)) 46 * atomic_add_short(P, V) (*(u_short*)(P) += (V)) 47 * atomic_subtract_short(P, V) (*(u_short*)(P) -= (V)) 48 * 49 * atomic_set_int(P, V) (*(u_int*)(P) |= (V)) 50 * atomic_clear_int(P, V) (*(u_int*)(P) &= ~(V)) 51 * atomic_add_int(P, V) (*(u_int*)(P) += (V)) 52 * atomic_subtract_int(P, V) (*(u_int*)(P) -= (V)) 53 * 54 * atomic_set_long(P, V) (*(u_long*)(P) |= (V)) 55 * atomic_clear_long(P, V) (*(u_long*)(P) &= ~(V)) 56 * atomic_add_long(P, V) (*(u_long*)(P) += (V)) 57 * atomic_subtract_long(P, V) (*(u_long*)(P) -= (V)) 58 * atomic_readandclear_long(P) (return (*(u_long*)(P)); *(u_long*)(P) = 0;) 59 * atomic_readandclear_int(P) (return (*(u_int*)(P)); *(u_int*)(P) = 0;) 60 */ 61 62 /* 63 * The above functions are expanded inline in the statically-linked 64 * kernel and lock prefixes are generated. 65 * 66 * Kernel modules call real functions which are built into the kernel. 67 */ 68 #if defined(KLD_MODULE) 69 #define ATOMIC_ASM(NAME, TYPE, OP, CONS, V) \ 70 extern void atomic_##NAME##_##TYPE \ 71 (volatile u_##TYPE *p, u_##TYPE v); \ 72 extern void atomic_##NAME##_##TYPE##_nonlocked \ 73 (volatile u_##TYPE *p, u_##TYPE v); \ 74 extern void atomic_##NAME##_##TYPE##_xacquire \ 75 (volatile u_##TYPE *p, u_##TYPE v); \ 76 extern void atomic_##NAME##_##TYPE##_xrelease \ 77 (volatile u_##TYPE *p, u_##TYPE v); \ 78 79 int atomic_testandset_int(volatile u_int *p, u_int v); 80 int atomic_testandset_long(volatile u_long *p, u_long v); 81 int atomic_testandclear_int(volatile u_int *p, u_int v); 82 int atomic_testandclear_long(volatile u_long *p, u_long v); 83 84 #else /* !KLD_MODULE */ 85 86 /* 87 * locked bus cycle 88 * lock elision (backwards compatible) 89 */ 90 #define MPLOCKED "lock ; " 91 #define XACQUIRE "repne; " /* lock elision */ 92 #define XRELEASE "repe; " /* lock elision */ 93 94 /* 95 * The assembly is volatilized to demark potential before-and-after side 96 * effects if an interrupt or SMP collision were to occur. The primary 97 * atomic instructions are MP safe, the nonlocked instructions are 98 * local-interrupt-safe (so we don't depend on C 'X |= Y' generating an 99 * atomic instruction). 100 * 101 * +m - memory is read and written (=m - memory is only written) 102 * iq - integer constant or %ax/%bx/%cx/%dx (ir = int constant or any reg) 103 * (Note: byte instructions only work on %ax,%bx,%cx, or %dx). iq 104 * is good enough for our needs so don't get fancy. 105 * r - any register. 106 * 107 * NOTE: 64-bit immediate values are not supported for most x86-64 108 * instructions so we have to use "r". 109 */ 110 111 /* egcs 1.1.2+ version */ 112 #define ATOMIC_ASM(NAME, TYPE, OP, CONS, V) \ 113 static __inline void \ 114 atomic_##NAME##_##TYPE(volatile u_##TYPE *p, u_##TYPE v)\ 115 { \ 116 __asm __volatile(MPLOCKED OP \ 117 : "+m" (*p) \ 118 : CONS (V)); \ 119 } \ 120 static __inline void \ 121 atomic_##NAME##_##TYPE##_xacquire(volatile u_##TYPE *p, u_##TYPE v)\ 122 { \ 123 __asm __volatile(XACQUIRE MPLOCKED OP \ 124 : "+m" (*p) \ 125 : CONS (V)); \ 126 } \ 127 static __inline void \ 128 atomic_##NAME##_##TYPE##_xrelease(volatile u_##TYPE *p, u_##TYPE v)\ 129 { \ 130 __asm __volatile(XRELEASE MPLOCKED OP \ 131 : "+m" (*p) \ 132 : CONS (V)); \ 133 } \ 134 static __inline void \ 135 atomic_##NAME##_##TYPE##_nonlocked(volatile u_##TYPE *p, u_##TYPE v)\ 136 { \ 137 __asm __volatile(OP \ 138 : "+m" (*p) \ 139 : CONS (V)); \ 140 } 141 142 #endif /* KLD_MODULE */ 143 144 /* egcs 1.1.2+ version */ 145 ATOMIC_ASM(set, char, "orb %b1,%0", "iq", v) 146 ATOMIC_ASM(clear, char, "andb %b1,%0", "iq", ~v) 147 ATOMIC_ASM(add, char, "addb %b1,%0", "iq", v) 148 ATOMIC_ASM(subtract, char, "subb %b1,%0", "iq", v) 149 150 ATOMIC_ASM(set, short, "orw %w1,%0", "iq", v) 151 ATOMIC_ASM(clear, short, "andw %w1,%0", "iq", ~v) 152 ATOMIC_ASM(add, short, "addw %w1,%0", "iq", v) 153 ATOMIC_ASM(subtract, short, "subw %w1,%0", "iq", v) 154 155 ATOMIC_ASM(set, int, "orl %1,%0", "iq", v) 156 ATOMIC_ASM(clear, int, "andl %1,%0", "iq", ~v) 157 ATOMIC_ASM(add, int, "addl %1,%0", "iq", v) 158 ATOMIC_ASM(subtract, int, "subl %1,%0", "iq", v) 159 160 ATOMIC_ASM(set, long, "orq %1,%0", "r", v) 161 ATOMIC_ASM(clear, long, "andq %1,%0", "r", ~v) 162 ATOMIC_ASM(add, long, "addq %1,%0", "r", v) 163 ATOMIC_ASM(subtract, long, "subq %1,%0", "r", v) 164 165 #if defined(KLD_MODULE) 166 167 u_long atomic_readandclear_long(volatile u_long *addr); 168 u_int atomic_readandclear_int(volatile u_int *addr); 169 170 #else /* !KLD_MODULE */ 171 172 static __inline u_long 173 atomic_readandclear_long(volatile u_long *addr) 174 { 175 u_long res; 176 177 res = 0; 178 __asm __volatile( 179 " xchgq %1,%0 ; " 180 "# atomic_readandclear_long" 181 : "+r" (res), /* 0 */ 182 "=m" (*addr) /* 1 */ 183 : "m" (*addr)); 184 185 return (res); 186 } 187 188 static __inline u_int 189 atomic_readandclear_int(volatile u_int *addr) 190 { 191 u_int res; 192 193 res = 0; 194 __asm __volatile( 195 " xchgl %1,%0 ; " 196 "# atomic_readandclear_int" 197 : "+r" (res), /* 0 */ 198 "=m" (*addr) /* 1 */ 199 : "m" (*addr)); 200 201 return (res); 202 } 203 204 #endif /* KLD_MODULE */ 205 206 /* 207 * atomic_poll_acquire_int(P) Returns non-zero on success, 0 if the lock 208 * has already been acquired. 209 * atomic_poll_release_int(P) 210 * 211 * These support the NDIS driver and are also used for IPIQ interlocks 212 * between cpus. Both the acquisition and release must be 213 * cache-synchronizing instructions. 214 */ 215 216 #if defined(KLD_MODULE) 217 218 extern int atomic_swap_int(volatile int *addr, int value); 219 extern long atomic_swap_long(volatile long *addr, long value); 220 extern void *atomic_swap_ptr(volatile void **addr, void *value); 221 extern int atomic_poll_acquire_int(volatile u_int *p); 222 extern void atomic_poll_release_int(volatile u_int *p); 223 224 #else 225 226 static __inline int 227 atomic_swap_int(volatile int *addr, int value) 228 { 229 __asm __volatile("xchgl %0, %1" : 230 "=r" (value), "=m" (*addr) : "0" (value) : "memory"); 231 return (value); 232 } 233 234 static __inline long 235 atomic_swap_long(volatile long *addr, long value) 236 { 237 __asm __volatile("xchgq %0, %1" : 238 "=r" (value), "=m" (*addr) : "0" (value) : "memory"); 239 return (value); 240 } 241 242 static __inline void * 243 atomic_swap_ptr(volatile void **addr, void *value) 244 { 245 __asm __volatile("xchgq %0, %1" : 246 "=r" (value), "=m" (*addr) : "0" (value) : "memory"); 247 return (value); 248 } 249 250 static __inline int 251 atomic_poll_acquire_int(volatile u_int *p) 252 { 253 u_int data; 254 255 __asm __volatile(MPLOCKED "btsl $0,%0; setnc %%al; andl $255,%%eax" : "+m" (*p), "=a" (data)); 256 return(data); 257 } 258 259 static __inline void 260 atomic_poll_release_int(volatile u_int *p) 261 { 262 __asm __volatile(MPLOCKED "btrl $0,%0" : "+m" (*p)); 263 } 264 265 #endif 266 267 /* 268 * These functions operate on a 32 bit interrupt interlock which is defined 269 * as follows: 270 * 271 * bit 0-29 interrupt handler wait counter 272 * bit 30 interrupt handler disabled bit 273 * bit 31 interrupt handler currently running bit (1 = run) 274 * 275 * atomic_intr_cond_test(P) Determine if the interlock is in an 276 * acquired state. Returns 0 if it not 277 * acquired, non-zero if it is. (not MPLOCKed) 278 * 279 * atomic_intr_cond_try(P) Attempt to set bit 31 to acquire the 280 * interlock. If we are unable to set bit 31 281 * we return 1, otherwise we return 0. 282 * 283 * atomic_intr_cond_enter(P, func, arg) 284 * Attempt to set bit 31 to acquire the 285 * interlock. If we are unable to set bit 31, 286 * the wait is incremented counter and func(arg) 287 * is called in a loop until we are able to set 288 * bit 31. Once we set bit 31, wait counter 289 * is decremented. 290 * 291 * atomic_intr_cond_exit(P, func, arg) 292 * Clear bit 31. If the wait counter is still 293 * non-zero call func(arg) once. 294 * 295 * atomic_intr_handler_disable(P) 296 * Set bit 30, indicating that the interrupt 297 * handler has been disabled. Must be called 298 * after the hardware is disabled. 299 * 300 * Returns bit 31 indicating whether a serialized 301 * accessor is active (typically the interrupt 302 * handler is running). 0 == not active, 303 * non-zero == active. 304 * 305 * atomic_intr_handler_enable(P) 306 * Clear bit 30, indicating that the interrupt 307 * handler has been enabled. Must be called 308 * before the hardware is actually enabled. 309 * 310 * atomic_intr_handler_is_enabled(P) 311 * Returns bit 30, 0 indicates that the handler 312 * is enabled, non-zero indicates that it is 313 * disabled. The request counter portion of 314 * the field is ignored. (not MPLOCKed) 315 * 316 * atomic_intr_cond_inc(P) Increment wait counter by 1. 317 * atomic_intr_cond_dec(P) Decrement wait counter by 1. 318 */ 319 320 #if defined(KLD_MODULE) 321 322 void atomic_intr_init(__atomic_intr_t *p); 323 int atomic_intr_handler_disable(__atomic_intr_t *p); 324 void atomic_intr_handler_enable(__atomic_intr_t *p); 325 int atomic_intr_handler_is_enabled(__atomic_intr_t *p); 326 int atomic_intr_cond_test(__atomic_intr_t *p); 327 int atomic_intr_cond_try(__atomic_intr_t *p); 328 void atomic_intr_cond_enter(__atomic_intr_t *p, void (*func)(void *), void *arg); 329 void atomic_intr_cond_exit(__atomic_intr_t *p, void (*func)(void *), void *arg); 330 void atomic_intr_cond_inc(__atomic_intr_t *p); 331 void atomic_intr_cond_dec(__atomic_intr_t *p); 332 333 #else 334 335 static __inline void 336 atomic_intr_init(__atomic_intr_t *p) 337 { 338 *p = 0; 339 } 340 341 static __inline int 342 atomic_intr_handler_disable(__atomic_intr_t *p) 343 { 344 int data; 345 346 __asm __volatile(MPLOCKED "orl $0x40000000,%1; movl %1,%%eax; " \ 347 "andl $0x80000000,%%eax" \ 348 : "=a"(data) , "+m"(*p)); 349 return(data); 350 } 351 352 static __inline void 353 atomic_intr_handler_enable(__atomic_intr_t *p) 354 { 355 __asm __volatile(MPLOCKED "andl $0xBFFFFFFF,%0" : "+m" (*p)); 356 } 357 358 static __inline int 359 atomic_intr_handler_is_enabled(__atomic_intr_t *p) 360 { 361 int data; 362 363 __asm __volatile("movl %1,%%eax; andl $0x40000000,%%eax" \ 364 : "=a"(data) : "m"(*p)); 365 return(data); 366 } 367 368 static __inline void 369 atomic_intr_cond_inc(__atomic_intr_t *p) 370 { 371 __asm __volatile(MPLOCKED "incl %0" : "+m" (*p)); 372 } 373 374 static __inline void 375 atomic_intr_cond_dec(__atomic_intr_t *p) 376 { 377 __asm __volatile(MPLOCKED "decl %0" : "+m" (*p)); 378 } 379 380 static __inline void 381 atomic_intr_cond_enter(__atomic_intr_t *p, void (*func)(void *), void *arg) 382 { 383 __asm __volatile(MPLOCKED "btsl $31,%0; jnc 3f; " \ 384 MPLOCKED "incl %0; " \ 385 "1: ;" \ 386 MPLOCKED "btsl $31,%0; jnc 2f; " \ 387 "movq %2,%%rdi; call *%1; " \ 388 "jmp 1b; " \ 389 "2: ;" \ 390 MPLOCKED "decl %0; " \ 391 "3: ;" \ 392 : "+m" (*p) \ 393 : "r"(func), "m"(arg) \ 394 : "ax", "cx", "dx", "rsi", "rdi", "r8", "r9", "r10", "r11"); 395 /* YYY the function call may clobber even more registers? */ 396 } 397 398 /* 399 * Attempt to enter the interrupt condition variable. Returns zero on 400 * success, 1 on failure. 401 */ 402 static __inline int 403 atomic_intr_cond_try(__atomic_intr_t *p) 404 { 405 int ret; 406 407 __asm __volatile("subl %%eax,%%eax; " \ 408 MPLOCKED "btsl $31,%0; jnc 2f; " \ 409 "movl $1,%%eax;" \ 410 "2: ;" 411 : "+m" (*p), "=&a"(ret) 412 : : "cx", "dx"); 413 return (ret); 414 } 415 416 417 static __inline int 418 atomic_intr_cond_test(__atomic_intr_t *p) 419 { 420 return((int)(*p & 0x80000000)); 421 } 422 423 static __inline void 424 atomic_intr_cond_exit(__atomic_intr_t *p, void (*func)(void *), void *arg) 425 { 426 __asm __volatile(MPLOCKED "btrl $31,%0; " \ 427 "testl $0x3FFFFFFF,%0; jz 1f; " \ 428 "movq %2,%%rdi; call *%1; " \ 429 "1: ;" \ 430 : "+m" (*p) \ 431 : "r"(func), "m"(arg) \ 432 : "ax", "cx", "dx", "rsi", "rdi", "r8", "r9", "r10", "r11"); 433 /* YYY the function call may clobber even more registers? */ 434 } 435 436 #endif 437 438 /* 439 * Atomic compare and set 440 * 441 * if (*_dst == _old) *_dst = _new (all 32 bit words) 442 * 443 * Returns 0 on failure, non-zero on success. The inline is designed to 444 * allow the compiler to optimize the common case where the caller calls 445 * these functions from inside a conditional. 446 */ 447 #if defined(KLD_MODULE) 448 449 extern int atomic_cmpxchg_int(volatile u_int *_dst, u_int _old, u_int _new); 450 extern int atomic_cmpxchg_long_test(volatile u_long *_dst, u_long _old, u_long _new); 451 extern int atomic_cmpset_short(volatile u_short *_dst, 452 u_short _old, u_short _new); 453 extern int atomic_cmpset_int(volatile u_int *_dst, u_int _old, u_int _new); 454 extern int atomic_cmpset_int_xacquire(volatile u_int *_dst, 455 u_int _old, u_int _new); 456 extern int atomic_cmpset_int_xrelease(volatile u_int *_dst, 457 u_int _old, u_int _new); 458 extern int atomic_cmpset_long(volatile u_long *_dst, u_long _exp, u_long _src); 459 extern int atomic_cmpset_long_xacquire(volatile u_long *_dst, 460 u_long _exp, u_long _src); 461 extern int atomic_cmpset_long_xrelease(volatile u_long *_dst, 462 u_long _exp, u_long _src); 463 extern u_int atomic_fetchadd_int(volatile u_int *_p, u_int _v); 464 extern u_int atomic_fetchadd_int_xacquire(volatile u_int *_p, u_int _v); 465 extern u_int atomic_fetchadd_int_xrelease(volatile u_int *_p, u_int _v); 466 extern u_long atomic_fetchadd_long(volatile u_long *_p, u_long _v); 467 extern u_long atomic_fetchadd_long_xacquire(volatile u_long *_p, u_long _v); 468 extern u_long atomic_fetchadd_long_xrelease(volatile u_long *_p, u_long _v); 469 470 #else 471 472 static __inline int 473 atomic_cmpxchg_int(volatile u_int *_dst, u_int _old, u_int _new) 474 { 475 u_int res = _old; 476 477 __asm __volatile(MPLOCKED "cmpxchgl %2,%1; " \ 478 : "+a" (res), "=m" (*_dst) \ 479 : "r" (_new), "m" (*_dst) \ 480 : "memory"); 481 return (res); 482 } 483 484 static __inline int 485 atomic_cmpxchg_long_test(volatile u_long *_dst, u_long _old, u_long _new) 486 { 487 u_int res = _old; 488 489 __asm __volatile(MPLOCKED "cmpxchgq %2,%1; " 490 " setz %%al;" 491 " movsbq %%al,%%rax" \ 492 : "+a" (res), "=m" (*_dst) \ 493 : "r" (_new), "m" (*_dst) \ 494 : "memory"); 495 return (res); 496 } 497 498 static __inline int 499 atomic_cmpset_short(volatile u_short *_dst, u_short _old, u_short _new) 500 { 501 u_short res = _old; 502 503 __asm __volatile(MPLOCKED "cmpxchgw %w2,%1; " \ 504 : "+a" (res), "=m" (*_dst) \ 505 : "r" (_new), "m" (*_dst) \ 506 : "memory"); 507 return (res == _old); 508 } 509 510 static __inline int 511 atomic_cmpset_int(volatile u_int *_dst, u_int _old, u_int _new) 512 { 513 u_int res = _old; 514 515 __asm __volatile(MPLOCKED "cmpxchgl %2,%1; " \ 516 : "+a" (res), "=m" (*_dst) \ 517 : "r" (_new), "m" (*_dst) \ 518 : "memory"); 519 return (res == _old); 520 } 521 522 static __inline int 523 atomic_cmpset_int_xacquire(volatile u_int *_dst, u_int _old, u_int _new) 524 { 525 u_int res = _old; 526 527 __asm __volatile(XACQUIRE MPLOCKED "cmpxchgl %2,%1; " \ 528 : "+a" (res), "=m" (*_dst) \ 529 : "r" (_new), "m" (*_dst) \ 530 : "memory"); 531 return (res == _old); 532 } 533 534 static __inline int 535 atomic_cmpset_int_xrelease(volatile u_int *_dst, u_int _old, u_int _new) 536 { 537 u_int res = _old; 538 539 __asm __volatile(XRELEASE MPLOCKED "cmpxchgl %2,%1; " \ 540 : "+a" (res), "=m" (*_dst) \ 541 : "r" (_new), "m" (*_dst) \ 542 : "memory"); 543 return (res == _old); 544 } 545 546 static __inline int 547 atomic_cmpset_long(volatile u_long *_dst, u_long _old, u_long _new) 548 { 549 u_long res = _old; 550 551 __asm __volatile(MPLOCKED "cmpxchgq %2,%1; " \ 552 : "+a" (res), "=m" (*_dst) \ 553 : "r" (_new), "m" (*_dst) \ 554 : "memory"); 555 return (res == _old); 556 } 557 558 static __inline int 559 atomic_cmpset_long_xacquire(volatile u_long *_dst, u_long _old, u_long _new) 560 { 561 u_long res = _old; 562 563 __asm __volatile(XACQUIRE MPLOCKED "cmpxchgq %2,%1; " \ 564 : "+a" (res), "=m" (*_dst) \ 565 : "r" (_new), "m" (*_dst) \ 566 : "memory"); 567 return (res == _old); 568 } 569 570 static __inline int 571 atomic_cmpset_long_xrelease(volatile u_long *_dst, u_long _old, u_long _new) 572 { 573 u_long res = _old; 574 575 __asm __volatile(XRELEASE MPLOCKED "cmpxchgq %2,%1; " \ 576 : "+a" (res), "=m" (*_dst) \ 577 : "r" (_new), "m" (*_dst) \ 578 : "memory"); 579 return (res == _old); 580 } 581 582 /* 583 * Atomically add the value of v to the integer pointed to by p and return 584 * the previous value of *p. 585 */ 586 static __inline u_int 587 atomic_fetchadd_int(volatile u_int *_p, u_int _v) 588 { 589 __asm __volatile(MPLOCKED "xaddl %0,%1; " \ 590 : "+r" (_v), "=m" (*_p) \ 591 : "m" (*_p) \ 592 : "memory"); 593 return (_v); 594 } 595 596 static __inline u_int 597 atomic_fetchadd_int_xacquire(volatile u_int *_p, u_int _v) 598 { 599 __asm __volatile(XACQUIRE MPLOCKED "xaddl %0,%1; " \ 600 : "+r" (_v), "=m" (*_p) \ 601 : "m" (*_p) \ 602 : "memory"); 603 return (_v); 604 } 605 606 static __inline u_int 607 atomic_fetchadd_int_xrelease(volatile u_int *_p, u_int _v) 608 { 609 __asm __volatile(XRELEASE MPLOCKED "xaddl %0,%1; " \ 610 : "+r" (_v), "=m" (*_p) \ 611 : "m" (*_p) \ 612 : "memory"); 613 return (_v); 614 } 615 616 static __inline u_long 617 atomic_fetchadd_long(volatile u_long *_p, u_long _v) 618 { 619 __asm __volatile(MPLOCKED "xaddq %0,%1; " \ 620 : "+r" (_v), "=m" (*_p) \ 621 : "m" (*_p) \ 622 : "memory"); 623 return (_v); 624 } 625 626 static __inline u_long 627 atomic_fetchadd_long_xacquire(volatile u_long *_p, u_long _v) 628 { 629 __asm __volatile(XACQUIRE MPLOCKED "xaddq %0,%1; " \ 630 : "+r" (_v), "=m" (*_p) \ 631 : "m" (*_p) \ 632 : "memory"); 633 return (_v); 634 } 635 636 static __inline u_long 637 atomic_fetchadd_long_xrelease(volatile u_long *_p, u_long _v) 638 { 639 __asm __volatile(XRELEASE MPLOCKED "xaddq %0,%1; " \ 640 : "+r" (_v), "=m" (*_p) \ 641 : "m" (*_p) \ 642 : "memory"); 643 return (_v); 644 } 645 646 static __inline int 647 atomic_testandset_int(volatile u_int *p, u_int v) 648 { 649 u_char res; 650 651 __asm __volatile( 652 " " MPLOCKED " " 653 " btsl %2,%1 ; " 654 " setc %0 ; " 655 "# atomic_testandset_int" 656 : "=q" (res), /* 0 */ 657 "+m" (*p) /* 1 */ 658 : "Ir" (v & 0x1f) /* 2 */ 659 : "cc"); 660 return (res); 661 } 662 663 static __inline int 664 atomic_testandset_long(volatile u_long *p, u_long v) 665 { 666 u_char res; 667 668 __asm __volatile( 669 " " MPLOCKED " " 670 " btsq %2,%1 ; " 671 " setc %0 ; " 672 "# atomic_testandset_int" 673 : "=q" (res), /* 0 */ 674 "+m" (*p) /* 1 */ 675 : "Ir" (v & 0x3f) /* 2 */ 676 : "cc"); 677 return (res); 678 } 679 680 static __inline int 681 atomic_testandclear_int(volatile u_int *p, u_int v) 682 { 683 u_char res; 684 685 __asm __volatile( 686 " " MPLOCKED " " 687 " btrl %2,%1 ; " 688 " setc %0 ; " 689 "# atomic_testandclear_int" 690 : "=q" (res), /* 0 */ 691 "+m" (*p) /* 1 */ 692 : "Ir" (v & 0x1f) /* 2 */ 693 : "cc"); 694 return (res); 695 } 696 697 static __inline int 698 atomic_testandclear_long(volatile u_long *p, u_long v) 699 { 700 u_char res; 701 702 __asm __volatile( 703 " " MPLOCKED " " 704 " btrq %2,%1 ; " 705 " setc %0 ; " 706 "# atomic_testandclear_int" 707 : "=q" (res), /* 0 */ 708 "+m" (*p) /* 1 */ 709 : "Ir" (v & 0x3f) /* 2 */ 710 : "cc"); 711 return (res); 712 } 713 714 #endif /* KLD_MODULE */ 715 716 #if defined(KLD_MODULE) 717 718 #define ATOMIC_STORE_LOAD(TYPE, LOP, SOP) \ 719 extern u_##TYPE atomic_load_acq_##TYPE(volatile u_##TYPE *p); \ 720 extern void atomic_store_rel_##TYPE(volatile u_##TYPE *p, u_##TYPE v); 721 722 #else /* !KLD_MODULE */ 723 724 #define ATOMIC_STORE_LOAD(TYPE, LOP, SOP) \ 725 static __inline u_##TYPE \ 726 atomic_load_acq_##TYPE(volatile u_##TYPE *p) \ 727 { \ 728 u_##TYPE res; /* accumulator can be anything */ \ 729 \ 730 __asm __volatile(MPLOCKED LOP \ 731 : "=a" (res), /* 0 */ \ 732 "=m" (*p) /* 1 */ \ 733 : "m" (*p) /* 2 */ \ 734 : "memory"); \ 735 \ 736 return (res); \ 737 } \ 738 \ 739 /* \ 740 * The XCHG instruction asserts LOCK automagically. \ 741 */ \ 742 static __inline void \ 743 atomic_store_rel_##TYPE(volatile u_##TYPE *p, u_##TYPE v)\ 744 { \ 745 __asm __volatile(SOP \ 746 : "=m" (*p), /* 0 */ \ 747 "+r" (v) /* 1 */ \ 748 : "m" (*p)); /* 2 */ \ 749 } \ 750 struct __hack 751 752 #endif /* !KLD_MODULE */ 753 754 ATOMIC_STORE_LOAD(char, "cmpxchgb %b0,%1", "xchgb %b1,%0"); 755 ATOMIC_STORE_LOAD(short,"cmpxchgw %w0,%1", "xchgw %w1,%0"); 756 ATOMIC_STORE_LOAD(int, "cmpxchgl %0,%1", "xchgl %1,%0"); 757 ATOMIC_STORE_LOAD(long, "cmpxchgq %0,%1", "xchgq %1,%0"); 758 759 #undef ATOMIC_ASM 760 #undef ATOMIC_STORE_LOAD 761 762 /* Acquire and release variants are identical to the normal ones. */ 763 #define atomic_set_acq_char atomic_set_char 764 #define atomic_set_rel_char atomic_set_char 765 #define atomic_clear_acq_char atomic_clear_char 766 #define atomic_clear_rel_char atomic_clear_char 767 #define atomic_add_acq_char atomic_add_char 768 #define atomic_add_rel_char atomic_add_char 769 #define atomic_subtract_acq_char atomic_subtract_char 770 #define atomic_subtract_rel_char atomic_subtract_char 771 772 #define atomic_set_acq_short atomic_set_short 773 #define atomic_set_rel_short atomic_set_short 774 #define atomic_clear_acq_short atomic_clear_short 775 #define atomic_clear_rel_short atomic_clear_short 776 #define atomic_add_acq_short atomic_add_short 777 #define atomic_add_rel_short atomic_add_short 778 #define atomic_subtract_acq_short atomic_subtract_short 779 #define atomic_subtract_rel_short atomic_subtract_short 780 781 #define atomic_set_acq_int atomic_set_int 782 #define atomic_set_rel_int atomic_set_int 783 #define atomic_clear_acq_int atomic_clear_int 784 #define atomic_clear_rel_int atomic_clear_int 785 #define atomic_add_acq_int atomic_add_int 786 #define atomic_add_rel_int atomic_add_int 787 #define atomic_subtract_acq_int atomic_subtract_int 788 #define atomic_subtract_rel_int atomic_subtract_int 789 #define atomic_cmpset_acq_int atomic_cmpset_int 790 #define atomic_cmpset_rel_int atomic_cmpset_int 791 792 #define atomic_set_acq_long atomic_set_long 793 #define atomic_set_rel_long atomic_set_long 794 #define atomic_clear_acq_long atomic_clear_long 795 #define atomic_clear_rel_long atomic_clear_long 796 #define atomic_add_acq_long atomic_add_long 797 #define atomic_add_rel_long atomic_add_long 798 #define atomic_subtract_acq_long atomic_subtract_long 799 #define atomic_subtract_rel_long atomic_subtract_long 800 #define atomic_cmpset_acq_long atomic_cmpset_long 801 #define atomic_cmpset_rel_long atomic_cmpset_long 802 803 /* cpumask_t is 64-bits on x86-64 */ 804 #define atomic_set_cpumask atomic_set_long 805 #define atomic_clear_cpumask atomic_clear_long 806 #define atomic_cmpset_cpumask atomic_cmpset_long 807 #define atomic_store_rel_cpumask atomic_store_rel_long 808 #define atomic_load_acq_cpumask atomic_load_acq_long 809 810 /* Operations on 8-bit bytes. */ 811 #define atomic_set_8 atomic_set_char 812 #define atomic_set_acq_8 atomic_set_acq_char 813 #define atomic_set_rel_8 atomic_set_rel_char 814 #define atomic_clear_8 atomic_clear_char 815 #define atomic_clear_acq_8 atomic_clear_acq_char 816 #define atomic_clear_rel_8 atomic_clear_rel_char 817 #define atomic_add_8 atomic_add_char 818 #define atomic_add_acq_8 atomic_add_acq_char 819 #define atomic_add_rel_8 atomic_add_rel_char 820 #define atomic_subtract_8 atomic_subtract_char 821 #define atomic_subtract_acq_8 atomic_subtract_acq_char 822 #define atomic_subtract_rel_8 atomic_subtract_rel_char 823 #define atomic_load_acq_8 atomic_load_acq_char 824 #define atomic_store_rel_8 atomic_store_rel_char 825 826 /* Operations on 16-bit words. */ 827 #define atomic_set_16 atomic_set_short 828 #define atomic_set_acq_16 atomic_set_acq_short 829 #define atomic_set_rel_16 atomic_set_rel_short 830 #define atomic_clear_16 atomic_clear_short 831 #define atomic_clear_acq_16 atomic_clear_acq_short 832 #define atomic_clear_rel_16 atomic_clear_rel_short 833 #define atomic_add_16 atomic_add_short 834 #define atomic_add_acq_16 atomic_add_acq_short 835 #define atomic_add_rel_16 atomic_add_rel_short 836 #define atomic_subtract_16 atomic_subtract_short 837 #define atomic_subtract_acq_16 atomic_subtract_acq_short 838 #define atomic_subtract_rel_16 atomic_subtract_rel_short 839 #define atomic_load_acq_16 atomic_load_acq_short 840 #define atomic_store_rel_16 atomic_store_rel_short 841 842 /* Operations on 32-bit double words. */ 843 #define atomic_set_32 atomic_set_int 844 #define atomic_set_acq_32 atomic_set_acq_int 845 #define atomic_set_rel_32 atomic_set_rel_int 846 #define atomic_clear_32 atomic_clear_int 847 #define atomic_clear_acq_32 atomic_clear_acq_int 848 #define atomic_clear_rel_32 atomic_clear_rel_int 849 #define atomic_add_32 atomic_add_int 850 #define atomic_add_acq_32 atomic_add_acq_int 851 #define atomic_add_rel_32 atomic_add_rel_int 852 #define atomic_subtract_32 atomic_subtract_int 853 #define atomic_subtract_acq_32 atomic_subtract_acq_int 854 #define atomic_subtract_rel_32 atomic_subtract_rel_int 855 #define atomic_load_acq_32 atomic_load_acq_int 856 #define atomic_store_rel_32 atomic_store_rel_int 857 #define atomic_cmpset_32 atomic_cmpset_int 858 #define atomic_cmpset_acq_32 atomic_cmpset_acq_int 859 #define atomic_cmpset_rel_32 atomic_cmpset_rel_int 860 #define atomic_readandclear_32 atomic_readandclear_int 861 #define atomic_fetchadd_32 atomic_fetchadd_int 862 863 /* Operations on 64-bit quad words. */ 864 #define atomic_load_acq_64 atomic_load_acq_long 865 #define atomic_store_rel_64 atomic_store_rel_long 866 #define atomic_swap_64 atomic_swap_long 867 #define atomic_fetchadd_64 atomic_fetchadd_long 868 #define atomic_cmpset_64 atomic_cmpset_long 869 #define atomic_set_64 atomic_set_long 870 #define atomic_clear_64 atomic_clear_long 871 872 /* Operations on pointers. */ 873 #define atomic_set_ptr(p, v) \ 874 atomic_set_long((volatile u_long *)(p), (u_long)(v)) 875 #define atomic_set_acq_ptr(p, v) \ 876 atomic_set_acq_long((volatile u_long *)(p), (u_long)(v)) 877 #define atomic_set_rel_ptr(p, v) \ 878 atomic_set_rel_long((volatile u_long *)(p), (u_long)(v)) 879 #define atomic_clear_ptr(p, v) \ 880 atomic_clear_long((volatile u_long *)(p), (u_long)(v)) 881 #define atomic_clear_acq_ptr(p, v) \ 882 atomic_clear_acq_long((volatile u_long *)(p), (u_long)(v)) 883 #define atomic_clear_rel_ptr(p, v) \ 884 atomic_clear_rel_long((volatile u_long *)(p), (u_long)(v)) 885 #define atomic_add_ptr(p, v) \ 886 atomic_add_long((volatile u_long *)(p), (u_long)(v)) 887 #define atomic_add_acq_ptr(p, v) \ 888 atomic_add_acq_long((volatile u_long *)(p), (u_long)(v)) 889 #define atomic_add_rel_ptr(p, v) \ 890 atomic_add_rel_long((volatile u_long *)(p), (u_long)(v)) 891 #define atomic_subtract_ptr(p, v) \ 892 atomic_subtract_long((volatile u_long *)(p), (u_long)(v)) 893 #define atomic_subtract_acq_ptr(p, v) \ 894 atomic_subtract_acq_long((volatile u_long *)(p), (u_long)(v)) 895 #define atomic_subtract_rel_ptr(p, v) \ 896 atomic_subtract_rel_long((volatile u_long *)(p), (u_long)(v)) 897 #define atomic_load_acq_ptr(p) \ 898 atomic_load_acq_long((volatile u_long *)(p)) 899 #define atomic_store_rel_ptr(p, v) \ 900 atomic_store_rel_long((volatile u_long *)(p), (v)) 901 #define atomic_cmpset_ptr(dst, old, new) \ 902 atomic_cmpset_long((volatile u_long *)(dst), (u_long)(old), \ 903 (u_long)(new)) 904 #define atomic_cmpset_acq_ptr(dst, old, new) \ 905 atomic_cmpset_acq_long((volatile u_long *)(dst), (u_long)(old), \ 906 (u_long)(new)) 907 #define atomic_cmpset_rel_ptr(dst, old, new) \ 908 atomic_cmpset_rel_long((volatile u_long *)(dst), (u_long)(old), \ 909 (u_long)(new)) 910 #define atomic_readandclear_ptr(p) \ 911 atomic_readandclear_long((volatile u_long *)(p)) 912 913 #endif /* ! _CPU_ATOMIC_H_ */ 914