nrelease - fix/improve livecd

[dragonfly.git] / usr.sbin / vidcontrol / vidcontrol.c

/*-
 * Copyright (c) 1994-1996 Søren Schmidt
 * All rights reserved.
 *
 * 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.
 *
 * $FreeBSD: src/usr.sbin/vidcontrol/vidcontrol.c,v 1.32.2.7 2002/09/15 22:31:50 dd Exp $
 */

#include <machine/console.h>

#include <sys/consio.h>
#include <sys/errno.h>
#include <sys/types.h>
#include <sys/stat.h>

#include <ctype.h>
#include <err.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "path.h"
#include "decode.h"


#define	DATASIZE(x)	((x).w * (x).h * 256 / 8)

#define	DUMP_RAW	0
#define	DUMP_TXT	1

#define	DUMP_FMT_REV	1


char legal_colors[16][16] = {
	"black", "blue", "green", "cyan",
	"red", "magenta", "brown", "white",
	"grey", "lightblue", "lightgreen", "lightcyan",
	"lightred", "lightmagenta", "yellow", "lightwhite"
};

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;

int hex = 0;
int number;
int g_cols;
int g_rows;
int font_height;
int colors_changed;
int video_mode_changed;
int normal_fore_color, normal_back_color;
int revers_fore_color, revers_back_color;
char letter;
struct vid_info info;
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");

	if (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, (caddr_t) (long) cur_info.active_vty);

	fprintf(stderr, "\e[=%dA", cur_info.console_info.mv_ovscan);
	fprintf(stderr, "\e[=%dF", cur_info.console_info.mv_norm.fore);
	fprintf(stderr, "\e[=%dG", cur_info.console_info.mv_norm.back);
	fprintf(stderr, "\e[=%dH", cur_info.console_info.mv_rev.fore);
	fprintf(stderr, "\e[=%dI", cur_info.console_info.mv_rev.back);

	ioctl(0, PIO_SCRNMAP, &cur_info.screen_map);
	ioctl(0, CONS_SET, &cur_info.video_mode_number);

	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)
{
	fprintf(stderr,
		"usage: vidcontrol [-CdLPpx] [-b color] [-c appearance]"
		" [-f [size] file]\n"
		"                  [-g geometry] [-h size] [-i adapter | mode]"
		" [-l screen_map]\n"
		"                  [-M char] [-m on | off] [-r foreground"
		" background]\n"
		"                  [-S on | off] [-s number] [-t N | off]"
		" [mode]\n"
		"                  [foreground [background]] [show]\n");

	exit(1);
}


/*
 * 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(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) {
			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;
}


/*
 * Load a font from file and set it.
 */

static void
load_font(char *type, 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 *b[] = {filename, NULL};
	const char *c[] = {"", size_sufx, NULL};
	const char *d[] = {"", ".fnt", NULL};
	vid_info_t vinfo;

	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}};

	vinfo.size = sizeof(vinfo);

	if (ioctl(0, CONS_GETINFO, &vinfo) == -1) {
		revert();
		errc(1, errno, "obtaining current video mode parameters");
	}

	snprintf(size_sufx, sizeof(size_sufx), "-8x%hu", vinfo.font_size);

	fd = openguess(a, b, c, d, &name);

	if (fd == NULL) {
		revert();
		errx(1, "%s: can't load font file", filename);
	}

	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) {
			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 void
video_mode(int argc, char **argv, int *mode_index)
{
	static struct {
		const char *name;
		unsigned long mode_num;
	} modes[] = {{ "80x25",		M_VGA_C80x25 },
		     { "80x30",		M_VGA_C80x30 },
		     { "80x43",		M_ENH_C80x43 },
		     { "80x50",		M_VGA_C80x50 },
		     { "80x60",		M_VGA_C80x60 },
		     { "VGA_40x25",	M_VGA_C40x25 },
		     { "VGA_80x25",	M_VGA_C80x25 },
		     { "VGA_80x30",	M_VGA_C80x30 },
		     { "VGA_80x50",	M_VGA_C80x50 },
		     { "VGA_80x60",	M_VGA_C80x60 },
#ifdef SW_VGA_C90x25
		     { "VGA_90x25",	M_VGA_C90x25 },
		     { "VGA_90x30",	M_VGA_C90x30 },
		     { "VGA_90x43",	M_VGA_C90x43 },
		     { "VGA_90x50",	M_VGA_C90x50 },
		     { "VGA_90x60",	M_VGA_C90x60 },
#endif
		     { "VGA_320x200",	M_CG320 },
		     { "EGA_80x25",	M_ENH_C80x25 },
		     { "EGA_80x43",	M_ENH_C80x43 },
		     { NULL, 0 },
	};

	int new_mode_num = 0;
	int size[3];
	int i;

	/*
	 * 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)) {
					new_mode_num = modes[i].mode_num;
					break;
				}
			}

			if (modes[i].name == NULL)
				errx(1, "invalid video mode name");
		}

		/*
		 * 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");
		}

		/*
		 * Try setting the new mode.
		 */

		if (ioctl(0, CONS_SET, &new_mode_num) == -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 ((g_cols * 8 > new_mode_info.vi_width) ||
			    (g_cols <= 0)) {
				size[0] = new_mode_info.vi_width / 8;
			} else {
				size[0] = g_cols;
			}

			/* adjust rows */

			if ((g_rows * font_height > new_mode_info.vi_height) ||
			    (g_rows <= 0)) {
				size[1] = new_mode_info.vi_height /
					  font_height;
			} else {
				size[1] = g_rows;
			}

			/* set raster mode */

			if (ioctl(0, KDRASTER, size)) {
				revert();
				errc(1, errno, "activating raster display");
			}
		}

		video_mode_changed = 1;

		(*mode_index)++;
	}
}


