[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]

[Xen-changelog] [xen master] bitmap_*() should cope with zero size bitmaps



commit d8a7694e5a415ac0a871f0ae58f50876ad30d619
Author:     Jan Beulich <jbeulich@xxxxxxxx>
AuthorDate: Thu Jul 4 10:26:24 2013 +0200
Commit:     Jan Beulich <jbeulich@xxxxxxxx>
CommitDate: Thu Jul 4 10:26:24 2013 +0200

    bitmap_*() should cope with zero size bitmaps
    
    ... to match expectations set by memset()/memcpy().
    
    Similarly for find_{first,next}_{,zero_}_bit() on x86.
    
    __bitmap_shift_{left,right}() would also need fixing (they more
    generally can't cope with the shift count being larger than the bitmap
    size, and they perform undefined operations by possibly shifting an
    unsigned long value by BITS_PER_LONG bits), but since these functions
    aren't really used anywhere I wonder if we wouldn't better simply get
    rid of them.
    
    Signed-off-by: Jan Beulich <jbeulich@xxxxxxxx>
    Acked-by: Keir Fraser <keir@xxxxxxx>
---
 xen/include/asm-x86/bitops.h |   54 ++++++++++++-----
 xen/include/xen/bitmap.h     |  128 +++++++++++++++++++++---------------------
 2 files changed, 102 insertions(+), 80 deletions(-)

diff --git a/xen/include/asm-x86/bitops.h b/xen/include/asm-x86/bitops.h
index f5a84ef..ab21d92 100644
--- a/xen/include/asm-x86/bitops.h
+++ b/xen/include/asm-x86/bitops.h
@@ -327,10 +327,7 @@ static inline unsigned int __scanbit(unsigned long val, 
unsigned long max)
  * Returns the bit-number of the first set bit, not the number of the byte
  * containing a bit.
  */
-#define find_first_bit(addr,size)                               \
-((__builtin_constant_p(size) && (size) <= BITS_PER_LONG ?       \
-  (__scanbit(*(const unsigned long *)addr, size)) :             \
-  __find_first_bit(addr,size)))
+#define find_first_bit(addr, size) find_next_bit(addr, size, 0)
 
 /**
  * find_next_bit - find the first set bit in a memory region
@@ -338,10 +335,24 @@ static inline unsigned int __scanbit(unsigned long val, 
unsigned long max)
  * @offset: The bitnumber to start searching at
  * @size: The maximum size to search
  */
-#define find_next_bit(addr,size,off)                                     \
-((__builtin_constant_p(size) && (size) <= BITS_PER_LONG ?                \
-  ((off) + (__scanbit((*(const unsigned long *)addr) >> (off), size))) : \
-  __find_next_bit(addr,size,off)))
+#define find_next_bit(addr, size, off) ({ \
+    unsigned int r__ = (size); \
+    unsigned int o__ = (off); \
+    switch ( -!__builtin_constant_p(size) | r__ ) \
+    { \
+    case 0: (void)(addr); break; \
+    case 1 ... BITS_PER_LONG: \
+        r__ = o__ + __scanbit(*(const unsigned long *)(addr) >> o__, r__); \
+        break; \
+    default: \
+        if ( __builtin_constant_p(off) && !o__ ) \
+            r__ = __find_first_bit(addr, r__); \
+        else \
+            r__ = __find_next_bit(addr, r__, o__); \
+        break; \
+    } \
+    r__; \
+})
 
 /**
  * find_first_zero_bit - find the first zero bit in a memory region
@@ -351,10 +362,7 @@ static inline unsigned int __scanbit(unsigned long val, 
unsigned long max)
  * Returns the bit-number of the first zero bit, not the number of the byte
  * containing a bit.
  */
-#define find_first_zero_bit(addr,size)                          \
-((__builtin_constant_p(size) && (size) <= BITS_PER_LONG ?       \
-  (__scanbit(~*(const unsigned long *)addr, size)) :            \
-  __find_first_zero_bit(addr,size)))
+#define find_first_zero_bit(addr, size) find_next_zero_bit(addr, size, 0)
 
 /**
  * find_next_zero_bit - find the first zero bit in a memory region
@@ -362,10 +370,24 @@ static inline unsigned int __scanbit(unsigned long val, 
unsigned long max)
  * @offset: The bitnumber to start searching at
  * @size: The maximum size to search
  */
