Commit 929659ac authored by Peter Zijlstra's avatar Peter Zijlstra
Browse files

sched/completion: Add wait_for_completion_state() 



Allows waiting with a custom @state.

Signed-off-by: default avatarPeter Zijlstra (Intel) <peterz@infradead.org>
Reviewed-by: default avatarIngo Molnar <mingo@kernel.org>
Link: https://lore.kernel.org/r/20220822114648.922711674@infradead.org
parent f9fc8cad

include/linux/completion.h

+1 −0
Original line number Diff line number Diff line
@@ -103,6 +103,7 @@ extern void wait_for_completion(struct completion *); 
extern void wait_for_completion_io(struct completion *);

extern int wait_for_completion_interruptible(struct completion *x);

extern int wait_for_completion_killable(struct completion *x);

extern int wait_for_completion_state(struct completion *x, unsigned int state);

extern unsigned long wait_for_completion_timeout(struct completion *x,

						   unsigned long timeout);

extern unsigned long wait_for_completion_io_timeout(struct completion *x,

kernel/sched/completion.c

+12 −0
Original line number Diff line number Diff line
@@ -204,6 +204,7 @@ EXPORT_SYMBOL(wait_for_completion_io_timeout); 
int __sched wait_for_completion_interruptible(struct completion *x)

{

	long t = wait_for_common(x, MAX_SCHEDULE_TIMEOUT, TASK_INTERRUPTIBLE);



	if (t == -ERESTARTSYS)

		return t;

	return 0;
@@ -241,12 +242,23 @@ EXPORT_SYMBOL(wait_for_completion_interruptible_timeout); 
int __sched wait_for_completion_killable(struct completion *x)

{

	long t = wait_for_common(x, MAX_SCHEDULE_TIMEOUT, TASK_KILLABLE);



	if (t == -ERESTARTSYS)

		return t;

	return 0;

}

EXPORT_SYMBOL(wait_for_completion_killable);



int __sched wait_for_completion_state(struct completion *x, unsigned int state)

{

	long t = wait_for_common(x, MAX_SCHEDULE_TIMEOUT, state);



	if (t == -ERESTARTSYS)

		return t;

	return 0;

}

EXPORT_SYMBOL(wait_for_completion_state);



/**

 * wait_for_completion_killable_timeout: - waits for completion of a task (w/(to,killable))

 * @x:  holds the state of this particular completion