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