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