parent
04382f7d6f
commit
bdb112d35f
@ -0,0 +1,89 @@
|
||||
/*
|
||||
* Bitfield
|
||||
* Copyright (c) 2013, Jouni Malinen <j@w1.fi>
|
||||
*
|
||||
* This software may be distributed under the terms of the BSD license.
|
||||
* See README for more details.
|
||||
*/
|
||||
|
||||
#include "includes.h"
|
||||
|
||||
#include "common.h"
|
||||
#include "bitfield.h"
|
||||
|
||||
|
||||
struct bitfield {
|
||||
u8 *bits;
|
||||
size_t max_bits;
|
||||
};
|
||||
|
||||
|
||||
struct bitfield * bitfield_alloc(size_t max_bits)
|
||||
{
|
||||
struct bitfield *bf;
|
||||
|
||||
bf = os_zalloc(sizeof(*bf) + (max_bits + 7) / 8);
|
||||
if (bf == NULL)
|
||||
return NULL;
|
||||
bf->bits = (u8 *) (bf + 1);
|
||||
bf->max_bits = max_bits;
|
||||
return bf;
|
||||
}
|
||||
|
||||
|
||||
void bitfield_free(struct bitfield *bf)
|
||||
{
|
||||
os_free(bf);
|
||||
}
|
||||
|
||||
|
||||
void bitfield_set(struct bitfield *bf, size_t bit)
|
||||
{
|
||||
if (bit >= bf->max_bits)
|
||||
return;
|
||||
bf->bits[bit / 8] |= BIT(bit % 8);
|
||||
}
|
||||
|
||||
|
||||
void bitfield_clear(struct bitfield *bf, size_t bit)
|
||||
{
|
||||
if (bit >= bf->max_bits)
|
||||
return;
|
||||
bf->bits[bit / 8] &= ~BIT(bit % 8);
|
||||
}
|
||||
|
||||
|
||||
int bitfield_is_set(struct bitfield *bf, size_t bit)
|
||||
{
|
||||
if (bit >= bf->max_bits)
|
||||
return 0;
|
||||
return !!(bf->bits[bit / 8] & BIT(bit % 8));
|
||||
}
|
||||
|
||||
|
||||
static int first_zero(u8 val)
|
||||
{
|
||||
int i;
|
||||
for (i = 0; i < 8; i++) {
|
||||
if (!(val & 0x01))
|
||||
return i;
|
||||
val >>= 1;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
|
||||
int bitfield_get_first_zero(struct bitfield *bf)
|
||||
{
|
||||
size_t i;
|
||||
for (i = 0; i <= (bf->max_bits + 7) / 8; i++) {
|
||||
if (bf->bits[i] != 0xff)
|
||||
break;
|
||||
}
|
||||
if (i > (bf->max_bits + 7) / 8)
|
||||
return -1;
|
||||
i = i * 8 + first_zero(bf->bits[i]);
|
||||
if (i >= bf->max_bits)
|
||||
return -1;
|
||||
return i;
|
||||
}
|
@ -0,0 +1,21 @@
|
||||
/*
|
||||
* Bitfield
|
||||
* Copyright (c) 2013, Jouni Malinen <j@w1.fi>
|
||||
*
|
||||
* This software may be distributed under the terms of the BSD license.
|
||||
* See README for more details.
|
||||
*/
|
||||
|
||||
#ifndef BITFIELD_H
|
||||
#define BITFIELD_H
|
||||
|
||||
struct bitfield;
|
||||
|
||||
struct bitfield * bitfield_alloc(size_t max_bits);
|
||||
void bitfield_free(struct bitfield *bf);
|
||||
void bitfield_set(struct bitfield *bf, size_t bit);
|
||||
void bitfield_clear(struct bitfield *bf, size_t bit);
|
||||
int bitfield_is_set(struct bitfield *bf, size_t bit);
|
||||
int bitfield_get_first_zero(struct bitfield *bf);
|
||||
|
||||
#endif /* BITFIELD_H */
|
@ -0,0 +1,90 @@
|
||||
/*
|
||||
* bitfield unit tests
|
||||
* Copyright (c) 2013, Jouni Malinen <j@w1.fi>
|
||||
*
|
||||
* This software may be distributed under the terms of the BSD license.
|
||||
* See README for more details.
|
||||
*/
|
||||
|
||||
#include "utils/includes.h"
|
||||
#include "utils/bitfield.h"
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
struct bitfield *bf;
|
||||
int i;
|
||||
int errors = 0;
|
||||
|
||||
bf = bitfield_alloc(123);
|
||||
if (bf == NULL)
|
||||
return -1;
|
||||
|
||||
for (i = 0; i < 123; i++) {
|
||||
if (bitfield_is_set(bf, i) || bitfield_is_set(bf, i + 1))
|
||||
errors++;
|
||||
if (i > 0 && bitfield_is_set(bf, i - 1))
|
||||
errors++;
|
||||
bitfield_set(bf, i);
|
||||
if (!bitfield_is_set(bf, i))
|
||||
errors++;
|
||||
bitfield_clear(bf, i);
|
||||
if (bitfield_is_set(bf, i))
|
||||
errors++;
|
||||
}
|
||||
|
||||
for (i = 123; i < 200; i++) {
|
||||
if (bitfield_is_set(bf, i) || bitfield_is_set(bf, i + 1))
|
||||
errors++;
|
||||
if (i > 0 && bitfield_is_set(bf, i - 1))
|
||||
errors++;
|
||||
bitfield_set(bf, i);
|
||||
if (bitfield_is_set(bf, i))
|
||||
errors++;
|
||||
bitfield_clear(bf, i);
|
||||
if (bitfield_is_set(bf, i))
|
||||
errors++;
|
||||
}
|
||||
|
||||
for (i = 0; i < 123; i++) {
|
||||
if (bitfield_is_set(bf, i) || bitfield_is_set(bf, i + 1))
|
||||
errors++;
|
||||
bitfield_set(bf, i);
|
||||
if (!bitfield_is_set(bf, i))
|
||||
errors++;
|
||||
}
|
||||
|
||||
for (i = 0; i < 123; i++) {
|
||||
if (!bitfield_is_set(bf, i))
|
||||
errors++;
|
||||
bitfield_clear(bf, i);
|
||||
if (bitfield_is_set(bf, i))
|
||||
errors++;
|
||||
}
|
||||
|
||||
for (i = 0; i < 123; i++) {
|
||||
if (bitfield_get_first_zero(bf) != i)
|
||||
errors++;
|
||||
bitfield_set(bf, i);
|
||||
}
|
||||
if (bitfield_get_first_zero(bf) != -1)
|
||||
errors++;
|
||||
for (i = 0; i < 123; i++) {
|
||||
if (!bitfield_is_set(bf, i))
|
||||
errors++;
|
||||
bitfield_clear(bf, i);
|
||||
if (bitfield_get_first_zero(bf) != i)
|
||||
errors++;
|
||||
bitfield_set(bf, i);
|
||||
}
|
||||
if (bitfield_get_first_zero(bf) != -1)
|
||||
errors++;
|
||||
|
||||
bitfield_free(bf);
|
||||
|
||||
if (errors) {
|
||||
printf("%d test(s) failed\n", errors);
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
Loading…
Reference in New Issue