1 /* 2 * fs/timerfd.c 3 * 4 * Copyright (C) 2007 Davide Libenzi <davidel@xmailserver.org> 5 * 6 * 7 * Thanks to Thomas Gleixner for code reviews and useful comments. 8 * 9 */ 10 11 #include <linux/alarmtimer.h> 12 #include <linux/file.h> 13 #include <linux/poll.h> 14 #include <linux/init.h> 15 #include <linux/fs.h> 16 #include <linux/sched.h> 17 #include <linux/kernel.h> 18 #include <linux/slab.h> 19 #include <linux/list.h> 20 #include <linux/spinlock.h> 21 #include <linux/time.h> 22 #include <linux/hrtimer.h> 23 #include <linux/anon_inodes.h> 24 #include <linux/timerfd.h> 25 #include <linux/syscalls.h> 26 #include <linux/compat.h> 27 #include <linux/rcupdate.h> 28 29 struct timerfd_ctx { 30 union { 31 struct hrtimer tmr; 32 struct alarm alarm; 33 } t; 34 ktime_t tintv; 35 ktime_t moffs; 36 wait_queue_head_t wqh; 37 u64 ticks; 38 int clockid; 39 short unsigned expired; 40 short unsigned settime_flags; /* to show in fdinfo */ 41 struct rcu_head rcu; 42 struct list_head clist; 43 bool might_cancel; 44 }; 45 46 static LIST_HEAD(cancel_list); 47 static DEFINE_SPINLOCK(cancel_lock); 48 49 static inline bool isalarm(struct timerfd_ctx *ctx) 50 { 51 return ctx->clockid == CLOCK_REALTIME_ALARM || 52 ctx->clockid == CLOCK_BOOTTIME_ALARM; 53 } 54 55 /* 56 * This gets called when the timer event triggers. We set the "expired" 57 * flag, but we do not re-arm the timer (in case it's necessary, 58 * tintv.tv64 != 0) until the timer is accessed. 59 */ 60 static void timerfd_triggered(struct timerfd_ctx *ctx) 61 { 62 unsigned long flags; 63 64 spin_lock_irqsave(&ctx->wqh.lock, flags); 65 ctx->expired = 1; 66 ctx->ticks++; 67 wake_up_locked(&ctx->wqh); 68 spin_unlock_irqrestore(&ctx->wqh.lock, flags); 69 } 70 71 static enum hrtimer_restart timerfd_tmrproc(struct hrtimer *htmr) 72 { 73 struct timerfd_ctx *ctx = container_of(htmr, struct timerfd_ctx, 74 t.tmr); 75 timerfd_triggered(ctx); 76 return HRTIMER_NORESTART; 77 } 78 79 static enum alarmtimer_restart timerfd_alarmproc(struct alarm *alarm, 80 ktime_t now) 81 { 82 struct timerfd_ctx *ctx = container_of(alarm, struct timerfd_ctx, 83 t.alarm); 84 timerfd_triggered(ctx); 85 return ALARMTIMER_NORESTART; 86 } 87 88 /* 89 * Called when the clock was set to cancel the timers in the cancel 90 * list. This will wake up processes waiting on these timers. The 91 * wake-up requires ctx->ticks to be non zero, therefore we increment 92 * it before calling wake_up_locked(). 93 */ 94 void timerfd_clock_was_set(void) 95 { 96 ktime_t moffs = ktime_mono_to_real((ktime_t){ .tv64 = 0 }); 97 struct timerfd_ctx *ctx; 98 unsigned long flags; 99 100 rcu_read_lock(); 101 list_for_each_entry_rcu(ctx, &cancel_list, clist) { 102 if (!ctx->might_cancel) 103 continue; 104 spin_lock_irqsave(&ctx->wqh.lock, flags); 105 if (ctx->moffs.tv64 != moffs.tv64) { 106 ctx->moffs.tv64 = KTIME_MAX; 107 ctx->ticks++; 108 wake_up_locked(&ctx->wqh); 109 } 110 spin_unlock_irqrestore(&ctx->wqh.lock, flags); 111 } 112 rcu_read_unlock(); 113 } 114 115 static void timerfd_remove_cancel(struct timerfd_ctx *ctx) 116 { 117 if (ctx->might_cancel) { 118 ctx->might_cancel = false; 119 spin_lock(&cancel_lock); 120 list_del_rcu(&ctx->clist); 121 spin_unlock(&cancel_lock); 122 } 123 } 124 125 static bool timerfd_canceled(struct timerfd_ctx *ctx) 126 { 127 if (!ctx->might_cancel || ctx->moffs.tv64 != KTIME_MAX) 128 return false; 129 ctx->moffs = ktime_mono_to_real((ktime_t){ .tv64 = 0 }); 130 return true; 131 } 132 133 static void timerfd_setup_cancel(struct timerfd_ctx *ctx, int flags) 134 { 135 if ((ctx->clockid == CLOCK_REALTIME || 136 ctx->clockid == CLOCK_REALTIME_ALARM) && 137 (flags & TFD_TIMER_ABSTIME) && (flags & TFD_TIMER_CANCEL_ON_SET)) { 138 if (!ctx->might_cancel) { 139 ctx->might_cancel = true; 140 spin_lock(&cancel_lock); 141 list_add_rcu(&ctx->clist, &cancel_list); 142 spin_unlock(&cancel_lock); 143 } 144 } else if (ctx->might_cancel) { 145 timerfd_remove_cancel(ctx); 146 } 147 } 148 149 static ktime_t timerfd_get_remaining(struct timerfd_ctx *ctx) 150 { 151 ktime_t remaining; 152 153 if (isalarm(ctx)) 154 remaining = alarm_expires_remaining(&ctx->t.alarm); 155 else 156 remaining = hrtimer_expires_remaining(&ctx->t.tmr); 157 158 return remaining.tv64 < 0 ? ktime_set(0, 0): remaining; 159 } 160 161 static int timerfd_setup(struct timerfd_ctx *ctx, int flags, 162 const struct itimerspec *ktmr) 163 { 164 enum hrtimer_mode htmode; 165 ktime_t texp; 166 int clockid = ctx->clockid; 167 168 htmode = (flags & TFD_TIMER_ABSTIME) ? 169 HRTIMER_MODE_ABS: HRTIMER_MODE_REL; 170 171 texp = timespec_to_ktime(ktmr->it_value); 172 ctx->expired = 0; 173 ctx->ticks = 0; 174 ctx->tintv = timespec_to_ktime(ktmr->it_interval); 175 176 if (isalarm(ctx)) { 177 alarm_init(&ctx->t.alarm, 178 ctx->clockid == CLOCK_REALTIME_ALARM ? 179 ALARM_REALTIME : ALARM_BOOTTIME, 180 timerfd_alarmproc); 181 } else { 182 hrtimer_init(&ctx->t.tmr, clockid, htmode); 183 hrtimer_set_expires(&ctx->t.tmr, texp); 184 ctx->t.tmr.function = timerfd_tmrproc; 185 } 186 187 if (texp.tv64 != 0) { 188 if (isalarm(ctx)) { 189 if (flags & TFD_TIMER_ABSTIME) 190 alarm_start(&ctx->t.alarm, texp); 191 else 192 alarm_start_relative(&ctx->t.alarm, texp); 193 } else { 194 hrtimer_start(&ctx->t.tmr, texp, htmode); 195 } 196 197 if (timerfd_canceled(ctx)) 198 return -ECANCELED; 199 } 200 201 ctx->settime_flags = flags & TFD_SETTIME_FLAGS; 202 return 0; 203 } 204 205 static int timerfd_release(struct inode *inode, struct file *file) 206 { 207 struct timerfd_ctx *ctx = file->private_data; 208 209 timerfd_remove_cancel(ctx); 210 211 if (isalarm(ctx)) 212 alarm_cancel(&ctx->t.alarm); 213 else 214 hrtimer_cancel(&ctx->t.tmr); 215 kfree_rcu(ctx, rcu); 216 return 0; 217 } 218 219 static unsigned int timerfd_poll(struct file *file, poll_table *wait) 220 { 221 struct timerfd_ctx *ctx = file->private_data; 222 unsigned int events = 0; 223 unsigned long flags; 224 225 poll_wait(file, &ctx->wqh, wait); 226 227 spin_lock_irqsave(&ctx->wqh.lock, flags); 228 if (ctx->ticks) 229 events |= POLLIN; 230 spin_unlock_irqrestore(&ctx->wqh.lock, flags); 231 232 return events; 233 } 234 235 static ssize_t timerfd_read(struct file *file, char __user *buf, size_t count, 236 loff_t *ppos) 237 { 238 struct timerfd_ctx *ctx = file->private_data; 239 ssize_t res; 240 u64 ticks = 0; 241 242 if (count < sizeof(ticks)) 243 return -EINVAL; 244 spin_lock_irq(&ctx->wqh.lock); 245 if (file->f_flags & O_NONBLOCK) 246 res = -EAGAIN; 247 else 248 res = wait_event_interruptible_locked_irq(ctx->wqh, ctx->ticks); 249 250 /* 251 * If clock has changed, we do not care about the 252 * ticks and we do not rearm the timer. Userspace must 253 * reevaluate anyway. 254 */ 255 if (timerfd_canceled(ctx)) { 256 ctx->ticks = 0; 257 ctx->expired = 0; 258 res = -ECANCELED; 259 } 260 261 if (ctx->ticks) { 262 ticks = ctx->ticks; 263 264 if (ctx->expired && ctx->tintv.tv64) { 265 /* 266 * If tintv.tv64 != 0, this is a periodic timer that 267 * needs to be re-armed. We avoid doing it in the timer 268 * callback to avoid DoS attacks specifying a very 269 * short timer period. 270 */ 271 if (isalarm(ctx)) { 272 ticks += alarm_forward_now( 273 &ctx->t.alarm, ctx->tintv) - 1; 274 alarm_restart(&ctx->t.alarm); 275 } else { 276 ticks += hrtimer_forward_now(&ctx->t.tmr, 277 ctx->tintv) - 1; 278 hrtimer_restart(&ctx->t.tmr); 279 } 280 } 281 ctx->expired = 0; 282 ctx->ticks = 0; 283 } 284 spin_unlock_irq(&ctx->wqh.lock); 285 if (ticks) 286 res = put_user(ticks, (u64 __user *) buf) ? -EFAULT: sizeof(ticks); 287 return res; 288 } 289 290 #ifdef CONFIG_PROC_FS 291 static int timerfd_show(struct seq_file *m, struct file *file) 292 { 293 struct timerfd_ctx *ctx = file->private_data; 294 struct itimerspec t; 295 296 spin_lock_irq(&ctx->wqh.lock); 297 t.it_value = ktime_to_timespec(timerfd_get_remaining(ctx)); 298 t.it_interval = ktime_to_timespec(ctx->tintv); 299 spin_unlock_irq(&ctx->wqh.lock); 300 301 return seq_printf(m, 302 "clockid: %d\n" 303 "ticks: %llu\n" 304 "settime flags: 0%o\n" 305 "it_value: (%llu, %llu)\n" 306 "it_interval: (%llu, %llu)\n", 307 ctx->clockid, (unsigned long long)ctx->ticks, 308 ctx->settime_flags, 309 (unsigned long long)t.it_value.tv_sec, 310 (unsigned long long)t.it_value.tv_nsec, 311 (unsigned long long)t.it_interval.tv_sec, 312 (unsigned long long)t.it_interval.tv_nsec); 313 } 314 #else 315 #define timerfd_show NULL 316 #endif 317 318 #ifdef CONFIG_CHECKPOINT_RESTORE 319 static long timerfd_ioctl(struct file *file, unsigned int cmd, unsigned long arg) 320 { 321 struct timerfd_ctx *ctx = file->private_data; 322 int ret = 0; 323 324 switch (cmd) { 325 case TFD_IOC_SET_TICKS: { 326 u64 ticks; 327 328 if (copy_from_user(&ticks, (u64 __user *)arg, sizeof(ticks))) 329 return -EFAULT; 330 if (!ticks) 331 return -EINVAL; 332 333 spin_lock_irq(&ctx->wqh.lock); 334 if (!timerfd_canceled(ctx)) { 335 ctx->ticks = ticks; 336 wake_up_locked(&ctx->wqh); 337 } else 338 ret = -ECANCELED; 339 spin_unlock_irq(&ctx->wqh.lock); 340 break; 341 } 342 default: 343 ret = -ENOTTY; 344 break; 345 } 346 347 return ret; 348 } 349 #else 350 #define timerfd_ioctl NULL 351 #endif 352 353 static const struct file_operations timerfd_fops = { 354 .release = timerfd_release, 355 .poll = timerfd_poll, 356 .read = timerfd_read, 357 .llseek = noop_llseek, 358 .show_fdinfo = timerfd_show, 359 .unlocked_ioctl = timerfd_ioctl, 360 }; 361 362 static int timerfd_fget(int fd, struct fd *p) 363 { 364 struct fd f = fdget(fd); 365 if (!f.file) 366 return -EBADF; 367 if (f.file->f_op != &timerfd_fops) { 368 fdput(f); 369 return -EINVAL; 370 } 371 *p = f; 372 return 0; 373 } 374 375 SYSCALL_DEFINE2(timerfd_create, int, clockid, int, flags) 376 { 377 int ufd; 378 struct timerfd_ctx *ctx; 379 380 /* Check the TFD_* constants for consistency. */ 381 BUILD_BUG_ON(TFD_CLOEXEC != O_CLOEXEC); 382 BUILD_BUG_ON(TFD_NONBLOCK != O_NONBLOCK); 383 384 if ((flags & ~TFD_CREATE_FLAGS) || 385 (clockid != CLOCK_MONOTONIC && 386 clockid != CLOCK_REALTIME && 387 clockid != CLOCK_REALTIME_ALARM && 388 clockid != CLOCK_BOOTTIME && 389 clockid != CLOCK_BOOTTIME_ALARM)) 390 return -EINVAL; 391 392 ctx = kzalloc(sizeof(*ctx), GFP_KERNEL); 393 if (!ctx) 394 return -ENOMEM; 395 396 init_waitqueue_head(&ctx->wqh); 397 ctx->clockid = clockid; 398 399 if (isalarm(ctx)) 400 alarm_init(&ctx->t.alarm, 401 ctx->clockid == CLOCK_REALTIME_ALARM ? 402 ALARM_REALTIME : ALARM_BOOTTIME, 403 timerfd_alarmproc); 404 else 405 hrtimer_init(&ctx->t.tmr, clockid, HRTIMER_MODE_ABS); 406 407 ctx->moffs = ktime_mono_to_real((ktime_t){ .tv64 = 0 }); 408 409 ufd = anon_inode_getfd("[timerfd]", &timerfd_fops, ctx, 410 O_RDWR | (flags & TFD_SHARED_FCNTL_FLAGS)); 411 if (ufd < 0) 412 kfree(ctx); 413 414 return ufd; 415 } 416 417 static int do_timerfd_settime(int ufd, int flags, 418 const struct itimerspec *new, 419 struct itimerspec *old) 420 { 421 struct fd f; 422 struct timerfd_ctx *ctx; 423 int ret; 424 425 if ((flags & ~TFD_SETTIME_FLAGS) || 426 !timespec_valid(&new->it_value) || 427 !timespec_valid(&new->it_interval)) 428 return -EINVAL; 429 430 ret = timerfd_fget(ufd, &f); 431 if (ret) 432 return ret; 433 ctx = f.file->private_data; 434 435 timerfd_setup_cancel(ctx, flags); 436 437 /* 438 * We need to stop the existing timer before reprogramming 439 * it to the new values. 440 */ 441 for (;;) { 442 spin_lock_irq(&ctx->wqh.lock); 443 444 if (isalarm(ctx)) { 445 if (alarm_try_to_cancel(&ctx->t.alarm) >= 0) 446 break; 447 } else { 448 if (hrtimer_try_to_cancel(&ctx->t.tmr) >= 0) 449 break; 450 } 451 spin_unlock_irq(&ctx->wqh.lock); 452 cpu_relax(); 453 } 454 455 /* 456 * If the timer is expired and it's periodic, we need to advance it 457 * because the caller may want to know the previous expiration time. 458 * We do not update "ticks" and "expired" since the timer will be 459 * re-programmed again in the following timerfd_setup() call. 460 */ 461 if (ctx->expired && ctx->tintv.tv64) { 462 if (isalarm(ctx)) 463 alarm_forward_now(&ctx->t.alarm, ctx->tintv); 464 else 465 hrtimer_forward_now(&ctx->t.tmr, ctx->tintv); 466 } 467 468 old->it_value = ktime_to_timespec(timerfd_get_remaining(ctx)); 469 old->it_interval = ktime_to_timespec(ctx->tintv); 470 471 /* 472 * Re-program the timer to the new value ... 473 */ 474 ret = timerfd_setup(ctx, flags, new); 475 476 spin_unlock_irq(&ctx->wqh.lock); 477 fdput(f); 478 return ret; 479 } 480 481 static int do_timerfd_gettime(int ufd, struct itimerspec *t) 482 { 483 struct fd f; 484 struct timerfd_ctx *ctx; 485 int ret = timerfd_fget(ufd, &f); 486 if (ret) 487 return ret; 488 ctx = f.file->private_data; 489 490 spin_lock_irq(&ctx->wqh.lock); 491 if (ctx->expired && ctx->tintv.tv64) { 492 ctx->expired = 0; 493 494 if (isalarm(ctx)) { 495 ctx->ticks += 496 alarm_forward_now( 497 &ctx->t.alarm, ctx->tintv) - 1; 498 alarm_restart(&ctx->t.alarm); 499 } else { 500 ctx->ticks += 501 hrtimer_forward_now(&ctx->t.tmr, ctx->tintv) 502 - 1; 503 hrtimer_restart(&ctx->t.tmr); 504 } 505 } 506 t->it_value = ktime_to_timespec(timerfd_get_remaining(ctx)); 507 t->it_interval = ktime_to_timespec(ctx->tintv); 508 spin_unlock_irq(&ctx->wqh.lock); 509 fdput(f); 510 return 0; 511 } 512 513 SYSCALL_DEFINE4(timerfd_settime, int, ufd, int, flags, 514 const struct itimerspec __user *, utmr, 515 struct itimerspec __user *, otmr) 516 { 517 struct itimerspec new, old; 518 int ret; 519 520 if (copy_from_user(&new, utmr, sizeof(new))) 521 return -EFAULT; 522 ret = do_timerfd_settime(ufd, flags, &new, &old); 523 if (ret) 524 return ret; 525 if (otmr && copy_to_user(otmr, &old, sizeof(old))) 526 return -EFAULT; 527 528 return ret; 529 } 530 531 SYSCALL_DEFINE2(timerfd_gettime, int, ufd, struct itimerspec __user *, otmr) 532 { 533 struct itimerspec kotmr; 534 int ret = do_timerfd_gettime(ufd, &kotmr); 535 if (ret) 536 return ret; 537 return copy_to_user(otmr, &kotmr, sizeof(kotmr)) ? -EFAULT: 0; 538 } 539 540 #ifdef CONFIG_COMPAT 541 COMPAT_SYSCALL_DEFINE4(timerfd_settime, int, ufd, int, flags, 542 const struct compat_itimerspec __user *, utmr, 543 struct compat_itimerspec __user *, otmr) 544 { 545 struct itimerspec new, old; 546 int ret; 547 548 if (get_compat_itimerspec(&new, utmr)) 549 return -EFAULT; 550 ret = do_timerfd_settime(ufd, flags, &new, &old); 551 if (ret) 552 return ret; 553 if (otmr && put_compat_itimerspec(otmr, &old)) 554 return -EFAULT; 555 return ret; 556 } 557 558 COMPAT_SYSCALL_DEFINE2(timerfd_gettime, int, ufd, 559 struct compat_itimerspec __user *, otmr) 560 { 561 struct itimerspec kotmr; 562 int ret = do_timerfd_gettime(ufd, &kotmr); 563 if (ret) 564 return ret; 565 return put_compat_itimerspec(otmr, &kotmr) ? -EFAULT: 0; 566 } 567 #endif 568