Linux Audio

Check our new training course

Loading...
v4.6
 
  1/*
  2 * builtin-inject.c
  3 *
  4 * Builtin inject command: Examine the live mode (stdin) event stream
  5 * and repipe it to stdout while optionally injecting additional
  6 * events into it.
  7 */
  8#include "builtin.h"
  9
 10#include "perf.h"
 11#include "util/color.h"
 12#include "util/evlist.h"
 13#include "util/evsel.h"
 14#include "util/session.h"
 15#include "util/tool.h"
 16#include "util/debug.h"
 17#include "util/build-id.h"
 18#include "util/data.h"
 19#include "util/auxtrace.h"
 20#include "util/jit.h"
 
 21
 22#include <subcmd/parse-options.h>
 23
 24#include <linux/list.h>
 
 
 25
 26struct perf_inject {
 27	struct perf_tool	tool;
 28	struct perf_session	*session;
 29	bool			build_ids;
 30	bool			sched_stat;
 31	bool			have_auxtrace;
 32	bool			strip;
 33	bool			jit_mode;
 34	const char		*input_name;
 35	struct perf_data_file	output;
 36	u64			bytes_written;
 37	u64			aux_id;
 38	struct list_head	samples;
 39	struct itrace_synth_opts itrace_synth_opts;
 40};
 41
 42struct event_entry {
 43	struct list_head node;
 44	u32		 tid;
 45	union perf_event event[0];
 46};
 47
 48static int output_bytes(struct perf_inject *inject, void *buf, size_t sz)
 49{
 50	ssize_t size;
 51
 52	size = perf_data_file__write(&inject->output, buf, sz);
 53	if (size < 0)
 54		return -errno;
 55
 56	inject->bytes_written += size;
 57	return 0;
 58}
 59
 60static int perf_event__repipe_synth(struct perf_tool *tool,
 61				    union perf_event *event)
 62{
 63	struct perf_inject *inject = container_of(tool, struct perf_inject,
 64						  tool);
 65
 66	return output_bytes(inject, event, event->header.size);
 67}
 68
 69static int perf_event__repipe_oe_synth(struct perf_tool *tool,
 70				       union perf_event *event,
 71				       struct ordered_events *oe __maybe_unused)
 72{
 73	return perf_event__repipe_synth(tool, event);
 74}
 75
 76#ifdef HAVE_JITDUMP
 77static int perf_event__drop_oe(struct perf_tool *tool __maybe_unused,
 78			       union perf_event *event __maybe_unused,
 79			       struct ordered_events *oe __maybe_unused)
 80{
 81	return 0;
 82}
 83#endif
 84
 85static int perf_event__repipe_op2_synth(struct perf_tool *tool,
 86					union perf_event *event,
 87					struct perf_session *session
 88					__maybe_unused)
 89{
 90	return perf_event__repipe_synth(tool, event);
 91}
 92
 93static int perf_event__repipe_attr(struct perf_tool *tool,
 94				   union perf_event *event,
 95				   struct perf_evlist **pevlist)
 96{
 97	struct perf_inject *inject = container_of(tool, struct perf_inject,
 98						  tool);
 99	int ret;
100
101	ret = perf_event__process_attr(tool, event, pevlist);
102	if (ret)
103		return ret;
104
105	if (!inject->output.is_pipe)
106		return 0;
107
108	return perf_event__repipe_synth(tool, event);
109}
110
111#ifdef HAVE_AUXTRACE_SUPPORT
112
113static int copy_bytes(struct perf_inject *inject, int fd, off_t size)
114{
115	char buf[4096];
116	ssize_t ssz;
117	int ret;
118
119	while (size > 0) {
120		ssz = read(fd, buf, min(size, (off_t)sizeof(buf)));
121		if (ssz < 0)
122			return -errno;
123		ret = output_bytes(inject, buf, ssz);
124		if (ret)
125			return ret;
126		size -= ssz;
127	}
128
129	return 0;
130}
131
132static s64 perf_event__repipe_auxtrace(struct perf_tool *tool,
133				       union perf_event *event,
134				       struct perf_session *session)
135{
136	struct perf_inject *inject = container_of(tool, struct perf_inject,
137						  tool);
138	int ret;
139
140	inject->have_auxtrace = true;
141
142	if (!inject->output.is_pipe) {
143		off_t offset;
144
145		offset = lseek(inject->output.fd, 0, SEEK_CUR);
146		if (offset == -1)
147			return -errno;
148		ret = auxtrace_index__auxtrace_event(&session->auxtrace_index,
149						     event, offset);
150		if (ret < 0)
151			return ret;
152	}
153
154	if (perf_data_file__is_pipe(session->file) || !session->one_mmap) {
155		ret = output_bytes(inject, event, event->header.size);
156		if (ret < 0)
157			return ret;
158		ret = copy_bytes(inject, perf_data_file__fd(session->file),
159				 event->auxtrace.size);
160	} else {
161		ret = output_bytes(inject, event,
162				   event->header.size + event->auxtrace.size);
163	}
164	if (ret < 0)
165		return ret;
166
167	return event->auxtrace.size;
168}
169
170#else
171
172static s64
173perf_event__repipe_auxtrace(struct perf_tool *tool __maybe_unused,
174			    union perf_event *event __maybe_unused,
175			    struct perf_session *session __maybe_unused)
176{
177	pr_err("AUX area tracing not supported\n");
178	return -EINVAL;
179}
180
181#endif
182
183static int perf_event__repipe(struct perf_tool *tool,
184			      union perf_event *event,
185			      struct perf_sample *sample __maybe_unused,
186			      struct machine *machine __maybe_unused)
187{
188	return perf_event__repipe_synth(tool, event);
189}
190
191static int perf_event__drop(struct perf_tool *tool __maybe_unused,
192			    union perf_event *event __maybe_unused,
193			    struct perf_sample *sample __maybe_unused,
194			    struct machine *machine __maybe_unused)
195{
196	return 0;
197}
198
199static int perf_event__drop_aux(struct perf_tool *tool,
200				union perf_event *event __maybe_unused,
201				struct perf_sample *sample,
202				struct machine *machine __maybe_unused)
203{
204	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
205
206	if (!inject->aux_id)
207		inject->aux_id = sample->id;
208
209	return 0;
210}
211
212typedef int (*inject_handler)(struct perf_tool *tool,
213			      union perf_event *event,
214			      struct perf_sample *sample,
215			      struct perf_evsel *evsel,
216			      struct machine *machine);
217
218static int perf_event__repipe_sample(struct perf_tool *tool,
219				     union perf_event *event,
220				     struct perf_sample *sample,
221				     struct perf_evsel *evsel,
222				     struct machine *machine)
223{
224	if (evsel->handler) {
225		inject_handler f = evsel->handler;
226		return f(tool, event, sample, evsel, machine);
227	}
228
229	build_id__mark_dso_hit(tool, event, sample, evsel, machine);
230
231	return perf_event__repipe_synth(tool, event);
232}
233
234static int perf_event__repipe_mmap(struct perf_tool *tool,
235				   union perf_event *event,
236				   struct perf_sample *sample,
237				   struct machine *machine)
238{
239	int err;
240
241	err = perf_event__process_mmap(tool, event, sample, machine);
242	perf_event__repipe(tool, event, sample, machine);
243
244	return err;
245}
246
247#ifdef HAVE_JITDUMP
248static int perf_event__jit_repipe_mmap(struct perf_tool *tool,
249				       union perf_event *event,
250				       struct perf_sample *sample,
251				       struct machine *machine)
252{
253	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
254	u64 n = 0;
255	int ret;
256
257	/*
258	 * if jit marker, then inject jit mmaps and generate ELF images
259	 */
260	ret = jit_process(inject->session, &inject->output, machine,
261			  event->mmap.filename, sample->pid, &n);
262	if (ret < 0)
263		return ret;
264	if (ret) {
265		inject->bytes_written += n;
266		return 0;
267	}
268	return perf_event__repipe_mmap(tool, event, sample, machine);
269}
270#endif
271
272static int perf_event__repipe_mmap2(struct perf_tool *tool,
273				   union perf_event *event,
274				   struct perf_sample *sample,
275				   struct machine *machine)
276{
277	int err;
278
279	err = perf_event__process_mmap2(tool, event, sample, machine);
280	perf_event__repipe(tool, event, sample, machine);
281
282	return err;
283}
284
285#ifdef HAVE_JITDUMP
286static int perf_event__jit_repipe_mmap2(struct perf_tool *tool,
287					union perf_event *event,
288					struct perf_sample *sample,
289					struct machine *machine)
290{
291	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
292	u64 n = 0;
293	int ret;
294
295	/*
296	 * if jit marker, then inject jit mmaps and generate ELF images
297	 */
298	ret = jit_process(inject->session, &inject->output, machine,
299			  event->mmap2.filename, sample->pid, &n);
300	if (ret < 0)
301		return ret;
302	if (ret) {
303		inject->bytes_written += n;
304		return 0;
305	}
306	return perf_event__repipe_mmap2(tool, event, sample, machine);
307}
308#endif
309
310static int perf_event__repipe_fork(struct perf_tool *tool,
311				   union perf_event *event,
312				   struct perf_sample *sample,
313				   struct machine *machine)
314{
315	int err;
316
317	err = perf_event__process_fork(tool, event, sample, machine);
318	perf_event__repipe(tool, event, sample, machine);
319
320	return err;
321}
322
323static int perf_event__repipe_comm(struct perf_tool *tool,
324				   union perf_event *event,
325				   struct perf_sample *sample,
326				   struct machine *machine)
327{
328	int err;
329
330	err = perf_event__process_comm(tool, event, sample, machine);
331	perf_event__repipe(tool, event, sample, machine);
332
333	return err;
334}
335
 
 
 
 
 
 
 
 
 
 
 
 
336static int perf_event__repipe_exit(struct perf_tool *tool,
337				   union perf_event *event,
338				   struct perf_sample *sample,
339				   struct machine *machine)
340{
341	int err;
342
343	err = perf_event__process_exit(tool, event, sample, machine);
344	perf_event__repipe(tool, event, sample, machine);
345
346	return err;
347}
348
349static int perf_event__repipe_tracing_data(struct perf_tool *tool,
350					   union perf_event *event,
351					   struct perf_session *session)
352{
353	int err;
354
355	perf_event__repipe_synth(tool, event);
356	err = perf_event__process_tracing_data(tool, event, session);
357
358	return err;
359}
360
361static int perf_event__repipe_id_index(struct perf_tool *tool,
362				       union perf_event *event,
363				       struct perf_session *session)
364{
365	int err;
366
367	perf_event__repipe_synth(tool, event);
368	err = perf_event__process_id_index(tool, event, session);
369
370	return err;
371}
372
373static int dso__read_build_id(struct dso *dso)
374{
375	if (dso->has_build_id)
376		return 0;
377
378	if (filename__read_build_id(dso->long_name, dso->build_id,
379				    sizeof(dso->build_id)) > 0) {
380		dso->has_build_id = true;
381		return 0;
382	}
383
384	return -1;
385}
386
387static int dso__inject_build_id(struct dso *dso, struct perf_tool *tool,
388				struct machine *machine)
389{
390	u16 misc = PERF_RECORD_MISC_USER;
391	int err;
392
393	if (dso__read_build_id(dso) < 0) {
394		pr_debug("no build_id found for %s\n", dso->long_name);
395		return -1;
396	}
397
398	if (dso->kernel)
399		misc = PERF_RECORD_MISC_KERNEL;
400
401	err = perf_event__synthesize_build_id(tool, dso, misc, perf_event__repipe,
402					      machine);
403	if (err) {
404		pr_err("Can't synthesize build_id event for %s\n", dso->long_name);
405		return -1;
406	}
407
408	return 0;
409}
410
411static int perf_event__inject_buildid(struct perf_tool *tool,
412				      union perf_event *event,
413				      struct perf_sample *sample,
414				      struct perf_evsel *evsel __maybe_unused,
415				      struct machine *machine)
416{
417	struct addr_location al;
418	struct thread *thread;
419
420	thread = machine__findnew_thread(machine, sample->pid, sample->tid);
421	if (thread == NULL) {
422		pr_err("problem processing %d event, skipping it.\n",
423		       event->header.type);
424		goto repipe;
425	}
426
427	thread__find_addr_map(thread, sample->cpumode, MAP__FUNCTION, sample->ip, &al);
428
429	if (al.map != NULL) {
430		if (!al.map->dso->hit) {
431			al.map->dso->hit = 1;
432			if (map__load(al.map, NULL) >= 0) {
433				dso__inject_build_id(al.map->dso, tool, machine);
434				/*
435				 * If this fails, too bad, let the other side
436				 * account this as unresolved.
437				 */
438			} else {
439#ifdef HAVE_LIBELF_SUPPORT
440				pr_warning("no symbols found in %s, maybe "
441					   "install a debug package?\n",
442					   al.map->dso->long_name);
443#endif
444			}
445		}
446	}
447
448	thread__put(thread);
449repipe:
450	perf_event__repipe(tool, event, sample, machine);
451	return 0;
452}
453
454static int perf_inject__sched_process_exit(struct perf_tool *tool,
455					   union perf_event *event __maybe_unused,
456					   struct perf_sample *sample,
457					   struct perf_evsel *evsel __maybe_unused,
458					   struct machine *machine __maybe_unused)
459{
460	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
461	struct event_entry *ent;
462
463	list_for_each_entry(ent, &inject->samples, node) {
464		if (sample->tid == ent->tid) {
465			list_del_init(&ent->node);
466			free(ent);
467			break;
468		}
469	}
470
471	return 0;
472}
473
474static int perf_inject__sched_switch(struct perf_tool *tool,
475				     union perf_event *event,
476				     struct perf_sample *sample,
477				     struct perf_evsel *evsel,
478				     struct machine *machine)
479{
480	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
481	struct event_entry *ent;
482
483	perf_inject__sched_process_exit(tool, event, sample, evsel, machine);
484
485	ent = malloc(event->header.size + sizeof(struct event_entry));
486	if (ent == NULL) {
487		color_fprintf(stderr, PERF_COLOR_RED,
488			     "Not enough memory to process sched switch event!");
489		return -1;
490	}
491
492	ent->tid = sample->tid;
493	memcpy(&ent->event, event, event->header.size);
494	list_add(&ent->node, &inject->samples);
495	return 0;
496}
497
498static int perf_inject__sched_stat(struct perf_tool *tool,
499				   union perf_event *event __maybe_unused,
500				   struct perf_sample *sample,
501				   struct perf_evsel *evsel,
502				   struct machine *machine)
503{
504	struct event_entry *ent;
505	union perf_event *event_sw;
506	struct perf_sample sample_sw;
507	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
508	u32 pid = perf_evsel__intval(evsel, sample, "pid");
509
510	list_for_each_entry(ent, &inject->samples, node) {
511		if (pid == ent->tid)
512			goto found;
513	}
514
515	return 0;
516found:
517	event_sw = &ent->event[0];
518	perf_evsel__parse_sample(evsel, event_sw, &sample_sw);
519
520	sample_sw.period = sample->period;
521	sample_sw.time	 = sample->time;
522	perf_event__synthesize_sample(event_sw, evsel->attr.sample_type,
523				      evsel->attr.read_format, &sample_sw,
524				      false);
525	build_id__mark_dso_hit(tool, event_sw, &sample_sw, evsel, machine);
526	return perf_event__repipe(tool, event_sw, &sample_sw, machine);
527}
528
529static void sig_handler(int sig __maybe_unused)
530{
531	session_done = 1;
532}
533
534static int perf_evsel__check_stype(struct perf_evsel *evsel,
535				   u64 sample_type, const char *sample_msg)
536{
537	struct perf_event_attr *attr = &evsel->attr;
538	const char *name = perf_evsel__name(evsel);
539
540	if (!(attr->sample_type & sample_type)) {
541		pr_err("Samples for %s event do not have %s attribute set.",
542			name, sample_msg);
543		return -EINVAL;
544	}
545
546	return 0;
547}
548
549static int drop_sample(struct perf_tool *tool __maybe_unused,
550		       union perf_event *event __maybe_unused,
551		       struct perf_sample *sample __maybe_unused,
552		       struct perf_evsel *evsel __maybe_unused,
553		       struct machine *machine __maybe_unused)
554{
555	return 0;
556}
557
558static void strip_init(struct perf_inject *inject)
559{
560	struct perf_evlist *evlist = inject->session->evlist;
561	struct perf_evsel *evsel;
562
563	inject->tool.context_switch = perf_event__drop;
564
565	evlist__for_each(evlist, evsel)
566		evsel->handler = drop_sample;
567}
568
569static bool has_tracking(struct perf_evsel *evsel)
570{
571	return evsel->attr.mmap || evsel->attr.mmap2 || evsel->attr.comm ||
572	       evsel->attr.task;
573}
574
575#define COMPAT_MASK (PERF_SAMPLE_ID | PERF_SAMPLE_TID | PERF_SAMPLE_TIME | \
576		     PERF_SAMPLE_ID | PERF_SAMPLE_CPU | PERF_SAMPLE_IDENTIFIER)
577
578/*
579 * In order that the perf.data file is parsable, tracking events like MMAP need
580 * their selected event to exist, except if there is only 1 selected event left
581 * and it has a compatible sample type.
582 */
583static bool ok_to_remove(struct perf_evlist *evlist,
584			 struct perf_evsel *evsel_to_remove)
585{
586	struct perf_evsel *evsel;
587	int cnt = 0;
588	bool ok = false;
589
590	if (!has_tracking(evsel_to_remove))
591		return true;
592
593	evlist__for_each(evlist, evsel) {
594		if (evsel->handler != drop_sample) {
595			cnt += 1;
596			if ((evsel->attr.sample_type & COMPAT_MASK) ==
597			    (evsel_to_remove->attr.sample_type & COMPAT_MASK))
598				ok = true;
599		}
600	}
601
602	return ok && cnt == 1;
603}
604
605static void strip_fini(struct perf_inject *inject)
606{
607	struct perf_evlist *evlist = inject->session->evlist;
608	struct perf_evsel *evsel, *tmp;
609
610	/* Remove non-synthesized evsels if possible */
611	evlist__for_each_safe(evlist, tmp, evsel) {
612		if (evsel->handler == drop_sample &&
613		    ok_to_remove(evlist, evsel)) {
614			pr_debug("Deleting %s\n", perf_evsel__name(evsel));
615			perf_evlist__remove(evlist, evsel);
616			perf_evsel__delete(evsel);
617		}
618	}
619}
620
621static int __cmd_inject(struct perf_inject *inject)
622{
623	int ret = -EINVAL;
624	struct perf_session *session = inject->session;
625	struct perf_data_file *file_out = &inject->output;
626	int fd = perf_data_file__fd(file_out);
627	u64 output_data_offset;
628
629	signal(SIGINT, sig_handler);
630
631	if (inject->build_ids || inject->sched_stat ||
632	    inject->itrace_synth_opts.set) {
633		inject->tool.mmap	  = perf_event__repipe_mmap;
634		inject->tool.mmap2	  = perf_event__repipe_mmap2;
635		inject->tool.fork	  = perf_event__repipe_fork;
636		inject->tool.tracing_data = perf_event__repipe_tracing_data;
637	}
638
639	output_data_offset = session->header.data_offset;
640
641	if (inject->build_ids) {
642		inject->tool.sample = perf_event__inject_buildid;
643	} else if (inject->sched_stat) {
644		struct perf_evsel *evsel;
645
646		evlist__for_each(session->evlist, evsel) {
647			const char *name = perf_evsel__name(evsel);
648
649			if (!strcmp(name, "sched:sched_switch")) {
650				if (perf_evsel__check_stype(evsel, PERF_SAMPLE_TID, "TID"))
651					return -EINVAL;
652
653				evsel->handler = perf_inject__sched_switch;
654			} else if (!strcmp(name, "sched:sched_process_exit"))
655				evsel->handler = perf_inject__sched_process_exit;
656			else if (!strncmp(name, "sched:sched_stat_", 17))
657				evsel->handler = perf_inject__sched_stat;
658		}
659	} else if (inject->itrace_synth_opts.set) {
660		session->itrace_synth_opts = &inject->itrace_synth_opts;
661		inject->itrace_synth_opts.inject = true;
662		inject->tool.comm	    = perf_event__repipe_comm;
 
663		inject->tool.exit	    = perf_event__repipe_exit;
664		inject->tool.id_index	    = perf_event__repipe_id_index;
665		inject->tool.auxtrace_info  = perf_event__process_auxtrace_info;
666		inject->tool.auxtrace	    = perf_event__process_auxtrace;
667		inject->tool.aux	    = perf_event__drop_aux;
668		inject->tool.itrace_start   = perf_event__drop_aux,
669		inject->tool.ordered_events = true;
670		inject->tool.ordering_requires_timestamps = true;
671		/* Allow space in the header for new attributes */
672		output_data_offset = 4096;
673		if (inject->strip)
674			strip_init(inject);
675	}
676
677	if (!inject->itrace_synth_opts.set)
678		auxtrace_index__free(&session->auxtrace_index);
679
680	if (!file_out->is_pipe)
681		lseek(fd, output_data_offset, SEEK_SET);
682
683	ret = perf_session__process_events(session);
 
 
684
685	if (!file_out->is_pipe) {
686		if (inject->build_ids)
687			perf_header__set_feat(&session->header,
688					      HEADER_BUILD_ID);
689		/*
690		 * Keep all buildids when there is unprocessed AUX data because
691		 * it is not known which ones the AUX trace hits.
692		 */
693		if (perf_header__has_feat(&session->header, HEADER_BUILD_ID) &&
694		    inject->have_auxtrace && !inject->itrace_synth_opts.set)
695			dsos__hit_all(session);
696		/*
697		 * The AUX areas have been removed and replaced with
698		 * synthesized hardware events, so clear the feature flag and
699		 * remove the evsel.
700		 */
701		if (inject->itrace_synth_opts.set) {
702			struct perf_evsel *evsel;
703
704			perf_header__clear_feat(&session->header,
705						HEADER_AUXTRACE);
706			if (inject->itrace_synth_opts.last_branch)
707				perf_header__set_feat(&session->header,
708						      HEADER_BRANCH_STACK);
709			evsel = perf_evlist__id2evsel_strict(session->evlist,
710							     inject->aux_id);
711			if (evsel) {
712				pr_debug("Deleting %s\n",
713					 perf_evsel__name(evsel));
714				perf_evlist__remove(session->evlist, evsel);
715				perf_evsel__delete(evsel);
716			}
717			if (inject->strip)
718				strip_fini(inject);
719		}
720		session->header.data_offset = output_data_offset;
721		session->header.data_size = inject->bytes_written;
722		perf_session__write_header(session, session->evlist, fd, true);
723	}
724
725	return ret;
726}
727
728int cmd_inject(int argc, const char **argv, const char *prefix __maybe_unused)
729{
730	struct perf_inject inject = {
731		.tool = {
732			.sample		= perf_event__repipe_sample,
733			.mmap		= perf_event__repipe,
734			.mmap2		= perf_event__repipe,
735			.comm		= perf_event__repipe,
736			.fork		= perf_event__repipe,
737			.exit		= perf_event__repipe,
738			.lost		= perf_event__repipe,
739			.lost_samples	= perf_event__repipe,
740			.aux		= perf_event__repipe,
741			.itrace_start	= perf_event__repipe,
742			.context_switch	= perf_event__repipe,
743			.read		= perf_event__repipe_sample,
744			.throttle	= perf_event__repipe,
745			.unthrottle	= perf_event__repipe,
746			.attr		= perf_event__repipe_attr,
747			.tracing_data	= perf_event__repipe_op2_synth,
748			.auxtrace_info	= perf_event__repipe_op2_synth,
749			.auxtrace	= perf_event__repipe_auxtrace,
750			.auxtrace_error	= perf_event__repipe_op2_synth,
 
751			.finished_round	= perf_event__repipe_oe_synth,
752			.build_id	= perf_event__repipe_op2_synth,
753			.id_index	= perf_event__repipe_op2_synth,
 
754		},
755		.input_name  = "-",
756		.samples = LIST_HEAD_INIT(inject.samples),
757		.output = {
758			.path = "-",
759			.mode = PERF_DATA_MODE_WRITE,
 
 
760		},
761	};
762	struct perf_data_file file = {
763		.mode = PERF_DATA_MODE_READ,
764	};
765	int ret;
766
767	struct option options[] = {
768		OPT_BOOLEAN('b', "build-ids", &inject.build_ids,
769			    "Inject build-ids into the output stream"),
770		OPT_STRING('i', "input", &inject.input_name, "file",
771			   "input file name"),
772		OPT_STRING('o', "output", &inject.output.path, "file",
773			   "output file name"),
774		OPT_BOOLEAN('s', "sched-stat", &inject.sched_stat,
775			    "Merge sched-stat and sched-switch for getting events "
776			    "where and how long tasks slept"),
777#ifdef HAVE_JITDUMP
778		OPT_BOOLEAN('j', "jit", &inject.jit_mode, "merge jitdump files into perf.data file"),
779#endif
780		OPT_INCR('v', "verbose", &verbose,
781			 "be more verbose (show build ids, etc)"),
782		OPT_STRING(0, "kallsyms", &symbol_conf.kallsyms_name, "file",
783			   "kallsyms pathname"),
784		OPT_BOOLEAN('f', "force", &file.force, "don't complain, do it"),
785		OPT_CALLBACK_OPTARG(0, "itrace", &inject.itrace_synth_opts,
786				    NULL, "opts", "Instruction Tracing options",
787				    itrace_parse_synth_opts),
788		OPT_BOOLEAN(0, "strip", &inject.strip,
789			    "strip non-synthesized events (use with --itrace)"),
790		OPT_END()
791	};
792	const char * const inject_usage[] = {
793		"perf inject [<options>]",
794		NULL
795	};
796#ifndef HAVE_JITDUMP
797	set_option_nobuild(options, 'j', "jit", "NO_LIBELF=1", true);
798#endif
799	argc = parse_options(argc, argv, options, inject_usage, 0);
800
801	/*
802	 * Any (unrecognized) arguments left?
803	 */
804	if (argc)
805		usage_with_options(inject_usage, options);
806
807	if (inject.strip && !inject.itrace_synth_opts.set) {
808		pr_err("--strip option requires --itrace option\n");
809		return -1;
810	}
811
812	if (perf_data_file__open(&inject.output)) {
813		perror("failed to create output file");
814		return -1;
815	}
816
817	inject.tool.ordered_events = inject.sched_stat;
818
819	file.path = inject.input_name;
820	inject.session = perf_session__new(&file, true, &inject.tool);
821	if (inject.session == NULL)
822		return -1;
823
824	if (inject.build_ids) {
825		/*
826		 * to make sure the mmap records are ordered correctly
827		 * and so that the correct especially due to jitted code
828		 * mmaps. We cannot generate the buildid hit list and
829		 * inject the jit mmaps at the same time for now.
830		 */
831		inject.tool.ordered_events = true;
832		inject.tool.ordering_requires_timestamps = true;
833	}
834#ifdef HAVE_JITDUMP
835	if (inject.jit_mode) {
836		inject.tool.mmap2	   = perf_event__jit_repipe_mmap2;
837		inject.tool.mmap	   = perf_event__jit_repipe_mmap;
838		inject.tool.ordered_events = true;
839		inject.tool.ordering_requires_timestamps = true;
840		/*
841		 * JIT MMAP injection injects all MMAP events in one go, so it
842		 * does not obey finished_round semantics.
843		 */
844		inject.tool.finished_round = perf_event__drop_oe;
845	}
846#endif
847	ret = symbol__init(&inject.session->header.env);
848	if (ret < 0)
849		goto out_delete;
850
851	ret = __cmd_inject(&inject);
852
853out_delete:
854	perf_session__delete(inject.session);
855	return ret;
856}
v4.17
  1// SPDX-License-Identifier: GPL-2.0
  2/*
  3 * builtin-inject.c
  4 *
  5 * Builtin inject command: Examine the live mode (stdin) event stream
  6 * and repipe it to stdout while optionally injecting additional
  7 * events into it.
  8 */
  9#include "builtin.h"
 10
 11#include "perf.h"
 12#include "util/color.h"
 13#include "util/evlist.h"
 14#include "util/evsel.h"
 15#include "util/session.h"
 16#include "util/tool.h"
 17#include "util/debug.h"
 18#include "util/build-id.h"
 19#include "util/data.h"
 20#include "util/auxtrace.h"
 21#include "util/jit.h"
 22#include "util/thread.h"
 23
 24#include <subcmd/parse-options.h>
 25
 26#include <linux/list.h>
 27#include <errno.h>
 28#include <signal.h>
 29
 30struct perf_inject {
 31	struct perf_tool	tool;
 32	struct perf_session	*session;
 33	bool			build_ids;
 34	bool			sched_stat;
 35	bool			have_auxtrace;
 36	bool			strip;
 37	bool			jit_mode;
 38	const char		*input_name;
 39	struct perf_data	output;
 40	u64			bytes_written;
 41	u64			aux_id;
 42	struct list_head	samples;
 43	struct itrace_synth_opts itrace_synth_opts;
 44};
 45
 46struct event_entry {
 47	struct list_head node;
 48	u32		 tid;
 49	union perf_event event[0];
 50};
 51
 52static int output_bytes(struct perf_inject *inject, void *buf, size_t sz)
 53{
 54	ssize_t size;
 55
 56	size = perf_data__write(&inject->output, buf, sz);
 57	if (size < 0)
 58		return -errno;
 59
 60	inject->bytes_written += size;
 61	return 0;
 62}
 63
 64static int perf_event__repipe_synth(struct perf_tool *tool,
 65				    union perf_event *event)
 66{
 67	struct perf_inject *inject = container_of(tool, struct perf_inject,
 68						  tool);
 69
 70	return output_bytes(inject, event, event->header.size);
 71}
 72
 73static int perf_event__repipe_oe_synth(struct perf_tool *tool,
 74				       union perf_event *event,
 75				       struct ordered_events *oe __maybe_unused)
 76{
 77	return perf_event__repipe_synth(tool, event);
 78}
 79
 80#ifdef HAVE_JITDUMP
 81static int perf_event__drop_oe(struct perf_tool *tool __maybe_unused,
 82			       union perf_event *event __maybe_unused,
 83			       struct ordered_events *oe __maybe_unused)
 84{
 85	return 0;
 86}
 87#endif
 88
 89static int perf_event__repipe_op2_synth(struct perf_tool *tool,
 90					union perf_event *event,
 91					struct perf_session *session
 92					__maybe_unused)
 93{
 94	return perf_event__repipe_synth(tool, event);
 95}
 96
 97static int perf_event__repipe_attr(struct perf_tool *tool,
 98				   union perf_event *event,
 99				   struct perf_evlist **pevlist)
