[linux.git] / fs / io-wq.h

#ifndef INTERNAL_IO_WQ_H
#define INTERNAL_IO_WQ_H

struct io_wq;

enum {
	IO_WQ_WORK_CANCEL	= 1,
	IO_WQ_WORK_HAS_MM	= 2,
	IO_WQ_WORK_HASHED	= 4,
	IO_WQ_WORK_NEEDS_USER	= 8,
	IO_WQ_WORK_NEEDS_FILES	= 16,
	IO_WQ_WORK_UNBOUND	= 32,
	IO_WQ_WORK_INTERNAL	= 64,
	IO_WQ_WORK_CB		= 128,

	IO_WQ_HASH_SHIFT	= 24,	/* upper 8 bits are used for hash key */
};

enum io_wq_cancel {
	IO_WQ_CANCEL_OK,	/* cancelled before started */
	IO_WQ_CANCEL_RUNNING,	/* found, running, and attempted cancelled */
	IO_WQ_CANCEL_NOTFOUND,	/* work not found */
};

struct io_wq_work_node {
	struct io_wq_work_node *next;
};

struct io_wq_work_list {
	struct io_wq_work_node *first;
	struct io_wq_work_node *last;
};

static inline void wq_list_add_tail(struct io_wq_work_node *node,
				    struct io_wq_work_list *list)
{
	if (!list->first) {
		list->last = node;
		WRITE_ONCE(list->first, node);
	} else {
		list->last->next = node;
		list->last = node;
	}
}

static inline void wq_node_del(struct io_wq_work_list *list,
			       struct io_wq_work_node *node,
			       struct io_wq_work_node *prev)
{
	if (node == list->first)
		WRITE_ONCE(list->first, node->next);
	if (node == list->last)
		list->last = prev;
	if (prev)
		prev->next = node->next;
	node->next = NULL;
}

#define wq_list_for_each(pos, prv, head)			\
	for (pos = (head)->first, prv = NULL; pos; prv = pos, pos = (pos)->next)

#define wq_list_empty(list)	(READ_ONCE((list)->first) == NULL)
#define INIT_WQ_LIST(list)	do {				\
	(list)->first = NULL;					\
	(list)->last = NULL;					\
} while (0)

struct io_wq_work {
	union {
		struct io_wq_work_node list;
		void *data;
	};
	void (*func)(struct io_wq_work **);
	struct files_struct *files;
	unsigned flags;
};

#define INIT_IO_WORK(work, _func)			\
	do {						\
		(work)->list.next = NULL;		\
		(work)->func = _func;			\
		(work)->flags = 0;			\
		(work)->files = NULL;			\
	} while (0)					\

typedef void (get_work_fn)(struct io_wq_work *);
typedef void (put_work_fn)(struct io_wq_work *);

struct io_wq_data {
	struct mm_struct *mm;
	struct user_struct *user;
	const struct cred *creds;

	get_work_fn *get_work;
	put_work_fn *put_work;
};

struct io_wq *io_wq_create(unsigned bounded, struct io_wq_data *data);
void io_wq_destroy(struct io_wq *wq);

void io_wq_enqueue(struct io_wq *wq, struct io_wq_work *work);
void io_wq_enqueue_hashed(struct io_wq *wq, struct io_wq_work *work, void *val);
void io_wq_flush(struct io_wq *wq);

void io_wq_cancel_all(struct io_wq *wq);
enum io_wq_cancel io_wq_cancel_work(struct io_wq *wq, struct io_wq_work *cwork);

typedef bool (work_cancel_fn)(struct io_wq_work *, void *);

enum io_wq_cancel io_wq_cancel_cb(struct io_wq *wq, work_cancel_fn *cancel,
					void *data);

#if defined(CONFIG_IO_WQ)
extern void io_wq_worker_sleeping(struct task_struct *);
extern void io_wq_worker_running(struct task_struct *);
#else
static inline void io_wq_worker_sleeping(struct task_struct *tsk)
{
}
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
Commit	Line	Data
771b53d0 JA	1	#ifndef INTERNAL_IO_WQ_H
	2	#define INTERNAL_IO_WQ_H
	3
	4	struct io_wq;
	5
	6	enum {
	7	IO_WQ_WORK_CANCEL = 1,
	8	IO_WQ_WORK_HAS_MM = 2,
	9	IO_WQ_WORK_HASHED = 4,
	10	IO_WQ_WORK_NEEDS_USER = 8,
fcb323cc	11	IO_WQ_WORK_NEEDS_FILES = 16,
c5def4ab	12	IO_WQ_WORK_UNBOUND = 32,
7d723065	13	IO_WQ_WORK_INTERNAL = 64,
b76da70f	14	IO_WQ_WORK_CB = 128,
771b53d0 JA	15
	16	IO_WQ_HASH_SHIFT = 24, /* upper 8 bits are used for hash key */
	17	};
	18
	19	enum io_wq_cancel {
	20	IO_WQ_CANCEL_OK, /* cancelled before started */
	21	IO_WQ_CANCEL_RUNNING, /* found, running, and attempted cancelled */
	22	IO_WQ_CANCEL_NOTFOUND, /* work not found */
	23	};
	24
