1 // SPDX-License-Identifier: GPL-2.0 2 #include <linux/compiler.h> 3 #include <linux/bitmap.h> 4 #include <perf/cpumap.h> 5 #include <internal/cpumap.h> 6 #include "tests.h" 7 #include "debug.h" 8 9 #define NBITS 100 10 11 static unsigned long *get_bitmap(const char *str, int nbits) 12 { 13 struct perf_cpu_map *map = perf_cpu_map__new(str); 14 unsigned long *bm = NULL; 15 int i; 16 17 bm = bitmap_zalloc(nbits); 18 19 if (map && bm) { 20 for (i = 0; i < perf_cpu_map__nr(map); i++) 21 set_bit(perf_cpu_map__cpu(map, i).cpu, bm); 22 } 23 24 if (map) 25 perf_cpu_map__put(map); 26 return bm; 27 } 28 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 43 static int test__bitmap_print(struct test_suite *test __maybe_unused, 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 55 DEFINE_SUITE("Print bitmap", bitmap_print); 56