alarmtimer.c 22.0 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
/*
 * Alarmtimer interface
 *
 * This interface provides a timer which is similarto hrtimers,
 * but triggers a RTC alarm if the box is suspend.
 *
 * This interface is influenced by the Android RTC Alarm timer
 * interface.
 *
 * Copyright (C) 2010 IBM Corperation
 *
 * Author: John Stultz <john.stultz@linaro.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */
#include <linux/time.h>
#include <linux/hrtimer.h>
#include <linux/timerqueue.h>
#include <linux/rtc.h>
22
#include <linux/sched/signal.h>
23
#include <linux/sched/debug.h>
24 25 26 27 28 29
#include <linux/alarmtimer.h>
#include <linux/mutex.h>
#include <linux/platform_device.h>
#include <linux/posix-timers.h>
#include <linux/workqueue.h>
#include <linux/freezer.h>
30
#include <linux/compat.h>
31

32 33
#include "posix-timers.h"

34 35 36
#define CREATE_TRACE_POINTS
#include <trace/events/alarmtimer.h>

37 38 39 40 41 42 43
/**
 * struct alarm_base - Alarm timer bases
 * @lock:		Lock for syncrhonized access to the base
 * @timerqueue:		Timerqueue head managing the list of events
 * @gettime:		Function to read the time correlating to the base
 * @base_clockid:	clockid for the base
 */
44 45 46 47 48 49 50
static struct alarm_base {
	spinlock_t		lock;
	struct timerqueue_head	timerqueue;
	ktime_t			(*gettime)(void);
	clockid_t		base_clockid;
} alarm_bases[ALARM_NUMTYPE];

51
#if defined(CONFIG_POSIX_TIMERS) || defined(CONFIG_RTC_CLASS)
52 53 54
/* freezer information to handle clock_nanosleep triggered wakeups */
static enum alarmtimer_type freezer_alarmtype;
static ktime_t freezer_expires;
55 56
static ktime_t freezer_delta;
static DEFINE_SPINLOCK(freezer_delta_lock);
57
#endif
58

59 60
static struct wakeup_source *ws;

61
#ifdef CONFIG_RTC_CLASS
62
/* rtc timer and device for setting alarm wakeups at suspend */
63
static struct rtc_timer		rtctimer;
64
static struct rtc_device	*rtcdev;
65
static DEFINE_SPINLOCK(rtcdev_lock);
66

67 68 69 70 71 72 73
/**
 * alarmtimer_get_rtcdev - Return selected rtcdevice
 *
 * This function returns the rtc device to use for wakealarms.
 * If one has not already been chosen, it checks to see if a
 * functional rtc device is available.
 */
74
struct rtc_device *alarmtimer_get_rtcdev(void)
75 76 77 78 79 80 81 82 83 84
{
	unsigned long flags;
	struct rtc_device *ret;

	spin_lock_irqsave(&rtcdev_lock, flags);
	ret = rtcdev;
	spin_unlock_irqrestore(&rtcdev_lock, flags);

	return ret;
}
85
EXPORT_SYMBOL_GPL(alarmtimer_get_rtcdev);
86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110

static int alarmtimer_rtc_add_device(struct device *dev,
				struct class_interface *class_intf)
{
	unsigned long flags;
	struct rtc_device *rtc = to_rtc_device(dev);

	if (rtcdev)
		return -EBUSY;

	if (!rtc->ops->set_alarm)
		return -1;
	if (!device_may_wakeup(rtc->dev.parent))
		return -1;

	spin_lock_irqsave(&rtcdev_lock, flags);
	if (!rtcdev) {
		rtcdev = rtc;
		/* hold a reference so it doesn't go away */
		get_device(dev);
	}
	spin_unlock_irqrestore(&rtcdev_lock, flags);
	return 0;
}

111 112 113 114 115
static inline void alarmtimer_rtc_timer_init(void)
{
	rtc_timer_init(&rtctimer, NULL, NULL);
}

116 117 118 119
static struct class_interface alarmtimer_rtc_interface = {
	.add_dev = &alarmtimer_rtc_add_device,
};

