Merge git://git.kernel.org/pub/scm/linux/kernel/git/sfrench/cifs-2.6
[linux-2.6] / kernel / itimer.c
1 /*
2  * linux/kernel/itimer.c
3  *
4  * Copyright (C) 1992 Darren Senn
5  */
6
7 /* These are all the functions necessary to implement itimers */
8
9 #include <linux/mm.h>
10 #include <linux/interrupt.h>
11 #include <linux/syscalls.h>
12 #include <linux/time.h>
13 #include <linux/posix-timers.h>
14 #include <linux/hrtimer.h>
15
16 #include <asm/uaccess.h>
17
18 /**
19  * itimer_get_remtime - get remaining time for the timer
20  *
21  * @timer: the timer to read
22  *
23  * Returns the delta between the expiry time and now, which can be
24  * less than zero or 1usec for an pending expired timer
25  */
26 static struct timeval itimer_get_remtime(struct hrtimer *timer)
27 {
28         ktime_t rem = hrtimer_get_remaining(timer);
29
30         /*
31          * Racy but safe: if the itimer expires after the above
32          * hrtimer_get_remtime() call but before this condition
33          * then we return 0 - which is correct.
34          */
35         if (hrtimer_active(timer)) {
36                 if (rem.tv64 <= 0)
37                         rem.tv64 = NSEC_PER_USEC;
38         } else
39                 rem.tv64 = 0;
40
41         return ktime_to_timeval(rem);
42 }
43
44 int do_getitimer(int which, struct itimerval *value)
45 {
46         struct task_struct *tsk = current;
47         cputime_t cinterval, cval;
48
49         switch (which) {
50         case ITIMER_REAL:
51                 spin_lock_irq(&tsk->sighand->siglock);
52                 value->it_value = itimer_get_remtime(&tsk->signal->real_timer);
53                 value->it_interval =
54                         ktime_to_timeval(tsk->signal->it_real_incr);
55                 spin_unlock_irq(&tsk->sighand->siglock);
56                 break;
57         case ITIMER_VIRTUAL:
58                 read_lock(&tasklist_lock);
59                 spin_lock_irq(&tsk->sighand->siglock);
60                 cval = tsk->signal->it_virt_expires;
61                 cinterval = tsk->signal->it_virt_incr;
62                 if (!cputime_eq(cval, cputime_zero)) {
63                         struct task_struct *t = tsk;
64                         cputime_t utime = tsk->signal->utime;
65                         do {
66                                 utime = cputime_add(utime, t->utime);
67                                 t = next_thread(t);
68                         } while (t != tsk);
69                         if (cputime_le(cval, utime)) { /* about to fire */
70                                 cval = jiffies_to_cputime(1);
71                         } else {
72                                 cval = cputime_sub(cval, utime);
73                         }
74                 }
75                 spin_unlock_irq(&tsk->sighand->siglock);
76                 read_unlock(&tasklist_lock);
77                 cputime_to_timeval(cval, &value->it_value);
78                 cputime_to_timeval(cinterval, &value->it_interval);
79                 break;
80         case ITIMER_PROF:
81                 read_lock(&tasklist_lock);
82                 spin_lock_irq(&tsk->sighand->siglock);
83                 cval = tsk->signal->it_prof_expires;
84                 cinterval = tsk->signal->it_prof_incr;
85                 if (!cputime_eq(cval, cputime_zero)) {
86                         struct task_struct *t = tsk;
87                         cputime_t ptime = cputime_add(tsk->signal->utime,
88                                                       tsk->signal->stime);
89                         do {
90                                 ptime = cputime_add(ptime,
91                                                     cputime_add(t->utime,
92                                                                 t->stime));
93                                 t = next_thread(t);
94                         } while (t != tsk);
95                         if (cputime_le(cval, ptime)) { /* about to fire */
96                                 cval = jiffies_to_cputime(1);
97                         } else {
98                                 cval = cputime_sub(cval, ptime);
99                         }
100                 }
101                 spin_unlock_irq(&tsk->sighand->siglock);
102                 read_unlock(&tasklist_lock);
103                 cputime_to_timeval(cval, &value->it_value);
104                 cputime_to_timeval(cinterval, &value->it_interval);
105                 break;
106         default:
107                 return(-EINVAL);
108         }
109         return 0;
110 }
111
112 asmlinkage long sys_getitimer(int which, struct itimerval __user *value)
113 {
114         int error = -EFAULT;
115         struct itimerval get_buffer;
116
117         if (value) {
118                 error = do_getitimer(which, &get_buffer);
119                 if (!error &&
120                     copy_to_user(value, &get_buffer, sizeof(get_buffer)))
121                         error = -EFAULT;
122         }
123         return error;
124 }
125
126
127 /*
128  * The timer is automagically restarted, when interval != 0
129  */
130 enum hrtimer_restart it_real_fn(struct hrtimer *timer)
131 {
132         struct signal_struct *sig =
133             container_of(timer, struct signal_struct, real_timer);
134
135         send_group_sig_info(SIGALRM, SEND_SIG_PRIV, sig->tsk);
136
137         return HRTIMER_NORESTART;
138 }
139
140 /*
141  * Returns true if the timeval is in canonical form
142  */
143 #define timeval_valid(t) \
144         (((t)->tv_sec >= 0) && (((unsigned long) (t)->tv_usec) < USEC_PER_SEC))
145
146 int do_setitimer(int which, struct itimerval *value, struct itimerval *ovalue)
147 {
148         struct task_struct *tsk = current;
149         struct hrtimer *timer;
150         ktime_t expires;
151         cputime_t cval, cinterval, nval, ninterval;
152
153         /*
154          * Validate the timevals in value.
155          */
156         if (!timeval_valid(&value->it_value) ||
157             !timeval_valid(&value->it_interval))
158                 return -EINVAL;
159
160         switch (which) {
161         case ITIMER_REAL:
162 again:
163                 spin_lock_irq(&tsk->sighand->siglock);
164                 timer = &tsk->signal->real_timer;
165                 if (ovalue) {
166                         ovalue->it_value = itimer_get_remtime(timer);
167                         ovalue->it_interval
168                                 = ktime_to_timeval(tsk->signal->it_real_incr);
169                 }
170                 /* We are sharing ->siglock with it_real_fn() */
171                 if (hrtimer_try_to_cancel(timer) < 0) {
172                         spin_unlock_irq(&tsk->sighand->siglock);
173                         goto again;
174                 }
175                 expires = timeval_to_ktime(value->it_value);
176                 if (expires.tv64 != 0) {
177                         tsk->signal->it_real_incr =
178                                 timeval_to_ktime(value->it_interval);
179                         hrtimer_start(timer, expires, HRTIMER_MODE_REL);
180                 } else
181                         tsk->signal->it_real_incr.tv64 = 0;
182
183                 spin_unlock_irq(&tsk->sighand->siglock);
184                 break;
185         case ITIMER_VIRTUAL:
186                 nval = timeval_to_cputime(&value->it_value);
187                 ninterval = timeval_to_cputime(&value->it_interval);
188                 read_lock(&tasklist_lock);
189                 spin_lock_irq(&tsk->sighand->siglock);
190                 cval = tsk->signal->it_virt_expires;
191                 cinterval = tsk->signal->it_virt_incr;
192                 if (!cputime_eq(cval, cputime_zero) ||
193                     !cputime_eq(nval, cputime_zero)) {
194                         if (cputime_gt(nval, cputime_zero))
195                                 nval = cputime_add(nval,
196                                                    jiffies_to_cputime(1));
197                         set_process_cpu_timer(tsk, CPUCLOCK_VIRT,
198                                               &nval, &cval);
199                 }
200                 tsk->signal->it_virt_expires = nval;
201                 tsk->signal->it_virt_incr = ninterval;
202                 spin_unlock_irq(&tsk->sighand->siglock);
203                 read_unlock(&tasklist_lock);
204                 if (ovalue) {
205                         cputime_to_timeval(cval, &ovalue->it_value);
206                         cputime_to_timeval(cinterval, &ovalue->it_interval);
207                 }
208                 break;
209         case ITIMER_PROF:
210                 nval = timeval_to_cputime(&value->it_value);
211                 ninterval = timeval_to_cputime(&value->it_interval);
212                 read_lock(&tasklist_lock);
213                 spin_lock_irq(&tsk->sighand->siglock);
214                 cval = tsk->signal->it_prof_expires;
215                 cinterval = tsk->signal->it_prof_incr;
216                 if (!cputime_eq(cval, cputime_zero) ||
217                     !cputime_eq(nval, cputime_zero)) {
218                         if (cputime_gt(nval, cputime_zero))
219                                 nval = cputime_add(nval,
220                                                    jiffies_to_cputime(1));
221                         set_process_cpu_timer(tsk, CPUCLOCK_PROF,
222                                               &nval, &cval);
223                 }
224                 tsk->signal->it_prof_expires = nval;
225                 tsk->signal->it_prof_incr = ninterval;
226                 spin_unlock_irq(&tsk->sighand->siglock);
227                 read_unlock(&tasklist_lock);
228                 if (ovalue) {
229                         cputime_to_timeval(cval, &ovalue->it_value);
230                         cputime_to_timeval(cinterval, &ovalue->it_interval);
231                 }
232                 break;
233         default:
234                 return -EINVAL;
235         }
236         return 0;
237 }
238
239 /**
240  * alarm_setitimer - set alarm in seconds
241  *
242  * @seconds:    number of seconds until alarm
243  *              0 disables the alarm
244  *
245  * Returns the remaining time in seconds of a pending timer or 0 when
246  * the timer is not active.
247  *
248  * On 32 bit machines the seconds value is limited to (INT_MAX/2) to avoid
249  * negative timeval settings which would cause immediate expiry.
250  */
251 unsigned int alarm_setitimer(unsigned int seconds)
252 {
253         struct itimerval it_new, it_old;
254
255 #if BITS_PER_LONG < 64
256         if (seconds > INT_MAX)
257                 seconds = INT_MAX;
258 #endif
259         it_new.it_value.tv_sec = seconds;
260         it_new.it_value.tv_usec = 0;
261         it_new.it_interval.tv_sec = it_new.it_interval.tv_usec = 0;
262
263         do_setitimer(ITIMER_REAL, &it_new, &it_old);
264
265         /*
266          * We can't return 0 if we have an alarm pending ...  And we'd
267          * better return too much than too little anyway
268          */
269         if ((!it_old.it_value.tv_sec && it_old.it_value.tv_usec) ||
270               it_old.it_value.tv_usec >= 500000)
271                 it_old.it_value.tv_sec++;
272
273         return it_old.it_value.tv_sec;
274 }
275
276 asmlinkage long sys_setitimer(int which,
277                               struct itimerval __user *value,
278                               struct itimerval __user *ovalue)
279 {
280         struct itimerval set_buffer, get_buffer;
281         int error;
282
283         if (value) {
284                 if(copy_from_user(&set_buffer, value, sizeof(set_buffer)))
285                         return -EFAULT;
286         } else
287                 memset((char *) &set_buffer, 0, sizeof(set_buffer));
288
289         error = do_setitimer(which, &set_buffer, ovalue ? &get_buffer : NULL);
290         if (error || !ovalue)
291                 return error;
292
293         if (copy_to_user(ovalue, &get_buffer, sizeof(get_buffer)))
294                 return -EFAULT; 
295         return 0;
296 }