Index: head/usr.sbin/vidcontrol/vidcontrol.c
===================================================================
--- head/usr.sbin/vidcontrol/vidcontrol.c	(revision 308311)
+++ head/usr.sbin/vidcontrol/vidcontrol.c	(revision 308312)
@@ -1,1499 +1,1507 @@
 /*-
  * Copyright (c) 1994-1996 Søren Schmidt
  * All rights reserved.
  *
  * Portions of this software are based in part on the work of
  * Sascha Wildner <saw@online.de> contributed to The DragonFly Project
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer,
  *    in this position and unchanged.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
  * 3. The name of the author may not be used to endorse or promote products
  *    derived from this software without specific prior written permission
  *
  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  *
  * $DragonFly: src/usr.sbin/vidcontrol/vidcontrol.c,v 1.10 2005/03/02 06:08:29 joerg Exp $
  */
 
 #ifndef lint
 static const char rcsid[] =
   "$FreeBSD$";
 #endif /* not lint */
 
 #include <ctype.h>
 #include <err.h>
 #include <limits.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 #include <sys/fbio.h>
 #include <sys/consio.h>
 #include <sys/endian.h>
 #include <sys/errno.h>
 #include <sys/param.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <sys/sysctl.h>
 #include "path.h"
 #include "decode.h"
 
 
 #define	DATASIZE(x)	((x).w * (x).h * 256 / 8)
 
 /* Screen dump modes */
 #define DUMP_FMT_RAW	1
 #define DUMP_FMT_TXT	2
 /* Screen dump options */
 #define DUMP_FBF	0
 #define DUMP_ALL	1
 /* Screen dump file format revision */
 #define DUMP_FMT_REV	1
 
 static const char *legal_colors[16] = {
 	"black", "blue", "green", "cyan",
 	"red", "magenta", "brown", "white",
 	"grey", "lightblue", "lightgreen", "lightcyan",
 	"lightred", "lightmagenta", "yellow", "lightwhite"
 };
 
 static struct {
 	int			active_vty;
 	vid_info_t		console_info;
 	unsigned char		screen_map[256];
 	int			video_mode_number;
 	struct video_info	video_mode_info;
 } cur_info;
 
 struct vt4font_header {
 	uint8_t		magic[8];
 	uint8_t		width;
 	uint8_t		height;
 	uint16_t	pad;
 	uint32_t	glyph_count;
 	uint32_t	map_count[4];
 } __packed;
 
 static int	hex = 0;
 static int	vesa_cols;
 static int	vesa_rows;
 static int	font_height;
 static int	colors_changed;
 static int	video_mode_changed;
 static int	normal_fore_color, normal_back_color;
 static int	revers_fore_color, revers_back_color;
 static int	vt4_mode = 0;
 static struct	vid_info info;
 static struct	video_info new_mode_info;
 
 
 /*
  * Initialize revert data.
  *
  * NOTE: the following parameters are not yet saved/restored:
  *
  *   screen saver timeout
  *   cursor type
  *   mouse character and mouse show/hide state
  *   vty switching on/off state
  *   history buffer size
  *   history contents
  *   font maps
  */
 
 static void
 init(void)
 {
 	if (ioctl(0, VT_GETACTIVE, &cur_info.active_vty) == -1)
 		errc(1, errno, "getting active vty");
 
 	cur_info.console_info.size = sizeof(cur_info.console_info);
 
 	if (ioctl(0, CONS_GETINFO, &cur_info.console_info) == -1)
 		errc(1, errno, "getting console information");
 
 	/* vt(4) use unicode, so no screen mapping required. */
 	if (vt4_mode == 0 &&
 	    ioctl(0, GIO_SCRNMAP, &cur_info.screen_map) == -1)
 		errc(1, errno, "getting screen map");
 
 	if (ioctl(0, CONS_GET, &cur_info.video_mode_number) == -1)
 		errc(1, errno, "getting video mode number");
 
 	cur_info.video_mode_info.vi_mode = cur_info.video_mode_number;
 
 	if (ioctl(0, CONS_MODEINFO, &cur_info.video_mode_info) == -1)
 		errc(1, errno, "getting video mode parameters");
 
 	normal_fore_color = cur_info.console_info.mv_norm.fore;
 	normal_back_color = cur_info.console_info.mv_norm.back;
 	revers_fore_color = cur_info.console_info.mv_rev.fore;
 	revers_back_color = cur_info.console_info.mv_rev.back;
 }
 
 
 /*
  * If something goes wrong along the way we call revert() to go back to the
  * console state we came from (which is assumed to be working).
  *
  * NOTE: please also read the comments of init().
  */
 
 static void
 revert(void)
 {
 	int size[3];
 
 	ioctl(0, VT_ACTIVATE, cur_info.active_vty);
 
 	fprintf(stderr, "\033[=%dA", cur_info.console_info.mv_ovscan);
 	fprintf(stderr, "\033[=%dF", cur_info.console_info.mv_norm.fore);
 	fprintf(stderr, "\033[=%dG", cur_info.console_info.mv_norm.back);
 	fprintf(stderr, "\033[=%dH", cur_info.console_info.mv_rev.fore);
 	fprintf(stderr, "\033[=%dI", cur_info.console_info.mv_rev.back);
 
 	if (vt4_mode == 0)
 		ioctl(0, PIO_SCRNMAP, &cur_info.screen_map);
 
 	if (cur_info.video_mode_number >= M_VESA_BASE)
 		ioctl(0, _IO('V', cur_info.video_mode_number - M_VESA_BASE),
 		      NULL);
 	else
 		ioctl(0, _IO('S', cur_info.video_mode_number), NULL);
 
 	if (cur_info.video_mode_info.vi_flags & V_INFO_GRAPHICS) {
 		size[0] = cur_info.video_mode_info.vi_width / 8;
 		size[1] = cur_info.video_mode_info.vi_height /
 			  cur_info.console_info.font_size;
 		size[2] = cur_info.console_info.font_size;
 
 		ioctl(0, KDRASTER, size);
 	}
 }
 
 
 /*
  * Print a short usage string describing all options, then exit.
  */
 
 static void
 usage(void)
 {
 	if (vt4_mode)
 		fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n",
 "usage: vidcontrol [-CHPpx] [-b color] [-c appearance] [-f [[size] file]]",
 "                  [-g geometry] [-h size] [-i active | adapter | mode]",
 "                  [-M char] [-m on | off] [-r foreground background]",
 "                  [-S on | off] [-s number] [-T xterm | cons25] [-t N | off]",
 "                  [mode] [foreground [background]] [show]");
 	else
 		fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n",
 "usage: vidcontrol [-CdHLPpx] [-b color] [-c appearance] [-f [size] file]",
 "                  [-g geometry] [-h size] [-i adapter | mode] [-l screen_map]",
 "                  [-M char] [-m on | off] [-r foreground background]",
 "                  [-S on | off] [-s number] [-T xterm | cons25] [-t N | off]",
 "                  [mode] [foreground [background]] [show]");
 	exit(1);
 }
 
 /* Detect presence of vt(4). */
 static int
 is_vt4(void)
 {
 	char vty_name[4] = "";
 	size_t len = sizeof(vty_name);
 
 	if (sysctlbyname("kern.vty", vty_name, &len, NULL, 0) != 0)
 		return (0);
 	return (strcmp(vty_name, "vt") == 0);
 }
 
 /*
  * Retrieve the next argument from the command line (for options that require
  * more than one argument).
  */
 
 static char *
 nextarg(int ac, char **av, int *indp, int oc, int strict)
 {
 	if (*indp < ac)
 		return(av[(*indp)++]);
 
 	if (strict != 0) {
 		revert();
 		errx(1, "option requires two arguments -- %c", oc);
 	}
 
 	return(NULL);
 }
 
 
 /*
  * Guess which file to open. Try to open each combination of a specified set
  * of file name components.
  */
 
 static FILE *
 openguess(const char *a[], const char *b[], const char *c[], const char *d[], char **name)
 {
 	FILE *f;
 	int i, j, k, l;
 
 	for (i = 0; a[i] != NULL; i++) {
 		for (j = 0; b[j] != NULL; j++) {
 			for (k = 0; c[k] != NULL; k++) {
 				for (l = 0; d[l] != NULL; l++) {
 					asprintf(name, "%s%s%s%s",
 						 a[i], b[j], c[k], d[l]);
 
 					f = fopen(*name, "r");
 
 					if (f != NULL)
 						return (f);
 
 					free(*name);
 				}
 			}
 		}
 	}
 	return (NULL);
 }
 
 
 /*
  * Load a screenmap from a file and set it.
  */
 
 static void
 load_scrnmap(const char *filename)
 {
 	FILE *fd;
 	int size;
 	char *name;
 	scrmap_t scrnmap;
 	const char *a[] = {"", SCRNMAP_PATH, NULL};
 	const char *b[] = {filename, NULL};
 	const char *c[] = {"", ".scm", NULL};
 	const char *d[] = {"", NULL};
 
 	fd = openguess(a, b, c, d, &name);
 
 	if (fd == NULL) {
 		revert();
 		errx(1, "screenmap file not found");
 	}
 
 	size = sizeof(scrnmap);
 
 	if (decode(fd, (char *)&scrnmap, size) != size) {
 		rewind(fd);
 
 		if (fread(&scrnmap, 1, size, fd) != (size_t)size) {
 			warnx("bad screenmap file");
 			fclose(fd);
 			revert();
 			errx(1, "bad screenmap file");
 		}
 	}
 
 	if (ioctl(0, PIO_SCRNMAP, &scrnmap) == -1) {
 		revert();
 		errc(1, errno, "loading screenmap");
 	}
 
 	fclose(fd);
 }
 
 
 /*
  * Set the default screenmap.
  */
 
 static void
 load_default_scrnmap(void)
 {
 	scrmap_t scrnmap;
 	int i;
 
 	for (i=0; i<256; i++)
 		*((char*)&scrnmap + i) = i;
 
 	if (ioctl(0, PIO_SCRNMAP, &scrnmap) == -1) {
 		revert();
 		errc(1, errno, "loading default screenmap");
 	}
 }
 
 
 /*
  * Print the current screenmap to stdout.
  */
 
 static void
 print_scrnmap(void)
 {
 	unsigned char map[256];
 	size_t i;
 
 	if (ioctl(0, GIO_SCRNMAP, &map) == -1) {
 		revert();
 		errc(1, errno, "getting screenmap");
 	}
 	for (i=0; i<sizeof(map); i++) {
 		if (i != 0 && i % 16 == 0)
 			fprintf(stdout, "\n");
 
 		if (hex != 0)
 			fprintf(stdout, " %02x", map[i]);
 		else
 			fprintf(stdout, " %03d", map[i]);
 	}
 	fprintf(stdout, "\n");
 
 }
 
 
 /*
  * Determine a file's size.
  */
 
 static int
 fsize(FILE *file)
 {
 	struct stat sb;
 
 	if (fstat(fileno(file), &sb) == 0)
 		return sb.st_size;
 	else
 		return -1;
 }
 
 static vfnt_map_t *
 load_vt4mappingtable(unsigned int nmappings, FILE *f)
 {
 	vfnt_map_t *t;
 	unsigned int i;
 
 	if (nmappings == 0)
 		return (NULL);
 
-	t = malloc(sizeof *t * nmappings);
+	if ((t = malloc(sizeof *t * nmappings)) == NULL) {
+		warn("malloc");
+		return (NULL);
+	}
 
 	if (fread(t, sizeof *t * nmappings, 1, f) != 1) {
-		perror("mappings");
-		exit(1);
+		warn("read mappings");
+		free(t);
+		return (NULL);
 	}
 
 	for (i = 0; i < nmappings; i++) {
 		t[i].src = be32toh(t[i].src);
 		t[i].dst = be16toh(t[i].dst);
 		t[i].len = be16toh(t[i].len);
 	}
 
 	return (t);
 }
 
 /*
  * Set the default vt font.
  */
 
 static void
 load_default_vt4font(void)
 {
 	if (ioctl(0, PIO_VFONT_DEFAULT) == -1) {
 		revert();
 		errc(1, errno, "loading default vt font");
 	}
 }
 