120
static int alarmtimer_rtc_interface_setup(void)
121 122
{
	alarmtimer_rtc_interface.class = rtc_class;
123 124 125 126 127
	return class_interface_register(&alarmtimer_rtc_interface);
}
static void alarmtimer_rtc_interface_remove(void)
{
	class_interface_unregister(&alarmtimer_rtc_interface);
128
}
129
#else
130
struct rtc_device *alarmtimer_get_rtcdev(void)
131 132 133 134 135 136
{
	return NULL;
}
#define rtcdev (NULL)
static inline int alarmtimer_rtc_interface_setup(void) { return 0; }
static inline void alarmtimer_rtc_interface_remove(void) { }
137
static inline void alarmtimer_rtc_timer_init(void) { }
138
#endif
139

140
/**
141 142 143 144
 * alarmtimer_enqueue - Adds an alarm timer to an alarm_base timerqueue
 * @base: pointer to the base where the timer is being run
 * @alarm: pointer to alarm being enqueued.
 *
145
 * Adds alarm to a alarm_base timerqueue
146 147 148 149 150
 *
 * Must hold base->lock when calling.
 */
static void alarmtimer_enqueue(struct alarm_base *base, struct alarm *alarm)
{
151 152 153
	if (alarm->state & ALARMTIMER_STATE_ENQUEUED)
		timerqueue_del(&base->timerqueue, &alarm->node);

154
	timerqueue_add(&base->timerqueue, &alarm->node);
155
	alarm->state |= ALARMTIMER_STATE_ENQUEUED;
156 157
}

158
/**
159
 * alarmtimer_dequeue - Removes an alarm timer from an alarm_base timerqueue
160 161 162
 * @base: pointer to the base where the timer is running
 * @alarm: pointer to alarm being removed
 *
163
 * Removes alarm to a alarm_base timerqueue
164 165 166
 *
 * Must hold base->lock when calling.
 */
167
static void alarmtimer_dequeue(struct alarm_base *base, struct alarm *alarm)
168
{
169 170 171
	if (!(alarm->state & ALARMTIMER_STATE_ENQUEUED))
		return;

172
	timerqueue_del(&base->timerqueue, &alarm->node);
173
	alarm->state &= ~ALARMTIMER_STATE_ENQUEUED;
174 175
}

176

177
/**
178 179
 * alarmtimer_fired - Handles alarm hrtimer being fired.
 * @timer: pointer to hrtimer being run
180
 *
181 182 183 184
 * When a alarm timer fires, this runs through the timerqueue to
 * see which alarms expired, and runs those. If there are more alarm
 * timers queued for the future, we set the hrtimer to fire when
 * when the next future alarm timer expires.
185
 */
186
static enum hrtimer_restart alarmtimer_fired(struct hrtimer *timer)
187
{
188 189
	struct alarm *alarm = container_of(timer, struct alarm, timer);
	struct alarm_base *base = &alarm_bases[alarm->type];
190
	unsigned long flags;
191
	int ret = HRTIMER_NORESTART;
192
	int restart = ALARMTIMER_NORESTART;
193 194

	spin_lock_irqsave(&base->lock, flags);
195
	alarmtimer_dequeue(base, alarm);
196
	spin_unlock_irqrestore(&base->lock, flags);
197

198 199
	if (alarm->function)
		restart = alarm->function(alarm, base->gettime());
200

201 202 203 204
	spin_lock_irqsave(&base->lock, flags);
	if (restart != ALARMTIMER_NORESTART) {
		hrtimer_set_expires(&alarm->timer, alarm->node.expires);
		alarmtimer_enqueue(base, alarm);
205
		ret = HRTIMER_RESTART;
206 207 208
	}
	spin_unlock_irqrestore(&base->lock, flags);

209
	trace_alarmtimer_fired(alarm, base->gettime());
210
	return ret;
211 212 213

}

214 215 216 217 218
ktime_t alarm_expires_remaining(const struct alarm *alarm)
{
	struct alarm_base *base = &alarm_bases[alarm->type];
	return ktime_sub(alarm->node.expires, base->gettime());
}
219
EXPORT_SYMBOL_GPL(alarm_expires_remaining);
220