/*
 * 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 *color_index)
{
	int color;

	if (*color_index < argc &&
	    (color = get_color_number(argv[*color_index])) != -1) {
		(*color_index)++;
		normal_fore_color = color;
		colors_changed = 1;
        
		if (*color_index < argc &&
		    (color = get_color_number(argv[*color_index])) != -1) {
			(*color_index)++;
			normal_back_color = color;            
		}
	}
}


/*
 * Get reverse text and background colors.
 */

static void
get_reverse_colors(int argc, char **argv, int *color_index)
{
	int color;

	if ((color = get_color_number(argv[*(color_index)-1])) != -1) {
		revers_fore_color = color;
		colors_changed = 1;

		if (*color_index < argc &&
		    (color = get_color_number(argv[*color_index])) != -1) {
			(*color_index)++;
			revers_back_color = color;            
		}
	}
}


/*
 * Set normal and reverse foreground and background colors.
 */

static void
set_colors(void)
{
	fprintf(stderr, "\e[=%dF", normal_fore_color);
	fprintf(stderr, "\e[=%dG", normal_back_color);
	fprintf(stderr, "\e[=%dH", revers_fore_color);
	fprintf(stderr, "\e[=%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, (caddr_t) (long) 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, "\e[=%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)) {
		revert();
		errx(1, "argument to -M must be 0 through %d", UCHAR_MAX);
	}

	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");
	}
	ioctl(0, CONS_MOUSECTL, &mouse);
}


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_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 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 (%d), flags:0x%x\n",
	       (int)sizeof(ad.va_name), ad.va_name, ad.va_unit,
	       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%x, 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%x\n", ad.va_unused0);
}


/*
 * Show video mode information.
 */

static void
show_mode_info(void)
{
	struct video_info vinfo;
	char buf[80];
	int mode;
	int c;

	printf("    mode#     flags   type    size       "
	       "font      window      linear buffer\n");
	printf("---------------------------------------"
	       "---------------------------------------\n");

	for (mode = 0; mode <= M_MCA_MODE; ++mode) {
		vinfo.vi_mode = mode;

		if (ioctl(0, CONS_MODEINFO, &vinfo))
			continue;
		if (vinfo.vi_mode != mode)
			continue;

		printf("%3d (0x%03x)", mode, mode);
		printf(" 0x%08x", vinfo.vi_flags);

		if (vinfo.vi_flags & V_INFO_GRAPHICS) {
			c = 'G';

			snprintf(buf, sizeof(buf), "%dx%dx%d %d",
				 vinfo.vi_width, vinfo.vi_height,
				 vinfo.vi_depth, vinfo.vi_planes);
		} else {
			c = 'T';

			snprintf(buf, sizeof(buf), "%dx%d",
				 vinfo.vi_width, vinfo.vi_height);
		}

		printf(" %c %-15s", c, buf);
		snprintf(buf, sizeof(buf), "%dx%d",
			 vinfo.vi_cwidth, vinfo.vi_cheight);
		printf(" %-5s", buf);
		printf(" 0x%05x %2dk %2dk",
		       vinfo.vi_window, (int)vinfo.vi_window_size / 1024,
		       (int)vinfo.vi_window_gran/1024);
		printf(" 0x%08x %dk\n",
		       vinfo.vi_buffer, (int)vinfo.vi_buffer_size / 1024);
	}
}


