tools/perf/tests/maps.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 #include <linux/compiler.h>
   3 #include <linux/kernel.h>
   4 #include "tests.h"
   5 #include "map.h"
   6 #include "maps.h"
   7 #include "dso.h"
   8 #include "debug.h"
   9
  10 struct map_def {
  11         const char *name;
  12         u64 start;
  13         u64 end;
  14 };
  15
  16 static int check_maps(struct map_def *merged, unsigned int size, struct maps *maps)
  17 {
  18         struct map *map;
  19         unsigned int i = 0;
  20
  21         maps__for_each_entry(maps, map) {
  22                 if (i > 0)
  23                         TEST_ASSERT_VAL("less maps expected", (map && i < size) || (!map && i == size));
  24
  25                 TEST_ASSERT_VAL("wrong map start",  map->start == merged[i].start);
  26                 TEST_ASSERT_VAL("wrong map end",    map->end == merged[i].end);
  27                 TEST_ASSERT_VAL("wrong map name",  !strcmp(map->dso->name, merged[i].name));
  28                 TEST_ASSERT_VAL("wrong map refcnt", refcount_read(&map->refcnt) == 1);
  29
  30                 i++;
  31         }
  32
  33         return TEST_OK;
  34 }
  35
  36 int test__maps__merge_in(struct test *t __maybe_unused, int subtest __maybe_unused)
  37 {
  38         struct maps maps;
  39         unsigned int i;
  40         struct map_def bpf_progs[] = {
  41                 { "bpf_prog_1", 200, 300 },
  42                 { "bpf_prog_2", 500, 600 },
  43                 { "bpf_prog_3", 800, 900 },
  44         };
  45         struct map_def merged12[] = {
  46                 { "kcore1",     100,  200 },
  47                 { "bpf_prog_1", 200,  300 },
  48                 { "kcore1",     300,  500 },
  49                 { "bpf_prog_2", 500,  600 },
  50                 { "kcore1",     600,  800 },
  51                 { "bpf_prog_3", 800,  900 },
  52                 { "kcore1",     900, 1000 },
  53         };
  54         struct map_def merged3[] = {
  55                 { "kcore1",      100,  200 },
  56                 { "bpf_prog_1",  200,  300 },
  57                 { "kcore1",      300,  500 },
  58                 { "bpf_prog_2",  500,  600 },
  59                 { "kcore1",      600,  800 },
  60                 { "bpf_prog_3",  800,  900 },
  61                 { "kcore1",      900, 1000 },
  62                 { "kcore3",     1000, 1100 },
  63         };
  64         struct map *map_kcore1, *map_kcore2, *map_kcore3;
  65         int ret;
  66
  67         maps__init(&maps, NULL);
  68
  69         for (i = 0; i < ARRAY_SIZE(bpf_progs); i++) {
  70                 struct map *map;
  71
  72                 map = dso__new_map(bpf_progs[i].name);
  73                 TEST_ASSERT_VAL("failed to create map", map);
  74
  75                 map->start = bpf_progs[i].start;
  76                 map->end   = bpf_progs[i].end;
  77                 maps__insert(&maps, map);
  78                 map__put(map);
  79         }
  80
  81         map_kcore1 = dso__new_map("kcore1");
  82         TEST_ASSERT_VAL("failed to create map", map_kcore1);
  83
  84         map_kcore2 = dso__new_map("kcore2");
  85         TEST_ASSERT_VAL("failed to create map", map_kcore2);
  86
  87         map_kcore3 = dso__new_map("kcore3");
  88         TEST_ASSERT_VAL("failed to create map", map_kcore3);
  89
  90         /* kcore1 map overlaps over all bpf maps */
  91         map_kcore1->start = 100;
  92         map_kcore1->end   = 1000;
  93
  94         /* kcore2 map hides behind bpf_prog_2 */
  95         map_kcore2->start = 550;
  96         map_kcore2->end   = 570;
  97
  98         /* kcore3 map hides behind bpf_prog_3, kcore1 and adds new map */
  99         map_kcore3->start = 880;
 100         map_kcore3->end   = 1100;
 101
 102         ret = maps__merge_in(&maps, map_kcore1);
 103         TEST_ASSERT_VAL("failed to merge map", !ret);
 104
 105         ret = check_maps(merged12, ARRAY_SIZE(merged12), &maps);
 106         TEST_ASSERT_VAL("merge check failed", !ret);
 107
 108         ret = maps__merge_in(&maps, map_kcore2);
 109         TEST_ASSERT_VAL("failed to merge map", !ret);
 110
 111         ret = check_maps(merged12, ARRAY_SIZE(merged12), &maps);
 112         TEST_ASSERT_VAL("merge check failed", !ret);
 113
 114         ret = maps__merge_in(&maps, map_kcore3);
 115         TEST_ASSERT_VAL("failed to merge map", !ret);
 116
 117         ret = check_maps(merged3, ARRAY_SIZE(merged3), &maps);
 118         TEST_ASSERT_VAL("merge check failed", !ret);
 119         return TEST_OK;
 120 }