LXR linux/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
  10static 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                bitmap_zero(bm, nbits);
  20
  21                for (i = 0; i < map->nr; i++)
  22                        set_bit(map->map[i], bm);
  23        }
  24
  25        if (map)
  26                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
  44int test__bitmap_print(struct test *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