busybox/util-linux/more.c

/* vi: set sw=4 ts=4: */
/*
 * Mini more implementation for busybox
 *
 * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
 * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
 *
 * Latest version blended together by Erik Andersen <andersen@codepoet.org>,
 * based on the original more implementation by Bruce, and code from the
 * Debian boot-floppies team.
 *
 * Termios corrects by Vladimir Oleynik <dzo@simtreas.ru>
 *
 * Licensed under GPLv2 or later, see file License in this tarball for details.
 */

#include "libbb.h"
#if ENABLE_FEATURE_USE_TERMIOS
#include <termios.h>
#endif /* FEATURE_USE_TERMIOS */


#if ENABLE_FEATURE_USE_TERMIOS

struct globals {
	int cin_fileno;
	struct termios initial_settings;
	struct termios new_settings;
};
#define G (*(struct globals*)bb_common_bufsiz1)
#define INIT_G() ((void)0)
#define initial_settings (G.initial_settings)
#define new_settings     (G.new_settings    )
#define cin_fileno       (G.cin_fileno      )

#define setTermSettings(fd, argp) tcsetattr(fd, TCSANOW, argp)
#define getTermSettings(fd, argp) tcgetattr(fd, argp)

static void gotsig(int sig UNUSED_PARAM)
{
	bb_putchar('\n');
	setTermSettings(cin_fileno, &initial_settings);
	exit(EXIT_FAILURE);
}

#else /* !FEATURE_USE_TERMIOS */
#define INIT_G() ((void)0)
#define setTermSettings(fd, argp) ((void)0)
#endif /* FEATURE_USE_TERMIOS */

#define CONVERTED_TAB_SIZE 8

