|
Packit |
d3489f |
/* SPDX-License-Identifier: MIT */
|
|
Packit |
d3489f |
/*
|
|
Packit |
d3489f |
* Description: run various file registration tests
|
|
Packit |
d3489f |
*
|
|
Packit |
d3489f |
*/
|
|
Packit |
d3489f |
#include <errno.h>
|
|
Packit |
d3489f |
#include <stdio.h>
|
|
Packit |
d3489f |
#include <unistd.h>
|
|
Packit |
d3489f |
#include <stdlib.h>
|
|
Packit |
d3489f |
#include <string.h>
|
|
Packit |
d3489f |
#include <fcntl.h>
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
#include "liburing.h"
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static void close_files(int *files, int nr_files, int add)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
char fname[32];
|
|
Packit |
d3489f |
int i;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < nr_files; i++) {
|
|
Packit |
d3489f |
if (files)
|
|
Packit |
d3489f |
close(files[i]);
|
|
Packit |
d3489f |
if (!add)
|
|
Packit |
d3489f |
sprintf(fname, ".reg.%d", i);
|
|
Packit |
d3489f |
else
|
|
Packit |
d3489f |
sprintf(fname, ".add.%d", i + add);
|
|
Packit |
d3489f |
unlink(fname);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
if (files)
|
|
Packit |
d3489f |
free(files);
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int *open_files(int nr_files, int extra, int add)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
char fname[32];
|
|
Packit |
d3489f |
int *files;
|
|
Packit |
d3489f |
int i;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
files = calloc(nr_files + extra, sizeof(int));
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
for (i = 0; i < nr_files; i++) {
|
|
Packit |
d3489f |
if (!add)
|
|
Packit |
d3489f |
sprintf(fname, ".reg.%d", i);
|
|
Packit |
d3489f |
else
|
|
Packit |
d3489f |
sprintf(fname, ".add.%d", i + add);
|
|
Packit |
d3489f |
files[i] = open(fname, O_RDWR | O_CREAT, 0644);
|
|
Packit |
d3489f |
if (files[i] < 0) {
|
|
Packit |
d3489f |
perror("open");
|
|
Packit |
d3489f |
free(files);
|
|
Packit |
d3489f |
files = NULL;
|
|
Packit |
d3489f |
break;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
if (extra) {
|
|
Packit |
d3489f |
for (i = nr_files; i < nr_files + extra; i++)
|
|
Packit |
d3489f |
files[i] = -1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return files;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int test_update_multiring(struct io_uring *r1, struct io_uring *r2,
|
|
Packit |
d3489f |
struct io_uring *r3, int do_unreg)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
int *fds, *newfds;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
fds = open_files(10, 0, 0);
|
|
Packit |
d3489f |
newfds = open_files(10, 0, 1);
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (io_uring_register_files(r1, fds, 10) ||
|
|
Packit |
d3489f |
io_uring_register_files(r2, fds, 10) ||
|
|
Packit |
d3489f |
io_uring_register_files(r3, fds, 10)) {
|
|
Packit |
d3489f |
fprintf(stderr, "%s: register files failed\n", __FUNCTION__);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (io_uring_register_files_update(r1, 0, newfds, 10) != 10 ||
|
|
Packit |
d3489f |
io_uring_register_files_update(r2, 0, newfds, 10) != 10 ||
|
|
Packit |
d3489f |
io_uring_register_files_update(r3, 0, newfds, 10) != 10) {
|
|
Packit |
d3489f |
fprintf(stderr, "%s: update files failed\n", __FUNCTION__);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (!do_unreg)
|
|
Packit |
d3489f |
goto done;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (io_uring_unregister_files(r1) ||
|
|
Packit |
d3489f |
io_uring_unregister_files(r2) ||
|
|
Packit |
d3489f |
io_uring_unregister_files(r3)) {
|
|
Packit |
d3489f |
fprintf(stderr, "%s: unregister files failed\n", __FUNCTION__);
|
|
Packit |
d3489f |
goto err;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
done:
|
|
Packit |
d3489f |
close_files(fds, 10, 0);
|
|
Packit |
d3489f |
close_files(newfds, 10, 1);
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
err:
|
|
Packit |
d3489f |
close_files(fds, 10, 0);
|
|
Packit |
d3489f |
close_files(newfds, 10, 1);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
static int test_sqe_update(struct io_uring *ring)
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct io_uring_sqe *sqe;
|
|
Packit |
d3489f |
struct io_uring_cqe *cqe;
|
|
Packit |
d3489f |
int *fds, i, ret;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
fds = malloc(sizeof(int) * 10);
|
|
Packit |
d3489f |
for (i = 0; i < 10; i++)
|
|
Packit |
d3489f |
fds[i] = -1;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
sqe = io_uring_get_sqe(ring);
|
|
Packit |
d3489f |
io_uring_prep_files_update(sqe, fds, 10, 0);
|
|
Packit |
d3489f |
ret = io_uring_submit(ring);
|
|
Packit |
d3489f |
if (ret != 1) {
|
|
Packit |
d3489f |
fprintf(stderr, "submit: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = io_uring_wait_cqe(ring, &cqe);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "wait: %d\n", ret);
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = cqe->res;
|
|
Packit |
d3489f |
io_uring_cqe_seen(ring, cqe);
|
|
Packit |
d3489f |
if (ret == -EINVAL) {
|
|
Packit |
d3489f |
fprintf(stdout, "IORING_OP_FILES_UPDATE not supported, skipping\n");
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
return ret != 10;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
int main(int argc, char *argv[])
|
|
Packit |
d3489f |
{
|
|
Packit |
d3489f |
struct io_uring r1, r2, r3;
|
|
Packit |
d3489f |
int ret;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (argc > 1)
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
if (io_uring_queue_init(8, &r1, 0) ||
|
|
Packit |
d3489f |
io_uring_queue_init(8, &r2, 0) ||
|
|
Packit |
d3489f |
io_uring_queue_init(8, &r3, 0)) {
|
|
Packit |
d3489f |
fprintf(stderr, "ring setup failed\n");
|
|
Packit |
d3489f |
return 1;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = test_update_multiring(&r1, &r2, &r3, 1);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "test_update_multiring w/unreg\n");
|
|
Packit |
d3489f |
return ret;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = test_update_multiring(&r1, &r2, &r3, 0);
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "test_update_multiring wo/unreg\n");
|
|
Packit |
d3489f |
return ret;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
ret = test_sqe_update(&r1;;
|
|
Packit |
d3489f |
if (ret) {
|
|
Packit |
d3489f |
fprintf(stderr, "test_sqe_update failed\n");
|
|
Packit |
d3489f |
return ret;
|
|
Packit |
d3489f |
}
|
|
Packit |
d3489f |
|
|
Packit |
d3489f |
return 0;
|
|
Packit |
d3489f |
}
|