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 | |