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