221
#ifdef CONFIG_RTC_CLASS
222
/**
223 224 225 226 227 228 229 230 231 232 233
 * alarmtimer_suspend - Suspend time callback
 * @dev: unused
 * @state: unused
 *
 * When we are going into suspend, we look through the bases
 * to see which is the soonest timer to expire. We then
 * set an rtc timer to fire that far into the future, which
 * will wake us from suspend.
 */
static int alarmtimer_suspend(struct device *dev)
{
234 235
	ktime_t min, now, expires;
	int i, ret, type;
236
	struct rtc_device *rtc;
237 238
	unsigned long flags;
	struct rtc_time tm;
239 240 241

	spin_lock_irqsave(&freezer_delta_lock, flags);
	min = freezer_delta;
242 243
	expires = freezer_expires;
	type = freezer_alarmtype;
T
Thomas Gleixner 已提交
244
	freezer_delta = 0;
245 246
	spin_unlock_irqrestore(&freezer_delta_lock, flags);

247
	rtc = alarmtimer_get_rtcdev();
248
	/* If we have no rtcdev, just return */
249
	if (!rtc)
250 251 252 253 254 255 256 257 258 259 260 261 262 263
		return 0;

	/* Find the soonest timer to expire*/
	for (i = 0; i < ALARM_NUMTYPE; i++) {
		struct alarm_base *base = &alarm_bases[i];
		struct timerqueue_node *next;
		ktime_t delta;

		spin_lock_irqsave(&base->lock, flags);
		next = timerqueue_getnext(&base->timerqueue);
		spin_unlock_irqrestore(&base->lock, flags);
		if (!next)
			continue;
		delta = ktime_sub(next->expires, base->gettime());
T
Thomas Gleixner 已提交
264
		if (!min || (delta < min)) {
265
			expires = next->expires;
266
			min = delta;
267 268
			type = i;
		}
269
	}
T
Thomas Gleixner 已提交
270
	if (min == 0)
271 272
		return 0;

273 274 275 276
	if (ktime_to_ns(min) < 2 * NSEC_PER_SEC) {
		__pm_wakeup_event(ws, 2 * MSEC_PER_SEC);
		return -EBUSY;
	}
277

278 279
	trace_alarmtimer_suspend(expires, type);

280
	/* Setup an rtc timer to fire that far in the future */
281 282
	rtc_timer_cancel(rtc, &rtctimer);
	rtc_read_time(rtc, &tm);
283 284 285
	now = rtc_tm_to_ktime(tm);
	now = ktime_add(now, min);

286
	/* Set alarm, if in the past reject suspend briefly to handle */
T
Thomas Gleixner 已提交
287
	ret = rtc_timer_start(rtc, &rtctimer, now, 0);
288 289 290
	if (ret < 0)
		__pm_wakeup_event(ws, MSEC_PER_SEC);
	return ret;
291
}
292 293 294 295 296 297 298 299 300 301 302

static int alarmtimer_resume(struct device *dev)
{
	struct rtc_device *rtc;

	rtc = alarmtimer_get_rtcdev();
	if (rtc)
		rtc_timer_cancel(rtc, &rtctimer);
	return 0;
}

303 304 305 306 307
#else
static int alarmtimer_suspend(struct device *dev)
{
	return 0;
}
308 309 310 311 312

static int alarmtimer_resume(struct device *dev)
{
	return 0;
}
313
#endif
314

315
/**
316 317 318 319 320 321
 * alarm_init - Initialize an alarm structure
 * @alarm: ptr to alarm to be initialized
 * @type: the type of the alarm
 * @function: callback that is run when the alarm fires
 */
void alarm_init(struct alarm *alarm, enum alarmtimer_type type,
322
		enum alarmtimer_restart (*function)(struct alarm *, ktime_t))
323 324
{
	timerqueue_init(&alarm->node);
325 326 327
	hrtimer_init(&alarm->timer, alarm_bases[type].base_clockid,
			HRTIMER_MODE_ABS);
	alarm->timer.function = alarmtimer_fired;
328 329
	alarm->function = function;
	alarm->type = type;
330
	alarm->state = ALARMTIMER_STATE_INACTIVE;
331
}
332
EXPORT_SYMBOL_GPL(alarm_init);
333

