Vladimir N. Oleynik (vodz) writes:
[oweals/busybox.git] / util-linux / more.c
index 974149177ad19372fdc68d45f991471115807f2a..5ece188b050af55e2069ab27355dd5d6d60abf04 100644 (file)
@@ -2,13 +2,15 @@
 /*
  * 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@lineo.com>,
- * based on the original more implementation by Bruce, and code from the 
+ * 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>
+ *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  * the Free Software Foundation; either version 2 of the License, or
  *
  */
 
-#include "internal.h"
 #include <stdio.h>
 #include <fcntl.h>
 #include <signal.h>
+#include <stdlib.h>
+#include <unistd.h>
 #include <sys/ioctl.h>
-#define BB_DECLARE_EXTERN
-#define bb_need_help
-#include "messages.c"
-
-static const char more_usage[] = "more [FILE ...]\n"
-#ifndef BB_FEATURE_TRIVIAL_HELP
-       "\nMore is a filter for viewing FILE one screenful at a time.\n"
-#endif
-       ;
+#include "busybox.h"
 
-/* ED: sparc termios is broken: revert back to old termio handling. */
-#ifdef BB_FEATURE_USE_TERMIOS
 
-#if #cpu(sparc)
-#      include <termio.h>
-#      define termios termio
-#      define setTermSettings(fd,argp) ioctl(fd,TCSETAF,argp)
-#      define getTermSettings(fd,argp) ioctl(fd,TCGETA,argp)
-#else
-#      include <termios.h>
-#      define setTermSettings(fd,argp) tcsetattr(fd,TCSANOW,argp)
-#      define getTermSettings(fd,argp) tcgetattr(fd, argp);
-#endif
+#ifdef CONFIG_FEATURE_USE_TERMIOS
+static int cin_fileno;
+#include <termios.h>
+#define setTermSettings(fd,argp) tcsetattr(fd,TCSANOW,argp)
+#define getTermSettings(fd,argp) tcgetattr(fd, argp);
 
-FILE *cin;
+static struct termios initial_settings, new_settings;
 
-struct termios initial_settings, new_settings;
-
-void gotsig(int sig)
+static void set_tty_to_initial_mode(void)
 {
-       setTermSettings(fileno(cin), &initial_settings);
-       fprintf(stdout, "\n");
-       exit(TRUE);
+       setTermSettings(cin_fileno, &initial_settings);
 }
-#endif
-
-
-
-#define TERMINAL_WIDTH 79              /* not 80 in case terminal has linefold bug */
-#define TERMINAL_HEIGHT        24
-
-
-#if defined BB_FEATURE_AUTOWIDTH
-#ifdef BB_FEATURE_USE_TERMIOS
-static int terminal_width = TERMINAL_WIDTH;
-#endif
-static int terminal_height = TERMINAL_HEIGHT;
-#else
-#define terminal_width TERMINAL_WIDTH
-#define terminal_height        TERMINAL_HEIGHT
-#endif
 
