tools/perf/tests/bitmap.c

   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
  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                 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);