334
/**
335
 * alarm_start - Sets an absolute alarm to fire
336 337 338
 * @alarm: ptr to alarm to set
 * @start: time to run the alarm
 */
339
void alarm_start(struct alarm *alarm, ktime_t start)
340 341 342 343 344 345 346
{
	struct alarm_base *base = &alarm_bases[alarm->type];
	unsigned long flags;

	spin_lock_irqsave(&base->lock, flags);
	alarm->node.expires = start;
	alarmtimer_enqueue(base, alarm);
347
	hrtimer_start(&alarm->timer, alarm->node.expires, HRTIMER_MODE_ABS);
348
	spin_unlock_irqrestore(&base->lock, flags);
349 350

	trace_alarmtimer_start(alarm, base->gettime());
351
}
352
EXPORT_SYMBOL_GPL(alarm_start);
353

354 355 356 357 358
/**
 * alarm_start_relative - Sets a relative alarm to fire
 * @alarm: ptr to alarm to set
 * @start: time relative to now to run the alarm
 */
359
void alarm_start_relative(struct alarm *alarm, ktime_t start)
360 361 362
{
	struct alarm_base *base = &alarm_bases[alarm->type];

363
	start = ktime_add_safe(start, base->gettime());
364
	alarm_start(alarm, start);
365
}
366
EXPORT_SYMBOL_GPL(alarm_start_relative);
367 368 369 370 371 372 373 374 375 376 377 378

void alarm_restart(struct alarm *alarm)
{
	struct alarm_base *base = &alarm_bases[alarm->type];
	unsigned long flags;

	spin_lock_irqsave(&base->lock, flags);
	hrtimer_set_expires(&alarm->timer, alarm->node.expires);
	hrtimer_restart(&alarm->timer);
	alarmtimer_enqueue(base, alarm);
	spin_unlock_irqrestore(&base->lock, flags);
}
379
EXPORT_SYMBOL_GPL(alarm_restart);
380

381
/**
382
 * alarm_try_to_cancel - Tries to cancel an alarm timer
383
 * @alarm: ptr to alarm to be canceled
384 385 386
 *
 * Returns 1 if the timer was canceled, 0 if it was not running,
 * and -1 if the callback was running
387
 */
388
int alarm_try_to_cancel(struct alarm *alarm)
389 390 391
{
	struct alarm_base *base = &alarm_bases[alarm->type];
	unsigned long flags;
392
	int ret;
393

394 395 396
	spin_lock_irqsave(&base->lock, flags);
	ret = hrtimer_try_to_cancel(&alarm->timer);
	if (ret >= 0)
397
		alarmtimer_dequeue(base, alarm);
398
	spin_unlock_irqrestore(&base->lock, flags);
399 400

	trace_alarmtimer_cancel(alarm, base->gettime());
401
	return ret;
402
}
403
EXPORT_SYMBOL_GPL(alarm_try_to_cancel);
404 405


406 407 408 409 410 411 412 413 414 415 416 417 418 419 420
/**
 * alarm_cancel - Spins trying to cancel an alarm timer until it is done
 * @alarm: ptr to alarm to be canceled
 *
 * Returns 1 if the timer was canceled, 0 if it was not active.
 */
int alarm_cancel(struct alarm *alarm)
{
	for (;;) {
		int ret = alarm_try_to_cancel(alarm);
		if (ret >= 0)
			return ret;
		cpu_relax();
	}
}
421
EXPORT_SYMBOL_GPL(alarm_cancel);
422

423 424 425 426 427 428 429 430

