iou 0.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (297) hide show
  1. checksums.yaml +7 -0
  2. data/.github/dependabot.yml +12 -0
  3. data/.gitignore +59 -0
  4. data/.gitmodules +3 -0
  5. data/Gemfile +3 -0
  6. data/LICENSE +21 -0
  7. data/README.md +106 -0
  8. data/Rakefile +39 -0
  9. data/TODO.md +4 -0
  10. data/examples/echo_server.rb +52 -0
  11. data/examples/event_loop.rb +69 -0
  12. data/examples/http_server.rb +56 -0
  13. data/examples/http_server_multishot.rb +59 -0
  14. data/ext/iou/extconf.rb +71 -0
  15. data/ext/iou/iou.c +729 -0
  16. data/ext/iou/iou.h +66 -0
  17. data/ext/iou/iou_ext.c +9 -0
  18. data/ext/iou/op_spec_data.c +61 -0
  19. data/iou.gemspec +27 -0
  20. data/lib/iou/version.rb +5 -0
  21. data/lib/iou.rb +3 -0
  22. data/test/helper.rb +59 -0
  23. data/test/test_iou.rb +794 -0
  24. data/vendor/liburing/.github/actions/codespell/stopwords +7 -0
  25. data/vendor/liburing/.github/pull_request_template.md +86 -0
  26. data/vendor/liburing/.github/workflows/build.yml +137 -0
  27. data/vendor/liburing/.github/workflows/codespell.yml +25 -0
  28. data/vendor/liburing/.github/workflows/shellcheck.yml +20 -0
  29. data/vendor/liburing/.gitignore +41 -0
  30. data/vendor/liburing/CHANGELOG +111 -0
  31. data/vendor/liburing/CITATION.cff +11 -0
  32. data/vendor/liburing/COPYING +502 -0
  33. data/vendor/liburing/COPYING.GPL +339 -0
  34. data/vendor/liburing/LICENSE +20 -0
  35. data/vendor/liburing/Makefile +96 -0
  36. data/vendor/liburing/Makefile.common +7 -0
  37. data/vendor/liburing/Makefile.quiet +11 -0
  38. data/vendor/liburing/README +106 -0
  39. data/vendor/liburing/SECURITY.md +6 -0
  40. data/vendor/liburing/configure +624 -0
  41. data/vendor/liburing/debian/README.Debian +7 -0
  42. data/vendor/liburing/debian/changelog +38 -0
  43. data/vendor/liburing/debian/control +39 -0
  44. data/vendor/liburing/debian/copyright +49 -0
  45. data/vendor/liburing/debian/liburing-dev.install +4 -0
  46. data/vendor/liburing/debian/liburing-dev.manpages +5 -0
  47. data/vendor/liburing/debian/liburing2.install +1 -0
  48. data/vendor/liburing/debian/liburing2.symbols +56 -0
  49. data/vendor/liburing/debian/patches/series +1 -0
  50. data/vendor/liburing/debian/rules +29 -0
  51. data/vendor/liburing/debian/source/format +1 -0
  52. data/vendor/liburing/debian/source/local-options +2 -0
  53. data/vendor/liburing/debian/source/options +1 -0
  54. data/vendor/liburing/debian/watch +3 -0
  55. data/vendor/liburing/examples/Makefile +53 -0
  56. data/vendor/liburing/examples/helpers.c +62 -0
  57. data/vendor/liburing/examples/helpers.h +7 -0
  58. data/vendor/liburing/examples/io_uring-close-test.c +123 -0
  59. data/vendor/liburing/examples/io_uring-cp.c +282 -0
  60. data/vendor/liburing/examples/io_uring-test.c +112 -0
  61. data/vendor/liburing/examples/io_uring-udp.c +403 -0
  62. data/vendor/liburing/examples/link-cp.c +193 -0
  63. data/vendor/liburing/examples/napi-busy-poll-client.c +509 -0
  64. data/vendor/liburing/examples/napi-busy-poll-server.c +450 -0
  65. data/vendor/liburing/examples/poll-bench.c +101 -0
  66. data/vendor/liburing/examples/proxy.c +2461 -0
  67. data/vendor/liburing/examples/proxy.h +102 -0
  68. data/vendor/liburing/examples/rsrc-update-bench.c +100 -0
  69. data/vendor/liburing/examples/send-zerocopy.c +658 -0
  70. data/vendor/liburing/examples/ucontext-cp.c +258 -0
  71. data/vendor/liburing/liburing-ffi.pc.in +12 -0
  72. data/vendor/liburing/liburing.pc.in +12 -0
  73. data/vendor/liburing/liburing.spec +66 -0
  74. data/vendor/liburing/make-debs.sh +55 -0
  75. data/vendor/liburing/src/Makefile +129 -0
  76. data/vendor/liburing/src/arch/aarch64/lib.h +47 -0
  77. data/vendor/liburing/src/arch/aarch64/syscall.h +91 -0
  78. data/vendor/liburing/src/arch/generic/lib.h +17 -0
  79. data/vendor/liburing/src/arch/generic/syscall.h +100 -0
  80. data/vendor/liburing/src/arch/riscv64/lib.h +48 -0
  81. data/vendor/liburing/src/arch/riscv64/syscall.h +100 -0
  82. data/vendor/liburing/src/arch/syscall-defs.h +94 -0
  83. data/vendor/liburing/src/arch/x86/lib.h +11 -0
  84. data/vendor/liburing/src/arch/x86/syscall.h +296 -0
  85. data/vendor/liburing/src/ffi.c +15 -0
  86. data/vendor/liburing/src/include/liburing/barrier.h +81 -0
  87. data/vendor/liburing/src/include/liburing/io_uring.h +818 -0
  88. data/vendor/liburing/src/include/liburing.h +1602 -0
  89. data/vendor/liburing/src/int_flags.h +11 -0
  90. data/vendor/liburing/src/lib.h +52 -0
  91. data/vendor/liburing/src/liburing-ffi.map +211 -0
  92. data/vendor/liburing/src/liburing.map +104 -0
  93. data/vendor/liburing/src/nolibc.c +55 -0
  94. data/vendor/liburing/src/queue.c +468 -0
  95. data/vendor/liburing/src/register.c +374 -0
  96. data/vendor/liburing/src/setup.c +689 -0
  97. data/vendor/liburing/src/setup.h +9 -0
  98. data/vendor/liburing/src/syscall.c +29 -0
  99. data/vendor/liburing/src/syscall.h +53 -0
  100. data/vendor/liburing/src/version.c +21 -0
  101. data/vendor/liburing/test/232c93d07b74.c +305 -0
  102. data/vendor/liburing/test/35fa71a030ca.c +329 -0
  103. data/vendor/liburing/test/500f9fbadef8.c +91 -0
  104. data/vendor/liburing/test/7ad0e4b2f83c.c +94 -0
  105. data/vendor/liburing/test/8a9973408177.c +107 -0
  106. data/vendor/liburing/test/917257daa0fe.c +54 -0
  107. data/vendor/liburing/test/Makefile +297 -0
  108. data/vendor/liburing/test/a0908ae19763.c +59 -0
  109. data/vendor/liburing/test/a4c0b3decb33.c +181 -0
  110. data/vendor/liburing/test/accept-link.c +255 -0
  111. data/vendor/liburing/test/accept-non-empty.c +256 -0
  112. data/vendor/liburing/test/accept-reuse.c +163 -0
  113. data/vendor/liburing/test/accept-test.c +83 -0
  114. data/vendor/liburing/test/accept.c +919 -0
  115. data/vendor/liburing/test/across-fork.c +284 -0
  116. data/vendor/liburing/test/b19062a56726.c +54 -0
  117. data/vendor/liburing/test/b5837bd5311d.c +78 -0
  118. data/vendor/liburing/test/bind-listen.c +408 -0
  119. data/vendor/liburing/test/buf-ring-nommap.c +123 -0
  120. data/vendor/liburing/test/buf-ring-put.c +83 -0
  121. data/vendor/liburing/test/buf-ring.c +473 -0
  122. data/vendor/liburing/test/ce593a6c480a.c +139 -0
  123. data/vendor/liburing/test/close-opath.c +123 -0
  124. data/vendor/liburing/test/config +14 -0
  125. data/vendor/liburing/test/connect-rep.c +204 -0
  126. data/vendor/liburing/test/connect.c +442 -0
  127. data/vendor/liburing/test/coredump.c +60 -0
  128. data/vendor/liburing/test/cq-full.c +97 -0
  129. data/vendor/liburing/test/cq-overflow.c +530 -0
  130. data/vendor/liburing/test/cq-peek-batch.c +103 -0
  131. data/vendor/liburing/test/cq-ready.c +95 -0
  132. data/vendor/liburing/test/cq-size.c +65 -0
  133. data/vendor/liburing/test/d4ae271dfaae.c +96 -0
  134. data/vendor/liburing/test/d77a67ed5f27.c +65 -0
  135. data/vendor/liburing/test/defer-taskrun.c +391 -0
  136. data/vendor/liburing/test/defer-tw-timeout.c +173 -0
  137. data/vendor/liburing/test/defer.c +319 -0
  138. data/vendor/liburing/test/double-poll-crash.c +195 -0
  139. data/vendor/liburing/test/drop-submit.c +94 -0
  140. data/vendor/liburing/test/eeed8b54e0df.c +120 -0
  141. data/vendor/liburing/test/empty-eownerdead.c +45 -0
  142. data/vendor/liburing/test/eploop.c +74 -0
  143. data/vendor/liburing/test/eventfd-disable.c +179 -0
  144. data/vendor/liburing/test/eventfd-reg.c +77 -0
  145. data/vendor/liburing/test/eventfd-ring.c +98 -0
  146. data/vendor/liburing/test/eventfd.c +113 -0
  147. data/vendor/liburing/test/evloop.c +73 -0
  148. data/vendor/liburing/test/exec-target.c +6 -0
  149. data/vendor/liburing/test/exit-no-cleanup.c +117 -0
  150. data/vendor/liburing/test/fadvise.c +202 -0
  151. data/vendor/liburing/test/fallocate.c +265 -0
  152. data/vendor/liburing/test/fc2a85cb02ef.c +132 -0
  153. data/vendor/liburing/test/fd-install.c +500 -0
  154. data/vendor/liburing/test/fd-pass.c +237 -0
  155. data/vendor/liburing/test/fdinfo.c +419 -0
  156. data/vendor/liburing/test/file-register.c +1189 -0
  157. data/vendor/liburing/test/file-update.c +231 -0
  158. data/vendor/liburing/test/file-verify.c +654 -0
  159. data/vendor/liburing/test/files-exit-hang-poll.c +114 -0
  160. data/vendor/liburing/test/files-exit-hang-timeout.c +137 -0
  161. data/vendor/liburing/test/fixed-buf-iter.c +115 -0
  162. data/vendor/liburing/test/fixed-buf-merge.c +101 -0
  163. data/vendor/liburing/test/fixed-hugepage.c +411 -0
  164. data/vendor/liburing/test/fixed-link.c +90 -0
  165. data/vendor/liburing/test/fixed-reuse.c +160 -0
  166. data/vendor/liburing/test/fpos.c +255 -0
  167. data/vendor/liburing/test/fsnotify.c +118 -0
  168. data/vendor/liburing/test/fsync.c +224 -0
  169. data/vendor/liburing/test/futex.c +571 -0
  170. data/vendor/liburing/test/hardlink.c +170 -0
  171. data/vendor/liburing/test/helpers.c +318 -0
  172. data/vendor/liburing/test/helpers.h +108 -0
  173. data/vendor/liburing/test/ignore-single-mmap.c +48 -0
  174. data/vendor/liburing/test/init-mem.c +164 -0
  175. data/vendor/liburing/test/io-cancel.c +561 -0
  176. data/vendor/liburing/test/io_uring_enter.c +264 -0
  177. data/vendor/liburing/test/io_uring_passthrough.c +482 -0
  178. data/vendor/liburing/test/io_uring_register.c +503 -0
  179. data/vendor/liburing/test/io_uring_setup.c +110 -0
  180. data/vendor/liburing/test/iopoll-leak.c +85 -0
  181. data/vendor/liburing/test/iopoll-overflow.c +118 -0
  182. data/vendor/liburing/test/iopoll.c +465 -0
  183. data/vendor/liburing/test/lfs-openat-write.c +119 -0
  184. data/vendor/liburing/test/lfs-openat.c +273 -0
  185. data/vendor/liburing/test/link-timeout.c +1108 -0
  186. data/vendor/liburing/test/link.c +497 -0
  187. data/vendor/liburing/test/link_drain.c +255 -0
  188. data/vendor/liburing/test/madvise.c +195 -0
  189. data/vendor/liburing/test/min-timeout-wait.c +354 -0
  190. data/vendor/liburing/test/min-timeout.c +233 -0
  191. data/vendor/liburing/test/mkdir.c +112 -0
  192. data/vendor/liburing/test/msg-ring-fd.c +331 -0
  193. data/vendor/liburing/test/msg-ring-flags.c +212 -0
  194. data/vendor/liburing/test/msg-ring-overflow.c +159 -0
  195. data/vendor/liburing/test/msg-ring.c +467 -0
  196. data/vendor/liburing/test/multicqes_drain.c +429 -0
  197. data/vendor/liburing/test/napi-test.c +215 -0
  198. data/vendor/liburing/test/napi-test.sh +48 -0
  199. data/vendor/liburing/test/no-mmap-inval.c +42 -0
  200. data/vendor/liburing/test/nolibc.c +62 -0
  201. data/vendor/liburing/test/nop-all-sizes.c +99 -0
  202. data/vendor/liburing/test/nop.c +177 -0
  203. data/vendor/liburing/test/nvme.h +169 -0
  204. data/vendor/liburing/test/ooo-file-unreg.c +82 -0
  205. data/vendor/liburing/test/open-close.c +261 -0
  206. data/vendor/liburing/test/open-direct-link.c +188 -0
  207. data/vendor/liburing/test/open-direct-pick.c +180 -0
  208. data/vendor/liburing/test/openat2.c +312 -0
  209. data/vendor/liburing/test/personality.c +204 -0
  210. data/vendor/liburing/test/pipe-bug.c +95 -0
  211. data/vendor/liburing/test/pipe-eof.c +83 -0
  212. data/vendor/liburing/test/pipe-reuse.c +105 -0
  213. data/vendor/liburing/test/poll-cancel-all.c +496 -0
  214. data/vendor/liburing/test/poll-cancel-ton.c +135 -0
  215. data/vendor/liburing/test/poll-cancel.c +228 -0
  216. data/vendor/liburing/test/poll-link.c +221 -0
  217. data/vendor/liburing/test/poll-many.c +230 -0
  218. data/vendor/liburing/test/poll-mshot-overflow.c +265 -0
  219. data/vendor/liburing/test/poll-mshot-update.c +323 -0
  220. data/vendor/liburing/test/poll-race-mshot.c +276 -0
  221. data/vendor/liburing/test/poll-race.c +105 -0
  222. data/vendor/liburing/test/poll-ring.c +48 -0
  223. data/vendor/liburing/test/poll-v-poll.c +353 -0
  224. data/vendor/liburing/test/poll.c +327 -0
  225. data/vendor/liburing/test/probe.c +135 -0
  226. data/vendor/liburing/test/read-before-exit.c +129 -0
  227. data/vendor/liburing/test/read-mshot-empty.c +153 -0
  228. data/vendor/liburing/test/read-mshot.c +404 -0
  229. data/vendor/liburing/test/read-write.c +1013 -0
  230. data/vendor/liburing/test/recv-msgall-stream.c +398 -0
  231. data/vendor/liburing/test/recv-msgall.c +263 -0
  232. data/vendor/liburing/test/recv-multishot.c +602 -0
  233. data/vendor/liburing/test/recvsend_bundle.c +691 -0
  234. data/vendor/liburing/test/reg-fd-only.c +131 -0
  235. data/vendor/liburing/test/reg-hint.c +56 -0
  236. data/vendor/liburing/test/reg-reg-ring.c +90 -0
  237. data/vendor/liburing/test/regbuf-merge.c +91 -0
  238. data/vendor/liburing/test/register-restrictions.c +633 -0
  239. data/vendor/liburing/test/rename.c +132 -0
  240. data/vendor/liburing/test/ring-leak.c +283 -0
  241. data/vendor/liburing/test/ring-leak2.c +249 -0
  242. data/vendor/liburing/test/ringbuf-read.c +196 -0
  243. data/vendor/liburing/test/ringbuf-status.c +242 -0
  244. data/vendor/liburing/test/rsrc_tags.c +461 -0
  245. data/vendor/liburing/test/runtests-loop.sh +16 -0
  246. data/vendor/liburing/test/runtests-quiet.sh +11 -0
  247. data/vendor/liburing/test/runtests.sh +168 -0
  248. data/vendor/liburing/test/rw_merge_test.c +98 -0
  249. data/vendor/liburing/test/self.c +91 -0
  250. data/vendor/liburing/test/send-zerocopy.c +971 -0
  251. data/vendor/liburing/test/send_recv.c +412 -0
  252. data/vendor/liburing/test/send_recvmsg.c +444 -0
  253. data/vendor/liburing/test/shared-wq.c +84 -0
  254. data/vendor/liburing/test/short-read.c +75 -0
  255. data/vendor/liburing/test/shutdown.c +165 -0
  256. data/vendor/liburing/test/sigfd-deadlock.c +88 -0
  257. data/vendor/liburing/test/single-issuer.c +169 -0
  258. data/vendor/liburing/test/skip-cqe.c +428 -0
  259. data/vendor/liburing/test/socket-getsetsock-cmd.c +346 -0
  260. data/vendor/liburing/test/socket-io-cmd.c +237 -0
  261. data/vendor/liburing/test/socket-rw-eagain.c +149 -0
  262. data/vendor/liburing/test/socket-rw-offset.c +149 -0
  263. data/vendor/liburing/test/socket-rw.c +137 -0
  264. data/vendor/liburing/test/socket.c +408 -0
  265. data/vendor/liburing/test/splice.c +512 -0
  266. data/vendor/liburing/test/sq-full-cpp.cc +45 -0
  267. data/vendor/liburing/test/sq-full.c +45 -0
  268. data/vendor/liburing/test/sq-poll-dup.c +211 -0
  269. data/vendor/liburing/test/sq-poll-kthread.c +169 -0
  270. data/vendor/liburing/test/sq-poll-share.c +138 -0
  271. data/vendor/liburing/test/sq-space_left.c +159 -0
  272. data/vendor/liburing/test/sqpoll-disable-exit.c +196 -0
  273. data/vendor/liburing/test/sqpoll-exec.c +132 -0
  274. data/vendor/liburing/test/sqpoll-exit-hang.c +78 -0
  275. data/vendor/liburing/test/sqpoll-sleep.c +69 -0
  276. data/vendor/liburing/test/statx.c +172 -0
  277. data/vendor/liburing/test/stdout.c +232 -0
  278. data/vendor/liburing/test/submit-and-wait.c +108 -0
  279. data/vendor/liburing/test/submit-link-fail.c +156 -0
  280. data/vendor/liburing/test/submit-reuse.c +237 -0
  281. data/vendor/liburing/test/symlink.c +117 -0
  282. data/vendor/liburing/test/sync-cancel.c +235 -0
  283. data/vendor/liburing/test/teardowns.c +58 -0
  284. data/vendor/liburing/test/test.h +36 -0
  285. data/vendor/liburing/test/thread-exit.c +143 -0
  286. data/vendor/liburing/test/timeout-new.c +256 -0
  287. data/vendor/liburing/test/timeout.c +1798 -0
  288. data/vendor/liburing/test/truncate.c +186 -0
  289. data/vendor/liburing/test/tty-write-dpoll.c +60 -0
  290. data/vendor/liburing/test/unlink.c +112 -0
  291. data/vendor/liburing/test/version.c +25 -0
  292. data/vendor/liburing/test/wait-timeout.c +287 -0
  293. data/vendor/liburing/test/waitid.c +373 -0
  294. data/vendor/liburing/test/wakeup-hang.c +162 -0
  295. data/vendor/liburing/test/wq-aff.c +146 -0
  296. data/vendor/liburing/test/xattr.c +442 -0
  297. metadata +402 -0
