/* $OpenBSD: apmd.c,v 1.54 2009/08/04 18:26:48 jmc Exp $ */ /* * Copyright (c) 1995, 1996 John T. Kohl * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * 3. The name of the author may not be used to endorse or promote products * derived from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR `AS IS'' AND ANY EXPRESS OR * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * POSSIBILITY OF SUCH DAMAGE. * */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "pathnames.h" #include "apm-proto.h" #define TRUE 1 #define FALSE 0 const char apmdev[] = _PATH_APM_CTLDEV; const char sockfile[] = _PATH_APM_SOCKET; int debug = 0; int doperf = PERF_NONE; #define PERFINC 50 #define PERFDEC 20 #define PERFMIN 0 #define PERFMAX 100 #define PERFINCTHRES 10 #define PERFDECTHRES 30 extern char *__progname; void usage(void); int power_status(int fd, int force, struct apm_power_info *pinfo); int bind_socket(const char *sn); enum apm_state handle_client(int sock_fd, int ctl_fd); int get_avg_idle_mp(int ncpu); int get_avg_idle_up(void); void perf_status(struct apm_power_info *pinfo, int ncpu); void suspend(int ctl_fd); void stand_by(int ctl_fd); void suspend_req(int ctl_fd); void stand_by_req(int ctl_fd); void setperf(int new_perf); void sigexit(int signo); void do_etc_file(const char *file); void sockunlink(void); /* ARGSUSED */ void sigexit(int signo) { sockunlink(); _exit(1); } void usage(void) { fprintf(stderr, "usage: %s [-AaCdHLs] [-f devname] [-S sockname] [-t seconds]\n", __progname); exit(1); } void error(const char *fmt, const char *arg) { char buf[128]; if (debug) err(1, fmt, arg); else { strlcpy(buf, fmt, sizeof(buf)); strlcat(buf, ": %m", sizeof(buf)); syslog(LOG_ERR, buf, arg); exit(1); } } /* * tell the driver if it should display messages or not. */ void set_driver_messages(int fd, int mode) { if (ioctl(fd, APM_IOC_PRN_CTL, &mode) == -1) syslog(LOG_DEBUG, "can't disable driver messages, error: %m"); } int power_status(int fd, int force, struct apm_power_info *pinfo) { struct apm_power_info bstate; static struct apm_power_info last; int acon = 0; if (fd == -1) { if (pinfo) { bstate.battery_state = 255; bstate.ac_state = 255; bstate.battery_life = 0; bstate.minutes_left = -1; *pinfo = bstate; } return 0; } if (ioctl(fd, APM_IOC_GETPOWER, &bstate) == 0) { /* various conditions under which we report status: something changed * enough since last report, or asked to force a print */ if (bstate.ac_state == APM_AC_ON) acon = 1; if (force || bstate.ac_state != last.ac_state || bstate.battery_state != last.battery_state || (bstate.minutes_left && bstate.minutes_left < 15) || abs(bstate.battery_life - last.battery_life) > 20) { #ifdef __powerpc__ /* * When the battery is charging, the estimated life * time is in fact the estimated remaining charge time * on Apple machines, so lie in the stats. * We still want an useful message if the battery or * ac status changes, however. */ if (bstate.minutes_left != 0 && bstate.battery_state != APM_BATT_CHARGING) #else if ((int)bstate.minutes_left > 0) #endif syslog(LOG_NOTICE, "battery status: %s. " "external power status: %s. " "estimated battery life %d%% (%u minutes)", battstate(bstate.battery_state), ac_state(bstate.ac_state), bstate.battery_life, bstate.minutes_left); else syslog(LOG_NOTICE, "battery status: %s. " "external power status: %s. " "estimated battery life %d%%", battstate(bstate.battery_state), ac_state(bstate.ac_state), bstate.battery_life); last = bstate; } if (pinfo) *pinfo = bstate; } else syslog(LOG_ERR, "cannot fetch power status: %m"); return acon; } /* multi- and uni-processor case */ int get_avg_idle_mp(int ncpu) { static int64_t **cp_time_old; static int64_t **cp_time; static int *avg_idle; int64_t change, sum, idle; int i, cpu, min_avg_idle; size_t cp_time_sz = CPUSTATES * sizeof(int64_t); if (!cp_time_old) if ((cp_time_old = calloc(sizeof(int64_t *), ncpu)) == NULL) return -1; if (!cp_time) if ((cp_time = calloc(sizeof(int64_t *), ncpu)) == NULL) return -1; if (!avg_idle) if ((avg_idle = calloc(sizeof(int), ncpu)) == NULL) return -1; min_avg_idle = 0; for (cpu = 0; cpu < ncpu; cpu++) { int cp_time_mib[] = {CTL_KERN, KERN_CPTIME2, cpu}; if (!cp_time_old[cpu]) if ((cp_time_old[cpu] = calloc(sizeof(int64_t), CPUSTATES)) == NULL) return -1; if (!cp_time[cpu]) if ((cp_time[cpu] = calloc(sizeof(int64_t), CPUSTATES)) == NULL) return -1; if (sysctl(cp_time_mib, 3, cp_time[cpu], &cp_time_sz, NULL, 0) < 0) syslog(LOG_INFO, "cannot read kern.cp_time2"); sum = 0; for (i = 0; i < CPUSTATES; i++) { if ((change = cp_time[cpu][i] - cp_time_old[cpu][i]) < 0) { /* counter wrapped */ change = ((uint64_t)cp_time[cpu][i] - (uint64_t)cp_time_old[cpu][i]); } sum += change; if (i == CP_IDLE) idle = change; } if (sum == 0) sum = 1; /* smooth data */ avg_idle[cpu] = (avg_idle[cpu] + (100 * idle) / sum) / 2; if (cpu == 0) min_avg_idle = avg_idle[cpu]; if (avg_idle[cpu] < min_avg_idle) min_avg_idle = avg_idle[cpu]; memcpy(cp_time_old[cpu], cp_time[cpu], cp_time_sz); } return min_avg_idle; } int get_avg_idle_up(void) { static long cp_time_old[CPUSTATES]; static int avg_idle; long change, cp_time[CPUSTATES]; int cp_time_mib[] = {CTL_KERN, KERN_CPTIME}; size_t cp_time_sz = sizeof(cp_time); int i, idle, sum = 0; if (sysctl(cp_time_mib, 2, &cp_time, &cp_time_sz, NULL, 0) < 0) syslog(LOG_INFO, "cannot read kern.cp_time"); for (i = 0; i < CPUSTATES; i++) { if ((change = cp_time[i] - cp_time_old[i]) < 0) { /* counter wrapped */ change = ((unsigned long)cp_time[i] - (unsigned long)cp_time_old[i]); } sum += change; if (i == CP_IDLE) idle = change; } if (sum == 0) sum = 1; /* smooth data */ avg_idle = (avg_idle + (100 * idle) / sum) / 2; memcpy(cp_time_old, cp_time, sizeof(cp_time_old)); return avg_idle; } void perf_status(struct apm_power_info *pinfo, int ncpu) { int avg_idle; int hw_perf_mib[] = {CTL_HW, HW_SETPERF}; int perf; int forcehi = 0; size_t perf_sz = sizeof(perf); if (ncpu > 1) { avg_idle = get_avg_idle_mp(ncpu); } else { avg_idle = get_avg_idle_up(); } if (avg_idle == -1) return; switch (doperf) { case PERF_AUTO: /* * force setperf towards the max if we are connected to AC * power and have a battery life greater than 15% */ if (pinfo->ac_state == APM_AC_ON && pinfo->battery_life > 15) forcehi = 1; break; case PERF_COOL: forcehi = 0; break; } if (sysctl(hw_perf_mib, 2, &perf, &perf_sz, NULL, 0) < 0) syslog(LOG_INFO, "cannot read hw.setperf"); if (forcehi || (avg_idle < PERFINCTHRES && perf < PERFMAX)) { perf += PERFINC; if (perf > PERFMAX) perf = PERFMAX; setperf(perf); } else if (avg_idle > PERFDECTHRES && perf > PERFMIN) { perf -= PERFDEC; if (perf < PERFMIN) perf = PERFMIN; setperf(perf); } } char socketname[MAXPATHLEN]; void sockunlink(void) { if (socketname[0]) remove(socketname); } int bind_socket(const char *sockname) { struct sockaddr_un s_un; mode_t old_umask; int sock; sock = socket(AF_UNIX, SOCK_STREAM, 0); if (sock == -1) error("cannot create local socket", NULL); s_un.sun_family = AF_UNIX; strncpy(s_un.sun_path, sockname, sizeof(s_un.sun_path)); s_un.sun_len = SUN_LEN(&s_un); /* remove it if present, we're moving in */ (void) remove(sockname); old_umask = umask(077); if (bind(sock, (struct sockaddr *)&s_un, s_un.sun_len) == -1) error("cannot connect to APM socket", NULL); umask(old_umask); if (chmod(sockname, 0660) == -1 || chown(sockname, 0, 0) == -1) error("cannot set socket mode/owner/group to 660/0/0", NULL); listen(sock, 1); strlcpy(socketname, sockname, sizeof socketname); atexit(sockunlink); return sock; } enum apm_state handle_client(int sock_fd, int ctl_fd) { /* accept a handle from the client, process it, then clean up */ int cli_fd; struct sockaddr_un from; socklen_t fromlen; struct apm_command cmd; struct apm_reply reply; int cpuspeed_mib[] = {CTL_HW, HW_CPUSPEED}; int cpuspeed = 0; size_t cpuspeed_sz = sizeof(cpuspeed); fromlen = sizeof(from); cli_fd = accept(sock_fd, (struct sockaddr *)&from, &fromlen); if (cli_fd == -1) { syslog(LOG_INFO, "client accept failure: %m"); return NORMAL; } if (recv(cli_fd, &cmd, sizeof(cmd), 0) != sizeof(cmd)) { (void) close(cli_fd); syslog(LOG_INFO, "client size botch"); return NORMAL; } if (cmd.vno != APMD_VNO) { close(cli_fd); /* terminate client */ /* no error message, just drop it. */ return NORMAL; } power_status(ctl_fd, 0, &reply.batterystate); switch (cmd.action) { case SUSPEND: reply.newstate = SUSPENDING; break; case STANDBY: reply.newstate = STANDING_BY; break; case SETPERF_LOW: doperf = PERF_MANUAL; reply.newstate = NORMAL; syslog(LOG_NOTICE, "setting hw.setperf to %d", PERFMIN); setperf(PERFMIN); break; case SETPERF_HIGH: doperf = PERF_MANUAL; reply.newstate = NORMAL; syslog(LOG_NOTICE, "setting hw.setperf to %d", PERFMAX); setperf(PERFMAX); break; case SETPERF_AUTO: doperf = PERF_AUTO; reply.newstate = NORMAL; syslog(LOG_NOTICE, "setting hw.setperf automatically"); break; case SETPERF_COOL: doperf = PERF_COOL; reply.newstate = NORMAL; syslog(LOG_NOTICE, "setting hw.setperf for cool running"); break; default: reply.newstate = NORMAL; break; } if (sysctl(cpuspeed_mib, 2, &cpuspeed, &cpuspeed_sz, NULL, 0) < 0) syslog(LOG_INFO, "cannot read hw.cpuspeed"); reply.cpuspeed = cpuspeed; reply.perfmode = doperf; reply.vno = APMD_VNO; if (send(cli_fd, &reply, sizeof(reply), 0) != sizeof(reply)) syslog(LOG_INFO, "client reply botch"); close(cli_fd); return reply.newstate; } void suspend(int ctl_fd) { do_etc_file(_PATH_APM_ETC_SUSPEND); sync(); sleep(1); ioctl(ctl_fd, APM_IOC_SUSPEND, 0); } void stand_by(int ctl_fd) { do_etc_file(_PATH_APM_ETC_STANDBY); sync(); sleep(1); ioctl(ctl_fd, APM_IOC_STANDBY, 0); } void suspend_req(int ctl_fd) { /* let the user get their finger off the enter key before we suspend */ sleep(1); ioctl(ctl_fd, APM_IOC_SUSPEND_REQ, 0); /* give X a chance to do things */ sleep(2); } void stand_by_req(int ctl_fd) { /* let the user get their finger off the enter key before we suspend */ sleep(1); ioctl(ctl_fd, APM_IOC_STANDBY_REQ, 0); /* give X a chance to do things */ sleep(2); } #define TIMO (10*60) /* 10 minutes */ int main(int argc, char *argv[]) { const char *fname = apmdev; int ctl_fd, sock_fd, ch, suspends, standbys, resumes; int statonly = 0; int powerstatus = 0, powerbak = 0, powerchange = 0; int noacsleep = 0; struct timespec ts = {TIMO, 0}, sts = {0, 0}; struct apm_power_info pinfo; time_t apmtimeout = 0; const char *sockname = sockfile; int kq, nchanges; struct kevent ev[2]; int ncpu_mib[2] = { CTL_HW, HW_NCPU }; int ncpu; size_t ncpu_sz = sizeof(ncpu); while ((ch = getopt(argc, argv, "aACdHLsf:t:S:")) != -1) switch(ch) { case 'a': noacsleep = 1; break; case 'd': debug = 1; break; case 'f': fname = optarg; break; case 'S': sockname = optarg; break; case 't': ts.tv_sec = strtoul(optarg, NULL, 0); if (ts.tv_sec == 0) usage(); break; case 's': /* status only */ statonly = 1; break; case 'A': if (doperf != PERF_NONE) usage(); doperf = PERF_AUTO; break; case 'C': if (doperf != PERF_NONE) usage(); doperf = PERF_COOL; break; case 'L': if (doperf != PERF_NONE) usage(); doperf = PERF_MANUAL; setperf(PERFMIN); break; case 'H': if (doperf != PERF_NONE) usage(); doperf = PERF_MANUAL; setperf(PERFMAX); break; case '?': default: usage(); } argc -= optind; argv += optind; if (argc != 0) usage(); if (doperf == PERF_NONE) doperf = PERF_MANUAL; if (debug) openlog(__progname, LOG_CONS, LOG_LOCAL1); else { daemon(0, 0); openlog(__progname, LOG_CONS, LOG_DAEMON); setlogmask(LOG_UPTO(LOG_NOTICE)); } (void) signal(SIGTERM, sigexit); (void) signal(SIGHUP, sigexit); (void) signal(SIGINT, sigexit); if ((ctl_fd = open(fname, O_RDWR)) == -1) { if (errno != ENXIO && errno != ENOENT) error("cannot open device file `%s'", fname); } else if (fcntl(ctl_fd, F_SETFD, 1) == -1) error("cannot set close-on-exec for `%s'", fname); sock_fd = bind_socket(sockname); if (fcntl(sock_fd, F_SETFD, 1) == -1) error("cannot set close-on-exec for the socket", NULL); power_status(ctl_fd, 1, &pinfo); if (statonly) exit(0); set_driver_messages(ctl_fd, APM_PRINT_OFF); kq = kqueue(); if (kq <= 0) error("kqueue", NULL); EV_SET(&ev[0], sock_fd, EVFILT_READ, EV_ADD | EV_ENABLE | EV_CLEAR, 0, 0, NULL); if (ctl_fd == -1) nchanges = 1; else { EV_SET(&ev[1], ctl_fd, EVFILT_READ, EV_ADD | EV_ENABLE | EV_CLEAR, 0, 0, NULL); nchanges = 2; } if (kevent(kq, ev, nchanges, NULL, 0, &sts) < 0) error("kevent", NULL); if (sysctl(ncpu_mib, 2, &ncpu, &ncpu_sz, NULL, 0) < 0) error("cannot read hw.ncpu", NULL); if (doperf == PERF_AUTO || doperf == PERF_COOL) { setperf(0); setperf(100); } for (;;) { int rv; sts = ts; if (doperf == PERF_AUTO || doperf == PERF_COOL) { sts.tv_sec = 1; perf_status(&pinfo, ncpu); } apmtimeout += sts.tv_sec; if ((rv = kevent(kq, NULL, 0, ev, 1, &sts)) < 0) break; if (apmtimeout >= ts.tv_sec) { apmtimeout = 0; /* wakeup for timeout: take status */ powerbak = power_status(ctl_fd, 0, &pinfo); if (powerstatus != powerbak) { powerstatus = powerbak; powerchange = 1; } } if (!rv) continue; if (ev->ident == ctl_fd) { suspends = standbys = resumes = 0; syslog(LOG_DEBUG, "apmevent %04x index %d", APM_EVENT_TYPE(ev->data), APM_EVENT_INDEX(ev->data)); switch (APM_EVENT_TYPE(ev->data)) { case APM_SUSPEND_REQ: case APM_USER_SUSPEND_REQ: case APM_CRIT_SUSPEND_REQ: case APM_BATTERY_LOW: suspends++; break; case APM_USER_STANDBY_REQ: case APM_STANDBY_REQ: standbys++; break; #if 0 case APM_CANCEL: suspends = standbys = 0; break; #endif case APM_NORMAL_RESUME: case APM_CRIT_RESUME: case APM_SYS_STANDBY_RESUME: powerbak = power_status(ctl_fd, 0, &pinfo); if (powerstatus != powerbak) { powerstatus = powerbak; powerchange = 1; } resumes++; break; case APM_POWER_CHANGE: powerbak = power_status(ctl_fd, 0, &pinfo); if (powerstatus != powerbak) { powerstatus = powerbak; powerchange = 1; } break; default: ; } if ((standbys || suspends) && noacsleep && power_status(ctl_fd, 0, &pinfo)) syslog(LOG_DEBUG, "no! sleep! till brooklyn!"); else if (suspends) suspend(ctl_fd); else if (standbys) stand_by(ctl_fd); else if (resumes) { do_etc_file(_PATH_APM_ETC_RESUME); syslog(LOG_NOTICE, "system resumed from APM sleep"); } if (powerchange) { if (powerstatus) do_etc_file(_PATH_APM_ETC_POWERUP); else do_etc_file(_PATH_APM_ETC_POWERDOWN); powerchange = 0; } } else if (ev->ident == sock_fd) switch (handle_client(sock_fd, ctl_fd)) { case NORMAL: break; case SUSPENDING: suspend_req(ctl_fd); break; case STANDING_BY: stand_by_req(ctl_fd); break; } } error("kevent loop", NULL); return 1; } void setperf(int new_perf) { int hw_perf_mib[] = {CTL_HW, HW_SETPERF}; int perf; size_t perf_sz = sizeof(perf); if (sysctl(hw_perf_mib, 2, &perf, &perf_sz, &new_perf, perf_sz) < 0) syslog(LOG_INFO, "cannot set hw.setperf"); } void do_etc_file(const char *file) { pid_t pid; int status; const char *prog; /* If file doesn't exist, do nothing. */ if (access(file, X_OK|R_OK)) { syslog(LOG_DEBUG, "do_etc_file(): cannot access file %s", file); return; } prog = strrchr(file, '/'); if (prog) prog++; else prog = file; pid = fork(); switch (pid) { case -1: syslog(LOG_ERR, "failed to fork(): %m"); return; case 0: /* We are the child. */ execl(file, prog, (char *)NULL); _exit(1); /* NOTREACHED */ default: /* We are the parent. */ wait4(pid, &status, 0, 0); if (WIFEXITED(status)) syslog(LOG_DEBUG, "%s exited with status %d", file, WEXITSTATUS(status)); else syslog(LOG_ERR, "%s exited abnormally.", file); } }