Vladimir N. Oleynik (vodz) writes:
[oweals/busybox.git] / util-linux / more.c
index 7d0ddb8ece069c2007c3de8fd53edd7f1025f788..5ece188b050af55e2069ab27355dd5d6d60abf04 100644 (file)
@@ -1,13 +1,16 @@
+/* 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@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
  *
  */
 
-
-/* Turning this off makes things a bit smaller (and less pretty) */
-#define BB_FEATURE_USE_TERMIOS
-/* Turning this off makes things a bit smaller (and less pretty) */
-#define BB_FEATURE_AUTOWIDTH
-
-
-
-#include "internal.h"
 #include <stdio.h>
 #include <fcntl.h>
 #include <signal.h>
+#include <stdlib.h>
+#include <unistd.h>
 #include <sys/ioctl.h>
+#include "busybox.h"
 
-static const char more_usage[] = "more [file ...]\n";
 
-/* ED: sparc termios is broken: revert back to old termio handling. */
-#ifdef BB_FEATURE_USE_TERMIOS
+#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);
 
-#if #cpu(sparc)
-#      define USE_OLD_TERMIO
-#      include <termio.h>
-#      define termios termio
-#      define stty(fd,argp) ioctl(fd,TCSETAF,argp)
-#else
-#      include <termios.h>
-#      define stty(fd,argp) tcsetattr(fd,TCSANOW,argp)
-#endif
+static struct termios initial_settings, new_settings;
 
-    FILE *cin;
-    struct termios initial_settings, new_settings;
-
-    void gotsig(int sig) { 
-           stty(fileno(cin), &initial_settings);
-           fprintf(stdout, "\n");
-           exit( TRUE);
-    }
-#endif
-
-
-
-#define TERMINAL_WIDTH 79      /* not 80 in case terminal has linefold bug */
-#define TERMINAL_HEIGHT        24
-
-
-#if defined BB_FEATURE_AUTOWIDTH
-static int terminal_width = 0, terminal_height = 0;
-#else
-#define terminal_width TERMINAL_WIDTH
-#define terminal_height        TERMINAL_HEIGHT
-#endif
+static void set_tty_to_initial_mode(void)
+{
+       setTermSettings(cin_fileno, &initial_settings);
+}
 
+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;
-    struct stat st;    
-    FILE *file;
-#ifdef BB_FEATURE_AUTOWIDTH
-    struct winsize win = {0,0};
+       int c, lines, input = 0;
+       int please_display_more_prompt = 0;
+       struct stat st;
+       FILE *file;
+       FILE *cin;
+       int len, page_height;
+       int terminal_width;
+       int terminal_height;
+#ifndef CONFIG_FEATURE_USE_TERMIOS
+       int cin_fileno;
 #endif
 
-    argc--;
-    argv++;
+       argc--;
+       argv++;
 
-    if ( argc > 0 && (strcmp(*argv,"--help")==0 || strcmp(*argv,"-h")==0) ) {
-       usage (more_usage);
-    }
-    do {
-       if (argc==0) {
-           file = stdin;
-       }
-       else
-           file = fopen(*argv, "r");
 
-       if (file == NULL) {
-           perror(*argv);
-           exit(FALSE);
+       /* 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;
        }
-       fstat(fileno(file), &st);
-
-#ifdef BB_FEATURE_USE_TERMIOS
-       cin = fopen("/dev/tty", "r");
-       if (!cin)
-           cin = fopen("/dev/console", "r");
-#ifdef USE_OLD_TERMIO
-       ioctl(fileno(cin),TCGETA,&initial_settings);
+
+       do {
+               if (argc == 0) {
+                       file = stdin;
+               } else
+                       file = bb_wfopen(*argv, "r");
+               if(file==0)
+                       goto loop;
+
+               st.st_size = 0;
+               fstat(fileno(file), &st);
+
+               please_display_more_prompt &= ~1;
+
+               get_terminal_width_height(cin_fileno, &terminal_width, &terminal_height);
+               if (terminal_height > 4)
+                       terminal_height -= 2;
+               if (terminal_width > 0)
+                       terminal_width -= 1;
+
+               len=0;
+               lines = 0;
+               page_height = terminal_height;
+               while ((c = getc(file)) != EOF) {
+
+                       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
-       tcgetattr(fileno(cin),&initial_settings);
-#endif
-       new_settings = initial_settings;
-       new_settings.c_lflag &= ~ICANON;
-       new_settings.c_lflag &= ~ECHO;
-       stty(fileno(cin), &new_settings);
-
-#ifdef BB_FEATURE_AUTOWIDTH    
-       ioctl(STDOUT_FILENO, TIOCGWINSZ, &win);
-       if (win.ws_row > 4) 
-           terminal_height = win.ws_row - 2;
-       if (win.ws_col > 0) 
-           terminal_width = win.ws_col - 1;
+                                       len += printf("(%d%% of %ld bytes)",
+                                                  (int) (100 * ((double) ftell(file) /
+                                                  (double) st.st_size)), (long)st.st_size);
 #endif
+                               }
 
-       (void) signal(SIGINT, gotsig);
-       (void) signal(SIGQUIT, gotsig);
-       (void) signal(SIGTERM, gotsig);
+                               fflush(stdout);
 
+                               /*
+                                * We've just displayed the "--More--" prompt, so now we need
+                                * to get input from the user.
+                                */
+                               input = getc(cin);
+#ifndef CONFIG_FEATURE_USE_TERMIOS
+                               printf("\033[A"); /* up cursor */
 #endif
-       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);
+                               /* Erase the "More" message */
+                               putc('\r', stdout);
+                               while (--len >= 0)
+                                       putc(' ', stdout);
+                               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') {
+                               /* 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 >= 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++;
                }
-               len += fprintf(stdout, "%s",
-#ifdef BB_FEATURE_USE_TERMIOS
-                       ""
-#else
-                       "\n"
-#endif
-                       );
-
-               fflush(stdout);
-               input = getc( cin);
-
-#ifdef BB_FEATURE_USE_TERMIOS
-               /* Erase the "More" message */
-               while(--len >= 0)
-                   putc('\b', stdout);
-               while(++len <= terminal_width)
-                   putc(' ', stdout);
-               while(--len >= 0)
-                   putc('\b', stdout);
+               fclose(file);
                fflush(stdout);
-#endif
-
-           }
-           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;
-               }
-               if ( ++lines == terminal_height )
-                   next_page = 1;
-           }
-           putc(c, stdout);
-       }
-       fclose(file);
-       fflush(stdout);
-
-       argv++;
-    } while (--argc > 0);
-end:
-#ifdef BB_FEATURE_USE_TERMIOS
-    gotsig(0);
-#endif 
-    exit(TRUE);
+loop:
+               argv++;
+       } while (--argc > 0);
+  end:
+       return 0;
 }
-