1) read from stdin work now

2) destroy segfault in number_process
3) removes dead code

complex patch my and Tito

1 files changed, 21 insertions, 34 deletions

diff --git a/miscutils/less.c b/miscutils/less.c
index 1b993ea2c..5bf59a130 100644
--- a/miscutils/less.c
+++ b/miscutils/less.c
@@ -108,11 +108,6 @@ static unsigned long flags;
 /* This is needed so that program behaviour changes when input comes from
    stdin */
 static int inp_stdin;
-/* This is required so that when a file is requested to be examined after
-   input has come from stdin (e.g. dmesg | less), the input stream from
-   the keyboard still stays the same. If it switched back to stdin, keyboard
-   input wouldn't work. */
-static int ea_inp_stdin;
 
 #ifdef CONFIG_FEATURE_LESS_MARKS
 static int mark_lines[15][2];
@@ -137,12 +132,12 @@ static FILE *inp;
 /* Reset terminal input to normal */
 static void set_tty_cooked(void) {
         fflush(stdout);
-        tcsetattr(0, TCSANOW, &term_orig);
+        tcsetattr(fileno(inp), TCSANOW, &term_orig);
 }
 
 /* Set terminal input to raw mode  (taken from vi.c) */
 static void set_tty_raw(void) {
-        tcsetattr(0, TCSANOW, &term_vi);
+        tcsetattr(fileno(inp), TCSANOW, &term_vi);
 }
 
 /* Exit the program gracefully */
@@ -224,35 +219,26 @@ static void data_readlines(void) {
         FILE *fp;
 
         fp = (inp_stdin) ? stdin : bb_xfopen(filename, "rt");
-
-        /* First of all, we need to know the number of lines so that flines can be initialised. */
-        for (i = 0; (!feof(fp)) && (i <= MAXLINES); i++)
-                fgets(current_line, 256, fp);
-        rewind(fp);
-        /* Initialise fp */
-        flines = malloc(i * sizeof(char *));
-
-        for (i = 0; (!feof(fp)) && (i <= MAXLINES); i++) {
+        flines = NULL;
+        for (i = 0; (feof(fp)==0) && (i <= MAXLINES); i++) {
                 strcpy(current_line, "");
                 fgets(current_line, 256, fp);
-                bb_xferror(fp, filename);
+                if(fp != stdin)
+                        bb_xferror(fp, filename);
+                flines = xrealloc(flines, (i+1) * sizeof(char *));
                 flines[i] = bb_xstrdup(current_line);
         }
         num_flines = i - 2;
 
-/* Reset variables for a new file */
+        /* Reset variables for a new file */
 
         line_pos = 0;
         past_eof = 0;
 
         fclose(fp);
 
-        inp = (inp_stdin) ? fopen(CURRENT_TTY, "r") : stdin;
-
-        if (ea_inp_stdin) {
-                fclose(inp);
-                inp = fopen(CURRENT_TTY, "r");
-        }
+        if(inp == NULL)
+                inp = (inp_stdin) ? fopen(CURRENT_TTY, "r") : stdin;
 
         if (flags & FLAG_N)
                 add_linenumbers();
@@ -544,7 +530,6 @@ static void examine_file(void) {
         num_files++;
 
         inp_stdin = 0;
-        ea_inp_stdin = 1;
         reinitialise();
 }
 
@@ -757,14 +742,16 @@ static void number_process(int first_digit) {
         int num;
         char num_input[80];
         char keypress;
+        char *endptr;
+
         num_input[0] = first_digit;
 
         /* Clear the current line, print a prompt, and then print the digit */
         clear_line();
         printf(":%c", first_digit);
 
-        /* Receive input until a letter is given */
-        while((num_input[i] = tless_getch()) && isdigit(num_input[i])) {
+        /* Receive input until a letter is given (max 80 chars)*/
+        while((i < 80) && (num_input[i] = tless_getch()) && isdigit(num_input[i])) {
                 printf("%c", num_input[i]);
                 i++;
         }
@@ -772,8 +759,9 @@ static void number_process(int first_digit) {
         /* Take the final letter out of the digits string */
         keypress = num_input[i];
         num_input[i] = '\0';
-        i--;
-        num = atoi(num_input);
+        num = strtol(num_input, &endptr, 10);
+        if (endptr==num_input || *endptr!='\0' || num < 1 || num > MAXLINES)
+                goto END;
 
         /* We now know the number and the letter entered, so we process them */
         switch (keypress) {
@@ -806,7 +794,7 @@ static void number_process(int first_digit) {
                 default:
                         break;
         }
-
+END:
         buffer_print();
 }
 
@@ -1165,17 +1153,16 @@ int less_main(int argc, char **argv) {
                 }
         }
 
-        strcpy(filename, (inp_stdin) ? "stdin" : files[0]);
+        strcpy(filename, (inp_stdin) ? bb_msg_standard_input : files[0]);
         tty_width_height();
-        tcgetattr(0, &term_orig);
+        data_readlines();
+        tcgetattr(fileno(inp), &term_orig);
         term_vi = term_orig;
         term_vi.c_lflag &= (~ICANON & ~ECHO);
         term_vi.c_iflag &= (~IXON & ~ICRNL);
         term_vi.c_oflag &= (~ONLCR);
         term_vi.c_cc[VMIN] = 1;
         term_vi.c_cc[VTIME] = 0;
-
-        data_readlines();
         buffer_init();
         buffer_print();