Rework option handling to save space.
[oweals/busybox.git] / util-linux / more.c
index 65409999b09d8aba7f1813773cb1ba9d13b18b2f..780cddf665e1b75dfb105047a5dd65b5f46c2ba3 100644 (file)
-#include "internal.h"
+/* vi: set sw=4 ts=4: */
+/*
+ * Mini more implementation for busybox
+ *
+ *
+ * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
+ *
+ * Latest version blended together by Erik Andersen <andersen@lineo.com>,
+ * based on the original more implementation by Bruce, and code from the 
+ * Debian boot-floppies team.
+ *
+ * Termios corrects by Vladimir Oleynik <vodz@usa.net>
+ *
+ * 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
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ *
+ */
+
 #include <stdio.h>
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <sys/ioctl.h>
 #include <fcntl.h>
+#include <signal.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <sys/ioctl.h>
+#include "busybox.h"
 
-#define BB_MORE_TERM
+static FILE *cin;
 
-#ifdef BB_MORE_TERM
-       #include <termios.h>
-       #include <signal.h>
+#ifdef BB_FEATURE_USE_TERMIOS
+#include <termios.h>
+#define setTermSettings(fd,argp) tcsetattr(fd,TCSANOW,argp)
+#define getTermSettings(fd,argp) tcgetattr(fd, argp);
 
-       FILE *cin;
-       struct termios initial_settings, new_settings;
+static struct termios initial_settings, new_settings;
 
-       void gotsig(int sig) { 
-               tcsetattr(fileno(cin), TCSANOW, &initial_settings);
-               exit(0);
-       }
-#endif
+static void set_tty_to_initial_mode(void)
+{
+       setTermSettings(fileno(cin), &initial_settings);
+}
+
+static void gotsig(int sig)
+{
+       putchar('\n');
+       exit(EXIT_FAILURE);
+}
+#endif /* BB_FEATURE_USE_TERMIOS */
+
+
+static int terminal_width = 79;        /* not 80 in case terminal has linefold bug */
+static int terminal_height = 24;
 
-const char     more_usage[] = "more [file]\n"
-"\n"
-"\tDisplays a file, one page at a time.\n"
-"\tIf there are no arguments, the standard input is displayed.\n";
 
