alarmtimer.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969
  1. /*
  2. * Alarmtimer interface
  3. *
  4. * This interface provides a timer which is similarto hrtimers,
  5. * but triggers a RTC alarm if the box is suspend.
  6. *
  7. * This interface is influenced by the Android RTC Alarm timer
  8. * interface.
  9. *
  10. * Copyright (C) 2010 IBM Corperation
  11. *
  12. * Author: John Stultz <john.stultz@linaro.org>
  13. *
  14. * This program is free software; you can redistribute it and/or modify
  15. * it under the terms of the GNU General Public License version 2 as
  16. * published by the Free Software Foundation.
  17. */
  18. #include <linux/time.h>
  19. #include <linux/hrtimer.h>
  20. #include <linux/timerqueue.h>
  21. #include <linux/rtc.h>
  22. #include <linux/sched/signal.h>
  23. #include <linux/sched/debug.h>
  24. #include <linux/alarmtimer.h>
  25. #include <linux/mutex.h>
  26. #include <linux/platform_device.h>
  27. #include <linux/posix-timers.h>
  28. #include <linux/workqueue.h>
  29. #include <linux/freezer.h>
  30. #include "posix-timers.h"
  31. #define CREATE_TRACE_POINTS
  32. #include <trace/events/alarmtimer.h>
  33. /**
  34. * struct alarm_base - Alarm timer bases
  35. * @lock: Lock for syncrhonized access to the base
  36. * @timerqueue: Timerqueue head managing the list of events
  37. * @gettime: Function to read the time correlating to the base
  38. * @base_clockid: clockid for the base
  39. */
  40. static struct alarm_base {
  41. spinlock_t lock;
  42. struct timerqueue_head timerqueue;
  43. ktime_t (*gettime)(void);
  44. clockid_t base_clockid;
  45. } alarm_bases[ALARM_NUMTYPE];
  46. #if defined(CONFIG_POSIX_TIMERS) || defined(CONFIG_RTC_CLASS)
  47. /* freezer information to handle clock_nanosleep triggered wakeups */
  48. static enum alarmtimer_type freezer_alarmtype;
  49. static ktime_t freezer_expires;
  50. static ktime_t freezer_delta;
  51. static DEFINE_SPINLOCK(freezer_delta_lock);
  52. #endif
  53. static struct wakeup_source *ws;
  54. #ifdef CONFIG_RTC_CLASS
  55. /* rtc timer and device for setting alarm wakeups at suspend */
  56. static struct rtc_timer rtctimer;
  57. static struct rtc_device *rtcdev;
  58. static DEFINE_SPINLOCK(rtcdev_lock);
  59. /**
  60. * alarmtimer_get_rtcdev - Return selected rtcdevice
  61. *
  62. * This function returns the rtc device to use for wakealarms.
  63. * If one has not already been chosen, it checks to see if a
  64. * functional rtc device is available.
  65. */
  66. struct rtc_device *alarmtimer_get_rtcdev(void)
  67. {
  68. unsigned long flags;
  69. struct rtc_device *ret;
  70. spin_lock_irqsave(&rtcdev_lock, flags);
  71. ret = rtcdev;
  72. spin_unlock_irqrestore(&rtcdev_lock, flags);
  73. return ret;
  74. }
  75. EXPORT_SYMBOL_GPL(alarmtimer_get_rtcdev);
  76. static int alarmtimer_rtc_add_device(struct device *dev,
  77. struct class_interface *class_intf)
  78. {
  79. unsigned long flags;
  80. struct rtc_device *rtc = to_rtc_device(dev);
  81. if (rtcdev)
  82. return -EBUSY;
  83. if (!rtc->ops->set_alarm)
  84. return -1;
  85. if (!device_may_wakeup(rtc->dev.parent))
  86. return -1;
  87. spin_lock_irqsave(&rtcdev_lock, flags);
  88. if (!rtcdev) {
  89. rtcdev = rtc;
  90. /* hold a reference so it doesn't go away */
  91. get_device(dev);
  92. }
  93. spin_unlock_irqrestore(&rtcdev_lock, flags);
  94. return 0;
  95. }
  96. static inline void alarmtimer_rtc_timer_init(void)
  97. {
  98. rtc_timer_init(&rtctimer, NULL, NULL);
  99. }
  100. static struct class_interface alarmtimer_rtc_interface = {
  101. .add_dev = &alarmtimer_rtc_add_device,
  102. };
  103. static int alarmtimer_rtc_interface_setup(void)
  104. {
  105. alarmtimer_rtc_interface.class = rtc_class;
  106. return class_interface_register(&alarmtimer_rtc_interface);
  107. }
  108. static void alarmtimer_rtc_interface_remove(void)
  109. {
  110. class_interface_unregister(&alarmtimer_rtc_interface);
  111. }
  112. #else
  113. struct rtc_device *alarmtimer_get_rtcdev(void)
  114. {
  115. return NULL;
  116. }
  117. #define rtcdev (NULL)
  118. static inline int alarmtimer_rtc_interface_setup(void) { return 0; }
  119. static inline void alarmtimer_rtc_interface_remove(void) { }
  120. static inline void alarmtimer_rtc_timer_init(void) { }
  121. #endif
  122. /**
  123. * alarmtimer_enqueue - Adds an alarm timer to an alarm_base timerqueue
  124. * @base: pointer to the base where the timer is being run
  125. * @alarm: pointer to alarm being enqueued.
  126. *
  127. * Adds alarm to a alarm_base timerqueue
  128. *
  129. * Must hold base->lock when calling.
  130. */
  131. static void alarmtimer_enqueue(struct alarm_base *base, struct alarm *alarm)
  132. {
  133. if (alarm->state & ALARMTIMER_STATE_ENQUEUED)
  134. timerqueue_del(&base->timerqueue, &alarm->node);
  135. timerqueue_add(&base->timerqueue, &alarm->node);
  136. alarm->state |= ALARMTIMER_STATE_ENQUEUED;
  137. }
  138. /**
  139. * alarmtimer_dequeue - Removes an alarm timer from an alarm_base timerqueue
  140. * @base: pointer to the base where the timer is running
  141. * @alarm: pointer to alarm being removed
  142. *
  143. * Removes alarm to a alarm_base timerqueue
  144. *
  145. * Must hold base->lock when calling.
  146. */
  147. static void alarmtimer_dequeue(struct alarm_base *base, struct alarm *alarm)
  148. {
  149. if (!(alarm->state & ALARMTIMER_STATE_ENQUEUED))
  150. return;
  151. timerqueue_del(&base->timerqueue, &alarm->node);
  152. alarm->state &= ~ALARMTIMER_STATE_ENQUEUED;
  153. }
  154. /**
  155. * alarmtimer_fired - Handles alarm hrtimer being fired.
  156. * @timer: pointer to hrtimer being run
  157. *
  158. * When a alarm timer fires, this runs through the timerqueue to
  159. * see which alarms expired, and runs those. If there are more alarm
  160. * timers queued for the future, we set the hrtimer to fire when
  161. * when the next future alarm timer expires.
  162. */
  163. static enum hrtimer_restart alarmtimer_fired(struct hrtimer *timer)
  164. {
  165. struct alarm *alarm = container_of(timer, struct alarm, timer);
  166. struct alarm_base *base = &alarm_bases[alarm->type];
  167. unsigned long flags;
  168. int ret = HRTIMER_NORESTART;
  169. int restart = ALARMTIMER_NORESTART;
  170. spin_lock_irqsave(&base->lock, flags);
  171. alarmtimer_dequeue(base, alarm);
  172. spin_unlock_irqrestore(&base->lock, flags);
  173. if (alarm->function)
  174. restart = alarm->function(alarm, base->gettime());
  175. spin_lock_irqsave(&base->lock, flags);
  176. if (restart != ALARMTIMER_NORESTART) {
  177. hrtimer_set_expires(&alarm->timer, alarm->node.expires);
  178. alarmtimer_enqueue(base, alarm);
  179. ret = HRTIMER_RESTART;
  180. }
  181. spin_unlock_irqrestore(&base->lock, flags);
  182. trace_alarmtimer_fired(alarm, base->gettime());
  183. return ret;
  184. }
  185. ktime_t alarm_expires_remaining(const struct alarm *alarm)
  186. {
  187. struct alarm_base *base = &alarm_bases[alarm->type];
  188. return ktime_sub(alarm->node.expires, base->gettime());
  189. }
  190. EXPORT_SYMBOL_GPL(alarm_expires_remaining);
  191. #ifdef CONFIG_RTC_CLASS
  192. /**
  193. * alarmtimer_suspend - Suspend time callback
  194. * @dev: unused
  195. * @state: unused
  196. *
  197. * When we are going into suspend, we look through the bases
  198. * to see which is the soonest timer to expire. We then
  199. * set an rtc timer to fire that far into the future, which
  200. * will wake us from suspend.
  201. */
  202. static int alarmtimer_suspend(struct device *dev)
  203. {
  204. ktime_t min, now, expires;
  205. int i, ret, type;
  206. struct rtc_device *rtc;
  207. unsigned long flags;
  208. struct rtc_time tm;
  209. spin_lock_irqsave(&freezer_delta_lock, flags);
  210. min = freezer_delta;
  211. expires = freezer_expires;
  212. type = freezer_alarmtype;
  213. freezer_delta = 0;
  214. spin_unlock_irqrestore(&freezer_delta_lock, flags);
  215. rtc = alarmtimer_get_rtcdev();
  216. /* If we have no rtcdev, just return */
  217. if (!rtc)
  218. return 0;
  219. /* Find the soonest timer to expire*/
  220. for (i = 0; i < ALARM_NUMTYPE; i++) {
  221. struct alarm_base *base = &alarm_bases[i];
  222. struct timerqueue_node *next;
  223. ktime_t delta;
  224. spin_lock_irqsave(&base->lock, flags);
  225. next = timerqueue_getnext(&base->timerqueue);
  226. spin_unlock_irqrestore(&base->lock, flags);
  227. if (!next)
  228. continue;
  229. delta = ktime_sub(next->expires, base->gettime());
  230. if (!min || (delta < min)) {
  231. expires = next->expires;
  232. min = delta;
  233. type = i;
  234. }
  235. }
  236. if (min == 0)
  237. return 0;
  238. if (ktime_to_ns(min) < 2 * NSEC_PER_SEC) {
  239. __pm_wakeup_event(ws, 2 * MSEC_PER_SEC);
  240. return -EBUSY;
  241. }
  242. trace_alarmtimer_suspend(expires, type);
  243. /* Setup an rtc timer to fire that far in the future */
  244. rtc_timer_cancel(rtc, &rtctimer);
  245. rtc_read_time(rtc, &tm);
  246. now = rtc_tm_to_ktime(tm);
  247. now = ktime_add(now, min);
  248. /* Set alarm, if in the past reject suspend briefly to handle */
  249. ret = rtc_timer_start(rtc, &rtctimer, now, 0);
  250. if (ret < 0)
  251. __pm_wakeup_event(ws, MSEC_PER_SEC);
  252. return ret;
  253. }
  254. static int alarmtimer_resume(struct device *dev)
  255. {
  256. struct rtc_device *rtc;
  257. rtc = alarmtimer_get_rtcdev();
  258. if (rtc)
  259. rtc_timer_cancel(rtc, &rtctimer);
  260. return 0;
  261. }
  262. #else
  263. static int alarmtimer_suspend(struct device *dev)
  264. {
  265. return 0;
  266. }
  267. static int alarmtimer_resume(struct device *dev)
  268. {
  269. return 0;
  270. }
  271. #endif
  272. /**
  273. * alarm_init - Initialize an alarm structure
  274. * @alarm: ptr to alarm to be initialized
  275. * @type: the type of the alarm
  276. * @function: callback that is run when the alarm fires
  277. */
  278. void alarm_init(struct alarm *alarm, enum alarmtimer_type type,
  279. enum alarmtimer_restart (*function)(struct alarm *, ktime_t))
  280. {
  281. timerqueue_init(&alarm->node);
  282. hrtimer_init(&alarm->timer, alarm_bases[type].base_clockid,
  283. HRTIMER_MODE_ABS);
  284. alarm->timer.function = alarmtimer_fired;
  285. alarm->function = function;
  286. alarm->type = type;
  287. alarm->state = ALARMTIMER_STATE_INACTIVE;
  288. }
  289. EXPORT_SYMBOL_GPL(alarm_init);
  290. /**
  291. * alarm_start - Sets an absolute alarm to fire
  292. * @alarm: ptr to alarm to set
  293. * @start: time to run the alarm
  294. */
  295. void alarm_start(struct alarm *alarm, ktime_t start)
  296. {
  297. struct alarm_base *base = &alarm_bases[alarm->type];
  298. unsigned long flags;
  299. spin_lock_irqsave(&base->lock, flags);
  300. alarm->node.expires = start;
  301. alarmtimer_enqueue(base, alarm);
  302. hrtimer_start(&alarm->timer, alarm->node.expires, HRTIMER_MODE_ABS);
  303. spin_unlock_irqrestore(&base->lock, flags);
  304. trace_alarmtimer_start(alarm, base->gettime());
  305. }
  306. EXPORT_SYMBOL_GPL(alarm_start);
  307. /**
  308. * alarm_start_relative - Sets a relative alarm to fire
  309. * @alarm: ptr to alarm to set
  310. * @start: time relative to now to run the alarm
  311. */
  312. void alarm_start_relative(struct alarm *alarm, ktime_t start)
  313. {
  314. struct alarm_base *base = &alarm_bases[alarm->type];
  315. start = ktime_add_safe(start, base->gettime());
  316. alarm_start(alarm, start);
  317. }
  318. EXPORT_SYMBOL_GPL(alarm_start_relative);
  319. void alarm_restart(struct alarm *alarm)
  320. {
  321. struct alarm_base *base = &alarm_bases[alarm->type];
  322. unsigned long flags;
  323. spin_lock_irqsave(&base->lock, flags);
  324. hrtimer_set_expires(&alarm->timer, alarm->node.expires);
  325. hrtimer_restart(&alarm->timer);
  326. alarmtimer_enqueue(base, alarm);
  327. spin_unlock_irqrestore(&base->lock, flags);
  328. }
  329. EXPORT_SYMBOL_GPL(alarm_restart);
  330. /**
  331. * alarm_try_to_cancel - Tries to cancel an alarm timer
  332. * @alarm: ptr to alarm to be canceled
  333. *
  334. * Returns 1 if the timer was canceled, 0 if it was not running,
  335. * and -1 if the callback was running
  336. */
  337. int alarm_try_to_cancel(struct alarm *alarm)
  338. {
  339. struct alarm_base *base = &alarm_bases[alarm->type];
  340. unsigned long flags;
  341. int ret;
  342. spin_lock_irqsave(&base->lock, flags);
  343. ret = hrtimer_try_to_cancel(&alarm->timer);
  344. if (ret >= 0)
  345. alarmtimer_dequeue(base, alarm);
  346. spin_unlock_irqrestore(&base->lock, flags);
  347. trace_alarmtimer_cancel(alarm, base->gettime());
  348. return ret;
  349. }
  350. EXPORT_SYMBOL_GPL(alarm_try_to_cancel);
  351. /**
  352. * alarm_cancel - Spins trying to cancel an alarm timer until it is done
  353. * @alarm: ptr to alarm to be canceled
  354. *
  355. * Returns 1 if the timer was canceled, 0 if it was not active.
  356. */
  357. int alarm_cancel(struct alarm *alarm)
  358. {
  359. for (;;) {
  360. int ret = alarm_try_to_cancel(alarm);
  361. if (ret >= 0)
  362. return ret;
  363. cpu_relax();
  364. }
  365. }
  366. EXPORT_SYMBOL_GPL(alarm_cancel);
  367. u64 alarm_forward(struct alarm *alarm, ktime_t now, ktime_t interval)
  368. {
  369. u64 overrun = 1;
  370. ktime_t delta;
  371. delta = ktime_sub(now, alarm->node.expires);
  372. if (delta < 0)
  373. return 0;
  374. if (unlikely(delta >= interval)) {
  375. s64 incr = ktime_to_ns(interval);
  376. overrun = ktime_divns(delta, incr);
  377. alarm->node.expires = ktime_add_ns(alarm->node.expires,
  378. incr*overrun);
  379. if (alarm->node.expires > now)
  380. return overrun;
  381. /*
  382. * This (and the ktime_add() below) is the
  383. * correction for exact:
  384. */
  385. overrun++;
  386. }
  387. alarm->node.expires = ktime_add_safe(alarm->node.expires, interval);
  388. return overrun;
  389. }
  390. EXPORT_SYMBOL_GPL(alarm_forward);
  391. u64 alarm_forward_now(struct alarm *alarm, ktime_t interval)
  392. {
  393. struct alarm_base *base = &alarm_bases[alarm->type];
  394. return alarm_forward(alarm, base->gettime(), interval);
  395. }
  396. EXPORT_SYMBOL_GPL(alarm_forward_now);
  397. #ifdef CONFIG_POSIX_TIMERS
  398. static void alarmtimer_freezerset(ktime_t absexp, enum alarmtimer_type type)
  399. {
  400. struct alarm_base *base;
  401. unsigned long flags;
  402. ktime_t delta;
  403. switch(type) {
  404. case ALARM_REALTIME:
  405. base = &alarm_bases[ALARM_REALTIME];
  406. type = ALARM_REALTIME_FREEZER;
  407. break;
  408. case ALARM_BOOTTIME:
  409. base = &alarm_bases[ALARM_BOOTTIME];
  410. type = ALARM_BOOTTIME_FREEZER;
  411. break;
  412. default:
  413. WARN_ONCE(1, "Invalid alarm type: %d\n", type);
  414. return;
  415. }
  416. delta = ktime_sub(absexp, base->gettime());
  417. spin_lock_irqsave(&freezer_delta_lock, flags);
  418. if (!freezer_delta || (delta < freezer_delta)) {
  419. freezer_delta = delta;
  420. freezer_expires = absexp;
  421. freezer_alarmtype = type;
  422. }
  423. spin_unlock_irqrestore(&freezer_delta_lock, flags);
  424. }
  425. /**
  426. * clock2alarm - helper that converts from clockid to alarmtypes
  427. * @clockid: clockid.
  428. */
  429. static enum alarmtimer_type clock2alarm(clockid_t clockid)
  430. {
  431. if (clockid == CLOCK_REALTIME_ALARM)
  432. return ALARM_REALTIME;
  433. if (clockid == CLOCK_BOOTTIME_ALARM)
  434. return ALARM_BOOTTIME;
  435. return -1;
  436. }
  437. /**
  438. * alarm_handle_timer - Callback for posix timers
  439. * @alarm: alarm that fired
  440. *
  441. * Posix timer callback for expired alarm timers.
  442. */
  443. static enum alarmtimer_restart alarm_handle_timer(struct alarm *alarm,
  444. ktime_t now)
  445. {
  446. unsigned long flags;
  447. struct k_itimer *ptr = container_of(alarm, struct k_itimer,
  448. it.alarm.alarmtimer);
  449. enum alarmtimer_restart result = ALARMTIMER_NORESTART;
  450. spin_lock_irqsave(&ptr->it_lock, flags);
  451. if ((ptr->it_sigev_notify & ~SIGEV_THREAD_ID) != SIGEV_NONE) {
  452. if (posix_timer_event(ptr, 0))
  453. ptr->it_overrun++;
  454. }
  455. /* Re-add periodic timers */
  456. if (ptr->it_interval) {
  457. ptr->it_overrun += alarm_forward(alarm, now, ptr->it_interval);
  458. result = ALARMTIMER_RESTART;
  459. }
  460. spin_unlock_irqrestore(&ptr->it_lock, flags);
  461. return result;
  462. }
  463. /**
  464. * alarm_timer_rearm - Posix timer callback for rearming timer
  465. * @timr: Pointer to the posixtimer data struct
  466. */
  467. static void alarm_timer_rearm(struct k_itimer *timr)
  468. {
  469. struct alarm *alarm = &timr->it.alarm.alarmtimer;
  470. timr->it_overrun += alarm_forward_now(alarm, timr->it_interval);
  471. alarm_start(alarm, alarm->node.expires);
  472. }
  473. /**
  474. * alarm_timer_forward - Posix timer callback for forwarding timer
  475. * @timr: Pointer to the posixtimer data struct
  476. * @now: Current time to forward the timer against
  477. */
  478. static int alarm_timer_forward(struct k_itimer *timr, ktime_t now)
  479. {
  480. struct alarm *alarm = &timr->it.alarm.alarmtimer;
  481. return (int) alarm_forward(alarm, timr->it_interval, now);
  482. }
  483. /**
  484. * alarm_timer_remaining - Posix timer callback to retrieve remaining time
  485. * @timr: Pointer to the posixtimer data struct
  486. * @now: Current time to calculate against
  487. */
  488. static ktime_t alarm_timer_remaining(struct k_itimer *timr, ktime_t now)
  489. {
  490. struct alarm *alarm = &timr->it.alarm.alarmtimer;
  491. return ktime_sub(now, alarm->node.expires);
  492. }
  493. /**
  494. * alarm_clock_getres - posix getres interface
  495. * @which_clock: clockid
  496. * @tp: timespec to fill
  497. *
  498. * Returns the granularity of underlying alarm base clock
  499. */
  500. static int alarm_clock_getres(const clockid_t which_clock, struct timespec64 *tp)
  501. {
  502. if (!alarmtimer_get_rtcdev())
  503. return -EINVAL;
  504. tp->tv_sec = 0;
  505. tp->tv_nsec = hrtimer_resolution;
  506. return 0;
  507. }
  508. /**
  509. * alarm_clock_get - posix clock_get interface
  510. * @which_clock: clockid
  511. * @tp: timespec to fill.
  512. *
  513. * Provides the underlying alarm base time.
  514. */
  515. static int alarm_clock_get(clockid_t which_clock, struct timespec64 *tp)
  516. {
  517. struct alarm_base *base = &alarm_bases[clock2alarm(which_clock)];
  518. if (!alarmtimer_get_rtcdev())
  519. return -EINVAL;
  520. *tp = ktime_to_timespec64(base->gettime());
  521. return 0;
  522. }
  523. /**
  524. * alarm_timer_create - posix timer_create interface
  525. * @new_timer: k_itimer pointer to manage
  526. *
  527. * Initializes the k_itimer structure.
  528. */
  529. static int alarm_timer_create(struct k_itimer *new_timer)
  530. {
  531. enum alarmtimer_type type;
  532. if (!alarmtimer_get_rtcdev())
  533. return -ENOTSUPP;
  534. if (!capable(CAP_WAKE_ALARM))
  535. return -EPERM;
  536. type = clock2alarm(new_timer->it_clock);
  537. alarm_init(&new_timer->it.alarm.alarmtimer, type, alarm_handle_timer);
  538. return 0;
  539. }
  540. /**
  541. * alarm_timer_get - posix timer_get interface
  542. * @timr: k_itimer pointer
  543. * @cur_setting: itimerspec data to fill
  544. *
  545. * Copies out the current itimerspec data
  546. */
  547. static void alarm_timer_get(struct k_itimer *timr,
  548. struct itimerspec64 *cur_setting)
  549. {
  550. ktime_t relative_expiry_time =
  551. alarm_expires_remaining(&(timr->it.alarm.alarmtimer));
  552. if (ktime_to_ns(relative_expiry_time) > 0) {
  553. cur_setting->it_value = ktime_to_timespec64(relative_expiry_time);
  554. } else {
  555. cur_setting->it_value.tv_sec = 0;
  556. cur_setting->it_value.tv_nsec = 0;
  557. }
  558. cur_setting->it_interval = ktime_to_timespec64(timr->it_interval);
  559. }
  560. /**
  561. * alarm_timer_del - posix timer_del interface
  562. * @timr: k_itimer pointer to be deleted
  563. *
  564. * Cancels any programmed alarms for the given timer.
  565. */
  566. static int alarm_timer_del(struct k_itimer *timr)
  567. {
  568. if (!rtcdev)
  569. return -ENOTSUPP;
  570. if (alarm_try_to_cancel(&timr->it.alarm.alarmtimer) < 0)
  571. return TIMER_RETRY;
  572. return 0;
  573. }
  574. /**
  575. * alarm_timer_set - posix timer_set interface
  576. * @timr: k_itimer pointer to be deleted
  577. * @flags: timer flags
  578. * @new_setting: itimerspec to be used
  579. * @old_setting: itimerspec being replaced
  580. *
  581. * Sets the timer to new_setting, and starts the timer.
  582. */
  583. static int alarm_timer_set(struct k_itimer *timr, int flags,
  584. struct itimerspec64 *new_setting,
  585. struct itimerspec64 *old_setting)
  586. {
  587. ktime_t exp;
  588. if (!rtcdev)
  589. return -ENOTSUPP;
  590. if (flags & ~TIMER_ABSTIME)
  591. return -EINVAL;
  592. if (old_setting)
  593. alarm_timer_get(timr, old_setting);
  594. /* If the timer was already set, cancel it */
  595. if (alarm_try_to_cancel(&timr->it.alarm.alarmtimer) < 0)
  596. return TIMER_RETRY;
  597. /* start the timer */
  598. timr->it_interval = timespec64_to_ktime(new_setting->it_interval);
  599. /*
  600. * Rate limit to the tick as a hot fix to prevent DOS. Will be
  601. * mopped up later.
  602. */
  603. if (timr->it_interval < TICK_NSEC)
  604. timr->it_interval = TICK_NSEC;
  605. exp = timespec64_to_ktime(new_setting->it_value);
  606. /* Convert (if necessary) to absolute time */
  607. if (flags != TIMER_ABSTIME) {
  608. ktime_t now;
  609. now = alarm_bases[timr->it.alarm.alarmtimer.type].gettime();
  610. exp = ktime_add_safe(now, exp);
  611. }
  612. alarm_start(&timr->it.alarm.alarmtimer, exp);
  613. return 0;
  614. }
  615. /**
  616. * alarmtimer_nsleep_wakeup - Wakeup function for alarm_timer_nsleep
  617. * @alarm: ptr to alarm that fired
  618. *
  619. * Wakes up the task that set the alarmtimer
  620. */
  621. static enum alarmtimer_restart alarmtimer_nsleep_wakeup(struct alarm *alarm,
  622. ktime_t now)
  623. {
  624. struct task_struct *task = (struct task_struct *)alarm->data;
  625. alarm->data = NULL;
  626. if (task)
  627. wake_up_process(task);
  628. return ALARMTIMER_NORESTART;
  629. }
  630. /**
  631. * alarmtimer_do_nsleep - Internal alarmtimer nsleep implementation
  632. * @alarm: ptr to alarmtimer
  633. * @absexp: absolute expiration time
  634. *
  635. * Sets the alarm timer and sleeps until it is fired or interrupted.
  636. */
  637. static int alarmtimer_do_nsleep(struct alarm *alarm, ktime_t absexp)
  638. {
  639. alarm->data = (void *)current;
  640. do {
  641. set_current_state(TASK_INTERRUPTIBLE);
  642. alarm_start(alarm, absexp);
  643. if (likely(alarm->data))
  644. schedule();
  645. alarm_cancel(alarm);
  646. } while (alarm->data && !signal_pending(current));
  647. __set_current_state(TASK_RUNNING);
  648. return (alarm->data == NULL);
  649. }
  650. /**
  651. * update_rmtp - Update remaining timespec value
  652. * @exp: expiration time
  653. * @type: timer type
  654. * @rmtp: user pointer to remaining timepsec value
  655. *
  656. * Helper function that fills in rmtp value with time between
  657. * now and the exp value
  658. */
  659. static int update_rmtp(ktime_t exp, enum alarmtimer_type type,
  660. struct timespec __user *rmtp)
  661. {
  662. struct timespec rmt;
  663. ktime_t rem;
  664. rem = ktime_sub(exp, alarm_bases[type].gettime());
  665. if (rem <= 0)
  666. return 0;
  667. rmt = ktime_to_timespec(rem);
  668. if (copy_to_user(rmtp, &rmt, sizeof(*rmtp)))
  669. return -EFAULT;
  670. return 1;
  671. }
  672. /**
  673. * alarm_timer_nsleep_restart - restartblock alarmtimer nsleep
  674. * @restart: ptr to restart block
  675. *
  676. * Handles restarted clock_nanosleep calls
  677. */
  678. static long __sched alarm_timer_nsleep_restart(struct restart_block *restart)
  679. {
  680. enum alarmtimer_type type = restart->nanosleep.clockid;
  681. ktime_t exp;
  682. struct timespec __user *rmtp;
  683. struct alarm alarm;
  684. int ret = 0;
  685. exp = restart->nanosleep.expires;
  686. alarm_init(&alarm, type, alarmtimer_nsleep_wakeup);
  687. if (alarmtimer_do_nsleep(&alarm, exp))
  688. goto out;
  689. if (freezing(current))
  690. alarmtimer_freezerset(exp, type);
  691. rmtp = restart->nanosleep.rmtp;
  692. if (rmtp) {
  693. ret = update_rmtp(exp, type, rmtp);
  694. if (ret <= 0)
  695. goto out;
  696. }
  697. /* The other values in restart are already filled in */
  698. ret = -ERESTART_RESTARTBLOCK;
  699. out:
  700. return ret;
  701. }
  702. /**
  703. * alarm_timer_nsleep - alarmtimer nanosleep
  704. * @which_clock: clockid
  705. * @flags: determins abstime or relative
  706. * @tsreq: requested sleep time (abs or rel)
  707. * @rmtp: remaining sleep time saved
  708. *
  709. * Handles clock_nanosleep calls against _ALARM clockids
  710. */
  711. static int alarm_timer_nsleep(const clockid_t which_clock, int flags,
  712. struct timespec64 *tsreq,
  713. struct timespec __user *rmtp)
  714. {
  715. enum alarmtimer_type type = clock2alarm(which_clock);
  716. struct restart_block *restart;
  717. struct alarm alarm;
  718. ktime_t exp;
  719. int ret = 0;
  720. if (!alarmtimer_get_rtcdev())
  721. return -ENOTSUPP;
  722. if (flags & ~TIMER_ABSTIME)
  723. return -EINVAL;
  724. if (!capable(CAP_WAKE_ALARM))
  725. return -EPERM;
  726. alarm_init(&alarm, type, alarmtimer_nsleep_wakeup);
  727. exp = timespec64_to_ktime(*tsreq);
  728. /* Convert (if necessary) to absolute time */
  729. if (flags != TIMER_ABSTIME) {
  730. ktime_t now = alarm_bases[type].gettime();
  731. exp = ktime_add(now, exp);
  732. }
  733. if (alarmtimer_do_nsleep(&alarm, exp))
  734. goto out;
  735. if (freezing(current))
  736. alarmtimer_freezerset(exp, type);
  737. /* abs timers don't set remaining time or restart */
  738. if (flags == TIMER_ABSTIME) {
  739. ret = -ERESTARTNOHAND;
  740. goto out;
  741. }
  742. if (rmtp) {
  743. ret = update_rmtp(exp, type, rmtp);
  744. if (ret <= 0)
  745. goto out;
  746. }
  747. restart = &current->restart_block;
  748. restart->fn = alarm_timer_nsleep_restart;
  749. restart->nanosleep.clockid = type;
  750. restart->nanosleep.expires = exp;
  751. restart->nanosleep.rmtp = rmtp;
  752. ret = -ERESTART_RESTARTBLOCK;
  753. out:
  754. return ret;
  755. }
  756. const struct k_clock alarm_clock = {
  757. .clock_getres = alarm_clock_getres,
  758. .clock_get = alarm_clock_get,
  759. .timer_create = alarm_timer_create,
  760. .timer_set = alarm_timer_set,
  761. .timer_del = alarm_timer_del,
  762. .timer_get = alarm_timer_get,
  763. .timer_rearm = alarm_timer_rearm,
  764. .timer_forward = alarm_timer_forward,
  765. .timer_remaining = alarm_timer_remaining,
  766. .nsleep = alarm_timer_nsleep,
  767. };
  768. #endif /* CONFIG_POSIX_TIMERS */
  769. /* Suspend hook structures */
  770. static const struct dev_pm_ops alarmtimer_pm_ops = {
  771. .suspend = alarmtimer_suspend,
  772. .resume = alarmtimer_resume,
  773. };
  774. static struct platform_driver alarmtimer_driver = {
  775. .driver = {
  776. .name = "alarmtimer",
  777. .pm = &alarmtimer_pm_ops,
  778. }
  779. };
  780. /**
  781. * alarmtimer_init - Initialize alarm timer code
  782. *
  783. * This function initializes the alarm bases and registers
  784. * the posix clock ids.
  785. */
  786. static int __init alarmtimer_init(void)
  787. {
  788. struct platform_device *pdev;
  789. int error = 0;
  790. int i;
  791. alarmtimer_rtc_timer_init();
  792. /* Initialize alarm bases */
  793. alarm_bases[ALARM_REALTIME].base_clockid = CLOCK_REALTIME;
  794. alarm_bases[ALARM_REALTIME].gettime = &ktime_get_real;
  795. alarm_bases[ALARM_BOOTTIME].base_clockid = CLOCK_BOOTTIME;
  796. alarm_bases[ALARM_BOOTTIME].gettime = &ktime_get_boottime;
  797. for (i = 0; i < ALARM_NUMTYPE; i++) {
  798. timerqueue_init_head(&alarm_bases[i].timerqueue);
  799. spin_lock_init(&alarm_bases[i].lock);
  800. }
  801. error = alarmtimer_rtc_interface_setup();
  802. if (error)
  803. return error;
  804. error = platform_driver_register(&alarmtimer_driver);
  805. if (error)
  806. goto out_if;
  807. pdev = platform_device_register_simple("alarmtimer", -1, NULL, 0);
  808. if (IS_ERR(pdev)) {
  809. error = PTR_ERR(pdev);
  810. goto out_drv;
  811. }
  812. ws = wakeup_source_register("alarmtimer");
  813. return 0;
  814. out_drv:
  815. platform_driver_unregister(&alarmtimer_driver);
  816. out_if:
  817. alarmtimer_rtc_interface_remove();
  818. return error;
  819. }
  820. device_initcall(alarmtimer_init);