io_uring: update to newer API
authorJens Axboe <axboe@kernel.dk>
Fri, 11 Jan 2019 17:33:28 +0000 (10:33 -0700)
committerJens Axboe <axboe@kernel.dk>
Fri, 11 Jan 2019 17:33:28 +0000 (10:33 -0700)
Signed-off-by: Jens Axboe <axboe@kernel.dk>
engines/io_uring.c
os/io_uring.h
t/io_uring.c

index 7591190a8cb4fee649ab6d67732612040e0aa644..47c53df9df09690b95505d6b2d63f5a068243e5b 100644 (file)
@@ -150,25 +150,31 @@ static int fio_ioring_prep(struct thread_data *td, struct io_u *io_u)
        sqe->buf_index = 0;
 
        if (io_u->ddir == DDIR_READ || io_u->ddir == DDIR_WRITE) {
-               if (io_u->ddir == DDIR_READ)
-                       sqe->opcode = IORING_OP_READV;
-               else
-                       sqe->opcode = IORING_OP_WRITEV;
-
                if (o->fixedbufs) {
-                       sqe->flags |= IOSQE_FIXED_BUFFER;
+                       if (io_u->ddir == DDIR_READ)
+                               sqe->opcode = IORING_OP_READ_FIXED;
+                       else
+                               sqe->opcode = IORING_OP_WRITE_FIXED;
                        sqe->addr = io_u->xfer_buf;
                        sqe->len = io_u->xfer_buflen;
                        sqe->buf_index = io_u->index;
                } else {
+                       if (io_u->ddir == DDIR_READ)
+                               sqe->opcode = IORING_OP_READV;
+                       else
+                               sqe->opcode = IORING_OP_WRITEV;
                        sqe->addr = &ld->iovecs[io_u->index];
                        sqe->len = 1;
                }
                sqe->off = io_u->offset;
-       } else if (ddir_sync(io_u->ddir))
+       } else if (ddir_sync(io_u->ddir)) {
+               sqe->fsync_flags = 0;
+               if (io_u->ddir == DDIR_DATASYNC)
+                       sqe->fsync_flags |= IORING_FSYNC_DATASYNC;
                sqe->opcode = IORING_OP_FSYNC;
+       }
 
-       sqe->data = (unsigned long) io_u;
+       sqe->user_data = (unsigned long) io_u;
        return 0;
 }
 
@@ -182,7 +188,7 @@ static struct io_u *fio_ioring_event(struct thread_data *td, int event)
        index = (event + ld->cq_ring_off) & ld->cq_ring_mask;
 
        cqe = &ld->cq_ring.cqes[index];
