libcfs/include/libcfs/user-bitops.h

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  19  *
  20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  21  * CA 95054 USA or visit www.sun.com if you need additional information or
  22  * have any questions.
  23  *
  24  * GPL HEADER END
  25  */
  26 /*
  27  * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  */
  30 /*
  31  * This file is part of Lustre, http://www.lustre.org/
  32  * Lustre is a trademark of Sun Microsystems, Inc.
  33  *
  34  * libcfs/include/libcfs/user-bitops.h
  35  *
  36  * Author: Nikita Danilov <nikita@clusterfs.com>
  37  */
  38
  39 #ifndef __LIBCFS_USER_BITOPS_H__
  40 #define __LIBCFS_USER_BITOPS_H__
  41
  42 /* test if bit nr is set in bitmap addr; returns previous value of bit nr */
  43 static inline int test_and_set_bit(int nr, unsigned long *addr)
  44 {
  45         unsigned long mask;
  46
  47         addr += nr / BITS_PER_LONG;
  48         mask = 1UL << (nr & (BITS_PER_LONG - 1));
  49         nr = (mask & *addr) != 0;
  50         *addr |= mask;
  51         return nr;
  52 }
  53
  54 #define set_bit(n, a) test_and_set_bit(n, a)
  55
  56 /* clear bit nr in bitmap addr; returns previous value of bit nr*/
  57 static inline int test_and_clear_bit(int nr, unsigned long *addr)
  58 {
  59         unsigned long mask;
  60
  61         addr += nr / BITS_PER_LONG;
  62         mask = 1UL << (nr & (BITS_PER_LONG - 1));
  63         nr = (mask & *addr) != 0;
  64         *addr &= ~mask;
  65         return nr;
  66 }
  67
  68 #define clear_bit(n, a) test_and_clear_bit(n, a)
  69
  70 static inline int test_bit(int nr, const unsigned long *addr)
  71 {
  72         return ((1UL << (nr & (BITS_PER_LONG - 1))) &
  73                 ((addr)[nr / BITS_PER_LONG])) != 0;
  74 }
  75
  76 /* using binary seach */
  77 static __inline__ unsigned long __cfs_fls(long data)
  78 {
  79         int pos = 32;
  80
  81         if (!data)
  82                 return 0;
  83
  84 #if BITS_PER_LONG == 64
  85         /* If any bit of the high 32 bits are set, shift the high
  86          * 32 bits down and pretend like it is a 32-bit value. */
  87         if ((data & 0xFFFFFFFF00000000llu)) {
  88                 data >>= 32;
  89                 pos += 32;
  90         }
  91 #endif
  92
  93         if (!(data & 0xFFFF0000u)) {
  94                 data <<= 16;
  95                 pos -= 16;
  96         }
  97         if (!(data & 0xFF000000u)) {
  98                 data <<= 8;
  99                 pos -= 8;
 100         }
 101         if (!(data & 0xF0000000u)) {
 102                 data <<= 4;
 103                 pos -= 4;
 104         }
 105         if (!(data & 0xC0000000u)) {
 106                 data <<= 2;
 107                 pos -= 2;
 108         }
 109         if (!(data & 0x80000000u)) {
 110                 data <<= 1;
 111                 pos -= 1;
 112         }
 113         return pos;
 114 }
 115
 116 static __inline__ unsigned long __cfs_ffs(long data)
 117 {
 118         int pos = 0;
 119
 120 #if BITS_PER_LONG == 64
 121         if ((data & 0xFFFFFFFF) == 0) {
 122                 pos += 32;
 123                 data >>= 32;
 124         }
 125 #endif
 126         if ((data & 0xFFFF) == 0) {
 127                 pos += 16;
 128                 data >>= 16;
 129         }
 130         if ((data & 0xFF) == 0) {
 131                 pos += 8;
 132                 data >>= 8;
 133         }
 134         if ((data & 0xF) == 0) {
 135                 pos += 4;
 136                 data >>= 4;
 137         }
 138         if ((data & 0x3) == 0) {
 139                 pos += 2;
 140                 data >>= 2;
 141         }
 142         if ((data & 0x1) == 0)
 143                 pos += 1;
 144
 145         return pos;
 146 }
 147
 148 #define __cfs_ffz(x)    __cfs_ffs(~(x))
 149 #define __cfs_flz(x)    __cfs_fls(~(x))
 150
 151 unsigned long find_next_bit(unsigned long *addr,
 152                             unsigned long size, unsigned long offset);
 153
 154 unsigned long find_next_zero_bit(unsigned long *addr,
 155                                  unsigned long size, unsigned long offset);
 156
 157 #define find_first_bit(addr, size)       find_next_bit((addr), (size),0)
 158 #define find_first_zero_bit(addr, size)  find_next_zero_bit((addr), (size),0)
 159
 160 #endif