1 // SPDX-License-Identifier: GPL-2.0 2 #include <linux/compiler.h> 3 #include <linux/bitmap.h> 4 #include "cpumap.h" 5 #include "mem2node.h" 6 #include "tests.h" 7 8 static struct node { 9 int node; 10 const char *map; 11 } test_nodes[] = { 12 { .node = 0, .map = "0" }, 13 { .node = 1, .map = "1-2" }, 14 { .node = 3, .map = "5-7,9" }, 15 }; 16 17 #define T TEST_ASSERT_VAL 18 19 static unsigned long *get_bitmap(const char *str, int nbits) 20 { 21 struct cpu_map *map = cpu_map__new(str); 22 unsigned long *bm = NULL; 23 int i; 24 25 bm = bitmap_alloc(nbits); 26 27 if (map && bm) { 28 for (i = 0; i < map->nr; i++) { 29 set_bit(map->map[i], bm); 30 } 31 } 32 33 if (map) 34 cpu_map__put(map); 35 else 36 free(bm); 37 38 return bm && map ? bm : NULL; 39 } 40 41 int test__mem2node(struct test *t __maybe_unused, int subtest __maybe_unused) 42 { 43 struct mem2node map; 44 struct memory_node nodes[3]; 45 struct perf_env env = { 46 .memory_nodes = (struct memory_node *) &nodes[0], 47 .nr_memory_nodes = ARRAY_SIZE(nodes), 48 .memory_bsize = 0x100, 49 }; 50 unsigned int i; 51 52 for (i = 0; i < ARRAY_SIZE(nodes); i++) { 53 nodes[i].node = test_nodes[i].node; 54 nodes[i].size = 10; 55 56 T("failed: alloc bitmap", 57 (nodes[i].set = get_bitmap(test_nodes[i].map, 10))); 58 } 59 60 T("failed: mem2node__init", !mem2node__init(&map, &env)); 61 T("failed: mem2node__node", 0 == mem2node__node(&map, 0x50)); 62 T("failed: mem2node__node", 1 == mem2node__node(&map, 0x100)); 63 T("failed: mem2node__node", 1 == mem2node__node(&map, 0x250)); 64 T("failed: mem2node__node", 3 == mem2node__node(&map, 0x500)); 65 T("failed: mem2node__node", 3 == mem2node__node(&map, 0x650)); 66 T("failed: mem2node__node", -1 == mem2node__node(&map, 0x450)); 67 T("failed: mem2node__node", -1 == mem2node__node(&map, 0x1050)); 68 69 for (i = 0; i < ARRAY_SIZE(nodes); i++) 70 free(nodes[i].set); 71 72 mem2node__exit(&map); 73 return 0; 74 } 75