test.c 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396
  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) exits("usage");
  43. r = e();
  44. if(c = nxtarg(1))
  45. synbad("unexpected operator/operand: ", c);
  46. exits(r?0:"false");
  47. }
  48. char *
  49. nxtarg(int mt)
  50. {
  51. if(ap>=ac){
  52. if(mt){
  53. ap++;
  54. return(0);
  55. }
  56. synbad("argument expected","");
  57. }
  58. return(av[ap++]);
  59. }
  60. int
  61. nxtintarg(int *pans)
  62. {
  63. if(ap<ac && isint(av[ap], pans)){
  64. ap++;
  65. return 1;
  66. }
  67. return 0;
  68. }
  69. int
  70. e(void) {
  71. int p1;
  72. p1 = e1();
  73. if (EQ(nxtarg(1), "-o")) return(p1 || e());
  74. ap--;
  75. return(p1);
  76. }
  77. int
  78. e1(void) {
  79. int p1;
  80. p1 = e2();
  81. if (EQ(nxtarg(1), "-a")) return (p1 && e1());
  82. ap--;
  83. return(p1);
  84. }
  85. int
  86. e2(void) {
  87. if (EQ(nxtarg(0), "!"))
  88. return(!e2());
  89. ap--;
  90. return(e3());
  91. }
  92. int
  93. e3(void) {
  94. int p1;
  95. char *a;
  96. char *p2;
  97. int int1, int2;
  98. a = nxtarg(0);
  99. if(EQ(a, "(")) {
  100. p1 = e();
  101. if(!EQ(nxtarg(0), ")")) synbad(") expected","");
  102. return(p1);
  103. }
  104. if(EQ(a, "-A"))
  105. return(hasmode(nxtarg(0), DMAPPEND));
  106. if(EQ(a, "-L"))
  107. return(hasmode(nxtarg(0), DMEXCL));
  108. if(EQ(a, "-T"))
  109. return(hasmode(nxtarg(0), DMTMP));
  110. if(EQ(a, "-f"))
  111. return(isreg(nxtarg(0)));
  112. if(EQ(a, "-d"))
  113. return(isdir(nxtarg(0)));
  114. if(EQ(a, "-r"))
  115. return(tio(nxtarg(0), 4));
  116. if(EQ(a, "-w"))
  117. return(tio(nxtarg(0), 2));
  118. if(EQ(a, "-x"))
  119. return(tio(nxtarg(0), 1));
  120. if(EQ(a, "-e"))
  121. return(tio(nxtarg(0), 0));
  122. if(EQ(a, "-c"))
  123. return(0);
  124. if(EQ(a, "-b"))
  125. return(0);
  126. if(EQ(a, "-u"))
  127. return(0);
  128. if(EQ(a, "-g"))
  129. return(0);
  130. if(EQ(a, "-s"))
  131. return(fsizep(nxtarg(0)));
  132. if(EQ(a, "-t"))
  133. if(ap>=ac)
  134. return(isatty(1));
  135. else if(nxtintarg(&int1))
  136. return(isatty(int1));
  137. else
  138. synbad("not a valid file descriptor number ", "");
  139. if(EQ(a, "-n"))
  140. return(!EQ(nxtarg(0), ""));
  141. if(EQ(a, "-z"))
  142. return(EQ(nxtarg(0), ""));
  143. p2 = nxtarg(1);
  144. if (p2==0)
  145. return(!EQ(a,""));
  146. if(EQ(p2, "="))
  147. return(EQ(nxtarg(0), a));
  148. if(EQ(p2, "!="))
  149. return(!EQ(nxtarg(0), a));
  150. if(EQ(p2, "-older"))
  151. return(isolder(nxtarg(0), a));
  152. if(EQ(p2, "-ot"))
  153. return(isolderthan(nxtarg(0), a));
  154. if(EQ(p2, "-nt"))
  155. return(isnewerthan(nxtarg(0), a));
  156. if(!isint(a, &int1))
  157. synbad("unexpected operator/operand: ", p2);
  158. if(nxtintarg(&int2)){
  159. if(EQ(p2, "-eq"))
  160. return(int1==int2);
  161. if(EQ(p2, "-ne"))
  162. return(int1!=int2);
  163. if(EQ(p2, "-gt"))
  164. return(int1>int2);
  165. if(EQ(p2, "-lt"))
  166. return(int1<int2);
  167. if(EQ(p2, "-ge"))
  168. return(int1>=int2);
  169. if(EQ(p2, "-le"))
  170. return(int1<=int2);
  171. }
  172. synbad("unknown operator ",p2);
  173. return 0; /* to shut ken up */
  174. }
  175. int
  176. tio(char *a, int f)
  177. {
  178. return access (a, f) >= 0;
  179. }
  180. /* copy to local memory; clear names for safety */
  181. int
  182. localstat(char *f, Dir *dir)
  183. {
  184. Dir *d;
  185. d = dirstat(f);
  186. if(d == 0)
  187. return(-1);
  188. *dir = *d;
  189. dir->name = 0;
  190. dir->uid = 0;
  191. dir->gid = 0;
  192. dir->muid = 0;
  193. return 0;
  194. }
  195. /* copy to local memory; clear names for safety */
  196. int
  197. localfstat(int f, Dir *dir)
  198. {
  199. Dir *d;
  200. d = dirfstat(f);
  201. if(d == 0)
  202. return(-1);
  203. *dir = *d;
  204. dir->name = 0;
  205. dir->uid = 0;
  206. dir->gid = 0;
  207. dir->muid = 0;
  208. return 0;
  209. }
  210. int
  211. hasmode(char *f, ulong m)
  212. {
  213. Dir dir;
  214. if(localstat(f,&dir)<0)
  215. return(0);
  216. return(dir.mode&m);
  217. }
  218. int
  219. isdir(char *f)
  220. {
  221. Dir dir;
  222. if(localstat(f,&dir)<0)
  223. return(0);
  224. return(dir.mode&DMDIR);
  225. }
  226. int
  227. isreg(char *f)
  228. {
  229. Dir dir;
  230. if(localstat(f,&dir)<0)
  231. return(0);
  232. return(!(dir.mode&DMDIR));
  233. }
  234. int
  235. isatty(int fd)
  236. {
  237. Dir d1, d2;
  238. if(localfstat(fd, &d1) < 0)
  239. return 0;
  240. if(localstat("/dev/cons", &d2) < 0)
  241. return 0;
  242. return d1.type==d2.type && d1.dev==d2.dev && d1.qid.path==d2.qid.path;
  243. }
  244. int
  245. fsizep(char *f)
  246. {
  247. Dir dir;
  248. if(localstat(f,&dir)<0)
  249. return(0);
  250. return(dir.length>0);
  251. }
  252. void
  253. synbad(char *s1, char *s2)
  254. {
  255. int len;
  256. write(2, "test: ", 6);
  257. if ((len = strlen(s1)) != 0)
  258. write(2, s1, len);
  259. if ((len = strlen(s2)) != 0)
  260. write(2, s2, len);
  261. write(2, "\n", 1);
  262. exits("bad syntax");
  263. }
  264. int
  265. isint(char *s, int *pans)
  266. {
  267. char *ep;
  268. *pans = strtol(s, &ep, 0);
  269. return (*ep == 0);
  270. }
  271. int
  272. isolder(char *pin, char *f)
  273. {
  274. char *p = pin;
  275. ulong n, m;
  276. Dir dir;
  277. if(localstat(f,&dir)<0)
  278. return(0);
  279. /* parse time */
  280. n = 0;
  281. while(*p){
  282. m = strtoul(p, &p, 0);
  283. switch(*p){
  284. case 0:
  285. n = m;
  286. break;
  287. case 'y':
  288. m *= 12;
  289. /* fall through */
  290. case 'M':
  291. m *= 30;
  292. /* fall through */
  293. case 'd':
  294. m *= 24;
  295. /* fall through */
  296. case 'h':
  297. m *= 60;
  298. /* fall through */
  299. case 'm':
  300. m *= 60;
  301. /* fall through */
  302. case 's':
  303. n += m;
  304. p++;
  305. break;
  306. default:
  307. synbad("bad time syntax, ", pin);
  308. }
  309. }
  310. return(dir.mtime+n < time(0));
  311. }
  312. int
  313. isolderthan(char *a, char *b)
  314. {
  315. Dir ad, bd;
  316. if(localstat(a, &ad)<0)
  317. return(0);
  318. if(localstat(b, &bd)<0)
  319. return(0);
  320. return ad.mtime > bd.mtime;
  321. }
  322. int
  323. isnewerthan(char *a, char *b)
  324. {
  325. Dir ad, bd;
  326. if(localstat(a, &ad)<0)
  327. return(0);
  328. if(localstat(b, &bd)<0)
  329. return(0);
  330. return ad.mtime < bd.mtime;
  331. }