/* vi: set sw=4 ts=4: */ /* * Termios command line History and Editing. * * Copyright (c) 1986-2003 may safely be consumed by a BSD or GPL license. * Written by: Vladimir Oleynik * * Used ideas: * Adam Rogoyski * Dave Cinege * Jakub Jelinek (c) 1995 * Erik Andersen (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) - not true viewing if length prompt less terminal width */ #include "busybox.h" #include #include #include #include #include #include #include #include #include #include "../shell/cmdedit.h" #ifdef CONFIG_LOCALE_SUPPORT #define Isprint(c) isprint((c)) #else #define Isprint(c) ( (c) >= ' ' && (c) != ((unsigned char)'\233') ) #endif #ifdef TEST /* pretect redefined for test */ #undef CONFIG_FEATURE_COMMAND_EDITING #undef CONFIG_FEATURE_COMMAND_TAB_COMPLETION #undef CONFIG_FEATURE_COMMAND_USERNAME_COMPLETION #undef CONFIG_FEATURE_NONPRINTABLE_INVERSE_PUT #undef CONFIG_FEATURE_CLEAN_UP #define CONFIG_FEATURE_COMMAND_EDITING #define CONFIG_FEATURE_COMMAND_TAB_COMPLETION #define CONFIG_FEATURE_COMMAND_USERNAME_COMPLETION #define CONFIG_FEATURE_NONPRINTABLE_INVERSE_PUT #define CONFIG_FEATURE_CLEAN_UP #endif /* TEST */ #ifdef CONFIG_FEATURE_COMMAND_TAB_COMPLETION #include #include #endif #ifdef CONFIG_FEATURE_COMMAND_EDITING #if defined(CONFIG_FEATURE_COMMAND_USERNAME_COMPLETION) || defined(CONFIG_FEATURE_SH_FANCY_PROMPT) #define CONFIG_FEATURE_GETUSERNAME_AND_HOMEDIR #endif #ifdef CONFIG_FEATURE_GETUSERNAME_AND_HOMEDIR #include "pwd_.h" #endif /* advanced FEATURES */ /* Maximum length of the linked list for the command line history */ #ifndef CONFIG_FEATURE_COMMAND_HISTORY #define MAX_HISTORY 15 #else #define MAX_HISTORY CONFIG_FEATURE_COMMAND_HISTORY #endif #if MAX_HISTORY < 1 #warning cmdedit: You set MAX_HISTORY < 1. The history algorithm switched off. #else static char *history[MAX_HISTORY+1]; /* history + current */ /* saved history lines */ static int n_history; /* current pointer to history line */ static int cur_history; #endif #include #define setTermSettings(fd,argp) tcsetattr(fd,TCSANOW,argp) #define getTermSettings(fd,argp) tcgetattr(fd, argp); /* Current termio and the previous termio before starting sh */ static struct termios initial_settings, new_settings; static volatile int cmdedit_termw = 80; /* actual terminal width */ 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_WCHG_HANDLERS = 2, /* winchg signal handler */ SET_RESET_TERM = 4, /* 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 CONFIG_FEATURE_SH_FANCY_PROMPT const #endif char *cmdedit_prompt; /* --- "" - - "" - -"- --""-- --""--- */ #ifdef CONFIG_FEATURE_GETUSERNAME_AND_HOMEDIR static char *user_buf = ""; static char *home_pwd_buf = ""; static int my_euid; #endif #ifdef CONFIG_FEATURE_SH_FANCY_PROMPT static char *hostname_buf; static int num_ok_lines = 1; #endif #ifdef CONFIG_FEATURE_COMMAND_TAB_COMPLETION #ifndef CONFIG_FEATURE_GETUSERNAME_AND_HOMEDIR static int my_euid; #endif static int my_uid; static int my_gid; #endif /* CONFIG_FEATURE_COMMAND_TAB_COMPLETION */ static void cmdedit_setwidth(int w, int redraw_flg); static void win_changed(int nsig) { static sighandler_t previous_SIGWINCH_handler; /* for reset */ /* emulate || signal call */ if (nsig == -SIGWINCH || nsig == SIGWINCH) { int width = 0; get_terminal_width_height(0, &width, NULL); cmdedit_setwidth(width, 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(STDIN_FILENO, (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); } /* special for recount position for scroll and remove terminal margin effect */ static void cmdedit_set_out_char(int next_char) { int c = (int)((unsigned char) command_ps[cursor]); if (c == 0) c = ' '; /* destroy end char? */ #ifdef CONFIG_FEATURE_NONPRINTABLE_INVERSE_PUT if (!Isprint(c)) { /* Inverse put non-printable characters */ if (c >= 128) c -= 128; if (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) { if ( s ) fputs(s, stdout); } static inline void beep(void) { putchar('\007'); } /* Move back one character */ /* 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; cmdedit_y = 0; /* new quasireal y */ } #ifndef CONFIG_FEATURE_SH_FANCY_PROMPT static void parse_prompt(const char *prmt_ptr) { cmdedit_prompt = prmt_ptr; cmdedit_prmt_len = strlen(prmt_ptr); put_prompt(); } #else static void parse_prompt(const char *prmt_ptr) { int prmt_len = 0; size_t cur_prmt_len = 0; char flg_not_length = '['; char *prmt_mem_ptr = xcalloc(1, 1); char *pwd_buf = xgetcwd(0); char buf2[PATH_MAX + 1]; char buf[2]; char c; char *pbuf; if (!pwd_buf) { pwd_buf=(char *)bb_msg_unknown; } while (*prmt_ptr) { pbuf = buf; pbuf[1] = 0; c = *prmt_ptr++; if (c == '\\') { const char *cp = prmt_ptr; int l; c = bb_process_escape_sequence(&prmt_ptr); if(prmt_ptr==cp) { if (*cp == 0) break; c = *prmt_ptr++; switch (c) { #ifdef CONFIG_FEATURE_GETUSERNAME_AND_HOMEDIR case 'u': pbuf = user_buf; break; #endif case 'h': pbuf = hostname_buf; if (pbuf == 0) { pbuf = xcalloc(256, 1); if (gethostname(pbuf, 255) < 0) { strcpy(pbuf, "?"); } else { char *s = strchr(pbuf, '.'); if (s) *s = 0; } hostname_buf = pbuf; } break; case '$': c = my_euid == 0 ? '#' : '$'; break; #ifdef CONFIG_FEATURE_GETUSERNAME_AND_HOMEDIR case 'w': pbuf = pwd_buf; l = strlen(home_pwd_buf); if (home_pwd_buf[0] != 0 && strncmp(home_pwd_buf, pbuf, l) == 0 && (pbuf[l]=='/' || pbuf[l]=='\0') && strlen(pwd_buf+l) UCHAR_MAX || (pbuf - buf2) < l) { l--; break; } prmt_ptr++; } buf2[l] = 0; c = (char)strtol(buf2, 0, 16); if(c==0) c = '?'; pbuf = buf; break; case '[': case ']': if (c == flg_not_length) { flg_not_length = flg_not_length == '[' ? ']' : '['; continue; } break; } } } if(pbuf == buf) *pbuf = c; cur_prmt_len = strlen(pbuf); prmt_len += cur_prmt_len; if (flg_not_length != ']') cmdedit_prmt_len += cur_prmt_len; prmt_mem_ptr = strcat(xrealloc(prmt_mem_ptr, prmt_len+1), pbuf); } if(pwd_buf!=(char *)bb_msg_unknown) free(pwd_buf); cmdedit_prompt = prmt_mem_ptr; put_prompt(); } #endif /* draw prompt, editor line, and clear tail */ static void redraw(int y, int back_cursor) { if (y > 0) /* up to start y */ printf("\033[%dA", y); putchar('\r'); put_prompt(); input_end(); /* rewrite */ printf("\033[J"); /* destroy tail after cursor */ input_backward(back_cursor); } #ifdef CONFIG_FEATURE_COMMAND_EDITING_VI #define DELBUFSIZ 128 static char *delbuf; /* a (malloced) place to store deleted characters */ static char *delp; static char newdelflag; /* whether delbuf should be reused yet */ #endif /* Delete the char in front of the cursor, optionally saving it * for later putback */ static void input_delete(int save) { int j = cursor; if (j == len) return; #ifdef CONFIG_FEATURE_COMMAND_EDITING_VI if (save) { if (newdelflag) { if (!delbuf) delbuf = malloc(DELBUFSIZ); /* safe if malloc fails */ delp = delbuf; newdelflag = 0; } if (delbuf && (delp - delbuf < DELBUFSIZ)) *delp++ = command_ps[j]; } #endif strcpy(command_ps + j, command_ps + j + 1); len--; input_end(); /* rewrite new line */ cmdedit_set_out_char(0); /* destroy end char */ input_backward(cursor - j); /* back to old pos cursor */ } #ifdef CONFIG_FEATURE_COMMAND_EDITING_VI static void put(void) { int ocursor, j = delp - delbuf; if (j == 0) return; ocursor = cursor; /* open hole and then fill it */ memmove(command_ps + cursor + j, command_ps + cursor, len - cursor + 1); strncpy(command_ps + cursor, delbuf, j); len += j; input_end(); /* rewrite new line */ input_backward(cursor-ocursor-j+1); /* at end of new text */ } #endif /* Delete the char in back of the cursor */ static void input_backspace(void) { if (cursor > 0) { input_backward(1); input_delete(0); } } /* Move forward one character */ static void input_forward(void) { if (cursor < len) cmdedit_set_out_char(command_ps[cursor + 1]); } 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); } } } static void cmdedit_init(void) { cmdedit_prmt_len = 0; 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 CONFIG_FEATURE_GETUSERNAME_AND_HOMEDIR struct passwd *entry; my_euid = geteuid(); entry = getpwuid(my_euid); if (entry) { user_buf = bb_xstrdup(entry->pw_name); home_pwd_buf = bb_xstrdup(entry->pw_dir); } #endif #ifdef CONFIG_FEATURE_COMMAND_TAB_COMPLETION #ifndef CONFIG_FEATURE_GETUSERNAME_AND_HOMEDIR my_euid = geteuid(); #endif my_uid = getuid(); my_gid = getgid(); #endif /* CONFIG_FEATURE_COMMAND_TAB_COMPLETION */ handlers_sets |= SET_ATEXIT; atexit(cmdedit_reset_term); /* be sure to do this only once */ } } #ifdef CONFIG_FEATURE_COMMAND_TAB_COMPLETION static char **matches; static int num_matches; static char *add_char_to_match; static void add_match(char *matched, int add_char) { int nm = num_matches; int nm1 = nm + 1; matches = xrealloc(matches, nm1 * sizeof(char *)); add_char_to_match = xrealloc(add_char_to_match, nm1); matches[nm] = matched; add_char_to_match[nm] = (char)add_char; num_matches++; } 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 CONFIG_FEATURE_COMMAND_USERNAME_COMPLETION static void username_tab_completion(char *ud, char *with_shash_flg) { struct passwd *entry; int userlen; ud++; /* ~user/... to user/... */ userlen = strlen(ud); if (with_shash_flg) { /* "~/..." or "~user/..." */ char *sav_ud = ud - 1; char *home = 0; char *temp; 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); } } } else { /* "~[^/]*" */ setpwent(); while ((entry = getpwent()) != NULL) { /* Null usernames should result in all users as possible completions. */ if ( /*!userlen || */ !strncmp(ud, entry->pw_name, userlen)) { add_match(bb_xasprintf("~%s", entry->pw_name), '/'); } } endpwent(); } } #endif /* CONFIG_FEATURE_COMMAND_USERNAME_COMPLETION */ enum { FIND_EXE_ONLY = 0, FIND_DIR_ONLY = 1, FIND_FILE_ONLY = 2, }; #ifdef CONFIG_ASH const char *cmdedit_path_lookup; #else #define cmdedit_path_lookup getenv("PATH") #endif static int path_parse(char ***p, int flags) { int npth; const char *tmp; const char *pth; /* if not setenv PATH variable, to search cur dir "." */ if (flags != FIND_EXE_ONLY || (pth = cmdedit_path_lookup) == 0 || /* PATH= or PATH=: */ *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; /* : */ } else break; } *p = xmalloc(npth * sizeof(char *)); tmp = pth; (*p)[0] = bb_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; /* : */ } 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 add) { int l = 0; char *s = xmalloc((strlen(found) + 1) * 2); while (*found) { if (strchr(" `\"#$%^&*()=+{}[]:;\'|\\<>", *found)) s[l++] = '\\'; s[l++] = *found++; } if(add) s[l++] = (char)add; s[l] = 0; return s; } static void exe_n_cwd_tab_completion(char *command, int type) { DIR *dir; struct dirent *next; char dirbuf[BUFSIZ]; struct stat st; char *path1[1]; char **paths = path1; int npaths; int i; char *found; 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 CONFIG_FEATURE_COMMAND_USERNAME_COMPLETION if (dirbuf[0] == '~') /* ~/... or ~user/... */ username_tab_completion(dirbuf, dirbuf); #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) { char *str_found = next->d_name; int add_chr = 0; /* 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; } found = concat_path_file(paths[i], str_found); /* hmm, remover in progress? */ if (stat(found, &st) < 0) goto cont; /* find with dirs ? */ if (paths[i] != dirbuf) strcpy(found, next->d_name); /* only name */ if (S_ISDIR(st.st_mode)) { /* name is directory */ char *e = found + strlen(found) - 1; add_chr = '/'; if(*e == '/') *e = '\0'; } else { /* not put found file if search only dirs for cd */ if (type == FIND_DIR_ONLY) goto cont; if (type == FIND_FILE_ONLY || (type == FIND_EXE_ONLY && is_execute(&st))) add_chr = ' '; } /* Add it to the list */ add_match(found, add_chr); continue; cont: free(found); } closedir(dir); } if (paths != path1) { free(paths[0]); /* allocated memory only in first member */ free(paths); } } #define QUOT (UCHAR_MAX+1) #define collapse_pos(is, in) { \ memmove(int_buf+(is), int_buf+(in), (BUFSIZ+1-(is)-(in))*sizeof(int)); \ memmove(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 CONFIG_FEATURE_NONPRINTABLE_INVERSE_PUT if (matchBuf[i] == '\t') /* algorithm equivalent */ int_buf[j] = ' ' | QUOT; #endif } #ifdef CONFIG_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 && matchBuf[pos_buf[0]]=='c' && matchBuf[pos_buf[1]]=='d' ) 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++; } /* set only match and destroy quotes */ j = 0; for (c = 0; pos_buf[i] >= 0; i++) { matchBuf[c++] = matchBuf[pos_buf[i]]; j = pos_buf[i] + 1; } matchBuf[c] = 0; /* old lenght matchBuf with quotes symbols */ *len_with_quotes = j ? j - pos_buf[0] : 0; return command_mode; } /* display by column original ideas from ls applet, very optimize by my :) */ static void showfiles(void) { int ncols, row; int column_width = 0; int nfiles = num_matches; int nrows = nfiles; char str_add_chr[2]; int l; /* find the longest file name- use that as the column width */ for (row = 0; row < nrows; row++) { l = strlen(matches[row]); if(add_char_to_match[row]) l++; if (column_width < l) column_width = l; } column_width += 2; /* min space for columns */ ncols = cmdedit_termw / column_width; if (ncols > 1) { nrows /= ncols; if(nfiles % ncols) nrows++; /* round up fractionals */ } else { ncols = 1; } str_add_chr[1] = 0; for (row = 0; row < nrows; row++) { int n = row; int nc; int acol; for(nc = 1; nc < ncols && n+nrows < nfiles; n += nrows, nc++) { str_add_chr[0] = add_char_to_match[n]; acol = str_add_chr[0] ? column_width - 1 : column_width; printf("%s%s", matches[n], str_add_chr); l = strlen(matches[n]); while(l < acol) { putchar(' '); l++; } } str_add_chr[0] = add_char_to_match[n]; printf("%s%s\n", matches[n], str_add_chr); } } static void input_tab(int *lastWasTab) { /* Do TAB completion */ if (lastWasTab == 0) { /* free all memory */ if (matches) { while (num_matches > 0) free(matches[--num_matches]); free(matches); matches = (char **) NULL; free(add_char_to_match); add_char_to_match = NULL; } return; } if (! *lastWasTab) { char *tmp, *tmp1; 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 CONFIG_FEATURE_COMMAND_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) username_tab_completion(matchBuf, NULL); if (!matches) #endif /* Try to match any executable in our path and everything * in the current working directory that matches. */ exe_n_cwd_tab_completion(matchBuf, find_type); /* Remove duplicate found and sort */ if(matches) { int i, j, n, srt; /* bubble */ n = num_matches; for(i=0; i<(n-1); i++) for(j=i+1; j 0) { tmp1 = matches[i]; matches[i] = matches[j]; matches[j] = tmp1; srt = add_char_to_match[i]; add_char_to_match[i] = add_char_to_match[j]; add_char_to_match[j] = srt; } } j = n; n = 0; for(i=0; i 1) { beep(); if (!matches) return; /* not found */ /* find minimal match */ tmp1 = bb_xstrdup(matches[0]); for (tmp = tmp1; *tmp; tmp++) for (len_found = 1; len_found < num_matches; len_found++) if (matches[len_found][(tmp - tmp1)] != *tmp) { *tmp = 0; break; } if (*tmp1 == 0) { /* have unique */ free(tmp1); return; } tmp = add_quote_for_spec_chars(tmp1, 0); free(tmp1); } else { /* one match */ tmp = add_quote_for_spec_chars(matches[0], add_char_to_match[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 */ redraw(cmdedit_y, len - recalc_pos); } 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 sav_cursor = cursor; /* change goto_new_line() */ /* Go to the next line */ goto_new_line(); showfiles(); redraw(0, len - sav_cursor); } } } #endif /* CONFIG_FEATURE_COMMAND_TAB_COMPLETION */ #if MAX_HISTORY >= 1 static void get_previous_history(void) { if(command_ps[0] != 0 || history[cur_history] == 0) { free(history[cur_history]); history[cur_history] = bb_xstrdup(command_ps); } cur_history--; } static int get_next_history(void) { int ch = cur_history; if (ch < n_history) { get_previous_history(); /* save the current history line */ return (cur_history = ch+1); } else { beep(); return 0; } } #ifdef CONFIG_FEATURE_COMMAND_SAVEHISTORY void load_history ( const char *fromfile ) { FILE *fp; int hi; /* cleanup old */ for(hi = n_history; hi > 0; ) { hi--; free ( history [hi] ); } if (( fp = fopen ( fromfile, "r" ))) { for ( hi = 0; hi < MAX_HISTORY; ) { char * hl = bb_get_chomped_line_from_file(fp); int l; if(!hl) break; l = strlen(hl); if(l >= BUFSIZ) hl[BUFSIZ-1] = 0; if(l == 0 || hl[0] == ' ') { free(hl); continue; } history [hi++] = hl; } fclose ( fp ); } cur_history = n_history = hi; } void save_history ( const char *tofile ) { FILE *fp = fopen ( tofile, "w" ); if ( fp ) { int i; for ( i = 0; i < n_history; i++ ) { fprintf(fp, "%s\n", history [i]); } fclose ( fp ); } } #endif #endif 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 (sort of 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 * * Minimalist vi-style command line editing available if configured. * vi mode implemented 2005 by Paul Fox * */ #ifdef CONFIG_FEATURE_COMMAND_EDITING_VI static int vi_mode; void setvimode ( int viflag ) { vi_mode = viflag; } static void vi_Word_motion(char *command, int eat) { while (cursor < len && !isspace(command[cursor])) input_forward(); if (eat) while (cursor < len && isspace(command[cursor])) input_forward(); } static void vi_word_motion(char *command, int eat) { if (isalnum(command[cursor]) || command[cursor] == '_') { while (cursor < len && (isalnum(command[cursor+1]) || command[cursor+1] == '_')) input_forward(); } else if (ispunct(command[cursor])) { while (cursor < len && (ispunct(command[cursor+1]))) input_forward(); } if (cursor < len) input_forward(); if (eat && cursor < len && isspace(command[cursor])) while (cursor < len && isspace(command[cursor])) input_forward(); } static void vi_End_motion(char *command) { input_forward(); while (cursor < len && isspace(command[cursor])) input_forward(); while (cursor < len-1 && !isspace(command[cursor+1])) input_forward(); } static void vi_end_motion(char *command) { if (cursor >= len-1) return; input_forward(); while (cursor < len-1 && isspace(command[cursor])) input_forward(); if (cursor >= len-1) return; if (isalnum(command[cursor]) || command[cursor] == '_') { while (cursor < len-1 && (isalnum(command[cursor+1]) || command[cursor+1] == '_')) input_forward(); } else if (ispunct(command[cursor])) { while (cursor < len-1 && (ispunct(command[cursor+1]))) input_forward(); } } static void vi_Back_motion(char *command) { while (cursor > 0 && isspace(command[cursor-1])) input_backward(1); while (cursor > 0 && !isspace(command[cursor-1])) input_backward(1); } static void vi_back_motion(char *command) { if (cursor <= 0) return; input_backward(1); while (cursor > 0 && isspace(command[cursor])) input_backward(1); if (cursor <= 0) return; if (isalnum(command[cursor]) || command[cursor] == '_') { while (cursor > 0 && (isalnum(command[cursor-1]) || command[cursor-1] == '_')) input_backward(1); } else if (ispunct(command[cursor])) { while (cursor > 0 && (ispunct(command[cursor-1]))) input_backward(1); } } #endif /* * the emacs and vi modes share much of the code in the big * command loop. commands entered when in vi's command mode (aka * "escape mode") get an extra bit added to distinguish them -- * this keeps them from being self-inserted. this clutters the * big switch a bit, but keeps all the code in one place. */ #define vbit 0x100 /* leave out the "vi-mode"-only case labels if vi editing isn't * configured. */ #define vi_case(caselabel) USE_FEATURE_COMMAND_EDITING(caselabel) /* convert uppercase ascii to equivalent control char, for readability */ #define CNTRL(uc_char) ((uc_char) - 0x40) int cmdedit_read_input(char *prompt, char command[BUFSIZ]) { int break_out = 0; int lastWasTab = FALSE; unsigned char c; unsigned int ic; #ifdef CONFIG_FEATURE_COMMAND_EDITING_VI unsigned int prevc; int vi_cmdmode = 0; #endif /* prepare before init handlers */ cmdedit_y = 0; /* quasireal y, not true work if line > xt*yt */ len = 0; command_ps = command; getTermSettings(0, (void *) &initial_settings); memcpy(&new_settings, &initial_settings, sizeof(struct termios)); new_settings.c_lflag &= ~ICANON; /* unbuffered input */ /* Turn off echoing and CTRL-C, so we can trap it */ new_settings.c_lflag &= ~(ECHO | ECHONL | ISIG); /* Hmm, in linux c_cc[] not parsed if set ~ICANON */ new_settings.c_cc[VMIN] = 1; new_settings.c_cc[VTIME] = 0; /* Turn off CTRL-C, so we can trap it */ # ifndef _POSIX_VDISABLE # define _POSIX_VDISABLE '\0' # endif new_settings.c_cc[VINTR] = _POSIX_VDISABLE; command[0] = 0; setTermSettings(0, (void *) &new_settings); handlers_sets |= SET_RESET_TERM; /* Now initialize things */ cmdedit_init(); /* Print out the command prompt */ parse_prompt(prompt); while (1) { fflush(stdout); /* buffered out to fast */ if (safe_read(0, &c, 1) < 1) /* if we can't read input then exit */ goto prepare_to_die; ic = c; #ifdef CONFIG_FEATURE_COMMAND_EDITING_VI newdelflag = 1; if (vi_cmdmode) ic |= vbit; #endif switch (ic) { case '\n': case '\r': vi_case( case '\n'|vbit: ) vi_case( case '\r'|vbit: ) /* Enter */ goto_new_line(); break_out = 1; break; case CNTRL('A'): vi_case( case '0'|vbit: ) /* Control-a -- Beginning of line */ input_backward(cursor); break; case CNTRL('B'): vi_case( case 'h'|vbit: ) vi_case( case '\b'|vbit: ) vi_case( case DEL|vbit: ) /* Control-b -- Move back one character */ input_backward(1); break; case CNTRL('C'): vi_case( case CNTRL('C')|vbit: ) /* Control-c -- stop gathering input */ goto_new_line(); #ifndef CONFIG_ASH command[0] = 0; len = 0; lastWasTab = FALSE; put_prompt(); #else len = 0; break_out = -1; /* to control traps */ #endif break; case CNTRL('D'): /* Control-d -- Delete one character, or exit * if the len=0 and no chars to delete */ if (len == 0) { errno = 0; prepare_to_die: #if !defined(CONFIG_ASH) printf("exit"); goto_new_line(); /* cmdedit_reset_term() called in atexit */ exit(EXIT_SUCCESS); #else /* to control stopped jobs */ len = break_out = -1; break; #endif } else { input_delete(0); } break; case CNTRL('E'): vi_case( case '$'|vbit: ) /* Control-e -- End of line */ input_end(); break; case CNTRL('F'): vi_case( case 'l'|vbit: ) vi_case( case ' '|vbit: ) /* Control-f -- Move forward one character */ input_forward(); break; case '\b': case DEL: /* Control-h and DEL */ input_backspace(); break; case '\t': #ifdef CONFIG_FEATURE_COMMAND_TAB_COMPLETION input_tab(&lastWasTab); #endif break; case CNTRL('K'): /* Control-k -- clear to end of line */ *(command + cursor) = 0; len = cursor; printf("\033[J"); break; case CNTRL('L'): vi_case( case CNTRL('L')|vbit: ) /* Control-l -- clear screen */ printf("\033[H"); redraw(0, len-cursor); break; #if MAX_HISTORY >= 1 case CNTRL('N'): vi_case( case CNTRL('N')|vbit: ) vi_case( case 'j'|vbit: ) /* Control-n -- Get next command in history */ if (get_next_history()) goto rewrite_line; break; case CNTRL('P'): vi_case( case CNTRL('P')|vbit: ) vi_case( case 'k'|vbit: ) /* Control-p -- Get previous command from history */ if (cur_history > 0) { get_previous_history(); goto rewrite_line; } else { beep(); } break; #endif case CNTRL('U'): vi_case( case CNTRL('U')|vbit: ) /* Control-U -- Clear line before cursor */ if (cursor) { strcpy(command, command + cursor); redraw(cmdedit_y, len -= cursor); } break; case CNTRL('W'): vi_case( case CNTRL('W')|vbit: ) /* Control-W -- Remove the last word */ while (cursor > 0 && isspace(command[cursor-1])) input_backspace(); while (cursor > 0 &&!isspace(command[cursor-1])) input_backspace(); break; #if CONFIG_FEATURE_COMMAND_EDITING_VI case 'i'|vbit: vi_cmdmode = 0; break; case 'I'|vbit: input_backward(cursor); vi_cmdmode = 0; break; case 'a'|vbit: input_forward(); vi_cmdmode = 0; break; case 'A'|vbit: input_end(); vi_cmdmode = 0; break; case 'x'|vbit: input_delete(1); break; case 'X'|vbit: if (cursor > 0) { input_backward(1); input_delete(1); } break; case 'W'|vbit: vi_Word_motion(command, 1); break; case 'w'|vbit: vi_word_motion(command, 1); break; case 'E'|vbit: vi_End_motion(command); break; case 'e'|vbit: vi_end_motion(command); break; case 'B'|vbit: vi_Back_motion(command); break; case 'b'|vbit: vi_back_motion(command); break; case 'C'|vbit: vi_cmdmode = 0; /* fall through */ case 'D'|vbit: goto clear_to_eol; case 'c'|vbit: vi_cmdmode = 0; /* fall through */ case 'd'|vbit: { int nc, sc; sc = cursor; prevc = ic; if (safe_read(0, &c, 1) < 1) goto prepare_to_die; if (c == (prevc & 0xff)) { /* "cc", "dd" */ input_backward(cursor); goto clear_to_eol; break; } switch(c) { case 'w': case 'W': case 'e': case 'E': switch (c) { case 'w': /* "dw", "cw" */ vi_word_motion(command, vi_cmdmode); break; case 'W': /* 'dW', 'cW' */ vi_Word_motion(command, vi_cmdmode); break; case 'e': /* 'de', 'ce' */ vi_end_motion(command); input_forward(); break; case 'E': /* 'dE', 'cE' */ vi_End_motion(command); input_forward(); break; } nc = cursor; input_backward(cursor - sc); while (nc-- > cursor) input_delete(1); break; case 'b': /* "db", "cb" */ case 'B': /* implemented as B */ if (c == 'b') vi_back_motion(command); else vi_Back_motion(command); while (sc-- > cursor) input_delete(1); break; case ' ': /* "d ", "c " */ input_delete(1); break; case '$': /* "d$", "c$" */ clear_to_eol: while (cursor < len) input_delete(1); break; } } break; case 'p'|vbit: input_forward(); /* fallthrough */ case 'P'|vbit: put(); break; case 'r'|vbit: if (safe_read(0, &c, 1) < 1) goto prepare_to_die; if (c == 0) beep(); else { *(command + cursor) = c; putchar(c); putchar('\b'); } break; #endif /* CONFIG_FEATURE_COMMAND_EDITING_VI */ case ESC: #if CONFIG_FEATURE_COMMAND_EDITING_VI if (vi_mode) { /* ESC: insert mode --> command mode */ vi_cmdmode = 1; input_backward(1); break; } #endif /* escape sequence follows */ if (safe_read(0, &c, 1) < 1) goto prepare_to_die; /* different vt100 emulations */ if (c == '[' || c == 'O') { vi_case( case '['|vbit: ) vi_case( case 'O'|vbit: ) if (safe_read(0, &c, 1) < 1) goto prepare_to_die; } if (c >= '1' && c <= '9') { unsigned char dummy; if (safe_read(0, &dummy, 1) < 1) goto prepare_to_die; if(dummy != '~') c = 0; } switch (c) { #ifdef CONFIG_FEATURE_COMMAND_TAB_COMPLETION case '\t': /* Alt-Tab */ input_tab(&lastWasTab); break; #endif #if MAX_HISTORY >= 1 case 'A': /* Up Arrow -- Get previous command from history */ if (cur_history > 0) { get_previous_history(); goto rewrite_line; } else { beep(); } break; case 'B': /* Down Arrow -- Get next command in history */ if (!get_next_history()) break; /* Rewrite the line with the selected history item */ rewrite_line: /* change command */ len = strlen(strcpy(command, history[cur_history])); /* redraw and go to eol (bol, in vi */ #if CONFIG_FEATURE_COMMAND_EDITING_VI redraw(cmdedit_y, vi_mode ? 9999:0); #else redraw(cmdedit_y, 0); #endif break; #endif 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(0); break; case '1': case 'H': /* */ input_backward(cursor); break; case '4': case 'F': /* */ input_end(); break; default: c = 0; beep(); } break; default: /* If it's regular input, do the normal thing */ #ifdef CONFIG_FEATURE_NONPRINTABLE_INVERSE_PUT /* Control-V -- Add non-printable symbol */ if (c == CNTRL('V')) { if (safe_read(0, &c, 1) < 1) goto prepare_to_die; if (c == 0) { beep(); break; } } else #endif { #if CONFIG_FEATURE_COMMAND_EDITING_VI if (vi_cmdmode) /* don't self-insert */ break; #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(0, (void *) &initial_settings); handlers_sets &= ~SET_RESET_TERM; #if MAX_HISTORY >= 1 /* Handle command history log */ /* cleanup may be saved current command line */ if (len> 0) { /* no put empty line */ int i = n_history; free(history[MAX_HISTORY]); history[MAX_HISTORY] = 0; /* After max history, remove the oldest command */ if (i >= MAX_HISTORY) { free(history[0]); for(i = 0; i < (MAX_HISTORY-1); i++) history[i] = history[i+1]; } history[i++] = bb_xstrdup(command); cur_history = i; n_history = i; #if defined(CONFIG_FEATURE_SH_FANCY_PROMPT) num_ok_lines++; #endif } #else /* MAX_HISTORY < 1 */ #if defined(CONFIG_FEATURE_SH_FANCY_PROMPT) if (len > 0) { /* no put empty line */ num_ok_lines++; } #endif #endif /* MAX_HISTORY >= 1 */ if (break_out > 0) { command[len++] = '\n'; /* set '\n' */ command[len] = 0; } #if defined(CONFIG_FEATURE_CLEAN_UP) && defined(CONFIG_FEATURE_COMMAND_TAB_COMPLETION) input_tab(0); /* strong free */ #endif #if defined(CONFIG_FEATURE_SH_FANCY_PROMPT) free(cmdedit_prompt); #endif cmdedit_reset_term(); return len; } #endif /* CONFIG_FEATURE_COMMAND_EDITING */ #ifdef TEST const char *bb_applet_name = "debug stuff usage"; #ifdef CONFIG_FEATURE_NONPRINTABLE_INVERSE_PUT #include #endif int main(int argc, char **argv) { char buff[BUFSIZ]; char *prompt = #if defined(CONFIG_FEATURE_SH_FANCY_PROMPT) "\\[\\033[32;1m\\]\\u@\\[\\x1b[33;1m\\]\\h:\ \\[\\033[34;1m\\]\\w\\[\\033[35;1m\\] \ \\!\\[\\e[36;1m\\]\\$ \\[\\E[0m\\]"; #else "% "; #endif #ifdef CONFIG_FEATURE_NONPRINTABLE_INVERSE_PUT setlocale(LC_ALL, ""); #endif while(1) { int l; l = cmdedit_read_input(prompt, buff); if(l > 0 && buff[l-1] == '\n') { buff[l-1] = 0; printf("*** cmdedit_read_input() returned line =%s=\n", buff); } else { break; } } printf("*** cmdedit_read_input() detect ^D\n"); return 0; } #endif /* TEST */