mem2node.c 1.9 KB

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