uringmachine 0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (307) hide show
  1. checksums.yaml +7 -0
  2. data/.github/dependabot.yml +12 -0
  3. data/.github/workflows/test.yml +35 -0
  4. data/.gitignore +59 -0
  5. data/.gitmodules +3 -0
  6. data/CHANGELOG.md +7 -0
  7. data/Gemfile +3 -0
  8. data/LICENSE +21 -0
  9. data/README.md +11 -0
  10. data/Rakefile +39 -0
  11. data/TODO.md +0 -0
  12. data/examples/echo_server.rb +52 -0
  13. data/examples/event_loop.rb +69 -0
  14. data/examples/fibers.rb +105 -0
  15. data/examples/http_server.rb +56 -0
  16. data/examples/http_server_multishot.rb +57 -0
  17. data/examples/http_server_simpler.rb +34 -0
  18. data/ext/um/extconf.rb +71 -0
  19. data/ext/um/iou.h +101 -0
  20. data/ext/um/op_ctx.c +138 -0
  21. data/ext/um/ring.c +755 -0
  22. data/ext/um/um.c +267 -0
  23. data/ext/um/um.h +97 -0
  24. data/ext/um/um_class.c +175 -0
  25. data/ext/um/um_ext.c +11 -0
  26. data/ext/um/um_op.c +87 -0
  27. data/ext/um/um_utils.c +23 -0
  28. data/lib/uringmachine/version.rb +3 -0
  29. data/lib/uringmachine.rb +8 -0
  30. data/test/helper.rb +70 -0
  31. data/test/test_iou.rb +876 -0
  32. data/test/test_um.rb +168 -0
  33. data/uringmachine.gemspec +27 -0
  34. data/vendor/liburing/.github/actions/codespell/stopwords +7 -0
  35. data/vendor/liburing/.github/pull_request_template.md +86 -0
  36. data/vendor/liburing/.github/workflows/build.yml +137 -0
  37. data/vendor/liburing/.github/workflows/codespell.yml +25 -0
  38. data/vendor/liburing/.github/workflows/shellcheck.yml +20 -0
  39. data/vendor/liburing/.gitignore +41 -0
  40. data/vendor/liburing/CHANGELOG +111 -0
  41. data/vendor/liburing/CITATION.cff +11 -0
  42. data/vendor/liburing/COPYING +502 -0
  43. data/vendor/liburing/COPYING.GPL +339 -0
  44. data/vendor/liburing/LICENSE +20 -0
  45. data/vendor/liburing/Makefile +96 -0
  46. data/vendor/liburing/Makefile.common +7 -0
  47. data/vendor/liburing/Makefile.quiet +11 -0
  48. data/vendor/liburing/README +106 -0
  49. data/vendor/liburing/SECURITY.md +6 -0
  50. data/vendor/liburing/configure +624 -0
  51. data/vendor/liburing/debian/README.Debian +7 -0
  52. data/vendor/liburing/debian/changelog +38 -0
  53. data/vendor/liburing/debian/control +39 -0
  54. data/vendor/liburing/debian/copyright +49 -0
  55. data/vendor/liburing/debian/liburing-dev.install +4 -0
  56. data/vendor/liburing/debian/liburing-dev.manpages +5 -0
  57. data/vendor/liburing/debian/liburing2.install +1 -0
  58. data/vendor/liburing/debian/liburing2.symbols +56 -0
  59. data/vendor/liburing/debian/patches/series +1 -0
  60. data/vendor/liburing/debian/rules +29 -0
  61. data/vendor/liburing/debian/source/format +1 -0
  62. data/vendor/liburing/debian/source/local-options +2 -0
  63. data/vendor/liburing/debian/source/options +1 -0
  64. data/vendor/liburing/debian/watch +3 -0
  65. data/vendor/liburing/examples/Makefile +53 -0
  66. data/vendor/liburing/examples/helpers.c +62 -0
  67. data/vendor/liburing/examples/helpers.h +7 -0
  68. data/vendor/liburing/examples/io_uring-close-test.c +123 -0
  69. data/vendor/liburing/examples/io_uring-cp.c +282 -0
  70. data/vendor/liburing/examples/io_uring-test.c +112 -0
  71. data/vendor/liburing/examples/io_uring-udp.c +403 -0
  72. data/vendor/liburing/examples/link-cp.c +193 -0
  73. data/vendor/liburing/examples/napi-busy-poll-client.c +509 -0
  74. data/vendor/liburing/examples/napi-busy-poll-server.c +450 -0
  75. data/vendor/liburing/examples/poll-bench.c +101 -0
  76. data/vendor/liburing/examples/proxy.c +2461 -0
  77. data/vendor/liburing/examples/proxy.h +102 -0
  78. data/vendor/liburing/examples/rsrc-update-bench.c +100 -0
  79. data/vendor/liburing/examples/send-zerocopy.c +658 -0
  80. data/vendor/liburing/examples/ucontext-cp.c +258 -0
  81. data/vendor/liburing/liburing-ffi.pc.in +12 -0
  82. data/vendor/liburing/liburing.pc.in +12 -0
  83. data/vendor/liburing/liburing.spec +66 -0
  84. data/vendor/liburing/make-debs.sh +55 -0
  85. data/vendor/liburing/src/Makefile +129 -0
  86. data/vendor/liburing/src/arch/aarch64/lib.h +47 -0
  87. data/vendor/liburing/src/arch/aarch64/syscall.h +91 -0
  88. data/vendor/liburing/src/arch/generic/lib.h +17 -0
  89. data/vendor/liburing/src/arch/generic/syscall.h +100 -0
  90. data/vendor/liburing/src/arch/riscv64/lib.h +48 -0
  91. data/vendor/liburing/src/arch/riscv64/syscall.h +100 -0
  92. data/vendor/liburing/src/arch/syscall-defs.h +94 -0
  93. data/vendor/liburing/src/arch/x86/lib.h +11 -0
  94. data/vendor/liburing/src/arch/x86/syscall.h +296 -0
  95. data/vendor/liburing/src/ffi.c +15 -0
  96. data/vendor/liburing/src/include/liburing/barrier.h +81 -0
  97. data/vendor/liburing/src/include/liburing/io_uring.h +818 -0
  98. data/vendor/liburing/src/include/liburing.h +1602 -0
  99. data/vendor/liburing/src/int_flags.h +11 -0
  100. data/vendor/liburing/src/lib.h +52 -0
  101. data/vendor/liburing/src/liburing-ffi.map +211 -0
  102. data/vendor/liburing/src/liburing.map +104 -0
  103. data/vendor/liburing/src/nolibc.c +55 -0
  104. data/vendor/liburing/src/queue.c +468 -0
  105. data/vendor/liburing/src/register.c +374 -0
  106. data/vendor/liburing/src/setup.c +689 -0
  107. data/vendor/liburing/src/setup.h +9 -0
  108. data/vendor/liburing/src/syscall.c +29 -0
  109. data/vendor/liburing/src/syscall.h +53 -0
  110. data/vendor/liburing/src/version.c +21 -0
  111. data/vendor/liburing/test/232c93d07b74.c +305 -0
  112. data/vendor/liburing/test/35fa71a030ca.c +329 -0
  113. data/vendor/liburing/test/500f9fbadef8.c +91 -0
  114. data/vendor/liburing/test/7ad0e4b2f83c.c +94 -0
  115. data/vendor/liburing/test/8a9973408177.c +107 -0
  116. data/vendor/liburing/test/917257daa0fe.c +54 -0
  117. data/vendor/liburing/test/Makefile +297 -0
  118. data/vendor/liburing/test/a0908ae19763.c +59 -0
  119. data/vendor/liburing/test/a4c0b3decb33.c +181 -0
  120. data/vendor/liburing/test/accept-link.c +255 -0
  121. data/vendor/liburing/test/accept-non-empty.c +256 -0
  122. data/vendor/liburing/test/accept-reuse.c +163 -0
  123. data/vendor/liburing/test/accept-test.c +83 -0
  124. data/vendor/liburing/test/accept.c +919 -0
  125. data/vendor/liburing/test/across-fork.c +284 -0
  126. data/vendor/liburing/test/b19062a56726.c +54 -0
  127. data/vendor/liburing/test/b5837bd5311d.c +78 -0
  128. data/vendor/liburing/test/bind-listen.c +408 -0
  129. data/vendor/liburing/test/buf-ring-nommap.c +123 -0
  130. data/vendor/liburing/test/buf-ring-put.c +83 -0
  131. data/vendor/liburing/test/buf-ring.c +473 -0
  132. data/vendor/liburing/test/ce593a6c480a.c +139 -0
  133. data/vendor/liburing/test/close-opath.c +123 -0
  134. data/vendor/liburing/test/config +14 -0
  135. data/vendor/liburing/test/connect-rep.c +204 -0
  136. data/vendor/liburing/test/connect.c +442 -0
  137. data/vendor/liburing/test/coredump.c +60 -0
  138. data/vendor/liburing/test/cq-full.c +97 -0
  139. data/vendor/liburing/test/cq-overflow.c +530 -0
  140. data/vendor/liburing/test/cq-peek-batch.c +103 -0
  141. data/vendor/liburing/test/cq-ready.c +95 -0
  142. data/vendor/liburing/test/cq-size.c +65 -0
  143. data/vendor/liburing/test/d4ae271dfaae.c +96 -0
  144. data/vendor/liburing/test/d77a67ed5f27.c +65 -0
  145. data/vendor/liburing/test/defer-taskrun.c +391 -0
  146. data/vendor/liburing/test/defer-tw-timeout.c +173 -0
  147. data/vendor/liburing/test/defer.c +319 -0
  148. data/vendor/liburing/test/double-poll-crash.c +195 -0
  149. data/vendor/liburing/test/drop-submit.c +94 -0
  150. data/vendor/liburing/test/eeed8b54e0df.c +120 -0
  151. data/vendor/liburing/test/empty-eownerdead.c +45 -0
  152. data/vendor/liburing/test/eploop.c +74 -0
  153. data/vendor/liburing/test/eventfd-disable.c +179 -0
  154. data/vendor/liburing/test/eventfd-reg.c +77 -0
  155. data/vendor/liburing/test/eventfd-ring.c +98 -0
  156. data/vendor/liburing/test/eventfd.c +113 -0
  157. data/vendor/liburing/test/evloop.c +73 -0
  158. data/vendor/liburing/test/exec-target.c +6 -0
  159. data/vendor/liburing/test/exit-no-cleanup.c +117 -0
  160. data/vendor/liburing/test/fadvise.c +202 -0
  161. data/vendor/liburing/test/fallocate.c +265 -0
  162. data/vendor/liburing/test/fc2a85cb02ef.c +132 -0
  163. data/vendor/liburing/test/fd-install.c +500 -0
  164. data/vendor/liburing/test/fd-pass.c +237 -0
  165. data/vendor/liburing/test/fdinfo.c +419 -0
  166. data/vendor/liburing/test/file-register.c +1189 -0
  167. data/vendor/liburing/test/file-update.c +231 -0
  168. data/vendor/liburing/test/file-verify.c +654 -0
  169. data/vendor/liburing/test/files-exit-hang-poll.c +114 -0
  170. data/vendor/liburing/test/files-exit-hang-timeout.c +137 -0
  171. data/vendor/liburing/test/fixed-buf-iter.c +115 -0
  172. data/vendor/liburing/test/fixed-buf-merge.c +101 -0
  173. data/vendor/liburing/test/fixed-hugepage.c +411 -0
  174. data/vendor/liburing/test/fixed-link.c +90 -0
  175. data/vendor/liburing/test/fixed-reuse.c +160 -0
  176. data/vendor/liburing/test/fpos.c +255 -0
  177. data/vendor/liburing/test/fsnotify.c +118 -0
  178. data/vendor/liburing/test/fsync.c +224 -0
  179. data/vendor/liburing/test/futex.c +571 -0
  180. data/vendor/liburing/test/hardlink.c +170 -0
  181. data/vendor/liburing/test/helpers.c +318 -0
  182. data/vendor/liburing/test/helpers.h +108 -0
  183. data/vendor/liburing/test/ignore-single-mmap.c +48 -0
  184. data/vendor/liburing/test/init-mem.c +164 -0
  185. data/vendor/liburing/test/io-cancel.c +561 -0
  186. data/vendor/liburing/test/io_uring_enter.c +264 -0
  187. data/vendor/liburing/test/io_uring_passthrough.c +482 -0
  188. data/vendor/liburing/test/io_uring_register.c +503 -0
  189. data/vendor/liburing/test/io_uring_setup.c +110 -0
  190. data/vendor/liburing/test/iopoll-leak.c +85 -0
  191. data/vendor/liburing/test/iopoll-overflow.c +118 -0
  192. data/vendor/liburing/test/iopoll.c +465 -0
  193. data/vendor/liburing/test/lfs-openat-write.c +119 -0
  194. data/vendor/liburing/test/lfs-openat.c +273 -0
  195. data/vendor/liburing/test/link-timeout.c +1108 -0
  196. data/vendor/liburing/test/link.c +497 -0
  197. data/vendor/liburing/test/link_drain.c +255 -0
  198. data/vendor/liburing/test/madvise.c +195 -0
  199. data/vendor/liburing/test/min-timeout-wait.c +354 -0
  200. data/vendor/liburing/test/min-timeout.c +233 -0
  201. data/vendor/liburing/test/mkdir.c +112 -0
  202. data/vendor/liburing/test/msg-ring-fd.c +331 -0
  203. data/vendor/liburing/test/msg-ring-flags.c +212 -0
  204. data/vendor/liburing/test/msg-ring-overflow.c +159 -0
  205. data/vendor/liburing/test/msg-ring.c +467 -0
  206. data/vendor/liburing/test/multicqes_drain.c +429 -0
  207. data/vendor/liburing/test/napi-test.c +215 -0
  208. data/vendor/liburing/test/napi-test.sh +48 -0
  209. data/vendor/liburing/test/no-mmap-inval.c +42 -0
  210. data/vendor/liburing/test/nolibc.c +62 -0
  211. data/vendor/liburing/test/nop-all-sizes.c +99 -0
  212. data/vendor/liburing/test/nop.c +177 -0
  213. data/vendor/liburing/test/nvme.h +169 -0
  214. data/vendor/liburing/test/ooo-file-unreg.c +82 -0
  215. data/vendor/liburing/test/open-close.c +261 -0
  216. data/vendor/liburing/test/open-direct-link.c +188 -0
  217. data/vendor/liburing/test/open-direct-pick.c +180 -0
  218. data/vendor/liburing/test/openat2.c +312 -0
  219. data/vendor/liburing/test/personality.c +204 -0
  220. data/vendor/liburing/test/pipe-bug.c +95 -0
  221. data/vendor/liburing/test/pipe-eof.c +83 -0
  222. data/vendor/liburing/test/pipe-reuse.c +105 -0
  223. data/vendor/liburing/test/poll-cancel-all.c +496 -0
  224. data/vendor/liburing/test/poll-cancel-ton.c +135 -0
  225. data/vendor/liburing/test/poll-cancel.c +228 -0
  226. data/vendor/liburing/test/poll-link.c +221 -0
  227. data/vendor/liburing/test/poll-many.c +230 -0
  228. data/vendor/liburing/test/poll-mshot-overflow.c +265 -0
  229. data/vendor/liburing/test/poll-mshot-update.c +323 -0
  230. data/vendor/liburing/test/poll-race-mshot.c +276 -0
  231. data/vendor/liburing/test/poll-race.c +105 -0
  232. data/vendor/liburing/test/poll-ring.c +48 -0
  233. data/vendor/liburing/test/poll-v-poll.c +353 -0
  234. data/vendor/liburing/test/poll.c +327 -0
  235. data/vendor/liburing/test/probe.c +135 -0
  236. data/vendor/liburing/test/read-before-exit.c +129 -0
  237. data/vendor/liburing/test/read-mshot-empty.c +153 -0
  238. data/vendor/liburing/test/read-mshot.c +404 -0
  239. data/vendor/liburing/test/read-write.c +1013 -0
  240. data/vendor/liburing/test/recv-msgall-stream.c +398 -0
  241. data/vendor/liburing/test/recv-msgall.c +263 -0
  242. data/vendor/liburing/test/recv-multishot.c +602 -0
  243. data/vendor/liburing/test/recvsend_bundle.c +691 -0
  244. data/vendor/liburing/test/reg-fd-only.c +131 -0
  245. data/vendor/liburing/test/reg-hint.c +56 -0
  246. data/vendor/liburing/test/reg-reg-ring.c +90 -0
  247. data/vendor/liburing/test/regbuf-merge.c +91 -0
  248. data/vendor/liburing/test/register-restrictions.c +633 -0
  249. data/vendor/liburing/test/rename.c +132 -0
  250. data/vendor/liburing/test/ring-leak.c +283 -0
  251. data/vendor/liburing/test/ring-leak2.c +249 -0
  252. data/vendor/liburing/test/ringbuf-read.c +196 -0
  253. data/vendor/liburing/test/ringbuf-status.c +242 -0
  254. data/vendor/liburing/test/rsrc_tags.c +461 -0
  255. data/vendor/liburing/test/runtests-loop.sh +16 -0
  256. data/vendor/liburing/test/runtests-quiet.sh +11 -0
  257. data/vendor/liburing/test/runtests.sh +168 -0
  258. data/vendor/liburing/test/rw_merge_test.c +98 -0
  259. data/vendor/liburing/test/self.c +91 -0
  260. data/vendor/liburing/test/send-zerocopy.c +971 -0
  261. data/vendor/liburing/test/send_recv.c +412 -0
  262. data/vendor/liburing/test/send_recvmsg.c +444 -0
  263. data/vendor/liburing/test/shared-wq.c +84 -0
  264. data/vendor/liburing/test/short-read.c +75 -0
  265. data/vendor/liburing/test/shutdown.c +165 -0
  266. data/vendor/liburing/test/sigfd-deadlock.c +88 -0
  267. data/vendor/liburing/test/single-issuer.c +169 -0
  268. data/vendor/liburing/test/skip-cqe.c +428 -0
  269. data/vendor/liburing/test/socket-getsetsock-cmd.c +346 -0
  270. data/vendor/liburing/test/socket-io-cmd.c +237 -0
  271. data/vendor/liburing/test/socket-rw-eagain.c +149 -0
  272. data/vendor/liburing/test/socket-rw-offset.c +149 -0
  273. data/vendor/liburing/test/socket-rw.c +137 -0
  274. data/vendor/liburing/test/socket.c +408 -0
  275. data/vendor/liburing/test/splice.c +512 -0
  276. data/vendor/liburing/test/sq-full-cpp.cc +45 -0
  277. data/vendor/liburing/test/sq-full.c +45 -0
  278. data/vendor/liburing/test/sq-poll-dup.c +211 -0
  279. data/vendor/liburing/test/sq-poll-kthread.c +169 -0
  280. data/vendor/liburing/test/sq-poll-share.c +138 -0
  281. data/vendor/liburing/test/sq-space_left.c +159 -0
  282. data/vendor/liburing/test/sqpoll-disable-exit.c +196 -0
  283. data/vendor/liburing/test/sqpoll-exec.c +132 -0
  284. data/vendor/liburing/test/sqpoll-exit-hang.c +78 -0
  285. data/vendor/liburing/test/sqpoll-sleep.c +69 -0
  286. data/vendor/liburing/test/statx.c +172 -0
  287. data/vendor/liburing/test/stdout.c +232 -0
  288. data/vendor/liburing/test/submit-and-wait.c +108 -0
  289. data/vendor/liburing/test/submit-link-fail.c +156 -0
  290. data/vendor/liburing/test/submit-reuse.c +237 -0
  291. data/vendor/liburing/test/symlink.c +117 -0
  292. data/vendor/liburing/test/sync-cancel.c +235 -0
  293. data/vendor/liburing/test/teardowns.c +58 -0
  294. data/vendor/liburing/test/test.h +36 -0
  295. data/vendor/liburing/test/thread-exit.c +143 -0
  296. data/vendor/liburing/test/timeout-new.c +256 -0
  297. data/vendor/liburing/test/timeout.c +1798 -0
  298. data/vendor/liburing/test/truncate.c +186 -0
  299. data/vendor/liburing/test/tty-write-dpoll.c +60 -0
  300. data/vendor/liburing/test/unlink.c +112 -0
  301. data/vendor/liburing/test/version.c +25 -0
  302. data/vendor/liburing/test/wait-timeout.c +287 -0
  303. data/vendor/liburing/test/waitid.c +373 -0
  304. data/vendor/liburing/test/wakeup-hang.c +162 -0
  305. data/vendor/liburing/test/wq-aff.c +146 -0
  306. data/vendor/liburing/test/xattr.c +442 -0
  307. metadata +412 -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
+ }