u64 alarm_forward(struct alarm *alarm, ktime_t now, ktime_t interval)
{
	u64 overrun = 1;
	ktime_t delta;

	delta = ktime_sub(now, alarm->node.expires);

T
Thomas Gleixner 已提交
431
	if (delta < 0)
432 433
		return 0;

T
Thomas Gleixner 已提交
434
	if (unlikely(delta >= interval)) {
435 436 437 438 439 440 441
		s64 incr = ktime_to_ns(interval);

		overrun = ktime_divns(delta, incr);

		alarm->node.expires = ktime_add_ns(alarm->node.expires,
							incr*overrun);

T
Thomas Gleixner 已提交
442
		if (alarm->node.expires > now)
443 444 445 446 447 448 449 450
			return overrun;
		/*
		 * This (and the ktime_add() below) is the
		 * correction for exact:
		 */
		overrun++;
	}

451
	alarm->node.expires = ktime_add_safe(alarm->node.expires, interval);
452 453
	return overrun;
}
454
EXPORT_SYMBOL_GPL(alarm_forward);
455

456 457 458 459 460 461
u64 alarm_forward_now(struct alarm *alarm, ktime_t interval)
{
	struct alarm_base *base = &alarm_bases[alarm->type];

	return alarm_forward(alarm, base->gettime(), interval);
}
462
EXPORT_SYMBOL_GPL(alarm_forward_now);
463

464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495
#ifdef CONFIG_POSIX_TIMERS

static void alarmtimer_freezerset(ktime_t absexp, enum alarmtimer_type type)
{
	struct alarm_base *base;
	unsigned long flags;
	ktime_t delta;

	switch(type) {
	case ALARM_REALTIME:
		base = &alarm_bases[ALARM_REALTIME];
		type = ALARM_REALTIME_FREEZER;
		break;
	case ALARM_BOOTTIME:
		base = &alarm_bases[ALARM_BOOTTIME];
		type = ALARM_BOOTTIME_FREEZER;
		break;
	default:
		WARN_ONCE(1, "Invalid alarm type: %d\n", type);
		return;
	}

	delta = ktime_sub(absexp, base->gettime());

	spin_lock_irqsave(&freezer_delta_lock, flags);
	if (!freezer_delta || (delta < freezer_delta)) {
		freezer_delta = delta;
		freezer_expires = absexp;
		freezer_alarmtype = type;
	}
	spin_unlock_irqrestore(&freezer_delta_lock, flags);
}
496

497
/**
498 499 500 501 502 503 504 505 506 507 508 509
 * clock2alarm - helper that converts from clockid to alarmtypes
 * @clockid: clockid.
 */
static enum alarmtimer_type clock2alarm(clockid_t clockid)
{
	if (clockid == CLOCK_REALTIME_ALARM)
		return ALARM_REALTIME;
	if (clockid == CLOCK_BOOTTIME_ALARM)
		return ALARM_BOOTTIME;
	return -1;
}

510
/**
511 512 513 514 515
 * alarm_handle_timer - Callback for posix timers
 * @alarm: alarm that fired
 *
 * Posix timer callback for expired alarm timers.
 */
516 517
static enum alarmtimer_restart alarm_handle_timer(struct alarm *alarm,
							ktime_t now)
518 519
{
	struct k_itimer *ptr = container_of(alarm, struct k_itimer,
520
					    it.alarm.alarmtimer);
521
	enum alarmtimer_restart result = ALARMTIMER_NORESTART;
522 523
	unsigned long flags;
	int si_private = 0;
524 525

	spin_lock_irqsave(&ptr->it_lock, flags);
526

527 528 529 530 531 532 533 534 535 536 537 538
	ptr->it_active = 0;
	if (ptr->it_interval)
		si_private = ++ptr->it_requeue_pending;

	if (posix_timer_event(ptr, si_private) && ptr->it_interval) {
		/*
		 * Handle ignored signals and rearm the timer. This will go
		 * away once we handle ignored signals proper.
		 */
		ptr->it_overrun += alarm_forward_now(alarm, ptr->it_interval);
		++ptr->it_requeue_pending;
		ptr->it_active = 1;
539
		result = ALARMTIMER_RESTART;
540
	}
541 542 543
	spin_unlock_irqrestore(&ptr->it_lock, flags);

	return result;
544 545
}

546 547 548 549 550 551 552 553 554 555 556 557
/**
 * alarm_timer_rearm - Posix timer callback for rearming timer
 * @timr:	Pointer to the posixtimer data struct
 */
