[31bb2e1] | 1 | //
|
---|
| 2 | // Cforall Version 1.0.0 Copyright (C) 2020 University of Waterloo
|
---|
| 3 | //
|
---|
| 4 | // The contents of this file are covered under the licence agreement in the
|
---|
| 5 | // file "LICENCE" distributed with Cforall.
|
---|
| 6 | //
|
---|
| 7 | // iocall.cfa --
|
---|
| 8 | //
|
---|
| 9 | // Author : Thierry Delisle
|
---|
| 10 | // Created On : Wed Jul 1 14:51:00 2020
|
---|
| 11 | // Last Modified By :
|
---|
| 12 | // Last Modified On :
|
---|
| 13 | // Update Count :
|
---|
| 14 | //
|
---|
| 15 |
|
---|
[5877b3e] | 16 | #include "bits/defs.hfa"
|
---|
| 17 |
|
---|
[4b84e35] | 18 | //=============================================================================================
|
---|
| 19 | // I/O uring backend
|
---|
| 20 | //=============================================================================================
|
---|
| 21 |
|
---|
[5751a56] | 22 | #if defined(CFA_HAVE_LINUX_IO_URING_H)
|
---|
[31bb2e1] | 23 | #include <stdint.h>
|
---|
| 24 | #include <linux/io_uring.h>
|
---|
| 25 |
|
---|
| 26 | #include "kernel_private.hfa"
|
---|
| 27 |
|
---|
| 28 | extern [* struct io_uring_sqe, uint32_t] __submit_alloc( struct __io_data & ring, uint64_t data );
|
---|
| 29 | extern void __submit( struct __io_data & ring, uint32_t idx );
|
---|
| 30 |
|
---|
[4b84e35] | 31 | static inline void ?{}(struct io_uring_sqe & this, uint8_t opcode, int fd) {
|
---|
| 32 | this.opcode = opcode;
|
---|
| 33 | #if !defined(IOSQE_ASYNC)
|
---|
| 34 | this.flags = 0;
|
---|
| 35 | #else
|
---|
| 36 | this.flags = IOSQE_ASYNC;
|
---|
| 37 | #endif
|
---|
| 38 | this.ioprio = 0;
|
---|
| 39 | this.fd = fd;
|
---|
| 40 | this.off = 0;
|
---|
| 41 | this.addr = 0;
|
---|
| 42 | this.len = 0;
|
---|
| 43 | this.rw_flags = 0;
|
---|
| 44 | this.__pad2[0] = this.__pad2[1] = this.__pad2[2] = 0;
|
---|
| 45 | }
|
---|
| 46 |
|
---|
| 47 | static inline void ?{}(struct io_uring_sqe & this, uint8_t opcode, int fd, void * addr, uint32_t len, uint64_t off ) {
|
---|
| 48 | (this){ opcode, fd };
|
---|
| 49 | this.off = off;
|
---|
| 50 | this.addr = (uint64_t)(uintptr_t)addr;
|
---|
| 51 | this.len = len;
|
---|
| 52 | }
|
---|
[31bb2e1] | 53 |
|
---|
| 54 | #define __submit_prelude \
|
---|
| 55 | __io_user_data_t data = { 0, active_thread() }; \
|
---|
| 56 | struct __io_data & ring = *data.thrd->curr_cluster->io; \
|
---|
| 57 | struct io_uring_sqe * sqe; \
|
---|
| 58 | uint32_t idx; \
|
---|
| 59 | [sqe, idx] = __submit_alloc( ring, (uint64_t)(uintptr_t)&data );
|
---|
| 60 |
|
---|
| 61 | #define __submit_wait \
|
---|
| 62 | /*__cfaabi_bits_print_safe( STDERR_FILENO, "Preparing user data %p for %p\n", &data, data.thrd );*/ \
|
---|
| 63 | verify( sqe->user_data == (uint64_t)(uintptr_t)&data ); \
|
---|
| 64 | __submit( ring, idx ); \
|
---|
| 65 | park( __cfaabi_dbg_ctx ); \
|
---|
| 66 | return data.result;
|
---|
[4b84e35] | 67 | #endif
|
---|
[31bb2e1] | 68 |
|
---|
[4b84e35] | 69 | //=============================================================================================
|
---|
| 70 | // I/O Forwards
|
---|
| 71 | //=============================================================================================
|
---|
[31bb2e1] | 72 |
|
---|
[4b84e35] | 73 | // Some forward declarations
|
---|
[cf48a14] | 74 | #include <errno.h>
|
---|
[4b84e35] | 75 | #include <unistd.h>
|
---|
| 76 |
|
---|
| 77 | extern "C" {
|
---|
| 78 | #include <sys/types.h>
|
---|
| 79 | #include <sys/socket.h>
|
---|
| 80 | #include <sys/syscall.h>
|
---|
| 81 |
|
---|
| 82 | #if defined(HAVE_PREADV2)
|
---|
| 83 | struct iovec;
|
---|
| 84 | extern ssize_t preadv2 (int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags);
|
---|
| 85 | #endif
|
---|
| 86 | #if defined(HAVE_PWRITEV2)
|
---|
| 87 | struct iovec;
|
---|
| 88 | extern ssize_t pwritev2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags);
|
---|
| 89 | #endif
|
---|
| 90 |
|
---|
| 91 | extern int fsync(int fd);
|
---|
| 92 | extern int sync_file_range(int fd, int64_t offset, int64_t nbytes, unsigned int flags);
|
---|
| 93 |
|
---|
| 94 | struct msghdr;
|
---|
| 95 | struct sockaddr;
|
---|
| 96 | extern ssize_t sendmsg(int sockfd, const struct msghdr *msg, int flags);
|
---|
| 97 | extern ssize_t recvmsg(int sockfd, struct msghdr *msg, int flags);
|
---|
| 98 | extern ssize_t send(int sockfd, const void *buf, size_t len, int flags);
|
---|
| 99 | extern ssize_t recv(int sockfd, void *buf, size_t len, int flags);
|
---|
| 100 | extern int accept4(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags);
|
---|
| 101 | extern int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
|
---|
| 102 |
|
---|
| 103 | extern int fallocate(int fd, int mode, uint64_t offset, uint64_t len);
|
---|
| 104 | extern int posix_fadvise(int fd, uint64_t offset, uint64_t len, int advice);
|
---|
| 105 | extern int madvise(void *addr, size_t length, int advice);
|
---|
| 106 |
|
---|
| 107 | extern int openat(int dirfd, const char *pathname, int flags, mode_t mode);
|
---|
| 108 | extern int close(int fd);
|
---|
| 109 |
|
---|
| 110 | extern ssize_t read (int fd, void *buf, size_t count);
|
---|
[0a92c78] | 111 |
|
---|
| 112 | extern ssize_t splice(int fd_in, loff_t *off_in, int fd_out, loff_t *off_out, size_t len, unsigned int flags);
|
---|
| 113 | extern ssize_t tee(int fd_in, int fd_out, size_t len, unsigned int flags);
|
---|
[4b84e35] | 114 | }
|
---|
[31bb2e1] | 115 |
|
---|
[4b84e35] | 116 | //=============================================================================================
|
---|
| 117 | // I/O Interface
|
---|
| 118 | //=============================================================================================
|
---|
[31bb2e1] | 119 |
|
---|
[4b84e35] | 120 | //-----------------------------------------------------------------------------
|
---|
| 121 | // Asynchronous operations
|
---|
| 122 | #if defined(HAVE_PREADV2)
|
---|
| 123 | ssize_t cfa_preadv2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags) {
|
---|
[5751a56] | 124 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_READV)
|
---|
[4b84e35] | 125 | return preadv2(fd, iov, iovcnt, offset, flags);
|
---|
[31bb2e1] | 126 | #else
|
---|
| 127 | __submit_prelude
|
---|
| 128 |
|
---|
[4b84e35] | 129 | (*sqe){ IORING_OP_READV, fd, iov, iovcnt, offset };
|
---|
[31bb2e1] | 130 |
|
---|
| 131 | __submit_wait
|
---|
| 132 | #endif
|
---|
| 133 | }
|
---|
[20ab637] | 134 |
|
---|
| 135 | ssize_t cfa_preadv2_fixed(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags) {
|
---|
[5751a56] | 136 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_READV)
|
---|
[20ab637] | 137 | return preadv2(fd, iov, iovcnt, offset, flags);
|
---|
| 138 | #else
|
---|
| 139 | __submit_prelude
|
---|
| 140 |
|
---|
| 141 | (*sqe){ IORING_OP_READV, fd, iov, iovcnt, offset };
|
---|
| 142 | sqe->flags |= IOSQE_FIXED_FILE;
|
---|
| 143 |
|
---|
| 144 | __submit_wait
|
---|
| 145 | #endif
|
---|
| 146 | }
|
---|
[4b84e35] | 147 | #endif
|
---|
[31bb2e1] | 148 |
|
---|
[4b84e35] | 149 | #if defined(HAVE_PWRITEV2)
|
---|
| 150 | ssize_t cfa_pwritev2(int fd, const struct iovec *iov, int iovcnt, off_t offset, int flags) {
|
---|
[5751a56] | 151 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_WRITEV)
|
---|
[4b84e35] | 152 | return pwritev2(fd, iov, iovcnt, offset, flags);
|
---|
[31bb2e1] | 153 | #else
|
---|
| 154 | __submit_prelude
|
---|
| 155 |
|
---|
[4b84e35] | 156 | (*sqe){ IORING_OP_WRITEV, fd, iov, iovcnt, offset };
|
---|
[31bb2e1] | 157 |
|
---|
| 158 | __submit_wait
|
---|
| 159 | #endif
|
---|
| 160 | }
|
---|
[4b84e35] | 161 | #endif
|
---|
[31bb2e1] | 162 |
|
---|
[4b84e35] | 163 | int cfa_fsync(int fd) {
|
---|
[5751a56] | 164 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_FSYNC)
|
---|
[4b84e35] | 165 | return fsync(fd);
|
---|
| 166 | #else
|
---|
| 167 | __submit_prelude
|
---|
[31bb2e1] | 168 |
|
---|
[4b84e35] | 169 | (*sqe){ IORING_OP_FSYNC, fd };
|
---|
[31bb2e1] | 170 |
|
---|
[4b84e35] | 171 | __submit_wait
|
---|
| 172 | #endif
|
---|
| 173 | }
|
---|
[31bb2e1] | 174 |
|
---|
[4b84e35] | 175 | int cfa_sync_file_range(int fd, int64_t offset, int64_t nbytes, unsigned int flags) {
|
---|
[5751a56] | 176 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_SYNC_FILE_RANGE)
|
---|
[4b84e35] | 177 | return sync_file_range(fd, offset, nbytes, flags);
|
---|
| 178 | #else
|
---|
| 179 | __submit_prelude
|
---|
[31bb2e1] | 180 |
|
---|
[4b84e35] | 181 | (*sqe){ IORING_OP_SYNC_FILE_RANGE, fd };
|
---|
| 182 | sqe->off = offset;
|
---|
| 183 | sqe->len = nbytes;
|
---|
| 184 | sqe->sync_range_flags = flags;
|
---|
[31bb2e1] | 185 |
|
---|
[4b84e35] | 186 | __submit_wait
|
---|
| 187 | #endif
|
---|
| 188 | }
|
---|
[31bb2e1] | 189 |
|
---|
| 190 |
|
---|
[4b84e35] | 191 | ssize_t cfa_sendmsg(int sockfd, const struct msghdr *msg, int flags) {
|
---|
[5751a56] | 192 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_SENDMSG)
|
---|
[4b84e35] | 193 | return sendmsg(sockfd, msg, flags);
|
---|
| 194 | #else
|
---|
| 195 | __submit_prelude
|
---|
[31bb2e1] | 196 |
|
---|
[4b84e35] | 197 | (*sqe){ IORING_OP_SENDMSG, sockfd, msg, 1, 0 };
|
---|
| 198 | sqe->msg_flags = flags;
|
---|
[31bb2e1] | 199 |
|
---|
[4b84e35] | 200 | __submit_wait
|
---|
| 201 | #endif
|
---|
| 202 | }
|
---|
[31bb2e1] | 203 |
|
---|
[4b84e35] | 204 | ssize_t cfa_recvmsg(int sockfd, struct msghdr *msg, int flags) {
|
---|
[5751a56] | 205 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_RECVMSG)
|
---|
[4b84e35] | 206 | return recvmsg(sockfd, msg, flags);
|
---|
| 207 | #else
|
---|
| 208 | __submit_prelude
|
---|
[31bb2e1] | 209 |
|
---|
[4b84e35] | 210 | (*sqe){ IORING_OP_RECVMSG, sockfd, msg, 1, 0 };
|
---|
| 211 | sqe->msg_flags = flags;
|
---|
[31bb2e1] | 212 |
|
---|
[4b84e35] | 213 | __submit_wait
|
---|
| 214 | #endif
|
---|
| 215 | }
|
---|
[31bb2e1] | 216 |
|
---|
[4b84e35] | 217 | ssize_t cfa_send(int sockfd, const void *buf, size_t len, int flags) {
|
---|
[5751a56] | 218 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_SEND)
|
---|
[4b84e35] | 219 | return send( sockfd, buf, len, flags );
|
---|
| 220 | #else
|
---|
| 221 | __submit_prelude
|
---|
[31bb2e1] | 222 |
|
---|
[4b84e35] | 223 | (*sqe){ IORING_OP_SEND, sockfd };
|
---|
| 224 | sqe->addr = (uint64_t)buf;
|
---|
| 225 | sqe->len = len;
|
---|
| 226 | sqe->msg_flags = flags;
|
---|
[31bb2e1] | 227 |
|
---|
[4b84e35] | 228 | __submit_wait
|
---|
| 229 | #endif
|
---|
| 230 | }
|
---|
[31bb2e1] | 231 |
|
---|
[4b84e35] | 232 | ssize_t cfa_recv(int sockfd, void *buf, size_t len, int flags) {
|
---|
[5751a56] | 233 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_RECV)
|
---|
[4b84e35] | 234 | return recv( sockfd, buf, len, flags );
|
---|
| 235 | #else
|
---|
| 236 | __submit_prelude
|
---|
[31bb2e1] | 237 |
|
---|
[4b84e35] | 238 | (*sqe){ IORING_OP_RECV, sockfd };
|
---|
| 239 | sqe->addr = (uint64_t)buf;
|
---|
| 240 | sqe->len = len;
|
---|
| 241 | sqe->msg_flags = flags;
|
---|
[31bb2e1] | 242 |
|
---|
[4b84e35] | 243 | __submit_wait
|
---|
| 244 | #endif
|
---|
| 245 | }
|
---|
[31bb2e1] | 246 |
|
---|
[4b84e35] | 247 | int cfa_accept4(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags) {
|
---|
[5751a56] | 248 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_ACCEPT)
|
---|
[4b84e35] | 249 | return accept4( sockfd, addr, addrlen, flags );
|
---|
| 250 | #else
|
---|
| 251 | __submit_prelude
|
---|
[31bb2e1] | 252 |
|
---|
[4b84e35] | 253 | (*sqe){ IORING_OP_ACCEPT, sockfd };
|
---|
| 254 | sqe->addr = addr;
|
---|
| 255 | sqe->addr2 = addrlen;
|
---|
| 256 | sqe->accept_flags = flags;
|
---|
[31bb2e1] | 257 |
|
---|
[4b84e35] | 258 | __submit_wait
|
---|
| 259 | #endif
|
---|
| 260 | }
|
---|
[31bb2e1] | 261 |
|
---|
[4b84e35] | 262 | int cfa_connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen) {
|
---|
[5751a56] | 263 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_CONNECT)
|
---|
[4b84e35] | 264 | return connect( sockfd, addr, addrlen );
|
---|
| 265 | #else
|
---|
| 266 | __submit_prelude
|
---|
[31bb2e1] | 267 |
|
---|
[4b84e35] | 268 | (*sqe){ IORING_OP_CONNECT, sockfd };
|
---|
| 269 | sqe->addr = (uint64_t)addr;
|
---|
| 270 | sqe->off = addrlen;
|
---|
[31bb2e1] | 271 |
|
---|
[4b84e35] | 272 | __submit_wait
|
---|
| 273 | #endif
|
---|
| 274 | }
|
---|
[31bb2e1] | 275 |
|
---|
[4b84e35] | 276 | int cfa_fallocate(int fd, int mode, uint64_t offset, uint64_t len) {
|
---|
[5751a56] | 277 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_FALLOCATE)
|
---|
[4b84e35] | 278 | return fallocate( fd, mode, offset, len );
|
---|
| 279 | #else
|
---|
| 280 | __submit_prelude
|
---|
[31bb2e1] | 281 |
|
---|
[4b84e35] | 282 | (*sqe){ IORING_OP_FALLOCATE, fd };
|
---|
| 283 | sqe->off = offset;
|
---|
| 284 | sqe->len = length;
|
---|
| 285 | sqe->mode = mode;
|
---|
[31bb2e1] | 286 |
|
---|
[4b84e35] | 287 | __submit_wait
|
---|
| 288 | #endif
|
---|
| 289 | }
|
---|
[31bb2e1] | 290 |
|
---|
[4b84e35] | 291 | int cfa_fadvise(int fd, uint64_t offset, uint64_t len, int advice) {
|
---|
[5751a56] | 292 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_FADVISE)
|
---|
[4b84e35] | 293 | return posix_fadvise( fd, offset, len, advice );
|
---|
| 294 | #else
|
---|
| 295 | __submit_prelude
|
---|
[31bb2e1] | 296 |
|
---|
[4b84e35] | 297 | (*sqe){ IORING_OP_FADVISE, fd };
|
---|
| 298 | sqe->off = (uint64_t)offset;
|
---|
| 299 | sqe->len = length;
|
---|
| 300 | sqe->fadvise_advice = advice;
|
---|
[31bb2e1] | 301 |
|
---|
[4b84e35] | 302 | __submit_wait
|
---|
| 303 | #endif
|
---|
| 304 | }
|
---|
[31bb2e1] | 305 |
|
---|
[4b84e35] | 306 | int cfa_madvise(void *addr, size_t length, int advice) {
|
---|
[5751a56] | 307 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_MADVISE)
|
---|
[4b84e35] | 308 | return madvise( addr, length, advice );
|
---|
| 309 | #else
|
---|
| 310 | __submit_prelude
|
---|
[31bb2e1] | 311 |
|
---|
[4b84e35] | 312 | (*sqe){ IORING_OP_MADVISE, 0 };
|
---|
| 313 | sqe->addr = (uint64_t)addr;
|
---|
| 314 | sqe->len = length;
|
---|
| 315 | sqe->fadvise_advice = advice;
|
---|
[31bb2e1] | 316 |
|
---|
[4b84e35] | 317 | __submit_wait
|
---|
| 318 | #endif
|
---|
| 319 | }
|
---|
[31bb2e1] | 320 |
|
---|
[4b84e35] | 321 | int cfa_openat(int dirfd, const char *pathname, int flags, mode_t mode) {
|
---|
[5751a56] | 322 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_OPENAT)
|
---|
[4b84e35] | 323 | return openat( dirfd, pathname, flags, mode );
|
---|
| 324 | #else
|
---|
| 325 | __submit_prelude
|
---|
[31bb2e1] | 326 |
|
---|
[4b84e35] | 327 | (*sqe){ IORING_OP_OPENAT, dirfd };
|
---|
| 328 | sqe->addr = (uint64_t)pathname;
|
---|
| 329 | sqe->open_flags = flags;
|
---|
| 330 | sqe->mode = mode;
|
---|
[31bb2e1] | 331 |
|
---|
[4b84e35] | 332 | __submit_wait
|
---|
| 333 | #endif
|
---|
| 334 | }
|
---|
[31bb2e1] | 335 |
|
---|
[4b84e35] | 336 | int cfa_close(int fd) {
|
---|
[5751a56] | 337 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_CLOSE)
|
---|
[4b84e35] | 338 | return close( fd );
|
---|
| 339 | #else
|
---|
| 340 | __submit_prelude
|
---|
[31bb2e1] | 341 |
|
---|
[4b84e35] | 342 | (*sqe){ IORING_OP_CLOSE, fd };
|
---|
| 343 |
|
---|
| 344 | __submit_wait
|
---|
| 345 | #endif
|
---|
| 346 | }
|
---|
[31bb2e1] | 347 |
|
---|
[efc171d1] | 348 | // Forward declare in case it is not supported
|
---|
| 349 | struct statx;
|
---|
[519f11c] | 350 | int cfa_statx(int dirfd, const char *pathname, int flags, unsigned int mask, struct statx *statxbuf) {
|
---|
[5751a56] | 351 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_STATX)
|
---|
[5db836e] | 352 | #if defined(__NR_statx)
|
---|
| 353 | return syscall( __NR_statx, dirfd, pathname, flags, mask, statxbuf );
|
---|
| 354 | #else
|
---|
| 355 | errno = ENOTSUP;
|
---|
| 356 | return -1;
|
---|
| 357 | #endif
|
---|
[519f11c] | 358 | #else
|
---|
| 359 | __submit_prelude
|
---|
| 360 |
|
---|
| 361 | (*sqe){ IORING_OP_STATX, dirfd, pathname, mask, (uint64_t)statxbuf };
|
---|
| 362 | sqe->flags = flags;
|
---|
| 363 |
|
---|
| 364 | __submit_wait
|
---|
| 365 | #endif
|
---|
| 366 | }
|
---|
| 367 |
|
---|
[4b84e35] | 368 | ssize_t cfa_read(int fd, void *buf, size_t count) {
|
---|
[5751a56] | 369 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_READ)
|
---|
[4b84e35] | 370 | return read( fd, buf, count );
|
---|
| 371 | #else
|
---|
| 372 | __submit_prelude
|
---|
| 373 |
|
---|
| 374 | (*sqe){ IORING_OP_READ, fd, buf, count, 0 };
|
---|
| 375 |
|
---|
| 376 | __submit_wait
|
---|
| 377 | #endif
|
---|
| 378 | }
|
---|
| 379 |
|
---|
| 380 | ssize_t cfa_write(int fd, void *buf, size_t count) {
|
---|
[5751a56] | 381 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_WRITE)
|
---|
[4b84e35] | 382 | return read( fd, buf, count );
|
---|
| 383 | #else
|
---|
| 384 | __submit_prelude
|
---|
| 385 |
|
---|
| 386 | (*sqe){ IORING_OP_WRITE, fd, buf, count, 0 };
|
---|
| 387 |
|
---|
| 388 | __submit_wait
|
---|
| 389 | #endif
|
---|
| 390 | }
|
---|
| 391 |
|
---|
[0a92c78] | 392 | ssize_t cfa_splice(int fd_in, loff_t *off_in, int fd_out, loff_t *off_out, size_t len, unsigned int flags) {
|
---|
[5751a56] | 393 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_SPLICE)
|
---|
[0a92c78] | 394 | return splice( fd_in, off_in, fd_out, off_out, len, flags );
|
---|
| 395 | #else
|
---|
| 396 | __submit_prelude
|
---|
| 397 |
|
---|
| 398 | (*sqe){ IORING_OP_SPLICE, fd_out, 0p, len, off_out };
|
---|
| 399 | sqe->splice_fd_in = fd_in;
|
---|
| 400 | sqe->splice_off_in = off_in;
|
---|
| 401 | sqe->splice_flags = flags;
|
---|
| 402 |
|
---|
| 403 | __submit_wait
|
---|
| 404 | #endif
|
---|
| 405 | }
|
---|
| 406 |
|
---|
[5751a56] | 407 | ssize_t cfa_splice(int fd_in, loff_t *off_in, int fd_out, loff_t *off_out, size_t len, unsigned int flags, int in_flags, int out_flags) {
|
---|
| 408 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_SPLICE)
|
---|
| 409 | return splice( fd_in, off_in, fd_out, off_out, len, flags );
|
---|
| 410 | #else
|
---|
| 411 | __submit_prelude
|
---|
| 412 |
|
---|
| 413 | (*sqe){ IORING_OP_SPLICE, fd_out, 0p, len, off_out };
|
---|
| 414 | sqe->splice_fd_in = fd_in;
|
---|
| 415 | sqe->splice_off_in = off_in;
|
---|
| 416 | sqe->splice_flags = flags | out_flags;
|
---|
| 417 | sqe->flags = in_flags;
|
---|
| 418 |
|
---|
| 419 | __submit_wait
|
---|
| 420 | #endif
|
---|
| 421 | }
|
---|
| 422 |
|
---|
[0a92c78] | 423 | ssize_t cfa_tee(int fd_in, int fd_out, size_t len, unsigned int flags) {
|
---|
[5751a56] | 424 | #if !defined(CFA_HAVE_LINUX_IO_URING_H) || !defined(CFA_HAVE_IORING_OP_TEE)
|
---|
[0a92c78] | 425 | return tee( fd_in, fd_out, len, flags );
|
---|
| 426 | #else
|
---|
| 427 | __submit_prelude
|
---|
| 428 |
|
---|
| 429 | (*sqe){ IORING_OP_TEE, fd_out, 0p, len, 0 };
|
---|
| 430 | sqe->splice_fd_in = fd_in;
|
---|
| 431 | sqe->splice_flags = flags;
|
---|
| 432 |
|
---|
| 433 | __submit_wait
|
---|
| 434 | #endif
|
---|
| 435 | }
|
---|
| 436 |
|
---|
[4b84e35] | 437 | //-----------------------------------------------------------------------------
|
---|
| 438 | // Check if a function is asynchronous
|
---|
| 439 |
|
---|
| 440 | // Macro magic to reduce the size of the following switch case
|
---|
| 441 | #define IS_DEFINED_APPLY(f, ...) f(__VA_ARGS__)
|
---|
| 442 | #define IS_DEFINED_SECOND(first, second, ...) second
|
---|
| 443 | #define IS_DEFINED_TEST(expansion) _CFA_IO_FEATURE_##expansion
|
---|
| 444 | #define IS_DEFINED(macro) IS_DEFINED_APPLY( IS_DEFINED_SECOND,IS_DEFINED_TEST(macro) false, true)
|
---|
| 445 |
|
---|
| 446 | bool has_user_level_blocking( fptr_t func ) {
|
---|
[5751a56] | 447 | #if defined(CFA_HAVE_LINUX_IO_URING_H)
|
---|
[4b84e35] | 448 | #if defined(HAVE_PREADV2)
|
---|
| 449 | if( /*func == (fptr_t)preadv2 || */
|
---|
| 450 | func == (fptr_t)cfa_preadv2 )
|
---|
[5751a56] | 451 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_READV ,
|
---|
| 452 | return IS_DEFINED(CFA_HAVE_IORING_OP_READV);
|
---|
[31bb2e1] | 453 | #endif
|
---|
| 454 |
|
---|
[4b84e35] | 455 | #if defined(HAVE_PWRITEV2)
|
---|
| 456 | if( /*func == (fptr_t)pwritev2 || */
|
---|
| 457 | func == (fptr_t)cfa_pwritev2 )
|
---|
[5751a56] | 458 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_WRITEV ,
|
---|
| 459 | return IS_DEFINED(CFA_HAVE_IORING_OP_WRITEV);
|
---|
[31bb2e1] | 460 | #endif
|
---|
| 461 |
|
---|
[4b84e35] | 462 | if( /*func == (fptr_t)fsync || */
|
---|
| 463 | func == (fptr_t)cfa_fsync )
|
---|
[5751a56] | 464 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_FSYNC ,
|
---|
| 465 | return IS_DEFINED(CFA_HAVE_IORING_OP_FSYNC);
|
---|
[4b84e35] | 466 |
|
---|
| 467 | if( /*func == (fptr_t)ync_file_range || */
|
---|
| 468 | func == (fptr_t)cfa_sync_file_range )
|
---|
[5751a56] | 469 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_SYNC_FILE_RANGE ,
|
---|
| 470 | return IS_DEFINED(CFA_HAVE_IORING_OP_SYNC_FILE_RANGE);
|
---|
[4b84e35] | 471 |
|
---|
| 472 | if( /*func == (fptr_t)sendmsg || */
|
---|
| 473 | func == (fptr_t)cfa_sendmsg )
|
---|
[5751a56] | 474 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_SENDMSG ,
|
---|
| 475 | return IS_DEFINED(CFA_HAVE_IORING_OP_SENDMSG);
|
---|
[4b84e35] | 476 |
|
---|
| 477 | if( /*func == (fptr_t)recvmsg || */
|
---|
| 478 | func == (fptr_t)cfa_recvmsg )
|
---|
[5751a56] | 479 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_RECVMSG ,
|
---|
| 480 | return IS_DEFINED(CFA_HAVE_IORING_OP_RECVMSG);
|
---|
[4b84e35] | 481 |
|
---|
| 482 | if( /*func == (fptr_t)send || */
|
---|
| 483 | func == (fptr_t)cfa_send )
|
---|
[5751a56] | 484 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_SEND ,
|
---|
| 485 | return IS_DEFINED(CFA_HAVE_IORING_OP_SEND);
|
---|
[4b84e35] | 486 |
|
---|
| 487 | if( /*func == (fptr_t)recv || */
|
---|
| 488 | func == (fptr_t)cfa_recv )
|
---|
[5751a56] | 489 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_RECV ,
|
---|
| 490 | return IS_DEFINED(CFA_HAVE_IORING_OP_RECV);
|
---|
[4b84e35] | 491 |
|
---|
| 492 | if( /*func == (fptr_t)accept4 || */
|
---|
| 493 | func == (fptr_t)cfa_accept4 )
|
---|
[5751a56] | 494 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_ACCEPT ,
|
---|
| 495 | return IS_DEFINED(CFA_HAVE_IORING_OP_ACCEPT);
|
---|
[4b84e35] | 496 |
|
---|
| 497 | if( /*func == (fptr_t)connect || */
|
---|
| 498 | func == (fptr_t)cfa_connect )
|
---|
[5751a56] | 499 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_CONNECT ,
|
---|
| 500 | return IS_DEFINED(CFA_HAVE_IORING_OP_CONNECT);
|
---|
[4b84e35] | 501 |
|
---|
| 502 | if( /*func == (fptr_t)fallocate || */
|
---|
| 503 | func == (fptr_t)cfa_fallocate )
|
---|
[5751a56] | 504 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_FALLOCATE ,
|
---|
| 505 | return IS_DEFINED(CFA_HAVE_IORING_OP_FALLOCATE);
|
---|
[4b84e35] | 506 |
|
---|
| 507 | if( /*func == (fptr_t)posix_fadvise || */
|
---|
| 508 | func == (fptr_t)cfa_fadvise )
|
---|
[5751a56] | 509 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_FADVISE ,
|
---|
| 510 | return IS_DEFINED(CFA_HAVE_IORING_OP_FADVISE);
|
---|
[4b84e35] | 511 |
|
---|
| 512 | if( /*func == (fptr_t)madvise || */
|
---|
| 513 | func == (fptr_t)cfa_madvise )
|
---|
[5751a56] | 514 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_MADVISE ,
|
---|
| 515 | return IS_DEFINED(CFA_HAVE_IORING_OP_MADVISE);
|
---|
[4b84e35] | 516 |
|
---|
| 517 | if( /*func == (fptr_t)openat || */
|
---|
| 518 | func == (fptr_t)cfa_openat )
|
---|
[5751a56] | 519 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_OPENAT ,
|
---|
| 520 | return IS_DEFINED(CFA_HAVE_IORING_OP_OPENAT);
|
---|
[4b84e35] | 521 |
|
---|
| 522 | if( /*func == (fptr_t)close || */
|
---|
| 523 | func == (fptr_t)cfa_close )
|
---|
[5751a56] | 524 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_CLOSE ,
|
---|
| 525 | return IS_DEFINED(CFA_HAVE_IORING_OP_CLOSE);
|
---|
[4b84e35] | 526 |
|
---|
| 527 | if( /*func == (fptr_t)read || */
|
---|
| 528 | func == (fptr_t)cfa_read )
|
---|
[5751a56] | 529 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_READ ,
|
---|
| 530 | return IS_DEFINED(CFA_HAVE_IORING_OP_READ);
|
---|
[4b84e35] | 531 |
|
---|
| 532 | if( /*func == (fptr_t)write || */
|
---|
| 533 | func == (fptr_t)cfa_write )
|
---|
[5751a56] | 534 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_WRITE ,
|
---|
| 535 | return IS_DEFINED(CFA_HAVE_IORING_OP_WRITE);
|
---|
[0a92c78] | 536 |
|
---|
| 537 | if( /*func == (fptr_t)splice || */
|
---|
[5751a56] | 538 | func == (fptr_t)(ssize_t (*)(int, loff_t *, int, loff_t *, size_t, unsigned int))cfa_splice,
|
---|
| 539 | func == (fptr_t)(ssize_t (*)(int, loff_t *, int, loff_t *, size_t, unsigned int, int, int))cfa_splice )
|
---|
| 540 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_SPLICE ,
|
---|
| 541 | return IS_DEFINED(CFA_HAVE_IORING_OP_SPLICE);
|
---|
[0a92c78] | 542 |
|
---|
| 543 | if( /*func == (fptr_t)tee || */
|
---|
| 544 | func == (fptr_t)cfa_tee )
|
---|
[5751a56] | 545 | #define _CFA_IO_FEATURE_CFA_HAVE_IORING_OP_TEE ,
|
---|
| 546 | return IS_DEFINED(CFA_HAVE_IORING_OP_TEE);
|
---|
[4b84e35] | 547 | #endif
|
---|
| 548 |
|
---|
| 549 | return false;
|
---|
| 550 | }
|
---|