alarmtimer.c 21.8 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
		struct timespec64 rmt;
716
		ktime_t rem;
717

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

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

724
		return nanosleep_copyout(restart, &rmt);
725 726
	}
	return -ERESTART_RESTARTBLOCK;
727 728 729 730 731 732 733 734 735 736
}

/**
 * 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)
{
737
	enum  alarmtimer_type type = restart->nanosleep.clockid;
738
	ktime_t exp = restart->nanosleep.expires;
739 740 741 742
	struct alarm alarm;

	alarm_init(&alarm, type, alarmtimer_nsleep_wakeup);

743
	return alarmtimer_do_nsleep(&alarm, exp, type);
744 745 746 747 748 749 750 751 752 753 754 755
}

/**
 * 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,
756
			      const struct timespec64 *tsreq)
757 758
{
	enum  alarmtimer_type type = clock2alarm(which_clock);
759
	struct restart_block *restart = &current->restart_block;
760 761 762 763
	struct alarm alarm;
	ktime_t exp;
	int ret = 0;

764 765 766
	if (!alarmtimer_get_rtcdev())
		return -ENOTSUPP;

767 768 769
	if (flags & ~TIMER_ABSTIME)
		return -EINVAL;

770 771 772 773 774
	if (!capable(CAP_WAKE_ALARM))
		return -EPERM;

	alarm_init(&alarm, type, alarmtimer_nsleep_wakeup);

775
	exp = timespec64_to_ktime(*tsreq);
776 777 778 779 780 781
	/* Convert (if necessary) to absolute time */
	if (flags != TIMER_ABSTIME) {
		ktime_t now = alarm_bases[type].gettime();
		exp = ktime_add(now, exp);
	}

782 783 784
	ret = alarmtimer_do_nsleep(&alarm, exp, type);
	if (ret != -ERESTART_RESTARTBLOCK)
		return ret;
785 786

	/* abs timers don't set remaining time or restart */
787 788
	if (flags == TIMER_ABSTIME)
		return -ERESTARTNOHAND;
789 790

	restart->fn = alarm_timer_nsleep_restart;
791
	restart->nanosleep.clockid = type;
T
Thomas Gleixner 已提交
792
	restart->nanosleep.expires = exp;
793 794
	return ret;
}
795

796
const struct k_clock alarm_clock = {
797 798 799
	.clock_getres		= alarm_clock_getres,
	.clock_get		= alarm_clock_get,
	.timer_create		= alarm_timer_create,
800 801 802
	.timer_set		= common_timer_set,
	.timer_del		= common_timer_del,
	.timer_get		= common_timer_get,
803
	.timer_arm		= alarm_timer_arm,
804 805 806
	.timer_rearm		= alarm_timer_rearm,
	.timer_forward		= alarm_timer_forward,
	.timer_remaining	= alarm_timer_remaining,
807
	.timer_try_to_cancel	= alarm_timer_try_to_cancel,
808
	.nsleep			= alarm_timer_nsleep,
809 810 811
};
#endif /* CONFIG_POSIX_TIMERS */

812 813 814 815

/* Suspend hook structures */
static const struct dev_pm_ops alarmtimer_pm_ops = {
	.suspend = alarmtimer_suspend,
816
	.resume = alarmtimer_resume,
817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833
};

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)
{
834
	struct platform_device *pdev;
835 836
	int error = 0;
	int i;
837

838
	alarmtimer_rtc_timer_init();
839

840 841 842 843 844 845 846 847 848
	/* 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);
	}
849

850 851 852 853
	error = alarmtimer_rtc_interface_setup();
	if (error)
		return error;

854
	error = platform_driver_register(&alarmtimer_driver);
855 856
	if (error)
		goto out_if;
857

858 859 860 861 862
	pdev = platform_device_register_simple("alarmtimer", -1, NULL, 0);
	if (IS_ERR(pdev)) {
		error = PTR_ERR(pdev);
		goto out_drv;
	}
863
	ws = wakeup_source_register("alarmtimer");
864 865 866 867 868 869
	return 0;

out_drv:
	platform_driver_unregister(&alarmtimer_driver);
out_if:
	alarmtimer_rtc_interface_remove();
870 871 872
	return error;
}
device_initcall(alarmtimer_init);