tools/perf/tests/bitmap.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 #include <linux/compiler.h>
   3 #include <linux/bitmap.h>
   4 #include "tests.h"
   5 #include "cpumap.h"
   6 #include "debug.h"
   7
   8 #define NBITS 100
   9
  10 static unsigned long *get_bitmap(const char *str, int nbits)
  11 {
  12         struct cpu_map *map = cpu_map__new(str);
  13         unsigned long *bm = NULL;
  14         int i;
  15
  16         bm = bitmap_alloc(nbits);
  17
  18         if (map && bm) {
  19                 for (i = 0; i < map->nr; i++)
  20                         set_bit(map->map[i], bm);
  21         }
  22
  23         if (map)
  24                 cpu_map__put(map);
  25         return bm;
  26 }
  27
  28 static int test_bitmap(const char *str)
  29 {
  30         unsigned long *bm = get_bitmap(str, NBITS);
  31         char buf[100];
  32         int ret;
  33
  34         bitmap_scnprintf(bm, NBITS, buf, sizeof(buf));
  35         pr_debug("bitmap: %s\n", buf);
  36
  37         ret = !strcmp(buf, str);
  38         free(bm);
  39         return ret;
  40 }
  41
  42 int test__bitmap_print(struct test *test __maybe_unused, int subtest __maybe_unused)
  43 {
  44         TEST_ASSERT_VAL("failed to convert map", test_bitmap("1"));
  45         TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,5"));
  46         TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3,5,7,9,11,13,15,17,19,21-40"));
  47         TEST_ASSERT_VAL("failed to convert map", test_bitmap("2-5"));
  48         TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3-6,8-10,24,35-37"));
  49         TEST_ASSERT_VAL("failed to convert map", test_bitmap("1,3-6,8-10,24,35-37"));
  50         TEST_ASSERT_VAL("failed to convert map", test_bitmap("1-10,12-20,22-30,32-40"));
  51         return 0;
  52 }