Commit 8fa374f9 authored by Jens Axboe's avatar Jens Axboe
Browse files

io_uring/cancel: add generic cancel helper 



Any opcode that is cancelable ends up defining its own cancel helper
for finding and canceling a specific request. Add a generic helper that
can be used for this purpose.

Signed-off-by: default avatarJens Axboe <axboe@kernel.dk>
parent 7d9944f5

io_uring/cancel.c

+21 −0
Original line number Diff line number Diff line
@@ -362,3 +362,24 @@ bool io_cancel_remove_all(struct io_ring_ctx *ctx, struct io_uring_task *tctx, 


	return found;

}



int io_cancel_remove(struct io_ring_ctx *ctx, struct io_cancel_data *cd,

		     unsigned int issue_flags, struct hlist_head *list,

		     bool (*cancel)(struct io_kiocb *))

{

	struct hlist_node *tmp;

	struct io_kiocb *req;

	int nr = 0;



	io_ring_submit_lock(ctx, issue_flags);

	hlist_for_each_entry_safe(req, tmp, list, hash_node) {

		if (!io_cancel_req_match(req, cd))

			continue;

		if (cancel(req))

			nr++;

		if (!(cd->flags & IORING_ASYNC_CANCEL_ALL))

			break;

	}

	io_ring_submit_unlock(ctx, issue_flags);

	return nr ?: -ENOENT;

}

io_uring/cancel.h

+4 −0
Original line number Diff line number Diff line
@@ -28,6 +28,10 @@ bool io_cancel_remove_all(struct io_ring_ctx *ctx, struct io_uring_task *tctx, 
			  struct hlist_head *list, bool cancel_all,

			  bool (*cancel)(struct io_kiocb *));



int io_cancel_remove(struct io_ring_ctx *ctx, struct io_cancel_data *cd,

		     unsigned int issue_flags, struct hlist_head *list,

		     bool (*cancel)(struct io_kiocb *));



static inline bool io_cancel_match_sequence(struct io_kiocb *req, int sequence)

{

	if (req->cancel_seq_set && sequence == req->work.cancel_seq)