mirror of
git://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2025-08-05 16:54:27 +00:00
92dda98424
Pass 'ublk_thread *' to more common helpers, then we can avoid to store this reference into 'struct ublk_io'. Prepare for supporting to handle IO via different task context. Signed-off-by: Ming Lei <ming.lei@redhat.com> Link: https://lore.kernel.org/r/20250713143415.2857561-14-ming.lei@redhat.com Signed-off-by: Jens Axboe <axboe@kernel.dk>
106 lines
2.9 KiB
C
106 lines
2.9 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
|
|
/*
|
|
* Fault injection ublk target. Hack this up however you like for
|
|
* testing specific behaviors of ublk_drv. Currently is a null target
|
|
* with a configurable delay before completing each I/O. This delay can
|
|
* be used to test ublk_drv's handling of I/O outstanding to the ublk
|
|
* server when it dies.
|
|
*/
|
|
|
|
#include "kublk.h"
|
|
|
|
static int ublk_fault_inject_tgt_init(const struct dev_ctx *ctx,
|
|
struct ublk_dev *dev)
|
|
{
|
|
const struct ublksrv_ctrl_dev_info *info = &dev->dev_info;
|
|
unsigned long dev_size = 250UL << 30;
|
|
|
|
if (ctx->auto_zc_fallback) {
|
|
ublk_err("%s: not support auto_zc_fallback\n", __func__);
|
|
return -EINVAL;
|
|
}
|
|
|
|
dev->tgt.dev_size = dev_size;
|
|
dev->tgt.params = (struct ublk_params) {
|
|
.types = UBLK_PARAM_TYPE_BASIC,
|
|
.basic = {
|
|
.logical_bs_shift = 9,
|
|
.physical_bs_shift = 12,
|
|
.io_opt_shift = 12,
|
|
.io_min_shift = 9,
|
|
.max_sectors = info->max_io_buf_bytes >> 9,
|
|
.dev_sectors = dev_size >> 9,
|
|
},
|
|
};
|
|
|
|
dev->private_data = (void *)(unsigned long)(ctx->fault_inject.delay_us * 1000);
|
|
return 0;
|
|
}
|
|
|
|
static int ublk_fault_inject_queue_io(struct ublk_thread *t,
|
|
struct ublk_queue *q, int tag)
|
|
{
|
|
const struct ublksrv_io_desc *iod = ublk_get_iod(q, tag);
|
|
struct io_uring_sqe *sqe;
|
|
struct __kernel_timespec ts = {
|
|
.tv_nsec = (long long)q->dev->private_data,
|
|
};
|
|
|
|
ublk_io_alloc_sqes(t, &sqe, 1);
|
|
io_uring_prep_timeout(sqe, &ts, 1, 0);
|
|
sqe->user_data = build_user_data(tag, ublksrv_get_op(iod), 0, q->q_id, 1);
|
|
|
|
ublk_queued_tgt_io(t, q, tag, 1);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void ublk_fault_inject_tgt_io_done(struct ublk_thread *t,
|
|
struct ublk_queue *q,
|
|
const struct io_uring_cqe *cqe)
|
|
{
|
|
unsigned tag = user_data_to_tag(cqe->user_data);
|
|
const struct ublksrv_io_desc *iod = ublk_get_iod(q, tag);
|
|
|
|
if (cqe->res != -ETIME)
|
|
ublk_err("%s: unexpected cqe res %d\n", __func__, cqe->res);
|
|
|
|
if (ublk_completed_tgt_io(t, q, tag))
|
|
ublk_complete_io(t, q, tag, iod->nr_sectors << 9);
|
|
else
|
|
ublk_err("%s: io not complete after 1 cqe\n", __func__);
|
|
}
|
|
|
|
static void ublk_fault_inject_cmd_line(struct dev_ctx *ctx, int argc, char *argv[])
|
|
{
|
|
static const struct option longopts[] = {
|
|
{ "delay_us", 1, NULL, 0 },
|
|
{ 0, 0, 0, 0 }
|
|
};
|
|
int option_idx, opt;
|
|
|
|
ctx->fault_inject.delay_us = 0;
|
|
while ((opt = getopt_long(argc, argv, "",
|
|
longopts, &option_idx)) != -1) {
|
|
switch (opt) {
|
|
case 0:
|
|
if (!strcmp(longopts[option_idx].name, "delay_us"))
|
|
ctx->fault_inject.delay_us = strtoll(optarg, NULL, 10);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void ublk_fault_inject_usage(const struct ublk_tgt_ops *ops)
|
|
{
|
|
printf("\tfault_inject: [--delay_us us (default 0)]\n");
|
|
}
|
|
|
|
const struct ublk_tgt_ops fault_inject_tgt_ops = {
|
|
.name = "fault_inject",
|
|
.init_tgt = ublk_fault_inject_tgt_init,
|
|
.queue_io = ublk_fault_inject_queue_io,
|
|
.tgt_io_done = ublk_fault_inject_tgt_io_done,
|
|
.parse_cmd_line = ublk_fault_inject_cmd_line,
|
|
.usage = ublk_fault_inject_usage,
|
|
};
|