1*25da2beaSAndroid Build Coastguard Worker /* SPDX-License-Identifier: MIT */
2*25da2beaSAndroid Build Coastguard Worker /*
3*25da2beaSAndroid Build Coastguard Worker * Description: test that sigfd reading/polling works. A regression test for
4*25da2beaSAndroid Build Coastguard Worker * the upstream commit:
5*25da2beaSAndroid Build Coastguard Worker *
6*25da2beaSAndroid Build Coastguard Worker * fd7d6de22414 ("io_uring: don't recurse on tsk->sighand->siglock with signalfd")
7*25da2beaSAndroid Build Coastguard Worker */
8*25da2beaSAndroid Build Coastguard Worker #include <unistd.h>
9*25da2beaSAndroid Build Coastguard Worker #include <sys/signalfd.h>
10*25da2beaSAndroid Build Coastguard Worker #include <sys/epoll.h>
11*25da2beaSAndroid Build Coastguard Worker #include <poll.h>
12*25da2beaSAndroid Build Coastguard Worker #include <stdio.h>
13*25da2beaSAndroid Build Coastguard Worker #include "liburing.h"
14*25da2beaSAndroid Build Coastguard Worker
setup_signal(void)15*25da2beaSAndroid Build Coastguard Worker static int setup_signal(void)
16*25da2beaSAndroid Build Coastguard Worker {
17*25da2beaSAndroid Build Coastguard Worker sigset_t mask;
18*25da2beaSAndroid Build Coastguard Worker int sfd;
19*25da2beaSAndroid Build Coastguard Worker
20*25da2beaSAndroid Build Coastguard Worker sigemptyset(&mask);
21*25da2beaSAndroid Build Coastguard Worker sigaddset(&mask, SIGINT);
22*25da2beaSAndroid Build Coastguard Worker
23*25da2beaSAndroid Build Coastguard Worker sigprocmask(SIG_BLOCK, &mask, NULL);
24*25da2beaSAndroid Build Coastguard Worker sfd = signalfd(-1, &mask, SFD_NONBLOCK);
25*25da2beaSAndroid Build Coastguard Worker if (sfd < 0)
26*25da2beaSAndroid Build Coastguard Worker perror("signalfd");
27*25da2beaSAndroid Build Coastguard Worker return sfd;
28*25da2beaSAndroid Build Coastguard Worker }
29*25da2beaSAndroid Build Coastguard Worker
test_uring(int sfd)30*25da2beaSAndroid Build Coastguard Worker static int test_uring(int sfd)
31*25da2beaSAndroid Build Coastguard Worker {
32*25da2beaSAndroid Build Coastguard Worker struct io_uring_sqe *sqe;
33*25da2beaSAndroid Build Coastguard Worker struct io_uring_cqe *cqe;
34*25da2beaSAndroid Build Coastguard Worker struct io_uring ring;
35*25da2beaSAndroid Build Coastguard Worker int ret;
36*25da2beaSAndroid Build Coastguard Worker
37*25da2beaSAndroid Build Coastguard Worker io_uring_queue_init(32, &ring, 0);
38*25da2beaSAndroid Build Coastguard Worker
39*25da2beaSAndroid Build Coastguard Worker sqe = io_uring_get_sqe(&ring);
40*25da2beaSAndroid Build Coastguard Worker io_uring_prep_poll_add(sqe, sfd, POLLIN);
41*25da2beaSAndroid Build Coastguard Worker io_uring_submit(&ring);
42*25da2beaSAndroid Build Coastguard Worker
43*25da2beaSAndroid Build Coastguard Worker kill(getpid(), SIGINT);
44*25da2beaSAndroid Build Coastguard Worker
45*25da2beaSAndroid Build Coastguard Worker io_uring_wait_cqe(&ring, &cqe);
46*25da2beaSAndroid Build Coastguard Worker if (cqe->res & POLLIN) {
47*25da2beaSAndroid Build Coastguard Worker ret = 0;
48*25da2beaSAndroid Build Coastguard Worker } else {
49*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "Unexpected poll mask %x\n", cqe->res);
50*25da2beaSAndroid Build Coastguard Worker ret = 1;
51*25da2beaSAndroid Build Coastguard Worker }
52*25da2beaSAndroid Build Coastguard Worker io_uring_cqe_seen(&ring, cqe);
53*25da2beaSAndroid Build Coastguard Worker io_uring_queue_exit(&ring);
54*25da2beaSAndroid Build Coastguard Worker return ret;
55*25da2beaSAndroid Build Coastguard Worker }
56*25da2beaSAndroid Build Coastguard Worker
main(int argc,char * argv[])57*25da2beaSAndroid Build Coastguard Worker int main(int argc, char *argv[])
58*25da2beaSAndroid Build Coastguard Worker {
59*25da2beaSAndroid Build Coastguard Worker int sfd, ret;
60*25da2beaSAndroid Build Coastguard Worker
61*25da2beaSAndroid Build Coastguard Worker if (argc > 1)
62*25da2beaSAndroid Build Coastguard Worker return 0;
63*25da2beaSAndroid Build Coastguard Worker
64*25da2beaSAndroid Build Coastguard Worker sfd = setup_signal();
65*25da2beaSAndroid Build Coastguard Worker if (sfd < 0)
66*25da2beaSAndroid Build Coastguard Worker return 1;
67*25da2beaSAndroid Build Coastguard Worker
68*25da2beaSAndroid Build Coastguard Worker ret = test_uring(sfd);
69*25da2beaSAndroid Build Coastguard Worker if (ret)
70*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "test_uring signalfd failed\n");
71*25da2beaSAndroid Build Coastguard Worker
72*25da2beaSAndroid Build Coastguard Worker close(sfd);
73*25da2beaSAndroid Build Coastguard Worker return ret;
74*25da2beaSAndroid Build Coastguard Worker }
75