| #include <inttypes.h> |
| #include <unistd.h> |
| #include <sys/syscall.h> |
| #include <sys/types.h> |
| #include <sys/mman.h> |
| #include <pthread.h> |
| #include <stdlib.h> |
| #include <stdio.h> |
| #include "debug.h" |
| #include "tests.h" |
| #include "machine.h" |
| #include "thread_map.h" |
| #include "symbol.h" |
| #include "thread.h" |
| |
| #define THREADS 4 |
| |
| static int go_away; |
| |
| struct thread_data { |
| pthread_t pt; |
| pid_t tid; |
| void *map; |
| int ready[2]; |
| }; |
| |
| static struct thread_data threads[THREADS]; |
| |
| static int thread_init(struct thread_data *td) |
| { |
| void *map; |
| |
| map = mmap(NULL, page_size, |
| PROT_READ|PROT_WRITE|PROT_EXEC, |
| MAP_SHARED|MAP_ANONYMOUS, -1, 0); |
| |
| if (map == MAP_FAILED) { |
| perror("mmap failed"); |
| return -1; |
| } |
| |
| td->map = map; |
| td->tid = syscall(SYS_gettid); |
| |
| pr_debug("tid = %d, map = %p\n", td->tid, map); |
| return 0; |
| } |
| |
| static void *thread_fn(void *arg) |
| { |
| struct thread_data *td = arg; |
| ssize_t ret; |
| int go; |
| |
| if (thread_init(td)) |
| return NULL; |
| |
| /* Signal thread_create thread is initialized. */ |
| ret = write(td->ready[1], &go, sizeof(int)); |
| if (ret != sizeof(int)) { |
| pr_err("failed to notify\n"); |
| return NULL; |
| } |
| |
| while (!go_away) { |
| /* Waiting for main thread to kill us. */ |
| usleep(100); |
| } |
| |
| munmap(td->map, page_size); |
| return NULL; |
| } |
| |
| static int thread_create(int i) |
| { |
| struct thread_data *td = &threads[i]; |
| int err, go; |
| |
| if (pipe(td->ready)) |
| return -1; |
| |
| err = pthread_create(&td->pt, NULL, thread_fn, td); |
| if (!err) { |
| /* Wait for thread initialization. */ |
| ssize_t ret = read(td->ready[0], &go, sizeof(int)); |
| err = ret != sizeof(int); |
| } |
| |
| close(td->ready[0]); |
| close(td->ready[1]); |
| return err; |
| } |
| |
| static int threads_create(void) |
| { |
| struct thread_data *td0 = &threads[0]; |
| int i, err = 0; |
| |
| go_away = 0; |
| |
| /* 0 is main thread */ |
| if (thread_init(td0)) |
| return -1; |
| |
| for (i = 1; !err && i < THREADS; i++) |
| err = thread_create(i); |
| |
| return err; |
| } |
| |
| static int threads_destroy(void) |
| { |
| struct thread_data *td0 = &threads[0]; |
| int i, err = 0; |
| |
| /* cleanup the main thread */ |
| munmap(td0->map, page_size); |
| |
| go_away = 1; |
| |
| for (i = 1; !err && i < THREADS; i++) |
| err = pthread_join(threads[i].pt, NULL); |
| |
| return err; |
| } |
| |
| typedef int (*synth_cb)(struct machine *machine); |
| |
| static int synth_all(struct machine *machine) |
| { |
| return perf_event__synthesize_threads(NULL, |
| perf_event__process, |
| machine, 0, 500); |
| } |
| |
| static int synth_process(struct machine *machine) |
| { |
| struct thread_map *map; |
| int err; |
| |
| map = thread_map__new_by_pid(getpid()); |
| |
| err = perf_event__synthesize_thread_map(NULL, map, |
| perf_event__process, |
| machine, 0, 500); |
| |
| thread_map__put(map); |
| return err; |
| } |
| |
| static int mmap_events(synth_cb synth) |
| { |
| struct machine *machine; |
| int err, i; |
| |
| /* |
| * The threads_create will not return before all threads |
| * are spawned and all created memory map. |
| * |
| * They will loop until threads_destroy is called, so we |
| * can safely run synthesizing function. |
| */ |
| TEST_ASSERT_VAL("failed to create threads", !threads_create()); |
| |
| machine = machine__new_host(); |
| |
| dump_trace = verbose > 1 ? 1 : 0; |
| |
| err = synth(machine); |
| |
| dump_trace = 0; |
| |
| TEST_ASSERT_VAL("failed to destroy threads", !threads_destroy()); |
| TEST_ASSERT_VAL("failed to synthesize maps", !err); |
| |
| /* |
| * All data is synthesized, try to find map for each |
| * thread object. |
| */ |
| for (i = 0; i < THREADS; i++) { |
| struct thread_data *td = &threads[i]; |
| struct addr_location al; |
| struct thread *thread; |
| |
| thread = machine__findnew_thread(machine, getpid(), td->tid); |
| |
| pr_debug("looking for map %p\n", td->map); |
| |
| thread__find_addr_map(thread, |
| PERF_RECORD_MISC_USER, MAP__FUNCTION, |
| (unsigned long) (td->map + 1), &al); |
| |
| thread__put(thread); |
| |
| if (!al.map) { |
| pr_debug("failed, couldn't find map\n"); |
| err = -1; |
| break; |
| } |
| |
| pr_debug("map %p, addr %" PRIx64 "\n", al.map, al.map->start); |
| } |
| |
| machine__delete_threads(machine); |
| machine__delete(machine); |
| return err; |
| } |
| |
| /* |
| * This test creates 'THREADS' number of threads (including |
| * main thread) and each thread creates memory map. |
| * |
| * When threads are created, we synthesize them with both |
| * (separate tests): |
| * perf_event__synthesize_thread_map (process based) |
| * perf_event__synthesize_threads (global) |
| * |
| * We test we can find all memory maps via: |
| * thread__find_addr_map |
| * |
| * by using all thread objects. |
| */ |
| int test__mmap_thread_lookup(int subtest __maybe_unused) |
| { |
| /* perf_event__synthesize_threads synthesize */ |
| TEST_ASSERT_VAL("failed with sythesizing all", |
| !mmap_events(synth_all)); |
| |
| /* perf_event__synthesize_thread_map synthesize */ |
| TEST_ASSERT_VAL("failed with sythesizing process", |
| !mmap_events(synth_process)); |
| |
| return 0; |
| } |