sched: refactor wait_for_completion_timeout()