xref: /aosp_15_r20/external/liburing/test/accept-test.c (revision 25da2bea747f3a93b4c30fd9708b0618ef55a0e6)
1 /* SPDX-License-Identifier: MIT */
2 /*
3  * Description: Check to see if accept handles addr and addrlen
4  */
5 #include <stdio.h>
6 #include <errno.h>
7 #include <sys/socket.h>
8 #include <sys/un.h>
9 #include <assert.h>
10 #include "liburing.h"
11 
main(int argc,char * argv[])12 int main(int argc, char *argv[])
13 {
14 	struct io_uring_sqe *sqe;
15 	struct io_uring_cqe *cqe;
16 	struct io_uring ring;
17 	struct sockaddr_un addr;
18 	socklen_t addrlen = sizeof(addr);
19 	int ret, fd;
20 	struct __kernel_timespec ts = {
21 		.tv_sec = 0,
22 		.tv_nsec = 1000000
23 	};
24 
25 	if (argc > 1)
26 		return 0;
27 
28 	if (io_uring_queue_init(4, &ring, 0) != 0) {
29 		fprintf(stderr, "ring setup failed\n");
30 		return 1;
31 	}
32 
33 	fd = socket(AF_UNIX, SOCK_STREAM, 0);
34 	assert(fd != -1);
35 
36 	memset(&addr, 0, sizeof(addr));
37 	addr.sun_family = AF_UNIX;
38 	memcpy(addr.sun_path, "\0sock2", 7);
39 
40 	ret = bind(fd, (struct sockaddr *)&addr, addrlen);
41 	assert(ret != -1);
42 	ret = listen(fd, 128);
43 	assert(ret != -1);
44 
45 	sqe = io_uring_get_sqe(&ring);
46 	if (!sqe) {
47 		fprintf(stderr, "get sqe failed\n");
48 		return 1;
49 	}
50 	io_uring_prep_accept(sqe, fd, (struct sockaddr*)&addr, &addrlen, 0);
51 	sqe->user_data = 1;
52 
53 	ret = io_uring_submit(&ring);
54 	if (ret != 1) {
55 		fprintf(stderr, "Got submit %d, expected 1\n", ret);
56 		return 1;
57 	}
58 
59 	ret = io_uring_wait_cqe_timeout(&ring, &cqe, &ts);
60 	if (!ret) {
61 		if (cqe->res == -EBADF || cqe->res == -EINVAL) {
62 			fprintf(stdout, "Accept not supported, skipping\n");
63 			goto out;
64 		} else if (cqe->res < 0) {
65 			fprintf(stderr, "cqe error %d\n", cqe->res);
66 			goto err;
67 		}
68 	} else if (ret != -ETIME) {
69 		fprintf(stderr, "accept() failed to use addr & addrlen parameters!\n");
70 		return 1;
71 	}
72 
73 out:
74 	io_uring_queue_exit(&ring);
75 	return 0;
76 err:
77 	io_uring_queue_exit(&ring);
78 	return 1;
79 }
80