1553 lines
		
	
	
		
			34 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			1553 lines
		
	
	
		
			34 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* 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 */
 |