libbb/lineedit.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Termios command line History and Editing.
   4  *
   5  * Copyright (c) 1986-2003 may safely be consumed by a BSD or GPL license.
   6  * Written by:   Vladimir Oleynik <dzo@simtreas.ru>
   7  *
   8  * Used ideas:
   9  *      Adam Rogoyski    <rogoyski@cs.utexas.edu>
  10  *      Dave Cinege      <dcinege@psychosis.com>
  11  *      Jakub Jelinek (c) 1995
  12  *      Erik Andersen    <andersen@codepoet.org> (Majorly adjusted for busybox)
  13  *
  14  * This code is 'as is' with no warranty.
  15  */
  16
  17 /*
  18  * Usage and known bugs:
  19  * Terminal key codes are not extensive, and more will probably
  20  * need to be added. This version was created on Debian GNU/Linux 2.x.
  21  * Delete, Backspace, Home, End, and the arrow keys were tested
  22  * to work in an Xterm and console. Ctrl-A also works as Home.
  23  * Ctrl-E also works as End.
  24  *
  25  * lineedit does not know that the terminal escape sequences do not
  26  * take up space on the screen. The redisplay code assumes, unless
  27  * told otherwise, that each character in the prompt is a printable
  28  * character that takes up one character position on the screen.
  29  * You need to tell lineedit that some sequences of characters
  30  * in the prompt take up no screen space. Compatibly with readline,
  31  * use the \[ escape to begin a sequence of non-printing characters,
  32  * and the \] escape to signal the end of such a sequence. Example:
  33  *
  34  * PS1='\[\033[01;32m\]\u@\h\[\033[01;34m\] \w \$\[\033[00m\] '
  35  */
  36
  37 #include "libbb.h"
  38
  39
  40 /* FIXME: obsolete CONFIG item? */
  41 #define ENABLE_FEATURE_NONPRINTABLE_INVERSE_PUT 0
  42
  43
  44 #ifdef TEST
  45
  46 #define ENABLE_FEATURE_EDITING 0
  47 #define ENABLE_FEATURE_TAB_COMPLETION 0
  48 #define ENABLE_FEATURE_USERNAME_COMPLETION 0
  49 #define ENABLE_FEATURE_NONPRINTABLE_INVERSE_PUT 0
  50
  51 #endif  /* TEST */
  52
  53
  54 /* Entire file (except TESTing part) sits inside this #if */
  55 #if ENABLE_FEATURE_EDITING
  56
  57 #if ENABLE_LOCALE_SUPPORT
  58 #define Isprint(c) isprint(c)
  59 #else
  60 #define Isprint(c) ((c) >= ' ' && (c) != ((unsigned char)'\233'))
  61 #endif
  62
  63 #define ENABLE_FEATURE_GETUSERNAME_AND_HOMEDIR \
  64         (ENABLE_FEATURE_USERNAME_COMPLETION || ENABLE_FEATURE_EDITING_FANCY_PROMPT)
  65 #define USE_FEATURE_GETUSERNAME_AND_HOMEDIR(...)
  66 #if ENABLE_FEATURE_GETUSERNAME_AND_HOMEDIR
  67 #undef USE_FEATURE_GETUSERNAME_AND_HOMEDIR
  68 #define USE_FEATURE_GETUSERNAME_AND_HOMEDIR(...) __VA_ARGS__
  69 #endif
  70
  71 enum {
  72         /* We use int16_t for positions, need to limit line len */
  73         MAX_LINELEN = CONFIG_FEATURE_EDITING_MAX_LEN < 0x7ff0
  74                       ? CONFIG_FEATURE_EDITING_MAX_LEN
  75                       : 0x7ff0
  76 };
  77
  78 #if ENABLE_FEATURE_GETUSERNAME_AND_HOMEDIR
  79 static const char null_str[] ALIGN1 = "";
  80 #endif
  81
  82 /* We try to minimize both static and stack usage. */
  83 struct lineedit_statics {
  84         line_input_t *state;
  85
  86         volatile unsigned cmdedit_termw; /* = 80; */ /* actual terminal width */
  87         sighandler_t previous_SIGWINCH_handler;
  88
  89         unsigned cmdedit_x;        /* real x terminal position */
  90         unsigned cmdedit_y;        /* pseudoreal y terminal position */
  91         unsigned cmdedit_prmt_len; /* length of prompt (without colors etc) */
  92
  93         unsigned cursor;
  94         unsigned command_len;
  95         char *command_ps;
  96
  97         const char *cmdedit_prompt;
  98 #if ENABLE_FEATURE_EDITING_FANCY_PROMPT
  99         int num_ok_lines; /* = 1; */
 100 #endif
 101
 102 #if ENABLE_FEATURE_GETUSERNAME_AND_HOMEDIR
 103         char *user_buf;
 104         char *home_pwd_buf; /* = (char*)null_str; */
 105 #endif
 106
 107 #if ENABLE_FEATURE_TAB_COMPLETION
 108         char **matches;
 109         unsigned num_matches;
 110 #endif
 111
 112 #if ENABLE_FEATURE_EDITING_VI
 113 #define DELBUFSIZ 128
 114         char *delptr;
 115         smallint newdelflag;     /* whether delbuf should be reused yet */
 116         char delbuf[DELBUFSIZ];  /* a place to store deleted characters */
 117 #endif
 118
 119         /* Formerly these were big buffers on stack: */
 120 #if ENABLE_FEATURE_TAB_COMPLETION
 121         char exe_n_cwd_tab_completion__dirbuf[MAX_LINELEN];
 122         char input_tab__matchBuf[MAX_LINELEN];
 123         int16_t find_match__int_buf[MAX_LINELEN + 1]; /* need to have 9 bits at least */
 124         int16_t find_match__pos_buf[MAX_LINELEN + 1];
 125 #endif
 126 };
 127
 128 /* See lineedit_ptr_hack.c */
 129 extern struct lineedit_statics *const lineedit_ptr_to_statics;
 130
 131 #define S (*lineedit_ptr_to_statics)
 132 #define state            (S.state           )
 133 #define cmdedit_termw    (S.cmdedit_termw   )
 134 #define previous_SIGWINCH_handler (S.previous_SIGWINCH_handler)
 135 #define cmdedit_x        (S.cmdedit_x       )
 136 #define cmdedit_y        (S.cmdedit_y       )
 137 #define cmdedit_prmt_len (S.cmdedit_prmt_len)
 138 #define cursor           (S.cursor          )
 139 #define command_len      (S.command_len     )
 140 #define command_ps       (S.command_ps      )
 141 #define cmdedit_prompt   (S.cmdedit_prompt  )
 142 #define num_ok_lines     (S.num_ok_lines    )
 143 #define user_buf         (S.user_buf        )
 144 #define home_pwd_buf     (S.home_pwd_buf    )
 145 #define matches          (S.matches         )
 146 #define num_matches      (S.num_matches     )
 147 #define delptr           (S.delptr          )
 148 #define newdelflag       (S.newdelflag      )
 149 #define delbuf           (S.delbuf          )
 150
 151 #define INIT_S() do { \
 152         (*(struct lineedit_statics**)&lineedit_ptr_to_statics) = xzalloc(sizeof(S)); \
 153         barrier(); \
 154         cmdedit_termw = 80; \
 155         USE_FEATURE_EDITING_FANCY_PROMPT(num_ok_lines = 1;) \
 156         USE_FEATURE_GETUSERNAME_AND_HOMEDIR(home_pwd_buf = (char*)null_str;) \
 157 } while (0)
 158 static void deinit_S(void)
 159 {
 160 #if ENABLE_FEATURE_EDITING_FANCY_PROMPT
 161         /* This one is allocated only if FANCY_PROMPT is on
 162          * (otherwise it points to verbatim prompt (NOT malloced) */
 163         free((char*)cmdedit_prompt);
 164 #endif
 165 #if ENABLE_FEATURE_GETUSERNAME_AND_HOMEDIR
 166         free(user_buf);
 167         if (home_pwd_buf != null_str)
 168                 free(home_pwd_buf);
 169 #endif
 170         free(lineedit_ptr_to_statics);
 171 }
 172 #define DEINIT_S() deinit_S()
 173
 174
 175 /* Put 'command_ps[cursor]', cursor++.
 176  * Advance cursor on screen. If we reached right margin, scroll text up
 177  * and remove terminal margin effect by printing 'next_char' */
 178 #define HACK_FOR_WRONG_WIDTH 1
 179 #if HACK_FOR_WRONG_WIDTH
 180 static void cmdedit_set_out_char(void)
 181 #define cmdedit_set_out_char(next_char) cmdedit_set_out_char()
 182 #else
 183 static void cmdedit_set_out_char(int next_char)
 184 #endif
 185 {
 186         int c = (unsigned char)command_ps[cursor];
 187
 188         if (c == '\0') {
 189                 /* erase character after end of input string */
 190                 c = ' ';
 191         }
 192 #if ENABLE_FEATURE_NONPRINTABLE_INVERSE_PUT
 193         /* Display non-printable characters in reverse */
 194         if (!Isprint(c)) {
 195                 if (c >= 128)
 196                         c -= 128;
 197                 if (c < ' ')
 198                         c += '@';
 199                 if (c == 127)
 200                         c = '?';
 201                 printf("\033[7m%c\033[0m", c);
 202         } else
 203 #endif
 204         {
 205                 bb_putchar(c);
 206         }
 207         if (++cmdedit_x >= cmdedit_termw) {
 208                 /* terminal is scrolled down */
 209                 cmdedit_y++;
 210                 cmdedit_x = 0;
 211 #if HACK_FOR_WRONG_WIDTH
 212                 /* This works better if our idea of term width is wrong
 213                  * and it is actually wider (often happens on serial lines).
 214                  * Printing CR,LF *forces* cursor to next line.
 215                  * OTOH if terminal width is correct AND terminal does NOT
 216                  * have automargin (IOW: it is moving cursor to next line
 217                  * by itself (which is wrong for VT-10x terminals)),
 218                  * this will break things: there will be one extra empty line */
 219                 puts("\r"); /* + implicit '\n' */
 220 #else
 221                 /* Works ok only if cmdedit_termw is correct */
 222                 /* destroy "(auto)margin" */
 223                 bb_putchar(next_char);
 224                 bb_putchar('\b');
 225 #endif
 226         }
 227 // Huh? What if command_ps[cursor] == '\0' (we are at the end already?)
 228         cursor++;
 229 }
 230
 231 /* Move to end of line (by printing all chars till the end) */
 232 static void input_end(void)
 233 {
 234         while (cursor < command_len)
 235                 cmdedit_set_out_char(' ');
 236 }
 237
 238 /* Go to the next line */
 239 static void goto_new_line(void)
 240 {
 241         input_end();
 242         if (cmdedit_x)
 243                 bb_putchar('\n');
 244 }
 245
 246
 247 static void out1str(const char *s)
 248 {
 249         if (s)
 250                 fputs(s, stdout);
 251 }
 252
 253 static void beep(void)
 254 {
 255         bb_putchar('\007');
 256 }
 257
 258 /* Move back one character */
 259 /* (optimized for slow terminals) */
 260 static void input_backward(unsigned num)
 261 {
 262         int count_y;
 263
 264         if (num > cursor)
 265                 num = cursor;
 266         if (!num)
 267                 return;
 268         cursor -= num;
 269
 270         if (cmdedit_x >= num) {
 271                 cmdedit_x -= num;
 272                 if (num <= 4) {
 273                         /* This is longer by 5 bytes on x86.
 274                          * Also gets miscompiled for ARM users
 275                          * (busybox.net/bugs/view.php?id=2274).
 276                          * printf(("\b\b\b\b" + 4) - num);
 277                          * return;
 278                          */
 279                         do {
 280                                 bb_putchar('\b');
 281                         } while (--num);
 282                         return;
 283                 }
 284                 printf("\033[%uD", num);
 285                 return;
 286         }
 287
 288         /* Need to go one or more lines up */
 289         num -= cmdedit_x;
 290         {
 291                 unsigned w = cmdedit_termw; /* volatile var */
 292                 count_y = 1 + (num / w);
 293                 cmdedit_y -= count_y;
 294                 cmdedit_x = w * count_y - num;
 295         }
 296         /* go to 1st column; go up; go to correct column */
 297         printf("\r" "\033[%dA" "\033[%dC", count_y, cmdedit_x);
 298 }
 299
 300 static void put_prompt(void)
 301 {
 302         out1str(cmdedit_prompt);
 303         cursor = 0;
 304         {
 305                 unsigned w = cmdedit_termw; /* volatile var */
 306                 cmdedit_y = cmdedit_prmt_len / w; /* new quasireal y */
 307                 cmdedit_x = cmdedit_prmt_len % w;
 308         }
 309 }
 310
 311 /* draw prompt, editor line, and clear tail */
 312 static void redraw(int y, int back_cursor)
 313 {
 314         if (y > 0)                              /* up to start y */
 315                 printf("\033[%dA", y);
 316         bb_putchar('\r');
 317         put_prompt();
 318         input_end();                            /* rewrite */
 319         printf("\033[J");                       /* erase after cursor */
 320         input_backward(back_cursor);
 321 }
 322
 323 /* Delete the char in front of the cursor, optionally saving it
 324  * for later putback */
 325 #if !ENABLE_FEATURE_EDITING_VI
 326 static void input_delete(void)
 327 #define input_delete(save) input_delete()
 328 #else
 329 static void input_delete(int save)
 330 #endif
 331 {
 332         int j = cursor;
 333
 334         if (j == (int)command_len)
 335                 return;
 336
 337 #if ENABLE_FEATURE_EDITING_VI
 338         if (save) {
 339                 if (newdelflag) {
 340                         delptr = delbuf;
 341                         newdelflag = 0;
 342                 }
 343                 if ((delptr - delbuf) < DELBUFSIZ)
 344                         *delptr++ = command_ps[j];
 345         }
 346 #endif
 347
 348         overlapping_strcpy(command_ps + j, command_ps + j + 1);
 349         command_len--;
 350         input_end();                    /* rewrite new line */
 351         cmdedit_set_out_char(' ');      /* erase char */
 352         input_backward(cursor - j);     /* back to old pos cursor */
 353 }
 354
 355 #if ENABLE_FEATURE_EDITING_VI
 356 static void put(void)
 357 {
 358         int ocursor;
 359         int j = delptr - delbuf;
 360
 361         if (j == 0)
 362                 return;
 363         ocursor = cursor;
 364         /* open hole and then fill it */
 365         memmove(command_ps + cursor + j, command_ps + cursor, command_len - cursor + 1);
 366         strncpy(command_ps + cursor, delbuf, j);
 367         command_len += j;
 368         input_end();                    /* rewrite new line */
 369         input_backward(cursor - ocursor - j + 1); /* at end of new text */
 370 }
 371 #endif
 372
 373 /* Delete the char in back of the cursor */
 374 static void input_backspace(void)
 375 {
 376         if (cursor > 0) {
 377                 input_backward(1);
 378                 input_delete(0);
 379         }
 380 }
 381
 382 /* Move forward one character */
 383 static void input_forward(void)
 384 {
 385         if (cursor < command_len)
 386                 cmdedit_set_out_char(command_ps[cursor + 1]);
 387 }
 388
 389 #if ENABLE_FEATURE_TAB_COMPLETION
 390
 391 static void free_tab_completion_data(void)
 392 {
 393         if (matches) {
 394                 while (num_matches)
 395                         free(matches[--num_matches]);
 396                 free(matches);
 397                 matches = NULL;
 398         }
 399 }
 400
 401 static void add_match(char *matched)
 402 {
 403         matches = xrealloc_vector(matches, 4, num_matches);
 404         matches[num_matches] = matched;
 405         num_matches++;
 406 }
 407
 408 #if ENABLE_FEATURE_USERNAME_COMPLETION
 409 static void username_tab_completion(char *ud, char *with_shash_flg)
 410 {
 411         struct passwd *entry;
 412         int userlen;
 413
 414         ud++;                           /* ~user/... to user/... */
 415         userlen = strlen(ud);
 416
 417         if (with_shash_flg) {           /* "~/..." or "~user/..." */
 418                 char *sav_ud = ud - 1;
 419                 char *home = NULL;
 420
 421                 if (*ud == '/') {       /* "~/..."     */
 422                         home = home_pwd_buf;
 423                 } else {
 424                         /* "~user/..." */
 425                         char *temp;
 426                         temp = strchr(ud, '/');
 427                         *temp = '\0';           /* ~user\0 */
 428                         entry = getpwnam(ud);
 429                         *temp = '/';            /* restore ~user/... */
 430                         ud = temp;
 431                         if (entry)
 432                                 home = entry->pw_dir;
 433                 }
 434                 if (home) {
 435                         if ((userlen + strlen(home) + 1) < MAX_LINELEN) {
 436                                 /* /home/user/... */
 437                                 sprintf(sav_ud, "%s%s", home, ud);
 438                         }
 439                 }
 440         } else {
 441                 /* "~[^/]*" */
 442                 /* Using _r function to avoid pulling in static buffers */
 443                 char line_buff[256];
 444                 struct passwd pwd;
 445                 struct passwd *result;
 446
 447                 setpwent();
 448                 while (!getpwent_r(&pwd, line_buff, sizeof(line_buff), &result)) {
 449                         /* Null usernames should result in all users as possible completions. */
 450                         if (/*!userlen || */ strncmp(ud, pwd.pw_name, userlen) == 0) {
 451                                 add_match(xasprintf("~%s/", pwd.pw_name));
 452                         }
 453                 }
 454                 endpwent();
 455         }
 456 }
 457 #endif  /* FEATURE_COMMAND_USERNAME_COMPLETION */
 458
 459 enum {
 460         FIND_EXE_ONLY = 0,
 461         FIND_DIR_ONLY = 1,
 462         FIND_FILE_ONLY = 2,
 463 };
 464
 465 static int path_parse(char ***p, int flags)
 466 {
 467         int npth;
 468         const char *pth;
 469         char *tmp;
 470         char **res;
 471
 472         /* if not setenv PATH variable, to search cur dir "." */
 473         if (flags != FIND_EXE_ONLY)
 474                 return 1;
 475
 476         if (state->flags & WITH_PATH_LOOKUP)
 477                 pth = state->path_lookup;
 478         else
 479                 pth = getenv("PATH");
 480         /* PATH=<empty> or PATH=:<empty> */
 481         if (!pth || !pth[0] || LONE_CHAR(pth, ':'))
 482                 return 1;
 483
 484         tmp = (char*)pth;
 485         npth = 1; /* path component count */
 486         while (1) {
 487                 tmp = strchr(tmp, ':');
 488                 if (!tmp)
 489                         break;
 490                 if (*++tmp == '\0')
 491                         break;  /* :<empty> */
 492                 npth++;
 493         }
 494
 495         res = xmalloc(npth * sizeof(char*));
 496         res[0] = tmp = xstrdup(pth);
 497         npth = 1;
 498         while (1) {
 499                 tmp = strchr(tmp, ':');
 500                 if (!tmp)
 501                         break;
 502                 *tmp++ = '\0'; /* ':' -> '\0' */
 503                 if (*tmp == '\0')
 504                         break; /* :<empty> */
 505                 res[npth++] = tmp;
 506         }
 507         *p = res;
 508         return npth;
 509 }
 510
 511 static void exe_n_cwd_tab_completion(char *command, int type)
 512 {
 513         DIR *dir;
 514         struct dirent *next;
 515         struct stat st;
 516         char *path1[1];
 517         char **paths = path1;
 518         int npaths;
 519         int i;
 520         char *found;
 521         char *pfind = strrchr(command, '/');
 522 /*      char dirbuf[MAX_LINELEN]; */
 523 #define dirbuf (S.exe_n_cwd_tab_completion__dirbuf)
 524
 525         npaths = 1;
 526         path1[0] = (char*)".";
 527
 528         if (pfind == NULL) {
 529                 /* no dir, if flags==EXE_ONLY - get paths, else "." */
 530                 npaths = path_parse(&paths, type);
 531                 pfind = command;
 532         } else {
 533                 /* dirbuf = ".../.../.../" */
 534                 safe_strncpy(dirbuf, command, (pfind - command) + 2);
 535 #if ENABLE_FEATURE_USERNAME_COMPLETION
 536                 if (dirbuf[0] == '~')   /* ~/... or ~user/... */
 537                         username_tab_completion(dirbuf, dirbuf);
 538 #endif
 539                 paths[0] = dirbuf;
 540                 /* point to 'l' in "..../last_component" */
 541                 pfind++;
 542         }
 543
 544         for (i = 0; i < npaths; i++) {
 545                 dir = opendir(paths[i]);
 546                 if (!dir)
 547                         continue; /* don't print an error */
 548
 549                 while ((next = readdir(dir)) != NULL) {
 550                         int len1;
 551                         const char *str_found = next->d_name;
 552
 553                         /* matched? */
 554                         if (strncmp(str_found, pfind, strlen(pfind)))
 555                                 continue;
 556                         /* not see .name without .match */
 557                         if (*str_found == '.' && *pfind == '\0') {
 558                                 if (NOT_LONE_CHAR(paths[i], '/') || str_found[1])
 559                                         continue;
 560                                 str_found = ""; /* only "/" */
 561                         }
 562                         found = concat_path_file(paths[i], str_found);
 563                         /* hmm, remove in progress? */
 564                         /* NB: stat() first so that we see is it a directory;
 565                          * but if that fails, use lstat() so that
 566                          * we still match dangling links */
 567                         if (stat(found, &st) && lstat(found, &st))
 568                                 goto cont;
 569                         /* find with dirs? */
 570                         if (paths[i] != dirbuf)
 571                                 strcpy(found, next->d_name); /* only name */
 572
 573                         len1 = strlen(found);
 574                         found = xrealloc(found, len1 + 2);
 575                         found[len1] = '\0';
 576                         found[len1+1] = '\0';
 577
 578                         if (S_ISDIR(st.st_mode)) {
 579                                 /* name is a directory */
 580                                 if (found[len1-1] != '/') {
 581                                         found[len1] = '/';
 582                                 }
 583                         } else {
 584                                 /* not put found file if search only dirs for cd */
 585                                 if (type == FIND_DIR_ONLY)
 586                                         goto cont;
 587                         }
 588                         /* Add it to the list */
 589                         add_match(found);
 590                         continue;
 591  cont:
 592                         free(found);
 593                 }
 594                 closedir(dir);
 595         }
 596         if (paths != path1) {
 597                 free(paths[0]); /* allocated memory is only in first member */
 598                 free(paths);
 599         }
 600 #undef dirbuf
 601 }
 602
 603 #define QUOT (UCHAR_MAX+1)
 604
 605 #define collapse_pos(is, in) do { \
 606         memmove(int_buf+(is), int_buf+(in), (MAX_LINELEN+1-(is)-(in)) * sizeof(pos_buf[0])); \
 607         memmove(pos_buf+(is), pos_buf+(in), (MAX_LINELEN+1-(is)-(in)) * sizeof(pos_buf[0])); \
 608 } while (0)
 609
 610 static int find_match(char *matchBuf, int *len_with_quotes)
 611 {
 612         int i, j;
 613         int command_mode;
 614         int c, c2;
 615 /*      int16_t int_buf[MAX_LINELEN + 1]; */
 616 /*      int16_t pos_buf[MAX_LINELEN + 1]; */
 617 #define int_buf (S.find_match__int_buf)
 618 #define pos_buf (S.find_match__pos_buf)
 619
 620         /* set to integer dimension characters and own positions */
 621         for (i = 0;; i++) {
 622                 int_buf[i] = (unsigned char)matchBuf[i];
 623                 if (int_buf[i] == 0) {
 624                         pos_buf[i] = -1;        /* indicator end line */
 625                         break;
 626                 }
 627                 pos_buf[i] = i;
 628         }
 629
 630         /* mask \+symbol and convert '\t' to ' ' */
 631         for (i = j = 0; matchBuf[i]; i++, j++)
 632                 if (matchBuf[i] == '\\') {
 633                         collapse_pos(j, j + 1);
 634                         int_buf[j] |= QUOT;
 635                         i++;
 636 #if ENABLE_FEATURE_NONPRINTABLE_INVERSE_PUT
 637                         if (matchBuf[i] == '\t')        /* algorithm equivalent */
 638                                 int_buf[j] = ' ' | QUOT;
 639 #endif
 640                 }
 641 #if ENABLE_FEATURE_NONPRINTABLE_INVERSE_PUT
 642                 else if (matchBuf[i] == '\t')
 643                         int_buf[j] = ' ';
 644 #endif
 645
 646         /* mask "symbols" or 'symbols' */
 647         c2 = 0;
 648         for (i = 0; int_buf[i]; i++) {
 649                 c = int_buf[i];
 650                 if (c == '\'' || c == '"') {
 651                         if (c2 == 0)
 652                                 c2 = c;
 653                         else {
 654                                 if (c == c2)
 655                                         c2 = 0;
 656                                 else
 657                                         int_buf[i] |= QUOT;
 658                         }
 659                 } else if (c2 != 0 && c != '$')
 660                         int_buf[i] |= QUOT;
 661         }
 662
 663         /* skip commands with arguments if line has commands delimiters */
 664         /* ';' ';;' '&' '|' '&&' '||' but `>&' `<&' `>|' */
 665         for (i = 0; int_buf[i]; i++) {
 666                 c = int_buf[i];
 667                 c2 = int_buf[i + 1];
 668                 j = i ? int_buf[i - 1] : -1;
 669                 command_mode = 0;
 670                 if (c == ';' || c == '&' || c == '|') {
 671                         command_mode = 1 + (c == c2);
 672                         if (c == '&') {
 673                                 if (j == '>' || j == '<')
 674                                         command_mode = 0;
 675                         } else if (c == '|' && j == '>')
 676                                 command_mode = 0;
 677                 }
 678                 if (command_mode) {
 679                         collapse_pos(0, i + command_mode);
 680                         i = -1;                         /* hack incremet */
 681                 }
 682         }
 683         /* collapse `command...` */
 684         for (i = 0; int_buf[i]; i++)
 685                 if (int_buf[i] == '`') {
 686                         for (j = i + 1; int_buf[j]; j++)
 687                                 if (int_buf[j] == '`') {
 688                                         collapse_pos(i, j + 1);
 689                                         j = 0;
 690                                         break;
 691                                 }
 692                         if (j) {
 693                                 /* not found close ` - command mode, collapse all previous */
 694                                 collapse_pos(0, i + 1);
 695                                 break;
 696                         } else
 697                                 i--;                    /* hack incremet */
 698                 }
 699
 700         /* collapse (command...(command...)...) or {command...{command...}...} */
 701         c = 0;                                          /* "recursive" level */
 702         c2 = 0;
 703         for (i = 0; int_buf[i]; i++)
 704                 if (int_buf[i] == '(' || int_buf[i] == '{') {
 705                         if (int_buf[i] == '(')
 706                                 c++;
 707                         else
 708                                 c2++;
 709                         collapse_pos(0, i + 1);
 710                         i = -1;                         /* hack incremet */
 711                 }
 712         for (i = 0; pos_buf[i] >= 0 && (c > 0 || c2 > 0); i++)
 713                 if ((int_buf[i] == ')' && c > 0) || (int_buf[i] == '}' && c2 > 0)) {
 714                         if (int_buf[i] == ')')
 715                                 c--;
 716                         else
 717                                 c2--;
 718                         collapse_pos(0, i + 1);
 719                         i = -1;                         /* hack incremet */
 720                 }
 721
 722         /* skip first not quote space */
 723         for (i = 0; int_buf[i]; i++)
 724                 if (int_buf[i] != ' ')
 725                         break;
 726         if (i)
 727                 collapse_pos(0, i);
 728
 729         /* set find mode for completion */
 730         command_mode = FIND_EXE_ONLY;
 731         for (i = 0; int_buf[i]; i++)
 732                 if (int_buf[i] == ' ' || int_buf[i] == '<' || int_buf[i] == '>') {
 733                         if (int_buf[i] == ' ' && command_mode == FIND_EXE_ONLY
 734                          && matchBuf[pos_buf[0]] == 'c'
 735                          && matchBuf[pos_buf[1]] == 'd'
 736                         ) {
 737                                 command_mode = FIND_DIR_ONLY;
 738                         } else {
 739                                 command_mode = FIND_FILE_ONLY;
 740                                 break;
 741                         }
 742                 }
 743         for (i = 0; int_buf[i]; i++)
 744                 /* "strlen" */;
 745         /* find last word */
 746         for (--i; i >= 0; i--) {
 747                 c = int_buf[i];
 748                 if (c == ' ' || c == '<' || c == '>' || c == '|' || c == '&') {
 749                         collapse_pos(0, i + 1);
 750                         break;
 751                 }
 752         }
 753         /* skip first not quoted '\'' or '"' */
 754         for (i = 0; int_buf[i] == '\'' || int_buf[i] == '"'; i++)
 755                 /*skip*/;
 756         /* collapse quote or unquote // or /~ */
 757         while ((int_buf[i] & ~QUOT) == '/'
 758          && ((int_buf[i+1] & ~QUOT) == '/' || (int_buf[i+1] & ~QUOT) == '~')
 759         ) {
 760                 i++;
 761         }
 762
 763         /* set only match and destroy quotes */
 764         j = 0;
 765         for (c = 0; pos_buf[i] >= 0; i++) {
 766                 matchBuf[c++] = matchBuf[pos_buf[i]];
 767                 j = pos_buf[i] + 1;
 768         }
 769         matchBuf[c] = '\0';
 770         /* old length matchBuf with quotes symbols */
 771         *len_with_quotes = j ? j - pos_buf[0] : 0;
 772
 773         return command_mode;
 774 #undef int_buf
 775 #undef pos_buf
 776 }
 777
 778 /*
 779  * display by column (original idea from ls applet,
 780  * very optimized by me :)
 781  */
 782 static void showfiles(void)
 783 {
 784         int ncols, row;
 785         int column_width = 0;
 786         int nfiles = num_matches;
 787         int nrows = nfiles;
 788         int l;
 789
 790         /* find the longest file name-  use that as the column width */
 791         for (row = 0; row < nrows; row++) {
 792                 l = strlen(matches[row]);
 793                 if (column_width < l)
 794                         column_width = l;
 795         }
 796         column_width += 2;              /* min space for columns */
 797         ncols = cmdedit_termw / column_width;
 798
 799         if (ncols > 1) {
 800                 nrows /= ncols;
 801                 if (nfiles % ncols)
 802                         nrows++;        /* round up fractionals */
 803         } else {
 804                 ncols = 1;
 805         }
 806         for (row = 0; row < nrows; row++) {
 807                 int n = row;
 808                 int nc;
 809
 810                 for (nc = 1; nc < ncols && n+nrows < nfiles; n += nrows, nc++) {
 811                         printf("%s%-*s", matches[n],
 812                                 (int)(column_width - strlen(matches[n])), "");
 813                 }
 814                 puts(matches[n]);
 815         }
 816 }
 817
 818 static char *add_quote_for_spec_chars(char *found)
 819 {
 820         int l = 0;
 821         char *s = xmalloc((strlen(found) + 1) * 2);
 822
 823         while (*found) {
 824                 if (strchr(" `\"#$%^&*()=+{}[]:;\'|\\<>", *found))
 825                         s[l++] = '\\';
 826                 s[l++] = *found++;
 827         }
 828         s[l] = 0;
 829         return s;
 830 }
 831
 832 /* Do TAB completion */
 833 static void input_tab(smallint *lastWasTab)
 834 {
 835         if (!(state->flags & TAB_COMPLETION))
 836                 return;
 837
 838         if (!*lastWasTab) {
 839                 char *tmp, *tmp1;
 840                 size_t len_found;
 841 /*              char matchBuf[MAX_LINELEN]; */
 842 #define matchBuf (S.input_tab__matchBuf)
 843                 int find_type;
 844                 int recalc_pos;
 845
 846                 *lastWasTab = TRUE;             /* flop trigger */
 847
 848                 /* Make a local copy of the string -- up
 849                  * to the position of the cursor */
 850                 tmp = strncpy(matchBuf, command_ps, cursor);
 851                 tmp[cursor] = '\0';
 852
 853                 find_type = find_match(matchBuf, &recalc_pos);
 854
 855                 /* Free up any memory already allocated */
 856                 free_tab_completion_data();
 857
 858 #if ENABLE_FEATURE_USERNAME_COMPLETION
 859                 /* If the word starts with `~' and there is no slash in the word,
 860                  * then try completing this word as a username. */
 861                 if (state->flags & USERNAME_COMPLETION)
 862                         if (matchBuf[0] == '~' && strchr(matchBuf, '/') == 0)
 863                                 username_tab_completion(matchBuf, NULL);
 864 #endif
 865                 /* Try to match any executable in our path and everything
 866                  * in the current working directory */
 867                 if (!matches)
 868                         exe_n_cwd_tab_completion(matchBuf, find_type);
 869                 /* Sort, then remove any duplicates found */
 870                 if (matches) {
 871                         unsigned i;
 872                         int n = 0;
 873                         qsort_string_vector(matches, num_matches);
 874                         for (i = 0; i < num_matches - 1; ++i) {
 875                                 if (matches[i] && matches[i+1]) { /* paranoia */
 876                                         if (strcmp(matches[i], matches[i+1]) == 0) {
 877                                                 free(matches[i]);
 878                                                 matches[i] = NULL; /* paranoia */
 879                                         } else {
 880                                                 matches[n++] = matches[i];
 881                                         }
 882                                 }
 883                         }
 884                         matches[n] = matches[i];
 885                         num_matches = n + 1;
 886                 }
 887                 /* Did we find exactly one match? */
 888                 if (!matches || num_matches > 1) {
 889                         beep();
 890                         if (!matches)
 891                                 return;         /* not found */
 892                         /* find minimal match */
 893                         tmp1 = xstrdup(matches[0]);
 894                         for (tmp = tmp1; *tmp; tmp++)
 895                                 for (len_found = 1; len_found < num_matches; len_found++)
 896                                         if (matches[len_found][(tmp - tmp1)] != *tmp) {
 897                                                 *tmp = '\0';
 898                                                 break;
 899                                         }
 900                         if (*tmp1 == '\0') {        /* have unique */
 901                                 free(tmp1);
 902                                 return;
 903                         }
 904                         tmp = add_quote_for_spec_chars(tmp1);
 905                         free(tmp1);
 906                 } else {                        /* one match */
 907                         tmp = add_quote_for_spec_chars(matches[0]);
 908                         /* for next completion current found */
 909                         *lastWasTab = FALSE;
 910
 911                         len_found = strlen(tmp);
 912                         if (tmp[len_found-1] != '/') {
 913                                 tmp[len_found] = ' ';
 914                                 tmp[len_found+1] = '\0';
 915                         }
 916                 }
 917                 len_found = strlen(tmp);
 918                 /* have space to placed match? */
 919                 if ((len_found - strlen(matchBuf) + command_len) < MAX_LINELEN) {
 920                         /* before word for match   */
 921                         command_ps[cursor - recalc_pos] = '\0';
 922                         /* save   tail line        */
 923                         strcpy(matchBuf, command_ps + cursor);
 924                         /* add    match            */
 925                         strcat(command_ps, tmp);
 926                         /* add    tail             */
 927                         strcat(command_ps, matchBuf);
 928                         /* back to begin word for match    */
 929                         input_backward(recalc_pos);
 930                         /* new pos                         */
 931                         recalc_pos = cursor + len_found;
 932                         /* new len                         */
 933                         command_len = strlen(command_ps);
 934                         /* write out the matched command   */
 935                         redraw(cmdedit_y, command_len - recalc_pos);
 936                 }
 937                 free(tmp);
 938 #undef matchBuf
 939         } else {
 940                 /* Ok -- the last char was a TAB.  Since they
 941                  * just hit TAB again, print a list of all the
 942                  * available choices... */
 943                 if (matches && num_matches > 0) {
 944                         int sav_cursor = cursor;        /* change goto_new_line() */
 945
 946                         /* Go to the next line */
 947                         goto_new_line();
 948                         showfiles();
 949                         redraw(0, command_len - sav_cursor);
 950                 }
 951         }
 952 }
 953
 954 #endif  /* FEATURE_COMMAND_TAB_COMPLETION */
 955
 956
 957 line_input_t* FAST_FUNC new_line_input_t(int flags)
 958 {
 959         line_input_t *n = xzalloc(sizeof(*n));
 960         n->flags = flags;
 961         return n;
 962 }
 963
 964
 965 #if MAX_HISTORY > 0
 966
 967 static void save_command_ps_at_cur_history(void)
 968 {
 969         if (command_ps[0] != '\0') {
 970                 int cur = state->cur_history;
 971                 free(state->history[cur]);
 972                 state->history[cur] = xstrdup(command_ps);
 973         }
 974 }
 975
 976 /* state->flags is already checked to be nonzero */
 977 static int get_previous_history(void)
 978 {
 979         if ((state->flags & DO_HISTORY) && state->cur_history) {
 980                 save_command_ps_at_cur_history();
 981                 state->cur_history--;
 982                 return 1;
 983         }
 984         beep();
 985         return 0;
 986 }
 987
 988 static int get_next_history(void)
 989 {
 990         if (state->flags & DO_HISTORY) {
 991                 if (state->cur_history < state->cnt_history) {
 992                         save_command_ps_at_cur_history(); /* save the current history line */
 993                         return ++state->cur_history;
 994                 }
 995         }
 996         beep();
 997         return 0;
 998 }
 999
