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
 12struct machine;
 13
 14static int process_event_mask(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
 50static int process_event_cpus(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(&map->refcnt) == 1);
 72	perf_cpu_map__put(map);
 73	return 0;
 74}
 75
 76static int process_event_range_cpus(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(&map->refcnt) == 1);
 98	perf_cpu_map__put(map);
 99	return 0;
100}
101
102
103static 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
133static 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
147static 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
159static int test__cpu_map_merge(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
160{
161	struct perf_cpu_map *a = perf_cpu_map__new("4,2,1");
162	struct perf_cpu_map *b = perf_cpu_map__new("4,5,7");
163	struct perf_cpu_map *c = perf_cpu_map__merge(a, b);
164	char buf[100];
165
166	TEST_ASSERT_VAL("failed to merge map: bad nr", perf_cpu_map__nr(c) == 5);
167	cpu_map__snprint(c, buf, sizeof(buf));
168	TEST_ASSERT_VAL("failed to merge map: bad result", !strcmp(buf, "1-2,4-5,7"));
169	perf_cpu_map__put(b);
170	perf_cpu_map__put(c);
171	return 0;
172}
173
174DEFINE_SUITE("Synthesize cpu map", cpu_map_synthesize);
175DEFINE_SUITE("Print cpu map", cpu_map_print);
176DEFINE_SUITE("Merge cpu map", cpu_map_merge);