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