static void alarm_timer_rearm(struct k_itimer *timr)
{
	struct alarm *alarm = &timr->it.alarm.alarmtimer;

	timr->it_overrun += alarm_forward_now(alarm, timr->it_interval);
	alarm_start(alarm, alarm->node.expires);
}

558 559 560 561 562 563 564 565 566 567 568 569
/**
 * alarm_timer_forward - Posix timer callback for forwarding timer
 * @timr:	Pointer to the posixtimer data struct
 * @now:	Current time to forward the timer against
 */
static int alarm_timer_forward(struct k_itimer *timr, ktime_t now)
{
	struct alarm *alarm = &timr->it.alarm.alarmtimer;

	return (int) alarm_forward(alarm, timr->it_interval, now);
}

570 571 572 573 574 575 576 577 578 579 580 581
/**
 * alarm_timer_remaining - Posix timer callback to retrieve remaining time
 * @timr:	Pointer to the posixtimer data struct
 * @now:	Current time to calculate against
 */
static ktime_t alarm_timer_remaining(struct k_itimer *timr, ktime_t now)
{
	struct alarm *alarm = &timr->it.alarm.alarmtimer;

	return ktime_sub(now, alarm->node.expires);
}

582 583 584 585 586 587 588 589 590
/**
 * alarm_timer_try_to_cancel - Posix timer callback to cancel a timer
 * @timr:	Pointer to the posixtimer data struct
 */
static int alarm_timer_try_to_cancel(struct k_itimer *timr)
{
	return alarm_try_to_cancel(&timr->it.alarm.alarmtimer);
}

591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611
/**
 * alarm_timer_arm - Posix timer callback to arm a timer
 * @timr:	Pointer to the posixtimer data struct
 * @expires:	The new expiry time
 * @absolute:	Expiry value is absolute time
 * @sigev_none:	Posix timer does not deliver signals
 */
static void alarm_timer_arm(struct k_itimer *timr, ktime_t expires,
			    bool absolute, bool sigev_none)
{
	struct alarm *alarm = &timr->it.alarm.alarmtimer;
	struct alarm_base *base = &alarm_bases[alarm->type];

	if (!absolute)
		expires = ktime_add_safe(expires, base->gettime());
	if (sigev_none)
		alarm->node.expires = expires;
	else
		alarm_start(&timr->it.alarm.alarmtimer, expires);
}

612
/**
613 614 615 616 617 618
 * alarm_clock_getres - posix getres interface
 * @which_clock: clockid
 * @tp: timespec to fill
 *
 * Returns the granularity of underlying alarm base clock
 */
619
static int alarm_clock_getres(const clockid_t which_clock, struct timespec64 *tp)
620
{
621
	if (!alarmtimer_get_rtcdev())
622
		return -EINVAL;
623

624 625 626
	tp->tv_sec = 0;
	tp->tv_nsec = hrtimer_resolution;
	return 0;
627 628 629 630 631 632 633 634 635
}

/**
 * alarm_clock_get - posix clock_get interface
 * @which_clock: clockid
 * @tp: timespec to fill.
 *
 * Provides the underlying alarm base time.
 */
636
static int alarm_clock_get(clockid_t which_clock, struct timespec64 *tp)
637 638 639
{
	struct alarm_base *base = &alarm_bases[clock2alarm(which_clock)];

640
	if (!alarmtimer_get_rtcdev())
641
		return -EINVAL;
642

643
	*tp = ktime_to_timespec64(base->gettime());
644 645 646 647 648 649 650 651 652 653 654 655 656
	return 0;
}

/**
 * alarm_timer_create - posix timer_create interface
 * @new_timer: k_itimer pointer to manage
 *
 * Initializes the k_itimer structure.
 */
static int alarm_timer_create(struct k_itimer *new_timer)
{
	enum  alarmtimer_type type;

657 658 659
	if (!alarmtimer_get_rtcdev())
		return -ENOTSUPP;

660 661 662 663
	if (!capable(CAP_WAKE_ALARM))
		return -EPERM;

	type = clock2alarm(new_timer->it_clock);
664
	alarm_init(&new_timer->it.alarm.alarmtimer, type, alarm_handle_timer);
665 666 667 668 669 670 671 672 673
	return 0;
}

