/* vi: set sw=4 ts=4: */ /* * Termios command line History and Editting. * * Copyright (c) 1986-2001 may safely be consumed by a BSD or GPL license. * Written by: Vladimir Oleynik <vodz@usa.net> * * Used ideas: * Adam Rogoyski <rogoyski@cs.utexas.edu> * Dave Cinege <dcinege@psychosis.com> * Jakub Jelinek (c) 1995 * Erik Andersen <andersee@debian.org> (Majorly adjusted for busybox) * * This code is 'as is' with no warranty. * * */ /* Usage and Known bugs: Terminal key codes are not extensive, and more will probably need to be added. This version was created on Debian GNU/Linux 2.x. Delete, Backspace, Home, End, and the arrow keys were tested to work in an Xterm and console. Ctrl-A also works as Home. Ctrl-E also works as End. Small bugs (simple effect): - not true viewing if terminal size (x*y symbols) less size (prompt + editor`s line + 2 symbols) */ //#define TEST #include <stdio.h> #include <errno.h> #include <unistd.h> #include <stdlib.h> #include <string.h> #include <sys/ioctl.h> #include <ctype.h> #include <signal.h> #include <limits.h> #ifndef TEST #include "busybox.h" #define D(x) #else #define BB_FEATURE_SH_COMMAND_EDITING #define BB_FEATURE_SH_TAB_COMPLETION #define BB_FEATURE_SH_USERNAME_COMPLETION #define BB_FEATURE_NONPRINTABLE_INVERSE_PUT #define BB_FEATURE_BASH_STYLE_PROMT #define BB_FEATURE_CLEAN_UP #define D(x) x #endif /* TEST */ #ifdef BB_FEATURE_SH_TAB_COMPLETION #include <dirent.h> #include <sys/stat.h> #endif #ifdef BB_FEATURE_SH_COMMAND_EDITING #ifndef BB_FEATURE_SH_TAB_COMPLETION #undef BB_FEATURE_SH_USERNAME_COMPLETION #endif #if defined(BB_FEATURE_SH_USERNAME_COMPLETION) || defined(BB_FEATURE_BASH_STYLE_PROMT) #define BB_FEATURE_GETUSERNAME_AND_HOMEDIR #endif #ifdef BB_FEATURE_GETUSERNAME_AND_HOMEDIR #ifndef TEST #include "pwd_grp/pwd.h" #else #include <pwd.h> #endif /* TEST */ #endif /* advanced FEATURES */ #ifdef TEST void *xrealloc(void *old, size_t size) { return realloc(old, size); } void *xmalloc(size_t size) { return malloc(size); } char *xstrdup(const char *s) { return strdup(s); } void *xcalloc(size_t size, size_t se) { return calloc(size, se); } #define error_msg(s, d) fprintf(stderr, s, d) #endif struct history { char *s; struct history *p; struct history *n; }; /* Maximum length of the linked list for the command line history */ static const int MAX_HISTORY = 15; /* First element in command line list */ static struct history *his_front = NULL; /* Last element in command line list */ static struct history *his_end = NULL; /* ED: sparc termios is broken: revert back to old termio handling. */ #if #cpu(sparc) # include <termio.h> # define termios termio # define setTermSettings(fd,argp) ioctl(fd,TCSETAF,argp) # define getTermSettings(fd,argp) ioctl(fd,TCGETA,argp) #else # include <termios.h> # define setTermSettings(fd,argp) tcsetattr(fd,TCSANOW,argp) # define getTermSettings(fd,argp) tcgetattr(fd, argp); #endif /* Current termio and the previous termio before starting sh */ static struct termios initial_settings, new_settings; #ifndef _POSIX_VDISABLE #define _POSIX_VDISABLE '\0' #endif static volatile int cmdedit_termw = 80; /* actual terminal width */ static int history_counter = 0; /* Number of commands in history list */ static volatile int handlers_sets = 0; /* Set next bites: */ enum { SET_ATEXIT = 1, /* when atexit() has been called and get euid,uid,gid to fast compare */ SET_TERM_HANDLERS = 2, /* set many terminates signal handlers */ SET_WCHG_HANDLERS = 4, /* winchg signal handler */ SET_RESET_TERM = 8, /* if the terminal needs to be reset upon exit */ }; static int cmdedit_x; /* real x terminal position */ static int cmdedit_y; /* pseudoreal y terminal position */ static int cmdedit_prmt_len; /* lenght prompt without colores string */ static int cursor; /* required global for signal handler */ static int len; /* --- "" - - "" - -"- --""-- --""--- */ static char *command_ps; /* --- "" - - "" - -"- --""-- --""--- */ static #ifndef BB_FEATURE_BASH_STYLE_PROMT const #endif char *cmdedit_prompt; /* --- "" - - "" - -"- --""-- --""--- */ /* Link into lash to reset context to 0 on ^C and such */ extern unsigned int shell_context; #ifdef BB_FEATURE_GETUSERNAME_AND_HOMEDIR static char *user_buf = ""; static char *home_pwd_buf = ""; static int my_euid; #endif #ifdef BB_FEATURE_BASH_STYLE_PROMT static char *hostname_buf = ""; static int num_ok_lines = 1; #endif #ifdef BB_FEATURE_SH_TAB_COMPLETION #ifndef BB_FEATURE_GETUSERNAME_AND_HOMEDIR static int my_euid; #endif static int my_uid; static int my_gid; #endif /* BB_FEATURE_SH_TAB_COMPLETION */ static void cmdedit_setwidth(int w, int redraw_flg); static void win_changed(int nsig) { struct winsize win = { 0, 0, 0, 0 }; static __sighandler_t previous_SIGWINCH_handler; /* for reset */ /* emulate || signal call */ if (nsig == -SIGWINCH || nsig == SIGWINCH) { ioctl(0, TIOCGWINSZ, &win); if (win.ws_col > 0) { cmdedit_setwidth(win.ws_col, nsig == SIGWINCH); } } /* Unix not all standart in recall signal */ if (nsig == -SIGWINCH) /* save previous handler */ previous_SIGWINCH_handler = signal(SIGWINCH, win_changed); else if (nsig == SIGWINCH) /* signaled called handler */ signal(SIGWINCH, win_changed); /* set for next call */ else /* nsig == 0 */ /* set previous handler */ signal(SIGWINCH, previous_SIGWINCH_handler); /* reset */ } static void cmdedit_reset_term(void) { if ((handlers_sets & SET_RESET_TERM) != 0) { /* sparc and other have broken termios support: use old termio handling. */ setTermSettings(fileno(stdin), (void *) &initial_settings); handlers_sets &= ~SET_RESET_TERM; } if ((handlers_sets & SET_WCHG_HANDLERS) != 0) { /* reset SIGWINCH handler to previous (default) */ win_changed(0); handlers_sets &= ~SET_WCHG_HANDLERS; } fflush(stdout); #ifdef BB_FEATURE_CLEAN_UP if (his_front) { struct history *n; //while(his_front!=his_end) { while (his_front != his_end) { n = his_front->n; free(his_front->s); free(his_front); his_front = n; } } #endif } /* special for recount position for scroll and remove terminal margin effect */ static void cmdedit_set_out_char(int next_char) { int c = command_ps[cursor]; if (c == 0) c = ' '; /* destroy end char? */ #ifdef BB_FEATURE_NONPRINTABLE_INVERSE_PUT if (!isprint(c)) { /* Inverse put non-printable characters */ if (((unsigned char) c) >= 128) c -= 128; if (((unsigned char) c) < ' ') c += '@'; if (c == 127) c = '?'; printf("\033[7m%c\033[0m", c); } else #endif putchar(c); if (++cmdedit_x >= cmdedit_termw) { /* terminal is scrolled down */ cmdedit_y++; cmdedit_x = 0; if (!next_char) next_char = ' '; /* destroy "(auto)margin" */ putchar(next_char); putchar('\b'); } cursor++; } /* Move to end line. Bonus: rewrite line from cursor */ static void input_end(void) { while (cursor < len) cmdedit_set_out_char(0); } /* Go to the next line */ static void goto_new_line(void) { input_end(); if (cmdedit_x) putchar('\n'); } static inline void out1str(const char *s) { fputs(s, stdout); } static inline void beep(void) { putchar('\007'); } /* Move back one charactor */ /* special for slow terminal */ static void input_backward(int num) { if (num > cursor) num = cursor; cursor -= num; /* new cursor (in command, not terminal) */ if (cmdedit_x >= num) { /* no to up line */ cmdedit_x -= num; if (num < 4) while (num-- > 0) putchar('\b'); else printf("\033[%dD", num); } else { int count_y; if (cmdedit_x) { putchar('\r'); /* back to first terminal pos. */ num -= cmdedit_x; /* set previous backward */ } count_y = 1 + num / cmdedit_termw; printf("\033[%dA", count_y); cmdedit_y -= count_y; /* require forward after uping */ cmdedit_x = cmdedit_termw * count_y - num; printf("\033[%dC", cmdedit_x); /* set term cursor */ } } static void put_prompt(void) { out1str(cmdedit_prompt); cmdedit_x = cmdedit_prmt_len; /* count real x terminal position */ cursor = 0; } #ifdef BB_FEATURE_BASH_STYLE_PROMT static void add_to_prompt(char **prmt_mem_ptr, int *alm, int *prmt_len, const char *addb) { int l = strlen(addb); *prmt_len += l; if (*alm < (*prmt_len) + 1) { *alm = (*prmt_len) + 1; *prmt_mem_ptr = xrealloc(*prmt_mem_ptr, *alm); } strcat(*prmt_mem_ptr, addb); } #endif static void parse_prompt(const char *prmt_ptr) { #ifdef BB_FEATURE_BASH_STYLE_PROMT int alm = strlen(prmt_ptr) + 1; /* supposedly require memory */ int prmt_len = 0; int sub_len = 0; int flg_not_length = '['; char *prmt_mem_ptr = xstrdup(prmt_ptr); char pwd_buf[PATH_MAX + 1]; char buf[16]; int c; pwd_buf[0] = 0; *prmt_mem_ptr = 0; while (*prmt_ptr) { c = *prmt_ptr++; if (c == '\\') { c = *prmt_ptr; if (c == 0) break; prmt_ptr++; switch (c) { case 'u': add_to_prompt(&prmt_mem_ptr, &alm, &prmt_len, user_buf); continue; case 'h': if (hostname_buf[0] == 0) { hostname_buf = xcalloc(256, 1); if (gethostname(hostname_buf, 255) < 0) { strcpy(hostname_buf, "?"); } else { char *s = strchr(hostname_buf, '.'); if (s) *s = 0; } } add_to_prompt(&prmt_mem_ptr, &alm, &prmt_len, hostname_buf); continue; case '$': c = my_euid == 0 ? '#' : '$'; break; case 'w': if (pwd_buf[0] == 0) { int l; getcwd(pwd_buf, PATH_MAX); l = strlen(home_pwd_buf); if (home_pwd_buf[0] != 0 && strncmp(home_pwd_buf, pwd_buf, l) == 0) { strcpy(pwd_buf + 1, pwd_buf + l); pwd_buf[0] = '~'; } } add_to_prompt(&prmt_mem_ptr, &alm, &prmt_len, pwd_buf); continue; case '!': snprintf(buf, sizeof(buf), "%d", num_ok_lines); add_to_prompt(&prmt_mem_ptr, &alm, &prmt_len, buf); continue; case 'e': case 'E': /* \e \E = \033 */ c = '\033'; break; case 'x': case 'X': case '0': case '1': case '2': case '3': case '4': case '5': case '6': case '7':{ int l; int ho = 0; char *eho; if (c == 'X') c = 'x'; for (l = 0; l < 3;) { buf[l++] = *prmt_ptr; buf[l] = 0; ho = strtol(buf, &eho, c == 'x' ? 16 : 8); if (ho > UCHAR_MAX || (eho - buf) < l) { l--; break; } prmt_ptr++; } buf[l] = 0; ho = strtol(buf, 0, c == 'x' ? 16 : 8); c = ho == 0 ? '?' : (char) ho; break; } case '[': case ']': if (c == flg_not_length) { flg_not_length = flg_not_length == '[' ? ']' : '['; continue; } break; } } buf[0] = c; buf[1] = 0; add_to_prompt(&prmt_mem_ptr, &alm, &prmt_len, buf); if (flg_not_length == ']') sub_len++; } cmdedit_prmt_len = prmt_len - sub_len; cmdedit_prompt = prmt_mem_ptr; #else cmdedit_prompt = prmt_ptr; cmdedit_prmt_len = strlen(prmt_ptr); #endif put_prompt(); } /* draw promt, editor line, and clear tail */ static void redraw(int y, int back_cursor) { if (y > 0) /* up to start y */ printf("\033[%dA", y); cmdedit_y = 0; /* new quasireal y */ putchar('\r'); put_prompt(); input_end(); /* rewrite */ printf("\033[J"); /* destroy tail after cursor */ input_backward(back_cursor); } /* Delete the char in front of the cursor */ static void input_delete(void) { int j = cursor; if (j == len) return; strcpy(command_ps + j, command_ps + j + 1); len--; input_end(); /* rewtite new line */ cmdedit_set_out_char(0); /* destroy end char */ input_backward(cursor - j); /* back to old pos cursor */ } /* Delete the char in back of the cursor */ static void input_backspace(void) { if (cursor > 0) { input_backward(1); input_delete(); } } /* Move forward one charactor */ static void input_forward(void) { if (cursor < len) cmdedit_set_out_char(command_ps[cursor + 1]); } static void clean_up_and_die(int sig) { goto_new_line(); if (sig != SIGINT) exit(EXIT_SUCCESS); /* cmdedit_reset_term() called in atexit */ cmdedit_reset_term(); } static void cmdedit_setwidth(int w, int redraw_flg) { cmdedit_termw = cmdedit_prmt_len + 2; if (w <= cmdedit_termw) { cmdedit_termw = cmdedit_termw % w; } if (w > cmdedit_termw) { cmdedit_termw = w; if (redraw_flg) { /* new y for current cursor */ int new_y = (cursor + cmdedit_prmt_len) / w; /* redraw */ redraw((new_y >= cmdedit_y ? new_y : cmdedit_y), len - cursor); fflush(stdout); } } } extern void cmdedit_init(void) { if ((handlers_sets & SET_WCHG_HANDLERS) == 0) { /* emulate usage handler to set handler and call yours work */ win_changed(-SIGWINCH); handlers_sets |= SET_WCHG_HANDLERS; } if ((handlers_sets & SET_ATEXIT) == 0) { #ifdef BB_FEATURE_GETUSERNAME_AND_HOMEDIR struct passwd *entry; my_euid = geteuid(); entry = getpwuid(my_euid); if (entry) { user_buf = xstrdup(entry->pw_name); home_pwd_buf = xstrdup(entry->pw_dir); } #endif #ifdef BB_FEATURE_SH_TAB_COMPLETION #ifndef BB_FEATURE_GETUSERNAME_AND_HOMEDIR my_euid = geteuid(); #endif my_uid = getuid(); my_gid = getgid(); #endif /* BB_FEATURE_SH_TAB_COMPLETION */ handlers_sets |= SET_ATEXIT; atexit(cmdedit_reset_term); /* be sure to do this only once */ } if ((handlers_sets & SET_TERM_HANDLERS) == 0) { signal(SIGKILL, clean_up_and_die); signal(SIGINT, clean_up_and_die); signal(SIGQUIT, clean_up_and_die); signal(SIGTERM, clean_up_and_die); handlers_sets |= SET_TERM_HANDLERS; } } #ifdef BB_FEATURE_SH_TAB_COMPLETION static int is_execute(const struct stat *st) { if ((!my_euid && (st->st_mode & (S_IXUSR | S_IXGRP | S_IXOTH))) || (my_uid == st->st_uid && (st->st_mode & S_IXUSR)) || (my_gid == st->st_gid && (st->st_mode & S_IXGRP)) || (st->st_mode & S_IXOTH)) return TRUE; return FALSE; } #ifdef BB_FEATURE_SH_USERNAME_COMPLETION static char **username_tab_completion(char *ud, int *num_matches) { struct passwd *entry; int userlen; char *temp; ud++; /* ~user/... to user/... */ userlen = strlen(ud); if (num_matches == 0) { /* "~/..." or "~user/..." */ char *sav_ud = ud - 1; char *home = 0; if (*ud == '/') { /* "~/..." */ home = home_pwd_buf; } else { /* "~user/..." */ temp = strchr(ud, '/'); *temp = 0; /* ~user\0 */ entry = getpwnam(ud); *temp = '/'; /* restore ~user/... */ ud = temp; if (entry) home = entry->pw_dir; } if (home) { if ((userlen + strlen(home) + 1) < BUFSIZ) { char temp2[BUFSIZ]; /* argument size */ /* /home/user/... */ sprintf(temp2, "%s%s", home, ud); strcpy(sav_ud, temp2); } } return 0; /* void, result save to argument :-) */ } else { /* "~[^/]*" */ char **matches = (char **) NULL; int nm = 0; setpwent(); while ((entry = getpwent()) != NULL) { /* Null usernames should result in all users as possible completions. */ if ( /*!userlen || */ !strncmp(ud, entry->pw_name, userlen)) { temp = xmalloc(3 + strlen(entry->pw_name)); sprintf(temp, "~%s/", entry->pw_name); matches = xrealloc(matches, (nm + 1) * sizeof(char *)); matches[nm++] = temp; } } endpwent(); (*num_matches) = nm; return (matches); } } #endif /* BB_FEATURE_SH_USERNAME_COMPLETION */ enum { FIND_EXE_ONLY = 0, FIND_DIR_ONLY = 1, FIND_FILE_ONLY = 2, }; static int path_parse(char ***p, int flags) { int npth; char *tmp; char *pth; /* if not setenv PATH variable, to search cur dir "." */ if (flags != FIND_EXE_ONLY || (pth = getenv("PATH")) == 0 || /* PATH=<empty> or PATH=:<empty> */ *pth == 0 || (*pth == ':' && *(pth + 1) == 0)) { return 1; } tmp = pth; npth = 0; for (;;) { npth++; /* count words is + 1 count ':' */ tmp = strchr(tmp, ':'); if (tmp) { if (*++tmp == 0) break; /* :<empty> */ } else break; } *p = xmalloc(npth * sizeof(char *)); tmp = pth; (*p)[0] = xstrdup(tmp); npth = 1; /* count words is + 1 count ':' */ for (;;) { tmp = strchr(tmp, ':'); if (tmp) { (*p)[0][(tmp - pth)] = 0; /* ':' -> '\0' */ if (*++tmp == 0) break; /* :<empty> */ } else break; (*p)[npth++] = &(*p)[0][(tmp - pth)]; /* p[next]=p[0][&'\0'+1] */ } return npth; } static char *add_quote_for_spec_chars(char *found) { int l = 0; char *s = xmalloc((strlen(found) + 1) * 2); while (*found) { if (strchr(" `\"#$%^&*()=+{}[]:;\'|\\<>", *found)) s[l++] = '\\'; s[l++] = *found++; } s[l] = 0; return s; } static char **exe_n_cwd_tab_completion(char *command, int *num_matches, int type) { char **matches = 0; DIR *dir; struct dirent *next; char dirbuf[BUFSIZ]; int nm = *num_matches; struct stat st; char *path1[1]; char **paths = path1; int npaths; int i; char found[BUFSIZ + 4 + PATH_MAX]; char *pfind = strrchr(command, '/'); path1[0] = "."; if (pfind == NULL) { /* no dir, if flags==EXE_ONLY - get paths, else "." */ npaths = path_parse(&paths, type); pfind = command; } else { /* with dir */ /* save for change */ strcpy(dirbuf, command); /* set dir only */ dirbuf[(pfind - command) + 1] = 0; #ifdef BB_FEATURE_SH_USERNAME_COMPLETION if (dirbuf[0] == '~') /* ~/... or ~user/... */ username_tab_completion(dirbuf, 0); #endif /* "strip" dirname in command */ pfind++; paths[0] = dirbuf; npaths = 1; /* only 1 dir */ } for (i = 0; i < npaths; i++) { dir = opendir(paths[i]); if (!dir) /* Don't print an error */ continue; while ((next = readdir(dir)) != NULL) { const char *str_merge = "%s/%s"; char *str_found = next->d_name; /* matched ? */ if (strncmp(str_found, pfind, strlen(pfind))) continue; /* not see .name without .match */ if (*str_found == '.' && *pfind == 0) { if (*paths[i] == '/' && paths[i][1] == 0 && str_found[1] == 0) str_found = ""; /* only "/" */ else continue; } if (paths[i][strlen(paths[i]) - 1] == '/') str_merge = "%s%s"; sprintf(found, str_merge, paths[i], str_found); /* hmm, remover in progress? */ if (stat(found, &st) < 0) continue; /* find with dirs ? */ if (paths[i] != dirbuf) strcpy(found, next->d_name); /* only name */ if (S_ISDIR(st.st_mode)) { /* name is directory */ /* algorithmic only "/" ? */ if (*str_found) strcat(found, "/"); str_found = add_quote_for_spec_chars(found); } else { /* not put found file if search only dirs for cd */ if (type == FIND_DIR_ONLY) continue; str_found = add_quote_for_spec_chars(found); if (type == FIND_FILE_ONLY || (type == FIND_EXE_ONLY && is_execute(&st) == TRUE)) strcat(str_found, " "); } /* Add it to the list */ matches = xrealloc(matches, (nm + 1) * sizeof(char *)); matches[nm++] = str_found; } closedir(dir); } if (paths != path1) { free(paths[0]); /* allocated memory only in first member */ free(paths); } *num_matches = nm; return (matches); } static int match_compare(const void *a, const void *b) { return strcmp(*(char **) a, *(char **) b); } #define QUOT (UCHAR_MAX+1) #define collapse_pos(is, in) { \ memcpy(int_buf+is, int_buf+in, (BUFSIZ+1-is-in)*sizeof(int)); \ memcpy(pos_buf+is, pos_buf+in, (BUFSIZ+1-is-in)*sizeof(int)); } static int find_match(char *matchBuf, int *len_with_quotes) { int i, j; int command_mode; int c, c2; int int_buf[BUFSIZ + 1]; int pos_buf[BUFSIZ + 1]; /* set to integer dimension characters and own positions */ for (i = 0;; i++) { int_buf[i] = (int) ((unsigned char) matchBuf[i]); if (int_buf[i] == 0) { pos_buf[i] = -1; /* indicator end line */ break; } else pos_buf[i] = i; } /* mask \+symbol and convert '\t' to ' ' */ for (i = j = 0; matchBuf[i]; i++, j++) if (matchBuf[i] == '\\') { collapse_pos(j, j + 1); int_buf[j] |= QUOT; i++; #ifdef BB_FEATURE_NONPRINTABLE_INVERSE_PUT if (matchBuf[i] == '\t') /* algorithm equivalent */ int_buf[j] = ' ' | QUOT; #endif } #ifdef BB_FEATURE_NONPRINTABLE_INVERSE_PUT else if (matchBuf[i] == '\t') int_buf[j] = ' '; #endif /* mask "symbols" or 'symbols' */ c2 = 0; for (i = 0; int_buf[i]; i++) { c = int_buf[i]; if (c == '\'' || c == '"') { if (c2 == 0) c2 = c; else { if (c == c2) c2 = 0; else int_buf[i] |= QUOT; } } else if (c2 != 0 && c != '$') int_buf[i] |= QUOT; } /* skip commands with arguments if line have commands delimiters */ /* ';' ';;' '&' '|' '&&' '||' but `>&' `<&' `>|' */ for (i = 0; int_buf[i]; i++) { c = int_buf[i]; c2 = int_buf[i + 1]; j = i ? int_buf[i - 1] : -1; command_mode = 0; if (c == ';' || c == '&' || c == '|') { command_mode = 1 + (c == c2); if (c == '&') { if (j == '>' || j == '<') command_mode = 0; } else if (c == '|' && j == '>') command_mode = 0; } if (command_mode) { collapse_pos(0, i + command_mode); i = -1; /* hack incremet */ } } /* collapse `command...` */ for (i = 0; int_buf[i]; i++) if (int_buf[i] == '`') { for (j = i + 1; int_buf[j]; j++) if (int_buf[j] == '`') { collapse_pos(i, j + 1); j = 0; break; } if (j) { /* not found close ` - command mode, collapse all previous */ collapse_pos(0, i + 1); break; } else i--; /* hack incremet */ } /* collapse (command...(command...)...) or {command...{command...}...} */ c = 0; /* "recursive" level */ c2 = 0; for (i = 0; int_buf[i]; i++) if (int_buf[i] == '(' || int_buf[i] == '{') { if (int_buf[i] == '(') c++; else c2++; collapse_pos(0, i + 1); i = -1; /* hack incremet */ } for (i = 0; pos_buf[i] >= 0 && (c > 0 || c2 > 0); i++) if ((int_buf[i] == ')' && c > 0) || (int_buf[i] == '}' && c2 > 0)) { if (int_buf[i] == ')') c--; else c2--; collapse_pos(0, i + 1); i = -1; /* hack incremet */ } /* skip first not quote space */ for (i = 0; int_buf[i]; i++) if (int_buf[i] != ' ') break; if (i) collapse_pos(0, i); /* set find mode for completion */ command_mode = FIND_EXE_ONLY; for (i = 0; int_buf[i]; i++) if (int_buf[i] == ' ' || int_buf[i] == '<' || int_buf[i] == '>') { if (int_buf[i] == ' ' && command_mode == FIND_EXE_ONLY && strncmp(&matchBuf[pos_buf[0]], "cd", 2) == 0) command_mode = FIND_DIR_ONLY; else { command_mode = FIND_FILE_ONLY; break; } } /* "strlen" */ for (i = 0; int_buf[i]; i++); /* find last word */ for (--i; i >= 0; i--) { c = int_buf[i]; if (c == ' ' || c == '<' || c == '>' || c == '|' || c == '&') { collapse_pos(0, i + 1); break; } } /* skip first not quoted '\'' or '"' */ for (i = 0; int_buf[i] == '\'' || int_buf[i] == '"'; i++); /* collapse quote or unquote // or /~ */ while ((int_buf[i] & ~QUOT) == '/' && ((int_buf[i + 1] & ~QUOT) == '/' || (int_buf[i + 1] & ~QUOT) == '~')) { i++; } if (i) { collapse_pos(0, i); } /* set only match and destroy quotes */ j = 0; for (i = 0; pos_buf[i] >= 0; i++) { matchBuf[i] = matchBuf[pos_buf[i]]; j = pos_buf[i] + 1; } matchBuf[i] = 0; /* old lenght matchBuf with quotes symbols */ *len_with_quotes = j ? j - pos_buf[0] : 0; return command_mode; } static void input_tab(int *lastWasTab) { /* Do TAB completion */ static int num_matches; static char **matches; if (lastWasTab == 0) { /* free all memory */ if (matches) { while (num_matches > 0) free(matches[--num_matches]); free(matches); matches = (char **) NULL; } return; } if (*lastWasTab == FALSE) { char *tmp; int len_found; char matchBuf[BUFSIZ]; int find_type; int recalc_pos; *lastWasTab = TRUE; /* flop trigger */ /* Make a local copy of the string -- up * to the position of the cursor */ tmp = strncpy(matchBuf, command_ps, cursor); tmp[cursor] = 0; find_type = find_match(matchBuf, &recalc_pos); /* Free up any memory already allocated */ input_tab(0); #ifdef BB_FEATURE_SH_USERNAME_COMPLETION /* If the word starts with `~' and there is no slash in the word, * then try completing this word as a username. */ if (matchBuf[0] == '~' && strchr(matchBuf, '/') == 0) matches = username_tab_completion(matchBuf, &num_matches); #endif /* Try to match any executable in our path and everything * in the current working directory that matches. */ if (!matches) matches = exe_n_cwd_tab_completion(matchBuf, &num_matches, find_type); /* Did we find exactly one match? */ if (!matches || num_matches > 1) { char *tmp1; beep(); if (!matches) return; /* not found */ /* sort */ qsort(matches, num_matches, sizeof(char *), match_compare); /* find minimal match */ tmp = xstrdup(matches[0]); for (tmp1 = tmp; *tmp1; tmp1++) for (len_found = 1; len_found < num_matches; len_found++) if (matches[len_found][(tmp1 - tmp)] != *tmp1) { *tmp1 = 0; break; } if (*tmp == 0) { /* have unique */ free(tmp); return; } } else { /* one match */ tmp = matches[0]; /* for next completion current found */ *lastWasTab = FALSE; } len_found = strlen(tmp); /* have space to placed match? */ if ((len_found - strlen(matchBuf) + len) < BUFSIZ) { /* before word for match */ command_ps[cursor - recalc_pos] = 0; /* save tail line */ strcpy(matchBuf, command_ps + cursor); /* add match */ strcat(command_ps, tmp); /* add tail */ strcat(command_ps, matchBuf); /* back to begin word for match */ input_backward(recalc_pos); /* new pos */ recalc_pos = cursor + len_found; /* new len */ len = strlen(command_ps); /* write out the matched command */ input_end(); input_backward(cursor - recalc_pos); } if (tmp != matches[0]) free(tmp); } else { /* Ok -- the last char was a TAB. Since they * just hit TAB again, print a list of all the * available choices... */ if (matches && num_matches > 0) { int i, col, l; int sav_cursor = cursor; /* change goto_new_line() */ /* Go to the next line */ goto_new_line(); for (i = 0, col = 0; i < num_matches; i++) { l = strlen(matches[i]); if (l < 14) l = 14; printf("%-14s ", matches[i]); if ((l += 2) > 16) while (l % 16) { putchar(' '); l++; } col += l; col -= (col / cmdedit_termw) * cmdedit_termw; if (col > 60 && matches[i + 1] != NULL) { putchar('\n'); col = 0; } } /* Go to the next line and rewrite */ putchar('\n'); redraw(0, len - sav_cursor); } } } #endif /* BB_FEATURE_SH_TAB_COMPLETION */ static void get_previous_history(struct history **hp, struct history *p) { if ((*hp)->s) free((*hp)->s); (*hp)->s = xstrdup(command_ps); *hp = p; } static inline void get_next_history(struct history **hp) { get_previous_history(hp, (*hp)->n); } enum { ESC = 27, DEL = 127, }; /* * This function is used to grab a character buffer * from the input file descriptor and allows you to * a string with full command editing (sortof like * a mini readline). * * The following standard commands are not implemented: * ESC-b -- Move back one word * ESC-f -- Move forward one word * ESC-d -- Delete back one word * ESC-h -- Delete forward one word * CTL-t -- Transpose two characters * * Furthermore, the "vi" command editing keys are not implemented. * */ extern void cmdedit_read_input(char *prompt, char command[BUFSIZ]) { int inputFd = fileno(stdin); int break_out = 0; int lastWasTab = FALSE; char c = 0; struct history *hp = his_end; /* prepare before init handlers */ cmdedit_y = 0; /* quasireal y, not true work if line > xt*yt */ len = 0; command_ps = command; if (new_settings.c_cc[VMIN] == 0) { /* first call */ getTermSettings(inputFd, (void *) &initial_settings); memcpy(&new_settings, &initial_settings, sizeof(struct termios)); new_settings.c_cc[VMIN] = 1; new_settings.c_cc[VTIME] = 0; new_settings.c_cc[VINTR] = _POSIX_VDISABLE; /* Turn off CTRL-C, so we can trap it */ new_settings.c_lflag &= ~ICANON; /* unbuffered input */ new_settings.c_lflag &= ~(ECHO | ECHOCTL | ECHONL); /* Turn off echoing */ } command[0] = 0; setTermSettings(inputFd, (void *) &new_settings); handlers_sets |= SET_RESET_TERM; /* Print out the command prompt */ parse_prompt(prompt); /* Now initialize things */ cmdedit_init(); while (1) { fflush(stdout); /* buffered out to fast */ if (read(inputFd, &c, 1) < 1) return; switch (c) { case '\n': case '\r': /* Enter */ goto_new_line(); break_out = 1; break; case 1: /* Control-a -- Beginning of line */ input_backward(cursor); break; case 2: /* Control-b -- Move back one character */ input_backward(1); break; case 3: /* Control-c -- stop gathering input */ /* Link into lash to reset context to 0 on ^C and such */ shell_context = 0; /* Go to the next line */ goto_new_line(); command[0] = 0; return; case 4: /* Control-d -- Delete one character, or exit * if the len=0 and no chars to delete */ if (len == 0) { printf("exit"); clean_up_and_die(0); } else { input_delete(); } break; case 5: /* Control-e -- End of line */ input_end(); break; case 6: /* Control-f -- Move forward one character */ input_forward(); break; case '\b': case DEL: /* Control-h and DEL */ input_backspace(); break; case '\t': #ifdef BB_FEATURE_SH_TAB_COMPLETION input_tab(&lastWasTab); #endif break; case 14: /* Control-n -- Get next command in history */ if (hp && hp->n && hp->n->s) { get_next_history(&hp); goto rewrite_line; } else { beep(); } break; case 16: /* Control-p -- Get previous command from history */ if (hp && hp->p) { get_previous_history(&hp, hp->p); goto rewrite_line; } else { beep(); } break; case 21: /* Control-U -- Clear line before cursor */ if (cursor) { strcpy(command, command + cursor); redraw(cmdedit_y, len -= cursor); } break; case ESC:{ /* escape sequence follows */ if (read(inputFd, &c, 1) < 1) return; /* different vt100 emulations */ if (c == '[' || c == 'O') { if (read(inputFd, &c, 1) < 1) return; } switch (c) { #ifdef BB_FEATURE_SH_TAB_COMPLETION case '\t': /* Alt-Tab */ input_tab(&lastWasTab); break; #endif case 'A': /* Up Arrow -- Get previous command from history */ if (hp && hp->p) { get_previous_history(&hp, hp->p); goto rewrite_line; } else { beep(); } break; case 'B': /* Down Arrow -- Get next command in history */ if (hp && hp->n && hp->n->s) { get_next_history(&hp); goto rewrite_line; } else { beep(); } break; /* Rewrite the line with the selected history item */ rewrite_line: /* change command */ len = strlen(strcpy(command, hp->s)); /* redraw and go to end line */ redraw(cmdedit_y, 0); break; case 'C': /* Right Arrow -- Move forward one character */ input_forward(); break; case 'D': /* Left Arrow -- Move back one character */ input_backward(1); break; case '3': /* Delete */ input_delete(); break; case '1': case 'H': /* Home (Ctrl-A) */ input_backward(cursor); break; case '4': case 'F': /* End (Ctrl-E) */ input_end(); break; default: if (!(c >= '1' && c <= '9')) c = 0; beep(); } if (c >= '1' && c <= '9') do if (read(inputFd, &c, 1) < 1) return; while (c != '~'); break; } default: /* If it's regular input, do the normal thing */ #ifdef BB_FEATURE_NONPRINTABLE_INVERSE_PUT /* Control-V -- Add non-printable symbol */ if (c == 22) { if (read(inputFd, &c, 1) < 1) return; if (c == 0) { beep(); break; } } else #endif if (!isprint(c)) /* Skip non-printable characters */ break; if (len >= (BUFSIZ - 2)) /* Need to leave space for enter */ break; len++; if (cursor == (len - 1)) { /* Append if at the end of the line */ *(command + cursor) = c; *(command + cursor + 1) = 0; cmdedit_set_out_char(0); } else { /* Insert otherwise */ int sc = cursor; memmove(command + sc + 1, command + sc, len - sc); *(command + sc) = c; sc++; /* rewrite from cursor */ input_end(); /* to prev x pos + 1 */ input_backward(cursor - sc); } break; } if (break_out) /* Enter is the command terminator, no more input. */ break; if (c != '\t') lastWasTab = FALSE; } setTermSettings(inputFd, (void *) &initial_settings); handlers_sets &= ~SET_RESET_TERM; /* Handle command history log */ if (len) { /* no put empty line */ struct history *h = his_end; char *ss; ss = xstrdup(command); /* duplicate */ if (h == 0) { /* No previous history -- this memory is never freed */ h = his_front = xmalloc(sizeof(struct history)); h->n = xmalloc(sizeof(struct history)); h->p = NULL; h->s = ss; h->n->p = h; h->n->n = NULL; h->n->s = NULL; his_end = h->n; history_counter++; } else { /* Add a new history command -- this memory is never freed */ h->n = xmalloc(sizeof(struct history)); h->n->p = h; h->n->n = NULL; h->n->s = NULL; h->s = ss; his_end = h->n; /* After max history, remove the oldest command */ if (history_counter >= MAX_HISTORY) { struct history *p = his_front->n; p->p = NULL; free(his_front->s); free(his_front); his_front = p; } else { history_counter++; } } #if defined(BB_FEATURE_BASH_STYLE_PROMT) num_ok_lines++; #endif } command[len++] = '\n'; /* set '\n' */ command[len] = 0; #if defined(BB_FEATURE_CLEAN_UP) && defined(BB_FEATURE_SH_TAB_COMPLETION) input_tab(0); /* strong free */ #endif #if defined(BB_FEATURE_BASH_STYLE_PROMT) free(cmdedit_prompt); #endif return; } /* Undo the effects of cmdedit_init(). */ extern void cmdedit_terminate(void) { cmdedit_reset_term(); if ((handlers_sets & SET_TERM_HANDLERS) != 0) { signal(SIGKILL, SIG_DFL); signal(SIGINT, SIG_DFL); signal(SIGQUIT, SIG_DFL); signal(SIGTERM, SIG_DFL); signal(SIGWINCH, SIG_DFL); handlers_sets &= ~SET_TERM_HANDLERS; } } #endif /* BB_FEATURE_SH_COMMAND_EDITING */ #ifdef TEST unsigned int shell_context; int main(int argc, char **argv) { char buff[BUFSIZ]; char *prompt = #if defined(BB_FEATURE_BASH_STYLE_PROMT) "\\[\\033[32;1m\\]\\u@\\[\\033[33;1m\\]\\h:\ \\[\\033[34;1m\\]\\w\\[\\033[35;1m\\] \ \\!\\[\\033[36;1m\\]\\$ \\[\\033[0m\\]"; #else "% "; #endif shell_context = 1; do { cmdedit_read_input(prompt, buff); printf("*** cmdedit_read_input() returned line =%s=\n", buff); } while (shell_context); printf("*** cmdedit_read_input() detect ^C\n"); return 0; } #endif /* TEST */