tools/perf/tests/perf-record.c - maze/linux - Git at Google

 #include <sched.h>
 #include "evlist.h"
 #include "evsel.h"
 #include "perf.h"
 #include "debug.h"
 #include "tests.h"

 static int sched__get_first_possible_cpu(pid_t pid, cpu_set_t *maskp)
 {
 	int i, cpu = -1, nrcpus = 1024;
 realloc:
 	CPU_ZERO(maskp);

 	if (sched_getaffinity(pid, sizeof(*maskp), maskp) == -1) {
 		if (errno == EINVAL && nrcpus < (1024 << 8)) {
 			nrcpus = nrcpus << 2;
 			goto realloc;
 		}
 		perror("sched_getaffinity");
 			return -1;
 	}

 	for (i = 0; i < nrcpus; i++) {
 		if (CPU_ISSET(i, maskp)) {
 			if (cpu == -1)
 				cpu = i;
 			else
 				CPU_CLR(i, maskp);
 		}
 	}

 	return cpu;
 }

 int test__PERF_RECORD(void)
 {
 	struct record_opts opts = {
 		.target = {
 			.uid = UINT_MAX,
 			.uses_mmap = true,
 		},
 		.no_buffering = true,
 		.freq	      = 10,
 		.mmap_pages   = 256,
 	};
 	cpu_set_t cpu_mask;
 	size_t cpu_mask_size = sizeof(cpu_mask);
 	struct perf_evlist *evlist = perf_evlist__new_default();
 	struct perf_evsel *evsel;
 	struct perf_sample sample;
 	const char *cmd = "sleep";
 	const char *argv[] = { cmd, "1", NULL, };
 	char *bname, *mmap_filename;
 	u64 prev_time = 0;
 	bool found_cmd_mmap = false,
 	     found_libc_mmap = false,
 	     found_vdso_mmap = false,
 	     found_ld_mmap = false;
 	int err = -1, errs = 0, i, wakeups = 0;
 	u32 cpu;
 	int total_events = 0, nr_events[PERF_RECORD_MAX] = { 0, };

 	if (evlist == NULL || argv == NULL) {
 		pr_debug("Not enough memory to create evlist\n");
 		goto out;
 	}

 	/*
 	 * Create maps of threads and cpus to monitor. In this case
 	 * we start with all threads and cpus (-1, -1) but then in
 	 * perf_evlist__prepare_workload we'll fill in the only thread
 	 * we're monitoring, the one forked there.
 	 */
 	err = perf_evlist__create_maps(evlist, &opts.target);
 	if (err < 0) {
 		pr_debug("Not enough memory to create thread/cpu maps\n");
 		goto out_delete_evlist;
 	}

 	/*
 	 * Prepare the workload in argv[] to run, it'll fork it, and then wait
 	 * for perf_evlist__start_workload() to exec it. This is done this way
 	 * so that we have time to open the evlist (calling sys_perf_event_open
 	 * on all the fds) and then mmap them.
 	 */
 	err = perf_evlist__prepare_workload(evlist, &opts.target, argv, false, NULL);
 	if (err < 0) {
 		pr_debug("Couldn't run the workload!\n");
 		goto out_delete_evlist;
 	}

 	/*
 	 * Config the evsels, setting attr->comm on the first one, etc.
 	 */
 	evsel = perf_evlist__first(evlist);
 	perf_evsel__set_sample_bit(evsel, CPU);
 	perf_evsel__set_sample_bit(evsel, TID);
 	perf_evsel__set_sample_bit(evsel, TIME);
 	perf_evlist__config(evlist, &opts);

 	err = sched__get_first_possible_cpu(evlist->workload.pid, &cpu_mask);
 	if (err < 0) {
 		pr_debug("sched__get_first_possible_cpu: %s\n", strerror(errno));
 		goto out_delete_evlist;
 	}

 	cpu = err;

 	/*
 	 * So that we can check perf_sample.cpu on all the samples.
 	 */
 	if (sched_setaffinity(evlist->workload.pid, cpu_mask_size, &cpu_mask) < 0) {
 		pr_debug("sched_setaffinity: %s\n", strerror(errno));
 		goto out_delete_evlist;
 	}

 	/*
 	 * Call sys_perf_event_open on all the fds on all the evsels,
 	 * grouping them if asked to.
 	 */
 	err = perf_evlist__open(evlist);
 	if (err < 0) {
 		pr_debug("perf_evlist__open: %s\n", strerror(errno));
 		goto out_delete_evlist;
 	}

 	/*
 	 * mmap the first fd on a given CPU and ask for events for the other
 	 * fds in the same CPU to be injected in the same mmap ring buffer
 	 * (using ioctl(PERF_EVENT_IOC_SET_OUTPUT)).
 	 */
 	err = perf_evlist__mmap(evlist, opts.mmap_pages, false);
 	if (err < 0) {
 		pr_debug("perf_evlist__mmap: %s\n", strerror(errno));
 		goto out_delete_evlist;
 	}

 	/*
 	 * Now that all is properly set up, enable the events, they will
 	 * count just on workload.pid, which will start...
 	 */
 	perf_evlist__enable(evlist);

 	/*
 	 * Now!
 	 */
 	perf_evlist__start_workload(evlist);

 	while (1) {
 		int before = total_events;

 		for (i = 0; i < evlist->nr_mmaps; i++) {
 			union perf_event *event;

 			while ((event = perf_evlist__mmap_read(evlist, i)) != NULL) {
 				const u32 type = event->header.type;
 				const char *name = perf_event__name(type);

 				++total_events;
 				if (type < PERF_RECORD_MAX)
 					nr_events[type]++;

 				err = perf_evlist__parse_sample(evlist, event, &sample);
 				if (err < 0) {
 					if (verbose)
 						perf_event__fprintf(event, stderr);
 					pr_debug("Couldn't parse sample\n");
 					goto out_delete_evlist;
 				}

 				if (verbose) {
 					pr_info("%" PRIu64" %d ", sample.time, sample.cpu);
 					perf_event__fprintf(event, stderr);
 				}

 				if (prev_time > sample.time) {
 					pr_debug("%s going backwards in time, prev=%" PRIu64 ", curr=%" PRIu64 "\n",
 						 name, prev_time, sample.time);
 					++errs;
 				}

 				prev_time = sample.time;

 				if (sample.cpu != cpu) {
 					pr_debug("%s with unexpected cpu, expected %d, got %d\n",
 						 name, cpu, sample.cpu);
 					++errs;
 				}

 				if ((pid_t)sample.pid != evlist->workload.pid) {
 					pr_debug("%s with unexpected pid, expected %d, got %d\n",
 						 name, evlist->workload.pid, sample.pid);
 					++errs;
 				}

 				if ((pid_t)sample.tid != evlist->workload.pid) {
 					pr_debug("%s with unexpected tid, expected %d, got %d\n",
 						 name, evlist->workload.pid, sample.tid);
 					++errs;
 				}

 				if ((type == PERF_RECORD_COMM ||
 				     type == PERF_RECORD_MMAP ||
 				     type == PERF_RECORD_MMAP2 ||
 				     type == PERF_RECORD_FORK ||
 				     type == PERF_RECORD_EXIT) &&
 				     (pid_t)event->comm.pid != evlist->workload.pid) {
 					pr_debug("%s with unexpected pid/tid\n", name);
 					++errs;
 				}

 				if ((type == PERF_RECORD_COMM ||
 				     type == PERF_RECORD_MMAP ||
 				     type == PERF_RECORD_MMAP2) &&
 				     event->comm.pid != event->comm.tid) {
 					pr_debug("%s with different pid/tid!\n", name);
 					++errs;
 				}

 				switch (type) {
 				case PERF_RECORD_COMM:
 					if (strcmp(event->comm.comm, cmd)) {
 						pr_debug("%s with unexpected comm!\n", name);
 						++errs;
 					}
 					break;
 				case PERF_RECORD_EXIT:
 					goto found_exit;
 				case PERF_RECORD_MMAP:
 					mmap_filename = event->mmap.filename;
 					goto check_bname;
 				case PERF_RECORD_MMAP2:
 					mmap_filename = event->mmap2.filename;
 				check_bname:
 					bname = strrchr(mmap_filename, '/');
 					if (bname != NULL) {
 						if (!found_cmd_mmap)
 							found_cmd_mmap = !strcmp(bname + 1, cmd);
 						if (!found_libc_mmap)
 							found_libc_mmap = !strncmp(bname + 1, "libc", 4);
 						if (!found_ld_mmap)
 							found_ld_mmap = !strncmp(bname + 1, "ld", 2);
 					} else if (!found_vdso_mmap)
 						found_vdso_mmap = !strcmp(mmap_filename, "[vdso]");
 					break;

 				case PERF_RECORD_SAMPLE:
 					/* Just ignore samples for now */
 					break;
 				default:
 					pr_debug("Unexpected perf_event->header.type %d!\n",
 						 type);
 					++errs;
 				}

 				perf_evlist__mmap_consume(evlist, i);
 			}
 		}

 		/*
 		 * We don't use poll here because at least at 3.1 times the
 		 * PERF_RECORD_{!SAMPLE} events don't honour
 		 * perf_event_attr.wakeup_events, just PERF_EVENT_SAMPLE does.
 		 */
 		if (total_events == before && false)
 			poll(evlist->pollfd, evlist->nr_fds, -1);

 		sleep(1);
 		if (++wakeups > 5) {
 			pr_debug("No PERF_RECORD_EXIT event!\n");
 			break;
 		}
 	}

 found_exit:
 	if (nr_events[PERF_RECORD_COMM] > 1) {
 		pr_debug("Excessive number of PERF_RECORD_COMM events!\n");
 		++errs;
 	}

 	if (nr_events[PERF_RECORD_COMM] == 0) {
 		pr_debug("Missing PERF_RECORD_COMM for %s!\n", cmd);
 		++errs;
 	}

 	if (!found_cmd_mmap) {
 		pr_debug("PERF_RECORD_MMAP for %s missing!\n", cmd);
 		++errs;
 	}

 	if (!found_libc_mmap) {
 		pr_debug("PERF_RECORD_MMAP for %s missing!\n", "libc");
 		++errs;
 	}

 	if (!found_ld_mmap) {
 		pr_debug("PERF_RECORD_MMAP for %s missing!\n", "ld");
 		++errs;
 	}

 	if (!found_vdso_mmap) {
 		pr_debug("PERF_RECORD_MMAP for %s missing!\n", "[vdso]");
 		++errs;
 	}
 out_delete_evlist:
 	perf_evlist__delete(evlist);
 out:
 	return (err < 0 || errs > 0) ? -1 : 0;
 }
	#include <sched.h>
	#include "evlist.h"
	#include "evsel.h"
	#include "perf.h"
	#include "debug.h"
	#include "tests.h"

	static int sched__get_first_possible_cpu(pid_t pid, cpu_set_t *maskp)
	{
	int i, cpu = -1, nrcpus = 1024;
	realloc:
	CPU_ZERO(maskp);

	if (sched_getaffinity(pid, sizeof(*maskp), maskp) == -1) {
	if (errno == EINVAL && nrcpus < (1024 << 8)) {
	nrcpus = nrcpus << 2;
	goto realloc;
	}
	perror("sched_getaffinity");
	return -1;
	}

	for (i = 0; i < nrcpus; i++) {
	if (CPU_ISSET(i, maskp)) {
	if (cpu == -1)
	cpu = i;
	else
	CPU_CLR(i, maskp);
	}
	}

	return cpu;
	}

	int test__PERF_RECORD(void)
	{
	struct record_opts opts = {
	.target = {
	.uid = UINT_MAX,
	.uses_mmap = true,
	},
	.no_buffering = true,
	.freq = 10,
	.mmap_pages = 256,
	};
	cpu_set_t cpu_mask;
	size_t cpu_mask_size = sizeof(cpu_mask);
	struct perf_evlist *evlist = perf_evlist__new_default();
	struct perf_evsel *evsel;
	struct perf_sample sample;
	const char *cmd = "sleep";
	const char *argv[] = { cmd, "1", NULL, };
	char bname, mmap_filename;
	u64 prev_time = 0;
	bool found_cmd_mmap = false,
	found_libc_mmap = false,
	found_vdso_mmap = false,
	found_ld_mmap = false;
	int err = -1, errs = 0, i, wakeups = 0;
	u32 cpu;
	int total_events = 0, nr_events[PERF_RECORD_MAX] = { 0, };

	if (evlist == NULL \|\| argv == NULL) {
	pr_debug("Not enough memory to create evlist\n");
	goto out;
	}

	/*
	* Create maps of threads and cpus to monitor. In this case
	* we start with all threads and cpus (-1, -1) but then in
	* perf_evlist__prepare_workload we'll fill in the only thread
	* we're monitoring, the one forked there.
	*/
	err = perf_evlist__create_maps(evlist, &opts.target);
	if (err < 0) {
	pr_debug("Not enough memory to create thread/cpu maps\n");
	goto out_delete_evlist;
	}

	/*
	* Prepare the workload in argv[] to run, it'll fork it, and then wait
	* for perf_evlist__start_workload() to exec it. This is done this way
	* so that we have time to open the evlist (calling sys_perf_event_open
	* on all the fds) and then mmap them.
	*/
	err = perf_evlist__prepare_workload(evlist, &opts.target, argv, false, NULL);
	if (err < 0) {
	pr_debug("Couldn't run the workload!\n");
	goto out_delete_evlist;
	}

	/*
	* Config the evsels, setting attr->comm on the first one, etc.
	*/
	evsel = perf_evlist__first(evlist);
	perf_evsel__set_sample_bit(evsel, CPU);
	perf_evsel__set_sample_bit(evsel, TID);
	perf_evsel__set_sample_bit(evsel, TIME);
	perf_evlist__config(evlist, &opts);

	err = sched__get_first_possible_cpu(evlist->workload.pid, &cpu_mask);
	if (err < 0) {
	pr_debug("sched__get_first_possible_cpu: %s\n", strerror(errno));
	goto out_delete_evlist;
	}

	cpu = err;

	/*
	* So that we can check perf_sample.cpu on all the samples.
	*/
	if (sched_setaffinity(evlist->workload.pid, cpu_mask_size, &cpu_mask) < 0) {
	pr_debug("sched_setaffinity: %s\n", strerror(errno));
	goto out_delete_evlist;
	}

	/*
	* Call sys_perf_event_open on all the fds on all the evsels,
	* grouping them if asked to.
	*/
	err = perf_evlist__open(evlist);
	if (err < 0) {
	pr_debug("perf_evlist__open: %s\n", strerror(errno));
	goto out_delete_evlist;
	}

	/*
	* mmap the first fd on a given CPU and ask for events for the other
	* fds in the same CPU to be injected in the same mmap ring buffer
	* (using ioctl(PERF_EVENT_IOC_SET_OUTPUT)).
	*/
	err = perf_evlist__mmap(evlist, opts.mmap_pages, false);
	if (err < 0) {
	pr_debug("perf_evlist__mmap: %s\n", strerror(errno));
	goto out_delete_evlist;
	}

	/*
	* Now that all is properly set up, enable the events, they will
	* count just on workload.pid, which will start...
	*/
	perf_evlist__enable(evlist);

	/*
	* Now!
	*/
	perf_evlist__start_workload(evlist);

	while (1) {
	int before = total_events;

	for (i = 0; i < evlist->nr_mmaps; i++) {
	union perf_event *event;

	while ((event = perf_evlist__mmap_read(evlist, i)) != NULL) {
	const u32 type = event->header.type;
	const char *name = perf_event__name(type);

	++total_events;
	if (type < PERF_RECORD_MAX)
	nr_events[type]++;

	err = perf_evlist__parse_sample(evlist, event, &sample);
	if (err < 0) {
	if (verbose)
	perf_event__fprintf(event, stderr);
	pr_debug("Couldn't parse sample\n");
	goto out_delete_evlist;
	}

	if (verbose) {
	pr_info("%" PRIu64" %d ", sample.time, sample.cpu);
	perf_event__fprintf(event, stderr);
	}

	if (prev_time > sample.time) {
	pr_debug("%s going backwards in time, prev=%" PRIu64 ", curr=%" PRIu64 "\n",
	name, prev_time, sample.time);
	++errs;
	}

	prev_time = sample.time;

	if (sample.cpu != cpu) {
	pr_debug("%s with unexpected cpu, expected %d, got %d\n",
	name, cpu, sample.cpu);
	++errs;
	}

	if ((pid_t)sample.pid != evlist->workload.pid) {
	pr_debug("%s with unexpected pid, expected %d, got %d\n",
	name, evlist->workload.pid, sample.pid);
	++errs;
	}

	if ((pid_t)sample.tid != evlist->workload.pid) {
	pr_debug("%s with unexpected tid, expected %d, got %d\n",
	name, evlist->workload.pid, sample.tid);
	++errs;
	}

	if ((type == PERF_RECORD_COMM \|\|
	type == PERF_RECORD_MMAP \|\|
	type == PERF_RECORD_MMAP2 \|\|
	type == PERF_RECORD_FORK \|\|
	type == PERF_RECORD_EXIT) &&
	(pid_t)event->comm.pid != evlist->workload.pid) {
	pr_debug("%s with unexpected pid/tid\n", name);
	++errs;
	}

	if ((type == PERF_RECORD_COMM \|\|
	type == PERF_RECORD_MMAP \|\|
	type == PERF_RECORD_MMAP2) &&
	event->comm.pid != event->comm.tid) {
	pr_debug("%s with different pid/tid!\n", name);
	++errs;
	}

	switch (type) {
	case PERF_RECORD_COMM:
	if (strcmp(event->comm.comm, cmd)) {
	pr_debug("%s with unexpected comm!\n", name);
	++errs;
	}
	break;
	case PERF_RECORD_EXIT:
	goto found_exit;
	case PERF_RECORD_MMAP:
	mmap_filename = event->mmap.filename;
	goto check_bname;
	case PERF_RECORD_MMAP2:
	mmap_filename = event->mmap2.filename;
	check_bname:
	bname = strrchr(mmap_filename, '/');
	if (bname != NULL) {
	if (!found_cmd_mmap)
	found_cmd_mmap = !strcmp(bname + 1, cmd);
	if (!found_libc_mmap)
	found_libc_mmap = !strncmp(bname + 1, "libc", 4);
	if (!found_ld_mmap)
	found_ld_mmap = !strncmp(bname + 1, "ld", 2);
	} else if (!found_vdso_mmap)
	found_vdso_mmap = !strcmp(mmap_filename, "[vdso]");
	break;

	case PERF_RECORD_SAMPLE:
	/* Just ignore samples for now */
	break;
	default:
	pr_debug("Unexpected perf_event->header.type %d!\n",
	type);
	++errs;
	}

	perf_evlist__mmap_consume(evlist, i);
	}
	}

	/*
	* We don't use poll here because at least at 3.1 times the
	* PERF_RECORD_{!SAMPLE} events don't honour
	* perf_event_attr.wakeup_events, just PERF_EVENT_SAMPLE does.
	*/
	if (total_events == before && false)
	poll(evlist->pollfd, evlist->nr_fds, -1);

	sleep(1);
	if (++wakeups > 5) {
	pr_debug("No PERF_RECORD_EXIT event!\n");
	break;
	}
	}

	found_exit:
	if (nr_events[PERF_RECORD_COMM] > 1) {
	pr_debug("Excessive number of PERF_RECORD_COMM events!\n");
	++errs;
	}

	if (nr_events[PERF_RECORD_COMM] == 0) {
	pr_debug("Missing PERF_RECORD_COMM for %s!\n", cmd);
	++errs;
	}

	if (!found_cmd_mmap) {
	pr_debug("PERF_RECORD_MMAP for %s missing!\n", cmd);
	++errs;
	}

	if (!found_libc_mmap) {
	pr_debug("PERF_RECORD_MMAP for %s missing!\n", "libc");
	++errs;
	}

	if (!found_ld_mmap) {
	pr_debug("PERF_RECORD_MMAP for %s missing!\n", "ld");
	++errs;
	}

	if (!found_vdso_mmap) {
	pr_debug("PERF_RECORD_MMAP for %s missing!\n", "[vdso]");
	++errs;
	}
	out_delete_evlist:
	perf_evlist__delete(evlist);
	out:
	return (err < 0 \|\| errs > 0) ? -1 : 0;
	}