less: switch off nonblock on kbd_fd before exit
[oweals/busybox.git] / miscutils / less.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini less implementation for busybox
4  *
5  * Copyright (C) 2005 by Rob Sullivan <cogito.ergo.cogito@gmail.com>
6  *
7  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
8  */
9
10 /*
11  * TODO:
12  * - Add more regular expression support - search modifiers, certain matches, etc.
13  * - Add more complex bracket searching - currently, nested brackets are
14  *   not considered.
15  * - Add support for "F" as an input. This causes less to act in
16  *   a similar way to tail -f.
17  * - Allow horizontal scrolling.
18  *
19  * Notes:
20  * - the inp file pointer is used so that keyboard input works after
21  *   redirected input has been read from stdin
22  */
23
24 //config:config LESS
25 //config:       bool "less"
26 //config:       default y
27 //config:       help
28 //config:         'less' is a pager, meaning that it displays text files. It possesses
29 //config:         a wide array of features, and is an improvement over 'more'.
30 //config:
31 //config:config FEATURE_LESS_MAXLINES
32 //config:       int "Max number of input lines less will try to eat"
33 //config:       default 9999999
34 //config:       depends on LESS
35 //config:
36 //config:config FEATURE_LESS_BRACKETS
37 //config:       bool "Enable bracket searching"
38 //config:       default y
39 //config:       depends on LESS
40 //config:       help
41 //config:         This option adds the capability to search for matching left and right
42 //config:         brackets, facilitating programming.
43 //config:
44 //config:config FEATURE_LESS_FLAGS
45 //config:       bool "Enable -m/-M"
46 //config:       default y
47 //config:       depends on LESS
48 //config:       help
49 //config:         The -M/-m flag enables a more sophisticated status line.
50 //config:
51 //config:config FEATURE_LESS_TRUNCATE
52 //config:       bool "Enable -S"
53 //config:       default y
54 //config:       depends on LESS
55 //config:       help
56 //config:         The -S flag causes long lines to be truncated rather than
57 //config:         wrapped.
58 //config:
59 //config:config FEATURE_LESS_MARKS
60 //config:       bool "Enable marks"
61 //config:       default y
62 //config:       depends on LESS
63 //config:       help
64 //config:         Marks enable positions in a file to be stored for easy reference.
65 //config:
66 //config:config FEATURE_LESS_REGEXP
67 //config:       bool "Enable regular expressions"
68 //config:       default y
69 //config:       depends on LESS
70 //config:       help
71 //config:         Enable regular expressions, allowing complex file searches.
72 //config:
73 //config:config FEATURE_LESS_WINCH
74 //config:       bool "Enable automatic resizing on window size changes"
75 //config:       default y
76 //config:       depends on LESS
77 //config:       help
78 //config:         Makes less track window size changes.
79 //config:
80 //config:config FEATURE_LESS_ASK_TERMINAL
81 //config:       bool "Use 'tell me cursor position' ESC sequence to measure window"
82 //config:       default y
83 //config:       depends on FEATURE_LESS_WINCH
84 //config:       help
85 //config:         Makes less track window size changes.
86 //config:         If terminal size can't be retrieved and $LINES/$COLUMNS are not set,
87 //config:         this option makes less perform a last-ditch effort to find it:
88 //config:         position cursor to 999,999 and ask terminal to report real
89 //config:         cursor position using "ESC [ 6 n" escape sequence, then read stdin.
90 //config:
91 //config:         This is not clean but helps a lot on serial lines and such.
92 //config:
93 //config:config FEATURE_LESS_DASHCMD
94 //config:       bool "Enable flag changes ('-' command)"
95 //config:       default y
96 //config:       depends on LESS
97 //config:       help
98 //config:         This enables the ability to change command-line flags within
99 //config:         less itself ('-' keyboard command).
100 //config:
101 //config:config FEATURE_LESS_LINENUMS
102 //config:       bool "Enable dynamic switching of line numbers"
103 //config:       default y
104 //config:       depends on FEATURE_LESS_DASHCMD
105 //config:       help
106 //config:         Enables "-N" command.
107
108 //usage:#define less_trivial_usage
109 //usage:       "[-E" IF_FEATURE_LESS_REGEXP("I")IF_FEATURE_LESS_FLAGS("Mm")
110 //usage:       "N" IF_FEATURE_LESS_TRUNCATE("S") "h~] [FILE]..."
111 //usage:#define less_full_usage "\n\n"
112 //usage:       "View FILE (or stdin) one screenful at a time\n"
113 //usage:     "\n        -E      Quit once the end of a file is reached"
114 //usage:        IF_FEATURE_LESS_REGEXP(
115 //usage:     "\n        -I      Ignore case in all searches"
116 //usage:        )
117 //usage:        IF_FEATURE_LESS_FLAGS(
118 //usage:     "\n        -M,-m   Display status line with line numbers"
119 //usage:     "\n                and percentage through the file"
120 //usage:        )
121 //usage:     "\n        -N      Prefix line number to each line"
122 //usage:        IF_FEATURE_LESS_TRUNCATE(
123 //usage:     "\n        -S      Truncate long lines"
124 //usage:        )
125 //usage:     "\n        -~      Suppress ~s displayed past EOF"
126
127 #include <sched.h>  /* sched_yield() */
128
129 #include "libbb.h"
130 #include "common_bufsiz.h"
131 #if ENABLE_FEATURE_LESS_REGEXP
132 #include "xregex.h"
133 #endif
134
135
136 #define ESC "\033"
137 /* The escape codes for highlighted and normal text */
138 #define HIGHLIGHT   ESC"[7m"
139 #define NORMAL      ESC"[0m"
140 /* The escape code to home and clear to the end of screen */
141 #define CLEAR       ESC"[H\033[J"
142 /* The escape code to clear to the end of line */
143 #define CLEAR_2_EOL ESC"[K"
144
145 enum {
146 /* Absolute max of lines eaten */
147         MAXLINES = CONFIG_FEATURE_LESS_MAXLINES,
148 /* This many "after the end" lines we will show (at max) */
149         TILDES = 1,
150 };
151
152 /* Command line options */
153 enum {
154         FLAG_E = 1 << 0,
155         FLAG_M = 1 << 1,
156         FLAG_m = 1 << 2,
157         FLAG_N = 1 << 3,
158         FLAG_TILDE = 1 << 4,
159         FLAG_I = 1 << 5,
160         FLAG_S = (1 << 6) * ENABLE_FEATURE_LESS_TRUNCATE,
161 /* hijack command line options variable for internal state vars */
162         LESS_STATE_MATCH_BACKWARDS = 1 << 15,
163 };
164
165 #if !ENABLE_FEATURE_LESS_REGEXP
166 enum { pattern_valid = 0 };
167 #endif
168
169 struct globals {
170         int cur_fline; /* signed */
171         int kbd_fd;  /* fd to get input from */
172         int kbd_fd_orig_flags;
173         int less_gets_pos;
174 /* last position in last line, taking into account tabs */
175         size_t last_line_pos;
176         unsigned max_fline;
177         unsigned max_lineno; /* this one tracks linewrap */
178         unsigned max_displayed_line;
179         unsigned width;
180 #if ENABLE_FEATURE_LESS_WINCH
181         unsigned winch_counter;
182 #endif
183         ssize_t eof_error; /* eof if 0, error if < 0 */
184         ssize_t readpos;
185         ssize_t readeof; /* must be signed */
186         const char **buffer;
187         const char **flines;
188         const char *empty_line_marker;
189         unsigned num_files;
190         unsigned current_file;
191         char *filename;
192         char **files;
193 #if ENABLE_FEATURE_LESS_FLAGS
194         int num_lines; /* a flag if < 0, line count if >= 0 */
195 # define REOPEN_AND_COUNT (-1)
196 # define REOPEN_STDIN     (-2)
197 # define NOT_REGULAR_FILE (-3)
198 #endif
199 #if ENABLE_FEATURE_LESS_MARKS
200         unsigned num_marks;
201         unsigned mark_lines[15][2];
202 #endif
203 #if ENABLE_FEATURE_LESS_REGEXP
204         unsigned *match_lines;
205         int match_pos; /* signed! */
206         int wanted_match; /* signed! */
207         int num_matches;
208         regex_t pattern;
209         smallint pattern_valid;
210 #endif
211 #if ENABLE_FEATURE_LESS_ASK_TERMINAL
212         smallint winsize_err;
213 #endif
214         smallint terminated;
215         struct termios term_orig, term_less;
216         char kbd_input[KEYCODE_BUFFER_SIZE];
217 };
218 #define G (*ptr_to_globals)
219 #define cur_fline           (G.cur_fline         )
220 #define kbd_fd              (G.kbd_fd            )
221 #define less_gets_pos       (G.less_gets_pos     )
222 #define last_line_pos       (G.last_line_pos     )
223 #define max_fline           (G.max_fline         )
224 #define max_lineno          (G.max_lineno        )
225 #define max_displayed_line  (G.max_displayed_line)
226 #define width               (G.width             )
227 #define winch_counter       (G.winch_counter     )
228 /* This one is 100% not cached by compiler on read access */
229 #define WINCH_COUNTER (*(volatile unsigned *)&winch_counter)
230 #define eof_error           (G.eof_error         )
231 #define readpos             (G.readpos           )
232 #define readeof             (G.readeof           )
233 #define buffer              (G.buffer            )
234 #define flines              (G.flines            )
235 #define empty_line_marker   (G.empty_line_marker )
236 #define num_files           (G.num_files         )
237 #define current_file        (G.current_file      )
238 #define filename            (G.filename          )
239 #define files               (G.files             )
240 #define num_lines           (G.num_lines         )
241 #define num_marks           (G.num_marks         )
242 #define mark_lines          (G.mark_lines        )
243 #if ENABLE_FEATURE_LESS_REGEXP
244 #define match_lines         (G.match_lines       )
245 #define match_pos           (G.match_pos         )
246 #define num_matches         (G.num_matches       )
247 #define wanted_match        (G.wanted_match      )
248 #define pattern             (G.pattern           )
249 #define pattern_valid       (G.pattern_valid     )
250 #endif
251 #define terminated          (G.terminated        )
252 #define term_orig           (G.term_orig         )
253 #define term_less           (G.term_less         )
254 #define kbd_input           (G.kbd_input         )
255 #define INIT_G() do { \
256         SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
257         less_gets_pos = -1; \
258         empty_line_marker = "~"; \
259         num_files = 1; \
260         current_file = 1; \
261         eof_error = 1; \
262         terminated = 1; \
263         IF_FEATURE_LESS_REGEXP(wanted_match = -1;) \
264 } while (0)
265
266 /* flines[] are lines read from stdin, each in malloc'ed buffer.
267  * Line numbers are stored as uint32_t prepended to each line.
268  * Pointer is adjusted so that flines[i] points directly past
269  * line number. Accesor: */
270 #define MEMPTR(p) ((char*)(p) - 4)
271 #define LINENO(p) (*(uint32_t*)((p) - 4))
272
273
274 /* Reset terminal input to normal */
275 static void set_tty_cooked(void)
276 {
277         fflush_all();
278         tcsetattr(kbd_fd, TCSANOW, &term_orig);
279 }
280
281 /* Move the cursor to a position (x,y), where (0,0) is the
282    top-left corner of the console */
283 static void move_cursor(int line, int row)
284 {
285         printf(ESC"[%u;%uH", line, row);
286 }
287
288 static void clear_line(void)
289 {
290         printf(ESC"[%u;0H" CLEAR_2_EOL, max_displayed_line + 2);
291 }
292
293 static void print_hilite(const char *str)
294 {
295         printf(HIGHLIGHT"%s"NORMAL, str);
296 }
297
298 static void print_statusline(const char *str)
299 {
300         clear_line();
301         printf(HIGHLIGHT"%.*s"NORMAL, width - 1, str);
302 }
303
304 /* Exit the program gracefully */
305 static void less_exit(int code)
306 {
307         set_tty_cooked();
308         if (!(G.kbd_fd_orig_flags & O_NONBLOCK))
309                 ndelay_off(kbd_fd);
310         clear_line();
311         if (code < 0)
312                 kill_myself_with_sig(- code); /* does not return */
313         exit(code);
314 }
315
316 #if (ENABLE_FEATURE_LESS_DASHCMD && ENABLE_FEATURE_LESS_LINENUMS) \
317  || ENABLE_FEATURE_LESS_WINCH
318 static void re_wrap(void)
319 {
320         int w = width;
321         int new_line_pos;
322         int src_idx;
323         int dst_idx;
324         int new_cur_fline = 0;
325         uint32_t lineno;
326         char linebuf[w + 1];
327         const char **old_flines = flines;
328         const char *s;
329         char **new_flines = NULL;
330         char *d;
331
332         if (option_mask32 & FLAG_N)
333                 w -= 8;
334
335         src_idx = 0;
336         dst_idx = 0;
337         s = old_flines[0];
338         lineno = LINENO(s);
339         d = linebuf;
340         new_line_pos = 0;
341         while (1) {
342                 *d = *s;
343                 if (*d != '\0') {
344                         new_line_pos++;
345                         if (*d == '\t') { /* tab */
346                                 new_line_pos += 7;
347                                 new_line_pos &= (~7);
348                         }
349                         s++;
350                         d++;
351                         if (new_line_pos >= w) {
352                                 int sz;
353                                 /* new line is full, create next one */
354                                 *d = '\0';
355  next_new:
356                                 sz = (d - linebuf) + 1; /* + 1: NUL */
357                                 d = ((char*)xmalloc(sz + 4)) + 4;
358                                 LINENO(d) = lineno;
359                                 memcpy(d, linebuf, sz);
360                                 new_flines = xrealloc_vector(new_flines, 8, dst_idx);
361                                 new_flines[dst_idx] = d;
362                                 dst_idx++;
363                                 if (new_line_pos < w) {
364                                         /* if we came here thru "goto next_new" */
365                                         if (src_idx > max_fline)
366                                                 break;
367                                         lineno = LINENO(s);
368                                 }
369                                 d = linebuf;
370                                 new_line_pos = 0;
371                         }
372                         continue;
373                 }
374                 /* *d == NUL: old line ended, go to next old one */
375                 free(MEMPTR(old_flines[src_idx]));
376                 /* btw, convert cur_fline... */
377                 if (cur_fline == src_idx)
378                         new_cur_fline = dst_idx;
379                 src_idx++;
380                 /* no more lines? finish last new line (and exit the loop) */
381                 if (src_idx > max_fline)
382                         goto next_new;
383                 s = old_flines[src_idx];
384                 if (lineno != LINENO(s)) {
385                         /* this is not a continuation line!
386                          * create next _new_ line too */
387                         goto next_new;
388                 }
389         }
390
391         free(old_flines);
392         flines = (const char **)new_flines;
393
394         max_fline = dst_idx - 1;
395         last_line_pos = new_line_pos;
396         cur_fline = new_cur_fline;
397         /* max_lineno is screen-size independent */
398 #if ENABLE_FEATURE_LESS_REGEXP
399         pattern_valid = 0;
400 #endif
401 }
402 #endif
403
404 #if ENABLE_FEATURE_LESS_REGEXP
405 static void fill_match_lines(unsigned pos);
406 #else
407 #define fill_match_lines(pos) ((void)0)
408 #endif
409
410 static int at_end(void)
411 {
412         return (option_mask32 & FLAG_S)
413                 ? !(cur_fline <= max_fline &&
414                         max_lineno > LINENO(flines[cur_fline]) + max_displayed_line)
415                 : !(max_fline > cur_fline + max_displayed_line);
416 }
417
418 /* Devilishly complex routine.
419  *
420  * Has to deal with EOF and EPIPE on input,
421  * with line wrapping, with last line not ending in '\n'
422  * (possibly not ending YET!), with backspace and tabs.
423  * It reads input again if last time we got an EOF (thus supporting
424  * growing files) or EPIPE (watching output of slow process like make).
425  *
426  * Variables used:
427  * flines[] - array of lines already read. Linewrap may cause
428  *      one source file line to occupy several flines[n].
429  * flines[max_fline] - last line, possibly incomplete.
430  * terminated - 1 if flines[max_fline] is 'terminated'
431  *      (if there was '\n' [which isn't stored itself, we just remember
432  *      that it was seen])
433  * max_lineno - last line's number, this one doesn't increment
434  *      on line wrap, only on "real" new lines.
435  * readbuf[0..readeof-1] - small preliminary buffer.
436  * readbuf[readpos] - next character to add to current line.
437  * last_line_pos - screen line position of next char to be read
438  *      (takes into account tabs and backspaces)
439  * eof_error - < 0 error, == 0 EOF, > 0 not EOF/error
440  *
441  * "git log -p | less -m" on the kernel git tree is a good test for EAGAINs,
442  * "/search on very long input" and "reaching max line count" corner cases.
443  */
444 static void read_lines(void)
445 {
446         char *current_line, *p;
447         int w = width;
448         char last_terminated = terminated;
449         time_t last_time = 0;
450         int retry_EAGAIN = 2;
451 #if ENABLE_FEATURE_LESS_REGEXP
452         unsigned old_max_fline = max_fline;
453 #endif
454
455 #define readbuf bb_common_bufsiz1
456         setup_common_bufsiz();
457
458         /* (careful: max_fline can be -1) */
459         if (max_fline + 1 > MAXLINES)
460                 return;
461
462         if (option_mask32 & FLAG_N)
463                 w -= 8;
464
465         p = current_line = ((char*)xmalloc(w + 5)) + 4;
466         if (!last_terminated) {
467                 const char *cp = flines[max_fline];
468                 p = stpcpy(p, cp);
469                 free(MEMPTR(cp));
470                 /* last_line_pos is still valid from previous read_lines() */
471         } else {
472                 max_fline++;
473                 last_line_pos = 0;
474         }
475
476         while (1) { /* read lines until we reach cur_fline or wanted_match */
477                 *p = '\0';
478                 terminated = 0;
479                 while (1) { /* read chars until we have a line */
480                         char c;
481                         /* if no unprocessed chars left, eat more */
482                         if (readpos >= readeof) {
483                                 int flags = ndelay_on(0);
484
485                                 while (1) {
486                                         time_t t;
487
488                                         errno = 0;
489                                         eof_error = safe_read(STDIN_FILENO, readbuf, COMMON_BUFSIZE);
490                                         if (errno != EAGAIN)
491                                                 break;
492                                         t = time(NULL);
493                                         if (t != last_time) {
494                                                 last_time = t;
495                                                 if (--retry_EAGAIN < 0)
496                                                         break;
497                                         }
498                                         sched_yield();
499                                 }
500                                 fcntl(0, F_SETFL, flags); /* ndelay_off(0) */
501                                 readpos = 0;
502                                 readeof = eof_error;
503                                 if (eof_error <= 0)
504                                         goto reached_eof;
505                                 retry_EAGAIN = 1;
506                         }
507                         c = readbuf[readpos];
508                         /* backspace? [needed for manpages] */
509                         /* <tab><bs> is (a) insane and */
510                         /* (b) harder to do correctly, so we refuse to do it */
511                         if (c == '\x8' && last_line_pos && p[-1] != '\t') {
512                                 readpos++; /* eat it */
513                                 last_line_pos--;
514                         /* was buggy (p could end up <= current_line)... */
515                                 *--p = '\0';
516                                 continue;
517                         }
518                         {
519                                 size_t new_last_line_pos = last_line_pos + 1;
520                                 if (c == '\t') {
521                                         new_last_line_pos += 7;
522                                         new_last_line_pos &= (~7);
523                                 }
524                                 if ((int)new_last_line_pos > w)
525                                         break;
526                                 last_line_pos = new_last_line_pos;
527                         }
528                         /* ok, we will eat this char */
529                         readpos++;
530                         if (c == '\n') {
531                                 terminated = 1;
532                                 last_line_pos = 0;
533                                 break;
534                         }
535                         /* NUL is substituted by '\n'! */
536                         if (c == '\0') c = '\n';
537                         *p++ = c;
538                         *p = '\0';
539                 } /* end of "read chars until we have a line" loop */
540 #if 0
541 //BUG: also triggers on this:
542 // { printf "\nfoo\n"; sleep 1; printf "\nbar\n"; } | less
543 // (resulting in lost empty line between "foo" and "bar" lines)
544 // the "terminated" logic needs fixing (or explaining)
545                 /* Corner case: linewrap with only "" wrapping to next line */
546                 /* Looks ugly on screen, so we do not store this empty line */
547                 if (!last_terminated && !current_line[0]) {
548                         last_terminated = 1;
549                         max_lineno++;
550                         continue;
551                 }
552 #endif
553  reached_eof:
554                 last_terminated = terminated;
555                 flines = xrealloc_vector(flines, 8, max_fline);
556
557                 flines[max_fline] = (char*)xrealloc(MEMPTR(current_line), strlen(current_line) + 1 + 4) + 4;
558                 LINENO(flines[max_fline]) = max_lineno;
559                 if (terminated)
560                         max_lineno++;
561
562                 if (max_fline >= MAXLINES) {
563                         eof_error = 0; /* Pretend we saw EOF */
564                         break;
565                 }
566                 if (!at_end()) {
567 #if !ENABLE_FEATURE_LESS_REGEXP
568                         break;
569 #else
570                         if (wanted_match >= num_matches) { /* goto_match called us */
571                                 fill_match_lines(old_max_fline);
572                                 old_max_fline = max_fline;
573                         }
574                         if (wanted_match < num_matches)
575                                 break;
576 #endif
577                 }
578                 if (eof_error <= 0) {
579                         break;
580                 }
581                 max_fline++;
582                 current_line = ((char*)xmalloc(w + 5)) + 4;
583                 p = current_line;
584                 last_line_pos = 0;
585         } /* end of "read lines until we reach cur_fline" loop */
586
587         if (eof_error < 0) {
588                 if (errno == EAGAIN) {
589                         eof_error = 1;
590                 } else {
591                         print_statusline(bb_msg_read_error);
592                 }
593         }
594 #if ENABLE_FEATURE_LESS_FLAGS
595         else if (eof_error == 0)
596                 num_lines = max_lineno;
597 #endif
598
599         fill_match_lines(old_max_fline);
600 #if ENABLE_FEATURE_LESS_REGEXP
601         /* prevent us from being stuck in search for a match */
602         wanted_match = -1;
603 #endif
604 #undef readbuf
605 }
606
607 #if ENABLE_FEATURE_LESS_FLAGS
608 static int safe_lineno(int fline)
609 {
610         if (fline >= max_fline)
611                 fline = max_fline - 1;
612
613         /* also catches empty file (max_fline == 0) */
614         if (fline < 0)
615                 return 0;
616
617         return LINENO(flines[fline]) + 1;
618 }
619
620 /* count number of lines in file */
621 static void update_num_lines(void)
622 {
623         int count, fd;
624         struct stat stbuf;
625         ssize_t len, i;
626         char buf[4096];
627
628         /* only do this for regular files */
629         if (num_lines == REOPEN_AND_COUNT || num_lines == REOPEN_STDIN) {
630                 count = 0;
631                 fd = open("/proc/self/fd/0", O_RDONLY);
632                 if (fd < 0 && num_lines == REOPEN_AND_COUNT) {
633                         /* "filename" is valid only if REOPEN_AND_COUNT */
634                         fd = open(filename, O_RDONLY);
635                 }
636                 if (fd < 0) {
637                         /* somebody stole my file! */
638                         num_lines = NOT_REGULAR_FILE;
639                         return;
640                 }
641                 if (fstat(fd, &stbuf) != 0 || !S_ISREG(stbuf.st_mode)) {
642                         num_lines = NOT_REGULAR_FILE;
643                         goto do_close;
644                 }
645                 while ((len = safe_read(fd, buf, sizeof(buf))) > 0) {
646                         for (i = 0; i < len; ++i) {
647                                 if (buf[i] == '\n' && ++count == MAXLINES)
648                                         goto done;
649                         }
650                 }
651  done:
652                 num_lines = count;
653  do_close:
654                 close(fd);
655         }
656 }
657
658 /* Print a status line if -M was specified */
659 static void m_status_print(void)
660 {
661         int first, last;
662         unsigned percent;
663
664         if (less_gets_pos >= 0) /* don't touch statusline while input is done! */
665                 return;
666
667         clear_line();
668         printf(HIGHLIGHT"%s", filename);
669         if (num_files > 1)
670                 printf(" (file %i of %i)", current_file, num_files);
671
672         first = safe_lineno(cur_fline);
673         last = (option_mask32 & FLAG_S)
674                         ? MIN(first + max_displayed_line, max_lineno)
675                         : safe_lineno(cur_fline + max_displayed_line);
676         printf(" lines %i-%i", first, last);
677
678         update_num_lines();
679         if (num_lines >= 0)
680                 printf("/%i", num_lines);
681
682         if (at_end()) {
683                 printf(" (END)");
684                 if (num_files > 1 && current_file != num_files)
685                         printf(" - next: %s", files[current_file]);
686         } else if (num_lines > 0) {
687                 percent = (100 * last + num_lines/2) / num_lines;
688                 printf(" %i%%", percent <= 100 ? percent : 100);
689         }
690         printf(NORMAL);
691 }
692 #endif
693
694 /* Print the status line */
695 static void status_print(void)
696 {
697         const char *p;
698
699         if (less_gets_pos >= 0) /* don't touch statusline while input is done! */
700                 return;
701
702         /* Change the status if flags have been set */
703 #if ENABLE_FEATURE_LESS_FLAGS
704         if (option_mask32 & (FLAG_M|FLAG_m)) {
705                 m_status_print();
706                 return;
707         }
708         /* No flags set */
709 #endif
710
711         clear_line();
712         if (cur_fline && !at_end()) {
713                 bb_putchar(':');
714                 return;
715         }
716         p = "(END)";
717         if (!cur_fline)
718                 p = filename;
719         if (num_files > 1) {
720                 printf(HIGHLIGHT"%s (file %i of %i)"NORMAL,
721                                 p, current_file, num_files);
722                 return;
723         }
724         print_hilite(p);
725 }
726
727 static const char controls[] ALIGN1 =
728         /* NUL: never encountered; TAB: not converted */
729         /**/"\x01\x02\x03\x04\x05\x06\x07\x08"  "\x0a\x0b\x0c\x0d\x0e\x0f"
730         "\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f"
731         "\x7f\x9b"; /* DEL and infamous Meta-ESC :( */
732 static const char ctrlconv[] ALIGN1 =
733         /* why 40 instead of 4a below? - it is a replacement for '\n'.
734          * '\n' is a former NUL - we subst it with @, not J */
735         "\x40\x41\x42\x43\x44\x45\x46\x47\x48\x49\x40\x4b\x4c\x4d\x4e\x4f"
736         "\x50\x51\x52\x53\x54\x55\x56\x57\x58\x59\x5a\x5b\x5c\x5d\x5e\x5f";
737
738 static void print_lineno(const char *line)
739 {
740         const char *fmt = "        ";
741         unsigned n = n; /* for compiler */
742
743         if (line != empty_line_marker) {
744                 /* Width of 7 preserves tab spacing in the text */
745                 fmt = "%7u ";
746                 n = LINENO(line) + 1;
747                 if (n > 9999999 && MAXLINES > 9999999) {
748                         n %= 10000000;
749                         fmt = "%07u ";
750                 }
751         }
752         printf(fmt, n);
753 }
754
755
756 #if ENABLE_FEATURE_LESS_REGEXP
757 static void print_found(const char *line)
758 {
759         int match_status;
760         int eflags;
761         char *growline;
762         regmatch_t match_structs;
763
764         char buf[width+1];
765         const char *str = line;
766         char *p = buf;
767         size_t n;
768
769         while (*str) {
770                 n = strcspn(str, controls);
771                 if (n) {
772                         if (!str[n]) break;
773                         memcpy(p, str, n);
774                         p += n;
775                         str += n;
776                 }
777                 n = strspn(str, controls);
778                 memset(p, '.', n);
779                 p += n;
780                 str += n;
781         }
782         strcpy(p, str);
783
784         /* buf[] holds quarantined version of str */
785
786         /* Each part of the line that matches has the HIGHLIGHT
787          * and NORMAL escape sequences placed around it.
788          * NB: we regex against line, but insert text
789          * from quarantined copy (buf[]) */
790         str = buf;
791         growline = NULL;
792         eflags = 0;
793         goto start;
794
795         while (match_status == 0) {
796                 char *new = xasprintf("%s%.*s"HIGHLIGHT"%.*s"NORMAL,
797                                 growline ? growline : "",
798                                 (int)match_structs.rm_so, str,
799                                 (int)(match_structs.rm_eo - match_structs.rm_so),
800                                                 str + match_structs.rm_so);
801                 free(growline);
802                 growline = new;
803                 str += match_structs.rm_eo;
804                 line += match_structs.rm_eo;
805                 eflags = REG_NOTBOL;
806  start:
807                 /* Most of the time doesn't find the regex, optimize for that */
808                 match_status = regexec(&pattern, line, 1, &match_structs, eflags);
809                 /* if even "" matches, treat it as "not a match" */
810                 if (match_structs.rm_so >= match_structs.rm_eo)
811                         match_status = 1;
812         }
813
814         printf("%s%s\n", growline ? growline : "", str);
815         free(growline);
816 }
817 #else
818 void print_found(const char *line);
819 #endif
820
821 static void print_ascii(const char *str)
822 {
823         char buf[width+1];
824         char *p;
825         size_t n;
826
827         while (*str) {
828                 n = strcspn(str, controls);
829                 if (n) {
830                         if (!str[n]) break;
831                         printf("%.*s", (int) n, str);
832                         str += n;
833                 }
834                 n = strspn(str, controls);
835                 p = buf;
836                 do {
837                         if (*str == 0x7f)
838                                 *p++ = '?';
839                         else if (*str == (char)0x9b)
840                         /* VT100's CSI, aka Meta-ESC. Who's inventor? */
841                         /* I want to know who committed this sin */
842                                 *p++ = '{';
843                         else
844                                 *p++ = ctrlconv[(unsigned char)*str];
845                         str++;
846                 } while (--n);
847                 *p = '\0';
848                 print_hilite(buf);
849         }
850         puts(str);
851 }
852
853 /* Print the buffer */
854 static void buffer_print(void)
855 {
856         unsigned i;
857
858         move_cursor(0, 0);
859         for (i = 0; i <= max_displayed_line; i++) {
860                 printf(CLEAR_2_EOL);
861                 if (option_mask32 & FLAG_N)
862                         print_lineno(buffer[i]);
863                 if (pattern_valid)
864                         print_found(buffer[i]);
865                 else
866                         print_ascii(buffer[i]);
867         }
868         if ((option_mask32 & FLAG_E)
869          && eof_error <= 0
870          && (max_fline - cur_fline) <= max_displayed_line
871         ) {
872                 less_exit(EXIT_SUCCESS);
873         }
874         status_print();
875 }
876
877 static void buffer_fill_and_print(void)
878 {
879         unsigned i;
880 #if ENABLE_FEATURE_LESS_TRUNCATE
881         int fpos = cur_fline;
882
883         if (option_mask32 & FLAG_S) {
884                 /* Go back to the beginning of this line */
885                 while (fpos && LINENO(flines[fpos]) == LINENO(flines[fpos-1]))
886                         fpos--;
887         }
888
889         i = 0;
890         while (i <= max_displayed_line && fpos <= max_fline) {
891                 int lineno = LINENO(flines[fpos]);
892                 buffer[i] = flines[fpos];
893                 i++;
894                 do {
895                         fpos++;
896                 } while ((fpos <= max_fline)
897                       && (option_mask32 & FLAG_S)
898                       && lineno == LINENO(flines[fpos])
899                 );
900         }
901 #else
902         for (i = 0; i <= max_displayed_line && cur_fline + i <= max_fline; i++) {
903                 buffer[i] = flines[cur_fline + i];
904         }
905 #endif
906         for (; i <= max_displayed_line; i++) {
907                 buffer[i] = empty_line_marker;
908         }
909         buffer_print();
910 }
911
912 /* move cur_fline to a given line number, reading lines if necessary */
913 static void goto_lineno(int target)
914 {
915         if (target <= 0 ) {
916                 cur_fline = 0;
917         }
918         else if (target > LINENO(flines[cur_fline])) {
919  retry:
920                 while (LINENO(flines[cur_fline]) != target && cur_fline < max_fline)
921                         ++cur_fline;
922                 /* target not reached but more input is available */
923                 if (LINENO(flines[cur_fline]) != target && eof_error > 0) {
924                         read_lines();
925                         goto retry;
926                 }
927         }
928         else {
929                 /* search backwards through already-read lines */
930                 while (LINENO(flines[cur_fline]) != target && cur_fline > 0)
931                         --cur_fline;
932         }
933 }
934
935 static void cap_cur_fline(void)
936 {
937         if ((option_mask32 & FLAG_S)) {
938                 if (cur_fline > max_fline)
939                         cur_fline = max_fline;
940                 if (LINENO(flines[cur_fline]) + max_displayed_line > max_lineno + TILDES) {
941                         goto_lineno(max_lineno - max_displayed_line + TILDES);
942                         read_lines();
943                 }
944         }
945         else {
946                 if (cur_fline + max_displayed_line > max_fline + TILDES)
947                         cur_fline = max_fline - max_displayed_line + TILDES;
948                 if (cur_fline < 0)
949                         cur_fline = 0;
950         }
951 }
952
953 /* Move the buffer up and down in the file in order to scroll */
954 static void buffer_down(int nlines)
955 {
956         if ((option_mask32 & FLAG_S))
957                 goto_lineno(LINENO(flines[cur_fline]) + nlines);
958         else
959                 cur_fline += nlines;
960         read_lines();
961         cap_cur_fline();
962         buffer_fill_and_print();
963 }
964
965 static void buffer_up(int nlines)
966 {
967         if ((option_mask32 & FLAG_S)) {
968                 goto_lineno(LINENO(flines[cur_fline]) - nlines);
969         }
970         else {
971                 cur_fline -= nlines;
972                 if (cur_fline < 0)
973                         cur_fline = 0;
974         }
975         read_lines();
976         buffer_fill_and_print();
977 }
978
979 /* display a given line where the argument can be either an index into
980  * the flines array or a line number */
981 static void buffer_to_line(int linenum, int is_lineno)
982 {
983         if (linenum <= 0)
984                 cur_fline = 0;
985         else if (is_lineno)
986                 goto_lineno(linenum);
987         else
988                 cur_fline = linenum;
989         read_lines();
990         cap_cur_fline();
991         buffer_fill_and_print();
992 }
993
994 static void buffer_line(int linenum)
995 {
996         buffer_to_line(linenum, FALSE);
997 }
998
999 static void buffer_lineno(int lineno)
1000 {
1001         buffer_to_line(lineno, TRUE);
1002 }
1003
1004 static void open_file_and_read_lines(void)
1005 {
1006         if (filename) {
1007                 xmove_fd(xopen(filename, O_RDONLY), STDIN_FILENO);
1008 #if ENABLE_FEATURE_LESS_FLAGS
1009                 num_lines = REOPEN_AND_COUNT;
1010 #endif
1011         } else {
1012                 /* "less" with no arguments in argv[] */
1013                 /* For status line only */
1014                 filename = xstrdup(bb_msg_standard_input);
1015 #if ENABLE_FEATURE_LESS_FLAGS
1016                 num_lines = REOPEN_STDIN;
1017 #endif
1018         }
1019         readpos = 0;
1020         readeof = 0;
1021         last_line_pos = 0;
1022         terminated = 1;
1023         read_lines();
1024 }
1025
1026 /* Reinitialize everything for a new file - free the memory and start over */
1027 static void reinitialize(void)
1028 {
1029         unsigned i;
1030
1031         if (flines) {
1032                 for (i = 0; i <= max_fline; i++)
1033                         free(MEMPTR(flines[i]));
1034                 free(flines);
1035                 flines = NULL;
1036         }
1037
1038         max_fline = -1;
1039         cur_fline = 0;
1040         max_lineno = 0;
1041         open_file_and_read_lines();
1042 #if ENABLE_FEATURE_LESS_ASK_TERMINAL
1043         if (G.winsize_err)
1044                 printf("\033[999;999H" "\033[6n");
1045 #endif
1046         buffer_fill_and_print();
1047 }
1048
1049 static int64_t getch_nowait(void)
1050 {
1051         int rd;
1052         int64_t key64;
1053         struct pollfd pfd[2];
1054
1055         pfd[0].fd = STDIN_FILENO;
1056         pfd[0].events = POLLIN;
1057         pfd[1].fd = kbd_fd;
1058         pfd[1].events = POLLIN;
1059  again:
1060         tcsetattr(kbd_fd, TCSANOW, &term_less);
1061         /* NB: select/poll returns whenever read will not block. Therefore:
1062          * if eof is reached, select/poll will return immediately
1063          * because read will immediately return 0 bytes.
1064          * Even if select/poll says that input is available, read CAN block
1065          * (switch fd into O_NONBLOCK'ed mode to avoid it)
1066          */
1067         rd = 1;
1068         /* Are we interested in stdin? */
1069         if (at_end()) {
1070                 if (eof_error > 0) /* did NOT reach eof yet */
1071                         rd = 0; /* yes, we are interested in stdin */
1072         }
1073         /* Position cursor if line input is done */
1074         if (less_gets_pos >= 0)
1075                 move_cursor(max_displayed_line + 2, less_gets_pos + 1);
1076         fflush_all();
1077
1078         if (kbd_input[0] == 0) { /* if nothing is buffered */
1079 #if ENABLE_FEATURE_LESS_WINCH
1080                 while (1) {
1081                         int r;
1082                         /* NB: SIGWINCH interrupts poll() */
1083                         r = poll(pfd + rd, 2 - rd, -1);
1084                         if (/*r < 0 && errno == EINTR &&*/ winch_counter)
1085                                 return '\\'; /* anything which has no defined function */
1086                         if (r) break;
1087                 }
1088 #else
1089                 safe_poll(pfd + rd, 2 - rd, -1);
1090 #endif
1091         }
1092
1093         /* We have kbd_fd in O_NONBLOCK mode, read inside read_key()
1094          * would not block even if there is no input available */
1095         key64 = read_key(kbd_fd, kbd_input, /*timeout off:*/ -2);
1096         if ((int)key64 == -1) {
1097                 if (errno == EAGAIN) {
1098                         /* No keyboard input available. Since poll() did return,
1099                          * we should have input on stdin */
1100                         read_lines();
1101                         buffer_fill_and_print();
1102                         goto again;
1103                 }
1104                 /* EOF/error (ssh session got killed etc) */
1105                 less_exit(0);
1106         }
1107         set_tty_cooked();
1108         return key64;
1109 }
1110
1111 /* Grab a character from input without requiring the return key.
1112  * May return KEYCODE_xxx values.
1113  * Note that this function works best with raw input. */
1114 static int64_t less_getch(int pos)
1115 {
1116         int64_t key64;
1117         int key;
1118
1119  again:
1120         less_gets_pos = pos;
1121         key = key64 = getch_nowait();
1122         less_gets_pos = -1;
1123
1124         /* Discard Ctrl-something chars.
1125          * (checking only lower 32 bits is a size optimization:
1126          * upper 32 bits are used only by KEYCODE_CURSOR_POS)
1127          */
1128         if (key >= 0 && key < ' ' && key != 0x0d && key != 8)
1129                 goto again;
1130
1131         return key64;
1132 }
1133
1134 static char* less_gets(int sz)
1135 {
1136         int c;
1137         unsigned i = 0;
1138         char *result = xzalloc(1);
1139
1140         while (1) {
1141                 c = '\0';
1142                 less_gets_pos = sz + i;
1143                 c = getch_nowait();
1144                 if (c == 0x0d) {
1145                         result[i] = '\0';
1146                         less_gets_pos = -1;
1147                         return result;
1148                 }
1149                 if (c == 0x7f)
1150                         c = 8;
1151                 if (c == 8 && i) {
1152                         printf("\x8 \x8");
1153                         i--;
1154                 }
1155                 if (c < ' ') /* filters out KEYCODE_xxx too (<0) */
1156                         continue;
1157                 if (i >= width - sz - 1)
1158                         continue; /* len limit */
1159                 bb_putchar(c);
1160                 result[i++] = c;
1161                 result = xrealloc(result, i+1);
1162         }
1163 }
1164
1165 static void examine_file(void)
1166 {
1167         char *new_fname;
1168
1169         print_statusline("Examine: ");
1170         new_fname = less_gets(sizeof("Examine: ") - 1);
1171         if (!new_fname[0]) {
1172                 status_print();
1173  err:
1174                 free(new_fname);
1175                 return;
1176         }
1177         if (access(new_fname, R_OK) != 0) {
1178                 print_statusline("Cannot read this file");
1179                 goto err;
1180         }
1181         free(filename);
1182         filename = new_fname;
1183         /* files start by = argv. why we assume that argv is infinitely long??
1184         files[num_files] = filename;
1185         current_file = num_files + 1;
1186         num_files++; */
1187         files[0] = filename;
1188         num_files = current_file = 1;
1189         reinitialize();
1190 }
1191
1192 /* This function changes the file currently being paged. direction can be one of the following:
1193  * -1: go back one file
1194  *  0: go to the first file
1195  *  1: go forward one file */
1196 static void change_file(int direction)
1197 {
1198         if (current_file != ((direction > 0) ? num_files : 1)) {
1199                 current_file = direction ? current_file + direction : 1;
1200                 free(filename);
1201                 filename = xstrdup(files[current_file - 1]);
1202                 reinitialize();
1203         } else {
1204                 print_statusline(direction > 0 ? "No next file" : "No previous file");
1205         }
1206 }
1207
1208 static void remove_current_file(void)
1209 {
1210         unsigned i;
1211
1212         if (num_files < 2)
1213                 return;
1214
1215         if (current_file != 1) {
1216                 change_file(-1);
1217                 for (i = 3; i <= num_files; i++)
1218                         files[i - 2] = files[i - 1];
1219                 num_files--;
1220         } else {
1221                 change_file(1);
1222                 for (i = 2; i <= num_files; i++)
1223                         files[i - 2] = files[i - 1];
1224                 num_files--;
1225                 current_file--;
1226         }
1227 }
1228
1229 static void colon_process(void)
1230 {
1231         int keypress;
1232
1233         /* Clear the current line and print a prompt */
1234         print_statusline(" :");
1235
1236         keypress = less_getch(2);
1237         switch (keypress) {
1238         case 'd':
1239                 remove_current_file();
1240                 break;
1241         case 'e':
1242                 examine_file();
1243                 break;
1244 #if ENABLE_FEATURE_LESS_FLAGS
1245         case 'f':
1246                 m_status_print();
1247                 break;
1248 #endif
1249         case 'n':
1250                 change_file(1);
1251                 break;
1252         case 'p':
1253                 change_file(-1);
1254                 break;
1255         case 'q':
1256                 less_exit(EXIT_SUCCESS);
1257                 break;
1258         case 'x':
1259                 change_file(0);
1260                 break;
1261         }
1262 }
1263
1264 #if ENABLE_FEATURE_LESS_REGEXP
1265 static void normalize_match_pos(int match)
1266 {
1267         if (match >= num_matches)
1268                 match = num_matches - 1;
1269         if (match < 0)
1270                 match = 0;
1271         match_pos = match;
1272 }
1273
1274 static void goto_match(int match)
1275 {
1276         if (!pattern_valid)
1277                 return;
1278         if (match < 0)
1279                 match = 0;
1280         /* Try to find next match if eof isn't reached yet */
1281         if (match >= num_matches && eof_error > 0) {
1282                 wanted_match = match; /* "I want to read until I see N'th match" */
1283                 read_lines();
1284         }
1285         if (num_matches) {
1286                 normalize_match_pos(match);
1287                 buffer_line(match_lines[match_pos]);
1288         } else {
1289                 print_statusline("No matches found");
1290         }
1291 }
1292
1293 static void fill_match_lines(unsigned pos)
1294 {
1295         if (!pattern_valid)
1296                 return;
1297         /* Run the regex on each line of the current file */
1298         while (pos <= max_fline) {
1299                 /* If this line matches */
1300                 if (regexec(&pattern, flines[pos], 0, NULL, 0) == 0
1301                 /* and we didn't match it last time */
1302                  && !(num_matches && match_lines[num_matches-1] == pos)
1303                 ) {
1304                         match_lines = xrealloc_vector(match_lines, 4, num_matches);
1305                         match_lines[num_matches++] = pos;
1306                 }
1307                 pos++;
1308         }
1309 }
1310
1311 static void regex_process(void)
1312 {
1313         char *uncomp_regex, *err;
1314
1315         /* Reset variables */
1316         free(match_lines);
1317         match_lines = NULL;
1318         match_pos = 0;
1319         num_matches = 0;
1320         if (pattern_valid) {
1321                 regfree(&pattern);
1322                 pattern_valid = 0;
1323         }
1324
1325         /* Get the uncompiled regular expression from the user */
1326         clear_line();
1327         bb_putchar((option_mask32 & LESS_STATE_MATCH_BACKWARDS) ? '?' : '/');
1328         uncomp_regex = less_gets(1);
1329         if (!uncomp_regex[0]) {
1330                 free(uncomp_regex);
1331                 buffer_print();
1332                 return;
1333         }
1334
1335         /* Compile the regex and check for errors */
1336         err = regcomp_or_errmsg(&pattern, uncomp_regex,
1337                                 (option_mask32 & FLAG_I) ? REG_ICASE : 0);
1338         free(uncomp_regex);
1339         if (err) {
1340                 print_statusline(err);
1341                 free(err);
1342                 return;
1343         }
1344
1345         pattern_valid = 1;
1346         match_pos = 0;
1347         fill_match_lines(0);
1348         while (match_pos < num_matches) {
1349                 if ((int)match_lines[match_pos] > cur_fline)
1350                         break;
1351                 match_pos++;
1352         }
1353         if (option_mask32 & LESS_STATE_MATCH_BACKWARDS)
1354                 match_pos--;
1355
1356         /* It's possible that no matches are found yet.
1357          * goto_match() will read input looking for match,
1358          * if needed */
1359         goto_match(match_pos);
1360 }
1361 #endif
1362
1363 static void number_process(int first_digit)
1364 {
1365         unsigned i;
1366         int num;
1367         int keypress;
1368         char num_input[sizeof(int)*4]; /* more than enough */
1369
1370         num_input[0] = first_digit;
1371
1372         /* Clear the current line, print a prompt, and then print the digit */
1373         clear_line();
1374         printf(":%c", first_digit);
1375
1376         /* Receive input until a letter is given */
1377         i = 1;
1378         while (i < sizeof(num_input)-1) {
1379                 keypress = less_getch(i + 1);
1380                 if ((unsigned)keypress > 255 || !isdigit(keypress))
1381                         break;
1382                 num_input[i] = keypress;
1383                 bb_putchar(keypress);
1384                 i++;
1385         }
1386
1387         num_input[i] = '\0';
1388         num = bb_strtou(num_input, NULL, 10);
1389         /* on format error, num == -1 */
1390         if (num < 1 || num > MAXLINES) {
1391                 buffer_print();
1392                 return;
1393         }
1394
1395         /* We now know the number and the letter entered, so we process them */
1396         switch (keypress) {
1397         case KEYCODE_DOWN: case 'z': case 'd': case 'e': case ' ': case '\015':
1398                 buffer_down(num);
1399                 break;
1400         case KEYCODE_UP: case 'b': case 'w': case 'y': case 'u':
1401                 buffer_up(num);
1402                 break;
1403         case 'g': case '<': case 'G': case '>':
1404                 buffer_lineno(num - 1);
1405                 break;
1406         case 'p': case '%':
1407 #if ENABLE_FEATURE_LESS_FLAGS
1408                 update_num_lines();
1409                 num = num * (num_lines > 0 ? num_lines : max_lineno) / 100;
1410 #else
1411                 num = num * max_lineno / 100;
1412 #endif
1413                 buffer_lineno(num);
1414                 break;
1415 #if ENABLE_FEATURE_LESS_REGEXP
1416         case 'n':
1417                 goto_match(match_pos + num);
1418                 break;
1419         case '/':
1420                 option_mask32 &= ~LESS_STATE_MATCH_BACKWARDS;
1421                 regex_process();
1422                 break;
1423         case '?':
1424                 option_mask32 |= LESS_STATE_MATCH_BACKWARDS;
1425                 regex_process();
1426                 break;
1427 #endif
1428         }
1429 }
1430
1431 #if ENABLE_FEATURE_LESS_DASHCMD
1432 static void flag_change(void)
1433 {
1434         int keypress;
1435
1436         clear_line();
1437         bb_putchar('-');
1438         keypress = less_getch(1);
1439
1440         switch (keypress) {
1441         case 'M':
1442                 option_mask32 ^= FLAG_M;
1443                 break;
1444         case 'm':
1445                 option_mask32 ^= FLAG_m;
1446                 break;
1447         case 'E':
1448                 option_mask32 ^= FLAG_E;
1449                 break;
1450         case '~':
1451                 option_mask32 ^= FLAG_TILDE;
1452                 break;
1453 #if ENABLE_FEATURE_LESS_TRUNCATE
1454         case 'S':
1455                 option_mask32 ^= FLAG_S;
1456                 buffer_fill_and_print();
1457                 break;
1458 #endif
1459 #if ENABLE_FEATURE_LESS_LINENUMS
1460         case 'N':
1461                 option_mask32 ^= FLAG_N;
1462                 re_wrap();
1463                 buffer_fill_and_print();
1464                 break;
1465 #endif
1466         }
1467 }
1468
1469 #ifdef BLOAT
1470 static void show_flag_status(void)
1471 {
1472         int keypress;
1473         int flag_val;
1474
1475         clear_line();
1476         bb_putchar('_');
1477         keypress = less_getch(1);
1478
1479         switch (keypress) {
1480         case 'M':
1481                 flag_val = option_mask32 & FLAG_M;
1482                 break;
1483         case 'm':
1484                 flag_val = option_mask32 & FLAG_m;
1485                 break;
1486         case '~':
1487                 flag_val = option_mask32 & FLAG_TILDE;
1488                 break;
1489         case 'N':
1490                 flag_val = option_mask32 & FLAG_N;
1491                 break;
1492         case 'E':
1493                 flag_val = option_mask32 & FLAG_E;
1494                 break;
1495         default:
1496                 flag_val = 0;
1497                 break;
1498         }
1499
1500         clear_line();
1501         printf(HIGHLIGHT"The status of the flag is: %u"NORMAL, flag_val != 0);
1502 }
1503 #endif
1504
1505 #endif /* ENABLE_FEATURE_LESS_DASHCMD */
1506
1507 static void save_input_to_file(void)
1508 {
1509         const char *msg = "";
1510         char *current_line;
1511         unsigned i;
1512         FILE *fp;
1513
1514         print_statusline("Log file: ");
1515         current_line = less_gets(sizeof("Log file: ")-1);
1516         if (current_line[0]) {
1517                 fp = fopen_for_write(current_line);
1518                 if (!fp) {
1519                         msg = "Error opening log file";
1520                         goto ret;
1521                 }
1522                 for (i = 0; i <= max_fline; i++)
1523                         fprintf(fp, "%s\n", flines[i]);
1524                 fclose(fp);
1525                 msg = "Done";
1526         }
1527  ret:
1528         print_statusline(msg);
1529         free(current_line);
1530 }
1531
1532 #if ENABLE_FEATURE_LESS_MARKS
1533 static void add_mark(void)
1534 {
1535         int letter;
1536
1537         print_statusline("Mark: ");
1538         letter = less_getch(sizeof("Mark: ") - 1);
1539
1540         if (isalpha(letter)) {
1541                 /* If we exceed 15 marks, start overwriting previous ones */
1542                 if (num_marks == 14)
1543                         num_marks = 0;
1544
1545                 mark_lines[num_marks][0] = letter;
1546                 mark_lines[num_marks][1] = cur_fline;
1547                 num_marks++;
1548         } else {
1549                 print_statusline("Invalid mark letter");
1550         }
1551 }
1552
1553 static void goto_mark(void)
1554 {
1555         int letter;
1556         int i;
1557
1558         print_statusline("Go to mark: ");
1559         letter = less_getch(sizeof("Go to mark: ") - 1);
1560         clear_line();
1561
1562         if (isalpha(letter)) {
1563                 for (i = 0; i <= num_marks; i++)
1564                         if (letter == mark_lines[i][0]) {
1565                                 buffer_line(mark_lines[i][1]);
1566                                 break;
1567                         }
1568                 if (num_marks == 14 && letter != mark_lines[14][0])
1569                         print_statusline("Mark not set");
1570         } else
1571                 print_statusline("Invalid mark letter");
1572 }
1573 #endif
1574
1575 #if ENABLE_FEATURE_LESS_BRACKETS
1576 static char opp_bracket(char bracket)
1577 {
1578         switch (bracket) {
1579                 case '{': case '[': /* '}' == '{' + 2. Same for '[' */
1580                         bracket++;
1581                 case '(':           /* ')' == '(' + 1 */
1582                         bracket++;
1583                         break;
1584                 case '}': case ']':
1585                         bracket--;
1586                 case ')':
1587                         bracket--;
1588                         break;
1589         };
1590         return bracket;
1591 }
1592
1593 static void match_right_bracket(char bracket)
1594 {
1595         unsigned i = cur_fline;
1596
1597         if (i >= max_fline
1598          || strchr(flines[i], bracket) == NULL
1599         ) {
1600                 print_statusline("No bracket in top line");
1601                 return;
1602         }
1603
1604         bracket = opp_bracket(bracket);
1605         for (; i < max_fline; i++) {
1606                 if (strchr(flines[i], bracket) != NULL) {
1607                         /*
1608                          * Line with matched right bracket becomes
1609                          * last visible line
1610                          */
1611                         buffer_line(i - max_displayed_line);
1612                         return;
1613                 }
1614         }
1615         print_statusline("No matching bracket found");
1616 }
1617
1618 static void match_left_bracket(char bracket)
1619 {
1620         int i = cur_fline + max_displayed_line;
1621
1622         if (i >= max_fline
1623          || strchr(flines[i], bracket) == NULL
1624         ) {
1625                 print_statusline("No bracket in bottom line");
1626                 return;
1627         }
1628
1629         bracket = opp_bracket(bracket);
1630         for (; i >= 0; i--) {
1631                 if (strchr(flines[i], bracket) != NULL) {
1632                         /*
1633                          * Line with matched left bracket becomes
1634                          * first visible line
1635                          */
1636                         buffer_line(i);
1637                         return;
1638                 }
1639         }
1640         print_statusline("No matching bracket found");
1641 }
1642 #endif  /* FEATURE_LESS_BRACKETS */
1643
1644 static void keypress_process(int keypress)
1645 {
1646         switch (keypress) {
1647         case KEYCODE_DOWN: case 'e': case 'j': case 0x0d:
1648                 buffer_down(1);
1649                 break;
1650         case KEYCODE_UP: case 'y': case 'k':
1651                 buffer_up(1);
1652                 break;
1653         case KEYCODE_PAGEDOWN: case ' ': case 'z': case 'f':
1654                 buffer_down(max_displayed_line + 1);
1655                 break;
1656         case KEYCODE_PAGEUP: case 'w': case 'b':
1657                 buffer_up(max_displayed_line + 1);
1658                 break;
1659         case 'd':
1660                 buffer_down((max_displayed_line + 1) / 2);
1661                 break;
1662         case 'u':
1663                 buffer_up((max_displayed_line + 1) / 2);
1664                 break;
1665         case KEYCODE_HOME: case 'g': case 'p': case '<': case '%':
1666                 buffer_line(0);
1667                 break;
1668         case KEYCODE_END: case 'G': case '>':
1669                 cur_fline = MAXLINES;
1670                 read_lines();
1671                 buffer_line(cur_fline);
1672                 break;
1673         case 'q': case 'Q':
1674                 less_exit(EXIT_SUCCESS);
1675                 break;
1676 #if ENABLE_FEATURE_LESS_MARKS
1677         case 'm':
1678                 add_mark();
1679                 buffer_print();
1680                 break;
1681         case '\'':
1682                 goto_mark();
1683                 buffer_print();
1684                 break;
1685 #endif
1686         case 'r': case 'R':
1687                 /* TODO: (1) also bind ^R, ^L to this?
1688                  * (2) re-measure window size?
1689                  */
1690                 buffer_print();
1691                 break;
1692         /*case 'R':
1693                 full_repaint();
1694                 break;*/
1695         case 's':
1696                 save_input_to_file();
1697                 break;
1698         case 'E':
1699                 examine_file();
1700                 break;
1701 #if ENABLE_FEATURE_LESS_FLAGS
1702         case '=':
1703                 m_status_print();
1704                 break;
1705 #endif
1706 #if ENABLE_FEATURE_LESS_REGEXP
1707         case '/':
1708                 option_mask32 &= ~LESS_STATE_MATCH_BACKWARDS;
1709                 regex_process();
1710                 break;
1711         case 'n':
1712                 goto_match(match_pos + 1);
1713                 break;
1714         case 'N':
1715                 goto_match(match_pos - 1);
1716                 break;
1717         case '?':
1718                 option_mask32 |= LESS_STATE_MATCH_BACKWARDS;
1719                 regex_process();
1720                 break;
1721 #endif
1722 #if ENABLE_FEATURE_LESS_DASHCMD
1723         case '-':
1724                 flag_change();
1725                 buffer_print();
1726                 break;
1727 #ifdef BLOAT
1728         case '_':
1729                 show_flag_status();
1730                 break;
1731 #endif
1732 #endif
1733 #if ENABLE_FEATURE_LESS_BRACKETS
1734         case '{': case '(': case '[':
1735                 match_right_bracket(keypress);
1736                 break;
1737         case '}': case ')': case ']':
1738                 match_left_bracket(keypress);
1739                 break;
1740 #endif
1741         case ':':
1742                 colon_process();
1743                 break;
1744         }
1745
1746         if (isdigit(keypress))
1747                 number_process(keypress);
1748 }
1749
1750 static void sig_catcher(int sig)
1751 {
1752         less_exit(- sig);
1753 }
1754
1755 #if ENABLE_FEATURE_LESS_WINCH
1756 static void sigwinch_handler(int sig UNUSED_PARAM)
1757 {
1758         winch_counter++;
1759 }
1760 #endif
1761
1762 int less_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1763 int less_main(int argc, char **argv)
1764 {
1765         char *tty_name;
1766         int tty_fd;
1767
1768         INIT_G();
1769
1770         /* TODO: -x: do not interpret backspace, -xx: tab also
1771          * -xxx: newline also
1772          * -w N: assume width N (-xxx -w 32: hex viewer of sorts)
1773          * -s: condense many empty lines to one
1774          *     (used by some setups for manpage display)
1775          */
1776         getopt32(argv, "EMmN~I" IF_FEATURE_LESS_TRUNCATE("S") /*ignored:*/"s");
1777         argc -= optind;
1778         argv += optind;
1779         num_files = argc;
1780         files = argv;
1781
1782         /* Another popular pager, most, detects when stdout
1783          * is not a tty and turns into cat. This makes sense. */
1784         if (!isatty(STDOUT_FILENO))
1785                 return bb_cat(argv);
1786
1787         if (!num_files) {
1788                 if (isatty(STDIN_FILENO)) {
1789                         /* Just "less"? No args and no redirection? */
1790                         bb_error_msg("missing filename");
1791                         bb_show_usage();
1792                 }
1793         } else {
1794                 filename = xstrdup(files[0]);
1795         }
1796
1797         if (option_mask32 & FLAG_TILDE)
1798                 empty_line_marker = "";
1799
1800         /* Some versions of less can survive w/o controlling tty,
1801          * try to do the same. This also allows to specify an alternative
1802          * tty via "less 1<>TTY".
1803          *
1804          * We prefer not to use STDOUT_FILENO directly,
1805          * since we want to set this fd to non-blocking mode,
1806          * and not interfere with other processes which share stdout with us.
1807          */
1808         tty_name = xmalloc_ttyname(STDOUT_FILENO);
1809         if (tty_name) {
1810                 tty_fd = open(tty_name, O_RDONLY);
1811                 free(tty_name);
1812                 if (tty_fd < 0)
1813                         goto try_ctty;
1814         } else {
1815                 /* Try controlling tty */
1816  try_ctty:
1817                 tty_fd = open(CURRENT_TTY, O_RDONLY);
1818                 if (tty_fd < 0) {
1819                         /* If all else fails, less 481 uses stdout. Mimic that */
1820                         tty_fd = STDOUT_FILENO;
1821                 }
1822         }
1823         G.kbd_fd_orig_flags = ndelay_on(tty_fd);
1824         kbd_fd = tty_fd; /* save in a global */
1825
1826         tcgetattr(kbd_fd, &term_orig);
1827         term_less = term_orig;
1828         term_less.c_lflag &= ~(ICANON | ECHO);
1829         term_less.c_iflag &= ~(IXON | ICRNL);
1830         /*term_less.c_oflag &= ~ONLCR;*/
1831         term_less.c_cc[VMIN] = 1;
1832         term_less.c_cc[VTIME] = 0;
1833
1834         IF_FEATURE_LESS_ASK_TERMINAL(G.winsize_err =) get_terminal_width_height(kbd_fd, &width, &max_displayed_line);
1835         /* 20: two tabstops + 4 */
1836         if (width < 20 || max_displayed_line < 3)
1837                 return bb_cat(argv);
1838         max_displayed_line -= 2;
1839
1840         /* We want to restore term_orig on exit */
1841         bb_signals(BB_FATAL_SIGS, sig_catcher);
1842 #if ENABLE_FEATURE_LESS_WINCH
1843         signal(SIGWINCH, sigwinch_handler);
1844 #endif
1845
1846         buffer = xmalloc((max_displayed_line+1) * sizeof(char *));
1847         reinitialize();
1848         while (1) {
1849                 int64_t keypress;
1850
1851 #if ENABLE_FEATURE_LESS_WINCH
1852                 while (WINCH_COUNTER) {
1853  again:
1854                         winch_counter--;
1855                         IF_FEATURE_LESS_ASK_TERMINAL(G.winsize_err =) get_terminal_width_height(kbd_fd, &width, &max_displayed_line);
1856  IF_FEATURE_LESS_ASK_TERMINAL(got_size:)
1857                         /* 20: two tabstops + 4 */
1858                         if (width < 20)
1859                                 width = 20;
1860                         if (max_displayed_line < 3)
1861                                 max_displayed_line = 3;
1862                         max_displayed_line -= 2;
1863                         free(buffer);
1864                         buffer = xmalloc((max_displayed_line+1) * sizeof(char *));
1865                         /* Avoid re-wrap and/or redraw if we already know
1866                          * we need to do it again. These ops are expensive */
1867                         if (WINCH_COUNTER)
1868                                 goto again;
1869                         re_wrap();
1870                         if (WINCH_COUNTER)
1871                                 goto again;
1872                         buffer_fill_and_print();
1873                         /* This took some time. Loop back and check,
1874                          * were there another SIGWINCH? */
1875                 }
1876                 keypress = less_getch(-1); /* -1: do not position cursor */
1877 # if ENABLE_FEATURE_LESS_ASK_TERMINAL
1878                 if ((int32_t)keypress == KEYCODE_CURSOR_POS) {
1879                         uint32_t rc = (keypress >> 32);
1880                         width = (rc & 0x7fff);
1881                         max_displayed_line = ((rc >> 16) & 0x7fff);
1882                         goto got_size;
1883                 }
1884 # endif
1885 #else
1886                 keypress = less_getch(-1); /* -1: do not position cursor */
1887 #endif
1888                 keypress_process(keypress);
1889         }
1890 }
1891
1892 /*
1893 Help text of less version 418 is below.
1894 If you are implementing something, keeping
1895 key and/or command line switch compatibility is a good idea:
1896
1897
1898                    SUMMARY OF LESS COMMANDS
1899
1900       Commands marked with * may be preceded by a number, N.
1901       Notes in parentheses indicate the behavior if N is given.
1902   h  H                 Display this help.
1903   q  :q  Q  :Q  ZZ     Exit.
1904  ---------------------------------------------------------------------------
1905                            MOVING
1906   e  ^E  j  ^N  CR  *  Forward  one line   (or N lines).
1907   y  ^Y  k  ^K  ^P  *  Backward one line   (or N lines).
1908   f  ^F  ^V  SPACE  *  Forward  one window (or N lines).
1909   b  ^B  ESC-v      *  Backward one window (or N lines).
1910   z                 *  Forward  one window (and set window to N).
1911   w                 *  Backward one window (and set window to N).
1912   ESC-SPACE         *  Forward  one window, but don't stop at end-of-file.
1913   d  ^D             *  Forward  one half-window (and set half-window to N).
1914   u  ^U             *  Backward one half-window (and set half-window to N).
1915   ESC-)  RightArrow *  Left  one half screen width (or N positions).
1916   ESC-(  LeftArrow  *  Right one half screen width (or N positions).
1917   F                    Forward forever; like "tail -f".
1918   r  ^R  ^L            Repaint screen.
1919   R                    Repaint screen, discarding buffered input.
1920         ---------------------------------------------------
1921         Default "window" is the screen height.
1922         Default "half-window" is half of the screen height.
1923  ---------------------------------------------------------------------------
1924                           SEARCHING
1925   /pattern          *  Search forward for (N-th) matching line.
1926   ?pattern          *  Search backward for (N-th) matching line.
1927   n                 *  Repeat previous search (for N-th occurrence).
1928   N                 *  Repeat previous search in reverse direction.
1929   ESC-n             *  Repeat previous search, spanning files.
1930   ESC-N             *  Repeat previous search, reverse dir. & spanning files.
1931   ESC-u                Undo (toggle) search highlighting.
1932         ---------------------------------------------------
1933         Search patterns may be modified by one or more of:
1934         ^N or !  Search for NON-matching lines.
1935         ^E or *  Search multiple files (pass thru END OF FILE).
1936         ^F or @  Start search at FIRST file (for /) or last file (for ?).
1937         ^K       Highlight matches, but don't move (KEEP position).
1938         ^R       Don't use REGULAR EXPRESSIONS.
1939  ---------------------------------------------------------------------------
1940                            JUMPING
1941   g  <  ESC-<       *  Go to first line in file (or line N).
1942   G  >  ESC->       *  Go to last line in file (or line N).
1943   p  %              *  Go to beginning of file (or N percent into file).
1944   t                 *  Go to the (N-th) next tag.
1945   T                 *  Go to the (N-th) previous tag.
1946   {  (  [           *  Find close bracket } ) ].
1947   }  )  ]           *  Find open bracket { ( [.
1948   ESC-^F <c1> <c2>  *  Find close bracket <c2>.
1949   ESC-^B <c1> <c2>  *  Find open bracket <c1>
1950         ---------------------------------------------------
1951         Each "find close bracket" command goes forward to the close bracket
1952           matching the (N-th) open bracket in the top line.
1953         Each "find open bracket" command goes backward to the open bracket
1954           matching the (N-th) close bracket in the bottom line.
1955   m<letter>            Mark the current position with <letter>.
1956   '<letter>            Go to a previously marked position.
1957   ''                   Go to the previous position.
1958   ^X^X                 Same as '.
1959         ---------------------------------------------------
1960         A mark is any upper-case or lower-case letter.
1961         Certain marks are predefined:
1962              ^  means  beginning of the file
1963              $  means  end of the file
1964  ---------------------------------------------------------------------------
1965                         CHANGING FILES
1966   :e [file]            Examine a new file.
1967   ^X^V                 Same as :e.
1968   :n                *  Examine the (N-th) next file from the command line.
1969   :p                *  Examine the (N-th) previous file from the command line.
1970   :x                *  Examine the first (or N-th) file from the command line.
1971   :d                   Delete the current file from the command line list.
1972   =  ^G  :f            Print current file name.
1973  ---------------------------------------------------------------------------
1974                     MISCELLANEOUS COMMANDS
1975   -<flag>              Toggle a command line option [see OPTIONS below].
1976   --<name>             Toggle a command line option, by name.
1977   _<flag>              Display the setting of a command line option.
1978   __<name>             Display the setting of an option, by name.
1979   +cmd                 Execute the less cmd each time a new file is examined.
1980   !command             Execute the shell command with $SHELL.
1981   |Xcommand            Pipe file between current pos & mark X to shell command.
1982   v                    Edit the current file with $VISUAL or $EDITOR.
1983   V                    Print version number of "less".
1984  ---------------------------------------------------------------------------
1985                            OPTIONS
1986         Most options may be changed either on the command line,
1987         or from within less by using the - or -- command.
1988         Options may be given in one of two forms: either a single
1989         character preceded by a -, or a name preceded by --.
1990   -?  ........  --help
1991                   Display help (from command line).
1992   -a  ........  --search-skip-screen
1993                   Forward search skips current screen.
1994   -b [N]  ....  --buffers=[N]
1995                   Number of buffers.
1996   -B  ........  --auto-buffers
1997                   Don't automatically allocate buffers for pipes.
1998   -c  ........  --clear-screen
1999                   Repaint by clearing rather than scrolling.
2000   -d  ........  --dumb
2001                   Dumb terminal.
2002   -D [xn.n]  .  --color=xn.n
2003                   Set screen colors. (MS-DOS only)
2004   -e  -E  ....  --quit-at-eof  --QUIT-AT-EOF
2005                   Quit at end of file.
2006   -f  ........  --force
2007                   Force open non-regular files.
2008   -F  ........  --quit-if-one-screen
2009                   Quit if entire file fits on first screen.
2010   -g  ........  --hilite-search
2011                   Highlight only last match for searches.
2012   -G  ........  --HILITE-SEARCH
2013                   Don't highlight any matches for searches.
2014   -h [N]  ....  --max-back-scroll=[N]
2015                   Backward scroll limit.
2016   -i  ........  --ignore-case
2017                   Ignore case in searches that do not contain uppercase.
2018   -I  ........  --IGNORE-CASE
2019                   Ignore case in all searches.
2020   -j [N]  ....  --jump-target=[N]
2021                   Screen position of target lines.
2022   -J  ........  --status-column
2023                   Display a status column at left edge of screen.
2024   -k [file]  .  --lesskey-file=[file]
2025                   Use a lesskey file.
2026   -L  ........  --no-lessopen
2027                   Ignore the LESSOPEN environment variable.
2028   -m  -M  ....  --long-prompt  --LONG-PROMPT
2029                   Set prompt style.
2030   -n  -N  ....  --line-numbers  --LINE-NUMBERS
2031                   Don't use line numbers.
2032   -o [file]  .  --log-file=[file]
2033                   Copy to log file (standard input only).
2034   -O [file]  .  --LOG-FILE=[file]
2035                   Copy to log file (unconditionally overwrite).
2036   -p [pattern]  --pattern=[pattern]
2037                   Start at pattern (from command line).
2038   -P [prompt]   --prompt=[prompt]
2039                   Define new prompt.
2040   -q  -Q  ....  --quiet  --QUIET  --silent --SILENT
2041                   Quiet the terminal bell.
2042   -r  -R  ....  --raw-control-chars  --RAW-CONTROL-CHARS
2043                   Output "raw" control characters.
2044   -s  ........  --squeeze-blank-lines
2045                   Squeeze multiple blank lines.
2046   -S  ........  --chop-long-lines
2047                   Chop long lines.
2048   -t [tag]  ..  --tag=[tag]
2049                   Find a tag.
2050   -T [tagsfile] --tag-file=[tagsfile]
2051                   Use an alternate tags file.
2052   -u  -U  ....  --underline-special  --UNDERLINE-SPECIAL
2053                   Change handling of backspaces.
2054   -V  ........  --version
2055                   Display the version number of "less".
2056   -w  ........  --hilite-unread
2057                   Highlight first new line after forward-screen.
2058   -W  ........  --HILITE-UNREAD
2059                   Highlight first new line after any forward movement.
2060   -x [N[,...]]  --tabs=[N[,...]]
2061                   Set tab stops.
2062   -X  ........  --no-init
2063                   Don't use termcap init/deinit strings.
2064                 --no-keypad
2065                   Don't use termcap keypad init/deinit strings.
2066   -y [N]  ....  --max-forw-scroll=[N]
2067                   Forward scroll limit.
2068   -z [N]  ....  --window=[N]
2069                   Set size of window.
2070   -" [c[c]]  .  --quotes=[c[c]]
2071                   Set shell quote characters.
2072   -~  ........  --tilde
2073                   Don't display tildes after end of file.
2074   -# [N]  ....  --shift=[N]
2075                   Horizontal scroll amount (0 = one half screen width)
2076
2077  ---------------------------------------------------------------------------
2078                           LINE EDITING
2079         These keys can be used to edit text being entered
2080         on the "command line" at the bottom of the screen.
2081  RightArrow                       ESC-l     Move cursor right one character.
2082  LeftArrow                        ESC-h     Move cursor left one character.
2083  CNTL-RightArrow  ESC-RightArrow  ESC-w     Move cursor right one word.
2084  CNTL-LeftArrow   ESC-LeftArrow   ESC-b     Move cursor left one word.
2085  HOME                             ESC-0     Move cursor to start of line.
2086  END                              ESC-$     Move cursor to end of line.
2087  BACKSPACE                                  Delete char to left of cursor.
2088  DELETE                           ESC-x     Delete char under cursor.
2089  CNTL-BACKSPACE   ESC-BACKSPACE             Delete word to left of cursor.
2090  CNTL-DELETE      ESC-DELETE      ESC-X     Delete word under cursor.
2091  CNTL-U           ESC (MS-DOS only)         Delete entire line.
2092  UpArrow                          ESC-k     Retrieve previous command line.
2093  DownArrow                        ESC-j     Retrieve next command line.
2094  TAB                                        Complete filename & cycle.
2095  SHIFT-TAB                        ESC-TAB   Complete filename & reverse cycle.
2096  CNTL-L                                     Complete filename, list all.
2097 */