18c2ecf20Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0 28c2ecf20Sopenharmony_ci#include <linux/compiler.h> 38c2ecf20Sopenharmony_ci#include <linux/bitmap.h> 48c2ecf20Sopenharmony_ci#include <perf/cpumap.h> 58c2ecf20Sopenharmony_ci#include <internal/cpumap.h> 68c2ecf20Sopenharmony_ci#include "tests.h" 78c2ecf20Sopenharmony_ci#include "debug.h" 88c2ecf20Sopenharmony_ci 98c2ecf20Sopenharmony_ci#define NBITS 100 108c2ecf20Sopenharmony_ci 118c2ecf20Sopenharmony_cistatic unsigned long *get_bitmap(const char *str, int nbits) 128c2ecf20Sopenharmony_ci{ 138c2ecf20Sopenharmony_ci struct perf_cpu_map *map = perf_cpu_map__new(str); 148c2ecf20Sopenharmony_ci unsigned long *bm = NULL; 158c2ecf20Sopenharmony_ci int i; 168c2ecf20Sopenharmony_ci 178c2ecf20Sopenharmony_ci bm = bitmap_alloc(nbits); 188c2ecf20Sopenharmony_ci 198c2ecf20Sopenharmony_ci if (map && bm) { 208c2ecf20Sopenharmony_ci for (i = 0; i < map->nr; i++) 218c2ecf20Sopenharmony_ci set_bit(map->map[i], bm); 228c2ecf20Sopenharmony_ci } 238c2ecf20Sopenharmony_ci 248c2ecf20Sopenharmony_ci if (map) 258c2ecf20Sopenharmony_ci perf_cpu_map__put(map); 268c2ecf20Sopenharmony_ci return bm; 278c2ecf20Sopenharmony_ci} 288c2ecf20Sopenharmony_ci 298c2ecf20Sopenharmony_cistatic int test_bitmap(const char *str) 308c2ecf20Sopenharmony_ci{ 318c2ecf20Sopenharmony_ci unsigned long *bm = get_bitmap(str, NBITS); 328c2ecf20Sopenharmony_ci char buf[100]; 338c2ecf20Sopenharmony_ci int ret; 348c2ecf20Sopenharmony_ci 358c2ecf20Sopenharmony_ci bitmap_scnprintf(bm, NBITS, buf, sizeof(buf)); 368c2ecf20Sopenharmony_ci pr_debug("bitmap: %s\n", buf); 378c2ecf20Sopenharmony_ci 388c2ecf20Sopenharmony_ci ret = !strcmp(buf, str); 398c2ecf20Sopenharmony_ci free(bm); 408c2ecf20Sopenharmony_ci return ret; 418c2ecf20Sopenharmony_ci} 428c2ecf20Sopenharmony_ci 438c2ecf20Sopenharmony_ciint test__bitmap_print(struct test *test __maybe_unused, int subtest __maybe_unused) 448c2ecf20Sopenharmony_ci{ 458c2ecf20Sopenharmony_ci TEST_ASSERT_VAL("failed to convert map", test_bitmap("1")); 468c2ecf20Sopenharmony_ci TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,5")); 478c2ecf20Sopenharmony_ci TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3,5,7,9,11,13,15,17,19,21-40")); 488c2ecf20Sopenharmony_ci TEST_ASSERT_VAL("failed to convert map", test_bitmap("2-5")); 498c2ecf20Sopenharmony_ci TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3-6,8-10,24,35-37")); 508c2ecf20Sopenharmony_ci TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3-6,8-10,24,35-37")); 518c2ecf20Sopenharmony_ci TEST_ASSERT_VAL("failed to convert map", test_bitmap("1-10,12-20,22-30,32-40")); 528c2ecf20Sopenharmony_ci return 0; 538c2ecf20Sopenharmony_ci} 54