1 /* $NetBSD: screenblank.c,v 1.19 2002/09/19 02:59:20 mycroft 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.19 2002/09/19 02:59:20 mycroft 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(*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