vidcontrol.c revision 146237 
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 without 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
29#ifndef lint
30static const char rcsid[] =
31  "$FreeBSD: head/usr.sbin/vidcontrol/vidcontrol.c 146237 2005-05-15 08:04:46Z nyan $";
32#endif /* not lint */
33
34#include <ctype.h>
35#include <err.h>
36#include <limits.h>
37#include <stdio.h>
38#include <stdlib.h>
39#include <string.h>
40#include <unistd.h>
41#include <sys/fbio.h>
42#include <sys/consio.h>
43#include <sys/errno.h>
44#include <sys/types.h>
45#include <sys/stat.h>
46#include "path.h"
47#include "decode.h"
48
49#define _VESA_800x600_DFL_COLS 80
50#define _VESA_800x600_DFL_ROWS 25
51#define _VESA_800x600_DFL_FNSZ 16
52
53/* Screen dump modes */
54#define DUMP_FMT_RAW	1
55#define DUMP_FMT_TXT	2
56/* Screen dump options */
57#define DUMP_FBF	0
58#define DUMP_ALL	1
59/* Screen dump file format revision */
60#define DUMP_FMT_REV	1
61
62char 	legal_colors[16][16] = {
63	"black", "blue", "green", "cyan",
64	"red", "magenta", "brown", "white",
65	"grey", "lightblue", "lightgreen", "lightcyan",
66	"lightred", "lightmagenta", "yellow", "lightwhite"
67};
68int 	hex = 0;
69int 	number;
70int	vesa_cols = _VESA_800x600_DFL_COLS;
71int	vesa_rows = _VESA_800x600_DFL_ROWS;
72char 	letter;
73struct 	vid_info info;
74
75
76static void
77usage(void)
78{
79	fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n",
80"usage: vidcontrol [-CdHLPpx] [-b color] [-c appearance] [-f [size] file]",
81"                  [-g geometry] [-h size] [-i adapter | mode] [-l screen_map]",
82"                  [-M char] [-m on | off] [-r foreground background]",
83"                  [-S on | off] [-s number] [-t N | off] [mode]",
84"                  [foreground [background]] [show]");
85	exit(1);
86}
87
88static char *
89nextarg(int ac, char **av, int *indp, int oc, int strict)
90{
91	if (*indp < ac)
92		return(av[(*indp)++]);
93	if (strict != 0)
94		errx(1, "option requires two arguments -- %c", oc);
95	return(NULL);
96}
97
98static FILE *
99openguess(const char *a[], const char *b[], const char *c[], const char *d[], char **name)
100{
101	FILE *f;
102	int i, j, k, l;
103
104	for (i = 0; a[i] != NULL; i++) {
105		for (j = 0; b[j] != NULL; j++) {
106			for (k = 0; c[k] != NULL; k++) {
107				for (l = 0; d[l] != NULL; l++) {
108					asprintf(name, "%s%s%s%s", a[i], b[j],
109					    c[k], d[l]);
110					f = fopen(*name, "r");
111					if (f != NULL)
112						return (f);
113					free(*name);
114				}
115			}
116		}
117	}
118	return (NULL);
119}
120
121static void
122load_scrnmap(const char *filename)
123{
124	FILE *fd;
125	int size;
126	char *name;
127	scrmap_t scrnmap;
128	const char *a[] = {"", SCRNMAP_PATH, NULL};
129	const char *b[] = {filename, NULL};
130	const char *c[] = {"", ".scm", NULL};
131	const char *d[] = {"", NULL};
132
133	fd = openguess(a, b, c, d, &name);
134	if (fd == NULL) {
135		warn("screenmap file not found");
136		return;
137	}
138	size = sizeof(scrnmap);
139	if (decode(fd, (char *)&scrnmap, size) != size) {
140		rewind(fd);
141		if (fread(&scrnmap, 1, size, fd) != (unsigned)size) {
142			warnx("bad screenmap file");
143			fclose(fd);
144			return;
145		}
146	}
147	if (ioctl(0, PIO_SCRNMAP, &scrnmap) < 0)
148		warn("can't load screenmap");
149	fclose(fd);
150}
151
152static void
153load_default_scrnmap(void)
154{
155	scrmap_t scrnmap;
156	int i;
157
158	for (i=0; i<256; i++)
159		*((char*)&scrnmap + i) = i;
160	if (ioctl(0, PIO_SCRNMAP, &scrnmap) < 0)
161		warn("can't load default screenmap");
162}
163
164static void
165print_scrnmap(void)
166{
167	unsigned char map[256];
168	size_t i;
169
170	if (ioctl(0, GIO_SCRNMAP, &map) < 0) {
171		warn("getting screenmap");
172		return;
173	}
174	for (i=0; i<sizeof(map); i++) {
175		if (i != 0 && i % 16 == 0)
176			fprintf(stdout, "\n");
177		if (hex)
178			fprintf(stdout, " %02x", map[i]);
179		else
180			fprintf(stdout, " %03d", map[i]);
181	}
182	fprintf(stdout, "\n");
183
184}
185
186static int
187fsize(FILE *file)
188{
189	struct stat sb;
190
191	if (fstat(fileno(file), &sb) == 0)
192		return sb.st_size;
193	else
194		return -1;
195}
196
197#define DATASIZE(x) ((x).w * (x).h * 256 / 8)
198
199static void
200load_font(const char *type, const char *filename)
201{
202	FILE	*fd;
203	int	h, i, size, w;
204	unsigned long io = 0;	/* silence stupid gcc(1) in the Wall mode */
205	char	*name, *fontmap, size_sufx[6];
206	const char	*a[] = {"", FONT_PATH, NULL};
207	const char	*b[] = {filename, NULL};
208	const char	*c[] = {"", size_sufx, NULL};
209	const char	*d[] = {"", ".fnt", NULL};
210	vid_info_t _info;
211
212	struct sizeinfo {
213		int w;
214		int h;
215		unsigned long io;
216	} sizes[] = {{8, 16, PIO_FONT8x16},
217		     {8, 14, PIO_FONT8x14},
218		     {8,  8,  PIO_FONT8x8},
219		     {0,  0,            0}};
220
221	_info.size = sizeof(_info);
222	if (ioctl(0, CONS_GETINFO, &_info) == -1) {
223		warn("failed to obtain current video mode parameters");
224		return;
225	}
226	snprintf(size_sufx, sizeof(size_sufx), "-8x%d", _info.font_size);
227	fd = openguess(a, b, c, d, &name);
228	if (fd == NULL) {
229		warn("%s: can't load font file", filename);
230		return;
231	}
232	if (type != NULL) {
233		size = 0;
234		if (sscanf(type, "%dx%d", &w, &h) == 2)
235			for (i = 0; sizes[i].w != 0; i++)
236				if (sizes[i].w == w && sizes[i].h == h) {
237					size = DATASIZE(sizes[i]);
238					io = sizes[i].io;
239				}
240
241		if (size == 0) {
242			warnx("%s: bad font size specification", type);
243			fclose(fd);
244			return;
245		}
246	} else {
247		/* Apply heuristics */
248		int j;
249		int dsize[2];
250
251		size = DATASIZE(sizes[0]);
252		fontmap = (char*) malloc(size);
253		dsize[0] = decode(fd, fontmap, size);
254		dsize[1] = fsize(fd);
255		free(fontmap);
256
257		size = 0;
258		for (j = 0; j < 2; j++)
259			for (i = 0; sizes[i].w != 0; i++)
260				if (DATASIZE(sizes[i]) == dsize[j]) {
261					size = dsize[j];
262					io = sizes[i].io;
263					j = 2;	/* XXX */
264					break;
265				}
266
267		if (size == 0) {
268			warnx("%s: can't guess font size", filename);
269			fclose(fd);
270			return;
271		}
272		rewind(fd);
273	}
274
275	fontmap = (char*) malloc(size);
276	if (decode(fd, fontmap, size) != size) {
277		rewind(fd);
278		if (fsize(fd) != size || fread(fontmap, 1, size, fd) != (unsigned)size) {
279			warnx("%s: bad font file", filename);
280			fclose(fd);
281			free(fontmap);
282			return;
283		}
284	}
285	if (ioctl(0, io, fontmap) < 0)
286		warn("can't load font");
287	fclose(fd);
288	free(fontmap);
289}
290
291static void
292set_screensaver_timeout(char *arg)
293{
294	int nsec;
295
296	if (!strcmp(arg, "off"))
297		nsec = 0;
298	else {
299		nsec = atoi(arg);
300		if ((*arg == '\0') || (nsec < 1)) {
301			warnx("argument must be a positive number");
302			return;
303		}
304	}
305	if (ioctl(0, CONS_BLANKTIME, &nsec) == -1)
306		warn("setting screensaver period");
307}
308
309static void
310set_cursor_type(char *appearence)
311{
312	int type;
313
314	if (!strcmp(appearence, "normal"))
315		type = 0;
316	else if (!strcmp(appearence, "blink"))
317		type = 1;
318	else if (!strcmp(appearence, "destructive"))
319		type = 3;
320	else {
321		warnx("argument to -c must be normal, blink or destructive");
322		return;
323	}
324	ioctl(0, CONS_CURSORTYPE, &type);
325}
326
327static int
328video_mode(int argc, char **argv, int *_index)
329{
330	static struct {
331		const char *name;
332		unsigned long mode;
333	} modes[] = {
334		{ "80x25",		SW_TEXT_80x25 },
335		{ "80x30",		SW_TEXT_80x30 },
336		{ "80x43",		SW_TEXT_80x43 },
337		{ "80x50",		SW_TEXT_80x50 },
338		{ "80x60",		SW_TEXT_80x60 },
339		{ "132x25",		SW_TEXT_132x25 },
340		{ "132x30",		SW_TEXT_132x30 },
341		{ "132x43",		SW_TEXT_132x43 },
342		{ "132x50",		SW_TEXT_132x50 },
343		{ "132x60",		SW_TEXT_132x60 },
344		{ "VGA_40x25",		SW_VGA_C40x25 },
345		{ "VGA_80x25",		SW_VGA_C80x25 },
346		{ "VGA_80x30",		SW_VGA_C80x30 },
347		{ "VGA_80x50",		SW_VGA_C80x50 },
348		{ "VGA_80x60",		SW_VGA_C80x60 },
349#ifdef SW_VGA_C90x25
350		{ "VGA_90x25",		SW_VGA_C90x25 },
351		{ "VGA_90x30",		SW_VGA_C90x30 },
352		{ "VGA_90x43",		SW_VGA_C90x43 },
353		{ "VGA_90x50",		SW_VGA_C90x50 },
354		{ "VGA_90x60",		SW_VGA_C90x60 },
355#endif
356		{ "VGA_320x200",	SW_VGA_CG320 },
357		{ "EGA_80x25",		SW_ENH_C80x25 },
358		{ "EGA_80x43",		SW_ENH_C80x43 },
359		{ "VESA_132x25",	SW_VESA_C132x25 },
360		{ "VESA_132x43",	SW_VESA_C132x43 },
361		{ "VESA_132x50",	SW_VESA_C132x50 },
362		{ "VESA_132x60",	SW_VESA_C132x60 },
363		{ "VESA_800x600",	SW_VESA_800x600 },
364		{ NULL,			0 },
365	};
366	unsigned long mode = 0;
367	int cur_mode;
368	int ioerr;
369	int size[3];
370	int i;
371
372	if (ioctl(0, CONS_GET, &cur_mode) < 0)
373		err(1, "cannot get the current video mode");
374	if (*_index < argc) {
375		for (i = 0; modes[i].name != NULL; ++i) {
376			if (!strcmp(argv[*_index], modes[i].name)) {
377				mode = modes[i].mode;
378				break;
379			}
380		}
381		if (modes[i].name == NULL)
382			return EXIT_FAILURE;
383		if (ioctl(0, mode, NULL) < 0) {
384			warn("cannot set videomode");
385			return EXIT_FAILURE;
386		}
387		if (mode == SW_VESA_800x600) {
388			/* columns */
389			if ((vesa_cols * 8 > 800) || (vesa_cols <= 0)) {
390				warnx("incorrect number of columns: %d",
391				      vesa_cols);
392				size[0] = _VESA_800x600_DFL_COLS;
393			} else {
394				size[0] = vesa_cols;
395			}
396			/* rows */
397			if ((vesa_rows * _VESA_800x600_DFL_FNSZ > 600) ||
398			    (vesa_rows <=0)) {
399				warnx("incorrect number of rows: %d",
400				      vesa_rows);
401				size[1] = _VESA_800x600_DFL_ROWS;
402			} else {
403				size[1] = vesa_rows;
404			}
405			/* font size */
406			size[2] = _VESA_800x600_DFL_FNSZ;
407			if (ioctl(0, KDRASTER, size)) {
408				ioerr = errno;
409				if (cur_mode >= M_VESA_BASE)
410					ioctl(0,
411					    _IO('V', cur_mode - M_VESA_BASE),
412					    NULL);
413				else
414					ioctl(0, _IO('S', cur_mode), NULL);
415				warnc(ioerr, "cannot activate raster display");
416				return EXIT_FAILURE;
417			}
418		}
419		(*_index)++;
420	}
421	return EXIT_SUCCESS;
422}
423
424static int
425get_color_number(char *color)
426{
427	int i;
428
429	for (i=0; i<16; i++)
430		if (!strcmp(color, legal_colors[i]))
431			return i;
432	return -1;
433}
434
435static void
436set_normal_colors(int argc, char **argv, int *_index)
437{
438	int color;
439
440	if (*_index < argc && (color = get_color_number(argv[*_index])) != -1) {
441		(*_index)++;
442		fprintf(stderr, "\e[=%dF", color);
443		if (*_index < argc
444		    && (color = get_color_number(argv[*_index])) != -1
445		    && color < 8) {
446			(*_index)++;
447			fprintf(stderr, "\e[=%dG", color);
448		}
449	}
450}
451
452static void
453set_reverse_colors(int argc, char **argv, int *_index)
454{
455	int color;
456
457	if ((color = get_color_number(argv[*(_index)-1])) != -1) {
458		fprintf(stderr, "\e[=%dH", color);
459		if (*_index < argc
460		    && (color = get_color_number(argv[*_index])) != -1
461		    && color < 8) {
462			(*_index)++;
463			fprintf(stderr, "\e[=%dI", color);
464		}
465	}
466}
467
468static void
469set_console(char *arg)
470{
471	int n;
472
473	if( !arg || strspn(arg,"0123456789") != strlen(arg)) {
474		warnx("bad console number");
475		return;
476	}
477
478	n = atoi(arg);
479	if (n < 1 || n > 16) {
480		warnx("console number out of range");
481	} else if (ioctl(0, VT_ACTIVATE, (caddr_t) (long) n) == -1)
482		warn("ioctl(VT_ACTIVATE)");
483}
484
485static void
486set_border_color(char *arg)
487{
488	int color;
489
490	if ((color = get_color_number(arg)) != -1) {
491		fprintf(stderr, "\e[=%dA", color);
492	}
493	else
494		usage();
495}
496
497static void
498set_mouse_char(char *arg)
499{
500	struct mouse_info mouse;
501	long l;
502
503	l = strtol(arg, NULL, 0);
504	if ((l < 0) || (l > UCHAR_MAX - 3)) {
505		warnx("argument to -M must be 0 through %d", UCHAR_MAX - 3);
506		return;
507	}
508	mouse.operation = MOUSE_MOUSECHAR;
509	mouse.u.mouse_char = (int)l;
510	ioctl(0, CONS_MOUSECTL, &mouse);
511}
512
513static void
514set_mouse(char *arg)
515{
516	struct mouse_info mouse;
517
518	if (!strcmp(arg, "on"))
519		mouse.operation = MOUSE_SHOW;
520	else if (!strcmp(arg, "off"))
521		mouse.operation = MOUSE_HIDE;
522	else {
523		warnx("argument to -m must be either on or off");
524		return;
525	}
526	ioctl(0, CONS_MOUSECTL, &mouse);
527}
528
529static void
530set_lockswitch(char *arg)
531{
532	int data;
533
534	if (!strcmp(arg, "off"))
535		data = 0x01;
536	else if (!strcmp(arg, "on"))
537		data = 0x02;
538	else {
539		warnx("argument to -S must be either on or off");
540		return;
541	}
542	if (ioctl(0, VT_LOCKSWITCH, &data) == -1)
543		warn("ioctl(VT_LOCKSWITCH)");
544}
545
546static const char
547*adapter_name(int type)
548{
549    static struct {
550	int type;
551	const char *name;
552    } names[] = {
553	{ KD_MONO,	"MDA" },
554	{ KD_HERCULES,	"Hercules" },
555	{ KD_CGA,	"CGA" },
556	{ KD_EGA,	"EGA" },
557	{ KD_VGA,	"VGA" },
558	{ KD_PC98,	"PC-98xx" },
559	{ KD_TGA,	"TGA" },
560	{ -1,		"Unknown" },
561    };
562    int i;
563
564    for (i = 0; names[i].type != -1; ++i)
565	if (names[i].type == type)
566	    break;
567    return names[i].name;
568}
569
570static void
571show_adapter_info(void)
572{
573	struct video_adapter_info ad;
574
575	ad.va_index = 0;
576	if (ioctl(0, CONS_ADPINFO, &ad)) {
577		warn("failed to obtain adapter information");
578		return;
579	}
580
581	printf("fb%d:\n", ad.va_index);
582	printf("    %.*s%d, type:%s%s (%d), flags:0x%x\n",
583	       (int)sizeof(ad.va_name), ad.va_name, ad.va_unit,
584	       (ad.va_flags & V_ADP_VESA) ? "VESA " : "",
585	       adapter_name(ad.va_type), ad.va_type, ad.va_flags);
586	printf("    initial mode:%d, current mode:%d, BIOS mode:%d\n",
587	       ad.va_initial_mode, ad.va_mode, ad.va_initial_bios_mode);
588	printf("    frame buffer window:0x%zx, buffer size:0x%zx\n",
589	       ad.va_window, ad.va_buffer_size);
590	printf("    window size:0x%zx, origin:0x%x\n",
591	       ad.va_window_size, ad.va_window_orig);
592	printf("    display start address (%d, %d), scan line width:%d\n",
593	       ad.va_disp_start.x, ad.va_disp_start.y, ad.va_line_width);
594	printf("    reserved:0x%zx\n", ad.va_unused0);
595}
596
597static void
598show_mode_info(void)
599{
600	struct video_info _info;
601	char buf[80];
602	int mode;
603	int c;
604
605	printf("    mode#     flags   type    size       "
606	       "font      window      linear buffer\n");
607	printf("---------------------------------------"
608	       "---------------------------------------\n");
609	for (mode = 0; mode < M_VESA_MODE_MAX; ++mode) {
610		_info.vi_mode = mode;
611		if (ioctl(0, CONS_MODEINFO, &_info))
612			continue;
613		if (_info.vi_mode != mode)
614			continue;
615
616		printf("%3d (0x%03x)", mode, mode);
617    		printf(" 0x%08x", _info.vi_flags);
618		if (_info.vi_flags & V_INFO_GRAPHICS) {
619			c = 'G';
620			snprintf(buf, sizeof(buf), "%dx%dx%d %d",
621				 _info.vi_width, _info.vi_height,
622				 _info.vi_depth, _info.vi_planes);
623		} else {
624			c = 'T';
625			snprintf(buf, sizeof(buf), "%dx%d",
626				 _info.vi_width, _info.vi_height);
627		}
628		printf(" %c %-15s", c, buf);
629		snprintf(buf, sizeof(buf), "%dx%d",
630			 _info.vi_cwidth, _info.vi_cheight);
631		printf(" %-5s", buf);
632    		printf(" 0x%05zx %2dk %2dk",
633		       _info.vi_window, (int)_info.vi_window_size/1024,
634		       (int)_info.vi_window_gran/1024);
635    		printf(" 0x%08zx %dk\n",
636		       _info.vi_buffer, (int)_info.vi_buffer_size/1024);
637	}
638}
639
640static void
641show_info(char *arg)
642{
643	if (!strcmp(arg, "adapter"))
644		show_adapter_info();
645	else if (!strcmp(arg, "mode"))
646		show_mode_info();
647	else {
648		warnx("argument to -i must be either adapter or mode");
649		return;
650	}
651}
652
653static void
654test_frame(void)
655{
656	int i, cur_mode, fore;
657
658	fore = 15;
659
660	if (ioctl(0, CONS_GET, &cur_mode) < 0)
661		err(1, "must be on a virtual console");
662	switch (cur_mode) {
663	case M_PC98_80x25:
664	case M_PC98_80x30:
665		fore = 7;
666		break;
667	}
668
669	fprintf(stdout, "\e[=0G\n\n");
670	for (i=0; i<8; i++) {
671		fprintf(stdout, "\e[=%dF\e[=0G        %2d \e[=%dF%-16s"
672				"\e[=%dF\e[=0G        %2d \e[=%dF%-16s        "
673				"\e[=%dF %2d \e[=%dGBACKGROUND\e[=0G\n",
674			fore, i, i, legal_colors[i],
675			fore, i+8, i+8, legal_colors[i+8],
676			fore, i, i);
677	}
678	fprintf(stdout, "\e[=%dF\e[=%dG\e[=%dH\e[=%dI\n",
679		info.mv_norm.fore, info.mv_norm.back,
680		info.mv_rev.fore, info.mv_rev.back);
681}
682
683/*
684 * Snapshot the video memory of that terminal, using the CONS_SCRSHOT
685 * ioctl, and writes the results to stdout either in the special
686 * binary format (see manual page for details), or in the plain
687 * text format.
688 */
689static void
690dump_screen(int mode, int opt)
691{
692	scrshot_t shot;
693	vid_info_t _info;
694
695	_info.size = sizeof(_info);
696	if (ioctl(0, CONS_GETINFO, &_info) == -1) {
697		warn("failed to obtain current video mode parameters");
698		return;
699	}
700
701	shot.x = shot.y = 0;
702	shot.xsize = _info.mv_csz;
703	shot.ysize = _info.mv_rsz;
704	if (opt == DUMP_ALL)
705		shot.ysize += _info.mv_hsz;
706
707	shot.buf = alloca(shot.xsize * shot.ysize * sizeof(u_int16_t));
708	if (shot.buf == NULL) {
709		warn("failed to allocate memory for dump");
710		return;
711	}
712
713	if (ioctl(0, CONS_SCRSHOT, &shot) == -1) {
714		warn("failed to get dump of the screen");
715		return;
716	}
717
718	if (mode == DUMP_FMT_RAW) {
719		printf("SCRSHOT_%c%c%c%c", DUMP_FMT_REV, 2,
720		       shot.xsize, shot.ysize);
721		fflush(stdout);
722
723		(void)write(STDOUT_FILENO, shot.buf,
724			    shot.xsize * shot.ysize * sizeof(u_int16_t));
725	} else {
726		char *line;
727		int x, y;
728		u_int16_t ch;
729
730		line = alloca(shot.xsize + 1);
731		if (line == NULL) {
732			warn("failed to allocate memory for line buffer");
733			return;
734		}
735
736		for (y = 0; y < shot.ysize; y++) {
737			for (x = 0; x < shot.xsize; x++) {
738				ch = shot.buf[x + (y * shot.xsize)];
739				ch &= 0xff;
740				if (isprint(ch) == 0)
741					ch = ' ';
742				line[x] = (char)ch;
743			}
744
745			/* Trim trailing spaces */
746			do {
747				line[x--] = '\0';
748			} while (line[x] == ' ' && x != 0);
749
750			puts(line);
751		}
752		fflush(stdout);
753	}
754
755	return;
756}
757
758static void
759set_history(char *opt)
760{
761	int size;
762
763	size = atoi(opt);
764	if ((*opt == '\0') || size < 0) {
765		warnx("argument must be a positive number");
766		return;
767	}
768	if (ioctl(0, CONS_HISTORY, &size) == -1)
769		warn("setting history buffer size");
770}
771
772static void
773clear_history(void)
774{
775
776	if (ioctl(0, CONS_CLRHIST) == -1)
777		warn("clear history buffer");
778}
779
780int
781main(int argc, char **argv)
782{
783	char	*font, *type;
784	int	dumpmod, dumpopt, opt;
785	int	reterr;
786
787	info.size = sizeof(info);
788	if (argc == 1)
789		usage();
790		/* Not reached */
791	if (ioctl(0, CONS_GETINFO, &info) < 0)
792		err(1, "must be on a virtual console");
793	dumpmod = 0;
794	dumpopt = DUMP_FBF;
795	while((opt = getopt(argc, argv, "b:Cc:df:g:h:Hi:l:LM:m:pPr:S:s:t:x")) != -1)
796		switch(opt) {
797		case 'b':
798			set_border_color(optarg);
799			break;
800		case 'C':
801			clear_history();
802			break;
803		case 'c':
804			set_cursor_type(optarg);
805			break;
806		case 'd':
807			print_scrnmap();
808			break;
809		case 'f':
810			type = optarg;
811			font = nextarg(argc, argv, &optind, 'f', 0);
812			if (font == NULL) {
813				type = NULL;
814				font = optarg;
815			}
816			load_font(type, font);
817			break;
818		case 'g':
819			if (sscanf(optarg, "%dx%d", &vesa_cols,
820			    &vesa_rows) != 2) {
821				warnx("incorrect geometry: %s", optarg);
822				usage();
823			}
824			break;
825		case 'h':
826			set_history(optarg);
827			break;
828		case 'H':
829			dumpopt = DUMP_ALL;
830			break;
831		case 'i':
832			show_info(optarg);
833			break;
834		case 'l':
835			load_scrnmap(optarg);
836			break;
837		case 'L':
838			load_default_scrnmap();
839			break;
840		case 'M':
841			set_mouse_char(optarg);
842			break;
843		case 'm':
844			set_mouse(optarg);
845			break;
846		case 'p':
847			dumpmod = DUMP_FMT_RAW;
848			break;
849		case 'P':
850			dumpmod = DUMP_FMT_TXT;
851			break;
852		case 'r':
853			set_reverse_colors(argc, argv, &optind);
854			break;
855		case 'S':
856			set_lockswitch(optarg);
857			break;
858		case 's':
859			set_console(optarg);
860			break;
861		case 't':
862			set_screensaver_timeout(optarg);
863			break;
864		case 'x':
865			hex = 1;
866			break;
867		default:
868			usage();
869		}
870	if (dumpmod != 0)
871		dump_screen(dumpmod, dumpopt);
872	reterr = video_mode(argc, argv, &optind);
873	set_normal_colors(argc, argv, &optind);
874	if (optind < argc && !strcmp(argv[optind], "show")) {
875		test_frame();
876		optind++;
877	}
878	if ((optind != argc) || (argc == 1))
879		usage();
880	return reterr;
881}
882
883