Home | History | Annotate | Line # | Download | only in apm
      1 /*	$NetBSD: apm.c,v 1.21 2011/11/25 12:51:28 joerg Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 1996 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 <sys/types.h>
     33 #include <sys/ioctl.h>
     34 #include <sys/socket.h>
     35 #include <sys/time.h>
     36 #include <sys/un.h>
     37 
     38 #include <machine/apmvar.h>
     39 
     40 #include <err.h>
     41 #include <errno.h>
     42 #include <fcntl.h>
     43 #include <stdio.h>
     44 #include <stdlib.h>
     45 #include <string.h>
     46 #include <unistd.h>
     47 
     48 #include "pathnames.h"
     49 #include "apm-proto.h"
     50 
     51 #define	FALSE 0
     52 #define	TRUE 1
     53 
     54 __dead static void	usage(void);
     55 __dead static void	zzusage(void);
     56 static int	do_zzz(const char *, enum apm_action);
     57 static int	open_socket(const char *);
     58 static int	send_command(int, struct apm_command *, struct apm_reply *);
     59 
     60 static void
     61 usage(void)
     62 {
     63 
     64 	fprintf(stderr,"usage: %s [-v] [-z | -S] [-abdlms] [-f socket]\n",
     65 	    getprogname());
     66 	exit(1);
     67 }
     68 
     69 static void
     70 zzusage(void)
     71 {
     72 
     73 	fprintf(stderr,"usage: %s [-z | -S] [-f socket]\n",
     74 	    getprogname());
     75 	exit(1);
     76 }
     77 
     78 static int
     79 send_command(int fd,
     80     struct apm_command *cmd,
     81     struct apm_reply *reply)
     82 {
     83 
     84 	/* send a command to the apm daemon */
     85 	cmd->vno = APMD_VNO;
     86 
     87 	if (send(fd, cmd, sizeof(*cmd), 0) == sizeof(*cmd)) {
     88 		if (recv(fd, reply, sizeof(*reply), 0) != sizeof(*reply)) {
     89 			warn("invalid reply from APM daemon");
     90 			return (1);
     91 		}
     92 	} else {
     93 		warn("invalid send to APM daemon");
     94 		return (1);
     95 	}
     96 	return (0);
     97 }
     98 
     99 static int
    100 do_zzz(const char *pn, enum apm_action action)
    101 {
    102 	struct apm_command command;
    103 	struct apm_reply reply;
    104 	int fd;
    105 
    106 	switch (action) {
    107 	case NONE:
    108 	case SUSPEND:
    109 		command.action = SUSPEND;
    110 		break;
    111 	case STANDBY:
    112 		command.action = STANDBY;
    113 		break;
    114 	default:
    115 		zzusage();
    116 	}
    117 
    118 	fd = open_socket(pn);
    119 	if (fd == -1)
    120 		err(1, "cannot open connection to APM daemon");
    121 	printf("Suspending system...\n");
    122 	exit(send_command(fd, &command, &reply));
    123 }
    124 
    125 static int
    126 open_socket(const char *sockname)
    127 {
    128 	struct sockaddr_un s_un;
    129 	int sock, errr;
    130 
    131 	sock = socket(AF_LOCAL, SOCK_STREAM, 0);
    132 	if (sock == -1)
    133 		err(1, "cannot create local socket");
    134 
    135 	s_un.sun_family = AF_LOCAL;
    136 	strncpy(s_un.sun_path, sockname, sizeof(s_un.sun_path));
    137 	s_un.sun_len = SUN_LEN(&s_un);
    138 	if (connect(sock, (struct sockaddr *)&s_un, s_un.sun_len) == -1) {
    139 		errr = errno;
    140 		close(sock);
    141 		errno = errr;
    142 		return (-1);
    143 	}
    144 	return (sock);
    145 }
    146 
    147 int
    148 main(int argc, char *argv[])
    149 {
    150 	struct apm_command command;
    151 	struct apm_reply reply;
    152 	struct apm_power_info *api = &reply.batterystate;
    153 	const char *sockname = _PATH_APM_SOCKET;
    154 	enum apm_action action = NONE;
    155 	int ch, doac, dobstate, domin, dopct, dostatus, fd, nodaemon,
    156 	    rval, verbose;
    157 
    158 	doac = dobstate = domin = dopct = dostatus = nodaemon =
    159 	    verbose = FALSE;
    160 	while ((ch = getopt(argc, argv, "Sabdf:lmsvz")) != -1)
    161 		switch (ch) {
    162 		case 'v':
    163 			verbose = TRUE;
    164 			break;
    165 		case 'f':
    166 			sockname = optarg;
    167 			break;
    168 		case 'z':
    169 			if (action != NONE)
    170 				usage();
    171 			action = SUSPEND;
    172 			break;
    173 		case 'S':
    174 			if (action != NONE)
    175 				usage();
    176 			action = STANDBY;
    177 			break;
    178 		case 's':
    179 			if (action != NONE && action != GETSTATUS)
    180 				usage();
    181 			dostatus = TRUE;
    182 			action = GETSTATUS;
    183 			break;
    184 		case 'b':
    185 			if (action != NONE && action != GETSTATUS)
    186 				usage();
    187 			dobstate = TRUE;
    188 			action = GETSTATUS;
    189 			break;
    190 		case 'l':
    191 			if (action != NONE && action != GETSTATUS)
    192 				usage();
    193 			dopct = TRUE;
    194 			action = GETSTATUS;
    195 			break;
    196 		case 'm':
    197 			if (action != NONE && action != GETSTATUS)
    198 				usage();
    199 			domin = TRUE;
    200 			action = GETSTATUS;
    201 			break;
    202 		case 'a':
    203 			if (action != NONE && action != GETSTATUS)
    204 				usage();
    205 			doac = TRUE;
    206 			action = GETSTATUS;
    207 			break;
    208 		case 'd':
    209 			nodaemon = TRUE;
    210 			break;
    211 		case '?':
    212 		default:
    213 			usage();
    214 		}
    215 
    216 	if (strcmp(getprogname(), "zzz") == 0)
    217 		exit(do_zzz(sockname, action));
    218 
    219 	if (nodaemon)
    220 		fd = -1;
    221 	else
    222 		fd = open_socket(sockname);
    223 
    224 	switch (action) {
    225 	case NONE:
    226 		verbose = doac = dopct = domin = dobstate = dostatus = TRUE;
    227 		action = GETSTATUS;
    228 		/* FALLTHROUGH */
    229 	case GETSTATUS:
    230 		if (fd == -1) {
    231 			/* open the device directly and get status */
    232 			fd = open(_PATH_APM_NORMAL, O_RDONLY);
    233 			if (fd == -1) {
    234 				err(1, "cannot contact APM daemon and "
    235 				    "cannot open "
    236 				    _PATH_APM_NORMAL);
    237 			}
    238 			memset(&reply, 0, sizeof(reply));
    239 			if (ioctl(fd, APM_IOC_GETPOWER,
    240 			    &reply.batterystate) == -1)
    241 				err(1, "ioctl(APM_IOC_GETPOWER)");
    242 			goto printval;
    243 		}
    244 		/* FALLTHROUGH */
    245 	case SUSPEND:
    246 	case STANDBY:
    247 		if (nodaemon && fd == -1) {
    248 			fd = open(_PATH_APM_CTLDEV, O_RDWR);
    249 			if (fd == -1)
    250 				err(1, "cannot open APM control device "
    251 				    _PATH_APM_CTLDEV);
    252 			sync();
    253 			sync();
    254 			sleep(1);
    255 			if (ioctl(fd, action == SUSPEND ?
    256 			    APM_IOC_SUSPEND : APM_IOC_STANDBY, 0) == -1)
    257 				err(1, "cannot enter requested power state");
    258 			printf("System will enter %s in a moment.\n",
    259 			    action == SUSPEND ? "suspend mode" :
    260 			    "standby mode");
    261 			exit(0);
    262 		} else if (fd == -1)
    263 			err(1, "cannot contact APM daemon at socket "
    264 			    _PATH_APM_SOCKET);
    265 		command.action = action;
    266 		break;
    267 	default:
    268 		usage();
    269 	}
    270 
    271 	if ((rval = send_command(fd, &command, &reply)) == 0) {
    272 		switch (action) {
    273 		case GETSTATUS:
    274 printval:
    275 			if (verbose) {
    276 				if (dobstate)
    277 					printf("Battery charge state: %s\n",
    278 					    battstate(api->battery_state));
    279 
    280 				if (dopct && domin && api->minutes_left == 0)
    281 					domin = FALSE;
    282 
    283 				if (dopct || domin) {
    284 					printf("Battery remaining: ");
    285 					if (dopct)
    286 						printf("%d percent",
    287 						    api->battery_life);
    288 					if (dopct && domin)
    289 						printf(" (");
    290 					if (domin)
    291 						printf("%d minutes",
    292 						    api->minutes_left);
    293 					if (dopct && domin)
    294 						printf(")");
    295 					printf("\n");
    296 				}
    297 				if (doac)
    298 					printf("A/C adapter state: %s\n",
    299 					    ac_state(api->ac_state));
    300 				if (dostatus)
    301 					printf("Power management enabled\n");
    302 				if (api->nbattery) {
    303 					printf("Number of batteries: %u\n",
    304 					    api->nbattery);
    305 				}
    306 			} else {
    307 				if (dobstate)
    308 					printf("%d\n", api->battery_state);
    309 				if (dopct)
    310 					printf("%d\n", api->battery_life);
    311 				if (domin)
    312 					printf("%d\n", api->minutes_left);
    313 				if (doac)
    314 					printf("%d\n", api->ac_state);
    315 				if (dostatus)
    316 					printf("1\n");
    317 			}
    318 			break;
    319 		default:
    320 			break;
    321 		}
    322 		switch (reply.newstate) {
    323 		case SUSPEND:
    324 			printf("System will enter suspend mode "
    325 			    "in a moment.\n");
    326 			break;
    327 		case STANDBY:
    328 			printf("System will enter standby mode "
    329 			    "in a moment.\n");
    330 			break;
    331 		default:
    332 			break;
    333 		}
    334 	} else
    335 		errx(rval, "cannot get reply from APM daemon");
    336 
    337 	exit(0);
    338 }
    339