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