-#define find_next_zero_bit(addr,size,off)                                   \
-((__builtin_constant_p(size) && (size) <= BITS_PER_LONG ?                   \
-  ((off)+(__scanbit(~(((*(const unsigned long *)addr)) >> (off)), size))) : \
-  __find_next_zero_bit(addr,size,off)))
+#define find_next_zero_bit(addr, size, off) ({ \
+    unsigned int r__ = (size); \
+    unsigned int o__ = (off); \
+    switch ( -!__builtin_constant_p(size) | r__ ) \
+    { \
+    case 0: (void)(addr); break; \
+    case 1 ... BITS_PER_LONG: \
+        r__ = o__ + __scanbit(~*(const unsigned long *)(addr) >> o__, r__); \
+        break; \
+    default: \
+        if ( __builtin_constant_p(off) && !o__ ) \
+            r__ = __find_first_zero_bit(addr, r__); \
+        else \
+            r__ = __find_next_zero_bit(addr, r__, o__); \
+        break; \
+    } \
+    r__; \
+})
 
 /**
  * find_first_set_bit - find the first set bit in @word
diff --git a/xen/include/xen/bitmap.h b/xen/include/xen/bitmap.h
index 56fe602..b5ec455 100644
--- a/xen/include/xen/bitmap.h
+++ b/xen/include/xen/bitmap.h
@@ -108,123 +108,122 @@ extern int bitmap_allocate_region(unsigned long 
*bitmap, int pos, int order);
                (1UL<<((nbits) % BITS_PER_LONG))-1 : ~0UL               \
 )
 
+#define bitmap_bytes(nbits) (BITS_TO_LONGS(nbits) * sizeof(unsigned long))
+
+#define bitmap_switch(nbits, zero_ret, small, large)                   \
+       switch (-!__builtin_constant_p(nbits) | (nbits)) {              \
+       case 0: return zero_ret;                                        \
+       case 1 ... BITS_PER_LONG:                                       \
+               small; break;                                           \
+       default:                                                        \
+               large; break;                                           \
+       }
+
 static inline void bitmap_zero(unsigned long *dst, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               *dst = 0UL;
-       else {
-               int len = BITS_TO_LONGS(nbits) * sizeof(unsigned long);
-               memset(dst, 0, len);
-       }
+       bitmap_switch(nbits,,
+               *dst = 0UL,
+               memset(dst, 0, bitmap_bytes(nbits)));
 }
 
 static inline void bitmap_fill(unsigned long *dst, int nbits)
 {
        size_t nlongs = BITS_TO_LONGS(nbits);
-       if (nlongs > 1) {
-               int len = (nlongs - 1) * sizeof(unsigned long);
-               memset(dst, 0xff,  len);
+
+       switch (nlongs) {
+       default:
+               memset(dst, -1, (nlongs - 1) * sizeof(unsigned long));
+               /* fall through */
+       case 1:
+               dst[nlongs - 1] = BITMAP_LAST_WORD_MASK(nbits);
+               break;
        }
