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
11static 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
29static 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
43static 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
55DEFINE_SUITE("Print bitmap", bitmap_print);
56

source code of linux/tools/perf/tests/bitmap.c