100{
101	struct perf_inject *inject = container_of(tool, struct perf_inject,
102						  tool);
103	int ret;
104
105	ret = perf_event__process_attr(tool, event, pevlist);
106	if (ret)
107		return ret;
108
109	if (!inject->output.is_pipe)
110		return 0;
111
112	return perf_event__repipe_synth(tool, event);
113}
114
115#ifdef HAVE_AUXTRACE_SUPPORT
116
117static int copy_bytes(struct perf_inject *inject, int fd, off_t size)
118{
119	char buf[4096];
120	ssize_t ssz;
121	int ret;
122
123	while (size > 0) {
124		ssz = read(fd, buf, min(size, (off_t)sizeof(buf)));
125		if (ssz < 0)
126			return -errno;
127		ret = output_bytes(inject, buf, ssz);
128		if (ret)
129			return ret;
130		size -= ssz;
131	}
132
133	return 0;
134}
135
136static s64 perf_event__repipe_auxtrace(struct perf_tool *tool,
137				       union perf_event *event,
138				       struct perf_session *session)
139{
140	struct perf_inject *inject = container_of(tool, struct perf_inject,
141						  tool);
142	int ret;
143
144	inject->have_auxtrace = true;
145
146	if (!inject->output.is_pipe) {
147		off_t offset;
148
149		offset = lseek(inject->output.file.fd, 0, SEEK_CUR);
150		if (offset == -1)
151			return -errno;
152		ret = auxtrace_index__auxtrace_event(&session->auxtrace_index,
153						     event, offset);
154		if (ret < 0)
155			return ret;
156	}
157
158	if (perf_data__is_pipe(session->data) || !session->one_mmap) {
159		ret = output_bytes(inject, event, event->header.size);
160		if (ret < 0)
161			return ret;
162		ret = copy_bytes(inject, perf_data__fd(session->data),
163				 event->auxtrace.size);
164	} else {
165		ret = output_bytes(inject, event,
166				   event->header.size + event->auxtrace.size);
167	}
168	if (ret < 0)
169		return ret;
170
171	return event->auxtrace.size;
172}
173
174#else
175
176static s64
177perf_event__repipe_auxtrace(struct perf_tool *tool __maybe_unused,
178			    union perf_event *event __maybe_unused,
179			    struct perf_session *session __maybe_unused)
180{
181	pr_err("AUX area tracing not supported\n");
182	return -EINVAL;
183}
184
185#endif
186
187static int perf_event__repipe(struct perf_tool *tool,
188			      union perf_event *event,
189			      struct perf_sample *sample __maybe_unused,
190			      struct machine *machine __maybe_unused)
191{
192	return perf_event__repipe_synth(tool, event);
193}
194
195static int perf_event__drop(struct perf_tool *tool __maybe_unused,
196			    union perf_event *event __maybe_unused,
197			    struct perf_sample *sample __maybe_unused,
198			    struct machine *machine __maybe_unused)
199{
200	return 0;
201}
202
203static int perf_event__drop_aux(struct perf_tool *tool,
204				union perf_event *event __maybe_unused,
205				struct perf_sample *sample,
206				struct machine *machine __maybe_unused)
207{
208	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
209
210	if (!inject->aux_id)
211		inject->aux_id = sample->id;
212
213	return 0;
214}
215
216typedef int (*inject_handler)(struct perf_tool *tool,
217			      union perf_event *event,
218			      struct perf_sample *sample,
219			      struct perf_evsel *evsel,
220			      struct machine *machine);
221
222static int perf_event__repipe_sample(struct perf_tool *tool,
223				     union perf_event *event,
224				     struct perf_sample *sample,
225				     struct perf_evsel *evsel,
226				     struct machine *machine)
227{
228	if (evsel->handler) {
229		inject_handler f = evsel->handler;
230		return f(tool, event, sample, evsel, machine);
231	}
232
233	build_id__mark_dso_hit(tool, event, sample, evsel, machine);
234
235	return perf_event__repipe_synth(tool, event);
236}
237
238static int perf_event__repipe_mmap(struct perf_tool *tool,
239				   union perf_event *event,
240				   struct perf_sample *sample,
241				   struct machine *machine)
242{
243	int err;
244
245	err = perf_event__process_mmap(tool, event, sample, machine);
246	perf_event__repipe(tool, event, sample, machine);
247
248	return err;
249}
250
251#ifdef HAVE_JITDUMP
252static int perf_event__jit_repipe_mmap(struct perf_tool *tool,
253				       union perf_event *event,
254				       struct perf_sample *sample,
255				       struct machine *machine)
256{
257	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
258	u64 n = 0;
259	int ret;
260
261	/*
262	 * if jit marker, then inject jit mmaps and generate ELF images
263	 */
264	ret = jit_process(inject->session, &inject->output, machine,
265			  event->mmap.filename, sample->pid, &n);
266	if (ret < 0)
267		return ret;
268	if (ret) {
269		inject->bytes_written += n;
270		return 0;
271	}
272	return perf_event__repipe_mmap(tool, event, sample, machine);
273}
274#endif
275
276static int perf_event__repipe_mmap2(struct perf_tool *tool,
277				   union perf_event *event,
278				   struct perf_sample *sample,
279				   struct machine *machine)
280{
281	int err;
282
283	err = perf_event__process_mmap2(tool, event, sample, machine);
284	perf_event__repipe(tool, event, sample, machine);
285
286	return err;
287}
288
289#ifdef HAVE_JITDUMP
290static int perf_event__jit_repipe_mmap2(struct perf_tool *tool,
291					union perf_event *event,
292					struct perf_sample *sample,
293					struct machine *machine)
294{
295	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
296	u64 n = 0;
297	int ret;
298
299	/*
300	 * if jit marker, then inject jit mmaps and generate ELF images
301	 */
302	ret = jit_process(inject->session, &inject->output, machine,
303			  event->mmap2.filename, sample->pid, &n);
304	if (ret < 0)
305		return ret;
306	if (ret) {
307		inject->bytes_written += n;
308		return 0;
309	}
310	return perf_event__repipe_mmap2(tool, event, sample, machine);
311}
312#endif
313
314static int perf_event__repipe_fork(struct perf_tool *tool,
315				   union perf_event *event,
316				   struct perf_sample *sample,
317				   struct machine *machine)
318{
319	int err;
320
321	err = perf_event__process_fork(tool, event, sample, machine);
322	perf_event__repipe(tool, event, sample, machine);
323
324	return err;
325}
326
327static int perf_event__repipe_comm(struct perf_tool *tool,
328				   union perf_event *event,
329				   struct perf_sample *sample,
330				   struct machine *machine)
331{
332	int err;
333
334	err = perf_event__process_comm(tool, event, sample, machine);
335	perf_event__repipe(tool, event, sample, machine);
336
337	return err;
338}
339
340static int perf_event__repipe_namespaces(struct perf_tool *tool,
341					 union perf_event *event,
342					 struct perf_sample *sample,
343					 struct machine *machine)
344{
345	int err = perf_event__process_namespaces(tool, event, sample, machine);
346
347	perf_event__repipe(tool, event, sample, machine);
348
349	return err;
350}
351
352static int perf_event__repipe_exit(struct perf_tool *tool,
353				   union perf_event *event,
354				   struct perf_sample *sample,
355				   struct machine *machine)
356{
357	int err;
358
359	err = perf_event__process_exit(tool, event, sample, machine);
360	perf_event__repipe(tool, event, sample, machine);
361
362	return err;
363}
364
365static int perf_event__repipe_tracing_data(struct perf_tool *tool,
366					   union perf_event *event,
367					   struct perf_session *session)
368{
369	int err;
370
371	perf_event__repipe_synth(tool, event);
372	err = perf_event__process_tracing_data(tool, event, session);
373
374	return err;
375}
376
377static int perf_event__repipe_id_index(struct perf_tool *tool,
378				       union perf_event *event,
379				       struct perf_session *session)
380{
381	int err;
382
383	perf_event__repipe_synth(tool, event);
384	err = perf_event__process_id_index(tool, event, session);
385
386	return err;
387}
388
389static int dso__read_build_id(struct dso *dso)
390{
391	if (dso->has_build_id)
392		return 0;
393
394	if (filename__read_build_id(dso->long_name, dso->build_id,
395				    sizeof(dso->build_id)) > 0) {
396		dso->has_build_id = true;
397		return 0;
398	}
399
400	return -1;
401}
402
403static int dso__inject_build_id(struct dso *dso, struct perf_tool *tool,
404				struct machine *machine)
405{
406	u16 misc = PERF_RECORD_MISC_USER;
407	int err;
408
409	if (dso__read_build_id(dso) < 0) {
410		pr_debug("no build_id found for %s\n", dso->long_name);
411		return -1;
412	}
413
414	if (dso->kernel)
415		misc = PERF_RECORD_MISC_KERNEL;
416
417	err = perf_event__synthesize_build_id(tool, dso, misc, perf_event__repipe,
418					      machine);
419	if (err) {
420		pr_err("Can't synthesize build_id event for %s\n", dso->long_name);
421		return -1;
422	}
423
424	return 0;
425}
426
427static int perf_event__inject_buildid(struct perf_tool *tool,
428				      union perf_event *event,
429				      struct perf_sample *sample,
430				      struct perf_evsel *evsel __maybe_unused,
431				      struct machine *machine)
432{
433	struct addr_location al;
434	struct thread *thread;
435
436	thread = machine__findnew_thread(machine, sample->pid, sample->tid);
437	if (thread == NULL) {
438		pr_err("problem processing %d event, skipping it.\n",
439		       event->header.type);
440		goto repipe;
441	}
442
443	thread__find_addr_map(thread, sample->cpumode, MAP__FUNCTION, sample->ip, &al);
444
445	if (al.map != NULL) {
446		if (!al.map->dso->hit) {
447			al.map->dso->hit = 1;
448			if (map__load(al.map) >= 0) {
449				dso__inject_build_id(al.map->dso, tool, machine);
450				/*
451				 * If this fails, too bad, let the other side
452				 * account this as unresolved.
453				 */
454			} else {
455#ifdef HAVE_LIBELF_SUPPORT
456				pr_warning("no symbols found in %s, maybe "
457					   "install a debug package?\n",
458					   al.map->dso->long_name);
459#endif
460			}
461		}
462	}
463
464	thread__put(thread);
465repipe:
466	perf_event__repipe(tool, event, sample, machine);
467	return 0;
468}
469
470static int perf_inject__sched_process_exit(struct perf_tool *tool,
471					   union perf_event *event __maybe_unused,
472					   struct perf_sample *sample,
473					   struct perf_evsel *evsel __maybe_unused,
474					   struct machine *machine __maybe_unused)
475{
476	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
477	struct event_entry *ent;
478
479	list_for_each_entry(ent, &inject->samples, node) {
480		if (sample->tid == ent->tid) {
481			list_del_init(&ent->node);
482			free(ent);
483			break;
484		}
485	}
486
487	return 0;
488}
489
490static int perf_inject__sched_switch(struct perf_tool *tool,
491				     union perf_event *event,
492				     struct perf_sample *sample,
493				     struct perf_evsel *evsel,
494				     struct machine *machine)
495{
496	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
497	struct event_entry *ent;
498
499	perf_inject__sched_process_exit(tool, event, sample, evsel, machine);
500
501	ent = malloc(event->header.size + sizeof(struct event_entry));
502	if (ent == NULL) {
503		color_fprintf(stderr, PERF_COLOR_RED,
504			     "Not enough memory to process sched switch event!");
505		return -1;
506	}
507
508	ent->tid = sample->tid;
509	memcpy(&ent->event, event, event->header.size);
510	list_add(&ent->node, &inject->samples);
511	return 0;
512}
513
514static int perf_inject__sched_stat(struct perf_tool *tool,
515				   union perf_event *event __maybe_unused,
516				   struct perf_sample *sample,
517				   struct perf_evsel *evsel,
518				   struct machine *machine)
519{
520	struct event_entry *ent;
521	union perf_event *event_sw;
522	struct perf_sample sample_sw;
523	struct perf_inject *inject = container_of(tool, struct perf_inject, tool);
524	u32 pid = perf_evsel__intval(evsel, sample, "pid");
525
526	list_for_each_entry(ent, &inject->samples, node) {
527		if (pid == ent->tid)
528			goto found;
529	}
530
531	return 0;
532found:
533	event_sw = &ent->event[0];
534	perf_evsel__parse_sample(evsel, event_sw, &sample_sw);
535
536	sample_sw.period = sample->period;
537	sample_sw.time	 = sample->time;
538	perf_event__synthesize_sample(event_sw, evsel->attr.sample_type,
539				      evsel->attr.read_format, &sample_sw);
 
540	build_id__mark_dso_hit(tool, event_sw, &sample_sw, evsel, machine);
541	return perf_event__repipe(tool, event_sw, &sample_sw, machine);
542}
543
544static void sig_handler(int sig __maybe_unused)
545{
546	session_done = 1;
547}
548
549static int perf_evsel__check_stype(struct perf_evsel *evsel,
550				   u64 sample_type, const char *sample_msg)
551{
552	struct perf_event_attr *attr = &evsel->attr;
553	const char *name = perf_evsel__name(evsel);
554
555	if (!(attr->sample_type & sample_type)) {
556		pr_err("Samples for %s event do not have %s attribute set.",
557			name, sample_msg);
558		return -EINVAL;
559	}
560
561	return 0;
562}
563
564static int drop_sample(struct perf_tool *tool __maybe_unused,
565		       union perf_event *event __maybe_unused,
566		       struct perf_sample *sample __maybe_unused,
567		       struct perf_evsel *evsel __maybe_unused,
568		       struct machine *machine __maybe_unused)
569{
570	return 0;
571}
572
573static void strip_init(struct perf_inject *inject)
574{
575	struct perf_evlist *evlist = inject->session->evlist;
576	struct perf_evsel *evsel;
577
578	inject->tool.context_switch = perf_event__drop;
579
580	evlist__for_each_entry(evlist, evsel)
581		evsel->handler = drop_sample;
582}
583
584static bool has_tracking(struct perf_evsel *evsel)
585{
586	return evsel->attr.mmap || evsel->attr.mmap2 || evsel->attr.comm ||
587	       evsel->attr.task;
588}
589
590#define COMPAT_MASK (PERF_SAMPLE_ID | PERF_SAMPLE_TID | PERF_SAMPLE_TIME | \
591		     PERF_SAMPLE_ID | PERF_SAMPLE_CPU | PERF_SAMPLE_IDENTIFIER)
592
593/*
594 * In order that the perf.data file is parsable, tracking events like MMAP need
595 * their selected event to exist, except if there is only 1 selected event left
596 * and it has a compatible sample type.
597 */
598static bool ok_to_remove(struct perf_evlist *evlist,
599			 struct perf_evsel *evsel_to_remove)
600{
601	struct perf_evsel *evsel;
602	int cnt = 0;
603	bool ok = false;
604
605	if (!has_tracking(evsel_to_remove))
606		return true;
607
608	evlist__for_each_entry(evlist, evsel) {
609		if (evsel->handler != drop_sample) {
610			cnt += 1;
611			if ((evsel->attr.sample_type & COMPAT_MASK) ==
612			    (evsel_to_remove->attr.sample_type & COMPAT_MASK))
613				ok = true;
614		}
615	}
616
617	return ok && cnt == 1;
618}
619
620static void strip_fini(struct perf_inject *inject)
621{
622	struct perf_evlist *evlist = inject->session->evlist;
623	struct perf_evsel *evsel, *tmp;
624
625	/* Remove non-synthesized evsels if possible */
626	evlist__for_each_entry_safe(evlist, tmp, evsel) {
627		if (evsel->handler == drop_sample &&
628		    ok_to_remove(evlist, evsel)) {
629			pr_debug("Deleting %s\n", perf_evsel__name(evsel));
630			perf_evlist__remove(evlist, evsel);
631			perf_evsel__delete(evsel);
632		}
633	}
634}
635
636static int __cmd_inject(struct perf_inject *inject)
637{
638	int ret = -EINVAL;
639	struct perf_session *session = inject->session;
640	struct perf_data *data_out = &inject->output;
641	int fd = perf_data__fd(data_out);
642	u64 output_data_offset;
643
644	signal(SIGINT, sig_handler);
645
646	if (inject->build_ids || inject->sched_stat ||
647	    inject->itrace_synth_opts.set) {
648		inject->tool.mmap	  = perf_event__repipe_mmap;
649		inject->tool.mmap2	  = perf_event__repipe_mmap2;
650		inject->tool.fork	  = perf_event__repipe_fork;
651		inject->tool.tracing_data = perf_event__repipe_tracing_data;
652	}
653
654	output_data_offset = session->header.data_offset;
655
656	if (inject->build_ids) {
657		inject->tool.sample = perf_event__inject_buildid;
658	} else if (inject->sched_stat) {
659		struct perf_evsel *evsel;
660
661		evlist__for_each_entry(session->evlist, evsel) {
662			const char *name = perf_evsel__name(evsel);
663
664			if (!strcmp(name, "sched:sched_switch")) {
665				if (perf_evsel__check_stype(evsel, PERF_SAMPLE_TID, "TID"))
666					return -EINVAL;
667
668				evsel->handler = perf_inject__sched_switch;
669			} else if (!strcmp(name, "sched:sched_process_exit"))
670				evsel->handler = perf_inject__sched_process_exit;
671			else if (!strncmp(name, "sched:sched_stat_", 17))
672				evsel->handler = perf_inject__sched_stat;
673		}
674	} else if (inject->itrace_synth_opts.set) {
675		session->itrace_synth_opts = &inject->itrace_synth_opts;
676		inject->itrace_synth_opts.inject = true;
677		inject->tool.comm	    = perf_event__repipe_comm;
678		inject->tool.namespaces	    = perf_event__repipe_namespaces;
679		inject->tool.exit	    = perf_event__repipe_exit;
680		inject->tool.id_index	    = perf_event__repipe_id_index;
681		inject->tool.auxtrace_info  = perf_event__process_auxtrace_info;
682		inject->tool.auxtrace	    = perf_event__process_auxtrace;
683		inject->tool.aux	    = perf_event__drop_aux;
684		inject->tool.itrace_start   = perf_event__drop_aux,
685		inject->tool.ordered_events = true;
686		inject->tool.ordering_requires_timestamps = true;
687		/* Allow space in the header for new attributes */
688		output_data_offset = 4096;
689		if (inject->strip)
690			strip_init(inject);
691	}
692
693	if (!inject->itrace_synth_opts.set)
694		auxtrace_index__free(&session->auxtrace_index);
695
696	if (!data_out->is_pipe)
697		lseek(fd, output_data_offset, SEEK_SET);
698
699	ret = perf_session__process_events(session);
700	if (ret)
701		return ret;
702
703	if (!data_out->is_pipe) {
704		if (inject->build_ids)
705			perf_header__set_feat(&session->header,
706					      HEADER_BUILD_ID);
707		/*
708		 * Keep all buildids when there is unprocessed AUX data because
709		 * it is not known which ones the AUX trace hits.
710		 */
711		if (perf_header__has_feat(&session->header, HEADER_BUILD_ID) &&
712		    inject->have_auxtrace && !inject->itrace_synth_opts.set)
713			dsos__hit_all(session);
714		/*
715		 * The AUX areas have been removed and replaced with
716		 * synthesized hardware events, so clear the feature flag and
717		 * remove the evsel.
718		 */
719		if (inject->itrace_synth_opts.set) {
720			struct perf_evsel *evsel;
721
722			perf_header__clear_feat(&session->header,
723						HEADER_AUXTRACE);
724			if (inject->itrace_synth_opts.last_branch)
725				perf_header__set_feat(&session->header,
726						      HEADER_BRANCH_STACK);
727			evsel = perf_evlist__id2evsel_strict(session->evlist,
728							     inject->aux_id);
729			if (evsel) {
730				pr_debug("Deleting %s\n",
731					 perf_evsel__name(evsel));
732				perf_evlist__remove(session->evlist, evsel);
733				perf_evsel__delete(evsel);
734			}
735			if (inject->strip)
736				strip_fini(inject);
737		}
738		session->header.data_offset = output_data_offset;
739		session->header.data_size = inject->bytes_written;
740		perf_session__write_header(session, session->evlist, fd, true);
741	}
742
743	return ret;
744}
745
746int cmd_inject(int argc, const char **argv)
747{
748	struct perf_inject inject = {
749		.tool = {
750			.sample		= perf_event__repipe_sample,
751			.mmap		= perf_event__repipe,
752			.mmap2		= perf_event__repipe,
753			.comm		= perf_event__repipe,
754			.fork		= perf_event__repipe,
755			.exit		= perf_event__repipe,
756			.lost		= perf_event__repipe,
757			.lost_samples	= perf_event__repipe,
758			.aux		= perf_event__repipe,
759			.itrace_start	= perf_event__repipe,
760			.context_switch	= perf_event__repipe,
761			.read		= perf_event__repipe_sample,
762			.throttle	= perf_event__repipe,
763			.unthrottle	= perf_event__repipe,
764			.attr		= perf_event__repipe_attr,
765			.tracing_data	= perf_event__repipe_op2_synth,
766			.auxtrace_info	= perf_event__repipe_op2_synth,
767			.auxtrace	= perf_event__repipe_auxtrace,
768			.auxtrace_error	= perf_event__repipe_op2_synth,
769			.time_conv	= perf_event__repipe_op2_synth,
770			.finished_round	= perf_event__repipe_oe_synth,
771			.build_id	= perf_event__repipe_op2_synth,
772			.id_index	= perf_event__repipe_op2_synth,
773			.feature	= perf_event__repipe_op2_synth,
774		},
775		.input_name  = "-",
776		.samples = LIST_HEAD_INIT(inject.samples),
777		.output = {
778			.file      = {
779				.path = "-",
780			},
781			.mode      = PERF_DATA_MODE_WRITE,
782		},
783	};
784	struct perf_data data = {
785		.mode = PERF_DATA_MODE_READ,
786	};
787	int ret;
788
789	struct option options[] = {
790		OPT_BOOLEAN('b', "build-ids", &inject.build_ids,
791			    "Inject build-ids into the output stream"),
792		OPT_STRING('i', "input", &inject.input_name, "file",
793			   "input file name"),
794		OPT_STRING('o', "output", &inject.output.file.path, "file",
795			   "output file name"),
796		OPT_BOOLEAN('s', "sched-stat", &inject.sched_stat,
797			    "Merge sched-stat and sched-switch for getting events "
798			    "where and how long tasks slept"),
799#ifdef HAVE_JITDUMP
800		OPT_BOOLEAN('j', "jit", &inject.jit_mode, "merge jitdump files into perf.data file"),
801#endif
802		OPT_INCR('v', "verbose", &verbose,
803			 "be more verbose (show build ids, etc)"),
804		OPT_STRING(0, "kallsyms", &symbol_conf.kallsyms_name, "file",
805			   "kallsyms pathname"),
806		OPT_BOOLEAN('f', "force", &data.force, "don't complain, do it"),
807		OPT_CALLBACK_OPTARG(0, "itrace", &inject.itrace_synth_opts,
808				    NULL, "opts", "Instruction Tracing options",
809				    itrace_parse_synth_opts),
810		OPT_BOOLEAN(0, "strip", &inject.strip,
811			    "strip non-synthesized events (use with --itrace)"),
812		OPT_END()
813	};
814	const char * const inject_usage[] = {
815		"perf inject [<options>]",
816		NULL
817	};
818#ifndef HAVE_JITDUMP
819	set_option_nobuild(options, 'j', "jit", "NO_LIBELF=1", true);
820#endif
821	argc = parse_options(argc, argv, options, inject_usage, 0);
822
823	/*
824	 * Any (unrecognized) arguments left?
825	 */
826	if (argc)
827		usage_with_options(inject_usage, options);
828
829	if (inject.strip && !inject.itrace_synth_opts.set) {
830		pr_err("--strip option requires --itrace option\n");
831		return -1;
832	}
833
834	if (perf_data__open(&inject.output)) {
835		perror("failed to create output file");
836		return -1;
837	}
838
839	inject.tool.ordered_events = inject.sched_stat;
840
841	data.file.path = inject.input_name;
842	inject.session = perf_session__new(&data, true, &inject.tool);
843	if (inject.session == NULL)
844		return -1;
845
846	if (inject.build_ids) {
847		/*
848		 * to make sure the mmap records are ordered correctly
849		 * and so that the correct especially due to jitted code
850		 * mmaps. We cannot generate the buildid hit list and
851		 * inject the jit mmaps at the same time for now.
852		 */
853		inject.tool.ordered_events = true;
854		inject.tool.ordering_requires_timestamps = true;
855	}
856#ifdef HAVE_JITDUMP
857	if (inject.jit_mode) {
858		inject.tool.mmap2	   = perf_event__jit_repipe_mmap2;
859		inject.tool.mmap	   = perf_event__jit_repipe_mmap;
860		inject.tool.ordered_events = true;
861		inject.tool.ordering_requires_timestamps = true;
862		/*
863		 * JIT MMAP injection injects all MMAP events in one go, so it
864		 * does not obey finished_round semantics.
865		 */
866		inject.tool.finished_round = perf_event__drop_oe;
867	}
868#endif
869	ret = symbol__init(&inject.session->header.env);
870	if (ret < 0)
871		goto out_delete;
872
873	ret = __cmd_inject(&inject);
874
875out_delete:
876	perf_session__delete(inject.session);
877	return ret;
878}