test.c 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400
  1. /*
  2. * POSIX standard
  3. * test expression
  4. * [ expression ]
  5. *
  6. * Plan 9 additions:
  7. * -A file exists and is append-only
  8. * -L file exists and is exclusive-use
  9. * -T file exists and is temporary
  10. */
  11. #include <u.h>
  12. #include <libc.h>
  13. #define EQ(a,b) ((tmp=a)==0?0:(strcmp(tmp,b)==0))
  14. int ap;
  15. int ac;
  16. char **av;
  17. char *tmp;
  18. void synbad(char *, char *);
  19. int fsizep(char *);
  20. int isdir(char *);
  21. int isreg(char *);
  22. int isatty(int);
  23. int isint(char *, int *);
  24. int isolder(char *, char *);
  25. int isolderthan(char *, char *);
  26. int isnewerthan(char *, char *);
  27. int hasmode(char *, ulong);
  28. int tio(char *, int);
  29. int e(void), e1(void), e2(void), e3(void);
  30. char *nxtarg(int);
  31. void
  32. main(int argc, char *argv[])
  33. {
  34. int r;
  35. char *c;
  36. ac = argc; av = argv; ap = 1;
  37. if(EQ(argv[0],"[")) {
  38. if(!EQ(argv[--ac],"]"))
  39. synbad("] missing","");
  40. }
  41. argv[ac] = 0;
  42. if (ac<=1)
  43. exits("usage");
  44. r = e();
  45. /*
  46. * nice idea but short-circuit -o and -a operators may have
  47. * not consumed their right-hand sides.
  48. */
  49. if(0 && (c = nxtarg(1)) != nil)
  50. synbad("unexpected operator/operand: ", c);
  51. exits(r?0:"false");
  52. }
  53. char *
  54. nxtarg(int mt)
  55. {
  56. if(ap>=ac){
  57. if(mt){
  58. ap++;
  59. return(0);
  60. }
  61. synbad("argument expected","");
  62. }
  63. return(av[ap++]);
  64. }
  65. int
  66. nxtintarg(int *pans)
  67. {
  68. if(ap<ac && isint(av[ap], pans)){
  69. ap++;
  70. return 1;
  71. }
  72. return 0;
  73. }
  74. int
  75. e(void)
  76. {
  77. int p1;
  78. p1 = e1();
  79. if (EQ(nxtarg(1), "-o"))
  80. return(p1 || e());
  81. ap--;
  82. return(p1);
  83. }
  84. int
  85. e1(void)
  86. {
  87. int p1;
  88. p1 = e2();
  89. if (EQ(nxtarg(1), "-a"))
  90. return (p1 && e1());
  91. ap--;
  92. return(p1);
  93. }
  94. int
  95. e2(void)
  96. {
  97. if (EQ(nxtarg(0), "!"))
  98. return(!e2());
  99. ap--;
  100. return(e3());
  101. }
  102. int
  103. e3(void)
  104. {
  105. int p1, int1, int2;
  106. char *a, *p2;
  107. a = nxtarg(0);
  108. if(EQ(a, "(")) {
  109. p1 = e();
  110. if(!EQ(nxtarg(0), ")"))
  111. synbad(") expected","");
  112. return(p1);
  113. }
  114. if(EQ(a, "-A"))
  115. return(hasmode(nxtarg(0), DMAPPEND));
  116. if(EQ(a, "-L"))
  117. return(hasmode(nxtarg(0), DMEXCL));
  118. if(EQ(a, "-T"))
  119. return(hasmode(nxtarg(0), DMTMP));
  120. if(EQ(a, "-f"))
  121. return(isreg(nxtarg(0)));
  122. if(EQ(a, "-d"))
  123. return(isdir(nxtarg(0)));
  124. if(EQ(a, "-r"))
  125. return(tio(nxtarg(0), 4));
  126. if(EQ(a, "-w"))
  127. return(tio(nxtarg(0), 2));
  128. if(EQ(a, "-x"))
  129. return(tio(nxtarg(0), 1));
  130. if(EQ(a, "-e"))
  131. return(tio(nxtarg(0), 0));
  132. if(EQ(a, "-c"))
  133. return(0);
  134. if(EQ(a, "-b"))
  135. return(0);
  136. if(EQ(a, "-u"))
  137. return(0);
  138. if(EQ(a, "-g"))
  139. return(0);
  140. if(EQ(a, "-s"))
  141. return(fsizep(nxtarg(0)));
  142. if(EQ(a, "-t"))
  143. if(ap>=ac)
  144. return(isatty(1));
  145. else if(nxtintarg(&int1))
  146. return(isatty(int1));
  147. else
  148. synbad("not a valid file descriptor number ", "");
  149. if(EQ(a, "-n"))
  150. return(!EQ(nxtarg(0), ""));
  151. if(EQ(a, "-z"))
  152. return(EQ(nxtarg(0), ""));
  153. p2 = nxtarg(1);
  154. if (p2==0)
  155. return(!EQ(a,""));
  156. if(EQ(p2, "="))
  157. return(EQ(nxtarg(0), a));
  158. if(EQ(p2, "!="))
  159. return(!EQ(nxtarg(0), a));
  160. if(EQ(p2, "-older"))
  161. return(isolder(nxtarg(0), a));
  162. if(EQ(p2, "-ot"))
  163. return(isolderthan(nxtarg(0), a));
  164. if(EQ(p2, "-nt"))
  165. return(isnewerthan(nxtarg(0), a));
  166. if(!isint(a, &int1))
  167. synbad("unexpected operator/operand: ", p2);
  168. if(nxtintarg(&int2)){
  169. if(EQ(p2, "-eq"))
  170. return(int1==int2);
  171. if(EQ(p2, "-ne"))
  172. return(int1!=int2);
  173. if(EQ(p2, "-gt"))
  174. return(int1>int2);
  175. if(EQ(p2, "-lt"))
  176. return(int1<int2);
  177. if(EQ(p2, "-ge"))
  178. return(int1>=int2);
  179. if(EQ(p2, "-le"))
  180. return(int1<=int2);
  181. }
  182. synbad("unknown operator ",p2);
  183. return 0; /* to shut ken up */
  184. }
  185. int
  186. tio(char *a, int f)
  187. {
  188. return access (a, f) >= 0;
  189. }
  190. /*
  191. * note that the name strings pointed to by Dir members are
  192. * allocated with the Dir itself (by the same call to malloc),
  193. * but are not included in sizeof(Dir), so copying a Dir won't
  194. * copy the strings it points to.
  195. */
  196. int
  197. hasmode(char *f, ulong m)
  198. {
  199. int r;
  200. Dir *dir;
  201. dir = dirstat(f);
  202. if (dir == nil)
  203. return 0;
  204. r = (dir->mode & m) != 0;
  205. free(dir);
  206. return r;
  207. }
  208. int
  209. isdir(char *f)
  210. {
  211. return hasmode(f, DMDIR);
  212. }
  213. int
  214. isreg(char *f)
  215. {
  216. return !isdir(f);
  217. }
  218. int
  219. isatty(int fd)
  220. {
  221. int r;
  222. Dir *d1, *d2;
  223. d1 = dirfstat(fd);
  224. d2 = dirstat("/dev/cons");
  225. if (d1 == nil || d2 == nil)
  226. r = 0;
  227. else
  228. r = d1->type == d2->type && d1->dev == d2->dev &&
  229. d1->qid.path == d2->qid.path;
  230. free(d1);
  231. free(d2);
  232. return r;
  233. }
  234. int
  235. fsizep(char *f)
  236. {
  237. int r;
  238. Dir *dir;
  239. dir = dirstat(f);
  240. if (dir == nil)
  241. return 0;
  242. r = dir->length > 0;
  243. free(dir);
  244. return r;
  245. }
  246. void
  247. synbad(char *s1, char *s2)
  248. {
  249. int len;
  250. write(2, "test: ", 6);
  251. if ((len = strlen(s1)) != 0)
  252. write(2, s1, len);
  253. if ((len = strlen(s2)) != 0)
  254. write(2, s2, len);
  255. write(2, "\n", 1);
  256. exits("bad syntax");
  257. }
  258. int
  259. isint(char *s, int *pans)
  260. {
  261. char *ep;
  262. *pans = strtol(s, &ep, 0);
  263. return (*ep == 0);
  264. }
  265. int
  266. isolder(char *pin, char *f)
  267. {
  268. int r;
  269. ulong n, m;
  270. char *p = pin;
  271. Dir *dir;
  272. dir = dirstat(f);
  273. if (dir == nil)
  274. return 0;
  275. /* parse time */
  276. n = 0;
  277. while(*p){
  278. m = strtoul(p, &p, 0);
  279. switch(*p){
  280. case 0:
  281. n = m;
  282. break;
  283. case 'y':
  284. m *= 12;
  285. /* fall through */
  286. case 'M':
  287. m *= 30;
  288. /* fall through */
  289. case 'd':
  290. m *= 24;
  291. /* fall through */
  292. case 'h':
  293. m *= 60;
  294. /* fall through */
  295. case 'm':
  296. m *= 60;
  297. /* fall through */
  298. case 's':
  299. n += m;
  300. p++;
  301. break;
  302. default:
  303. synbad("bad time syntax, ", pin);
  304. }
  305. }
  306. r = dir->mtime + n < time(0);
  307. free(dir);
  308. return r;
  309. }
  310. int
  311. isolderthan(char *a, char *b)
  312. {
  313. int r;
  314. Dir *ad, *bd;
  315. ad = dirstat(a);
  316. bd = dirstat(b);
  317. if (ad == nil || bd == nil)
  318. r = 0;
  319. else
  320. r = ad->mtime > bd->mtime;
  321. free(ad);
  322. free(bd);
  323. return r;
  324. }
  325. int
  326. isnewerthan(char *a, char *b)
  327. {
  328. int r;
  329. Dir *ad, *bd;
  330. ad = dirstat(a);
  331. bd = dirstat(b);
  332. if (ad == nil || bd == nil)
  333. r = 0;
  334. else
  335. r = ad->mtime < bd->mtime;
  336. free(ad);
  337. free(bd);
  338. return r;
  339. }