test.c 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303
  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. */
  10. #include <u.h>
  11. #include <libc.h>
  12. #define EQ(a,b) ((tmp=a)==0?0:(strcmp(tmp,b)==0))
  13. int ap;
  14. int ac;
  15. char **av;
  16. char *tmp;
  17. void synbad(char *, char *);
  18. int fsizep(char *);
  19. int isdir(char *);
  20. int isreg(char *);
  21. int isatty(int);
  22. int isint(char *, int *);
  23. int hasmode(char *, ulong);
  24. int tio(char *, int);
  25. int e(void), e1(void), e2(void), e3(void);
  26. void
  27. main(int argc, char *argv[])
  28. {
  29. ac = argc; av = argv; ap = 1;
  30. if(EQ(argv[0],"[")) {
  31. if(!EQ(argv[--ac],"]"))
  32. synbad("] missing","");
  33. }
  34. argv[ac] = 0;
  35. if (ac<=1) exits("usage");
  36. exits(e()?0:"false");
  37. }
  38. char *
  39. nxtarg(int mt)
  40. {
  41. if(ap>=ac){
  42. if(mt){
  43. ap++;
  44. return(0);
  45. }
  46. synbad("argument expected","");
  47. }
  48. return(av[ap++]);
  49. }
  50. int
  51. nxtintarg(int *pans)
  52. {
  53. if(ap<ac && isint(av[ap], pans)){
  54. ap++;
  55. return 1;
  56. }
  57. return 0;
  58. }
  59. int
  60. e(void) {
  61. int p1;
  62. p1 = e1();
  63. if (EQ(nxtarg(1), "-o")) return(p1 || e());
  64. ap--;
  65. return(p1);
  66. }
  67. int
  68. e1(void) {
  69. int p1;
  70. p1 = e2();
  71. if (EQ(nxtarg(1), "-a")) return (p1 && e1());
  72. ap--;
  73. return(p1);
  74. }
  75. int
  76. e2(void) {
  77. if (EQ(nxtarg(0), "!"))
  78. return(!e2());
  79. ap--;
  80. return(e3());
  81. }
  82. int
  83. e3(void) {
  84. int p1;
  85. char *a;
  86. char *p2;
  87. int int1, int2;
  88. a = nxtarg(0);
  89. if(EQ(a, "(")) {
  90. p1 = e();
  91. if(!EQ(nxtarg(0), ")")) synbad(") expected","");
  92. return(p1);
  93. }
  94. if(EQ(a, "-A"))
  95. return(hasmode(nxtarg(0), DMAPPEND));
  96. if(EQ(a, "-L"))
  97. return(hasmode(nxtarg(0), DMEXCL));
  98. if(EQ(a, "-f"))
  99. return(isreg(nxtarg(0)));
  100. if(EQ(a, "-d"))
  101. return(isdir(nxtarg(0)));
  102. if(EQ(a, "-r"))
  103. return(tio(nxtarg(0), 4));
  104. if(EQ(a, "-w"))
  105. return(tio(nxtarg(0), 2));
  106. if(EQ(a, "-x"))
  107. return(tio(nxtarg(0), 1));
  108. if(EQ(a, "-e"))
  109. return(tio(nxtarg(0), 0));
  110. if(EQ(a, "-c"))
  111. return(0);
  112. if(EQ(a, "-b"))
  113. return(0);
  114. if(EQ(a, "-u"))
  115. return(0);
  116. if(EQ(a, "-g"))
  117. return(0);
  118. if(EQ(a, "-s"))
  119. return(fsizep(nxtarg(0)));
  120. if(EQ(a, "-t"))
  121. if(ap>=ac || !nxtintarg(&int1))
  122. return(isatty(1));
  123. else
  124. return(isatty(int1));
  125. if(EQ(a, "-n"))
  126. return(!EQ(nxtarg(0), ""));
  127. if(EQ(a, "-z"))
  128. return(EQ(nxtarg(0), ""));
  129. p2 = nxtarg(1);
  130. if (p2==0)
  131. return(!EQ(a,""));
  132. if(EQ(p2, "="))
  133. return(EQ(nxtarg(0), a));
  134. if(EQ(p2, "!="))
  135. return(!EQ(nxtarg(0), a));
  136. if(!isint(a, &int1))
  137. return(!EQ(a,""));
  138. if(nxtintarg(&int2)){
  139. if(EQ(p2, "-eq"))
  140. return(int1==int2);
  141. if(EQ(p2, "-ne"))
  142. return(int1!=int2);
  143. if(EQ(p2, "-gt"))
  144. return(int1>int2);
  145. if(EQ(p2, "-lt"))
  146. return(int1<int2);
  147. if(EQ(p2, "-ge"))
  148. return(int1>=int2);
  149. if(EQ(p2, "-le"))
  150. return(int1<=int2);
  151. }
  152. synbad("unknown operator ",p2);
  153. return 0; /* to shut ken up */
  154. }
  155. int
  156. tio(char *a, int f)
  157. {
  158. return access (a, f) >= 0;
  159. }
  160. /* copy to local memory; clear names for safety */
  161. int
  162. localstat(char *f, Dir *dir)
  163. {
  164. Dir *d;
  165. d = dirstat(f);
  166. if(d == 0)
  167. return(-1);
  168. *dir = *d;
  169. dir->name = 0;
  170. dir->uid = 0;
  171. dir->gid = 0;
  172. dir->muid = 0;
  173. return 0;
  174. }
  175. /* copy to local memory; clear names for safety */
  176. int
  177. localfstat(int f, Dir *dir)
  178. {
  179. Dir *d;
  180. d = dirfstat(f);
  181. if(d == 0)
  182. return(-1);
  183. *dir = *d;
  184. dir->name = 0;
  185. dir->uid = 0;
  186. dir->gid = 0;
  187. dir->muid = 0;
  188. return 0;
  189. }
  190. int
  191. hasmode(char *f, ulong m)
  192. {
  193. Dir dir;
  194. if(localstat(f,&dir)<0)
  195. return(0);
  196. return(dir.mode&m);
  197. }
  198. int
  199. isdir(char *f)
  200. {
  201. Dir dir;
  202. if(localstat(f,&dir)<0)
  203. return(0);
  204. return(dir.mode&DMDIR);
  205. }
  206. int
  207. isreg(char *f)
  208. {
  209. Dir dir;
  210. if(localstat(f,&dir)<0)
  211. return(0);
  212. return(!(dir.mode&DMDIR));
  213. }
  214. int
  215. isatty(int fd)
  216. {
  217. Dir d1, d2;
  218. if(localfstat(fd, &d1) < 0)
  219. return 0;
  220. if(localstat("/dev/cons", &d2) < 0)
  221. return 0;
  222. return d1.type==d2.type && d1.dev==d2.dev && d1.qid.path==d2.qid.path;
  223. }
  224. int
  225. fsizep(char *f)
  226. {
  227. Dir dir;
  228. if(localstat(f,&dir)<0)
  229. return(0);
  230. return(dir.length>0);
  231. }
  232. void
  233. synbad(char *s1, char *s2)
  234. {
  235. int len;
  236. write(2, "test: ", 6);
  237. if ((len = strlen(s1)) != 0)
  238. write(2, s1, len);
  239. if ((len = strlen(s2)) != 0)
  240. write(2, s2, len);
  241. write(2, "\n", 1);
  242. exits("bad syntax");
  243. }
  244. int
  245. isint(char *s, int *pans)
  246. {
  247. char *ep;
  248. *pans = strtol(s, &ep, 0);
  249. return (*ep == 0);
  250. }