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