1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16#include <linux/sched.h>
17#include <linux/slab.h>
18#include <linux/taskstats.h>
19#include <linux/time.h>
20#include <linux/sysctl.h>
21#include <linux/delayacct.h>
22
23int delayacct_on __read_mostly = 1;
24struct kmem_cache *delayacct_cache;
25
26static int __init delayacct_setup_disable(char *str)
27{
28 delayacct_on = 0;
29 return 1;
30}
31__setup("nodelayacct", delayacct_setup_disable);
32
33void delayacct_init(void)
34{
35 delayacct_cache = KMEM_CACHE(task_delay_info, SLAB_PANIC);
36 delayacct_tsk_init(&init_task);
37}
38
39void __delayacct_tsk_init(struct task_struct *tsk)
40{
41 tsk->delays = kmem_cache_zalloc(delayacct_cache, GFP_KERNEL);
42 if (tsk->delays)
43 spin_lock_init(&tsk->delays->lock);
44}
45
46
47
48
49
50
51static inline void delayacct_start(struct timespec *start)
52{
53 do_posix_clock_monotonic_gettime(start);
54}
55
56
57
58
59
60
61static void delayacct_end(struct timespec *start, struct timespec *end,
62 u64 *total, u32 *count)
63{
64 struct timespec ts;
65 s64 ns;
66 unsigned long flags;
67
68 do_posix_clock_monotonic_gettime(end);
69 ts = timespec_sub(*end, *start);
70 ns = timespec_to_ns(&ts);
71 if (ns < 0)
72 return;
73
74 spin_lock_irqsave(¤t->delays->lock, flags);
75 *total += ns;
76 (*count)++;
77 spin_unlock_irqrestore(¤t->delays->lock, flags);
78}
79
80void __delayacct_blkio_start(void)
81{
82 delayacct_start(¤t->delays->blkio_start);
83}
84
85void __delayacct_blkio_end(void)
86{
87 if (current->delays->flags & DELAYACCT_PF_SWAPIN)
88
89 delayacct_end(¤t->delays->blkio_start,
90 ¤t->delays->blkio_end,
91 ¤t->delays->swapin_delay,
92 ¤t->delays->swapin_count);
93 else
94 delayacct_end(¤t->delays->blkio_start,
95 ¤t->delays->blkio_end,
96 ¤t->delays->blkio_delay,
97 ¤t->delays->blkio_count);
98}
99
100int __delayacct_add_tsk(struct taskstats *d, struct task_struct *tsk)
101{
102 s64 tmp;
103 unsigned long t1;
104 unsigned long long t2, t3;
105 unsigned long flags;
106 struct timespec ts;
107
108
109
110
111 if (!tsk->delays)
112 goto done;
113
114 tmp = (s64)d->cpu_run_real_total;
115 cputime_to_timespec(tsk->utime + tsk->stime, &ts);
116 tmp += timespec_to_ns(&ts);
117 d->cpu_run_real_total = (tmp < (s64)d->cpu_run_real_total) ? 0 : tmp;
118
119 tmp = (s64)d->cpu_scaled_run_real_total;
120 cputime_to_timespec(tsk->utimescaled + tsk->stimescaled, &ts);
121 tmp += timespec_to_ns(&ts);
122 d->cpu_scaled_run_real_total =
123 (tmp < (s64)d->cpu_scaled_run_real_total) ? 0 : tmp;
124
125
126
127
128
129 t1 = tsk->sched_info.pcount;
130 t2 = tsk->sched_info.run_delay;
131 t3 = tsk->se.sum_exec_runtime;
132
133 d->cpu_count += t1;
134
135 tmp = (s64)d->cpu_delay_total + t2;
136 d->cpu_delay_total = (tmp < (s64)d->cpu_delay_total) ? 0 : tmp;
137
138 tmp = (s64)d->cpu_run_virtual_total + t3;
139 d->cpu_run_virtual_total =
140 (tmp < (s64)d->cpu_run_virtual_total) ? 0 : tmp;
141
142
143
144 spin_lock_irqsave(&tsk->delays->lock, flags);
145 tmp = d->blkio_delay_total + tsk->delays->blkio_delay;
146 d->blkio_delay_total = (tmp < d->blkio_delay_total) ? 0 : tmp;
147 tmp = d->swapin_delay_total + tsk->delays->swapin_delay;
148 d->swapin_delay_total = (tmp < d->swapin_delay_total) ? 0 : tmp;
149 tmp = d->freepages_delay_total + tsk->delays->freepages_delay;
150 d->freepages_delay_total = (tmp < d->freepages_delay_total) ? 0 : tmp;
151 d->blkio_count += tsk->delays->blkio_count;
152 d->swapin_count += tsk->delays->swapin_count;
153 d->freepages_count += tsk->delays->freepages_count;
154 spin_unlock_irqrestore(&tsk->delays->lock, flags);
155
156done:
157 return 0;
158}
159
160__u64 __delayacct_blkio_ticks(struct task_struct *tsk)
161{
162 __u64 ret;
163 unsigned long flags;
164
165 spin_lock_irqsave(&tsk->delays->lock, flags);
166 ret = nsec_to_clock_t(tsk->delays->blkio_delay +
167 tsk->delays->swapin_delay);
168 spin_unlock_irqrestore(&tsk->delays->lock, flags);
169 return ret;
170}
171
172void __delayacct_freepages_start(void)
173{
174 delayacct_start(¤t->delays->freepages_start);
175}
176
177void __delayacct_freepages_end(void)
178{
179 delayacct_end(¤t->delays->freepages_start,
180 ¤t->delays->freepages_end,
181 ¤t->delays->freepages_delay,
182 ¤t->delays->freepages_count);
183}
184
185