-extern int
-more_fn(const struct FileInfo * i)
+extern int more_main(int argc, char **argv)
 {
-       FILE *  f = stdin;
-       int     c;
-       int     lines = 0, tlines = 0;
-       int     next_page = 0;
-       int     rows = 24, cols = 79;
-#ifdef BB_MORE_TERM
-       long sizeb = 0;
-       struct stat st; 
-       struct winsize win;
+       int c, lines, input = 0;
+       int please_display_more_prompt = -1;
+       struct stat st;
+       FILE *file;
+       int len, page_height;
+
+#if defined BB_FEATURE_AUTOWIDTH && defined BB_FEATURE_USE_TERMIOS
+       struct winsize win = { 0, 0, 0, 0 };
 #endif
-       
-       if ( i ) {
-               if (! (f = fopen(i->source, "r") )) {
-                       name_and_error(i->source);
-                       return 1;
-               }
-               fstat(fileno(f), &st);
-               sizeb = st.st_size / 100;
-       }
-               
-#ifdef BB_MORE_TERM
-       cin = fopen("/dev/tty", "r");
-       tcgetattr(fileno(cin),&initial_settings);
-       new_settings = initial_settings;
-       new_settings.c_lflag &= ~ICANON;
-       new_settings.c_lflag &= ~ECHO;
-       tcsetattr(fileno(cin), TCSANOW, &new_settings);
-       
-       (void) signal(SIGINT, gotsig);
-
-       ioctl(STDOUT_FILENO, TIOCGWINSZ, &win);
-       if (win.ws_row > 4)     rows = win.ws_row - 2;
-       if (win.ws_col > 0)     cols = win.ws_col - 1;
 
+       argc--;
+       argv++;
 
+
+       /* not use inputing from terminal if usage: more > outfile */
+       if(isatty(fileno(stdout))) {
+               cin = fopen(CURRENT_TTY, "r");
+               if (!cin)
+                       cin = xfopen(CONSOLE_DEV, "r");
+               please_display_more_prompt = 0;
+#ifdef BB_FEATURE_USE_TERMIOS
+               getTermSettings(fileno(cin), &initial_settings);
+               new_settings = initial_settings;
+               new_settings.c_lflag &= ~ICANON;
+               new_settings.c_lflag &= ~ECHO;
+#ifndef linux
+                /* Hmm, in linux c_cc[] not parsed if set ~ICANON */
+               new_settings.c_cc[VMIN] = 1;
+               new_settings.c_cc[VTIME] = 0;
+#endif
+               setTermSettings(fileno(cin), &new_settings);
+               atexit(set_tty_to_initial_mode);
+               (void) signal(SIGINT, gotsig);
+               (void) signal(SIGQUIT, gotsig);
+               (void) signal(SIGTERM, gotsig);
 #endif
+       }
 
-       while ( (c = getc(f)) != EOF ) {
-               if ( next_page ) {
-                       char    garbage;
-                       int     len;
-                       tlines += lines;
-                       lines = 0;
-                       next_page = 0;          //Percentage is based on bytes, not lines.
-                       if ( i && i->source )   //It is not very acurate, but still useful.
-                               len = printf("%s - %%%2ld - line: %d", i->source, (ftell(f) - sizeb - sizeb) / sizeb, tlines);
-                       else
-                               len = printf("line: %d", tlines);
-                               
-                       fflush(stdout);
-#ifndef BB_MORE_TERM           
-                       read(2, &garbage, 1);
-#else                          
-                       do {
-                               fread(&garbage, 1, 1, cin);     
-                       } while ((garbage != ' ') && (garbage != '\n'));
+       do {
+               if (argc == 0) {
+                       file = stdin;
+               } else
+                       file = wfopen(*argv, "r");
+               if(file==0)
+                       goto loop;
                        
-                       if (garbage == '\n') {
-                               lines = rows;
-                               tlines -= rows;
-                       }                                       
-                       garbage = 0;                            
-                       //clear line, since tabs don't overwrite.
-                       while(len-- > 0)        putchar('\b');
-                       while(len++ < cols)     putchar(' ');
-                       while(len-- > 0)        putchar('\b');
-                       fflush(stdout);
-#endif                                                         
+               fstat(fileno(file), &st);
+
+               if(please_display_more_prompt>0)
+                       please_display_more_prompt = 0;
+
+#if defined BB_FEATURE_AUTOWIDTH && defined BB_FEATURE_USE_TERMIOS
+               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
+               len=0;
+               lines = 0;
+               page_height = terminal_height;
+               while ((c = getc(file)) != EOF) {
+
+                       if (please_display_more_prompt>0) {
+                               len = printf("--More-- ");
+                               if (file != stdin) {
+#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
+                                       len += printf("(%d%% of %ld bytes)",
+                                                  (int) (100 * ((double) ftell(file) /
+                                                  (double) st.st_size)), (long)st.st_size);
+#endif
+                               }
+
+                               fflush(stdout);
+
+                               /*
+                                * We've just displayed the "--More--" prompt, so now we need
+                                * to get input from the user.
+                                */
+                               input = getc(cin);
+#ifndef BB_FEATURE_USE_TERMIOS
+                               printf("\033[A"); /* up cursor */
+#endif
+                               /* Erase the "More" message */
+                               putc('\r', stdout);
+                               while (--len >= 0)
+                                       putc(' ', stdout);
+                               putc('\r', stdout);
+                               fflush(stdout);
+                               len=0;
+                               lines = 0;
+                               page_height = terminal_height;
+                               please_display_more_prompt = 0;
+
+                               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')
+                                       if(please_display_more_prompt==0)
+                                       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) {
+                                       if(please_display_more_prompt==0)
+                                       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++;
                }
-               putchar(c);
-               if ( c == '\n' && ++lines == (rows + 1) )
-                       next_page = 1;
-       }
-       if ( f != stdin )
-               fclose(f);
-#ifdef BB_MORE_TERM
-       gotsig(0);
-#endif 
+               fclose(file);
+               fflush(stdout);
+loop:
+               argv++;
+       } while (--argc > 0);
+  end:
        return 0;
 }