1 #include <linux/compiler.h>
2 #include <linux/bitmap.h>
3 #include "tests.h"
4 #include "cpumap.h"
5 #include "debug.h"
6
7 #define NBITS 100
8
get_bitmap(const char * str,int nbits)9 static unsigned long *get_bitmap(const char *str, int nbits)
10 {
11 struct cpu_map *map = cpu_map__new(str);
12 unsigned long *bm = NULL;
13 int i;
14
15 bm = bitmap_alloc(nbits);
16
17 if (map && bm) {
18 bitmap_zero(bm, nbits);
19
20 for (i = 0; i < map->nr; i++)
21 set_bit(map->map[i], bm);
22 }
23
24 if (map)
25 cpu_map__put(map);
26 return bm;
27 }
28
test_bitmap(const char * str)29 static int test_bitmap(const char *str)
30 {
31 unsigned long *bm = get_bitmap(str, NBITS);
32 char buf[100];
33 int ret;
34
35 bitmap_scnprintf(bm, NBITS, buf, sizeof(buf));
36 pr_debug("bitmap: %s\n", buf);
37
38 ret = !strcmp(buf, str);
39 free(bm);
40 return ret;
41 }
42
test__bitmap_print(int subtest __maybe_unused)43 int test__bitmap_print(int subtest __maybe_unused)
44 {
45 TEST_ASSERT_VAL("failed to convert map", test_bitmap("1"));
46 TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,5"));
47 TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3,5,7,9,11,13,15,17,19,21-40"));
48 TEST_ASSERT_VAL("failed to convert map", test_bitmap("2-5"));
49 TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3-6,8-10,24,35-37"));
50 TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3-6,8-10,24,35-37"));
51 TEST_ASSERT_VAL("failed to convert map", test_bitmap("1-10,12-20,22-30,32-40"));
52 return 0;
53 }
54