diff options
author | Con Kolivas <kernel@kolivas.org> | 2016-11-04 08:17:23 +1100 |
---|---|---|
committer | Con Kolivas <kernel@kolivas.org> | 2016-11-04 10:24:03 +1100 |
commit | 7dc96b7d399d1c5107b989d18e6adeb4f51781e9 (patch) | |
tree | 497373e3540f5eb06347bbd62959ba94e255d99b | |
parent | da7832bb05a0819f0a056657b1eead8e501e00ff (diff) |
Add functions for min and msec hrtimeout un/interruptible.
-rw-r--r-- | include/linux/sched.h | 6 | ||||
-rw-r--r-- | kernel/time/hrtimer.c | 62 |
2 files changed, 68 insertions, 0 deletions
diff --git a/include/linux/sched.h b/include/linux/sched.h index 7b9e719bc103..62dd30441dd3 100644 --- a/include/linux/sched.h +++ b/include/linux/sched.h @@ -437,6 +437,12 @@ extern signed long schedule_timeout_interruptible(signed long timeout); extern signed long schedule_timeout_killable(signed long timeout); extern signed long schedule_timeout_uninterruptible(signed long timeout); extern signed long schedule_timeout_idle(signed long timeout); + +extern signed long schedule_msec_hrtimeout(signed long timeout); +extern signed long schedule_min_hrtimeout(void); +extern signed long schedule_msec_hrtimeout_interruptible(signed long timeout); +extern signed long schedule_msec_hrtimeout_uninterruptible(signed long timeout); + asmlinkage void schedule(void); extern void schedule_preempt_disabled(void); diff --git a/kernel/time/hrtimer.c b/kernel/time/hrtimer.c index 9ba7c820fc23..53291d7e72f9 100644 --- a/kernel/time/hrtimer.c +++ b/kernel/time/hrtimer.c @@ -1788,3 +1788,65 @@ int __sched schedule_hrtimeout(ktime_t *expires, return schedule_hrtimeout_range(expires, 0, mode); } EXPORT_SYMBOL_GPL(schedule_hrtimeout); + +/* + * As per schedule_hrtimeout but taskes a millisecond value and returns how + * many milliseconds are left. + */ +signed long __sched schedule_msec_hrtimeout(signed long timeout) +{ + struct hrtimer_sleeper t; + ktime_t expires; + int delta, secs; + + if (!timeout) { + __set_current_state(TASK_RUNNING); + return 0; + } + + secs = timeout / 1000; + delta = (timeout % 1000) * NSEC_PER_MSEC; + expires = ktime_set(secs, delta); + + hrtimer_init_on_stack(&t.timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL); + hrtimer_set_expires_range_ns(&t.timer, expires, delta); + + hrtimer_init_sleeper(&t, current); + + hrtimer_start_expires(&t.timer, HRTIMER_MODE_REL); + + if (likely(t.task)) + schedule(); + + hrtimer_cancel(&t.timer); + destroy_hrtimer_on_stack(&t.timer); + + __set_current_state(TASK_RUNNING); + + expires = hrtimer_expires_remaining(&t.timer); + timeout = ktime_to_ms(expires); + return timeout < 0 ? 0 : timeout; +} + +EXPORT_SYMBOL(schedule_msec_hrtimeout); + +signed long __sched schedule_min_hrtimeout(void) +{ + return schedule_msec_hrtimeout(1); +} + +EXPORT_SYMBOL(schedule_min_hrtimeout); + +signed long __sched schedule_msec_hrtimeout_interruptible(signed long timeout) +{ + __set_current_state(TASK_INTERRUPTIBLE); + return schedule_msec_hrtimeout(timeout); +} +EXPORT_SYMBOL(schedule_msec_hrtimeout_interruptible); + +signed long __sched schedule_msec_hrtimeout_uninterruptible(signed long timeout) +{ + __set_current_state(TASK_UNINTERRUPTIBLE); + return schedule_msec_hrtimeout(timeout); +} +EXPORT_SYMBOL(schedule_msec_hrtimeout_uninterruptible); |