1000 #if ENABLE_FEATURE_EDITING_SAVEHISTORY
1001 /* We try to ensure that concurrent additions to the history
1002  * do not overwrite each other.
1003  * Otherwise shell users get unhappy.
1004  *
1005  * History file is trimmed lazily, when it grows several times longer
1006  * than configured MAX_HISTORY lines.
1007  */
1008
1009 static void free_line_input_t(line_input_t *n)
1010 {
1011         int i = n->cnt_history;
1012         while (i > 0)
1013                 free(n->history[--i]);
1014         free(n);
1015 }
1016
1017 /* state->flags is already checked to be nonzero */
1018 static void load_history(line_input_t *st_parm)
1019 {
1020         char *temp_h[MAX_HISTORY];
1021         char *line;
1022         FILE *fp;
1023         unsigned idx, i, line_len;
1024
1025         /* NB: do not trash old history if file can't be opened */
1026
1027         fp = fopen_for_read(st_parm->hist_file);
1028         if (fp) {
1029                 /* clean up old history */
1030                 for (idx = st_parm->cnt_history; idx > 0;) {
1031                         idx--;
1032                         free(st_parm->history[idx]);
1033                         st_parm->history[idx] = NULL;
1034                 }
1035
1036                 /* fill temp_h[], retaining only last MAX_HISTORY lines */
1037                 memset(temp_h, 0, sizeof(temp_h));
1038                 st_parm->cnt_history_in_file = idx = 0;
1039                 while ((line = xmalloc_fgetline(fp)) != NULL) {
1040                         if (line[0] == '\0') {
1041                                 free(line);
1042                                 continue;
1043                         }
1044                         free(temp_h[idx]);
1045                         temp_h[idx] = line;
1046                         st_parm->cnt_history_in_file++;
1047                         idx++;
1048                         if (idx == MAX_HISTORY)
1049                                 idx = 0;
1050                 }
1051                 fclose(fp);
1052
1053                 /* find first non-NULL temp_h[], if any */
1054                 if (st_parm->cnt_history_in_file) {
1055                         while (temp_h[idx] == NULL) {
1056                                 idx++;
1057                                 if (idx == MAX_HISTORY)
1058                                         idx = 0;
1059                         }
1060                 }
1061
1062                 /* copy temp_h[] to st_parm->history[] */
1063                 for (i = 0; i < MAX_HISTORY;) {
1064                         line = temp_h[idx];
1065                         if (!line)
1066                                 break;
1067                         idx++;
1068                         if (idx == MAX_HISTORY)
1069                                 idx = 0;
1070                         line_len = strlen(line);
1071                         if (line_len >= MAX_LINELEN)
1072                                 line[MAX_LINELEN-1] = '\0';
1073                         st_parm->history[i++] = line;
1074                 }
1075                 st_parm->cnt_history = i;
1076         }
1077 }
1078
1079 /* state->flags is already checked to be nonzero */
1080 static void save_history(char *str)
1081 {
1082         int fd;
1083         int len, len2;
1084
1085         fd = open(state->hist_file, O_WRONLY | O_CREAT | O_APPEND, 0666);
1086         if (fd < 0)
1087                 return;
1088         xlseek(fd, 0, SEEK_END); /* paranoia */
1089         len = strlen(str);
1090         str[len] = '\n'; /* we (try to) do atomic write */
1091         len2 = full_write(fd, str, len + 1);
1092         str[len] = '\0';
1093         close(fd);
1094         if (len2 != len + 1)
1095                 return; /* "wtf?" */
1096
1097         /* did we write so much that history file needs trimming? */
1098         state->cnt_history_in_file++;
1099         if (state->cnt_history_in_file > MAX_HISTORY * 4) {
1100                 FILE *fp;
1101                 char *new_name;
1102                 line_input_t *st_temp;
1103                 int i;
1104
1105                 /* we may have concurrently written entries from others.
1106                  * load them */
1107                 st_temp = new_line_input_t(state->flags);
1108                 st_temp->hist_file = state->hist_file;
1109                 load_history(st_temp);
1110
1111                 /* write out temp file and replace hist_file atomically */
1112                 new_name = xasprintf("%s.%u.new", state->hist_file, (int) getpid());
1113                 fp = fopen_for_write(new_name);
1114                 if (fp) {
1115                         for (i = 0; i < st_temp->cnt_history; i++)
1116                                 fprintf(fp, "%s\n", st_temp->history[i]);
1117                         fclose(fp);
1118                         if (rename(new_name, state->hist_file) == 0)
1119                                 state->cnt_history_in_file = st_temp->cnt_history;
1120                 }
1121                 free(new_name);
1122                 free_line_input_t(st_temp);
1123         }
1124 }
1125 #else
1126 #define load_history(a) ((void)0)
1127 #define save_history(a) ((void)0)
1128 #endif /* FEATURE_COMMAND_SAVEHISTORY */
1129
1130 static void remember_in_history(char *str)
1131 {
1132         int i;
1133
1134         if (!(state->flags & DO_HISTORY))
1135                 return;
1136         if (str[0] == '\0')
1137                 return;
1138         i = state->cnt_history;
1139         /* Don't save dupes */
1140         if (i && strcmp(state->history[i-1], str) == 0)
1141                 return;
1142
1143         free(state->history[MAX_HISTORY]); /* redundant, paranoia */
1144         state->history[MAX_HISTORY] = NULL; /* redundant, paranoia */
1145
1146         /* If history[] is full, remove the oldest command */
1147         /* we need to keep history[MAX_HISTORY] empty, hence >=, not > */
1148         if (i >= MAX_HISTORY) {
1149                 free(state->history[0]);
1150                 for (i = 0; i < MAX_HISTORY-1; i++)
1151                         state->history[i] = state->history[i+1];
1152                 /* i == MAX_HISTORY-1 */
1153         }
1154         /* i <= MAX_HISTORY-1 */
1155         state->history[i++] = xstrdup(str);
1156         /* i <= MAX_HISTORY */
1157         state->cur_history = i;
1158         state->cnt_history = i;
1159 #if ENABLE_FEATURE_EDITING_SAVEHISTORY
1160         if ((state->flags & SAVE_HISTORY) && state->hist_file)
1161                 save_history(str);
1162 #endif
1163         USE_FEATURE_EDITING_FANCY_PROMPT(num_ok_lines++;)
1164 }
1165
1166 #else /* MAX_HISTORY == 0 */
1167 #define remember_in_history(a) ((void)0)
1168 #endif /* MAX_HISTORY */
1169
1170
1171 /*
1172  * This function is used to grab a character buffer
1173  * from the input file descriptor and allows you to
1174  * a string with full command editing (sort of like
1175  * a mini readline).
1176  *
1177  * The following standard commands are not implemented:
1178  * ESC-b -- Move back one word
1179  * ESC-f -- Move forward one word
1180  * ESC-d -- Delete back one word
1181  * ESC-h -- Delete forward one word
1182  * CTL-t -- Transpose two characters
1183  *
1184  * Minimalist vi-style command line editing available if configured.
1185  * vi mode implemented 2005 by Paul Fox <pgf@foxharp.boston.ma.us>
1186  */
1187
1188 #if ENABLE_FEATURE_EDITING_VI
1189 static void
1190 vi_Word_motion(char *command, int eat)
1191 {
1192         while (cursor < command_len && !isspace(command[cursor]))
1193                 input_forward();
1194         if (eat) while (cursor < command_len && isspace(command[cursor]))
1195                 input_forward();
1196 }
1197
1198 static void
1199 vi_word_motion(char *command, int eat)
1200 {
1201         if (isalnum(command[cursor]) || command[cursor] == '_') {
1202                 while (cursor < command_len
1203                  && (isalnum(command[cursor+1]) || command[cursor+1] == '_'))
1204                         input_forward();
1205         } else if (ispunct(command[cursor])) {
1206                 while (cursor < command_len && ispunct(command[cursor+1]))
1207                         input_forward();
1208         }
1209
1210         if (cursor < command_len)
1211                 input_forward();
1212
1213         if (eat && cursor < command_len && isspace(command[cursor]))
1214                 while (cursor < command_len && isspace(command[cursor]))
1215                         input_forward();
1216 }
1217
1218 static void
1219 vi_End_motion(char *command)
1220 {
1221         input_forward();
1222         while (cursor < command_len && isspace(command[cursor]))
1223                 input_forward();
1224         while (cursor < command_len-1 && !isspace(command[cursor+1]))
1225                 input_forward();
1226 }
1227
1228 static void
1229 vi_end_motion(char *command)
1230 {
1231         if (cursor >= command_len-1)
1232                 return;
1233         input_forward();
1234         while (cursor < command_len-1 && isspace(command[cursor]))
1235                 input_forward();
1236         if (cursor >= command_len-1)
1237                 return;
1238         if (isalnum(command[cursor]) || command[cursor] == '_') {
1239                 while (cursor < command_len-1
1240                  && (isalnum(command[cursor+1]) || command[cursor+1] == '_')
1241                 ) {
1242                         input_forward();
1243                 }
1244         } else if (ispunct(command[cursor])) {
1245                 while (cursor < command_len-1 && ispunct(command[cursor+1]))
1246                         input_forward();
1247         }
1248 }
1249
1250 static void
1251 vi_Back_motion(char *command)
1252 {
1253         while (cursor > 0 && isspace(command[cursor-1]))
1254                 input_backward(1);
1255         while (cursor > 0 && !isspace(command[cursor-1]))
1256                 input_backward(1);
1257 }
1258
1259 static void
1260 vi_back_motion(char *command)
1261 {
1262         if (cursor <= 0)
1263                 return;
1264         input_backward(1);
1265         while (cursor > 0 && isspace(command[cursor]))
1266                 input_backward(1);
1267         if (cursor <= 0)
1268                 return;
1269         if (isalnum(command[cursor]) || command[cursor] == '_') {
1270                 while (cursor > 0
1271                  && (isalnum(command[cursor-1]) || command[cursor-1] == '_')
1272                 ) {
1273                         input_backward(1);
1274                 }
1275         } else if (ispunct(command[cursor])) {
1276                 while (cursor > 0 && ispunct(command[cursor-1]))
1277                         input_backward(1);
1278         }
1279 }
1280 #endif
1281
1282
1283 /*
1284  * read_line_input and its helpers
1285  */
1286
1287 #if !ENABLE_FEATURE_EDITING_FANCY_PROMPT
1288 static void parse_and_put_prompt(const char *prmt_ptr)
1289 {
1290         cmdedit_prompt = prmt_ptr;
1291         cmdedit_prmt_len = strlen(prmt_ptr);
1292         put_prompt();
1293 }
1294 #else
1295 static void parse_and_put_prompt(const char *prmt_ptr)
1296 {
1297         int prmt_len = 0;
1298         size_t cur_prmt_len = 0;
1299         char flg_not_length = '[';
1300         char *prmt_mem_ptr = xzalloc(1);
1301         char *cwd_buf = xrealloc_getcwd_or_warn(NULL);
1302         char cbuf[2];
1303         char c;
1304         char *pbuf;
1305
1306         cmdedit_prmt_len = 0;
1307
1308         if (!cwd_buf) {
1309                 cwd_buf = (char *)bb_msg_unknown;
1310         }
1311
1312         cbuf[1] = '\0'; /* never changes */
1313
1314         while (*prmt_ptr) {
1315                 char *free_me = NULL;
1316
1317                 pbuf = cbuf;
1318                 c = *prmt_ptr++;
1319                 if (c == '\\') {
1320                         const char *cp = prmt_ptr;
1321                         int l;
1322
1323                         c = bb_process_escape_sequence(&prmt_ptr);
1324                         if (prmt_ptr == cp) {
1325                                 if (*cp == '\0')
1326                                         break;
1327                                 c = *prmt_ptr++;
1328
1329                                 switch (c) {
1330 #if ENABLE_FEATURE_GETUSERNAME_AND_HOMEDIR
1331                                 case 'u':
1332                                         pbuf = user_buf ? user_buf : (char*)"";
1333                                         break;
1334 #endif
1335                                 case 'h':
1336                                         pbuf = free_me = safe_gethostname();
1337                                         *strchrnul(pbuf, '.') = '\0';
1338                                         break;
1339                                 case '$':
1340                                         c = (geteuid() == 0 ? '#' : '$');
1341                                         break;
1342 #if ENABLE_FEATURE_GETUSERNAME_AND_HOMEDIR
1343                                 case 'w':
1344                                         /* /home/user[/something] -> ~[/something] */
1345                                         pbuf = cwd_buf;
1346                                         l = strlen(home_pwd_buf);
1347                                         if (l != 0
1348                                          && strncmp(home_pwd_buf, cwd_buf, l) == 0
1349                                          && (cwd_buf[l]=='/' || cwd_buf[l]=='\0')
1350                                          && strlen(cwd_buf + l) < PATH_MAX
1351                                         ) {
1352                                                 pbuf = free_me = xasprintf("~%s", cwd_buf + l);
1353                                         }
1354                                         break;
1355 #endif
1356                                 case 'W':
1357                                         pbuf = cwd_buf;
1358                                         cp = strrchr(pbuf, '/');
1359                                         if (cp != NULL && cp != pbuf)
1360                                                 pbuf += (cp-pbuf) + 1;
1361                                         break;
1362                                 case '!':
1363                                         pbuf = free_me = xasprintf("%d", num_ok_lines);
1364                                         break;
1365                                 case 'e': case 'E':     /* \e \E = \033 */
1366                                         c = '\033';
1367                                         break;
1368                                 case 'x': case 'X': {
1369                                         char buf2[4];
1370                                         for (l = 0; l < 3;) {
1371                                                 unsigned h;
1372                                                 buf2[l++] = *prmt_ptr;
1373                                                 buf2[l] = '\0';
1374                                                 h = strtoul(buf2, &pbuf, 16);
1375                                                 if (h > UCHAR_MAX || (pbuf - buf2) < l) {
1376                                                         buf2[--l] = '\0';
1377                                                         break;
1378                                                 }
1379                                                 prmt_ptr++;
1380                                         }
1381                                         c = (char)strtoul(buf2, NULL, 16);
1382                                         if (c == 0)
1383                                                 c = '?';
1384                                         pbuf = cbuf;
1385                                         break;
1386                                 }
1387                                 case '[': case ']':
1388                                         if (c == flg_not_length) {
1389                                                 flg_not_length = (flg_not_length == '[' ? ']' : '[');
1390                                                 continue;
1391                                         }
1392                                         break;
1393                                 } /* switch */
1394                         } /* if */
1395                 } /* if */
1396                 cbuf[0] = c;
1397                 cur_prmt_len = strlen(pbuf);
1398                 prmt_len += cur_prmt_len;
1399                 if (flg_not_length != ']')
1400                         cmdedit_prmt_len += cur_prmt_len;
1401                 prmt_mem_ptr = strcat(xrealloc(prmt_mem_ptr, prmt_len+1), pbuf);
1402                 free(free_me);
1403         } /* while */
1404
1405         if (cwd_buf != (char *)bb_msg_unknown)
1406                 free(cwd_buf);
1407         cmdedit_prompt = prmt_mem_ptr;
1408         put_prompt();
1409 }
1410 #endif
1411
1412 static void cmdedit_setwidth(unsigned w, int redraw_flg)
1413 {
1414         cmdedit_termw = w;
1415         if (redraw_flg) {
1416                 /* new y for current cursor */
1417                 int new_y = (cursor + cmdedit_prmt_len) / w;
1418                 /* redraw */
1419                 redraw((new_y >= cmdedit_y ? new_y : cmdedit_y), command_len - cursor);
1420                 fflush(stdout);
1421         }
1422 }
1423
1424 static void win_changed(int nsig)
1425 {
1426         unsigned width;
1427         get_terminal_width_height(0, &width, NULL);
1428         cmdedit_setwidth(width, nsig /* - just a yes/no flag */);
1429         if (nsig == SIGWINCH)
1430                 signal(SIGWINCH, win_changed); /* rearm ourself */
1431 }
1432
1433 /*
1434  * The emacs and vi modes share much of the code in the big
1435  * command loop.  Commands entered when in vi's command mode (aka
1436  * "escape mode") get an extra bit added to distinguish them --
1437  * this keeps them from being self-inserted.  This clutters the
1438  * big switch a bit, but keeps all the code in one place.
1439  */
1440
1441 #define vbit 0x100
1442
1443 /* leave out the "vi-mode"-only case labels if vi editing isn't
1444  * configured. */
1445 #define vi_case(caselabel) USE_FEATURE_EDITING(case caselabel)
1446
1447 /* convert uppercase ascii to equivalent control char, for readability */
1448 #undef CTRL
1449 #define CTRL(a) ((a) & ~0x40)
1450
1451 /* Returns:
1452  * -1 on read errors or EOF, or on bare Ctrl-D,
1453  * 0  on ctrl-C (the line entered is still returned in 'command'),
1454  * >0 length of input string, including terminating '\n'
1455  */
1456 int FAST_FUNC read_line_input(const char *prompt, char *command, int maxsize, line_input_t *st)
1457 {
1458         int len;
1459 #if ENABLE_FEATURE_TAB_COMPLETION
1460         smallint lastWasTab = FALSE;
1461 #endif
1462         unsigned ic;
1463         unsigned char c;
1464         smallint break_out = 0;
1465 #if ENABLE_FEATURE_EDITING_VI
1466         smallint vi_cmdmode = 0;
1467         smalluint prevc;
1468 #endif
1469         struct termios initial_settings;
1470         struct termios new_settings;
1471
1472         INIT_S();
1473
1474         if (tcgetattr(STDIN_FILENO, &initial_settings) < 0
1475          || !(initial_settings.c_lflag & ECHO)
1476         ) {
1477                 /* Happens when e.g. stty -echo was run before */
1478                 parse_and_put_prompt(prompt);
1479                 fflush(stdout);
1480                 if (fgets(command, maxsize, stdin) == NULL)
1481                         len = -1; /* EOF or error */
1482                 else
1483                         len = strlen(command);
1484                 DEINIT_S();
1485                 return len;
1486         }
1487
1488 // FIXME: audit & improve this
1489         if (maxsize > MAX_LINELEN)
1490                 maxsize = MAX_LINELEN;
1491
1492         /* With null flags, no other fields are ever used */
1493         state = st ? st : (line_input_t*) &const_int_0;
1494 #if ENABLE_FEATURE_EDITING_SAVEHISTORY
1495         if ((state->flags & SAVE_HISTORY) && state->hist_file)
1496                 if (state->cnt_history == 0)
1497                         load_history(state);
1498 #endif
1499         if (state->flags & DO_HISTORY)
1500                 state->cur_history = state->cnt_history;
1501
1502         /* prepare before init handlers */
1503         cmdedit_y = 0;  /* quasireal y, not true if line > xt*yt */
1504         command_len = 0;
1505         command_ps = command;
1506         command[0] = '\0';
1507
1508         new_settings = initial_settings;
1509         new_settings.c_lflag &= ~ICANON;        /* unbuffered input */
1510         /* Turn off echoing and CTRL-C, so we can trap it */
1511         new_settings.c_lflag &= ~(ECHO | ECHONL | ISIG);
1512         /* Hmm, in linux c_cc[] is not parsed if ICANON is off */
1513         new_settings.c_cc[VMIN] = 1;
1514         new_settings.c_cc[VTIME] = 0;
1515         /* Turn off CTRL-C, so we can trap it */
1516 #ifndef _POSIX_VDISABLE
1517 #define _POSIX_VDISABLE '\0'
1518 #endif
1519         new_settings.c_cc[VINTR] = _POSIX_VDISABLE;
1520         tcsetattr_stdin_TCSANOW(&new_settings);
1521
1522         /* Now initialize things */
1523         previous_SIGWINCH_handler = signal(SIGWINCH, win_changed);
1524         win_changed(0); /* do initial resizing */
1525 #if ENABLE_FEATURE_GETUSERNAME_AND_HOMEDIR
1526         {
1527                 struct passwd *entry;
1528
1529                 entry = getpwuid(geteuid());
1530                 if (entry) {
1531                         user_buf = xstrdup(entry->pw_name);
1532                         home_pwd_buf = xstrdup(entry->pw_dir);
1533                 }
1534         }
1535 #endif
1536
1537 #if 0
1538         for (ic = 0; ic <= MAX_HISTORY; ic++)
1539                 bb_error_msg("history[%d]:'%s'", ic, state->history[ic]);
1540         bb_error_msg("cur_history:%d cnt_history:%d", state->cur_history, state->cnt_history);
1541 #endif
1542
1543         /* Print out the command prompt */
1544         parse_and_put_prompt(prompt);
1545
1546         while (1) {
1547                 fflush(NULL);
1548
1549                 if (nonblock_safe_read(STDIN_FILENO, &c, 1) < 1) {
1550                         /* if we can't read input then exit */
1551                         goto prepare_to_die;
1552                 }
1553
1554                 ic = c;
1555
1556 #if ENABLE_FEATURE_EDITING_VI
1557                 newdelflag = 1;
1558                 if (vi_cmdmode)
1559                         ic |= vbit;
1560 #endif
1561                 switch (ic) {
1562                 case '\n':
1563                 case '\r':
1564                 vi_case('\n'|vbit:)
1565                 vi_case('\r'|vbit:)
1566                         /* Enter */
1567                         goto_new_line();
1568                         break_out = 1;
1569                         break;
1570                 case CTRL('A'):
1571                 vi_case('0'|vbit:)
1572                         /* Control-a -- Beginning of line */
1573                         input_backward(cursor);
1574                         break;
1575                 case CTRL('B'):
1576                 vi_case('h'|vbit:)
1577                 vi_case('\b'|vbit:)
1578                 vi_case('\x7f'|vbit:) /* DEL */
1579                         /* Control-b -- Move back one character */
1580                         input_backward(1);
1581                         break;
1582                 case CTRL('C'):
1583                 vi_case(CTRL('C')|vbit:)
1584                         /* Control-c -- stop gathering input */
1585                         goto_new_line();
1586                         command_len = 0;
1587                         break_out = -1; /* "do not append '\n'" */
1588                         break;
1589                 case CTRL('D'):
1590                         /* Control-d -- Delete one character, or exit
1591                          * if the len=0 and no chars to delete */
1592                         if (command_len == 0) {
1593                                 errno = 0;
1594  prepare_to_die:
1595                                 /* to control stopped jobs */
1596                                 break_out = command_len = -1;
1597                                 break;
1598                         }
1599                         input_delete(0);
1600                         break;
1601
1602                 case CTRL('E'):
1603                 vi_case('$'|vbit:)
1604                         /* Control-e -- End of line */
1605                         input_end();
1606                         break;
1607                 case CTRL('F'):
1608                 vi_case('l'|vbit:)
1609                 vi_case(' '|vbit:)
1610                         /* Control-f -- Move forward one character */
1611                         input_forward();
1612                         break;
1613
1614                 case '\b':
1615                 case '\x7f': /* DEL */
1616                         /* Control-h and DEL */
1617                         input_backspace();
1618                         break;
1619
1620 #if ENABLE_FEATURE_TAB_COMPLETION
1621                 case '\t':
1622                         input_tab(&lastWasTab);
1623                         break;
1624 #endif
1625
1626                 case CTRL('K'):
1627                         /* Control-k -- clear to end of line */
1628                         command[cursor] = 0;
1629                         command_len = cursor;
1630                         printf("\033[J");
1631                         break;
1632                 case CTRL('L'):
1633                 vi_case(CTRL('L')|vbit:)
1634                         /* Control-l -- clear screen */
1635                         printf("\033[H");
1636                         redraw(0, command_len - cursor);
1637                         break;
1638
1639 #if MAX_HISTORY > 0
1640                 case CTRL('N'):
1641                 vi_case(CTRL('N')|vbit:)
1642                 vi_case('j'|vbit:)
1643                         /* Control-n -- Get next command in history */
1644                         if (get_next_history())
1645                                 goto rewrite_line;
1646                         break;
1647                 case CTRL('P'):
1648                 vi_case(CTRL('P')|vbit:)
1649                 vi_case('k'|vbit:)
1650                         /* Control-p -- Get previous command from history */
1651                         if (get_previous_history())
1652                                 goto rewrite_line;
1653                         break;
1654 #endif
1655
1656                 case CTRL('U'):
1657                 vi_case(CTRL('U')|vbit:)
1658                         /* Control-U -- Clear line before cursor */
1659                         if (cursor) {
1660                                 overlapping_strcpy(command, command + cursor);
1661                                 command_len -= cursor;
1662                                 redraw(cmdedit_y, command_len);
1663                         }
1664                         break;
1665                 case CTRL('W'):
1666                 vi_case(CTRL('W')|vbit:)
1667                         /* Control-W -- Remove the last word */
1668                         while (cursor > 0 && isspace(command[cursor-1]))
1669                                 input_backspace();
1670                         while (cursor > 0 && !isspace(command[cursor-1]))
1671                                 input_backspace();
1672                         break;
1673
1674 #if ENABLE_FEATURE_EDITING_VI
1675                 case 'i'|vbit:
1676                         vi_cmdmode = 0;
1677                         break;
1678                 case 'I'|vbit:
1679                         input_backward(cursor);
1680                         vi_cmdmode = 0;
1681                         break;
1682                 case 'a'|vbit:
1683                         input_forward();
1684                         vi_cmdmode = 0;
1685                         break;
1686                 case 'A'|vbit:
1687                         input_end();
1688                         vi_cmdmode = 0;
1689                         break;
1690                 case 'x'|vbit:
1691                         input_delete(1);
1692                         break;
1693                 case 'X'|vbit:
1694                         if (cursor > 0) {
1695                                 input_backward(1);
1696                                 input_delete(1);
1697                         }
1698                         break;
1699                 case 'W'|vbit:
1700                         vi_Word_motion(command, 1);
1701                         break;
1702                 case 'w'|vbit:
1703                         vi_word_motion(command, 1);
1704                         break;
1705                 case 'E'|vbit:
1706                         vi_End_motion(command);
1707                         break;
1708                 case 'e'|vbit:
1709                         vi_end_motion(command);
1710                         break;
1711                 case 'B'|vbit:
1712                         vi_Back_motion(command);
1713                         break;
1714                 case 'b'|vbit:
1715                         vi_back_motion(command);
1716                         break;
1717                 case 'C'|vbit:
1718                         vi_cmdmode = 0;
1719                         /* fall through */
1720                 case 'D'|vbit:
1721                         goto clear_to_eol;
1722
1723                 case 'c'|vbit:
1724                         vi_cmdmode = 0;
1725                         /* fall through */
1726                 case 'd'|vbit: {
1727                         int nc, sc;
1728                         sc = cursor;
1729                         prevc = ic;
1730                         if (safe_read(STDIN_FILENO, &c, 1) < 1)
1731                                 goto prepare_to_die;
1732                         if (c == (prevc & 0xff)) {
1733                                 /* "cc", "dd" */
1734                                 input_backward(cursor);
1735                                 goto clear_to_eol;
1736                                 break;
1737                         }
1738                         switch (c) {
1739                         case 'w':
1740                         case 'W':
1741                         case 'e':
1742                         case 'E':
1743                                 switch (c) {
1744                                 case 'w':   /* "dw", "cw" */
1745                                         vi_word_motion(command, vi_cmdmode);
1746                                         break;
1747                                 case 'W':   /* 'dW', 'cW' */
1748                                         vi_Word_motion(command, vi_cmdmode);
1749                                         break;
1750                                 case 'e':   /* 'de', 'ce' */
1751                                         vi_end_motion(command);
1752                                         input_forward();
1753                                         break;
1754                                 case 'E':   /* 'dE', 'cE' */
1755                                         vi_End_motion(command);
1756                                         input_forward();
1757                                         break;
1758                                 }
1759                                 nc = cursor;
1760                                 input_backward(cursor - sc);
1761                                 while (nc-- > cursor)
1762                                         input_delete(1);
1763                                 break;
1764                         case 'b':  /* "db", "cb" */
1765                         case 'B':  /* implemented as B */
1766                                 if (c == 'b')
1767                                         vi_back_motion(command);
1768                                 else
1769                                         vi_Back_motion(command);
1770                                 while (sc-- > cursor)
1771                                         input_delete(1);
1772                                 break;
1773                         case ' ':  /* "d ", "c " */
1774                                 input_delete(1);
1775                                 break;
1776                         case '$':  /* "d$", "c$" */
1777                         clear_to_eol:
1778                                 while (cursor < command_len)
1779                                         input_delete(1);
1780                                 break;
1781                         }
1782                         break;
1783                 }
1784                 case 'p'|vbit:
1785                         input_forward();
1786                         /* fallthrough */
1787                 case 'P'|vbit:
1788                         put();
1789                         break;
1790                 case 'r'|vbit:
1791                         if (safe_read(STDIN_FILENO, &c, 1) < 1)
1792                                 goto prepare_to_die;
1793                         if (c == 0)
1794                                 beep();
1795                         else {
1796                                 *(command + cursor) = c;
1797                                 bb_putchar(c);
1798                                 bb_putchar('\b');
1799                         }
1800                         break;
1801 #endif /* FEATURE_COMMAND_EDITING_VI */
1802
1803                 case '\x1b': /* ESC */
1804
1805 #if ENABLE_FEATURE_EDITING_VI
1806                         if (state->flags & VI_MODE) {
1807                                 /* ESC: insert mode --> command mode */
1808                                 vi_cmdmode = 1;
1809                                 input_backward(1);
1810                                 break;
1811                         }
1812 #endif
1813                         /* escape sequence follows */
1814                         if (safe_read(STDIN_FILENO, &c, 1) < 1)
1815                                 goto prepare_to_die;
1816                         /* different vt100 emulations */
1817                         if (c == '[' || c == 'O') {
1818                 vi_case('['|vbit:)
1819                 vi_case('O'|vbit:)
1820                                 if (safe_read(STDIN_FILENO, &c, 1) < 1)
1821                                         goto prepare_to_die;
1822                         }
1823                         if (c >= '1' && c <= '9') {
1824                                 unsigned char dummy;
1825
1826                                 if (safe_read(STDIN_FILENO, &dummy, 1) < 1)
1827                                         goto prepare_to_die;
1828                                 if (dummy != '~')
1829                                         c = '\0';
1830                         }
1831
1832                         switch (c) {
1833 #if ENABLE_FEATURE_TAB_COMPLETION
1834                         case '\t':                      /* Alt-Tab */
1835                                 input_tab(&lastWasTab);
1836                                 break;
1837 #endif
1838 #if MAX_HISTORY > 0
1839                         case 'A':
1840                                 /* Up Arrow -- Get previous command from history */
1841                                 if (get_previous_history())
1842                                         goto rewrite_line;
1843                                 beep();
1844                                 break;
1845                         case 'B':
1846                                 /* Down Arrow -- Get next command in history */
1847                                 if (!get_next_history())
1848                                         break;
1849  rewrite_line:
1850                                 /* Rewrite the line with the selected history item */
1851                                 /* change command */
1852                                 command_len = strlen(strcpy(command, state->history[state->cur_history] ? : ""));
1853                                 /* redraw and go to eol (bol, in vi */
1854                                 redraw(cmdedit_y, (state->flags & VI_MODE) ? 9999 : 0);
1855                                 break;
1856 #endif
1857                         case 'C':
1858                                 /* Right Arrow -- Move forward one character */
1859                                 input_forward();
1860                                 break;
1861                         case 'D':
1862                                 /* Left Arrow -- Move back one character */
1863                                 input_backward(1);
1864                                 break;
1865                         case '3':
1866                                 /* Delete */
1867                                 input_delete(0);
1868                                 break;
1869                         case '1': // vt100? linux vt? or what?
1870                         case '7': // vt100? linux vt? or what?
1871                         case 'H': /* xterm's <Home> */
1872                                 input_backward(cursor);
1873                                 break;
1874                         case '4': // vt100? linux vt? or what?
1875                         case '8': // vt100? linux vt? or what?
1876                         case 'F': /* xterm's <End> */
1877                                 input_end();
1878                                 break;
1879                         default:
1880                                 c = '\0';
1881                                 beep();
1882                         }
1883                         break;
1884
1885                 default:        /* If it's regular input, do the normal thing */
1886
1887                         /* Control-V -- force insert of next char */
1888                         if (c == CTRL('V')) {
1889                                 if (safe_read(STDIN_FILENO, &c, 1) < 1)
1890                                         goto prepare_to_die;
1891                                 if (c == 0) {
1892                                         beep();
1893                                         break;
1894                                 }
1895                         }
1896
1897 #if ENABLE_FEATURE_EDITING_VI
1898                         if (vi_cmdmode)  /* Don't self-insert */
1899                                 break;
1900 #endif
1901                         if ((int)command_len >= (maxsize - 2))        /* Need to leave space for enter */
1902                                 break;
1903
1904                         command_len++;
1905                         if (cursor == (command_len - 1)) {      /* Append if at the end of the line */
1906                                 command[cursor] = c;
1907                                 command[cursor+1] = '\0';
1908                                 cmdedit_set_out_char(' ');
1909                         } else {                        /* Insert otherwise */
1910                                 int sc = cursor;
1911
1912                                 memmove(command + sc + 1, command + sc, command_len - sc);
1913                                 command[sc] = c;
1914                                 sc++;
1915                                 /* rewrite from cursor */
1916                                 input_end();
1917                                 /* to prev x pos + 1 */
1918                                 input_backward(cursor - sc);
1919                         }
1920                         break;
1921                 }
1922                 if (break_out)                  /* Enter is the command terminator, no more input. */
1923                         break;
1924
1925 #if ENABLE_FEATURE_TAB_COMPLETION
1926                 if (c != '\t')
1927                         lastWasTab = FALSE;
1928 #endif
1929         }
1930
1931         if (command_len > 0)
1932                 remember_in_history(command);
1933
1934         if (break_out > 0) {
1935                 command[command_len++] = '\n';
1936                 command[command_len] = '\0';
1937         }
1938
1939 #if ENABLE_FEATURE_TAB_COMPLETION
1940         free_tab_completion_data();
1941 #endif
1942
1943         /* restore initial_settings */
1944         tcsetattr_stdin_TCSANOW(&initial_settings);
1945         /* restore SIGWINCH handler */
1946         signal(SIGWINCH, previous_SIGWINCH_handler);
1947         fflush(stdout);
1948
1949         len = command_len;
1950         DEINIT_S();
1951
1952         return len; /* can't return command_len, DEINIT_S() destroys it */
1953 }
1954
1955 #else
1956
1957 #undef read_line_input
1958 int FAST_FUNC read_line_input(const char* prompt, char* command, int maxsize)
1959 {
1960         fputs(prompt, stdout);
1961         fflush(stdout);
1962         fgets(command, maxsize, stdin);
1963         return strlen(command);
1964 }
1965
1966 #endif  /* FEATURE_COMMAND_EDITING */
1967
1968
1969 /*
1970  * Testing
1971  */
1972
1973 #ifdef TEST
1974
1975 #include <locale.h>
1976
1977 const char *applet_name = "debug stuff usage";
1978
1979 int main(int argc, char **argv)
1980 {
1981         char buff[MAX_LINELEN];
1982         char *prompt =
1983 #if ENABLE_FEATURE_EDITING_FANCY_PROMPT
1984                 "\\[\\033[32;1m\\]\\u@\\[\\x1b[33;1m\\]\\h:"
1985                 "\\[\\033[34;1m\\]\\w\\[\\033[35;1m\\] "
1986                 "\\!\\[\\e[36;1m\\]\\$ \\[\\E[0m\\]";
1987 #else
1988                 "% ";
1989 #endif
1990
1991 #if ENABLE_FEATURE_NONPRINTABLE_INVERSE_PUT
1992         setlocale(LC_ALL, "");
1993 #endif
1994         while (1) {
1995                 int l;
1996                 l = read_line_input(prompt, buff);
1997                 if (l <= 0 || buff[l-1] != '\n')
1998                         break;
1999                 buff[l-1] = 0;
2000                 printf("*** read_line_input() returned line =%s=\n", buff);
2001         }
2002         printf("*** read_line_input() detect ^D\n");
2003         return 0;
2004 }
2005
2006 #endif  /* TEST */