builtin-inject.c 23 KB

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