/**
 * alarmtimer_nsleep_wakeup - Wakeup function for alarm_timer_nsleep
 * @alarm: ptr to alarm that fired
 *
 * Wakes up the task that set the alarmtimer
 */
674 675
static enum alarmtimer_restart alarmtimer_nsleep_wakeup(struct alarm *alarm,
								ktime_t now)
676 677 678 679 680 681
{
	struct task_struct *task = (struct task_struct *)alarm->data;

	alarm->data = NULL;
	if (task)
		wake_up_process(task);
682
	return ALARMTIMER_NORESTART;
683 684 685 686 687 688 689 690 691
}

/**
 * alarmtimer_do_nsleep - Internal alarmtimer nsleep implementation
 * @alarm: ptr to alarmtimer
 * @absexp: absolute expiration time
 *
 * Sets the alarm timer and sleeps until it is fired or interrupted.
 */
692 693
static int alarmtimer_do_nsleep(struct alarm *alarm, ktime_t absexp,
				enum alarmtimer_type type)
694
{
695
	struct restart_block *restart;
696 697 698
	alarm->data = (void *)current;
	do {
		set_current_state(TASK_INTERRUPTIBLE);
699
		alarm_start(alarm, absexp);
700 701 702 703 704 705 706 707
		if (likely(alarm->data))
			schedule();

		alarm_cancel(alarm);
	} while (alarm->data && !signal_pending(current));

	__set_current_state(TASK_RUNNING);

708
	if (!alarm->data)
709 710
		return 0;

711 712
	if (freezing(current))
		alarmtimer_freezerset(absexp, type);
713 714
	restart = &current->restart_block;
	if (restart->nanosleep.type != TT_NONE) {
715 716
		struct timespec rmt;
		ktime_t rem;
717

718
		rem = ktime_sub(absexp, alarm_bases[type].gettime());
719

720 721 722 723
		if (rem <= 0)
			return 0;
		rmt = ktime_to_timespec(rem);

724 725 726 727 728 729 730 731
#ifdef CONFIG_COMPAT
		if (restart->nanosleep.type == TT_COMPAT) {
			if (compat_put_timespec(&rmt,
						restart->nanosleep.compat_rmtp))
				return -EFAULT;
		} else
#endif
		if (copy_to_user(restart->nanosleep.rmtp, &rmt, sizeof(rmt)))
732 733 734
			return -EFAULT;
	}
	return -ERESTART_RESTARTBLOCK;
735 736 737 738 739 740 741 742 743 744
}

/**
 * alarm_timer_nsleep_restart - restartblock alarmtimer nsleep
 * @restart: ptr to restart block
 *
 * Handles restarted clock_nanosleep calls
 */
static long __sched alarm_timer_nsleep_restart(struct restart_block *restart)
{
745
	enum  alarmtimer_type type = restart->nanosleep.clockid;
746
	ktime_t exp = restart->nanosleep.expires;
747 748 749 750
	struct alarm alarm;

	alarm_init(&alarm, type, alarmtimer_nsleep_wakeup);

751
	return alarmtimer_do_nsleep(&alarm, exp, type);
752 753 754 755 756 757 758 759 760 761 762 763
}

/**
 * alarm_timer_nsleep - alarmtimer nanosleep
 * @which_clock: clockid
 * @flags: determins abstime or relative
 * @tsreq: requested sleep time (abs or rel)
 * @rmtp: remaining sleep time saved
 *
 * Handles clock_nanosleep calls against _ALARM clockids
 */
static int alarm_timer_nsleep(const clockid_t which_clock, int flags,
764
			      struct timespec64 *tsreq)
