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