Linux Audio

Check our new training course

Loading...
v5.4
  1// SPDX-License-Identifier: GPL-2.0-only
  2/*
  3 * Copyright (C) 2008,2009, Steven Rostedt <srostedt@redhat.com>
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  4 */
 
  5#include <dirent.h>
  6#include <mntent.h>
  7#include <stdio.h>
  8#include <stdlib.h>
  9#include <string.h>
 10#include <stdarg.h>
 11#include <sys/types.h>
 12#include <sys/stat.h>
 13#include <sys/wait.h>
 14#include <fcntl.h>
 15#include <unistd.h>
 16#include <errno.h>
 17#include <stdbool.h>
 18#include <linux/list.h>
 19#include <linux/kernel.h>
 20#include <linux/zalloc.h>
 21#include <internal/lib.h> // page_size
 22
 
 23#include "trace-event.h"
 24#include <api/fs/tracing_path.h>
 25#include "evsel.h"
 26#include "debug.h"
 27
 28#define VERSION "0.6"
 29
 30static int output_fd;
 31
 32
 33int bigendian(void)
 34{
 35	unsigned char str[] = { 0x1, 0x2, 0x3, 0x4, 0x0, 0x0, 0x0, 0x0};
 36	unsigned int *ptr;
 37
 38	ptr = (unsigned int *)(void *)str;
 39	return *ptr == 0x01020304;
 40}
 41
 42/* unfortunately, you can not stat debugfs or proc files for size */
 43static int record_file(const char *file, ssize_t hdr_sz)
 44{
 45	unsigned long long size = 0;
 46	char buf[BUFSIZ], *sizep;
 47	off_t hdr_pos = lseek(output_fd, 0, SEEK_CUR);
 48	int r, fd;
 49	int err = -EIO;
 50
 51	fd = open(file, O_RDONLY);
 52	if (fd < 0) {
 53		pr_debug("Can't read '%s'", file);
 54		return -errno;
 55	}
 56
 57	/* put in zeros for file size, then fill true size later */
 58	if (hdr_sz) {
 59		if (write(output_fd, &size, hdr_sz) != hdr_sz)
 60			goto out;
 61	}
 62
 63	do {
 64		r = read(fd, buf, BUFSIZ);
 65		if (r > 0) {
 66			size += r;
 67			if (write(output_fd, buf, r) != r)
 68				goto out;
 69		}
 70	} while (r > 0);
 71
 72	/* ugh, handle big-endian hdr_size == 4 */
 73	sizep = (char*)&size;
 74	if (bigendian())
 75		sizep += sizeof(u64) - hdr_sz;
 76
 77	if (hdr_sz && pwrite(output_fd, sizep, hdr_sz, hdr_pos) < 0) {
 78		pr_debug("writing file size failed\n");
 79		goto out;
 80	}
 81
 82	err = 0;
 83out:
 84	close(fd);
 85	return err;
 86}
 87
 88static int record_header_files(void)
 89{
 90	char *path = get_events_file("header_page");
 91	struct stat st;
 92	int err = -EIO;
 93
 
 94	if (!path) {
 95		pr_debug("can't get tracing/events/header_page");
 96		return -ENOMEM;
 97	}
 98
 99	if (stat(path, &st) < 0) {
100		pr_debug("can't read '%s'", path);
101		goto out;
102	}
103
104	if (write(output_fd, "header_page", 12) != 12) {
105		pr_debug("can't write header_page\n");
106		goto out;
107	}
108
109	if (record_file(path, 8) < 0) {
110		pr_debug("can't record header_page file\n");
111		goto out;
112	}
113
114	put_events_file(path);
115
116	path = get_events_file("header_event");
117	if (!path) {
118		pr_debug("can't get tracing/events/header_event");
119		err = -ENOMEM;
120		goto out;
121	}
122
123	if (stat(path, &st) < 0) {
124		pr_debug("can't read '%s'", path);
125		goto out;
126	}
127
128	if (write(output_fd, "header_event", 13) != 13) {
129		pr_debug("can't write header_event\n");
130		goto out;
131	}
132
133	if (record_file(path, 8) < 0) {
134		pr_debug("can't record header_event file\n");
135		goto out;
136	}
137
138	err = 0;
139out:
140	put_events_file(path);
141	return err;
142}
143
144static bool name_in_tp_list(char *sys, struct tracepoint_path *tps)
145{
146	while (tps) {
147		if (!strcmp(sys, tps->name))
148			return true;
149		tps = tps->next;
150	}
151
152	return false;
153}
154
155#define for_each_event(dir, dent, tps)				\
156	while ((dent = readdir(dir)))				\
157		if (dent->d_type == DT_DIR &&			\
158		    (strcmp(dent->d_name, ".")) &&		\
159		    (strcmp(dent->d_name, "..")))		\
160
161static int copy_event_system(const char *sys, struct tracepoint_path *tps)
162{
163	struct dirent *dent;
164	struct stat st;
165	char *format;
166	DIR *dir;
167	int count = 0;
168	int ret;
169	int err;
170
171	dir = opendir(sys);
172	if (!dir) {
173		pr_debug("can't read directory '%s'", sys);
174		return -errno;
175	}
176
177	for_each_event(dir, dent, tps) {
178		if (!name_in_tp_list(dent->d_name, tps))
179			continue;
180
181		if (asprintf(&format, "%s/%s/format", sys, dent->d_name) < 0) {
182			err = -ENOMEM;
183			goto out;
184		}
185		ret = stat(format, &st);
186		free(format);
187		if (ret < 0)
188			continue;
189		count++;
190	}
191
192	if (write(output_fd, &count, 4) != 4) {
193		err = -EIO;
194		pr_debug("can't write count\n");
195		goto out;
196	}
197
198	rewinddir(dir);
199	for_each_event(dir, dent, tps) {
200		if (!name_in_tp_list(dent->d_name, tps))
201			continue;
202
203		if (asprintf(&format, "%s/%s/format", sys, dent->d_name) < 0) {
204			err = -ENOMEM;
205			goto out;
206		}
207		ret = stat(format, &st);
208
209		if (ret >= 0) {
210			err = record_file(format, 8);
211			if (err) {
212				free(format);
213				goto out;
214			}
215		}
216		free(format);
217	}
218	err = 0;
219out:
220	closedir(dir);
221	return err;
222}
223
224static int record_ftrace_files(struct tracepoint_path *tps)
225{
226	char *path;
227	int ret;
228
229	path = get_events_file("ftrace");
230	if (!path) {
231		pr_debug("can't get tracing/events/ftrace");
232		return -ENOMEM;
233	}
234
235	ret = copy_event_system(path, tps);
236
237	put_tracing_file(path);
238
239	return ret;
240}
241
242static bool system_in_tp_list(char *sys, struct tracepoint_path *tps)
243{
244	while (tps) {
245		if (!strcmp(sys, tps->system))
246			return true;
247		tps = tps->next;
248	}
249
250	return false;
251}
252
253static int record_event_files(struct tracepoint_path *tps)
254{
255	struct dirent *dent;
256	struct stat st;
257	char *path;
258	char *sys;
259	DIR *dir;
260	int count = 0;
261	int ret;
262	int err;
263
264	path = get_tracing_file("events");
265	if (!path) {
266		pr_debug("can't get tracing/events");
267		return -ENOMEM;
268	}
269
270	dir = opendir(path);
271	if (!dir) {
272		err = -errno;
273		pr_debug("can't read directory '%s'", path);
274		goto out;
275	}
276
277	for_each_event(dir, dent, tps) {
278		if (strcmp(dent->d_name, "ftrace") == 0 ||
279		    !system_in_tp_list(dent->d_name, tps))
280			continue;
281
282		count++;
283	}
284
285	if (write(output_fd, &count, 4) != 4) {
286		err = -EIO;
287		pr_debug("can't write count\n");
288		goto out;
289	}
290
291	rewinddir(dir);
292	for_each_event(dir, dent, tps) {
293		if (strcmp(dent->d_name, "ftrace") == 0 ||
294		    !system_in_tp_list(dent->d_name, tps))
295			continue;
296
297		if (asprintf(&sys, "%s/%s", path, dent->d_name) < 0) {
298			err = -ENOMEM;
299			goto out;
300		}
301		ret = stat(sys, &st);
302		if (ret >= 0) {
303			ssize_t size = strlen(dent->d_name) + 1;
304
305			if (write(output_fd, dent->d_name, size) != size ||
306			    copy_event_system(sys, tps) < 0) {
307				err = -EIO;
308				free(sys);
309				goto out;
310			}
311		}
312		free(sys);
313	}
314	err = 0;
315out:
316	closedir(dir);
317	put_tracing_file(path);
318
319	return err;
320}
321
322static int record_proc_kallsyms(void)
323{
324	unsigned long long size = 0;
325	/*
326	 * Just to keep older perf.data file parsers happy, record a zero
327	 * sized kallsyms file, i.e. do the same thing that was done when
328	 * /proc/kallsyms (or something specified via --kallsyms, in a
329	 * different path) couldn't be read.
330	 */
331	return write(output_fd, &size, 4) != 4 ? -EIO : 0;
332}
333
334static int record_ftrace_printk(void)
335{
336	unsigned int size;
337	char *path;
338	struct stat st;
339	int ret, err = 0;
340
341	path = get_tracing_file("printk_formats");
342	if (!path) {
343		pr_debug("can't get tracing/printk_formats");
344		return -ENOMEM;
345	}
346
347	ret = stat(path, &st);
348	if (ret < 0) {
349		/* not found */
350		size = 0;
351		if (write(output_fd, &size, 4) != 4)
352			err = -EIO;
353		goto out;
354	}
355	err = record_file(path, 4);
356
357out:
358	put_tracing_file(path);
359	return err;
360}
361
362static int record_saved_cmdline(void)
363{
364	unsigned long long size;
365	char *path;
366	struct stat st;
367	int ret, err = 0;
368
369	path = get_tracing_file("saved_cmdlines");
370	if (!path) {
371		pr_debug("can't get tracing/saved_cmdline");
372		return -ENOMEM;
373	}
374
375	ret = stat(path, &st);
376	if (ret < 0) {
377		/* not found */
378		size = 0;
379		if (write(output_fd, &size, 8) != 8)
380			err = -EIO;
381		goto out;
382	}
383	err = record_file(path, 8);
384
385out:
386	put_tracing_file(path);
387	return err;
388}
389
390static void
391put_tracepoints_path(struct tracepoint_path *tps)
392{
393	while (tps) {
394		struct tracepoint_path *t = tps;
395
396		tps = tps->next;
397		zfree(&t->name);
398		zfree(&t->system);
399		free(t);
400	}
401}
402
403static struct tracepoint_path *
404get_tracepoints_path(struct list_head *pattrs)
405{
406	struct tracepoint_path path, *ppath = &path;
407	struct evsel *pos;
408	int nr_tracepoints = 0;
409
410	list_for_each_entry(pos, pattrs, core.node) {
411		if (pos->core.attr.type != PERF_TYPE_TRACEPOINT)
412			continue;
413		++nr_tracepoints;
414
415		if (pos->name) {
416			ppath->next = tracepoint_name_to_path(pos->name);
417			if (ppath->next)
418				goto next;
419
420			if (strchr(pos->name, ':') == NULL)
421				goto try_id;
422
423			goto error;
424		}
425
426try_id:
427		ppath->next = tracepoint_id_to_path(pos->core.attr.config);
428		if (!ppath->next) {
429error:
430			pr_debug("No memory to alloc tracepoints list\n");
431			put_tracepoints_path(&path);
432			return NULL;
433		}
434next:
435		ppath = ppath->next;
436	}
437
438	return nr_tracepoints > 0 ? path.next : NULL;
439}
440
441bool have_tracepoints(struct list_head *pattrs)
442{
443	struct evsel *pos;
444
445	list_for_each_entry(pos, pattrs, core.node)
446		if (pos->core.attr.type == PERF_TYPE_TRACEPOINT)
447			return true;
448
449	return false;
450}
451
452static int tracing_data_header(void)
453{
454	char buf[20];
455	ssize_t size;
456
457	/* just guessing this is someone's birthday.. ;) */
458	buf[0] = 23;
459	buf[1] = 8;
460	buf[2] = 68;
461	memcpy(buf + 3, "tracing", 7);
462
463	if (write(output_fd, buf, 10) != 10)
464		return -1;
465
466	size = strlen(VERSION) + 1;
467	if (write(output_fd, VERSION, size) != size)
468		return -1;
469
470	/* save endian */
471	if (bigendian())
472		buf[0] = 1;
473	else
474		buf[0] = 0;
475
476	if (write(output_fd, buf, 1) != 1)
477		return -1;
478
479	/* save size of long */
480	buf[0] = sizeof(long);
481	if (write(output_fd, buf, 1) != 1)
482		return -1;
483
484	/* save page_size */
485	if (write(output_fd, &page_size, 4) != 4)
486		return -1;
487
488	return 0;
489}
490
491struct tracing_data *tracing_data_get(struct list_head *pattrs,
492				      int fd, bool temp)
493{
494	struct tracepoint_path *tps;
495	struct tracing_data *tdata;
496	int err;
497
498	output_fd = fd;
499
500	tps = get_tracepoints_path(pattrs);
501	if (!tps)
502		return NULL;
503
504	tdata = malloc(sizeof(*tdata));
505	if (!tdata)
506		return NULL;
507
508	tdata->temp = temp;
509	tdata->size = 0;
510
511	if (temp) {
512		int temp_fd;
513
514		snprintf(tdata->temp_file, sizeof(tdata->temp_file),
515			 "/tmp/perf-XXXXXX");
516		if (!mkstemp(tdata->temp_file)) {
517			pr_debug("Can't make temp file");
518			free(tdata);
519			return NULL;
520		}
521
522		temp_fd = open(tdata->temp_file, O_RDWR);
523		if (temp_fd < 0) {
524			pr_debug("Can't read '%s'", tdata->temp_file);
525			free(tdata);
526			return NULL;
527		}
528
529		/*
530		 * Set the temp file the default output, so all the
531		 * tracing data are stored into it.
532		 */
533		output_fd = temp_fd;
534	}
535
536	err = tracing_data_header();
537	if (err)
538		goto out;
539	err = record_header_files();
540	if (err)
541		goto out;
542	err = record_ftrace_files(tps);
543	if (err)
544		goto out;
545	err = record_event_files(tps);
546	if (err)
547		goto out;
548	err = record_proc_kallsyms();
549	if (err)
550		goto out;
551	err = record_ftrace_printk();
552	if (err)
553		goto out;
554	err = record_saved_cmdline();
555
556out:
557	/*
558	 * All tracing data are stored by now, we can restore
559	 * the default output file in case we used temp file.
560	 */
561	if (temp) {
562		tdata->size = lseek(output_fd, 0, SEEK_CUR);
563		close(output_fd);
564		output_fd = fd;
565	}
566
567	if (err)
568		zfree(&tdata);
569
570	put_tracepoints_path(tps);
571	return tdata;
572}
573
574int tracing_data_put(struct tracing_data *tdata)
575{
576	int err = 0;
577
578	if (tdata->temp) {
579		err = record_file(tdata->temp_file, 0);
580		unlink(tdata->temp_file);
581	}
582
583	free(tdata);
584	return err;
585}
586
587int read_tracing_data(int fd, struct list_head *pattrs)
588{
589	int err;
590	struct tracing_data *tdata;
591
592	/*
593	 * We work over the real file, so we can write data
594	 * directly, no temp file is needed.
595	 */
596	tdata = tracing_data_get(pattrs, fd, false);
597	if (!tdata)
598		return -ENOMEM;
599
600	err = tracing_data_put(tdata);
601	return err;
602}
v4.17
 
  1/*
  2 * Copyright (C) 2008,2009, Steven Rostedt <srostedt@redhat.com>
  3 *
  4 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  5 *
  6 * This program is free software; you can redistribute it and/or modify
  7 * it under the terms of the GNU General Public License as published by
  8 * the Free Software Foundation; version 2 of the License (not later!)
  9 *
 10 * This program is distributed in the hope that it will be useful,
 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 13 * GNU General Public License for more details.
 14 *
 15 * You should have received a copy of the GNU General Public License
 16 * along with this program; if not, write to the Free Software
 17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 18 *
 19 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 20 */
 21#include "util.h"
 22#include <dirent.h>
 23#include <mntent.h>
 24#include <stdio.h>
 25#include <stdlib.h>
 26#include <string.h>
 27#include <stdarg.h>
 28#include <sys/types.h>
 29#include <sys/stat.h>
 30#include <sys/wait.h>
 31#include <fcntl.h>
 32#include <unistd.h>
 33#include <errno.h>
 34#include <stdbool.h>
 35#include <linux/list.h>
 36#include <linux/kernel.h>
 
 
 37
 38#include "../perf.h"
 39#include "trace-event.h"
 40#include <api/fs/tracing_path.h>
 41#include "evsel.h"
 42#include "debug.h"
 43
 44#define VERSION "0.6"
 45
 46static int output_fd;
 47
 48
 49int bigendian(void)
 50{
 51	unsigned char str[] = { 0x1, 0x2, 0x3, 0x4, 0x0, 0x0, 0x0, 0x0};
 52	unsigned int *ptr;
 53
 54	ptr = (unsigned int *)(void *)str;
 55	return *ptr == 0x01020304;
 56}
 57
 58/* unfortunately, you can not stat debugfs or proc files for size */
 59static int record_file(const char *file, ssize_t hdr_sz)
 60{
 61	unsigned long long size = 0;
 62	char buf[BUFSIZ], *sizep;
 63	off_t hdr_pos = lseek(output_fd, 0, SEEK_CUR);
 64	int r, fd;
 65	int err = -EIO;
 66
 67	fd = open(file, O_RDONLY);
 68	if (fd < 0) {
 69		pr_debug("Can't read '%s'", file);
 70		return -errno;
 71	}
 72
 73	/* put in zeros for file size, then fill true size later */
 74	if (hdr_sz) {
 75		if (write(output_fd, &size, hdr_sz) != hdr_sz)
 76			goto out;
 77	}
 78
 79	do {
 80		r = read(fd, buf, BUFSIZ);
 81		if (r > 0) {
 82			size += r;
 83			if (write(output_fd, buf, r) != r)
 84				goto out;
 85		}
 86	} while (r > 0);
 87
 88	/* ugh, handle big-endian hdr_size == 4 */
 89	sizep = (char*)&size;
 90	if (bigendian())
 91		sizep += sizeof(u64) - hdr_sz;
 92
 93	if (hdr_sz && pwrite(output_fd, sizep, hdr_sz, hdr_pos) < 0) {
 94		pr_debug("writing file size failed\n");
 95		goto out;
 96	}
 97
 98	err = 0;
 99out:
100	close(fd);
101	return err;
102}
103
104static int record_header_files(void)
105{
106	char *path;
107	struct stat st;
108	int err = -EIO;
109
110	path = get_tracing_file("events/header_page");
111	if (!path) {
112		pr_debug("can't get tracing/events/header_page");
113		return -ENOMEM;
114	}
115
116	if (stat(path, &st) < 0) {
117		pr_debug("can't read '%s'", path);
118		goto out;
119	}
120
121	if (write(output_fd, "header_page", 12) != 12) {
122		pr_debug("can't write header_page\n");
123		goto out;
124	}
125
126	if (record_file(path, 8) < 0) {
127		pr_debug("can't record header_page file\n");
128		goto out;
129	}
130
131	put_tracing_file(path);
132
133	path = get_tracing_file("events/header_event");
134	if (!path) {
135		pr_debug("can't get tracing/events/header_event");
136		err = -ENOMEM;
137		goto out;
138	}
139
140	if (stat(path, &st) < 0) {
141		pr_debug("can't read '%s'", path);
142		goto out;
143	}
144
145	if (write(output_fd, "header_event", 13) != 13) {
146		pr_debug("can't write header_event\n");
147		goto out;
148	}
149
150	if (record_file(path, 8) < 0) {
151		pr_debug("can't record header_event file\n");
152		goto out;
153	}
154
155	err = 0;
156out:
157	put_tracing_file(path);
158	return err;
159}
160
161static bool name_in_tp_list(char *sys, struct tracepoint_path *tps)
162{
163	while (tps) {
164		if (!strcmp(sys, tps->name))
165			return true;
166		tps = tps->next;
167	}
168
169	return false;
170}
171
172#define for_each_event(dir, dent, tps)				\
173	while ((dent = readdir(dir)))				\
174		if (dent->d_type == DT_DIR &&			\
175		    (strcmp(dent->d_name, ".")) &&		\
176		    (strcmp(dent->d_name, "..")))		\
177
178static int copy_event_system(const char *sys, struct tracepoint_path *tps)
179{
180	struct dirent *dent;
181	struct stat st;
182	char *format;
183	DIR *dir;
184	int count = 0;
185	int ret;
186	int err;
187
188	dir = opendir(sys);
189	if (!dir) {
190		pr_debug("can't read directory '%s'", sys);
191		return -errno;
192	}
193
194	for_each_event(dir, dent, tps) {
195		if (!name_in_tp_list(dent->d_name, tps))
196			continue;
197
198		if (asprintf(&format, "%s/%s/format", sys, dent->d_name) < 0) {
199			err = -ENOMEM;
200			goto out;
201		}
202		ret = stat(format, &st);
203		free(format);
204		if (ret < 0)
205			continue;
206		count++;
207	}
208
209	if (write(output_fd, &count, 4) != 4) {
210		err = -EIO;
211		pr_debug("can't write count\n");
212		goto out;
213	}
214
215	rewinddir(dir);
216	for_each_event(dir, dent, tps) {
217		if (!name_in_tp_list(dent->d_name, tps))
218			continue;
219
220		if (asprintf(&format, "%s/%s/format", sys, dent->d_name) < 0) {
221			err = -ENOMEM;
222			goto out;
223		}
224		ret = stat(format, &st);
225
226		if (ret >= 0) {
227			err = record_file(format, 8);
228			if (err) {
229				free(format);
230				goto out;
231			}
232		}
233		free(format);
234	}
235	err = 0;
236out:
237	closedir(dir);
238	return err;
239}
240
241static int record_ftrace_files(struct tracepoint_path *tps)
242{
243	char *path;
244	int ret;
245
246	path = get_tracing_file("events/ftrace");
247	if (!path) {
248		pr_debug("can't get tracing/events/ftrace");
249		return -ENOMEM;
250	}
251
252	ret = copy_event_system(path, tps);
253
254	put_tracing_file(path);
255
256	return ret;
257}
258
259static bool system_in_tp_list(char *sys, struct tracepoint_path *tps)
260{
261	while (tps) {
262		if (!strcmp(sys, tps->system))
263			return true;
264		tps = tps->next;
265	}
266
267	return false;
268}
269
270static int record_event_files(struct tracepoint_path *tps)
271{
272	struct dirent *dent;
273	struct stat st;
274	char *path;
275	char *sys;
276	DIR *dir;
277	int count = 0;
278	int ret;
279	int err;
280
281	path = get_tracing_file("events");
282	if (!path) {
283		pr_debug("can't get tracing/events");
284		return -ENOMEM;
285	}
286
287	dir = opendir(path);
288	if (!dir) {
289		err = -errno;
290		pr_debug("can't read directory '%s'", path);
291		goto out;
292	}
293
294	for_each_event(dir, dent, tps) {
295		if (strcmp(dent->d_name, "ftrace") == 0 ||
296		    !system_in_tp_list(dent->d_name, tps))
297			continue;
298
299		count++;
300	}
301
302	if (write(output_fd, &count, 4) != 4) {
303		err = -EIO;
304		pr_debug("can't write count\n");
305		goto out;
306	}
307
308	rewinddir(dir);
309	for_each_event(dir, dent, tps) {
310		if (strcmp(dent->d_name, "ftrace") == 0 ||
311		    !system_in_tp_list(dent->d_name, tps))
312			continue;
313
314		if (asprintf(&sys, "%s/%s", path, dent->d_name) < 0) {
315			err = -ENOMEM;
316			goto out;
317		}
318		ret = stat(sys, &st);
319		if (ret >= 0) {
320			ssize_t size = strlen(dent->d_name) + 1;
321
322			if (write(output_fd, dent->d_name, size) != size ||
323			    copy_event_system(sys, tps) < 0) {
324				err = -EIO;
325				free(sys);
326				goto out;
327			}
328		}
329		free(sys);
330	}
331	err = 0;
332out:
333	closedir(dir);
334	put_tracing_file(path);
335
336	return err;
337}
338
339static int record_proc_kallsyms(void)
340{
341	unsigned long long size = 0;
342	/*
343	 * Just to keep older perf.data file parsers happy, record a zero
344	 * sized kallsyms file, i.e. do the same thing that was done when
345	 * /proc/kallsyms (or something specified via --kallsyms, in a
346	 * different path) couldn't be read.
347	 */
348	return write(output_fd, &size, 4) != 4 ? -EIO : 0;
349}
350
351static int record_ftrace_printk(void)
352{
353	unsigned int size;
354	char *path;
355	struct stat st;
356	int ret, err = 0;
357
358	path = get_tracing_file("printk_formats");
359	if (!path) {
360		pr_debug("can't get tracing/printk_formats");
361		return -ENOMEM;
362	}
363
364	ret = stat(path, &st);
365	if (ret < 0) {
366		/* not found */
367		size = 0;
368		if (write(output_fd, &size, 4) != 4)
369			err = -EIO;
370		goto out;
371	}
372	err = record_file(path, 4);
373
374out:
375	put_tracing_file(path);
376	return err;
377}
378
379static int record_saved_cmdline(void)
380{
381	unsigned int size;
382	char *path;
383	struct stat st;
384	int ret, err = 0;
385
386	path = get_tracing_file("saved_cmdlines");
387	if (!path) {
388		pr_debug("can't get tracing/saved_cmdline");
389		return -ENOMEM;
390	}
391
392	ret = stat(path, &st);
393	if (ret < 0) {
394		/* not found */
395		size = 0;
396		if (write(output_fd, &size, 8) != 8)
397			err = -EIO;
398		goto out;
399	}
400	err = record_file(path, 8);
401
402out:
403	put_tracing_file(path);
404	return err;
405}
406
407static void
408put_tracepoints_path(struct tracepoint_path *tps)
409{
410	while (tps) {
411		struct tracepoint_path *t = tps;
412
413		tps = tps->next;
414		zfree(&t->name);
415		zfree(&t->system);
416		free(t);
417	}
418}
419
420static struct tracepoint_path *
421get_tracepoints_path(struct list_head *pattrs)
422{
423	struct tracepoint_path path, *ppath = &path;
424	struct perf_evsel *pos;
425	int nr_tracepoints = 0;
426
427	list_for_each_entry(pos, pattrs, node) {
428		if (pos->attr.type != PERF_TYPE_TRACEPOINT)
429			continue;
430		++nr_tracepoints;
431
432		if (pos->name) {
433			ppath->next = tracepoint_name_to_path(pos->name);
434			if (ppath->next)
435				goto next;
436
437			if (strchr(pos->name, ':') == NULL)
438				goto try_id;
439
440			goto error;
441		}
442
443try_id:
444		ppath->next = tracepoint_id_to_path(pos->attr.config);
445		if (!ppath->next) {
446error:
447			pr_debug("No memory to alloc tracepoints list\n");
448			put_tracepoints_path(&path);
449			return NULL;
450		}
451next:
452		ppath = ppath->next;
453	}
454
455	return nr_tracepoints > 0 ? path.next : NULL;
456}
457
458bool have_tracepoints(struct list_head *pattrs)
459{
460	struct perf_evsel *pos;
461
462	list_for_each_entry(pos, pattrs, node)
463		if (pos->attr.type == PERF_TYPE_TRACEPOINT)
464			return true;
465
466	return false;
467}
468
469static int tracing_data_header(void)
470{
471	char buf[20];
472	ssize_t size;
473
474	/* just guessing this is someone's birthday.. ;) */
475	buf[0] = 23;
476	buf[1] = 8;
477	buf[2] = 68;
478	memcpy(buf + 3, "tracing", 7);
479
480	if (write(output_fd, buf, 10) != 10)
481		return -1;
482
483	size = strlen(VERSION) + 1;
484	if (write(output_fd, VERSION, size) != size)
485		return -1;
486
487	/* save endian */
488	if (bigendian())
489		buf[0] = 1;
490	else
491		buf[0] = 0;
492
493	if (write(output_fd, buf, 1) != 1)
494		return -1;
495
496	/* save size of long */
497	buf[0] = sizeof(long);
498	if (write(output_fd, buf, 1) != 1)
499		return -1;
500
501	/* save page_size */
502	if (write(output_fd, &page_size, 4) != 4)
503		return -1;
504
505	return 0;
506}
507
508struct tracing_data *tracing_data_get(struct list_head *pattrs,
509				      int fd, bool temp)
510{
511	struct tracepoint_path *tps;
512	struct tracing_data *tdata;
513	int err;
514
515	output_fd = fd;
516
517	tps = get_tracepoints_path(pattrs);
518	if (!tps)
519		return NULL;
520
521	tdata = malloc(sizeof(*tdata));
522	if (!tdata)
523		return NULL;
524
525	tdata->temp = temp;
526	tdata->size = 0;
527
528	if (temp) {
529		int temp_fd;
530
531		snprintf(tdata->temp_file, sizeof(tdata->temp_file),
532			 "/tmp/perf-XXXXXX");
533		if (!mkstemp(tdata->temp_file)) {
534			pr_debug("Can't make temp file");
 
535			return NULL;
536		}
537
538		temp_fd = open(tdata->temp_file, O_RDWR);
539		if (temp_fd < 0) {
540			pr_debug("Can't read '%s'", tdata->temp_file);
 
541			return NULL;
542		}
543
544		/*
545		 * Set the temp file the default output, so all the
546		 * tracing data are stored into it.
547		 */
548		output_fd = temp_fd;
549	}
550
551	err = tracing_data_header();
552	if (err)
553		goto out;
554	err = record_header_files();
555	if (err)
556		goto out;
557	err = record_ftrace_files(tps);
558	if (err)
559		goto out;
560	err = record_event_files(tps);
561	if (err)
562		goto out;
563	err = record_proc_kallsyms();
564	if (err)
565		goto out;
566	err = record_ftrace_printk();
567	if (err)
568		goto out;
569	err = record_saved_cmdline();
570
571out:
572	/*
573	 * All tracing data are stored by now, we can restore
574	 * the default output file in case we used temp file.
575	 */
576	if (temp) {
577		tdata->size = lseek(output_fd, 0, SEEK_CUR);
578		close(output_fd);
579		output_fd = fd;
580	}
581
582	if (err)
583		zfree(&tdata);
584
585	put_tracepoints_path(tps);
586	return tdata;
587}
588
589int tracing_data_put(struct tracing_data *tdata)
590{
591	int err = 0;
592
593	if (tdata->temp) {
594		err = record_file(tdata->temp_file, 0);
595		unlink(tdata->temp_file);
596	}
597
598	free(tdata);
599	return err;
600}
601
602int read_tracing_data(int fd, struct list_head *pattrs)
603{
604	int err;
605	struct tracing_data *tdata;
606
607	/*
608	 * We work over the real file, so we can write data
609	 * directly, no temp file is needed.
610	 */
611	tdata = tracing_data_get(pattrs, fd, false);
612	if (!tdata)
613		return -ENOMEM;
614
615	err = tracing_data_put(tdata);
616	return err;
617}