exec.c revision 1.4 1 /* $NetBSD: exec.c,v 1.4 1998/07/28 05:31:25 mycroft Exp $ */
2
3 /*
4 * execute command tree
5 */
6
7 #include "sh.h"
8 #include "c_test.h"
9 #include <ctype.h>
10 #include "ksh_stat.h"
11
12 /* Does ps4 get parameter substitutions done? */
13 #ifdef KSH
14 # define PS4_SUBSTITUTE(s) substitute((s), 0)
15 #else
16 # define PS4_SUBSTITUTE(s) (s)
17 #endif /* KSH */
18
19 static int comexec ARGS((struct op *t, struct tbl *volatile tp, char **ap,
20 int volatile flags));
21 static void scriptexec ARGS((struct op *tp, char **ap));
22 static int call_builtin ARGS((struct tbl *tp, char **wp));
23 static int iosetup ARGS((struct ioword *iop, struct tbl *tp));
24 static int herein ARGS((char *hname, int sub));
25 #ifdef KSH
26 static char *do_selectargs ARGS((char **ap, bool_t print_menu));
27 #endif /* KSH */
28 #ifdef KSH
29 static int dbteste_isa ARGS((Test_env *te, Test_meta meta));
30 static const char *dbteste_getopnd ARGS((Test_env *te, Test_op op,
31 int do_eval));
32 static int dbteste_eval ARGS((Test_env *te, Test_op op, const char *opnd1,
33 const char *opnd2, int do_eval));
34 static void dbteste_error ARGS((Test_env *te, int offset, const char *msg));
35 #endif /* KSH */
36 #ifdef OS2
37 static int search_access1 ARGS((const char *path, int mode, int *errnop));
38 #endif /* OS2 */
39
40
41 /*
42 * handle systems that don't have F_SETFD
43 */
44 #ifndef F_SETFD
45 # ifndef MAXFD
46 # define MAXFD 64
47 # endif
48 /* a bit field would be smaller, but this will work */
49 static char clexec_tab[MAXFD+1];
50 #endif
51
52 /*
53 * we now use this function always.
54 */
55 int
56 fd_clexec(fd)
57 int fd;
58 {
59 #ifndef F_SETFD
60 if (fd >= 0 && fd < sizeof(clexec_tab)) {
61 clexec_tab[fd] = 1;
62 return 0;
63 }
64 return -1;
65 #else
66 return fcntl(fd, F_SETFD, 1);
67 #endif
68 }
69
70
71 /*
72 * execute command tree
73 */
74 int
75 execute(t, flags)
76 struct op * volatile t;
77 volatile int flags; /* if XEXEC don't fork */
78 {
79 int i;
80 volatile int rv = 0;
81 int pv[2];
82 char ** volatile ap;
83 char *s, *cp;
84 struct ioword **iowp;
85 struct tbl *tp = NULL;
86
87 if (t == NULL)
88 return 0;
89
90 /* Is this the end of a pipeline? If so, we want to evaluate the
91 * command arguments
92 bool_t eval_done = FALSE;
93 if ((flags&XFORK) && !(flags&XEXEC) && (flags&XPCLOSE)) {
94 eval_done = TRUE;
95 tp = eval_execute_args(t, &ap);
96 }
97 */
98 if ((flags&XFORK) && !(flags&XEXEC) && t->type != TPIPE)
99 return exchild(t, flags, -1); /* run in sub-process */
100
101 newenv(E_EXEC);
102 if (trap)
103 runtraps(0);
104
105 if (t->type == TCOM) {
106 /* Clear subst_exstat before argument expansion. Used by
107 * null commands (see comexec()) and by c_set().
108 */
109 subst_exstat = 0;
110
111 /* POSIX says expand command words first, then redirections,
112 * and assignments last..
113 */
114 ap = eval(t->args, t->u.evalflags | DOBLANK | DOGLOB | DOTILDE);
115 if (Flag(FXTRACE) && ap[0]) {
116 shf_fprintf(shl_out, "%s",
117 PS4_SUBSTITUTE(str_val(global("PS4"))));
118 for (i = 0; ap[i]; i++)
119 shf_fprintf(shl_out, "%s%s", ap[i],
120 ap[i + 1] ? space : newline);
121 shf_flush(shl_out);
122 }
123 if (ap[0])
124 tp = findcom(ap[0], FC_BI|FC_FUNC);
125 }
126
127 if (t->ioact != NULL || t->type == TPIPE || t->type == TCOPROC) {
128 e->savefd = (short *) alloc(sizeofN(short, NUFILE), ATEMP);
129 /* initialize to not redirected */
130 memset(e->savefd, 0, sizeofN(short, NUFILE));
131 }
132
133 /* do redirection, to be restored in quitenv() */
134 if (t->ioact != NULL)
135 for (iowp = t->ioact; *iowp != NULL; iowp++) {
136 if (iosetup(*iowp, tp) < 0) {
137 exstat = rv = 1;
138 /* Redirection failures for special commands
139 * cause (non-interactive) shell to exit.
140 */
141 if (tp && tp->type == CSHELL
142 && (tp->flag & SPEC_BI))
143 errorf(null);
144 /* Deal with FERREXIT, quitenv(), etc. */
145 goto Break;
146 }
147 }
148
149 switch(t->type) {
150 case TCOM:
151 rv = comexec(t, tp, ap, flags);
152 break;
153
154 case TPAREN:
155 rv = execute(t->left, flags|XFORK);
156 break;
157
158 case TPIPE:
159 flags |= XFORK;
160 flags &= ~XEXEC;
161 e->savefd[0] = savefd(0, 0);
162 (void) ksh_dup2(e->savefd[0], 0, FALSE); /* stdin of first */
163 e->savefd[1] = savefd(1, 0);
164 while (t->type == TPIPE) {
165 openpipe(pv);
166 (void) ksh_dup2(pv[1], 1, FALSE); /* stdout of curr */
167 /* Let exchild() close pv[0] in child
168 * (if this isn't done, commands like
169 * (: ; cat /etc/termcap) | sleep 1
170 * will hang forever).
171 */
172 exchild(t->left, flags|XPIPEO|XCCLOSE, pv[0]);
173 (void) ksh_dup2(pv[0], 0, FALSE); /* stdin of next */
174 closepipe(pv);
175 flags |= XPIPEI;
176 t = t->right;
177 }
178 restfd(1, e->savefd[1]); /* stdout of last */
179 e->savefd[1] = 0; /* no need to re-restore this */
180 /* Let exchild() close 0 in parent, after fork, before wait */
181 i = exchild(t, flags|XPCLOSE, 0);
182 if (!(flags&XBGND) && !(flags&XXCOM))
183 rv = i;
184 break;
185
186 case TLIST:
187 while (t->type == TLIST) {
188 execute(t->left, flags & XERROK);
189 t = t->right;
190 }
191 rv = execute(t, flags & XERROK);
192 break;
193
194 #ifdef KSH
195 case TCOPROC:
196 {
197 # ifdef JOB_SIGS
198 sigset_t omask;
199 # endif /* JOB_SIGS */
200
201 # ifdef JOB_SIGS
202 /* Block sigchild as we are using things changed in the
203 * signal handler
204 */
205 sigprocmask(SIG_BLOCK, &sm_sigchld, &omask);
206 e->type = E_ERRH;
207 i = ksh_sigsetjmp(e->jbuf, 0);
208 if (i) {
209 sigprocmask(SIG_SETMASK, &omask, (sigset_t *) 0);
210 quitenv();
211 unwind(i);
212 /*NOTREACHED*/
213 }
214 # endif /* JOB_SIGS */
215 /* Already have a (live) co-process? */
216 if (coproc.job && coproc.write >= 0)
217 errorf("coprocess already exists");
218
219 /* Can we re-use the existing co-process pipe? */
220 coproc_cleanup(TRUE);
221
222 /* do this before opening pipes, in case these fail */
223 e->savefd[0] = savefd(0, 0);
224 e->savefd[1] = savefd(1, 0);
225
226 openpipe(pv);
227 ksh_dup2(pv[0], 0, FALSE);
228 close(pv[0]);
229 coproc.write = pv[1];
230 coproc.job = (void *) 0;
231
232 if (coproc.readw >= 0)
233 ksh_dup2(coproc.readw, 1, FALSE);
234 else {
235 openpipe(pv);
236 coproc.read = pv[0];
237 ksh_dup2(pv[1], 1, FALSE);
238 coproc.readw = pv[1]; /* closed before first read */
239 coproc.njobs = 0;
240 /* create new coprocess id */
241 ++coproc.id;
242 }
243 # ifdef JOB_SIGS
244 sigprocmask(SIG_SETMASK, &omask, (sigset_t *) 0);
245 e->type = E_EXEC; /* no more need for error handler */
246 # endif /* JOB_SIGS */
247
248 /* exchild() closes coproc.* in child after fork,
249 * will also increment coproc.njobs when the
250 * job is actually created.
251 */
252 flags &= ~XEXEC;
253 exchild(t->left, flags|XBGND|XFORK|XCOPROC|XCCLOSE,
254 coproc.readw);
255 break;
256 }
257 #endif /* KSH */
258
259 case TASYNC:
260 /* XXX non-optimal, I think - "(foo &)", forks for (),
261 * forks again for async... parent should optimize
262 * this to "foo &"...
263 */
264 rv = execute(t->left, (flags&~XEXEC)|XBGND|XFORK);
265 break;
266
267 case TOR:
268 case TAND:
269 rv = execute(t->left, XERROK);
270 if (t->right != NULL && (rv == 0) == (t->type == TAND))
271 rv = execute(t->right, flags & XERROK);
272 else
273 flags |= XERROK;
274 break;
275
276 case TBANG:
277 rv = !execute(t->right, XERROK);
278 break;
279
280 #ifdef KSH
281 case TDBRACKET:
282 {
283 Test_env te;
284
285 te.flags = TEF_DBRACKET;
286 te.pos.wp = t->args;
287 te.isa = dbteste_isa;
288 te.getopnd = dbteste_getopnd;
289 te.eval = dbteste_eval;
290 te.error = dbteste_error;
291
292 rv = test_parse(&te);
293 break;
294 }
295 #endif /* KSH */
296
297 case TFOR:
298 #ifdef KSH
299 case TSELECT:
300 {
301 volatile bool_t is_first = TRUE;
302 #endif /* KSH */
303 ap = (t->vars != NULL) ?
304 eval(t->vars, DOBLANK|DOGLOB|DOTILDE)
305 : e->loc->argv + 1;
306 e->type = E_LOOP;
307 while (1) {
308 i = ksh_sigsetjmp(e->jbuf, 0);
309 if (!i)
310 break;
311 if ((e->flags&EF_BRKCONT_PASS)
312 || (i != LBREAK && i != LCONTIN))
313 {
314 quitenv();
315 unwind(i);
316 } else if (i == LBREAK) {
317 rv = 0;
318 goto Break;
319 }
320 }
321 rv = 0; /* in case of a continue */
322 if (t->type == TFOR) {
323 struct tbl *vq;
324
325 while (*ap != NULL) {
326 vq = global(t->str);
327 if (vq->flag & RDONLY)
328 errorf("%s is read only", t->str);
329 setstr(vq, *ap++);
330 rv = execute(t->left, flags & XERROK);
331 }
332 }
333 #ifdef KSH
334 else { /* TSELECT */
335 struct tbl *vq;
336
337 for (;;) {
338 if (!(cp = do_selectargs(ap, is_first))) {
339 rv = 1;
340 break;
341 }
342 is_first = FALSE;
343 vq = global(t->str);
344 if (vq->flag & RDONLY)
345 errorf("%s is read only", t->str);
346 setstr(vq, cp);
347 rv = execute(t->left, flags & XERROK);
348 }
349 }
350 }
351 #endif /* KSH */
352 break;
353
354 case TWHILE:
355 case TUNTIL:
356 e->type = E_LOOP;
357 while (1) {
358 i = ksh_sigsetjmp(e->jbuf, 0);
359 if (!i)
360 break;
361 if ((e->flags&EF_BRKCONT_PASS)
362 || (i != LBREAK && i != LCONTIN))
363 {
364 quitenv();
365 unwind(i);
366 } else if (i == LBREAK) {
367 rv = 0;
368 goto Break;
369 }
370 }
371 rv = 0; /* in case of a continue */
372 while ((execute(t->left, XERROK) == 0) == (t->type == TWHILE))
373 rv = execute(t->right, flags & XERROK);
374 break;
375
376 case TIF:
377 case TELIF:
378 if (t->right == NULL)
379 break; /* should be error */
380 rv = execute(t->left, XERROK) == 0 ?
381 execute(t->right->left, flags & XERROK) :
382 execute(t->right->right, flags & XERROK);
383 break;
384
385 case TCASE:
386 cp = evalstr(t->str, DOTILDE);
387 for (t = t->left; t != NULL && t->type == TPAT; t = t->right)
388 for (ap = t->vars; *ap; ap++)
389 if ((s = evalstr(*ap, DOTILDE|DOPAT))
390 && gmatch(cp, s, FALSE))
391 goto Found;
392 break;
393 Found:
394 rv = execute(t->left, flags & XERROK);
395 break;
396
397 case TBRACE:
398 rv = execute(t->left, flags & XERROK);
399 break;
400
401 case TFUNCT:
402 rv = define(t->str, t);
403 break;
404
405 case TTIME:
406 rv = timex(t, flags);
407 break;
408
409 case TEXEC: /* an eval'd TCOM */
410 s = t->args[0];
411 ap = makenv();
412 #ifndef F_SETFD
413 for (i = 0; i < sizeof(clexec_tab); i++)
414 if (clexec_tab[i]) {
415 close(i);
416 clexec_tab[i] = 0;
417 }
418 #endif
419 restoresigs();
420 cleanup_proc_env();
421 ksh_execve(t->str, t->args, ap);
422 if (errno == ENOEXEC)
423 scriptexec(t, ap);
424 else
425 errorf("%s: %s", s, strerror(errno));
426 }
427 Break:
428 exstat = rv;
429
430 quitenv(); /* restores IO */
431 if ((flags&XEXEC)) {
432 exit(rv); /* exit child */
433 /* NOTREACHED */
434 }
435 if (rv != 0 && !(flags & XERROK)) {
436 if (Flag(FERREXIT))
437 unwind(LERROR);
438 trapsig(SIGERR_);
439 }
440 return rv;
441 }
442
443 /*
444 * execute simple command
445 */
446
447 static int
448 comexec(t, tp, ap, flags)
449 struct op *t;
450 struct tbl *volatile tp;
451 register char **ap;
452 int volatile flags;
453 {
454 int i;
455 int rv = 0;
456 register char *cp;
457 register char **lastp;
458 static struct op texec; /* Must be static (XXX but why?) */
459 int type_flags;
460 int keepasn_ok;
461 int fcflags = FC_BI|FC_FUNC|FC_PATH;
462 #ifdef __GNUC__
463 (void) &rv;
464 #endif
465
466 /* snag the last argument for $_ XXX not the same as at&t ksh,
467 * which only seems to set $_ after a newline (but not in
468 * functions/dot scripts, but in interactive and scipt) -
469 * perhaps save last arg here and set it in shell()?.
470 */
471 if (*(lastp = ap)) {
472 while (*++lastp)
473 ;
474 setstr(typeset("_", LOCAL, 0, 0, 0), *--lastp);
475 }
476
477 /* Deal with the shell builtins builtin, exec and command since
478 * they can be followed by other commands. This must be done before
479 * we know if we should create a local block, which must be done
480 * before we can do a path search (in case the assignments change
481 * PATH).
482 * Odd cases:
483 * FOO=bar exec > /dev/null FOO is kept but not exported
484 * FOO=bar exec foobar FOO is exported
485 * FOO=bar command exec > /dev/null FOO is neither kept nor exported
486 * FOO=bar command FOO is neither kept nor exported
487 * PATH=... foobar use new PATH in foobar search
488 */
489 keepasn_ok = 1;
490 while (tp && tp->type == CSHELL) {
491 fcflags = FC_BI|FC_FUNC|FC_PATH;/* undo effects of command */
492 if (tp->val.f == c_builtin) {
493 if ((cp = *++ap) == NULL) {
494 tp = NULL;
495 break;
496 }
497 tp = findcom(cp, FC_BI);
498 if (tp == NULL)
499 errorf("builtin: %s: not a builtin", cp);
500 continue;
501 } else if (tp->val.f == c_exec) {
502 if (ap[1] == NULL)
503 break;
504 ap++;
505 flags |= XEXEC;
506 } else if (tp->val.f == c_command) {
507 int optc, saw_p = 0;
508
509 /* Ugly dealing with options in two places (here and
510 * in c_command(), but such is life)
511 */
512 ksh_getopt_reset(&builtin_opt, 0);
513 while ((optc = ksh_getopt(ap, &builtin_opt, ":p"))
514 == 'p')
515 saw_p = 1;
516 if (optc != EOF)
517 break; /* command -vV or something */
518 /* don't look for functions */
519 fcflags = FC_BI|FC_PATH;
520 if (saw_p) {
521 if (Flag(FRESTRICTED)) {
522 warningf(TRUE,
523 "command -p: restricted");
524 rv = 1;
525 goto Leave;
526 }
527 fcflags |= FC_DEFPATH;
528 }
529 ap += builtin_opt.optind;
530 /* POSIX says special builtins lose their status
531 * if accessed using command.
532 */
533 keepasn_ok = 0;
534 if (!ap[0]) {
535 /* ensure command with no args exits with 0 */
536 subst_exstat = 0;
537 break;
538 }
539 } else
540 break;
541 tp = findcom(ap[0], fcflags & (FC_BI|FC_FUNC));
542 }
543 if (keepasn_ok && (!ap[0] || (tp && (tp->flag & KEEPASN))))
544 type_flags = 0;
545 else {
546 /* create new variable/function block */
547 newblock();
548 /* ksh functions don't keep assignments, POSIX functions do. */
549 if (keepasn_ok && tp && tp->type == CFUNC
550 && !(tp->flag & FKSH))
551 type_flags = 0;
552 else
553 type_flags = LOCAL|LOCAL_COPY|EXPORT;
554 }
555 if (Flag(FEXPORT))
556 type_flags |= EXPORT;
557 for (i = 0; t->vars[i]; i++) {
558 cp = evalstr(t->vars[i], DOASNTILDE);
559 if (Flag(FXTRACE)) {
560 if (i == 0)
561 shf_fprintf(shl_out, "%s",
562 PS4_SUBSTITUTE(str_val(global("PS4"))));
563 shf_fprintf(shl_out, "%s%s", cp,
564 t->vars[i + 1] ? space : newline);
565 if (!t->vars[i + 1])
566 shf_flush(shl_out);
567 }
568 typeset(cp, type_flags, 0, 0, 0);
569 }
570
571 if ((cp = *ap) == NULL) {
572 rv = subst_exstat;
573 goto Leave;
574 } else if (!tp) {
575 if (Flag(FRESTRICTED) && ksh_strchr_dirsep(cp)) {
576 warningf(TRUE, "%s: restricted", cp);
577 rv = 1;
578 goto Leave;
579 }
580 tp = findcom(cp, fcflags);
581 }
582
583 switch (tp->type) {
584 case CSHELL: /* shell built-in */
585 rv = call_builtin(tp, ap);
586 break;
587
588 case CFUNC: /* function call */
589 {
590 volatile int old_xflag;
591 volatile Tflag old_inuse;
592 const char *volatile old_kshname;
593
594 if (!(tp->flag & ISSET)) {
595 struct tbl *ftp;
596
597 if (!tp->u.fpath) {
598 if (tp->u2.errno_) {
599 warningf(TRUE,
600 "%s: can't find function definition file - %s",
601 cp, strerror(tp->u2.errno_));
602 rv = 126;
603 } else {
604 warningf(TRUE,
605 "%s: can't find function definition file", cp);
606 rv = 127;
607 }
608 break;
609 }
610 if (include(tp->u.fpath, 0, (char **) 0, 0) < 0) {
611 warningf(TRUE,
612 "%s: can't open function definition file %s - %s",
613 cp, tp->u.fpath, strerror(errno));
614 rv = 127;
615 break;
616 }
617 if (!(ftp = findfunc(cp, hash(cp), FALSE))
618 || !(ftp->flag & ISSET))
619 {
620 warningf(TRUE,
621 "%s: function not defined by %s",
622 cp, tp->u.fpath);
623 rv = 127;
624 break;
625 }
626 tp = ftp;
627 }
628
629 /* ksh functions set $0 to function name, POSIX functions leave
630 * $0 unchanged.
631 */
632 old_kshname = kshname;
633 if (tp->flag & FKSH)
634 kshname = ap[0];
635 e->loc->argv = ap;
636 for (i = 0; *ap++ != NULL; i++)
637 ;
638 e->loc->argc = i - 1;
639 getopts_reset(1);
640
641 old_xflag = Flag(FXTRACE);
642 Flag(FXTRACE) = tp->flag & TRACE ? TRUE : FALSE;
643
644 old_inuse = tp->flag & FINUSE;
645 tp->flag |= FINUSE;
646
647 e->type = E_FUNC;
648 i = ksh_sigsetjmp(e->jbuf, 0);
649 if (i == 0) {
650 /* seems odd to pass XERROK here, but at&t ksh does */
651 exstat = execute(tp->val.t, flags & XERROK);
652 i = LRETURN;
653 }
654 kshname = old_kshname;
655 Flag(FXTRACE) = old_xflag;
656 tp->flag = (tp->flag & ~FINUSE) | old_inuse;
657 /* Were we deleted while executing? If so, free the execution
658 * tree. todo: Unfortunately, the table entry is never re-used
659 * until the lookup table is expanded.
660 */
661 if ((tp->flag & (FDELETE|FINUSE)) == FDELETE) {
662 if (tp->flag & ALLOC) {
663 tp->flag &= ~ALLOC;
664 tfree(tp->val.t, tp->areap);
665 }
666 tp->flag = 0;
667 }
668 switch (i) {
669 case LRETURN:
670 case LERROR:
671 rv = exstat;
672 break;
673 case LINTR:
674 case LEXIT:
675 case LLEAVE:
676 case LSHELL:
677 quitenv();
678 unwind(i);
679 /*NOTREACHED*/
680 default:
681 quitenv();
682 internal_errorf(1, "CFUNC %d", i);
683 }
684 break;
685 }
686
687 case CEXEC: /* executable command */
688 case CTALIAS: /* tracked alias */
689 if (!(tp->flag&ISSET)) {
690 /* errno_ will be set if the named command was found
691 * but could not be executed (permissions, no execute
692 * bit, directory, etc). Print out a (hopefully)
693 * useful error message and set the exit status to 126.
694 */
695 if (tp->u2.errno_) {
696 warningf(TRUE, "%s: cannot execute - %s", cp,
697 strerror(tp->u2.errno_));
698 rv = 126; /* POSIX */
699 } else {
700 warningf(TRUE, "%s: not found", cp);
701 rv = 127;
702 }
703 break;
704 }
705
706 /* set $_ to program's full path */
707 setstr(typeset("_", LOCAL|EXPORT, 0, 0, 0), tp->val.s);
708
709 if (flags&XEXEC) {
710 j_exit();
711 if (!(flags&XBGND) || Flag(FMONITOR)) {
712 setexecsig(&sigtraps[SIGINT], SS_RESTORE_ORIG);
713 setexecsig(&sigtraps[SIGQUIT], SS_RESTORE_ORIG);
714 }
715 }
716
717 /* to fork we set up a TEXEC node and call execute */
718 texec.type = TEXEC;
719 texec.left = t; /* for tprint */
720 texec.str = tp->val.s;
721 texec.args = ap;
722 rv = exchild(&texec, flags, -1);
723 break;
724 }
725 Leave:
726 if (flags & XEXEC) {
727 exstat = rv;
728 unwind(LLEAVE);
729 }
730 return rv;
731 }
732
733 static void
734 scriptexec(tp, ap)
735 register struct op *tp;
736 register char **ap;
737 {
738 char *shell;
739
740 shell = str_val(global(EXECSHELL_STR));
741 if (shell && *shell)
742 shell = search(shell, path, X_OK, (int *) 0);
743 if (!shell || !*shell)
744 shell = EXECSHELL;
745
746 *tp->args-- = tp->str;
747 #ifdef SHARPBANG
748 {
749 char buf[LINE];
750 register char *cp;
751 register int fd, n;
752
753 buf[0] = '\0';
754 if ((fd = open(tp->str, O_RDONLY)) >= 0) {
755 if ((n = read(fd, buf, LINE - 1)) > 0)
756 buf[n] = '\0';
757 (void) close(fd);
758 }
759 if ((buf[0] == '#' && buf[1] == '!' && (cp = &buf[2]))
760 # ifdef OS2
761 || (strncmp(buf, "extproc", 7) == 0 && isspace(buf[7])
762 && (cp = &buf[7]))
763 # endif /* OS2 */
764 )
765 {
766 while (*cp && (*cp == ' ' || *cp == '\t'))
767 cp++;
768 if (*cp && *cp != '\n') {
769 char *a0 = cp, *a1 = (char *) 0;
770 # ifdef OS2
771 char *a2 = cp;
772 # endif /* OS2 */
773
774 while (*cp && *cp != '\n' && *cp != ' '
775 && *cp != '\t')
776 {
777 # ifdef OS2
778 /* Allow shell search without prepended path
779 * if shell with / in pathname cannot be found.
780 * Use / explicitly so \ can be used if explicit
781 * needs to be forced.
782 */
783 if (*cp == '/')
784 a2 = cp + 1;
785 # endif /* OS2 */
786 cp++;
787 }
788 if (*cp && *cp != '\n') {
789 *cp++ = '\0';
790 while (*cp
791 && (*cp == ' ' || *cp == '\t'))
792 cp++;
793 if (*cp && *cp != '\n') {
794 a1 = cp;
795 /* all one argument */
796 while (*cp && *cp != '\n')
797 cp++;
798 }
799 }
800 if (*cp == '\n') {
801 *cp = '\0';
802 if (a1)
803 *tp->args-- = a1;
804 # ifdef OS2
805 if (a0 != a2 && search_access(a0, X_OK, (int *) 0))
806 a0 = a2;
807 # endif /* OS2 */
808 shell = a0;
809 }
810 }
811 # ifdef OS2
812 } else {
813 /* Use ksh documented shell default if present
814 * else use OS2_SHELL which is assumed to need
815 * the /c option and '\' as dir separater.
816 */
817 char *p = shell;
818
819 shell = str_val(global("EXECSHELL"));
820 if (shell && *shell)
821 shell = search(shell, path, X_OK, (int *) 0);
822 if (!shell || !*shell) {
823 shell = p;
824 *tp->args-- = "/c";
825 for (p = tp->str; *p; p++)
826 if (*p == '/')
827 *p = '\\';
828 }
829 # endif /* OS2 */
830 }
831 }
832 #endif /* SHARPBANG */
833 *tp->args = shell;
834
835 ksh_execve(tp->args[0], tp->args, ap);
836
837 /* report both the program that was run and the bogus shell */
838 errorf("%s: %s: %s", tp->str, shell, strerror(errno));
839 }
840
841 int
842 shcomexec(wp)
843 register char **wp;
844 {
845 register struct tbl *tp;
846
847 tp = tsearch(&builtins, *wp, hash(*wp));
848 if (tp == NULL)
849 internal_errorf(1, "shcomexec: %s", *wp);
850 return call_builtin(tp, wp);
851 }
852
853 /*
854 * Search function tables for a function. If create set, a table entry
855 * is created if none is found.
856 */
857 struct tbl *
858 findfunc(name, h, create)
859 const char *name;
860 unsigned int h;
861 int create;
862 {
863 struct block *l;
864 struct tbl *tp = (struct tbl *) 0;
865
866 for (l = e->loc; l; l = l->next) {
867 tp = tsearch(&l->funs, name, h);
868 if (tp)
869 break;
870 if (!l->next && create) {
871 tp = tenter(&l->funs, name, h);
872 tp->flag = DEFINED;
873 tp->type = CFUNC;
874 tp->val.t = (struct op *) 0;
875 break;
876 }
877 }
878 return tp;
879 }
880
881 /*
882 * define function. Returns 1 if function is being undefined (t == 0) and
883 * function did not exist, returns 0 otherwise.
884 */
885 int
886 define(name, t)
887 const char *name;
888 struct op *t;
889 {
890 struct tbl *tp;
891 int was_set = 0;
892
893 while (1) {
894 tp = findfunc(name, hash(name), TRUE);
895
896 if (tp->flag & ISSET)
897 was_set = 1;
898 /* If this function is currently being executed, we zap this
899 * table entry so findfunc() won't see it
900 */
901 if (tp->flag & FINUSE) {
902 tp->name[0] = '\0';
903 tp->flag &= ~DEFINED; /* ensure it won't be found */
904 tp->flag |= FDELETE;
905 } else
906 break;
907 }
908
909 if (tp->flag & ALLOC) {
910 tp->flag &= ~(ISSET|ALLOC);
911 tfree(tp->val.t, tp->areap);
912 }
913
914 if (t == NULL) { /* undefine */
915 tdelete(tp);
916 return was_set ? 0 : 1;
917 }
918
919 tp->val.t = tcopy(t->left, tp->areap);
920 tp->flag |= (ISSET|ALLOC);
921 if (t->u.ksh_func)
922 tp->flag |= FKSH;
923
924 return 0;
925 }
926
927 /*
928 * add builtin
929 */
930 void
931 builtin(name, func)
932 const char *name;
933 int (*func) ARGS((char **));
934 {
935 register struct tbl *tp;
936 Tflag flag;
937
938 /* see if any flags should be set for this builtin */
939 for (flag = 0; ; name++) {
940 if (*name == '=') /* command does variable assignment */
941 flag |= KEEPASN;
942 else if (*name == '*') /* POSIX special builtin */
943 flag |= SPEC_BI;
944 else if (*name == '+') /* POSIX regular builtin */
945 flag |= REG_BI;
946 else
947 break;
948 }
949
950 tp = tenter(&builtins, name, hash(name));
951 tp->flag = DEFINED | flag;
952 tp->type = CSHELL;
953 tp->val.f = func;
954 }
955
956 /*
957 * find command
958 * either function, hashed command, or built-in (in that order)
959 */
960 struct tbl *
961 findcom(name, flags)
962 const char *name;
963 int flags; /* FC_* */
964 {
965 static struct tbl temp;
966 unsigned int h = hash(name);
967 struct tbl *tp = NULL, *tbi;
968 int insert = Flag(FTRACKALL); /* insert if not found */
969 char *fpath; /* for function autoloading */
970 char *npath;
971
972 if (ksh_strchr_dirsep(name) != NULL) {
973 insert = 0;
974 /* prevent FPATH search below */
975 flags &= ~FC_FUNC;
976 goto Search;
977 }
978 tbi = (flags & FC_BI) ? tsearch(&builtins, name, h) : NULL;
979 /* POSIX says special builtins first, then functions, then
980 * POSIX regular builtins, then search path...
981 */
982 if ((flags & FC_SPECBI) && tbi && (tbi->flag & SPEC_BI))
983 tp = tbi;
984 if (!tp && (flags & FC_FUNC)) {
985 tp = findfunc(name, h, FALSE);
986 if (tp && !(tp->flag & ISSET)) {
987 if ((fpath = str_val(global("FPATH"))) == null) {
988 tp->u.fpath = (char *) 0;
989 tp->u2.errno_ = 0;
990 } else
991 tp->u.fpath = search(name, fpath, R_OK,
992 &tp->u2.errno_);
993 }
994 }
995 if (!tp && (flags & FC_REGBI) && tbi && (tbi->flag & REG_BI))
996 tp = tbi;
997 /* todo: posix says non-special/non-regular builtins must
998 * be triggered by some user-controllable means like a
999 * special directory in PATH. Requires modifications to
1000 * the search() function. Tracked aliases should be
1001 * modified to allow tracking of builtin commands.
1002 * This should be under control of the FPOSIX flag.
1003 * If this is changed, also change c_whence...
1004 */
1005 if (!tp && (flags & FC_UNREGBI) && tbi)
1006 tp = tbi;
1007 if (!tp && (flags & FC_PATH) && !(flags & FC_DEFPATH)) {
1008 tp = tsearch(&taliases, name, h);
1009 if (tp && (tp->flag & ISSET) && eaccess(tp->val.s, X_OK) != 0) {
1010 if (tp->flag & ALLOC) {
1011 tp->flag &= ~ALLOC;
1012 afree(tp->val.s, APERM);
1013 }
1014 tp->flag &= ~ISSET;
1015 }
1016 }
1017
1018 Search:
1019 if ((!tp || (tp->type == CTALIAS && !(tp->flag&ISSET)))
1020 && (flags & FC_PATH))
1021 {
1022 if (!tp) {
1023 if (insert && !(flags & FC_DEFPATH)) {
1024 tp = tenter(&taliases, name, h);
1025 tp->type = CTALIAS;
1026 } else {
1027 tp = &temp;
1028 tp->type = CEXEC;
1029 }
1030 tp->flag = DEFINED; /* make ~ISSET */
1031 }
1032 npath = search(name, flags & FC_DEFPATH ? def_path : path,
1033 X_OK, &tp->u2.errno_);
1034 if (npath) {
1035 tp->val.s = tp == &temp ? npath : str_save(npath, APERM);
1036 tp->flag |= ISSET|ALLOC;
1037 } else if ((flags & FC_FUNC)
1038 && (fpath = str_val(global("FPATH"))) != null
1039 && (npath = search(name, fpath, R_OK,
1040 &tp->u2.errno_)) != (char *) 0)
1041 {
1042 /* An undocumented feature of at&t ksh is that it
1043 * searches FPATH if a command is not found, even
1044 * if the command hasn't been set up as an autoloaded
1045 * function (ie, no typeset -uf).
1046 */
1047 tp = &temp;
1048 tp->type = CFUNC;
1049 tp->flag = DEFINED; /* make ~ISSET */
1050 tp->u.fpath = npath;
1051 }
1052 }
1053 return tp;
1054 }
1055
1056 /*
1057 * flush executable commands with relative paths
1058 */
1059 void
1060 flushcom(all)
1061 int all; /* just relative or all */
1062 {
1063 struct tbl *tp;
1064 struct tstate ts;
1065
1066 for (twalk(&ts, &taliases); (tp = tnext(&ts)) != NULL; )
1067 if ((tp->flag&ISSET) && (all || !ISDIRSEP(tp->val.s[0]))) {
1068 if (tp->flag&ALLOC) {
1069 tp->flag &= ~(ALLOC|ISSET);
1070 afree(tp->val.s, APERM);
1071 }
1072 tp->flag = ~ISSET;
1073 }
1074 }
1075
1076 /* Check if path is something we want to find. Returns -1 for failure. */
1077 int
1078 search_access(path, mode, errnop)
1079 const char *path;
1080 int mode;
1081 int *errnop; /* set if candidate found, but not suitable */
1082 {
1083 #ifndef OS2
1084 int ret, err = 0;
1085 struct stat statb;
1086
1087 if (stat(path, &statb) < 0)
1088 return -1;
1089 ret = eaccess(path, mode);
1090 if (ret < 0)
1091 err = errno; /* File exists, but we can't access it */
1092 else if (mode == X_OK && (!S_ISREG(statb.st_mode)
1093 /* This 'cause access() says root can execute everything */
1094 || !(statb.st_mode & (S_IXUSR|S_IXGRP|S_IXOTH))))
1095 {
1096 ret = -1;
1097 err = S_ISDIR(statb.st_mode) ? EISDIR : EACCES;
1098 }
1099 if (err && errnop)
1100 *errnop = err;
1101 return ret;
1102 #else /* !OS2 */
1103 /*
1104 * NOTE: ASSUMES path can be modified and has enough room at the
1105 * end of the string for a suffix (ie, 4 extra characters).
1106 * Certain code knows this (eg, eval.c(globit()),
1107 * exec.c(search())).
1108 */
1109 static char *xsuffixes[] = { ".ksh", ".exe", ".", ".sh", ".cmd",
1110 ".com", ".bat", (char *) 0
1111 };
1112 static char *rsuffixes[] = { ".ksh", ".", ".sh", ".cmd", ".bat",
1113 (char *) 0
1114 };
1115 int i;
1116 char *mpath = (char *) path;
1117 char *tp = mpath + strlen(mpath);
1118 char *p;
1119 char **sfx;
1120
1121 /* If a suffix has been specified, check if it is one of the
1122 * suffixes that indicate the file is executable - if so, change
1123 * the access test to R_OK...
1124 * This code assumes OS/2 files can have only one suffix...
1125 */
1126 if ((p = strrchr((p = ksh_strrchr_dirsep(mpath)) ? p : mpath, '.'))) {
1127 if (mode == X_OK)
1128 mode = R_OK;
1129 return search_access1(mpath, mode, errnop);
1130 }
1131 /* Try appending the various suffixes. Different suffixes for
1132 * read and execute 'cause we don't want to read an executable...
1133 */
1134 sfx = mode == R_OK ? rsuffixes : xsuffixes;
1135 for (i = 0; sfx[i]; i++) {
1136 strcpy(tp, p = sfx[i]);
1137 if (search_access1(mpath, R_OK, errnop) == 0)
1138 return 0;
1139 *tp = '\0';
1140 }
1141 return -1;
1142 #endif /* !OS2 */
1143 }
1144
1145 #ifdef OS2
1146 static int
1147 search_access1(path, mode, errnop)
1148 const char *path;
1149 int mode;
1150 int *errnop; /* set if candidate found, but not suitable */
1151 {
1152 int ret, err = 0;
1153 struct stat statb;
1154
1155 if (stat(path, &statb) < 0)
1156 return -1;
1157 ret = eaccess(path, mode);
1158 if (ret < 0)
1159 err = errno; /* File exists, but we can't access it */
1160 else if (!S_ISREG(statb.st_mode)) {
1161 ret = -1;
1162 err = S_ISDIR(statb.st_mode) ? EISDIR : EACCES;
1163 }
1164 if (err && errnop)
1165 *errnop = err;
1166 return ret;
1167 }
1168 #endif /* OS2 */
1169
1170 /*
1171 * search for command with PATH
1172 */
1173 char *
1174 search(name, path, mode, errnop)
1175 const char *name;
1176 const char *path;
1177 int mode; /* R_OK or X_OK */
1178 int *errnop; /* set if candidate found, but not suitable */
1179 {
1180 const char *sp, *p;
1181 char *xp;
1182 XString xs;
1183 int namelen;
1184
1185 if (errnop)
1186 *errnop = 0;
1187 #ifdef OS2
1188 /* Xinit() allocates 8 additional bytes, so appended suffixes won't
1189 * overflow the memory.
1190 */
1191 namelen = strlen(name) + 1;
1192 Xinit(xs, xp, namelen, ATEMP);
1193 memcpy(Xstring(xs, xp), name, namelen);
1194
1195 if (ksh_strchr_dirsep(name)) {
1196 if (search_access(Xstring(xs, xp), mode, errnop) >= 0)
1197 return Xstring(xs, xp); /* not Xclose() - see above */
1198 Xfree(xs, xp);
1199 return NULL;
1200 }
1201
1202 /* Look in current context always. (os2 style) */
1203 if (search_access(Xstring(xs, xp), mode, errnop) == 0)
1204 return Xstring(xs, xp); /* not Xclose() - xp may be wrong */
1205 #else /* OS2 */
1206 if (ksh_strchr_dirsep(name)) {
1207 if (search_access(name, mode, errnop) == 0)
1208 return (char *) name;
1209 return NULL;
1210 }
1211
1212 namelen = strlen(name) + 1;
1213 Xinit(xs, xp, 128, ATEMP);
1214 #endif /* OS2 */
1215
1216 sp = path;
1217 while (sp != NULL) {
1218 xp = Xstring(xs, xp);
1219 if (!(p = strchr(sp, PATHSEP)))
1220 p = sp + strlen(sp);
1221 if (p != sp) {
1222 XcheckN(xs, xp, p - sp);
1223 memcpy(xp, sp, p - sp);
1224 xp += p - sp;
1225 *xp++ = DIRSEP;
1226 }
1227 sp = p;
1228 XcheckN(xs, xp, namelen);
1229 memcpy(xp, name, namelen);
1230 if (search_access(Xstring(xs, xp), mode, errnop) == 0)
1231 #ifdef OS2
1232 return Xstring(xs, xp); /* Not Xclose() - see above */
1233 #else /* OS2 */
1234 return Xclose(xs, xp + namelen);
1235 #endif /* OS2 */
1236 if (*sp++ == '\0')
1237 sp = NULL;
1238 }
1239 Xfree(xs, xp);
1240 return NULL;
1241 }
1242
1243 static int
1244 call_builtin(tp, wp)
1245 struct tbl *tp;
1246 char **wp;
1247 {
1248 int rv;
1249
1250 builtin_argv0 = wp[0];
1251 builtin_flag = tp->flag;
1252 shf_reopen(1, SHF_WR, shl_stdout);
1253 shl_stdout_ok = 1;
1254 ksh_getopt_reset(&builtin_opt, GF_ERROR);
1255 rv = (*tp->val.f)(wp);
1256 shf_flush(shl_stdout);
1257 shl_stdout_ok = 0;
1258 builtin_flag = 0;
1259 builtin_argv0 = (char *) 0;
1260 return rv;
1261 }
1262
1263 /*
1264 * set up redirection, saving old fd's in e->savefd
1265 */
1266 static int
1267 iosetup(iop, tp)
1268 register struct ioword *iop;
1269 struct tbl *tp;
1270 {
1271 register int u = -1;
1272 char *cp = iop->name;
1273 int iotype = iop->flag & IOTYPE;
1274 int do_open = 1, do_close = 0, UNINITIALIZED(flags);
1275 struct ioword iotmp;
1276 struct stat statb;
1277
1278 if (iotype != IOHERE)
1279 cp = evalonestr(cp, DOTILDE|(Flag(FTALKING) ? DOGLOB : 0));
1280
1281 /* Used for tracing and error messages to print expanded cp */
1282 iotmp = *iop;
1283 iotmp.name = (iotype == IOHERE) ? (char *) 0 : cp;
1284 iotmp.flag |= IONAMEXP;
1285
1286 if (Flag(FXTRACE))
1287 shellf("%s%s\n",
1288 PS4_SUBSTITUTE(str_val(global("PS4"))),
1289 snptreef((char *) 0, 32, "%R", &iotmp));
1290
1291 switch (iotype) {
1292 case IOREAD:
1293 flags = O_RDONLY;
1294 break;
1295
1296 case IOCAT:
1297 flags = O_WRONLY | O_APPEND | O_CREAT;
1298 break;
1299
1300 case IOWRITE:
1301 flags = O_WRONLY | O_CREAT | O_TRUNC;
1302 if (Flag(FNOCLOBBER) && !(iop->flag & IOCLOB)
1303 && (stat(cp, &statb) < 0 || S_ISREG(statb.st_mode)))
1304 flags |= O_EXCL;
1305 break;
1306
1307 case IORDWR:
1308 flags = O_RDWR | O_CREAT;
1309 break;
1310
1311 case IOHERE:
1312 do_open = 0;
1313 /* herein() returns -2 if error has been printed */
1314 u = herein(cp, iop->flag & IOEVAL);
1315 /* cp may have wrong name */
1316 break;
1317
1318 case IODUP:
1319 {
1320 const char *emsg;
1321
1322 do_open = 0;
1323 if (*cp == '-' && !cp[1]) {
1324 u = 1009; /* prevent error return below */
1325 do_close = 1;
1326 } else if ((u = check_fd(cp,
1327 X_OK | ((iop->flag & IORDUP) ? R_OK : W_OK),
1328 &emsg)) < 0)
1329 {
1330 warningf(TRUE, "%s: %s",
1331 snptreef((char *) 0, 32, "%R", &iotmp), emsg);
1332 return -1;
1333 }
1334 break;
1335 }
1336 }
1337 if (do_open) {
1338 if (Flag(FRESTRICTED) && (flags & O_CREAT)) {
1339 warningf(TRUE, "%s: restricted", cp);
1340 return -1;
1341 }
1342 u = open(cp, flags, 0666);
1343 #ifdef OS2
1344 if (u < 0 && strcmp(cp, "/dev/null") == 0)
1345 u = open("nul", flags, 0666);
1346 #endif /* OS2 */
1347 }
1348 if (u < 0) {
1349 /* herein() may already have printed message */
1350 if (u == -1)
1351 warningf(TRUE, "cannot %s %s: %s",
1352 iotype == IODUP ? "dup"
1353 : (iotype == IOREAD || iotype == IOHERE) ?
1354 "open" : "create", cp, strerror(errno));
1355 return -1;
1356 }
1357 /* Do not save if it has already been redirected (i.e. "cat >x >y"). */
1358 if (e->savefd[iop->unit] == 0)
1359 /* c_exec() assumes e->savefd[fd] set for any redirections.
1360 * Ask savefd() not to close iop->unit - allows error messages
1361 * to be seen if iop->unit is 2; also means we can't lose
1362 * the fd (eg, both dup2 below and dup2 in restfd() failing).
1363 */
1364 e->savefd[iop->unit] = savefd(iop->unit, 1);
1365
1366 if (do_close)
1367 close(iop->unit);
1368 else if (u != iop->unit) {
1369 if (ksh_dup2(u, iop->unit, TRUE) < 0) {
1370 warningf(TRUE,
1371 "could not finish (dup) redirection %s: %s",
1372 snptreef((char *) 0, 32, "%R", &iotmp),
1373 strerror(errno));
1374 if (iotype != IODUP)
1375 close(u);
1376 return -1;
1377 }
1378 if (iotype != IODUP)
1379 close(u);
1380 #ifdef KSH
1381 /* Touching any co-process fd in an empty exec
1382 * causes the shell to close its copies
1383 */
1384 else if (tp && tp->type == CSHELL && tp->val.f == c_exec) {
1385 if (iop->flag & IORDUP) /* possible exec <&p */
1386 coproc_read_close(u);
1387 else /* possible exec >&p */
1388 coproc_write_close(u);
1389 }
1390 #endif /* KSH */
1391 }
1392 if (u == 2) /* Clear any write errors */
1393 shf_reopen(2, SHF_WR, shl_out);
1394 return 0;
1395 }
1396
1397 /*
1398 * open here document temp file.
1399 * if unquoted here, expand here temp file into second temp file.
1400 */
1401 static int
1402 herein(hname, sub)
1403 char *hname;
1404 int sub;
1405 {
1406 int fd;
1407
1408 /* ksh -c 'cat << EOF' can cause this... */
1409 if (hname == (char *) 0) {
1410 warningf(TRUE, "here document missing");
1411 return -2; /* special to iosetup(): don't print error */
1412 }
1413 if (sub) {
1414 char *cp;
1415 struct source *s, *volatile osource = source;
1416 struct temp *h;
1417 struct shf *volatile shf;
1418 int i;
1419
1420 /* must be before newenv() 'cause shf uses ATEMP */
1421 shf = shf_open(hname, O_RDONLY, 0, SHF_MAPHI|SHF_CLEXEC);
1422 if (shf == NULL)
1423 return -1;
1424 newenv(E_ERRH);
1425 i = ksh_sigsetjmp(e->jbuf, 0);
1426 if (i) {
1427 if (shf)
1428 shf_close(shf);
1429 source = osource;
1430 quitenv(); /* after shf_close() due to alloc */
1431 return -2; /* special to iosetup(): don't print error */
1432 }
1433 /* set up yylex input from here file */
1434 s = pushs(SFILE, ATEMP);
1435 s->u.shf = shf;
1436 source = s;
1437 if (yylex(ONEWORD) != LWORD)
1438 internal_errorf(1, "herein: yylex");
1439 shf_close(shf);
1440 shf = (struct shf *) 0;
1441 cp = evalstr(yylval.cp, 0);
1442
1443 /* write expanded input to another temp file */
1444 h = maketemp(ATEMP);
1445 h->next = e->temps; e->temps = h;
1446 if (!(shf = h->shf) || (fd = open(h->name, O_RDONLY, 0)) < 0)
1447 /* shf closeed by error handler */
1448 errorf("%s: %s", h->name, strerror(errno));
1449 shf_puts(cp, shf);
1450 if (shf_close(shf) == EOF) {
1451 close(fd);
1452 shf = (struct shf *) 0;
1453 errorf("error writing %s: %s", h->name,
1454 strerror(errno));
1455 }
1456 shf = (struct shf *) 0;
1457
1458 quitenv();
1459 } else {
1460 fd = open(hname, O_RDONLY, 0);
1461 if (fd < 0)
1462 return -1;
1463 }
1464
1465 return fd;
1466 }
1467
1468 #ifdef KSH
1469 /*
1470 * ksh special - the select command processing section
1471 * print the args in column form - assuming that we can
1472 */
1473 static char *
1474 do_selectargs(ap, print_menu)
1475 register char **ap;
1476 bool_t print_menu;
1477 {
1478 static const char *const read_args[] = {
1479 "read", "-r", "REPLY", (char *) 0
1480 };
1481 char *s;
1482 int i, argct;
1483
1484 for (argct = 0; ap[argct]; argct++)
1485 ;
1486 while (1) {
1487 /* Menu is printed if
1488 * - this is the first time around the select loop
1489 * - the user enters a blank line
1490 * - the REPLY parameter is empty
1491 */
1492 if (print_menu || !*str_val(global("REPLY")))
1493 pr_menu(ap);
1494 shellf("%s", str_val(global("PS3")));
1495 if (call_builtin(findcom("read", FC_BI), (char **) read_args))
1496 return (char *) 0;
1497 s = str_val(global("REPLY"));
1498 if (*s) {
1499 i = atoi(s);
1500 return (i >= 1 && i <= argct) ? ap[i - 1] : null;
1501 }
1502 print_menu = 1;
1503 }
1504 }
1505
1506 struct select_menu_info {
1507 char *const *args;
1508 int arg_width;
1509 int num_width;
1510 } info;
1511
1512 static char *select_fmt_entry ARGS((void *arg, int i, char *buf, int buflen));
1513
1514 /* format a single select menu item */
1515 static char *
1516 select_fmt_entry(arg, i, buf, buflen)
1517 void *arg;
1518 int i;
1519 char *buf;
1520 int buflen;
1521 {
1522 struct select_menu_info *smi = (struct select_menu_info *) arg;
1523
1524 shf_snprintf(buf, buflen, "%*d) %s",
1525 smi->num_width, i + 1, smi->args[i]);
1526 return buf;
1527 }
1528
1529 /*
1530 * print a select style menu
1531 */
1532 int
1533 pr_menu(ap)
1534 char *const *ap;
1535 {
1536 struct select_menu_info smi;
1537 char *const *pp;
1538 int nwidth, dwidth;
1539 int i, n;
1540
1541 /* Width/column calculations were done once and saved, but this
1542 * means select can't be used recursively so we re-calculate each
1543 * time (could save in a structure that is returned, but its probably
1544 * not worth the bother).
1545 */
1546
1547 /*
1548 * get dimensions of the list
1549 */
1550 for (n = 0, nwidth = 0, pp = ap; *pp; n++, pp++) {
1551 i = strlen(*pp);
1552 nwidth = (i > nwidth) ? i : nwidth;
1553 }
1554 /*
1555 * we will print an index of the form
1556 * %d)
1557 * in front of each entry
1558 * get the max width of this
1559 */
1560 for (i = n, dwidth = 1; i >= 10; i /= 10)
1561 dwidth++;
1562
1563 smi.args = ap;
1564 smi.arg_width = nwidth;
1565 smi.num_width = dwidth;
1566 print_columns(shl_out, n, select_fmt_entry, (void *) &smi,
1567 dwidth + nwidth + 2);
1568
1569 return n;
1570 }
1571 #endif /* KSH */
1572 #ifdef KSH
1573
1574 /*
1575 * [[ ... ]] evaluation routines
1576 */
1577
1578 extern const char *const dbtest_tokens[];
1579 extern const char db_close[];
1580
1581 /* Test if the current token is a whatever. Accepts the current token if
1582 * it is. Returns 0 if it is not, non-zero if it is (in the case of
1583 * TM_UNOP and TM_BINOP, the returned value is a Test_op).
1584 */
1585 static int
1586 dbteste_isa(te, meta)
1587 Test_env *te;
1588 Test_meta meta;
1589 {
1590 int ret = 0;
1591 int uqword;
1592 char *p;
1593
1594 if (!*te->pos.wp)
1595 return meta == TM_END;
1596
1597 /* unquoted word? */
1598 for (p = *te->pos.wp; *p == CHAR; p += 2)
1599 ;
1600 uqword = *p == EOS;
1601
1602 if (meta == TM_UNOP || meta == TM_BINOP) {
1603 if (uqword) {
1604 char buf[8]; /* longer than the longest operator */
1605 char *q = buf;
1606 for (p = *te->pos.wp; *p == CHAR
1607 && q < &buf[sizeof(buf) - 1];
1608 p += 2)
1609 *q++ = p[1];
1610 *q = '\0';
1611 ret = (int) test_isop(te, meta, buf);
1612 }
1613 } else if (meta == TM_END)
1614 ret = 0;
1615 else
1616 ret = uqword
1617 && strcmp(*te->pos.wp, dbtest_tokens[(int) meta]) == 0;
1618
1619 /* Accept the token? */
1620 if (ret)
1621 te->pos.wp++;
1622
1623 return ret;
1624 }
1625
1626 static const char *
1627 dbteste_getopnd(te, op, do_eval)
1628 Test_env *te;
1629 Test_op op;
1630 int do_eval;
1631 {
1632 char *s = *te->pos.wp;
1633
1634 if (!s)
1635 return (char *) 0;
1636
1637 te->pos.wp++;
1638
1639 if (!do_eval)
1640 return null;
1641
1642 if (op == TO_STEQL || op == TO_STNEQ)
1643 s = evalstr(s, DOTILDE | DOPAT);
1644 else
1645 s = evalstr(s, DOTILDE);
1646
1647 return s;
1648 }
1649
1650 static int
1651 dbteste_eval(te, op, opnd1, opnd2, do_eval)
1652 Test_env *te;
1653 Test_op op;
1654 const char *opnd1;
1655 const char *opnd2;
1656 int do_eval;
1657 {
1658 return test_eval(te, op, opnd1, opnd2, do_eval);
1659 }
1660
1661 static void
1662 dbteste_error(te, offset, msg)
1663 Test_env *te;
1664 int offset;
1665 const char *msg;
1666 {
1667 te->flags |= TEF_ERROR;
1668 internal_errorf(0, "dbteste_error: %s (offset %d)", msg, offset);
1669 }
1670 #endif /* KSH */
1671