@@ -0,0 +1,1013 @@
1
+ /* SPDX-License-Identifier: MIT */
2
+ /*
3
+ * Description: basic read/write tests with buffered, O_DIRECT, and SQPOLL
4
+ */
5
+ #include <errno.h>
6
+ #include <stdio.h>
7
+ #include <unistd.h>
8
+ #include <stdlib.h>
9
+ #include <string.h>
10
+ #include <fcntl.h>
11
+ #include <sys/types.h>
12
+ #include <poll.h>
13
+ #include <sys/eventfd.h>
14
+ #include <sys/resource.h>
15
+
16
+ #include "helpers.h"
17
+ #include "liburing.h"
18
+
19
+ #define FILE_SIZE (256 * 1024)
20
+ #define BS 8192
21
+ #define BUFFERS (FILE_SIZE / BS)
22
+
23
+ static struct iovec *vecs;
24
+ static int no_read;
25
+ static int no_buf_select;
26
+ static int warned;
27
+
28
+ static int create_nonaligned_buffers(void)
29
+ {
30
+ int i;
31
+
32
+ vecs = t_malloc(BUFFERS * sizeof(struct iovec));
33
+ for (i = 0; i < BUFFERS; i++) {
34
+ char *p = t_malloc(3 * BS);
35
+
36
+ if (!p)
37
+ return 1;
38
+ vecs[i].iov_base = p + (rand() % BS);
39
+ vecs[i].iov_len = 1 + (rand() % BS);
40
+ }
41
+
42
+ return 0;
43
+ }
44
+
45
+ static int __test_io(const char *file, struct io_uring *ring, int write,
46
+ int buffered, int sqthread, int fixed, int nonvec,
47
+ int buf_select, int seq, int exp_len)
48
+ {
49
+ struct io_uring_sqe *sqe;
50
+ struct io_uring_cqe *cqe;
51
+ int open_flags;
52
+ int i, fd = -1, ret;
53
+ off_t offset;
54
+
55
+ #ifdef VERBOSE
56
+ fprintf(stdout, "%s: start %d/%d/%d/%d/%d: ", __FUNCTION__, write,
57
+ buffered, sqthread,
58
+ fixed, nonvec);
59
+ #endif
60
+ if (write)
61
+ open_flags = O_WRONLY;
62
+ else
63
+ open_flags = O_RDONLY;
64
+ if (!buffered)
65
+ open_flags |= O_DIRECT;
66
+
67
+ if (fixed) {
68
+ ret = t_register_buffers(ring, vecs, BUFFERS);
69
+ if (ret == T_SETUP_SKIP)
70
+ return 0;
71
+ if (ret != T_SETUP_OK) {
72
+ fprintf(stderr, "buffer reg failed: %d\n", ret);
73
+ goto err;
74
+ }
75
+ }
76
+
77
+ fd = open(file, open_flags);
78
+ if (fd < 0) {
79
+ if (errno == EINVAL)
80
+ return 0;
81
+ perror("file open");
82
+ goto err;
83
+ }
84
+
85
+ if (sqthread) {
86
+ ret = io_uring_register_files(ring, &fd, 1);
87
+ if (ret) {
88
+ fprintf(stderr, "file reg failed: %d\n", ret);
89
+ goto err;
90
+ }
91
+ }
92
+
93
+ offset = 0;
94
+ for (i = 0; i < BUFFERS; i++) {
95
+ sqe = io_uring_get_sqe(ring);
96
+ if (!sqe) {
97
+ fprintf(stderr, "sqe get failed\n");
98
+ goto err;
99
+ }
100
+ if (!seq)
101
+ offset = BS * (rand() % BUFFERS);
102
+ if (write) {
103
+ int do_fixed = fixed;
104
+ int use_fd = fd;
105
+
106
+ if (sqthread)
107
+ use_fd = 0;
108
+ if (fixed && (i & 1))
109
+ do_fixed = 0;
110
+ if (do_fixed) {
111
+ io_uring_prep_write_fixed(sqe, use_fd, vecs[i].iov_base,
112
+ vecs[i].iov_len,
113
+ offset, i);
114
+ } else if (nonvec) {
115
+ io_uring_prep_write(sqe, use_fd, vecs[i].iov_base,
116
+ vecs[i].iov_len, offset);
117
+ } else {
118
+ io_uring_prep_writev(sqe, use_fd, &vecs[i], 1,
119
+ offset);
120
+ }
121
+ } else {
122
+ int do_fixed = fixed;
123
+ int use_fd = fd;
124
+
125
+ if (sqthread)
126
+ use_fd = 0;
127
+ if (fixed && (i & 1))
128
+ do_fixed = 0;
129
+ if (do_fixed) {
130
+ io_uring_prep_read_fixed(sqe, use_fd, vecs[i].iov_base,
131
+ vecs[i].iov_len,
132
+ offset, i);
133
+ } else if (nonvec) {
134
+ io_uring_prep_read(sqe, use_fd, vecs[i].iov_base,
135
+ vecs[i].iov_len, offset);
136
+ } else {
137
+ io_uring_prep_readv(sqe, use_fd, &vecs[i], 1,
138
+ offset);
139
+ }
140
+
141
+ }
142
+ sqe->user_data = i;
143
+ if (sqthread)
144
+ sqe->flags |= IOSQE_FIXED_FILE;
145
+ if (buf_select) {
146
+ if (nonvec)
147
+ sqe->addr = 0;
148
+ sqe->flags |= IOSQE_BUFFER_SELECT;
149
+ sqe->buf_group = buf_select;
150
+ }
151
+ if (seq)
152
+ offset += BS;
153
+ }
154
+
155
+ ret = io_uring_submit(ring);
156
+ if (ret != BUFFERS) {
157
+ fprintf(stderr, "submit got %d, wanted %d\n", ret, BUFFERS);
158
+ goto err;
159
+ }
160
+
161
+ for (i = 0; i < BUFFERS; i++) {
162
+ ret = io_uring_wait_cqe(ring, &cqe);
163
+ if (ret) {
164
+ fprintf(stderr, "wait_cqe=%d\n", ret);
165
+ goto err;
166
+ }
167
+ if (cqe->res == -EINVAL && nonvec) {
168
+ if (!warned) {
169
+ fprintf(stdout, "Non-vectored IO not "
170
+ "supported, skipping\n");
171
+ warned = 1;
172
+ no_read = 1;
173
+ }
174
+ } else if (exp_len == -1) {
175
+ int iov_len = vecs[cqe->user_data].iov_len;
176
+
177
+ if (cqe->res != iov_len) {
178
+ fprintf(stderr, "cqe res %d, wanted %d\n",
179
+ cqe->res, iov_len);
180
+ goto err;
181
+ }
182
+ } else if (cqe->res != exp_len) {
183
+ fprintf(stderr, "cqe res %d, wanted %d\n", cqe->res, exp_len);
184
+ goto err;
185
+ }
186
+ if (buf_select && exp_len == BS) {
187
+ int bid = cqe->flags >> 16;
188
+ unsigned char *ptr = vecs[bid].iov_base;
189
+ int j;
190
+
191
+ for (j = 0; j < BS; j++) {
192
+ if (ptr[j] == cqe->user_data)
193
+ continue;
194
+
195
+ fprintf(stderr, "Data mismatch! bid=%d, "
196
+ "wanted=%d, got=%d\n", bid,
197
+ (int)cqe->user_data, ptr[j]);
198
+ return 1;
199
+ }
200
+ }
201
+ io_uring_cqe_seen(ring, cqe);
202
+ }
203
+
204
+ if (fixed) {
205
+ ret = io_uring_unregister_buffers(ring);
206
+ if (ret) {
207
+ fprintf(stderr, "buffer unreg failed: %d\n", ret);
208
+ goto err;
209
+ }
210
+ }
211
+ if (sqthread) {
212
+ ret = io_uring_unregister_files(ring);
213
+ if (ret) {
214
+ fprintf(stderr, "file unreg failed: %d\n", ret);
215
+ goto err;
216
+ }
217
+ }
218
+
219
+ close(fd);
220
+ #ifdef VERBOSE
221
+ fprintf(stdout, "PASS\n");
222
+ #endif
223
+ return 0;
224
+ err:
225
+ #ifdef VERBOSE
226
+ fprintf(stderr, "FAILED\n");
227
+ #endif
228
+ if (fd != -1)
229
+ close(fd);
230
+ return 1;
231
+ }
232
+ static int test_io(const char *file, int write, int buffered, int sqthread,
233
+ int fixed, int nonvec, int exp_len)
234
+ {
235
+ struct io_uring ring;
236
+ int ret, ring_flags = 0;
237
+
238
+ if (sqthread)
239
+ ring_flags = IORING_SETUP_SQPOLL;
240
+
241
+ ret = t_create_ring(64, &ring, ring_flags);
242
+ if (ret == T_SETUP_SKIP)
243
+ return 0;
244
+ if (ret != T_SETUP_OK) {
245
+ fprintf(stderr, "ring create failed: %d\n", ret);
246
+ return 1;
247
+ }
248
+
249
+ ret = __test_io(file, &ring, write, buffered, sqthread, fixed, nonvec,
250
+ 0, 0, exp_len);
251
+ io_uring_queue_exit(&ring);
252
+ return ret;
253
+ }
254
+
255
+ static int read_poll_link(const char *file)
256
+ {
257
+ struct __kernel_timespec ts;
258
+ struct io_uring_sqe *sqe;
259
+ struct io_uring_cqe *cqe;
260
+ struct io_uring ring;
261
+ int i, fd, ret, fds[2];
262
+
263
+ ret = io_uring_queue_init(8, &ring, 0);
264
+ if (ret)
265
+ return ret;
266
+
267
+ fd = open(file, O_WRONLY);
268
+ if (fd < 0) {
269
+ perror("open");
270
+ return 1;
271
+ }
272
+
273
+ if (pipe(fds)) {
274
+ perror("pipe");
275
+ return 1;
276
+ }
277
+
278
+ sqe = io_uring_get_sqe(&ring);
279
+ io_uring_prep_writev(sqe, fd, &vecs[0], 1, 0);
280
+ sqe->flags |= IOSQE_IO_LINK;
281
+ sqe->user_data = 1;
282
+
283
+ sqe = io_uring_get_sqe(&ring);
284
+ io_uring_prep_poll_add(sqe, fds[0], POLLIN);
285
+ sqe->flags |= IOSQE_IO_LINK;
286
+ sqe->user_data = 2;
287
+
288
+ ts.tv_sec = 1;
289
+ ts.tv_nsec = 0;
290
+ sqe = io_uring_get_sqe(&ring);
291
+ io_uring_prep_link_timeout(sqe, &ts, 0);
292
+ sqe->user_data = 3;
293
+
294
+ ret = io_uring_submit(&ring);
295
+ if (ret != 3) {
296
+ fprintf(stderr, "submitted %d\n", ret);
297
+ return 1;
298
+ }
299
+
300
+ for (i = 0; i < 3; i++) {
301
+ ret = io_uring_wait_cqe(&ring, &cqe);
302
+ if (ret) {
303
+ fprintf(stderr, "wait_cqe=%d\n", ret);
304
+ return 1;
305
+ }
306
+ io_uring_cqe_seen(&ring, cqe);
307
+ }
308
+
309
+ return 0;
310
+ }
311
+
312
+ static int has_nonvec_read(void)
313
+ {
314
+ struct io_uring_probe *p;
315
+ struct io_uring ring;
316
+ int ret;
317
+
318
+ ret = io_uring_queue_init(1, &ring, 0);
319
+ if (ret) {
320
+ fprintf(stderr, "queue init failed: %d\n", ret);
321
+ exit(ret);
322
+ }
323
+
324
+ p = t_calloc(1, sizeof(*p) + 256 * sizeof(struct io_uring_probe_op));
325
+ ret = io_uring_register_probe(&ring, p, 256);
326
+ /* if we don't have PROBE_REGISTER, we don't have OP_READ/WRITE */
327
+ if (ret == -EINVAL) {
328
+ out:
329
+ io_uring_queue_exit(&ring);
330
+ return 0;
331
+ } else if (ret) {
332
+ fprintf(stderr, "register_probe: %d\n", ret);
333
+ goto out;
334
+ }
335
+
336
+ if (p->ops_len <= IORING_OP_READ)
337
+ goto out;
338
+ if (!(p->ops[IORING_OP_READ].flags & IO_URING_OP_SUPPORTED))
339
+ goto out;
340
+ io_uring_queue_exit(&ring);
341
+ return 1;
342
+ }
343
+
344
+ static int test_eventfd_read(void)
345
+ {
346
+ struct io_uring ring;
347
+ int fd, ret;
348
+ eventfd_t event;
349
+ struct io_uring_sqe *sqe;
350
+ struct io_uring_cqe *cqe;
351
+
352
+ if (no_read)
353
+ return 0;
354
+ ret = io_uring_queue_init(8, &ring, 0);
355
+ if (ret)
356
+ return ret;
357
+
358
+ fd = eventfd(1, 0);
359
+ if (fd < 0) {
360
+ perror("eventfd");
361
+ return 1;
362
+ }
363
+ sqe = io_uring_get_sqe(&ring);
364
+ io_uring_prep_read(sqe, fd, &event, sizeof(eventfd_t), 0);
365
+ ret = io_uring_submit(&ring);
366
+ if (ret != 1) {
367
+ fprintf(stderr, "submitted %d\n", ret);
368
+ return 1;
369
+ }
370
+ eventfd_write(fd, 1);
371
+ ret = io_uring_wait_cqe(&ring, &cqe);
372
+ if (ret) {
373
+ fprintf(stderr, "wait_cqe=%d\n", ret);
374
+ return 1;
375
+ }
376
+ if (cqe->res == -EINVAL) {
377
+ fprintf(stdout, "eventfd IO not supported, skipping\n");
378
+ } else if (cqe->res != sizeof(eventfd_t)) {
379
+ fprintf(stderr, "cqe res %d, wanted %d\n", cqe->res,
380
+ (int) sizeof(eventfd_t));
381
+ return 1;
382
+ }
383
+ io_uring_cqe_seen(&ring, cqe);
384
+ return 0;
385
+ }
386
+
387
+ static int test_buf_select_short(const char *filename, int nonvec)
388
+ {
389
+ struct io_uring_sqe *sqe;
390
+ struct io_uring_cqe *cqe;
391
+ struct io_uring ring;
392
+ int ret, i, exp_len;
393
+
394
+ if (no_buf_select)
395
+ return 0;
396
+
397
+ ret = io_uring_queue_init(64, &ring, 0);
398
+ if (ret) {
399
+ fprintf(stderr, "ring create failed: %d\n", ret);
400
+ return 1;
401
+ }
402
+
403
+ exp_len = 0;
404
+ for (i = 0; i < BUFFERS; i++) {
405
+ sqe = io_uring_get_sqe(&ring);
406
+ io_uring_prep_provide_buffers(sqe, vecs[i].iov_base,
407
+ vecs[i].iov_len / 2, 1, 1, i);
408
+ if (!exp_len)
409
+ exp_len = vecs[i].iov_len / 2;
410
+ }
411
+
412
+ ret = io_uring_submit(&ring);
413
+ if (ret != BUFFERS) {
414
+ fprintf(stderr, "submit: %d\n", ret);
415
+ return -1;
416
+ }
417
+
418
+ for (i = 0; i < BUFFERS; i++) {
419
+ ret = io_uring_wait_cqe(&ring, &cqe);
420
+ if (cqe->res < 0) {
421
+ fprintf(stderr, "cqe->res=%d\n", cqe->res);
422
+ return 1;
423
+ }
424
+ io_uring_cqe_seen(&ring, cqe);
425
+ }
426
+
427
+ ret = __test_io(filename, &ring, 0, 0, 0, 0, nonvec, 1, 1, exp_len);
428
+
429
+ io_uring_queue_exit(&ring);
430
+ return ret;
431
+ }
432
+
433
+ static int provide_buffers_iovec(struct io_uring *ring, int bgid)
434
+ {
435
+ struct io_uring_sqe *sqe;
436
+ struct io_uring_cqe *cqe;
437
+ int i, ret;
438
+
439
+ for (i = 0; i < BUFFERS; i++) {
440
+ sqe = io_uring_get_sqe(ring);
441
+ io_uring_prep_provide_buffers(sqe, vecs[i].iov_base,
442
+ vecs[i].iov_len, 1, bgid, i);
443
+ }
444
+
445
+ ret = io_uring_submit(ring);
446
+ if (ret != BUFFERS) {
447
+ fprintf(stderr, "submit: %d\n", ret);
448
+ return -1;
449
+ }
450
+
451
+ for (i = 0; i < BUFFERS; i++) {
452
+ ret = io_uring_wait_cqe(ring, &cqe);
453
+ if (ret) {
454
+ fprintf(stderr, "wait_cqe=%d\n", ret);
455
+ return 1;
456
+ }
457
+ if (cqe->res < 0) {
458
+ fprintf(stderr, "cqe->res=%d\n", cqe->res);
459
+ return 1;
460
+ }
461
+ io_uring_cqe_seen(ring, cqe);
462
+ }
463
+
464
+ return 0;
465
+ }
466
+
467
+ static int test_buf_select_pipe(void)
468
+ {
469
+ struct io_uring_sqe *sqe;
470
+ struct io_uring_cqe *cqe;
471
+ struct io_uring ring;
472
+ int ret, i;
473
+ int fds[2];
474
+
475
+ if (no_buf_select)
476
+ return 0;
477
+
478
+ ret = io_uring_queue_init(64, &ring, 0);
479
+ if (ret) {
480
+ fprintf(stderr, "ring create failed: %d\n", ret);
481
+ return 1;
482
+ }
483
+
484
+ ret = provide_buffers_iovec(&ring, 0);
485
+ if (ret) {
486
+ fprintf(stderr, "provide buffers failed: %d\n", ret);
487
+ return 1;
488
+ }
489
+
490
+ ret = pipe(fds);
491
+ if (ret) {
492
+ fprintf(stderr, "pipe failed: %d\n", ret);
493
+ return 1;
494
+ }
495
+
496
+ for (i = 0; i < 5; i++) {
497
+ sqe = io_uring_get_sqe(&ring);
498
+ io_uring_prep_read(sqe, fds[0], NULL, 1 /* max read 1 per go */, -1);
499
+ sqe->flags |= IOSQE_BUFFER_SELECT;
500
+ sqe->buf_group = 0;
501
+ }
502
+ io_uring_submit(&ring);
503
+
504
+ ret = write(fds[1], "01234", 5);
505
+ if (ret != 5) {
506
+ fprintf(stderr, "pipe write failed %d\n", ret);
507
+ return 1;
508
+ }
509
+
510
+ for (i = 0; i < 5; i++) {
511
+ const char *buff;
512
+
513
+ if (io_uring_wait_cqe(&ring, &cqe)) {
514
+ fprintf(stderr, "bad wait %d\n", i);
515
+ return 1;
516
+ }
517
+ if (cqe->res != 1) {
518
+ fprintf(stderr, "expected read %d\n", cqe->res);
519
+ return 1;
520
+ }
521
+ if (!(cqe->flags & IORING_CQE_F_BUFFER)) {
522
+ fprintf(stderr, "no buffer %d\n", cqe->res);
523
+ return 1;
524
+ }
525
+ buff = vecs[cqe->flags >> 16].iov_base;
526
+ if (*buff != '0' + i) {
527
+ fprintf(stderr, "%d: expected %c, got %c\n", i, '0' + i, *buff);
528
+ return 1;
529
+ }
530
+ io_uring_cqe_seen(&ring, cqe);
531
+ }
532
+
533
+
534
+ close(fds[0]);
535
+ close(fds[1]);
536
+ io_uring_queue_exit(&ring);
537
+ return 0;
538
+ }
539
+
540
+ static int test_buf_select(const char *filename, int nonvec)
541
+ {
542
+ struct io_uring_probe *p;
543
+ struct io_uring ring;
544
+ int ret, i;
545
+
546
+ ret = io_uring_queue_init(64, &ring, 0);
547
+ if (ret) {
548
+ fprintf(stderr, "ring create failed: %d\n", ret);
549
+ return 1;
550
+ }
551
+
552
+ p = io_uring_get_probe_ring(&ring);
553
+ if (!p || !io_uring_opcode_supported(p, IORING_OP_PROVIDE_BUFFERS)) {
554
+ no_buf_select = 1;
555
+ fprintf(stdout, "Buffer select not supported, skipping\n");
556
+ return 0;
557
+ }
558
+ io_uring_free_probe(p);
559
+
560
+ /*
561
+ * Write out data with known pattern
562
+ */
563
+ for (i = 0; i < BUFFERS; i++)
564
+ memset(vecs[i].iov_base, i, vecs[i].iov_len);
565
+
566
+ ret = __test_io(filename, &ring, 1, 0, 0, 0, 0, 0, 1, BS);
567
+ if (ret) {
568
+ fprintf(stderr, "failed writing data\n");
569
+ return 1;
570
+ }
571
+
572
+ for (i = 0; i < BUFFERS; i++)
573
+ memset(vecs[i].iov_base, 0x55, vecs[i].iov_len);
574
+
575
+ ret = provide_buffers_iovec(&ring, 1);
576
+ if (ret)
577
+ return ret;
578
+
579
+ ret = __test_io(filename, &ring, 0, 0, 0, 0, nonvec, 1, 1, BS);
580
+ io_uring_queue_exit(&ring);
581
+ return ret;
582
+ }
583
+
584
+ static int test_rem_buf(int batch, int sqe_flags)
585
+ {
586
+ struct io_uring_sqe *sqe;
587
+ struct io_uring_cqe *cqe;
588
+ struct io_uring ring;
589
+ int left, ret, nr = 0;
590
+ int bgid = 1;
591
+
592
+ if (no_buf_select)
593
+ return 0;
594
+
595
+ ret = io_uring_queue_init(64, &ring, 0);
596
+ if (ret) {
597
+ fprintf(stderr, "ring create failed: %d\n", ret);
598
+ return 1;
599
+ }
600
+
601
+ ret = provide_buffers_iovec(&ring, bgid);
602
+ if (ret)
603
+ return ret;
604
+
605
+ left = BUFFERS;
606
+ while (left) {
607
+ int to_rem = (left < batch) ? left : batch;
608
+
609
+ left -= to_rem;
610
+ sqe = io_uring_get_sqe(&ring);
611
+ io_uring_prep_remove_buffers(sqe, to_rem, bgid);
612
+ sqe->user_data = to_rem;
613
+ sqe->flags |= sqe_flags;
614
+ ++nr;
615
+ }
616
+
617
+ ret = io_uring_submit(&ring);
618
+ if (ret != nr) {
619
+ fprintf(stderr, "submit: %d\n", ret);
620
+ return -1;
621
+ }
622
+
623
+ for (; nr > 0; nr--) {
624
+ ret = io_uring_wait_cqe(&ring, &cqe);
625
+ if (ret) {
626
+ fprintf(stderr, "wait_cqe=%d\n", ret);
627
+ return 1;
628
+ }
629
+ if (cqe->res != cqe->user_data) {
630
+ fprintf(stderr, "cqe->res=%d\n", cqe->res);
631
+ return 1;
632
+ }
633
+ io_uring_cqe_seen(&ring, cqe);
634
+ }
635
+
636
+ io_uring_queue_exit(&ring);
637
+ return ret;
638
+ }
639
+
640
+ static int test_rem_buf_single(int to_rem)
641
+ {
642
+ struct io_uring_sqe *sqe;
643
+ struct io_uring_cqe *cqe;
644
+ struct io_uring ring;
645
+ int ret, expected;
646
+ int bgid = 1;
647
+
648
+ if (no_buf_select)
649
+ return 0;
650
+
651
+ ret = io_uring_queue_init(64, &ring, 0);
652
+ if (ret) {
653
+ fprintf(stderr, "ring create failed: %d\n", ret);
654
+ return 1;
655
+ }
656
+
657
+ ret = provide_buffers_iovec(&ring, bgid);
658
+ if (ret)
659
+ return ret;
660
+
661
+ expected = (to_rem > BUFFERS) ? BUFFERS : to_rem;
662
+
663
+ sqe = io_uring_get_sqe(&ring);
664
+ io_uring_prep_remove_buffers(sqe, to_rem, bgid);
665
+
666
+ ret = io_uring_submit(&ring);
667
+ if (ret != 1) {
668
+ fprintf(stderr, "submit: %d\n", ret);
669
+ return -1;
670
+ }
671
+
672
+ ret = io_uring_wait_cqe(&ring, &cqe);
673
+ if (ret) {
674
+ fprintf(stderr, "wait_cqe=%d\n", ret);
675
+ return 1;
676
+ }
677
+ if (cqe->res != expected) {
678
+ fprintf(stderr, "cqe->res=%d, expected=%d\n", cqe->res, expected);
679
+ return 1;
680
+ }
681
+ io_uring_cqe_seen(&ring, cqe);
682
+
683
+ io_uring_queue_exit(&ring);
684
+ return ret;
685
+ }
686
+
687
+ static int test_io_link(const char *file)
688
+ {
689
+ const int nr_links = 100;
690
+ const int link_len = 100;
691
+ const int nr_sqes = nr_links * link_len;
692
+ struct io_uring_sqe *sqe;
693
+ struct io_uring_cqe *cqe;
694
+ struct io_uring ring;
695
+ int i, j, fd, ret;
696
+
697
+ fd = open(file, O_WRONLY);
698
+ if (fd < 0) {
699
+ perror("file open");
700
+ goto err;
701
+ }
702
+
703
+ ret = io_uring_queue_init(nr_sqes, &ring, 0);
704
+ if (ret) {
705
+ fprintf(stderr, "ring create failed: %d\n", ret);
706
+ goto err;
707
+ }
708
+
709
+ for (i = 0; i < nr_links; ++i) {
710
+ for (j = 0; j < link_len; ++j) {
711
+ sqe = io_uring_get_sqe(&ring);
712
+ if (!sqe) {
713
+ fprintf(stderr, "sqe get failed\n");
714
+ goto err;
715
+ }
716
+ io_uring_prep_writev(sqe, fd, &vecs[0], 1, 0);
717
+ sqe->flags |= IOSQE_ASYNC;
718
+ if (j != link_len - 1)
719
+ sqe->flags |= IOSQE_IO_LINK;
720
+ }
721
+ }
722
+
723
+ ret = io_uring_submit(&ring);
724
+ if (ret != nr_sqes) {
725
+ ret = io_uring_peek_cqe(&ring, &cqe);
726
+ if (!ret && cqe->res == -EINVAL) {
727
+ fprintf(stdout, "IOSQE_ASYNC not supported, skipped\n");
728
+ goto out;
729
+ }
730
+ fprintf(stderr, "submit got %d, wanted %d\n", ret, nr_sqes);
731
+ goto err;
732
+ }
733
+
734
+ for (i = 0; i < nr_sqes; i++) {
735
+ ret = io_uring_wait_cqe(&ring, &cqe);
736
+ if (ret) {
737
+ fprintf(stderr, "wait_cqe=%d\n", ret);
738
+ goto err;
739
+ }
740
+ if (cqe->res == -EINVAL) {
741
+ if (!warned) {
742
+ fprintf(stdout, "Non-vectored IO not "
743
+ "supported, skipping\n");
744
+ warned = 1;
745
+ no_read = 1;
746
+ }
747
+ } else if (cqe->res != BS) {
748
+ fprintf(stderr, "cqe res %d, wanted %d\n", cqe->res, BS);
749
+ goto err;
750
+ }
751
+ io_uring_cqe_seen(&ring, cqe);
752
+ }
753
+
754
+ out:
755
+ io_uring_queue_exit(&ring);
756
+ close(fd);
757
+ return 0;
758
+ err:
759
+ if (fd != -1)
760
+ close(fd);
761
+ return 1;
762
+ }
763
+
764
+ static int test_write_efbig(void)
765
+ {
766
+ struct io_uring_sqe *sqe;
767
+ struct io_uring_cqe *cqe;
768
+ struct io_uring ring;
769
+ struct rlimit rlim, old_rlim;
770
+ int i, fd, ret;
771
+ loff_t off;
772
+
773
+ if (geteuid()) {
774
+ fprintf(stdout, "Not root, skipping %s\n", __FUNCTION__);
775
+ return 0;
776
+ }
777
+
778
+ if (getrlimit(RLIMIT_FSIZE, &old_rlim) < 0) {
779
+ perror("getrlimit");
780
+ return 1;
781
+ }
782
+ rlim = old_rlim;
783
+ rlim.rlim_cur = 128 * 1024;
784
+ rlim.rlim_max = 128 * 1024;
785
+ if (setrlimit(RLIMIT_FSIZE, &rlim) < 0) {
786
+ perror("setrlimit");
787
+ return 1;
788
+ }
789
+
790
+ fd = open(".efbig", O_WRONLY | O_CREAT, 0644);
791
+ if (fd < 0) {
792
+ perror("file open");
793
+ goto err;
794
+ }
795
+ unlink(".efbig");
796
+
797
+ ret = io_uring_queue_init(32, &ring, 0);
798
+ if (ret) {
799
+ fprintf(stderr, "ring create failed: %d\n", ret);
800
+ goto err;
801
+ }
802
+
803
+ off = 0;
804
+ for (i = 0; i < 32; i++) {
805
+ sqe = io_uring_get_sqe(&ring);
806
+ if (!sqe) {
807
+ fprintf(stderr, "sqe get failed\n");
808
+ goto err;
809
+ }
810
+ io_uring_prep_writev(sqe, fd, &vecs[i], 1, off);
811
+ io_uring_sqe_set_data64(sqe, i);
812
+ off += BS;
813
+ }
814
+
815
+ ret = io_uring_submit(&ring);
816
+ if (ret != 32) {
817
+ fprintf(stderr, "submit got %d, wanted %d\n", ret, 32);
818
+ goto err;
819
+ }
820
+
821
+ for (i = 0; i < 32; i++) {
822
+ ret = io_uring_wait_cqe(&ring, &cqe);
823
+ if (ret) {
824
+ fprintf(stderr, "wait_cqe=%d\n", ret);
825
+ goto err;
826
+ }
827
+ if (cqe->user_data < 16) {
828
+ if (cqe->res != BS) {
829
+ fprintf(stderr, "bad write: %d\n", cqe->res);
830
+ goto err;
831
+ }
832
+ } else {
833
+ if (cqe->res != -EFBIG) {
834
+ fprintf(stderr, "Expected -EFBIG: %d\n", cqe->res);
835
+ goto err;
836
+ }
837
+ }
838
+ io_uring_cqe_seen(&ring, cqe);
839
+ }
840
+
841
+ io_uring_queue_exit(&ring);
842
+ close(fd);
843
+ unlink(".efbig");
844
+
845
+ if (setrlimit(RLIMIT_FSIZE, &old_rlim) < 0) {
846
+ perror("setrlimit");
847
+ return 1;
848
+ }
849
+ return 0;
850
+ err:
851
+ if (fd != -1)
852
+ close(fd);
853
+ return 1;
854
+ }
855
+
856
+ int main(int argc, char *argv[])
857
+ {
858
+ int i, ret, nr;
859
+ char buf[256];
860
+ char *fname;
861
+
862
+ if (argc > 1) {
863
+ fname = argv[1];
864
+ } else {
865
+ srand((unsigned)time(NULL));
866
+ snprintf(buf, sizeof(buf), ".basic-rw-%u-%u",
867
+ (unsigned)rand(), (unsigned)getpid());
868
+ fname = buf;
869
+ t_create_file(fname, FILE_SIZE);
870
+ }
871
+
872
+ signal(SIGXFSZ, SIG_IGN);
873
+
874
+ vecs = t_create_buffers(BUFFERS, BS);
875
+
876
+ /* if we don't have nonvec read, skip testing that */
877
+ nr = has_nonvec_read() ? 32 : 16;
878
+
879
+ for (i = 0; i < nr; i++) {
880
+ int write = (i & 1) != 0;
881
+ int buffered = (i & 2) != 0;
882
+ int sqthread = (i & 4) != 0;
883
+ int fixed = (i & 8) != 0;
884
+ int nonvec = (i & 16) != 0;
885
+
886
+ ret = test_io(fname, write, buffered, sqthread, fixed, nonvec,
887
+ BS);
888
+ if (ret) {
889
+ fprintf(stderr, "test_io failed %d/%d/%d/%d/%d\n",
890
+ write, buffered, sqthread, fixed, nonvec);
891
+ goto err;
892
+ }
893
+ }
894
+
895
+ ret = test_buf_select(fname, 1);
896
+ if (ret) {
897
+ fprintf(stderr, "test_buf_select nonvec failed\n");
898
+ goto err;
899
+ }
900
+
901
+ ret = test_buf_select(fname, 0);
902
+ if (ret) {
903
+ fprintf(stderr, "test_buf_select vec failed\n");
904
+ goto err;
905
+ }
906
+
907
+ ret = test_buf_select_short(fname, 1);
908
+ if (ret) {
909
+ fprintf(stderr, "test_buf_select_short nonvec failed\n");
910
+ goto err;
911
+ }
912
+
913
+ ret = test_buf_select_short(fname, 0);
914
+ if (ret) {
915
+ fprintf(stderr, "test_buf_select_short vec failed\n");
916
+ goto err;
917
+ }
918
+
919
+ ret = test_buf_select_pipe();
920
+ if (ret) {
921
+ fprintf(stderr, "test_buf_select_pipe failed\n");
922
+ goto err;
923
+ }
924
+
925
+ ret = test_eventfd_read();
926
+ if (ret) {
927
+ fprintf(stderr, "test_eventfd_read failed\n");
928
+ goto err;
929
+ }
930
+
931
+ ret = read_poll_link(fname);
932
+ if (ret) {
933
+ fprintf(stderr, "read_poll_link failed\n");
934
+ goto err;
935
+ }
936
+
937
+ ret = test_io_link(fname);
938
+ if (ret) {
939
+ fprintf(stderr, "test_io_link failed\n");
940
+ goto err;
941
+ }
942
+
943
+ ret = test_write_efbig();
944
+ if (ret) {
945
+ fprintf(stderr, "test_write_efbig failed\n");
946
+ goto err;
947
+ }
948
+
949
+ ret = test_rem_buf(1, 0);
950
+ if (ret) {
951
+ fprintf(stderr, "test_rem_buf by 1 failed\n");
952
+ goto err;
953
+ }
954
+
955
+ ret = test_rem_buf(10, 0);
956
+ if (ret) {
957
+ fprintf(stderr, "test_rem_buf by 10 failed\n");
958
+ goto err;
959
+ }
960
+
961
+ ret = test_rem_buf(2, IOSQE_IO_LINK);
962
+ if (ret) {
963
+ fprintf(stderr, "test_rem_buf link failed\n");
964
+ goto err;
965
+ }
966
+
967
+ ret = test_rem_buf(2, IOSQE_ASYNC);
968
+ if (ret) {
969
+ fprintf(stderr, "test_rem_buf async failed\n");
970
+ goto err;
971
+ }
972
+
973
+ srand((unsigned)time(NULL));
974
+ if (create_nonaligned_buffers()) {
975
+ fprintf(stderr, "file creation failed\n");
976
+ goto err;
977
+ }
978
+
979
+ /* test fixed bufs with non-aligned len/offset */
980
+ for (i = 0; i < nr; i++) {
981
+ int write = (i & 1) != 0;
982
+ int buffered = (i & 2) != 0;
983
+ int sqthread = (i & 4) != 0;
984
+ int fixed = (i & 8) != 0;
985
+ int nonvec = (i & 16) != 0;
986
+
987
+ /* direct IO requires alignment, skip it */
988
+ if (!buffered || !fixed || nonvec)
989
+ continue;
990
+
991
+ ret = test_io(fname, write, buffered, sqthread, fixed, nonvec,
992
+ -1);
993
+ if (ret) {
994
+ fprintf(stderr, "test_io failed %d/%d/%d/%d/%d\n",
995
+ write, buffered, sqthread, fixed, nonvec);
996
+ goto err;
997
+ }
998
+ }
999
+
1000
+ ret = test_rem_buf_single(BUFFERS + 1);
1001
+ if (ret) {
1002
+ fprintf(stderr, "test_rem_buf_single(BUFFERS + 1) failed\n");
1003
+ goto err;
1004
+ }
1005
+
1006
+ if (fname != argv[1])
1007
+ unlink(fname);
1008
+ return 0;
1009
+ err:
1010
+ if (fname != argv[1])
1011
+ unlink(fname);
1012
+ return 1;
1013
+ }