1 /*-
2  * Copyright (c) 1994-1996 S�ren Schmidt
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer,
10  *    in this position and unchanged.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  * 3. The name of the author may not be used to endorse or promote products
15  *    derived from this software withough specific prior written permission
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  *
28  * $FreeBSD: src/usr.sbin/vidcontrol/vidcontrol.c,v 1.32.2.7 2002/09/15 22:31:50 dd Exp $
29  * $DragonFly: src/usr.sbin/vidcontrol/vidcontrol.c,v 1.9 2005/02/19 01:42:16 swildner Exp $
30  */
31 
32 #include <machine/console.h>
33 
34 #include <sys/consio.h>
35 #include <sys/errno.h>
36 #include <sys/types.h>
37 #include <sys/stat.h>
38 
39 #include <ctype.h>
40 #include <err.h>
41 #include <limits.h>
42 #include <stdio.h>
43 #include <stdlib.h>
44 #include <string.h>
45 #include <unistd.h>
46 
47 #include "path.h"
48 #include "decode.h"
49 
50 
51 #define	DATASIZE(x)	((x).w * (x).h * 256 / 8)
52 
53 #define	DUMP_RAW	0
54 #define	DUMP_TXT	1
55 
56 #define	DUMP_FMT_REV	1
57 
58 
59 char legal_colors[16][16] = {
60 	"black", "blue", "green", "cyan",
61 	"red", "magenta", "brown", "white",
62 	"grey", "lightblue", "lightgreen", "lightcyan",
63 	"lightred", "lightmagenta", "yellow", "lightwhite"
64 };
65 
66 struct {
67 	int			active_vty;
68 	vid_info_t		console_info;
69 	unsigned char		screen_map[256];
70 	int			video_mode_number;
71 	struct video_info	video_mode_info;
72 } cur_info;
73 
74 int hex = 0;
75 int number;
76 int vesa_cols;
77 int vesa_rows;
78 int font_height;
79 int colors_changed;
80 int video_mode_changed;
81 int normal_fore_color, normal_back_color;
82 int revers_fore_color, revers_back_color;
83 char letter;
84 struct vid_info info;
85 struct video_info new_mode_info;
86 
87 
88 /*
89  * Initialize revert data.
90  *
91  * NOTE: the following parameters are not yet saved/restored:
92  *
93  *   screen saver timeout
94  *   cursor type
95  *   mouse character and mouse show/hide state
96  *   vty switching on/off state
97  *   history buffer size
98  *   history contents
99  *   font maps
100  */
101 
102 static void
103 init(void)
104 {
105 	if (ioctl(0, VT_GETACTIVE, &cur_info.active_vty) == -1)
106 		errc(1, errno, "getting active vty");
107 
108 	cur_info.console_info.size = sizeof(cur_info.console_info);
109 
110 	if (ioctl(0, CONS_GETINFO, &cur_info.console_info) == -1)
111 		errc(1, errno, "getting console information");
112 
113 	if (ioctl(0, GIO_SCRNMAP, &cur_info.screen_map) == -1)
114 		errc(1, errno, "getting screen map");
115 
116 	if (ioctl(0, CONS_GET, &cur_info.video_mode_number) == -1)
117 		errc(1, errno, "getting video mode number");
118 
119 	cur_info.video_mode_info.vi_mode = cur_info.video_mode_number;
120 
121 	if (ioctl(0, CONS_MODEINFO, &cur_info.video_mode_info) == -1)
122 		errc(1, errno, "getting video mode parameters");
123 
124 	normal_fore_color = cur_info.console_info.mv_norm.fore;
125 	normal_back_color = cur_info.console_info.mv_norm.back;
126 	revers_fore_color = cur_info.console_info.mv_rev.fore;
127 	revers_back_color = cur_info.console_info.mv_rev.back;
128 }
129 
130 
131 /*
132  * If something goes wrong along the way we call revert() to go back to the
133  * console state we came from (which is assumed to be working).
134  *
135  * NOTE: please also read the comments of init().
136  */
137 
138 static void
139 revert(void)
140 {
141 	int size[3];
142 
143 	ioctl(0, VT_ACTIVATE, (caddr_t) (long) cur_info.active_vty);
144 
145 	fprintf(stderr, "[=%dA", cur_info.console_info.mv_ovscan);
146 	fprintf(stderr, "[=%dF", cur_info.console_info.mv_norm.fore);
147 	fprintf(stderr, "[=%dG", cur_info.console_info.mv_norm.back);
148 	fprintf(stderr, "[=%dH", cur_info.console_info.mv_rev.fore);
149 	fprintf(stderr, "[=%dI", cur_info.console_info.mv_rev.back);
150 
151 	ioctl(0, PIO_SCRNMAP, &cur_info.screen_map);
152 
153 	if (cur_info.video_mode_number >= M_VESA_BASE)
154 		ioctl(0, _IO('V', cur_info.video_mode_number - M_VESA_BASE),
155 		      NULL);
156 	else
157 		ioctl(0, _IO('S', cur_info.video_mode_number), NULL);
158 
159 	if (cur_info.video_mode_info.vi_flags & V_INFO_GRAPHICS) {
160 		size[0] = cur_info.video_mode_info.vi_width / 8;
161 		size[1] = cur_info.video_mode_info.vi_height /
162 			  cur_info.console_info.font_size;
163 		size[2] = cur_info.console_info.font_size;
164 
165 		ioctl(0, KDRASTER, size);
166 	}
167 }
168 
169 
170 /*
171  * Print a short usage string describing all options, then exit.
172  */
173 
174 static void
175 usage(void)
176 {
177 	fprintf(stderr,
178 		"usage: vidcontrol [-CdLPpx] [-b color] [-c appearance]"
179 		" [-f [size] file]\n"
180 		"                  [-g geometry] [-h size] [-i adapter | mode]"
181 		" [-l screen_map]\n"
182 		"                  [-M char] [-m on | off] [-r foreground"
183 		" background]\n"
184 		"                  [-S on | off] [-s number] [-t N | off]"
185 		" [mode]\n"
186 		"                  [foreground [background]] [show]\n");
187 
188 	exit(1);
189 }
190 
191 
192 /*
193  * Retrieve the next argument from the command line (for options that require
194  * more than one argument).
195  */
196 
197 static char *
198 nextarg(int ac, char **av, int *indp, int oc, int strict)
199 {
200 	if (*indp < ac)
201 		return(av[(*indp)++]);
202 
203 	if (strict != 0) {
204 		revert();
205 		errx(1, "option requires two arguments -- %c", oc);
206 	}
207 
208 	return(NULL);
209 }
210 
211 
212 /*
213  * Guess which file to open. Try to open each combination of a specified set
214  * of file name components.
215  */
216 
217 static FILE *
218 openguess(const char *a[], const char *b[], const char *c[], const char *d[], char **name)
219 {
220 	FILE *f;
221 	int i, j, k, l;
222 
223 	for (i = 0; a[i] != NULL; i++) {
224 		for (j = 0; b[j] != NULL; j++) {
225 			for (k = 0; c[k] != NULL; k++) {
226 				for (l = 0; d[l] != NULL; l++) {
227 					asprintf(name, "%s%s%s%s",
228 						 a[i], b[j], c[k], d[l]);
229 
230 					f = fopen(*name, "r");
231 
232 					if (f != NULL)
233 						return (f);
234 
235 					free(*name);
236 				}
237 			}
238 		}
239 	}
240 
241 	return (NULL);
242 }
243 
244 
245 /*
246  * Load a screenmap from a file and set it.
247  */
248 
249 static void
250 load_scrnmap(char *filename)
251 {
252 	FILE *fd;
253 	int size;
254 	char *name;
255 	scrmap_t scrnmap;
256 	const char *a[] = {"", SCRNMAP_PATH, NULL};
257 	const char *b[] = {filename, NULL};
258 	const char *c[] = {"", ".scm", NULL};
259 	const char *d[] = {"", NULL};
260 
261 	fd = openguess(a, b, c, d, &name);
262 
263 	if (fd == NULL) {
264 		revert();
265 		errx(1, "screenmap file not found");
266 	}
267 
268 	size = sizeof(scrnmap);
269 
270 	if (decode(fd, (char *)&scrnmap, size) != size) {
271 		rewind(fd);
272 
273 		if (fread(&scrnmap, 1, size, fd) != (size_t)size) {
274 			fclose(fd);
275 			revert();
276 			errx(1, "bad screenmap file");
277 		}
278 	}
279 
280 	if (ioctl(0, PIO_SCRNMAP, &scrnmap) == -1) {
281 		revert();
282 		errc(1, errno, "loading screenmap");
283 	}
284 
285 	fclose(fd);
286 }
287 
288 
289 /*
290  * Set the default screenmap.
291  */
292 
293 static void
294 load_default_scrnmap(void)
295 {
296 	scrmap_t scrnmap;
297 	int i;
298 
299 	for (i = 0; i < 256; i++)
300 		*((char*)&scrnmap + i) = i;
301 
302 	if (ioctl(0, PIO_SCRNMAP, &scrnmap) == -1) {
303 		revert();
304 		errc(1, errno, "loading default screenmap");
305 	}
306 }
307 
308 
309 /*
310  * Print the current screenmap to stdout.
311  */
312 
313 static void
314 print_scrnmap(void)
315 {
316 	unsigned char map[256];
317 	size_t i;
318 
319 	if (ioctl(0, GIO_SCRNMAP, &map) == -1) {
320 		revert();
321 		errc(1, errno, "getting screenmap");
322 	}
323 
324 	for (i=0; i<sizeof(map); i++) {
325 		if (i > 0 && i % 16 == 0)
326 			fprintf(stdout, "\n");
327 
328 		if (hex != NULL)
329 			fprintf(stdout, " %02x", map[i]);
330 		else
331 			fprintf(stdout, " %03d", map[i]);
332 	}
333 
334 	fprintf(stdout, "\n");
335 }
336 
337 
338 /*
339  * Determine a file's size.
340  */
341 
342 static int
343 fsize(FILE *file)
344 {
345 	struct stat sb;
346 
347 	if (fstat(fileno(file), &sb) == 0)
348 		return sb.st_size;
349 	else
350 		return -1;
351 }
352 
353 
354 /*
355  * Load a font from file and set it.
356  */
357 
358 static void
359 load_font(char *type, char *filename)
360 {
361 	FILE *fd;
362 	int h, i, size, w;
363 	unsigned long io = 0;	/* silence stupid gcc(1) in the Wall mode */
364 	char *name, *fontmap, size_sufx[6];
365 	const char *a[] = {"", FONT_PATH, NULL};
366 	const char *b[] = {filename, NULL};
367 	const char *c[] = {"", size_sufx, NULL};
368 	const char *d[] = {"", ".fnt", NULL};
369 	vid_info_t vinfo;
370 
371 	struct sizeinfo {
372 		int w;
373 		int h;
374 		unsigned long io;
375 	} sizes[] = {{8, 16, PIO_FONT8x16},
376 		     {8, 14, PIO_FONT8x14},
377 		     {8,  8, PIO_FONT8x8},
378 		     {0,  0, 0}};
379 
380 	vinfo.size = sizeof(vinfo);
381 
382 	if (ioctl(0, CONS_GETINFO, &vinfo) == -1) {
383 		revert();
384 		errc(1, errno, "obtaining current video mode parameters");
385 	}
386 
387 	snprintf(size_sufx, sizeof(size_sufx), "-8x%d", vinfo.font_size);
388 
389 	fd = openguess(a, b, c, d, &name);
390 
391 	if (fd == NULL) {
392 		revert();
393 		errx(1, "%s: can't load font file", filename);
394 	}
395 
396 	if (type != NULL) {
397 		size = 0;
398 		if (sscanf(type, "%dx%d", &w, &h) == 2) {
399 			for (i = 0; sizes[i].w != 0; i++) {
400 				if (sizes[i].w == w && sizes[i].h == h) {
401 					size = DATASIZE(sizes[i]);
402 					io = sizes[i].io;
403 					font_height = sizes[i].h;
404 				}
405 			}
406 		}
407 		if (size == 0) {
408 			fclose(fd);
409 			revert();
410 			errx(1, "%s: bad font size specification", type);
411 		}
412 	} else {
413 		/* Apply heuristics */
414 
415 		int j;
416 		int dsize[2];
417 
418 		size = DATASIZE(sizes[0]);
419 		fontmap = (char*) malloc(size);
420 		dsize[0] = decode(fd, fontmap, size);
421 		dsize[1] = fsize(fd);
422 		free(fontmap);
423 
424 		size = 0;
425 		for (j = 0; j < 2; j++) {
426 			for (i = 0; sizes[i].w != 0; i++) {
427 				if (DATASIZE(sizes[i]) == dsize[j]) {
428 					size = dsize[j];
429 					io = sizes[i].io;
430 					font_height = sizes[i].h;
431 					j = 2;	/* XXX */
432 					break;
433 				}
434 			}
435 		}
436 
437 		if (size == 0) {
438 			fclose(fd);
439 			revert();
440 			errx(1, "%s: can't guess font size", filename);
441 		}
442 
443 		rewind(fd);
444 	}
445 
446 	fontmap = (char*) malloc(size);
447 
448 	if (decode(fd, fontmap, size) != size) {
449 		rewind(fd);
450 		if (fsize(fd) != size ||
451 		    fread(fontmap, 1, size, fd) != (size_t)size) {
452 			fclose(fd);
453 			free(fontmap);
454 			revert();
455 			errx(1, "%s: bad font file", filename);
456 		}
457 	}
458 
459 	if (ioctl(0, io, fontmap) == -1) {
460 		revert();
461 		errc(1, errno, "loading font");
462 	}
463 
464 	fclose(fd);
465 	free(fontmap);
466 }
467 
468 
469 /*
470  * Set the timeout for the screensaver.
471  */
472 
473 static void
474 set_screensaver_timeout(char *arg)
475 {
476 	int nsec;
477 
478 	if (!strcmp(arg, "off")) {
479 		nsec = 0;
480 	} else {
481 		nsec = atoi(arg);
482 
483 		if ((*arg == '\0') || (nsec < 1)) {
484 			revert();
485 			errx(1, "argument must be a positive number");
486 		}
487 	}
488 
489 	if (ioctl(0, CONS_BLANKTIME, &nsec) == -1) {
490 		revert();
491 		errc(1, errno, "setting screensaver period");
492 	}
493 }
494 
495 
496 /*
497  * Set the cursor's shape/type.
498  */
499 
500 static void
501 set_cursor_type(char *appearance)
502 {
503 	int type;
504 
505 	if (!strcmp(appearance, "normal"))
506 		type = 0;
507 	else if (!strcmp(appearance, "blink"))
508 		type = 1;
509 	else if (!strcmp(appearance, "destructive"))
510 		type = 3;
511 	else {
512 		revert();
513 		errx(1, "argument to -c must be normal, blink or destructive");
514 	}
515 
516 	if (ioctl(0, CONS_CURSORTYPE, &type) == -1) {
517 		revert();
518 		errc(1, errno, "setting cursor type");
519 	}
520 }
521 
522 
523 /*
524  * Set the video mode.
525  */
526 
527 static void
528 video_mode(int argc, char **argv, int *mode_index)
529 {
530 	static struct {
531 		const char *name;
532 		unsigned long mode;
533 		unsigned long mode_num;
534 	} modes[] = {{ "80x25",        SW_TEXT_80x25,   M_TEXT_80x25 },
535 		     { "80x30",        SW_TEXT_80x30,   M_TEXT_80x30 },
536 		     { "80x43",        SW_TEXT_80x43,   M_TEXT_80x43 },
537 		     { "80x50",        SW_TEXT_80x50,   M_TEXT_80x50 },
538 		     { "80x60",        SW_TEXT_80x60,   M_TEXT_80x60 },
539 		     { "132x25",       SW_TEXT_132x25,  M_TEXT_132x25 },
540 		     { "132x30",       SW_TEXT_132x30,  M_TEXT_132x30 },
541 		     { "132x43",       SW_TEXT_132x43,  M_TEXT_132x43 },
542 		     { "132x50",       SW_TEXT_132x50,  M_TEXT_132x50 },
543 		     { "132x60",       SW_TEXT_132x60,  M_TEXT_132x60 },
544 		     { "VGA_40x25",    SW_VGA_C40x25,   M_VGA_C40x25 },
545 		     { "VGA_80x25",    SW_VGA_C80x25,   M_VGA_C80x25 },
546 		     { "VGA_80x30",    SW_VGA_C80x30,   M_VGA_C80x30 },
547 		     { "VGA_80x50",    SW_VGA_C80x50,   M_VGA_C80x50 },
548 		     { "VGA_80x60",    SW_VGA_C80x60,   M_VGA_C80x60 },
549 #ifdef SW_VGA_C90x25
550 		     { "VGA_90x25",    SW_VGA_C90x25,   M_VGA_C90x25 },
551 		     { "VGA_90x30",    SW_VGA_C90x30,   M_VGA_C90x30 },
552 		     { "VGA_90x43",    SW_VGA_C90x43,   M_VGA_C90x43 },
553 		     { "VGA_90x50",    SW_VGA_C90x50,   M_VGA_C90x50 },
554 		     { "VGA_90x60",    SW_VGA_C90x60,   M_VGA_C90x60 },
555 #endif
556 		     { "VGA_320x200",  SW_VGA_CG320,    M_CG320 },
557 		     { "EGA_80x25",    SW_ENH_C80x25,   M_ENH_C80x25 },
558 		     { "EGA_80x43",    SW_ENH_C80x43,   M_ENH_C80x43 },
559 		     { "VESA_132x25",  SW_VESA_C132x25, M_VESA_C132x25 },
560 		     { "VESA_132x43",  SW_VESA_C132x43, M_VESA_C132x43 },
561 		     { "VESA_132x50",  SW_VESA_C132x50, M_VESA_C132x50 },
562 		     { "VESA_132x60",  SW_VESA_C132x60, M_VESA_C132x60 },
563 		     { "VESA_800x600", SW_VESA_800x600, M_VESA_800x600 },
564 		     { NULL, NULL, NULL },
565 	};
566 
567 	int new_mode_num = 0;
568 	unsigned long mode = 0;
569 	int size[3];
570 	int i;
571 
572 	/*
573 	 * Parse the video mode argument...
574 	 */
575 
576 	if (*mode_index < argc) {
577 		if (!strncmp(argv[*mode_index], "MODE_", 5)) {
578 			if (!isnumber(argv[*mode_index][5]))
579 				errx(1, "invalid video mode number");
580 
581 			new_mode_num = atoi(&argv[*mode_index][5]);
582 		} else {
583 			for (i = 0; modes[i].name != NULL; ++i) {
584 				if (!strcmp(argv[*mode_index], modes[i].name)) {
585 					mode = modes[i].mode;
586 					new_mode_num = modes[i].mode_num;
587 					break;
588 				}
589 			}
590 
591 			if (modes[i].name == NULL)
592 				errx(1, "invalid video mode name");
593 		}
594 
595 		/*
596 		 * Collect enough information about the new video mode...
597 		 */
598 
599 		new_mode_info.vi_mode = new_mode_num;
600 
601 		if (ioctl(0, CONS_MODEINFO, &new_mode_info) == -1) {
602 			revert();
603 			errc(1, errno, "obtaining new video mode parameters");
604 		}
605 
606 		if (mode == 0) {
607 			if (new_mode_num >= M_VESA_BASE)
608 				mode = _IO('V', new_mode_num - M_VESA_BASE);
609 			else
610 				mode = _IO('S', new_mode_num);
611 		}
612 
613 		/*
614 		 * Try setting the new mode.
615 		 */
616 
617 		if (ioctl(0, mode, NULL) == -1) {
618 			revert();
619 			errc(1, errno, "setting video mode");
620 		}
621 
622 		/*
623 		 * For raster modes it's not enough to just set the mode.
624 		 * We also need to explicitly set the raster mode.
625 		 */
626 
627 		if (new_mode_info.vi_flags & V_INFO_GRAPHICS) {
628 			/* font size */
629 
630 			if (font_height == 0)
631 				font_height = cur_info.console_info.font_size;
632 
633 			size[2] = font_height;
634 
635 			/* adjust columns */
636 
637 			if ((vesa_cols * 8 > new_mode_info.vi_width) ||
638 			    (vesa_cols <= 0)) {
639 				size[0] = new_mode_info.vi_width / 8;
640 			} else {
641 				size[0] = vesa_cols;
642 			}
643 
644 			/* adjust rows */
645 
646 			if ((vesa_rows * font_height > new_mode_info.vi_height) ||
647 			    (vesa_rows <= 0)) {
648 				size[1] = new_mode_info.vi_height /
649 					  font_height;
650 			} else {
651 				size[1] = vesa_rows;
652 			}
653 
654 			/* set raster mode */
655 
656 			if (ioctl(0, KDRASTER, size)) {
657 				revert();
658 				errc(1, errno, "activating raster display");
659 			}
660 		}
661 
662 		video_mode_changed = 1;
663 
664 		(*mode_index)++;
665 	}
666 }
667 
668 
669 /*
670  * Return the number for a specified color name.
671  */
672 
673 static int
674 get_color_number(char *color)
675 {
676 	int i;
677 
678 	for (i=0; i<16; i++) {
679 		if (!strcmp(color, legal_colors[i]))
680 			return i;
681 	}
682 	return -1;
683 }
684 
685 
686 /*
687  * Get normal text and background colors.
688  */
689 
690 static void
691 get_normal_colors(int argc, char **argv, int *color_index)
692 {
693 	int color;
694 
695 	if (*color_index < argc &&
696 	    (color = get_color_number(argv[*color_index])) != -1) {
697 		(*color_index)++;
698 		normal_fore_color = color;
699 		colors_changed = 1;
700 
701 		if (*color_index < argc &&
702 		    (color = get_color_number(argv[*color_index])) != -1) {
703 			(*color_index)++;
704 			normal_back_color = color;
705 		}
706 	}
707 }
708 
709 
710 /*
711  * Get reverse text and background colors.
712  */
713 
714 static void
715 get_reverse_colors(int argc, char **argv, int *color_index)
716 {
717 	int color;
718 
719 	if ((color = get_color_number(argv[*(color_index)-1])) != -1) {
720 		revers_fore_color = color;
721 		colors_changed = 1;
722 
723 		if (*color_index < argc &&
724 		    (color = get_color_number(argv[*color_index])) != -1) {
725 			(*color_index)++;
726 			revers_back_color = color;
727 		}
728 	}
729 }
730 
731 
732 /*
733  * Set normal and reverse foreground and background colors.
734  */
735 
736 static void
737 set_colors(void)
738 {
739 	fprintf(stderr, "[=%dF", normal_fore_color);
740 	fprintf(stderr, "[=%dG", normal_back_color);
741 	fprintf(stderr, "[=%dH", revers_fore_color);
742 	fprintf(stderr, "[=%dI", revers_back_color);
743 }
744 
745 
746 /*
747  * Switch to virtual terminal #arg.
748  */
749 
750 static void
751 set_console(char *arg)
752 {
753 	int n;
754 
755 	if(!arg || strspn(arg,"0123456789") != strlen(arg)) {
756 		revert();
757 		errx(1, "bad console number");
758 	}
759 
760 	n = atoi(arg);
761 
762 	if (n < 1 || n > 16) {
763 		revert();
764 		errx(1, "console number out of range");
765 	} else if (ioctl(0, VT_ACTIVATE, (caddr_t) (long) n) == -1) {
766 		revert();
767 		errc(1, errno, "switching vty");
768 	}
769 }
770 
771 
772 /*
773  * Sets the border color.
774  */
775 
776 static void
777 set_border_color(char *arg)
778 {
779 	int color;
780 
781 	if ((color = get_color_number(arg)) != -1)
782 		fprintf(stderr, "[=%dA", color);
783 	else
784 		usage();
785 }
786 
787 
788 static void
789 set_mouse_char(char *arg)
790 {
791 	struct mouse_info mouse;
792 	long l;
793 
794 	l = strtol(arg, NULL, 0);
795 
796 	if ((l < 0) || (l > UCHAR_MAX)) {
797 		revert();
798 		errx(1, "argument to -M must be 0 through %d", UCHAR_MAX);
799 	}
800 
801 	mouse.operation = MOUSE_MOUSECHAR;
802 	mouse.u.mouse_char = (int)l;
803 
804 	if (ioctl(0, CONS_MOUSECTL, &mouse) == -1) {
805 		revert();
806 		errc(1, errno, "setting mouse character");
807 	}
808 }
809 
810 
811 /*
812  * Show/hide the mouse.
813  */
814 
815 static void
816 set_mouse(char *arg)
817 {
818 	struct mouse_info mouse;
819 
820 	if (!strcmp(arg, "on")) {
821 		mouse.operation = MOUSE_SHOW;
822 	} else if (!strcmp(arg, "off")) {
823 		mouse.operation = MOUSE_HIDE;
824 	} else {
825 		revert();
826 		errx(1, "argument to -m must be either on or off");
827 	}
828 
829 	if (ioctl(0, CONS_MOUSECTL, &mouse) == -1) {
830 		revert();
831 		errc(1, errno, "%sing the mouse",
832 		     mouse.operation == MOUSE_SHOW ? "show" : "hid");
833 	}
834 }
835 
836 
837 static void
838 set_lockswitch(char *arg)
839 {
840 	int data;
841 
842 	if (!strcmp(arg, "off")) {
843 		data = 0x01;
844 	} else if (!strcmp(arg, "on")) {
845 		data = 0x02;
846 	} else {
847 		revert();
848 		errx(1, "argument to -S must be either on or off");
849 	}
850 
851 	if (ioctl(0, VT_LOCKSWITCH, &data) == -1) {
852 		revert();
853 		errc(1, errno, "turning %s vty switching",
854 		     data == 0x01 ? "off" : "on");
855 	}
856 }
857 
858 
859 /*
860  * Return the adapter name for a specified type.
861  */
862 
863 static const char *
864 adapter_name(int type)
865 {
866 	static struct {
867 		int type;
868 		const char *name;
869 	} names[] = {{ KD_MONO,     "MDA" },
870 		     { KD_HERCULES, "Hercules" },
871 		     { KD_CGA,      "CGA" },
872 		     { KD_EGA,      "EGA" },
873 		     { KD_VGA,      "VGA" },
874 		     { KD_TGA,      "TGA" },
875 		     { -1,          "Unknown" },
876 	};
877 
878 	int i;
879 
880 	for (i = 0; names[i].type != -1; ++i) {
881 		if (names[i].type == type)
882 			break;
883 	}
884 
885 	return names[i].name;
886 }
887 
888 
889 /*
890  * Show graphics adapter information.
891  */
892 
893 static void
894 show_adapter_info(void)
895 {
896 	struct video_adapter_info ad;
897 
898 	ad.va_index = 0;
899 
900 	if (ioctl(0, CONS_ADPINFO, &ad) == -1) {
901 		revert();
902 		errc(1, errno, "obtaining adapter information");
903 	}
904 
905 	printf("fb%d:\n", ad.va_index);
906 	printf("    %.*s%d, type:%s%s (%d), flags:0x%x\n",
907 	       (int)sizeof(ad.va_name), ad.va_name, ad.va_unit,
908 	       (ad.va_flags & V_ADP_VESA) ? "VESA " : "",
909 	       adapter_name(ad.va_type), ad.va_type, ad.va_flags);
910 	printf("    initial mode:%d, current mode:%d, BIOS mode:%d\n",
911 	       ad.va_initial_mode, ad.va_mode, ad.va_initial_bios_mode);
912 	printf("    frame buffer window:0x%x, buffer size:0x%x\n",
913 	       ad.va_window, ad.va_buffer_size);
914 	printf("    window size:0x%x, origin:0x%x\n",
915 	       ad.va_window_size, ad.va_window_orig);
916 	printf("    display start address (%d, %d), scan line width:%d\n",
917 	       ad.va_disp_start.x, ad.va_disp_start.y, ad.va_line_width);
918 	printf("    reserved:0x%x\n", ad.va_unused0);
919 }
920 
921 
922 /*
923  * Show video mode information.
924  */
925 
926 static void
927 show_mode_info(void)
928 {
929 	struct video_info vinfo;
930 	char buf[80];
931 	int mode;
932 	int c;
933 
934 	printf("    mode#     flags   type    size       "
935 	       "font      window      linear buffer\n");
936 	printf("---------------------------------------"
937 	       "---------------------------------------\n");
938 
939 	for (mode = 0; mode < M_VESA_MODE_MAX; ++mode) {
940 		vinfo.vi_mode = mode;
941 
942 		if (ioctl(0, CONS_MODEINFO, &vinfo))
943 			continue;
944 		if (vinfo.vi_mode != mode)
945 			continue;
946 
947 		printf("%3d (0x%03x)", mode, mode);
948 		printf(" 0x%08x", vinfo.vi_flags);
949 
950 		if (vinfo.vi_flags & V_INFO_GRAPHICS) {
951 			c = 'G';
952 
953 			snprintf(buf, sizeof(buf), "%dx%dx%d %d",
954 				 vinfo.vi_width, vinfo.vi_height,
955 				 vinfo.vi_depth, vinfo.vi_planes);
956 		} else {
957 			c = 'T';
958 
959 			snprintf(buf, sizeof(buf), "%dx%d",
960 				 vinfo.vi_width, vinfo.vi_height);
961 		}
962 
963 		printf(" %c %-15s", c, buf);
964 		snprintf(buf, sizeof(buf), "%dx%d",
965 			 vinfo.vi_cwidth, vinfo.vi_cheight);
966 		printf(" %-5s", buf);
967 		printf(" 0x%05x %2dk %2dk",
968 		       vinfo.vi_window, (int)vinfo.vi_window_size / 1024,
969 		       (int)vinfo.vi_window_gran/1024);
970 		printf(" 0x%08x %dk\n",
971 		       vinfo.vi_buffer, (int)vinfo.vi_buffer_size / 1024);
972 	}
973 }
974 
975 
976 static void
977 show_info(char *arg)
978 {
979 	if (!strcmp(arg, "adapter")) {
980 		show_adapter_info();
981 	} else if (!strcmp(arg, "mode")) {
982 		show_mode_info();
983 	} else {
984 		revert();
985 		errx(1, "argument to -i must be either adapter or mode");
986 	}
987 }
988 
989 
990 static void
991 test_frame(void)
992 {
993 	int i;
994 
995 	fprintf(stdout, "[=0G\n\n");
996 
997 	for (i = 0; i < 8; i++) {
998 		fprintf(stdout, "[=15F[=0G        %2d [=%dF%-16s"
999 			"[=15F[=0G        %2d [=%dF%-16s        "
1000 			"[=15F %2d [=%dGBACKGROUND[=0G\n",
1001 			i, i, legal_colors[i], i+8, i+8,
1002 			legal_colors[i+8], i, i);
1003 	}
1004 
1005 	fprintf(stdout, "[=%dF[=%dG[=%dH[=%dI\n",
1006 		info.mv_norm.fore, info.mv_norm.back,
1007 		info.mv_rev.fore, info.mv_rev.back);
1008 }
1009 
1010 
1011 /*
1012  * Snapshot the video memory of that terminal, using the CONS_SCRSHOT
1013  * ioctl, and writes the results to stdout either in the special
1014  * binary format (see manual page for details), or in the plain
1015  * text format.
1016  */
1017 
1018 static void
1019 dump_screen(int mode)
1020 {
1021 	scrshot_t shot;
1022 	vid_info_t vinfo;
1023 
1024 	vinfo.size = sizeof(vinfo);
1025 
1026 	if (ioctl(0, CONS_GETINFO, &vinfo) == -1) {
1027 		revert();
1028 		errc(1, errno, "obtaining current video mode parameters");
1029 	}
1030 
1031 	shot.buf = alloca(vinfo.mv_csz * vinfo.mv_rsz * sizeof(u_int16_t));
1032 
1033 	if (shot.buf == NULL) {
1034 		revert();
1035 		errx(1, "failed to allocate memory for dump");
1036 	}
1037 
1038 	shot.xsize = vinfo.mv_csz;
1039 	shot.ysize = vinfo.mv_rsz;
1040 
1041 	if (ioctl(0, CONS_SCRSHOT, &shot) == -1) {
1042 		revert();
1043 		errc(1, errno, "dumping screen");
1044 	}
1045 
1046 	if (mode == DUMP_RAW) {
1047 		printf("SCRSHOT_%c%c%c%c", DUMP_FMT_REV, 2,
1048 		       shot.xsize, shot.ysize);
1049 
1050 		fflush(stdout);
1051 
1052 		write(STDOUT_FILENO, shot.buf,
1053 		      shot.xsize * shot.ysize * sizeof(u_int16_t));
1054 	} else {
1055 		char *line;
1056 		int x, y;
1057 		u_int16_t ch;
1058 
1059 		line = alloca(shot.xsize + 1);
1060 
1061 		if (line == NULL) {
1062 			revert();
1063 			errx(1, "failed to allocate memory for line buffer");
1064 		}
1065 
1066 		for (y = 0; y < shot.ysize; y++) {
1067 			for (x = 0; x < shot.xsize; x++) {
1068 				ch = shot.buf[x + (y * shot.xsize)];
1069 				ch &= 0xff;
1070 
1071 				if (isprint(ch) == 0)
1072 					ch = ' ';
1073 
1074 				line[x] = (char)ch;
1075 			}
1076 
1077 			/* Trim trailing spaces */
1078 
1079 			do {
1080 				line[x--] = '\0';
1081 			} while (line[x] == ' ' && x != 0);
1082 
1083 			puts(line);
1084 		}
1085 
1086 		fflush(stdout);
1087 	}
1088 }
1089 
1090 
1091 /*
1092  * Set the console history buffer size.
1093  */
1094 
1095 static void
1096 set_history(char *opt)
1097 {
1098 	int size;
1099 
1100 	size = atoi(opt);
1101 
1102 	if ((*opt == '\0') || size < 0) {
1103 		revert();
1104 		errx(1, "argument must be a positive number");
1105 	}
1106 
1107 	if (ioctl(0, CONS_HISTORY, &size) == -1) {
1108 		revert();
1109 		errc(1, errno, "setting history buffer size");
1110 	}
1111 }
1112 
1113 
1114 /*
1115  * Clear the console history buffer.
1116  */
1117 
1118 static void
1119 clear_history(void)
1120 {
1121 	if (ioctl(0, CONS_CLRHIST) == -1) {
1122 		revert();
1123 		errc(1, errno, "clearing history buffer");
1124 	}
1125 }
1126 
1127 
1128 int
1129 main(int argc, char **argv)
1130 {
1131 	char *font, *type;
1132 	int opt;
1133 
1134 	if (argc == 1)
1135 		usage();
1136 
1137 	init();
1138 
1139 	info.size = sizeof(info);
1140 
1141 	if (ioctl(0, CONS_GETINFO, &info) == -1)
1142 		err(1, "must be on a virtual console");
1143 
1144 	while((opt = getopt(argc, argv, "b:Cc:df:g:h:i:l:LM:m:pPr:S:s:t:x")) != -1) {
1145 		switch(opt) {
1146 		case 'b':
1147 			set_border_color(optarg);
1148 			break;
1149 		case 'C':
1150 			clear_history();
1151 			break;
1152 		case 'c':
1153 			set_cursor_type(optarg);
1154 			break;
1155 		case 'd':
1156 			print_scrnmap();
1157 			break;
1158 		case 'f':
1159 			type = optarg;
1160 			font = nextarg(argc, argv, &optind, 'f', 0);
1161 
1162 			if (font == NULL) {
1163 				type = NULL;
1164 				font = optarg;
1165 			}
1166 
1167 			load_font(type, font);
1168 			break;
1169 		case 'g':
1170 			if (sscanf(optarg, "%dx%d",
1171 			    &vesa_cols, &vesa_rows) != 2) {
1172 				revert();
1173 				warnx("incorrect geometry: %s", optarg);
1174 				usage();
1175 			}
1176                 	break;
1177 		case 'h':
1178 			set_history(optarg);
1179 			break;
1180 		case 'i':
1181 			show_info(optarg);
1182 			break;
1183 		case 'l':
1184 			load_scrnmap(optarg);
1185 			break;
1186 		case 'L':
1187 			load_default_scrnmap();
1188 			break;
1189 		case 'M':
1190 			set_mouse_char(optarg);
1191 			break;
1192 		case 'm':
1193 			set_mouse(optarg);
1194 			break;
1195 		case 'p':
1196 			dump_screen(DUMP_RAW);
1197 			break;
1198 		case 'P':
1199 			dump_screen(DUMP_TXT);
1200 			break;
1201 		case 'r':
1202 			get_reverse_colors(argc, argv, &optind);
1203 			break;
1204 		case 'S':
1205 			set_lockswitch(optarg);
1206 			break;
1207 		case 's':
1208 			set_console(optarg);
1209 			break;
1210 		case 't':
1211 			set_screensaver_timeout(optarg);
1212 			break;
1213 		case 'x':
1214 			hex = 1;
1215 			break;
1216 		default:
1217 			usage();
1218 		}
1219 	}
1220 
1221 	if (optind < argc && !strcmp(argv[optind], "show")) {
1222 		test_frame();
1223 		optind++;
1224 	}
1225 
1226 	video_mode(argc, argv, &optind);
1227 
1228 	get_normal_colors(argc, argv, &optind);
1229 
1230 	if (colors_changed || video_mode_changed) {
1231 		if (!(new_mode_info.vi_flags & V_INFO_GRAPHICS)) {
1232 			if ((normal_back_color < 8) && (revers_back_color < 8)) {
1233 				set_colors();
1234 			} else {
1235 				revert();
1236 				errx(1, "bg color for text modes must be < 8");
1237 			}
1238 		} else {
1239 			set_colors();
1240 		}
1241 	}
1242 
1243 	if ((optind != argc) || (argc == 1))
1244 		usage();
1245 
1246 	return 0;
1247 }
1248