1 // SPDX-License-Identifier: GPL-2.0 2 #include "tests.h" 3 #include <stdio.h> 4 #include "cpumap.h" 5 #include "event.h" 6 #include "util/synthetic-events.h" 7 #include <string.h> 8 #include <linux/bitops.h> 9 #include <internal/cpumap.h> 10 #include "debug.h" 11 12 struct machine; 13 14 static int process_event_mask(const struct perf_tool *tool __maybe_unused, 15 union perf_event *event, 16 struct perf_sample *sample __maybe_unused, 17 struct machine *machine __maybe_unused) 18 { 19 struct perf_record_cpu_map *map_event = &event->cpu_map; 20 struct perf_record_cpu_map_data *data; 21 struct perf_cpu_map *map; 22 unsigned int long_size; 23 24 data = &map_event->data; 25 26 TEST_ASSERT_VAL("wrong type", data->type == PERF_CPU_MAP__MASK); 27 28 long_size = data->mask32_data.long_size; 29 30 TEST_ASSERT_VAL("wrong long_size", long_size == 4 || long_size == 8); 31 32 TEST_ASSERT_VAL("wrong nr", data->mask32_data.nr == 1); 33 34 TEST_ASSERT_VAL("wrong cpu", perf_record_cpu_map_data__test_bit(0, data)); 35 TEST_ASSERT_VAL("wrong cpu", !perf_record_cpu_map_data__test_bit(1, data)); 36 for (int i = 2; i <= 20; i++) 37 TEST_ASSERT_VAL("wrong cpu", perf_record_cpu_map_data__test_bit(i, data)); 38 39 map = cpu_map__new_data(data); 40 TEST_ASSERT_VAL("wrong nr", perf_cpu_map__nr(map) == 20); 41 42 TEST_ASSERT_VAL("wrong cpu", perf_cpu_map__cpu(map, 0).cpu == 0); 43 for (int i = 2; i <= 20; i++) 44 TEST_ASSERT_VAL("wrong cpu", perf_cpu_map__cpu(map, i - 1).cpu == i); 45 46 perf_cpu_map__put(map); 47 return 0; 48 } 49 50 static int process_event_cpus(const struct perf_tool *tool __maybe_unused, 51 union perf_event *event, 52 struct perf_sample *sample __maybe_unused, 53 struct machine *machine __maybe_unused) 54 { 55 struct perf_record_cpu_map *map_event = &event->cpu_map; 56 struct perf_record_cpu_map_data *data; 57 struct perf_cpu_map *map; 58 59 data = &map_event->data; 60 61 TEST_ASSERT_VAL("wrong type", data->type == PERF_CPU_MAP__CPUS); 62 63 TEST_ASSERT_VAL("wrong nr", data->cpus_data.nr == 2); 64 TEST_ASSERT_VAL("wrong cpu", data->cpus_data.cpu[0] == 1); 65 TEST_ASSERT_VAL("wrong cpu", data->cpus_data.cpu[1] == 256); 66 67 map = cpu_map__new_data(data); 68 TEST_ASSERT_VAL("wrong nr", perf_cpu_map__nr(map) == 2); 69 TEST_ASSERT_VAL("wrong cpu", perf_cpu_map__cpu(map, 0).cpu == 1); 70 TEST_ASSERT_VAL("wrong cpu", perf_cpu_map__cpu(map, 1).cpu == 256); 71 TEST_ASSERT_VAL("wrong refcnt", refcount_read(perf_cpu_map__refcnt(map)) == 1); 72 perf_cpu_map__put(map); 73 return 0; 74 } 75 76 static int process_event_range_cpus(const struct perf_tool *tool __maybe_unused, 77 union perf_event *event, 78 struct perf_sample *sample __maybe_unused, 79 struct machine *machine __maybe_unused) 80 { 81 struct perf_record_cpu_map *map_event = &event->cpu_map; 82 struct perf_record_cpu_map_data *data; 83 struct perf_cpu_map *map; 84 85 data = &map_event->data; 86 87 TEST_ASSERT_VAL("wrong type", data->type == PERF_CPU_MAP__RANGE_CPUS); 88 89 TEST_ASSERT_VAL("wrong any_cpu", data->range_cpu_data.any_cpu == 0); 90 TEST_ASSERT_VAL("wrong start_cpu", data->range_cpu_data.start_cpu == 1); 91 TEST_ASSERT_VAL("wrong end_cpu", data->range_cpu_data.end_cpu == 256); 92 93 map = cpu_map__new_data(data); 94 TEST_ASSERT_VAL("wrong nr", perf_cpu_map__nr(map) == 256); 95 TEST_ASSERT_VAL("wrong cpu", perf_cpu_map__cpu(map, 0).cpu == 1); 96 TEST_ASSERT_VAL("wrong cpu", perf_cpu_map__max(map).cpu == 256); 97 TEST_ASSERT_VAL("wrong refcnt", refcount_read(perf_cpu_map__refcnt(map)) == 1); 98 perf_cpu_map__put(map); 99 return 0; 100 } 101 102 103 static int test__cpu_map_synthesize(struct test_suite *test __maybe_unused, int subtest __maybe_unused) 104 { 105 struct perf_cpu_map *cpus; 106 107 /* This one is better stored in a mask. */ 108 cpus = perf_cpu_map__new("0,2-20"); 109 110 TEST_ASSERT_VAL("failed to synthesize map", 111 !perf_event__synthesize_cpu_map(NULL, cpus, process_event_mask, NULL)); 112 113 perf_cpu_map__put(cpus); 114 115 /* This one is better stored in cpu values. */ 116 cpus = perf_cpu_map__new("1,256"); 117 118 TEST_ASSERT_VAL("failed to synthesize map", 119 !perf_event__synthesize_cpu_map(NULL, cpus, process_event_cpus, NULL)); 120 121 perf_cpu_map__put(cpus); 122 123 /* This one is better stored as a range. */ 124 cpus = perf_cpu_map__new("1-256"); 125 126 TEST_ASSERT_VAL("failed to synthesize map", 127 !perf_event__synthesize_cpu_map(NULL, cpus, process_event_range_cpus, NULL)); 128 129 perf_cpu_map__put(cpus); 130 return 0; 131 } 132 133 static int cpu_map_print(const char *str) 134 { 135 struct perf_cpu_map *map = perf_cpu_map__new(str); 136 char buf[100]; 137 138 if (!map) 139 return -1; 140 141 cpu_map__snprint(map, buf, sizeof(buf)); 142 perf_cpu_map__put(map); 143 144 return !strcmp(buf, str); 145 } 146 147 static int test__cpu_map_print(struct test_suite *test __maybe_unused, int subtest __maybe_unused) 148 { 149 TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1")); 150 TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1,5")); 151 TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1,3,5,7,9,11,13,15,17,19,21-40")); 152 TEST_ASSERT_VAL("failed to convert map", cpu_map_print("2-5")); 153 TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1,3-6,8-10,24,35-37")); 154 TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1,3-6,8-10,24,35-37")); 155 TEST_ASSERT_VAL("failed to convert map", cpu_map_print("1-10,12-20,22-30,32-40")); 156 return 0; 157 } 158 159 static int __test__cpu_map_merge(const char *lhs, const char *rhs, unsigned int nr, 160 const char *expected) 161 { 162 struct perf_cpu_map *a = perf_cpu_map__new(lhs); 163 struct perf_cpu_map *b = perf_cpu_map__new(rhs); 164 char buf[100]; 165 166 perf_cpu_map__merge(&a, b); 167 TEST_ASSERT_VAL("failed to merge map: bad nr", perf_cpu_map__nr(a) == nr); 168 cpu_map__snprint(a, buf, sizeof(buf)); 169 TEST_ASSERT_VAL("failed to merge map: bad result", !strcmp(buf, expected)); 170 perf_cpu_map__put(b); 171 172 /* 173 * If 'b' is a superset of 'a', 'a' points to the same map with the 174 * map 'b'. In this case, the owner 'b' has released the resource above 175 * but 'a' still keeps the ownership, the reference counter should be 1. 176 */ 177 TEST_ASSERT_VAL("unexpected refcnt: bad result", 178 refcount_read(perf_cpu_map__refcnt(a)) == 1); 179 180 perf_cpu_map__put(a); 181 return 0; 182 } 183 184 static int test__cpu_map_merge(struct test_suite *test __maybe_unused, 185 int subtest __maybe_unused) 186 { 187 int ret; 188 189 ret = __test__cpu_map_merge("4,2,1", "4,5,7", 5, "1-2,4-5,7"); 190 if (ret) 191 return ret; 192 ret = __test__cpu_map_merge("1-8", "6-9", 9, "1-9"); 193 if (ret) 194 return ret; 195 ret = __test__cpu_map_merge("1-8,12-20", "6-9,15", 18, "1-9,12-20"); 196 if (ret) 197 return ret; 198 ret = __test__cpu_map_merge("4,2,1", "1", 3, "1-2,4"); 199 if (ret) 200 return ret; 201 ret = __test__cpu_map_merge("1", "4,2,1", 3, "1-2,4"); 202 if (ret) 203 return ret; 204 ret = __test__cpu_map_merge("1", "1", 1, "1"); 205 return ret; 206 } 207 208 static int __test__cpu_map_intersect(const char *lhs, const char *rhs, unsigned int nr, 209 const char *expected) 210 { 211 struct perf_cpu_map *a = perf_cpu_map__new(lhs); 212 struct perf_cpu_map *b = perf_cpu_map__new(rhs); 213 struct perf_cpu_map *c = perf_cpu_map__intersect(a, b); 214 char buf[100]; 215 216 TEST_ASSERT_EQUAL("failed to intersect map: bad nr", perf_cpu_map__nr(c), nr); 217 cpu_map__snprint(c, buf, sizeof(buf)); 218 TEST_ASSERT_VAL("failed to intersect map: bad result", !strcmp(buf, expected)); 219 perf_cpu_map__put(a); 220 perf_cpu_map__put(b); 221 perf_cpu_map__put(c); 222 return 0; 223 } 224 225 static int test__cpu_map_intersect(struct test_suite *test __maybe_unused, 226 int subtest __maybe_unused) 227 { 228 int ret; 229 230 ret = __test__cpu_map_intersect("4,2,1", "4,5,7", 1, "4"); 231 if (ret) 232 return ret; 233 ret = __test__cpu_map_intersect("1-8", "6-9", 3, "6-8"); 234 if (ret) 235 return ret; 236 ret = __test__cpu_map_intersect("1-8,12-20", "6-9,15", 4, "6-8,15"); 237 if (ret) 238 return ret; 239 ret = __test__cpu_map_intersect("4,2,1", "1", 1, "1"); 240 if (ret) 241 return ret; 242 ret = __test__cpu_map_intersect("1", "4,2,1", 1, "1"); 243 if (ret) 244 return ret; 245 ret = __test__cpu_map_intersect("1", "1", 1, "1"); 246 return ret; 247 } 248 249 static int test__cpu_map_equal(struct test_suite *test __maybe_unused, int subtest __maybe_unused) 250 { 251 struct perf_cpu_map *any = perf_cpu_map__new_any_cpu(); 252 struct perf_cpu_map *one = perf_cpu_map__new("1"); 253 struct perf_cpu_map *two = perf_cpu_map__new("2"); 254 struct perf_cpu_map *empty = perf_cpu_map__intersect(one, two); 255 struct perf_cpu_map *pair = perf_cpu_map__new("1-2"); 256 struct perf_cpu_map *tmp; 257 struct perf_cpu_map **maps[] = {&empty, &any, &one, &two, &pair}; 258 259 for (size_t i = 0; i < ARRAY_SIZE(maps); i++) { 260 /* Maps equal themself. */ 261 TEST_ASSERT_VAL("equal", perf_cpu_map__equal(*maps[i], *maps[i])); 262 for (size_t j = 0; j < ARRAY_SIZE(maps); j++) { 263 /* Maps dont't equal each other. */ 264 if (i == j) 265 continue; 266 TEST_ASSERT_VAL("not equal", !perf_cpu_map__equal(*maps[i], *maps[j])); 267 } 268 } 269 270 /* Maps equal made maps. */ 271 perf_cpu_map__merge(&two, one); 272 TEST_ASSERT_VAL("pair", perf_cpu_map__equal(pair, two)); 273 274 tmp = perf_cpu_map__intersect(pair, one); 275 TEST_ASSERT_VAL("one", perf_cpu_map__equal(one, tmp)); 276 perf_cpu_map__put(tmp); 277 278 for (size_t i = 0; i < ARRAY_SIZE(maps); i++) 279 perf_cpu_map__put(*maps[i]); 280 281 return TEST_OK; 282 } 283 284 static struct test_case tests__cpu_map[] = { 285 TEST_CASE("Synthesize cpu map", cpu_map_synthesize), 286 TEST_CASE("Print cpu map", cpu_map_print), 287 TEST_CASE("Merge cpu map", cpu_map_merge), 288 TEST_CASE("Intersect cpu map", cpu_map_intersect), 289 TEST_CASE("Equal cpu map", cpu_map_equal), 290 { .name = NULL, } 291 }; 292 293 struct test_suite suite__cpu_map = { 294 .desc = "CPU map", 295 .test_cases = tests__cpu_map, 296 }; 297