Jens Axboe | e502435 | 2020-02-11 20:34:12 -0700 | [diff] [blame] | 1 | /* SPDX-License-Identifier: MIT */ |
Hrvoje Zeba | ba88810 | 2019-11-25 12:28:57 -0700 | [diff] [blame] | 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 | |
| 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 | |
Jens Axboe | a2141fc | 2020-05-19 17:36:19 -0600 | [diff] [blame] | 25 | if (argc > 1) |
| 26 | return 0; |
| 27 | |
Hrvoje Zeba | ba88810 | 2019-11-25 12:28:57 -0700 | [diff] [blame] | 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, "\0sock", 6); |
| 39 | |
Ryan Sharpelletti | e3adbfc | 2020-10-27 21:04:11 +0000 | [diff] [blame] | 40 | ret = bind(fd, (struct sockaddr *)&addr, addrlen); |
| 41 | assert(ret != -1); |
| 42 | ret = listen(fd, 128); |
| 43 | assert(ret != -1); |
Hrvoje Zeba | ba88810 | 2019-11-25 12:28:57 -0700 | [diff] [blame] | 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); |
Jens Axboe | 007841f | 2020-10-07 09:04:08 -0600 | [diff] [blame] | 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) { |
Hrvoje Zeba | ba88810 | 2019-11-25 12:28:57 -0700 | [diff] [blame] | 69 | fprintf(stderr, "accept() failed to use addr & addrlen parameters!\n"); |
| 70 | return 1; |
| 71 | } |
| 72 | |
Jens Axboe | 007841f | 2020-10-07 09:04:08 -0600 | [diff] [blame] | 73 | out: |
Hrvoje Zeba | ba88810 | 2019-11-25 12:28:57 -0700 | [diff] [blame] | 74 | io_uring_queue_exit(&ring); |
| 75 | return 0; |
Jens Axboe | 007841f | 2020-10-07 09:04:08 -0600 | [diff] [blame] | 76 | err: |
| 77 | io_uring_queue_exit(&ring); |
| 78 | return 1; |
Hrvoje Zeba | ba88810 | 2019-11-25 12:28:57 -0700 | [diff] [blame] | 79 | } |