[PATCHv2 bpf-next 3/6] selftests/bpf: Fix kprobe multi stacktrace_ips test

[Jiri Olsa]

We now include the attached function in the stack trace,
fixing the test accordingly.

Fixes: c9e208fa93cd ("selftests/bpf: Add stacktrace ips test for 
kprobe_multi/kretprobe_multi")
Signed-off-by: Jiri Olsa <jo...@kernel.org>
---
 .../selftests/bpf/prog_tests/stacktrace_ips.c | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/tools/testing/selftests/bpf/prog_tests/stacktrace_ips.c 
b/tools/testing/selftests/bpf/prog_tests/stacktrace_ips.c
index c9efdd2a5b18..c93718dafd9b 100644
--- a/tools/testing/selftests/bpf/prog_tests/stacktrace_ips.c
+++ b/tools/testing/selftests/bpf/prog_tests/stacktrace_ips.c
@@ -74,11 +74,20 @@ static void test_stacktrace_ips_kprobe_multi(bool retprobe)
 
        load_kallsyms();
 
-       check_stacktrace_ips(bpf_map__fd(skel->maps.stackmap), 
skel->bss->stack_key, 4,
-                            ksym_get_addr("bpf_testmod_stacktrace_test_3"),
-                            ksym_get_addr("bpf_testmod_stacktrace_test_2"),
-                            ksym_get_addr("bpf_testmod_stacktrace_test_1"),
-                            ksym_get_addr("bpf_testmod_test_read"));
+       if (retprobe) {
+               check_stacktrace_ips(bpf_map__fd(skel->maps.stackmap), 
skel->bss->stack_key, 4,
+                                    
ksym_get_addr("bpf_testmod_stacktrace_test_3"),
+                                    
ksym_get_addr("bpf_testmod_stacktrace_test_2"),
+                                    
ksym_get_addr("bpf_testmod_stacktrace_test_1"),
+                                    ksym_get_addr("bpf_testmod_test_read"));
+       } else {
+               check_stacktrace_ips(bpf_map__fd(skel->maps.stackmap), 
skel->bss->stack_key, 5,
+                                    
ksym_get_addr("bpf_testmod_stacktrace_test"),
+                                    
ksym_get_addr("bpf_testmod_stacktrace_test_3"),
+                                    
ksym_get_addr("bpf_testmod_stacktrace_test_2"),
+                                    
ksym_get_addr("bpf_testmod_stacktrace_test_1"),
+                                    ksym_get_addr("bpf_testmod_test_read"));
+       }
 
 cleanup:
        stacktrace_ips__destroy(skel);
-- 
2.52.0