《操作系统》的实验代码。

57 righe
1.8 KiB

  1. #ifndef __LIBS_ATOMIC_H__
  2. #define __LIBS_ATOMIC_H__
  3. /* Atomic operations that C can't guarantee us. Useful for resource counting etc.. */
  4. static inline void set_bit(int nr, volatile void *addr) __attribute__((always_inline));
  5. static inline void clear_bit(int nr, volatile void *addr) __attribute__((always_inline));
  6. static inline void change_bit(int nr, volatile void *addr) __attribute__((always_inline));
  7. static inline bool test_bit(int nr, volatile void *addr) __attribute__((always_inline));
  8. /* *
  9. * set_bit - Atomically set a bit in memory
  10. * @nr: the bit to set
  11. * @addr: the address to start counting from
  12. *
  13. * Note that @nr may be almost arbitrarily large; this function is not
  14. * restricted to acting on a single-word quantity.
  15. * */
  16. static inline void
  17. set_bit(int nr, volatile void *addr) {
  18. asm volatile ("btsl %1, %0" :"=m" (*(volatile long *)addr) : "Ir" (nr));
  19. }
  20. /* *
  21. * clear_bit - Atomically clears a bit in memory
  22. * @nr: the bit to clear
  23. * @addr: the address to start counting from
  24. * */
  25. static inline void
  26. clear_bit(int nr, volatile void *addr) {
  27. asm volatile ("btrl %1, %0" :"=m" (*(volatile long *)addr) : "Ir" (nr));
  28. }
  29. /* *
  30. * change_bit - Atomically toggle a bit in memory
  31. * @nr: the bit to change
  32. * @addr: the address to start counting from
  33. * */
  34. static inline void
  35. change_bit(int nr, volatile void *addr) {
  36. asm volatile ("btcl %1, %0" :"=m" (*(volatile long *)addr) : "Ir" (nr));
  37. }
  38. /* *
  39. * test_bit - Determine whether a bit is set
  40. * @nr: the bit to test
  41. * @addr: the address to count from
  42. * */
  43. static inline bool
  44. test_bit(int nr, volatile void *addr) {
  45. int oldbit;
  46. asm volatile ("btl %2, %1; sbbl %0,%0" : "=r" (oldbit) : "m" (*(volatile long *)addr), "Ir" (nr));
  47. return oldbit != 0;
  48. }
  49. #endif /* !__LIBS_ATOMIC_H__ */