data_map.c 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252
  1. #include "data_map.h"
  2. #include "symbol.h"
  3. #include "util.h"
  4. #include "debug.h"
  5. static struct perf_file_handler *curr_handler;
  6. static unsigned long mmap_window = 32;
  7. static char __cwd[PATH_MAX];
  8. static int
  9. process_event_stub(event_t *event __used,
  10. unsigned long offset __used,
  11. unsigned long head __used)
  12. {
  13. return 0;
  14. }
  15. void register_perf_file_handler(struct perf_file_handler *handler)
  16. {
  17. if (!handler->process_sample_event)
  18. handler->process_sample_event = process_event_stub;
  19. if (!handler->process_mmap_event)
  20. handler->process_mmap_event = process_event_stub;
  21. if (!handler->process_comm_event)
  22. handler->process_comm_event = process_event_stub;
  23. if (!handler->process_fork_event)
  24. handler->process_fork_event = process_event_stub;
  25. if (!handler->process_exit_event)
  26. handler->process_exit_event = process_event_stub;
  27. if (!handler->process_lost_event)
  28. handler->process_lost_event = process_event_stub;
  29. if (!handler->process_read_event)
  30. handler->process_read_event = process_event_stub;
  31. if (!handler->process_throttle_event)
  32. handler->process_throttle_event = process_event_stub;
  33. if (!handler->process_unthrottle_event)
  34. handler->process_unthrottle_event = process_event_stub;
  35. curr_handler = handler;
  36. }
  37. static int
  38. process_event(event_t *event, unsigned long offset, unsigned long head)
  39. {
  40. trace_event(event);
  41. switch (event->header.type) {
  42. case PERF_RECORD_SAMPLE:
  43. return curr_handler->process_sample_event(event, offset, head);
  44. case PERF_RECORD_MMAP:
  45. return curr_handler->process_mmap_event(event, offset, head);
  46. case PERF_RECORD_COMM:
  47. return curr_handler->process_comm_event(event, offset, head);
  48. case PERF_RECORD_FORK:
  49. return curr_handler->process_fork_event(event, offset, head);
  50. case PERF_RECORD_EXIT:
  51. return curr_handler->process_exit_event(event, offset, head);
  52. case PERF_RECORD_LOST:
  53. return curr_handler->process_lost_event(event, offset, head);
  54. case PERF_RECORD_READ:
  55. return curr_handler->process_read_event(event, offset, head);
  56. case PERF_RECORD_THROTTLE:
  57. return curr_handler->process_throttle_event(event, offset, head);
  58. case PERF_RECORD_UNTHROTTLE:
  59. return curr_handler->process_unthrottle_event(event, offset, head);
  60. default:
  61. curr_handler->total_unknown++;
  62. return -1;
  63. }
  64. }
  65. int perf_header__read_build_ids(int input, off_t size)
  66. {
  67. struct build_id_event bev;
  68. char filename[PATH_MAX];
  69. off_t offset = lseek(input, 0, SEEK_CUR);
  70. off_t limit = offset + size;
  71. int err = -1;
  72. while (offset < limit) {
  73. struct dso *dso;
  74. ssize_t len;
  75. if (read(input, &bev, sizeof(bev)) != sizeof(bev))
  76. goto out;
  77. len = bev.header.size - sizeof(bev);
  78. if (read(input, filename, len) != len)
  79. goto out;
  80. dso = dsos__findnew(filename);
  81. if (dso != NULL)
  82. dso__set_build_id(dso, &bev.build_id);
  83. offset += bev.header.size;
  84. }
  85. err = 0;
  86. out:
  87. return err;
  88. }
  89. int mmap_dispatch_perf_file(struct perf_header **pheader,
  90. const char *input_name,
  91. int force,
  92. int full_paths,
  93. int *cwdlen,
  94. char **cwd)
  95. {
  96. int ret, rc = EXIT_FAILURE;
  97. struct perf_header *header;
  98. unsigned long head, shift;
  99. unsigned long offset = 0;
  100. struct stat input_stat;
  101. size_t page_size;
  102. u64 sample_type;
  103. event_t *event;
  104. uint32_t size;
  105. int input;
  106. char *buf;
  107. if (!curr_handler)
  108. die("Forgot to register perf file handler");
  109. page_size = getpagesize();
  110. input = open(input_name, O_RDONLY);
  111. if (input < 0) {
  112. fprintf(stderr, " failed to open file: %s", input_name);
  113. if (!strcmp(input_name, "perf.data"))
  114. fprintf(stderr, " (try 'perf record' first)");
  115. fprintf(stderr, "\n");
  116. exit(-1);
  117. }
  118. ret = fstat(input, &input_stat);
  119. if (ret < 0) {
  120. perror("failed to stat file");
  121. exit(-1);
  122. }
  123. if (!force && input_stat.st_uid && (input_stat.st_uid != geteuid())) {
  124. fprintf(stderr, "file: %s not owned by current user or root\n",
  125. input_name);
  126. exit(-1);
  127. }
  128. if (!input_stat.st_size) {
  129. fprintf(stderr, "zero-sized file, nothing to do!\n");
  130. exit(0);
  131. }
  132. *pheader = perf_header__read(input);
  133. header = *pheader;
  134. head = header->data_offset;
  135. sample_type = perf_header__sample_type(header);
  136. if (curr_handler->sample_type_check)
  137. if (curr_handler->sample_type_check(sample_type) < 0)
  138. exit(-1);
  139. if (load_kernel(NULL) < 0) {
  140. perror("failed to load kernel symbols");
  141. return EXIT_FAILURE;
  142. }
  143. if (!full_paths) {
  144. if (getcwd(__cwd, sizeof(__cwd)) == NULL) {
  145. perror("failed to get the current directory");
  146. return EXIT_FAILURE;
  147. }
  148. *cwd = __cwd;
  149. *cwdlen = strlen(*cwd);
  150. } else {
  151. *cwd = NULL;
  152. *cwdlen = 0;
  153. }
  154. shift = page_size * (head / page_size);
  155. offset += shift;
  156. head -= shift;
  157. remap:
  158. buf = (char *)mmap(NULL, page_size * mmap_window, PROT_READ,
  159. MAP_SHARED, input, offset);
  160. if (buf == MAP_FAILED) {
  161. perror("failed to mmap file");
  162. exit(-1);
  163. }
  164. more:
  165. event = (event_t *)(buf + head);
  166. size = event->header.size;
  167. if (!size)
  168. size = 8;
  169. if (head + event->header.size >= page_size * mmap_window) {
  170. int munmap_ret;
  171. shift = page_size * (head / page_size);
  172. munmap_ret = munmap(buf, page_size * mmap_window);
  173. assert(munmap_ret == 0);
  174. offset += shift;
  175. head -= shift;
  176. goto remap;
  177. }
  178. size = event->header.size;
  179. dump_printf("\n%p [%p]: event: %d\n",
  180. (void *)(offset + head),
  181. (void *)(long)event->header.size,
  182. event->header.type);
  183. if (!size || process_event(event, offset, head) < 0) {
  184. dump_printf("%p [%p]: skipping unknown header type: %d\n",
  185. (void *)(offset + head),
  186. (void *)(long)(event->header.size),
  187. event->header.type);
  188. /*
  189. * assume we lost track of the stream, check alignment, and
  190. * increment a single u64 in the hope to catch on again 'soon'.
  191. */
  192. if (unlikely(head & 7))
  193. head &= ~7ULL;
  194. size = 8;
  195. }
  196. head += size;
  197. if (offset + head >= header->data_offset + header->data_size)
  198. goto done;
  199. if (offset + head < (unsigned long)input_stat.st_size)
  200. goto more;
  201. done:
  202. rc = EXIT_SUCCESS;
  203. close(input);
  204. return rc;
  205. }