Commit ea15c79c authored by David Gibson's avatar David Gibson

bitmap: Use a typesafe(ish) parameters instead of void *

This also fixes a bug which will hit on platforms where it's not safe to
dereference a (long *) which is not correctly aligned.
Signed-off-by: default avatarDavid Gibson <david@gibson.dropbear.id.au>
parent ba88f53b
...@@ -8,6 +8,14 @@ ...@@ -8,6 +8,14 @@
#define BITS_PER_LONG (sizeof(unsigned long) * 8) #define BITS_PER_LONG (sizeof(unsigned long) * 8)
/*
* We use an unsigned long to get alignment, but wrap it in a
* structure for type checking
*/
typedef struct {
unsigned long l;
} bitmap;
#define BYTE(_bm, _n) (((unsigned char *)(_bm))[(_n) / 8]) #define BYTE(_bm, _n) (((unsigned char *)(_bm))[(_n) / 8])
#define LONG(_bm, _n) (((unsigned long *)(_bm))[(_n) / BITS_PER_LONG]) #define LONG(_bm, _n) (((unsigned long *)(_bm))[(_n) / BITS_PER_LONG])
#define BIT(_n) (0x80 >> ((_n) % 8)) #define BIT(_n) (0x80 >> ((_n) % 8))
...@@ -15,42 +23,42 @@ ...@@ -15,42 +23,42 @@
#define BYTES(_nbits) ((_nbits) / 8) #define BYTES(_nbits) ((_nbits) / 8)
#define BITS(_nbits) ((~(0xff >> ((_nbits) % 8))) & 0xff) #define BITS(_nbits) ((~(0xff >> ((_nbits) % 8))) & 0xff)
static inline void bitmap_set_bit(void *bitmap, int n) static inline void bitmap_set_bit(bitmap *bitmap, int n)
{ {
BYTE(bitmap, n) |= BIT(n); BYTE(bitmap, n) |= BIT(n);
} }
static inline void bitmap_clear_bit(void *bitmap, int n) static inline void bitmap_clear_bit(bitmap *bitmap, int n)
{ {
BYTE(bitmap, n) &= ~BIT(n); BYTE(bitmap, n) &= ~BIT(n);
} }
static inline void bitmap_change_bit(void *bitmap, int n) static inline void bitmap_change_bit(bitmap *bitmap, int n)
{ {
BYTE(bitmap, n) ^= BIT(n); BYTE(bitmap, n) ^= BIT(n);
} }
static inline bool bitmap_test_bit(void *bitmap, int n) static inline bool bitmap_test_bit(bitmap *bitmap, int n)
{ {
return !!(BYTE(bitmap, n) & BIT(n)); return !!(BYTE(bitmap, n) & BIT(n));
} }
static inline void bitmap_zero(void *bitmap, int nbits) static inline void bitmap_zero(bitmap *bitmap, int nbits)
{ {
memset(bitmap, 0, BYTES(nbits)); memset(bitmap, 0, BYTES(nbits));
if (BITS(nbits)) if (BITS(nbits))
BYTE(bitmap, nbits) &= ~BITS(nbits); BYTE(bitmap, nbits) &= ~BITS(nbits);
} }
static inline void bitmap_fill(void *bitmap, int nbits) static inline void bitmap_fill(bitmap *bitmap, int nbits)
{ {
memset(bitmap, 0xff, BYTES(nbits)); memset(bitmap, 0xff, BYTES(nbits));
if (BITS(nbits)) if (BITS(nbits))
BYTE(bitmap, nbits) |= BITS(nbits); BYTE(bitmap, nbits) |= BITS(nbits);
} }
static inline void bitmap_copy(void *dst, void *src, int nbits) static inline void bitmap_copy(bitmap *dst, bitmap *src, int nbits)
{ {
memcpy(dst, src, BYTES(nbits)); memcpy(dst, src, BYTES(nbits));
if (BITS(nbits)) { if (BITS(nbits)) {
...@@ -60,7 +68,7 @@ static inline void bitmap_copy(void *dst, void *src, int nbits) ...@@ -60,7 +68,7 @@ static inline void bitmap_copy(void *dst, void *src, int nbits)
} }
#define DEF_BINOP(_name, _op) \ #define DEF_BINOP(_name, _op) \
static inline void bitmap_##_name(void *dst, void *src1, void *src2, \ static inline void bitmap_##_name(bitmap *dst, bitmap *src1, bitmap *src2, \
int nbits) \ int nbits) \
{ \ { \
int n = 0; \ int n = 0; \
...@@ -86,7 +94,7 @@ DEF_BINOP(andnot, & ~) ...@@ -86,7 +94,7 @@ DEF_BINOP(andnot, & ~)
#undef DEF_BINOP #undef DEF_BINOP
static inline void bitmap_complement(void *dst, void *src, int nbits) static inline void bitmap_complement(bitmap *dst, bitmap *src, int nbits)
{ {
int n = 0; int n = 0;
...@@ -104,7 +112,7 @@ static inline void bitmap_complement(void *dst, void *src, int nbits) ...@@ -104,7 +112,7 @@ static inline void bitmap_complement(void *dst, void *src, int nbits)
} }
} }
static inline bool bitmap_equal(void *src1, void *src2, int nbits) static inline bool bitmap_equal(bitmap *src1, bitmap *src2, int nbits)
{ {
if (memcmp(src1, src2, BYTES(nbits)) != 0) if (memcmp(src1, src2, BYTES(nbits)) != 0)
return false; return false;
...@@ -114,7 +122,7 @@ static inline bool bitmap_equal(void *src1, void *src2, int nbits) ...@@ -114,7 +122,7 @@ static inline bool bitmap_equal(void *src1, void *src2, int nbits)
return true; return true;
} }
static inline bool bitmap_intersects(void *src1, void *src2, int nbits) static inline bool bitmap_intersects(bitmap *src1, bitmap *src2, int nbits)
{ {
int n = 0; int n = 0;
...@@ -134,7 +142,7 @@ static inline bool bitmap_intersects(void *src1, void *src2, int nbits) ...@@ -134,7 +142,7 @@ static inline bool bitmap_intersects(void *src1, void *src2, int nbits)
return false; return false;
} }
static inline bool bitmap_subset(void *src1, void *src2, int nbits) static inline bool bitmap_subset(bitmap *src1, bitmap *src2, int nbits)
{ {
int n = 0; int n = 0;
...@@ -154,7 +162,7 @@ static inline bool bitmap_subset(void *src1, void *src2, int nbits) ...@@ -154,7 +162,7 @@ static inline bool bitmap_subset(void *src1, void *src2, int nbits)
return true; return true;
} }
static inline bool bitmap_full(void *bitmap, int nbits) static inline bool bitmap_full(bitmap *bitmap, int nbits)
{ {
int n = 0; int n = 0;
...@@ -175,7 +183,7 @@ static inline bool bitmap_full(void *bitmap, int nbits) ...@@ -175,7 +183,7 @@ static inline bool bitmap_full(void *bitmap, int nbits)
return true; return true;
} }
static inline bool bitmap_empty(void *bitmap, int nbits) static inline bool bitmap_empty(bitmap *bitmap, int nbits)
{ {
int n = 0; int n = 0;
...@@ -196,7 +204,7 @@ static inline bool bitmap_empty(void *bitmap, int nbits) ...@@ -196,7 +204,7 @@ static inline bool bitmap_empty(void *bitmap, int nbits)
} }
static inline void *bitmap_alloc(int nbits) static inline bitmap *bitmap_alloc(int nbits)
{ {
return malloc((nbits + 7) / 8); return malloc((nbits + 7) / 8);
} }
......
...@@ -13,7 +13,7 @@ int bitmap_sizes[] = { ...@@ -13,7 +13,7 @@ int bitmap_sizes[] = {
static void test_sizes(int nbits) static void test_sizes(int nbits)
{ {
void *bitmap = bitmap_alloc(nbits); bitmap *bitmap = bitmap_alloc(nbits);
int i, j; int i, j;
bool wrong; bool wrong;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment