1 // SPDX-License-Identifier: GPL-2.0
2 #include <inttypes.h>
3 #include <stdio.h>
4 #include <stdlib.h>
5 #include <string.h>
6 #include <errno.h>
7 #include <linux/zalloc.h>
8 
9 #include "values.h"
10 #include "debug.h"
11 
perf_read_values_init(struct perf_read_values * values)12 int perf_read_values_init(struct perf_read_values *values)
13 {
14 	values->threads_max = 16;
15 	values->pid = malloc(values->threads_max * sizeof(*values->pid));
16 	values->tid = malloc(values->threads_max * sizeof(*values->tid));
17 	values->value = zalloc(values->threads_max * sizeof(*values->value));
18 	if (!values->pid || !values->tid || !values->value) {
19 		pr_debug("failed to allocate read_values threads arrays");
20 		goto out_free_pid;
21 	}
22 	values->threads = 0;
23 
24 	values->counters_max = 16;
25 	values->counterrawid = malloc(values->counters_max
26 				      * sizeof(*values->counterrawid));
27 	values->countername = malloc(values->counters_max
28 				     * sizeof(*values->countername));
29 	if (!values->counterrawid || !values->countername) {
30 		pr_debug("failed to allocate read_values counters arrays");
31 		goto out_free_counter;
32 	}
33 	values->counters = 0;
34 
35 	return 0;
36 
37 out_free_counter:
38 	zfree(&values->counterrawid);
39 	zfree(&values->countername);
40 out_free_pid:
41 	zfree(&values->pid);
42 	zfree(&values->tid);
43 	zfree(&values->value);
44 	return -ENOMEM;
45 }
46 
perf_read_values_destroy(struct perf_read_values * values)47 void perf_read_values_destroy(struct perf_read_values *values)
48 {
49 	int i;
50 
51 	if (!values->threads_max || !values->counters_max)
52 		return;
53 
54 	for (i = 0; i < values->threads; i++)
55 		zfree(&values->value[i]);
56 	zfree(&values->value);
57 	zfree(&values->pid);
58 	zfree(&values->tid);
59 	zfree(&values->counterrawid);
60 	for (i = 0; i < values->counters; i++)
61 		zfree(&values->countername[i]);
62 	zfree(&values->countername);
63 }
64 
perf_read_values__enlarge_threads(struct perf_read_values * values)65 static int perf_read_values__enlarge_threads(struct perf_read_values *values)
66 {
67 	int nthreads_max = values->threads_max * 2;
68 	void *npid = realloc(values->pid, nthreads_max * sizeof(*values->pid)),
69 	     *ntid = realloc(values->tid, nthreads_max * sizeof(*values->tid)),
70 	     *nvalue = realloc(values->value, nthreads_max * sizeof(*values->value));
71 
72 	if (!npid || !ntid || !nvalue)
73 		goto out_err;
74 
75 	values->threads_max = nthreads_max;
76 	values->pid = npid;
77 	values->tid = ntid;
78 	values->value = nvalue;
79 	return 0;
80 out_err:
81 	free(npid);
82 	free(ntid);
83 	free(nvalue);
84 	pr_debug("failed to enlarge read_values threads arrays");
85 	return -ENOMEM;
86 }
87 
perf_read_values__findnew_thread(struct perf_read_values * values,u32 pid,u32 tid)88 static int perf_read_values__findnew_thread(struct perf_read_values *values,
89 					    u32 pid, u32 tid)
90 {
91 	int i;
92 
93 	for (i = 0; i < values->threads; i++)
94 		if (values->pid[i] == pid && values->tid[i] == tid)
95 			return i;
96 
97 	if (values->threads == values->threads_max) {
98 		i = perf_read_values__enlarge_threads(values);
99 		if (i < 0)
100 			return i;
101 	}
102 
103 	i = values->threads;
104 
105 	values->value[i] = zalloc(values->counters_max * sizeof(**values->value));
106 	if (!values->value[i]) {
107 		pr_debug("failed to allocate read_values counters array");
108 		return -ENOMEM;
109 	}
110 	values->pid[i] = pid;
111 	values->tid[i] = tid;
112 	values->threads = i + 1;
113 
114 	return i;
115 }
116 
perf_read_values__enlarge_counters(struct perf_read_values * values)117 static int perf_read_values__enlarge_counters(struct perf_read_values *values)
118 {
119 	char **countername;
120 	int i, counters_max = values->counters_max * 2;
121 	u64 *counterrawid = realloc(values->counterrawid, counters_max * sizeof(*values->counterrawid));
122 
123 	if (!counterrawid) {
124 		pr_debug("failed to enlarge read_values rawid array");
125 		goto out_enomem;
126 	}
127 
128 	countername = realloc(values->countername, counters_max * sizeof(*values->countername));
129 	if (!countername) {
130 		pr_debug("failed to enlarge read_values rawid array");
131 		goto out_free_rawid;
132 	}
133 
134 	for (i = 0; i < values->threads; i++) {
135 		u64 *value = realloc(values->value[i], counters_max * sizeof(**values->value));
136 		int j;
137 
138 		if (!value) {
139 			pr_debug("failed to enlarge read_values ->values array");
140 			goto out_free_name;
141 		}
142 
143 		for (j = values->counters_max; j < counters_max; j++)
144 			value[j] = 0;
145 
146 		values->value[i] = value;
147 	}
148 
149 	values->counters_max = counters_max;
150 	values->counterrawid = counterrawid;
151 	values->countername  = countername;
152 
153 	return 0;
154 out_free_name:
155 	free(countername);
156 out_free_rawid:
157 	free(counterrawid);
158 out_enomem:
159 	return -ENOMEM;
160 }
161 
perf_read_values__findnew_counter(struct perf_read_values * values,u64 rawid,const char * name)162 static int perf_read_values__findnew_counter(struct perf_read_values *values,
163 					     u64 rawid, const char *name)
164 {
165 	int i;
166 
167 	for (i = 0; i < values->counters; i++)
168 		if (values->counterrawid[i] == rawid)
169 			return i;
170 
171 	if (values->counters == values->counters_max) {
172 		i = perf_read_values__enlarge_counters(values);
173 		if (i)
174 			return i;
175 	}
176 
177 	i = values->counters++;
178 	values->counterrawid[i] = rawid;
179 	values->countername[i] = strdup(name);
180 
181 	return i;
182 }
183 
perf_read_values_add_value(struct perf_read_values * values,u32 pid,u32 tid,u64 rawid,const char * name,u64 value)184 int perf_read_values_add_value(struct perf_read_values *values,
185 				u32 pid, u32 tid,
186 				u64 rawid, const char *name, u64 value)
187 {
188 	int tindex, cindex;
189 
190 	tindex = perf_read_values__findnew_thread(values, pid, tid);
191 	if (tindex < 0)
192 		return tindex;
193 	cindex = perf_read_values__findnew_counter(values, rawid, name);
194 	if (cindex < 0)
195 		return cindex;
196 
197 	values->value[tindex][cindex] += value;
198 	return 0;
199 }
200 
perf_read_values__display_pretty(FILE * fp,struct perf_read_values * values)201 static void perf_read_values__display_pretty(FILE *fp,
202 					     struct perf_read_values *values)
203 {
204 	int i, j;
205 	int pidwidth, tidwidth;
206 	int *counterwidth;
207 
208 	counterwidth = malloc(values->counters * sizeof(*counterwidth));
209 	if (!counterwidth) {
210 		fprintf(fp, "INTERNAL ERROR: Failed to allocate counterwidth array\n");
211 		return;
212 	}
213 	tidwidth = 3;
214 	pidwidth = 3;
215 	for (j = 0; j < values->counters; j++)
216 		counterwidth[j] = strlen(values->countername[j]);
217 	for (i = 0; i < values->threads; i++) {
218 		int width;
219 
220 		width = snprintf(NULL, 0, "%d", values->pid[i]);
221 		if (width > pidwidth)
222 			pidwidth = width;
223 		width = snprintf(NULL, 0, "%d", values->tid[i]);
224 		if (width > tidwidth)
225 			tidwidth = width;
226 		for (j = 0; j < values->counters; j++) {
227 			width = snprintf(NULL, 0, "%" PRIu64, values->value[i][j]);
228 			if (width > counterwidth[j])
229 				counterwidth[j] = width;
230 		}
231 	}
232 
233 	fprintf(fp, "# %*s  %*s", pidwidth, "PID", tidwidth, "TID");
234 	for (j = 0; j < values->counters; j++)
235 		fprintf(fp, "  %*s", counterwidth[j], values->countername[j]);
236 	fprintf(fp, "\n");
237 
238 	for (i = 0; i < values->threads; i++) {
239 		fprintf(fp, "  %*d  %*d", pidwidth, values->pid[i],
240 			tidwidth, values->tid[i]);
241 		for (j = 0; j < values->counters; j++)
242 			fprintf(fp, "  %*" PRIu64,
243 				counterwidth[j], values->value[i][j]);
244 		fprintf(fp, "\n");
245 	}
246 	free(counterwidth);
247 }
248 
perf_read_values__display_raw(FILE * fp,struct perf_read_values * values)249 static void perf_read_values__display_raw(FILE *fp,
250 					  struct perf_read_values *values)
251 {
252 	int width, pidwidth, tidwidth, namewidth, rawwidth, countwidth;
253 	int i, j;
254 
255 	tidwidth = 3; /* TID */
256 	pidwidth = 3; /* PID */
257 	namewidth = 4; /* "Name" */
258 	rawwidth = 3; /* "Raw" */
259 	countwidth = 5; /* "Count" */
260 
261 	for (i = 0; i < values->threads; i++) {
262 		width = snprintf(NULL, 0, "%d", values->pid[i]);
263 		if (width > pidwidth)
264 			pidwidth = width;
265 		width = snprintf(NULL, 0, "%d", values->tid[i]);
266 		if (width > tidwidth)
267 			tidwidth = width;
268 	}
269 	for (j = 0; j < values->counters; j++) {
270 		width = strlen(values->countername[j]);
271 		if (width > namewidth)
272 			namewidth = width;
273 		width = snprintf(NULL, 0, "%" PRIx64, values->counterrawid[j]);
274 		if (width > rawwidth)
275 			rawwidth = width;
276 	}
277 	for (i = 0; i < values->threads; i++) {
278 		for (j = 0; j < values->counters; j++) {
279 			width = snprintf(NULL, 0, "%" PRIu64, values->value[i][j]);
280 			if (width > countwidth)
281 				countwidth = width;
282 		}
283 	}
284 
285 	fprintf(fp, "# %*s  %*s  %*s  %*s  %*s\n",
286 		pidwidth, "PID", tidwidth, "TID",
287 		namewidth, "Name", rawwidth, "Raw",
288 		countwidth, "Count");
289 	for (i = 0; i < values->threads; i++)
290 		for (j = 0; j < values->counters; j++)
291 			fprintf(fp, "  %*d  %*d  %*s  %*" PRIx64 "  %*" PRIu64,
292 				pidwidth, values->pid[i],
293 				tidwidth, values->tid[i],
294 				namewidth, values->countername[j],
295 				rawwidth, values->counterrawid[j],
296 				countwidth, values->value[i][j]);
297 }
298 
perf_read_values_display(FILE * fp,struct perf_read_values * values,int raw)299 void perf_read_values_display(FILE *fp, struct perf_read_values *values, int raw)
300 {
301 	if (raw)
302 		perf_read_values__display_raw(fp, values);
303 	else
304 		perf_read_values__display_pretty(fp, values);
305 }
306