blob: 2ecec3a4f1eeccf44bf9de57b0f8ad242c438c88 [file] [log] [blame]
Frederic Weisbecker73fbec62012-06-16 15:57:37 +02001#include <linux/export.h>
2#include <linux/sched.h>
3#include <linux/tsacct_kern.h>
4#include <linux/kernel_stat.h>
5#include <linux/static_key.h>
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +02006#include <linux/context_tracking.h>
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +01007#include <linux/cputime.h>
Frederic Weisbecker73fbec62012-06-16 15:57:37 +02008#include "sched.h"
Stefano Stabellini1fe7c4e2015-11-10 12:36:46 +00009#ifdef CONFIG_PARAVIRT
10#include <asm/paravirt.h>
11#endif
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020012
13
14#ifdef CONFIG_IRQ_TIME_ACCOUNTING
15
16/*
17 * There are no locks covering percpu hardirq/softirq time.
Frederic Weisbeckerbf9fae92012-09-08 15:23:11 +020018 * They are only modified in vtime_account, on corresponding CPU
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020019 * with interrupts disabled. So, writes are safe.
20 * They are read and saved off onto struct rq in update_rq_clock().
21 * This may result in other CPU reading this CPU's irq time and can
Frederic Weisbeckerbf9fae92012-09-08 15:23:11 +020022 * race with irq/vtime_account on this CPU. We would either get old
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020023 * or new value with a side effect of accounting a slice of irq time to wrong
24 * task when irq is in progress while we read rq->clock. That is a worthy
25 * compromise in place of having locks on each irq in account_system_time.
26 */
Frederic Weisbecker19d23dbf2016-09-26 02:29:20 +020027DEFINE_PER_CPU(struct irqtime, cpu_irqtime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020028
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020029static int sched_clock_irqtime;
30
31void enable_sched_clock_irqtime(void)
32{
33 sched_clock_irqtime = 1;
34}
35
36void disable_sched_clock_irqtime(void)
37{
38 sched_clock_irqtime = 0;
39}
40
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020041/*
42 * Called before incrementing preempt_count on {soft,}irq_enter
43 * and before decrementing preempt_count on {soft,}irq_exit.
44 */
Frederic Weisbecker3e1df4f52012-10-06 05:23:22 +020045void irqtime_account_irq(struct task_struct *curr)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020046{
Frederic Weisbecker19d23dbf2016-09-26 02:29:20 +020047 struct irqtime *irqtime = this_cpu_ptr(&cpu_irqtime);
Frederic Weisbeckera499a5a2017-01-31 04:09:32 +010048 u64 *cpustat = kcpustat_this_cpu->cpustat;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020049 s64 delta;
50 int cpu;
51
52 if (!sched_clock_irqtime)
53 return;
54
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020055 cpu = smp_processor_id();
Frederic Weisbecker19d23dbf2016-09-26 02:29:20 +020056 delta = sched_clock_cpu(cpu) - irqtime->irq_start_time;
57 irqtime->irq_start_time += delta;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020058
Frederic Weisbecker19d23dbf2016-09-26 02:29:20 +020059 u64_stats_update_begin(&irqtime->sync);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020060 /*
61 * We do not account for softirq time from ksoftirqd here.
62 * We want to continue accounting softirq time to ksoftirqd thread
63 * in that case, so as not to confuse scheduler with a special task
64 * that do not consume any time, but still wants to run.
65 */
Frederic Weisbeckera499a5a2017-01-31 04:09:32 +010066 if (hardirq_count()) {
67 cpustat[CPUTIME_IRQ] += delta;
68 irqtime->tick_delta += delta;
69 } else if (in_serving_softirq() && curr != this_cpu_ksoftirqd()) {
70 cpustat[CPUTIME_SOFTIRQ] += delta;
71 irqtime->tick_delta += delta;
72 }
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020073
Frederic Weisbecker19d23dbf2016-09-26 02:29:20 +020074 u64_stats_update_end(&irqtime->sync);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020075}
Frederic Weisbecker3e1df4f52012-10-06 05:23:22 +020076EXPORT_SYMBOL_GPL(irqtime_account_irq);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020077
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +010078static u64 irqtime_tick_accounted(u64 maxtime)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020079{
Frederic Weisbeckera499a5a2017-01-31 04:09:32 +010080 struct irqtime *irqtime = this_cpu_ptr(&cpu_irqtime);
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +010081 u64 delta;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020082
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +010083 delta = min(irqtime->tick_delta, maxtime);
84 irqtime->tick_delta -= delta;
Frederic Weisbecker2810f612016-09-26 02:29:18 +020085
Frederic Weisbeckera499a5a2017-01-31 04:09:32 +010086 return delta;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020087}
88
89#else /* CONFIG_IRQ_TIME_ACCOUNTING */
90
91#define sched_clock_irqtime (0)
92
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +010093static u64 irqtime_tick_accounted(u64 dummy)
Rik van Riel57430212016-07-13 16:50:01 +020094{
95 return 0;
96}
97
Frederic Weisbecker73fbec62012-06-16 15:57:37 +020098#endif /* !CONFIG_IRQ_TIME_ACCOUNTING */
99
100static inline void task_group_account_field(struct task_struct *p, int index,
101 u64 tmp)
102{
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200103 /*
104 * Since all updates are sure to touch the root cgroup, we
105 * get ourselves ahead and touch it first. If the root cgroup
106 * is the only cgroup, then nothing else should be necessary.
107 *
108 */
Christoph Lametera4f61cc2013-08-07 15:38:24 +0000109 __this_cpu_add(kernel_cpustat.cpustat[index], tmp);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200110
Li Zefan1966aaf2013-03-29 14:37:06 +0800111 cpuacct_account_field(p, index, tmp);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200112}
113
114/*
115 * Account user cpu time to a process.
116 * @p: the process that the cpu time gets accounted to
117 * @cputime: the cpu time spent in user space since the last update
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200118 */
Frederic Weisbecker23244a52017-01-31 04:09:37 +0100119void account_user_time(struct task_struct *p, u64 cputime)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200120{
121 int index;
122
123 /* Add user time to process. */
Frederic Weisbecker23244a52017-01-31 04:09:37 +0100124 p->utime += cputime;
125 account_group_user_time(p, cputime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200126
Dongsheng Yangd0ea0262014-01-27 22:00:45 -0500127 index = (task_nice(p) > 0) ? CPUTIME_NICE : CPUTIME_USER;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200128
129 /* Add user time to cpustat. */
Frederic Weisbecker23244a52017-01-31 04:09:37 +0100130 task_group_account_field(p, index, cputime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200131
132 /* Account for user time used */
Frederic Weisbecker6fac4822012-11-13 14:20:55 +0100133 acct_account_cputime(p);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200134}
135
136/*
137 * Account guest cpu time to a process.
138 * @p: the process that the cpu time gets accounted to
139 * @cputime: the cpu time spent in virtual machine since the last update
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200140 */
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100141void account_guest_time(struct task_struct *p, u64 cputime)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200142{
143 u64 *cpustat = kcpustat_this_cpu->cpustat;
144
145 /* Add guest time to process. */
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100146 p->utime += cputime;
147 account_group_user_time(p, cputime);
148 p->gtime += cputime;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200149
150 /* Add guest time to cpustat. */
Dongsheng Yangd0ea0262014-01-27 22:00:45 -0500151 if (task_nice(p) > 0) {
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100152 cpustat[CPUTIME_NICE] += cputime;
153 cpustat[CPUTIME_GUEST_NICE] += cputime;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200154 } else {
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100155 cpustat[CPUTIME_USER] += cputime;
156 cpustat[CPUTIME_GUEST] += cputime;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200157 }
158}
159
160/*
161 * Account system cpu time to a process and desired cpustat field
162 * @p: the process that the cpu time gets accounted to
163 * @cputime: the cpu time spent in kernel space since the last update
Stanislaw Gruszka40565b52016-11-15 03:06:51 +0100164 * @index: pointer to cpustat field that has to be updated
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200165 */
Frederic Weisbeckerc31cc6a2017-01-05 18:11:43 +0100166void account_system_index_time(struct task_struct *p,
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100167 u64 cputime, enum cpu_usage_stat index)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200168{
169 /* Add system time to process. */
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100170 p->stime += cputime;
171 account_group_system_time(p, cputime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200172
173 /* Add system time to cpustat. */
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100174 task_group_account_field(p, index, cputime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200175
176 /* Account for system time used */
Frederic Weisbecker6fac4822012-11-13 14:20:55 +0100177 acct_account_cputime(p);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200178}
179
180/*
181 * Account system cpu time to a process.
182 * @p: the process that the cpu time gets accounted to
183 * @hardirq_offset: the offset to subtract from hardirq_count()
184 * @cputime: the cpu time spent in kernel space since the last update
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200185 */
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100186void account_system_time(struct task_struct *p, int hardirq_offset, u64 cputime)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200187{
188 int index;
189
190 if ((p->flags & PF_VCPU) && (irq_count() - hardirq_offset == 0)) {
Stanislaw Gruszka40565b52016-11-15 03:06:51 +0100191 account_guest_time(p, cputime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200192 return;
193 }
194
195 if (hardirq_count() - hardirq_offset)
196 index = CPUTIME_IRQ;
197 else if (in_serving_softirq())
198 index = CPUTIME_SOFTIRQ;
199 else
200 index = CPUTIME_SYSTEM;
201
Frederic Weisbeckerc31cc6a2017-01-05 18:11:43 +0100202 account_system_index_time(p, cputime, index);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200203}
204
205/*
206 * Account for involuntary wait time.
207 * @cputime: the cpu time spent in involuntary wait
208 */
Frederic Weisbeckerbe9095e2017-01-31 04:09:38 +0100209void account_steal_time(u64 cputime)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200210{
211 u64 *cpustat = kcpustat_this_cpu->cpustat;
212
Frederic Weisbeckerbe9095e2017-01-31 04:09:38 +0100213 cpustat[CPUTIME_STEAL] += cputime;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200214}
215
216/*
217 * Account for idle time.
218 * @cputime: the cpu time spent in idle wait
219 */
Frederic Weisbecker18b43a92017-01-31 04:09:39 +0100220void account_idle_time(u64 cputime)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200221{
222 u64 *cpustat = kcpustat_this_cpu->cpustat;
223 struct rq *rq = this_rq();
224
225 if (atomic_read(&rq->nr_iowait) > 0)
Frederic Weisbecker18b43a92017-01-31 04:09:39 +0100226 cpustat[CPUTIME_IOWAIT] += cputime;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200227 else
Frederic Weisbecker18b43a92017-01-31 04:09:39 +0100228 cpustat[CPUTIME_IDLE] += cputime;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200229}
230
Wanpeng Li03cbc732016-08-17 10:05:46 +0800231/*
232 * When a guest is interrupted for a longer amount of time, missed clock
233 * ticks are not redelivered later. Due to that, this function may on
234 * occasion account more time than the calling functions think elapsed.
235 */
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100236static __always_inline u64 steal_account_process_time(u64 maxtime)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200237{
238#ifdef CONFIG_PARAVIRT
239 if (static_key_false(&paravirt_steal_enabled)) {
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100240 u64 steal;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200241
242 steal = paravirt_steal_clock(smp_processor_id());
243 steal -= this_rq()->prev_steal_time;
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100244 steal = min(steal, maxtime);
245 account_steal_time(steal);
246 this_rq()->prev_steal_time += steal;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200247
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100248 return steal;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200249 }
250#endif
Wanpeng Li807e5b82016-06-13 18:32:46 +0800251 return 0;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200252}
253
Frederic Weisbeckera634f932012-11-21 15:55:59 +0100254/*
Rik van Riel57430212016-07-13 16:50:01 +0200255 * Account how much elapsed time was spent in steal, irq, or softirq time.
256 */
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100257static inline u64 account_other_time(u64 max)
Rik van Riel57430212016-07-13 16:50:01 +0200258{
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100259 u64 accounted;
Rik van Riel57430212016-07-13 16:50:01 +0200260
Frederic Weisbecker2810f612016-09-26 02:29:18 +0200261 /* Shall be converted to a lockdep-enabled lightweight check */
262 WARN_ON_ONCE(!irqs_disabled());
263
Rik van Riel57430212016-07-13 16:50:01 +0200264 accounted = steal_account_process_time(max);
265
266 if (accounted < max)
Frederic Weisbeckera499a5a2017-01-31 04:09:32 +0100267 accounted += irqtime_tick_accounted(max - accounted);
Rik van Riel57430212016-07-13 16:50:01 +0200268
269 return accounted;
270}
271
Stanislaw Gruszkaa1eb1412016-08-17 11:30:44 +0200272#ifdef CONFIG_64BIT
273static inline u64 read_sum_exec_runtime(struct task_struct *t)
274{
275 return t->se.sum_exec_runtime;
276}
277#else
278static u64 read_sum_exec_runtime(struct task_struct *t)
279{
280 u64 ns;
281 struct rq_flags rf;
282 struct rq *rq;
283
284 rq = task_rq_lock(t, &rf);
285 ns = t->se.sum_exec_runtime;
286 task_rq_unlock(rq, t, &rf);
287
288 return ns;
289}
290#endif
291
Rik van Riel57430212016-07-13 16:50:01 +0200292/*
Frederic Weisbeckera634f932012-11-21 15:55:59 +0100293 * Accumulate raw cputime values of dead tasks (sig->[us]time) and live
294 * tasks (sum on group iteration) belonging to @tsk's group.
295 */
296void thread_group_cputime(struct task_struct *tsk, struct task_cputime *times)
297{
298 struct signal_struct *sig = tsk->signal;
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100299 u64 utime, stime;
Frederic Weisbeckera634f932012-11-21 15:55:59 +0100300 struct task_struct *t;
Rik van Riele78c3492014-08-16 13:40:10 -0400301 unsigned int seq, nextseq;
Rik van Riel9c368b52014-09-12 09:12:15 -0400302 unsigned long flags;
Frederic Weisbeckera634f932012-11-21 15:55:59 +0100303
Stanislaw Gruszkaa1eb1412016-08-17 11:30:44 +0200304 /*
305 * Update current task runtime to account pending time since last
306 * scheduler action or thread_group_cputime() call. This thread group
307 * might have other running tasks on different CPUs, but updating
308 * their runtime can affect syscall performance, so we skip account
309 * those pending times and rely only on values updated on tick or
310 * other scheduler action.
311 */
312 if (same_thread_group(current, tsk))
313 (void) task_sched_runtime(current);
314
Frederic Weisbeckera634f932012-11-21 15:55:59 +0100315 rcu_read_lock();
Rik van Riele78c3492014-08-16 13:40:10 -0400316 /* Attempt a lockless read on the first round. */
317 nextseq = 0;
318 do {
319 seq = nextseq;
Rik van Riel9c368b52014-09-12 09:12:15 -0400320 flags = read_seqbegin_or_lock_irqsave(&sig->stats_lock, &seq);
Rik van Riele78c3492014-08-16 13:40:10 -0400321 times->utime = sig->utime;
322 times->stime = sig->stime;
323 times->sum_exec_runtime = sig->sum_sched_runtime;
324
325 for_each_thread(tsk, t) {
326 task_cputime(t, &utime, &stime);
327 times->utime += utime;
328 times->stime += stime;
Stanislaw Gruszkaa1eb1412016-08-17 11:30:44 +0200329 times->sum_exec_runtime += read_sum_exec_runtime(t);
Rik van Riele78c3492014-08-16 13:40:10 -0400330 }
331 /* If lockless access failed, take the lock. */
332 nextseq = 1;
333 } while (need_seqretry(&sig->stats_lock, seq));
Rik van Riel9c368b52014-09-12 09:12:15 -0400334 done_seqretry_irqrestore(&sig->stats_lock, seq, flags);
Frederic Weisbeckera634f932012-11-21 15:55:59 +0100335 rcu_read_unlock();
336}
337
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200338#ifdef CONFIG_IRQ_TIME_ACCOUNTING
339/*
340 * Account a tick to a process and cpustat
341 * @p: the process that the cpu time gets accounted to
342 * @user_tick: is the tick from userspace
343 * @rq: the pointer to rq
344 *
345 * Tick demultiplexing follows the order
346 * - pending hardirq update
347 * - pending softirq update
348 * - user_time
349 * - idle_time
350 * - system time
351 * - check for guest_time
352 * - else account as system_time
353 *
354 * Check for hardirq is done both for system and user time as there is
355 * no timer going off while we are on hardirq and hence we may never get an
356 * opportunity to update it solely in system time.
357 * p->stime and friends are only updated on system time and not on irq
358 * softirq as those do not count in task exec_runtime any more.
359 */
360static void irqtime_account_process_tick(struct task_struct *p, int user_tick,
Thomas Gleixner2d513862014-05-02 23:26:24 +0200361 struct rq *rq, int ticks)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200362{
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100363 u64 other, cputime = TICK_NSEC * ticks;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200364
Rik van Riel57430212016-07-13 16:50:01 +0200365 /*
366 * When returning from idle, many ticks can get accounted at
367 * once, including some ticks of steal, irq, and softirq time.
368 * Subtract those ticks from the amount of time accounted to
369 * idle, or potentially user or system time. Due to rounding,
370 * other time can exceed ticks occasionally.
371 */
Wanpeng Li03cbc732016-08-17 10:05:46 +0800372 other = account_other_time(ULONG_MAX);
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100373 if (other >= cputime)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200374 return;
Frederic Weisbecker23244a52017-01-31 04:09:37 +0100375
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100376 cputime -= other;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200377
Rik van Riel57430212016-07-13 16:50:01 +0200378 if (this_cpu_ksoftirqd() == p) {
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200379 /*
380 * ksoftirqd time do not get accounted in cpu_softirq_time.
381 * So, we have to handle it separately here.
382 * Also, p->stime needs to be updated for ksoftirqd.
383 */
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100384 account_system_index_time(p, cputime, CPUTIME_SOFTIRQ);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200385 } else if (user_tick) {
Stanislaw Gruszka40565b52016-11-15 03:06:51 +0100386 account_user_time(p, cputime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200387 } else if (p == rq->idle) {
Frederic Weisbecker18b43a92017-01-31 04:09:39 +0100388 account_idle_time(cputime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200389 } else if (p->flags & PF_VCPU) { /* System time or guest time */
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100390 account_guest_time(p, cputime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200391 } else {
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100392 account_system_index_time(p, cputime, CPUTIME_SYSTEM);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200393 }
394}
395
396static void irqtime_account_idle_ticks(int ticks)
397{
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200398 struct rq *rq = this_rq();
399
Thomas Gleixner2d513862014-05-02 23:26:24 +0200400 irqtime_account_process_tick(current, 0, rq, ticks);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200401}
402#else /* CONFIG_IRQ_TIME_ACCOUNTING */
Frederic Weisbecker3f4724e2012-07-16 18:00:34 +0200403static inline void irqtime_account_idle_ticks(int ticks) {}
404static inline void irqtime_account_process_tick(struct task_struct *p, int user_tick,
Thomas Gleixner2d513862014-05-02 23:26:24 +0200405 struct rq *rq, int nr_ticks) {}
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200406#endif /* CONFIG_IRQ_TIME_ACCOUNTING */
407
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200408/*
409 * Use precise platform statistics if available:
410 */
411#ifdef CONFIG_VIRT_CPU_ACCOUNTING
Frederic Weisbeckera7e1a9e2012-09-08 16:14:02 +0200412
Frederic Weisbeckere3942ba2012-11-14 00:24:25 +0100413#ifndef __ARCH_HAS_VTIME_TASK_SWITCH
Frederic Weisbeckerb0493402013-07-12 03:10:15 +0200414void vtime_common_task_switch(struct task_struct *prev)
Frederic Weisbeckere3942ba2012-11-14 00:24:25 +0100415{
416 if (is_idle_task(prev))
417 vtime_account_idle(prev);
418 else
419 vtime_account_system(prev);
420
Frederic Weisbeckerc8d7dab2017-01-05 18:11:50 +0100421 vtime_flush(prev);
Frederic Weisbeckere3942ba2012-11-14 00:24:25 +0100422 arch_vtime_task_switch(prev);
423}
424#endif
Frederic Weisbecker11113332012-10-24 18:05:51 +0200425
Frederic Weisbecker0cfdf9a2016-07-13 16:50:03 +0200426#endif /* CONFIG_VIRT_CPU_ACCOUNTING */
427
428
429#ifdef CONFIG_VIRT_CPU_ACCOUNTING_NATIVE
Frederic Weisbeckera7e1a9e2012-09-08 16:14:02 +0200430/*
431 * Archs that account the whole time spent in the idle task
432 * (outside irq) as idle time can rely on this and just implement
Frederic Weisbeckerfd25b4c2012-11-13 18:21:22 +0100433 * vtime_account_system() and vtime_account_idle(). Archs that
Frederic Weisbeckera7e1a9e2012-09-08 16:14:02 +0200434 * have other meaning of the idle time (s390 only includes the
435 * time spent by the CPU when it's in low power mode) must override
436 * vtime_account().
437 */
438#ifndef __ARCH_HAS_VTIME_ACCOUNT
Frederic Weisbecker0cfdf9a2016-07-13 16:50:03 +0200439void vtime_account_irq_enter(struct task_struct *tsk)
Frederic Weisbeckera7e1a9e2012-09-08 16:14:02 +0200440{
Frederic Weisbecker0cfdf9a2016-07-13 16:50:03 +0200441 if (!in_interrupt() && is_idle_task(tsk))
442 vtime_account_idle(tsk);
443 else
444 vtime_account_system(tsk);
Frederic Weisbeckera7e1a9e2012-09-08 16:14:02 +0200445}
Frederic Weisbecker0cfdf9a2016-07-13 16:50:03 +0200446EXPORT_SYMBOL_GPL(vtime_account_irq_enter);
Frederic Weisbeckera7e1a9e2012-09-08 16:14:02 +0200447#endif /* __ARCH_HAS_VTIME_ACCOUNT */
448
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100449void task_cputime_adjusted(struct task_struct *p, u64 *ut, u64 *st)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200450{
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100451 *ut = p->utime;
452 *st = p->stime;
453}
Andrey Smetanin9eec50b2015-09-16 12:29:50 +0300454EXPORT_SYMBOL_GPL(task_cputime_adjusted);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200455
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100456void thread_group_cputime_adjusted(struct task_struct *p, u64 *ut, u64 *st)
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100457{
458 struct task_cputime cputime;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200459
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100460 thread_group_cputime(p, &cputime);
461
462 *ut = cputime.utime;
463 *st = cputime.stime;
464}
465#else /* !CONFIG_VIRT_CPU_ACCOUNTING_NATIVE */
466/*
467 * Account a single tick of cpu time.
468 * @p: the process that the cpu time gets accounted to
469 * @user_tick: indicates if the tick is a user or a system tick
470 */
471void account_process_tick(struct task_struct *p, int user_tick)
472{
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100473 u64 cputime, steal;
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100474 struct rq *rq = this_rq();
475
Frederic Weisbecker55dbdcf2015-11-19 16:47:32 +0100476 if (vtime_accounting_cpu_enabled())
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100477 return;
478
479 if (sched_clock_irqtime) {
Thomas Gleixner2d513862014-05-02 23:26:24 +0200480 irqtime_account_process_tick(p, user_tick, rq, 1);
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100481 return;
482 }
483
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100484 cputime = TICK_NSEC;
Wanpeng Li03cbc732016-08-17 10:05:46 +0800485 steal = steal_account_process_time(ULONG_MAX);
Rik van Riel57430212016-07-13 16:50:01 +0200486
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100487 if (steal >= cputime)
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100488 return;
489
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100490 cputime -= steal;
Rik van Riel57430212016-07-13 16:50:01 +0200491
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100492 if (user_tick)
Stanislaw Gruszka40565b52016-11-15 03:06:51 +0100493 account_user_time(p, cputime);
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100494 else if ((p != rq->idle) || (irq_count() != HARDIRQ_OFFSET))
Frederic Weisbeckerfb8b0492017-01-31 04:09:40 +0100495 account_system_time(p, HARDIRQ_OFFSET, cputime);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200496 else
Frederic Weisbecker18b43a92017-01-31 04:09:39 +0100497 account_idle_time(cputime);
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100498}
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200499
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100500/*
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100501 * Account multiple ticks of idle time.
502 * @ticks: number of stolen ticks
503 */
504void account_idle_ticks(unsigned long ticks)
505{
Frederic Weisbecker18b43a92017-01-31 04:09:39 +0100506 u64 cputime, steal;
Frederic Weisbecker26f2c752016-08-11 14:58:24 +0200507
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100508 if (sched_clock_irqtime) {
509 irqtime_account_idle_ticks(ticks);
510 return;
511 }
512
Frederic Weisbecker18b43a92017-01-31 04:09:39 +0100513 cputime = ticks * TICK_NSEC;
Frederic Weisbecker2b1f9672017-01-31 04:09:41 +0100514 steal = steal_account_process_time(ULONG_MAX);
Wanpeng Lif9bcf1e2016-08-11 13:36:35 +0800515
516 if (steal >= cputime)
517 return;
518
519 cputime -= steal;
520 account_idle_time(cputime);
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100521}
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200522
Frederic Weisbeckerd9a3c982013-02-20 18:54:55 +0100523/*
Stanislaw Gruszka55eaa7c2013-04-30 17:14:42 +0200524 * Perform (stime * rtime) / total, but avoid multiplication overflow by
525 * loosing precision when the numbers are big.
Frederic Weisbeckerd9a3c982013-02-20 18:54:55 +0100526 */
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100527static u64 scale_stime(u64 stime, u64 rtime, u64 total)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200528{
Stanislaw Gruszka55eaa7c2013-04-30 17:14:42 +0200529 u64 scaled;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200530
Stanislaw Gruszka55eaa7c2013-04-30 17:14:42 +0200531 for (;;) {
532 /* Make sure "rtime" is the bigger of stime/rtime */
Stanislaw Gruszka84f9f3a2013-05-02 15:34:33 +0200533 if (stime > rtime)
534 swap(rtime, stime);
Stanislaw Gruszka55eaa7c2013-04-30 17:14:42 +0200535
536 /* Make sure 'total' fits in 32 bits */
537 if (total >> 32)
538 goto drop_precision;
539
540 /* Does rtime (and thus stime) fit in 32 bits? */
541 if (!(rtime >> 32))
542 break;
543
544 /* Can we just balance rtime/stime rather than dropping bits? */
545 if (stime >> 31)
546 goto drop_precision;
547
548 /* We can grow stime and shrink rtime and try to make them both fit */
549 stime <<= 1;
550 rtime >>= 1;
551 continue;
552
553drop_precision:
554 /* We drop from rtime, it has more bits than stime */
555 rtime >>= 1;
556 total >>= 1;
Frederic Weisbeckerd9a3c982013-02-20 18:54:55 +0100557 }
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200558
Stanislaw Gruszka55eaa7c2013-04-30 17:14:42 +0200559 /*
560 * Make sure gcc understands that this is a 32x32->64 multiply,
561 * followed by a 64/32->64 divide.
562 */
563 scaled = div_u64((u64) (u32) stime * (u64) (u32) rtime, (u32)total);
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100564 return scaled;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200565}
566
Frederic Weisbeckerfa092052012-11-28 17:00:57 +0100567/*
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200568 * Adjust tick based cputime random precision against scheduler runtime
569 * accounting.
Rik van Riel347abad2014-09-30 15:59:47 -0400570 *
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200571 * Tick based cputime accounting depend on random scheduling timeslices of a
572 * task to be interrupted or not by the timer. Depending on these
573 * circumstances, the number of these interrupts may be over or
574 * under-optimistic, matching the real user and system cputime with a variable
575 * precision.
576 *
577 * Fix this by scaling these tick based values against the total runtime
578 * accounted by the CFS scheduler.
579 *
580 * This code provides the following guarantees:
581 *
582 * stime + utime == rtime
583 * stime_i+1 >= stime_i, utime_i+1 >= utime_i
584 *
585 * Assuming that rtime_i+1 >= rtime_i.
Frederic Weisbeckerfa092052012-11-28 17:00:57 +0100586 */
Frederic Weisbeckerd37f761d2012-11-22 00:58:35 +0100587static void cputime_adjust(struct task_cputime *curr,
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200588 struct prev_cputime *prev,
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100589 u64 *ut, u64 *st)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200590{
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100591 u64 rtime, stime, utime;
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200592 unsigned long flags;
Frederic Weisbeckerfa092052012-11-28 17:00:57 +0100593
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200594 /* Serialize concurrent callers such that we can honour our guarantees */
595 raw_spin_lock_irqsave(&prev->lock, flags);
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100596 rtime = curr->sum_exec_runtime;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200597
Stanislaw Gruszka772c8082013-04-30 11:35:05 +0200598 /*
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200599 * This is possible under two circumstances:
600 * - rtime isn't monotonic after all (a bug);
601 * - we got reordered by the lock.
602 *
603 * In both cases this acts as a filter such that the rest of the code
604 * can assume it is monotonic regardless of anything else.
Stanislaw Gruszka772c8082013-04-30 11:35:05 +0200605 */
606 if (prev->stime + prev->utime >= rtime)
607 goto out;
608
Stanislaw Gruszka5a8e01f2013-09-04 15:16:03 +0200609 stime = curr->stime;
610 utime = curr->utime;
611
Peter Zijlstra173be9a2016-08-15 18:38:42 +0200612 /*
613 * If either stime or both stime and utime are 0, assume all runtime is
614 * userspace. Once a task gets some ticks, the monotonicy code at
615 * 'update' will ensure things converge to the observed ratio.
616 */
617 if (stime == 0) {
618 utime = rtime;
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200619 goto update;
Frederic Weisbeckerd9a3c982013-02-20 18:54:55 +0100620 }
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200621
Peter Zijlstra173be9a2016-08-15 18:38:42 +0200622 if (utime == 0) {
623 stime = rtime;
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200624 goto update;
625 }
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200626
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100627 stime = scale_stime(stime, rtime, stime + utime);
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200628
Peter Zijlstra173be9a2016-08-15 18:38:42 +0200629update:
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200630 /*
631 * Make sure stime doesn't go backwards; this preserves monotonicity
632 * for utime because rtime is monotonic.
633 *
634 * utime_i+1 = rtime_i+1 - stime_i
635 * = rtime_i+1 - (rtime_i - utime_i)
636 * = (rtime_i+1 - rtime_i) + utime_i
637 * >= utime_i
638 */
639 if (stime < prev->stime)
640 stime = prev->stime;
641 utime = rtime - stime;
642
643 /*
644 * Make sure utime doesn't go backwards; this still preserves
645 * monotonicity for stime, analogous argument to above.
646 */
647 if (utime < prev->utime) {
648 utime = prev->utime;
649 stime = rtime - utime;
650 }
651
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200652 prev->stime = stime;
653 prev->utime = utime;
Stanislaw Gruszka772c8082013-04-30 11:35:05 +0200654out:
Frederic Weisbeckerd37f761d2012-11-22 00:58:35 +0100655 *ut = prev->utime;
656 *st = prev->stime;
Peter Zijlstra9d7fb042015-06-30 11:30:54 +0200657 raw_spin_unlock_irqrestore(&prev->lock, flags);
Frederic Weisbeckerd37f761d2012-11-22 00:58:35 +0100658}
659
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100660void task_cputime_adjusted(struct task_struct *p, u64 *ut, u64 *st)
Frederic Weisbeckerd37f761d2012-11-22 00:58:35 +0100661{
662 struct task_cputime cputime = {
Frederic Weisbeckerd37f761d2012-11-22 00:58:35 +0100663 .sum_exec_runtime = p->se.sum_exec_runtime,
664 };
665
Frederic Weisbecker6fac4822012-11-13 14:20:55 +0100666 task_cputime(p, &cputime.utime, &cputime.stime);
Frederic Weisbeckerd37f761d2012-11-22 00:58:35 +0100667 cputime_adjust(&cputime, &p->prev_cputime, ut, st);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200668}
Andrey Smetanin9eec50b2015-09-16 12:29:50 +0300669EXPORT_SYMBOL_GPL(task_cputime_adjusted);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200670
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100671void thread_group_cputime_adjusted(struct task_struct *p, u64 *ut, u64 *st)
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200672{
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200673 struct task_cputime cputime;
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200674
675 thread_group_cputime(p, &cputime);
Frederic Weisbeckerd37f761d2012-11-22 00:58:35 +0100676 cputime_adjust(&cputime, &p->signal->prev_cputime, ut, st);
Frederic Weisbecker73fbec62012-06-16 15:57:37 +0200677}
Frederic Weisbecker9fbc42e2013-02-25 17:25:39 +0100678#endif /* !CONFIG_VIRT_CPU_ACCOUNTING_NATIVE */
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200679
680#ifdef CONFIG_VIRT_CPU_ACCOUNTING_GEN
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100681static u64 vtime_delta(struct task_struct *tsk)
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200682{
Rik van Rielff9a9b42016-02-10 20:08:27 -0500683 unsigned long now = READ_ONCE(jiffies);
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200684
Rik van Rielff9a9b42016-02-10 20:08:27 -0500685 if (time_before(now, (unsigned long)tsk->vtime_snap))
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100686 return 0;
687
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100688 return jiffies_to_nsecs(now - tsk->vtime_snap);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100689}
690
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100691static u64 get_vtime_delta(struct task_struct *tsk)
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100692{
Rik van Rielff9a9b42016-02-10 20:08:27 -0500693 unsigned long now = READ_ONCE(jiffies);
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100694 u64 delta, other;
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100695
Wanpeng Li03cbc732016-08-17 10:05:46 +0800696 /*
697 * Unlike tick based timing, vtime based timing never has lost
698 * ticks, and no need for steal time accounting to make up for
699 * lost ticks. Vtime accounts a rounded version of actual
700 * elapsed time. Limit account_other_time to prevent rounding
701 * errors from causing elapsed vtime to go negative.
702 */
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100703 delta = jiffies_to_nsecs(now - tsk->vtime_snap);
Rik van Rielb58c3582016-07-13 16:50:02 +0200704 other = account_other_time(delta);
Frederic Weisbecker7098c1e2015-11-19 16:47:30 +0100705 WARN_ON_ONCE(tsk->vtime_snap_whence == VTIME_INACTIVE);
Rik van Rielff9a9b42016-02-10 20:08:27 -0500706 tsk->vtime_snap = now;
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200707
Rik van Rielb58c3582016-07-13 16:50:02 +0200708 return delta - other;
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200709}
710
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100711static void __vtime_account_system(struct task_struct *tsk)
712{
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100713 account_system_time(tsk, irq_count(), get_vtime_delta(tsk));
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100714}
715
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200716void vtime_account_system(struct task_struct *tsk)
717{
Rik van Rielff9a9b42016-02-10 20:08:27 -0500718 if (!vtime_delta(tsk))
719 return;
720
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100721 write_seqcount_begin(&tsk->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100722 __vtime_account_system(tsk);
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100723 write_seqcount_end(&tsk->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100724}
725
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200726void vtime_account_user(struct task_struct *tsk)
727{
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100728 write_seqcount_begin(&tsk->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100729 tsk->vtime_snap_whence = VTIME_SYS;
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100730 if (vtime_delta(tsk))
731 account_user_time(tsk, get_vtime_delta(tsk));
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100732 write_seqcount_end(&tsk->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100733}
734
735void vtime_user_enter(struct task_struct *tsk)
736{
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100737 write_seqcount_begin(&tsk->vtime_seqcount);
Rik van Rielff9a9b42016-02-10 20:08:27 -0500738 if (vtime_delta(tsk))
739 __vtime_account_system(tsk);
Frederic Weisbeckeraf2350b2013-07-15 16:35:55 +0200740 tsk->vtime_snap_whence = VTIME_USER;
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100741 write_seqcount_end(&tsk->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100742}
743
744void vtime_guest_enter(struct task_struct *tsk)
745{
Frederic Weisbecker5b206d42013-07-12 19:05:14 +0200746 /*
747 * The flags must be updated under the lock with
748 * the vtime_snap flush and update.
749 * That enforces a right ordering and update sequence
750 * synchronization against the reader (task_gtime())
751 * that can thus safely catch up with a tickless delta.
752 */
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100753 write_seqcount_begin(&tsk->vtime_seqcount);
Rik van Rielff9a9b42016-02-10 20:08:27 -0500754 if (vtime_delta(tsk))
755 __vtime_account_system(tsk);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100756 current->flags |= PF_VCPU;
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100757 write_seqcount_end(&tsk->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100758}
Frederic Weisbecker48d6a812013-07-10 02:44:35 +0200759EXPORT_SYMBOL_GPL(vtime_guest_enter);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100760
761void vtime_guest_exit(struct task_struct *tsk)
762{
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100763 write_seqcount_begin(&tsk->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100764 __vtime_account_system(tsk);
765 current->flags &= ~PF_VCPU;
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100766 write_seqcount_end(&tsk->vtime_seqcount);
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200767}
Frederic Weisbecker48d6a812013-07-10 02:44:35 +0200768EXPORT_SYMBOL_GPL(vtime_guest_exit);
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200769
770void vtime_account_idle(struct task_struct *tsk)
771{
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100772 account_idle_time(get_vtime_delta(tsk));
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200773}
Frederic Weisbecker3f4724e2012-07-16 18:00:34 +0200774
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100775void arch_vtime_task_switch(struct task_struct *prev)
776{
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100777 write_seqcount_begin(&prev->vtime_seqcount);
Frederic Weisbecker7098c1e2015-11-19 16:47:30 +0100778 prev->vtime_snap_whence = VTIME_INACTIVE;
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100779 write_seqcount_end(&prev->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100780
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100781 write_seqcount_begin(&current->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100782 current->vtime_snap_whence = VTIME_SYS;
Rik van Rielff9a9b42016-02-10 20:08:27 -0500783 current->vtime_snap = jiffies;
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100784 write_seqcount_end(&current->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100785}
786
Frederic Weisbecker45eacc62013-05-15 22:16:32 +0200787void vtime_init_idle(struct task_struct *t, int cpu)
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100788{
789 unsigned long flags;
790
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100791 local_irq_save(flags);
792 write_seqcount_begin(&t->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100793 t->vtime_snap_whence = VTIME_SYS;
Rik van Rielff9a9b42016-02-10 20:08:27 -0500794 t->vtime_snap = jiffies;
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100795 write_seqcount_end(&t->vtime_seqcount);
796 local_irq_restore(flags);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100797}
798
Frederic Weisbecker16a6d9b2017-01-31 04:09:21 +0100799u64 task_gtime(struct task_struct *t)
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100800{
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100801 unsigned int seq;
Frederic Weisbecker16a6d9b2017-01-31 04:09:21 +0100802 u64 gtime;
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100803
Frederic Weisbeckere5925392015-11-19 16:47:33 +0100804 if (!vtime_accounting_enabled())
Hiroshi Shimamoto25411172015-11-19 16:47:28 +0100805 return t->gtime;
806
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100807 do {
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100808 seq = read_seqcount_begin(&t->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100809
810 gtime = t->gtime;
Frederic Weisbeckercab245d2015-11-19 16:47:31 +0100811 if (t->vtime_snap_whence == VTIME_SYS && t->flags & PF_VCPU)
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100812 gtime += vtime_delta(t);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100813
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100814 } while (read_seqcount_retry(&t->vtime_seqcount, seq));
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100815
816 return gtime;
817}
818
819/*
820 * Fetch cputime raw values from fields of task_struct and
821 * add up the pending nohz execution time since the last
822 * cputime snapshot.
823 */
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100824void task_cputime(struct task_struct *t, u64 *utime, u64 *stime)
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100825{
Frederic Weisbecker5613fda2017-01-31 04:09:23 +0100826 u64 delta;
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100827 unsigned int seq;
Stanislaw Gruszka353c50e2016-11-15 03:06:52 +0100828
829 if (!vtime_accounting_enabled()) {
830 *utime = t->utime;
831 *stime = t->stime;
832 return;
833 }
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100834
835 do {
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100836 seq = read_seqcount_begin(&t->vtime_seqcount);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100837
Stanislaw Gruszka353c50e2016-11-15 03:06:52 +0100838 *utime = t->utime;
839 *stime = t->stime;
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100840
841 /* Task is sleeping, nothing to add */
Stanislaw Gruszka353c50e2016-11-15 03:06:52 +0100842 if (t->vtime_snap_whence == VTIME_INACTIVE || is_idle_task(t))
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100843 continue;
844
Frederic Weisbeckerbfce1d62017-01-31 04:09:42 +0100845 delta = vtime_delta(t);
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100846
847 /*
848 * Task runs either in user or kernel space, add pending nohz time to
849 * the right place.
850 */
Stanislaw Gruszka353c50e2016-11-15 03:06:52 +0100851 if (t->vtime_snap_whence == VTIME_USER || t->flags & PF_VCPU)
852 *utime += delta;
853 else if (t->vtime_snap_whence == VTIME_SYS)
854 *stime += delta;
Frederic Weisbeckerb7ce2272015-11-19 16:47:34 +0100855 } while (read_seqcount_retry(&t->vtime_seqcount, seq));
Frederic Weisbecker6a616712012-12-16 20:00:34 +0100856}
Frederic Weisbeckerabf917c2012-07-25 07:56:04 +0200857#endif /* CONFIG_VIRT_CPU_ACCOUNTING_GEN */