| 1 | |
| 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;
|
| 15 |
|
| 16 | bm = bitmap_zalloc(nbits);
|
| 17 |
|
| 18 | if (map && bm) {
|
| 19 | unsigned 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 |
|
| 30 | static 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 |
|
| 44 | static 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 |
|
| 56 | DEFINE_SUITE("Print bitmap", bitmap_print);
|