Home | History | Annotate | Line # | Download | only in screenblank
screenblank.c revision 1.21
      1 /*	$NetBSD: screenblank.c,v 1.21 2004/10/30 15:43:25 dsl Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 1996-2002 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Jason R. Thorpe.
      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  * 3. All advertising materials mentioning features or use of this software
     19  *    must display the following acknowledgement:
     20  *        This product includes software developed by the NetBSD
     21  *        Foundation, Inc. and its contributors.
     22  * 4. Neither the name of The NetBSD Foundation nor the names of its
     23  *    contributors may be used to endorse or promote products derived
     24  *    from this software without specific prior written permission.
     25  *
     26  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     27  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     28  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     29  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     30  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     31  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     32  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     33  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     34  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     35  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     36  * POSSIBILITY OF SUCH DAMAGE.
     37  */
     38 
     39 /*
     40  * Screensaver daemon for the Sun 3 and SPARC, and platforms using WSCONS.
     41  */
     42 
     43 #include <sys/cdefs.h>
     44 #ifndef lint
     45 __COPYRIGHT(
     46 "@(#) Copyright (c) 1996-2002 \
     47 	The NetBSD Foundation, Inc.  All rights reserved.");
     48 __RCSID("$NetBSD: screenblank.c,v 1.21 2004/10/30 15:43:25 dsl Exp $");
     49 #endif
     50 
     51 #include <sys/types.h>
     52 #include <sys/time.h>
     53 #include <sys/stat.h>
     54 #include <sys/ioctl.h>
     55 #include <sys/queue.h>
     56 #include <ctype.h>
     57 #include <err.h>
     58 #include <errno.h>
     59 #include <fcntl.h>
     60 #include <limits.h>
     61 #include <math.h>
     62 #include <paths.h>
     63 #include <stdlib.h>
     64 #include <stdio.h>
     65 #include <string.h>
     66 #include <signal.h>
     67 #include <syslog.h>
     68 #include <unistd.h>
     69 #include <util.h>
     70 
     71 #include <dev/wscons/wsconsio.h>
     72 
     73 #ifdef HAVE_FBIO
     74 #include <dev/sun/fbio.h>
     75 #endif
     76 
     77 #include "pathnames.h"
     78 
     79 u_long	setvideo = WSDISPLAYIO_SVIDEO;		/* "set video" ioctl */
     80 int	videoon  = WSDISPLAYIO_VIDEO_ON;	/* value for "on" */
     81 int	videooff = WSDISPLAYIO_VIDEO_OFF;	/* value for "off" */
     82 
     83 struct	dev_stat {
     84 	LIST_ENTRY(dev_stat) ds_link;	/* linked list */
     85 	const char *ds_path;		/* path to device */
     86 	int	ds_isfb;		/* boolean; framebuffer? */
     87 	time_t	ds_atime;		/* time device last accessed */
     88 	time_t	ds_mtime;		/* time device last modified */
     89 };
     90 LIST_HEAD(ds_list, dev_stat) ds_list;
     91 
     92 int	main(int, char *[]);
     93 static	void add_dev(const char *, int);
     94 static	void change_state(int);
     95 static	void cvt_arg(char *, struct timespec *);
     96 static	void sighandler(int);
     97 static	void usage(void);
     98 
     99 int
    100 main(int argc, char *argv[])
    101 {
    102 	struct dev_stat *dsp;
    103 	struct timespec timo_on, timo_off, *tvp, tv;
    104 	struct sigaction sa;
    105 	struct stat st;
    106 	int ch, change, fflag = 0, kflag = 0, mflag = 0, state;
    107 	const char *kbd, *mouse, *display;
    108 
    109 	LIST_INIT(&ds_list);
    110 
    111 	/*
    112 	 * Set the default timeouts: 10 minutes on, .25 seconds off.
    113 	 */
    114 	timo_on.tv_sec = 600;
    115 	timo_on.tv_nsec = 0;
    116 	timo_off.tv_sec = 0;
    117 	timo_off.tv_nsec = 250000000;
    118 
    119 	while ((ch = getopt(argc, argv, "d:e:f:i:km")) != -1) {
    120 		switch (ch) {
    121 		case 'd':
    122 			cvt_arg(optarg, &timo_on);
    123 			break;
    124 
    125 		case 'e':
    126 			cvt_arg(optarg, &timo_off);
    127 			break;
    128 
    129 		case 'f':
    130 			fflag = 1;
    131 			add_dev(optarg, 1);
    132 			break;
    133 
    134 		case 'i':
    135 			add_dev(optarg, 0);
    136 			break;
    137 
    138 		case 'k':
    139 			if (mflag || kflag)
    140 				usage();
    141 			kflag = 1;
    142 			break;
    143 
    144 		case 'm':
    145 			if (kflag || mflag)
    146 				usage();
    147 			mflag = 1;
    148 			break;
    149 
    150 		default:
    151 			usage();
    152 		}
    153 	}
    154 	argc -= optind;
    155 	if (argc)
    156 		usage();
    157 
    158 	/*
    159 	 * Default to WSCONS support.
    160 	 */
    161 	kbd = _PATH_WSKBD;
    162 	mouse = _PATH_WSMOUSE;
    163 	display = _PATH_WSDISPLAY;
    164 
    165 #ifdef HAVE_FBIO
    166 	/*
    167 	 * If a display device wasn't specified, check to see which we
    168 	 * have.  If we can't open the WSCONS display, fall back to fbio.
    169 	 */
    170 	if (!fflag) {
    171 		int fd;
    172 
    173 		if ((fd = open(display, O_RDONLY, 0666)) == -1)
    174 			setvideo = FBIOSVIDEO;
    175 		else
    176 			(void) close(fd);
    177 	}
    178 
    179 	/*
    180 	 * Do this here so that -f ... args above can influence us.
    181 	 */
    182 	if (setvideo == FBIOSVIDEO) {
    183 		videoon = FBVIDEO_ON;
    184 		videooff = FBVIDEO_OFF;
    185 		kbd = _PATH_KEYBOARD;
    186 		mouse = _PATH_MOUSE;
    187 		display = _PATH_FB;
    188 	}
    189 #endif
    190 
    191 	/*
    192 	 * Add the keyboard, mouse, and default framebuffer devices
    193 	 * as necessary.  We _always_ check the console device.
    194 	 */
    195 	add_dev(_PATH_CONSOLE, 0);
    196 	if (!kflag)
    197 		add_dev(kbd, 0);
    198 	if (!mflag)
    199 		add_dev(mouse, 0);
    200 	if (!fflag)
    201 		add_dev(display, 1);
    202 
    203 	/* Ensure that the framebuffer is on. */
    204 	state = videoon;
    205 	change_state(state);
    206 	tvp = &timo_on;
    207 
    208 	/*
    209 	 * Make sure the framebuffer gets turned back on when we're
    210 	 * killed.
    211 	 */
    212 	sa.sa_handler = sighandler;
    213 	sa.sa_flags = SA_NOCLDSTOP;
    214 	if (sigemptyset(&sa.sa_mask))
    215 		err(1, "sigemptyset");
    216 	if (sigaction(SIGINT, &sa, NULL) || sigaction(SIGTERM, &sa, NULL) ||
    217 	    sigaction(SIGHUP, &sa, NULL))
    218 		err(1, "sigaction");
    219 
    220 	openlog("screenblank", LOG_PID, LOG_DAEMON);
    221 	/* Detach. */
    222 	if (daemon(0, 0))
    223 		err(1, "daemon");
    224 	pidfile(NULL);
    225 
    226 	/* Start the state machine. */
    227 	for (;;) {
    228 		change = 0;
    229 		for (dsp = ds_list.lh_first; dsp != NULL;
    230 		    dsp = dsp->ds_link.le_next) {
    231 			/* Don't check framebuffers. */
    232 			if (dsp->ds_isfb)
    233 				continue;
    234 			if (stat(dsp->ds_path, &st) == -1) {
    235 				syslog(LOG_CRIT,
    236 				    "Can't stat `%s' (%m)", dsp->ds_path);
    237 				exit(1);
    238 			}
    239 			if (st.st_atime > dsp->ds_atime) {
    240 				change = 1;
    241 				dsp->ds_atime = st.st_atime;
    242 			}
    243 			if (st.st_mtime > dsp->ds_mtime) {
    244 				change = 1;
    245 				dsp->ds_mtime = st.st_mtime;
    246 			}
    247 		}
    248 
    249 		if (state == videoon) {
    250 			if (!change) {
    251 				state = videooff;
    252 				change_state(state);
    253 				tvp = &timo_off;
    254 			}
    255 		} else {
    256 			if (change) {
    257 				state = videoon;
    258 				change_state(state);
    259 				tvp = &timo_on;
    260 			}
    261 		}
    262 
    263 		tv = *tvp;
    264 		if (nanosleep(&tv, NULL) == -1)
    265 			err(1, "nanosleep");
    266 	}
    267 	/* NOTREACHED */
    268 }
    269 
    270 static void
    271 add_dev(const char *path, int isfb)
    272 {
    273 	struct dev_stat *dsp;
    274 	struct stat sb;
    275 
    276 	/* Make sure we can stat the device. */
    277 	if (stat(path, &sb) == -1) {
    278 		warn("Can't stat `%s'", path);
    279 		return;
    280 	}
    281 
    282 #ifdef HAVE_FBIO
    283 	/*
    284 	 * We default to WSCONS.  If this is a frame buffer
    285 	 * device, check to see if it responds to the old
    286 	 * Sun-style fbio ioctls.  If so, switch to fbio mode.
    287 	 */
    288 	if (isfb && setvideo != FBIOSVIDEO) {
    289 		int onoff, fd;
    290 
    291 		if ((fd = open(path, O_RDWR, 0666)) == -1) {
    292 			warn("Can't open `%s'", path);
    293 			return;
    294 		}
    295 		if ((ioctl(fd, FBIOGVIDEO, &onoff)) == 0)
    296 			setvideo = FBIOSVIDEO;
    297 		(void)close(fd);
    298 	}
    299 #endif
    300 
    301 	/* Create the entry... */
    302 	dsp = malloc(sizeof(struct dev_stat));
    303 	if (dsp == NULL)
    304 		err(1, "Can't allocate memory for `%s'", path);
    305 	(void)memset(dsp, 0, sizeof(struct dev_stat));
    306 	dsp->ds_path = path;
    307 	dsp->ds_isfb = isfb;
    308 
    309 	/* ...and put it in the list. */
    310 	LIST_INSERT_HEAD(&ds_list, dsp, ds_link);
    311 }
    312 
    313 /* ARGSUSED */
    314 static void
    315 sighandler(int sig)
    316 {
    317 
    318 	/* Kill the pid file and re-enable the framebuffer before exit. */
    319 	change_state(videoon);
    320 	exit(0);
    321 }
    322 
    323 static void
    324 change_state(int state)
    325 {
    326 	struct dev_stat *dsp;
    327 	int fd;
    328 	int fail = 1;
    329 
    330 	for (dsp = ds_list.lh_first; dsp != NULL; dsp = dsp->ds_link.le_next) {
    331 		/* Don't change the state of non-framebuffers! */
    332 		if (dsp->ds_isfb == 0)
    333 			continue;
    334 		if ((fd = open(dsp->ds_path, O_RDWR, 0)) == -1) {
    335 			syslog(LOG_WARNING, "Can't open `%s' (%m)",
    336 			    dsp->ds_path);
    337 			continue;
    338 		}
    339 		if (ioctl(fd, setvideo, &state) == -1)
    340 			syslog(LOG_WARNING, "Can't set video on `%s' (%m)",
    341 			    dsp->ds_path);
    342 		else
    343 			fail = 0;
    344 		(void)close(fd);
    345 	}
    346 	if (fail) {
    347 		syslog(LOG_CRIT, "No frame buffer devices, exiting\n");
    348 		exit(1);
    349 	}
    350 }
    351 
    352 static void
    353 cvt_arg(char *arg, struct timespec *tvp)
    354 {
    355 	char *cp;
    356 	int seconds, nanoseconds, factor;
    357 	int period = 0;
    358 	factor = 1000000000;
    359 	nanoseconds = 0;
    360 	seconds = 0;
    361 
    362 	for (cp = arg; *cp != '\0'; ++cp) {
    363 		if (*cp == '.') {
    364 			if (period)
    365 				errx(1, "Invalid argument: %s", arg);
    366 			period = 1;
    367 			continue;
    368 		}
    369 
    370 		if (!isdigit((unsigned char)*cp))
    371 			errx(1, "Invalid argument: %s", arg);
    372 
    373 		if (period) {
    374 			if (factor > 1) {
    375 				nanoseconds = nanoseconds * 10 + (*cp - '0');
    376 				factor /= 10;
    377 			}
    378 		} else
    379 			seconds = (seconds * 10) + (*cp - '0');
    380 	}
    381 
    382 	tvp->tv_sec = seconds;
    383 	if (factor > 1)
    384 		nanoseconds *= factor;
    385 
    386 	tvp->tv_nsec = nanoseconds;
    387 }
    388 
    389 static void
    390 usage(void)
    391 {
    392 
    393 	(void)fprintf(stderr,
    394 	    "usage: %s [-k | -m] [-d inactivity-timeout] [-e wakeup-delay]\n"
    395 	    "\t\t[-f framebuffer] [-i input-device]\n",
    396 	    getprogname());
    397 	exit(1);
    398 }
    399