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;
15
16	bm = bitmap_zalloc(nbits);
17
18	if (map && bm) {
19		int i;
20		struct perf_cpu cpu;
21
22		perf_cpu_map__for_each_cpu(cpu, i, map)
23			__set_bit(cpu.cpu, bm);
24	}
25
26	perf_cpu_map__put(map);
27	return bm;
28}
29
30static int test_bitmap(const char *str)
31{
32	unsigned long *bm = get_bitmap(str, NBITS);
33	char buf[100];
34	int ret;
35
36	bitmap_scnprintf(bm, NBITS, buf, sizeof(buf));
37	pr_debug("bitmap: %s\n", buf);
38
39	ret = !strcmp(buf, str);
40	free(bm);
41	return ret;
42}
43
44static int test__bitmap_print(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
45{
46	TEST_ASSERT_VAL("failed to convert map", test_bitmap("1"));
47	TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,5"));
48	TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3,5,7,9,11,13,15,17,19,21-40"));
49	TEST_ASSERT_VAL("failed to convert map", test_bitmap("2-5"));
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,3-6,8-10,24,35-37"));
52	TEST_ASSERT_VAL("failed to convert map", test_bitmap("1-10,12-20,22-30,32-40"));
53	return 0;
54}
55
56DEFINE_SUITE("Print bitmap", bitmap_print);