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