int more_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int more_main(int argc UNUSED_PARAM, char **argv)
{
	int c = c; /* for gcc */
	int lines;
	int input = 0;
	int spaces = 0;
	int please_display_more_prompt;
	struct stat st;
	FILE *file;
	FILE *cin;
	int len;
	unsigned terminal_width;
	unsigned terminal_height;

	INIT_G();

	argv++;
	/* Another popular pager, most, detects when stdout
	 * is not a tty and turns into cat. This makes sense. */
	if (!isatty(STDOUT_FILENO))
		return bb_cat(argv);
	cin = fopen_for_read(CURRENT_TTY);
	if (!cin)
		return bb_cat(argv);

#if ENABLE_FEATURE_USE_TERMIOS
	cin_fileno = fileno(cin);
	getTermSettings(cin_fileno, &initial_settings);
	new_settings = initial_settings;
	new_settings.c_lflag &= ~ICANON;
	new_settings.c_lflag &= ~ECHO;
	new_settings.c_cc[VMIN] = 1;
	new_settings.c_cc[VTIME] = 0;
	setTermSettings(cin_fileno, &new_settings);
	bb_signals(0
		+ (1 << SIGINT)
		+ (1 << SIGQUIT)
		+ (1 << SIGTERM)
		, gotsig);
#endif

	do {
		file = stdin;
		if (*argv) {
			file = fopen_or_warn(*argv, "r");
			if (!file)
				continue;
		}
		st.st_size = 0;
		fstat(fileno(file), &st);

		please_display_more_prompt = 0;
		/* never returns w, h <= 1 */
		get_terminal_width_height(fileno(cin), &terminal_width, &terminal_height);
		terminal_height -= 1;

		len = 0;
		lines = 0;
		while (spaces || (c = getc(file)) != EOF) {
			int wrap;
			if (spaces)
				spaces--;
 loop_top:
			if (input != 'r' && please_display_more_prompt) {
				len = printf("--More-- ");
				if (st.st_size > 0) {
					len += printf("(%d%% of %"OFF_FMT"d bytes)",
						(int) (ftello(file)*100 / st.st_size),
						st.st_size);
				}
				fflush(stdout);

				/*
				 * We've just displayed the "--More--" prompt, so now we need
				 * to get input from the user.
				 */
				for (;;) {
					input = getc(cin);
					input = tolower(input);
#if !ENABLE_FEATURE_USE_TERMIOS
					printf("\033[A"); /* up cursor */
#endif
					/* Erase the last message */
					printf("\r%*s\r", len, "");

					/* Due to various multibyte escape
					 * sequences, it's not ok to accept
					 * any input as a command to scroll
					 * the screen. We only allow known
					 * commands, else we show help msg. */
					if (input == ' ' || input == '\n' || input == 'q' || input == 'r')
						break;
					len = printf("(Enter:next line Space:next page Q:quit R:show the rest)");
				}
				len = 0;
				lines = 0;
				please_display_more_prompt = 0;

				if (input == 'q')
					goto end;

				/* The user may have resized the terminal.
				 * Re-read the dimensions. */
#if ENABLE_FEATURE_USE_TERMIOS
				get_terminal_width_height(cin_fileno, &terminal_width, &terminal_height);
				terminal_height -= 1;
#endif
			}

			/* Crudely convert tabs into spaces, which are
			 * a bajillion times easier to deal with. */
			if (c == '\t') {
				spaces = CONVERTED_TAB_SIZE - 1;
				c = ' ';
			}

			/*
			 * There are two input streams to worry about here:
			 *
			 * c    : the character we are reading from the file being "mored"
			 * input: a character received from the keyboard
			 *
			 * If we hit a newline in the _file_ stream, we want to test and
			 * see if any characters have been hit in the _input_ stream. This
			 * allows the user to quit while in the middle of a file.
			 */
			wrap = (++len > terminal_width);
			if (c == '\n' || wrap) {
				/* Then outputting this character
				 * will move us to a new line. */
				if (++lines >= terminal_height || input == '\n')
					please_display_more_prompt = 1;
				len = 0;
			}
			if (c != '\n' && wrap) {
				/* Then outputting this will also put a character on
				 * the beginning of that new line. Thus we first want to
				 * display the prompt (if any), so we skip the putchar()
				 * and go back to the top of the loop, without reading
				 * a new character. */
				goto loop_top;
			}
			/* My small mind cannot fathom backspaces and UTF-8 */
			putchar(c);
		}
		fclose(file);
		fflush(stdout);
	} while (*argv && *++argv);
 end:
	setTermSettings(cin_fileno, &initial_settings);
	return 0;
}
1	/* vi: set sw=4 ts=4: */
2	/*
3	* Mini more implementation for busybox
4	*
5	* Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
6	* Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
7	*
8	* Latest version blended together by Erik Andersen <andersen@codepoet.org>,
9	* based on the original more implementation by Bruce, and code from the
10	* Debian boot-floppies team.
11	*
12	* Termios corrects by Vladimir Oleynik <dzo@simtreas.ru>
13	*
14	* Licensed under GPLv2 or later, see file License in this tarball for details.
15	*/
16
17	#include "libbb.h"
18	#if ENABLE_FEATURE_USE_TERMIOS
19	#include <termios.h>
20	#endif /* FEATURE_USE_TERMIOS */
21
22
23	#if ENABLE_FEATURE_USE_TERMIOS
24
25	struct globals {
26	int cin_fileno;
27	struct termios initial_settings;
28	struct termios new_settings;
29	};
30	#define G ((struct globals)bb_common_bufsiz1)
31	#define INIT_G() ((void)0)
32	#define initial_settings (G.initial_settings)
33	#define new_settings (G.new_settings )
34	#define cin_fileno (G.cin_fileno )
35
36	#define setTermSettings(fd, argp) tcsetattr(fd, TCSANOW, argp)
37	#define getTermSettings(fd, argp) tcgetattr(fd, argp)
38
39	static void gotsig(int sig UNUSED_PARAM)
40	{
41	bb_putchar('\n');
42	setTermSettings(cin_fileno, &initial_settings);
43	exit(EXIT_FAILURE);
44	}
45
46	#else /* !FEATURE_USE_TERMIOS */
47	#define INIT_G() ((void)0)
48	#define setTermSettings(fd, argp) ((void)0)
49	#endif /* FEATURE_USE_TERMIOS */
50
51	#define CONVERTED_TAB_SIZE 8
52
53	int more_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
54	int more_main(int argc UNUSED_PARAM, char **argv)
55	{
56	int c = c; /* for gcc */
57	int lines;
58	int input = 0;
59	int spaces = 0;
60	int please_display_more_prompt;
61	struct stat st;
62	FILE *file;
63	FILE *cin;
64	int len;
65	unsigned terminal_width;
66	unsigned terminal_height;
67
68	INIT_G();
69
70	argv++;
71	/* Another popular pager, most, detects when stdout
72	* is not a tty and turns into cat. This makes sense. */
73	if (!isatty(STDOUT_FILENO))
74	return bb_cat(argv);
75	cin = fopen_for_read(CURRENT_TTY);
76	if (!cin)
77	return bb_cat(argv);
78
79	#if ENABLE_FEATURE_USE_TERMIOS
80	cin_fileno = fileno(cin);
81	getTermSettings(cin_fileno, &initial_settings);
82	new_settings = initial_settings;
83	new_settings.c_lflag &= ~ICANON;
84	new_settings.c_lflag &= ~ECHO;
85	new_settings.c_cc[VMIN] = 1;
86	new_settings.c_cc[VTIME] = 0;
87	setTermSettings(cin_fileno, &new_settings);
88	bb_signals(0
89	+ (1 << SIGINT)
90	+ (1 << SIGQUIT)
91	+ (1 << SIGTERM)
92	, gotsig);
93	#endif
94
95	do {
96	file = stdin;
97	if (*argv) {
98	file = fopen_or_warn(*argv, "r");
99	if (!file)
100	continue;
101	}
102	st.st_size = 0;
103	fstat(fileno(file), &st);
104
105	please_display_more_prompt = 0;
106	/* never returns w, h <= 1 */
107	get_terminal_width_height(fileno(cin), &terminal_width, &terminal_height);
108	terminal_height -= 1;
109
110	len = 0;
111	lines = 0;
112	while (spaces \|\| (c = getc(file)) != EOF) {
113	int wrap;
114	if (spaces)
115	spaces--;
116	loop_top:
117	if (input != 'r' && please_display_more_prompt) {
118	len = printf("--More-- ");
119	if (st.st_size > 0) {
120	len += printf("(%d%% of %"OFF_FMT"d bytes)",
121	(int) (ftello(file)*100 / st.st_size),
122	st.st_size);
123	}
124	fflush(stdout);
125
126	/*
127	* We've just displayed the "--More--" prompt, so now we need
128	* to get input from the user.
129	*/
130	for (;;) {
131	input = getc(cin);
132	input = tolower(input);
133	#if !ENABLE_FEATURE_USE_TERMIOS
134	printf("\033[A"); /* up cursor */
135	#endif
136	/* Erase the last message */
137	printf("\r%*s\r", len, "");
138
139	/* Due to various multibyte escape
140	* sequences, it's not ok to accept
141	* any input as a command to scroll
142	* the screen. We only allow known
143	* commands, else we show help msg. */
144	if (input == ' ' \|\| input == '\n' \|\| input == 'q' \|\| input == 'r')
145	break;
146	len = printf("(Enter:next line Space:next page Q:quit R:show the rest)");
147	}
148	len = 0;
149	lines = 0;
150	please_display_more_prompt = 0;
151
152	if (input == 'q')
153	goto end;
154
155	/* The user may have resized the terminal.
156	* Re-read the dimensions. */
157	#if ENABLE_FEATURE_USE_TERMIOS
158	get_terminal_width_height(cin_fileno, &terminal_width, &terminal_height);
159	terminal_height -= 1;
160	#endif
161	}
162
163	/* Crudely convert tabs into spaces, which are
164	* a bajillion times easier to deal with. */
165	if (c == '\t') {
166	spaces = CONVERTED_TAB_SIZE - 1;
167	c = ' ';
168	}
169
170	/*
171	* There are two input streams to worry about here:
172	*
173	* c : the character we are reading from the file being "mored"
174	* input: a character received from the keyboard
175	*
176	* If we hit a newline in the _file_ stream, we want to test and
177	* see if any characters have been hit in the _input_ stream. This
178	* allows the user to quit while in the middle of a file.
179	*/
180	wrap = (++len > terminal_width);
181	if (c == '\n' \|\| wrap) {
182	/* Then outputting this character
183	* will move us to a new line. */
184	if (++lines >= terminal_height \|\| input == '\n')
185	please_display_more_prompt = 1;
186	len = 0;
187	}
188	if (c != '\n' && wrap) {
189	/* Then outputting this will also put a character on
190	* the beginning of that new line. Thus we first want to
191	* display the prompt (if any), so we skip the putchar()
192	* and go back to the top of the loop, without reading
193	* a new character. */
194	goto loop_top;
195	}
196	/* My small mind cannot fathom backspaces and UTF-8 */
197	putchar(c);
198	}
199	fclose(file);
200	fflush(stdout);
201	} while (argv && ++argv);
202	end:
203	setTermSettings(cin_fileno, &initial_settings);
204	return 0;
205	}