summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--fs/io-wq.h4
-rw-r--r--fs/io_uring.c2
2 files changed, 5 insertions, 1 deletions
diff --git a/fs/io-wq.h b/fs/io-wq.h
index 8cb345256f35..cc50754d028c 100644
--- a/fs/io-wq.h
+++ b/fs/io-wq.h
@@ -62,4 +62,8 @@ static inline void io_wq_worker_running(struct task_struct *tsk)
}
#endif
+static inline bool io_wq_current_is_worker(void)
+{
+ return in_task() && (current->flags & PF_IO_WORKER);
+}
#endif
diff --git a/fs/io_uring.c b/fs/io_uring.c
index ad7f569319c2..3c573f0578a8 100644
--- a/fs/io_uring.c
+++ b/fs/io_uring.c
@@ -872,7 +872,7 @@ static void io_req_link_next(struct io_kiocb *req, struct io_kiocb **nxtptr)
/* we dropped this link, get next */
nxt = list_first_entry_or_null(&req->link_list,
struct io_kiocb, list);
- } else if (nxtptr && current_work()) {
+ } else if (nxtptr && io_wq_current_is_worker()) {
*nxtptr = nxt;
break;
} else {