niansa-misc/linux
1
0
Fork 0
mirror of synced 2025-03-06 20:59:54 +01:00
Code Issues Projects Releases Packages Wiki Activity

io_uring/nop: add support for testing registered files and buffers

Browse source 
Useful for testing performance/efficiency impact of registered files
and buffers, vs (particularly) non-registered files.

Signed-off-by: Jens Axboe <axboe@kernel.dk>
This commit is contained in:
Jens Axboe 2024-10-27 08:59:10 -06:00
parent aa00f67adc
commit a85f31052b
2 changed files with 47 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
include/uapi/linux/io_uring.h
View file

@ -416,6 +416,9 @@ enum io_uring_msg_ring_flags {
* IORING_NOP_INJECT_RESULT Inject result from sqe->result * IORING_NOP_INJECT_RESULT Inject result from sqe->result
*/ */
#define IORING_NOP_INJECT_RESULT (1U << 0) #define IORING_NOP_INJECT_RESULT (1U << 0)
#define IORING_NOP_FILE (1U << 1)
#define IORING_NOP_FIXED_FILE (1U << 2)
#define IORING_NOP_FIXED_BUFFER (1U << 3)
/* /*
* IO completion data structure (Completion Queue Entry) * IO completion data structure (Completion Queue Entry)

49
io_uring/nop.c
View file

@ -8,35 +8,74 @@
#include <uapi/linux/io_uring.h> #include <uapi/linux/io_uring.h>
#include "io_uring.h" #include "io_uring.h"
#include "rsrc.h"
#include "nop.h" #include "nop.h"
struct io_nop { struct io_nop {
/* NOTE: kiocb has the file as the first member, so don't do it here */ /* NOTE: kiocb has the file as the first member, so don't do it here */
struct file *file; struct file *file;
int result; int result;
int fd;
int buffer;
unsigned int flags;
}; };
#define NOP_FLAGS (IORING_NOP_INJECT_RESULT | IORING_NOP_FIXED_FILE | \
IORING_NOP_FIXED_BUFFER | IORING_NOP_FILE)
int io_nop_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) int io_nop_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe)
{ {
unsigned int flags;
struct io_nop *nop = io_kiocb_to_cmd(req, struct io_nop); struct io_nop *nop = io_kiocb_to_cmd(req, struct io_nop);
flags = READ_ONCE(sqe->nop_flags); nop->flags = READ_ONCE(sqe->nop_flags);
if (flags & ~IORING_NOP_INJECT_RESULT) if (nop->flags & ~NOP_FLAGS)
return -EINVAL; return -EINVAL;
if (flags & IORING_NOP_INJECT_RESULT) if (nop->flags & IORING_NOP_INJECT_RESULT)
nop->result = READ_ONCE(sqe->len); nop->result = READ_ONCE(sqe->len);
else else
nop->result = 0; nop->result = 0;
if (nop->flags & IORING_NOP_FIXED_FILE)
nop->fd = READ_ONCE(sqe->fd);
if (nop->flags & IORING_NOP_FIXED_BUFFER)
nop->buffer = READ_ONCE(sqe->buf_index);
return 0; return 0;
} }
int io_nop(struct io_kiocb *req, unsigned int issue_flags) int io_nop(struct io_kiocb *req, unsigned int issue_flags)
{ {
struct io_nop *nop = io_kiocb_to_cmd(req, struct io_nop); struct io_nop *nop = io_kiocb_to_cmd(req, struct io_nop);
int ret = nop->result;
if (nop->result < 0) if (nop->flags & IORING_NOP_FILE) {
if (nop->flags & IORING_NOP_FIXED_FILE) {
req->file = io_file_get_fixed(req, nop->fd, issue_flags);
req->flags |= REQ_F_FIXED_FILE;
} else {
req->file = io_file_get_normal(req, nop->fd);
}
if (!req->file) {
ret = -EBADF;
goto done;
}
}
if (nop->flags & IORING_NOP_FIXED_BUFFER) {
struct io_ring_ctx *ctx = req->ctx;
struct io_mapped_ubuf *imu;
int idx;
ret = -EFAULT;
io_ring_submit_lock(ctx, issue_flags);
if (nop->buffer < ctx->nr_user_bufs) {
idx = array_index_nospec(nop->buffer, ctx->nr_user_bufs);
imu = READ_ONCE(ctx->user_bufs[idx]);
io_req_set_rsrc_node(req, ctx);
ret = 0;
}
io_ring_submit_unlock(ctx, issue_flags);
}
done:
if (ret < 0)
req_set_fail(req); req_set_fail(req);
io_req_set_res(req, nop->result, 0); io_req_set_res(req, nop->result, 0);
return IOU_OK; return IOU_OK;