-static int
+static void
 load_vt4font(FILE *f)
 {
 	struct vt4font_header fh;
 	static vfnt_t vfnt;
 	size_t glyphsize;
 	unsigned int i;
 
 	if (fread(&fh, sizeof fh, 1, f) != 1) {
-		perror("file_header");
-		return (1);
+		warn("read file_header");
+		return;
 	}
 
 	if (memcmp(fh.magic, "VFNT0002", 8) != 0) {
-		fprintf(stderr, "Bad magic\n");
-		return (1);
+		warnx("bad magic in font file\n");
+		return;
 	}
 
 	for (i = 0; i < VFNT_MAPS; i++)
 		vfnt.map_count[i] = be32toh(fh.map_count[i]);
 	vfnt.glyph_count = be32toh(fh.glyph_count);
 	vfnt.width = fh.width;
 	vfnt.height = fh.height;
 
 	glyphsize = howmany(vfnt.width, 8) * vfnt.height * vfnt.glyph_count;
-	vfnt.glyphs = malloc(glyphsize);
+	if ((vfnt.glyphs = malloc(glyphsize)) == NULL) {
+		warn("malloc");
+		return;
+	}
 
 	if (fread(vfnt.glyphs, glyphsize, 1, f) != 1) {
-		perror("glyphs");
-		return (1);
+		warn("read glyphs");
+		free(vfnt.glyphs);
+		return;
 	}
 
 	for (i = 0; i < VFNT_MAPS; i++)
 		vfnt.map[i] = load_vt4mappingtable(vfnt.map_count[i], f);
 
-	if (ioctl(STDIN_FILENO, PIO_VFONT, &vfnt) == -1) {
-		perror("PIO_VFONT");
-		return (1);
-	}
-	return (0);
+	if (ioctl(STDIN_FILENO, PIO_VFONT, &vfnt) == -1)
+		warn("PIO_VFONT");
+
+	for (i = 0; i < VFNT_MAPS; i++)
+		free(vfnt.map[i]);
+	free(vfnt.glyphs);
 }
 
 /*
  * Load a font from file and set it.
  */
 
 static void
 load_font(const char *type, const char *filename)
 {
 	FILE	*fd;
 	int	h, i, size, w;
 	unsigned long io = 0;	/* silence stupid gcc(1) in the Wall mode */
 	char	*name, *fontmap, size_sufx[6];
 	const char	*a[] = {"", FONT_PATH, NULL};
 	const char	*vt4a[] = {"", VT_FONT_PATH, NULL};
 	const char	*b[] = {filename, NULL};
 	const char	*c[] = {"", size_sufx, NULL};
 	const char	*d[] = {"", ".fnt", NULL};
 	vid_info_t _info;
 
 	struct sizeinfo {
 		int w;
 		int h;
 		unsigned long io;
 	} sizes[] = {{8, 16, PIO_FONT8x16},
 		     {8, 14, PIO_FONT8x14},
 		     {8,  8,  PIO_FONT8x8},
 		     {0,  0,            0}};
 
 	if (vt4_mode) {
 		size_sufx[0] = '\0';
 	} else {
 		_info.size = sizeof(_info);
 		if (ioctl(0, CONS_GETINFO, &_info) == -1) {
 			revert();
 			warn("failed to obtain current video mode parameters");
 			return;
 		}
 
 		snprintf(size_sufx, sizeof(size_sufx), "-8x%d", _info.font_size);
 	}
 	fd = openguess((vt4_mode == 0) ? a : vt4a, b, c, d, &name);
 
 	if (fd == NULL) {
 		revert();
 		errx(1, "%s: can't load font file", filename);
 	}
 
 	if (vt4_mode) {
-		if(load_vt4font(fd))
-			warn("failed to load font \"%s\"", filename);
+		load_vt4font(fd);
 		fclose(fd);
 		return;
 	}
 
 	if (type != NULL) {
 		size = 0;
 		if (sscanf(type, "%dx%d", &w, &h) == 2) {
 			for (i = 0; sizes[i].w != 0; i++) {
 				if (sizes[i].w == w && sizes[i].h == h) {
 					size = DATASIZE(sizes[i]);
 					io = sizes[i].io;
 					font_height = sizes[i].h;
 				}
 			}
 		}
 		if (size == 0) {
 			fclose(fd);
 			revert();
 			errx(1, "%s: bad font size specification", type);
 		}
 	} else {
 		/* Apply heuristics */
 
 		int j;
 		int dsize[2];
 
 		size = DATASIZE(sizes[0]);
 		fontmap = (char*) malloc(size);
 		dsize[0] = decode(fd, fontmap, size);
 		dsize[1] = fsize(fd);
 		free(fontmap);
 
 		size = 0;
 		for (j = 0; j < 2; j++) {
 			for (i = 0; sizes[i].w != 0; i++) {
 				if (DATASIZE(sizes[i]) == dsize[j]) {
 					size = dsize[j];
 					io = sizes[i].io;
 					font_height = sizes[i].h;
 					j = 2;	/* XXX */
 					break;
 				}
 			}
 		}
 
 		if (size == 0) {
 			fclose(fd);
 			revert();
 			errx(1, "%s: can't guess font size", filename);
 		}
 
 		rewind(fd);
 	}
 
 	fontmap = (char*) malloc(size);
 
 	if (decode(fd, fontmap, size) != size) {
 		rewind(fd);
 		if (fsize(fd) != size ||
 		    fread(fontmap, 1, size, fd) != (size_t)size) {
 			warnx("%s: bad font file", filename);
 			fclose(fd);
 			free(fontmap);
 			revert();
 			errx(1, "%s: bad font file", filename);
 		}
 	}
 
 	if (ioctl(0, io, fontmap) == -1) {
 		revert();
 		errc(1, errno, "loading font");
 	}
 
 	fclose(fd);
 	free(fontmap);
 }
 
 
 /*
  * Set the timeout for the screensaver.
  */
 
 static void
 set_screensaver_timeout(char *arg)
 {
 	int nsec;
 
 	if (!strcmp(arg, "off")) {
 		nsec = 0;
 	} else {
 		nsec = atoi(arg);
 
 		if ((*arg == '\0') || (nsec < 1)) {
 			revert();
 			errx(1, "argument must be a positive number");
 		}
 	}
 
 	if (ioctl(0, CONS_BLANKTIME, &nsec) == -1) {
 		revert();
 		errc(1, errno, "setting screensaver period");
 	}
 }
 
 
 /*
  * Set the cursor's shape/type.
  */
 
 static void
 set_cursor_type(char *appearance)
 {
 	int type;
 
 	if (!strcmp(appearance, "normal"))
 		type = 0;
 	else if (!strcmp(appearance, "blink"))
 		type = 1;
 	else if (!strcmp(appearance, "destructive"))
 		type = 3;
 	else {
 		revert();
 		errx(1, "argument to -c must be normal, blink or destructive");
 	}
 
 	if (ioctl(0, CONS_CURSORTYPE, &type) == -1) {
 		revert();
 		errc(1, errno, "setting cursor type");
 	}
 }
 
 
 /*
  * Set the video mode.
  */
 
 static int
 video_mode(int argc, char **argv, int *mode_index)
 {
 	static struct {
 		const char *name;
 		unsigned long mode;
 		unsigned long mode_num;
 	} modes[] = {
 		{ "80x25",        SW_TEXT_80x25,   M_TEXT_80x25 },
 		{ "80x30",        SW_TEXT_80x30,   M_TEXT_80x30 },
 		{ "80x43",        SW_TEXT_80x43,   M_TEXT_80x43 },
 		{ "80x50",        SW_TEXT_80x50,   M_TEXT_80x50 },
 		{ "80x60",        SW_TEXT_80x60,   M_TEXT_80x60 },
 		{ "132x25",       SW_TEXT_132x25,  M_TEXT_132x25 },
 		{ "132x30",       SW_TEXT_132x30,  M_TEXT_132x30 },
 		{ "132x43",       SW_TEXT_132x43,  M_TEXT_132x43 },
 		{ "132x50",       SW_TEXT_132x50,  M_TEXT_132x50 },
 		{ "132x60",       SW_TEXT_132x60,  M_TEXT_132x60 },
 		{ "VGA_40x25",    SW_VGA_C40x25,   M_VGA_C40x25 },
 		{ "VGA_80x25",    SW_VGA_C80x25,   M_VGA_C80x25 },
 		{ "VGA_80x30",    SW_VGA_C80x30,   M_VGA_C80x30 },
 		{ "VGA_80x50",    SW_VGA_C80x50,   M_VGA_C80x50 },
 		{ "VGA_80x60",    SW_VGA_C80x60,   M_VGA_C80x60 },
 #ifdef SW_VGA_C90x25
 		{ "VGA_90x25",    SW_VGA_C90x25,   M_VGA_C90x25 },
 		{ "VGA_90x30",    SW_VGA_C90x30,   M_VGA_C90x30 },
 		{ "VGA_90x43",    SW_VGA_C90x43,   M_VGA_C90x43 },
 		{ "VGA_90x50",    SW_VGA_C90x50,   M_VGA_C90x50 },
 		{ "VGA_90x60",    SW_VGA_C90x60,   M_VGA_C90x60 },
 #endif
 		{ "VGA_320x200",	SW_VGA_CG320,	M_CG320 },
 		{ "EGA_80x25",		SW_ENH_C80x25,	M_ENH_C80x25 },
 		{ "EGA_80x43",		SW_ENH_C80x43,	M_ENH_C80x43 },
 		{ "VESA_132x25",	SW_VESA_C132x25,M_VESA_C132x25 },
 		{ "VESA_132x43",	SW_VESA_C132x43,M_VESA_C132x43 },
 		{ "VESA_132x50",	SW_VESA_C132x50,M_VESA_C132x50 },
 		{ "VESA_132x60",	SW_VESA_C132x60,M_VESA_C132x60 },
 		{ "VESA_800x600",	SW_VESA_800x600,M_VESA_800x600 },
 		{ NULL, 0, 0 },
 	};
 
 	int new_mode_num = 0;
 	unsigned long mode = 0;
 	int cur_mode; 
 	int ioerr;
 	int size[3];
 	int i;
 
 	if (ioctl(0, CONS_GET, &cur_mode) < 0)
 		err(1, "cannot get the current video mode");
 
 	/*
 	 * Parse the video mode argument...
 	 */
 
 	if (*mode_index < argc) {
 		if (!strncmp(argv[*mode_index], "MODE_", 5)) {
 			if (!isdigit(argv[*mode_index][5]))
 				errx(1, "invalid video mode number");
 
 			new_mode_num = atoi(&argv[*mode_index][5]);
 		} else {
 			for (i = 0; modes[i].name != NULL; ++i) {
 				if (!strcmp(argv[*mode_index], modes[i].name)) {
 					mode = modes[i].mode;
 					new_mode_num = modes[i].mode_num;
 					break;
 				}
 			}
 
 			if (modes[i].name == NULL)
 				return EXIT_FAILURE;
 			if (ioctl(0, mode, NULL) < 0) {
 				warn("cannot set videomode");
 				return EXIT_FAILURE;
 			}
 		}
 
 		/*
 		 * Collect enough information about the new video mode...
 		 */
 
 		new_mode_info.vi_mode = new_mode_num;
 
 		if (ioctl(0, CONS_MODEINFO, &new_mode_info) == -1) {
 			revert();
 			errc(1, errno, "obtaining new video mode parameters");
 		}
 
 		if (mode == 0) {
 			if (new_mode_num >= M_VESA_BASE)
 				mode = _IO('V', new_mode_num - M_VESA_BASE);
 			else
 				mode = _IO('S', new_mode_num);
 		}
 
 		/*
 		 * Try setting the new mode.
 		 */
 
 		if (ioctl(0, mode, NULL) == -1) {
 			revert();
 			errc(1, errno, "setting video mode");
 		}
 
 		/*
 		 * For raster modes it's not enough to just set the mode.
 		 * We also need to explicitly set the raster mode.
 		 */
 
 		if (new_mode_info.vi_flags & V_INFO_GRAPHICS) {
 			/* font size */
 
 			if (font_height == 0)
 				font_height = cur_info.console_info.font_size;
 
 			size[2] = font_height;
 
 			/* adjust columns */
 
 			if ((vesa_cols * 8 > new_mode_info.vi_width) ||
 			    (vesa_cols <= 0)) {
 				size[0] = new_mode_info.vi_width / 8;
 			} else {
 				size[0] = vesa_cols;
 			}
 
 			/* adjust rows */
 
 			if ((vesa_rows * font_height > new_mode_info.vi_height) ||
 			    (vesa_rows <= 0)) {
 				size[1] = new_mode_info.vi_height /
 					  font_height;
 			} else {
 				size[1] = vesa_rows;
 			}
 
 			/* set raster mode */
 
 			if (ioctl(0, KDRASTER, size)) {
 				ioerr = errno;
 				if (cur_mode >= M_VESA_BASE)
 					ioctl(0,
 					    _IO('V', cur_mode - M_VESA_BASE),
 					    NULL);
 				else
 					ioctl(0, _IO('S', cur_mode), NULL);
 				revert();
 				warnc(ioerr, "cannot activate raster display");
 				return EXIT_FAILURE;
 			}
 		}
 
 		video_mode_changed = 1;
 
 		(*mode_index)++;
 	}
 	return EXIT_SUCCESS;
 }
 
 
 /*
  * Return the number for a specified color name.
  */
 
 static int
 get_color_number(char *color)
 {
 	int i;
 
 	for (i=0; i<16; i++) {
 		if (!strcmp(color, legal_colors[i]))
 			return i;
 	}
 	return -1;
 }
 
 
 /*
  * Get normal text and background colors.
  */
 
 static void
 get_normal_colors(int argc, char **argv, int *_index)
 {
 	int color;
 
 	if (*_index < argc && (color = get_color_number(argv[*_index])) != -1) {
 		(*_index)++;
 		fprintf(stderr, "\033[=%dF", color);
 		normal_fore_color=color;
 		colors_changed = 1;
 		if (*_index < argc
 		    && (color = get_color_number(argv[*_index])) != -1
 		    && color < 8) {
 			(*_index)++;
 			fprintf(stderr, "\033[=%dG", color);
 			normal_back_color=color;
 		}
 	}
 }
 
 
 /*
  * Get reverse text and background colors.
  */
 
 static void
 get_reverse_colors(int argc, char **argv, int *_index)
 {
 	int color;
 
 	if ((color = get_color_number(argv[*(_index)-1])) != -1) {
 		fprintf(stderr, "\033[=%dH", color);
 		revers_fore_color=color;
 		colors_changed = 1;
 		if (*_index < argc
 		    && (color = get_color_number(argv[*_index])) != -1
 		    && color < 8) {
 			(*_index)++;
 			fprintf(stderr, "\033[=%dI", color);
 			revers_back_color=color;
 		}
 	}
 }
 
 
 /*
  * Set normal and reverse foreground and background colors.
  */
 
 static void
 set_colors(void)
 {
 	fprintf(stderr, "\033[=%dF", normal_fore_color);
 	fprintf(stderr, "\033[=%dG", normal_back_color);
 	fprintf(stderr, "\033[=%dH", revers_fore_color);
 	fprintf(stderr, "\033[=%dI", revers_back_color);
 }
 
 
 /*
  * Switch to virtual terminal #arg.
  */
 
 static void
 set_console(char *arg)
 {
 	int n;
 
 	if(!arg || strspn(arg,"0123456789") != strlen(arg)) {
 		revert();
 		errx(1, "bad console number");
 	}
 
 	n = atoi(arg);
 
 	if (n < 1 || n > 16) {
 		revert();
 		errx(1, "console number out of range");
 	} else if (ioctl(0, VT_ACTIVATE, n) == -1) {
 		revert();
 		errc(1, errno, "switching vty");
 	}
 }
 
 
 /*
  * Sets the border color.
  */
 
 static void
 set_border_color(char *arg)
 {
 	int color;
 
 	if ((color = get_color_number(arg)) != -1) {
 		fprintf(stderr, "\033[=%dA", color);
 	}
 	else
 		usage();
 }
 
 static void
 set_mouse_char(char *arg)
 {
 	struct mouse_info mouse;
 	long l;
 
 	l = strtol(arg, NULL, 0);
 
 	if ((l < 0) || (l > UCHAR_MAX - 3)) {
 		revert();
 		warnx("argument to -M must be 0 through %d", UCHAR_MAX - 3);
 		return;
 	}
 
 	mouse.operation = MOUSE_MOUSECHAR;
 	mouse.u.mouse_char = (int)l;
 
 	if (ioctl(0, CONS_MOUSECTL, &mouse) == -1) {
 		revert();
 		errc(1, errno, "setting mouse character");
 	}
 }
 
 
 /*
  * Show/hide the mouse.
  */
 
 static void
 set_mouse(char *arg)
 {
 	struct mouse_info mouse;
 
 	if (!strcmp(arg, "on")) {
 		mouse.operation = MOUSE_SHOW;
 	} else if (!strcmp(arg, "off")) {
 		mouse.operation = MOUSE_HIDE;
 	} else {
 		revert();
 		errx(1, "argument to -m must be either on or off");
 	}
 
 	if (ioctl(0, CONS_MOUSECTL, &mouse) == -1) {
 		revert();
 		errc(1, errno, "%sing the mouse",
 		     mouse.operation == MOUSE_SHOW ? "show" : "hid");
 	}
 }
 
 
 static void
 set_lockswitch(char *arg)
 {
 	int data;
 
 	if (!strcmp(arg, "off")) {
 		data = 0x01;
 	} else if (!strcmp(arg, "on")) {
 		data = 0x02;
 	} else {
 		revert();
 		errx(1, "argument to -S must be either on or off");
 	}
 
 	if (ioctl(0, VT_LOCKSWITCH, &data) == -1) {
 		revert();
 		errc(1, errno, "turning %s vty switching",
 		     data == 0x01 ? "off" : "on");
 	}
 }
 
 
 /*
  * Return the adapter name for a specified type.
  */
 
 static const char
 *adapter_name(int type)
 {
     static struct {
 	int type;
 	const char *name;
     } names[] = {
 	{ KD_MONO,	"MDA" },
 	{ KD_HERCULES,	"Hercules" },
 	{ KD_CGA,	"CGA" },
 	{ KD_EGA,	"EGA" },
 	{ KD_VGA,	"VGA" },
 	{ KD_PC98,	"PC-98xx" },
 	{ KD_TGA,	"TGA" },
 	{ -1,		"Unknown" },
     };
 
     int i;
 
     for (i = 0; names[i].type != -1; ++i)
 	if (names[i].type == type)
 	    break;
     return names[i].name;
 }
 
 
 /*
  * Show active VTY, ie current console number.
  */
 
 static void
 show_active_info(void)
 {
 
 	printf("%d\n", cur_info.active_vty);
 }
 
 
 /*
  * Show graphics adapter information.
  */
 
 static void
 show_adapter_info(void)
 {
 	struct video_adapter_info ad;
 
 	ad.va_index = 0;
 
 	if (ioctl(0, CONS_ADPINFO, &ad) == -1) {
 		revert();
 		errc(1, errno, "obtaining adapter information");
 	}
 
 	printf("fb%d:\n", ad.va_index);
 	printf("    %.*s%d, type:%s%s (%d), flags:0x%x\n",
 	       (int)sizeof(ad.va_name), ad.va_name, ad.va_unit,
 	       (ad.va_flags & V_ADP_VESA) ? "VESA " : "",
 	       adapter_name(ad.va_type), ad.va_type, ad.va_flags);
 	printf("    initial mode:%d, current mode:%d, BIOS mode:%d\n",
 	       ad.va_initial_mode, ad.va_mode, ad.va_initial_bios_mode);
 	printf("    frame buffer window:0x%zx, buffer size:0x%zx\n",
 	       ad.va_window, ad.va_buffer_size);
 	printf("    window size:0x%zx, origin:0x%x\n",
 	       ad.va_window_size, ad.va_window_orig);
 	printf("    display start address (%d, %d), scan line width:%d\n",
 	       ad.va_disp_start.x, ad.va_disp_start.y, ad.va_line_width);
 	printf("    reserved:0x%zx\n", ad.va_unused0);
 }
 
 
 /*
  * Show video mode information.
  */
 
 static void
 show_mode_info(void)
 {
 	char buf[80];
 	struct video_info _info;
 	int c;
 	int mm;
 	int mode;
 
 	printf("    mode#     flags   type    size       "
 	       "font      window      linear buffer\n");
 	printf("---------------------------------------"
 	       "---------------------------------------\n");
 
 	memset(&_info, 0, sizeof(_info));
 	for (mode = 0; mode <= M_VESA_MODE_MAX; ++mode) {
 		_info.vi_mode = mode;
 		if (ioctl(0, CONS_MODEINFO, &_info))
 			continue;
 		if (_info.vi_mode != mode)
 			continue;
 		if (_info.vi_width == 0 && _info.vi_height == 0 &&
 		    _info.vi_cwidth == 0 && _info.vi_cheight == 0)
 			continue;
 
 		printf("%3d (0x%03x)", mode, mode);
     		printf(" 0x%08x", _info.vi_flags);
 		if (_info.vi_flags & V_INFO_GRAPHICS) {
 			c = 'G';
 
 			if (_info.vi_mem_model == V_INFO_MM_PLANAR)
 				snprintf(buf, sizeof(buf), "%dx%dx%d %d",
 				    _info.vi_width, _info.vi_height, 
 				    _info.vi_depth, _info.vi_planes);
 			else {
 				switch (_info.vi_mem_model) {
 				case V_INFO_MM_PACKED:
 					mm = 'P';
 					break;
 				case V_INFO_MM_DIRECT:
 					mm = 'D';
 					break;
 				case V_INFO_MM_CGA:
 					mm = 'C';
 					break;
 				case V_INFO_MM_HGC:
 					mm = 'H';
 					break;
 				case V_INFO_MM_VGAX:
 					mm = 'V';
 					break;
 				default:
 					mm = ' ';
 					break;
 				}
 				snprintf(buf, sizeof(buf), "%dx%dx%d %c",
 				    _info.vi_width, _info.vi_height, 
 				    _info.vi_depth, mm);
 			}
 		} else {
 			c = 'T';
 
 			snprintf(buf, sizeof(buf), "%dx%d",
 				 _info.vi_width, _info.vi_height);
 		}
 
 		printf(" %c %-15s", c, buf);
 		snprintf(buf, sizeof(buf), "%dx%d", 
 			 _info.vi_cwidth, _info.vi_cheight); 
 		printf(" %-5s", buf);
     		printf(" 0x%05zx %2dk %2dk", 
 		       _info.vi_window, (int)_info.vi_window_size/1024, 
 		       (int)_info.vi_window_gran/1024);
     		printf(" 0x%08zx %dk\n",
 		       _info.vi_buffer, (int)_info.vi_buffer_size/1024);
 	}
 }
 
 
 static void
 show_info(char *arg)
 {
 
 	if (!strcmp(arg, "active")) {
 		show_active_info();
 	} else if (!strcmp(arg, "adapter")) {
 		show_adapter_info();
 	} else if (!strcmp(arg, "mode")) {
 		show_mode_info();
 	} else {
 		revert();
 		errx(1, "argument to -i must be active, adapter, or mode");
 	}
 }
 
 
 static void
 test_frame(void)
 {
 	int i, cur_mode, fore;
 
 	fore = 15;
 
 	if (ioctl(0, CONS_GET, &cur_mode) < 0)
 		err(1, "must be on a virtual console");
 	switch (cur_mode) {
 	case M_PC98_80x25:
 	case M_PC98_80x30:
 		fore = 7;
 		break;
 	}
 
 	fprintf(stdout, "\033[=0G\n\n");
 	for (i=0; i<8; i++) {
 		fprintf(stdout, "\033[=%dF\033[=0G        %2d \033[=%dF%-16s"
 				"\033[=%dF\033[=0G        %2d \033[=%dF%-16s        "
 				"\033[=%dF %2d \033[=%dGBACKGROUND\033[=0G\n",
 			fore, i, i, legal_colors[i],
 			fore, i+8, i+8, legal_colors[i+8],
 			fore, i, i);
 	}
 	fprintf(stdout, "\033[=%dF\033[=%dG\033[=%dH\033[=%dI\n",
 		info.mv_norm.fore, info.mv_norm.back,
 		info.mv_rev.fore, info.mv_rev.back);
 }
 
 
 /*
  * Snapshot the video memory of that terminal, using the CONS_SCRSHOT
  * ioctl, and writes the results to stdout either in the special
  * binary format (see manual page for details), or in the plain
  * text format.
  */
 
 static void
 dump_screen(int mode, int opt)
 {
 	scrshot_t shot;
 	vid_info_t _info;
 
 	_info.size = sizeof(_info);
 
 	if (ioctl(0, CONS_GETINFO, &_info) == -1) {
 		revert();
 		errc(1, errno, "obtaining current video mode parameters");
 		return;
 	}
 
 	shot.x = shot.y = 0;
 	shot.xsize = _info.mv_csz;
 	shot.ysize = _info.mv_rsz;
 	if (opt == DUMP_ALL)
 		shot.ysize += _info.mv_hsz;
 
 	shot.buf = alloca(shot.xsize * shot.ysize * sizeof(u_int16_t));
 	if (shot.buf == NULL) {
 		revert();
 		errx(1, "failed to allocate memory for dump");
 	}
 
 	if (ioctl(0, CONS_SCRSHOT, &shot) == -1) {
 		revert();
 		errc(1, errno, "dumping screen");
 	}
 
 	if (mode == DUMP_FMT_RAW) {
 		printf("SCRSHOT_%c%c%c%c", DUMP_FMT_REV, 2,
 		       shot.xsize, shot.ysize);
 
 		fflush(stdout);
 
 		write(STDOUT_FILENO, shot.buf,
 		      shot.xsize * shot.ysize * sizeof(u_int16_t));
 	} else {
 		char *line;
 		int x, y;
 		u_int16_t ch;
 
 		line = alloca(shot.xsize + 1);
 
 		if (line == NULL) {
 			revert();
 			errx(1, "failed to allocate memory for line buffer");
 		}
 
 		for (y = 0; y < shot.ysize; y++) {
 			for (x = 0; x < shot.xsize; x++) {
 				ch = shot.buf[x + (y * shot.xsize)];
 				ch &= 0xff;
 
 				if (isprint(ch) == 0)
 					ch = ' ';
 
 				line[x] = (char)ch;
 			}
 
 			/* Trim trailing spaces */
 
 			do {
 				line[x--] = '\0';
 			} while (line[x] == ' ' && x != 0);
 
 			puts(line);
 		}
 
 		fflush(stdout);
 	}
 }
 
 
 /*
  * Set the console history buffer size.
  */
 
 static void
 set_history(char *opt)
 {
 	int size;
 
 	size = atoi(opt);
 
 	if ((*opt == '\0') || size < 0) {
 		revert();
 		errx(1, "argument must be a positive number");
 	}
 
 	if (ioctl(0, CONS_HISTORY, &size) == -1) {
 		revert();
 		errc(1, errno, "setting history buffer size");
 	}
 }
 
 
 /*
  * Clear the console history buffer.
  */
 
 static void
 clear_history(void)
 {
 	if (ioctl(0, CONS_CLRHIST) == -1) {
 		revert();
 		errc(1, errno, "clearing history buffer");
 	}
 }
 
 static void
 set_terminal_mode(char *arg)
 {
 
 	if (strcmp(arg, "xterm") == 0)
 		fprintf(stderr, "\033[=T");
 	else if (strcmp(arg, "cons25") == 0)
 		fprintf(stderr, "\033[=1T");
 }
 
 
 int
 main(int argc, char **argv)
 {
 	char    *font, *type, *termmode;
 	const char *opts;
 	int	dumpmod, dumpopt, opt;
 	int	reterr;
 
 	vt4_mode = is_vt4();
 
 	init();
 
 	info.size = sizeof(info);
 
 	if (ioctl(0, CONS_GETINFO, &info) == -1)
 		err(1, "must be on a virtual console");
 	dumpmod = 0;
 	dumpopt = DUMP_FBF;
 	termmode = NULL;
 	if (vt4_mode)
 		opts = "b:Cc:fg:h:Hi:M:m:pPr:S:s:T:t:x";
 	else
 		opts = "b:Cc:dfg:h:Hi:l:LM:m:pPr:S:s:T:t:x";
 
 	while ((opt = getopt(argc, argv, opts)) != -1)
 		switch(opt) {
 		case 'b':
 			set_border_color(optarg);
 			break;
 		case 'C':
 			clear_history();
 			break;
 		case 'c':
 			set_cursor_type(optarg);
 			break;
 		case 'd':
 			if (vt4_mode)
 				break;
 			print_scrnmap();
 			break;
 		case 'f':
 			optarg = nextarg(argc, argv, &optind, 'f', 0);
 			if (optarg != NULL) {
 				font = nextarg(argc, argv, &optind, 'f', 0);
 
 				if (font == NULL) {
 					type = NULL;
 					font = optarg;
 				} else
 					type = optarg;
 
 				load_font(type, font);
 			} else {
 				if (!vt4_mode)
 					usage(); /* Switch syscons to ROM? */
 
 				load_default_vt4font();
 			}
 			break;
 		case 'g':
 			if (sscanf(optarg, "%dx%d",
 			    &vesa_cols, &vesa_rows) != 2) {
 				revert();
 				warnx("incorrect geometry: %s", optarg);
 				usage();
 			}
                 	break;
 		case 'h':
 			set_history(optarg);
 			break;
 		case 'H':
 			dumpopt = DUMP_ALL;
 			break;
 		case 'i':
 			show_info(optarg);
 			break;
 		case 'l':
 			if (vt4_mode)
 				break;
 			load_scrnmap(optarg);
 			break;
 		case 'L':
 			if (vt4_mode)
 				break;
 			load_default_scrnmap();
 			break;
 		case 'M':
 			set_mouse_char(optarg);
 			break;
 		case 'm':
 			set_mouse(optarg);
 			break;
 		case 'p':
 			dumpmod = DUMP_FMT_RAW;
 			break;
 		case 'P':
 			dumpmod = DUMP_FMT_TXT;
 			break;
 		case 'r':
 			get_reverse_colors(argc, argv, &optind);
 			break;
 		case 'S':
 			set_lockswitch(optarg);
 			break;
 		case 's':
 			set_console(optarg);
 			break;
 		case 'T':
 			if (strcmp(optarg, "xterm") != 0 &&
 			    strcmp(optarg, "cons25") != 0)
 				usage();
 			termmode = optarg;
 			break;
 		case 't':
 			set_screensaver_timeout(optarg);
 			break;
 		case 'x':
 			hex = 1;
 			break;
 		default:
 			usage();
 		}
 
 	if (dumpmod != 0)
 		dump_screen(dumpmod, dumpopt);
 	reterr = video_mode(argc, argv, &optind);
 	get_normal_colors(argc, argv, &optind);
 
 	if (optind < argc && !strcmp(argv[optind], "show")) {
 		test_frame();
 		optind++;
 	}
 
 	video_mode(argc, argv, &optind);
 	if (termmode != NULL)
 		set_terminal_mode(termmode);
 
 	get_normal_colors(argc, argv, &optind);
 
 	if (colors_changed || video_mode_changed) {
 		if (!(new_mode_info.vi_flags & V_INFO_GRAPHICS)) {
 			if ((normal_back_color < 8) && (revers_back_color < 8)) {
 				set_colors();
 			} else {
 				revert();
 				errx(1, "bg color for text modes must be < 8");
 			}
 		} else {
 			set_colors();
 		}
 	}
 
 	if ((optind != argc) || (argc == 1))
 		usage();
 	return reterr;
 }