1 /* 2 * test(1); version 7-like -- author Erik Baalbergen 3 * modified by Eric Gisin to be used as built-in. 4 * modified by Arnold Robbins to add SVR3 compatibility 5 * (-x -c -b -p -u -g -k) plus Korn's -L -nt -ot -ef and new -S (socket). 6 * modified by Michael Rendell to add Korn's [[ .. ]] expressions. 7 * modified by J.T. Conklin to add POSIX compatibility. 8 */ 9 10 #include "sh.h" 11 #include "ksh_stat.h" 12 #include "c_test.h" 13 14 /* test(1) accepts the following grammar: 15 oexpr ::= aexpr | aexpr "-o" oexpr ; 16 aexpr ::= nexpr | nexpr "-a" aexpr ; 17 nexpr ::= primary | "!" nexpr ; 18 primary ::= unary-operator operand 19 | operand binary-operator operand 20 | operand 21 | "(" oexpr ")" 22 ; 23 24 unary-operator ::= "-a"|"-r"|"-w"|"-x"|"-e"|"-f"|"-d"|"-c"|"-b"|"-p"| 25 "-u"|"-g"|"-k"|"-s"|"-t"|"-z"|"-n"|"-o"|"-O"|"-G"| 26 "-L"|"-h"|"-S"|"-H"; 27 28 binary-operator ::= "="|"=="|"!="|"-eq"|"-ne"|"-ge"|"-gt"|"-le"|"-lt"| 29 "-nt"|"-ot"|"-ef"| 30 "<"|">" # rules used for [[ .. ]] expressions 31 ; 32 operand ::= <any thing> 33 */ 34 35 #define T_ERR_EXIT 2 /* POSIX says > 1 for errors */ 36 37 struct t_op { 38 char op_text[4]; 39 Test_op op_num; 40 }; 41 static const struct t_op u_ops [] = { 42 {"-a", TO_FILAXST }, 43 {"-b", TO_FILBDEV }, 44 {"-c", TO_FILCDEV }, 45 {"-d", TO_FILID }, 46 {"-e", TO_FILEXST }, 47 {"-f", TO_FILREG }, 48 {"-G", TO_FILGID }, 49 {"-g", TO_FILSETG }, 50 {"-h", TO_FILSYM }, 51 {"-H", TO_FILCDF }, 52 {"-k", TO_FILSTCK }, 53 {"-L", TO_FILSYM }, 54 {"-n", TO_STNZE }, 55 {"-O", TO_FILUID }, 56 {"-o", TO_OPTION }, 57 {"-p", TO_FILFIFO }, 58 {"-r", TO_FILRD }, 59 {"-s", TO_FILGZ }, 60 {"-S", TO_FILSOCK }, 61 {"-t", TO_FILTT }, 62 {"-u", TO_FILSETU }, 63 {"-w", TO_FILWR }, 64 {"-x", TO_FILEX }, 65 {"-z", TO_STZER }, 66 {"", TO_NONOP } 67 }; 68 static const struct t_op b_ops [] = { 69 {"=", TO_STEQL }, 70 #ifdef KSH 71 {"==", TO_STEQL }, 72 #endif /* KSH */ 73 {"!=", TO_STNEQ }, 74 {"<", TO_STLT }, 75 {">", TO_STGT }, 76 {"-eq", TO_INTEQ }, 77 {"-ne", TO_INTNE }, 78 {"-gt", TO_INTGT }, 79 {"-ge", TO_INTGE }, 80 {"-lt", TO_INTLT }, 81 {"-le", TO_INTLE }, 82 {"-ef", TO_FILEQ }, 83 {"-nt", TO_FILNT }, 84 {"-ot", TO_FILOT }, 85 {"", TO_NONOP } 86 }; 87 88 static int test_stat ARGS((const char *path, struct stat *statb)); 89 static int test_eaccess ARGS((const char *path, int mode)); 90 static int test_oexpr ARGS((Test_env *te, int do_eval)); 91 static int test_aexpr ARGS((Test_env *te, int do_eval)); 92 static int test_nexpr ARGS((Test_env *te, int do_eval)); 93 static int test_primary ARGS((Test_env *te, int do_eval)); 94 static int ptest_isa ARGS((Test_env *te, Test_meta meta)); 95 static const char *ptest_getopnd ARGS((Test_env *te, Test_op op, int do_eval)); 96 static int ptest_eval ARGS((Test_env *te, Test_op op, const char *opnd1, 97 const char *opnd2, int do_eval)); 98 static void ptest_error ARGS((Test_env *te, int offset, const char *msg)); 99 100 int 101 c_test(wp) 102 char **wp; 103 { 104 int argc; 105 int res; 106 Test_env te; 107 108 te.flags = 0; 109 te.isa = ptest_isa; 110 te.getopnd = ptest_getopnd; 111 te.eval = ptest_eval; 112 te.error = ptest_error; 113 114 for (argc = 0; wp[argc]; argc++) 115 ; 116 117 if (strcmp(wp[0], "[") == 0) { 118 if (strcmp(wp[--argc], "]") != 0) { 119 bi_errorf("missing ]"); 120 return T_ERR_EXIT; 121 } 122 } 123 124 te.pos.wp = wp + 1; 125 te.wp_end = wp + argc; 126 127 /* 128 * Handle the special cases from POSIX.2, section 4.62.4. 129 * Implementation of all the rules isn't necessary since 130 * our parser does the right thing for the ommited steps. 131 */ 132 if (argc <= 5) { 133 char **owp = wp; 134 int invert = 0; 135 Test_op op; 136 const char *opnd1, *opnd2; 137 138 while (--argc >= 0) { 139 if ((*te.isa)(&te, TM_END)) 140 return !0; 141 if (argc == 3) { 142 opnd1 = (*te.getopnd)(&te, TO_NONOP, 1); 143 if ((op = (Test_op) (*te.isa)(&te, TM_BINOP))) { 144 opnd2 = (*te.getopnd)(&te, op, 1); 145 res = (*te.eval)(&te, op, opnd1, opnd2, 146 1); 147 if (te.flags & TEF_ERROR) 148 return T_ERR_EXIT; 149 if (invert & 1) 150 res = !res; 151 return !res; 152 } 153 /* back up to opnd1 */ 154 te.pos.wp--; 155 } 156 if (argc == 1) { 157 opnd1 = (*te.getopnd)(&te, TO_NONOP, 1); 158 res = (*te.eval)(&te, TO_STNZE, opnd1, 159 (char *) 0, 1); 160 if (invert & 1) 161 res = !res; 162 return !res; 163 } 164 if ((*te.isa)(&te, TM_NOT)) { 165 invert++; 166 } else 167 break; 168 } 169 te.pos.wp = owp + 1; 170 } 171 172 return test_parse(&te); 173 } 174 175 /* 176 * Generic test routines. 177 */ 178 179 Test_op 180 test_isop(te, meta, s) 181 Test_env *te; 182 Test_meta meta; 183 const char *s; 184 { 185 char sc1; 186 const struct t_op *otab; 187 188 otab = meta == TM_UNOP ? u_ops : b_ops; 189 if (*s) { 190 sc1 = s[1]; 191 for (; otab->op_text[0]; otab++) 192 if (sc1 == otab->op_text[1] 193 && strcmp(s, otab->op_text) == 0 194 && ((te->flags & TEF_DBRACKET) 195 || (otab->op_num != TO_STLT 196 && otab->op_num != TO_STGT))) 197 return otab->op_num; 198 } 199 return TO_NONOP; 200 } 201 202 int 203 test_eval(te, op, opnd1, opnd2, do_eval) 204 Test_env *te; 205 Test_op op; 206 const char *opnd1; 207 const char *opnd2; 208 int do_eval; 209 { 210 int res; 211 int not; 212 struct stat b1, b2; 213 214 if (!do_eval) 215 return 0; 216 217 switch ((int) op) { 218 /* 219 * Unary Operators 220 */ 221 case TO_STNZE: /* -n */ 222 return *opnd1 != '\0'; 223 case TO_STZER: /* -z */ 224 return *opnd1 == '\0'; 225 case TO_OPTION: /* -o */ 226 if ((not = *opnd1 == '!')) 227 opnd1++; 228 if ((res = option(opnd1)) < 0) 229 res = 0; 230 else { 231 res = Flag(res); 232 if (not) 233 res = !res; 234 } 235 return res; 236 case TO_FILRD: /* -r */ 237 return test_eaccess(opnd1, R_OK) == 0; 238 case TO_FILWR: /* -w */ 239 return test_eaccess(opnd1, W_OK) == 0; 240 case TO_FILEX: /* -x */ 241 return test_eaccess(opnd1, X_OK) == 0; 242 case TO_FILAXST: /* -a */ 243 return test_stat(opnd1, &b1) == 0; 244 case TO_FILEXST: /* -e */ 245 /* at&t ksh does not appear to do the /dev/fd/ thing for 246 * this (unless the os itself handles it) 247 */ 248 return stat(opnd1, &b1) == 0; 249 case TO_FILREG: /* -r */ 250 return test_stat(opnd1, &b1) == 0 && S_ISREG(b1.st_mode); 251 case TO_FILID: /* -d */ 252 return test_stat(opnd1, &b1) == 0 && S_ISDIR(b1.st_mode); 253 case TO_FILCDEV: /* -c */ 254 #ifdef S_ISCHR 255 return test_stat(opnd1, &b1) == 0 && S_ISCHR(b1.st_mode); 256 #else 257 return 0; 258 #endif 259 case TO_FILBDEV: /* -b */ 260 #ifdef S_ISBLK 261 return test_stat(opnd1, &b1) == 0 && S_ISBLK(b1.st_mode); 262 #else 263 return 0; 264 #endif 265 case TO_FILFIFO: /* -p */ 266 #ifdef S_ISFIFO 267 return test_stat(opnd1, &b1) == 0 && S_ISFIFO(b1.st_mode); 268 #else 269 return 0; 270 #endif 271 case TO_FILSYM: /* -h -L */ 272 #ifdef S_ISLNK 273 return lstat(opnd1, &b1) == 0 && S_ISLNK(b1.st_mode); 274 #else 275 return 0; 276 #endif 277 case TO_FILSOCK: /* -S */ 278 #ifdef S_ISSOCK 279 return test_stat(opnd1, &b1) == 0 && S_ISSOCK(b1.st_mode); 280 #else 281 return 0; 282 #endif 283 case TO_FILCDF:/* -H HP context dependent files (directories) */ 284 #ifdef S_ISCDF 285 { 286 /* Append a + to filename and check to see if result is a 287 * setuid directory. CDF stuff in general is hookey, since 288 * it breaks for the following sequence: echo hi > foo+; 289 * mkdir foo; echo bye > foo/default; chmod u+s foo 290 * (foo+ refers to the file with hi in it, there is no way 291 * to get at the file with bye in it - please correct me if 292 * I'm wrong about this). 293 */ 294 int len = strlen(opnd1); 295 char *p = str_nsave(opnd1, len + 1, ATEMP); 296 297 p[len++] = '+'; 298 p[len] = '\0'; 299 return stat(p, &b1) == 0 && S_ISCDF(b1.st_mode); 300 } 301 #else 302 return 0; 303 #endif 304 case TO_FILSETU: /* -u */ 305 #ifdef S_ISUID 306 return test_stat(opnd1, &b1) == 0 307 && (b1.st_mode & S_ISUID) == S_ISUID; 308 #else 309 return 0; 310 #endif 311 case TO_FILSETG: /* -g */ 312 #ifdef S_ISGID 313 return test_stat(opnd1, &b1) == 0 314 && (b1.st_mode & S_ISGID) == S_ISGID; 315 #else 316 return 0; 317 #endif 318 case TO_FILSTCK: /* -k */ 319 return test_stat(opnd1, &b1) == 0 320 && (b1.st_mode & S_ISVTX) == S_ISVTX; 321 case TO_FILGZ: /* -s */ 322 return test_stat(opnd1, &b1) == 0 && b1.st_size > 0L; 323 case TO_FILTT: /* -t */ 324 if (opnd1 && !bi_getn(opnd1, &res)) { 325 te->flags |= TEF_ERROR; 326 res = 0; 327 } else 328 res = isatty(opnd1 ? res : 0); 329 return res; 330 case TO_FILUID: /* -O */ 331 return test_stat(opnd1, &b1) == 0 && b1.st_uid == geteuid(); 332 case TO_FILGID: /* -G */ 333 return test_stat(opnd1, &b1) == 0 && b1.st_gid == getegid(); 334 /* 335 * Binary Operators 336 */ 337 case TO_STEQL: /* = */ 338 if (te->flags & TEF_DBRACKET) 339 return gmatch(opnd1, opnd2, FALSE); 340 return strcmp(opnd1, opnd2) == 0; 341 case TO_STNEQ: /* != */ 342 if (te->flags & TEF_DBRACKET) 343 return !gmatch(opnd1, opnd2, FALSE); 344 return strcmp(opnd1, opnd2) != 0; 345 case TO_STLT: /* < */ 346 return strcmp(opnd1, opnd2) < 0; 347 case TO_STGT: /* > */ 348 return strcmp(opnd1, opnd2) > 0; 349 case TO_INTEQ: /* -eq */ 350 case TO_INTNE: /* -ne */ 351 case TO_INTGE: /* -ge */ 352 case TO_INTGT: /* -gt */ 353 case TO_INTLE: /* -le */ 354 case TO_INTLT: /* -lt */ 355 { 356 long v1, v2; 357 358 if (!evaluate(opnd1, &v1, TRUE) 359 || !evaluate(opnd2, &v2, TRUE)) 360 { 361 /* error already printed.. */ 362 te->flags |= TEF_ERROR; 363 return 1; 364 } 365 switch ((int) op) { 366 case TO_INTEQ: 367 return v1 == v2; 368 case TO_INTNE: 369 return v1 != v2; 370 case TO_INTGE: 371 return v1 >= v2; 372 case TO_INTGT: 373 return v1 > v2; 374 case TO_INTLE: 375 return v1 <= v2; 376 case TO_INTLT: 377 return v1 < v2; 378 } 379 } 380 case TO_FILNT: /* -nt */ 381 return stat (opnd1, &b1) == 0 && stat (opnd2, &b2) == 0 382 && b1.st_mtime > b2.st_mtime; 383 case TO_FILOT: /* -ot */ 384 return stat (opnd1, &b1) == 0 && stat (opnd2, &b2) == 0 385 && b1.st_mtime < b2.st_mtime; 386 case TO_FILEQ: /* -ef */ 387 return stat (opnd1, &b1) == 0 && stat (opnd2, &b2) == 0 388 && b1.st_dev == b2.st_dev 389 && b1.st_ino == b2.st_ino; 390 } 391 (*te->error)(te, 0, "internal error: unknown op"); 392 return 1; 393 } 394 395 /* Nasty kludge to handle Korn's bizarre /dev/fd hack */ 396 static int 397 test_stat(path, statb) 398 const char *path; 399 struct stat *statb; 400 { 401 #if !defined(HAVE_DEV_FD) 402 int fd; 403 404 if (strncmp(path, "/dev/fd/", 8) == 0 && getn(path + 8, &fd)) 405 return fstat(fd, statb); 406 #endif /* !HAVE_DEV_FD */ 407 408 return stat(path, statb); 409 } 410 411 /* Another nasty kludge to handle Korn's bizarre /dev/fd hack */ 412 static int 413 test_eaccess(path, mode) 414 const char *path; 415 int mode; 416 { 417 #if !defined(HAVE_DEV_FD) 418 int fd; 419 420 if (strncmp(path, "/dev/fd/", 8) == 0 && getn(path + 8, &fd)) { 421 int flags; 422 423 if ((flags = fcntl(fd, F_GETFL, 0)) < 0 424 || (mode & X_OK) 425 || ((mode & W_OK) && (flags & O_ACCMODE) == O_RDONLY) 426 || ((mode & R_OK) && (flags & O_ACCMODE) == O_WRONLY)) 427 return -1; 428 return 0; 429 } 430 #endif /* !HAVE_DEV_FD */ 431 432 return eaccess(path, mode); 433 } 434 435 int 436 test_parse(te) 437 Test_env *te; 438 { 439 int res; 440 441 res = test_oexpr(te, 1); 442 443 if (!(te->flags & TEF_ERROR) && !(*te->isa)(te, TM_END)) 444 (*te->error)(te, 0, "unexpected operator/operand"); 445 446 return (te->flags & TEF_ERROR) ? T_ERR_EXIT : !res; 447 } 448 449 static int 450 test_oexpr(te, do_eval) 451 Test_env *te; 452 int do_eval; 453 { 454 int res; 455 456 res = test_aexpr(te, do_eval); 457 if (res) 458 do_eval = 0; 459 if (!(te->flags & TEF_ERROR) && (*te->isa)(te, TM_OR)) 460 return test_oexpr(te, do_eval) || res; 461 return res; 462 } 463 464 static int 465 test_aexpr(te, do_eval) 466 Test_env *te; 467 int do_eval; 468 { 469 int res; 470 471 res = test_nexpr(te, do_eval); 472 if (!res) 473 do_eval = 0; 474 if (!(te->flags & TEF_ERROR) && (*te->isa)(te, TM_AND)) 475 return test_aexpr(te, do_eval) && res; 476 return res; 477 } 478 479 static int 480 test_nexpr(te, do_eval) 481 Test_env *te; 482 int do_eval; 483 { 484 if (!(te->flags & TEF_ERROR) && (*te->isa)(te, TM_NOT)) 485 return !test_nexpr(te, do_eval); 486 return test_primary(te, do_eval); 487 } 488 489 static int 490 test_primary(te, do_eval) 491 Test_env *te; 492 int do_eval; 493 { 494 const char *opnd1, *opnd2; 495 int res; 496 Test_op op; 497 498 if (te->flags & TEF_ERROR) 499 return 0; 500 if ((*te->isa)(te, TM_OPAREN)) { 501 res = test_oexpr(te, do_eval); 502 if (te->flags & TEF_ERROR) 503 return 0; 504 if (!(*te->isa)(te, TM_CPAREN)) { 505 (*te->error)(te, 0, "missing closing paren"); 506 return 0; 507 } 508 return res; 509 } 510 if ((op = (Test_op) (*te->isa)(te, TM_UNOP))) { 511 /* unary expression */ 512 opnd1 = (*te->getopnd)(te, op, do_eval); 513 if (!opnd1) { 514 (*te->error)(te, -1, "missing argument"); 515 return 0; 516 } 517 518 return (*te->eval)(te, op, opnd1, (const char *) 0, do_eval); 519 } 520 opnd1 = (*te->getopnd)(te, TO_NONOP, do_eval); 521 if (!opnd1) { 522 (*te->error)(te, 0, "expression expected"); 523 return 0; 524 } 525 if ((op = (Test_op) (*te->isa)(te, TM_BINOP))) { 526 /* binary expression */ 527 opnd2 = (*te->getopnd)(te, op, do_eval); 528 if (!opnd2) { 529 (*te->error)(te, -1, "missing second argument"); 530 return 0; 531 } 532 533 return (*te->eval)(te, op, opnd1, opnd2, do_eval); 534 } 535 if (te->flags & TEF_DBRACKET) { 536 (*te->error)(te, -1, "missing expression operator"); 537 return 0; 538 } 539 return (*te->eval)(te, TO_STNZE, opnd1, (const char *) 0, do_eval); 540 } 541 542 /* 543 * Plain test (test and [ .. ]) specific routines. 544 */ 545 546 /* Test if the current token is a whatever. Accepts the current token if 547 * it is. Returns 0 if it is not, non-zero if it is (in the case of 548 * TM_UNOP and TM_BINOP, the returned value is a Test_op). 549 */ 550 static int 551 ptest_isa(te, meta) 552 Test_env *te; 553 Test_meta meta; 554 { 555 /* Order important - indexed by Test_meta values */ 556 static const char *const tokens[] = { 557 "-o", "-a", "!", "(", ")" 558 }; 559 int ret; 560 561 if (te->pos.wp >= te->wp_end) 562 return meta == TM_END; 563 564 if (meta == TM_UNOP || meta == TM_BINOP) 565 ret = (int) test_isop(te, meta, *te->pos.wp); 566 else if (meta == TM_END) 567 ret = 0; 568 else 569 ret = strcmp(*te->pos.wp, tokens[(int) meta]) == 0; 570 571 /* Accept the token? */ 572 if (ret) 573 te->pos.wp++; 574 575 return ret; 576 } 577 578 static const char * 579 ptest_getopnd(te, op, do_eval) 580 Test_env *te; 581 Test_op op; 582 int do_eval; 583 { 584 if (te->pos.wp >= te->wp_end) 585 return op == TO_FILTT ? "1" : (const char *) 0; 586 return *te->pos.wp++; 587 } 588 589 static int 590 ptest_eval(te, op, opnd1, opnd2, do_eval) 591 Test_env *te; 592 Test_op op; 593 const char *opnd1; 594 const char *opnd2; 595 int do_eval; 596 { 597 return test_eval(te, op, opnd1, opnd2, do_eval); 598 } 599 600 static void 601 ptest_error(te, offset, msg) 602 Test_env *te; 603 int offset; 604 const char *msg; 605 { 606 const char *op = te->pos.wp + offset >= te->wp_end ? 607 (const char *) 0 : te->pos.wp[offset]; 608 609 te->flags |= TEF_ERROR; 610 if (op) 611 bi_errorf("%s: %s", op, msg); 612 else 613 bi_errorf("%s", msg); 614 } 615