aboutsummaryrefslogtreecommitdiff
path: root/toys/pending/sh.c
diff options
context:
space:
mode:
Diffstat (limited to 'toys/pending/sh.c')
-rw-r--r--toys/pending/sh.c384
1 files changed, 256 insertions, 128 deletions
diff --git a/toys/pending/sh.c b/toys/pending/sh.c
index 27dda822..df8edd4f 100644
--- a/toys/pending/sh.c
+++ b/toys/pending/sh.c
@@ -19,11 +19,9 @@
*
* Things like the bash man page are good to read too.
*
- * TODO: "make sh" doesn't work (nofork builtins need to be included)
* TODO: test that $PS1 color changes work without stupid \[ \] hack
* TODO: make fake pty wrapper for test infrastructure
* TODO: // Handle embedded NUL bytes in the command line.
- * TODO: var=val command
* existing but considered builtins: false kill pwd true time
* buitins: alias bg command fc fg getopts jobs newgrp read umask unalias wait
* "special" builtins: break continue : . eval exec export readonly return set
@@ -37,6 +35,7 @@
* label:
* TODO: test exit from "trap EXIT" doesn't recurse
* TODO: ! history expansion
+ * TODO: getuid() vs geteuid()
*
* bash man page:
* control operators || & && ; ;; ;& ;;& ( ) | |& <newline>
@@ -44,9 +43,7 @@
* ! case coproc do done elif else esac fi for function if in select
* then until while { } time [[ ]]
-
-
-USE_SH(NEWTOY(cd, NULL, TOYFLAG_NOFORK))
+USE_SH(NEWTOY(cd, ">1LP[-LP]", TOYFLAG_NOFORK))
USE_SH(NEWTOY(exit, NULL, TOYFLAG_NOFORK))
USE_SH(NEWTOY(sh, "(noediting)(noprofile)(norc)sc:i", TOYFLAG_BIN))
@@ -75,7 +72,7 @@ config CD
default n
depends on SH
help
- usage: cd [-PL] [path]
+ usage: cd [-PLe] [path]
Change current directory. With no arguments, go $HOME.
@@ -100,11 +97,10 @@ GLOBALS(
char *command;
long lineno;
-
char **locals;
-
struct double_list functions;
- unsigned options;
+ unsigned options, jobcnt;
+ int hfd; // next high filehandle (>= 10)
// Running jobs.
struct sh_job {
@@ -120,13 +116,11 @@ GLOBALS(
// null terminated array of running processes in pipeline
struct sh_process {
struct sh_process *next, *prev;
- struct string_list *delete; // expanded strings
+ struct arg_list *delete; // expanded strings
int *urd, envlen, pid, exit; // undo redirects, child PID, exit status
struct sh_arg arg;
} *procs, *proc;
} *jobs, *job;
- unsigned jobcnt;
- int hfd, *hfdclose; // next high filehandle (>= 10)
)
#define BUGBUG 0
@@ -139,20 +133,6 @@ static const char *redirectors[] = {"<<<", "<<-", "<<", "<&", "<>", "<", ">>",
#define SH_NOCLOBBER 1 // set -C
-void cd_main(void)
-{
- char *dest = *toys.optargs ? *toys.optargs : getenv("HOME");
-
-// TODO: -LPE@
-// TODO: cd .. goes up $PWD path we used to get here, not ./..
- xchdir(dest ? dest : "/");
-}
-
-void exit_main(void)
-{
- exit(*toys.optargs ? atoi(*toys.optargs) : 0);
-}
-
// like error_msg() but exit from shell scripts
static void syntax_err(char *msg, ...)
{
@@ -165,59 +145,54 @@ static void syntax_err(char *msg, ...)
if (*toys.optargs) xexit();
}
-// Print prompt to stderr, parsing escapes
-// Truncated to 4k at the moment, waiting for somebody to complain.
-static void do_prompt(char *prompt)
+void array_add(char ***list, unsigned count, char *data)
{
- char *s, c, cc, *pp = toybuf;
- int len;
+ if (!(count&31)) *list = xrealloc(*list, sizeof(char *)*(count+33));
+ (*list)[count] = data;
+ (*list)[count+1] = 0;
+}
- if (!prompt) prompt = "\\$ ";
- while ((len = sizeof(toybuf)-(pp-toybuf))>0 && *prompt) {
- c = *(prompt++);
+// TODO local variables
- if (c=='!') {
- if (*prompt=='!') prompt++;
- else {
- pp += snprintf(pp, len, "%ld", TT.lineno);
- continue;
- }
- } else if (c=='\\') {
- cc = *(prompt++);
- if (!cc) {
- *pp++ = c;
- break;
- }
+// Assign one variable
+// s: key=val
+// type: 0 = whatever it was before, local otherwise
+#define TAKE_MEM 0x80000000
+// declare -aAilnrux
+// ft
+static void setvar(char *s, unsigned type)
+{
+ if (type&TAKE_MEM) type ^= TAKE_MEM;
+ else s = xstrdup(s);
- // \nnn \dD{}hHjlstT@AuvVwW!#$
- // Ignore bash's "nonprintable" hack; query our cursor position instead.
- if (cc=='[' || cc==']') continue;
- else if (cc=='$') *pp++ = getuid() ? '$' : '#';
- else if (cc=='h' || cc=='H') {
- *pp = 0;
- gethostname(pp, len);
- pp[len-1] = 0;
- if (cc=='h' && (s = strchr(pp, '.'))) *s = 0;
- pp += strlen(pp);
- } else if (cc=='s') {
- s = getbasename(*toys.argv);
- while (*s && len--) *pp++ = *s++;
- } else if (!(c = unescape(cc))) {
- *pp++ = '\\';
- if (--len) *pp++ = c;
- } else *pp++ = c;
- } else *pp++ = c;
+ // local, export, readonly, integer...
+ xsetenv(s, 0);
+}
+
+// get variable of length len starting at s.
+static char *getvarlen(char *s, int len)
+{
+ unsigned uu;
+ char **ss = TT.locals;
+
+ // loop through local, then global variables
+ for (uu = 0; ; uu++) {
+ if (!ss || !ss[uu]) {
+ if (ss != TT.locals) return 0;
+ ss = environ;
+ uu = 0;
+ }
+
+ // Use UHF rubik's cube protocol to find match.
+ if (!strncmp(ss[uu], s, len) && ss[uu][len] == '=') return ss[uu]+len+1;
}
- len = pp-toybuf;
- if (len>=sizeof(toybuf)) len = sizeof(toybuf);
- writeall(2, toybuf, len);
+
+ return 0;
}
-void array_add(char ***list, unsigned count, char *data)
+static char *getvar(char *s)
{
- if (!(count&31)) *list = xrealloc(*list, sizeof(char *)*(count+33));
- (*list)[count] = data;
- (*list)[count+1] = 0;
+ return getvarlen(s, strlen(s));
}
// quote removal, brace, tilde, parameter/variable, $(command),
@@ -227,6 +202,7 @@ void array_add(char ***list, unsigned count, char *data)
#define NO_BRACE (1<<2)
#define NO_TILDE (1<<3)
#define NO_QUOTE (1<<4)
+#define FORCE_COPY (1<<31)
// TODO: ${name:?error} causes an error/abort here (syntax_err longjmp?)
// TODO: $1 $@ $* need args marshalled down here: function+structure?
// arg = append to this
@@ -234,15 +210,34 @@ void array_add(char ***list, unsigned count, char *data)
// flags = type of expansions (not) to do
// delete = append new allocations to this so they can be freed later
// TODO: at_args: $1 $2 $3 $* $@
-static void expand_arg(struct sh_arg *arg, char *new, unsigned flags,
- struct string_list **delete)
+static void expand_arg(struct sh_arg *arg, char *old, unsigned flags,
+ struct arg_list **delete)
{
-// char *s = new, quote = 0;
+ char *new = old, *s, *ss, quote = 0;
// TODO ls -l /proc/$$/fd
// ${ $(( $( $[ $' `
+ // Tilde expansion
+ if (!(flags&NO_TILDE) && *new == '~') {
+ struct passwd *pw = 0;
+
+ // first expansion so don't need to free previous new
+ ss = 0;
+ for (s = new; *s && *s!=':' && *s!='/'; s++);
+ if (s-new==1) {
+ if (!(ss = getvar("HOME")) || !*ss) pw = bufgetpwuid(getuid());
+ } else {
+ // TODO bufgetpwnam
+ pw = getpwnam(new = xstrndup(new+1, (s-new)-1));
+ free(new);
+ }
+ if (pw && pw->pw_dir) ss = pw->pw_dir;
+ if (!ss || !*ss) ss = "/";
+ new = xmprintf("%s%s", ss, s);
+ }
+
/*
while (*s) {
if (!quote && !(flags&NO_BRACE) && *s == '{') {
@@ -259,12 +254,6 @@ TODO this recurses
// PS0 PS1='$ ' PS2='> ' PS3
}
}
-*/
- // literal passthrough
- array_add(&arg->v, arg->c++, new);
-
-/*
- char *s = word, *new = 0;
// replacement
while (*s) {
@@ -276,14 +265,23 @@ TODO this recurses
s++;
} else s++;
}
-
- return new;
*/
+
+ // We have a result. Append it.
+ if (old==new && (flags&FORCE_COPY)) new = xstrdup(new);
+ if (old!=new && delete) {
+ struct arg_list *al = xmalloc(sizeof(struct arg_list));
+
+ al->next = *delete;
+ al->arg = new;
+ *delete = al;
+ }
+ array_add(&arg->v, arg->c++, new);
}
// Expand exactly one arg, returning NULL if it split.
// If return != new you need to free it.
-static char *expand_one_arg(char *new, unsigned flags, struct string_list **del)
+static char *expand_one_arg(char *new, unsigned flags, struct arg_list **del)
{
struct sh_arg arg;
char *s = 0;
@@ -300,53 +298,17 @@ static char *expand_one_arg(char *new, unsigned flags, struct string_list **del)
return s;
}
-// Assign one variable
-// s: key=val
-// type: 0 = whatever it was before, local otherwise
-#define TAKE_MEM 0x80000000
-// declare -aAilnrux
-// ft
-static void setvar(char *s, unsigned type)
-{
- if (type&TAKE_MEM) type ^= TAKE_MEM;
- else s = xstrdup(s);
-
- // local, export, readonly, integer...
- xsetenv(s, 0);
-}
-
-// get variable of length len starting at s.
-static char *getvar(char *s, int len)
-{
- unsigned uu;
- char **ss = TT.locals;
-
- // loop through local, then global variables
- for (uu = 0; ; uu++) {
- if (!ss[uu]) {
- if (ss != TT.locals) return 0;
- ss = environ;
- uu = 0;
- }
- // Use UHF rubik's cube protocol to find match.
- if (!strncmp(ss[uu], s, len) && ss[uu][len] == '=') return ss[uu]+len+1;
- }
-}
-
-// return length of match found at this point
+// return length of match found at this point (try is null terminated array)
static int anystart(char *s, char **try)
{
char *ss = s;
- while (*try) {
- if (strstart(&s, *try)) return s-ss;
- try++;
- }
+ while (*try) if (strstart(&s, *try++)) return s-ss;
return 0;
}
-// is this one of the strings in try[] (null terminated array)
+// does this entire string match one of the strings in try[]
static int anystr(char *s, char **try)
{
while (*try) if (!strcmp(s, *try++)) return 1;
@@ -546,7 +508,7 @@ static struct sh_process *expand_redir(struct sh_arg *arg, int envlen, int *urd)
else if (*s == '{') {
// when we close a filehandle, we _read_ from {var}, not write to it
if ((!strcmp(ss, "<&") || !strcmp(ss, ">&")) && !strcmp(sss, "-")) {
- if (!(ss = getvar(s+1, ss-s-2))) break;
+ if (!(ss = getvarlen(s+1, ss-s-2))) break;
to = atoi(ss); // TODO trailing garbage?
if (save_redirect(&pp->urd, -1, to)) break;
close(to);
@@ -562,7 +524,7 @@ static struct sh_process *expand_redir(struct sh_arg *arg, int envlen, int *urd)
// HERE documents?
if (!strcmp(ss, "<<<") || !strcmp(ss, "<<-") || !strcmp(ss, "<<")) {
- char *tmp = getvar("TMPDIR", 6);
+ char *tmp = getvar("TMPDIR");
int i, len, zap = (ss[2] == '-'), x = !ss[strcspn(ss, "\"'")];
// store contents in open-but-deleted /tmp file.
@@ -755,8 +717,7 @@ if (BUGBUG) { int i; dprintf(255, "envlen=%d arg->c=%d run=", envlen, arg->c); f
static void free_process(void *ppp)
{
struct sh_process *pp = ppp;
-
- llist_traverse(pp->delete, free);
+ llist_traverse(pp->delete, llist_free_arg);
free(pp);
}
@@ -999,7 +960,7 @@ static int parse_line(char *line, struct sh_function *sp)
// Did we hit end of line or ) outside a function declaration?
// ) is only saved at start of a statement, ends current statement
if (end == start || (arg->c && *start == ')' && pl->type!='f')) {
- arg->v[arg->c] = 0;
+ if (!arg->v) array_add(&arg->v, arg->c, 0);
if (pl->type == 'f' && arg->c<3) {
s = "function()";
@@ -1306,7 +1267,7 @@ static void run_function(struct sh_function *sp)
struct sh_process *pin; // processes piping into this block
int run, loop, *urd, pout;
struct sh_arg farg; // for/select arg stack
- struct string_list *fdelete; // farg's cleanup list
+ struct arg_list *fdelete; // farg's cleanup list
char *fvar; // for/select's iteration variable name
} *blk = 0, *new;
struct sh_process *pplist = 0; // processes piping into current level
@@ -1524,6 +1485,82 @@ dprintf(2, "TODO skipped running for((;;)), need math parser\n");
return;
}
+// Parse and run a self-contained command line with no prompt/continuation
+static int sh_run(char *new)
+{
+ struct sh_function scratch;
+ int rc;
+
+// TODO: parse with len? (End early?)
+
+ if (!parse_line(new, &scratch)) run_function(&scratch);
+ free_function(&scratch);
+ rc = toys.exitval;
+ toys.exitval = 0;
+
+ return rc;
+}
+
+// Print prompt to stderr, parsing escapes
+// Truncated to 4k at the moment, waiting for somebody to complain.
+static void do_prompt(char *prompt)
+{
+ char *s, *ss, c, cc, *pp = toybuf;
+ int len, ll;
+
+ if (!prompt) prompt = "\\$ ";
+ while ((len = sizeof(toybuf)-(pp-toybuf))>0 && *prompt) {
+ c = *(prompt++);
+
+ if (c=='!') {
+ if (*prompt=='!') prompt++;
+ else {
+ pp += snprintf(pp, len, "%ld", TT.lineno);
+ continue;
+ }
+ } else if (c=='\\') {
+ cc = *(prompt++);
+ if (!cc) {
+ *pp++ = c;
+ break;
+ }
+
+ // \nnn \dD{}hHjlstT@AuvVwW!#$
+ // Ignore bash's "nonprintable" hack; query our cursor position instead.
+ if (cc=='[' || cc==']') continue;
+ else if (cc=='$') *pp++ = getuid() ? '$' : '#';
+ else if (cc=='h' || cc=='H') {
+ *pp = 0;
+ gethostname(pp, len);
+ pp[len-1] = 0;
+ if (cc=='h' && (s = strchr(pp, '.'))) *s = 0;
+ pp += strlen(pp);
+ } else if (cc=='s') {
+ s = getbasename(*toys.argv);
+ while (*s && len--) *pp++ = *s++;
+ } else if (cc=='w') {
+ if ((s = getvar("PWD"))) {
+ if ((ss = getvar("HOME")) && strstart(&s, ss)) {
+ *pp++ = '~';
+ if (--len && *s!='/') *pp++ = '/';
+ len--;
+ }
+ if (len>0) {
+ ll = strlen(s);
+ pp = stpncpy(pp, s, ll>len ? len : ll);
+ }
+ }
+ } else if (!(c = unescape(cc))) {
+ *pp++ = '\\';
+ if (--len) *pp++ = c;
+ } else *pp++ = c;
+ } else *pp++ = c;
+ }
+ len = pp-toybuf;
+ if (len>=sizeof(toybuf)) len = sizeof(toybuf);
+ writeall(2, toybuf, len);
+}
+
void subshell_imports(void)
{
/*
@@ -1587,6 +1624,20 @@ void sh_main(void)
TT.hfd = 10;
signal(SIGPIPE, SIG_IGN);
+ // TODO euid stuff?
+
+ // TODO read profile, read rc
+
+ // if (!FLAG(noprofile)) { }
+
+ // Set local variable $HOME to user's login path
+ if (!(new = getenv("HOME"))) {
+ struct passwd *pw = getpwuid(getuid());
+
+ setvar(xmprintf("HOME=%s", (pw && *pw->pw_dir)?pw->pw_dir:"/"), TAKE_MEM);
+ }
+ sh_run("cd .");
+
if (BUGBUG) { int fd = open("/dev/tty", O_RDWR); dup2(fd, 255); close(fd); }
// Is this an interactive shell?
// if (FLAG(i) || (!FLAG(c)&&(FLAG(S)||!toys.optc) && isatty(0) && isatty(1)))
@@ -1633,3 +1684,80 @@ if (BUGBUG) dump_state(&scratch);
toys.exitval = f && ferror(f);
clearerr(stdout);
}
+
+/********************* shell builtin functions *************************/
+
+#define CLEANUP_sh
+#define FOR_cd
+#include "generated/flags.h"
+void cd_main(void)
+{
+ char *home = getvar("HOME"), *pwd = getvar("PWD"), *dd = 0, *from, *to,
+ *dest = (*toys.optargs && **toys.optargs) ? *toys.optargs : "~";
+ int bad = 0;
+
+ // TODO: CDPATH? Really?
+
+ if (!home) home = "/";
+
+ // expand variables
+ if (dest) dd = expand_one_arg(dest, FORCE_COPY|NO_SPLIT, 0);
+ if (!dd || !*dd) {
+ free(dd);
+ dd = xstrdup("/");
+ }
+
+ // prepend cwd or $PWD to relative path
+ if (*dd != '/') {
+ to = 0;
+ from = pwd ? pwd : (to = getcwd(0, 0));
+ if (!from) xsetenv("PWD", "(nowhere)");
+ else {
+ from = xmprintf("%s/%s", from, dd);
+ free(dd);
+ free(to);
+ dd = from;
+ }
+ }
+
+ if (FLAG(P)) {
+ struct stat st;
+ char *pp;
+
+ // Does this directory exist?
+ if ((pp = xabspath(dd, 1)) && stat(pp, &st) && !S_ISDIR(st.st_mode))
+ bad++, errno = ENOTDIR;
+ else {
+ free(dd);
+ dd = pp;
+ }
+ } else {
+
+ // cancel out . and .. in the string
+ for (from = to = dd; *from;) {
+ while (*from=='/' && from[1]=='/') from++;
+ if (*from!='/' || from[1]!='.') *to++ = *from++;
+ else if (!from[2] || from[2]=='/') from += 2;
+ else if (from[2]=='.' && (!from[3] || from[3]=='/')) {
+ from += 3;
+ while (to>dd && *--to != '/');
+ } else *to++ = *from++;
+ }
+ if (to-dd>1 && to[-1]=='/') to--;
+ *to = 0;
+ }
+
+ if (bad || chdir(dd)) perror_msg("chdir '%s'", dd);
+ else {
+ if (pwd) xsetenv("OLD", pwd);
+ xsetenv("PWD", dd);
+ }
+ free(dd);
+}
+
+void exit_main(void)
+{
+ exit(*toys.optargs ? atoi(*toys.optargs) : 0);
+}
+
+