touch_softlockup_watchdog_sched();
 }
 
-/*
- * Updates the per-CPU time idle statistics counters
- */
-static void
-update_ts_time_stats(int cpu, struct tick_sched *ts, ktime_t now, u64 *last_update_time)
+static void tick_nohz_stop_idle(struct tick_sched *ts, ktime_t now)
 {
        ktime_t delta;
 
-       if (ts->idle_active) {
-               delta = ktime_sub(now, ts->idle_entrytime);
-               if (nr_iowait_cpu(cpu) > 0)
-                       ts->iowait_sleeptime = ktime_add(ts->iowait_sleeptime, delta);
-               else
-                       ts->idle_sleeptime = ktime_add(ts->idle_sleeptime, delta);
-               ts->idle_entrytime = now;
-       }
+       if (WARN_ON_ONCE(!ts->idle_active))
+               return;
 
-       if (last_update_time)
-               *last_update_time = ktime_to_us(now);
+       delta = ktime_sub(now, ts->idle_entrytime);
 
-}
+       if (nr_iowait_cpu(smp_processor_id()) > 0)
+               ts->iowait_sleeptime = ktime_add(ts->iowait_sleeptime, delta);
+       else
+               ts->idle_sleeptime = ktime_add(ts->idle_sleeptime, delta);
 
-static void tick_nohz_stop_idle(struct tick_sched *ts, ktime_t now)
-{
-       update_ts_time_stats(smp_processor_id(), ts, now, NULL);
+       ts->idle_entrytime = now;
        ts->idle_active = 0;
 
        sched_clock_idle_wakeup_event();
        sched_clock_idle_sleep_event();
 }
 
+static u64 get_cpu_sleep_time_us(struct tick_sched *ts, ktime_t *sleeptime,
+                                bool compute_delta, u64 *last_update_time)
+{
+       ktime_t now, idle;
+
+       if (!tick_nohz_active)
+               return -1;
+
+       now = ktime_get();
+       if (last_update_time)
+               *last_update_time = ktime_to_us(now);
+
+       if (ts->idle_active && compute_delta) {
+               ktime_t delta = ktime_sub(now, ts->idle_entrytime);
+
+               idle = ktime_add(*sleeptime, delta);
+       } else {
+               idle = *sleeptime;
+       }
+
+       return ktime_to_us(idle);
+
+}
+
 /**
  * get_cpu_idle_time_us - get the total idle time of a CPU
  * @cpu: CPU number to query
 u64 get_cpu_idle_time_us(int cpu, u64 *last_update_time)
 {
        struct tick_sched *ts = &per_cpu(tick_cpu_sched, cpu);
-       ktime_t now, idle;
-
-       if (!tick_nohz_active)
-               return -1;
-
-       now = ktime_get();
-       if (last_update_time) {
-               update_ts_time_stats(cpu, ts, now, last_update_time);
-               idle = ts->idle_sleeptime;
-       } else {
-               if (ts->idle_active && !nr_iowait_cpu(cpu)) {
-                       ktime_t delta = ktime_sub(now, ts->idle_entrytime);
-
-                       idle = ktime_add(ts->idle_sleeptime, delta);
-               } else {
-                       idle = ts->idle_sleeptime;
-               }
-       }
-
-       return ktime_to_us(idle);
 
+       return get_cpu_sleep_time_us(ts, &ts->idle_sleeptime,
+                                    !nr_iowait_cpu(cpu), last_update_time);
 }
 EXPORT_SYMBOL_GPL(get_cpu_idle_time_us);
 
 u64 get_cpu_iowait_time_us(int cpu, u64 *last_update_time)
 {
        struct tick_sched *ts = &per_cpu(tick_cpu_sched, cpu);
-       ktime_t now, iowait;
-
-       if (!tick_nohz_active)
-               return -1;
-
-       now = ktime_get();
-       if (last_update_time) {
-               update_ts_time_stats(cpu, ts, now, last_update_time);
-               iowait = ts->iowait_sleeptime;
-       } else {
-               if (ts->idle_active && nr_iowait_cpu(cpu) > 0) {
-                       ktime_t delta = ktime_sub(now, ts->idle_entrytime);
-
-                       iowait = ktime_add(ts->iowait_sleeptime, delta);
-               } else {
-                       iowait = ts->iowait_sleeptime;
-               }
-       }
 
-       return ktime_to_us(iowait);
+       return get_cpu_sleep_time_us(ts, &ts->iowait_sleeptime,
+                                    nr_iowait_cpu(cpu), last_update_time);
 }
 EXPORT_SYMBOL_GPL(get_cpu_iowait_time_us);