1 // SPDX-License-Identifier: GPL-2.0 2 // Copyright (c) 2020 Facebook 3 #define _GNU_SOURCE 4 #include <pthread.h> 5 #include <sched.h> 6 #include <test_progs.h> 7 #include "perf_event_stackmap.skel.h" 8 9 #ifndef noinline 10 #define noinline __attribute__((noinline)) 11 #endif 12 func_1(void)13 noinline int func_1(void) 14 { 15 static int val = 1; 16 17 val += 1; 18 19 usleep(100); 20 return val; 21 } 22 func_2(void)23 noinline int func_2(void) 24 { 25 return func_1(); 26 } 27 func_3(void)28 noinline int func_3(void) 29 { 30 return func_2(); 31 } 32 func_4(void)33 noinline int func_4(void) 34 { 35 return func_3(); 36 } 37 func_5(void)38 noinline int func_5(void) 39 { 40 return func_4(); 41 } 42 func_6(void)43 noinline int func_6(void) 44 { 45 int i, val = 1; 46 47 for (i = 0; i < 100; i++) 48 val += func_5(); 49 50 return val; 51 } 52 test_perf_event_stackmap(void)53 void test_perf_event_stackmap(void) 54 { 55 struct perf_event_attr attr = { 56 /* .type = PERF_TYPE_SOFTWARE, */ 57 .type = PERF_TYPE_HARDWARE, 58 .config = PERF_COUNT_HW_CPU_CYCLES, 59 .precise_ip = 2, 60 .sample_type = PERF_SAMPLE_IP | PERF_SAMPLE_BRANCH_STACK | 61 PERF_SAMPLE_CALLCHAIN, 62 .branch_sample_type = PERF_SAMPLE_BRANCH_USER | 63 PERF_SAMPLE_BRANCH_NO_FLAGS | 64 PERF_SAMPLE_BRANCH_NO_CYCLES | 65 PERF_SAMPLE_BRANCH_CALL_STACK, 66 .freq = 1, 67 .sample_freq = read_perf_max_sample_freq(), 68 .size = sizeof(struct perf_event_attr), 69 }; 70 struct perf_event_stackmap *skel; 71 __u32 duration = 0; 72 cpu_set_t cpu_set; 73 int pmu_fd, err; 74 75 skel = perf_event_stackmap__open(); 76 77 if (CHECK(!skel, "skel_open", "skeleton open failed\n")) 78 return; 79 80 err = perf_event_stackmap__load(skel); 81 if (CHECK(err, "skel_load", "skeleton load failed: %d\n", err)) 82 goto cleanup; 83 84 CPU_ZERO(&cpu_set); 85 CPU_SET(0, &cpu_set); 86 err = pthread_setaffinity_np(pthread_self(), sizeof(cpu_set), &cpu_set); 87 if (CHECK(err, "set_affinity", "err %d, errno %d\n", err, errno)) 88 goto cleanup; 89 90 pmu_fd = syscall(__NR_perf_event_open, &attr, -1 /* pid */, 91 0 /* cpu 0 */, -1 /* group id */, 92 0 /* flags */); 93 if (pmu_fd < 0) { 94 printf("%s:SKIP:cpu doesn't support the event\n", __func__); 95 test__skip(); 96 goto cleanup; 97 } 98 99 skel->links.oncpu = bpf_program__attach_perf_event(skel->progs.oncpu, 100 pmu_fd); 101 if (!ASSERT_OK_PTR(skel->links.oncpu, "attach_perf_event")) { 102 close(pmu_fd); 103 goto cleanup; 104 } 105 106 /* create kernel and user stack traces for testing */ 107 func_6(); 108 109 CHECK(skel->data->stackid_kernel != 2, "get_stackid_kernel", "failed\n"); 110 CHECK(skel->data->stackid_user != 2, "get_stackid_user", "failed\n"); 111 CHECK(skel->data->stack_kernel != 2, "get_stack_kernel", "failed\n"); 112 CHECK(skel->data->stack_user != 2, "get_stack_user", "failed\n"); 113 114 cleanup: 115 perf_event_stackmap__destroy(skel); 116 } 117