1 /* $OpenBSD: exec.c,v 1.52 2015/09/10 22:48:58 nicm Exp $ */
2
3 /*-
4 * Copyright (c) 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010,
5 * 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018,
6 * 2019, 2020
7 * mirabilos <m@mirbsd.org>
8 *
9 * Provided that these terms and disclaimer and all copyright notices
10 * are retained or reproduced in an accompanying document, permission
11 * is granted to deal in this work without restriction, including un-
12 * limited rights to use, publicly perform, distribute, sell, modify,
13 * merge, give away, or sublicence.
14 *
15 * This work is provided "AS IS" and WITHOUT WARRANTY of any kind, to
16 * the utmost extent permitted by applicable law, neither express nor
17 * implied; without malicious intent or gross negligence. In no event
18 * may a licensor, author or contributor be held liable for indirect,
19 * direct, other damage, loss, or other issues arising in any way out
20 * of dealing in the work, even if advised of the possibility of such
21 * damage or existence of a defect, except proven that it results out
22 * of said person's immediate fault when using the work as intended.
23 */
24
25 #include "sh.h"
26
27 __RCSID("$MirOS: src/bin/mksh/exec.c,v 1.224 2020/08/27 19:52:43 tg Exp $");
28
29 #ifndef MKSH_DEFAULT_EXECSHELL
30 #define MKSH_DEFAULT_EXECSHELL MKSH_UNIXROOT "/bin/sh"
31 #endif
32
33 static int comexec(struct op *, struct tbl * volatile, const char **,
34 int volatile, volatile int *);
35 static void scriptexec(struct op *, const char **) MKSH_A_NORETURN;
36 static int call_builtin(struct tbl *, const char **, const char *, bool);
37 static int iosetup(struct ioword *, struct tbl *);
38 static const char *do_selectargs(const char **, bool);
39 static Test_op dbteste_isa(Test_env *, Test_meta);
40 static const char *dbteste_getopnd(Test_env *, Test_op, bool);
41 static void dbteste_error(Test_env *, int, const char *);
42 /* XXX: horrible kludge to fit within the framework */
43 static void plain_fmt_entry(char *, size_t, unsigned int, const void *);
44 static void select_fmt_entry(char *, size_t, unsigned int, const void *);
45
46 /*
47 * execute command tree
48 */
49 int
execute(struct op * volatile t,volatile int flags,volatile int * volatile xerrok)50 execute(struct op * volatile t,
51 /* if XEXEC don't fork */
52 volatile int flags,
53 volatile int * volatile xerrok)
54 {
55 int i;
56 volatile int rv = 0, dummy = 0;
57 int pv[2];
58 const char ** volatile ap = NULL;
59 char ** volatile up;
60 const char *s, *ccp;
61 struct ioword **iowp;
62 struct tbl *tp = NULL;
63
64 if (t == NULL)
65 return (0);
66
67 /* Caller doesn't care if XERROK should propagate. */
68 if (xerrok == NULL)
69 xerrok = &dummy;
70
71 if ((flags&XFORK) && !(flags&XEXEC) && t->type != TPIPE)
72 /* run in sub-process */
73 return (exchild(t, flags & ~XTIME, xerrok, -1));
74
75 newenv(E_EXEC);
76 if (trap)
77 runtraps(0);
78
79 /* we want to run an executable, do some variance checks */
80 if (t->type == TCOM) {
81 /*
82 * Clear subst_exstat before argument expansion. Used by
83 * null commands (see comexec() and c_eval()) and by c_set().
84 */
85 subst_exstat = 0;
86
87 /* for $LINENO */
88 current_lineno = t->lineno;
89
90 /* check if this is 'var=<<EOF' */
91 if (
92 /* we have zero arguments, i.e. no program to run */
93 t->args[0] == NULL &&
94 /* we have exactly one variable assignment */
95 t->vars[0] != NULL && t->vars[1] == NULL &&
96 /* we have exactly one I/O redirection */
97 t->ioact != NULL && t->ioact[0] != NULL &&
98 t->ioact[1] == NULL &&
99 /* of type "here document" (or "here string") */
100 (t->ioact[0]->ioflag & IOTYPE) == IOHERE &&
101 /* the variable assignment begins with a valid varname */
102 (ccp = skip_wdvarname(t->vars[0], true)) != t->vars[0] &&
103 /* and has no right-hand side (i.e. "varname=") */
104 ccp[0] == CHAR && ((ccp[1] == '=' && ccp[2] == EOS) ||
105 /* or "varname+=" */ (ccp[1] == '+' && ccp[2] == CHAR &&
106 ccp[3] == '=' && ccp[4] == EOS))) {
107 char *cp, *dp;
108
109 if ((rv = herein(t->ioact[0], &cp) /*? 1 : 0*/))
110 cp = NULL;
111 strdup2x(dp, evalstr(t->vars[0], DOASNTILDE | DOSCALAR),
112 rv ? null : cp);
113 typeset(dp, Flag(FEXPORT) ? EXPORT : 0, 0, 0, 0);
114 /* free the expanded value */
115 afree(cp, APERM);
116 afree(dp, ATEMP);
117 goto Break;
118 }
119
120 /*
121 * POSIX says expand command words first, then redirections,
122 * and assignments last..
123 */
124 up = eval(t->args, t->u.evalflags | DOBLANK | DOGLOB | DOTILDE);
125 if (flags & XTIME)
126 /* Allow option parsing (bizarre, but POSIX) */
127 timex_hook(t, &up);
128 ap = (const char **)up;
129 if (ap[0])
130 tp = findcom(ap[0], FC_BI | FC_FUNC);
131 }
132 flags &= ~XTIME;
133
134 if (t->ioact != NULL || t->type == TPIPE || t->type == TCOPROC) {
135 e->savefd = alloc2(NUFILE, sizeof(short), ATEMP);
136 /* initialise to not redirected */
137 memset(e->savefd, 0, NUFILE * sizeof(short));
138 }
139
140 /* mark for replacement later (unless TPIPE) */
141 vp_pipest->flag |= INT_L;
142
143 /* do redirection, to be restored in quitenv() */
144 if (t->ioact != NULL)
145 for (iowp = t->ioact; *iowp != NULL; iowp++) {
146 if (iosetup(*iowp, tp) < 0) {
147 exstat = rv = 1;
148 /*
149 * Redirection failures for special commands
150 * cause (non-interactive) shell to exit.
151 */
152 if (tp && tp->type == CSHELL &&
153 (tp->flag & SPEC_BI))
154 errorfz();
155 /* Deal with FERREXIT, quitenv(), etc. */
156 goto Break;
157 }
158 }
159
160 switch (t->type) {
161 case TCOM:
162 rv = comexec(t, tp, (const char **)ap, flags, xerrok);
163 break;
164
165 case TPAREN:
166 rv = execute(t->left, flags | XFORK, xerrok);
167 break;
168
169 case TPIPE:
170 flags |= XFORK;
171 flags &= ~XEXEC;
172 e->savefd[0] = savefd(0);
173 e->savefd[1] = savefd(1);
174 while (t->type == TPIPE) {
175 openpipe(pv);
176 /* stdout of curr */
177 ksh_dup2(pv[1], 1, false);
178 /**
179 * Let exchild() close pv[0] in child
180 * (if this isn't done, commands like
181 * (: ; cat /etc/termcap) | sleep 1
182 * will hang forever).
183 */
184 exchild(t->left, flags | XPIPEO | XCCLOSE,
185 NULL, pv[0]);
186 /* stdin of next */
187 ksh_dup2(pv[0], 0, false);
188 closepipe(pv);
189 flags |= XPIPEI;
190 t = t->right;
191 }
192 /* stdout of last */
193 restfd(1, e->savefd[1]);
194 /* no need to re-restore this */
195 e->savefd[1] = 0;
196 /* Let exchild() close 0 in parent, after fork, before wait */
197 i = exchild(t, flags | XPCLOSE | XPIPEST, xerrok, 0);
198 if (!(flags&XBGND) && !(flags&XXCOM))
199 rv = i;
200 break;
201
202 case TLIST:
203 while (t->type == TLIST) {
204 execute(t->left, flags & XERROK, NULL);
205 t = t->right;
206 }
207 rv = execute(t, flags & XERROK, xerrok);
208 break;
209
210 case TCOPROC: {
211 #ifndef MKSH_NOPROSPECTOFWORK
212 sigset_t omask;
213
214 /*
215 * Block sigchild as we are using things changed in the
216 * signal handler
217 */
218 sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
219 e->type = E_ERRH;
220 if ((i = kshsetjmp(e->jbuf))) {
221 sigprocmask(SIG_SETMASK, &omask, NULL);
222 quitenv(NULL);
223 unwind(i);
224 /* NOTREACHED */
225 }
226 #endif
227 /* Already have a (live) co-process? */
228 if (coproc.job && coproc.write >= 0)
229 errorf("coprocess already exists");
230
231 /* Can we re-use the existing co-process pipe? */
232 coproc_cleanup(true);
233
234 /* do this before opening pipes, in case these fail */
235 e->savefd[0] = savefd(0);
236 e->savefd[1] = savefd(1);
237
238 openpipe(pv);
239 if (pv[0] != 0) {
240 ksh_dup2(pv[0], 0, false);
241 close(pv[0]);
242 }
243 coproc.write = pv[1];
244 coproc.job = NULL;
245
246 if (coproc.readw >= 0)
247 ksh_dup2(coproc.readw, 1, false);
248 else {
249 openpipe(pv);
250 coproc.read = pv[0];
251 ksh_dup2(pv[1], 1, false);
252 /* closed before first read */
253 coproc.readw = pv[1];
254 coproc.njobs = 0;
255 /* create new coprocess id */
256 ++coproc.id;
257 }
258 #ifndef MKSH_NOPROSPECTOFWORK
259 sigprocmask(SIG_SETMASK, &omask, NULL);
260 /* no more need for error handler */
261 e->type = E_EXEC;
262 #endif
263
264 /*
265 * exchild() closes coproc.* in child after fork,
266 * will also increment coproc.njobs when the
267 * job is actually created.
268 */
269 flags &= ~XEXEC;
270 exchild(t->left, flags | XBGND | XFORK | XCOPROC | XCCLOSE,
271 NULL, coproc.readw);
272 break;
273 }
274
275 case TASYNC:
276 /*
277 * XXX non-optimal, I think - "(foo &)", forks for (),
278 * forks again for async... parent should optimise
279 * this to "foo &"...
280 */
281 rv = execute(t->left, (flags&~XEXEC)|XBGND|XFORK, xerrok);
282 break;
283
284 case TOR:
285 case TAND:
286 rv = execute(t->left, XERROK, NULL);
287 if ((rv == 0) == (t->type == TAND))
288 rv = execute(t->right, flags & XERROK, xerrok);
289 else {
290 flags |= XERROK;
291 if (xerrok)
292 *xerrok = 1;
293 }
294 break;
295
296 case TBANG:
297 rv = !execute(t->right, XERROK, xerrok);
298 flags |= XERROK;
299 if (xerrok)
300 *xerrok = 1;
301 break;
302
303 case TDBRACKET: {
304 Test_env te;
305
306 te.flags = TEF_DBRACKET;
307 te.pos.wp = t->args;
308 te.isa = dbteste_isa;
309 te.getopnd = dbteste_getopnd;
310 te.eval = test_eval;
311 te.error = dbteste_error;
312
313 rv = test_parse(&te);
314 break;
315 }
316
317 case TFOR:
318 case TSELECT: {
319 volatile bool is_first = true;
320
321 ap = (t->vars == NULL) ? e->loc->argv + 1 :
322 (const char **)eval((const char **)t->vars,
323 DOBLANK | DOGLOB | DOTILDE);
324 e->type = E_LOOP;
325 while ((i = kshsetjmp(e->jbuf))) {
326 if ((e->flags&EF_BRKCONT_PASS) ||
327 (i != LBREAK && i != LCONTIN)) {
328 quitenv(NULL);
329 unwind(i);
330 } else if (i == LBREAK) {
331 rv = 0;
332 goto Break;
333 }
334 }
335 /* in case of a continue */
336 rv = 0;
337 if (t->type == TFOR) {
338 while (*ap != NULL) {
339 setstr(global(t->str), *ap++, KSH_UNWIND_ERROR);
340 rv = execute(t->left, flags & XERROK, xerrok);
341 }
342 } else {
343 do_TSELECT:
344 if ((ccp = do_selectargs(ap, is_first))) {
345 is_first = false;
346 setstr(global(t->str), ccp, KSH_UNWIND_ERROR);
347 execute(t->left, flags & XERROK, xerrok);
348 goto do_TSELECT;
349 }
350 rv = 1;
351 }
352 break;
353 }
354
355 case TWHILE:
356 case TUNTIL:
357 e->type = E_LOOP;
358 while ((i = kshsetjmp(e->jbuf))) {
359 if ((e->flags&EF_BRKCONT_PASS) ||
360 (i != LBREAK && i != LCONTIN)) {
361 quitenv(NULL);
362 unwind(i);
363 } else if (i == LBREAK) {
364 rv = 0;
365 goto Break;
366 }
367 }
368 /* in case of a continue */
369 rv = 0;
370 while ((execute(t->left, XERROK, NULL) == 0) ==
371 (t->type == TWHILE))
372 rv = execute(t->right, flags & XERROK, xerrok);
373 break;
374
375 case TIF:
376 case TELIF:
377 if (t->right == NULL)
378 /* should be error */
379 break;
380 rv = execute(execute(t->left, XERROK, NULL) == 0 ?
381 t->right->left : t->right->right, flags & XERROK, xerrok);
382 break;
383
384 case TCASE:
385 i = 0;
386 ccp = evalstr(t->str, DOTILDE | DOSCALAR);
387 for (t = t->left; t != NULL && t->type == TPAT; t = t->right) {
388 for (ap = (const char **)t->vars; *ap; ap++) {
389 if (i || ((s = evalstr(*ap, DOTILDE|DOPAT)) &&
390 gmatchx(ccp, s, false))) {
391 record_match(ccp);
392 rv = execute(t->left, flags & XERROK,
393 xerrok);
394 i = 0;
395 switch (t->u.charflag) {
396 case '&':
397 i = 1;
398 /* FALLTHROUGH */
399 case '|':
400 goto TCASE_next;
401 }
402 goto TCASE_out;
403 }
404 }
405 i = 0;
406 TCASE_next:
407 /* empty */;
408 }
409 TCASE_out:
410 break;
411
412 case TBRACE:
413 rv = execute(t->left, flags & XERROK, xerrok);
414 break;
415
416 case TFUNCT:
417 rv = define(t->str, t);
418 break;
419
420 case TTIME:
421 /*
422 * Clear XEXEC so nested execute() call doesn't exit
423 * (allows "ls -l | time grep foo").
424 */
425 rv = timex(t, flags & ~XEXEC, xerrok);
426 break;
427
428 case TEXEC:
429 /* an eval'd TCOM */
430 up = makenv();
431 restoresigs();
432 cleanup_proc_env();
433 /* I/O redirection cleanup to be done in child process */
434 if (!Flag(FPOSIX) && !Flag(FSH) && t->left->ioact != NULL)
435 for (iowp = t->left->ioact; *iowp != NULL; iowp++)
436 if ((*iowp)->ioflag & IODUPSELF)
437 fcntl((*iowp)->unit, F_SETFD, 0);
438 /* try to execute */
439 {
440 union mksh_ccphack cargs;
441
442 cargs.ro = t->args;
443 execve(t->str, cargs.rw, up);
444 rv = errno;
445 }
446 if (rv == ENOEXEC)
447 scriptexec(t, (const char **)up);
448 else
449 errorfx(126, Tf_sD_s, t->str, cstrerror(rv));
450 }
451 Break:
452 exstat = rv & 0xFF;
453 if (vp_pipest->flag & INT_L) {
454 unset(vp_pipest, 1);
455 vp_pipest->flag = DEFINED | ISSET | INTEGER | RDONLY |
456 ARRAY | INT_U | INT_L;
457 vp_pipest->val.i = rv;
458 }
459
460 /* restores IO */
461 quitenv(NULL);
462 if ((flags&XEXEC))
463 /* exit child */
464 unwind(LEXIT);
465 if (rv != 0 && !(flags & XERROK) &&
466 (xerrok == NULL || !*xerrok)) {
467 trapsig(ksh_SIGERR);
468 if (Flag(FERREXIT))
469 unwind(LERREXT);
470 }
471 return (rv);
472 }
473
474 /*
475 * execute simple command
476 */
477
478 static int
comexec(struct op * t,struct tbl * volatile tp,const char ** ap,volatile int flags,volatile int * xerrok)479 comexec(struct op *t, struct tbl * volatile tp, const char **ap,
480 volatile int flags, volatile int *xerrok)
481 {
482 int i;
483 volatile int rv = 0;
484 const char *cp;
485 const char **lastp;
486 /* Must be static (XXX but why?) */
487 static struct op texec;
488 int type_flags;
489 bool resetspec;
490 int fcflags = FC_BI | FC_FUNC | FC_PATH;
491 struct block *l_expand, *l_assign;
492 int optc;
493 const char *exec_argv0 = NULL;
494 bool exec_clrenv = false;
495
496 /* snag the last argument for $_ */
497 if (Flag(FTALKING) && *(lastp = ap)) {
498 /*
499 * XXX not the same as AT&T ksh, which only seems to set $_
500 * after a newline (but not in functions/dot scripts, but in
501 * interactive and script) - perhaps save last arg here and
502 * set it in shell()?.
503 */
504 while (*++lastp)
505 ;
506 /* setstr() can't fail here */
507 setstr(typeset("_", LOCAL, 0, INTEGER, 0), *--lastp,
508 KSH_RETURN_ERROR);
509 }
510
511 /**
512 * Deal with the shell builtins builtin, exec and command since
513 * they can be followed by other commands. This must be done before
514 * we know if we should create a local block which must be done
515 * before we can do a path search (in case the assignments change
516 * PATH).
517 * Odd cases:
518 * FOO=bar exec >/dev/null FOO is kept but not exported
519 * FOO=bar exec foobar FOO is exported
520 * FOO=bar command exec >/dev/null FOO is neither kept nor exported
521 * FOO=bar command FOO is neither kept nor exported
522 * PATH=... foobar use new PATH in foobar search
523 */
524 resetspec = false;
525 while (tp && tp->type == CSHELL) {
526 /* undo effects of command */
527 fcflags = FC_BI | FC_FUNC | FC_PATH;
528 if (tp->val.f == c_builtin) {
529 if ((cp = *++ap) == NULL ||
530 (!strcmp(cp, "--") && (cp = *++ap) == NULL)) {
531 tp = NULL;
532 break;
533 }
534 if ((tp = findcom(cp, FC_BI)) == NULL)
535 errorf(Tf_sD_sD_s, Tbuiltin, cp, Tnot_found);
536 if (tp->type == CSHELL && (tp->flag & LOW_BI))
537 break;
538 continue;
539 } else if (tp->val.f == c_exec) {
540 if (ap[1] == NULL)
541 break;
542 ksh_getopt_reset(&builtin_opt, GF_ERROR);
543 while ((optc = ksh_getopt(ap, &builtin_opt, "a:c")) != -1)
544 switch (optc) {
545 case 'a':
546 exec_argv0 = builtin_opt.optarg;
547 break;
548 case 'c':
549 exec_clrenv = true;
550 /* ensure we can actually do this */
551 resetspec = true;
552 break;
553 default:
554 rv = 2;
555 goto Leave;
556 }
557 ap += builtin_opt.optind;
558 flags |= XEXEC;
559 /* POSuX demands ksh88-like behaviour here */
560 if (Flag(FPOSIX))
561 fcflags = FC_PATH;
562 #ifndef MKSH_LESS_BUILDINS
563 } else if (tp->val.f == c_command) {
564 bool saw_p = false;
565
566 /*
567 * Ugly dealing with options in two places (here
568 * and in c_command(), but such is life)
569 */
570 ksh_getopt_reset(&builtin_opt, 0);
571 while ((optc = ksh_getopt(ap, &builtin_opt, ":p")) == 'p')
572 saw_p = true;
573 if (optc != -1)
574 /* command -vV or something */
575 break;
576 /* don't look for functions */
577 fcflags = FC_BI | FC_PATH;
578 if (saw_p) {
579 if (Flag(FRESTRICTED)) {
580 warningf(true, Tf_sD_s,
581 "command -p", "restricted");
582 rv = 1;
583 goto Leave;
584 }
585 fcflags |= FC_DEFPATH;
586 }
587 ap += builtin_opt.optind;
588 /*
589 * POSIX says special builtins lose their status
590 * if accessed using command.
591 */
592 resetspec = true;
593 if (!ap[0]) {
594 /* ensure command with no args exits with 0 */
595 subst_exstat = 0;
596 break;
597 }
598 #endif // MKSH_LESS_BUILDINS
599 } else if (tp->flag & LOW_BI) {
600 /* if we have any flags, do not use the builtin */
601 if ((ap[1] && ap[1][0] == '-' && ap[1][1] != '\0' &&
602 /* argument, begins with -, is not - or -- */
603 (ap[1][1] != '-' || ap[1][2] != '\0')) ||
604 /* always prefer the external utility */
605 (tp->flag & LOWER_BI)) {
606 struct tbl *ext_cmd;
607
608 ext_cmd = findcom(tp->name, FC_FUNC | FC_PATH);
609 if (ext_cmd && (ext_cmd->type == CFUNC ||
610 (ext_cmd->flag & ISSET)))
611 tp = ext_cmd;
612 }
613 break;
614 } else if (tp->val.f == c_trap) {
615 t->u.evalflags &= ~DOTCOMEXEC;
616 break;
617 } else
618 break;
619 tp = findcom(ap[0], fcflags & (FC_BI | FC_FUNC));
620 }
621 if (t->u.evalflags & DOTCOMEXEC)
622 flags |= XEXEC;
623 l_expand = e->loc;
624 if (!resetspec && (!ap[0] || (tp && (tp->flag & KEEPASN))))
625 type_flags = 0;
626 else {
627 /* create new variable/function block */
628 newblock();
629 /* all functions keep assignments */
630 type_flags = LOCAL | LOCAL_COPY | EXPORT;
631 }
632 l_assign = e->loc;
633 if (exec_clrenv)
634 l_assign->flags |= BF_STOPENV;
635 if (Flag(FEXPORT))
636 type_flags |= EXPORT;
637 if (Flag(FXTRACE))
638 change_xtrace(2, false);
639 for (i = 0; t->vars[i]; i++) {
640 /* do NOT lookup in the new var/fn block just created */
641 e->loc = l_expand;
642 cp = evalstr(t->vars[i], DOASNTILDE | DOSCALAR);
643 e->loc = l_assign;
644 if (Flag(FXTRACE)) {
645 const char *ccp;
646
647 ccp = skip_varname(cp, true);
648 if (*ccp == '+')
649 ++ccp;
650 if (*ccp == '=')
651 ++ccp;
652 shf_write(cp, ccp - cp, shl_xtrace);
653 print_value_quoted(shl_xtrace, ccp);
654 shf_putc(' ', shl_xtrace);
655 }
656 /* but assign in there as usual */
657 typeset(cp, type_flags, 0, 0, 0);
658 }
659
660 if (Flag(FXTRACE)) {
661 change_xtrace(2, false);
662 if (ap[rv = 0]) {
663 xtrace_ap_loop:
664 print_value_quoted(shl_xtrace, ap[rv]);
665 if (ap[++rv]) {
666 shf_putc(' ', shl_xtrace);
667 goto xtrace_ap_loop;
668 }
669 }
670 change_xtrace(1, false);
671 }
672
673 if ((cp = *ap) == NULL) {
674 rv = subst_exstat;
675 goto Leave;
676 } else if (!tp) {
677 if (Flag(FRESTRICTED) && mksh_vdirsep(cp)) {
678 warningf(true, Tf_sD_s, cp, "restricted");
679 rv = 1;
680 goto Leave;
681 }
682 tp = findcom(cp, fcflags);
683 }
684
685 switch (tp->type) {
686
687 /* shell built-in */
688 case CSHELL:
689 do_call_builtin:
690 if (l_expand != l_assign)
691 l_assign->flags |= (tp->flag & NEXTLOC_BI);
692 rv = call_builtin(tp, (const char **)ap, null, resetspec);
693 break;
694
695 /* function call */
696 case CFUNC: {
697 volatile uint32_t old_inuse;
698 const char * volatile old_kshname;
699 volatile uint8_t old_flags[FNFLAGS];
700
701 if (!(tp->flag & ISSET)) {
702 struct tbl *ftp;
703
704 if (!tp->u.fpath) {
705 fpath_error:
706 rv = (tp->u2.errnov == ENOENT) ? 127 : 126;
707 warningf(true, Tf_sD_s_sD_s, cp,
708 Tcant_find, Tfile_fd,
709 cstrerror(tp->u2.errnov));
710 break;
711 }
712 errno = 0;
713 if (include(tp->u.fpath, 0, NULL, false) < 0 ||
714 !(ftp = findfunc(cp, hash(cp), false)) ||
715 !(ftp->flag & ISSET)) {
716 rv = errno;
717 if ((ftp = findcom(cp, FC_BI)) &&
718 (ftp->type == CSHELL) &&
719 (ftp->flag & LOW_BI)) {
720 tp = ftp;
721 goto do_call_builtin;
722 }
723 if (rv) {
724 tp->u2.errnov = rv;
725 cp = tp->u.fpath;
726 goto fpath_error;
727 }
728 warningf(true, Tf_sD_s_s, cp,
729 "function not defined by", tp->u.fpath);
730 rv = 127;
731 break;
732 }
733 tp = ftp;
734 }
735
736 /*
737 * ksh functions set $0 to function name, POSIX
738 * functions leave $0 unchanged.
739 */
740 old_kshname = kshname;
741 if (tp->flag & FKSH)
742 kshname = ap[0];
743 else
744 ap[0] = kshname;
745 e->loc->argv = ap;
746 for (i = 0; *ap++ != NULL; i++)
747 ;
748 e->loc->argc = i - 1;
749 /*
750 * ksh-style functions handle getopts sanely,
751 * Bourne/POSIX functions are insane...
752 */
753 if (tp->flag & FKSH) {
754 e->loc->flags |= BF_DOGETOPTS;
755 e->loc->getopts_state = user_opt;
756 getopts_reset(1);
757 }
758
759 for (type_flags = 0; type_flags < FNFLAGS; ++type_flags)
760 old_flags[type_flags] = shell_flags[type_flags];
761 change_xtrace((Flag(FXTRACEREC) ? Flag(FXTRACE) : 0) |
762 ((tp->flag & TRACE) ? 1 : 0), false);
763 old_inuse = tp->flag & FINUSE;
764 tp->flag |= FINUSE;
765
766 e->type = E_FUNC;
767 if (!(i = kshsetjmp(e->jbuf))) {
768 execute(tp->val.t, flags & XERROK, NULL);
769 i = LRETURN;
770 }
771
772 kshname = old_kshname;
773 change_xtrace(old_flags[(int)FXTRACE], false);
774 #ifndef MKSH_LEGACY_MODE
775 if (tp->flag & FKSH) {
776 /* Korn style functions restore Flags on return */
777 old_flags[(int)FXTRACE] = Flag(FXTRACE);
778 for (type_flags = 0; type_flags < FNFLAGS; ++type_flags)
779 shell_flags[type_flags] = old_flags[type_flags];
780 }
781 #endif
782 tp->flag = (tp->flag & ~FINUSE) | old_inuse;
783
784 /*
785 * Were we deleted while executing? If so, free the
786 * execution tree.
787 */
788 if ((tp->flag & (FDELETE|FINUSE)) == FDELETE) {
789 if (tp->flag & ALLOC) {
790 tp->flag &= ~ALLOC;
791 tfree(tp->val.t, tp->areap);
792 }
793 tp->flag = 0;
794 }
795 switch (i) {
796 case LRETURN:
797 case LERROR:
798 case LERREXT:
799 rv = exstat & 0xFF;
800 break;
801 case LINTR:
802 case LEXIT:
803 case LLEAVE:
804 case LSHELL:
805 quitenv(NULL);
806 unwind(i);
807 /* NOTREACHED */
808 default:
809 quitenv(NULL);
810 internal_errorf(Tunexpected_type, Tunwind, Tfunction, i);
811 }
812 break;
813 }
814
815 /* executable command */
816 case CEXEC:
817 /* tracked alias */
818 case CTALIAS:
819 if (!(tp->flag&ISSET)) {
820 if (tp->u2.errnov == ENOENT) {
821 rv = 127;
822 warningf(true, Tf_sD_s_s, cp,
823 "inaccessible or", Tnot_found);
824 } else {
825 rv = 126;
826 warningf(true, Tf_sD_sD_s, cp, "can't execute",
827 cstrerror(tp->u2.errnov));
828 }
829 break;
830 }
831
832 /* set $_ to program's full path */
833 /* setstr() can't fail here */
834 setstr(typeset("_", LOCAL | EXPORT, 0, INTEGER, 0),
835 tp->val.s, KSH_RETURN_ERROR);
836
837 /* to fork, we set up a TEXEC node and call execute */
838 texec.type = TEXEC;
839 /* for vistree/dumptree */
840 texec.left = t;
841 texec.str = tp->val.s;
842 texec.args = ap;
843
844 /* in this case we do not fork, of course */
845 if (flags & XEXEC) {
846 if (exec_argv0)
847 texec.args[0] = exec_argv0;
848 j_exit();
849 if (!(flags & XBGND)
850 #ifndef MKSH_UNEMPLOYED
851 || Flag(FMONITOR)
852 #endif
853 ) {
854 setexecsig(&sigtraps[SIGINT], SS_RESTORE_ORIG);
855 setexecsig(&sigtraps[SIGQUIT], SS_RESTORE_ORIG);
856 }
857 }
858
859 rv = exchild(&texec, flags, xerrok, -1);
860 break;
861 }
862 Leave:
863 if (flags & XEXEC) {
864 exstat = rv & 0xFF;
865 unwind(LEXIT);
866 }
867 return (rv);
868 }
869
870 static void
scriptexec(struct op * tp,const char ** ap)871 scriptexec(struct op *tp, const char **ap)
872 {
873 const char *sh;
874 #ifndef MKSH_SMALL
875 int fd;
876 unsigned char buf[68];
877 #endif
878 union mksh_ccphack args, cap;
879
880 sh = str_val(global(TEXECSHELL));
881 if (sh && *sh)
882 sh = search_path(sh, path, X_OK, NULL);
883 if (!sh || !*sh)
884 sh = MKSH_DEFAULT_EXECSHELL;
885
886 *tp->args-- = tp->str;
887
888 #ifndef MKSH_SMALL
889 if ((fd = binopen2(tp->str, O_RDONLY | O_MAYEXEC)) >= 0) {
890 unsigned char *cp;
891 #ifndef MKSH_EBCDIC
892 unsigned short m;
893 #endif
894 ssize_t n;
895
896 #if defined(__OS2__) && defined(MKSH_WITH_TEXTMODE)
897 setmode(fd, O_TEXT);
898 #endif
899 /* read first couple of octets from file */
900 n = read(fd, buf, sizeof(buf) - 1);
901 close(fd);
902 /* read error or short read? */
903 if (n < 5)
904 goto nomagic;
905 /* terminate buffer */
906 buf[n] = '\0';
907
908 /* skip UTF-8 Byte Order Mark, if present */
909 cp = buf + (n = ((buf[0] == 0xEF) && (buf[1] == 0xBB) &&
910 (buf[2] == 0xBF)) ? 3 : 0);
911
912 /* scan for newline or NUL (end of buffer) */
913 while (!ctype(*cp, C_NL | C_NUL))
914 ++cp;
915 /* if the shebang line is longer than MAXINTERP, bail out */
916 if (!*cp)
917 goto noshebang;
918 /* replace newline by NUL */
919 *cp = '\0';
920
921 /* restore start of shebang position (buf+0 or buf+3) */
922 cp = buf + n;
923 /* bail out if no shebang magic found */
924 if (cp[0] == '#' && cp[1] == '!')
925 cp += 2;
926 #ifdef __OS2__
927 else if (!strncmp(cp, Textproc, 7) &&
928 ctype(cp[7], C_BLANK))
929 cp += 8;
930 #endif
931 else
932 goto noshebang;
933 /* skip whitespace before shell name */
934 while (ctype(*cp, C_BLANK))
935 ++cp;
936 /* just whitespace on the line? */
937 if (*cp == '\0')
938 goto noshebang;
939 /* no, we actually found an interpreter name */
940 sh = (char *)cp;
941 /* look for end of shell/interpreter name */
942 while (!ctype(*cp, C_BLANK | C_NUL))
943 ++cp;
944 /* any arguments? */
945 if (*cp) {
946 *cp++ = '\0';
947 /* skip spaces before arguments */
948 while (ctype(*cp, C_BLANK))
949 ++cp;
950 /* pass it all in ONE argument (historic reasons) */
951 if (*cp)
952 *tp->args-- = (char *)cp;
953 }
954 #ifdef __OS2__
955 /*
956 * On OS/2, the directory structure differs from normal
957 * Unix, which can make many scripts whose shebang
958 * hardcodes the path to an interpreter fail (and there
959 * might be no /usr/bin/env); for user convenience, if
960 * the specified interpreter is not usable, do a PATH
961 * search to find it.
962 */
963 if (mksh_vdirsep(sh) && !search_path(sh, path, X_OK, NULL)) {
964 cp = search_path(_getname(sh), path, X_OK, NULL);
965 if (cp)
966 sh = cp;
967 }
968 #endif
969 goto nomagic;
970 noshebang:
971 #ifndef MKSH_EBCDIC
972 m = buf[0] << 8 | buf[1];
973 if (m == 0x7F45 && buf[2] == 'L' && buf[3] == 'F')
974 errorf("%s: not executable: %d-bit ELF file", tp->str,
975 32 * buf[4]);
976 if ((m == /* OMAGIC */ 0407) ||
977 (m == /* NMAGIC */ 0410) ||
978 (m == /* ZMAGIC */ 0413) ||
979 (m == /* QMAGIC */ 0314) ||
980 (m == /* ECOFF_I386 */ 0x4C01) ||
981 (m == /* ECOFF_M68K */ 0x0150 || m == 0x5001) ||
982 (m == /* ECOFF_SH */ 0x0500 || m == 0x0005) ||
983 (m == /* bzip */ 0x425A) || (m == /* "MZ" */ 0x4D5A) ||
984 (m == /* "NE" */ 0x4E45) || (m == /* "LX" */ 0x4C58) ||
985 (m == /* ksh93 */ 0x0B13) || (m == /* LZIP */ 0x4C5A) ||
986 (m == /* xz */ 0xFD37 && buf[2] == 'z' && buf[3] == 'X' &&
987 buf[4] == 'Z') || (m == /* 7zip */ 0x377A) ||
988 (m == /* gzip */ 0x1F8B) || (m == /* .Z */ 0x1F9D))
989 errorf("%s: not executable: magic %04X", tp->str, m);
990 #endif
991 #ifdef __OS2__
992 cp = _getext(tp->str);
993 if (cp && (!stricmp(cp, ".cmd") || !stricmp(cp, ".bat"))) {
994 /* execute .cmd and .bat with OS2_SHELL, usually CMD.EXE */
995 sh = str_val(global("OS2_SHELL"));
996 *tp->args-- = "/c";
997 /* convert slahes to backslashes */
998 for (cp = tp->str; *cp; cp++) {
999 if (*cp == '/')
1000 *cp = '\\';
1001 }
1002 }
1003 #endif
1004 nomagic:
1005 ;
1006 }
1007 #endif
1008 args.ro = tp->args;
1009 *args.ro = sh;
1010
1011 cap.ro = ap;
1012 execve(args.rw[0], args.rw, cap.rw);
1013
1014 /* report both the program that was run and the bogus interpreter */
1015 errorf(Tf_sD_sD_s, tp->str, sh, cstrerror(errno));
1016 }
1017
1018 /* actual 'builtin' built-in utility call is handled in comexec() */
1019 int
c_builtin(const char ** wp)1020 c_builtin(const char **wp)
1021 {
1022 return (call_builtin(get_builtin(*wp), wp, Tbuiltin, false));
1023 }
1024
1025 struct tbl *
get_builtin(const char * s)1026 get_builtin(const char *s)
1027 {
1028 return (s && *s ? ktsearch(&builtins, s, hash(s)) : NULL);
1029 }
1030
1031 /*
1032 * Search function tables for a function. If create set, a table entry
1033 * is created if none is found.
1034 */
1035 struct tbl *
findfunc(const char * name,uint32_t h,bool create)1036 findfunc(const char *name, uint32_t h, bool create)
1037 {
1038 struct block *l;
1039 struct tbl *tp = NULL;
1040
1041 for (l = e->loc; l; l = l->next) {
1042 tp = ktsearch(&l->funs, name, h);
1043 if (tp)
1044 break;
1045 if (!l->next && create) {
1046 tp = ktenter(&l->funs, name, h);
1047 tp->flag = DEFINED;
1048 tp->type = CFUNC;
1049 tp->val.t = NULL;
1050 break;
1051 }
1052 }
1053 return (tp);
1054 }
1055
1056 /*
1057 * define function. Returns 1 if function is being undefined (t == 0) and
1058 * function did not exist, returns 0 otherwise.
1059 */
1060 int
define(const char * name,struct op * t)1061 define(const char *name, struct op *t)
1062 {
1063 uint32_t nhash;
1064 struct tbl *tp;
1065 bool was_set = false;
1066
1067 nhash = hash(name);
1068
1069 while (/* CONSTCOND */ 1) {
1070 tp = findfunc(name, nhash, true);
1071
1072 if (tp->flag & ISSET)
1073 was_set = true;
1074 /*
1075 * If this function is currently being executed, we zap
1076 * this table entry so findfunc() won't see it
1077 */
1078 if (tp->flag & FINUSE) {
1079 tp->name[0] = '\0';
1080 /* ensure it won't be found */
1081 tp->flag &= ~DEFINED;
1082 tp->flag |= FDELETE;
1083 } else
1084 break;
1085 }
1086
1087 if (tp->flag & ALLOC) {
1088 tp->flag &= ~(ISSET|ALLOC|FKSH);
1089 tfree(tp->val.t, tp->areap);
1090 }
1091
1092 if (t == NULL) {
1093 /* undefine */
1094 ktdelete(tp);
1095 return (was_set ? 0 : 1);
1096 }
1097
1098 tp->val.t = tcopy(t->left, tp->areap);
1099 tp->flag |= (ISSET|ALLOC);
1100 if (t->u.ksh_func)
1101 tp->flag |= FKSH;
1102
1103 return (0);
1104 }
1105
1106 /*
1107 * add builtin
1108 */
1109 const char *
builtin(const char * name,int (* func)(const char **))1110 builtin(const char *name, int (*func) (const char **))
1111 {
1112 struct tbl *tp;
1113 uint32_t flag = DEFINED;
1114
1115 /* see if any flags should be set for this builtin */
1116 flags_loop:
1117 switch (*name) {
1118 case '=':
1119 /* command does variable assignment */
1120 flag |= KEEPASN;
1121 break;
1122 case '*':
1123 /* POSIX special builtin */
1124 flag |= SPEC_BI;
1125 break;
1126 case '~':
1127 /* external utility overrides built-in utility, always */
1128 flag |= LOWER_BI;
1129 /* FALLTHROUGH */
1130 case '!':
1131 /* external utility overrides built-in utility, with flags */
1132 flag |= LOW_BI;
1133 break;
1134 case '-':
1135 /* is declaration utility if argv[1] is one (POSIX: command) */
1136 flag |= DECL_FWDR;
1137 break;
1138 case '^':
1139 /* is declaration utility (POSIX: export, readonly) */
1140 flag |= DECL_UTIL;
1141 break;
1142 case '#':
1143 /* is set or shift */
1144 flag |= NEXTLOC_BI;
1145 break;
1146 default:
1147 goto flags_seen;
1148 }
1149 ++name;
1150 goto flags_loop;
1151 flags_seen:
1152
1153 /* enter into the builtins hash table */
1154 tp = ktenter(&builtins, name, hash(name));
1155 tp->flag = flag;
1156 tp->type = CSHELL;
1157 tp->val.f = func;
1158
1159 /* return name, for direct builtin call check in main.c */
1160 return (name);
1161 }
1162
1163 /*
1164 * find command
1165 * either function, hashed command, or built-in (in that order)
1166 */
1167 struct tbl *
findcom(const char * name,int flags)1168 findcom(const char *name, int flags)
1169 {
1170 #ifdef MKSH_OH_ADAPT
1171 // fix crash for 'exec -a0'
1172 if (name == NULL) {
1173 return NULL;
1174 }
1175 #endif
1176 static struct tbl temp;
1177 uint32_t h = hash(name);
1178 struct tbl *tp = NULL, *tbi;
1179 /* insert if not found */
1180 unsigned char insert = Flag(FTRACKALL);
1181 /* for function autoloading */
1182 char *fpath;
1183 union mksh_cchack npath;
1184
1185 if (mksh_vdirsep(name)) {
1186 insert = 0;
1187 /* prevent FPATH search below */
1188 flags &= ~FC_FUNC;
1189 goto Search;
1190 }
1191 tbi = (flags & FC_BI) ? ktsearch(&builtins, name, h) : NULL;
1192 /*
1193 * POSIX says special builtins first, then functions, then
1194 * regular builtins, then search path...
1195 */
1196 if ((flags & FC_SPECBI) && tbi && (tbi->flag & SPEC_BI))
1197 tp = tbi;
1198 if (!tp && (flags & FC_FUNC)) {
1199 tp = findfunc(name, h, false);
1200 if (tp && !(tp->flag & ISSET)) {
1201 if ((fpath = str_val(global(TFPATH))) == null) {
1202 tp->u.fpath = NULL;
1203 tp->u2.errnov = ENOENT;
1204 } else
1205 tp->u.fpath = search_path(name, fpath, R_OK,
1206 &tp->u2.errnov);
1207 }
1208 }
1209 if (!tp && (flags & FC_NORMBI) && tbi)
1210 tp = tbi;
1211 if (!tp && (flags & FC_PATH) && !(flags & FC_DEFPATH)) {
1212 tp = ktsearch(&taliases, name, h);
1213 if (tp && (tp->flag & ISSET) &&
1214 ksh_access(tp->val.s, X_OK) != 0) {
1215 if (tp->flag & ALLOC) {
1216 tp->flag &= ~ALLOC;
1217 afree(tp->val.s, APERM);
1218 }
1219 tp->flag &= ~ISSET;
1220 }
1221 }
1222
1223 Search:
1224 if ((!tp || (tp->type == CTALIAS && !(tp->flag & ISSET))) &&
1225 (flags & FC_PATH)) {
1226 if (!tp) {
1227 if (insert && !(flags & FC_DEFPATH)) {
1228 tp = ktenter(&taliases, name, h);
1229 tp->type = CTALIAS;
1230 } else {
1231 tp = &temp;
1232 tp->type = CEXEC;
1233 }
1234 /* make ~ISSET */
1235 tp->flag = DEFINED;
1236 }
1237 npath.ro = search_path(name,
1238 (flags & FC_DEFPATH) ? def_path : path,
1239 X_OK, &tp->u2.errnov);
1240 if (npath.ro) {
1241 strdupx(tp->val.s, npath.ro, APERM);
1242 if (npath.ro != name)
1243 afree(npath.rw, ATEMP);
1244 tp->flag |= ISSET|ALLOC;
1245 } else if ((flags & FC_FUNC) &&
1246 (fpath = str_val(global(TFPATH))) != null &&
1247 (npath.ro = search_path(name, fpath, R_OK,
1248 &tp->u2.errnov)) != NULL) {
1249 /*
1250 * An undocumented feature of AT&T ksh is that
1251 * it searches FPATH if a command is not found,
1252 * even if the command hasn't been set up as an
1253 * autoloaded function (ie, no typeset -uf).
1254 */
1255 tp = &temp;
1256 tp->type = CFUNC;
1257 /* make ~ISSET */
1258 tp->flag = DEFINED;
1259 tp->u.fpath = npath.ro;
1260 }
1261 }
1262 return (tp);
1263 }
1264
1265 /*
1266 * flush executable commands with relative paths
1267 * (just relative or all?)
1268 */
1269 void
flushcom(bool all)1270 flushcom(bool all)
1271 {
1272 struct tbl *tp;
1273 struct tstate ts;
1274
1275 for (ktwalk(&ts, &taliases); (tp = ktnext(&ts)) != NULL; )
1276 if ((tp->flag&ISSET) && (all || !mksh_abspath(tp->val.s))) {
1277 if (tp->flag&ALLOC) {
1278 tp->flag &= ~(ALLOC|ISSET);
1279 afree(tp->val.s, APERM);
1280 }
1281 tp->flag &= ~ISSET;
1282 }
1283 }
1284
1285 /* check if path is something we want to find */
1286 int
search_access(const char * fn,int mode)1287 search_access(const char *fn, int mode)
1288 {
1289 struct stat sb;
1290
1291 if (stat(fn, &sb) < 0)
1292 /* file does not exist */
1293 return (ENOENT);
1294 /* LINTED use of access */
1295 if (access(fn, mode) < 0) {
1296 /* file exists, but we can't access it */
1297 int eno;
1298
1299 eno = errno;
1300 return (eno ? eno : EACCES);
1301 }
1302 #ifdef __OS2__
1303 /* treat all files as executable on OS/2 */
1304 sb.st_mode |= S_IXUSR | S_IXGRP | S_IXOTH;
1305 #endif
1306 if (mode == X_OK && (!S_ISREG(sb.st_mode) ||
1307 !(sb.st_mode & (S_IXUSR | S_IXGRP | S_IXOTH))))
1308 /* access(2) may say root can execute everything */
1309 return (S_ISDIR(sb.st_mode) ? EISDIR : EACCES);
1310 return (0);
1311 }
1312
1313 #ifdef __OS2__
1314 /* check if path is something we want to find adding executable extensions */
1315 #define search_access(fn,mode) access_ex((search_access), (fn), (mode))
1316 #else
1317 #define search_access(fn,mode) (search_access)((fn), (mode))
1318 #endif
1319
1320 /*
1321 * search for command with PATH
1322 */
1323 const char *
search_path(const char * name,const char * lpath,int mode,int * errnop)1324 search_path(const char *name, const char *lpath,
1325 /* R_OK or X_OK */
1326 int mode,
1327 /* set if candidate found, but not suitable */
1328 int *errnop)
1329 {
1330 const char *sp, *p;
1331 char *xp;
1332 XString xs;
1333 size_t namelen;
1334 int ec = 0, ev;
1335
1336 if (mksh_vdirsep(name)) {
1337 if ((ec = search_access(name, mode)) == 0) {
1338 search_path_ok:
1339 if (errnop)
1340 *errnop = 0;
1341 #ifndef __OS2__
1342 return (name);
1343 #else
1344 return (real_exec_name(name));
1345 #endif
1346 }
1347 goto search_path_err;
1348 }
1349
1350 namelen = strlen(name) + 1;
1351 Xinit(xs, xp, 128, ATEMP);
1352
1353 sp = lpath;
1354 while (sp != NULL) {
1355 xp = Xstring(xs, xp);
1356 if (!(p = cstrchr(sp, MKSH_PATHSEPC)))
1357 p = strnul(sp);
1358 if (p != sp) {
1359 XcheckN(xs, xp, p - sp);
1360 memcpy(xp, sp, p - sp);
1361 xp += p - sp;
1362 if (mksh_cdirsep(xp[-1]))
1363 xp--;
1364 *xp++ = '/';
1365 }
1366 sp = p;
1367 XcheckN(xs, xp, namelen);
1368 memcpy(xp, name, namelen);
1369 if ((ev = search_access(Xstring(xs, xp), mode)) == 0) {
1370 name = Xclose(xs, xp + namelen);
1371 goto search_path_ok;
1372 }
1373 /* accumulate non-ENOENT errors only */
1374 if (ev != ENOENT && ec == 0)
1375 ec = ev;
1376 if (*sp++ == '\0')
1377 sp = NULL;
1378 }
1379 Xfree(xs, xp);
1380 search_path_err:
1381 if (errnop)
1382 *errnop = ec ? ec : ENOENT;
1383 return (NULL);
1384 }
1385
1386 static int
call_builtin(struct tbl * tp,const char ** wp,const char * where,bool resetspec)1387 call_builtin(struct tbl *tp, const char **wp, const char *where, bool resetspec)
1388 {
1389 int rv;
1390
1391 if (!tp)
1392 internal_errorf(Tf_sD_s, where, wp[0]);
1393 builtin_argv0 = wp[0];
1394 builtin_spec = tobool(!resetspec && (tp->flag & SPEC_BI));
1395 shf_reopen(1, SHF_WR, shl_stdout);
1396 shl_stdout_ok = true;
1397 ksh_getopt_reset(&builtin_opt, GF_ERROR);
1398 rv = (*tp->val.f)(wp);
1399 shf_flush(shl_stdout);
1400 shl_stdout_ok = false;
1401 builtin_argv0 = NULL;
1402 builtin_spec = false;
1403 return (rv);
1404 }
1405
1406 /*
1407 * set up redirection, saving old fds in e->savefd
1408 */
1409 static int
iosetup(struct ioword * iop,struct tbl * tp)1410 iosetup(struct ioword *iop, struct tbl *tp)
1411 {
1412 int u = -1;
1413 char *cp = iop->ioname;
1414 int iotype = iop->ioflag & IOTYPE;
1415 bool do_open = true, do_close = false, do_fstat = false;
1416 int flags = 0;
1417 struct ioword iotmp;
1418 struct stat statb;
1419
1420 if (iotype != IOHERE)
1421 cp = evalonestr(cp, DOTILDE|(Flag(FTALKING_I) ? DOGLOB : 0));
1422
1423 /* Used for tracing and error messages to print expanded cp */
1424 iotmp = *iop;
1425 iotmp.ioname = (iotype == IOHERE) ? NULL : cp;
1426 iotmp.ioflag |= IONAMEXP;
1427
1428 if (Flag(FXTRACE)) {
1429 change_xtrace(2, false);
1430 fptreef(shl_xtrace, 0, Tft_R, &iotmp);
1431 change_xtrace(1, false);
1432 }
1433
1434 switch (iotype) {
1435 case IOREAD:
1436 flags = O_RDONLY;
1437 break;
1438
1439 case IOCAT:
1440 flags = O_WRONLY | O_APPEND | O_CREAT;
1441 break;
1442
1443 case IOWRITE:
1444 if (Flag(FNOCLOBBER) && !(iop->ioflag & IOCLOB)) {
1445 /* >file under set -C */
1446 if (stat(cp, &statb)) {
1447 /* nonexistent file */
1448 flags = O_WRONLY | O_CREAT | O_EXCL;
1449 } else if (S_ISREG(statb.st_mode)) {
1450 /* regular file, refuse clobbering */
1451 goto clobber_refused;
1452 } else {
1453 /*
1454 * allow redirections to things
1455 * like /dev/null without error
1456 */
1457 flags = O_WRONLY;
1458 /* but check again after opening */
1459 do_fstat = true;
1460 }
1461 } else {
1462 /* >|file or set +C */
1463 flags = O_WRONLY | O_CREAT | O_TRUNC;
1464 }
1465 break;
1466
1467 case IORDWR:
1468 flags = O_RDWR | O_CREAT;
1469 break;
1470
1471 case IOHERE:
1472 do_open = false;
1473 /* herein() returns -2 if error has been printed */
1474 u = herein(iop, NULL);
1475 /* cp may have wrong name */
1476 break;
1477
1478 case IODUP: {
1479 const char *emsg;
1480
1481 do_open = false;
1482 if (ksh_isdash(cp)) {
1483 /* prevent error return below */
1484 u = 1009;
1485 do_close = true;
1486 } else if ((u = check_fd(cp,
1487 X_OK | ((iop->ioflag & IORDUP) ? R_OK : W_OK),
1488 &emsg)) < 0) {
1489 char *sp;
1490
1491 warningf(true, Tf_sD_s,
1492 (sp = snptreef(NULL, 32, Tft_R, &iotmp)), emsg);
1493 afree(sp, ATEMP);
1494 return (-1);
1495 }
1496 if (u == (int)iop->unit) {
1497 /* "dup from" == "dup to" */
1498 iop->ioflag |= IODUPSELF;
1499 return (0);
1500 }
1501 break;
1502 }
1503 }
1504
1505 if (do_open) {
1506 if (Flag(FRESTRICTED) && (flags & O_CREAT)) {
1507 warningf(true, Tf_sD_s, cp, "restricted");
1508 return (-1);
1509 }
1510 u = binopen3(cp, flags, 0666);
1511 if (do_fstat && u >= 0) {
1512 /* prevent race conditions */
1513 if (fstat(u, &statb) || S_ISREG(statb.st_mode)) {
1514 close(u);
1515 clobber_refused:
1516 u = -1;
1517 errno = EEXIST;
1518 }
1519 }
1520 }
1521 if (u < 0) {
1522 /* herein() may already have printed message */
1523 if (u == -1) {
1524 u = errno;
1525 warningf(true, Tf_cant_ss_s,
1526 #if 0
1527 /* can't happen */
1528 iotype == IODUP ? "dup" :
1529 #endif
1530 (iotype == IOREAD || iotype == IOHERE) ?
1531 Topen : Tcreate, cp, cstrerror(u));
1532 }
1533 return (-1);
1534 }
1535 /* Do not save if it has already been redirected (i.e. "cat >x >y"). */
1536 if (e->savefd[iop->unit] == 0) {
1537 /* If these are the same, it means unit was previously closed */
1538 if (u == (int)iop->unit)
1539 e->savefd[iop->unit] = -1;
1540 else
1541 /*
1542 * c_exec() assumes e->savefd[fd] set for any
1543 * redirections. Ask savefd() not to close iop->unit;
1544 * this allows error messages to be seen if iop->unit
1545 * is 2; also means we can't lose the fd (eg, both
1546 * dup2 below and dup2 in restfd() failing).
1547 */
1548 e->savefd[iop->unit] = savefd(iop->unit);
1549 }
1550
1551 if (do_close)
1552 close(iop->unit);
1553 else if (u != (int)iop->unit) {
1554 if (ksh_dup2(u, iop->unit, true) < 0) {
1555 int eno;
1556 char *sp;
1557
1558 eno = errno;
1559 warningf(true, Tf_s_sD_s, Tredirection_dup,
1560 (sp = snptreef(NULL, 32, Tft_R, &iotmp)),
1561 cstrerror(eno));
1562 afree(sp, ATEMP);
1563 if (iotype != IODUP)
1564 close(u);
1565 return (-1);
1566 }
1567 if (iotype != IODUP)
1568 close(u);
1569 /*
1570 * Touching any co-process fd in an empty exec
1571 * causes the shell to close its copies
1572 */
1573 else if (tp && tp->type == CSHELL && tp->val.f == c_exec) {
1574 if (iop->ioflag & IORDUP)
1575 /* possible exec <&p */
1576 coproc_read_close(u);
1577 else
1578 /* possible exec >&p */
1579 coproc_write_close(u);
1580 }
1581 }
1582 if (u == 2)
1583 /* Clear any write errors */
1584 shf_reopen(2, SHF_WR, shl_out);
1585 return (0);
1586 }
1587
1588 /*
1589 * Process here documents by providing the content, either as
1590 * result (globally allocated) string or in a temp file; if
1591 * unquoted, the string is expanded first.
1592 */
1593 static int
hereinval(struct ioword * iop,int sub,char ** resbuf,struct shf * shf)1594 hereinval(struct ioword *iop, int sub, char **resbuf, struct shf *shf)
1595 {
1596 const char * volatile ccp = iop->heredoc;
1597 struct source *s, *osource;
1598
1599 osource = source;
1600 newenv(E_ERRH);
1601 if (kshsetjmp(e->jbuf)) {
1602 source = osource;
1603 quitenv(shf);
1604 /* special to iosetup(): don't print error */
1605 return (-2);
1606 }
1607 if (iop->ioflag & IOHERESTR) {
1608 ccp = evalstr(iop->delim, DOHERESTR | DOSCALAR);
1609 } else if (sub) {
1610 /* do substitutions on the content of heredoc */
1611 s = pushs(SSTRING, ATEMP);
1612 s->start = s->str = ccp;
1613 source = s;
1614 if (yylex(sub) != LWORD)
1615 internal_errorf("herein: yylex");
1616 source = osource;
1617 ccp = evalstr(yylval.cp, DOSCALAR | DOHEREDOC);
1618 }
1619
1620 if (resbuf == NULL)
1621 shf_puts(ccp, shf);
1622 else
1623 strdupx(*resbuf, ccp, APERM);
1624
1625 quitenv(NULL);
1626 return (0);
1627 }
1628
1629 int
herein(struct ioword * iop,char ** resbuf)1630 herein(struct ioword *iop, char **resbuf)
1631 {
1632 int fd = -1;
1633 struct shf *shf;
1634 struct temp *h;
1635 int i;
1636
1637 /* lexer substitution flags */
1638 i = (iop->ioflag & IOEVAL) ? (ONEWORD | HEREDOC) : 0;
1639
1640 /* skip all the fd setup if we just want the value */
1641 if (resbuf != NULL)
1642 return (hereinval(iop, i, resbuf, NULL));
1643
1644 /*
1645 * Create temp file to hold content (done before newenv
1646 * so temp doesn't get removed too soon).
1647 */
1648 h = maketemp(ATEMP, TT_HEREDOC_EXP, &e->temps);
1649 if (!(shf = h->shf) || (fd = binopen3(h->tffn, O_RDONLY, 0)) < 0) {
1650 i = errno;
1651 warningf(true, Tf_temp,
1652 !shf ? Tcreate : Topen, h->tffn, cstrerror(i));
1653 if (shf)
1654 shf_close(shf);
1655 /* special to iosetup(): don't print error */
1656 return (-2);
1657 }
1658
1659 if (hereinval(iop, i, NULL, shf) == -2) {
1660 close(fd);
1661 /* special to iosetup(): don't print error */
1662 return (-2);
1663 }
1664
1665 if (shf_close(shf) == -1) {
1666 i = errno;
1667 close(fd);
1668 warningf(true, Tf_temp,
1669 Twrite, h->tffn, cstrerror(i));
1670 /* special to iosetup(): don't print error */
1671 return (-2);
1672 }
1673
1674 return (fd);
1675 }
1676
1677 /*
1678 * ksh special - the select command processing section
1679 * print the args in column form - assuming that we can
1680 */
1681 static const char *
do_selectargs(const char ** ap,bool print_menu)1682 do_selectargs(const char **ap, bool print_menu)
1683 {
1684 static const char *read_args[] = {
1685 Tread, Tdr, TREPLY, NULL
1686 };
1687 char *s;
1688 int i, argct;
1689
1690 for (argct = 0; ap[argct]; argct++)
1691 ;
1692 while (/* CONSTCOND */ 1) {
1693 /*-
1694 * Menu is printed if
1695 * - this is the first time around the select loop
1696 * - the user enters a blank line
1697 * - the REPLY parameter is empty
1698 */
1699 if (print_menu || !*str_val(global(TREPLY)))
1700 pr_menu(ap);
1701 shellf(Tf_s, str_val(global("PS3")));
1702 if (call_builtin(findcom(Tread, FC_BI), read_args, Tselect,
1703 false))
1704 return (NULL);
1705 if (*(s = str_val(global(TREPLY))))
1706 return ((getn(s, &i) && i >= 1 && i <= argct) ?
1707 ap[i - 1] : null);
1708 print_menu = true;
1709 }
1710 }
1711
1712 struct select_menu_info {
1713 const char * const *args;
1714 int num_width;
1715 };
1716
1717 /* format a single select menu item */
1718 static void
select_fmt_entry(char * buf,size_t buflen,unsigned int i,const void * arg)1719 select_fmt_entry(char *buf, size_t buflen, unsigned int i, const void *arg)
1720 {
1721 const struct select_menu_info *smi =
1722 (const struct select_menu_info *)arg;
1723
1724 shf_snprintf(buf, buflen, "%*u) %s",
1725 smi->num_width, i + 1, smi->args[i]);
1726 }
1727
1728 /*
1729 * print a select style menu
1730 */
1731 void
pr_menu(const char * const * ap)1732 pr_menu(const char * const *ap)
1733 {
1734 struct select_menu_info smi;
1735 const char * const *pp;
1736 size_t acols = 0, aocts = 0, i;
1737 unsigned int n;
1738 struct columnise_opts co;
1739
1740 /*
1741 * width/column calculations were done once and saved, but this
1742 * means select can't be used recursively so we re-calculate
1743 * each time (could save in a structure that is returned, but
1744 * it's probably not worth the bother)
1745 */
1746
1747 /*
1748 * get dimensions of the list
1749 */
1750 for (n = 0, pp = ap; *pp; n++, pp++) {
1751 i = strlen(*pp);
1752 if (i > aocts)
1753 aocts = i;
1754 i = utf_mbswidth(*pp);
1755 if (i > acols)
1756 acols = i;
1757 }
1758
1759 /*
1760 * we will print an index of the form "%d) " in front of
1761 * each entry, so get the maximum width of this
1762 */
1763 for (i = n, smi.num_width = 1; i >= 10; i /= 10)
1764 smi.num_width++;
1765
1766 smi.args = ap;
1767 co.shf = shl_out;
1768 co.linesep = '\n';
1769 co.prefcol = co.do_last = true;
1770 print_columns(&co, n, select_fmt_entry, (void *)&smi,
1771 smi.num_width + 2 + aocts, smi.num_width + 2 + acols);
1772 }
1773
1774 static void
plain_fmt_entry(char * buf,size_t buflen,unsigned int i,const void * arg)1775 plain_fmt_entry(char *buf, size_t buflen, unsigned int i, const void *arg)
1776 {
1777 strlcpy(buf, ((const char * const *)arg)[i], buflen);
1778 }
1779
1780 void
pr_list(struct columnise_opts * cop,char * const * ap)1781 pr_list(struct columnise_opts *cop, char * const *ap)
1782 {
1783 size_t acols = 0, aocts = 0, i;
1784 unsigned int n;
1785 char * const *pp;
1786
1787 for (n = 0, pp = ap; *pp; n++, pp++) {
1788 i = strlen(*pp);
1789 if (i > aocts)
1790 aocts = i;
1791 i = utf_mbswidth(*pp);
1792 if (i > acols)
1793 acols = i;
1794 }
1795
1796 print_columns(cop, n, plain_fmt_entry, (const void *)ap,
1797 aocts, acols);
1798 }
1799
1800 /*
1801 * [[ ... ]] evaluation routines
1802 */
1803
1804 /*
1805 * Test if the current token is a whatever. Accepts the current token if
1806 * it is. Returns 0 if it is not, non-zero if it is (in the case of
1807 * TM_UNOP and TM_BINOP, the returned value is a Test_op).
1808 */
1809 static Test_op
dbteste_isa(Test_env * te,Test_meta meta)1810 dbteste_isa(Test_env *te, Test_meta meta)
1811 {
1812 Test_op ret = TO_NONOP;
1813 bool uqword;
1814 const char *p;
1815
1816 if (!*te->pos.wp)
1817 return (meta == TM_END ? TO_NONNULL : TO_NONOP);
1818
1819 /* unquoted word? */
1820 for (p = *te->pos.wp; *p == CHAR; p += 2)
1821 ;
1822 uqword = *p == EOS;
1823
1824 if (meta == TM_UNOP || meta == TM_BINOP) {
1825 if (uqword) {
1826 /* longer than the longest operator */
1827 char buf[8];
1828 char *q = buf;
1829
1830 p = *te->pos.wp;
1831 while (*p++ == CHAR &&
1832 (size_t)(q - buf) < sizeof(buf) - 1)
1833 *q++ = *p++;
1834 *q = '\0';
1835 ret = test_isop(meta, buf);
1836 }
1837 } else if (meta == TM_END)
1838 ret = TO_NONOP;
1839 else
1840 ret = (uqword && !strcmp(*te->pos.wp,
1841 dbtest_tokens[(int)meta])) ? TO_NONNULL : TO_NONOP;
1842
1843 /* Accept the token? */
1844 if (ret != TO_NONOP)
1845 te->pos.wp++;
1846
1847 return (ret);
1848 }
1849
1850 static const char *
dbteste_getopnd(Test_env * te,Test_op op,bool do_eval)1851 dbteste_getopnd(Test_env *te, Test_op op, bool do_eval)
1852 {
1853 const char *s = *te->pos.wp;
1854 int flags = DOTILDE | DOSCALAR;
1855
1856 if (!s)
1857 return (NULL);
1858
1859 te->pos.wp++;
1860
1861 if (!do_eval)
1862 return (null);
1863
1864 if (op == TO_STEQL || op == TO_STNEQ) {
1865 flags |= DOPAT;
1866 if (!Flag(FSH))
1867 flags |= DODBMAGIC;
1868 }
1869
1870 return (evalstr(s, flags));
1871 }
1872
1873 static void
dbteste_error(Test_env * te,int offset,const char * msg)1874 dbteste_error(Test_env *te, int offset, const char *msg)
1875 {
1876 te->flags |= TEF_ERROR;
1877 internal_warningf("dbteste_error: %s (offset %d)", msg, offset);
1878 }
1879