| /* $OpenBSD: exec.c,v 1.52 2015/09/10 22:48:58 nicm Exp $ */ | |
| /*- | |
| * Copyright (c) 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010, | |
| * 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, | |
| * 2019, 2020 | |
| * mirabilos <m@mirbsd.org> | |
| * | |
| * Provided that these terms and disclaimer and all copyright notices | |
| * are retained or reproduced in an accompanying document, permission | |
| * is granted to deal in this work without restriction, including un- | |
| * limited rights to use, publicly perform, distribute, sell, modify, | |
| * merge, give away, or sublicence. | |
| * | |
| * This work is provided "AS IS" and WITHOUT WARRANTY of any kind, to | |
| * the utmost extent permitted by applicable law, neither express nor | |
| * implied; without malicious intent or gross negligence. In no event | |
| * may a licensor, author or contributor be held liable for indirect, | |
| * direct, other damage, loss, or other issues arising in any way out | |
| * of dealing in the work, even if advised of the possibility of such | |
| * damage or existence of a defect, except proven that it results out | |
| * of said person's immediate fault when using the work as intended. | |
| */ | |
| #include "sh.h" | |
| __RCSID("$MirOS: src/bin/mksh/exec.c,v 1.224 2020/08/27 19:52:43 tg Exp $"); | |
| #ifndef MKSH_DEFAULT_EXECSHELL | |
| #define MKSH_DEFAULT_EXECSHELL MKSH_UNIXROOT "/bin/sh" | |
| #endif | |
| static int comexec(struct op *, struct tbl * volatile, const char **, | |
| int volatile, volatile int *); | |
| static void scriptexec(struct op *, const char **) MKSH_A_NORETURN; | |
| static int call_builtin(struct tbl *, const char **, const char *, bool); | |
| static int iosetup(struct ioword *, struct tbl *); | |
| static const char *do_selectargs(const char **, bool); | |
| static Test_op dbteste_isa(Test_env *, Test_meta); | |
| static const char *dbteste_getopnd(Test_env *, Test_op, bool); | |
| static void dbteste_error(Test_env *, int, const char *); | |
| /* XXX: horrible kludge to fit within the framework */ | |
| static void plain_fmt_entry(char *, size_t, unsigned int, const void *); | |
| static void select_fmt_entry(char *, size_t, unsigned int, const void *); | |
| /* | |
| * execute command tree | |
| */ | |
| int | |
| execute(struct op * volatile t, | |
| /* if XEXEC don't fork */ | |
| volatile int flags, | |
| volatile int * volatile xerrok) | |
| { | |
| int i; | |
| volatile int rv = 0, dummy = 0; | |
| int pv[2]; | |
| const char ** volatile ap = NULL; | |
| char ** volatile up; | |
| const char *s, *ccp; | |
| struct ioword **iowp; | |
| struct tbl *tp = NULL; | |
| if (t == NULL) | |
| return (0); | |
| /* Caller doesn't care if XERROK should propagate. */ | |
| if (xerrok == NULL) | |
| xerrok = &dummy; | |
| if ((flags&XFORK) && !(flags&XEXEC) && t->type != TPIPE) | |
| /* run in sub-process */ | |
| return (exchild(t, flags & ~XTIME, xerrok, -1)); | |
| newenv(E_EXEC); | |
| if (trap) | |
| runtraps(0); | |
| /* we want to run an executable, do some variance checks */ | |
| if (t->type == TCOM) { | |
| /* | |
| * Clear subst_exstat before argument expansion. Used by | |
| * null commands (see comexec() and c_eval()) and by c_set(). | |
| */ | |
| subst_exstat = 0; | |
| /* for $LINENO */ | |
| current_lineno = t->lineno; | |
| /* check if this is 'var=<<EOF' */ | |
| if ( | |
| /* we have zero arguments, i.e. no program to run */ | |
| t->args[0] == NULL && | |
| /* we have exactly one variable assignment */ | |
| t->vars[0] != NULL && t->vars[1] == NULL && | |
| /* we have exactly one I/O redirection */ | |
| t->ioact != NULL && t->ioact[0] != NULL && | |
| t->ioact[1] == NULL && | |
| /* of type "here document" (or "here string") */ | |
| (t->ioact[0]->ioflag & IOTYPE) == IOHERE && | |
| /* the variable assignment begins with a valid varname */ | |
| (ccp = skip_wdvarname(t->vars[0], true)) != t->vars[0] && | |
| /* and has no right-hand side (i.e. "varname=") */ | |
| ccp[0] == CHAR && ((ccp[1] == '=' && ccp[2] == EOS) || | |
| /* or "varname+=" */ (ccp[1] == '+' && ccp[2] == CHAR && | |
| ccp[3] == '=' && ccp[4] == EOS))) { | |
| char *cp, *dp; | |
| if ((rv = herein(t->ioact[0], &cp) /*? 1 : 0*/)) | |
| cp = NULL; | |
| strdup2x(dp, evalstr(t->vars[0], DOASNTILDE | DOSCALAR), | |
| rv ? null : cp); | |
| typeset(dp, Flag(FEXPORT) ? EXPORT : 0, 0, 0, 0); | |
| /* free the expanded value */ | |
| afree(cp, APERM); | |
| afree(dp, ATEMP); | |
| goto Break; | |
| } | |
| /* | |
| * POSIX says expand command words first, then redirections, | |
| * and assignments last.. | |
| */ | |
| up = eval(t->args, t->u.evalflags | DOBLANK | DOGLOB | DOTILDE); | |
| if (flags & XTIME) | |
| /* Allow option parsing (bizarre, but POSIX) */ | |
| timex_hook(t, &up); | |
| ap = (const char **)up; | |
| if (ap[0]) | |
| tp = findcom(ap[0], FC_BI | FC_FUNC); | |
| } | |
| flags &= ~XTIME; | |
| if (t->ioact != NULL || t->type == TPIPE || t->type == TCOPROC) { | |
| e->savefd = alloc2(NUFILE, sizeof(short), ATEMP); | |
| /* initialise to not redirected */ | |
| memset(e->savefd, 0, NUFILE * sizeof(short)); | |
| } | |
| /* mark for replacement later (unless TPIPE) */ | |
| vp_pipest->flag |= INT_L; | |
| /* do redirection, to be restored in quitenv() */ | |
| if (t->ioact != NULL) | |
| for (iowp = t->ioact; *iowp != NULL; iowp++) { | |
| if (iosetup(*iowp, tp) < 0) { | |
| exstat = rv = 1; | |
| /* | |
| * Redirection failures for special commands | |
| * cause (non-interactive) shell to exit. | |
| */ | |
| if (tp && tp->type == CSHELL && | |
| (tp->flag & SPEC_BI)) | |
| errorfz(); | |
| /* Deal with FERREXIT, quitenv(), etc. */ | |
| goto Break; | |
| } | |
| } | |
| switch (t->type) { | |
| case TCOM: | |
| rv = comexec(t, tp, (const char **)ap, flags, xerrok); | |
| break; | |
| case TPAREN: | |
| rv = execute(t->left, flags | XFORK, xerrok); | |
| break; | |
| case TPIPE: | |
| flags |= XFORK; | |
| flags &= ~XEXEC; | |
| e->savefd[0] = savefd(0); | |
| e->savefd[1] = savefd(1); | |
| while (t->type == TPIPE) { | |
| openpipe(pv); | |
| /* stdout of curr */ | |
| ksh_dup2(pv[1], 1, false); | |
| /** | |
| * Let exchild() close pv[0] in child | |
| * (if this isn't done, commands like | |
| * (: ; cat /etc/termcap) | sleep 1 | |
| * will hang forever). | |
| */ | |
| exchild(t->left, flags | XPIPEO | XCCLOSE, | |
| NULL, pv[0]); | |
| /* stdin of next */ | |
| ksh_dup2(pv[0], 0, false); | |
| closepipe(pv); | |
| flags |= XPIPEI; | |
| t = t->right; | |
| } | |
| /* stdout of last */ | |
| restfd(1, e->savefd[1]); | |
| /* no need to re-restore this */ | |
| e->savefd[1] = 0; | |
| /* Let exchild() close 0 in parent, after fork, before wait */ | |
| i = exchild(t, flags | XPCLOSE | XPIPEST, xerrok, 0); | |
| if (!(flags&XBGND) && !(flags&XXCOM)) | |
| rv = i; | |
| break; | |
| case TLIST: | |
| while (t->type == TLIST) { | |
| execute(t->left, flags & XERROK, NULL); | |
| t = t->right; | |
| } | |
| rv = execute(t, flags & XERROK, xerrok); | |
| break; | |
| case TCOPROC: { | |
| #ifndef MKSH_NOPROSPECTOFWORK | |
| sigset_t omask; | |
| /* | |
| * Block sigchild as we are using things changed in the | |
| * signal handler | |
| */ | |
| sigprocmask(SIG_BLOCK, &sm_sigchld, &omask); | |
| e->type = E_ERRH; | |
| if ((i = kshsetjmp(e->jbuf))) { | |
| sigprocmask(SIG_SETMASK, &omask, NULL); | |
| quitenv(NULL); | |
| unwind(i); | |
| /* NOTREACHED */ | |
| } | |
| #endif | |
| /* Already have a (live) co-process? */ | |
| if (coproc.job && coproc.write >= 0) | |
| errorf("coprocess already exists"); | |
| /* Can we re-use the existing co-process pipe? */ | |
| coproc_cleanup(true); | |
| /* do this before opening pipes, in case these fail */ | |
| e->savefd[0] = savefd(0); | |
| e->savefd[1] = savefd(1); | |
| openpipe(pv); | |
| if (pv[0] != 0) { | |
| ksh_dup2(pv[0], 0, false); | |
| close(pv[0]); | |
| } | |
| coproc.write = pv[1]; | |
| coproc.job = NULL; | |
| if (coproc.readw >= 0) | |
| ksh_dup2(coproc.readw, 1, false); | |
| else { | |
| openpipe(pv); | |
| coproc.read = pv[0]; | |
| ksh_dup2(pv[1], 1, false); | |
| /* closed before first read */ | |
| coproc.readw = pv[1]; | |
| coproc.njobs = 0; | |
| /* create new coprocess id */ | |
| ++coproc.id; | |
| } | |
| #ifndef MKSH_NOPROSPECTOFWORK | |
| sigprocmask(SIG_SETMASK, &omask, NULL); | |
| /* no more need for error handler */ | |
| e->type = E_EXEC; | |
| #endif | |
| /* | |
| * exchild() closes coproc.* in child after fork, | |
| * will also increment coproc.njobs when the | |
| * job is actually created. | |
| */ | |
| flags &= ~XEXEC; | |
| exchild(t->left, flags | XBGND | XFORK | XCOPROC | XCCLOSE, | |
| NULL, coproc.readw); | |
| break; | |
| } | |
| case TASYNC: | |
| /* | |
| * XXX non-optimal, I think - "(foo &)", forks for (), | |
| * forks again for async... parent should optimise | |
| * this to "foo &"... | |
| */ | |
| rv = execute(t->left, (flags&~XEXEC)|XBGND|XFORK, xerrok); | |
| break; | |
| case TOR: | |
| case TAND: | |
| rv = execute(t->left, XERROK, NULL); | |
| if ((rv == 0) == (t->type == TAND)) | |
| rv = execute(t->right, flags & XERROK, xerrok); | |
| else { | |
| flags |= XERROK; | |
| if (xerrok) | |
| *xerrok = 1; | |
| } | |
| break; | |
| case TBANG: | |
| rv = !execute(t->right, XERROK, xerrok); | |
| flags |= XERROK; | |
| if (xerrok) | |
| *xerrok = 1; | |
| break; | |
| case TDBRACKET: { | |
| Test_env te; | |
| te.flags = TEF_DBRACKET; | |
| te.pos.wp = t->args; | |
| te.isa = dbteste_isa; | |
| te.getopnd = dbteste_getopnd; | |
| te.eval = test_eval; | |
| te.error = dbteste_error; | |
| rv = test_parse(&te); | |
| break; | |
| } | |
| case TFOR: | |
| case TSELECT: { | |
| volatile bool is_first = true; | |
| ap = (t->vars == NULL) ? e->loc->argv + 1 : | |
| (const char **)eval((const char **)t->vars, | |
| DOBLANK | DOGLOB | DOTILDE); | |
| e->type = E_LOOP; | |
| while ((i = kshsetjmp(e->jbuf))) { | |
| if ((e->flags&EF_BRKCONT_PASS) || | |
| (i != LBREAK && i != LCONTIN)) { | |
| quitenv(NULL); | |
| unwind(i); | |
| } else if (i == LBREAK) { | |
| rv = 0; | |
| goto Break; | |
| } | |
| } | |
| /* in case of a continue */ | |
| rv = 0; | |
| if (t->type == TFOR) { | |
| while (*ap != NULL) { | |
| setstr(global(t->str), *ap++, KSH_UNWIND_ERROR); | |
| rv = execute(t->left, flags & XERROK, xerrok); | |
| } | |
| } else { | |
| do_TSELECT: | |
| if ((ccp = do_selectargs(ap, is_first))) { | |
| is_first = false; | |
| setstr(global(t->str), ccp, KSH_UNWIND_ERROR); | |
| execute(t->left, flags & XERROK, xerrok); | |
| goto do_TSELECT; | |
| } | |
| rv = 1; | |
| } | |
| break; | |
| } | |
| case TWHILE: | |
| case TUNTIL: | |
| e->type = E_LOOP; | |
| while ((i = kshsetjmp(e->jbuf))) { | |
| if ((e->flags&EF_BRKCONT_PASS) || | |
| (i != LBREAK && i != LCONTIN)) { | |
| quitenv(NULL); | |
| unwind(i); | |
| } else if (i == LBREAK) { | |
| rv = 0; | |
| goto Break; | |
| } | |
| } | |
| /* in case of a continue */ | |
| rv = 0; | |
| while ((execute(t->left, XERROK, NULL) == 0) == | |
| (t->type == TWHILE)) | |
| rv = execute(t->right, flags & XERROK, xerrok); | |
| break; | |
| case TIF: | |
| case TELIF: | |
| if (t->right == NULL) | |
| /* should be error */ | |
| break; | |
| rv = execute(execute(t->left, XERROK, NULL) == 0 ? | |
| t->right->left : t->right->right, flags & XERROK, xerrok); | |
| break; | |
| case TCASE: | |
| i = 0; | |
| ccp = evalstr(t->str, DOTILDE | DOSCALAR); | |
| for (t = t->left; t != NULL && t->type == TPAT; t = t->right) { | |
| for (ap = (const char **)t->vars; *ap; ap++) { | |
| if (i || ((s = evalstr(*ap, DOTILDE|DOPAT)) && | |
| gmatchx(ccp, s, false))) { | |
| record_match(ccp); | |
| rv = execute(t->left, flags & XERROK, | |
| xerrok); | |
| i = 0; | |
| switch (t->u.charflag) { | |
| case '&': | |
| i = 1; | |
| /* FALLTHROUGH */ | |
| case '|': | |
| goto TCASE_next; | |
| } | |
| goto TCASE_out; | |
| } | |
| } | |
| i = 0; | |
| TCASE_next: | |
| /* empty */; | |
| } | |
| TCASE_out: | |
| break; | |
| case TBRACE: | |
| rv = execute(t->left, flags & XERROK, xerrok); | |
| break; | |
| case TFUNCT: | |
| rv = define(t->str, t); | |
| break; | |
| case TTIME: | |
| /* | |
| * Clear XEXEC so nested execute() call doesn't exit | |
| * (allows "ls -l | time grep foo"). | |
| */ | |
| rv = timex(t, flags & ~XEXEC, xerrok); | |
| break; | |
| case TEXEC: | |
| /* an eval'd TCOM */ | |
| up = makenv(); | |
| restoresigs(); | |
| cleanup_proc_env(); | |
| /* I/O redirection cleanup to be done in child process */ | |
| if (!Flag(FPOSIX) && !Flag(FSH) && t->left->ioact != NULL) | |
| for (iowp = t->left->ioact; *iowp != NULL; iowp++) | |
| if ((*iowp)->ioflag & IODUPSELF) | |
| fcntl((*iowp)->unit, F_SETFD, 0); | |
| /* try to execute */ | |
| { | |
| union mksh_ccphack cargs; | |
| cargs.ro = t->args; | |
| execve(t->str, cargs.rw, up); | |
| rv = errno; | |
| } | |
| if (rv == ENOEXEC) | |
| scriptexec(t, (const char **)up); | |
| else | |
| errorfx(126, Tf_sD_s, t->str, cstrerror(rv)); | |
| } | |
| Break: | |
| exstat = rv & 0xFF; | |
| if (vp_pipest->flag & INT_L) { | |
| unset(vp_pipest, 1); | |
| vp_pipest->flag = DEFINED | ISSET | INTEGER | RDONLY | | |
| ARRAY | INT_U | INT_L; | |
| vp_pipest->val.i = rv; | |
| } | |
| /* restores IO */ | |
| quitenv(NULL); | |
| if ((flags&XEXEC)) | |
| /* exit child */ | |
| unwind(LEXIT); | |
| if (rv != 0 && !(flags & XERROK) && | |
| (xerrok == NULL || !*xerrok)) { | |
| trapsig(ksh_SIGERR); | |
| if (Flag(FERREXIT)) | |
| unwind(LERREXT); | |
| } | |
| return (rv); | |
| } | |
| /* | |
| * execute simple command | |
| */ | |
| static int | |
| comexec(struct op *t, struct tbl * volatile tp, const char **ap, | |
| volatile int flags, volatile int *xerrok) | |
| { | |
| int i; | |
| volatile int rv = 0; | |
| const char *cp; | |
| const char **lastp; | |
| /* Must be static (XXX but why?) */ | |
| static struct op texec; | |
| int type_flags; | |
| bool resetspec; | |
| int fcflags = FC_BI | FC_FUNC | FC_PATH; | |
| struct block *l_expand, *l_assign; | |
| int optc; | |
| const char *exec_argv0 = NULL; | |
| bool exec_clrenv = false; | |
| /* snag the last argument for $_ */ | |
| if (Flag(FTALKING) && *(lastp = ap)) { | |
| /* | |
| * XXX not the same as AT&T ksh, which only seems to set $_ | |
| * after a newline (but not in functions/dot scripts, but in | |
| * interactive and script) - perhaps save last arg here and | |
| * set it in shell()?. | |
| */ | |
| while (*++lastp) | |
| ; | |
| /* setstr() can't fail here */ | |
| setstr(typeset("_", LOCAL, 0, INTEGER, 0), *--lastp, | |
| KSH_RETURN_ERROR); | |
| } | |
| /** | |
| * Deal with the shell builtins builtin, exec and command since | |
| * they can be followed by other commands. This must be done before | |
| * we know if we should create a local block which must be done | |
| * before we can do a path search (in case the assignments change | |
| * PATH). | |
| * Odd cases: | |
| * FOO=bar exec >/dev/null FOO is kept but not exported | |
| * FOO=bar exec foobar FOO is exported | |
| * FOO=bar command exec >/dev/null FOO is neither kept nor exported | |
| * FOO=bar command FOO is neither kept nor exported | |
| * PATH=... foobar use new PATH in foobar search | |
| */ | |
| resetspec = false; | |
| while (tp && tp->type == CSHELL) { | |
| /* undo effects of command */ | |
| fcflags = FC_BI | FC_FUNC | FC_PATH; | |
| if (tp->val.f == c_builtin) { | |
| if ((cp = *++ap) == NULL || | |
| (!strcmp(cp, "--") && (cp = *++ap) == NULL)) { | |
| tp = NULL; | |
| break; | |
| } | |
| if ((tp = findcom(cp, FC_BI)) == NULL) | |
| errorf(Tf_sD_sD_s, Tbuiltin, cp, Tnot_found); | |
| if (tp->type == CSHELL && (tp->flag & LOW_BI)) | |
| break; | |
| continue; | |
| } else if (tp->val.f == c_exec) { | |
| if (ap[1] == NULL) | |
| break; | |
| ksh_getopt_reset(&builtin_opt, GF_ERROR); | |
| while ((optc = ksh_getopt(ap, &builtin_opt, "a:c")) != -1) | |
| switch (optc) { | |
| case 'a': | |
| exec_argv0 = builtin_opt.optarg; | |
| break; | |
| case 'c': | |
| exec_clrenv = true; | |
| /* ensure we can actually do this */ | |
| resetspec = true; | |
| break; | |
| default: | |
| rv = 2; | |
| goto Leave; | |
| } | |
| ap += builtin_opt.optind; | |
| flags |= XEXEC; | |
| /* POSuX demands ksh88-like behaviour here */ | |
| if (Flag(FPOSIX)) | |
| fcflags = FC_PATH; | |
| } else if (tp->val.f == c_command) { | |
| bool saw_p = false; | |
| /* | |
| * Ugly dealing with options in two places (here | |
| * and in c_command(), but such is life) | |
| */ | |
| ksh_getopt_reset(&builtin_opt, 0); | |
| while ((optc = ksh_getopt(ap, &builtin_opt, ":p")) == 'p') | |
| saw_p = true; | |
| if (optc != -1) | |
| /* command -vV or something */ | |
| break; | |
| /* don't look for functions */ | |
| fcflags = FC_BI | FC_PATH; | |
| if (saw_p) { | |
| if (Flag(FRESTRICTED)) { | |
| warningf(true, Tf_sD_s, | |
| "command -p", "restricted"); | |
| rv = 1; | |
| goto Leave; | |
| } | |
| fcflags |= FC_DEFPATH; | |
| } | |
| ap += builtin_opt.optind; | |
| /* | |
| * POSIX says special builtins lose their status | |
| * if accessed using command. | |
| */ | |
| resetspec = true; | |
| if (!ap[0]) { | |
| /* ensure command with no args exits with 0 */ | |
| subst_exstat = 0; | |
| break; | |
| } | |
| } else if (tp->flag & LOW_BI) { | |
| /* if we have any flags, do not use the builtin */ | |
| if ((ap[1] && ap[1][0] == '-' && ap[1][1] != '\0' && | |
| /* argument, begins with -, is not - or -- */ | |
| (ap[1][1] != '-' || ap[1][2] != '\0')) || | |
| /* always prefer the external utility */ | |
| (tp->flag & LOWER_BI)) { | |
| struct tbl *ext_cmd; | |
| ext_cmd = findcom(tp->name, FC_FUNC | FC_PATH); | |
| if (ext_cmd && (ext_cmd->type == CFUNC || | |
| (ext_cmd->flag & ISSET))) | |
| tp = ext_cmd; | |
| } | |
| break; | |
| } else if (tp->val.f == c_trap) { | |
| t->u.evalflags &= ~DOTCOMEXEC; | |
| break; | |
| } else | |
| break; | |
| tp = findcom(ap[0], fcflags & (FC_BI | FC_FUNC)); | |
| } | |
| if (t->u.evalflags & DOTCOMEXEC) | |
| flags |= XEXEC; | |
| l_expand = e->loc; | |
| if (!resetspec && (!ap[0] || (tp && (tp->flag & KEEPASN)))) | |
| type_flags = 0; | |
| else { | |
| /* create new variable/function block */ | |
| newblock(); | |
| /* all functions keep assignments */ | |
| type_flags = LOCAL | LOCAL_COPY | EXPORT; | |
| } | |
| l_assign = e->loc; | |
| if (exec_clrenv) | |
| l_assign->flags |= BF_STOPENV; | |
| if (Flag(FEXPORT)) | |
| type_flags |= EXPORT; | |
| if (Flag(FXTRACE)) | |
| change_xtrace(2, false); | |
| for (i = 0; t->vars[i]; i++) { | |
| /* do NOT lookup in the new var/fn block just created */ | |
| e->loc = l_expand; | |
| cp = evalstr(t->vars[i], DOASNTILDE | DOSCALAR); | |
| e->loc = l_assign; | |
| if (Flag(FXTRACE)) { | |
| const char *ccp; | |
| ccp = skip_varname(cp, true); | |
| if (*ccp == '+') | |
| ++ccp; | |
| if (*ccp == '=') | |
| ++ccp; | |
| shf_write(cp, ccp - cp, shl_xtrace); | |
| print_value_quoted(shl_xtrace, ccp); | |
| shf_putc(' ', shl_xtrace); | |
| } | |
| /* but assign in there as usual */ | |
| typeset(cp, type_flags, 0, 0, 0); | |
| } | |
| if (Flag(FXTRACE)) { | |
| change_xtrace(2, false); | |
| if (ap[rv = 0]) { | |
| xtrace_ap_loop: | |
| print_value_quoted(shl_xtrace, ap[rv]); | |
| if (ap[++rv]) { | |
| shf_putc(' ', shl_xtrace); | |
| goto xtrace_ap_loop; | |
| } | |
| } | |
| change_xtrace(1, false); | |
| } | |
| if ((cp = *ap) == NULL) { | |
| rv = subst_exstat; | |
| goto Leave; | |
| } else if (!tp) { | |
| if (Flag(FRESTRICTED) && mksh_vdirsep(cp)) { | |
| warningf(true, Tf_sD_s, cp, "restricted"); | |
| rv = 1; | |
| goto Leave; | |
| } | |
| tp = findcom(cp, fcflags); | |
| } | |
| switch (tp->type) { | |
| /* shell built-in */ | |
| case CSHELL: | |
| do_call_builtin: | |
| if (l_expand != l_assign) | |
| l_assign->flags |= (tp->flag & NEXTLOC_BI); | |
| rv = call_builtin(tp, (const char **)ap, null, resetspec); | |
| break; | |
| /* function call */ | |
| case CFUNC: { | |
| volatile uint32_t old_inuse; | |
| const char * volatile old_kshname; | |
| volatile uint8_t old_flags[FNFLAGS]; | |
| if (!(tp->flag & ISSET)) { | |
| struct tbl *ftp; | |
| if (!tp->u.fpath) { | |
| fpath_error: | |
| rv = (tp->u2.errnov == ENOENT) ? 127 : 126; | |
| warningf(true, Tf_sD_s_sD_s, cp, | |
| Tcant_find, Tfile_fd, | |
| cstrerror(tp->u2.errnov)); | |
| break; | |
| } | |
| errno = 0; | |
| if (include(tp->u.fpath, 0, NULL, false) < 0 || | |
| !(ftp = findfunc(cp, hash(cp), false)) || | |
| !(ftp->flag & ISSET)) { | |
| rv = errno; | |
| if ((ftp = findcom(cp, FC_BI)) && | |
| (ftp->type == CSHELL) && | |
| (ftp->flag & LOW_BI)) { | |
| tp = ftp; | |
| goto do_call_builtin; | |
| } | |
| if (rv) { | |
| tp->u2.errnov = rv; | |
| cp = tp->u.fpath; | |
| goto fpath_error; | |
| } | |
| warningf(true, Tf_sD_s_s, cp, | |
| "function not defined by", tp->u.fpath); | |
| rv = 127; | |
| break; | |
| } | |
| tp = ftp; | |
| } | |
| /* | |
| * ksh functions set $0 to function name, POSIX | |
| * functions leave $0 unchanged. | |
| */ | |
| old_kshname = kshname; | |
| if (tp->flag & FKSH) | |
| kshname = ap[0]; | |
| else | |
| ap[0] = kshname; | |
| e->loc->argv = ap; | |
| for (i = 0; *ap++ != NULL; i++) | |
| ; | |
| e->loc->argc = i - 1; | |
| /* | |
| * ksh-style functions handle getopts sanely, | |
| * Bourne/POSIX functions are insane... | |
| */ | |
| if (tp->flag & FKSH) { | |
| e->loc->flags |= BF_DOGETOPTS; | |
| e->loc->getopts_state = user_opt; | |
| getopts_reset(1); | |
| } | |
| for (type_flags = 0; type_flags < FNFLAGS; ++type_flags) | |
| old_flags[type_flags] = shell_flags[type_flags]; | |
| change_xtrace((Flag(FXTRACEREC) ? Flag(FXTRACE) : 0) | | |
| ((tp->flag & TRACE) ? 1 : 0), false); | |
| old_inuse = tp->flag & FINUSE; | |
| tp->flag |= FINUSE; | |
| e->type = E_FUNC; | |
| if (!(i = kshsetjmp(e->jbuf))) { | |
| execute(tp->val.t, flags & XERROK, NULL); | |
| i = LRETURN; | |
| } | |
| kshname = old_kshname; | |
| change_xtrace(old_flags[(int)FXTRACE], false); | |
| #ifndef MKSH_LEGACY_MODE | |
| if (tp->flag & FKSH) { | |
| /* Korn style functions restore Flags on return */ | |
| old_flags[(int)FXTRACE] = Flag(FXTRACE); | |
| for (type_flags = 0; type_flags < FNFLAGS; ++type_flags) | |
| shell_flags[type_flags] = old_flags[type_flags]; | |
| } | |
| #endif | |
| tp->flag = (tp->flag & ~FINUSE) | old_inuse; | |
| /* | |
| * Were we deleted while executing? If so, free the | |
| * execution tree. | |
| */ | |
| if ((tp->flag & (FDELETE|FINUSE)) == FDELETE) { | |
| if (tp->flag & ALLOC) { | |
| tp->flag &= ~ALLOC; | |
| tfree(tp->val.t, tp->areap); | |
| } | |
| tp->flag = 0; | |
| } | |
| switch (i) { | |
| case LRETURN: | |
| case LERROR: | |
| case LERREXT: | |
| rv = exstat & 0xFF; | |
| break; | |
| case LINTR: | |
| case LEXIT: | |
| case LLEAVE: | |
| case LSHELL: | |
| quitenv(NULL); | |
| unwind(i); | |
| /* NOTREACHED */ | |
| default: | |
| quitenv(NULL); | |
| internal_errorf(Tunexpected_type, Tunwind, Tfunction, i); | |
| } | |
| break; | |
| } | |
| /* executable command */ | |
| case CEXEC: | |
| /* tracked alias */ | |
| case CTALIAS: | |
| if (!(tp->flag&ISSET)) { | |
| if (tp->u2.errnov == ENOENT) { | |
| rv = 127; | |
| warningf(true, Tf_sD_s_s, cp, | |
| "inaccessible or", Tnot_found); | |
| } else { | |
| rv = 126; | |
| warningf(true, Tf_sD_sD_s, cp, "can't execute", | |
| cstrerror(tp->u2.errnov)); | |
| } | |
| break; | |
| } | |
| /* set $_ to program's full path */ | |
| /* setstr() can't fail here */ | |
| setstr(typeset("_", LOCAL | EXPORT, 0, INTEGER, 0), | |
| tp->val.s, KSH_RETURN_ERROR); | |
| /* to fork, we set up a TEXEC node and call execute */ | |
| texec.type = TEXEC; | |
| /* for vistree/dumptree */ | |
| texec.left = t; | |
| texec.str = tp->val.s; | |
| texec.args = ap; | |
| /* in this case we do not fork, of course */ | |
| if (flags & XEXEC) { | |
| if (exec_argv0) | |
| texec.args[0] = exec_argv0; | |
| j_exit(); | |
| if (!(flags & XBGND) | |
| #ifndef MKSH_UNEMPLOYED | |
| || Flag(FMONITOR) | |
| #endif | |
| ) { | |
| setexecsig(&sigtraps[SIGINT], SS_RESTORE_ORIG); | |
| setexecsig(&sigtraps[SIGQUIT], SS_RESTORE_ORIG); | |
| } | |
| } | |
| rv = exchild(&texec, flags, xerrok, -1); | |
| break; | |
| } | |
| Leave: | |
| if (flags & XEXEC) { | |
| exstat = rv & 0xFF; | |
| unwind(LEXIT); | |
| } | |
| return (rv); | |
| } | |
| static void | |
| scriptexec(struct op *tp, const char **ap) | |
| { | |
| const char *sh; | |
| #ifndef MKSH_SMALL | |
| int fd; | |
| unsigned char buf[68]; | |
| #endif | |
| union mksh_ccphack args, cap; | |
| sh = str_val(global(TEXECSHELL)); | |
| if (sh && *sh) | |
| sh = search_path(sh, path, X_OK, NULL); | |
| if (!sh || !*sh) | |
| sh = MKSH_DEFAULT_EXECSHELL; | |
| *tp->args-- = tp->str; | |
| #ifndef MKSH_SMALL | |
| if ((fd = binopen2(tp->str, O_RDONLY | O_MAYEXEC)) >= 0) { | |
| unsigned char *cp; | |
| #ifndef MKSH_EBCDIC | |
| unsigned short m; | |
| #endif | |
| ssize_t n; | |
| #if defined(__OS2__) && defined(MKSH_WITH_TEXTMODE) | |
| setmode(fd, O_TEXT); | |
| #endif | |
| /* read first couple of octets from file */ | |
| n = read(fd, buf, sizeof(buf) - 1); | |
| close(fd); | |
| /* read error or short read? */ | |
| if (n < 5) | |
| goto nomagic; | |
| /* terminate buffer */ | |
| buf[n] = '\0'; | |
| /* skip UTF-8 Byte Order Mark, if present */ | |
| cp = buf + (n = ((buf[0] == 0xEF) && (buf[1] == 0xBB) && | |
| (buf[2] == 0xBF)) ? 3 : 0); | |
| /* scan for newline or NUL (end of buffer) */ | |
| while (!ctype(*cp, C_NL | C_NUL)) | |
| ++cp; | |
| /* if the shebang line is longer than MAXINTERP, bail out */ | |
| if (!*cp) | |
| goto noshebang; | |
| /* replace newline by NUL */ | |
| *cp = '\0'; | |
| /* restore start of shebang position (buf+0 or buf+3) */ | |
| cp = buf + n; | |
| /* bail out if no shebang magic found */ | |
| if (cp[0] == '#' && cp[1] == '!') | |
| cp += 2; | |
| #ifdef __OS2__ | |
| else if (!strncmp(cp, Textproc, 7) && | |
| ctype(cp[7], C_BLANK)) | |
| cp += 8; | |
| #endif | |
| else | |
| goto noshebang; | |
| /* skip whitespace before shell name */ | |
| while (ctype(*cp, C_BLANK)) | |
| ++cp; | |
| /* just whitespace on the line? */ | |
| if (*cp == '\0') | |
| goto noshebang; | |
| /* no, we actually found an interpreter name */ | |
| sh = (char *)cp; | |
| /* look for end of shell/interpreter name */ | |
| while (!ctype(*cp, C_BLANK | C_NUL)) | |
| ++cp; | |
| /* any arguments? */ | |
| if (*cp) { | |
| *cp++ = '\0'; | |
| /* skip spaces before arguments */ | |
| while (ctype(*cp, C_BLANK)) | |
| ++cp; | |
| /* pass it all in ONE argument (historic reasons) */ | |
| if (*cp) | |
| *tp->args-- = (char *)cp; | |
| } | |
| #ifdef __OS2__ | |
| /* | |
| * On OS/2, the directory structure differs from normal | |
| * Unix, which can make many scripts whose shebang | |
| * hardcodes the path to an interpreter fail (and there | |
| * might be no /usr/bin/env); for user convenience, if | |
| * the specified interpreter is not usable, do a PATH | |
| * search to find it. | |
| */ | |
| if (mksh_vdirsep(sh) && !search_path(sh, path, X_OK, NULL)) { | |
| cp = search_path(_getname(sh), path, X_OK, NULL); | |
| if (cp) | |
| sh = cp; | |
| } | |
| #endif | |
| goto nomagic; | |
| noshebang: | |
| #ifndef MKSH_EBCDIC | |
| m = buf[0] << 8 | buf[1]; | |
| if (m == 0x7F45 && buf[2] == 'L' && buf[3] == 'F') | |
| errorf("%s: not executable: %d-bit ELF file", tp->str, | |
| 32 * buf[4]); | |
| if ((m == /* OMAGIC */ 0407) || | |
| (m == /* NMAGIC */ 0410) || | |
| (m == /* ZMAGIC */ 0413) || | |
| (m == /* QMAGIC */ 0314) || | |
| (m == /* ECOFF_I386 */ 0x4C01) || | |
| (m == /* ECOFF_M68K */ 0x0150 || m == 0x5001) || | |
| (m == /* ECOFF_SH */ 0x0500 || m == 0x0005) || | |
| (m == /* bzip */ 0x425A) || (m == /* "MZ" */ 0x4D5A) || | |
| (m == /* "NE" */ 0x4E45) || (m == /* "LX" */ 0x4C58) || | |
| (m == /* ksh93 */ 0x0B13) || (m == /* LZIP */ 0x4C5A) || | |
| (m == /* xz */ 0xFD37 && buf[2] == 'z' && buf[3] == 'X' && | |
| buf[4] == 'Z') || (m == /* 7zip */ 0x377A) || | |
| (m == /* gzip */ 0x1F8B) || (m == /* .Z */ 0x1F9D)) | |
| errorf("%s: not executable: magic %04X", tp->str, m); | |
| #endif | |
| #ifdef __OS2__ | |
| cp = _getext(tp->str); | |
| if (cp && (!stricmp(cp, ".cmd") || !stricmp(cp, ".bat"))) { | |
| /* execute .cmd and .bat with OS2_SHELL, usually CMD.EXE */ | |
| sh = str_val(global("OS2_SHELL")); | |
| *tp->args-- = "/c"; | |
| /* convert slahes to backslashes */ | |
| for (cp = tp->str; *cp; cp++) { | |
| if (*cp == '/') | |
| *cp = '\\'; | |
| } | |
| } | |
| #endif | |
| nomagic: | |
| ; | |
| } | |
| #endif | |
| args.ro = tp->args; | |
| *args.ro = sh; | |
| cap.ro = ap; | |
| execve(args.rw[0], args.rw, cap.rw); | |
| /* report both the program that was run and the bogus interpreter */ | |
| errorf(Tf_sD_sD_s, tp->str, sh, cstrerror(errno)); | |
| } | |
| /* actual 'builtin' built-in utility call is handled in comexec() */ | |
| int | |
| c_builtin(const char **wp) | |
| { | |
| return (call_builtin(get_builtin(*wp), wp, Tbuiltin, false)); | |
| } | |
| struct tbl * | |
| get_builtin(const char *s) | |
| { | |
| return (s && *s ? ktsearch(&builtins, s, hash(s)) : NULL); | |
| } | |
| /* | |
| * Search function tables for a function. If create set, a table entry | |
| * is created if none is found. | |
| */ | |
| struct tbl * | |
| findfunc(const char *name, uint32_t h, bool create) | |
| { | |
| struct block *l; | |
| struct tbl *tp = NULL; | |
| for (l = e->loc; l; l = l->next) { | |
| tp = ktsearch(&l->funs, name, h); | |
| if (tp) | |
| break; | |
| if (!l->next && create) { | |
| tp = ktenter(&l->funs, name, h); | |
| tp->flag = DEFINED; | |
| tp->type = CFUNC; | |
| tp->val.t = NULL; | |
| break; | |
| } | |
| } | |
| return (tp); | |
| } | |
| /* | |
| * define function. Returns 1 if function is being undefined (t == 0) and | |
| * function did not exist, returns 0 otherwise. | |
| */ | |
| int | |
| define(const char *name, struct op *t) | |
| { | |
| uint32_t nhash; | |
| struct tbl *tp; | |
| bool was_set = false; | |
| nhash = hash(name); | |
| while (/* CONSTCOND */ 1) { | |
| tp = findfunc(name, nhash, true); | |
| if (tp->flag & ISSET) | |
| was_set = true; | |
| /* | |
| * If this function is currently being executed, we zap | |
| * this table entry so findfunc() won't see it | |
| */ | |
| if (tp->flag & FINUSE) { | |
| tp->name[0] = '\0'; | |
| /* ensure it won't be found */ | |
| tp->flag &= ~DEFINED; | |
| tp->flag |= FDELETE; | |
| } else | |
| break; | |
| } | |
| if (tp->flag & ALLOC) { | |
| tp->flag &= ~(ISSET|ALLOC|FKSH); | |
| tfree(tp->val.t, tp->areap); | |
| } | |
| if (t == NULL) { | |
| /* undefine */ | |
| ktdelete(tp); | |
| return (was_set ? 0 : 1); | |
| } | |
| tp->val.t = tcopy(t->left, tp->areap); | |
| tp->flag |= (ISSET|ALLOC); | |
| if (t->u.ksh_func) | |
| tp->flag |= FKSH; | |
| return (0); | |
| } | |
| /* | |
| * add builtin | |
| */ | |
| const char * | |
| builtin(const char *name, int (*func) (const char **)) | |
| { | |
| struct tbl *tp; | |
| uint32_t flag = DEFINED; | |
| /* see if any flags should be set for this builtin */ | |
| flags_loop: | |
| switch (*name) { | |
| case '=': | |
| /* command does variable assignment */ | |
| flag |= KEEPASN; | |
| break; | |
| case '*': | |
| /* POSIX special builtin */ | |
| flag |= SPEC_BI; | |
| break; | |
| case '~': | |
| /* external utility overrides built-in utility, always */ | |
| flag |= LOWER_BI; | |
| /* FALLTHROUGH */ | |
| case '!': | |
| /* external utility overrides built-in utility, with flags */ | |
| flag |= LOW_BI; | |
| break; | |
| case '-': | |
| /* is declaration utility if argv[1] is one (POSIX: command) */ | |
| flag |= DECL_FWDR; | |
| break; | |
| case '^': | |
| /* is declaration utility (POSIX: export, readonly) */ | |
| flag |= DECL_UTIL; | |
| break; | |
| case '#': | |
| /* is set or shift */ | |
| flag |= NEXTLOC_BI; | |
| break; | |
| default: | |
| goto flags_seen; | |
| } | |
| ++name; | |
| goto flags_loop; | |
| flags_seen: | |
| /* enter into the builtins hash table */ | |
| tp = ktenter(&builtins, name, hash(name)); | |
| tp->flag = flag; | |
| tp->type = CSHELL; | |
| tp->val.f = func; | |
| /* return name, for direct builtin call check in main.c */ | |
| return (name); | |
| } | |
| /* | |
| * find command | |
| * either function, hashed command, or built-in (in that order) | |
| */ | |
| struct tbl * | |
| findcom(const char *name, int flags) | |
| { | |
| static struct tbl temp; | |
| uint32_t h = hash(name); | |
| struct tbl *tp = NULL, *tbi; | |
| /* insert if not found */ | |
| unsigned char insert = Flag(FTRACKALL); | |
| /* for function autoloading */ | |
| char *fpath; | |
| union mksh_cchack npath; | |
| if (mksh_vdirsep(name)) { | |
| insert = 0; | |
| /* prevent FPATH search below */ | |
| flags &= ~FC_FUNC; | |
| goto Search; | |
| } | |
| tbi = (flags & FC_BI) ? ktsearch(&builtins, name, h) : NULL; | |
| /* | |
| * POSIX says special builtins first, then functions, then | |
| * regular builtins, then search path... | |
| */ | |
| if ((flags & FC_SPECBI) && tbi && (tbi->flag & SPEC_BI)) | |
| tp = tbi; | |
| if (!tp && (flags & FC_FUNC)) { | |
| tp = findfunc(name, h, false); | |
| if (tp && !(tp->flag & ISSET)) { | |
| if ((fpath = str_val(global(TFPATH))) == null) { | |
| tp->u.fpath = NULL; | |
| tp->u2.errnov = ENOENT; | |
| } else | |
| tp->u.fpath = search_path(name, fpath, R_OK, | |
| &tp->u2.errnov); | |
| } | |
| } | |
| if (!tp && (flags & FC_NORMBI) && tbi) | |
| tp = tbi; | |
| if (!tp && (flags & FC_PATH) && !(flags & FC_DEFPATH)) { | |
| tp = ktsearch(&taliases, name, h); | |
| if (tp && (tp->flag & ISSET) && | |
| ksh_access(tp->val.s, X_OK) != 0) { | |
| if (tp->flag & ALLOC) { | |
| tp->flag &= ~ALLOC; | |
| afree(tp->val.s, APERM); | |
| } | |
| tp->flag &= ~ISSET; | |
| } | |
| } | |
| Search: | |
| if ((!tp || (tp->type == CTALIAS && !(tp->flag & ISSET))) && | |
| (flags & FC_PATH)) { | |
| if (!tp) { | |
| if (insert && !(flags & FC_DEFPATH)) { | |
| tp = ktenter(&taliases, name, h); | |
| tp->type = CTALIAS; | |
| } else { | |
| tp = &temp; | |
| tp->type = CEXEC; | |
| } | |
| /* make ~ISSET */ | |
| tp->flag = DEFINED; | |
| } | |
| npath.ro = search_path(name, | |
| (flags & FC_DEFPATH) ? def_path : path, | |
| X_OK, &tp->u2.errnov); | |
| if (npath.ro) { | |
| strdupx(tp->val.s, npath.ro, APERM); | |
| if (npath.ro != name) | |
| afree(npath.rw, ATEMP); | |
| tp->flag |= ISSET|ALLOC; | |
| } else if ((flags & FC_FUNC) && | |
| (fpath = str_val(global(TFPATH))) != null && | |
| (npath.ro = search_path(name, fpath, R_OK, | |
| &tp->u2.errnov)) != NULL) { | |
| /* | |
| * An undocumented feature of AT&T ksh is that | |
| * it searches FPATH if a command is not found, | |
| * even if the command hasn't been set up as an | |
| * autoloaded function (ie, no typeset -uf). | |
| */ | |
| tp = &temp; | |
| tp->type = CFUNC; | |
| /* make ~ISSET */ | |
| tp->flag = DEFINED; | |
| tp->u.fpath = npath.ro; | |
| } | |
| } | |
| return (tp); | |
| } | |
| /* | |
| * flush executable commands with relative paths | |
| * (just relative or all?) | |
| */ | |
| void | |
| flushcom(bool all) | |
| { | |
| struct tbl *tp; | |
| struct tstate ts; | |
| for (ktwalk(&ts, &taliases); (tp = ktnext(&ts)) != NULL; ) | |
| if ((tp->flag&ISSET) && (all || !mksh_abspath(tp->val.s))) { | |
| if (tp->flag&ALLOC) { | |
| tp->flag &= ~(ALLOC|ISSET); | |
| afree(tp->val.s, APERM); | |
| } | |
| tp->flag &= ~ISSET; | |
| } | |
| } | |
| /* check if path is something we want to find */ | |
| int | |
| search_access(const char *fn, int mode) | |
| { | |
| struct stat sb; | |
| if (stat(fn, &sb) < 0) | |
| /* file does not exist */ | |
| return (ENOENT); | |
| /* LINTED use of access */ | |
| if (access(fn, mode) < 0) { | |
| /* file exists, but we can't access it */ | |
| int eno; | |
| eno = errno; | |
| return (eno ? eno : EACCES); | |
| } | |
| #ifdef __OS2__ | |
| /* treat all files as executable on OS/2 */ | |
| sb.st_mode |= S_IXUSR | S_IXGRP | S_IXOTH; | |
| #endif | |
| if (mode == X_OK && (!S_ISREG(sb.st_mode) || | |
| !(sb.st_mode & (S_IXUSR | S_IXGRP | S_IXOTH)))) | |
| /* access(2) may say root can execute everything */ | |
| return (S_ISDIR(sb.st_mode) ? EISDIR : EACCES); | |
| return (0); | |
| } | |
| #ifdef __OS2__ | |
| /* check if path is something we want to find adding executable extensions */ | |
| #define search_access(fn,mode) access_ex((search_access), (fn), (mode)) | |
| #else | |
| #define search_access(fn,mode) (search_access)((fn), (mode)) | |
| #endif | |
| /* | |
| * search for command with PATH | |
| */ | |
| const char * | |
| search_path(const char *name, const char *lpath, | |
| /* R_OK or X_OK */ | |
| int mode, | |
| /* set if candidate found, but not suitable */ | |
| int *errnop) | |
| { | |
| const char *sp, *p; | |
| char *xp; | |
| XString xs; | |
| size_t namelen; | |
| int ec = 0, ev; | |
| if (mksh_vdirsep(name)) { | |
| if ((ec = search_access(name, mode)) == 0) { | |
| search_path_ok: | |
| if (errnop) | |
| *errnop = 0; | |
| #ifndef __OS2__ | |
| return (name); | |
| #else | |
| return (real_exec_name(name)); | |
| #endif | |
| } | |
| goto search_path_err; | |
| } | |
| namelen = strlen(name) + 1; | |
| Xinit(xs, xp, 128, ATEMP); | |
| sp = lpath; | |
| while (sp != NULL) { | |
| xp = Xstring(xs, xp); | |
| if (!(p = cstrchr(sp, MKSH_PATHSEPC))) | |
| p = strnul(sp); | |
| if (p != sp) { | |
| XcheckN(xs, xp, p - sp); | |
| memcpy(xp, sp, p - sp); | |
| xp += p - sp; | |
| if (mksh_cdirsep(xp[-1])) | |
| xp--; | |
| *xp++ = '/'; | |
| } | |
| sp = p; | |
| XcheckN(xs, xp, namelen); | |
| memcpy(xp, name, namelen); | |
| if ((ev = search_access(Xstring(xs, xp), mode)) == 0) { | |
| name = Xclose(xs, xp + namelen); | |
| goto search_path_ok; | |
| } | |
| /* accumulate non-ENOENT errors only */ | |
| if (ev != ENOENT && ec == 0) | |
| ec = ev; | |
| if (*sp++ == '\0') | |
| sp = NULL; | |
| } | |
| Xfree(xs, xp); | |
| search_path_err: | |
| if (errnop) | |
| *errnop = ec ? ec : ENOENT; | |
| return (NULL); | |
| } | |
| static int | |
| call_builtin(struct tbl *tp, const char **wp, const char *where, bool resetspec) | |
| { | |
| int rv; | |
| if (!tp) | |
| internal_errorf(Tf_sD_s, where, wp[0]); | |
| builtin_argv0 = wp[0]; | |
| builtin_spec = tobool(!resetspec && (tp->flag & SPEC_BI)); | |
| shf_reopen(1, SHF_WR, shl_stdout); | |
| shl_stdout_ok = true; | |
| ksh_getopt_reset(&builtin_opt, GF_ERROR); | |
| rv = (*tp->val.f)(wp); | |
| shf_flush(shl_stdout); | |
| shl_stdout_ok = false; | |
| builtin_argv0 = NULL; | |
| builtin_spec = false; | |
| return (rv); | |
| } | |
| /* | |
| * set up redirection, saving old fds in e->savefd | |
| */ | |
| static int | |
| iosetup(struct ioword *iop, struct tbl *tp) | |
| { | |
| int u = -1; | |
| char *cp = iop->ioname; | |
| int iotype = iop->ioflag & IOTYPE; | |
| bool do_open = true, do_close = false, do_fstat = false; | |
| int flags = 0; | |
| struct ioword iotmp; | |
| struct stat statb; | |
| if (iotype != IOHERE) | |
| cp = evalonestr(cp, DOTILDE|(Flag(FTALKING_I) ? DOGLOB : 0)); | |
| /* Used for tracing and error messages to print expanded cp */ | |
| iotmp = *iop; | |
| iotmp.ioname = (iotype == IOHERE) ? NULL : cp; | |
| iotmp.ioflag |= IONAMEXP; | |
| if (Flag(FXTRACE)) { | |
| change_xtrace(2, false); | |
| fptreef(shl_xtrace, 0, Tft_R, &iotmp); | |
| change_xtrace(1, false); | |
| } | |
| switch (iotype) { | |
| case IOREAD: | |
| flags = O_RDONLY; | |
| break; | |
| case IOCAT: | |
| flags = O_WRONLY | O_APPEND | O_CREAT; | |
| break; | |
| case IOWRITE: | |
| if (Flag(FNOCLOBBER) && !(iop->ioflag & IOCLOB)) { | |
| /* >file under set -C */ | |
| if (stat(cp, &statb)) { | |
| /* nonexistent file */ | |
| flags = O_WRONLY | O_CREAT | O_EXCL; | |
| } else if (S_ISREG(statb.st_mode)) { | |
| /* regular file, refuse clobbering */ | |
| goto clobber_refused; | |
| } else { | |
| /* | |
| * allow redirections to things | |
| * like /dev/null without error | |
| */ | |
| flags = O_WRONLY; | |
| /* but check again after opening */ | |
| do_fstat = true; | |
| } | |
| } else { | |
| /* >|file or set +C */ | |
| flags = O_WRONLY | O_CREAT | O_TRUNC; | |
| } | |
| break; | |
| case IORDWR: | |
| flags = O_RDWR | O_CREAT; | |
| break; | |
| case IOHERE: | |
| do_open = false; | |
| /* herein() returns -2 if error has been printed */ | |
| u = herein(iop, NULL); | |
| /* cp may have wrong name */ | |
| break; | |
| case IODUP: { | |
| const char *emsg; | |
| do_open = false; | |
| if (ksh_isdash(cp)) { | |
| /* prevent error return below */ | |
| u = 1009; | |
| do_close = true; | |
| } else if ((u = check_fd(cp, | |
| X_OK | ((iop->ioflag & IORDUP) ? R_OK : W_OK), | |
| &emsg)) < 0) { | |
| char *sp; | |
| warningf(true, Tf_sD_s, | |
| (sp = snptreef(NULL, 32, Tft_R, &iotmp)), emsg); | |
| afree(sp, ATEMP); | |
| return (-1); | |
| } | |
| if (u == (int)iop->unit) { | |
| /* "dup from" == "dup to" */ | |
| iop->ioflag |= IODUPSELF; | |
| return (0); | |
| } | |
| break; | |
| } | |
| } | |
| if (do_open) { | |
| if (Flag(FRESTRICTED) && (flags & O_CREAT)) { | |
| warningf(true, Tf_sD_s, cp, "restricted"); | |
| return (-1); | |
| } | |
| u = binopen3(cp, flags, 0666); | |
| if (do_fstat && u >= 0) { | |
| /* prevent race conditions */ | |
| if (fstat(u, &statb) || S_ISREG(statb.st_mode)) { | |
| close(u); | |
| clobber_refused: | |
| u = -1; | |
| errno = EEXIST; | |
| } | |
| } | |
| } | |
| if (u < 0) { | |
| /* herein() may already have printed message */ | |
| if (u == -1) { | |
| u = errno; | |
| warningf(true, Tf_cant_ss_s, | |
| #if 0 | |
| /* can't happen */ | |
| iotype == IODUP ? "dup" : | |
| #endif | |
| (iotype == IOREAD || iotype == IOHERE) ? | |
| Topen : Tcreate, cp, cstrerror(u)); | |
| } | |
| return (-1); | |
| } | |
| /* Do not save if it has already been redirected (i.e. "cat >x >y"). */ | |
| if (e->savefd[iop->unit] == 0) { | |
| /* If these are the same, it means unit was previously closed */ | |
| if (u == (int)iop->unit) | |
| e->savefd[iop->unit] = -1; | |
| else | |
| /* | |
| * c_exec() assumes e->savefd[fd] set for any | |
| * redirections. Ask savefd() not to close iop->unit; | |
| * this allows error messages to be seen if iop->unit | |
| * is 2; also means we can't lose the fd (eg, both | |
| * dup2 below and dup2 in restfd() failing). | |
| */ | |
| e->savefd[iop->unit] = savefd(iop->unit); | |
| } | |
| if (do_close) | |
| close(iop->unit); | |
| else if (u != (int)iop->unit) { | |
| if (ksh_dup2(u, iop->unit, true) < 0) { | |
| int eno; | |
| char *sp; | |
| eno = errno; | |
| warningf(true, Tf_s_sD_s, Tredirection_dup, | |
| (sp = snptreef(NULL, 32, Tft_R, &iotmp)), | |
| cstrerror(eno)); | |
| afree(sp, ATEMP); | |
| if (iotype != IODUP) | |
| close(u); | |
| return (-1); | |
| } | |
| if (iotype != IODUP) | |
| close(u); | |
| /* | |
| * Touching any co-process fd in an empty exec | |
| * causes the shell to close its copies | |
| */ | |
| else if (tp && tp->type == CSHELL && tp->val.f == c_exec) { | |
| if (iop->ioflag & IORDUP) | |
| /* possible exec <&p */ | |
| coproc_read_close(u); | |
| else | |
| /* possible exec >&p */ | |
| coproc_write_close(u); | |
| } | |
| } | |
| if (u == 2) | |
| /* Clear any write errors */ | |
| shf_reopen(2, SHF_WR, shl_out); | |
| return (0); | |
| } | |
| /* | |
| * Process here documents by providing the content, either as | |
| * result (globally allocated) string or in a temp file; if | |
| * unquoted, the string is expanded first. | |
| */ | |
| static int | |
| hereinval(struct ioword *iop, int sub, char **resbuf, struct shf *shf) | |
| { | |
| const char * volatile ccp = iop->heredoc; | |
| struct source *s, *osource; | |
| osource = source; | |
| newenv(E_ERRH); | |
| if (kshsetjmp(e->jbuf)) { | |
| source = osource; | |
| quitenv(shf); | |
| /* special to iosetup(): don't print error */ | |
| return (-2); | |
| } | |
| if (iop->ioflag & IOHERESTR) { | |
| ccp = evalstr(iop->delim, DOHERESTR | DOSCALAR); | |
| } else if (sub) { | |
| /* do substitutions on the content of heredoc */ | |
| s = pushs(SSTRING, ATEMP); | |
| s->start = s->str = ccp; | |
| source = s; | |
| if (yylex(sub) != LWORD) | |
| internal_errorf("herein: yylex"); | |
| source = osource; | |
| ccp = evalstr(yylval.cp, DOSCALAR | DOHEREDOC); | |
| } | |
| if (resbuf == NULL) | |
| shf_puts(ccp, shf); | |
| else | |
| strdupx(*resbuf, ccp, APERM); | |
| quitenv(NULL); | |
| return (0); | |
| } | |
| int | |
| herein(struct ioword *iop, char **resbuf) | |
| { | |
| int fd = -1; | |
| struct shf *shf; | |
| struct temp *h; | |
| int i; | |
| /* lexer substitution flags */ | |
| i = (iop->ioflag & IOEVAL) ? (ONEWORD | HEREDOC) : 0; | |
| /* skip all the fd setup if we just want the value */ | |
| if (resbuf != NULL) | |
| return (hereinval(iop, i, resbuf, NULL)); | |
| /* | |
| * Create temp file to hold content (done before newenv | |
| * so temp doesn't get removed too soon). | |
| */ | |
| h = maketemp(ATEMP, TT_HEREDOC_EXP, &e->temps); | |
| if (!(shf = h->shf) || (fd = binopen3(h->tffn, O_RDONLY, 0)) < 0) { | |
| i = errno; | |
| warningf(true, Tf_temp, | |
| !shf ? Tcreate : Topen, h->tffn, cstrerror(i)); | |
| if (shf) | |
| shf_close(shf); | |
| /* special to iosetup(): don't print error */ | |
| return (-2); | |
| } | |
| if (hereinval(iop, i, NULL, shf) == -2) { | |
| close(fd); | |
| /* special to iosetup(): don't print error */ | |
| return (-2); | |
| } | |
| if (shf_close(shf) == -1) { | |
| i = errno; | |
| close(fd); | |
| warningf(true, Tf_temp, | |
| Twrite, h->tffn, cstrerror(i)); | |
| /* special to iosetup(): don't print error */ | |
| return (-2); | |
| } | |
| return (fd); | |
| } | |
| /* | |
| * ksh special - the select command processing section | |
| * print the args in column form - assuming that we can | |
| */ | |
| static const char * | |
| do_selectargs(const char **ap, bool print_menu) | |
| { | |
| static const char *read_args[] = { | |
| Tread, Tdr, TREPLY, NULL | |
| }; | |
| char *s; | |
| int i, argct; | |
| for (argct = 0; ap[argct]; argct++) | |
| ; | |
| while (/* CONSTCOND */ 1) { | |
| /*- | |
| * Menu is printed if | |
| * - this is the first time around the select loop | |
| * - the user enters a blank line | |
| * - the REPLY parameter is empty | |
| */ | |
| if (print_menu || !*str_val(global(TREPLY))) | |
| pr_menu(ap); | |
| shellf(Tf_s, str_val(global("PS3"))); | |
| if (call_builtin(findcom(Tread, FC_BI), read_args, Tselect, | |
| false)) | |
| return (NULL); | |
| if (*(s = str_val(global(TREPLY)))) | |
| return ((getn(s, &i) && i >= 1 && i <= argct) ? | |
| ap[i - 1] : null); | |
| print_menu = true; | |
| } | |
| } | |
| struct select_menu_info { | |
| const char * const *args; | |
| int num_width; | |
| }; | |
| /* format a single select menu item */ | |
| static void | |
| select_fmt_entry(char *buf, size_t buflen, unsigned int i, const void *arg) | |
| { | |
| const struct select_menu_info *smi = | |
| (const struct select_menu_info *)arg; | |
| shf_snprintf(buf, buflen, "%*u) %s", | |
| smi->num_width, i + 1, smi->args[i]); | |
| } | |
| /* | |
| * print a select style menu | |
| */ | |
| void | |
| pr_menu(const char * const *ap) | |
| { | |
| struct select_menu_info smi; | |
| const char * const *pp; | |
| size_t acols = 0, aocts = 0, i; | |
| unsigned int n; | |
| struct columnise_opts co; | |
| /* | |
| * width/column calculations were done once and saved, but this | |
| * means select can't be used recursively so we re-calculate | |
| * each time (could save in a structure that is returned, but | |
| * it's probably not worth the bother) | |
| */ | |
| /* | |
| * get dimensions of the list | |
| */ | |
| for (n = 0, pp = ap; *pp; n++, pp++) { | |
| i = strlen(*pp); | |
| if (i > aocts) | |
| aocts = i; | |
| i = utf_mbswidth(*pp); | |
| if (i > acols) | |
| acols = i; | |
| } | |
| /* | |
| * we will print an index of the form "%d) " in front of | |
| * each entry, so get the maximum width of this | |
| */ | |
| for (i = n, smi.num_width = 1; i >= 10; i /= 10) | |
| smi.num_width++; | |
| smi.args = ap; | |
| co.shf = shl_out; | |
| co.linesep = '\n'; | |
| co.prefcol = co.do_last = true; | |
| print_columns(&co, n, select_fmt_entry, (void *)&smi, | |
| smi.num_width + 2 + aocts, smi.num_width + 2 + acols); | |
| } | |
| static void | |
| plain_fmt_entry(char *buf, size_t buflen, unsigned int i, const void *arg) | |
| { | |
| strlcpy(buf, ((const char * const *)arg)[i], buflen); | |
| } | |
| void | |
| pr_list(struct columnise_opts *cop, char * const *ap) | |
| { | |
| size_t acols = 0, aocts = 0, i; | |
| unsigned int n; | |
| char * const *pp; | |
| for (n = 0, pp = ap; *pp; n++, pp++) { | |
| i = strlen(*pp); | |
| if (i > aocts) | |
| aocts = i; | |
| i = utf_mbswidth(*pp); | |
| if (i > acols) | |
| acols = i; | |
| } | |
| print_columns(cop, n, plain_fmt_entry, (const void *)ap, | |
| aocts, acols); | |
| } | |
| /* | |
| * [[ ... ]] evaluation routines | |
| */ | |
| /* | |
| * Test if the current token is a whatever. Accepts the current token if | |
| * it is. Returns 0 if it is not, non-zero if it is (in the case of | |
| * TM_UNOP and TM_BINOP, the returned value is a Test_op). | |
| */ | |
| static Test_op | |
| dbteste_isa(Test_env *te, Test_meta meta) | |
| { | |
| Test_op ret = TO_NONOP; | |
| bool uqword; | |
| const char *p; | |
| if (!*te->pos.wp) | |
| return (meta == TM_END ? TO_NONNULL : TO_NONOP); | |
| /* unquoted word? */ | |
| for (p = *te->pos.wp; *p == CHAR; p += 2) | |
| ; | |
| uqword = *p == EOS; | |
| if (meta == TM_UNOP || meta == TM_BINOP) { | |
| if (uqword) { | |
| /* longer than the longest operator */ | |
| char buf[8]; | |
| char *q = buf; | |
| p = *te->pos.wp; | |
| while (*p++ == CHAR && | |
| (size_t)(q - buf) < sizeof(buf) - 1) | |
| *q++ = *p++; | |
| *q = '\0'; | |
| ret = test_isop(meta, buf); | |
| } | |
| } else if (meta == TM_END) | |
| ret = TO_NONOP; | |
| else | |
| ret = (uqword && !strcmp(*te->pos.wp, | |
| dbtest_tokens[(int)meta])) ? TO_NONNULL : TO_NONOP; | |
| /* Accept the token? */ | |
| if (ret != TO_NONOP) | |
| te->pos.wp++; | |
| return (ret); | |
| } | |
| static const char * | |
| dbteste_getopnd(Test_env *te, Test_op op, bool do_eval) | |
| { | |
| const char *s = *te->pos.wp; | |
| int flags = DOTILDE | DOSCALAR; | |
| if (!s) | |
| return (NULL); | |
| te->pos.wp++; | |
| if (!do_eval) | |
| return (null); | |
| if (op == TO_STEQL || op == TO_STNEQ) { | |
| flags |= DOPAT; | |
| if (!Flag(FSH)) | |
| flags |= DODBMAGIC; | |
| } | |
| return (evalstr(s, flags)); | |
| } | |
| static void | |
| dbteste_error(Test_env *te, int offset, const char *msg) | |
| { | |
| te->flags |= TEF_ERROR; | |
| internal_warningf("dbteste_error: %s (offset %d)", msg, offset); | |
| } |