Jens Axboe | e502435 | 2020-02-11 20:34:12 -0700 | [diff] [blame] | 1 | /* SPDX-License-Identifier: MIT */ |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 2 | /* |
| 3 | * Description: run various openat(2) tests |
| 4 | * |
| 5 | */ |
| 6 | #include <errno.h> |
| 7 | #include <stdio.h> |
| 8 | #include <unistd.h> |
| 9 | #include <stdlib.h> |
| 10 | #include <string.h> |
| 11 | #include <fcntl.h> |
| 12 | |
| 13 | #include "liburing.h" |
| 14 | |
| 15 | static int create_file(const char *file, size_t size) |
| 16 | { |
| 17 | ssize_t ret; |
| 18 | char *buf; |
| 19 | int fd; |
| 20 | |
| 21 | buf = malloc(size); |
| 22 | memset(buf, 0xaa, size); |
| 23 | |
| 24 | fd = open(file, O_WRONLY | O_CREAT, 0644); |
| 25 | if (fd < 0) { |
| 26 | perror("open file"); |
| 27 | return 1; |
| 28 | } |
| 29 | ret = write(fd, buf, size); |
| 30 | close(fd); |
| 31 | return ret != size; |
| 32 | } |
| 33 | |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 34 | static int test_openat2(struct io_uring *ring, const char *path, int dfd) |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 35 | { |
| 36 | struct io_uring_cqe *cqe; |
| 37 | struct io_uring_sqe *sqe; |
| 38 | struct open_how how; |
| 39 | int ret; |
| 40 | |
| 41 | sqe = io_uring_get_sqe(ring); |
| 42 | if (!sqe) { |
| 43 | fprintf(stderr, "get sqe failed\n"); |
| 44 | goto err; |
| 45 | } |
| 46 | memset(&how, 0, sizeof(how)); |
| 47 | how.flags = O_RDONLY; |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 48 | io_uring_prep_openat2(sqe, dfd, path, &how); |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 49 | |
| 50 | ret = io_uring_submit(ring); |
| 51 | if (ret <= 0) { |
| 52 | fprintf(stderr, "sqe submit failed: %d\n", ret); |
| 53 | goto err; |
| 54 | } |
| 55 | |
| 56 | ret = io_uring_wait_cqe(ring, &cqe); |
| 57 | if (ret < 0) { |
| 58 | fprintf(stderr, "wait completion %d\n", ret); |
| 59 | goto err; |
| 60 | } |
| 61 | ret = cqe->res; |
| 62 | io_uring_cqe_seen(ring, cqe); |
| 63 | return ret; |
| 64 | err: |
| 65 | return -1; |
| 66 | } |
| 67 | |
| 68 | int main(int argc, char *argv[]) |
| 69 | { |
| 70 | struct io_uring ring; |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 71 | const char *path, *path_rel; |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 72 | int ret, do_unlink; |
| 73 | |
| 74 | ret = io_uring_queue_init(8, &ring, 0); |
| 75 | if (ret) { |
| 76 | fprintf(stderr, "ring setup failed\n"); |
| 77 | return 1; |
| 78 | } |
| 79 | |
| 80 | if (argc > 1) { |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 81 | path = "/tmp/.open.close"; |
| 82 | path_rel = argv[1]; |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 83 | do_unlink = 0; |
| 84 | } else { |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 85 | path = "/tmp/.open.close"; |
| 86 | path_rel = ".open.close"; |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 87 | do_unlink = 1; |
| 88 | } |
| 89 | |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 90 | if (create_file(path, 4096)) { |
| 91 | fprintf(stderr, "file create failed\n"); |
| 92 | return 1; |
| 93 | } |
| 94 | if (do_unlink && create_file(path_rel, 4096)) { |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 95 | fprintf(stderr, "file create failed\n"); |
| 96 | return 1; |
| 97 | } |
| 98 | |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 99 | ret = test_openat2(&ring, path, -1); |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 100 | if (ret < 0) { |
| 101 | if (ret == -EINVAL) { |
| 102 | fprintf(stdout, "openat2 not supported, skipping\n"); |
| 103 | goto done; |
| 104 | } |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 105 | fprintf(stderr, "test_openat2 absolute failed: %d\n", ret); |
| 106 | goto err; |
| 107 | } |
| 108 | |
| 109 | ret = test_openat2(&ring, path_rel, AT_FDCWD); |
| 110 | if (ret < 0) { |
| 111 | fprintf(stderr, "test_openat2 relative failed: %d\n", ret); |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 112 | goto err; |
| 113 | } |
| 114 | |
| 115 | done: |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 116 | unlink(path); |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 117 | if (do_unlink) |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 118 | unlink(path_rel); |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 119 | return 0; |
| 120 | err: |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 121 | unlink(path); |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 122 | if (do_unlink) |
Jens Axboe | e93e341 | 2020-02-07 09:01:32 -0700 | [diff] [blame] | 123 | unlink(path_rel); |
Jens Axboe | 7772c81 | 2020-01-08 18:52:56 -0700 | [diff] [blame] | 124 | return 1; |
| 125 | } |