static void
show_info(char *arg)
{
	if (!strcmp(arg, "adapter")) {
		show_adapter_info();
	} else if (!strcmp(arg, "mode")) {
		show_mode_info();
	} else {
		revert();
		errx(1, "argument to -i must be either adapter or mode");
	}
}


static void
test_frame(void)
{
	int i;

	fprintf(stdout, "\e[=0G\n\n");

	for (i = 0; i < 8; i++) {
		fprintf(stdout, "\e[=15F\e[=0G        %2d \e[=%dF%-16s"
			"\e[=15F\e[=0G        %2d \e[=%dF%-16s        "
			"\e[=15F %2d \e[=%dGBACKGROUND\e[=0G\n",
			i, i, legal_colors[i], i+8, i+8,
			legal_colors[i+8], i, i);
	}

	fprintf(stdout, "\e[=%dF\e[=%dG\e[=%dH\e[=%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)
{
	scrshot_t shot;
	vid_info_t vinfo;

	vinfo.size = sizeof(vinfo);

	if (ioctl(0, CONS_GETINFO, &vinfo) == -1) {
		revert();
		errc(1, errno, "obtaining current video mode parameters");
	}

	shot.buf = alloca(vinfo.mv_csz * vinfo.mv_rsz * sizeof(u_int16_t));

	if (shot.buf == NULL) {
		revert();
		errx(1, "failed to allocate memory for dump");
	}

	shot.xsize = vinfo.mv_csz;
	shot.ysize = vinfo.mv_rsz;

	if (ioctl(0, CONS_SCRSHOT, &shot) == -1) {
		revert();
		errc(1, errno, "dumping screen");
	}

	if (mode == DUMP_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");
	}
}


int
main(int argc, char **argv)
{
	char *font, *type;
	int opt;

	if (argc == 1)
		usage();

	init();

	info.size = sizeof(info);

	if (ioctl(0, CONS_GETINFO, &info) == -1)
		err(1, "must be on a virtual console");

	while((opt = getopt(argc, argv, "b:Cc:df:g:h:i:l:LM:m:pPr:S:s:t:x")) != -1) {
		switch(opt) {
		case 'b':
			set_border_color(optarg);
			break;
		case 'C':
			clear_history();
			break;
		case 'c':
			set_cursor_type(optarg);
			break;
		case 'd':
			print_scrnmap();
			break;
		case 'f':
			type = optarg;
			font = nextarg(argc, argv, &optind, 'f', 0);

			if (font == NULL) {
				type = NULL;
				font = optarg;
			}

			load_font(type, font);
			break;
		case 'g':
			if (sscanf(optarg, "%dx%d", &g_cols, &g_rows) != 2) {
				revert();
				warnx("incorrect geometry: %s", optarg);
				usage();
			}
                	break;
		case 'h':
			set_history(optarg);
			break;
		case 'i':
			show_info(optarg);
			break;
		case 'l':
			load_scrnmap(optarg);
			break;
		case 'L':
			load_default_scrnmap();
			break;
		case 'M':
			set_mouse_char(optarg);
			break;
		case 'm':
			set_mouse(optarg);
			break;
		case 'p':
			dump_screen(DUMP_RAW);
			break;
		case 'P':
			dump_screen(DUMP_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':
			set_screensaver_timeout(optarg);
			break;
		case 'x':
			hex = 1;
			break;
		default:
			usage();
		}
	}

	if (optind < argc && !strcmp(argv[optind], "show")) {
		test_frame();
		optind++;
	}

	video_mode(argc, argv, &optind);

	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 0;
}