57 lines · c
1// SPDX-License-Identifier: GPL-2.02#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 10010 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);57