6206f0e1 JA	25	struct io_wq_work_node {
	26	struct io_wq_work_node *next;
	27	};
	28
	29	struct io_wq_work_list {
	30	struct io_wq_work_node *first;
	31	struct io_wq_work_node *last;
	32	};
	33
	34	static inline void wq_list_add_tail(struct io_wq_work_node *node,
	35	struct io_wq_work_list *list)
	36	{
	37	if (!list->first) {
e995d512 JA	38	list->last = node;
e995d512 JA	39	WRITE_ONCE(list->first, node);
6206f0e1 JA	40	} else {
	41	list->last->next = node;
	42	list->last = node;
	43	}
	44	}
	45
	46	static inline void wq_node_del(struct io_wq_work_list *list,
	47	struct io_wq_work_node *node,
	48	struct io_wq_work_node *prev)
	49	{
	50	if (node == list->first)
e995d512	51	WRITE_ONCE(list->first, node->next);
6206f0e1 JA	52	if (node == list->last)
	53	list->last = prev;
	54	if (prev)
	55	prev->next = node->next;
08bdcc35	56	node->next = NULL;
6206f0e1 JA	57	}
	58
	59	#define wq_list_for_each(pos, prv, head) \
	60	for (pos = (head)->first, prv = NULL; pos; prv = pos, pos = (pos)->next)
	61
e995d512	62	#define wq_list_empty(list) (READ_ONCE((list)->first) == NULL)
6206f0e1 JA	63	#define INIT_WQ_LIST(list) do { \
	64	(list)->first = NULL; \
	65	(list)->last = NULL; \
	66	} while (0)
	67
771b53d0	68	struct io_wq_work {
b76da70f	69	union {
6206f0e1	70	struct io_wq_work_node list;
b76da70f JA	71	void *data;
b76da70f JA	72	};
771b53d0	73	void (func)(struct io_wq_work *);
fcb323cc	74	struct files_struct *files;
6206f0e1	75	unsigned flags;
771b53d0 JA	76	};
	77
	78	#define INIT_IO_WORK(work, _func) \
	79	do { \
6206f0e1	80	(work)->list.next = NULL; \
771b53d0 JA	81	(work)->func = _func; \
771b53d0 JA	82	(work)->flags = 0; \
fcb323cc	83	(work)->files = NULL; \
771b53d0 JA	84	} while (0) \
771b53d0 JA	85
7d723065 JA	86	typedef void (get_work_fn)(struct io_wq_work *);
	87	typedef void (put_work_fn)(struct io_wq_work *);
	88
576a347b JA	89	struct io_wq_data {
	90	struct mm_struct *mm;
	91	struct user_struct *user;
0b8c0ec7	92	const struct cred *creds;
576a347b JA	93
	94	get_work_fn *get_work;
	95	put_work_fn *put_work;
	96	};
	97
	98	struct io_wq io_wq_create(unsigned bounded, struct io_wq_data data);
771b53d0 JA	99	void io_wq_destroy(struct io_wq *wq);
	100
	101	void io_wq_enqueue(struct io_wq wq, struct io_wq_work work);
	102	void io_wq_enqueue_hashed(struct io_wq wq, struct io_wq_work work, void *val);
	103	void io_wq_flush(struct io_wq *wq);
	104
	105	void io_wq_cancel_all(struct io_wq *wq);
	106	enum io_wq_cancel io_wq_cancel_work(struct io_wq wq, struct io_wq_work cwork);
	107
62755e35 JA	108	typedef bool (work_cancel_fn)(struct io_wq_work , void );
	109
	110	enum io_wq_cancel io_wq_cancel_cb(struct io_wq wq, work_cancel_fn cancel,
	111	void *data);
	112
771b53d0 JA	113	#if defined(CONFIG_IO_WQ)
	114	extern void io_wq_worker_sleeping(struct task_struct *);
	115	extern void io_wq_worker_running(struct task_struct *);
	116	#else
	117	static inline void io_wq_worker_sleeping(struct task_struct *tsk)
	118	{
	119	}
	120	static inline void io_wq_worker_running(struct task_struct *tsk)
	121	{
	122	}
525b305d	123	#endif
771b53d0	124
525b305d JA	125	static inline bool io_wq_current_is_worker(void)
	126	{
	127	return in_task() && (current->flags & PF_IO_WORKER);
	128	}
	129	#endif