+static void gotsig(int sig)
+{
+       putchar('\n');
+       exit(EXIT_FAILURE);
+}
+#endif /* CONFIG_FEATURE_USE_TERMIOS */
 
 
 extern int more_main(int argc, char **argv)
 {
-       int c, lines = 0, input = 0;
-       int next_page = 0;
+       int c, lines, input = 0;
+       int please_display_more_prompt = 0;
        struct stat st;
        FILE *file;
-
-#if defined BB_FEATURE_AUTOWIDTH && defined BB_FEATURE_USE_TERMIOS
-       struct winsize win = { 0, 0 };
+       FILE *cin;
+       int len, page_height;
+       int terminal_width;
+       int terminal_height;
+#ifndef CONFIG_FEATURE_USE_TERMIOS
+       int cin_fileno;
 #endif
 
        argc--;
        argv++;
 
-       if (argc > 0
-               && (strcmp(*argv, dash_dash_help) == 0 || strcmp(*argv, "-h") == 0)) {
-               usage(more_usage);
+
+       /* not use inputing from terminal if usage: more > outfile */
+       if(isatty(fileno(stdout))) {
+               cin = fopen(CURRENT_TTY, "r");
+               if (!cin)
+                       cin = bb_xfopen(CONSOLE_DEV, "r");
+               cin_fileno = fileno(cin);
+               please_display_more_prompt = 2;
+#ifdef CONFIG_FEATURE_USE_TERMIOS
+               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);
+               atexit(set_tty_to_initial_mode);
+               (void) signal(SIGINT, gotsig);
+               (void) signal(SIGQUIT, gotsig);
+               (void) signal(SIGTERM, gotsig);
+#endif
+       } else {
+               cin = stdin;
        }
+
        do {
                if (argc == 0) {
                        file = stdin;
                } else
-                       file = fopen(*argv, "r");
+                       file = bb_wfopen(*argv, "r");
+               if(file==0)
+                       goto loop;
 
-               if (file == NULL) {
-                       perror(*argv);
-                       exit(FALSE);
-               }
+               st.st_size = 0;
                fstat(fileno(file), &st);
 
-#ifdef BB_FEATURE_USE_TERMIOS
-               cin = fopen("/dev/tty", "r");
-               if (!cin)
-                       cin = fopen("/dev/console", "r");
-               getTermSettings(fileno(cin), &initial_settings);
-               new_settings = initial_settings;
-               new_settings.c_cc[VMIN] = 1;
-               new_settings.c_cc[VTIME] = 0;
-               new_settings.c_lflag &= ~ICANON;
-               new_settings.c_lflag &= ~ECHO;
-               setTermSettings(fileno(cin), &new_settings);
-
-#ifdef BB_FEATURE_AUTOWIDTH
-               ioctl(fileno(stdout), TIOCGWINSZ, &win);
-               if (win.ws_row > 4)
-                       terminal_height = win.ws_row - 2;
-               if (win.ws_col > 0)
-                       terminal_width = win.ws_col - 1;
-#endif
+               please_display_more_prompt &= ~1;
 
-               (void) signal(SIGINT, gotsig);
-               (void) signal(SIGQUIT, gotsig);
-               (void) signal(SIGTERM, gotsig);
+               get_terminal_width_height(cin_fileno, &terminal_width, &terminal_height);
+               if (terminal_height > 4)
+                       terminal_height -= 2;
+               if (terminal_width > 0)
+                       terminal_width -= 1;
 
-#endif
+               len=0;
+               lines = 0;
+               page_height = terminal_height;
                while ((c = getc(file)) != EOF) {
-                       if (next_page) {
-                               int len = 0;
 
-                               next_page = 0;
-                               lines = 0;
-                               len = fprintf(stdout, "--More-- ");
-                               if (file != stdin) {
-                                       len += fprintf(stdout, "(%d%% of %ld bytes)",
-                                                                  (int) (100 *
-                                                                                 ((double) ftell(file) /
-                                                                                  (double) st.st_size)),
-                                                                  st.st_size);
-                               }
-                               len += fprintf(stdout, "%s",
-#ifdef BB_FEATURE_USE_TERMIOS
-                                                          ""
+                       if ((please_display_more_prompt & 3) == 3) {
+                               len = printf("--More-- ");
+                               if (file != stdin && st.st_size > 0) {
+#if _FILE_OFFSET_BITS == 64
+                                       len += printf("(%d%% of %lld bytes)",
+                                                  (int) (100 * ((double) ftell(file) /
+                                                  (double) st.st_size)), (long long)st.st_size);
 #else
-                                                          "\n"
+                                       len += printf("(%d%% of %ld bytes)",
+                                                  (int) (100 * ((double) ftell(file) /
+                                                  (double) st.st_size)), (long)st.st_size);
 #endif
-                                       );
+                               }
 
                                fflush(stdout);
-#ifdef BB_FEATURE_USE_TERMIOS
+
+                               /*
+                                * We've just displayed the "--More--" prompt, so now we need
+                                * to get input from the user.
+                                */
                                input = getc(cin);
-#else
-                               input = getc(stdin);
+#ifndef CONFIG_FEATURE_USE_TERMIOS
+                               printf("\033[A"); /* up cursor */
 #endif
-
-#ifdef BB_FEATURE_USE_TERMIOS
                                /* Erase the "More" message */
+                               putc('\r', stdout);
                                while (--len >= 0)
-                                       putc('\b', stdout);
-                               while (++len <= terminal_width)
                                        putc(' ', stdout);
-                               while (--len >= 0)
-                                       putc('\b', stdout);
-                               fflush(stdout);
-#endif
+                               putc('\r', stdout);
+                               len=0;
+                               lines = 0;
+                               page_height = terminal_height;
+                               please_display_more_prompt &= ~1;
 
+                               if (input == 'q')
+                                       goto end;
                        }
+
+                       /*
+                        * 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.
+                        */
                        if (c == '\n') {
-                               switch (input) {
-                               case 'q':
-                                       goto end;
-                               case '\n':
-                                       /* increment by just one line if we are at 
-                                        * the end of this line*/
-                                       next_page = 1;
-                                       break;
+                               /* increment by just one line if we are at
+                                * the end of this line */
+                               if (input == '\n')
+                                       please_display_more_prompt |= 1;
+                               /* Adjust the terminal height for any overlap, so that
+                                * no lines get lost off the top. */
+                               if (len >= terminal_width) {
+                                       int quot, rem;
+                                       quot = len / terminal_width;
+                                       rem  = len - (quot * terminal_width);
+                                       if (quot) {
+                                               if (rem)
+                                                       page_height-=quot;
+                                               else
+                                                       page_height-=(quot-1);
+                                       }
                                }
-                               if (++lines == terminal_height)
-                                       next_page = 1;
+                               if (++lines >= page_height) {
+                                       please_display_more_prompt |= 1;
+                               }
+                               len=0;
                        }
+                       /*
+                        * If we just read a newline from the file being 'mored' and any
+                        * key other than a return is hit, scroll by one page
+                        */
                        putc(c, stdout);
+                       len++;
                }
                fclose(file);
                fflush(stdout);
-
+loop:
                argv++;
        } while (--argc > 0);
   end:
-#ifdef BB_FEATURE_USE_TERMIOS
-       gotsig(0);
-#endif
-       exit(TRUE);
+       return 0;
 }