OpenBSD-4.6/usr.sbin/apmd/apmd.c
/* $OpenBSD: apmd.c,v 1.53 2009/02/26 17:21:56 oga 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 <sys/param.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <sys/wait.h>
#include <sys/event.h>
#include <sys/time.h>
#include <sys/dkstat.h>
#include <sys/sysctl.h>
#include <stdio.h>
#include <syslog.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <errno.h>
#include <err.h>
#include <machine/apmvar.h>
#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 [-AaCdeHLmps] [-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 enableonly = 0;
int pctonly = 0;
int powerstatus = 0, powerbak = 0, powerchange = 0;
int messages = 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, "aACdHLsepmf: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 'e':
enableonly = 1;
break;
case 'p':
pctonly = 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 'm':
messages = 1;
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);
if (enableonly) {
set_driver_messages(ctl_fd, APM_PRINT_ON);
exit(0);
}
if (pctonly) {
set_driver_messages(ctl_fd, APM_PRINT_PCT);
exit(0);
}
if (!messages)
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);
}
}