1/*	$NetBSD: apmd.c,v 1.31 2008/04/28 20:24:15 martin Exp $	*/
2
3/*-
4 * Copyright (c) 1996, 2000 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by John Kohl.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 *    notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 *    notice, this list of conditions and the following disclaimer in the
17 *    documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32#include <stdio.h>
33#include <errno.h>
34#include <syslog.h>
35#include <fcntl.h>
36#include <unistd.h>
37#include <util.h>
38#include <stdlib.h>
39#include <string.h>
40#include <signal.h>
41#include <sys/types.h>
42#include <pwd.h>
43#include <grp.h>
44#include <sys/stat.h>
45#include <sys/ioctl.h>
46#include <sys/time.h>
47#include <sys/socket.h>
48#include <sys/un.h>
49#include <sys/wait.h>
50#include <poll.h>
51#include <machine/apmvar.h>
52#include <err.h>
53#include "pathnames.h"
54#include "apm-proto.h"
55
56#define TRUE 1
57#define FALSE 0
58
59#define POWER_STATUS_ACON	0x1
60#define POWER_STATUS_LOWBATTNOW	0x2
61
62static const char apmdev[] = _PATH_APM_CTLDEV;
63static const char sockfile[] = _PATH_APM_SOCKET;
64
65static int debug = 0;
66static int verbose = 0;
67
68__dead static void usage (void);
69static int power_status (int fd, int force, struct apm_power_info *pinfo);
70static int bind_socket (const char *sn, mode_t mode, uid_t uid, gid_t gid);
71static enum apm_state handle_client(int sock_fd, int ctl_fd);
72static void suspend(int ctl_fd);
73static void stand_by(int ctl_fd);
74static void resume(int ctl_fd);
75__dead static void sigexit(int signo);
76static void make_noise(int howmany);
77static void do_etc_file(const char *file);
78static void do_ac_state(int state);
79
80static void
81sigexit(int signo)
82{
83    exit(1);
84}
85
86static void
87usage(void)
88{
89    fprintf(stderr,"usage: %s [-adlqsv] [-t seconds] [-S sockname]\n\t[-m sockmode] [-o sockowner:sockgroup] [-f devname]\n", getprogname());
90    exit(1);
91}
92
93
94static int
95power_status(int fd, int force, struct apm_power_info *pinfo)
96{
97    struct apm_power_info bstate;
98    static struct apm_power_info last;
99    int acon = 0;
100    int lowbattnow = 0;
101
102    memset(&bstate, 0, sizeof(bstate));
103    if (ioctl(fd, APM_IOC_GETPOWER, &bstate) == 0) {
104	/* various conditions under which we report status:  something changed
105	   enough since last report, or asked to force a print */
106	if (bstate.ac_state == APM_AC_ON)
107	    acon = 1;
108	if (bstate.battery_state != last.battery_state  &&
109	    bstate.battery_state == APM_BATT_LOW)
110		lowbattnow = 1;
111	if (force ||
112	    bstate.ac_state != last.ac_state ||
113	    bstate.battery_state != last.battery_state ||
114	    (bstate.minutes_left && bstate.minutes_left < 15) ||
115	    abs(bstate.battery_life - last.battery_life) > 20) {
116	    if (verbose) {
117		if (bstate.minutes_left)
118		    syslog(LOG_NOTICE,
119		           "battery status: %s. external power status: %s. "
120		           "estimated battery life %d%% (%d minutes)",
121		           battstate(bstate.battery_state),
122		           ac_state(bstate.ac_state), bstate.battery_life,
123		           bstate.minutes_left);
124		else
125		    syslog(LOG_NOTICE,
126		           "battery status: %s. external power status: %s. "
127		           "estimated battery life %d%%",
128		           battstate(bstate.battery_state),
129		           ac_state(bstate.ac_state), bstate.battery_life);
130	    }
131	    last = bstate;
132	}
133	if (pinfo)
134	    *pinfo = bstate;
135    } else
136	syslog(LOG_ERR, "cannot fetch power status: %m");
137    return ((acon?POWER_STATUS_ACON:0) |
138	(lowbattnow?POWER_STATUS_LOWBATTNOW:0));
139}
140
141static char *socketname;
142
143static void sockunlink(void);
144
145static void
146sockunlink(void)
147{
148    if (socketname)
149	(void) remove(socketname);
150}
151
152static int
153bind_socket(const char *sockname, mode_t mode, uid_t uid, gid_t gid)
154{
155    int sock;
156    struct sockaddr_un s_un;
157
158    sock = socket(AF_LOCAL, SOCK_STREAM, 0);
159    if (sock == -1)
160	err(1, "cannot create local socket");
161
162    s_un.sun_family = AF_LOCAL;
163    strncpy(s_un.sun_path, sockname, sizeof(s_un.sun_path));
164    s_un.sun_len = SUN_LEN(&s_un);
165    /* remove it if present, we're moving in */
166    (void) remove(sockname);
167    if (bind(sock, (struct sockaddr *)&s_un, s_un.sun_len) == -1)
168	err(1, "cannot create APM socket");
169    if (chmod(sockname, mode) == -1 || chown(sockname, uid, gid) == -1)
170	err(1, "cannot set socket mode/owner/group to %o/%d/%d",
171	    mode, uid, gid);
172    listen(sock, 1);
173    socketname = strdup(sockname);
174    atexit(sockunlink);
175    return sock;
176}
177
178static enum apm_state
179handle_client(int sock_fd, int ctl_fd)
180{
181    /* accept a handle from the client, process it, then clean up */
182    int cli_fd;
183    struct sockaddr_un from;
184    socklen_t fromlen = sizeof(from);
185    struct apm_command cmd;
186    struct apm_reply reply;
187
188    cli_fd = accept(sock_fd, (struct sockaddr *)&from, &fromlen);
189    if (cli_fd == -1) {
190	syslog(LOG_INFO, "client accept failure: %m");
191	return NORMAL;
192    }
193    if (recv(cli_fd, &cmd, sizeof(cmd), 0) != sizeof(cmd)) {
194	(void) close(cli_fd);
195	syslog(LOG_INFO, "client size botch");
196	return NORMAL;
197    }
198    if (cmd.vno != APMD_VNO) {
199	close(cli_fd);			/* terminate client */
200	/* no error message, just drop it. */
201	return NORMAL;
202    }
203    power_status(ctl_fd, 0, &reply.batterystate);
204    switch (cmd.action) {
205    default:
206	reply.newstate = NORMAL;
207	break;
208    case SUSPEND:
209	reply.newstate = SUSPENDING;
210	break;
211    case STANDBY:
212	reply.newstate = STANDING_BY;
213	break;
214    }
215    reply.vno = APMD_VNO;
216    if (send(cli_fd, &reply, sizeof(reply), 0) != sizeof(reply)) {
217	syslog(LOG_INFO, "client reply botch");
218    }
219    close(cli_fd);
220    return reply.newstate;
221}
222
223static int speaker_ok = TRUE;
224
225static void
226make_noise(int howmany)
227{
228    int spkrfd;
229    int trycnt;
230
231    if (!speaker_ok)		/* don't bother after sticky errors */
232	return;
233
234    for (trycnt = 0; trycnt < 3; trycnt++) {
235	spkrfd = open(_PATH_DEV_SPEAKER, O_WRONLY);
236	if (spkrfd == -1) {
237	    switch (errno) {
238	    case EBUSY:
239		usleep(500000);
240		errno = EBUSY;
241		continue;
242	    case ENOENT:
243	    case ENODEV:
244	    case ENXIO:
245	    case EPERM:
246	    case EACCES:
247		syslog(LOG_INFO,
248		       "speaker device " _PATH_DEV_SPEAKER " unavailable: %m");
249		speaker_ok = FALSE;
250		return;
251	    }
252	} else
253	    break;
254    }
255    if (spkrfd == -1) {
256	syslog(LOG_WARNING, "cannot open " _PATH_DEV_SPEAKER ": %m");
257	return;
258    }
259    syslog(LOG_DEBUG, "sending %d tones to speaker\n", howmany);
260    write (spkrfd, "o4cc", 2 + howmany);
261    close(spkrfd);
262    return;
263}
264
265
266static void
267suspend(int ctl_fd)
268{
269    do_etc_file(_PATH_APM_ETC_SUSPEND);
270    sync();
271    make_noise(2);
272    sync();
273    sync();
274    sleep(1);
275    ioctl(ctl_fd, APM_IOC_SUSPEND, 0);
276}
277
278static void
279stand_by(int ctl_fd)
280{
281    do_etc_file(_PATH_APM_ETC_STANDBY);
282    sync();
283    make_noise(1);
284    sync();
285    sync();
286    sleep(1);
287    ioctl(ctl_fd, APM_IOC_STANDBY, 0);
288}
289
290#define TIMO (10*60)			/* 10 minutes */
291
292static void
293resume(int ctl_fd)
294{
295    do_etc_file(_PATH_APM_ETC_RESUME);
296}
297
298int
299main(int argc, char *argv[])
300{
301    const char *fname = apmdev;
302    int ctl_fd, sock_fd, ch, ready;
303    int statonly = 0;
304    struct pollfd set[2];
305    struct apm_event_info apmevent;
306    int suspends, standbys, resumes;
307    int ac_is_off;
308    int noacsleep = 0;
309    int lowbattsleep = 0;
310    mode_t mode = 0660;
311    unsigned long timeout = TIMO;
312    const char *sockname = sockfile;
313    char *user, *group;
314    char *scratch;
315    uid_t uid = 2; /* operator */
316    gid_t gid = 5; /* operator */
317    struct passwd *pw;
318    struct group *gr;
319
320    while ((ch = getopt(argc, argv, "adlqsvf:t:S:m:o:")) != -1)
321	switch(ch) {
322	case 'q':
323	    speaker_ok = FALSE;
324	    break;
325	case 'a':
326	    noacsleep = 1;
327	    break;
328	case 'l':
329	    lowbattsleep = 1;
330	    break;
331	case 'd':
332	    debug = 1;
333	    break;
334	case 'v':
335	    verbose = 1;
336	    break;
337	case 'f':
338	    fname = optarg;
339	    break;
340	case 'S':
341	    sockname = optarg;
342	    break;
343	case 't':
344	    timeout = strtoul(optarg, 0, 0);
345	    if (timeout == 0)
346		usage();
347	    break;
348	case 'm':
349	    mode = strtoul(optarg, 0, 8);
350	    if (mode == 0)
351		usage();
352	    break;
353	case 'o':
354	    /* (user):(group) */
355	    user = optarg;
356	    group = strchr(user, ':');
357	    if (group)
358		*group++ = '\0';
359	    if (*user) {
360		uid = strtoul(user, &scratch, 0);
361		if (*scratch != '\0') {
362		    pw = getpwnam(user);
363		    if (pw)
364			uid = pw->pw_uid;
365		    else
366			errx(1, "user name `%s' unknown", user);
367		}
368	    }
369	    if (group && *group) {
370		gid = strtoul(group, &scratch, 0);
371		if (*scratch != '\0') {
372		    gr = getgrnam(group);
373		    if (gr)
374			gid = gr->gr_gid;
375		    else
376			errx(1, "group name `%s' unknown", group);
377		}
378	    }
379	    break;
380	case 's':			/* status only */
381	    statonly = 1;
382	    break;
383	case '?':
384
385	default:
386	    usage();
387	}
388    argc -= optind;
389    argv += optind;
390    if (debug) {
391	openlog("apmd", 0, LOG_LOCAL1);
392    } else {
393	daemon(0, 0);
394	openlog("apmd", 0, LOG_DAEMON);
395	setlogmask(LOG_UPTO(LOG_NOTICE));
396	pidfile(NULL);
397    }
398    if ((ctl_fd = open(fname, O_RDWR)) == -1) {
399	syslog(LOG_ERR, "cannot open device file `%s'", fname);
400	exit(1);
401    }
402
403    if (statonly) {
404        power_status(ctl_fd, 1, 0);
405	exit(0);
406    } else {
407	struct apm_power_info pinfo;
408	power_status(ctl_fd, 1, &pinfo);
409	do_ac_state(pinfo.ac_state);
410	ac_is_off = (pinfo.ac_state == APM_AC_OFF);
411    }
412
413    (void) signal(SIGTERM, sigexit);
414    (void) signal(SIGHUP, sigexit);
415    (void) signal(SIGINT, sigexit);
416    (void) signal(SIGPIPE, SIG_IGN);
417
418
419    sock_fd = bind_socket(sockname, mode, uid, gid);
420
421    set[0].fd = ctl_fd;
422    set[0].events = POLLIN;
423    set[1].fd = sock_fd;
424    set[1].events = POLLIN;
425
426
427    for (errno = 0;
428	 (ready = poll(set, 2, timeout * 1000)) >= 0 || errno == EINTR;
429	 errno = 0) {
430	if (errno == EINTR)
431	    continue;
432	if (ready == 0) {
433		int status;
434		/* wakeup for timeout: take status */
435		status = power_status(ctl_fd, 0, 0);
436		if (lowbattsleep && status&POWER_STATUS_LOWBATTNOW) {
437			if (noacsleep && status&POWER_STATUS_ACON) {
438				if (debug)
439					syslog(LOG_DEBUG,
440					    "not sleeping because "
441					    "AC is connected");
442			} else
443				suspend(ctl_fd);
444		}
445	}
446	if (set[0].revents & POLLIN) {
447	    suspends = standbys = resumes = 0;
448	    while (ioctl(ctl_fd, APM_IOC_NEXTEVENT, &apmevent) == 0) {
449		if (debug)
450		    syslog(LOG_DEBUG, "apmevent %04x index %d", apmevent.type,
451		           apmevent.index);
452		switch (apmevent.type) {
453		case APM_SUSPEND_REQ:
454		case APM_USER_SUSPEND_REQ:
455		case APM_CRIT_SUSPEND_REQ:
456		    suspends++;
457		    break;
458		case APM_BATTERY_LOW:
459		    if (lowbattsleep)
460			suspends++;
461		    break;
462		case APM_USER_STANDBY_REQ:
463		case APM_STANDBY_REQ:
464		    standbys++;
465		    break;
466#if 0
467		case APM_CANCEL:
468		    suspends = standbys = 0;
469		    break;
470#endif
471		case APM_NORMAL_RESUME:
472		case APM_CRIT_RESUME:
473		case APM_SYS_STANDBY_RESUME:
474		    resumes++;
475		    break;
476		case APM_POWER_CHANGE:
477		{
478		    struct apm_power_info pinfo;
479		    power_status(ctl_fd, 0, &pinfo);
480		    /* power status can change without ac status changing */
481		    if (ac_is_off != (pinfo.ac_state == APM_AC_OFF)) {
482		    	do_ac_state(pinfo.ac_state);
483			ac_is_off = (pinfo.ac_state == APM_AC_OFF);
484		    }
485		    break;
486		}
487		default:
488		    break;
489		}
490	    }
491	    if ((standbys || suspends) && noacsleep &&
492		(power_status(ctl_fd, 0, 0) & POWER_STATUS_ACON)) {
493		if (debug)
494		    syslog(LOG_DEBUG, "not sleeping because AC is connected");
495	    } else if (suspends) {
496		suspend(ctl_fd);
497	    } else if (standbys) {
498		stand_by(ctl_fd);
499	    } else if (resumes) {
500		resume(ctl_fd);
501		if (verbose)
502		    syslog(LOG_NOTICE, "system resumed from APM sleep");
503	    }
504	    ready--;
505	}
506	if (ready == 0)
507	    continue;
508	if (set[1].revents & POLLIN) {
509	    switch (handle_client(sock_fd, ctl_fd)) {
510	    case NORMAL:
511		break;
512	    case SUSPENDING:
513		suspend(ctl_fd);
514		break;
515	    case STANDING_BY:
516		stand_by(ctl_fd);
517		break;
518	    }
519	}
520    }
521    syslog(LOG_ERR, "poll failed: %m");
522    exit(1);
523}
524
525static void
526do_etc_file(const char *file)
527{
528    pid_t pid;
529    int status;
530    const char *prog;
531
532    /* If file doesn't exist, do nothing. */
533    if (access(file, X_OK|R_OK)) {
534	if (debug)
535	    syslog(LOG_DEBUG, "do_etc_file(): cannot access file %s", file);
536	return;
537    }
538
539    prog = strrchr(file, '/');
540    if (prog)
541	prog++;
542    else
543	prog = file;
544
545    pid = fork();
546    switch (pid) {
547    case -1:
548	syslog(LOG_ERR, "failed to fork(): %m");
549	return;
550    case 0:
551	/* We are the child. */
552	if (execl(file, prog, NULL) == -1)
553		syslog(LOG_ERR, "could not execute \"%s\": %m", file);
554	_exit(1);
555	/* NOTREACHED */
556    default:
557	/* We are the parent. */
558	wait4(pid, &status, 0, 0);
559	if (WIFEXITED(status)) {
560	    if (debug)
561		syslog(LOG_DEBUG, "%s exited with status %d", file,
562		       WEXITSTATUS(status));
563	} else
564	    syslog(LOG_ERR, "%s exited abnormally.", file);
565	break;
566    }
567}
568
569static void
570do_ac_state(int state)
571{
572	switch (state) {
573	case APM_AC_OFF:
574		do_etc_file(_PATH_APM_ETC_BATTERY);
575		break;
576	case APM_AC_ON:
577	case APM_AC_BACKUP:
578		do_etc_file(_PATH_APM_ETC_LINE);
579		break;
580	default:
581		/* Silently ignore */ ;
582	}
583}
584