summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--test/io_uring-cp.c251
1 files changed, 137 insertions, 114 deletions
diff --git a/test/io_uring-cp.c b/test/io_uring-cp.c
index 0b75208..e129d84 100644
--- a/test/io_uring-cp.c
+++ b/test/io_uring-cp.c
@@ -6,21 +6,24 @@
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
+#include <assert.h>
+#include <errno.h>
#include <inttypes.h>
#include <sys/types.h>
#include <sys/stat.h>
+#include <sys/ioctl.h>
#include "../src/liburing.h"
#define QD 64
-#define BS 4096
+#define BS (32*1024)
-static struct io_uring in_ring;
-static struct io_uring out_ring;
-static struct iovec iovecs[QD];
+static int infd, outfd;
struct io_data {
- off_t offset;
- struct iovec *iov;
+ int read;
+ off_t first_offset, offset;
+ size_t first_len;
+ struct iovec iov;
};
static int setup_context(unsigned entries, struct io_uring *ring)
@@ -45,155 +48,124 @@ static int get_file_size(int fd, off_t *size)
if (S_ISREG(st.st_mode)) {
*size = st.st_size;
return 0;
+ } else if (S_ISBLK(st.st_mode)) {
+ unsigned long long bytes;
+
+ if (ioctl(fd, BLKGETSIZE64, &bytes) != 0)
+ return -1;
+
+ *size = bytes;
+ return 0;
}
return -1;
}
-static unsigned sqe_index(struct io_uring_sqe *sqe)
-{
- return sqe - in_ring.sq.sqes;
-}
-
-static int queue_read(int fd, off_t size, off_t offset)
+static void queue_prepped(struct io_uring *ring, struct io_data *data)
{
struct io_uring_sqe *sqe;
- struct io_data *data;
- sqe = io_uring_get_sqe(&in_ring);
- if (!sqe)
- return 1;
+ sqe = io_uring_get_sqe(ring);
+ assert(sqe);
- data = malloc(sizeof(*data));
- data->offset = offset;
- data->iov = &iovecs[sqe_index(sqe)];
+ if (data->read)
+ io_uring_prep_readv(sqe, infd, &data->iov, 1, data->offset);
+ else
+ io_uring_prep_writev(sqe, outfd, &data->iov, 1, data->offset);
- io_uring_prep_readv(sqe, fd, data->iov, 1, offset);
io_uring_sqe_set_data(sqe, data);
- iovecs[sqe_index(sqe)].iov_len = size;
- return 0;
}
-static int complete_writes(unsigned *writes)
+static int queue_read(struct io_uring *ring, off_t size, off_t offset)
{
- int ret, nr;
+ struct io_uring_sqe *sqe;
+ struct io_data *data;
- ret = io_uring_submit(&out_ring);
- if (ret < 0) {
- fprintf(stderr, "io_uring_submit: %s\n", strerror(-ret));
+ sqe = io_uring_get_sqe(ring);
+ if (!sqe)
return 1;
- }
- nr = ret;
- while (nr) {
- struct io_uring_cqe *cqe;
+ data = malloc(size + sizeof(*data));
+ data->read = 1;
+ data->offset = data->first_offset = offset;
- ret = io_uring_wait_completion(&out_ring, &cqe);
- if (ret < 0) {
- fprintf(stderr, "io_uring_wait_completion: %s\n",
- strerror(-ret));
- return 1;
- }
- if (cqe->res < 0) {
- fprintf(stderr, "cqe failed: %s\n", strerror(-cqe->res));
- return 1;
- }
- (*writes)--;
- nr--;
- }
+ data->iov.iov_base = data + 1;
+ data->iov.iov_len = size;
+ data->first_len = size;
+ io_uring_prep_readv(sqe, infd, &data->iov, 1, offset);
+ io_uring_sqe_set_data(sqe, data);
return 0;
}
-static void queue_write(int fd, struct io_uring_cqe *cqe)
+static void queue_write(struct io_uring *ring, struct io_data *data)
{
- struct io_data *data = (struct io_data *) (uintptr_t) cqe->user_data;
- struct io_uring_sqe *sqe;
+ data->read = 0;
+ data->offset = data->first_offset;
+
+ data->iov.iov_base = data + 1;
+ data->iov.iov_len = data->first_len;
- sqe = io_uring_get_sqe(&out_ring);
- io_uring_prep_writev(sqe, fd, data->iov, 1, data->offset);
- data->iov->iov_len = cqe->res;
- free(data);
+ queue_prepped(ring, data);
+ io_uring_submit(ring);
}
-int main(int argc, char *argv[])
+static int copy_file(struct io_uring *ring, off_t insize)
{
- off_t read_left, write_left, offset;
+ unsigned long reads, writes;
struct io_uring_cqe *cqe;
- int i, infd, outfd, ret;
- unsigned reads, writes;
-
- if (argc < 3) {
- printf("%s: infile outfile\n", argv[0]);
- return 1;
- }
-
- infd = open(argv[1], O_RDONLY);
- if (infd < 0) {
- perror("open infile");
- return 1;
- }
- outfd = open(argv[2], O_WRONLY | O_CREAT | O_TRUNC, 0644);
- if (outfd < 0) {
- perror("open outfile");
- return 1;
- }
-
- for (i = 0; i < QD; i++) {
- void *buf;
-
- if (posix_memalign(&buf, BS, BS))
- return 1;
- iovecs[i].iov_base = buf;
- iovecs[i].iov_len = BS;
- }
+ off_t write_left, offset;
+ int ret;
- if (setup_context(QD, &in_ring))
- return 1;
- if (setup_context(QD, &out_ring))
- return 1;
- if (get_file_size(infd, &read_left))
- return 1;
+ write_left = insize;
+ writes = reads = offset = 0;
- offset = 0;
- writes = reads = 0;
- write_left = read_left;
- while (read_left || write_left) {
+ while (insize || write_left) {
+ int had_reads, got_comp;
/*
* Queue up as many reads as we can
*/
- while (read_left) {
- off_t this_size = read_left;
+ had_reads = reads;
+ while (insize) {
+ off_t this_size = insize;
+ if (reads + writes >= QD)
+ break;
if (this_size > BS)
this_size = BS;
else if (!this_size)
break;
- if (queue_read(infd, this_size, offset))
+ if (queue_read(ring, this_size, offset))
break;
- read_left -= this_size;
+ insize -= this_size;
offset += this_size;
reads++;
}
- ret = io_uring_submit(&in_ring);
- if (ret < 0) {
- fprintf(stderr, "io_uring_submit: %s\n", strerror(-ret));
- break;
+ if (had_reads != reads) {
+ ret = io_uring_submit(ring);
+ if (ret < 0) {
+ fprintf(stderr, "io_uring_submit: %s\n", strerror(-ret));
+ break;
+ }
}
/*
* read queue full, get at least one completion and queue up
* a write
*/
- while (reads || write_left) {
- if (reads)
- ret = io_uring_wait_completion(&in_ring, &cqe);
- else
- ret = io_uring_get_completion(&in_ring, &cqe);
+ got_comp = 0;
+ while (write_left) {
+ struct io_data *data;
+
+ if (!got_comp) {
+ ret = io_uring_wait_completion(ring, &cqe);
+ got_comp = 1;
+ } else
+ ret = io_uring_get_completion(ring, &cqe);
if (ret < 0) {
fprintf(stderr, "io_uring_get_completion: %s\n",
strerror(-ret));
@@ -201,23 +173,74 @@ int main(int argc, char *argv[])
}
if (!cqe)
break;
- reads--;
+
+ data = (struct io_data *) (uintptr_t) cqe->user_data;
if (cqe->res < 0) {
+ if (cqe->res == -EAGAIN) {
+ queue_prepped(ring, data);
+ continue;
+ }
fprintf(stderr, "cqe failed: %s\n",
strerror(-cqe->res));
return 1;
+ } else if (cqe->res != data->iov.iov_len) {
+ data->iov.iov_base += cqe->res;
+ data->iov.iov_len -= cqe->res;
+ data->offset += cqe->res;
+ queue_prepped(ring, data);
+ continue;
}
- queue_write(outfd, cqe);
- write_left -= cqe->res;
- writes++;
- };
- if (complete_writes(&writes))
- break;
- };
+
+ /*
+ * All done. if write, nothing else to do. if read,
+ * queue up corresponding write.
+ */
+ if (data->read) {
+ queue_write(ring, data);
+ write_left -= data->first_len;
+ reads--;
+ writes++;
+ } else {
+ free(data);
+ writes--;
+ }
+ }
+ }
+
+ return 0;
+}
+
+int main(int argc, char *argv[])
+{
+ struct io_uring ring;
+ off_t insize;
+ int ret;
+
+ if (argc < 3) {
+ printf("%s: infile outfile\n", argv[0]);
+ return 1;
+ }
+
+ infd = open(argv[1], O_RDONLY);
+ if (infd < 0) {
+ perror("open infile");
+ return 1;
+ }
+ outfd = open(argv[2], O_WRONLY | O_CREAT | O_TRUNC, 0644);
+ if (outfd < 0) {
+ perror("open outfile");
+ return 1;
+ }
+
+ if (setup_context(QD, &ring))
+ return 1;
+ if (get_file_size(infd, &insize))
+ return 1;
+
+ ret = copy_file(&ring, insize);
close(infd);
close(outfd);
- io_uring_queue_exit(&in_ring);
- io_uring_queue_exit(&out_ring);
- return 0;
+ io_uring_queue_exit(&ring);
+ return ret;
}