X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=lib%2Fbitmap.c;h=51ad5bf919eaa2bb39e46704bbfe1dee658da6cd;hb=4b0424809b823101c969a0691fc1db0c880ae64a;hp=d607526d5f55be3234ac0be98fe8b324819c5674;hpb=8706009e555bb9fa04a5679e4be2c7c67506802b;p=sliver-openvswitch.git

diff --git a/lib/bitmap.c b/lib/bitmap.c
index d607526d5..51ad5bf91 100644
--- a/lib/bitmap.c
+++ b/lib/bitmap.c
@@ -17,6 +17,7 @@
 #include <config.h>
 #include "bitmap.h"
 #include <string.h>
+#include "util.h"
 
 /* Allocates and returns a bitmap initialized to all-1-bits. */
 unsigned long *
@@ -24,6 +25,7 @@ bitmap_allocate1(size_t n_bits)
 {
     size_t n_bytes = bitmap_n_bytes(n_bits);
     size_t n_longs = bitmap_n_longs(n_bits);
+    size_t r_bits = n_bits % BITMAP_ULONG_BITS;
     unsigned long *bitmap;
 
     /* Allocate and initialize most of the bitmap. */
@@ -32,7 +34,9 @@ bitmap_allocate1(size_t n_bits)
 
     /* Ensure that the last "unsigned long" in the bitmap only has as many
      * 1-bits as there actually should be. */
-    bitmap[n_longs - 1] = (1UL << (n_bits % BITMAP_ULONG_BITS)) - 1;
+    if (r_bits) {
+        bitmap[n_longs - 1] = (1UL << r_bits) - 1;
+    }
 
     return bitmap;
 }
@@ -89,3 +93,18 @@ bitmap_scan(const unsigned long int *bitmap, size_t start, size_t end)
     }
     return i;
 }
+
+/* Returns the number of 1-bits in the 'n'-bit bitmap at 'bitmap'. */
+size_t
+bitmap_count1(const unsigned long int *bitmap, size_t n)
+{
+    size_t i;
+    size_t count = 0;
+
+    BUILD_ASSERT(ULONG_MAX <= UINT64_MAX);
+    for (i = 0; i < BITMAP_N_LONGS(n); i++) {
+        count += count_1bits(bitmap[i]);
+    }
+
+    return count;
+}