c_test.c revision 1.1 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