aboutsummaryrefslogtreecommitdiff
path: root/toys
diff options
context:
space:
mode:
Diffstat (limited to 'toys')
-rw-r--r--toys/pending/sh.c177
1 files changed, 96 insertions, 81 deletions
diff --git a/toys/pending/sh.c b/toys/pending/sh.c
index e5d7227f..632f75b6 100644
--- a/toys/pending/sh.c
+++ b/toys/pending/sh.c
@@ -168,7 +168,7 @@ GLOBALS(
// keep lineno here, we use it to work around a compiler bug
long lineno;
- char *ifs;
+ char *ifs, *isexec;
struct double_list functions;
unsigned options, jobcnt;
int hfd, pid, varslen, shift, cdcount;
@@ -401,6 +401,19 @@ static struct sh_vars *setvar(char *s)
return var;
}
+static void unsetvar(char *name)
+{
+ struct sh_vars *var = findvar(name);
+ int ii = var-TT.vars;
+
+ if (!var) return;
+ if (var->flags&VAR_GLOBAL) xunsetenv(name);
+ else free(var->str);
+
+ memmove(TT.vars+ii, TT.vars+ii+1, TT.varslen-ii);
+ TT.varslen--;
+}
+
static struct sh_vars *setvarval(char *name, char *val)
{
return setvar(xmprintf("%s=%s", name, val));
@@ -427,21 +440,6 @@ static char *getvar(char *s)
return varend(var->str)+1;
}
-static void unsetvar(char *name)
-{
- struct sh_vars *var = findvar(name);
- int ii;
-
- if (!var) return;
- if (var->flags&VAR_GLOBAL) {
- *varend(var->str) = 0;
- xunsetenv(var->str);
- } else free(var->str);
-
- ii = var-TT.vars;
- memmove(TT.vars+ii, TT.vars+ii+1, TT.varslen-ii);
-}
-
// malloc declare -x "escaped string"
static char *declarep(struct sh_vars *var)
{
@@ -641,7 +639,7 @@ if (BUGBUG) dprintf(255, "%d schedule close %d\n", getpid(), to);
}
// TODO: waitpid(WNOHANG) to clean up zombies and catch background& ending
-static void subshell_callback(void)
+static void subshell_callback(char **argv)
{
char *s;
@@ -674,7 +672,7 @@ if (BUGBUG) dprintf(255, "run_subshell %.*s\n", len, str);
// On nommu vfork, exec /proc/self/exe, and pipe state data to ourselves.
} else {
- int pipes[2], i;
+ int pipes[2], i, c;
// open pipe to child
if (pipe(pipes) || 254 != dup2(pipes[0], 254)) return 1;
@@ -686,7 +684,8 @@ if (BUGBUG) dprintf(255, "run_subshell %.*s\n", len, str);
// free entries added to end of environment by callback (shared heap)
for (i = 0; environ[i]; i++) {
- if (!ispunct(environ[i][0])) continue;
+ c = environ[i][0];
+ if (c == '_' || !ispunct(c)) continue;
free(environ[i]);
environ[i] = 0;
}
@@ -791,8 +790,6 @@ static int expand_arg_nobrace(struct sh_arg *arg, char *str, unsigned flags,
if (BUGBUG) dprintf(255, "expand %s\n", str);
-// TODO ls -l /proc/$$/fd
-
// Tilde expansion
if (!(flags&NO_TILDE) && *str == '~') {
struct passwd *pw = 0;
@@ -1467,13 +1464,33 @@ notfd:
return pp;
}
+static void shexec(char *cmd, char **argv)
+{
+ xsetenv(xmprintf("_=%s", cmd), 0);
+ execve(cmd, argv, environ);
+ if (errno == ENOEXEC) run_subshell("source \"$_\"", 11);
+}
+
+// Call binary, or run via child shell
+static void sh_exec(char **argv)
+{
+ char *pp = getvar("PATH" ? : _PATH_DEFPATH), *cc = TT.isexec ? : *argv;
+ struct string_list *sl;
+
+ if (strchr(cc, '/')) shexec(cc, argv);
+ else for (sl = find_in_path(pp, cc); sl; free(llist_pop(&sl)))
+ shexec(sl->str, argv);
+
+ if (!TT.isexec) _exit(127);
+}
+
// Execute a single command
static struct sh_process *run_command(struct sh_arg *arg)
{
+ char *s, *ss = 0, *sss, **env = 0, **old = environ;
+ int envlen, jj, kk, ll;
struct sh_process *pp;
struct toy_list *tl;
- int envlen, j;
- char *s;
if (BUGBUG) dprintf(255, "run_command %s\n", arg->v[0]);
@@ -1489,14 +1506,36 @@ if (BUGBUG) dprintf(255, "run_command %s\n", arg->v[0]);
if (BUGBUG) { int i; dprintf(255, "envlen=%d arg->c=%d run=", envlen, arg->c); for (i=0; i<pp->arg.c; i++) dprintf(255, "'%s' ", pp->arg.v[i]); dprintf(255, "\n"); }
// perform assignments locally if there's no command
if (envlen == arg->c) {
- for (j = 0; j<envlen; j++) {
- if (!(s = expand_one_arg(arg->v[j], NO_PATH|NO_SPLIT, 0))) break;
- if (s == arg->v[j]) s = xstrdup(s);
+ for (jj = 0; jj<envlen; jj++) {
+ if (!(s = expand_one_arg(arg->v[jj], NO_PATH|NO_SPLIT, 0))) break;
+ if (s == arg->v[jj]) s = xstrdup(s);
setvar(s);
}
+ goto out;
+ }
+
+ if (envlen) {
+ for (kk = 0; environ[kk]; kk++);
+ env = xmalloc(sizeof(char *)*(kk+33));
+ memcpy(env, environ, sizeof(char *)*(kk+1));
+ environ = env;
+
+ // assign leading environment variables
+ for (jj = 0; jj<envlen; jj++) {
+ if (!(sss = expand_one_arg(arg->v[jj], NO_PATH|NO_SPLIT, &pp->delete)))
+ break;
+ for (ll = 0; ll<kk; ll++) {
+ for (s = sss, ss = environ[ll]; *s == *ss && *s != '='; s++, ss++);
+ if (*s != '=') continue;
+ environ[ll] = sss;
+ break;
+ }
+ if (ll == kk) array_add(&environ, kk++, sss);
+ }
+ } else jj = 0;
// Do nothing if nothing to do
- } else if (pp->exit || !pp->arg.v);
+ if (jj != envlen || pp->exit || !pp->arg.v);
// else if (!strcmp(*pp->arg.v, "(("))
// TODO: handle ((math)) currently totally broken
// TODO: call functions()
@@ -1505,11 +1544,11 @@ if (BUGBUG) { int i; dprintf(255, "envlen=%d arg->c=%d run=", envlen, arg->c); f
&& (tl->flags & (TOYFLAG_NOFORK|TOYFLAG_MAYFORK)))
{
sigjmp_buf rebound;
- char temp[j = offsetof(struct toy_context, rebound)];
+ char temp[jj = offsetof(struct toy_context, rebound)];
// This fakes lots of what toybox_main() does.
- memcpy(&temp, &toys, j);
- memset(&toys, 0, j);
+ memcpy(&temp, &toys, jj);
+ memset(&toys, 0, jj);
// If we give the union in TT a name, the compiler complains
// "declaration does not declare anything", but if we DON'T give it a name
@@ -1529,45 +1568,16 @@ if (BUGBUG) { int i; dprintf(255, "envlen=%d arg->c=%d run=", envlen, arg->c); f
pp->exit = toys.exitval;
if (toys.optargs != toys.argv+1) free(toys.optargs);
if (toys.old_umask) umask(toys.old_umask);
- memcpy(&toys, &temp, j);
- } else {
- char **env = 0, **old = environ, *ss = 0, *sss;
- int kk = 0, ll, mm = -1;
+ memcpy(&toys, &temp, jj);
+ } else if (-1==(pp->pid = xpopen_setup(pp->arg.v+envlen, 0, sh_exec)))
+ perror_msg("%s: vfork", *pp->arg.v);
- // We don't allocate/free any array members, just the array
- if (environ) while (environ[kk]) {
- if (strncmp(environ[kk], "SHLVL=", 6)) mm = kk;
- kk++;
- }
- if (kk) {
- env = xmalloc(sizeof(char *)*(kk+33));
- memcpy(env, environ, sizeof(char *)*(kk+1));
- if (mm != -1) env[mm] = xmprintf("SHLVL=%d", atoi(env[mm]+6)+1);
- environ = env;
- }
-
- // assign leading environment variables
- for (j = 0; j<envlen; j++) {
- if (!(sss = expand_one_arg(arg->v[j], NO_PATH|NO_SPLIT, &pp->delete)))
- break;
- for (ll = 0; ll<kk; ll++) {
- for (s = sss, ss = env[ll]; *s == *ss && *s != '='; s++, ss++);
- if (*s != '=') continue;
- env[ll] = sss;
- break;
- }
- if (ll == kk) array_add(&environ, kk++, sss);
- }
-
- if (j == envlen && -1 == (pp->pid = xpopen_both(pp->arg.v, 0)))
- perror_msg("%s: vfork", *pp->arg.v);
-
- // Restore environment variables
- environ = old;
- if (mm != -1) free(env[mm]);
- free(env);
- }
+ // Restore environment variables
+ environ = old;
+ free(env);
+out:
+ setvarval("_", (envlen == arg->c) ? "" : s);
// cleanup process
unredirect(pp->urd);
@@ -2201,7 +2211,7 @@ dprintf(2, "TODO skipped init for((;;)), need math parser\n");
} else expand_arg(&blk->farg, "\"$@\"", 0, &blk->fdelete);
}
-// TODO case/esac [[/]] (/) ((/)) function/}
+// TODO case/esac [[/]] ((/)) function/}
/*
TODO: a | b | c needs subshell for builtins?
@@ -2358,7 +2368,7 @@ static struct sh_vars *initlocal(char *name, char *val)
return addvar(xmprintf("%s=%s", name, val ? val : ""));
}
-// export malloced name=value string
+// export existing "name" or assign/export name=value string (making new copy)
static void export(char *str)
{
struct sh_vars *shv = 0;
@@ -2366,7 +2376,7 @@ static void export(char *str)
// Make sure variable exists and is updated
if (strchr(str, '=')) shv = setvar(xstrdup(str));
- else if (!findvar(str)) shv = addvar(str = xmprintf("%s=", str));
+ else if (!(shv = findvar(str))) shv = addvar(str = xmprintf("%s=", str));
if (!shv || (shv->flags&VAR_GLOBAL)) return;
// Resolve local magic for export
@@ -2433,6 +2443,7 @@ static void subshell_setup(void)
// Ensure environ copied and toys.envc set, and clean out illegal entries
TT.ifs = " \t\n";
+ xsetenv("", 0);
for (to = from = pid = ppid = zpid = 0; (s = environ[from]); from++) {
// If nommu subshell gets handoff
@@ -2479,9 +2490,13 @@ static void subshell_setup(void)
s = xsetenv("_", s);
if (!findvar(s)) addvar(s)->flags = VAR_GLOBAL;
free(ss);
- if (!getvar("SHLVL")) {
- s = xsetenv("SHLVL", "1");
- if (!findvar(s)) addvar(s)->flags = VAR_GLOBAL;
+ if (!(ss = getvar("SHLVL"))) export("SHLVL=1");
+ else {
+ char buf[16];
+
+ sprintf(buf, "%u", atoi(ss+6)+1);
+ xsetenv("SHLVL", buf);
+ export("SHLVL");
}
//TODO indexed array,associative array,integer,local,nameref,readonly,uppercase
@@ -2732,8 +2747,7 @@ void eval_main(void)
void exec_main(void)
{
- char *ee[1] = {0}, **env = FLAG(c) ? ee : environ, *cc, *pp = getvar("PATH");
- struct string_list *sl;
+ char *ee[1] = {0}, **old = environ;
// discard redirects and return if nothing to exec
free(TT.pp->urd);
@@ -2741,16 +2755,17 @@ void exec_main(void)
if (!toys.optc) return;
// exec, handling -acl
- cc = *toys.optargs;
+ TT.isexec = *toys.optargs;
+ if (FLAG(c)) environ = ee;
if (TT.exec.a || FLAG(l))
- *toys.optargs = xmprintf("%s%s", FLAG(l) ? "-" : "", TT.exec.a ? : cc);
- if (strchr(cc, '/')) execve(cc, toys.optargs, env);
- else for (sl = find_in_path(pp?:_PATH_DEFPATH, cc); sl; free(llist_pop(&sl)))
- execve(sl->str, toys.optargs, env);
+ *toys.optargs = xmprintf("%s%s", FLAG(l) ? "-" : "", TT.exec.a?:TT.isexec);
+ sh_exec(toys.optargs);
// report error (usually ENOENT) and return
- perror_msg("%s", cc);
+ perror_msg("%s", TT.isexec);
toys.exitval = 127;
+ environ = old;
+ TT.isexec = 0;
}
void shift_main(void)