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,512 @@
1
+ /* SPDX-License-Identifier: MIT */
2
+ #include <errno.h>
3
+ #include <stdio.h>
4
+ #include <unistd.h>
5
+ #include <stdlib.h>
6
+ #include <string.h>
7
+ #include <fcntl.h>
8
+ #include <sys/mman.h>
9
+
10
+ #include "helpers.h"
11
+ #include "liburing.h"
12
+
13
+ #define BUF_SIZE (16 * 4096)
14
+
15
+ struct test_ctx {
16
+ int real_pipe1[2];
17
+ int real_pipe2[2];
18
+ int real_fd_in;
19
+ int real_fd_out;
20
+
21
+ /* fds or for registered files */
22
+ int pipe1[2];
23
+ int pipe2[2];
24
+ int fd_in;
25
+ int fd_out;
26
+
27
+ void *buf_in;
28
+ void *buf_out;
29
+ };
30
+
31
+ static unsigned int splice_flags = 0;
32
+ static unsigned int sqe_flags = 0;
33
+ static int has_splice = 0;
34
+ static int has_tee = 0;
35
+
36
+ static int read_buf(int fd, void *buf, int len)
37
+ {
38
+ int ret;
39
+
40
+ while (len) {
41
+ ret = read(fd, buf, len);
42
+ if (ret < 0)
43
+ return ret;
44
+ len -= ret;
45
+ buf += ret;
46
+ }
47
+ return 0;
48
+ }
49
+
50
+ static int write_buf(int fd, const void *buf, int len)
51
+ {
52
+ int ret;
53
+
54
+ while (len) {
55
+ ret = write(fd, buf, len);
56
+ if (ret < 0)
57
+ return ret;
58
+ len -= ret;
59
+ buf += ret;
60
+ }
61
+ return 0;
62
+ }
63
+
64
+ static int check_content(int fd, void *buf, int len, const void *src)
65
+ {
66
+ int ret;
67
+
68
+ ret = read_buf(fd, buf, len);
69
+ if (ret)
70
+ return ret;
71
+
72
+ ret = memcmp(buf, src, len);
73
+ return (ret != 0) ? -1 : 0;
74
+ }
75
+
76
+ static int create_file(const char *filename)
77
+ {
78
+ int fd, save_errno;
79
+
80
+ fd = open(filename, O_RDWR | O_CREAT, 0644);
81
+ save_errno = errno;
82
+ unlink(filename);
83
+ errno = save_errno;
84
+ return fd;
85
+ }
86
+
87
+ static int init_splice_ctx(struct test_ctx *ctx)
88
+ {
89
+ int ret, rnd_fd;
90
+
91
+ ctx->buf_in = t_calloc(BUF_SIZE, 1);
92
+ ctx->buf_out = t_calloc(BUF_SIZE, 1);
93
+
94
+ ctx->fd_in = create_file(".splice-test-in");
95
+ if (ctx->fd_in < 0) {
96
+ perror("file open");
97
+ return 1;
98
+ }
99
+
100
+ ctx->fd_out = create_file(".splice-test-out");
101
+ if (ctx->fd_out < 0) {
102
+ perror("file open");
103
+ return 1;
104
+ }
105
+
106
+ /* get random data */
107
+ rnd_fd = open("/dev/urandom", O_RDONLY);
108
+ if (rnd_fd < 0)
109
+ return 1;
110
+
111
+ ret = read_buf(rnd_fd, ctx->buf_in, BUF_SIZE);
112
+ if (ret != 0)
113
+ return 1;
114
+ close(rnd_fd);
115
+
116
+ /* populate file */
117
+ ret = write_buf(ctx->fd_in, ctx->buf_in, BUF_SIZE);
118
+ if (ret)
119
+ return ret;
120
+
121
+ if (pipe(ctx->pipe1) < 0)
122
+ return 1;
123
+ if (pipe(ctx->pipe2) < 0)
124
+ return 1;
125
+
126
+ ctx->real_pipe1[0] = ctx->pipe1[0];
127
+ ctx->real_pipe1[1] = ctx->pipe1[1];
128
+ ctx->real_pipe2[0] = ctx->pipe2[0];
129
+ ctx->real_pipe2[1] = ctx->pipe2[1];
130
+ ctx->real_fd_in = ctx->fd_in;
131
+ ctx->real_fd_out = ctx->fd_out;
132
+ return 0;
133
+ }
134
+
135
+ static int do_splice_op(struct io_uring *ring,
136
+ int fd_in, loff_t off_in,
137
+ int fd_out, loff_t off_out,
138
+ unsigned int len,
139
+ __u8 opcode)
140
+ {
141
+ struct io_uring_cqe *cqe;
142
+ struct io_uring_sqe *sqe;
143
+ int ret = -1;
144
+
145
+ do {
146
+ sqe = io_uring_get_sqe(ring);
147
+ if (!sqe) {
148
+ fprintf(stderr, "get sqe failed\n");
149
+ return -1;
150
+ }
151
+ io_uring_prep_splice(sqe, fd_in, off_in, fd_out, off_out,
152
+ len, splice_flags);
153
+ sqe->flags |= sqe_flags;
154
+ sqe->user_data = 42;
155
+ sqe->opcode = opcode;
156
+
157
+ ret = io_uring_submit(ring);
158
+ if (ret != 1) {
159
+ fprintf(stderr, "sqe submit failed: %d\n", ret);
160
+ return ret;
161
+ }
162
+
163
+ ret = io_uring_wait_cqe(ring, &cqe);
164
+ if (ret < 0) {
165
+ fprintf(stderr, "wait completion %d\n", cqe->res);
166
+ return ret;
167
+ }
168
+
169
+ if (cqe->res <= 0) {
170
+ io_uring_cqe_seen(ring, cqe);
171
+ return cqe->res;
172
+ }
173
+
174
+ len -= cqe->res;
175
+ if (off_in != -1)
176
+ off_in += cqe->res;
177
+ if (off_out != -1)
178
+ off_out += cqe->res;
179
+ io_uring_cqe_seen(ring, cqe);
180
+ } while (len);
181
+
182
+ return 0;
183
+ }
184
+
185
+ static int do_splice(struct io_uring *ring,
186
+ int fd_in, loff_t off_in,
187
+ int fd_out, loff_t off_out,
188
+ unsigned int len)
189
+ {
190
+ return do_splice_op(ring, fd_in, off_in, fd_out, off_out, len,
191
+ IORING_OP_SPLICE);
192
+ }
193
+
194
+ static int do_tee(struct io_uring *ring, int fd_in, int fd_out,
195
+ unsigned int len)
196
+ {
197
+ return do_splice_op(ring, fd_in, 0, fd_out, 0, len, IORING_OP_TEE);
198
+ }
199
+
200
+ static void check_splice_support(struct io_uring *ring, struct test_ctx *ctx)
201
+ {
202
+ int ret;
203
+
204
+ ret = do_splice(ring, -1, 0, -1, 0, BUF_SIZE);
205
+ has_splice = (ret == -EBADF);
206
+ }
207
+
208
+ static void check_tee_support(struct io_uring *ring, struct test_ctx *ctx)
209
+ {
210
+ int ret;
211
+
212
+ ret = do_tee(ring, -1, -1, BUF_SIZE);
213
+ has_tee = (ret == -EBADF);
214
+ }
215
+
216
+ static int check_zero_splice(struct io_uring *ring, struct test_ctx *ctx)
217
+ {
218
+ int ret;
219
+
220
+ ret = do_splice(ring, ctx->fd_in, -1, ctx->pipe1[1], -1, 0);
221
+ if (ret)
222
+ return ret;
223
+
224
+ ret = do_splice(ring, ctx->pipe2[0], -1, ctx->pipe1[1], -1, 0);
225
+ if (ret)
226
+ return ret;
227
+
228
+ return 0;
229
+ }
230
+
231
+ static int splice_to_pipe(struct io_uring *ring, struct test_ctx *ctx)
232
+ {
233
+ int ret;
234
+
235
+ ret = lseek(ctx->real_fd_in, 0, SEEK_SET);
236
+ if (ret)
237
+ return ret;
238
+
239
+ /* implicit file offset */
240
+ ret = do_splice(ring, ctx->fd_in, -1, ctx->pipe1[1], -1, BUF_SIZE);
241
+ if (ret)
242
+ return ret;
243
+
244
+ ret = check_content(ctx->real_pipe1[0], ctx->buf_out, BUF_SIZE,
245
+ ctx->buf_in);
246
+ if (ret)
247
+ return ret;
248
+
249
+ /* explicit file offset */
250
+ ret = do_splice(ring, ctx->fd_in, 0, ctx->pipe1[1], -1, BUF_SIZE);
251
+ if (ret)
252
+ return ret;
253
+
254
+ return check_content(ctx->real_pipe1[0], ctx->buf_out, BUF_SIZE,
255
+ ctx->buf_in);
256
+ }
257
+
258
+ static int splice_from_pipe(struct io_uring *ring, struct test_ctx *ctx)
259
+ {
260
+ int ret;
261
+
262
+ ret = write_buf(ctx->real_pipe1[1], ctx->buf_in, BUF_SIZE);
263
+ if (ret)
264
+ return ret;
265
+ ret = do_splice(ring, ctx->pipe1[0], -1, ctx->fd_out, 0, BUF_SIZE);
266
+ if (ret)
267
+ return ret;
268
+ ret = check_content(ctx->real_fd_out, ctx->buf_out, BUF_SIZE,
269
+ ctx->buf_in);
270
+ if (ret)
271
+ return ret;
272
+
273
+ ret = ftruncate(ctx->real_fd_out, 0);
274
+ if (ret)
275
+ return ret;
276
+ return lseek(ctx->real_fd_out, 0, SEEK_SET);
277
+ }
278
+
279
+ static int splice_pipe_to_pipe(struct io_uring *ring, struct test_ctx *ctx)
280
+ {
281
+ int ret;
282
+
283
+ ret = do_splice(ring, ctx->fd_in, 0, ctx->pipe1[1], -1, BUF_SIZE);
284
+ if (ret)
285
+ return ret;
286
+ ret = do_splice(ring, ctx->pipe1[0], -1, ctx->pipe2[1], -1, BUF_SIZE);
287
+ if (ret)
288
+ return ret;
289
+
290
+ return check_content(ctx->real_pipe2[0], ctx->buf_out, BUF_SIZE,
291
+ ctx->buf_in);
292
+ }
293
+
294
+ static int fail_splice_pipe_offset(struct io_uring *ring, struct test_ctx *ctx)
295
+ {
296
+ int ret;
297
+
298
+ ret = do_splice(ring, ctx->fd_in, 0, ctx->pipe1[1], 0, BUF_SIZE);
299
+ if (ret != -ESPIPE && ret != -EINVAL)
300
+ return ret;
301
+
302
+ ret = do_splice(ring, ctx->pipe1[0], 0, ctx->fd_out, 0, BUF_SIZE);
303
+ if (ret != -ESPIPE && ret != -EINVAL)
304
+ return ret;
305
+
306
+ return 0;
307
+ }
308
+
309
+ static int fail_tee_nonpipe(struct io_uring *ring, struct test_ctx *ctx)
310
+ {
311
+ int ret;
312
+
313
+ ret = do_tee(ring, ctx->fd_in, ctx->pipe1[1], BUF_SIZE);
314
+ if (ret != -ESPIPE && ret != -EINVAL)
315
+ return ret;
316
+
317
+ return 0;
318
+ }
319
+
320
+ static int fail_tee_offset(struct io_uring *ring, struct test_ctx *ctx)
321
+ {
322
+ int ret;
323
+
324
+ ret = do_splice_op(ring, ctx->pipe2[0], -1, ctx->pipe1[1], 0,
325
+ BUF_SIZE, IORING_OP_TEE);
326
+ if (ret != -ESPIPE && ret != -EINVAL)
327
+ return ret;
328
+
329
+ ret = do_splice_op(ring, ctx->pipe2[0], 0, ctx->pipe1[1], -1,
330
+ BUF_SIZE, IORING_OP_TEE);
331
+ if (ret != -ESPIPE && ret != -EINVAL)
332
+ return ret;
333
+
334
+ return 0;
335
+ }
336
+
337
+ static int check_tee(struct io_uring *ring, struct test_ctx *ctx)
338
+ {
339
+ int ret;
340
+
341
+ ret = write_buf(ctx->real_pipe1[1], ctx->buf_in, BUF_SIZE);
342
+ if (ret)
343
+ return ret;
344
+ ret = do_tee(ring, ctx->pipe1[0], ctx->pipe2[1], BUF_SIZE);
345
+ if (ret)
346
+ return ret;
347
+
348
+ ret = check_content(ctx->real_pipe1[0], ctx->buf_out, BUF_SIZE,
349
+ ctx->buf_in);
350
+ if (ret) {
351
+ fprintf(stderr, "tee(), invalid src data\n");
352
+ return ret;
353
+ }
354
+
355
+ ret = check_content(ctx->real_pipe2[0], ctx->buf_out, BUF_SIZE,
356
+ ctx->buf_in);
357
+ if (ret) {
358
+ fprintf(stderr, "tee(), invalid dst data\n");
359
+ return ret;
360
+ }
361
+
362
+ return 0;
363
+ }
364
+
365
+ static int check_zero_tee(struct io_uring *ring, struct test_ctx *ctx)
366
+ {
367
+ return do_tee(ring, ctx->pipe2[0], ctx->pipe1[1], 0);
368
+ }
369
+
370
+ static int test_splice(struct io_uring *ring, struct test_ctx *ctx)
371
+ {
372
+ int ret;
373
+
374
+ if (has_splice) {
375
+ ret = check_zero_splice(ring, ctx);
376
+ if (ret) {
377
+ fprintf(stderr, "check_zero_splice failed %i %i\n",
378
+ ret, errno);
379
+ return ret;
380
+ }
381
+
382
+ ret = splice_to_pipe(ring, ctx);
383
+ if (ret) {
384
+ fprintf(stderr, "splice_to_pipe failed %i %i\n",
385
+ ret, errno);
386
+ return ret;
387
+ }
388
+
389
+ ret = splice_from_pipe(ring, ctx);
390
+ if (ret) {
391
+ fprintf(stderr, "splice_from_pipe failed %i %i\n",
392
+ ret, errno);
393
+ return ret;
394
+ }
395
+
396
+ ret = splice_pipe_to_pipe(ring, ctx);
397
+ if (ret) {
398
+ fprintf(stderr, "splice_pipe_to_pipe failed %i %i\n",
399
+ ret, errno);
400
+ return ret;
401
+ }
402
+
403
+ ret = fail_splice_pipe_offset(ring, ctx);
404
+ if (ret) {
405
+ fprintf(stderr, "fail_splice_pipe_offset failed %i %i\n",
406
+ ret, errno);
407
+ return ret;
408
+ }
409
+ }
410
+
411
+ if (has_tee) {
412
+ ret = check_zero_tee(ring, ctx);
413
+ if (ret) {
414
+ fprintf(stderr, "check_zero_tee() failed %i %i\n",
415
+ ret, errno);
416
+ return ret;
417
+ }
418
+
419
+ ret = fail_tee_nonpipe(ring, ctx);
420
+ if (ret) {
421
+ fprintf(stderr, "fail_tee_nonpipe() failed %i %i\n",
422
+ ret, errno);
423
+ return ret;
424
+ }
425
+
426
+ ret = fail_tee_offset(ring, ctx);
427
+ if (ret) {
428
+ fprintf(stderr, "fail_tee_offset failed %i %i\n",
429
+ ret, errno);
430
+ return ret;
431
+ }
432
+
433
+ ret = check_tee(ring, ctx);
434
+ if (ret) {
435
+ fprintf(stderr, "check_tee() failed %i %i\n",
436
+ ret, errno);
437
+ return ret;
438
+ }
439
+ }
440
+
441
+ return 0;
442
+ }
443
+
444
+ int main(int argc, char *argv[])
445
+ {
446
+ struct io_uring ring;
447
+ struct io_uring_params p = { };
448
+ struct test_ctx ctx;
449
+ int ret;
450
+ int reg_fds[6];
451
+
452
+ if (argc > 1)
453
+ return 0;
454
+
455
+ ret = io_uring_queue_init_params(8, &ring, &p);
456
+ if (ret) {
457
+ fprintf(stderr, "ring setup failed\n");
458
+ return 1;
459
+ }
460
+ if (!(p.features & IORING_FEAT_FAST_POLL)) {
461
+ fprintf(stdout, "No splice support, skipping\n");
462
+ return 0;
463
+ }
464
+
465
+ ret = init_splice_ctx(&ctx);
466
+ if (ret) {
467
+ fprintf(stderr, "init failed %i %i\n", ret, errno);
468
+ return 1;
469
+ }
470
+
471
+ check_splice_support(&ring, &ctx);
472
+ if (!has_splice)
473
+ fprintf(stdout, "skip, doesn't support splice()\n");
474
+ check_tee_support(&ring, &ctx);
475
+ if (!has_tee)
476
+ fprintf(stdout, "skip, doesn't support tee()\n");
477
+
478
+ ret = test_splice(&ring, &ctx);
479
+ if (ret) {
480
+ fprintf(stderr, "basic splice tests failed\n");
481
+ return ret;
482
+ }
483
+
484
+ reg_fds[0] = ctx.real_pipe1[0];
485
+ reg_fds[1] = ctx.real_pipe1[1];
486
+ reg_fds[2] = ctx.real_pipe2[0];
487
+ reg_fds[3] = ctx.real_pipe2[1];
488
+ reg_fds[4] = ctx.real_fd_in;
489
+ reg_fds[5] = ctx.real_fd_out;
490
+ ret = io_uring_register_files(&ring, reg_fds, 6);
491
+ if (ret) {
492
+ fprintf(stderr, "%s: register ret=%d\n", __FUNCTION__, ret);
493
+ return 1;
494
+ }
495
+
496
+ /* remap fds to registered */
497
+ ctx.pipe1[0] = 0;
498
+ ctx.pipe1[1] = 1;
499
+ ctx.pipe2[0] = 2;
500
+ ctx.pipe2[1] = 3;
501
+ ctx.fd_in = 4;
502
+ ctx.fd_out = 5;
503
+
504
+ splice_flags = SPLICE_F_FD_IN_FIXED;
505
+ sqe_flags = IOSQE_FIXED_FILE;
506
+ ret = test_splice(&ring, &ctx);
507
+ if (ret) {
508
+ fprintf(stderr, "registered fds splice tests failed\n");
509
+ return ret;
510
+ }
511
+ return 0;
512
+ }
@@ -0,0 +1,45 @@
1
+ /* SPDX-License-Identifier: MIT */
2
+ /*
3
+ * Description: test SQ queue full condition
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
+ int main(int argc, char *argv[])
16
+ {
17
+ struct io_uring_sqe *sqe;
18
+ struct io_uring ring;
19
+ int ret, i;
20
+
21
+ if (argc > 1)
22
+ return 0;
23
+
24
+ ret = io_uring_queue_init(8, &ring, 0);
25
+ if (ret) {
26
+ fprintf(stderr, "ring setup failed: %d\n", ret);
27
+ return 1;
28
+
29
+ }
30
+
31
+ i = 0;
32
+ while ((sqe = io_uring_get_sqe(&ring)) != NULL)
33
+ i++;
34
+
35
+ if (i != 8) {
36
+ fprintf(stderr, "Got %d SQEs, wanted 8\n", i);
37
+ goto err;
38
+ }
39
+
40
+ io_uring_queue_exit(&ring);
41
+ return 0;
42
+ err:
43
+ io_uring_queue_exit(&ring);
44
+ return 1;
45
+ }
@@ -0,0 +1,45 @@
1
+ /* SPDX-License-Identifier: MIT */
2
+ /*
3
+ * Description: test SQ queue full condition
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
+ int main(int argc, char *argv[])
16
+ {
17
+ struct io_uring_sqe *sqe;
18
+ struct io_uring ring;
19
+ int ret, i;
20
+
21
+ if (argc > 1)
22
+ return 0;
23
+
24
+ ret = io_uring_queue_init(8, &ring, 0);
25
+ if (ret) {
26
+ fprintf(stderr, "ring setup failed: %d\n", ret);
27
+ return 1;
28
+
29
+ }
30
+
31
+ i = 0;
32
+ while ((sqe = io_uring_get_sqe(&ring)) != NULL)
33
+ i++;
34
+
35
+ if (i != 8) {
36
+ fprintf(stderr, "Got %d SQEs, wanted 8\n", i);
37
+ goto err;
38
+ }
39
+
40
+ io_uring_queue_exit(&ring);
41
+ return 0;
42
+ err:
43
+ io_uring_queue_exit(&ring);
44
+ return 1;
45
+ }