494 lines
14 KiB
Diff
494 lines
14 KiB
Diff
|
--- a/libbb/lineedit.c
|
||
|
+++ b/libbb/lineedit.c
|
||
|
@@ -156,7 +156,6 @@ struct lineedit_statics {
|
||
|
|
||
|
/* Formerly these were big buffers on stack: */
|
||
|
#if ENABLE_FEATURE_TAB_COMPLETION
|
||
|
- char exe_n_cwd_tab_completion__dirbuf[MAX_LINELEN];
|
||
|
char input_tab__matchBuf[MAX_LINELEN];
|
||
|
int16_t find_match__int_buf[MAX_LINELEN + 1]; /* need to have 9 bits at least */
|
||
|
int16_t find_match__pos_buf[MAX_LINELEN + 1];
|
||
|
@@ -235,6 +234,8 @@ static unsigned save_string(char *dst, u
|
||
|
while (dstpos < maxsize) {
|
||
|
wchar_t wc;
|
||
|
int n = srcpos;
|
||
|
+
|
||
|
+ /* Convert up to 1st invalid byte (or up to end) */
|
||
|
while ((wc = command_ps[srcpos]) != 0
|
||
|
&& !unicode_is_raw_byte(wc)
|
||
|
) {
|
||
|
@@ -247,6 +248,7 @@ static unsigned save_string(char *dst, u
|
||
|
dstpos += n;
|
||
|
if (wc == 0) /* usually is */
|
||
|
break;
|
||
|
+
|
||
|
/* We do have invalid byte here! */
|
||
|
command_ps[srcpos] = wc; /* restore it */
|
||
|
srcpos++;
|
||
|
@@ -608,53 +610,56 @@ static void add_match(char *matched)
|
||
|
}
|
||
|
|
||
|
#if ENABLE_FEATURE_USERNAME_COMPLETION
|
||
|
-static void username_tab_completion(char *ud, char *with_shash_flg)
|
||
|
+/* Replace "~user/..." with "/homedir/...".
|
||
|
+ * The parameter is malloced, free it or return it
|
||
|
+ * unchanged if no user is matched.
|
||
|
+ */
|
||
|
+static char *username_path_completion(char *ud)
|
||
|
{
|
||
|
struct passwd *entry;
|
||
|
+ char *tilde_name = ud;
|
||
|
+ char *home = NULL;
|
||
|
+
|
||
|
+ ud++; /* skip ~ */
|
||
|
+ if (*ud == '/') { /* "~/..." */
|
||
|
+ home = home_pwd_buf;
|
||
|
+ } else {
|
||
|
+ /* "~user/..." */
|
||
|
+ ud = strchr(ud, '/');
|
||
|
+ *ud = '\0'; /* "~user" */
|
||
|
+ entry = getpwnam(tilde_name + 1);
|
||
|
+ *ud = '/'; /* restore "~user/..." */
|
||
|
+ if (entry)
|
||
|
+ home = entry->pw_dir;
|
||
|
+ }
|
||
|
+ if (home) {
|
||
|
+ ud = concat_path_file(home, ud);
|
||
|
+ free(tilde_name);
|
||
|
+ tilde_name = ud;
|
||
|
+ }
|
||
|
+ return tilde_name;
|
||
|
+}
|
||
|
+
|
||
|
+/* ~use<tab> - find all users with this prefix */
|
||
|
+static NOINLINE void username_completion(const char *ud)
|
||
|
+{
|
||
|
+ /* Using _r function to avoid pulling in static buffers */
|
||
|
+ char line_buff[256];
|
||
|
+ struct passwd pwd;
|
||
|
+ struct passwd *result;
|
||
|
int userlen;
|
||
|
|
||
|
- ud++; /* ~user/... to user/... */
|
||
|
+ ud++; /* skip ~ */
|
||
|
userlen = strlen(ud);
|
||
|
|
||
|
- if (with_shash_flg) { /* "~/..." or "~user/..." */
|
||
|
- char *sav_ud = ud - 1;
|
||
|
- char *home = NULL;
|
||
|
-
|
||
|
- if (*ud == '/') { /* "~/..." */
|
||
|
- home = home_pwd_buf;
|
||
|
- } else {
|
||
|
- /* "~user/..." */
|
||
|
- char *temp;
|
||
|
- 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) < MAX_LINELEN) {
|
||
|
- /* /home/user/... */
|
||
|
- sprintf(sav_ud, "%s%s", home, ud);
|
||
|
- }
|
||
|
+ setpwent();
|
||
|
+ while (!getpwent_r(&pwd, line_buff, sizeof(line_buff), &result)) {
|
||
|
+ /* Null usernames should result in all users as possible completions. */
|
||
|
+ if (/*!userlen || */ strncmp(ud, pwd.pw_name, userlen) == 0) {
|
||
|
+ add_match(xasprintf("~%s/", pwd.pw_name));
|
||
|
}
|
||
|
- } else {
|
||
|
- /* "~[^/]*" */
|
||
|
- /* Using _r function to avoid pulling in static buffers */
|
||
|
- char line_buff[256];
|
||
|
- struct passwd pwd;
|
||
|
- struct passwd *result;
|
||
|
-
|
||
|
- setpwent();
|
||
|
- while (!getpwent_r(&pwd, line_buff, sizeof(line_buff), &result)) {
|
||
|
- /* Null usernames should result in all users as possible completions. */
|
||
|
- if (/*!userlen || */ strncmp(ud, pwd.pw_name, userlen) == 0) {
|
||
|
- add_match(xasprintf("~%s/", pwd.pw_name));
|
||
|
- }
|
||
|
- }
|
||
|
- endpwent();
|
||
|
}
|
||
|
+ endpwent();
|
||
|
}
|
||
|
#endif /* FEATURE_COMMAND_USERNAME_COMPLETION */
|
||
|
|
||
|
@@ -664,22 +669,19 @@ enum {
|
||
|
FIND_FILE_ONLY = 2,
|
||
|
};
|
||
|
|
||
|
-static int path_parse(char ***p, int flags)
|
||
|
+static int path_parse(char ***p)
|
||
|
{
|
||
|
int npth;
|
||
|
const char *pth;
|
||
|
char *tmp;
|
||
|
char **res;
|
||
|
|
||
|
- /* if not setenv PATH variable, to search cur dir "." */
|
||
|
- if (flags != FIND_EXE_ONLY)
|
||
|
- return 1;
|
||
|
-
|
||
|
if (state->flags & WITH_PATH_LOOKUP)
|
||
|
pth = state->path_lookup;
|
||
|
else
|
||
|
pth = getenv("PATH");
|
||
|
- /* PATH=<empty> or PATH=:<empty> */
|
||
|
+
|
||
|
+ /* PATH="" or PATH=":"? */
|
||
|
if (!pth || !pth[0] || LONE_CHAR(pth, ':'))
|
||
|
return 1;
|
||
|
|
||
|
@@ -689,12 +691,13 @@ static int path_parse(char ***p, int fla
|
||
|
tmp = strchr(tmp, ':');
|
||
|
if (!tmp)
|
||
|
break;
|
||
|
- if (*++tmp == '\0')
|
||
|
+ tmp++;
|
||
|
+ if (*tmp == '\0')
|
||
|
break; /* :<empty> */
|
||
|
npth++;
|
||
|
}
|
||
|
|
||
|
- res = xmalloc(npth * sizeof(char*));
|
||
|
+ *p = res = xmalloc(npth * sizeof(res[0]));
|
||
|
res[0] = tmp = xstrdup(pth);
|
||
|
npth = 1;
|
||
|
while (1) {
|
||
|
@@ -706,100 +709,96 @@ static int path_parse(char ***p, int fla
|
||
|
break; /* :<empty> */
|
||
|
res[npth++] = tmp;
|
||
|
}
|
||
|
- *p = res;
|
||
|
return npth;
|
||
|
}
|
||
|
|
||
|
static void exe_n_cwd_tab_completion(char *command, int type)
|
||
|
{
|
||
|
- DIR *dir;
|
||
|
- struct dirent *next;
|
||
|
- struct stat st;
|
||
|
char *path1[1];
|
||
|
char **paths = path1;
|
||
|
int npaths;
|
||
|
int i;
|
||
|
- char *found;
|
||
|
- char *pfind = strrchr(command, '/');
|
||
|
-/* char dirbuf[MAX_LINELEN]; */
|
||
|
-#define dirbuf (S.exe_n_cwd_tab_completion__dirbuf)
|
||
|
+ char *pfind;
|
||
|
+ char *dirbuf = NULL;
|
||
|
|
||
|
npaths = 1;
|
||
|
path1[0] = (char*)".";
|
||
|
|
||
|
- if (pfind == NULL) {
|
||
|
- /* no dir, if flags==EXE_ONLY - get paths, else "." */
|
||
|
- npaths = path_parse(&paths, type);
|
||
|
+ pfind = strrchr(command, '/');
|
||
|
+ if (!pfind) {
|
||
|
+ if (type == FIND_EXE_ONLY)
|
||
|
+ npaths = path_parse(&paths);
|
||
|
pfind = command;
|
||
|
} else {
|
||
|
+ /* point to 'l' in "..../last_component" */
|
||
|
+ pfind++;
|
||
|
/* dirbuf = ".../.../.../" */
|
||
|
- safe_strncpy(dirbuf, command, (pfind - command) + 2);
|
||
|
+ dirbuf = xstrndup(command, pfind - command);
|
||
|
#if ENABLE_FEATURE_USERNAME_COMPLETION
|
||
|
if (dirbuf[0] == '~') /* ~/... or ~user/... */
|
||
|
- username_tab_completion(dirbuf, dirbuf);
|
||
|
+ dirbuf = username_path_completion(dirbuf);
|
||
|
#endif
|
||
|
- paths[0] = dirbuf;
|
||
|
- /* point to 'l' in "..../last_component" */
|
||
|
- pfind++;
|
||
|
+ path1[0] = dirbuf;
|
||
|
}
|
||
|
|
||
|
for (i = 0; i < npaths; i++) {
|
||
|
+ DIR *dir;
|
||
|
+ struct dirent *next;
|
||
|
+ struct stat st;
|
||
|
+ char *found;
|
||
|
+
|
||
|
dir = opendir(paths[i]);
|
||
|
if (!dir)
|
||
|
continue; /* don't print an error */
|
||
|
|
||
|
while ((next = readdir(dir)) != NULL) {
|
||
|
- int len1;
|
||
|
- const char *str_found = next->d_name;
|
||
|
+ const char *name_found = next->d_name;
|
||
|
|
||
|
- /* matched? */
|
||
|
- if (strncmp(str_found, pfind, strlen(pfind)))
|
||
|
+ /* .../<tab>: bash 3.2.0 shows dotfiles, but not . and .. */
|
||
|
+ if (!pfind[0] && DOT_OR_DOTDOT(name_found))
|
||
|
continue;
|
||
|
- /* not see .name without .match */
|
||
|
- if (*str_found == '.' && *pfind == '\0') {
|
||
|
- if (NOT_LONE_CHAR(paths[i], '/') || str_found[1])
|
||
|
- continue;
|
||
|
- str_found = ""; /* only "/" */
|
||
|
- }
|
||
|
- found = concat_path_file(paths[i], str_found);
|
||
|
- /* hmm, remove in progress? */
|
||
|
+ /* match? */
|
||
|
+ if (strncmp(name_found, pfind, strlen(pfind)) != 0)
|
||
|
+ continue; /* no */
|
||
|
+
|
||
|
+ found = concat_path_file(paths[i], name_found);
|
||
|
/* NB: stat() first so that we see is it a directory;
|
||
|
* but if that fails, use lstat() so that
|
||
|
* we still match dangling links */
|
||
|
if (stat(found, &st) && lstat(found, &st))
|
||
|
- goto cont;
|
||
|
- /* find with dirs? */
|
||
|
- if (paths[i] != dirbuf)
|
||
|
- strcpy(found, next->d_name); /* only name */
|
||
|
+ goto cont; /* hmm, remove in progress? */
|
||
|
|
||
|
- len1 = strlen(found);
|
||
|
- found = xrealloc(found, len1 + 2);
|
||
|
- found[len1] = '\0';
|
||
|
- found[len1+1] = '\0';
|
||
|
+ /* save only name if we scan PATH */
|
||
|
+ if (paths[i] != dirbuf)
|
||
|
+ strcpy(found, name_found);
|
||
|
|
||
|
if (S_ISDIR(st.st_mode)) {
|
||
|
+ unsigned len1 = strlen(found);
|
||
|
/* name is a directory */
|
||
|
if (found[len1-1] != '/') {
|
||
|
+ found = xrealloc(found, len1 + 2);
|
||
|
found[len1] = '/';
|
||
|
+ found[len1 + 1] = '\0';
|
||
|
}
|
||
|
} else {
|
||
|
- /* not put found file if search only dirs for cd */
|
||
|
+ /* skip files if looking for dirs only (example: cd) */
|
||
|
if (type == FIND_DIR_ONLY)
|
||
|
goto cont;
|
||
|
}
|
||
|
- /* Add it to the list */
|
||
|
+ /* add it to the list */
|
||
|
add_match(found);
|
||
|
continue;
|
||
|
cont:
|
||
|
free(found);
|
||
|
}
|
||
|
closedir(dir);
|
||
|
- }
|
||
|
+ } /* for every path */
|
||
|
+
|
||
|
if (paths != path1) {
|
||
|
free(paths[0]); /* allocated memory is only in first member */
|
||
|
free(paths);
|
||
|
}
|
||
|
-#undef dirbuf
|
||
|
+ free(dirbuf);
|
||
|
}
|
||
|
|
||
|
/* QUOT is used on elements of int_buf[], which are bytes,
|
||
|
@@ -812,15 +811,20 @@ static void exe_n_cwd_tab_completion(cha
|
||
|
/* is must be <= in */
|
||
|
static void collapse_pos(int is, int in)
|
||
|
{
|
||
|
- memmove(int_buf+is, int_buf+in, (MAX_LINELEN+1-in)*sizeof(int_buf[0]));
|
||
|
- memmove(pos_buf+is, pos_buf+in, (MAX_LINELEN+1-in)*sizeof(pos_buf[0]));
|
||
|
+ memmove(int_buf+is, int_buf+in, (MAX_LINELEN+1-in) * sizeof(int_buf[0]));
|
||
|
+ memmove(pos_buf+is, pos_buf+in, (MAX_LINELEN+1-in) * sizeof(pos_buf[0]));
|
||
|
}
|
||
|
-static NOINLINE int find_match(char *matchBuf, int *len_with_quotes)
|
||
|
+/* On entry, matchBuf contains everything up to cursor at the moment <tab>
|
||
|
+ * was pressed. This function looks at it, figures out what part of it
|
||
|
+ * constitutes the command/file/directory prefix to use for completion,
|
||
|
+ * and rewrites matchBuf to contain only that part.
|
||
|
+ */
|
||
|
+static NOINLINE int build_match_prefix(char *matchBuf, int *len_with_quotes)
|
||
|
{
|
||
|
int i, j;
|
||
|
int command_mode;
|
||
|
int c, c2;
|
||
|
-/* Were local, but it uses too much stack */
|
||
|
+/* Were local, but it used too much stack */
|
||
|
/* int16_t int_buf[MAX_LINELEN + 1]; */
|
||
|
/* int16_t pos_buf[MAX_LINELEN + 1]; */
|
||
|
|
||
|
@@ -860,22 +864,23 @@ static NOINLINE int find_match(char *mat
|
||
|
}
|
||
|
|
||
|
/* skip commands with arguments if line has commands delimiters */
|
||
|
- /* ';' ';;' '&' '|' '&&' '||' but `>&' `<&' `>|' */
|
||
|
+ /* ';' ';;' '&' '|' '&&' '||' but '>&' '<&' '>|' */
|
||
|
for (i = 0; int_buf[i]; i++) {
|
||
|
+ int n;
|
||
|
c = int_buf[i];
|
||
|
c2 = int_buf[i + 1];
|
||
|
j = i ? int_buf[i - 1] : -1;
|
||
|
- command_mode = 0;
|
||
|
+ n = 0;
|
||
|
if (c == ';' || c == '&' || c == '|') {
|
||
|
- command_mode = 1 + (c == c2);
|
||
|
+ n = 1 + (c == c2);
|
||
|
if (c == '&') {
|
||
|
if (j == '>' || j == '<')
|
||
|
- command_mode = 0;
|
||
|
+ n = 0;
|
||
|
} else if (c == '|' && j == '>')
|
||
|
- command_mode = 0;
|
||
|
+ n = 0;
|
||
|
}
|
||
|
- if (command_mode) {
|
||
|
- collapse_pos(0, i + command_mode);
|
||
|
+ if (n) {
|
||
|
+ collapse_pos(0, i + n);
|
||
|
i = -1; /* hack incremet */
|
||
|
}
|
||
|
}
|
||
|
@@ -943,8 +948,8 @@ static NOINLINE int find_match(char *mat
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
- for (i = 0; int_buf[i]; i++)
|
||
|
- /* "strlen" */;
|
||
|
+ for (i = 0; int_buf[i]; i++) /* quasi-strlen(int_buf) */
|
||
|
+ continue;
|
||
|
/* find last word */
|
||
|
for (--i; i >= 0; i--) {
|
||
|
c = int_buf[i];
|
||
|
@@ -955,7 +960,7 @@ static NOINLINE int find_match(char *mat
|
||
|
}
|
||
|
/* skip first not quoted '\'' or '"' */
|
||
|
for (i = 0; int_buf[i] == '\'' || int_buf[i] == '"'; i++)
|
||
|
- /*skip*/;
|
||
|
+ continue;
|
||
|
/* collapse quote or unquote // or /~ */
|
||
|
while ((int_buf[i] & ~QUOT) == '/'
|
||
|
&& ((int_buf[i+1] & ~QUOT) == '/' || (int_buf[i+1] & ~QUOT) == '~')
|
||
|
@@ -1058,13 +1063,20 @@ static void input_tab(smallint *lastWasT
|
||
|
|
||
|
/* Make a local copy of the string --
|
||
|
* up to the position of the cursor */
|
||
|
+#if !ENABLE_UNICODE_SUPPORT
|
||
|
save_string(matchBuf, cursor + 1);
|
||
|
-#if ENABLE_UNICODE_SUPPORT
|
||
|
- cursor_mb = strlen(matchBuf);
|
||
|
+#else
|
||
|
+ {
|
||
|
+ CHAR_T wc = command_ps[cursor];
|
||
|
+ command_ps[cursor] = 0;
|
||
|
+ save_string(matchBuf, MAX_LINELEN);
|
||
|
+ command_ps[cursor] = wc;
|
||
|
+ cursor_mb = strlen(matchBuf);
|
||
|
+ }
|
||
|
#endif
|
||
|
tmp = matchBuf;
|
||
|
|
||
|
- find_type = find_match(matchBuf, &recalc_pos);
|
||
|
+ find_type = build_match_prefix(matchBuf, &recalc_pos);
|
||
|
|
||
|
/* Free up any memory already allocated */
|
||
|
free_tab_completion_data();
|
||
|
@@ -1074,7 +1086,7 @@ static void input_tab(smallint *lastWasT
|
||
|
* then try completing this word as a username. */
|
||
|
if (state->flags & USERNAME_COMPLETION)
|
||
|
if (matchBuf[0] == '~' && strchr(matchBuf, '/') == NULL)
|
||
|
- username_tab_completion(matchBuf, NULL);
|
||
|
+ username_completion(matchBuf);
|
||
|
#endif
|
||
|
/* Try to match any executable in our path and everything
|
||
|
* in the current working directory */
|
||
|
@@ -1083,7 +1095,7 @@ static void input_tab(smallint *lastWasT
|
||
|
/* Sort, then remove any duplicates found */
|
||
|
if (matches) {
|
||
|
unsigned i;
|
||
|
- int n = 0;
|
||
|
+ unsigned n = 0;
|
||
|
qsort_string_vector(matches, num_matches);
|
||
|
for (i = 0; i < num_matches - 1; ++i) {
|
||
|
if (matches[i] && matches[i+1]) { /* paranoia */
|
||
|
@@ -1095,14 +1107,14 @@ static void input_tab(smallint *lastWasT
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
- matches[n] = matches[i];
|
||
|
- num_matches = n + 1;
|
||
|
+ matches[n++] = matches[i];
|
||
|
+ num_matches = n;
|
||
|
}
|
||
|
/* Did we find exactly one match? */
|
||
|
- if (!matches || num_matches > 1) { /* no */
|
||
|
+ if (num_matches != 1) { /* no */
|
||
|
beep();
|
||
|
if (!matches)
|
||
|
- return; /* not found */
|
||
|
+ return; /* no matches at all */
|
||
|
/* find minimal match */
|
||
|
tmp1 = xstrdup(matches[0]);
|
||
|
for (tmp = tmp1; *tmp; tmp++) {
|
||
|
@@ -1113,13 +1125,14 @@ static void input_tab(smallint *lastWasT
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
- if (*tmp1 == '\0') { /* have unique */
|
||
|
- free(tmp1);
|
||
|
+ if (*tmp1 == '\0') { /* have unique pfx? */
|
||
|
+ free(tmp1); /* no */
|
||
|
return;
|
||
|
}
|
||
|
tmp = add_quote_for_spec_chars(tmp1);
|
||
|
free(tmp1);
|
||
|
- } else { /* one match */
|
||
|
+ len_found = strlen(tmp);
|
||
|
+ } else { /* exactly one match */
|
||
|
tmp = add_quote_for_spec_chars(matches[0]);
|
||
|
/* for next completion current found */
|
||
|
*lastWasTab = FALSE;
|
||
|
@@ -1127,11 +1140,10 @@ static void input_tab(smallint *lastWasT
|
||
|
len_found = strlen(tmp);
|
||
|
if (tmp[len_found-1] != '/') {
|
||
|
tmp[len_found] = ' ';
|
||
|
- tmp[len_found+1] = '\0';
|
||
|
+ tmp[++len_found] = '\0';
|
||
|
}
|
||
|
}
|
||
|
|
||
|
- len_found = strlen(tmp);
|
||
|
#if !ENABLE_UNICODE_SUPPORT
|
||
|
/* have space to place the match? */
|
||
|
/* The result consists of three parts with these lengths: */
|
||
|
@@ -1164,7 +1176,10 @@ static void input_tab(smallint *lastWasT
|
||
|
sprintf(&command[cursor_mb - recalc_pos], "%s%s", tmp, matchBuf);
|
||
|
command_len = load_string(command, S.maxsize);
|
||
|
/* write out the matched command */
|
||
|
- redraw(cmdedit_y, command_len - len);
|
||
|
+ /* paranoia: load_string can return 0 on conv error,
|
||
|
+ * prevent passing len = (0 - 12) to redraw */
|
||
|
+ len = command_len - len;
|
||
|
+ redraw(cmdedit_y, len >= 0 ? len : 0);
|
||
|
}
|
||
|
}
|
||
|
#endif
|
||
|
--- a/libbb/unicode.c
|
||
|
+++ b/libbb/unicode.c
|
||
|
@@ -131,7 +131,7 @@ size_t FAST_FUNC wcstombs(char *dest, co
|
||
|
size_t len = wcrtomb_internal(tbuf, wc);
|
||
|
|
||
|
if (len > n)
|
||
|
- len = n;
|
||
|
+ break;
|
||
|
memcpy(dest, tbuf, len);
|
||
|
if (wc == L'\0')
|
||
|
return org_n - n;
|