1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2021 Facebook */
3 #include <test_progs.h>
4 #include "timer.skel.h"
5 #include "timer_failure.skel.h"
6
7 #define NUM_THR 8
8
spin_lock_thread(void * arg)9 static void *spin_lock_thread(void *arg)
10 {
11 int i, err, prog_fd = *(int *)arg;
12 LIBBPF_OPTS(bpf_test_run_opts, topts);
13
14 for (i = 0; i < 10000; i++) {
15 err = bpf_prog_test_run_opts(prog_fd, &topts);
16 if (!ASSERT_OK(err, "test_run_opts err") ||
17 !ASSERT_OK(topts.retval, "test_run_opts retval"))
18 break;
19 }
20
21 pthread_exit(arg);
22 }
23
timer(struct timer * timer_skel)24 static int timer(struct timer *timer_skel)
25 {
26 int i, err, prog_fd;
27 LIBBPF_OPTS(bpf_test_run_opts, topts);
28 pthread_t thread_id[NUM_THR];
29 void *ret;
30
31 err = timer__attach(timer_skel);
32 if (!ASSERT_OK(err, "timer_attach"))
33 return err;
34
35 ASSERT_EQ(timer_skel->data->callback_check, 52, "callback_check1");
36 ASSERT_EQ(timer_skel->data->callback2_check, 52, "callback2_check1");
37 ASSERT_EQ(timer_skel->bss->pinned_callback_check, 0, "pinned_callback_check1");
38
39 prog_fd = bpf_program__fd(timer_skel->progs.test1);
40 err = bpf_prog_test_run_opts(prog_fd, &topts);
41 ASSERT_OK(err, "test_run");
42 ASSERT_EQ(topts.retval, 0, "test_run");
43 timer__detach(timer_skel);
44
45 usleep(50); /* 10 usecs should be enough, but give it extra */
46 /* check that timer_cb1() was executed 10+10 times */
47 ASSERT_EQ(timer_skel->data->callback_check, 42, "callback_check2");
48 ASSERT_EQ(timer_skel->data->callback2_check, 42, "callback2_check2");
49
50 /* check that timer_cb2() was executed twice */
51 ASSERT_EQ(timer_skel->bss->bss_data, 10, "bss_data");
52
53 /* check that timer_cb3() was executed twice */
54 ASSERT_EQ(timer_skel->bss->abs_data, 12, "abs_data");
55
56 /* check that timer_cb_pinned() was executed twice */
57 ASSERT_EQ(timer_skel->bss->pinned_callback_check, 2, "pinned_callback_check");
58
59 /* check that there were no errors in timer execution */
60 ASSERT_EQ(timer_skel->bss->err, 0, "err");
61
62 /* check that code paths completed */
63 ASSERT_EQ(timer_skel->bss->ok, 1 | 2 | 4, "ok");
64
65 prog_fd = bpf_program__fd(timer_skel->progs.race);
66 for (i = 0; i < NUM_THR; i++) {
67 err = pthread_create(&thread_id[i], NULL,
68 &spin_lock_thread, &prog_fd);
69 if (!ASSERT_OK(err, "pthread_create"))
70 break;
71 }
72
73 while (i) {
74 err = pthread_join(thread_id[--i], &ret);
75 if (ASSERT_OK(err, "pthread_join"))
76 ASSERT_EQ(ret, (void *)&prog_fd, "pthread_join");
77 }
78
79 return 0;
80 }
81
82 /* TODO: use pid filtering */
serial_test_timer(void)83 void serial_test_timer(void)
84 {
85 struct timer *timer_skel = NULL;
86 int err;
87
88 timer_skel = timer__open_and_load();
89 if (!ASSERT_OK_PTR(timer_skel, "timer_skel_load"))
90 return;
91
92 err = timer(timer_skel);
93 ASSERT_OK(err, "timer");
94 timer__destroy(timer_skel);
95
96 RUN_TESTS(timer_failure);
97 }
98