-       dst[nlongs - 1] = BITMAP_LAST_WORD_MASK(nbits);
 }
 
 static inline void bitmap_copy(unsigned long *dst, const unsigned long *src,
                        int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               *dst = *src;
-       else {
-               int len = BITS_TO_LONGS(nbits) * sizeof(unsigned long);
-               memcpy(dst, src, len);
-       }
+       bitmap_switch(nbits,,
+               *dst = *src,
+               memcpy(dst, src, bitmap_bytes(nbits)));
 }
 
 static inline void bitmap_and(unsigned long *dst, const unsigned long *src1,
                        const unsigned long *src2, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               *dst = *src1 & *src2;
-       else
-               __bitmap_and(dst, src1, src2, nbits);
+       bitmap_switch(nbits,,
+               *dst = *src1 & *src2,
+               __bitmap_and(dst, src1, src2, nbits));
 }
 
 static inline void bitmap_or(unsigned long *dst, const unsigned long *src1,
                        const unsigned long *src2, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               *dst = *src1 | *src2;
-       else
-               __bitmap_or(dst, src1, src2, nbits);
+       bitmap_switch(nbits,,
+               *dst = *src1 | *src2,
+               __bitmap_or(dst, src1, src2, nbits));
 }
 
 static inline void bitmap_xor(unsigned long *dst, const unsigned long *src1,
                        const unsigned long *src2, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               *dst = *src1 ^ *src2;
-       else
-               __bitmap_xor(dst, src1, src2, nbits);
+       bitmap_switch(nbits,,
+               *dst = *src1 ^ *src2,
+               __bitmap_xor(dst, src1, src2, nbits));
 }
 
 static inline void bitmap_andnot(unsigned long *dst, const unsigned long *src1,
                        const unsigned long *src2, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               *dst = *src1 & ~(*src2);
-       else
-               __bitmap_andnot(dst, src1, src2, nbits);
+       bitmap_switch(nbits,,
+               *dst = *src1 & ~*src2,
+               __bitmap_andnot(dst, src1, src2, nbits));
 }
 
 static inline void bitmap_complement(unsigned long *dst, const unsigned long 
*src,
                        int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               *dst = ~(*src) & BITMAP_LAST_WORD_MASK(nbits);
-       else
-               __bitmap_complement(dst, src, nbits);
+       bitmap_switch(nbits,,
+               *dst = ~*src & BITMAP_LAST_WORD_MASK(nbits),
+               __bitmap_complement(dst, src, nbits));
 }
 
 static inline int bitmap_equal(const unsigned long *src1,
                        const unsigned long *src2, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               return ! ((*src1 ^ *src2) & BITMAP_LAST_WORD_MASK(nbits));
-       else
-               return __bitmap_equal(src1, src2, nbits);
+       bitmap_switch(nbits, -1,
+               return !((*src1 ^ *src2) & BITMAP_LAST_WORD_MASK(nbits)),
+               return __bitmap_equal(src1, src2, nbits));
 }
 
 static inline int bitmap_intersects(const unsigned long *src1,
                        const unsigned long *src2, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               return ((*src1 & *src2) & BITMAP_LAST_WORD_MASK(nbits)) != 0;
-       else
-               return __bitmap_intersects(src1, src2, nbits);
+       bitmap_switch(nbits, -1,
+               return ((*src1 & *src2) & BITMAP_LAST_WORD_MASK(nbits)) != 0,
+               return __bitmap_intersects(src1, src2, nbits));
 }
 
 static inline int bitmap_subset(const unsigned long *src1,
                        const unsigned long *src2, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               return ! ((*src1 & ~(*src2)) & BITMAP_LAST_WORD_MASK(nbits));
-       else
-               return __bitmap_subset(src1, src2, nbits);
+       bitmap_switch(nbits, -1,
+               return !((*src1 & ~*src2) & BITMAP_LAST_WORD_MASK(nbits)),
+               return __bitmap_subset(src1, src2, nbits));
 }
 
 static inline int bitmap_empty(const unsigned long *src, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               return ! (*src & BITMAP_LAST_WORD_MASK(nbits));
-       else
-               return __bitmap_empty(src, nbits);
+       bitmap_switch(nbits, -1,
+               return !(*src & BITMAP_LAST_WORD_MASK(nbits)),
+               return __bitmap_empty(src, nbits));
 }
 
 static inline int bitmap_full(const unsigned long *src, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               return ! (~(*src) & BITMAP_LAST_WORD_MASK(nbits));
-       else
-               return __bitmap_full(src, nbits);
+       bitmap_switch(nbits, -1,
+               return !(~*src & BITMAP_LAST_WORD_MASK(nbits)),
+               return __bitmap_full(src, nbits));
 }
 
 static inline int bitmap_weight(const unsigned long *src, int nbits)
@@ -235,21 +234,22 @@ static inline int bitmap_weight(const unsigned long *src, 
int nbits)
 static inline void bitmap_shift_right(unsigned long *dst,
                        const unsigned long *src, int n, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               *dst = *src >> n;
-       else
-               __bitmap_shift_right(dst, src, n, nbits);
+       bitmap_switch(nbits,,
+               *dst = *src >> n,
+               __bitmap_shift_right(dst, src, n, nbits));
 }
 
 static inline void bitmap_shift_left(unsigned long *dst,
                        const unsigned long *src, int n, int nbits)
 {
-       if (nbits <= BITS_PER_LONG)
-               *dst = (*src << n) & BITMAP_LAST_WORD_MASK(nbits);
-       else
-               __bitmap_shift_left(dst, src, n, nbits);
+       bitmap_switch(nbits,,
+               *dst = (*src << n) & BITMAP_LAST_WORD_MASK(nbits),
+               __bitmap_shift_left(dst, src, n, nbits));
 }
 
+#undef bitmap_switch
+#undef bitmap_bytes
+
 void bitmap_long_to_byte(uint8_t *bp, const unsigned long *lp, int nbits);
 void bitmap_byte_to_long(unsigned long *lp, const uint8_t *bp, int nbits);
 
--
generated by git-patchbot for /home/xen/git/xen.git#master

_______________________________________________
Xen-changelog mailing list
Xen-changelog@xxxxxxxxxxxxx
http://lists.xensource.com/xen-changelog


 


Rackspace

Lists.xenproject.org is hosted with RackSpace, monitoring our
servers 24x7x365 and backed by RackSpace's Fanatical Support®.