blob: 202e4034fe26c41157dda26af230aba553891f1f [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001#ifndef __LINUX_BITMAP_H
2#define __LINUX_BITMAP_H
3
4#ifndef __ASSEMBLY__
5
6#include <linux/types.h>
7#include <linux/bitops.h>
8#include <linux/string.h>
Jiri Slaby14ed9d22007-10-18 23:40:37 -07009#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070010
11/*
12 * bitmaps provide bit arrays that consume one or more unsigned
13 * longs. The bitmap interface and available operations are listed
14 * here, in bitmap.h
15 *
16 * Function implementations generic to all architectures are in
17 * lib/bitmap.c. Functions implementations that are architecture
18 * specific are in various include/asm-<arch>/bitops.h headers
19 * and other arch/<arch> specific files.
20 *
21 * See lib/bitmap.c for more details.
22 */
23
24/*
25 * The available bitmap operations and their rough meaning in the
26 * case that the bitmap is a single unsigned long are thus:
27 *
Andi Kleen08cd3652006-06-26 13:57:10 +020028 * Note that nbits should be always a compile time evaluable constant.
29 * Otherwise many inlines will generate horrible code.
30 *
Linus Torvalds1da177e2005-04-16 15:20:36 -070031 * bitmap_zero(dst, nbits) *dst = 0UL
32 * bitmap_fill(dst, nbits) *dst = ~0UL
33 * bitmap_copy(dst, src, nbits) *dst = *src
34 * bitmap_and(dst, src1, src2, nbits) *dst = *src1 & *src2
35 * bitmap_or(dst, src1, src2, nbits) *dst = *src1 | *src2
36 * bitmap_xor(dst, src1, src2, nbits) *dst = *src1 ^ *src2
37 * bitmap_andnot(dst, src1, src2, nbits) *dst = *src1 & ~(*src2)
38 * bitmap_complement(dst, src, nbits) *dst = ~(*src)
39 * bitmap_equal(src1, src2, nbits) Are *src1 and *src2 equal?
40 * bitmap_intersects(src1, src2, nbits) Do *src1 and *src2 overlap?
41 * bitmap_subset(src1, src2, nbits) Is *src1 a subset of *src2?
42 * bitmap_empty(src, nbits) Are all bits zero in *src?
43 * bitmap_full(src, nbits) Are all bits set in *src?
44 * bitmap_weight(src, nbits) Hamming Weight: number set bits
Akinobu Mitac1a2a962009-12-15 16:48:25 -080045 * bitmap_set(dst, pos, nbits) Set specified bit area
46 * bitmap_clear(dst, pos, nbits) Clear specified bit area
47 * bitmap_find_next_zero_area(buf, len, pos, n, mask) Find bit free area
Michal Nazarewicz5e19b012014-12-12 16:54:45 -080048 * bitmap_find_next_zero_area_off(buf, len, pos, n, mask) as above
Linus Torvalds1da177e2005-04-16 15:20:36 -070049 * bitmap_shift_right(dst, src, n, nbits) *dst = *src >> n
50 * bitmap_shift_left(dst, src, n, nbits) *dst = *src << n
Paul Jacksonfb5eeee2005-10-30 15:02:33 -080051 * bitmap_remap(dst, src, old, new, nbits) *dst = map(old, new)(src)
52 * bitmap_bitremap(oldbit, old, new, nbits) newbit = map(old, new)(oldbit)
Paul Jackson7ea931c2008-04-28 02:12:29 -070053 * bitmap_onto(dst, orig, relmap, nbits) *dst = orig relative to relmap
54 * bitmap_fold(dst, orig, sz, nbits) dst bits = orig bits mod sz
Linus Torvalds1da177e2005-04-16 15:20:36 -070055 * bitmap_scnprintf(buf, len, src, nbits) Print bitmap src to buf
Reinette Chatre01a3ee22006-10-11 01:21:55 -070056 * bitmap_parse(buf, buflen, dst, nbits) Parse bitmap dst from kernel buf
57 * bitmap_parse_user(ubuf, ulen, dst, nbits) Parse bitmap dst from user buf
Linus Torvalds1da177e2005-04-16 15:20:36 -070058 * bitmap_scnlistprintf(buf, len, src, nbits) Print bitmap src as list to buf
Mike Travis4b060422011-05-24 17:13:12 -070059 * bitmap_parselist(buf, dst, nbits) Parse bitmap dst from kernel buf
60 * bitmap_parselist_user(buf, dst, nbits) Parse bitmap dst from user buf
Paul Jackson87e24802006-03-24 03:15:44 -080061 * bitmap_find_free_region(bitmap, bits, order) Find and allocate bit region
62 * bitmap_release_region(bitmap, pos, order) Free specified bit region
63 * bitmap_allocate_region(bitmap, pos, order) Allocate specified bit region
Sudeep Holla5aaba362014-09-30 14:48:22 +010064 * bitmap_print_to_pagebuf(list, buf, mask, nbits) Print bitmap src as list/hex
Linus Torvalds1da177e2005-04-16 15:20:36 -070065 */
66
67/*
68 * Also the following operations in asm/bitops.h apply to bitmaps.
69 *
70 * set_bit(bit, addr) *addr |= bit
71 * clear_bit(bit, addr) *addr &= ~bit
72 * change_bit(bit, addr) *addr ^= bit
73 * test_bit(bit, addr) Is bit set in *addr?
74 * test_and_set_bit(bit, addr) Set bit and return old value
75 * test_and_clear_bit(bit, addr) Clear bit and return old value
76 * test_and_change_bit(bit, addr) Change bit and return old value
77 * find_first_zero_bit(addr, nbits) Position first zero bit in *addr
78 * find_first_bit(addr, nbits) Position first set bit in *addr
79 * find_next_zero_bit(addr, nbits, bit) Position next zero bit in *addr >= bit
80 * find_next_bit(addr, nbits, bit) Position next set bit in *addr >= bit
81 */
82
83/*
84 * The DECLARE_BITMAP(name,bits) macro, in linux/types.h, can be used
85 * to declare an array named 'name' of just enough unsigned longs to
86 * contain all bit positions from 0 to 'bits' - 1.
87 */
88
89/*
90 * lib/bitmap.c provides these functions:
91 */
92
Rasmus Villemoes0679cc42014-08-06 16:09:49 -070093extern int __bitmap_empty(const unsigned long *bitmap, unsigned int nbits);
Rasmus Villemoes83979272014-08-06 16:09:51 -070094extern int __bitmap_full(const unsigned long *bitmap, unsigned int nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -070095extern int __bitmap_equal(const unsigned long *bitmap1,
Rasmus Villemoes5e0680692014-08-06 16:09:53 -070096 const unsigned long *bitmap2, unsigned int nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -070097extern void __bitmap_complement(unsigned long *dst, const unsigned long *src,
Rasmus Villemoes3d6684f2014-08-06 16:09:55 -070098 unsigned int nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -070099extern void __bitmap_shift_right(unsigned long *dst,
100 const unsigned long *src, int shift, int bits);
101extern void __bitmap_shift_left(unsigned long *dst,
102 const unsigned long *src, int shift, int bits);
Linus Torvaldsf4b03732009-08-21 09:26:15 -0700103extern int __bitmap_and(unsigned long *dst, const unsigned long *bitmap1,
Rasmus Villemoes2f9305e2014-08-06 16:09:59 -0700104 const unsigned long *bitmap2, unsigned int nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700105extern void __bitmap_or(unsigned long *dst, const unsigned long *bitmap1,
Rasmus Villemoes2f9305e2014-08-06 16:09:59 -0700106 const unsigned long *bitmap2, unsigned int nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700107extern void __bitmap_xor(unsigned long *dst, const unsigned long *bitmap1,
Rasmus Villemoes2f9305e2014-08-06 16:09:59 -0700108 const unsigned long *bitmap2, unsigned int nbits);
Linus Torvaldsf4b03732009-08-21 09:26:15 -0700109extern int __bitmap_andnot(unsigned long *dst, const unsigned long *bitmap1,
Rasmus Villemoes2f9305e2014-08-06 16:09:59 -0700110 const unsigned long *bitmap2, unsigned int nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700111extern int __bitmap_intersects(const unsigned long *bitmap1,
Rasmus Villemoes6dfe9792014-08-06 16:10:01 -0700112 const unsigned long *bitmap2, unsigned int nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700113extern int __bitmap_subset(const unsigned long *bitmap1,
Rasmus Villemoes5be20212014-08-06 16:10:03 -0700114 const unsigned long *bitmap2, unsigned int nbits);
Rasmus Villemoes877d9f32014-08-06 16:10:05 -0700115extern int __bitmap_weight(const unsigned long *bitmap, unsigned int nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700116
Rasmus Villemoesfb5ac542014-08-06 16:10:07 -0700117extern void bitmap_set(unsigned long *map, unsigned int start, int len);
Rasmus Villemoes154f5e32014-08-06 16:10:10 -0700118extern void bitmap_clear(unsigned long *map, unsigned int start, int len);
Michal Nazarewicz5e19b012014-12-12 16:54:45 -0800119
120extern unsigned long bitmap_find_next_zero_area_off(unsigned long *map,
121 unsigned long size,
122 unsigned long start,
123 unsigned int nr,
124 unsigned long align_mask,
125 unsigned long align_offset);
126
127/**
128 * bitmap_find_next_zero_area - find a contiguous aligned zero area
129 * @map: The address to base the search on
130 * @size: The bitmap size in bits
131 * @start: The bitnumber to start searching at
132 * @nr: The number of zeroed bits we're looking for
133 * @align_mask: Alignment mask for zero area
134 *
135 * The @align_mask should be one less than a power of 2; the effect is that
136 * the bit offset of all zero areas this function finds is multiples of that
137 * power of 2. A @align_mask of 0 means no alignment is required.
138 */
139static inline unsigned long
140bitmap_find_next_zero_area(unsigned long *map,
141 unsigned long size,
142 unsigned long start,
143 unsigned int nr,
144 unsigned long align_mask)
145{
146 return bitmap_find_next_zero_area_off(map, size, start, nr,
147 align_mask, 0);
148}
Akinobu Mitac1a2a962009-12-15 16:48:25 -0800149
Linus Torvalds1da177e2005-04-16 15:20:36 -0700150extern int bitmap_scnprintf(char *buf, unsigned int len,
151 const unsigned long *src, int nbits);
Reinette Chatre01a3ee22006-10-11 01:21:55 -0700152extern int __bitmap_parse(const char *buf, unsigned int buflen, int is_user,
153 unsigned long *dst, int nbits);
154extern int bitmap_parse_user(const char __user *ubuf, unsigned int ulen,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700155 unsigned long *dst, int nbits);
156extern int bitmap_scnlistprintf(char *buf, unsigned int len,
157 const unsigned long *src, int nbits);
158extern int bitmap_parselist(const char *buf, unsigned long *maskp,
159 int nmaskbits);
Mike Travis4b060422011-05-24 17:13:12 -0700160extern int bitmap_parselist_user(const char __user *ubuf, unsigned int ulen,
161 unsigned long *dst, int nbits);
Paul Jacksonfb5eeee2005-10-30 15:02:33 -0800162extern void bitmap_remap(unsigned long *dst, const unsigned long *src,
163 const unsigned long *old, const unsigned long *new, int bits);
164extern int bitmap_bitremap(int oldbit,
165 const unsigned long *old, const unsigned long *new, int bits);
Paul Jackson7ea931c2008-04-28 02:12:29 -0700166extern void bitmap_onto(unsigned long *dst, const unsigned long *orig,
167 const unsigned long *relmap, int bits);
168extern void bitmap_fold(unsigned long *dst, const unsigned long *orig,
169 int sz, int bits);
Rasmus Villemoes9279d322014-08-06 16:10:16 -0700170extern int bitmap_find_free_region(unsigned long *bitmap, unsigned int bits, int order);
171extern void bitmap_release_region(unsigned long *bitmap, unsigned int pos, int order);
172extern int bitmap_allocate_region(unsigned long *bitmap, unsigned int pos, int order);
David Vrabelccbe3292008-09-17 16:34:03 +0100173extern void bitmap_copy_le(void *dst, const unsigned long *src, int nbits);
Michal Hocko778d3b02011-07-26 16:08:30 -0700174extern int bitmap_ord_to_pos(const unsigned long *bitmap, int n, int bits);
Sudeep Holla5aaba362014-09-30 14:48:22 +0100175extern int bitmap_print_to_pagebuf(bool list, char *buf,
176 const unsigned long *maskp, int nmaskbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700177
Huang Ying7f184272011-07-13 13:14:24 +0800178#define BITMAP_FIRST_WORD_MASK(start) (~0UL << ((start) % BITS_PER_LONG))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700179#define BITMAP_LAST_WORD_MASK(nbits) \
180( \
181 ((nbits) % BITS_PER_LONG) ? \
182 (1UL<<((nbits) % BITS_PER_LONG))-1 : ~0UL \
183)
184
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030185#define small_const_nbits(nbits) \
186 (__builtin_constant_p(nbits) && (nbits) <= BITS_PER_LONG)
187
Linus Torvalds1da177e2005-04-16 15:20:36 -0700188static inline void bitmap_zero(unsigned long *dst, int nbits)
189{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030190 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700191 *dst = 0UL;
192 else {
193 int len = BITS_TO_LONGS(nbits) * sizeof(unsigned long);
194 memset(dst, 0, len);
195 }
196}
197
198static inline void bitmap_fill(unsigned long *dst, int nbits)
199{
200 size_t nlongs = BITS_TO_LONGS(nbits);
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030201 if (!small_const_nbits(nbits)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700202 int len = (nlongs - 1) * sizeof(unsigned long);
203 memset(dst, 0xff, len);
204 }
205 dst[nlongs - 1] = BITMAP_LAST_WORD_MASK(nbits);
206}
207
208static inline void bitmap_copy(unsigned long *dst, const unsigned long *src,
209 int nbits)
210{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030211 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700212 *dst = *src;
213 else {
214 int len = BITS_TO_LONGS(nbits) * sizeof(unsigned long);
215 memcpy(dst, src, len);
216 }
217}
218
Linus Torvaldsf4b03732009-08-21 09:26:15 -0700219static inline int bitmap_and(unsigned long *dst, const unsigned long *src1,
Rasmus Villemoes2f9305e2014-08-06 16:09:59 -0700220 const unsigned long *src2, unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700221{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030222 if (small_const_nbits(nbits))
Rasmus Villemoes7e5f97d2014-08-06 16:10:22 -0700223 return (*dst = *src1 & *src2 & BITMAP_LAST_WORD_MASK(nbits)) != 0;
Linus Torvaldsf4b03732009-08-21 09:26:15 -0700224 return __bitmap_and(dst, src1, src2, nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700225}
226
227static inline void bitmap_or(unsigned long *dst, const unsigned long *src1,
Rasmus Villemoes2f9305e2014-08-06 16:09:59 -0700228 const unsigned long *src2, unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700229{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030230 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700231 *dst = *src1 | *src2;
232 else
233 __bitmap_or(dst, src1, src2, nbits);
234}
235
236static inline void bitmap_xor(unsigned long *dst, const unsigned long *src1,
Rasmus Villemoes2f9305e2014-08-06 16:09:59 -0700237 const unsigned long *src2, unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700238{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030239 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700240 *dst = *src1 ^ *src2;
241 else
242 __bitmap_xor(dst, src1, src2, nbits);
243}
244
Linus Torvaldsf4b03732009-08-21 09:26:15 -0700245static inline int bitmap_andnot(unsigned long *dst, const unsigned long *src1,
Rasmus Villemoes2f9305e2014-08-06 16:09:59 -0700246 const unsigned long *src2, unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700247{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030248 if (small_const_nbits(nbits))
Rasmus Villemoes74e76532014-08-06 16:10:24 -0700249 return (*dst = *src1 & ~(*src2) & BITMAP_LAST_WORD_MASK(nbits)) != 0;
Linus Torvaldsf4b03732009-08-21 09:26:15 -0700250 return __bitmap_andnot(dst, src1, src2, nbits);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700251}
252
253static inline void bitmap_complement(unsigned long *dst, const unsigned long *src,
Rasmus Villemoes3d6684f2014-08-06 16:09:55 -0700254 unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700255{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030256 if (small_const_nbits(nbits))
Rasmus Villemoes65b4ee62014-08-06 16:09:57 -0700257 *dst = ~(*src);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700258 else
259 __bitmap_complement(dst, src, nbits);
260}
261
262static inline int bitmap_equal(const unsigned long *src1,
Rasmus Villemoes3d6684f2014-08-06 16:09:55 -0700263 const unsigned long *src2, unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700264{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030265 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700266 return ! ((*src1 ^ *src2) & BITMAP_LAST_WORD_MASK(nbits));
267 else
268 return __bitmap_equal(src1, src2, nbits);
269}
270
271static inline int bitmap_intersects(const unsigned long *src1,
Rasmus Villemoes6dfe9792014-08-06 16:10:01 -0700272 const unsigned long *src2, unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700273{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030274 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700275 return ((*src1 & *src2) & BITMAP_LAST_WORD_MASK(nbits)) != 0;
276 else
277 return __bitmap_intersects(src1, src2, nbits);
278}
279
280static inline int bitmap_subset(const unsigned long *src1,
Rasmus Villemoes5be20212014-08-06 16:10:03 -0700281 const unsigned long *src2, unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700282{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030283 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700284 return ! ((*src1 & ~(*src2)) & BITMAP_LAST_WORD_MASK(nbits));
285 else
286 return __bitmap_subset(src1, src2, nbits);
287}
288
Rasmus Villemoes0679cc42014-08-06 16:09:49 -0700289static inline int bitmap_empty(const unsigned long *src, unsigned nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700290{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030291 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700292 return ! (*src & BITMAP_LAST_WORD_MASK(nbits));
293 else
294 return __bitmap_empty(src, nbits);
295}
296
Rasmus Villemoes83979272014-08-06 16:09:51 -0700297static inline int bitmap_full(const unsigned long *src, unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700298{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030299 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700300 return ! (~(*src) & BITMAP_LAST_WORD_MASK(nbits));
301 else
302 return __bitmap_full(src, nbits);
303}
304
Rasmus Villemoes877d9f32014-08-06 16:10:05 -0700305static inline int bitmap_weight(const unsigned long *src, unsigned int nbits)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700306{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030307 if (small_const_nbits(nbits))
Andi Kleen08cd3652006-06-26 13:57:10 +0200308 return hweight_long(*src & BITMAP_LAST_WORD_MASK(nbits));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700309 return __bitmap_weight(src, nbits);
310}
311
312static inline void bitmap_shift_right(unsigned long *dst,
313 const unsigned long *src, int n, int nbits)
314{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030315 if (small_const_nbits(nbits))
Rasmus Villemoesc5341ec892014-08-06 16:10:20 -0700316 *dst = (*src & BITMAP_LAST_WORD_MASK(nbits)) >> n;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700317 else
318 __bitmap_shift_right(dst, src, n, nbits);
319}
320
321static inline void bitmap_shift_left(unsigned long *dst,
322 const unsigned long *src, int n, int nbits)
323{
Rusty Russell4b0bc0b2008-12-30 09:05:13 +1030324 if (small_const_nbits(nbits))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700325 *dst = (*src << n) & BITMAP_LAST_WORD_MASK(nbits);
326 else
327 __bitmap_shift_left(dst, src, n, nbits);
328}
329
Reinette Chatre01a3ee22006-10-11 01:21:55 -0700330static inline int bitmap_parse(const char *buf, unsigned int buflen,
331 unsigned long *maskp, int nmaskbits)
332{
333 return __bitmap_parse(buf, buflen, 0, maskp, nmaskbits);
334}
335
Linus Torvalds1da177e2005-04-16 15:20:36 -0700336#endif /* __ASSEMBLY__ */
337
338#endif /* __LINUX_BITMAP_H */