1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Description: test io_uring mkdirat handling
4 */
5 #include <fcntl.h>
6 #include <stdio.h>
7 #include <string.h>
8 #include <sys/stat.h>
9 #include <sys/types.h>
10 #include <unistd.h>
11
12 #include "liburing.h"
13 #include "helpers.h"
14
do_mkdirat(struct io_uring * ring,const char * fn)15 static int do_mkdirat(struct io_uring *ring, const char *fn)
16 {
17 int ret;
18 struct io_uring_sqe *sqe;
19 struct io_uring_cqe *cqe;
20
21 sqe = io_uring_get_sqe(ring);
22 if (!sqe) {
23 fprintf(stderr, "sqe get failed\n");
24 goto err;
25 }
26 io_uring_prep_mkdirat(sqe, AT_FDCWD, fn, 0700);
27
28 ret = io_uring_submit(ring);
29 if (ret != 1) {
30 fprintf(stderr, "submit failed: %d\n", ret);
31 goto err;
32 }
33
34 ret = io_uring_wait_cqes(ring, &cqe, 1, 0, 0);
35 if (ret) {
36 fprintf(stderr, "wait_cqe failed: %d\n", ret);
37 goto err;
38 }
39 ret = cqe->res;
40 io_uring_cqe_seen(ring, cqe);
41 return ret;
42 err:
43 return 1;
44 }
45
stat_file(const char * fn)46 static int stat_file(const char *fn)
47 {
48 struct stat sb;
49
50 if (!stat(fn, &sb))
51 return 0;
52
53 return errno;
54 }
55
main(int argc,char * argv[])56 int main(int argc, char *argv[])
57 {
58 static const char fn[] = "io_uring-mkdirat-test";
59 int ret;
60 struct io_uring ring;
61
62 if (argc > 1)
63 return T_EXIT_SKIP;
64
65 ret = io_uring_queue_init(8, &ring, 0);
66 if (ret) {
67 fprintf(stderr, "queue init failed: %d\n", ret);
68 return ret;
69 }
70
71 ret = do_mkdirat(&ring, fn);
72 if (ret < 0) {
73 if (ret == -EBADF || ret == -EINVAL) {
74 fprintf(stdout, "mkdirat not supported, skipping\n");
75 goto skip;
76 }
77 fprintf(stderr, "mkdirat: %s\n", strerror(-ret));
78 goto err;
79 } else if (ret) {
80 goto err;
81 }
82
83 if (stat_file(fn)) {
84 perror("stat");
85 goto err;
86 }
87
88 ret = do_mkdirat(&ring, fn);
89 if (ret != -EEXIST) {
90 fprintf(stderr, "do_mkdirat already exists failed: %d\n", ret);
91 goto err1;
92 }
93
94 ret = do_mkdirat(&ring, "surely/this/wont/exist");
95 if (ret != -ENOENT) {
96 fprintf(stderr, "do_mkdirat no parent failed: %d\n", ret);
97 goto err1;
98 }
99
100 ret = do_mkdirat(&ring, (const char *) (uintptr_t) 0x1234);
101 if (ret != -EFAULT) {
102 fprintf(stderr, "do_mkdirat bad address: %d\n", ret);
103 goto err1;
104 }
105
106 unlinkat(AT_FDCWD, fn, AT_REMOVEDIR);
107 io_uring_queue_exit(&ring);
108 return T_EXIT_PASS;
109 skip:
110 unlinkat(AT_FDCWD, fn, AT_REMOVEDIR);
111 io_uring_queue_exit(&ring);
112 return T_EXIT_SKIP;
113 err1:
114 unlinkat(AT_FDCWD, fn, AT_REMOVEDIR);
115 err:
116 io_uring_queue_exit(&ring);
117 return T_EXIT_FAIL;
118 }
119