bitops.h
上传用户:jlfgdled
上传日期:2013-04-10
资源大小:33168k
文件大小:10k
源码类别:

Linux/Unix编程

开发平台:

Unix_Linux

  1. #ifndef _X86_64_BITOPS_H
  2. #define _X86_64_BITOPS_H
  3. /*
  4.  * Copyright 1992, Linus Torvalds.
  5.  */
  6. #include <linux/config.h>
  7. /*
  8.  * These have to be done with inline assembly: that way the bit-setting
  9.  * is guaranteed to be atomic. All bit operations return 0 if the bit
  10.  * was cleared before the operation and != 0 if it was not.
  11.  *
  12.  * bit 0 is the LSB of addr; bit 32 is the LSB of (addr+1).
  13.  */
  14. #ifdef CONFIG_SMP
  15. #define LOCK_PREFIX "lock ; "
  16. #else
  17. #define LOCK_PREFIX ""
  18. #endif
  19. #define ADDR (*(volatile long *) addr)
  20. /**
  21.  * set_bit - Atomically set a bit in memory
  22.  * @nr: the bit to set
  23.  * @addr: the address to start counting from
  24.  *
  25.  * This function is atomic and may not be reordered.  See __set_bit()
  26.  * if you do not require the atomic guarantees.
  27.  * Note that @nr may be almost arbitrarily large; this function is not
  28.  * restricted to acting on a single-word quantity.
  29.  */
  30. static __inline__ void set_bit(long nr, volatile void * addr)
  31. {
  32. __asm__ __volatile__( LOCK_PREFIX
  33. "btsq %1,%0"
  34. :"=m" (ADDR)
  35. :"dIr" (nr));
  36. }
  37. /**
  38.  * __set_bit - Set a bit in memory
  39.  * @nr: the bit to set
  40.  * @addr: the address to start counting from
  41.  *
  42.  * Unlike set_bit(), this function is non-atomic and may be reordered.
  43.  * If it's called on the same region of memory simultaneously, the effect
  44.  * may be that only one operation succeeds.
  45.  */
  46. static __inline__ void __set_bit(long nr, volatile void * addr)
  47. {
  48. __asm__(
  49. "btsq %1,%0"
  50. :"=m" (ADDR)
  51. :"dIr" (nr));
  52. }
  53. /**
  54.  * clear_bit - Clears a bit in memory
  55.  * @nr: Bit to clear
  56.  * @addr: Address to start counting from
  57.  *
  58.  * clear_bit() is atomic and may not be reordered.  However, it does
  59.  * not contain a memory barrier, so if it is used for locking purposes,
  60.  * you should call smp_mb__before_clear_bit() and/or smp_mb__after_clear_bit()
  61.  * in order to ensure changes are visible on other processors.
  62.  */
  63. static __inline__ void clear_bit(long nr, volatile void * addr)
  64. {
  65. __asm__ __volatile__( LOCK_PREFIX
  66. "btrq %1,%0"
  67. :"=m" (ADDR)
  68. :"dIr" (nr));
  69. }
  70. #define smp_mb__before_clear_bit() barrier()
  71. #define smp_mb__after_clear_bit() barrier()
  72. /**
  73.  * __change_bit - Toggle a bit in memory
  74.  * @nr: the bit to set
  75.  * @addr: the address to start counting from
  76.  *
  77.  * Unlike change_bit(), this function is non-atomic and may be reordered.
  78.  * If it's called on the same region of memory simultaneously, the effect
  79.  * may be that only one operation succeeds.
  80.  */
  81. static __inline__ void __change_bit(long nr, volatile void * addr)
  82. {
  83. __asm__ __volatile__(
  84. "btcq %1,%0"
  85. :"=m" (ADDR)
  86. :"dIr" (nr));
  87. }
  88. /**
  89.  * change_bit - Toggle a bit in memory
  90.  * @nr: Bit to clear
  91.  * @addr: Address to start counting from
  92.  *
  93.  * change_bit() is atomic and may not be reordered.
  94.  * Note that @nr may be almost arbitrarily large; this function is not
  95.  * restricted to acting on a single-word quantity.
  96.  */
  97. static __inline__ void change_bit(long nr, volatile void * addr)
  98. {
  99. __asm__ __volatile__( LOCK_PREFIX
  100. "btcq %1,%0"
  101. :"=m" (ADDR)
  102. :"dIr" (nr));
  103. }
  104. /**
  105.  * test_and_set_bit - Set a bit and return its old value
  106.  * @nr: Bit to set
  107.  * @addr: Address to count from
  108.  *
  109.  * This operation is atomic and cannot be reordered.  
  110.  * It also implies a memory barrier.
  111.  */
  112. static __inline__ int test_and_set_bit(long nr, volatile void * addr)
  113. {
  114.         long oldbit;
  115. __asm__ __volatile__( LOCK_PREFIX
  116. "btsq %2,%1ntsbbq %0,%0"
  117. :"=r" (oldbit),"=m" (ADDR)
  118. :"dIr" (nr) : "memory");
  119. return oldbit;
  120. }
  121. /**
  122.  * __test_and_set_bit - Set a bit and return its old value
  123.  * @nr: Bit to set
  124.  * @addr: Address to count from
  125.  *
  126.  * This operation is non-atomic and can be reordered.  
  127.  * If two examples of this operation race, one can appear to succeed
  128.  * but actually fail.  You must protect multiple accesses with a lock.
  129.  */
  130. static __inline__ int __test_and_set_bit(long nr, volatile void * addr)
  131. {
  132. long oldbit;
  133. __asm__(
  134. "btsq %2,%1ntsbbq %0,%0"
  135. :"=r" (oldbit),"=m" (ADDR)
  136. :"dIr" (nr));
  137. return oldbit;
  138. }
  139. /**
  140.  * test_and_clear_bit - Clear a bit and return its old value
  141.  * @nr: Bit to set
  142.  * @addr: Address to count from
  143.  *
  144.  * This operation is atomic and cannot be reordered.  
  145.  * It also implies a memory barrier.
  146.  */
  147. static __inline__ int test_and_clear_bit(long nr, volatile void * addr)
  148. {
  149. long oldbit;
  150. __asm__ __volatile__( LOCK_PREFIX
  151. "btrq %2,%1ntsbbq %0,%0"
  152. :"=r" (oldbit),"=m" (ADDR)
  153. :"dIr" (nr) : "memory");
  154. return oldbit;
  155. }
  156. /**
  157.  * __test_and_clear_bit - Clear a bit and return its old value
  158.  * @nr: Bit to set
  159.  * @addr: Address to count from
  160.  *
  161.  * This operation is non-atomic and can be reordered.  
  162.  * If two examples of this operation race, one can appear to succeed
  163.  * but actually fail.  You must protect multiple accesses with a lock.
  164.  */
  165. static __inline__ int __test_and_clear_bit(long nr, volatile void * addr)
  166. {
  167. long oldbit;
  168. __asm__(
  169. "btrq %2,%1ntsbbq %0,%0"
  170. :"=r" (oldbit),"=m" (ADDR)
  171. :"dIr" (nr));
  172. return oldbit;
  173. }
  174. /* WARNING: non atomic and it can be reordered! */
  175. static __inline__ int __test_and_change_bit(long nr, volatile void * addr)
  176. {
  177. long oldbit;
  178. __asm__ __volatile__(
  179. "btcq %2,%1ntsbbq %0,%0"
  180. :"=r" (oldbit),"=m" (ADDR)
  181. :"dIr" (nr) : "memory");
  182. return oldbit;
  183. }
  184. /**
  185.  * test_and_change_bit - Change a bit and return its new value
  186.  * @nr: Bit to set
  187.  * @addr: Address to count from
  188.  *
  189.  * This operation is atomic and cannot be reordered.  
  190.  * It also implies a memory barrier.
  191.  */
  192. static __inline__ int test_and_change_bit(long nr, volatile void * addr)
  193. {
  194. long oldbit;
  195. __asm__ __volatile__( LOCK_PREFIX
  196. "btcq %2,%1ntsbbq %0,%0"
  197. :"=r" (oldbit),"=m" (ADDR)
  198. :"dIr" (nr) : "memory");
  199. return oldbit;
  200. }
  201. #if 0 /* Fool kernel-doc since it doesn't do macros yet */
  202. /**
  203.  * test_bit - Determine whether a bit is set
  204.  * @nr: bit number to test
  205.  * @addr: Address to start counting from
  206.  */
  207. static int test_bit(int nr, const volatile void * addr);
  208. #endif
  209. static __inline__ int constant_test_bit(long nr, const volatile void * addr)
  210. {
  211. return ((1UL << (nr & 31)) & (((const volatile unsigned int *) addr)[nr >> 5])) != 0;
  212. }
  213. static __inline__ int variable_test_bit(long nr, volatile void * addr)
  214. {
  215. long oldbit;
  216. __asm__ __volatile__(
  217. "btq %2,%1ntsbbq %0,%0"
  218. :"=r" (oldbit)
  219. :"m" (ADDR),"dIr" (nr));
  220. return oldbit;
  221. }
  222. #define test_bit(nr,addr) 
  223. (__builtin_constant_p(nr) ? 
  224.  constant_test_bit((nr),(addr)) : 
  225.  variable_test_bit((nr),(addr)))
  226. /**
  227.  * find_first_zero_bit - find the first zero bit in a memory region
  228.  * @addr: The address to start the search at
  229.  * @size: The maximum bitnumber to search
  230.  *
  231.  * Returns the bit-number of the first zero bit, not the number of the byte
  232.  * containing a bit. -1 when none found.
  233.  */
  234. static __inline__ int find_first_zero_bit(void * addr, unsigned size)
  235. {
  236. int d0, d1, d2;
  237. int res;
  238. if (!size)
  239. return 0;
  240. __asm__ __volatile__(
  241. "movl $-1,%%eaxnt"
  242. "xorl %%edx,%%edxnt"
  243. "repe; scaslnt"
  244. "je 1fnt"
  245. "xorl -4(%%rdi),%%eaxnt"
  246. "subq $4,%%rdint"
  247. "bsfl %%eax,%%edxn"
  248. "1:tsubq %%rbx,%%rdint"
  249. "shlq $3,%%rdint"
  250. "addq %%rdi,%%rdx"
  251. :"=d" (res), "=&c" (d0), "=&D" (d1), "=&a" (d2)
  252. :"1" ((size + 31) >> 5), "2" (addr), "b" (addr) : "memory");
  253. return res;
  254. }
  255. /**
  256.  * find_next_zero_bit - find the first zero bit in a memory region
  257.  * @addr: The address to base the search on
  258.  * @offset: The bitnumber to start searching at
  259.  * @size: The maximum size to search
  260.  */
  261. static __inline__ int find_next_zero_bit (void * addr, int size, int offset)
  262. {
  263. unsigned int * p = ((unsigned int *) addr) + (offset >> 5);
  264. int set = 0, bit = offset & 31, res;
  265. if (bit) {
  266. /*
  267.  * Look for zero in first byte
  268.  */
  269. __asm__("bsfl %1,%0nt"
  270. "jne 1fnt"
  271. "movl $32, %0n"
  272. "1:"
  273. : "=r" (set)
  274. : "r" (~(*p >> bit)));
  275. if (set < (32 - bit))
  276. return set + offset;
  277. set = 32 - bit;
  278. p++;
  279. }
  280. /*
  281.  * No zero yet, search remaining full bytes for a zero
  282.  */
  283. res = find_first_zero_bit (p, size - 32 * (p - (unsigned int *) addr));
  284. return (offset + set + res);
  285. }
  286. /* 
  287.  * Find string of zero bits in a bitmap. -1 when not found.
  288.  */ 
  289. extern unsigned long 
  290. find_next_zero_string(unsigned long *bitmap, long start, long nbits, int len);
  291. static inline void set_bit_string(unsigned long *bitmap, unsigned long i, 
  292.   int len) 
  293. unsigned long end = i + len; 
  294. while (i < end) {
  295. __set_bit(i, bitmap); 
  296. i++;
  297. }
  298. static inline void clear_bit_string(unsigned long *bitmap, unsigned long i, 
  299.     int len) 
  300. unsigned long end = i + len; 
  301. while (i < end) {
  302. clear_bit(i, bitmap); 
  303. i++;
  304. }
  305. /**
  306.  * ffz - find first zero in word.
  307.  * @word: The word to search
  308.  *
  309.  * Undefined if no zero exists, so code should check against ~0UL first.
  310.  */
  311. static __inline__ unsigned long ffz(unsigned long word)
  312. {
  313. __asm__("bsfq %1,%0"
  314. :"=r" (word)
  315. :"r" (~word));
  316. return word;
  317. }
  318. #ifdef __KERNEL__
  319. /**
  320.  * ffs - find first bit set
  321.  * @x: the word to search
  322.  *
  323.  * This is defined the same way as
  324.  * the libc and compiler builtin ffs routines, therefore
  325.  * differs in spirit from the above ffz (man ffs).
  326.  */
  327. static __inline__ int ffs(int x)
  328. {
  329. int r;
  330. __asm__("bsfl %1,%0nt"
  331. "jnz 1fnt"
  332. "movl $-1,%0n"
  333. "1:" : "=r" (r) : "rm" (x));
  334. return r+1;
  335. }
  336. /**
  337.  * hweightN - returns the hamming weight of a N-bit word
  338.  * @x: the word to weigh
  339.  *
  340.  * The Hamming Weight of a number is the total number of bits set in it.
  341.  */
  342. #define hweight32(x) generic_hweight32(x)
  343. #define hweight16(x) generic_hweight16(x)
  344. #define hweight8(x) generic_hweight8(x)
  345. #endif /* __KERNEL__ */
  346. #ifdef __KERNEL__
  347. #define ext2_set_bit                 __test_and_set_bit
  348. #define ext2_clear_bit               __test_and_clear_bit
  349. #define ext2_test_bit                test_bit
  350. #define ext2_find_first_zero_bit     find_first_zero_bit
  351. #define ext2_find_next_zero_bit      find_next_zero_bit
  352. /* Bitmap functions for the minix filesystem.  */
  353. #define minix_test_and_set_bit(nr,addr) __test_and_set_bit(nr,addr)
  354. #define minix_set_bit(nr,addr) __set_bit(nr,addr)
  355. #define minix_test_and_clear_bit(nr,addr) __test_and_clear_bit(nr,addr)
  356. #define minix_test_bit(nr,addr) test_bit(nr,addr)
  357. #define minix_find_first_zero_bit(addr,size) find_first_zero_bit(addr,size)
  358. #endif /* __KERNEL__ */
  359. #endif /* _X86_64_BITOPS_H */