builtin-inject.c 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277
  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. #include "perf.h"
  10. #include "util/session.h"
  11. #include "util/tool.h"
  12. #include "util/debug.h"
  13. #include "util/parse-options.h"
  14. static char const *input_name = "-";
  15. static bool inject_build_ids;
  16. static int perf_event__repipe_synth(struct perf_tool *tool __used,
  17. union perf_event *event,
  18. struct machine *machine __used)
  19. {
  20. uint32_t size;
  21. void *buf = event;
  22. size = event->header.size;
  23. while (size) {
  24. int ret = write(STDOUT_FILENO, buf, size);
  25. if (ret < 0)
  26. return -errno;
  27. size -= ret;
  28. buf += ret;
  29. }
  30. return 0;
  31. }
  32. static int perf_event__repipe_op2_synth(struct perf_tool *tool,
  33. union perf_event *event,
  34. struct perf_session *session __used)
  35. {
  36. return perf_event__repipe_synth(tool, event, NULL);
  37. }
  38. static int perf_event__repipe_event_type_synth(struct perf_tool *tool,
  39. union perf_event *event)
  40. {
  41. return perf_event__repipe_synth(tool, event, NULL);
  42. }
  43. static int perf_event__repipe_tracing_data_synth(union perf_event *event,
  44. struct perf_session *session __used)
  45. {
  46. return perf_event__repipe_synth(NULL, event, NULL);
  47. }
  48. static int perf_event__repipe_attr(union perf_event *event,
  49. struct perf_evlist **pevlist __used)
  50. {
  51. return perf_event__repipe_synth(NULL, event, NULL);
  52. }
  53. static int perf_event__repipe(struct perf_tool *tool,
  54. union perf_event *event,
  55. struct perf_sample *sample __used,
  56. struct machine *machine)
  57. {
  58. return perf_event__repipe_synth(tool, event, machine);
  59. }
  60. static int perf_event__repipe_sample(struct perf_tool *tool,
  61. union perf_event *event,
  62. struct perf_sample *sample __used,
  63. struct perf_evsel *evsel __used,
  64. struct machine *machine)
  65. {
  66. return perf_event__repipe_synth(tool, event, machine);
  67. }
  68. static int perf_event__repipe_mmap(struct perf_tool *tool,
  69. union perf_event *event,
  70. struct perf_sample *sample,
  71. struct machine *machine)
  72. {
  73. int err;
  74. err = perf_event__process_mmap(tool, event, sample, machine);
  75. perf_event__repipe(tool, event, sample, machine);
  76. return err;
  77. }
  78. static int perf_event__repipe_task(struct perf_tool *tool,
  79. union perf_event *event,
  80. struct perf_sample *sample,
  81. struct machine *machine)
  82. {
  83. int err;
  84. err = perf_event__process_task(tool, event, sample, machine);
  85. perf_event__repipe(tool, event, sample, machine);
  86. return err;
  87. }
  88. static int perf_event__repipe_tracing_data(union perf_event *event,
  89. struct perf_session *session)
  90. {
  91. int err;
  92. perf_event__repipe_synth(NULL, event, NULL);
  93. err = perf_event__process_tracing_data(event, session);
  94. return err;
  95. }
  96. static int dso__read_build_id(struct dso *self)
  97. {
  98. if (self->has_build_id)
  99. return 0;
  100. if (filename__read_build_id(self->long_name, self->build_id,
  101. sizeof(self->build_id)) > 0) {
  102. self->has_build_id = true;
  103. return 0;
  104. }
  105. return -1;
  106. }
  107. static int dso__inject_build_id(struct dso *self, struct perf_tool *tool,
  108. struct machine *machine)
  109. {
  110. u16 misc = PERF_RECORD_MISC_USER;
  111. int err;
  112. if (dso__read_build_id(self) < 0) {
  113. pr_debug("no build_id found for %s\n", self->long_name);
  114. return -1;
  115. }
  116. if (self->kernel)
  117. misc = PERF_RECORD_MISC_KERNEL;
  118. err = perf_event__synthesize_build_id(tool, self, misc, perf_event__repipe,
  119. machine);
  120. if (err) {
  121. pr_err("Can't synthesize build_id event for %s\n", self->long_name);
  122. return -1;
  123. }
  124. return 0;
  125. }
  126. static int perf_event__inject_buildid(struct perf_tool *tool,
  127. union perf_event *event,
  128. struct perf_sample *sample,
  129. struct perf_evsel *evsel __used,
  130. struct machine *machine)
  131. {
  132. struct addr_location al;
  133. struct thread *thread;
  134. u8 cpumode;
  135. cpumode = event->header.misc & PERF_RECORD_MISC_CPUMODE_MASK;
  136. thread = machine__findnew_thread(machine, event->ip.pid);
  137. if (thread == NULL) {
  138. pr_err("problem processing %d event, skipping it.\n",
  139. event->header.type);
  140. goto repipe;
  141. }
  142. thread__find_addr_map(thread, machine, cpumode, MAP__FUNCTION,
  143. event->ip.ip, &al);
  144. if (al.map != NULL) {
  145. if (!al.map->dso->hit) {
  146. al.map->dso->hit = 1;
  147. if (map__load(al.map, NULL) >= 0) {
  148. dso__inject_build_id(al.map->dso, tool, machine);
  149. /*
  150. * If this fails, too bad, let the other side
  151. * account this as unresolved.
  152. */
  153. } else
  154. pr_warning("no symbols found in %s, maybe "
  155. "install a debug package?\n",
  156. al.map->dso->long_name);
  157. }
  158. }
  159. repipe:
  160. perf_event__repipe(tool, event, sample, machine);
  161. return 0;
  162. }
  163. struct perf_tool perf_inject = {
  164. .sample = perf_event__repipe_sample,
  165. .mmap = perf_event__repipe,
  166. .comm = perf_event__repipe,
  167. .fork = perf_event__repipe,
  168. .exit = perf_event__repipe,
  169. .lost = perf_event__repipe,
  170. .read = perf_event__repipe_sample,
  171. .throttle = perf_event__repipe,
  172. .unthrottle = perf_event__repipe,
  173. .attr = perf_event__repipe_attr,
  174. .event_type = perf_event__repipe_event_type_synth,
  175. .tracing_data = perf_event__repipe_tracing_data_synth,
  176. .build_id = perf_event__repipe_op2_synth,
  177. };
  178. extern volatile int session_done;
  179. static void sig_handler(int sig __attribute__((__unused__)))
  180. {
  181. session_done = 1;
  182. }
  183. static int __cmd_inject(void)
  184. {
  185. struct perf_session *session;
  186. int ret = -EINVAL;
  187. signal(SIGINT, sig_handler);
  188. if (inject_build_ids) {
  189. perf_inject.sample = perf_event__inject_buildid;
  190. perf_inject.mmap = perf_event__repipe_mmap;
  191. perf_inject.fork = perf_event__repipe_task;
  192. perf_inject.tracing_data = perf_event__repipe_tracing_data;
  193. }
  194. session = perf_session__new(input_name, O_RDONLY, false, true, &perf_inject);
  195. if (session == NULL)
  196. return -ENOMEM;
  197. ret = perf_session__process_events(session, &perf_inject);
  198. perf_session__delete(session);
  199. return ret;
  200. }
  201. static const char * const report_usage[] = {
  202. "perf inject [<options>]",
  203. NULL
  204. };
  205. static const struct option options[] = {
  206. OPT_BOOLEAN('b', "build-ids", &inject_build_ids,
  207. "Inject build-ids into the output stream"),
  208. OPT_INCR('v', "verbose", &verbose,
  209. "be more verbose (show build ids, etc)"),
  210. OPT_END()
  211. };
  212. int cmd_inject(int argc, const char **argv, const char *prefix __used)
  213. {
  214. argc = parse_options(argc, argv, options, report_usage, 0);
  215. /*
  216. * Any (unrecognized) arguments left?
  217. */
  218. if (argc)
  219. usage_with_options(report_usage, options);
  220. if (symbol__init() < 0)
  221. return -1;
  222. return __cmd_inject();
  223. }