116 lines
2.3 KiB
C
116 lines
2.3 KiB
C
|
/* SPDX-License-Identifier: MIT */
|
||
|
|
/*
|
||
|
|
* Test fixed buffers with non-iterators.
|
||
|
|
*
|
||
|
|
* Taken from: https://github.com/axboe/liburing/issues/549
|
||
|
|
*/
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <fcntl.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
|
||
|
|
#include "liburing.h"
|
||
|
|
#include "helpers.h"
|
||
|
|
|
||
|
|
#define BUF_SIZE 4096
|
||
|
|
#define BUFFERS 1
|
||
|
|
#define IN_FD "/dev/urandom"
|
||
|
|
#define OUT_FD "/dev/zero"
|
||
|
|
|
||
|
|
static int test(struct io_uring *ring)
|
||
|
|
{
|
||
|
|
struct iovec iov[BUFFERS];
|
||
|
|
struct io_uring_sqe *sqe;
|
||
|
|
struct io_uring_cqe *cqe;
|
||
|
|
int ret, fd_in, fd_out, i;
|
||
|
|
|
||
|
|
fd_in = open(IN_FD, O_RDONLY, 0644);
|
||
|
|
if (fd_in < 0) {
|
||
|
|
perror("open in");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
fd_out = open(OUT_FD, O_RDWR, 0644);
|
||
|
|
if (fd_out < 0) {
|
||
|
|
perror("open out");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (i = 0; i < BUFFERS; i++) {
|
||
|
|
iov[i].iov_base = malloc(BUF_SIZE);
|
||
|
|
iov[i].iov_len = BUF_SIZE;
|
||
|
|
memset(iov[i].iov_base, 0, BUF_SIZE);
|
||
|
|
}
|
||
|
|
|
||
|
|
ret = io_uring_register_buffers(ring, iov, BUFFERS);
|
||
|
|
if (ret) {
|
||
|
|
fprintf(stderr, "Error registering buffers: %s", strerror(-ret));
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
sqe = io_uring_get_sqe(ring);
|
||
|
|
if (!sqe) {
|
||
|
|
fprintf(stderr, "Could not get SQE.\n");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
io_uring_prep_read_fixed(sqe, fd_in, iov[0].iov_base, BUF_SIZE, 0, 0);
|
||
|
|
io_uring_submit(ring);
|
||
|
|
|
||
|
|
ret = io_uring_wait_cqe(ring, &cqe);
|
||
|
|
if (ret < 0) {
|
||
|
|
fprintf(stderr, "Error waiting for completion: %s\n", strerror(-ret));
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (cqe->res < 0) {
|
||
|
|
fprintf(stderr, "Error in async operation: %s\n", strerror(-cqe->res));
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
io_uring_cqe_seen(ring, cqe);
|
||
|
|
|
||
|
|
sqe = io_uring_get_sqe(ring);
|
||
|
|
if (!sqe) {
|
||
|
|
fprintf(stderr, "Could not get SQE.\n");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
io_uring_prep_write_fixed(sqe, fd_out, iov[0].iov_base, BUF_SIZE, 0, 0);
|
||
|
|
io_uring_submit(ring);
|
||
|
|
|
||
|
|
ret = io_uring_wait_cqe(ring, &cqe);
|
||
|
|
if (ret < 0) {
|
||
|
|
fprintf(stderr, "Error waiting for completion: %s\n", strerror(-ret));
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
if (cqe->res < 0) {
|
||
|
|
fprintf(stderr, "Error in async operation: %s\n", strerror(-cqe->res));
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
io_uring_cqe_seen(ring, cqe);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(int argc, char *argv[])
|
||
|
|
{
|
||
|
|
struct io_uring ring;
|
||
|
|
int ret;
|
||
|
|
|
||
|
|
if (argc > 1)
|
||
|
|
return 0;
|
||
|
|
|
||
|
|
ret = t_create_ring(8, &ring, 0);
|
||
|
|
if (ret == T_SETUP_SKIP)
|
||
|
|
return 0;
|
||
|
|
else if (ret < 0)
|
||
|
|
return 1;
|
||
|
|
|
||
|
|
ret = test(&ring);
|
||
|
|
if (ret) {
|
||
|
|
fprintf(stderr, "Test failed\n");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
io_uring_queue_exit(&ring);
|
||
|
|
return 0;
|
||
|
|
}
|