765 766
{
	enum  alarmtimer_type type = clock2alarm(which_clock);
767
	struct restart_block *restart = &current->restart_block;
768 769 770 771
	struct alarm alarm;
	ktime_t exp;
	int ret = 0;

772 773 774
	if (!alarmtimer_get_rtcdev())
		return -ENOTSUPP;

775 776 777
	if (flags & ~TIMER_ABSTIME)
		return -EINVAL;

778 779 780 781 782
	if (!capable(CAP_WAKE_ALARM))
		return -EPERM;

	alarm_init(&alarm, type, alarmtimer_nsleep_wakeup);

783
	exp = timespec64_to_ktime(*tsreq);
784 785 786 787 788 789
	/* Convert (if necessary) to absolute time */
	if (flags != TIMER_ABSTIME) {
		ktime_t now = alarm_bases[type].gettime();
		exp = ktime_add(now, exp);
	}

790 791 792
	ret = alarmtimer_do_nsleep(&alarm, exp, type);
	if (ret != -ERESTART_RESTARTBLOCK)
		return ret;
793 794

	/* abs timers don't set remaining time or restart */
795 796
	if (flags == TIMER_ABSTIME)
		return -ERESTARTNOHAND;
797 798

	restart->fn = alarm_timer_nsleep_restart;
799
	restart->nanosleep.clockid = type;
T
Thomas Gleixner 已提交
800
	restart->nanosleep.expires = exp;
801 802
	return ret;
}
803

804
const struct k_clock alarm_clock = {
805 806 807
	.clock_getres		= alarm_clock_getres,
	.clock_get		= alarm_clock_get,
	.timer_create		= alarm_timer_create,
808 809 810
	.timer_set		= common_timer_set,
	.timer_del		= common_timer_del,
	.timer_get		= common_timer_get,
811
	.timer_arm		= alarm_timer_arm,
812 813 814
	.timer_rearm		= alarm_timer_rearm,
	.timer_forward		= alarm_timer_forward,
	.timer_remaining	= alarm_timer_remaining,
815
	.timer_try_to_cancel	= alarm_timer_try_to_cancel,
816
	.nsleep			= alarm_timer_nsleep,
817 818 819
};
#endif /* CONFIG_POSIX_TIMERS */

820 821 822 823

/* Suspend hook structures */
static const struct dev_pm_ops alarmtimer_pm_ops = {
	.suspend = alarmtimer_suspend,
824
	.resume = alarmtimer_resume,
825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841
};

static struct platform_driver alarmtimer_driver = {
	.driver = {
		.name = "alarmtimer",
		.pm = &alarmtimer_pm_ops,
	}
};

/**
 * alarmtimer_init - Initialize alarm timer code
 *
 * This function initializes the alarm bases and registers
 * the posix clock ids.
 */
static int __init alarmtimer_init(void)
{
842
	struct platform_device *pdev;
843 844
	int error = 0;
	int i;
845

846
	alarmtimer_rtc_timer_init();
847

848 849 850 851 852 853 854 855 856
	/* Initialize alarm bases */
	alarm_bases[ALARM_REALTIME].base_clockid = CLOCK_REALTIME;
	alarm_bases[ALARM_REALTIME].gettime = &ktime_get_real;
	alarm_bases[ALARM_BOOTTIME].base_clockid = CLOCK_BOOTTIME;
	alarm_bases[ALARM_BOOTTIME].gettime = &ktime_get_boottime;
	for (i = 0; i < ALARM_NUMTYPE; i++) {
		timerqueue_init_head(&alarm_bases[i].timerqueue);
		spin_lock_init(&alarm_bases[i].lock);
	}
857

858 859 860 861
	error = alarmtimer_rtc_interface_setup();
	if (error)
		return error;

862
	error = platform_driver_register(&alarmtimer_driver);
863 864
	if (error)
		goto out_if;
865

866 867 868 869 870
	pdev = platform_device_register_simple("alarmtimer", -1, NULL, 0);
	if (IS_ERR(pdev)) {
		error = PTR_ERR(pdev);
		goto out_drv;
	}
871
	ws = wakeup_source_register("alarmtimer");
872 873 874 875 876 877
	return 0;

out_drv:
	platform_driver_unregister(&alarmtimer_driver);
out_if:
	alarmtimer_rtc_interface_remove();
878 879 880
	return error;
}
device_initcall(alarmtimer_init);