xref: /openbmc/linux/tools/perf/tests/bitmap.c (revision 9cf0666f)
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 < map->nr; i++)
21 			set_bit(map->map[i], 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