1// SPDX-License-Identifier: GPL-2.0
 2#include <test_progs.h>
 3
 4static void sigalrm_handler(int s) {}
 5static struct sigaction sigalrm_action = {
 6	.sa_handler = sigalrm_handler,
 7};
 8
 9static void test_signal_pending_by_type(enum bpf_prog_type prog_type)
10{
11	struct bpf_insn prog[4096];
12	struct itimerval timeo = {
13		.it_value.tv_usec = 100000, /* 100ms */
14	};
15	__u32 duration = 0, retval;
16	int prog_fd;
17	int err;
18	int i;
19
20	for (i = 0; i < ARRAY_SIZE(prog); i++)
21		prog[i] = BPF_ALU64_IMM(BPF_MOV, BPF_REG_0, 0);
22	prog[ARRAY_SIZE(prog) - 1] = BPF_EXIT_INSN();
23
24	prog_fd = bpf_load_program(prog_type, prog, ARRAY_SIZE(prog),
25				   "GPL", 0, NULL, 0);
26	CHECK(prog_fd < 0, "test-run", "errno %d\n", errno);
27
28	err = sigaction(SIGALRM, &sigalrm_action, NULL);
29	CHECK(err, "test-run-signal-sigaction", "errno %d\n", errno);
30
31	err = setitimer(ITIMER_REAL, &timeo, NULL);
32	CHECK(err, "test-run-signal-timer", "errno %d\n", errno);
33
34	err = bpf_prog_test_run(prog_fd, 0xffffffff, &pkt_v4, sizeof(pkt_v4),
35				NULL, NULL, &retval, &duration);
36	CHECK(duration > 500000000, /* 500ms */
37	      "test-run-signal-duration",
38	      "duration %dns > 500ms\n",
39	      duration);
40
41	signal(SIGALRM, SIG_DFL);
42}
43
44void test_signal_pending(enum bpf_prog_type prog_type)
45{
46	test_signal_pending_by_type(BPF_PROG_TYPE_SOCKET_FILTER);
47	test_signal_pending_by_type(BPF_PROG_TYPE_FLOW_DISSECTOR);
48}