-       io_u = (struct io_u *) cqe->data;
+       io_u = (struct io_u *) cqe->user_data;
 
        if (cqe->res != io_u->xfer_buflen) {
                if (cqe->res > io_u->xfer_buflen)
index e1d3df0bef60d63cdb579b6f97cd363438fc1e9f..74370aed5f7125c9508d661e4df251a74e536e22 100644 (file)
@@ -16,7 +16,7 @@
  */
 struct io_uring_sqe {
        __u8    opcode;         /* type of operation for this sqe */
-       __u8    flags;          /* IOSQE_ flags below */
+       __u8    flags;          /* IOSQE_ flags */
        __u16   ioprio;         /* ioprio for the request */
        __s32   fd;             /* file descriptor to do IO on */
        __u64   off;            /* offset into file */
@@ -27,18 +27,18 @@ struct io_uring_sqe {
        __u32   len;            /* buffer size or number of iovecs */
        union {
                __kernel_rwf_t  rw_flags;
-               __u32           __resv;
+               __u32           fsync_flags;
        };
        __u16   buf_index;      /* index into fixed buffers, if used */
-       __u16   __pad2[3];
-       __u64   data;           /* data to be passed back at completion time */
+       __u16   __pad2;
+       __u32   __pad3;
+       __u64   user_data;      /* data to be passed back at completion time */
 };
 
 /*
  * sqe->flags
  */
-#define IOSQE_FIXED_BUFFER     (1 << 0)        /* use fixed buffer */
-#define IOSQE_FIXED_FILE       (1 << 1)        /* use fixed fileset */
+#define IOSQE_FIXED_FILE       (1 << 0)        /* use fixed fileset */
 
 /*
  * io_uring_setup() flags
@@ -50,13 +50,19 @@ struct io_uring_sqe {
 #define IORING_OP_READV                1
 #define IORING_OP_WRITEV       2
 #define IORING_OP_FSYNC                3
-#define IORING_OP_FDSYNC       4
+#define IORING_OP_READ_FIXED   4
+#define IORING_OP_WRITE_FIXED  5
+
+/*
+ * sqe->fsync_flags
+ */
+#define IORING_FSYNC_DATASYNC  (1 << 0)
 
 /*
  * IO completion data structure (Completion Queue Entry)
  */
 struct io_uring_cqe {
-       __u64   data;           /* sqe->data submission passed back */
+       __u64   user_data;      /* sqe->data submission passed back */
        __s32   res;            /* result code for this event */
        __u32   flags;
 };
@@ -87,6 +93,9 @@ struct io_sqring_offsets {
        __u32 resv[3];
 };
 
+/*
+ * sq_ring->flags
+ */
 #define IORING_SQ_NEED_WAKEUP  (1 << 0) /* needs io_uring_enter wakeup */
 
 struct io_cqring_offsets {
@@ -127,12 +136,12 @@ struct io_uring_params {
 
 struct io_uring_register_buffers {
        struct iovec *iovecs;
-       unsigned nr_iovecs;
+       __u32 nr_iovecs;
 };
 
 struct io_uring_register_files {
-       int *fds;
-       unsigned nr_fds;
+       __s32 *fds;
+       __u32 nr_fds;
 };
 
 #endif
index 0461329b64d2267ae240fca1545d605338b73a0e..0c8058ba089ed997274005edb871e00e4324be90 100644 (file)
@@ -59,7 +59,8 @@ static unsigned sq_ring_mask, cq_ring_mask;
 struct file {
        unsigned long max_blocks;
        unsigned pending_ios;
-       int fd;
+       int real_fd;
+       int fixed_fd;
 };
 
 struct submitter {
@@ -77,6 +78,8 @@ struct submitter {
        unsigned long cachehit, cachemiss;
        volatile int finish;
 
+       __s32 *fds;
+
        struct file files[MAX_FDS];
        unsigned nr_files;
        unsigned cur_file;
@@ -105,17 +108,18 @@ static int io_uring_register_buffers(struct submitter *s)
 static int io_uring_register_files(struct submitter *s)
 {
        struct io_uring_register_files reg;
-       int i, ret;
+       int i;
 
-       reg.fds = calloc(s->nr_files, sizeof(int));
-       for (i = 0; i < s->nr_files; i++)
-               reg.fds[i] = s->files[i].fd;
+       s->fds = calloc(s->nr_files, sizeof(__s32));
+       for (i = 0; i < s->nr_files; i++) {
+               s->fds[i] = s->files[i].real_fd;
+               s->files[i].fixed_fd = i;
+       }
+       reg.fds = s->fds;
        reg.nr_fds = s->nr_files;
 
-       ret = syscall(__NR_sys_io_uring_register, s->ring_fd,
+       return syscall(__NR_sys_io_uring_register, s->ring_fd,
                        IORING_REGISTER_FILES, &reg);
-       free(reg.fds);
-       return ret;
 }
 
 static int io_uring_setup(unsigned entries, struct io_uring_params *p)
@@ -163,21 +167,21 @@ static void init_io(struct submitter *s, unsigned index)
        offset = (r % (f->max_blocks - 1)) * BS;
 
        sqe->flags = IOSQE_FIXED_FILE;
-       sqe->opcode = IORING_OP_READV;
        if (fixedbufs) {
+               sqe->opcode = IORING_OP_READ_FIXED;
                sqe->addr = s->iovecs[index].iov_base;
                sqe->len = BS;
                sqe->buf_index = index;
-               sqe->flags |= IOSQE_FIXED_BUFFER;
        } else {
+               sqe->opcode = IORING_OP_READV;
                sqe->addr = &s->iovecs[index];
                sqe->len = 1;
                sqe->buf_index = 0;
        }
        sqe->ioprio = 0;
-       sqe->fd = f->fd;
+       sqe->fd = f->fixed_fd;
        sqe->off = offset;
-       sqe->data = (unsigned long) f;
+       sqe->user_data = (unsigned long) f;
 }
 
 static int prep_more_ios(struct submitter *s, int max_ios)
@@ -212,12 +216,12 @@ static int get_file_size(struct file *f)
 {
        struct stat st;
 
-       if (fstat(f->fd, &st) < 0)
+       if (fstat(f->real_fd, &st) < 0)
                return -1;
        if (S_ISBLK(st.st_mode)) {
                unsigned long long bytes;
 
-               if (ioctl(f->fd, BLKGETSIZE64, &bytes) != 0)
+               if (ioctl(f->real_fd, BLKGETSIZE64, &bytes) != 0)
                        return -1;
 
                f->max_blocks = bytes / BS;
@@ -244,7 +248,7 @@ static int reap_events(struct submitter *s)
                if (head == *ring->tail)
                        break;
                cqe = &ring->cqes[head & cq_ring_mask];
-               f = (struct file *) cqe->data;
+               f = (struct file *) cqe->user_data;
                f->pending_ios--;
                if (cqe->res != BS) {
                        printf("io: unexpected ret=%d\n", cqe->res);
@@ -463,7 +467,7 @@ int main(int argc, char *argv[])
                        perror("open");
                        return 1;
                }
-               f->fd = fd;
+               f->real_fd = fd;
                if (get_file_size(f)) {
                        printf("failed getting size of device/file\n");
                        return 1;