diff options
Diffstat (limited to 'toys/other')
-rw-r--r-- | toys/other/rtcwake.c | 118 |
1 files changed, 118 insertions, 0 deletions
diff --git a/toys/other/rtcwake.c b/toys/other/rtcwake.c new file mode 100644 index 00000000..161bd881 --- /dev/null +++ b/toys/other/rtcwake.c @@ -0,0 +1,118 @@ +/* rtcwake.c - enter sleep state until given time. + * + * Copyright 2020 The Android Open Source Project + +USE_RTCWAKE(NEWTOY(rtcwake, "(list-modes);(auto)a(device)d:(local)l(mode)m:(seconds)s#(time)t#(utc)u(verbose)v[!alu]", TOYFLAG_USR|TOYFLAG_BIN)) + +config RTCWAKE + bool "rtcwake" + default y + help + usage: rtcwake [-aluv] [-d FILE] [-m MODE] [-s SECS] [-t UNIX] + + Enter the given sleep state until the given time. + + -a RTC uses time specified in /etc/adjtime + -d FILE Device to use (default /dev/rtc) + -l RTC uses local time + -m Mode (--list-modes to see those supported by your kernel): + standby S1: default mem S3: suspend to RAM + disk S4: suspend to disk off S5: power off + disable Cancel current alarm freeze stop processes/processors + no just set wakeup time on just poll RTC for alarm + show just show current alarm + -s SECS Wake SECS seconds from now + -t UNIX Wake UNIX seconds from epoch + -u RTC uses UTC + -v Verbose +*/ + +#define FOR_rtcwake +#include "toys.h" +#include <linux/rtc.h> + +GLOBALS( + long t, s; + char *m, *d; +) + +void rtcwake_main(void) +{ + struct rtc_wkalrm *alarm = (void *)(toybuf+2048); + struct tm rtc_tm; + time_t now, rtc_now, then; + int fd, utc; + + if (FLAG(list_modes)) { + printf("off no on disable show %s", + xreadfile("/sys/power/state", toybuf, 2048)); + return; + } + + // util-linux defaults to "suspend", even though I don't have anything that + // supports that (testing everything from a ~2010 laptop to a 2019 desktop). + if (!TT.m) TT.m = "suspend"; + + if (FLAG(u)) utc = 1; + else if (FLAG(l)) utc = 0; + else utc = !!strstr(xreadfile("/etc/adjtime", toybuf, 2048), "UTC"); + if (FLAG(v)) xprintf("RTC time: %s\n", utc ? "UTC" : "local"); + + if (!TT.d) TT.d = "/dev/rtc0"; + if (FLAG(v)) xprintf("Device: %s\n", TT.d); + fd = xopen(TT.d, O_RDWR); + + now = time(0); + xioctl(fd, RTC_RD_TIME, &rtc_tm); + rtc_now = xmktime(&rtc_tm, utc); + if (FLAG(v)) { + xprintf("System time:\t%lld / %s", (long long)now, ctime(&now)); + xprintf("RTC time:\t%lld / %s", (long long)rtc_now, ctime(&rtc_now)); + } + + if (!strcmp(TT.m, "show")) { // Don't suspend, just show current alarm. + xioctl(fd, RTC_WKALM_RD, alarm); + if (!alarm->enabled) xputs("alarm: off"); + else { + if ((then = mktime((void *)&alarm->time)) < 0) perror_exit("mktime"); + xprintf("alarm: on %s", ctime(&then)); + } + return; + } else if (!strcmp(TT.m, "disable")) { // Cancel current alarm. + xioctl(fd, RTC_WKALM_RD, alarm); + alarm->enabled = 0; + xioctl(fd, RTC_WKALM_SET, alarm); + return; + } + + if (FLAG(s)) then = rtc_now + TT.s + 1; // strace shows util-linux adds 1. + else if (FLAG(t)) { + then = TT.t + (rtc_now - now); + if (then<=rtc_now) error_exit("rtc %lld >= %ld", (long long)rtc_now, TT.t); + } else help_exit("-m %s needs -s or -t", TT.m); + if (FLAG(v)) xprintf("Wake time:\t%lld / %s", (long long)then, ctime(&then)); + + if (!(utc ? gmtime_r : localtime_r)(&then, (void *)&alarm->time)) + error_exit("%s failed", utc ? "gmtime_r" : "localtime_r"); + + alarm->enabled = 1; + xioctl(fd, RTC_WKALM_SET, alarm); + sync(); + + xprintf("wakeup using \"%s\" from %s at %s", TT.m, TT.d, ctime(&then)); + msleep(10); + + if (!strcmp(TT.m, "no")); // Don't suspend, just set wakeup time. + else if (!strcmp(TT.m, "on")) { // Don't suspend, poll RTC for alarm. + unsigned long data = 0; + + if (FLAG(v)) xputs("Reading RTC..."); + while (!(data & RTC_AF)) { + if (read(fd, &data, sizeof(data)) != sizeof(data)) perror_exit("read"); + if (FLAG(v)) xprintf("... %s: %lx\n", TT.d, data); + } + } else if (!strcmp(TT.m, "off")) xexec((char *[]){"poweroff", 0}); + // Everything else lands here for one final step. The write will fail with + // EINVAL if the mode is not supported. + else xwrite(xopen("/sys/power/state", O_WRONLY), TT.m, strlen(TT.m)); +} |