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